Upreckon: win32.py annotate

annotate win32.py @ 82:06356af50bf9

Finished testcases reorganization and CPU time limit implementation We now have: * Win32-specific code in the win32 module (including bug fixes), * UNIX-specific and generic code in the unix module, * a much cleaner testcases module, * wait4-based resource limits working on Python 3 (this is a bug fix), * no warning/error reported on non-Win32 when -x is not passed but standard input does not come from a terminal, * the maxtime configuration variable replaced with two new variables named maxcputime and maxwalltime, * CPU time reported if it can be determined unless an error occurs sooner than it is determined (e. g. if the wall-clock time limit is exceeded), * memory limits enforced even if Upreckon's forking already breaks them, * CPU time limits and private virtual memory limits honoured on Win32, * CPU time limits honoured on UNIX(-like) platforms supporting wait4 or getrusage, * address space limits honoured on UNIX(-like) platforms supporting setrlimit with RLIMIT_AS/RLIMIT_VMEM, * resident set size limits honoured on UNIX(-like) platforms supporting wait4.

author	Oleg Oshmyan <chortos@inbox.lv>
date	Wed, 23 Feb 2011 23:35:27 +0000
parents	24752db487c5
children	741ae3391b61

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

Mercurial > ~astiob > upreckon > hgweb

annotate win32.py @ 82:06356af50bf9