[PATCH V2 10/46] nvmftests-nvmf: add support for target ns

[ckulkarnilinux@gmail.com (Chaitanya Kulkarni)]

From: Chaitanya Kulkarni <chaitanya.kulkarni@wdc.com>

This adds a new class to handle the target namespace operations.

Signed-off-by: Chaitanya Kulkarni <chaitanya.kulkarni at wdc.com>
---
 .../selftests/nvmftests/nvmf/target/target_ns.py   | 117 +++++++++++++++++++++
 1 file changed, 117 insertions(+)
 create mode 100644 tools/testing/selftests/nvmftests/nvmf/target/target_ns.py

diff --git a/tools/testing/selftests/nvmftests/nvmf/target/target_ns.py b/tools/testing/selftests/nvmftests/nvmf/target/target_ns.py
new file mode 100644
index 0000000..0e3ba65
--- /dev/null
+++ b/tools/testing/selftests/nvmftests/nvmf/target/target_ns.py
@@ -0,0 +1,117 @@
+# Copyright (c) 2016-2017 Western Digital Corporation or its affiliates.
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+# MA  02110-1301, USA.
+#
+#   Author: Chaitanya Kulkarni <chaitanya.kulkarni at wdc.com>
+#
+""" Represents NVMe Over Fabric Target Namespace.
+"""
+
+import os
+import shutil
+
+from utils.shell import Cmd
+from utils.const import Const
+from utils.log import Log
+
+
+class NVMFTargetNamespace(object):
+    """
+    Represents a target namespace.
+
+        - Attributes :
+            - ns_attr : namespace attributes.
+            - cfgfs : configfs path.
+            - nqn : subsystem nqn.
+            - ns_id : namespace identifier.
+            - ns_path : namespace path in configfs tree.
+            - ns_attr : namespace attributes.
+    """
+    def __init__(self, cfgfs, nqn, ns_id, **ns_attr):
+        self.ns_attr = {}
+        self.cfgfs = cfgfs
+        self.nqn = nqn
+        self.ns_id = ns_id
+        self.ns_path = (self.cfgfs + Const.SYSFS_NVMET_SUBSYS +
+                        nqn + Const.SYSFS_NVMET_SUBSYS_NS + str(ns_id) + "/")
+        self.ns_attr['device_nguid'] = ns_attr['device_nguid']
+        self.ns_attr['device_path'] = ns_attr['device_path']
+        self.ns_attr['enable'] = ns_attr['enable']
+        self.logger = Log.get_logger(__name__, 'target_ns')
+
+    def init(self):
+        """ Create and initialize namespace.
+            - Args :
+                - None.
+            - Returns :
+                - True on success, False on failure.
+        """
+        self.logger.info("Creating ns " + self.ns_path + " ...")
+        try:
+            os.makedirs(self.ns_path)
+        except Exception, err:
+            self.logger.error(str(err) + ".")
+            return False
+
+        cmd = "echo -n " + self.ns_attr['device_path'] + " > " + \
+              self.ns_path + "/device_path"
+
+        if Cmd.exec_cmd(cmd) is False:
+            self.logger.error("failed to configure device path.")
+            return False
+
+        if self.ns_attr['enable'] == '1':
+            if self.enable() is False:
+                self.logger.error("enable ns " + self.ns_path + " failed.")
+                return False
+
+        self.logger.info("NS " + self.ns_path + " enabled.")
+        return True
+
+    def disable(self):
+        """ Disable Namespace.
+            - Args :
+                - None.
+            - Returns :
+                - True on success, False on failure.
+        """
+        self.ns_attr['enable'] = '0'
+        return Cmd.exec_cmd("echo 0 > " + self.ns_path + "/enable")
+
+    def enable(self):
+        """ Enable Namespace.
+            - Args :
+                - None.
+            - Returns :
+                - True on success, False on failure.
+        """
+        self.ns_attr['enable'] = '1'
+        return Cmd.exec_cmd("echo 1 > " + self.ns_path + "/enable")
+
+    def delete(self):
+        """ Delete namespace.
+            - Args :
+                 - None.
+            - Returns :
+                 - True on success, False on failure.
+        """
+        self.logger.info("Removing NS " + self.ns_path + ".")
+        ret = os.path.exists(self.ns_path)
+        if ret is True:
+            shutil.rmtree(self.ns_path, ignore_errors=True)
+        else:
+            self.logger.error("path " + self.ns_path + " doesn't exists.")
+        return ret
-- 
1.8.3.1