1 #!/usr/bin/env python |
|
2 |
|
3 import os, sys, time, errno, signal |
|
4 |
|
5 if os.name =='nt': |
|
6 import ctypes |
|
7 |
|
8 def _check(ret, expectederr=None): |
|
9 if ret == 0: |
|
10 winerrno = ctypes.GetLastError() |
|
11 if winerrno == expectederr: |
|
12 return True |
|
13 raise ctypes.WinError(winerrno) |
|
14 |
|
15 def kill(pid, logfn, tryhard=True): |
|
16 logfn('# Killing daemon process %d' % pid) |
|
17 PROCESS_TERMINATE = 1 |
|
18 PROCESS_QUERY_INFORMATION = 0x400 |
|
19 SYNCHRONIZE = 0x00100000 |
|
20 WAIT_OBJECT_0 = 0 |
|
21 WAIT_TIMEOUT = 258 |
|
22 handle = ctypes.windll.kernel32.OpenProcess( |
|
23 PROCESS_TERMINATE|SYNCHRONIZE|PROCESS_QUERY_INFORMATION, |
|
24 False, pid) |
|
25 if handle == 0: |
|
26 _check(0, 87) # err 87 when process not found |
|
27 return # process not found, already finished |
|
28 try: |
|
29 r = ctypes.windll.kernel32.WaitForSingleObject(handle, 100) |
|
30 if r == WAIT_OBJECT_0: |
|
31 pass # terminated, but process handle still available |
|
32 elif r == WAIT_TIMEOUT: |
|
33 _check(ctypes.windll.kernel32.TerminateProcess(handle, -1)) |
|
34 else: |
|
35 _check(r) |
|
36 |
|
37 # TODO?: forcefully kill when timeout |
|
38 # and ?shorter waiting time? when tryhard==True |
|
39 r = ctypes.windll.kernel32.WaitForSingleObject(handle, 100) |
|
40 # timeout = 100 ms |
|
41 if r == WAIT_OBJECT_0: |
|
42 pass # process is terminated |
|
43 elif r == WAIT_TIMEOUT: |
|
44 logfn('# Daemon process %d is stuck') |
|
45 else: |
|
46 _check(r) # any error |
|
47 except: #re-raises |
|
48 ctypes.windll.kernel32.CloseHandle(handle) # no _check, keep error |
|
49 raise |
|
50 _check(ctypes.windll.kernel32.CloseHandle(handle)) |
|
51 |
|
52 else: |
|
53 def kill(pid, logfn, tryhard=True): |
|
54 try: |
|
55 os.kill(pid, 0) |
|
56 logfn('# Killing daemon process %d' % pid) |
|
57 os.kill(pid, signal.SIGTERM) |
|
58 if tryhard: |
|
59 for i in range(10): |
|
60 time.sleep(0.05) |
|
61 os.kill(pid, 0) |
|
62 else: |
|
63 time.sleep(0.1) |
|
64 os.kill(pid, 0) |
|
65 logfn('# Daemon process %d is stuck - really killing it' % pid) |
|
66 os.kill(pid, signal.SIGKILL) |
|
67 except OSError, err: |
|
68 if err.errno != errno.ESRCH: |
|
69 raise |
|
70 |
|
71 def killdaemons(pidfile, tryhard=True, remove=False, logfn=None): |
|
72 if not logfn: |
|
73 logfn = lambda s: s |
|
74 # Kill off any leftover daemon processes |
|
75 try: |
|
76 fp = open(pidfile) |
|
77 for line in fp: |
|
78 try: |
|
79 pid = int(line) |
|
80 except ValueError: |
|
81 continue |
|
82 kill(pid, logfn, tryhard) |
|
83 fp.close() |
|
84 if remove: |
|
85 os.unlink(pidfile) |
|
86 except IOError: |
|
87 pass |
|
88 |
|
89 if __name__ == '__main__': |
|
90 path, = sys.argv[1:] |
|
91 killdaemons(path) |
|