1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- #!@PYTHON@
- # This file is part of GNUnet.
- # (C) 2011, 2018 Christian Grothoff (and other contributing authors)
- #
- # GNUnet is free software: you can redistribute it and/or modify it
- # under the terms of the GNU Affero General Public License as published
- # by the Free Software Foundation, either version 3 of the License, or
- # (at your option) any later version.
- #
- # GNUnet is distributed in the hope that it will be useful, but
- # WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- # Affero General Public License for more details.
- #
- # You should have received a copy of the GNU Affero General Public License
- # along with this program. If not, see <http://www.gnu.org/licenses/>.
- #
- # SPDX-License-Identifier: AGPL3.0-or-later
- #
- # Utility module that implements safe process termination for W32.
- # For other platforms it's equivalent to Popen.kill ()
- # Requires pywin32 on W32.
- import sys
- import subprocess
- import os
- if os.name == 'nt':
- import win32api
- import win32process
- class dummyobj (object):
- pass
- def safe_terminate_process_by_pid(pid, code):
- if os.name == 'nt':
- p = dummyobj()
- p._handle = win32api.OpenProcess(2 | 1024 | 8 | 32 | 16, 0, pid)
- result = safe_terminate_process(p, code)
- win32api.CloseHandle(p._handle)
- return result
- else:
- # XXX (F821): Undefined name 'SIGKILL'
- return os.kill(int(pid), SIGKILL)
- def safe_terminate_process(proc, code):
- if os.name == 'nt':
- cp = win32api.GetCurrentProcess()
- result = False
- dupproc = win32api.DuplicateHandle(cp, proc._handle, cp, 2 | 1024 | 8 | 32 | 16, 0, 0)
- try:
- exitcode = win32process.GetExitCodeProcess(dupproc)
- if exitcode == 0x103:
- kernel32 = win32api.GetModuleHandle("kernel32")
- exitprocess = win32api.GetProcAddress(kernel32, "ExitProcess")
- th, tid = win32process.CreateRemoteThread(dupproc, None, 0, exitprocess, code, 0)
- win32api.CloseHandle(th)
- result = True
- else:
- result = True
- # except failed to get exit code? failed to get module handle?
- finally:
- win32api.CloseHandle(dupproc)
- return result
- else:
- return proc.kill()
|