From 5db1650b424ff8850b0a5a80b0358fa06a0d05ef Mon Sep 17 00:00:00 2001
From: jeff-lien-sndk <jeff.lien@sandisk.com>
Date: Fri, 25 Jul 2025 14:14:34 -0500
Subject: [PATCH] linux:  Add functions to set and clear etdas bit

To retrieve telemetry log data area 4, the Extended Telemetry Data Area 4
Supported in the Host Behavior Support feature needs to be set.  The bit
will be cleared after the log has been retrieved.  This commit will add
functions to set and clear that bit.

Signed-off-by: jeff-lien-sndk <jeff.lien@sandisk.com>
---
 src/libnvme.map  |  6 ++++++
 src/nvme/linux.c | 35 +++++++++++++++++++++++++++++++++++
 src/nvme/linux.h | 21 +++++++++++++++++++++
 3 files changed, 62 insertions(+)

diff --git a/src/libnvme.map b/src/libnvme.map
index bae5962ce..95f2435cf 100644
--- a/src/libnvme.map
+++ b/src/libnvme.map
@@ -2,6 +2,12 @@
 LIBNVME_UNRELEASED {
 };
 
+LIBNVME_1_15 {
+	global:
+		nvme_set_etdas;
+		nvme_clear_etdas;
+};
+
 LIBNVME_1_14 {
 	global:
 		nvme_get_features_temp_thresh2;
diff --git a/src/nvme/linux.c b/src/nvme/linux.c
index ae4aa526f..4db0a7af0 100644
--- a/src/nvme/linux.c
+++ b/src/nvme/linux.c
@@ -121,6 +121,41 @@ int nvme_fw_download_seq(int fd, __u32 size, __u32 xfer, __u32 offset,
 	return err;
 }
 
+int nvme_set_etdas(int fd,
+		bool *host_behavior_changed)
+{
+	__u32 result;
+	int err;
+	struct nvme_feat_host_behavior da4_enable;
+
+	*host_behavior_changed = false;
+	err = nvme_get_features_host_behavior(fd, 0, &da4_enable, &result);
+
+	if (!err && !da4_enable.etdas) {
+		da4_enable.etdas = 1;
+		err = nvme_set_features_host_behavior(fd, 0, &da4_enable);
+		*host_behavior_changed = true;
+	}
+
+	return err;
+}
+
+int nvme_clear_etdas(int fd)
+{
+	__u32 result;
+	int err;
+	struct nvme_feat_host_behavior da4_disable;
+
+	err = nvme_get_features_host_behavior(fd, 0, &da4_disable, &result);
+
+	if (!err && da4_disable.etdas) {
+		da4_disable.etdas = 0;
+		err = nvme_set_features_host_behavior(fd, 0, &da4_disable);
+	}
+
+	return err;
+}
+
 int nvme_get_telemetry_max(int fd, enum nvme_telemetry_da *da, size_t *data_tx)
 {
 	_cleanup_free_ struct nvme_id_ctrl *id_ctrl = NULL;
diff --git a/src/nvme/linux.h b/src/nvme/linux.h
index e865a8c8a..e7ca29637 100644
--- a/src/nvme/linux.h
+++ b/src/nvme/linux.h
@@ -34,6 +34,27 @@
 int nvme_fw_download_seq(int fd, __u32 size, __u32 xfer, __u32 offset,
 			 void *buf);
 
+/**
+ * nvme_set_etdas() - Set the Extended Telemetry Data Area 4 Supported bit
+ * @fd:		File descriptor of nvme device
+ * @host_behavior_changed: boolean to indicate whether or not the host
+ *          behavior support feature had been changed
+ *
+ * Return: The nvme command status if a response was received (see
+ * &enum nvme_status_field) or -1 with errno set otherwise.
+ */
+int nvme_set_etdas(int fd,
+		bool *host_behavior_changed);
+
+/**
+ * nvme_clear_etdas() - Clear the Extended Telemetry Data Area 4 Supported bit
+ * @fd:		File descriptor of nvme device
+ *
+ * Return: The nvme command status if a response was received (see
+ * &enum nvme_status_field) or -1 with errno set otherwise.
+ */
+int nvme_clear_etdas(int fd);
+
 /**
  * nvme_get_telemetry_max() - Get telemetry limits
  * @fd:		File descriptor of nvme device