1 files changed, 45 insertions, 0 deletions
diff --git a/tests/utils/pidof.py b/tests/utils/pidof.py
new file mode 100755
index 00000000000..4b7071c0a48
--- /dev/null
+++ b/tests/utils/pidof.py
@@ -0,0 +1,45 @@
+
+from __future__ import print_function
+import sys
+
+try:
+    import psutil
+except ImportError:
+    print("Please install psutil --> pip install psutil")
+    sys.exit(1)
+
+def pmap_find(p, name):
+    for m in p.memory_maps(grouped=True):
+        if m.path.endswith("%s.so" % name):
+            return True
+        continue
+    return False
+
+def pidof(processname):
+    for p in psutil.process_iter():
+        if p.pid == 0:
+            continue
+        if "gluster" in processname:
+            if processname == "glusterd" and pmap_find(p, "glusterd"):
+                print((p.pid))
+            if processname == "glusterfs" and pmap_find(p, "client"):
+                print((p.pid))
+            if processname == "glusterfsd" and pmap_find(p, "posix-acl"):
+                print((p.pid))
+            continue
+        if processname.strip() == p.name():
+            print((p.pid))
+
+def main(argv):
+    if len(argv) < 2:
+        sys.stderr.write("Usage: %s <processname>\n" % (argv[0],))
+        return 1
+    try:
+        pidof(argv[1])
+    except Exception as err:
+        print(err)
+        sys.stderr.write("Please be root - %s\n" % err);
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main(sys.argv)