From: Martin Belanger <martin.belanger@dell.com>
Date: Wed, 15 Mar 2023 20:05:53 +0000 (-0400)
Subject: Python: Add support for setting/getting Host DHCHAP key
X-Git-Tag: v1.4~21
X-Git-Url: https://www.infradead.org/git/?a=commitdiff_plain;h=e71cf10e33c93afac1d0d99c00e3ce99c4c45364;p=users%2Fsagi%2Flibnvme.git

Python: Add support for setting/getting Host DHCHAP key

Signed-off-by: Martin Belanger <martin.belanger@dell.com>
---

diff --git a/libnvme/nvme.i b/libnvme/nvme.i
index 0409afc0..6c10c87b 100644
--- a/libnvme/nvme.i
+++ b/libnvme/nvme.i
@@ -463,19 +463,32 @@ struct nvme_ns {
 @return: None"
 %enddef
 
+%define SET_KEY_DOCSTRING
+"@brief Set or Clear Host's DHCHAP Key
+
+@param key: A DHCHAP key, or None to clear the key.
+@type key: str|None
+
+@return: None"
+%enddef
+
 %pythonappend nvme_host::nvme_host(struct nvme_root *r,
 				   const char *hostnqn,
 				   const char *hostid,
+				   const char *hostkey,
 				   const char *hostsymname) {
 	self.__parent = r  # Keep a reference to parent to ensure garbage collection happens in the right order}
 %extend nvme_host {
 	nvme_host(struct nvme_root *r,
 		  const char *hostnqn = NULL,
 		  const char *hostid = NULL,
+		  const char *hostkey = NULL,
 		  const char *hostsymname = NULL) {
 		nvme_host_t h = hostnqn ? nvme_lookup_host(r, hostnqn, hostid) : nvme_default_host(r);
 		if (hostsymname)
 			nvme_host_set_hostsymname(h, hostsymname);
+		if (hostkey)
+			nvme_host_set_dhchap_key(h, hostkey);
 		return h;
 	}
 	~nvme_host() {
@@ -486,6 +499,10 @@ struct nvme_ns {
 		nvme_host_set_hostsymname($self, hostsymname);
 	}
 
+	%feature("autodoc", SET_KEY_DOCSTRING) set_key;
+	void set_key(const char *key) {
+		nvme_host_set_dhchap_key($self, key);
+	}
 	PyObject* __str__() {
 		return PyUnicode_FromFormat("nvme.host(%s,%s)", STR_OR_NONE($self->hostnqn), STR_OR_NONE($self->hostid));
 	}