Mercurial > hg > config
comparison python/attach.py @ 431:72f33eb3aa23
add an attach.py script
| author | Jeff Hammel <jhammel@mozilla.com> |
|---|---|
| date | Thu, 08 Aug 2013 00:24:41 -0700 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 430:92ee74af9456 | 431:72f33eb3aa23 |
|---|---|
| 1 #!/usr/bin/env python | |
| 2 | |
| 3 """ | |
| 4 attach strace to a process: | |
| 5 | |
| 6 624 sudo strace -p $(pidof firefox) | |
| 7 625 kill 6976 | |
| 8 626 isrunning firefox | |
| 9 | |
| 10 Optionally kill the process when detached | |
| 11 """ | |
| 12 | |
| 13 | |
| 14 import optparse | |
| 15 import os | |
| 16 import subprocess | |
| 17 import sys | |
| 18 | |
| 19 from subprocess import check_output | |
| 20 | |
| 21 def main(args=sys.argv[1:]): | |
| 22 | |
| 23 usage = '%prog [options] ProgramName_or_PID' | |
| 24 parser = optparse.OptionParser(usage=usage, description=__doc__) | |
| 25 parser.add_option('-k', '--kill', | |
| 26 action='store_true', default=False, | |
| 27 help="kill process after strace is done") | |
| 28 options, args = parser.parse_args(args) | |
| 29 if len(args) != 1: | |
| 30 parser.print_usage() | |
| 31 parser.error("Please specify program or PID to attach to") | |
| 32 | |
| 33 # get the PID | |
| 34 try: | |
| 35 pid = int(args[0]) | |
| 36 except ValueError: | |
| 37 pid = check_output(["pidof", args[0]]).strip().split() | |
| 38 if len(pid) > 1: | |
| 39 parser.error("Multiple PIDs found for %s:\n%s" % (args[0], | |
| 40 '\n'.join(pid))) | |
| 41 # TODO: handle not matching case | |
| 42 pid = int(pid[0]) | |
| 43 | |
| 44 # invoke strace | |
| 45 subprocess.call(['sudo', 'strace', '-p', str(pid)]) | |
| 46 | |
| 47 # kill if specified | |
| 48 os.kill(pid, 9) # TODO: better than SIGKILL | |
| 49 | |
| 50 if __name__ == '__main__': | |
| 51 main() |