contrib/terminate.py.in

   1 #!@PYTHON@
   2 #    This file is part of GNUnet.
   3 #    (C) 2011 Christian Grothoff (and other contributing authors)
   4 #
   5 #    GNUnet is free software; you can redistribute it and/or modify
   6 #    it under the terms of the GNU General Public License as published
   7 #    by the Free Software Foundation; either version 2, or (at your
   8 #    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 #    General Public License for more details.
  14 #
  15 #    You should have received a copy of the GNU General Public License
  16 #    along with GNUnet; see the file COPYING.  If not, write to the
  17 #    Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18 #    Boston, MA 02111-1307, USA.
  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 \r
  24 import sys\r
  25 import os\r
  26 import subprocess\r
  27 if os.name == 'nt':\r
  28   import win32api\r
  29   import win32process\r
  30 \r
  31 class dummyobj (object):\r
  32   pass\r
  33 \r
  34 def safe_terminate_process_by_pid (pid, code):\r
  35   if os.name == 'nt':\r
  36     p = dummyobj ()\r
  37     p._handle = win32api.OpenProcess (2 | 1024 | 8 | 32 | 16, 0, pid)\r
  38     result = safe_terminate_process (p, code)\r
  39     win32api.CloseHandle (p._handle)\r
  40     return result\r
  41   else:\r
  42     return os.kill (int (pid), SIGKILL)\r
  43 \r
  44 def safe_terminate_process (proc, code):\r
  45   if os.name == 'nt':\r
  46     cp = win32api.GetCurrentProcess ()\r
  47     result = False\r
  48     dupproc = win32api.DuplicateHandle (cp, proc._handle, cp, 2 | 1024 | 8 | 32 | 16, 0, 0)\r
  49     try:\r
  50       exitcode = win32process.GetExitCodeProcess (dupproc)\r
  51       if exitcode == 0x103:\r
  52         kernel32 = win32api.GetModuleHandle ("kernel32")\r
  53         exitprocess = win32api.GetProcAddress (kernel32, "ExitProcess")\r
  54         th, tid = win32process.CreateRemoteThread (dupproc, None, 0, exitprocess, code, 0)\r
  55         win32api.CloseHandle (th)\r
  56         result = True\r
  57       else:\r
  58         result = True\r
  59     # except failed to get exit code? failed to get module handle?\r
  60     finally:\r
  61       win32api.CloseHandle (dupproc)\r
  62     return result\r
  63   else:\r
  64     return proc.kill ()\r