killdaemons.py
106 lines
| 3.4 KiB
| text/x-python
|
PythonLexer
/ tests / killdaemons.py
|
r7344 | #!/usr/bin/env python | ||
|
r28942 | from __future__ import absolute_import | ||
import errno | ||||
import os | ||||
import signal | ||||
import sys | ||||
import time | ||||
|
r7344 | |||
|
r17465 | if os.name =='nt': | ||
import ctypes | ||||
|
r20493 | |||
def _check(ret, expectederr=None): | ||||
if ret == 0: | ||||
winerrno = ctypes.GetLastError() | ||||
if winerrno == expectederr: | ||||
return True | ||||
raise ctypes.WinError(winerrno) | ||||
|
r17465 | def kill(pid, logfn, tryhard=True): | ||
logfn('# Killing daemon process %d' % pid) | ||||
PROCESS_TERMINATE = 1 | ||||
|
r20496 | PROCESS_QUERY_INFORMATION = 0x400 | ||
|
r20698 | SYNCHRONIZE = 0x00100000 | ||
|
r20494 | WAIT_OBJECT_0 = 0 | ||
WAIT_TIMEOUT = 258 | ||||
|
r17465 | handle = ctypes.windll.kernel32.OpenProcess( | ||
|
r20496 | PROCESS_TERMINATE|SYNCHRONIZE|PROCESS_QUERY_INFORMATION, | ||
False, pid) | ||||
|
r20493 | if handle == 0: | ||
|
r20495 | _check(0, 87) # err 87 when process not found | ||
|
r20493 | return # process not found, already finished | ||
try: | ||||
|
r20496 | r = ctypes.windll.kernel32.WaitForSingleObject(handle, 100) | ||
if r == WAIT_OBJECT_0: | ||||
pass # terminated, but process handle still available | ||||
elif r == WAIT_TIMEOUT: | ||||
_check(ctypes.windll.kernel32.TerminateProcess(handle, -1)) | ||||
else: | ||||
_check(r) | ||||
|
r20494 | |||
# TODO?: forcefully kill when timeout | ||||
# and ?shorter waiting time? when tryhard==True | ||||
r = ctypes.windll.kernel32.WaitForSingleObject(handle, 100) | ||||
# timeout = 100 ms | ||||
if r == WAIT_OBJECT_0: | ||||
pass # process is terminated | ||||
elif r == WAIT_TIMEOUT: | ||||
logfn('# Daemon process %d is stuck') | ||||
else: | ||||
|
r21194 | _check(r) # any error | ||
|
r20493 | except: #re-raises | ||
ctypes.windll.kernel32.CloseHandle(handle) # no _check, keep error | ||||
raise | ||||
_check(ctypes.windll.kernel32.CloseHandle(handle)) | ||||
|
r17465 | else: | ||
def kill(pid, logfn, tryhard=True): | ||||
try: | ||||
os.kill(pid, 0) | ||||
logfn('# Killing daemon process %d' % pid) | ||||
os.kill(pid, signal.SIGTERM) | ||||
if tryhard: | ||||
for i in range(10): | ||||
time.sleep(0.05) | ||||
os.kill(pid, 0) | ||||
else: | ||||
time.sleep(0.1) | ||||
os.kill(pid, 0) | ||||
logfn('# Daemon process %d is stuck - really killing it' % pid) | ||||
os.kill(pid, signal.SIGKILL) | ||||
|
r25031 | except OSError as err: | ||
|
r17465 | if err.errno != errno.ESRCH: | ||
raise | ||||
|
r17464 | def killdaemons(pidfile, tryhard=True, remove=False, logfn=None): | ||
if not logfn: | ||||
logfn = lambda s: s | ||||
# Kill off any leftover daemon processes | ||||
try: | ||||
|
r32677 | pids = [] | ||
with open(pidfile) as fp: | ||||
for line in fp: | ||||
try: | ||||
pid = int(line) | ||||
if pid <= 0: | ||||
raise ValueError | ||||
except ValueError: | ||||
logfn('# Not killing daemon process %s - invalid pid' | ||||
% line.rstrip()) | ||||
continue | ||||
pids.append(pid) | ||||
for pid in pids: | ||||
|
r17465 | kill(pid, logfn, tryhard) | ||
|
r17464 | if remove: | ||
os.unlink(pidfile) | ||||
except IOError: | ||||
pass | ||||
if __name__ == '__main__': | ||||
|
r25473 | if len(sys.argv) > 1: | ||
path, = sys.argv[1:] | ||||
else: | ||||
path = os.environ["DAEMON_PIDS"] | ||||
|
r17466 | killdaemons(path) | ||