1 """Script used to test os.kill on Windows, for issue #1220212 2 3 This script is started as a subprocess in test_os and is used to test the 4 CTRL_C_EVENT and CTRL_BREAK_EVENT signals, which requires a custom handler 5 to be written into the kill target. 6 7 See http://msdn.microsoft.com/en-us/library/ms685049%28v=VS.85%29.aspx for a 8 similar example in C. 9 """ 10 11 from ctypes import wintypes, WINFUNCTYPE 12 import signal 13 import ctypes 14 import mmap 15 import sys 16 17 # Function prototype for the handler function. Returns BOOL, takes a DWORD. 18 HandlerRoutine = WINFUNCTYPE(wintypes.BOOL, wintypes.DWORD) 19 20 def _ctrl_handler(sig): 21 """Handle a sig event and return 0 to terminate the process""" 22 if sig == signal.CTRL_C_EVENT: 23 pass 24 elif sig == signal.CTRL_BREAK_EVENT: 25 pass 26 else: 27 print("UNKNOWN EVENT") 28 return 0 29 30 ctrl_handler = HandlerRoutine(_ctrl_handler) 31 32 33 SetConsoleCtrlHandler = ctypes.windll.kernel32.SetConsoleCtrlHandler 34 SetConsoleCtrlHandler.argtypes = (HandlerRoutine, wintypes.BOOL) 35 SetConsoleCtrlHandler.restype = wintypes.BOOL 36 37 if __name__ == "__main__": 38 # Add our console control handling function with value 1 39 if not SetConsoleCtrlHandler(ctrl_handler, 1): 40 print("Unable to add SetConsoleCtrlHandler") 41 exit(-1) 42 43 # Awake main process 44 m = mmap.mmap(-1, 1, sys.argv[1]) 45 m[0] = 1 46 47 # Do nothing but wait for the signal 48 while True: 49 pass 50
