1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#!/usr/bin/env python
# ===-- klee-control ------------------------------------------------------===##
#
# The KLEE Symbolic Virtual Machine
#
# This file is distributed under the University of Illinois Open Source
# License. See LICENSE.TXT for details.
#
# ===----------------------------------------------------------------------===##
import os, signal, popen2
def getPID(dir):
f = open(os.path.join(dir,'info'))
for ln in f.readlines():
if ln.startswith('PID: '):
return int(ln[5:])
return None
def execCmd(pid, gdbCmd, opts):
cmd = ("gdb " +
"--batch " +
"--pid=%d " +
"--eval-command=\"%s\" " +
"--eval-command=detach")%(pid,gdbCmd)
cout,cin = popen2.popen2(cmd)
cin.close()
return cout.read()
def main():
from optparse import OptionParser
op = OptionParser("usage: %prog <PID | test directory>")
op.add_option('','--backtrace', dest='backtrace',
action='store_true', default=False)
op.add_option('-s','--stop-forking', dest='stopForking',
action='store_true', default=False)
op.add_option('-H','--halt-execution', dest='haltExecution',
action='store_true', default=False)
op.add_option('-d','--dump-states', dest='dumpStates',
action='store_true', default=False)
op.add_option('-t','--dump-tree', dest='dumpTree',
action='store_true', default=False)
op.add_option('-i','--int', dest='int',
action='store_true', default=False)
op.add_option('-k','--kill', dest='kill',
action='store_true', default=False)
op.add_option('','--print-pid', dest='printPid',
action='store_true', default=False)
op.add_option('','--print-ticks', dest='printTicks',
action='store_true', default=False)
opts,args = op.parse_args()
if len(args) != 1:
op.error("invalid arguments")
try:
pid = int(args[0])
except:
pid = None
if pid is None:
try:
pid = getPID(args[0])
except:
pid = None
if pid is None:
op.error("unable to determine PID (bad pid or test directory)")
if opts.printPid:
print pid
return
print 'pid: %d'%pid
if opts.backtrace:
execCmd(pid, 'bt', opts)
if opts.dumpStates:
execCmd(pid, "p dumpStates = 1", opts)
if opts.dumpTree:
execCmd(pid, "p dumpExecutionTree = 1", opts)
if opts.stopForking:
execCmd(pid, 'p stop_forking()', opts)
if opts.haltExecution:
execCmd(pid, 'p halt_execution()', opts)
if opts.printTicks:
res = execCmd(pid, 'p timerTicks', opts)
lns = res.split('\n')
for ln in lns:
if ln.startswith('$1') and '=' in ln:
print ln.split('=',1)[1].strip()
if opts.int:
os.kill(pid, signal.SIGINT)
if opts.kill:
os.kill(pid, signal.SIGKILL)
if __name__=='__main__':
main()
|
