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
|
import sys, os
import threading
import ctypes
def childkill(pid):
global timedout
timedout = True
sys.stderr.write("==== test running for %d seconds ====\n" % timeout)
sys.stderr.write("="*26 + "timedout" + "="*26 + "\n")
ctypes.windll.kernel32.TerminateProcess(pid, 1)
if __name__ == '__main__':
PROCESS_TERMINATE = 0x1
timeout = float(sys.argv[1])
timedout = False
pid = os.spawnv(os.P_NOWAIT, sys.argv[2], sys.argv[2:])
t = threading.Timer(timeout, childkill, (pid,))
t.start()
while True:
try:
pid, status = os.waitpid(pid, 0)
except KeyboardInterrupt:
continue
else:
t.cancel()
break
#print 'status ', status >> 8
sys.exit(status >> 8)
|
