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 0409afc0de7908aad489b304a09a88f1a5bd97ad..6c10c87b95e70a02d3537dc67bc8843500c681cd 100644 (file)
--- 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));
        }