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 from builtins import object
  25 import sys
  26 import subprocess
  27 import os
  28 if os.name == 'nt':
  29     import win32api
  30     import win32process
  31
  32
  33 class dummyobj (object):
  34     pass
  35
  36
  37 def safe_terminate_process_by_pid(pid, code):
  38     if os.name == 'nt':
  39         p = dummyobj()
  40         p._handle = win32api.OpenProcess(2 | 1024 | 8 | 32 | 16, 0, pid)
  41         result = safe_terminate_process(p, code)
  42         win32api.CloseHandle(p._handle)
  43         return result
  44     else:
  45         # XXX (F821): Undefined name 'SIGKILL'
  46         return os.kill(int(pid), SIGKILL)
  47
  48
  49 def safe_terminate_process(proc, code):
  50     if os.name == 'nt':
  51         cp = win32api.GetCurrentProcess()
  52         result = False
  53         dupproc = win32api.DuplicateHandle(cp, proc._handle, cp, 2 | 1024 | 8 | 32 | 16, 0, 0)
  54         try:
  55             exitcode = win32process.GetExitCodeProcess(dupproc)
  56             if exitcode == 0x103:
  57                 kernel32 = win32api.GetModuleHandle("kernel32")
  58                 exitprocess = win32api.GetProcAddress(kernel32, "ExitProcess")
  59                 th, tid = win32process.CreateRemoteThread(dupproc, None, 0, exitprocess, code, 0)
  60                 win32api.CloseHandle(th)
  61                 result = True
  62             else:
  63                 result = True
  64         # except failed to get exit code? failed to get module handle?
  65         finally:
  66             win32api.CloseHandle(dupproc)
  67         return result
  68     else:
  69         return proc.kill()