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