GNUnet  0.10.x
terminate.py
Go to the documentation of this file.
1 #!/usr/bin/python3
2 # This file is part of GNUnet.
3 # (C) 2011, 2018 Christian Grothoff (and other contributing authors)
4 #
5 # GNUnet is free software: you can redistribute it and/or modify it
6 # under the terms of the GNU Affero General Public License as published
7 # by the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # GNUnet is distributed in the hope that it will be useful, but
11 # WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 # Affero General Public License for more details.
14 #
15 # You should have received a copy of the GNU Affero General Public License
16 # along with this program. If not, see <http://www.gnu.org/licenses/>.
17 #
18 # SPDX-License-Identifier: AGPL3.0-or-later
19 #
20 # Utility module that implements safe process termination for W32.
21 # For other platforms it's equivalent to Popen.kill ()
22 # Requires pywin32 on W32.
23 
24 import sys
25 import subprocess
26 import os
27 if os.name == 'nt':
28  import win32api
29  import win32process
30 
31 
32 class dummyobj (object):
33  pass
34 
35 
37  if os.name == 'nt':
38  p = dummyobj()
39  p._handle = win32api.OpenProcess(2 | 1024 | 8 | 32 | 16, 0, pid)
40  result = safe_terminate_process(p, code)
41  win32api.CloseHandle(p._handle)
42  return result
43  else:
44  # XXX (F821): Undefined name 'SIGKILL'
45  return os.kill(int(pid), SIGKILL)
46 
47 
48 def safe_terminate_process(proc, code):
49  if os.name == 'nt':
50  cp = win32api.GetCurrentProcess()
51  result = False
52  dupproc = win32api.DuplicateHandle(cp, proc._handle, cp, 2 | 1024 | 8 | 32 | 16, 0, 0)
53  try:
54  exitcode = win32process.GetExitCodeProcess(dupproc)
55  if exitcode == 0x103:
56  kernel32 = win32api.GetModuleHandle("kernel32")
57  exitprocess = win32api.GetProcAddress(kernel32, "ExitProcess")
58  th, tid = win32process.CreateRemoteThread(dupproc, None, 0, exitprocess, code, 0)
59  win32api.CloseHandle(th)
60  result = True
61  else:
62  result = True
63  # except failed to get exit code? failed to get module handle?
64  finally:
65  win32api.CloseHandle(dupproc)
66  return result
67  else:
68  return proc.kill()
def safe_terminate_process_by_pid(pid, code)
Definition: terminate.py:36
def safe_terminate_process(proc, code)
Definition: terminate.py:48