python (3.11.7)
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