annotate unix.py @ 142:98bccf81db4d

win32.call() now trusts its own case.time_started
author Oleg Oshmyan <chortos@inbox.lv>
date Wed, 25 May 2011 00:27:12 +0100
parents ed4035661b85
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
1 # Copyright (c) 2010-2011 Chortos-2 <chortos@inbox.lv>
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
2
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
3 from __future__ import division, with_statement
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
4
91
c62c9bfd614a Removed import_error
Oleg Oshmyan <chortos@inbox.lv>
parents: 86
diff changeset
5 from compat import *
c62c9bfd614a Removed import_error
Oleg Oshmyan <chortos@inbox.lv>
parents: 86
diff changeset
6 import testcases # mutual import
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
7
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
8 from subprocess import Popen
86
8cd7a732f2f3 Fixed a crash in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 85
diff changeset
9 import os, sys, time
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
10
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
11 try:
85
741ae3391b61 Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents: 82
diff changeset
12 from testcases import clock
741ae3391b61 Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents: 82
diff changeset
13 except ImportError:
741ae3391b61 Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents: 82
diff changeset
14 if sys.platform.startswith('java'):
741ae3391b61 Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents: 82
diff changeset
15 from time import clock
741ae3391b61 Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents: 82
diff changeset
16 else:
741ae3391b61 Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents: 82
diff changeset
17 from time import time as clock
741ae3391b61 Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents: 82
diff changeset
18
741ae3391b61 Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents: 82
diff changeset
19 try:
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
20 from signal import SIGTERM, SIGKILL
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
21 except ImportError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
22 SIGTERM = 15
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
23 SIGKILL = 9
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
24
134
e84f33a60a5c Moved process killing logic into platform-specific modules
Oleg Oshmyan <chortos@inbox.lv>
parents: 129
diff changeset
25 __all__ = 'call', 'kill', 'pause', 'clock'
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
26
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
27
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
28 try:
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
29 from signal import SIGCHLD, SIG_DFL, signal, set_wakeup_fd
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
30 from select import select, error as SelectError
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
31 from errno import EAGAIN, EINTR
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
32 from fcntl import fcntl, F_SETFD, F_GETFD, F_SETFL, F_GETFL
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
33 from os import O_NONBLOCK
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
34 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
35 import cPickle as pickle
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
36 except ImportError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
37 import pickle
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
38 except ImportError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
39 def call(*args, **kwargs):
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
40 case = kwargs.pop('case')
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
41 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
42 case.process = Popen(*args, **kwargs)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
43 except OSError:
94
b7fb64ce03d9 Fixed a couple of bugs
Oleg Oshmyan <chortos@inbox.lv>
parents: 91
diff changeset
44 raise testcases.CannotStartTestee(sys.exc_info()[1])
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
45 case.time_started = clock()
100
995502cdafc6 Fixed the generic implementation of call to use maxwalltime
Oleg Oshmyan <chortos@inbox.lv>
parents: 94
diff changeset
46 if not case.maxwalltime:
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
47 while True:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
48 exitcode, now = case.process.poll(), clock()
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
49 if exitcode is not None:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
50 case.time_stopped = now
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
51 break
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
52 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
53 time.sleep(.001)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
54 else:
100
995502cdafc6 Fixed the generic implementation of call to use maxwalltime
Oleg Oshmyan <chortos@inbox.lv>
parents: 94
diff changeset
55 time_end = case.time_started + case.maxwalltime
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
56 while True:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
57 exitcode, now = case.process.poll(), clock()
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
58 if exitcode is not None:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
59 case.time_stopped = now
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
60 break
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
61 elif now >= time_end:
126
fcdcd0c95552 Fixed the generic implementation of call to use testcases.WallTimeLimitExceeded
Oleg Oshmyan <chortos@inbox.lv>
parents: 124
diff changeset
62 raise testcases.WallTimeLimitExceeded
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
63 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
64 time.sleep(.001)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
65 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
66 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
67 from fcntl import FD_CLOEXEC
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
68 except ImportError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
69 FD_CLOEXEC = 1
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
70
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
71 try:
127
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
72 from signal import siginterrupt
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
73 except ImportError:
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
74 # Sucks.
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
75 siginterrupt = lambda signalnum, flag: None
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
76
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
77 try:
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
78 from resource import getrusage, RUSAGE_SELF, RUSAGE_CHILDREN
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
79 except ImportError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
80 from time import clock as cpuclock
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
81 getrusage = lambda who: None
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
82 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
83 def cpuclock():
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
84 rusage = getrusage(RUSAGE_SELF)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
85 return rusage.ru_utime + rusage.ru_stime
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
86
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
87 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
88 from resource import setrlimit
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
89 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
90 from resource import RLIMIT_AS
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
91 except ImportError:
127
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
92 from resource import RLIMIT_VMEM as RLIMIT_AS
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
93 except ImportError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
94 setrlimit = None
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
95
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
96 # Make SIGCHLD interrupt sleep() and select()
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
97 sigchld_pipe_read, sigchld_pipe_write = os.pipe()
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
98 fcntl(sigchld_pipe_read, F_SETFL,
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
99 fcntl(sigchld_pipe_read, F_GETFL) | O_NONBLOCK)
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
100 fcntl(sigchld_pipe_write, F_SETFL,
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
101 fcntl(sigchld_pipe_write, F_GETFL) | O_NONBLOCK)
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
102 def bury_child(signum, frame):
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
103 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
104 bury_child.case.time_stopped = clock()
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
105 except Exception:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
106 pass
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
107 signal(SIGCHLD, bury_child)
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
108 set_wakeup_fd(sigchld_pipe_write)
108
218b8c28549c Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents: 100
diff changeset
109 class SignalIgnorer(object):
218b8c28549c Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents: 100
diff changeset
110 def __enter__(self):
218b8c28549c Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents: 100
diff changeset
111 signal(SIGCHLD, SIG_DFL)
218b8c28549c Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents: 100
diff changeset
112 def __exit__(self, exc_type, exc_value, traceback):
218b8c28549c Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents: 100
diff changeset
113 signal(SIGCHLD, bury_child)
218b8c28549c Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents: 100
diff changeset
114 signal_ignorer = SignalIgnorer()
218b8c28549c Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents: 100
diff changeset
115 __all__ += 'signal_ignorer',
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
116
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
117 # If you want this to work portably, don't set any stdio argument to PIPE
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
118 def call(*args, **kwargs):
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
119 global last_rusage
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
120 bury_child.case = case = kwargs.pop('case')
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
121 read, write = os.pipe()
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
122 fcntl(write, F_SETFD, fcntl(write, F_GETFD) | FD_CLOEXEC)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
123 def preexec_fn():
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
124 os.close(read)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
125 if setrlimit and case.maxmemory:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
126 maxmemory = ceil(case.maxmemory * 1048576)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
127 setrlimit(RLIMIT_AS, (maxmemory, maxmemory))
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
128 # I would also set a CPU time limit but I do not want the time
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
129 # passing between the calls to fork and exec to be counted in
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
130 os.write(write, pickle.dumps((clock(), cpuclock()), 1))
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
131 kwargs['preexec_fn'] = preexec_fn
127
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
132 # So how the hell do I actually make use of pass_fds?
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
133 # On 3.1-, calling Popen with pass_fds prints an exception
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
134 # from Popen.__del__ to stderr. On 3.2, Popen without close_fds
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
135 # or pass_fds creates a child and fails but that of course
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
136 # generates a SIGCHLD, which causes problems, and I have
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
137 # no process ID to wait upon to negate the changes made
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
138 # by the SIGCHLD handler.
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
139 kwargs['close_fds'] = False
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
140 old_rusage = getrusage(RUSAGE_CHILDREN)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
141 last_rusage = None
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
142 while True:
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
143 try:
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
144 os.read(sigchld_pipe_read, 512)
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
145 except OSError:
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
146 if sys.exc_info()[1].errno == EAGAIN:
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
147 break
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
148 else:
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
149 raise
124
19c42a3cd962 Very fast testees no longer break Popen on some UNIX systems on Python 2.6+
Oleg Oshmyan <chortos@inbox.lv>
parents: 123
diff changeset
150 siginterrupt(SIGCHLD, False)
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
151 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
152 case.process = Popen(*args, **kwargs)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
153 except OSError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
154 os.close(read)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
155 raise testcases.CannotStartTestee(sys.exc_info()[1])
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
156 finally:
124
19c42a3cd962 Very fast testees no longer break Popen on some UNIX systems on Python 2.6+
Oleg Oshmyan <chortos@inbox.lv>
parents: 123
diff changeset
157 siginterrupt(SIGCHLD, True)
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
158 os.close(write)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
159 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
160 if not catch_escape:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
161 if case.maxwalltime:
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
162 try:
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
163 select((sigchld_pipe_read,), (), (), case.maxwalltime)
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
164 except SelectError:
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
165 if sys.exc_info()[1].args[0] != EINTR:
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
166 raise
129
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
167 # subprocess in Python 2.6- is not guarded against EINTR
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
168 try:
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
169 if case.process.poll() is None:
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
170 raise testcases.WallTimeLimitExceeded
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
171 except OSError:
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
172 if sys.exc_info()[1].errno != EINTR:
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
173 raise
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
174 else:
580f0f4687c3 Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents: 128
diff changeset
175 case.process.poll()
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
176 else:
128
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
177 wait(case.process)
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
178 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
179 if not case.maxwalltime:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
180 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
181 while case.process.poll() is None:
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
182 s = select((sys.stdin, sigchld_pipe_read), (), ())
127
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
183 if (s[0] == [sys.stdin] and
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
184 sys.stdin.read(1) == '\33'):
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
185 raise testcases.CanceledByUser
128
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
186 except (SelectError, IOError, OSError):
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
187 if sys.exc_info()[1].args[0] != EINTR:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
188 raise
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
189 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
190 case.process.poll()
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
191 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
192 time_end = clock() + case.maxwalltime
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
193 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
194 while case.process.poll() is None:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
195 remaining = time_end - clock()
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
196 if remaining > 0:
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
197 s = select((sys.stdin, sigchld_pipe_read),
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
198 (), (), remaining)
127
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
199 if (s[0] == [sys.stdin] and
118
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
200 sys.stdin.read(1) == '\33'):
16fe21d6582e Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents: 108
diff changeset
201 raise testcases.CanceledByUser
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
202 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
203 raise testcases.WallTimeLimitExceeded
128
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
204 except (SelectError, IOError, OSError):
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
205 if sys.exc_info()[1].args[0] != EINTR:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
206 raise
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
207 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
208 case.process.poll()
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
209 finally:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
210 case.time_started, cpustart = pickle.loads(os.read(read, 512))
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
211 os.close(read)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
212 del bury_child.case
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
213 new_rusage = getrusage(RUSAGE_CHILDREN)
119
0b265fe9c81f Wall-clock time limit is now enforced on the output wall-clock time
Oleg Oshmyan <chortos@inbox.lv>
parents: 118
diff changeset
214 if (case.maxwalltime and
0b265fe9c81f Wall-clock time limit is now enforced on the output wall-clock time
Oleg Oshmyan <chortos@inbox.lv>
parents: 118
diff changeset
215 case.time_stopped - case.time_started > case.maxwalltime):
0b265fe9c81f Wall-clock time limit is now enforced on the output wall-clock time
Oleg Oshmyan <chortos@inbox.lv>
parents: 118
diff changeset
216 raise testcases.WallTimeLimitExceeded
123
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
217 if new_rusage:
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
218 time_started = old_rusage.ru_utime + old_rusage.ru_stime + cpustart
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
219 time_stopped = new_rusage.ru_utime + new_rusage.ru_stime
127
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
220 # Yes, this actually happens
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
221 if time_started > time_stopped:
f5b8a0c0e3cb Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents: 126
diff changeset
222 time_started = time_stopped
123
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
223 if case.maxcputime or not case.maxwalltime:
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
224 case.time_started = time_started
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
225 case.time_stopped = time_stopped
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
226 case.time_limit_string = case.cpu_time_limit_string
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
227 if (case.maxcputime and
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
228 time_stopped - time_started > case.maxcputime):
90c002c960cb Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents: 119
diff changeset
229 raise testcases.CPUTimeLimitExceeded
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
230 if case.maxmemory:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
231 if sys.platform != 'darwin':
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
232 maxrss = case.maxmemory * 1024
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
233 else:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
234 maxrss = case.maxmemory * 1048576
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
235 if last_rusage and last_rusage.ru_maxrss > maxrss:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
236 raise testcases.MemoryLimitExceeded
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
237 elif (new_rusage and
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
238 new_rusage.ru_maxrss > old_rusage.ru_maxrss and
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
239 new_rusage.ru_maxrss > maxrss):
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
240 raise testcases.MemoryLimitExceeded
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
241
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
242 # Emulate memory limits on platforms compatible with 4.3BSD but not XSI
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
243 # I say 'emulate' because the OS will allow excessive memory usage
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
244 # anyway; Upreckon will just treat the test case as not passed.
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
245 # To do this, we not only require os.wait4 to be present but also
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
246 # assume things about the implementation of subprocess.Popen.
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
247 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
248 def waitpid_emu(pid, options, _wait4=os.wait4):
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
249 global last_rusage
128
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
250 pid, status, last_rusage = _wait4(pid, options)
82
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
251 return pid, status
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
252 _waitpid = os.waitpid
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
253 os.waitpid = waitpid_emu
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
254 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
255 defaults = Popen._internal_poll.__func__.__defaults__
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
256 except AttributeError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
257 # Python 2.5
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
258 defaults = Popen._internal_poll.im_func.func_defaults
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
259 i = defaults.index(_waitpid)
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
260 defaults = defaults[:i] + (waitpid_emu,) + defaults[i+1:]
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
261 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
262 Popen._internal_poll.__func__.__defaults__ = defaults
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
263 except AttributeError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
264 # Python 2.5 again
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
265 Popen._internal_poll.im_func.func_defaults = defaults
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
266 except (AttributeError, ValueError):
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
267 pass
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
268
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
269
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
270 def kill(process):
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
271 try:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
272 process.kill()
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
273 except AttributeError:
06356af50bf9 Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff changeset
274 os.kill(process.pid, SIGKILL)
134
e84f33a60a5c Moved process killing logic into platform-specific modules
Oleg Oshmyan <chortos@inbox.lv>
parents: 129
diff changeset
275 wait(process)
128
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
276
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
277
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
278 # subprocess in Python 2.6- is not guarded against EINTR
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
279 try:
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
280 from errno import EINTR
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
281 except ImportError:
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
282 wait = Popen.wait
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
283 else:
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
284 def wait(process):
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
285 while True:
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
286 try:
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
287 return process.wait()
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
288 except OSError:
42c8f5c152a5 Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents: 127
diff changeset
289 if sys.exc_info()[1].errno != EINTR:
136
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
290 raise
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
291
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
292
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
293 try:
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
294 from _unix import *
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
295 except ImportError:
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
296 if not sys.stdin.isatty():
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
297 pause = lambda: sys.stdin.read(1)
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
298 catch_escape = False
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
299 else:
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
300 try:
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
301 from select import select
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
302 import termios, tty, atexit
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
303 except ImportError:
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
304 pause = lambda: sys.stdin.read(1)
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
305 catch_escape = False
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
306 else:
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
307 catch_escape = True
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
308 def cleanup(old=termios.tcgetattr(sys.stdin.fileno())):
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
309 termios.tcsetattr(sys.stdin.fileno(), termios.TCSAFLUSH, old)
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
310 atexit.register(cleanup)
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
311 tty.setcbreak(sys.stdin.fileno())
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
312 def pause():
ed4035661b85 Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents: 134
diff changeset
313 sys.stdin.read(1)