1 files changed, 82 insertions, 0 deletions
diff --git a/cyborg_enhancement/mitaka_version/cyborg/cyborg/accelerator/drivers/spdk/util/pyspdk/py_spdk.py b/cyborg_enhancement/mitaka_version/cyborg/cyborg/accelerator/drivers/spdk/util/pyspdk/py_spdk.py
new file mode 100644
index 0000000..a298c18
--- /dev/null
+++ b/cyborg_enhancement/mitaka_version/cyborg/cyborg/accelerator/drivers/spdk/util/pyspdk/py_spdk.py
@@ -0,0 +1,82 @@
+import psutil
+import re
+import os
+import subprocess
+
+
+class PySPDK(object):
+
+    def __init__(self, pname):
+        super(PySPDK, self).__init__()
+        self.pid = None
+        self.pname = pname
+
+    def start_server(self, spdk_dir, server_name):
+        if not self.is_alive():
+            self.init_hugepages(spdk_dir)
+            server_dir = os.path.join(spdk_dir, 'app/')
+            file_dir = self._search_file(server_dir, server_name)
+            print file_dir
+            os.chdir(file_dir)
+            p = subprocess.Popen(
+                'sudo ./%s' % server_name,
+                shell=True, stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE)
+            out, err = p.communicate()
+            return out
+
+    def init_hugepages(self, spdk_dir):
+        huge_dir = os.path.join(spdk_dir, 'scripts/')
+        file_dir = self._search_file(huge_dir, 'setup.sh')
+        print file_dir
+        os.chdir(file_dir)
+        p = subprocess.Popen(
+            'sudo ./setup.sh',
+            shell=True, stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE)
+        out, err = p.communicate()
+        return out
+
+    @staticmethod
+    def _search_file(spdk_dir, file_name):
+        for dirpath, dirnames, filenames in os.walk(spdk_dir):
+            for filename in filenames:
+                if filename == file_name:
+                    return dirpath
+
+    def _get_process_id(self):
+        for proc in psutil.process_iter():
+            try:
+                pinfo = proc.as_dict(attrs=['pid', 'cmdline'])
+                if re.search(self.pname, str(pinfo.get('cmdline'))):
+                    self.pid = pinfo.get('pid')
+                    return self.pid
+            except psutil.NoSuchProcess:
+                print "NoSuchProcess:%s" % self.pname
+        print "NoSuchProcess:%s" % self.pname
+        return self.pid
+
+    def is_alive(self):
+        self.pid = self._get_process_id()
+        if self.pid:
+            p = psutil.Process(self.pid)
+            if p.is_running():
+                return True
+        return False
+
+    @staticmethod
+    def exec_rpc(method, server='127.0.0.1', port=5260, sub_args=None):
+        exec_cmd = ["./rpc.py", "-s", "-p"]
+        exec_cmd.insert(2, server)
+        exec_cmd.insert(4, str(port))
+        exec_cmd.insert(5, method)
+        if sub_args is None:
+            sub_args = []
+        exec_cmd.extend(sub_args)
+        p = subprocess.Popen(
+            exec_cmd,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE
+        )
+        out, err = p.communicate()
+        return out
+\ No newline at end of file