431
+ − 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()
