Upreckon: win32.py annotate

annotate win32.py @ 80:809b77302b21

Win32-specific module with memory and CPU time limits The Win32-specific implementation of call() and friends now lives in module win32, looks clean and in addition is able to enforce memory and CPU time limits on NT kernels, in particular on Windows 2000 and up asking the system to terminate the process as soon as or (in the case of CPU time) almost as soon as the limits are broken. According to my observations, malloc() in the limited process does not return NULL when memory usage is close to the limit and instead crashes the process (which Upreckon happily translates into 'memory limit exceeded'). The catch is that the module is not actually used yet; coming soon.

author	Oleg Oshmyan <chortos@inbox.lv>
date	Wed, 16 Feb 2011 00:01:33 +0000
parents
children	24752db487c5

rev	line source
80 809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	1 # Copyright (c) 2011 Chortos-2 <chortos@inbox.lv>
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	2
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	3 from __future__ import division, with_statement
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	4
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	5 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	6 from compat import *
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	7 except ImportError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	8 import __main__
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	9 __main__.import_error(sys.exc_info()[1])
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	10
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	11 from __main__ import clock
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	12 from ctypes import *
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	13 from ctypes.wintypes import *
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	14 from subprocess import Popen
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	15
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	16 # Defaults that may be overwritten by values from _subprocess
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	17 INFINITE = -1
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	18 STD_INPUT_HANDLE = -10
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	19 WAIT_OBJECT_0 = 0
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	20
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	21 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	22 from _subprocess import *
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	23 except ImportError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	24 pass
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	25
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	26 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	27 from numbers import Integral
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	28 except ImportError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	29 Integral = int, long
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	30
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	31 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	32 from collections import namedtuple
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	33 except ImportError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	34 from operator import itemgetter
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	35 class ProcessTimes(tuple):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	36 __slots__ = ()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	37 __new__ = lambda cls, kernel, user: tuple.__new__(cls, (kernel, user))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	38 __getnewargs__ = lambda self: tuple(self)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	39 kernel, user = (property(itemgetter(i)) for i in (0, 1))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	40 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	41 ProcessTimes = namedtuple('ProcessTimes', 'kernel user')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	42
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	43 __all__ = 'call', 'kill', 'terminate'
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	44
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	45
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	46 # Automatically convert _subprocess handle objects into low-level HANDLEs
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	47 # and replicate their functionality for our own use
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	48 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	49 _subprocess_handle = type(GetCurrentProcess())
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	50 except NameError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	51 _subprocess_handle = Integral
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	52 class Handle(object):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	53 @staticmethod
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	54 def from_param(handle):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	55 if isinstance(handle, (_subprocess_handle, Integral)):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	56 return HANDLE(int(handle))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	57 elif isinstance(handle, Handle):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	58 return HANDLE(handle.handle)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	59 elif isinstance(handle, HANDLE):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	60 return handle
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	61 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	62 raise TypeError('cannot convert %s to a handle' %
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	63 type(handle).__name__)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	64
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	65 __slots__ = 'handle'
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	66
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	67 def __init__(self, handle):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	68 if isinstance(handle, Integral):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	69 self.handle = handle
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	70 elif isinstance(handle, HANDLE):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	71 self.handle = handle.value
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	72 elif isinstance(handle, Handle):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	73 self.handle = handle.handle
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	74 elif isinstance(handle, _subprocess_handle):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	75 handle = HANDLE(int(handle))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	76 flags = DWORD()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	77 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	78 if windll.kernel32.GetHandleInformation(handle, byref(flags)):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	79 flags = flags.value
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	80 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	81 flags = 0
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	82 except AttributeError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	83 # Available on NT 3.51 and up, NT line only
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	84 flags = 0
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	85 proc = HANDLE(int(GetCurrentProcess()))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	86 handle = DuplicateHandle(proc, handle, proc, 0, flags & 1, 2)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	87 self.handle = handle.Detach()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	88 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	89 raise TypeError("Handle() argument must be a handle, not '%s'" %
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	90 type(name).__name__)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	91
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	92 def __int__(self):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	93 return int(self.handle)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	94
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	95 def Detach(self):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	96 handle = self.handle
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	97 self.handle = None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	98 return handle
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	99
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	100 # This is also __del__, so only locals are accessed
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	101 def Close(self, _CloseHandle=windll.kernel32.CloseHandle, _HANDLE=HANDLE):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	102 if self.handle:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	103 _CloseHandle(_HANDLE(self.handle))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	104 self.handle = None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	105 __del__ = Close
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	106
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	107 CHAR = c_char
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	108 INVALID_HANDLE_VALUE = HANDLE(-1).value
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	109 LPDWORD = POINTER(DWORD)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	110 LPFILETIME = POINTER(FILETIME)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	111 SIZE_T = ULONG_PTR = WPARAM
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	112 ULONGLONG = c_ulonglong
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	113
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	114 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	115 unicode
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	116 except NameError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	117 LPCTSTR = LPCWSTR
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	118 unisuffix = 'W'
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	119 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	120 LPCTSTR = LPCSTR
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	121 unisuffix = 'A'
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	122
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	123
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	124 prototype = WINFUNCTYPE(BOOL, Handle,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	125 LPFILETIME, LPFILETIME, LPFILETIME, LPFILETIME)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	126 flags = ((1, 'process'),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	127 (2, 'creation'), (2, 'exit'), (2, 'kernel'), (2, 'user'))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	128 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	129 GetProcessTimes = prototype(('GetProcessTimes', windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	130 except AttributeError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	131 # Available on NT 3.5 and up, NT line only
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	132 GetProcessTimes = None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	133 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	134 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	135 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	136 ftimes = [t.dwHighDateTime << 32 \| t.dwLowDateTime for t in args[3:]]
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	137 kernel = ftimes[0] / 10000000
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	138 user = ftimes[1] / 10000000
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	139 return ProcessTimes(kernel, user)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	140 GetProcessTimes.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	141
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	142
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	143 class PROCESS_MEMORY_COUNTERS(Structure):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	144 _fields_ = (('cb', DWORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	145 ('PageFaultCount', DWORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	146 ('PeakWorkingSetSize', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	147 ('WorkingSetSize', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	148 ('QuotaPeakPagedPoolUsage', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	149 ('QuotaPagedPoolUsage', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	150 ('QuotaPeakNonPagedPoolUsage', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	151 ('QuotaNonPagedPoolUsage', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	152 ('PagefileUsage', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	153 ('PeakPagefileUsage', SIZE_T))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	154
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	155 prototype = WINFUNCTYPE(BOOL, Handle, POINTER(PROCESS_MEMORY_COUNTERS), DWORD)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	156 flags = ((1, 'process'), (2, 'counters'),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	157 (5, 'cb', sizeof(PROCESS_MEMORY_COUNTERS)))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	158 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	159 GetProcessMemoryInfo = prototype(('GetProcessMemoryInfo', windll.psapi),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	160 flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	161 except AttributeError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	162 # Available on NT 4.0 and up, NT line only
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	163 GetProcessMemoryInfo = None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	164 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	165 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	166 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	167 return args
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	168 GetProcessMemoryInfo.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	169
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	170
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	171 class _uChar_union(Union):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	172 _fields_ = (('UnicodeChar', WCHAR),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	173 ('AsciiChar', CHAR))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	174
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	175 class KEY_EVENT_RECORD(Structure):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	176 _fields_ = (('bKeyDown', BOOL),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	177 ('wRepeatCount', WORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	178 ('wVirtualKeyCode', WORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	179 ('wVirtualScanCode', WORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	180 ('uChar', _uChar_union),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	181 ('dwControlKeyState', DWORD))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	182
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	183 RIGHT_ALT_PRESSED = 0x001
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	184 LEFT_ALT_PRESSED = 0x002
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	185 RIGHT_CTRL_PRESSED = 0x004
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	186 LEFT_CTRL_PRESSED = 0x008
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	187 SHIFT_PRESSED = 0x010
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	188 NUMLOCK_ON = 0x020
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	189 SCROLLLOCK_ON = 0x040
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	190 CAPSLOCK_ON = 0x080
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	191 ENHANCED_KEY = 0x100
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	192
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	193 class _Event_union(Union):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	194 _fields_ = ('KeyEvent', KEY_EVENT_RECORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	195
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	196 class INPUT_RECORD(Structure):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	197 _fields_ = (('EventType', WORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	198 ('Event', _Event_union))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	199
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	200 KEY_EVENT = 0x01
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	201 MOUSE_EVENT = 0x02
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	202 WINDOW_BUFFER_SIZE_EVENT = 0x04
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	203 MENU_EVENT = 0x08
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	204 FOCUS_EVENT = 0x10
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	205
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	206 prototype = WINFUNCTYPE(BOOL, Handle, POINTER(INPUT_RECORD), DWORD, LPDWORD)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	207 flags = (1, 'input'), (2, 'buffer'), (5, 'length', 1), (2, 'number_read')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	208 ReadConsoleInput = prototype(('ReadConsoleInputA', windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	209 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	210 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	211 return args[1] if args[3] else None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	212 ReadConsoleInput.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	213
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	214
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	215 prototype = WINFUNCTYPE(BOOL, Handle)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	216 flags = (1, 'input'),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	217 FlushConsoleInputBuffer = prototype(('FlushConsoleInputBuffer',
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	218 windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	219 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	220 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	221 FlushConsoleInputBuffer.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	222
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	223
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	224 prototype = WINFUNCTYPE(BOOL, Handle, DWORD)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	225 flags = (1, 'console'), (1, 'mode')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	226 SetConsoleMode = prototype(('SetConsoleMode', windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	227 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	228 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	229 SetConsoleMode.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	230
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	231 ENABLE_PROCESSED_INPUT = 0x001
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	232 ENABLE_LINE_INPUT = 0x002
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	233 ENABLE_ECHO_INPUT = 0x004
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	234 ENABLE_WINDOW_INPUT = 0x008
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	235 ENABLE_MOUSE_INPUT = 0x010
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	236 ENABLE_INSERT_MODE = 0x020
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	237 ENABLE_QUICK_EDIT_MODE = 0x040
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	238 ENABLE_EXTENDED_FLAGS = 0x080
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	239
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	240 ENABLE_PROCESSED_OUTPUT = 1
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	241 ENABLE_WRAP_AT_EOL_OUTPUT = 2
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	242
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	243
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	244 prototype = WINFUNCTYPE(HANDLE, c_void_p, LPCTSTR)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	245 flags = (5, 'attributes'), (1, 'name')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	246 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	247 CreateJobObject = prototype(('CreateJobObject'+unisuffix, windll.kernel32),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	248 flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	249 except AttributeError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	250 # Available on 2000 and up, NT line only
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	251 CreateJobObject = lambda name: None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	252 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	253 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	254 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	255 return Handle(result)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	256 CreateJobObject.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	257
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	258
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	259 prototype = WINFUNCTYPE(BOOL, Handle, Handle)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	260 flags = (1, 'job'), (1, 'handle')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	261 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	262 AssignProcessToJobObject = prototype(('AssignProcessToJobObject',
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	263 windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	264 except AttributeError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	265 # Available on 2000 and up, NT line only
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	266 AssignProcessToJobObject = lambda job, handle: None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	267 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	268 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	269 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	270 AssignProcessToJobObject.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	271
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	272
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	273 class JOBOBJECT_BASIC_LIMIT_INFORMATION(Structure):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	274 _fields_ = (('PerProcessUserTimeLimit', LARGE_INTEGER),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	275 ('PerJobUserTimeLimit', LARGE_INTEGER),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	276 ('LimitFlags', DWORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	277 ('MinimumWorkingSetSize', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	278 ('MaximumWorkingSetSize', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	279 ('ActiveProcessLimit', DWORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	280 ('Affinity', ULONG_PTR),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	281 ('PriorityClass', DWORD),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	282 ('SchedulingClass', DWORD))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	283
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	284 JOB_OBJECT_LIMIT_WORKINGSET = 0x0001
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	285 JOB_OBJECT_LIMIT_PROCESS_TIME = 0x0002
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	286 JOB_OBJECT_LIMIT_JOB_TIME = 0x0004
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	287 JOB_OBJECT_LIMIT_ACTIVE_PROCESS = 0x0008
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	288 JOB_OBJECT_LIMIT_AFFINITY = 0x0010
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	289 JOB_OBJECT_LIMIT_PRIORITY_CLASS = 0x0020
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	290 JOB_OBJECT_LIMIT_PRESERVE_JOB_TIME = 0x0040
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	291 JOB_OBJECT_LIMIT_SCHEDULING_CLASS = 0x0080
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	292 JOB_OBJECT_LIMIT_PROCESS_MEMORY = 0x0100
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	293 JOB_OBJECT_LIMIT_JOB_MEMORY = 0x0200
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	294 JOB_OBJECT_LIMIT_DIE_ON_UNHANDLED_EXCEPTION = 0x0400
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	295 JOB_OBJECT_LIMIT_BREAKAWAY_OK = 0x0800
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	296 JOB_OBJECT_LIMIT_SILENT_BREAKAWAY_OK = 0x1000
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	297 JOB_OBJECT_LIMIT_KILL_ON_JOB_CLOSE = 0x2000
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	298 JOB_OBJECT_LIMIT_SUBSET_AFFINITY = 0x4000
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	299
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	300 class IO_COUNTERS(Structure):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	301 _fields_ = (('ReadOperationCount', ULONGLONG),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	302 ('WriteOperationCount', ULONGLONG),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	303 ('OtherOperationCount', ULONGLONG),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	304 ('ReadTransferCount', ULONGLONG),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	305 ('WriteTransferCount', ULONGLONG),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	306 ('OtherTransferCount', ULONGLONG))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	307
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	308 class JOBOBJECT_EXTENDED_LIMIT_INFORMATION(Structure):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	309 _fields_ = (('BasicLimitInformation', JOBOBJECT_BASIC_LIMIT_INFORMATION),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	310 ('IoInfo', IO_COUNTERS),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	311 ('ProcessMemoryLimit', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	312 ('JobMemoryLimit', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	313 ('PeakProcessMemoryUsed', SIZE_T),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	314 ('PeakJobMemoryUsed', SIZE_T))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	315
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	316 prototype = WINFUNCTYPE(BOOL, HANDLE, c_int, c_void_p, DWORD)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	317 flags = (1, 'job'), (1, 'infoclass'), (1, 'info'), (1, 'infosize')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	318 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	319 _setjobinfo = prototype(('SetInformationJobObject',windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	320 except AttributeError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	321 # Available on 2000 and up, NT line only
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	322 SetInformationJobObject = lambda job, infoclass, info: None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	323 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	324 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	325 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	326 _setjobinfo.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	327 def SetInformationJobObject(job, infoclass, info):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	328 return _setjobinfo(job, infoclass, info, sizeof(info))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	329
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	330 (
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	331 JobObjectBasicAccountingInformation,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	332 JobObjectBasicLimitInformation,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	333 JobObjectBasicProcessIdList,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	334 JobObjectBasicUIRestrictions,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	335 JobObjectSecurityLimitInformation,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	336 JobObjectEndOfJobTimeInformation,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	337 JobObjectAssociateCompletionPortInformation,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	338 JobObjectBasicAndIoAccountingInformation,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	339 JobObjectExtendedLimitInformation,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	340 JobObjectJobSetInformation,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	341 MaxJobObjectInfoClass
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	342 ) = range(1, 12)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	343
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	344
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	345 prototype = WINFUNCTYPE(DWORD, DWORD, POINTER(HANDLE), BOOL, DWORD)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	346 flags = (1, 'count'), (1, 'handles'), (1, 'wait_all'), (1, 'milliseconds')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	347 _wait_multiple = prototype(('WaitForMultipleObjects', windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	348 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	349 if result == WAIT_FAILED: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	350 return args
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	351 _wait_multiple.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	352 def WaitForMultipleObjects(handles, wait_all, timeout):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	353 n = len(handles)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	354 handles = (Handle.from_param(handle) for handle in handles)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	355 timeout = ceil(timeout * 1000)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	356 return _wait_multiple(n, (HANDLE * n)(*handles), wait_all, timeout)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	357
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	358 # WAIT_OBJECT_0 defined at the top of the file
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	359 WAIT_ABANDONED_0 = 0x00000080
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	360 WAIT_TIMEOUT = 0x00000102
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	361 WAIT_FAILED = 0xFFFFFFFF
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	362
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	363
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	364 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	365 _wait_single = WaitForSingleObject
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	366 except NameError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	367 prototype = WINFUNCTYPE(DWORD, Handle, DWORD)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	368 flags = (1, 'handle'), (1, 'milliseconds')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	369 _wait_single = prototype(('WaitForSingleObject', windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	370 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	371 if result == WAIT_FAILED: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	372 return args
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	373 _wait_single.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	374 def WaitForSingleObject(handle, timeout):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	375 return _wait_single(handle, ceil(timeout * 1000))
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	376
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	377
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	378 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	379 GetStdHandle
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	380 except NameError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	381 prototype = WINFUNCTYPE(HANDLE, DWORD)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	382 flags = (1, 'which'),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	383 GetStdHandle = prototype(('GetStdHandle', windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	384 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	385 if result == INVALID_HANDLE_VALUE: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	386 return args if result else None
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	387 GetStdHandle.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	388
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	389
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	390 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	391 TerminateProcess
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	392 except NameError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	393 prototype = WINFUNCTYPE(BOOL, Handle, UINT)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	394 flags = (1, 'process'), (1, 'exitcode')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	395 TerminateProcess = prototype(('TerminateProcess', windll.kernel32), flags)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	396 def errcheck(result, func, args):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	397 if not result: raise WinError()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	398 TerminateProcess.errcheck = errcheck
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	399
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	400
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	401 # Do not show error messages due to errors in the program being tested
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	402 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	403 errmode = ctypes.windll.kernel32.GetErrorMode()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	404 except AttributeError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	405 # GetErrorMode is available on Vista/2008 and up
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	406 errmode = ctypes.windll.kernel32.SetErrorMode(0)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	407 ctypes.windll.kernel32.SetErrorMode(errmode \| 0x8003)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	408
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	409 stdin = GetStdHandle(STD_INPUT_HANDLE)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	410 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	411 SetConsoleMode(stdin, ENABLE_PROCESSED_INPUT)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	412 except WindowsError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	413 console_input = False
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	414 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	415 console_input = True
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	416 FlushConsoleInputBuffer(stdin)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	417
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	418 def kill(process):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	419 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	420 process.terminate()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	421 except AttributeError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	422 TerminateProcess(process._handle)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	423 terminate = kill
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	424
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	425 def call(args, *kwargs):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	426 case = kwargs.pop('case')
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	427 job = CreateJobObject(None)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	428 flags = 0
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	429 if case.maxcputime:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	430 flags \|= JOB_OBJECT_LIMIT_PROCESS_TIME
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	431 if case.maxmemory:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	432 flags \|= JOB_OBJECT_LIMIT_PROCESS_MEMORY
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	433 limits = JOBOBJECT_EXTENDED_LIMIT_INFORMATION(
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	434 JOBOBJECT_BASIC_LIMIT_INFORMATION(
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	435 PerProcessUserTimeLimit=ceil((case.maxcputime or 0)*10000000),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	436 LimitFlags=flags,
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	437 ),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	438 ProcessMemoryLimit=ceil((case.maxmemory or 0)*1048576),
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	439 )
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	440 SetInformationJobObject(job, JobObjectExtendedLimitInformation, limits)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	441 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	442 case.process = Popen(args, *kwargs)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	443 except OSError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	444 raise CannotStartTestee(sys.exc_info()[1])
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	445 case.time_started = clock()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	446 AssignProcessToJobObject(job, case.process._handle)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	447 if not console_input:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	448 if case.maxwalltime:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	449 if (WaitForSingleObject(case.process._handle, case.maxwalltime) !=
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	450 WAIT_OBJECT_0):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	451 raise TimeLimitExceeded
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	452 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	453 case.process.wait()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	454 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	455 handles = stdin, case.process._handle
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	456 if case.maxwalltime:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	457 time_end = clock() + case.maxwalltime
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	458 while case.process.poll() is None:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	459 remaining = time_end - clock()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	460 if remaining > 0:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	461 if (WaitForMultipleObjects(handles, False, remaining) ==
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	462 WAIT_OBJECT_0):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	463 ir = ReadConsoleInput(stdin)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	464 if (ir and
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	465 ir.EventType == 1 and
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	466 ir.Event.KeyEvent.bKeyDown and
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	467 ir.Event.KeyEvent.wVirtualKeyCode == 27):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	468 raise CanceledByUser
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	469 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	470 raise TimeLimitExceeded
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	471 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	472 while case.process.poll() is None:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	473 if (WaitForMultipleObjects(handles, False, INFINITE) ==
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	474 WAIT_OBJECT_0):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	475 ir = ReadConsoleInput(stdin)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	476 if (ir and
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	477 ir.EventType == 1 and
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	478 ir.Event.KeyEvent.bKeyDown and
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	479 ir.Event.KeyEvent.wVirtualKeyCode == 27):
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	480 raise CanceledByUser
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	481 case.time_stopped = clock()
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	482 if case.maxcputime and GetProcessTimes:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	483 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	484 times = GetProcessTimes(case.process._handle)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	485 except WindowsError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	486 pass
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	487 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	488 if times.kernel + times.user > case.maxcputime:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	489 raise TimeLimitExceeded
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	490 if case.maxmemory and GetProcessMemoryInfo:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	491 try:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	492 counters = GetProcessMemoryInfo(case.process._handle)
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	493 except WindowsError:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	494 pass
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	495 else:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	496 if counters.PeakPagefileUsage > case.maxmemory * 1048576:
809b77302b21 Win32-specific module with memory and CPU time limits Oleg Oshmyan <chortos@inbox.lv> parents: diff changeset	497 raise MemoryLimitExceeded

Mercurial > ~astiob > upreckon > hgweb

annotate win32.py @ 80:809b77302b21