# 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/>.
#
import signal
import terminate
-if os.name == 'nt':
- from win32com.client import GetObject
- WMI = GetObject('winmgmts:')
-
-
def get_process_list():
result = []
- if os.name == 'nt':
- processes = WMI.InstancesOf('Win32_Process')
- for p in processes:
- result.append((p.Properties_('ProcessId').Value, re.sub(r'(.+)\.exe', r'\1', p.Properties_('Name').Value)))
- else:
- pids = [pid for pid in os.listdir('/proc') if pid.isdigit()]
- for pid in pids:
- with open(os.path.join('/proc', pid, 'cmdline'), 'rb') as p:
- cmdline = p.read().split('\x00')
- if len(cmdline) > 0:
- result.append((pid, cmdline[0]))
+ pids = [pid for pid in os.listdir('/proc') if pid.isdigit()]
+ for pid in pids:
+ with open(os.path.join('/proc', pid, 'cmdline'), 'rb') as p:
+ cmdline = p.read().split('\x00')
+ if len(cmdline) > 0:
+ result.append((pid, cmdline[0]))
return result
# 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/>.
#
import sys
import subprocess
import os
-if os.name == 'nt':
- import win32api
- import win32process
class dummyobj (object):
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)
+ # 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()
+ return proc.kill()