Mercurial > ~astiob > upreckon > hgweb
annotate upreckon/unix.py @ 228:715e3525a904 2.03
Fixed crashing on testconfs inside archives on Python 3
author | Oleg Oshmyan <chortos@inbox.lv> |
---|---|
date | Wed, 06 Jun 2012 20:41:44 +0100 |
parents | d5b6708c1955 |
children | 65b5c9390010 |
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 |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
5 from .compat import * |
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
6 from .exceptions 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 |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
11 if sys.platform.startswith('java'): |
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
12 from time import clock |
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
13 else: |
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
14 from time import time as clock |
85
741ae3391b61
Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents:
82
diff
changeset
|
15 |
741ae3391b61
Moved clock/time detection into platform-specific modules and testcases
Oleg Oshmyan <chortos@inbox.lv>
parents:
82
diff
changeset
|
16 try: |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
17 from signal import SIGTERM, SIGKILL |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
18 except ImportError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
19 SIGTERM = 15 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
20 SIGKILL = 9 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
21 |
134
e84f33a60a5c
Moved process killing logic into platform-specific modules
Oleg Oshmyan <chortos@inbox.lv>
parents:
129
diff
changeset
|
22 __all__ = 'call', 'kill', 'pause', 'clock' |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
23 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
24 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
25 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
|
26 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
|
27 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
|
28 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
|
29 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
|
30 from os import O_NONBLOCK |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
31 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
32 import cPickle as pickle |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
33 except ImportError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
34 import pickle |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
35 except ImportError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
36 def call(*args, **kwargs): |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
37 case = kwargs.pop('case') |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
38 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
39 case.process = Popen(*args, **kwargs) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
40 except OSError: |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
41 raise CannotStartTestee(sys.exc_info()[1]) |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
42 case.time_started = clock() |
100
995502cdafc6
Fixed the generic implementation of call to use maxwalltime
Oleg Oshmyan <chortos@inbox.lv>
parents:
94
diff
changeset
|
43 if not case.maxwalltime: |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
44 while True: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
45 exitcode, now = case.process.poll(), clock() |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
46 if exitcode is not None: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
47 case.time_stopped = now |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
48 break |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
49 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
50 time.sleep(.001) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
51 else: |
100
995502cdafc6
Fixed the generic implementation of call to use maxwalltime
Oleg Oshmyan <chortos@inbox.lv>
parents:
94
diff
changeset
|
52 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
|
53 while True: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
54 exitcode, now = case.process.poll(), clock() |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
55 if exitcode is not None: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
56 case.time_stopped = now |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
57 break |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
58 elif now >= time_end: |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
59 raise WallTimeLimitExceeded |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
60 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
61 time.sleep(.001) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
62 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
63 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
64 from fcntl import FD_CLOEXEC |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
65 except ImportError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
66 FD_CLOEXEC = 1 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
67 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
68 try: |
127
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
69 from signal import siginterrupt |
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
70 except ImportError: |
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
71 # Sucks. |
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
72 siginterrupt = lambda signalnum, flag: None |
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
73 |
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
74 try: |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
75 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
|
76 except ImportError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
77 from time import clock as cpuclock |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
78 getrusage = lambda who: None |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
79 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
80 def cpuclock(): |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
81 rusage = getrusage(RUSAGE_SELF) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
82 return rusage.ru_utime + rusage.ru_stime |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
83 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
84 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
85 from resource import setrlimit |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
86 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
87 from resource import RLIMIT_AS |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
88 except ImportError: |
127
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
89 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
|
90 except ImportError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
91 setrlimit = None |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
92 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
93 # 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
|
94 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
|
95 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
|
96 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
|
97 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
|
98 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
|
99 def bury_child(signum, frame): |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
100 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
101 bury_child.case.time_stopped = clock() |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
102 except Exception: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
103 pass |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
104 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
|
105 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
|
106 class SignalIgnorer(object): |
218b8c28549c
Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents:
100
diff
changeset
|
107 def __enter__(self): |
218b8c28549c
Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents:
100
diff
changeset
|
108 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
|
109 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
|
110 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
|
111 signal_ignorer = SignalIgnorer() |
218b8c28549c
Fixed a crash due to SIGCHLD interrupting validator output pipe reads
Oleg Oshmyan <chortos@inbox.lv>
parents:
100
diff
changeset
|
112 __all__ += 'signal_ignorer', |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
113 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
114 # 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
|
115 def call(*args, **kwargs): |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
116 global last_rusage |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
117 bury_child.case = case = kwargs.pop('case') |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
118 read, write = os.pipe() |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
119 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
|
120 def preexec_fn(): |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
121 os.close(read) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
122 if setrlimit and case.maxmemory: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
123 maxmemory = ceil(case.maxmemory * 1048576) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
124 setrlimit(RLIMIT_AS, (maxmemory, maxmemory)) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
125 # 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
|
126 # 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
|
127 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
|
128 kwargs['preexec_fn'] = preexec_fn |
127
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
129 # 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
|
130 # 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
|
131 # 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
|
132 # 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
|
133 # 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
|
134 # 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
|
135 # by the SIGCHLD handler. |
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
136 kwargs['close_fds'] = False |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
137 old_rusage = getrusage(RUSAGE_CHILDREN) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
138 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
|
139 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
|
140 try: |
16fe21d6582e
Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents:
108
diff
changeset
|
141 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
|
142 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
|
143 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
|
144 break |
16fe21d6582e
Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents:
108
diff
changeset
|
145 else: |
16fe21d6582e
Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents:
108
diff
changeset
|
146 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
|
147 siginterrupt(SIGCHLD, False) |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
148 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
149 case.process = Popen(*args, **kwargs) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
150 except OSError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
151 os.close(read) |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
152 raise CannotStartTestee(sys.exc_info()[1]) |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
153 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
|
154 siginterrupt(SIGCHLD, True) |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
155 os.close(write) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
156 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
157 if not catch_escape: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
158 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
|
159 try: |
16fe21d6582e
Fixed a few race conditions in unix.call triggered by very fast testees
Oleg Oshmyan <chortos@inbox.lv>
parents:
108
diff
changeset
|
160 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
|
161 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
|
162 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
|
163 raise |
129
580f0f4687c3
Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents:
128
diff
changeset
|
164 # 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
|
165 try: |
580f0f4687c3
Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents:
128
diff
changeset
|
166 if case.process.poll() is None: |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
167 raise WallTimeLimitExceeded |
129
580f0f4687c3
Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents:
128
diff
changeset
|
168 except OSError: |
580f0f4687c3
Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents:
128
diff
changeset
|
169 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
|
170 raise |
580f0f4687c3
Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents:
128
diff
changeset
|
171 else: |
580f0f4687c3
Fixed EINTR fatally breaking another poll() call on Python 2.6-
Oleg Oshmyan <chortos@inbox.lv>
parents:
128
diff
changeset
|
172 case.process.poll() |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
173 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
|
174 wait(case.process) |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
175 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
176 if not case.maxwalltime: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
177 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
178 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
|
179 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
|
180 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
|
181 sys.stdin.read(1) == '\33'): |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
182 raise 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
|
183 except (SelectError, IOError, OSError): |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
184 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
|
185 raise |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
186 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
187 case.process.poll() |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
188 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
189 time_end = clock() + case.maxwalltime |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
190 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
191 while case.process.poll() is None: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
192 remaining = time_end - clock() |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
193 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
|
194 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
|
195 (), (), remaining) |
127
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
196 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
|
197 sys.stdin.read(1) == '\33'): |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
198 raise CanceledByUser |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
199 else: |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
200 raise 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
|
201 except (SelectError, IOError, OSError): |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
202 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
|
203 raise |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
204 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
205 case.process.poll() |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
206 finally: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
207 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
|
208 os.close(read) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
209 del bury_child.case |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
210 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
|
211 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
|
212 case.time_stopped - case.time_started > case.maxwalltime): |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
213 raise WallTimeLimitExceeded |
123
90c002c960cb
Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents:
119
diff
changeset
|
214 if new_rusage: |
90c002c960cb
Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents:
119
diff
changeset
|
215 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
|
216 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
|
217 # Yes, this actually happens |
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
218 if time_started > time_stopped: |
f5b8a0c0e3cb
Multiple bug fixes in the unix module
Oleg Oshmyan <chortos@inbox.lv>
parents:
126
diff
changeset
|
219 time_started = time_stopped |
123
90c002c960cb
Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents:
119
diff
changeset
|
220 if case.maxcputime or not case.maxwalltime: |
90c002c960cb
Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents:
119
diff
changeset
|
221 case.time_started = time_started |
90c002c960cb
Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents:
119
diff
changeset
|
222 case.time_stopped = time_stopped |
90c002c960cb
Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents:
119
diff
changeset
|
223 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
|
224 if (case.maxcputime and |
90c002c960cb
Fixed CPU time display on UNIX
Oleg Oshmyan <chortos@inbox.lv>
parents:
119
diff
changeset
|
225 time_stopped - time_started > case.maxcputime): |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
226 raise CPUTimeLimitExceeded |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
227 if case.maxmemory: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
228 if sys.platform != 'darwin': |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
229 maxrss = case.maxmemory * 1024 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
230 else: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
231 maxrss = case.maxmemory * 1048576 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
232 if last_rusage and last_rusage.ru_maxrss > maxrss: |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
233 raise MemoryLimitExceeded |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
234 elif (new_rusage and |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
235 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
|
236 new_rusage.ru_maxrss > maxrss): |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
237 raise MemoryLimitExceeded |
82
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
238 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
239 # 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
|
240 # 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
|
241 # 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
|
242 # 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
|
243 # 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
|
244 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
245 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
|
246 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
|
247 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
|
248 return pid, status |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
249 _waitpid = os.waitpid |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
250 os.waitpid = waitpid_emu |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
251 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
252 defaults = Popen._internal_poll.__func__.__defaults__ |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
253 except AttributeError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
254 # Python 2.5 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
255 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
|
256 i = defaults.index(_waitpid) |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
257 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
|
258 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
259 Popen._internal_poll.__func__.__defaults__ = defaults |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
260 except AttributeError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
261 # Python 2.5 again |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
262 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
|
263 except (AttributeError, ValueError): |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
264 pass |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
265 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
266 |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
267 def kill(process): |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
268 try: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
269 process.kill() |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
270 except AttributeError: |
06356af50bf9
Finished testcases reorganization and CPU time limit implementation
Oleg Oshmyan <chortos@inbox.lv>
parents:
diff
changeset
|
271 os.kill(process.pid, SIGKILL) |
134
e84f33a60a5c
Moved process killing logic into platform-specific modules
Oleg Oshmyan <chortos@inbox.lv>
parents:
129
diff
changeset
|
272 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
|
273 |
42c8f5c152a5
Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents:
127
diff
changeset
|
274 |
42c8f5c152a5
Fixed EINTR fatally breaking poll/wait on Python 2.6- (for real this time)
Oleg Oshmyan <chortos@inbox.lv>
parents:
127
diff
changeset
|
275 # 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
|
276 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
|
277 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
|
278 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
|
279 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
|
280 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
|
281 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
|
282 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
|
283 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
|
284 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
|
285 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
|
286 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
|
287 raise |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
288 |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
289 |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
290 try: |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
136
diff
changeset
|
291 from ._unix import * |
136
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
292 except ImportError: |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
293 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
|
294 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
|
295 catch_escape = False |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
296 else: |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
297 try: |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
298 from select import select |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
299 import termios, tty, atexit |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
300 except ImportError: |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
301 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
|
302 catch_escape = False |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
303 else: |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
304 catch_escape = True |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
305 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
|
306 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
|
307 atexit.register(cleanup) |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
308 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
|
309 def pause(): |
ed4035661b85
Added a C implementation of the unix module (called _unix)
Oleg Oshmyan <chortos@inbox.lv>
parents:
134
diff
changeset
|
310 sys.stdin.read(1) |