summaryrefslogtreecommitdiffstats
path: root/testing/mozharness/mozprocess/pid.py
diff options
context:
space:
mode:
authorMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
committerMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
commit5f8de423f190bbb79a62f804151bc24824fa32d8 (patch)
tree10027f336435511475e392454359edea8e25895d /testing/mozharness/mozprocess/pid.py
parent49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff)
downloadUXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip
Add m-esr52 at 52.6.0
Diffstat (limited to 'testing/mozharness/mozprocess/pid.py')
-rwxr-xr-xtesting/mozharness/mozprocess/pid.py88
1 files changed, 88 insertions, 0 deletions
diff --git a/testing/mozharness/mozprocess/pid.py b/testing/mozharness/mozprocess/pid.py
new file mode 100755
index 000000000..d1f0d9336
--- /dev/null
+++ b/testing/mozharness/mozprocess/pid.py
@@ -0,0 +1,88 @@
+#!/usr/bin/env python
+
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this file,
+# You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import os
+import mozinfo
+import shlex
+import subprocess
+import sys
+
+# determine the platform-specific invocation of `ps`
+if mozinfo.isMac:
+ psarg = '-Acj'
+elif mozinfo.isLinux:
+ psarg = 'axwww'
+else:
+ psarg = 'ax'
+
+def ps(arg=psarg):
+ """
+ python front-end to `ps`
+ http://en.wikipedia.org/wiki/Ps_%28Unix%29
+ returns a list of process dicts based on the `ps` header
+ """
+ retval = []
+ process = subprocess.Popen(['ps', arg], stdout=subprocess.PIPE)
+ stdout, _ = process.communicate()
+ header = None
+ for line in stdout.splitlines():
+ line = line.strip()
+ if header is None:
+ # first line is the header
+ header = line.split()
+ continue
+ split = line.split(None, len(header)-1)
+ process_dict = dict(zip(header, split))
+ retval.append(process_dict)
+ return retval
+
+def running_processes(name, psarg=psarg, defunct=True):
+ """
+ returns a list of
+ {'PID': PID of process (int)
+ 'command': command line of process (list)}
+ with the executable named `name`.
+ - defunct: whether to return defunct processes
+ """
+ retval = []
+ for process in ps(psarg):
+ # Support for both BSD and UNIX syntax
+ # `ps aux` returns COMMAND, `ps -ef` returns CMD
+ try:
+ command = process['COMMAND']
+ except KeyError:
+ command = process['CMD']
+
+ command = shlex.split(command)
+ if command[-1] == '<defunct>':
+ command = command[:-1]
+ if not command or not defunct:
+ continue
+ if 'STAT' in process and not defunct:
+ if process['STAT'] == 'Z+':
+ continue
+ prog = command[0]
+ basename = os.path.basename(prog)
+ if basename == name:
+ retval.append((int(process['PID']), command))
+ return retval
+
+def get_pids(name):
+ """Get all the pids matching name"""
+
+ if mozinfo.isWin:
+ # use the windows-specific implementation
+ import wpk
+ return wpk.get_pids(name)
+ else:
+ return [pid for pid,_ in running_processes(name)]
+
+if __name__ == '__main__':
+ pids = set()
+ for i in sys.argv[1:]:
+ pids.update(get_pids(i))
+ for i in sorted(pids):
+ print i