contrib/scripts/terminate.py.in

   1 #!@PYTHON@
   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
  36 def safe_terminate_process_by_pid(pid, code):
  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()