From 8d264dd98b6dbea4543fc26fc6c26614509c4f2d Mon Sep 17 00:00:00 2001
From: Leonardo da Cunha <leonardo.da.cunha@solidigm.com>
Date: Fri, 15 Dec 2023 07:53:28 -0800
Subject: [PATCH] util: Added function to find specific UUID in UUID list.
 Finds a given UUID in the UUID list returned by identify UUID.

Signed-off-by: Leonardo da Cunha <leonardo.da.cunha@solidigm.com>
---
 src/libnvme.map |  1 +
 src/nvme/util.c | 16 ++++++++++++++++
 src/nvme/util.h |  9 +++++++++
 3 files changed, 26 insertions(+)

diff --git a/src/libnvme.map b/src/libnvme.map
index 742f6356e..ebefbb069 100644
--- a/src/libnvme.map
+++ b/src/libnvme.map
@@ -5,6 +5,7 @@ LIBNVME_1_7 {
 		nvme_init_copy_range_f3;
 		nvme_insert_tls_key_versioned;
 		nvme_generate_tls_key_identity;
+		nvme_uuid_find;
 };
 
 LIBNVME_1_6 {
diff --git a/src/nvme/util.c b/src/nvme/util.c
index 45512ff90..6bc4626ce 100644
--- a/src/nvme/util.c
+++ b/src/nvme/util.c
@@ -951,6 +951,22 @@ int nvme_uuid_random(unsigned char uuid[NVME_UUID_LEN])
 	return 0;
 }
 
+__u8 nvme_uuid_find(struct nvme_id_uuid_list *uuid_list, const unsigned char uuid[NVME_UUID_LEN])
+{
+	const unsigned char uuid_end[NVME_UUID_LEN] = {0};
+
+	if ((!uuid_list) || (!uuid))
+		return 0;
+
+	for (int i = 0; i < NVME_ID_UUID_LIST_MAX; i++) {
+		if (memcmp(uuid_end, &uuid_list->entry[i].uuid, NVME_UUID_LEN) == 0)
+			return 0;
+		if (memcmp(uuid, &uuid_list->entry[i].uuid, NVME_UUID_LEN) == 0)
+			return i + 1;
+	}
+	return 0;
+}
+
 #ifdef HAVE_NETDB
 static bool _nvme_ipaddrs_eq(struct sockaddr *addr1, struct sockaddr *addr2)
 {
diff --git a/src/nvme/util.h b/src/nvme/util.h
index 16d5b9cad..dce1ffe76 100644
--- a/src/nvme/util.h
+++ b/src/nvme/util.h
@@ -666,6 +666,15 @@ int nvme_uuid_from_string(const char *str, unsigned char uuid[NVME_UUID_LEN]);
  */
 int nvme_uuid_random(unsigned char uuid[NVME_UUID_LEN]);
 
+/**
+ * nvme_uuid_find - Find UUID position on UUID list
+ * @uuid_list:	UUID list returned by identify UUID
+ * @uuid:	Binary encoded input UUID
+ *
+ * Return: The array position where given UUID resides, or zero if UUID is not present.
+ */
+__u8 nvme_uuid_find(struct nvme_id_uuid_list *uuid_list, const unsigned char uuid[NVME_UUID_LEN]);
+
 /**
  * nvme_ipaddrs_eq - Check if 2 IP addresses are equal.
  * @addr1: IP address (can be IPv4 or IPv6)