Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
G
gevent
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
Analytics
Analytics
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Commits
Issue Boards
Open sidebar
Kirill Smelkov
gevent
Commits
b7ba2f41
Commit
b7ba2f41
authored
Dec 14, 2021
by
Jason Madden
Committed by
GitHub
Dec 14, 2021
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #1848 from gevent/issue1798
Add pypy3.7 to Appveyor.
parents
9338965a
42fdaecc
Changes
15
Hide whitespace changes
Inline
Side-by-side
Showing
15 changed files
with
6582 additions
and
28 deletions
+6582
-28
appveyor.yml
appveyor.yml
+16
-3
docs/changes/1798.feature
docs/changes/1798.feature
+4
-0
docs/install.rst
docs/install.rst
+5
-3
src/gevent/testing/patched_tests_setup.py
src/gevent/testing/patched_tests_setup.py
+37
-1
src/gevent/testing/testrunner.py
src/gevent/testing/testrunner.py
+31
-7
src/gevent/tests/known_failures.py
src/gevent/tests/known_failures.py
+40
-8
src/gevent/tests/test__issues461_471.py
src/gevent/tests/test__issues461_471.py
+7
-4
src/gevent/tests/test__server.py
src/gevent/tests/test__server.py
+5
-0
src/gevent/tests/test__threading_vs_settrace.py
src/gevent/tests/test__threading_vs_settrace.py
+10
-2
src/greentest/3.10/signalinterproctester.py
src/greentest/3.10/signalinterproctester.py
+84
-0
src/greentest/3.10/test_signal.py
src/greentest/3.10/test_signal.py
+1369
-0
src/greentest/3.6/test_signal.py
src/greentest/3.6/test_signal.py
+1092
-0
src/greentest/3.7/test_signal.py
src/greentest/3.7/test_signal.py
+1181
-0
src/greentest/3.8/test_signal.py
src/greentest/3.8/test_signal.py
+1340
-0
src/greentest/3.9/test_signal.py
src/greentest/3.9/test_signal.py
+1361
-0
No files found.
appveyor.yml
View file @
b7ba2f41
...
...
@@ -41,6 +41,13 @@ environment:
# a later point release.
# 64-bit
-
PYTHON
:
"
C:
\\
pypy3.7-v7.3.7-win64"
PYTHON_ID
:
"
pypy3"
PYTHON_EXE
:
pypy3w
PYTHON_VERSION
:
"
3.7.x"
PYTHON_ARCH
:
"
64"
APPVEYOR_BUILD_WORKER_IMAGE
:
Visual Studio
2019
-
PYTHON
:
"
C:
\\
pypy2.7-v7.3.6-win64"
PYTHON_ID
:
"
pypy"
PYTHON_EXE
:
pypy
...
...
@@ -168,6 +175,12 @@ install:
}
7z x -y "${env:PYTMP}\pypy2-v7.3.6-win64.zip" -oC:\ | Out-Null;
}
elseif ("${env:PYTHON_ID}" -eq "pypy3") {
if (!(Test-Path "${env:PYTMP}\pypy3.7-v7.3.7-win64.zip")) {
(New-Object Net.WebClient).DownloadFile("https://downloads.python.org/pypy/pypy3.7-v7.3.7-win64.zip", "${env:PYTMP}\pypy3.7-v7.3.7-win64.zip");
}
7z x -y "${env:PYTMP}\pypy3.7-v7.3.7-win64.zip" -oC:\ | Out-Null;
}
elseif (-not(Test-Path($env:PYTHON))) {
& appveyor\install.ps1;
}
...
...
@@ -198,9 +211,8 @@ cache:
build_script
:
# Build the compiled extension
# Try to get some things that don't wind up in the pip cache as
# built wheels if they're built during an isolated build.
-
"
%CMD_IN_ENV%
%PYEXE%
-m
pip
install
-U
wheel
cython
setuptools
cffi"
-
"
%CMD_IN_ENV%
%PYEXE%
-m
pip
install
-U
wheel"
-
"
%CMD_IN_ENV%
%PYEXE%
-m
pip
install
-U
setuptools"
-
if not "%GWHEEL_ONLY%"=="true" %PYEXE% -m pip install -U -e .[test]
test_script
:
...
...
@@ -209,6 +221,7 @@ test_script:
-
if not "%GWHEEL_ONLY%"=="true" %PYEXE% -c "import gevent.core; print(gevent.core.loop)"
-
if not "%GWHEEL_ONLY%"=="true" %PYEXE% -c "import gevent; print(gevent.config.settings['resolver'].get_options())"
-
if not "%GWHEEL_ONLY%"=="true" %PYEXE% -c "from gevent._compat import get_clock_info; print(get_clock_info('perf_counter'))"
-
if not "%GWHEEL_ONLY%"=="true" %PYEXE% -mgevent.tests.known_failures
-
if not "%GWHEEL_ONLY%"=="true" %PYEXE% -mgevent.tests --second-chance --config known_failures.py
after_test
:
...
...
docs/changes/1798.feature
0 → 100644
View file @
b7ba2f41
Windows
:
Test
and
provide
binary
wheels
for
PyPy3.7.
Note
that
there
may
be
issues
with
subprocesses,
signals,
and
it
may
be
slow.
docs/install.rst
View file @
b7ba2f41
...
...
@@ -26,8 +26,8 @@ library and will install the `cffi`_ library by default on Windows.
The cffi library will become the default on all platforms in a future
release of gevent.
This version of gevent also runs on PyPy 7.
0 or above. On PyPy, there
are no external dependencies.
This version of gevent also runs on PyPy 7.
3.7 (7.3.6 for PyPy2) or
a
bove. On PyPy, there a
re no external dependencies.
gevent is tested on Windows, macOS, and Linux, and should run on most
other Unix-like operating systems (e.g., FreeBSD, Solaris, etc.)
...
...
@@ -35,7 +35,9 @@ other Unix-like operating systems (e.g., FreeBSD, Solaris, etc.)
.. note::
Windows is supported as a tier 2, "best effort," platform. It is
suitable for development, but not recommended for production.
suitable for development, but not recommended for production. In
particular, PyPy3 on Windows may have issues, especially with
subprocesses.
On Windows using the deprecated libev backend, gevent is
limited to a maximum of 1024 open sockets due to
...
...
src/gevent/testing/patched_tests_setup.py
View file @
b7ba2f41
...
...
@@ -180,9 +180,15 @@ disabled_tests = [
# (unless signal handler raises an error) maybe it should?
'test_signal.WakeupSignalTests.test_wakeup_fd_during'
,
# these rely on os.read raising EINTR which never happens with gevent.os.read
'test_signal.SiginterruptTest.test_without_siginterrupt'
,
'test_signal.SiginterruptTest.test_siginterrupt_on'
,
# these rely on os.read raising EINTR which never happens with gevent.os.read
'test_signal.SiginterruptTest.test_siginterrupt_off'
,
# This one takes forever and relies on threading details
'test_signal.StressTest.test_stress_modifying_handlers'
,
# This uses an external file, and launches it. This means that it's not
# actually testing gevent because there's no monkey-patch.
'test_signal.PosixTests.test_interprocess_signal'
,
'test_subprocess.ProcessTestCase.test_leak_fast_process_del_killed'
,
'test_subprocess.ProcessTestCase.test_zombie_fast_process_del'
,
...
...
@@ -680,6 +686,34 @@ if PYPY3 and TRAVIS:
'test_socket.InheritanceTest.test_SOCK_CLOEXEC'
,
]
if
PYPY3
and
WIN
:
disabled_tests
+=
[
# test_httpservers.CGIHTTPServerTestCase all seem to hang.
# There seem to be some general subprocess issues. This is
# ignored entirely from known_failures.py
# This produces:
#
# OSError: [Errno 10014] The system detected an invalid
# pointer address in attempting to use a pointer argument in
# a call
#
# When calling socket.socket(fileno=fd) when we actually
# call ``self._socket =self._gevent_sock_class()``.
'test_socket.GeneralModuleTests.test_socket_fileno'
,
# This doesn't respect the scope properly
#
# self.assertEqual(sockaddr, ('ff02::1de:c0:face:8d', 1234, 0, ifindex))
# AssertionError: Tuples differ: ('ff02::1de:c0:face:8d%42', 1234, 0, 42) != ('ff02::1de:c0:face:8d', 1234, 0, 42
#
'test_socket.GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_numeric'
,
# self.assertEqual(newsock.get_inheritable(), False)
# AssertionError: True != False
'test_socket.InheritanceTest.test_dup'
,
]
def
_make_run_with_original
(
mod_name
,
func_name
):
@
contextlib
.
contextmanager
def
with_orig
():
...
...
@@ -1412,6 +1446,8 @@ if PY310:
# We don't currently implement pipesize.
'test_subprocess.ProcessTestCase.test_pipesize_default'
,
'test_subprocess.ProcessTestCase.test_pipesizes'
,
# Unknown
'test_signal.SiginterruptTest.test_siginterrupt_off'
,
]
if
TRAVIS
:
...
...
src/gevent/testing/testrunner.py
View file @
b7ba2f41
...
...
@@ -808,15 +808,39 @@ def main():
import
argparse
parser
=
argparse
.
ArgumentParser
()
parser
.
add_argument
(
'--ignore'
)
parser
.
add_argument
(
'--discover'
,
action
=
'store_true'
)
parser
.
add_argument
(
'--full'
,
action
=
'store_true'
)
parser
.
add_argument
(
'--config'
,
default
=
'known_failures.py'
)
parser
.
add_argument
(
"--coverage"
,
action
=
"store_true"
)
parser
.
add_argument
(
"--quiet"
,
action
=
"store_true"
,
default
=
True
)
parser
.
add_argument
(
'--discover'
,
action
=
'store_true'
,
help
=
"Only print the tests found."
)
parser
.
add_argument
(
'--config'
,
default
=
'known_failures.py'
,
help
=
"The path to the config file containing "
"FAILING_TESTS, IGNORED_TESTS and RUN_ALONE. "
"Defaults to %(default)s."
)
parser
.
add_argument
(
"--coverage"
,
action
=
"store_true"
,
help
=
"Enable coverage recording with coverage.py."
)
# TODO: Quiet and verbose should be mutually exclusive
parser
.
add_argument
(
"--quiet"
,
action
=
"store_true"
,
default
=
True
,
help
=
"Be quiet. Defaults to %(default)s. Also the "
"GEVENTTEST_QUIET environment variable."
)
parser
.
add_argument
(
"--verbose"
,
action
=
"store_false"
,
dest
=
'quiet'
)
parser
.
add_argument
(
"--debug"
,
action
=
"store_true"
,
default
=
False
)
parser
.
add_argument
(
"--package"
,
default
=
"gevent.tests"
)
parser
.
add_argument
(
"--debug"
,
action
=
"store_true"
,
default
=
False
,
help
=
"Enable debug settings. If the GEVENT_DEBUG environment variable is not set, "
"this sets it to 'debug'. This can also enable PYTHONTRACEMALLOC and the debug PYTHONMALLOC "
"allocators, if not already set. Defaults to %(default)s."
)
parser
.
add_argument
(
"--package"
,
default
=
"gevent.tests"
,
help
=
"Load tests from the given package. Defaults to %(default)s."
)
parser
.
add_argument
(
"--processes"
,
"-j"
,
default
=
DEFAULT_NWORKERS
,
type
=
int
,
help
=
"Use up to the given number of parallel processes to execute tests. "
...
...
src/gevent/tests/known_failures.py
View file @
b7ba2f41
...
...
@@ -80,6 +80,7 @@ class _AttrCondition(ConstantCondition):
ConstantCondition
.
__init__
(
self
,
getattr
(
sysinfo
,
name
),
name
)
PYPY
=
_AttrCondition
(
'PYPY'
)
PYPY3
=
_AttrCondition
(
'PYPY3'
)
PY3
=
_AttrCondition
(
'PY3'
)
PY2
=
_AttrCondition
(
'PY2'
)
OSX
=
_AttrCondition
(
'OSX'
)
...
...
@@ -160,12 +161,18 @@ class Multi(object):
def
__init__
(
self
):
self
.
_conds
=
[]
def
flaky
(
self
,
reason
=
''
,
when
=
True
):
self
.
_conds
.
append
(
Flaky
(
reason
,
when
))
def
flaky
(
self
,
reason
=
''
,
when
=
True
,
ignore_coverage
=
NEVER
,
run_alone
=
NEVER
):
self
.
_conds
.
append
(
Flaky
(
reason
,
when
=
when
,
ignore_coverage
=
ignore_coverage
,
run_alone
=
run_alone
,
)
)
return
self
def
ignored
(
self
,
reason
=
''
,
when
=
True
):
self
.
_conds
.
append
(
Ignored
(
reason
,
when
))
self
.
_conds
.
append
(
Ignored
(
reason
,
when
=
when
))
return
self
def
__set_name__
(
self
,
owner
,
name
):
...
...
@@ -279,13 +286,19 @@ class Definitions(DefinitionsBase):
test__backdoor
=
Flaky
(
when
=
LEAKTEST
|
PYPY
)
test__socket_errors
=
Flaky
(
when
=
LEAKTEST
)
test_signal
=
F
laky
(
test_signal
=
Multi
().
f
laky
(
"On Travis, this very frequently fails due to timing"
,
when
=
TRAVIS
&
LEAKTEST
,
# Partial workaround for the _testcapi issue on PyPy,
# but also because signal delivery can sometimes be slow, and this
# spawn processes of its own
run_alone
=
APPVEYOR
,
).
ignored
(
"""
This fails to run a single test. It looks like just importing the module
can hang. All I see is the output from patch_all()
"""
,
when
=
APPVEYOR
&
PYPY3
)
test__monkey_sigchld_2
=
Ignored
(
...
...
@@ -303,10 +316,22 @@ class Definitions(DefinitionsBase):
allocate SSL Context objects, either in Python 2.7 or 3.6.
There must be some library incompatibility. No point even
running them. XXX: Remember to turn this back on.
On Windows, with PyPy3.7 7.3.7, there seem to be all kind of certificate
errors.
"""
,
when
=
(
PYPY
&
TRAVIS
)
|
(
PYPY3
&
WIN
)
)
test_httpservers
=
Ignored
(
"""
All the CGI tests hang. There appear to be subprocess problems.
"""
,
when
=
PYPY
&
TRAVIS
when
=
PYPY
3
&
WIN
)
test__pywsgi
=
Ignored
(
"""
XXX: Re-enable this when we can investigate more. This has
...
...
@@ -319,16 +344,23 @@ class Definitions(DefinitionsBase):
On Appveyor 3.8.0, for some reason this takes *way* too long, about 100s, which
often goes just over the default timeout of 100s. This makes no sense.
But it also takes nearly that long in 3.7. 3.6 and earlier are much faster.
It also takes just over 100s on PyPy 3.7.
"""
,
when
=
(
PYPY
&
TRAVIS
&
LIBUV
)
|
PY380_EXACTLY
,
# https://bitbucket.org/pypy/pypy/issues/2769/systemerror-unexpected-internal-exception
run_alone
=
(
CI
&
LEAKTEST
&
PY3
)
|
(
PYPY
&
LIBUV
),
# This often takes much longer on PyPy on CI.
options
=
{
'timeout'
:
(
CI
&
PYPY
,
180
)},
)
test_subprocess
=
F
laky
(
test_subprocess
=
Multi
().
f
laky
(
"Unknown, can't reproduce locally; times out one test"
,
when
=
PYPY
&
PY3
&
TRAVIS
,
ignore_coverage
=
ALWAYS
,
).
ignored
(
"Tests don't even start before the process times out."
,
when
=
PYPY3
&
WIN
)
test__threadpool
=
Ignored
(
...
...
@@ -388,7 +420,7 @@ class Definitions(DefinitionsBase):
off.
"""
,
when
=
COVERAGE
,
ignore_coverage
=
ALWAYS
ignore_coverage
=
ALWAYS
,
)
test__hub_join_timeout
=
Ignored
(
...
...
@@ -417,7 +449,7 @@ class Definitions(DefinitionsBase):
"""
On a heavily loaded box, these can all take upwards of 200s.
"""
,
when
=
CI
&
LEAKTEST
when
=
(
CI
&
LEAKTEST
)
|
(
PYPY3
&
APPVEYOR
)
)
test_socket
=
RunAlone
(
...
...
src/gevent/tests/test__issues461_471.py
View file @
b7ba2f41
...
...
@@ -33,13 +33,16 @@ else:
from
gevent.testing.sysinfo
import
CFFI_BACKEND
from
gevent.testing.sysinfo
import
RUN_COVERAGE
from
gevent.testing.sysinfo
import
WIN
from
gevent.testing.sysinfo
import
PYPY3
class
Test
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
CFFI_BACKEND
and
RUN_COVERAGE
,
"Interferes with the timing"
)
@
unittest
.
skipIf
(
(
CFFI_BACKEND
and
RUN_COVERAGE
)
or
(
PYPY3
and
WIN
),
"Interferes with the timing; times out waiting for the child"
)
def
test_hang
(
self
):
# XXX: Why does PyPy3 on Win fail to kill the child? (This was before we switched
# to pypy3w; perhaps that makes a difference?)
if
WIN
:
from
subprocess
import
CREATE_NEW_PROCESS_GROUP
kwargs
=
{
'creationflags'
:
CREATE_NEW_PROCESS_GROUP
}
...
...
@@ -63,7 +66,7 @@ else:
p
.
send_signal
(
signal_to_send
)
# Wait a few seconds for child process to die. Sometimes signal delivery is delayed
# or even swallowed by Python, so send the signal a few more times if necessary
wait_seconds
=
1
5.0
wait_seconds
=
2
5.0
now
=
time
.
time
()
midtime
=
now
+
(
wait_seconds
/
2.0
)
endtime
=
time
.
time
()
+
wait_seconds
...
...
src/gevent/tests/test__server.py
View file @
b7ba2f41
...
...
@@ -158,6 +158,11 @@ class TestCase(greentest.TestCase):
if
greentest
.
OSX
:
# A kernel bug in OS X sometimes results in this
LOCAL_CONN_REFUSED_ERRORS
=
(
errno
.
EPROTOTYPE
,)
elif
greentest
.
WIN
and
greentest
.
PYPY3
:
# We see WinError 10049: The requested address is not valid
# which is not one of the errors we get anywhere else.
# Not sure which errno constant this is?
LOCAL_CONN_REFUSED_ERRORS
=
(
10049
,)
def
assertConnectionRefused
(
self
,
in_proc_server
=
True
):
try
:
...
...
src/gevent/tests/test__threading_vs_settrace.py
View file @
b7ba2f41
...
...
@@ -14,7 +14,7 @@ import sys, os, threading, time
# A deadlock-killer, to prevent the
# testsuite to hang forever
def killer():
time.sleep(0.
1
)
time.sleep(0.
2
)
sys.stdout.write('..program blocked; aborting!')
sys.stdout.flush()
os._exit(2)
...
...
@@ -137,7 +137,15 @@ class TestTrace(unittest.TestCase):
self
.
assertTrue
(
isinstance
(
e
,
LoopExit
))
def
run_script
(
self
,
more_args
=
()):
args
=
[
sys
.
executable
,
"-c"
,
script
]
if
(
greentest
.
PYPY3
and
greentest
.
RUNNING_ON_APPVEYOR
and
sys
.
version_info
[:
2
]
==
(
3
,
7
)
):
# Somehow launching the subprocess fails with exit code 1, and
# produces no output. It's not clear why.
self
.
skipTest
(
"Known to hang on AppVeyor"
)
args
=
[
sys
.
executable
,
"-u"
,
"-c"
,
script
]
args
.
extend
(
more_args
)
rc
=
subprocess
.
call
(
args
)
self
.
assertNotEqual
(
rc
,
2
,
"interpreter was blocked"
)
...
...
src/greentest/3.10/signalinterproctester.py
0 → 100644
View file @
b7ba2f41
import
os
import
signal
import
subprocess
import
sys
import
time
import
unittest
from
test
import
support
class
SIGUSR1Exception
(
Exception
):
pass
class
InterProcessSignalTests
(
unittest
.
TestCase
):
def
setUp
(
self
):
self
.
got_signals
=
{
'SIGHUP'
:
0
,
'SIGUSR1'
:
0
,
'SIGALRM'
:
0
}
def
sighup_handler
(
self
,
signum
,
frame
):
self
.
got_signals
[
'SIGHUP'
]
+=
1
def
sigusr1_handler
(
self
,
signum
,
frame
):
self
.
got_signals
[
'SIGUSR1'
]
+=
1
raise
SIGUSR1Exception
def
wait_signal
(
self
,
child
,
signame
):
if
child
is
not
None
:
# This wait should be interrupted by exc_class
# (if set)
child
.
wait
()
timeout
=
support
.
SHORT_TIMEOUT
deadline
=
time
.
monotonic
()
+
timeout
while
time
.
monotonic
()
<
deadline
:
if
self
.
got_signals
[
signame
]:
return
signal
.
pause
()
self
.
fail
(
'signal %s not received after %s seconds'
%
(
signame
,
timeout
))
def
subprocess_send_signal
(
self
,
pid
,
signame
):
code
=
'import os, signal; os.kill(%s, signal.%s)'
%
(
pid
,
signame
)
args
=
[
sys
.
executable
,
'-I'
,
'-c'
,
code
]
return
subprocess
.
Popen
(
args
)
def
test_interprocess_signal
(
self
):
# Install handlers. This function runs in a sub-process, so we
# don't worry about re-setting the default handlers.
signal
.
signal
(
signal
.
SIGHUP
,
self
.
sighup_handler
)
signal
.
signal
(
signal
.
SIGUSR1
,
self
.
sigusr1_handler
)
signal
.
signal
(
signal
.
SIGUSR2
,
signal
.
SIG_IGN
)
signal
.
signal
(
signal
.
SIGALRM
,
signal
.
default_int_handler
)
# Let the sub-processes know who to send signals to.
pid
=
str
(
os
.
getpid
())
with
self
.
subprocess_send_signal
(
pid
,
"SIGHUP"
)
as
child
:
self
.
wait_signal
(
child
,
'SIGHUP'
)
self
.
assertEqual
(
self
.
got_signals
,
{
'SIGHUP'
:
1
,
'SIGUSR1'
:
0
,
'SIGALRM'
:
0
})
with
self
.
assertRaises
(
SIGUSR1Exception
):
with
self
.
subprocess_send_signal
(
pid
,
"SIGUSR1"
)
as
child
:
self
.
wait_signal
(
child
,
'SIGUSR1'
)
self
.
assertEqual
(
self
.
got_signals
,
{
'SIGHUP'
:
1
,
'SIGUSR1'
:
1
,
'SIGALRM'
:
0
})
with
self
.
subprocess_send_signal
(
pid
,
"SIGUSR2"
)
as
child
:
# Nothing should happen: SIGUSR2 is ignored
child
.
wait
()
try
:
with
self
.
assertRaises
(
KeyboardInterrupt
):
signal
.
alarm
(
1
)
self
.
wait_signal
(
None
,
'SIGALRM'
)
self
.
assertEqual
(
self
.
got_signals
,
{
'SIGHUP'
:
1
,
'SIGUSR1'
:
1
,
'SIGALRM'
:
0
})
finally
:
signal
.
alarm
(
0
)
if
__name__
==
"__main__"
:
unittest
.
main
()
src/greentest/3.10/test_signal.py
0 → 100644
View file @
b7ba2f41
import
errno
import
os
import
random
import
signal
import
socket
import
statistics
import
subprocess
import
sys
import
threading
import
time
import
unittest
from
test
import
support
from
test.support
import
os_helper
from
test.support.script_helper
import
assert_python_ok
,
spawn_python
try
:
import
_testcapi
except
ImportError
:
_testcapi
=
None
class
GenericTests
(
unittest
.
TestCase
):
def
test_enums
(
self
):
for
name
in
dir
(
signal
):
sig
=
getattr
(
signal
,
name
)
if
name
in
{
'SIG_DFL'
,
'SIG_IGN'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Handlers
)
elif
name
in
{
'SIG_BLOCK'
,
'SIG_UNBLOCK'
,
'SIG_SETMASK'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Sigmasks
)
elif
name
.
startswith
(
'SIG'
)
and
not
name
.
startswith
(
'SIG_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
elif
name
.
startswith
(
'CTRL_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
self
.
assertEqual
(
sys
.
platform
,
"win32"
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
PosixTests
(
unittest
.
TestCase
):
def
trivial_signal_handler
(
self
,
*
args
):
pass
def
test_out_of_range_signal_number_raises_error
(
self
):
self
.
assertRaises
(
ValueError
,
signal
.
getsignal
,
4242
)
self
.
assertRaises
(
ValueError
,
signal
.
signal
,
4242
,
self
.
trivial_signal_handler
)
self
.
assertRaises
(
ValueError
,
signal
.
strsignal
,
4242
)
def
test_setting_signal_handler_to_none_raises_error
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
signal
,
signal
.
SIGUSR1
,
None
)
def
test_getsignal
(
self
):
hup
=
signal
.
signal
(
signal
.
SIGHUP
,
self
.
trivial_signal_handler
)
self
.
assertIsInstance
(
hup
,
signal
.
Handlers
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
self
.
trivial_signal_handler
)
signal
.
signal
(
signal
.
SIGHUP
,
hup
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
hup
)
def
test_strsignal
(
self
):
self
.
assertIn
(
"Interrupt"
,
signal
.
strsignal
(
signal
.
SIGINT
))
self
.
assertIn
(
"Terminated"
,
signal
.
strsignal
(
signal
.
SIGTERM
))
self
.
assertIn
(
"Hangup"
,
signal
.
strsignal
(
signal
.
SIGHUP
))
# Issue 3864, unknown if this affects earlier versions of freebsd also
def
test_interprocess_signal
(
self
):
dirname
=
os
.
path
.
dirname
(
__file__
)
script
=
os
.
path
.
join
(
dirname
,
'signalinterproctester.py'
)
assert_python_ok
(
script
)
def
test_valid_signals
(
self
):
s
=
signal
.
valid_signals
()
self
.
assertIsInstance
(
s
,
set
)
self
.
assertIn
(
signal
.
Signals
.
SIGINT
,
s
)
self
.
assertIn
(
signal
.
Signals
.
SIGALRM
,
s
)
self
.
assertNotIn
(
0
,
s
)
self
.
assertNotIn
(
signal
.
NSIG
,
s
)
self
.
assertLess
(
len
(
s
),
signal
.
NSIG
)
@
unittest
.
skipUnless
(
sys
.
executable
,
"sys.executable required."
)
def
test_keyboard_interrupt_exit_code
(
self
):
"""KeyboardInterrupt triggers exit via SIGINT."""
process
=
subprocess
.
run
(
[
sys
.
executable
,
"-c"
,
"import os, signal, time
\
n
"
"os.kill(os.getpid(), signal.SIGINT)
\
n
"
"for _ in range(999): time.sleep(0.01)"
],
stderr
=
subprocess
.
PIPE
)
self
.
assertIn
(
b"KeyboardInterrupt"
,
process
.
stderr
)
self
.
assertEqual
(
process
.
returncode
,
-
signal
.
SIGINT
)
# Caveat: The exit code is insufficient to guarantee we actually died
# via a signal. POSIX shells do more than look at the 8 bit value.
# Writing an automation friendly test of an interactive shell
# to confirm that our process died via a SIGINT proved too complex.
@
unittest
.
skipUnless
(
sys
.
platform
==
"win32"
,
"Windows specific"
)
class
WindowsSignalTests
(
unittest
.
TestCase
):
def
test_valid_signals
(
self
):
s
=
signal
.
valid_signals
()
self
.
assertIsInstance
(
s
,
set
)
self
.
assertGreaterEqual
(
len
(
s
),
6
)
self
.
assertIn
(
signal
.
Signals
.
SIGINT
,
s
)
self
.
assertNotIn
(
0
,
s
)
self
.
assertNotIn
(
signal
.
NSIG
,
s
)
self
.
assertLess
(
len
(
s
),
signal
.
NSIG
)
def
test_issue9324
(
self
):
# Updated for issue #10003, adding SIGBREAK
handler
=
lambda
x
,
y
:
None
checked
=
set
()
for
sig
in
(
signal
.
SIGABRT
,
signal
.
SIGBREAK
,
signal
.
SIGFPE
,
signal
.
SIGILL
,
signal
.
SIGINT
,
signal
.
SIGSEGV
,
signal
.
SIGTERM
):
# Set and then reset a handler for signals that work on windows.
# Issue #18396, only for signals without a C-level handler.
if
signal
.
getsignal
(
sig
)
is
not
None
:
signal
.
signal
(
sig
,
signal
.
signal
(
sig
,
handler
))
checked
.
add
(
sig
)
# Issue #18396: Ensure the above loop at least tested *something*
self
.
assertTrue
(
checked
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
-
1
,
handler
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
7
,
handler
)
@
unittest
.
skipUnless
(
sys
.
executable
,
"sys.executable required."
)
def
test_keyboard_interrupt_exit_code
(
self
):
"""KeyboardInterrupt triggers an exit using STATUS_CONTROL_C_EXIT."""
# We don't test via os.kill(os.getpid(), signal.CTRL_C_EVENT) here
# as that requires setting up a console control handler in a child
# in its own process group. Doable, but quite complicated. (see
# @eryksun on https://github.com/python/cpython/pull/11862)
process
=
subprocess
.
run
(
[
sys
.
executable
,
"-c"
,
"raise KeyboardInterrupt"
],
stderr
=
subprocess
.
PIPE
)
self
.
assertIn
(
b"KeyboardInterrupt"
,
process
.
stderr
)
STATUS_CONTROL_C_EXIT
=
0xC000013A
self
.
assertEqual
(
process
.
returncode
,
STATUS_CONTROL_C_EXIT
)
class
WakeupFDTests
(
unittest
.
TestCase
):
def
test_invalid_call
(
self
):
# First parameter is positional-only
with
self
.
assertRaises
(
TypeError
):
signal
.
set_wakeup_fd
(
signum
=
signal
.
SIGINT
)
# warn_on_full_buffer is a keyword-only parameter
with
self
.
assertRaises
(
TypeError
):
signal
.
set_wakeup_fd
(
signal
.
SIGINT
,
False
)
def
test_invalid_fd
(
self
):
fd
=
os_helper
.
make_bad_fd
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_invalid_socket
(
self
):
sock
=
socket
.
socket
()
fd
=
sock
.
fileno
()
sock
.
close
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_set_wakeup_fd_result
(
self
):
r1
,
w1
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r1
)
self
.
addCleanup
(
os
.
close
,
w1
)
r2
,
w2
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r2
)
self
.
addCleanup
(
os
.
close
,
w2
)
if
hasattr
(
os
,
'set_blocking'
):
os
.
set_blocking
(
w1
,
False
)
os
.
set_blocking
(
w2
,
False
)
signal
.
set_wakeup_fd
(
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
w2
),
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
w2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
def
test_set_wakeup_fd_socket_result
(
self
):
sock1
=
socket
.
socket
()
self
.
addCleanup
(
sock1
.
close
)
sock1
.
setblocking
(
False
)
fd1
=
sock1
.
fileno
()
sock2
=
socket
.
socket
()
self
.
addCleanup
(
sock2
.
close
)
sock2
.
setblocking
(
False
)
fd2
=
sock2
.
fileno
()
signal
.
set_wakeup_fd
(
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
fd2
),
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
fd2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
# On Windows, files are always blocking and Windows does not provide a
# function to test if a socket is in non-blocking mode.
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"tests specific to POSIX"
)
def
test_set_wakeup_fd_blocking
(
self
):
rfd
,
wfd
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
rfd
)
self
.
addCleanup
(
os
.
close
,
wfd
)
# fd must be non-blocking
os
.
set_blocking
(
wfd
,
True
)
with
self
.
assertRaises
(
ValueError
)
as
cm
:
signal
.
set_wakeup_fd
(
wfd
)
self
.
assertEqual
(
str
(
cm
.
exception
),
"the fd %s must be in non-blocking mode"
%
wfd
)
# non-blocking is ok
os
.
set_blocking
(
wfd
,
False
)
signal
.
set_wakeup_fd
(
wfd
)
signal
.
set_wakeup_fd
(
-
1
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
WakeupSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
check_wakeup
(
self
,
test_body
,
*
signals
,
ordered
=
True
):
# use a subprocess to have only one thread
code
=
"""if 1:
import _testcapi
import os
import signal
import struct
signals = {!r}
def handler(signum, frame):
pass
def check_signum(signals):
data = os.read(read, len(signals)+1)
raised = struct.unpack('%uB' % len(data), data)
if not {!r}:
raised = set(raised)
signals = set(signals)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
{}
signal.signal(signal.SIGALRM, handler)
read, write = os.pipe()
os.set_blocking(write, False)
signal.set_wakeup_fd(write)
test()
check_signum(signals)
os.close(read)
os.close(write)
"""
.
format
(
tuple
(
map
(
int
,
signals
)),
ordered
,
test_body
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_wakeup_write_error
(
self
):
# Issue #16105: write() errors in the C signal handler should not
# pass silently.
# Use a subprocess to have only one thread.
code
=
"""if 1:
import _testcapi
import errno
import os
import signal
import sys
from test.support import captured_stderr
def handler(signum, frame):
1/0
signal.signal(signal.SIGALRM, handler)
r, w = os.pipe()
os.set_blocking(r, False)
# Set wakeup_fd a read-only file descriptor to trigger the error
signal.set_wakeup_fd(r)
try:
with captured_stderr() as err:
signal.raise_signal(signal.SIGALRM)
except ZeroDivisionError:
# An ignored exception should have been printed out on stderr
err = err.getvalue()
if ('Exception ignored when trying to write to the signal wakeup fd'
not in err):
raise AssertionError(err)
if ('OSError: [Errno %d]' % errno.EBADF) not in err:
raise AssertionError(err)
else:
raise AssertionError("ZeroDivisionError not raised")
os.close(r)
os.close(w)
"""
r
,
w
=
os
.
pipe
()
try
:
os
.
write
(
r
,
b'x'
)
except
OSError
:
pass
else
:
self
.
skipTest
(
"OS doesn't report write() error on the read end of a pipe"
)
finally
:
os
.
close
(
r
)
os
.
close
(
w
)
assert_python_ok
(
'-c'
,
code
)
def
test_wakeup_fd_early
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
# We attempt to get a signal during the sleep,
# before select is called
try:
select.select([], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
before_time = time.monotonic()
select.select([read], [], [], TIMEOUT_FULL)
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_wakeup_fd_during
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
before_time = time.monotonic()
# We attempt to get a signal during the select call
try:
select.select([read], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_signum
(
self
):
self
.
check_wakeup
(
"""def test():
signal.signal(signal.SIGUSR1, handler)
signal.raise_signal(signal.SIGUSR1)
signal.raise_signal(signal.SIGALRM)
"""
,
signal
.
SIGUSR1
,
signal
.
SIGALRM
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pending
(
self
):
self
.
check_wakeup
(
"""def test():
signum1 = signal.SIGUSR1
signum2 = signal.SIGUSR2
signal.signal(signum1, handler)
signal.signal(signum2, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, (signum1, signum2))
signal.raise_signal(signum1)
signal.raise_signal(signum2)
# Unblocking the 2 signals calls the C signal handler twice
signal.pthread_sigmask(signal.SIG_UNBLOCK, (signum1, signum2))
"""
,
signal
.
SIGUSR1
,
signal
.
SIGUSR2
,
ordered
=
False
)
@
unittest
.
skipUnless
(
hasattr
(
socket
,
'socketpair'
),
'need socket.socketpair'
)
class
WakeupSocketSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_socket
(
self
):
# use a subprocess to have only one thread
code
=
"""if 1:
import signal
import socket
import struct
import _testcapi
signum = signal.SIGINT
signals = (signum,)
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
signal.raise_signal(signum)
data = read.recv(1)
if not data:
raise Exception("no signum written")
raised = struct.unpack('B', data)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
read.close()
write.close()
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_send_error
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
read.setblocking(False)
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
# Close sockets: send() will fail
read.close()
write.close()
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if ('Exception ignored when trying to {action} to the signal wakeup fd'
not in err):
raise AssertionError(err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_warn_on_full_buffer
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
# This handler will be called, but we intentionally won't read from
# the wakeup fd.
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
# Fill the socketpair buffer
if sys.platform == 'win32':
# bpo-34130: On Windows, sometimes non-blocking send fails to fill
# the full socketpair buffer, so use a timeout of 50 ms instead.
write.settimeout(0.050)
else:
write.setblocking(False)
written = 0
if sys.platform == "vxworks":
CHUNK_SIZES = (1,)
else:
# Start with large chunk size to reduce the
# number of send needed to fill the buffer.
CHUNK_SIZES = (2 ** 16, 2 ** 8, 1)
for chunk_size in CHUNK_SIZES:
chunk = b"x" * chunk_size
try:
while True:
write.send(chunk)
written += chunk_size
except (BlockingIOError, TimeoutError):
pass
print(f"%s bytes written into the socketpair" % written, flush=True)
write.setblocking(False)
try:
write.send(b"x")
except BlockingIOError:
# The socketpair buffer seems full
pass
else:
raise AssertionError("%s bytes failed to fill the socketpair "
"buffer" % written)
# By default, we get a warning when a signal arrives
msg = ('Exception ignored when trying to {action} '
'to the signal wakeup fd')
signal.set_wakeup_fd(write.fileno())
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("first set_wakeup_fd() test failed, "
"stderr: %r" % err)
# And also if warn_on_full_buffer=True
signal.set_wakeup_fd(write.fileno(), warn_on_full_buffer=True)
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("set_wakeup_fd(warn_on_full_buffer=True) "
"test failed, stderr: %r" % err)
# But not if warn_on_full_buffer=False
signal.set_wakeup_fd(write.fileno(), warn_on_full_buffer=False)
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if err != "":
raise AssertionError("set_wakeup_fd(warn_on_full_buffer=False) "
"test failed, stderr: %r" % err)
# And then check the default again, to make sure warn_on_full_buffer
# settings don't leak across calls.
signal.set_wakeup_fd(write.fileno())
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("second set_wakeup_fd() test failed, "
"stderr: %r" % err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'siginterrupt'
),
"needs signal.siginterrupt()"
)
class
SiginterruptTest
(
unittest
.
TestCase
):
def
readpipe_interrupted
(
self
,
interrupt
):
"""Perform a read during which a signal will arrive. Return True if the
read is interrupted by the signal and raises an exception. Return False
if it returns normally.
"""
# use a subprocess to have only one thread, to have a timeout on the
# blocking read and to not touch signal handling in this process
code
=
"""if 1:
import errno
import os
import signal
import sys
interrupt = %r
r, w = os.pipe()
def handler(signum, frame):
1 / 0
signal.signal(signal.SIGALRM, handler)
if interrupt is not None:
signal.siginterrupt(signal.SIGALRM, interrupt)
print("ready")
sys.stdout.flush()
# run the test twice
try:
for loop in range(2):
# send a SIGALRM in a second (during the read)
signal.alarm(1)
try:
# blocking call: read from a pipe without data
os.read(r, 1)
except ZeroDivisionError:
pass
else:
sys.exit(2)
sys.exit(3)
finally:
os.close(r)
os.close(w)
"""
%
(
interrupt
,)
with
spawn_python
(
'-c'
,
code
)
as
process
:
try
:
# wait until the child process is loaded and has started
first_line
=
process
.
stdout
.
readline
()
stdout
,
stderr
=
process
.
communicate
(
timeout
=
support
.
SHORT_TIMEOUT
)
except
subprocess
.
TimeoutExpired
:
process
.
kill
()
return
False
else
:
stdout
=
first_line
+
stdout
exitcode
=
process
.
wait
()
if
exitcode
not
in
(
2
,
3
):
raise
Exception
(
"Child error (exit code %s): %r"
%
(
exitcode
,
stdout
))
return
(
exitcode
==
3
)
def
test_without_siginterrupt
(
self
):
# If a signal handler is installed and siginterrupt is not called
# at all, when that signal arrives, it interrupts a syscall that's in
# progress.
interrupted
=
self
.
readpipe_interrupted
(
None
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_on
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a true value for the second argument, when that signal arrives, it
# interrupts a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
True
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_off
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a false value for the second argument, when that signal arrives, it
# does not interrupt a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
False
)
self
.
assertFalse
(
interrupted
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'getitimer'
)
and
hasattr
(
signal
,
'setitimer'
),
"needs signal.getitimer() and signal.setitimer()"
)
class
ItimerTest
(
unittest
.
TestCase
):
def
setUp
(
self
):
self
.
hndl_called
=
False
self
.
hndl_count
=
0
self
.
itimer
=
None
self
.
old_alarm
=
signal
.
signal
(
signal
.
SIGALRM
,
self
.
sig_alrm
)
def
tearDown
(
self
):
signal
.
signal
(
signal
.
SIGALRM
,
self
.
old_alarm
)
if
self
.
itimer
is
not
None
:
# test_itimer_exc doesn't change this attr
# just ensure that itimer is stopped
signal
.
setitimer
(
self
.
itimer
,
0
)
def
sig_alrm
(
self
,
*
args
):
self
.
hndl_called
=
True
def
sig_vtalrm
(
self
,
*
args
):
self
.
hndl_called
=
True
if
self
.
hndl_count
>
3
:
# it shouldn't be here, because it should have been disabled.
raise
signal
.
ItimerError
(
"setitimer didn't disable ITIMER_VIRTUAL "
"timer."
)
elif
self
.
hndl_count
==
3
:
# disable ITIMER_VIRTUAL, this function shouldn't be called anymore
signal
.
setitimer
(
signal
.
ITIMER_VIRTUAL
,
0
)
self
.
hndl_count
+=
1
def
sig_prof
(
self
,
*
args
):
self
.
hndl_called
=
True
signal
.
setitimer
(
signal
.
ITIMER_PROF
,
0
)
def
test_itimer_exc
(
self
):
# XXX I'm assuming -1 is an invalid itimer, but maybe some platform
# defines it ?
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
-
1
,
0
)
# Negative times are treated as zero on some platforms.
if
0
:
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
-
1
)
def
test_itimer_real
(
self
):
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1.0
)
signal
.
pause
()
self
.
assertEqual
(
self
.
hndl_called
,
True
)
# Issue 3864, unknown if this affects earlier versions of freebsd also
@
unittest
.
skipIf
(
sys
.
platform
in
(
'netbsd5'
,),
'itimer not reliable (does not mix well with threading) on some BSDs.'
)
def
test_itimer_virtual
(
self
):
self
.
itimer
=
signal
.
ITIMER_VIRTUAL
signal
.
signal
(
signal
.
SIGVTALRM
,
self
.
sig_vtalrm
)
signal
.
setitimer
(
self
.
itimer
,
0.3
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# use up some virtual time by doing real work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_vtalrm handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# virtual itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_itimer_prof
(
self
):
self
.
itimer
=
signal
.
ITIMER_PROF
signal
.
signal
(
signal
.
SIGPROF
,
self
.
sig_prof
)
signal
.
setitimer
(
self
.
itimer
,
0.2
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# do some work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_prof handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# profiling itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_setitimer_tiny
(
self
):
# bpo-30807: C setitimer() takes a microsecond-resolution interval.
# Check that float -> timeval conversion doesn't round
# the interval down to zero, which would disable the timer.
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1e-6
)
time
.
sleep
(
1
)
self
.
assertEqual
(
self
.
hndl_called
,
True
)
class
PendingSignalsTests
(
unittest
.
TestCase
):
"""
Test pthread_sigmask(), pthread_kill(), sigpending() and sigwait()
functions.
"""
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending_empty
(
self
):
self
.
assertEqual
(
signal
.
sigpending
(),
set
())
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending
(
self
):
code
=
"""if 1:
import os
import signal
def handler(signum, frame):
1/0
signum = signal.SIGUSR1
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
os.kill(os.getpid(), signum)
pending = signal.sigpending()
for sig in pending:
assert isinstance(sig, signal.Signals), repr(pending)
if pending != {signum}:
raise Exception('%s != {%s}' % (pending, signum))
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill
(
self
):
code
=
"""if 1:
import signal
import threading
import sys
signum = signal.SIGUSR1
def handler(signum, frame):
1/0
signal.signal(signum, handler)
tid = threading.get_ident()
try:
signal.pthread_kill(tid, signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
wait_helper
(
self
,
blocked
,
test
):
"""
test: body of the "def test(signum):" function.
blocked: number of the blocked signal
"""
code
=
'''if 1:
import signal
import sys
from signal import Signals
def handler(signum, frame):
1/0
%s
blocked = %s
signum = signal.SIGALRM
# child: block and wait the signal
try:
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [blocked])
# Do the tests
test(signum)
# The handler must not be called on unblock
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [blocked])
except ZeroDivisionError:
print("the signal handler has been called",
file=sys.stderr)
sys.exit(1)
except BaseException as err:
print("error: {}".format(err), file=sys.stderr)
sys.stderr.flush()
sys.exit(1)
'''
%
(
test
.
strip
(),
blocked
)
# sig*wait* must be called with the signal blocked: since the current
# process might have several threads running, use a subprocess to have
# a single thread.
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
def
test_sigwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
received = signal.sigwait([signum])
assert isinstance(received, signal.Signals), received
if received != signum:
raise Exception('received %s, not %s' % (received, signum))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwaitinfo'
),
'need signal.sigwaitinfo()'
)
def
test_sigwaitinfo
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigwaitinfo([signum])
if info.si_signo != signum:
raise Exception("info.si_signo != %s" % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigtimedwait([signum], 10.1000)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_poll
(
self
):
# check that polling with sigtimedwait works
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
import os
os.kill(os.getpid(), signum)
info = signal.sigtimedwait([signum], 0)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_timeout
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
received = signal.sigtimedwait([signum], 1.0)
if received is not None:
raise Exception("received=%r" % (received,))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_negative_timeout
(
self
):
signum
=
signal
.
SIGALRM
self
.
assertRaises
(
ValueError
,
signal
.
sigtimedwait
,
[
signum
],
-
1.0
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_sigwait_thread
(
self
):
# Check that calling sigwait() from a thread doesn't suspend the whole
# process. A new interpreter is spawned to avoid problems when mixing
# threads and fork(): only async-safe functions are allowed between
# fork() and exec().
assert_python_ok
(
"-c"
,
"""if True:
import os, threading, sys, time, signal
# the default handler terminates the process
signum = signal.SIGUSR1
def kill_later():
# wait until the main thread is waiting in sigwait()
time.sleep(1)
os.kill(os.getpid(), signum)
# the signal must be blocked by all the threads
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
killer = threading.Thread(target=kill_later)
killer.start()
received = signal.sigwait([signum])
if received != signum:
print("sigwait() received %s, not %s" % (received, signum),
file=sys.stderr)
sys.exit(1)
killer.join()
# unblock the signal, which should have been cleared by sigwait()
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
"""
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask_arguments
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
,
2
,
3
)
self
.
assertRaises
(
OSError
,
signal
.
pthread_sigmask
,
1700
,
[])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
signal
.
NSIG
])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
0
])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
1
<<
1000
])
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask_valid_signals
(
self
):
s
=
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
signal
.
valid_signals
())
self
.
addCleanup
(
signal
.
pthread_sigmask
,
signal
.
SIG_SETMASK
,
s
)
# Get current blocked set
s
=
signal
.
pthread_sigmask
(
signal
.
SIG_UNBLOCK
,
signal
.
valid_signals
())
self
.
assertLessEqual
(
s
,
signal
.
valid_signals
())
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask
(
self
):
code
=
"""if 1:
import signal
import os; import threading
def handler(signum, frame):
1/0
def kill(signum):
os.kill(os.getpid(), signum)
def check_mask(mask):
for sig in mask:
assert isinstance(sig, signal.Signals), repr(sig)
def read_sigmask():
sigmask = signal.pthread_sigmask(signal.SIG_BLOCK, [])
check_mask(sigmask)
return sigmask
signum = signal.SIGUSR1
# Install our signal handler
old_handler = signal.signal(signum, handler)
# Unblock SIGUSR1 (and copy the old mask) to test our signal handler
old_mask = signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
check_mask(old_mask)
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Block and then raise SIGUSR1. The signal is blocked: the signal
# handler is not called, and the signal is now pending
mask = signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
check_mask(mask)
kill(signum)
# Check the new mask
blocked = read_sigmask()
check_mask(blocked)
if signum not in blocked:
raise Exception("%s not in %s" % (signum, blocked))
if old_mask ^ blocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (old_mask, blocked, signum))
# Unblock SIGUSR1
try:
# unblock the pending signal calls immediately the signal handler
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Check the new mask
unblocked = read_sigmask()
if signum in unblocked:
raise Exception("%s in %s" % (signum, unblocked))
if blocked ^ unblocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (blocked, unblocked, signum))
if old_mask != unblocked:
raise Exception("%s != %s" % (old_mask, unblocked))
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill_main_thread
(
self
):
# Test that a signal can be sent to the main thread with pthread_kill()
# before any other thread has been created (see issue #12392).
code
=
"""if True:
import threading
import signal
import sys
def handler(signum, frame):
sys.exit(3)
signal.signal(signal.SIGUSR1, handler)
signal.pthread_kill(threading.get_ident(), signal.SIGUSR1)
sys.exit(2)
"""
with
spawn_python
(
'-c'
,
code
)
as
process
:
stdout
,
stderr
=
process
.
communicate
()
exitcode
=
process
.
wait
()
if
exitcode
!=
3
:
raise
Exception
(
"Child error (exit code %s): %s"
%
(
exitcode
,
stdout
))
class
StressTest
(
unittest
.
TestCase
):
"""
Stress signal delivery, especially when a signal arrives in
the middle of recomputing the signal state or executing
previously tripped signal handlers.
"""
def
setsig
(
self
,
signum
,
handler
):
old_handler
=
signal
.
signal
(
signum
,
handler
)
self
.
addCleanup
(
signal
.
signal
,
signum
,
old_handler
)
def
measure_itimer_resolution
(
self
):
N
=
20
times
=
[]
def
handler
(
signum
=
None
,
frame
=
None
):
if
len
(
times
)
<
N
:
times
.
append
(
time
.
perf_counter
())
# 1 µs is the smallest possible timer interval,
# we want to measure what the concrete duration
# will be on this platform
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
)
self
.
addCleanup
(
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
0
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
handler
()
while
len
(
times
)
<
N
:
time
.
sleep
(
1e-3
)
durations
=
[
times
[
i
+
1
]
-
times
[
i
]
for
i
in
range
(
len
(
times
)
-
1
)]
med
=
statistics
.
median
(
durations
)
if
support
.
verbose
:
print
(
"detected median itimer() resolution: %.6f s."
%
(
med
,))
return
med
def
decide_itimer_count
(
self
):
# Some systems have poor setitimer() resolution (for example
# measured around 20 ms. on FreeBSD 9), so decide on a reasonable
# number of sequential timers based on that.
reso
=
self
.
measure_itimer_resolution
()
if
reso
<=
1e-4
:
return
10000
elif
reso
<=
1e-2
:
return
100
else
:
self
.
skipTest
(
"detected itimer resolution (%.3f s.) too high "
"(> 10 ms.) on this platform (or system too busy)"
%
(
reso
,))
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_dependent
(
self
):
"""
This test uses dependent signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
first_handler
(
signum
,
frame
):
# 1e-6 is the minimum non-zero value for `setitimer()`.
# Choose a random delay so as to improve chances of
# triggering a race condition. Ideally the signal is received
# when inside critical signal-handling routines such as
# Py_MakePendingCalls().
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
def
second_handler
(
signum
=
None
,
frame
=
None
):
sigs
.
append
(
signum
)
# Here on Linux, SIGPROF > SIGALRM > SIGUSR1. By using both
# ascending and descending sequences (SIGUSR1 then SIGALRM,
# SIGPROF then SIGALRM), we maximize chances of hitting a bug.
self
.
setsig
(
signal
.
SIGPROF
,
first_handler
)
self
.
setsig
(
signal
.
SIGUSR1
,
first_handler
)
self
.
setsig
(
signal
.
SIGALRM
,
second_handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
monotonic
()
+
support
.
SHORT_TIMEOUT
while
expected_sigs
<
N
:
os
.
kill
(
os
.
getpid
(),
signal
.
SIGPROF
)
expected_sigs
+=
1
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
1
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_simultaneous
(
self
):
"""
This test uses simultaneous signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
handler
(
signum
,
frame
):
sigs
.
append
(
signum
)
self
.
setsig
(
signal
.
SIGUSR1
,
handler
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
monotonic
()
+
support
.
SHORT_TIMEOUT
while
expected_sigs
<
N
:
# Hopefully the SIGALRM will be received somewhere during
# initial processing of SIGUSR1.
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
2
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"SIGUSR1"
),
"test needs SIGUSR1"
)
def
test_stress_modifying_handlers
(
self
):
# bpo-43406: race condition between trip_signal() and signal.signal
signum
=
signal
.
SIGUSR1
num_sent_signals
=
0
num_received_signals
=
0
do_stop
=
False
def
custom_handler
(
signum
,
frame
):
nonlocal
num_received_signals
num_received_signals
+=
1
def
set_interrupts
():
nonlocal
num_sent_signals
while
not
do_stop
:
signal
.
raise_signal
(
signum
)
num_sent_signals
+=
1
def
cycle_handlers
():
while
num_sent_signals
<
100
:
for
i
in
range
(
20000
):
# Cycle between a Python-defined and a non-Python handler
for
handler
in
[
custom_handler
,
signal
.
SIG_IGN
]:
signal
.
signal
(
signum
,
handler
)
old_handler
=
signal
.
signal
(
signum
,
custom_handler
)
self
.
addCleanup
(
signal
.
signal
,
signum
,
old_handler
)
t
=
threading
.
Thread
(
target
=
set_interrupts
)
try
:
ignored
=
False
with
support
.
catch_unraisable_exception
()
as
cm
:
t
.
start
()
cycle_handlers
()
do_stop
=
True
t
.
join
()
if
cm
.
unraisable
is
not
None
:
# An unraisable exception may be printed out when
# a signal is ignored due to the aforementioned
# race condition, check it.
self
.
assertIsInstance
(
cm
.
unraisable
.
exc_value
,
OSError
)
self
.
assertIn
(
f"Signal
{
signum
:
d
}
ignored due to race condition"
,
str
(
cm
.
unraisable
.
exc_value
))
ignored
=
True
# bpo-43406: Even if it is unlikely, it's technically possible that
# all signals were ignored because of race conditions.
if
not
ignored
:
# Sanity check that some signals were received, but not all
self
.
assertGreater
(
num_received_signals
,
0
)
self
.
assertLess
(
num_received_signals
,
num_sent_signals
)
finally
:
do_stop
=
True
t
.
join
()
class
RaiseSignalTest
(
unittest
.
TestCase
):
def
test_sigint
(
self
):
with
self
.
assertRaises
(
KeyboardInterrupt
):
signal
.
raise_signal
(
signal
.
SIGINT
)
@
unittest
.
skipIf
(
sys
.
platform
!=
"win32"
,
"Windows specific test"
)
def
test_invalid_argument
(
self
):
try
:
SIGHUP
=
1
# not supported on win32
signal
.
raise_signal
(
SIGHUP
)
self
.
fail
(
"OSError (Invalid argument) expected"
)
except
OSError
as
e
:
if
e
.
errno
==
errno
.
EINVAL
:
pass
else
:
raise
def
test_handler
(
self
):
is_ok
=
False
def
handler
(
a
,
b
):
nonlocal
is_ok
is_ok
=
True
old_signal
=
signal
.
signal
(
signal
.
SIGINT
,
handler
)
self
.
addCleanup
(
signal
.
signal
,
signal
.
SIGINT
,
old_signal
)
signal
.
raise_signal
(
signal
.
SIGINT
)
self
.
assertTrue
(
is_ok
)
class
PidfdSignalTest
(
unittest
.
TestCase
):
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"pidfd_send_signal"
),
"pidfd support not built in"
,
)
def
test_pidfd_send_signal
(
self
):
with
self
.
assertRaises
(
OSError
)
as
cm
:
signal
.
pidfd_send_signal
(
0
,
signal
.
SIGINT
)
if
cm
.
exception
.
errno
==
errno
.
ENOSYS
:
self
.
skipTest
(
"kernel does not support pidfds"
)
elif
cm
.
exception
.
errno
==
errno
.
EPERM
:
self
.
skipTest
(
"Not enough privileges to use pidfs"
)
self
.
assertEqual
(
cm
.
exception
.
errno
,
errno
.
EBADF
)
my_pidfd
=
os
.
open
(
f'/proc/
{
os
.
getpid
()
}
'
,
os
.
O_DIRECTORY
)
self
.
addCleanup
(
os
.
close
,
my_pidfd
)
with
self
.
assertRaisesRegex
(
TypeError
,
"^siginfo must be None$"
):
signal
.
pidfd_send_signal
(
my_pidfd
,
signal
.
SIGINT
,
object
(),
0
)
with
self
.
assertRaises
(
KeyboardInterrupt
):
signal
.
pidfd_send_signal
(
my_pidfd
,
signal
.
SIGINT
)
def
tearDownModule
():
support
.
reap_children
()
if
__name__
==
"__main__"
:
unittest
.
main
()
src/greentest/3.6/test_signal.py
0 → 100644
View file @
b7ba2f41
import
unittest
from
test
import
support
from
contextlib
import
closing
import
enum
import
gc
import
os
import
pickle
import
random
import
select
import
signal
import
socket
import
statistics
import
subprocess
import
traceback
import
sys
,
os
,
time
,
errno
from
test.support.script_helper
import
assert_python_ok
,
spawn_python
try
:
import
threading
except
ImportError
:
threading
=
None
try
:
import
_testcapi
except
ImportError
:
_testcapi
=
None
class
GenericTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
threading
is
None
,
"test needs threading module"
)
def
test_enums
(
self
):
for
name
in
dir
(
signal
):
sig
=
getattr
(
signal
,
name
)
if
name
in
{
'SIG_DFL'
,
'SIG_IGN'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Handlers
)
elif
name
in
{
'SIG_BLOCK'
,
'SIG_UNBLOCK'
,
'SIG_SETMASK'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Sigmasks
)
elif
name
.
startswith
(
'SIG'
)
and
not
name
.
startswith
(
'SIG_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
elif
name
.
startswith
(
'CTRL_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
self
.
assertEqual
(
sys
.
platform
,
"win32"
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
PosixTests
(
unittest
.
TestCase
):
def
trivial_signal_handler
(
self
,
*
args
):
pass
def
test_out_of_range_signal_number_raises_error
(
self
):
self
.
assertRaises
(
ValueError
,
signal
.
getsignal
,
4242
)
self
.
assertRaises
(
ValueError
,
signal
.
signal
,
4242
,
self
.
trivial_signal_handler
)
def
test_setting_signal_handler_to_none_raises_error
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
signal
,
signal
.
SIGUSR1
,
None
)
def
test_getsignal
(
self
):
hup
=
signal
.
signal
(
signal
.
SIGHUP
,
self
.
trivial_signal_handler
)
self
.
assertIsInstance
(
hup
,
signal
.
Handlers
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
self
.
trivial_signal_handler
)
signal
.
signal
(
signal
.
SIGHUP
,
hup
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
hup
)
# Issue 3864, unknown if this affects earlier versions of freebsd also
@
unittest
.
skipIf
(
sys
.
platform
==
'freebsd6'
,
'inter process signals not reliable (do not mix well with threading) '
'on freebsd6'
)
def
test_interprocess_signal
(
self
):
dirname
=
os
.
path
.
dirname
(
__file__
)
script
=
os
.
path
.
join
(
dirname
,
'signalinterproctester.py'
)
assert_python_ok
(
script
)
@
unittest
.
skipUnless
(
sys
.
platform
==
"win32"
,
"Windows specific"
)
class
WindowsSignalTests
(
unittest
.
TestCase
):
def
test_issue9324
(
self
):
# Updated for issue #10003, adding SIGBREAK
handler
=
lambda
x
,
y
:
None
checked
=
set
()
for
sig
in
(
signal
.
SIGABRT
,
signal
.
SIGBREAK
,
signal
.
SIGFPE
,
signal
.
SIGILL
,
signal
.
SIGINT
,
signal
.
SIGSEGV
,
signal
.
SIGTERM
):
# Set and then reset a handler for signals that work on windows.
# Issue #18396, only for signals without a C-level handler.
if
signal
.
getsignal
(
sig
)
is
not
None
:
signal
.
signal
(
sig
,
signal
.
signal
(
sig
,
handler
))
checked
.
add
(
sig
)
# Issue #18396: Ensure the above loop at least tested *something*
self
.
assertTrue
(
checked
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
-
1
,
handler
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
7
,
handler
)
class
WakeupFDTests
(
unittest
.
TestCase
):
def
test_invalid_fd
(
self
):
fd
=
support
.
make_bad_fd
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_invalid_socket
(
self
):
sock
=
socket
.
socket
()
fd
=
sock
.
fileno
()
sock
.
close
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_set_wakeup_fd_result
(
self
):
r1
,
w1
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r1
)
self
.
addCleanup
(
os
.
close
,
w1
)
r2
,
w2
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r2
)
self
.
addCleanup
(
os
.
close
,
w2
)
if
hasattr
(
os
,
'set_blocking'
):
os
.
set_blocking
(
w1
,
False
)
os
.
set_blocking
(
w2
,
False
)
signal
.
set_wakeup_fd
(
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
w2
),
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
w2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
def
test_set_wakeup_fd_socket_result
(
self
):
sock1
=
socket
.
socket
()
self
.
addCleanup
(
sock1
.
close
)
sock1
.
setblocking
(
False
)
fd1
=
sock1
.
fileno
()
sock2
=
socket
.
socket
()
self
.
addCleanup
(
sock2
.
close
)
sock2
.
setblocking
(
False
)
fd2
=
sock2
.
fileno
()
signal
.
set_wakeup_fd
(
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
fd2
),
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
fd2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
# On Windows, files are always blocking and Windows does not provide a
# function to test if a socket is in non-blocking mode.
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"tests specific to POSIX"
)
def
test_set_wakeup_fd_blocking
(
self
):
rfd
,
wfd
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
rfd
)
self
.
addCleanup
(
os
.
close
,
wfd
)
# fd must be non-blocking
os
.
set_blocking
(
wfd
,
True
)
with
self
.
assertRaises
(
ValueError
)
as
cm
:
signal
.
set_wakeup_fd
(
wfd
)
self
.
assertEqual
(
str
(
cm
.
exception
),
"the fd %s must be in non-blocking mode"
%
wfd
)
# non-blocking is ok
os
.
set_blocking
(
wfd
,
False
)
signal
.
set_wakeup_fd
(
wfd
)
signal
.
set_wakeup_fd
(
-
1
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
WakeupSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
check_wakeup
(
self
,
test_body
,
*
signals
,
ordered
=
True
):
# use a subprocess to have only one thread
code
=
"""if 1:
import _testcapi
import os
import signal
import struct
signals = {!r}
def handler(signum, frame):
pass
def check_signum(signals):
data = os.read(read, len(signals)+1)
raised = struct.unpack('%uB' % len(data), data)
if not {!r}:
raised = set(raised)
signals = set(signals)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
{}
signal.signal(signal.SIGALRM, handler)
read, write = os.pipe()
os.set_blocking(write, False)
signal.set_wakeup_fd(write)
test()
check_signum(signals)
os.close(read)
os.close(write)
"""
.
format
(
tuple
(
map
(
int
,
signals
)),
ordered
,
test_body
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_wakeup_write_error
(
self
):
# Issue #16105: write() errors in the C signal handler should not
# pass silently.
# Use a subprocess to have only one thread.
code
=
"""if 1:
import _testcapi
import errno
import os
import signal
import sys
from test.support import captured_stderr
def handler(signum, frame):
1/0
signal.signal(signal.SIGALRM, handler)
r, w = os.pipe()
os.set_blocking(r, False)
# Set wakeup_fd a read-only file descriptor to trigger the error
signal.set_wakeup_fd(r)
try:
with captured_stderr() as err:
_testcapi.raise_signal(signal.SIGALRM)
except ZeroDivisionError:
# An ignored exception should have been printed out on stderr
err = err.getvalue()
if ('Exception ignored when trying to write to the signal wakeup fd'
not in err):
raise AssertionError(err)
if ('OSError: [Errno %d]' % errno.EBADF) not in err:
raise AssertionError(err)
else:
raise AssertionError("ZeroDivisionError not raised")
os.close(r)
os.close(w)
"""
r
,
w
=
os
.
pipe
()
try
:
os
.
write
(
r
,
b'x'
)
except
OSError
:
pass
else
:
self
.
skipTest
(
"OS doesn't report write() error on the read end of a pipe"
)
finally
:
os
.
close
(
r
)
os
.
close
(
w
)
assert_python_ok
(
'-c'
,
code
)
def
test_wakeup_fd_early
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
# We attempt to get a signal during the sleep,
# before select is called
try:
select.select([], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
before_time = time.monotonic()
select.select([read], [], [], TIMEOUT_FULL)
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_wakeup_fd_during
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
before_time = time.monotonic()
# We attempt to get a signal during the select call
try:
select.select([read], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_signum
(
self
):
self
.
check_wakeup
(
"""def test():
import _testcapi
signal.signal(signal.SIGUSR1, handler)
_testcapi.raise_signal(signal.SIGUSR1)
_testcapi.raise_signal(signal.SIGALRM)
"""
,
signal
.
SIGUSR1
,
signal
.
SIGALRM
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pending
(
self
):
self
.
check_wakeup
(
"""def test():
signum1 = signal.SIGUSR1
signum2 = signal.SIGUSR2
signal.signal(signum1, handler)
signal.signal(signum2, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, (signum1, signum2))
_testcapi.raise_signal(signum1)
_testcapi.raise_signal(signum2)
# Unblocking the 2 signals calls the C signal handler twice
signal.pthread_sigmask(signal.SIG_UNBLOCK, (signum1, signum2))
"""
,
signal
.
SIGUSR1
,
signal
.
SIGUSR2
,
ordered
=
False
)
@
unittest
.
skipUnless
(
hasattr
(
socket
,
'socketpair'
),
'need socket.socketpair'
)
class
WakeupSocketSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_socket
(
self
):
# use a subprocess to have only one thread
code
=
"""if 1:
import signal
import socket
import struct
import _testcapi
signum = signal.SIGINT
signals = (signum,)
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
_testcapi.raise_signal(signum)
data = read.recv(1)
if not data:
raise Exception("no signum written")
raised = struct.unpack('B', data)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
read.close()
write.close()
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_send_error
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
read.setblocking(False)
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
# Close sockets: send() will fail
read.close()
write.close()
with captured_stderr() as err:
_testcapi.raise_signal(signum)
err = err.getvalue()
if ('Exception ignored when trying to {action} to the signal wakeup fd'
not in err):
raise AssertionError(err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
SiginterruptTest
(
unittest
.
TestCase
):
def
readpipe_interrupted
(
self
,
interrupt
):
"""Perform a read during which a signal will arrive. Return True if the
read is interrupted by the signal and raises an exception. Return False
if it returns normally.
"""
# use a subprocess to have only one thread, to have a timeout on the
# blocking read and to not touch signal handling in this process
code
=
"""if 1:
import errno
import os
import signal
import sys
interrupt = %r
r, w = os.pipe()
def handler(signum, frame):
1 / 0
signal.signal(signal.SIGALRM, handler)
if interrupt is not None:
signal.siginterrupt(signal.SIGALRM, interrupt)
print("ready")
sys.stdout.flush()
# run the test twice
try:
for loop in range(2):
# send a SIGALRM in a second (during the read)
signal.alarm(1)
try:
# blocking call: read from a pipe without data
os.read(r, 1)
except ZeroDivisionError:
pass
else:
sys.exit(2)
sys.exit(3)
finally:
os.close(r)
os.close(w)
"""
%
(
interrupt
,)
with
spawn_python
(
'-c'
,
code
)
as
process
:
try
:
# wait until the child process is loaded and has started
first_line
=
process
.
stdout
.
readline
()
stdout
,
stderr
=
process
.
communicate
(
timeout
=
5.0
)
except
subprocess
.
TimeoutExpired
:
process
.
kill
()
return
False
else
:
stdout
=
first_line
+
stdout
exitcode
=
process
.
wait
()
if
exitcode
not
in
(
2
,
3
):
raise
Exception
(
"Child error (exit code %s): %r"
%
(
exitcode
,
stdout
))
return
(
exitcode
==
3
)
def
test_without_siginterrupt
(
self
):
# If a signal handler is installed and siginterrupt is not called
# at all, when that signal arrives, it interrupts a syscall that's in
# progress.
interrupted
=
self
.
readpipe_interrupted
(
None
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_on
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a true value for the second argument, when that signal arrives, it
# interrupts a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
True
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_off
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a false value for the second argument, when that signal arrives, it
# does not interrupt a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
False
)
self
.
assertFalse
(
interrupted
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
ItimerTest
(
unittest
.
TestCase
):
def
setUp
(
self
):
self
.
hndl_called
=
False
self
.
hndl_count
=
0
self
.
itimer
=
None
self
.
old_alarm
=
signal
.
signal
(
signal
.
SIGALRM
,
self
.
sig_alrm
)
def
tearDown
(
self
):
signal
.
signal
(
signal
.
SIGALRM
,
self
.
old_alarm
)
if
self
.
itimer
is
not
None
:
# test_itimer_exc doesn't change this attr
# just ensure that itimer is stopped
signal
.
setitimer
(
self
.
itimer
,
0
)
def
sig_alrm
(
self
,
*
args
):
self
.
hndl_called
=
True
def
sig_vtalrm
(
self
,
*
args
):
self
.
hndl_called
=
True
if
self
.
hndl_count
>
3
:
# it shouldn't be here, because it should have been disabled.
raise
signal
.
ItimerError
(
"setitimer didn't disable ITIMER_VIRTUAL "
"timer."
)
elif
self
.
hndl_count
==
3
:
# disable ITIMER_VIRTUAL, this function shouldn't be called anymore
signal
.
setitimer
(
signal
.
ITIMER_VIRTUAL
,
0
)
self
.
hndl_count
+=
1
def
sig_prof
(
self
,
*
args
):
self
.
hndl_called
=
True
signal
.
setitimer
(
signal
.
ITIMER_PROF
,
0
)
def
test_itimer_exc
(
self
):
# XXX I'm assuming -1 is an invalid itimer, but maybe some platform
# defines it ?
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
-
1
,
0
)
# Negative times are treated as zero on some platforms.
if
0
:
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
-
1
)
def
test_itimer_real
(
self
):
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1.0
)
signal
.
pause
()
self
.
assertEqual
(
self
.
hndl_called
,
True
)
# Issue 3864, unknown if this affects earlier versions of freebsd also
@
unittest
.
skipIf
(
sys
.
platform
in
(
'freebsd6'
,
'netbsd5'
),
'itimer not reliable (does not mix well with threading) on some BSDs.'
)
def
test_itimer_virtual
(
self
):
self
.
itimer
=
signal
.
ITIMER_VIRTUAL
signal
.
signal
(
signal
.
SIGVTALRM
,
self
.
sig_vtalrm
)
signal
.
setitimer
(
self
.
itimer
,
0.3
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# use up some virtual time by doing real work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_vtalrm handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# virtual itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
# Issue 3864, unknown if this affects earlier versions of freebsd also
@
unittest
.
skipIf
(
sys
.
platform
==
'freebsd6'
,
'itimer not reliable (does not mix well with threading) on freebsd6'
)
def
test_itimer_prof
(
self
):
self
.
itimer
=
signal
.
ITIMER_PROF
signal
.
signal
(
signal
.
SIGPROF
,
self
.
sig_prof
)
signal
.
setitimer
(
self
.
itimer
,
0.2
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# do some work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_prof handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# profiling itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_setitimer_tiny
(
self
):
# bpo-30807: C setitimer() takes a microsecond-resolution interval.
# Check that float -> timeval conversion doesn't round
# the interval down to zero, which would disable the timer.
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1e-6
)
time
.
sleep
(
1
)
self
.
assertEqual
(
self
.
hndl_called
,
True
)
class
PendingSignalsTests
(
unittest
.
TestCase
):
"""
Test pthread_sigmask(), pthread_kill(), sigpending() and sigwait()
functions.
"""
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending_empty
(
self
):
self
.
assertEqual
(
signal
.
sigpending
(),
set
())
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending
(
self
):
code
=
"""if 1:
import os
import signal
def handler(signum, frame):
1/0
signum = signal.SIGUSR1
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
os.kill(os.getpid(), signum)
pending = signal.sigpending()
for sig in pending:
assert isinstance(sig, signal.Signals), repr(pending)
if pending != {signum}:
raise Exception('%s != {%s}' % (pending, signum))
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill
(
self
):
code
=
"""if 1:
import signal
import threading
import sys
signum = signal.SIGUSR1
def handler(signum, frame):
1/0
signal.signal(signum, handler)
if sys.platform == 'freebsd6':
# Issue #12392 and #12469: send a signal to the main thread
# doesn't work before the creation of the first thread on
# FreeBSD 6
def noop():
pass
thread = threading.Thread(target=noop)
thread.start()
thread.join()
tid = threading.get_ident()
try:
signal.pthread_kill(tid, signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
wait_helper
(
self
,
blocked
,
test
):
"""
test: body of the "def test(signum):" function.
blocked: number of the blocked signal
"""
code
=
'''if 1:
import signal
import sys
from signal import Signals
def handler(signum, frame):
1/0
%s
blocked = %s
signum = signal.SIGALRM
# child: block and wait the signal
try:
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [blocked])
# Do the tests
test(signum)
# The handler must not be called on unblock
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [blocked])
except ZeroDivisionError:
print("the signal handler has been called",
file=sys.stderr)
sys.exit(1)
except BaseException as err:
print("error: {}".format(err), file=sys.stderr)
sys.stderr.flush()
sys.exit(1)
'''
%
(
test
.
strip
(),
blocked
)
# sig*wait* must be called with the signal blocked: since the current
# process might have several threads running, use a subprocess to have
# a single thread.
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
def
test_sigwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
received = signal.sigwait([signum])
assert isinstance(received, signal.Signals), received
if received != signum:
raise Exception('received %s, not %s' % (received, signum))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwaitinfo'
),
'need signal.sigwaitinfo()'
)
def
test_sigwaitinfo
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigwaitinfo([signum])
if info.si_signo != signum:
raise Exception("info.si_signo != %s" % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigtimedwait([signum], 10.1000)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_poll
(
self
):
# check that polling with sigtimedwait works
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
import os
os.kill(os.getpid(), signum)
info = signal.sigtimedwait([signum], 0)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_timeout
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
received = signal.sigtimedwait([signum], 1.0)
if received is not None:
raise Exception("received=%r" % (received,))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_negative_timeout
(
self
):
signum
=
signal
.
SIGALRM
self
.
assertRaises
(
ValueError
,
signal
.
sigtimedwait
,
[
signum
],
-
1.0
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
@
unittest
.
skipIf
(
threading
is
None
,
"test needs threading module"
)
def
test_sigwait_thread
(
self
):
# Check that calling sigwait() from a thread doesn't suspend the whole
# process. A new interpreter is spawned to avoid problems when mixing
# threads and fork(): only async-safe functions are allowed between
# fork() and exec().
assert_python_ok
(
"-c"
,
"""if True:
import os, threading, sys, time, signal
# the default handler terminates the process
signum = signal.SIGUSR1
def kill_later():
# wait until the main thread is waiting in sigwait()
time.sleep(1)
os.kill(os.getpid(), signum)
# the signal must be blocked by all the threads
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
killer = threading.Thread(target=kill_later)
killer.start()
received = signal.sigwait([signum])
if received != signum:
print("sigwait() received %s, not %s" % (received, signum),
file=sys.stderr)
sys.exit(1)
killer.join()
# unblock the signal, which should have been cleared by sigwait()
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
"""
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask_arguments
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
,
2
,
3
)
self
.
assertRaises
(
OSError
,
signal
.
pthread_sigmask
,
1700
,
[])
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask
(
self
):
code
=
"""if 1:
import signal
import os; import threading
def handler(signum, frame):
1/0
def kill(signum):
os.kill(os.getpid(), signum)
def check_mask(mask):
for sig in mask:
assert isinstance(sig, signal.Signals), repr(sig)
def read_sigmask():
sigmask = signal.pthread_sigmask(signal.SIG_BLOCK, [])
check_mask(sigmask)
return sigmask
signum = signal.SIGUSR1
# Install our signal handler
old_handler = signal.signal(signum, handler)
# Unblock SIGUSR1 (and copy the old mask) to test our signal handler
old_mask = signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
check_mask(old_mask)
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Block and then raise SIGUSR1. The signal is blocked: the signal
# handler is not called, and the signal is now pending
mask = signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
check_mask(mask)
kill(signum)
# Check the new mask
blocked = read_sigmask()
check_mask(blocked)
if signum not in blocked:
raise Exception("%s not in %s" % (signum, blocked))
if old_mask ^ blocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (old_mask, blocked, signum))
# Unblock SIGUSR1
try:
# unblock the pending signal calls immediately the signal handler
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Check the new mask
unblocked = read_sigmask()
if signum in unblocked:
raise Exception("%s in %s" % (signum, unblocked))
if blocked ^ unblocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (blocked, unblocked, signum))
if old_mask != unblocked:
raise Exception("%s != %s" % (old_mask, unblocked))
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
sys
.
platform
==
'freebsd6'
,
"issue #12392: send a signal to the main thread doesn't work "
"before the creation of the first thread on FreeBSD 6"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill_main_thread
(
self
):
# Test that a signal can be sent to the main thread with pthread_kill()
# before any other thread has been created (see issue #12392).
code
=
"""if True:
import threading
import signal
import sys
def handler(signum, frame):
sys.exit(3)
signal.signal(signal.SIGUSR1, handler)
signal.pthread_kill(threading.get_ident(), signal.SIGUSR1)
sys.exit(2)
"""
with
spawn_python
(
'-c'
,
code
)
as
process
:
stdout
,
stderr
=
process
.
communicate
()
exitcode
=
process
.
wait
()
if
exitcode
!=
3
:
raise
Exception
(
"Child error (exit code %s): %s"
%
(
exitcode
,
stdout
))
class
StressTest
(
unittest
.
TestCase
):
"""
Stress signal delivery, especially when a signal arrives in
the middle of recomputing the signal state or executing
previously tripped signal handlers.
"""
def
setsig
(
self
,
signum
,
handler
):
old_handler
=
signal
.
signal
(
signum
,
handler
)
self
.
addCleanup
(
signal
.
signal
,
signum
,
old_handler
)
def
measure_itimer_resolution
(
self
):
N
=
20
times
=
[]
def
handler
(
signum
=
None
,
frame
=
None
):
if
len
(
times
)
<
N
:
times
.
append
(
time
.
perf_counter
())
# 1 µs is the smallest possible timer interval,
# we want to measure what the concrete duration
# will be on this platform
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
)
self
.
addCleanup
(
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
0
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
handler
()
while
len
(
times
)
<
N
:
time
.
sleep
(
1e-3
)
durations
=
[
times
[
i
+
1
]
-
times
[
i
]
for
i
in
range
(
len
(
times
)
-
1
)]
med
=
statistics
.
median
(
durations
)
if
support
.
verbose
:
print
(
"detected median itimer() resolution: %.6f s."
%
(
med
,))
return
med
def
decide_itimer_count
(
self
):
# Some systems have poor setitimer() resolution (for example
# measured around 20 ms. on FreeBSD 9), so decide on a reasonable
# number of sequential timers based on that.
reso
=
self
.
measure_itimer_resolution
()
if
reso
<=
1e-4
:
return
10000
elif
reso
<=
1e-2
:
return
100
else
:
self
.
skipTest
(
"detected itimer resolution (%.3f s.) too high "
"(> 10 ms.) on this platform (or system too busy)"
%
(
reso
,))
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_dependent
(
self
):
"""
This test uses dependent signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
first_handler
(
signum
,
frame
):
# 1e-6 is the minimum non-zero value for `setitimer()`.
# Choose a random delay so as to improve chances of
# triggering a race condition. Ideally the signal is received
# when inside critical signal-handling routines such as
# Py_MakePendingCalls().
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
def
second_handler
(
signum
=
None
,
frame
=
None
):
sigs
.
append
(
signum
)
# Here on Linux, SIGPROF > SIGALRM > SIGUSR1. By using both
# ascending and descending sequences (SIGUSR1 then SIGALRM,
# SIGPROF then SIGALRM), we maximize chances of hitting a bug.
self
.
setsig
(
signal
.
SIGPROF
,
first_handler
)
self
.
setsig
(
signal
.
SIGUSR1
,
first_handler
)
self
.
setsig
(
signal
.
SIGALRM
,
second_handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
time
()
+
15.0
while
expected_sigs
<
N
:
os
.
kill
(
os
.
getpid
(),
signal
.
SIGPROF
)
expected_sigs
+=
1
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
time
()
<
deadline
:
time
.
sleep
(
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
1
while
len
(
sigs
)
<
expected_sigs
and
time
.
time
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_simultaneous
(
self
):
"""
This test uses simultaneous signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
handler
(
signum
,
frame
):
sigs
.
append
(
signum
)
self
.
setsig
(
signal
.
SIGUSR1
,
handler
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
time
()
+
15.0
while
expected_sigs
<
N
:
# Hopefully the SIGALRM will be received somewhere during
# initial processing of SIGUSR1.
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
2
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
time
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
def
tearDownModule
():
support
.
reap_children
()
if
__name__
==
"__main__"
:
unittest
.
main
()
src/greentest/3.7/test_signal.py
0 → 100644
View file @
b7ba2f41
import
os
import
random
import
signal
import
socket
import
statistics
import
subprocess
import
sys
import
threading
import
time
import
unittest
from
test
import
support
from
test.support.script_helper
import
assert_python_ok
,
spawn_python
try
:
import
_testcapi
except
ImportError
:
_testcapi
=
None
class
GenericTests
(
unittest
.
TestCase
):
def
test_enums
(
self
):
for
name
in
dir
(
signal
):
sig
=
getattr
(
signal
,
name
)
if
name
in
{
'SIG_DFL'
,
'SIG_IGN'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Handlers
)
elif
name
in
{
'SIG_BLOCK'
,
'SIG_UNBLOCK'
,
'SIG_SETMASK'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Sigmasks
)
elif
name
.
startswith
(
'SIG'
)
and
not
name
.
startswith
(
'SIG_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
elif
name
.
startswith
(
'CTRL_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
self
.
assertEqual
(
sys
.
platform
,
"win32"
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
PosixTests
(
unittest
.
TestCase
):
def
trivial_signal_handler
(
self
,
*
args
):
pass
def
test_out_of_range_signal_number_raises_error
(
self
):
self
.
assertRaises
(
ValueError
,
signal
.
getsignal
,
4242
)
self
.
assertRaises
(
ValueError
,
signal
.
signal
,
4242
,
self
.
trivial_signal_handler
)
def
test_setting_signal_handler_to_none_raises_error
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
signal
,
signal
.
SIGUSR1
,
None
)
def
test_getsignal
(
self
):
hup
=
signal
.
signal
(
signal
.
SIGHUP
,
self
.
trivial_signal_handler
)
self
.
assertIsInstance
(
hup
,
signal
.
Handlers
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
self
.
trivial_signal_handler
)
signal
.
signal
(
signal
.
SIGHUP
,
hup
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
hup
)
# Issue 3864, unknown if this affects earlier versions of freebsd also
def
test_interprocess_signal
(
self
):
dirname
=
os
.
path
.
dirname
(
__file__
)
script
=
os
.
path
.
join
(
dirname
,
'signalinterproctester.py'
)
assert_python_ok
(
script
)
@
unittest
.
skipUnless
(
sys
.
platform
==
"win32"
,
"Windows specific"
)
class
WindowsSignalTests
(
unittest
.
TestCase
):
def
test_issue9324
(
self
):
# Updated for issue #10003, adding SIGBREAK
handler
=
lambda
x
,
y
:
None
checked
=
set
()
for
sig
in
(
signal
.
SIGABRT
,
signal
.
SIGBREAK
,
signal
.
SIGFPE
,
signal
.
SIGILL
,
signal
.
SIGINT
,
signal
.
SIGSEGV
,
signal
.
SIGTERM
):
# Set and then reset a handler for signals that work on windows.
# Issue #18396, only for signals without a C-level handler.
if
signal
.
getsignal
(
sig
)
is
not
None
:
signal
.
signal
(
sig
,
signal
.
signal
(
sig
,
handler
))
checked
.
add
(
sig
)
# Issue #18396: Ensure the above loop at least tested *something*
self
.
assertTrue
(
checked
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
-
1
,
handler
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
7
,
handler
)
class
WakeupFDTests
(
unittest
.
TestCase
):
def
test_invalid_call
(
self
):
# First parameter is positional-only
with
self
.
assertRaises
(
TypeError
):
signal
.
set_wakeup_fd
(
signum
=
signal
.
SIGINT
)
# warn_on_full_buffer is a keyword-only parameter
with
self
.
assertRaises
(
TypeError
):
signal
.
set_wakeup_fd
(
signal
.
SIGINT
,
False
)
def
test_invalid_fd
(
self
):
fd
=
support
.
make_bad_fd
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_invalid_socket
(
self
):
sock
=
socket
.
socket
()
fd
=
sock
.
fileno
()
sock
.
close
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_set_wakeup_fd_result
(
self
):
r1
,
w1
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r1
)
self
.
addCleanup
(
os
.
close
,
w1
)
r2
,
w2
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r2
)
self
.
addCleanup
(
os
.
close
,
w2
)
if
hasattr
(
os
,
'set_blocking'
):
os
.
set_blocking
(
w1
,
False
)
os
.
set_blocking
(
w2
,
False
)
signal
.
set_wakeup_fd
(
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
w2
),
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
w2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
def
test_set_wakeup_fd_socket_result
(
self
):
sock1
=
socket
.
socket
()
self
.
addCleanup
(
sock1
.
close
)
sock1
.
setblocking
(
False
)
fd1
=
sock1
.
fileno
()
sock2
=
socket
.
socket
()
self
.
addCleanup
(
sock2
.
close
)
sock2
.
setblocking
(
False
)
fd2
=
sock2
.
fileno
()
signal
.
set_wakeup_fd
(
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
fd2
),
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
fd2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
# On Windows, files are always blocking and Windows does not provide a
# function to test if a socket is in non-blocking mode.
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"tests specific to POSIX"
)
def
test_set_wakeup_fd_blocking
(
self
):
rfd
,
wfd
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
rfd
)
self
.
addCleanup
(
os
.
close
,
wfd
)
# fd must be non-blocking
os
.
set_blocking
(
wfd
,
True
)
with
self
.
assertRaises
(
ValueError
)
as
cm
:
signal
.
set_wakeup_fd
(
wfd
)
self
.
assertEqual
(
str
(
cm
.
exception
),
"the fd %s must be in non-blocking mode"
%
wfd
)
# non-blocking is ok
os
.
set_blocking
(
wfd
,
False
)
signal
.
set_wakeup_fd
(
wfd
)
signal
.
set_wakeup_fd
(
-
1
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
WakeupSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
check_wakeup
(
self
,
test_body
,
*
signals
,
ordered
=
True
):
# use a subprocess to have only one thread
code
=
"""if 1:
import _testcapi
import os
import signal
import struct
signals = {!r}
def handler(signum, frame):
pass
def check_signum(signals):
data = os.read(read, len(signals)+1)
raised = struct.unpack('%uB' % len(data), data)
if not {!r}:
raised = set(raised)
signals = set(signals)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
{}
signal.signal(signal.SIGALRM, handler)
read, write = os.pipe()
os.set_blocking(write, False)
signal.set_wakeup_fd(write)
test()
check_signum(signals)
os.close(read)
os.close(write)
"""
.
format
(
tuple
(
map
(
int
,
signals
)),
ordered
,
test_body
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_wakeup_write_error
(
self
):
# Issue #16105: write() errors in the C signal handler should not
# pass silently.
# Use a subprocess to have only one thread.
code
=
"""if 1:
import _testcapi
import errno
import os
import signal
import sys
from test.support import captured_stderr
def handler(signum, frame):
1/0
signal.signal(signal.SIGALRM, handler)
r, w = os.pipe()
os.set_blocking(r, False)
# Set wakeup_fd a read-only file descriptor to trigger the error
signal.set_wakeup_fd(r)
try:
with captured_stderr() as err:
_testcapi.raise_signal(signal.SIGALRM)
except ZeroDivisionError:
# An ignored exception should have been printed out on stderr
err = err.getvalue()
if ('Exception ignored when trying to write to the signal wakeup fd'
not in err):
raise AssertionError(err)
if ('OSError: [Errno %d]' % errno.EBADF) not in err:
raise AssertionError(err)
else:
raise AssertionError("ZeroDivisionError not raised")
os.close(r)
os.close(w)
"""
r
,
w
=
os
.
pipe
()
try
:
os
.
write
(
r
,
b'x'
)
except
OSError
:
pass
else
:
self
.
skipTest
(
"OS doesn't report write() error on the read end of a pipe"
)
finally
:
os
.
close
(
r
)
os
.
close
(
w
)
assert_python_ok
(
'-c'
,
code
)
def
test_wakeup_fd_early
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
# We attempt to get a signal during the sleep,
# before select is called
try:
select.select([], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
before_time = time.monotonic()
select.select([read], [], [], TIMEOUT_FULL)
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_wakeup_fd_during
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
before_time = time.monotonic()
# We attempt to get a signal during the select call
try:
select.select([read], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_signum
(
self
):
self
.
check_wakeup
(
"""def test():
import _testcapi
signal.signal(signal.SIGUSR1, handler)
_testcapi.raise_signal(signal.SIGUSR1)
_testcapi.raise_signal(signal.SIGALRM)
"""
,
signal
.
SIGUSR1
,
signal
.
SIGALRM
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pending
(
self
):
self
.
check_wakeup
(
"""def test():
signum1 = signal.SIGUSR1
signum2 = signal.SIGUSR2
signal.signal(signum1, handler)
signal.signal(signum2, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, (signum1, signum2))
_testcapi.raise_signal(signum1)
_testcapi.raise_signal(signum2)
# Unblocking the 2 signals calls the C signal handler twice
signal.pthread_sigmask(signal.SIG_UNBLOCK, (signum1, signum2))
"""
,
signal
.
SIGUSR1
,
signal
.
SIGUSR2
,
ordered
=
False
)
@
unittest
.
skipUnless
(
hasattr
(
socket
,
'socketpair'
),
'need socket.socketpair'
)
class
WakeupSocketSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_socket
(
self
):
# use a subprocess to have only one thread
code
=
"""if 1:
import signal
import socket
import struct
import _testcapi
signum = signal.SIGINT
signals = (signum,)
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
_testcapi.raise_signal(signum)
data = read.recv(1)
if not data:
raise Exception("no signum written")
raised = struct.unpack('B', data)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
read.close()
write.close()
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_send_error
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
read.setblocking(False)
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
# Close sockets: send() will fail
read.close()
write.close()
with captured_stderr() as err:
_testcapi.raise_signal(signum)
err = err.getvalue()
if ('Exception ignored when trying to {action} to the signal wakeup fd'
not in err):
raise AssertionError(err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_warn_on_full_buffer
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
# This handler will be called, but we intentionally won't read from
# the wakeup fd.
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
# Fill the socketpair buffer
if sys.platform == 'win32':
# bpo-34130: On Windows, sometimes non-blocking send fails to fill
# the full socketpair buffer, so use a timeout of 50 ms instead.
write.settimeout(0.050)
else:
write.setblocking(False)
# Start with large chunk size to reduce the
# number of send needed to fill the buffer.
written = 0
for chunk_size in (2 ** 16, 2 ** 8, 1):
chunk = b"x" * chunk_size
try:
while True:
write.send(chunk)
written += chunk_size
except (BlockingIOError, socket.timeout):
pass
print(f"%s bytes written into the socketpair" % written, flush=True)
write.setblocking(False)
try:
write.send(b"x")
except BlockingIOError:
# The socketpair buffer seems full
pass
else:
raise AssertionError("%s bytes failed to fill the socketpair "
"buffer" % written)
# By default, we get a warning when a signal arrives
msg = ('Exception ignored when trying to {action} '
'to the signal wakeup fd')
signal.set_wakeup_fd(write.fileno())
with captured_stderr() as err:
_testcapi.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("first set_wakeup_fd() test failed, "
"stderr: %r" % err)
# And also if warn_on_full_buffer=True
signal.set_wakeup_fd(write.fileno(), warn_on_full_buffer=True)
with captured_stderr() as err:
_testcapi.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("set_wakeup_fd(warn_on_full_buffer=True) "
"test failed, stderr: %r" % err)
# But not if warn_on_full_buffer=False
signal.set_wakeup_fd(write.fileno(), warn_on_full_buffer=False)
with captured_stderr() as err:
_testcapi.raise_signal(signum)
err = err.getvalue()
if err != "":
raise AssertionError("set_wakeup_fd(warn_on_full_buffer=False) "
"test failed, stderr: %r" % err)
# And then check the default again, to make sure warn_on_full_buffer
# settings don't leak across calls.
signal.set_wakeup_fd(write.fileno())
with captured_stderr() as err:
_testcapi.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("second set_wakeup_fd() test failed, "
"stderr: %r" % err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
SiginterruptTest
(
unittest
.
TestCase
):
def
readpipe_interrupted
(
self
,
interrupt
):
"""Perform a read during which a signal will arrive. Return True if the
read is interrupted by the signal and raises an exception. Return False
if it returns normally.
"""
# use a subprocess to have only one thread, to have a timeout on the
# blocking read and to not touch signal handling in this process
code
=
"""if 1:
import errno
import os
import signal
import sys
interrupt = %r
r, w = os.pipe()
def handler(signum, frame):
1 / 0
signal.signal(signal.SIGALRM, handler)
if interrupt is not None:
signal.siginterrupt(signal.SIGALRM, interrupt)
print("ready")
sys.stdout.flush()
# run the test twice
try:
for loop in range(2):
# send a SIGALRM in a second (during the read)
signal.alarm(1)
try:
# blocking call: read from a pipe without data
os.read(r, 1)
except ZeroDivisionError:
pass
else:
sys.exit(2)
sys.exit(3)
finally:
os.close(r)
os.close(w)
"""
%
(
interrupt
,)
with
spawn_python
(
'-c'
,
code
)
as
process
:
try
:
# wait until the child process is loaded and has started
first_line
=
process
.
stdout
.
readline
()
stdout
,
stderr
=
process
.
communicate
(
timeout
=
5.0
)
except
subprocess
.
TimeoutExpired
:
process
.
kill
()
return
False
else
:
stdout
=
first_line
+
stdout
exitcode
=
process
.
wait
()
if
exitcode
not
in
(
2
,
3
):
raise
Exception
(
"Child error (exit code %s): %r"
%
(
exitcode
,
stdout
))
return
(
exitcode
==
3
)
def
test_without_siginterrupt
(
self
):
# If a signal handler is installed and siginterrupt is not called
# at all, when that signal arrives, it interrupts a syscall that's in
# progress.
interrupted
=
self
.
readpipe_interrupted
(
None
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_on
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a true value for the second argument, when that signal arrives, it
# interrupts a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
True
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_off
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a false value for the second argument, when that signal arrives, it
# does not interrupt a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
False
)
self
.
assertFalse
(
interrupted
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
ItimerTest
(
unittest
.
TestCase
):
def
setUp
(
self
):
self
.
hndl_called
=
False
self
.
hndl_count
=
0
self
.
itimer
=
None
self
.
old_alarm
=
signal
.
signal
(
signal
.
SIGALRM
,
self
.
sig_alrm
)
def
tearDown
(
self
):
signal
.
signal
(
signal
.
SIGALRM
,
self
.
old_alarm
)
if
self
.
itimer
is
not
None
:
# test_itimer_exc doesn't change this attr
# just ensure that itimer is stopped
signal
.
setitimer
(
self
.
itimer
,
0
)
def
sig_alrm
(
self
,
*
args
):
self
.
hndl_called
=
True
def
sig_vtalrm
(
self
,
*
args
):
self
.
hndl_called
=
True
if
self
.
hndl_count
>
3
:
# it shouldn't be here, because it should have been disabled.
raise
signal
.
ItimerError
(
"setitimer didn't disable ITIMER_VIRTUAL "
"timer."
)
elif
self
.
hndl_count
==
3
:
# disable ITIMER_VIRTUAL, this function shouldn't be called anymore
signal
.
setitimer
(
signal
.
ITIMER_VIRTUAL
,
0
)
self
.
hndl_count
+=
1
def
sig_prof
(
self
,
*
args
):
self
.
hndl_called
=
True
signal
.
setitimer
(
signal
.
ITIMER_PROF
,
0
)
def
test_itimer_exc
(
self
):
# XXX I'm assuming -1 is an invalid itimer, but maybe some platform
# defines it ?
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
-
1
,
0
)
# Negative times are treated as zero on some platforms.
if
0
:
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
-
1
)
def
test_itimer_real
(
self
):
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1.0
)
signal
.
pause
()
self
.
assertEqual
(
self
.
hndl_called
,
True
)
# Issue 3864, unknown if this affects earlier versions of freebsd also
@
unittest
.
skipIf
(
sys
.
platform
in
(
'netbsd5'
,),
'itimer not reliable (does not mix well with threading) on some BSDs.'
)
def
test_itimer_virtual
(
self
):
self
.
itimer
=
signal
.
ITIMER_VIRTUAL
signal
.
signal
(
signal
.
SIGVTALRM
,
self
.
sig_vtalrm
)
signal
.
setitimer
(
self
.
itimer
,
0.3
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# use up some virtual time by doing real work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_vtalrm handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# virtual itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_itimer_prof
(
self
):
self
.
itimer
=
signal
.
ITIMER_PROF
signal
.
signal
(
signal
.
SIGPROF
,
self
.
sig_prof
)
signal
.
setitimer
(
self
.
itimer
,
0.2
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# do some work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_prof handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# profiling itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_setitimer_tiny
(
self
):
# bpo-30807: C setitimer() takes a microsecond-resolution interval.
# Check that float -> timeval conversion doesn't round
# the interval down to zero, which would disable the timer.
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1e-6
)
time
.
sleep
(
1
)
self
.
assertEqual
(
self
.
hndl_called
,
True
)
class
PendingSignalsTests
(
unittest
.
TestCase
):
"""
Test pthread_sigmask(), pthread_kill(), sigpending() and sigwait()
functions.
"""
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending_empty
(
self
):
self
.
assertEqual
(
signal
.
sigpending
(),
set
())
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending
(
self
):
code
=
"""if 1:
import os
import signal
def handler(signum, frame):
1/0
signum = signal.SIGUSR1
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
os.kill(os.getpid(), signum)
pending = signal.sigpending()
for sig in pending:
assert isinstance(sig, signal.Signals), repr(pending)
if pending != {signum}:
raise Exception('%s != {%s}' % (pending, signum))
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill
(
self
):
code
=
"""if 1:
import signal
import threading
import sys
signum = signal.SIGUSR1
def handler(signum, frame):
1/0
signal.signal(signum, handler)
tid = threading.get_ident()
try:
signal.pthread_kill(tid, signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
wait_helper
(
self
,
blocked
,
test
):
"""
test: body of the "def test(signum):" function.
blocked: number of the blocked signal
"""
code
=
'''if 1:
import signal
import sys
from signal import Signals
def handler(signum, frame):
1/0
%s
blocked = %s
signum = signal.SIGALRM
# child: block and wait the signal
try:
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [blocked])
# Do the tests
test(signum)
# The handler must not be called on unblock
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [blocked])
except ZeroDivisionError:
print("the signal handler has been called",
file=sys.stderr)
sys.exit(1)
except BaseException as err:
print("error: {}".format(err), file=sys.stderr)
sys.stderr.flush()
sys.exit(1)
'''
%
(
test
.
strip
(),
blocked
)
# sig*wait* must be called with the signal blocked: since the current
# process might have several threads running, use a subprocess to have
# a single thread.
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
def
test_sigwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
received = signal.sigwait([signum])
assert isinstance(received, signal.Signals), received
if received != signum:
raise Exception('received %s, not %s' % (received, signum))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwaitinfo'
),
'need signal.sigwaitinfo()'
)
def
test_sigwaitinfo
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigwaitinfo([signum])
if info.si_signo != signum:
raise Exception("info.si_signo != %s" % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigtimedwait([signum], 10.1000)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_poll
(
self
):
# check that polling with sigtimedwait works
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
import os
os.kill(os.getpid(), signum)
info = signal.sigtimedwait([signum], 0)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_timeout
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
received = signal.sigtimedwait([signum], 1.0)
if received is not None:
raise Exception("received=%r" % (received,))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_negative_timeout
(
self
):
signum
=
signal
.
SIGALRM
self
.
assertRaises
(
ValueError
,
signal
.
sigtimedwait
,
[
signum
],
-
1.0
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_sigwait_thread
(
self
):
# Check that calling sigwait() from a thread doesn't suspend the whole
# process. A new interpreter is spawned to avoid problems when mixing
# threads and fork(): only async-safe functions are allowed between
# fork() and exec().
assert_python_ok
(
"-c"
,
"""if True:
import os, threading, sys, time, signal
# the default handler terminates the process
signum = signal.SIGUSR1
def kill_later():
# wait until the main thread is waiting in sigwait()
time.sleep(1)
os.kill(os.getpid(), signum)
# the signal must be blocked by all the threads
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
killer = threading.Thread(target=kill_later)
killer.start()
received = signal.sigwait([signum])
if received != signum:
print("sigwait() received %s, not %s" % (received, signum),
file=sys.stderr)
sys.exit(1)
killer.join()
# unblock the signal, which should have been cleared by sigwait()
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
"""
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask_arguments
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
,
2
,
3
)
self
.
assertRaises
(
OSError
,
signal
.
pthread_sigmask
,
1700
,
[])
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask
(
self
):
code
=
"""if 1:
import signal
import os; import threading
def handler(signum, frame):
1/0
def kill(signum):
os.kill(os.getpid(), signum)
def check_mask(mask):
for sig in mask:
assert isinstance(sig, signal.Signals), repr(sig)
def read_sigmask():
sigmask = signal.pthread_sigmask(signal.SIG_BLOCK, [])
check_mask(sigmask)
return sigmask
signum = signal.SIGUSR1
# Install our signal handler
old_handler = signal.signal(signum, handler)
# Unblock SIGUSR1 (and copy the old mask) to test our signal handler
old_mask = signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
check_mask(old_mask)
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Block and then raise SIGUSR1. The signal is blocked: the signal
# handler is not called, and the signal is now pending
mask = signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
check_mask(mask)
kill(signum)
# Check the new mask
blocked = read_sigmask()
check_mask(blocked)
if signum not in blocked:
raise Exception("%s not in %s" % (signum, blocked))
if old_mask ^ blocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (old_mask, blocked, signum))
# Unblock SIGUSR1
try:
# unblock the pending signal calls immediately the signal handler
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Check the new mask
unblocked = read_sigmask()
if signum in unblocked:
raise Exception("%s in %s" % (signum, unblocked))
if blocked ^ unblocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (blocked, unblocked, signum))
if old_mask != unblocked:
raise Exception("%s != %s" % (old_mask, unblocked))
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill_main_thread
(
self
):
# Test that a signal can be sent to the main thread with pthread_kill()
# before any other thread has been created (see issue #12392).
code
=
"""if True:
import threading
import signal
import sys
def handler(signum, frame):
sys.exit(3)
signal.signal(signal.SIGUSR1, handler)
signal.pthread_kill(threading.get_ident(), signal.SIGUSR1)
sys.exit(2)
"""
with
spawn_python
(
'-c'
,
code
)
as
process
:
stdout
,
stderr
=
process
.
communicate
()
exitcode
=
process
.
wait
()
if
exitcode
!=
3
:
raise
Exception
(
"Child error (exit code %s): %s"
%
(
exitcode
,
stdout
))
class
StressTest
(
unittest
.
TestCase
):
"""
Stress signal delivery, especially when a signal arrives in
the middle of recomputing the signal state or executing
previously tripped signal handlers.
"""
def
setsig
(
self
,
signum
,
handler
):
old_handler
=
signal
.
signal
(
signum
,
handler
)
self
.
addCleanup
(
signal
.
signal
,
signum
,
old_handler
)
def
measure_itimer_resolution
(
self
):
N
=
20
times
=
[]
def
handler
(
signum
=
None
,
frame
=
None
):
if
len
(
times
)
<
N
:
times
.
append
(
time
.
perf_counter
())
# 1 µs is the smallest possible timer interval,
# we want to measure what the concrete duration
# will be on this platform
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
)
self
.
addCleanup
(
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
0
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
handler
()
while
len
(
times
)
<
N
:
time
.
sleep
(
1e-3
)
durations
=
[
times
[
i
+
1
]
-
times
[
i
]
for
i
in
range
(
len
(
times
)
-
1
)]
med
=
statistics
.
median
(
durations
)
if
support
.
verbose
:
print
(
"detected median itimer() resolution: %.6f s."
%
(
med
,))
return
med
def
decide_itimer_count
(
self
):
# Some systems have poor setitimer() resolution (for example
# measured around 20 ms. on FreeBSD 9), so decide on a reasonable
# number of sequential timers based on that.
reso
=
self
.
measure_itimer_resolution
()
if
reso
<=
1e-4
:
return
10000
elif
reso
<=
1e-2
:
return
100
else
:
self
.
skipTest
(
"detected itimer resolution (%.3f s.) too high "
"(> 10 ms.) on this platform (or system too busy)"
%
(
reso
,))
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_dependent
(
self
):
"""
This test uses dependent signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
first_handler
(
signum
,
frame
):
# 1e-6 is the minimum non-zero value for `setitimer()`.
# Choose a random delay so as to improve chances of
# triggering a race condition. Ideally the signal is received
# when inside critical signal-handling routines such as
# Py_MakePendingCalls().
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
def
second_handler
(
signum
=
None
,
frame
=
None
):
sigs
.
append
(
signum
)
# Here on Linux, SIGPROF > SIGALRM > SIGUSR1. By using both
# ascending and descending sequences (SIGUSR1 then SIGALRM,
# SIGPROF then SIGALRM), we maximize chances of hitting a bug.
self
.
setsig
(
signal
.
SIGPROF
,
first_handler
)
self
.
setsig
(
signal
.
SIGUSR1
,
first_handler
)
self
.
setsig
(
signal
.
SIGALRM
,
second_handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
monotonic
()
+
15.0
while
expected_sigs
<
N
:
os
.
kill
(
os
.
getpid
(),
signal
.
SIGPROF
)
expected_sigs
+=
1
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
1
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_simultaneous
(
self
):
"""
This test uses simultaneous signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
handler
(
signum
,
frame
):
sigs
.
append
(
signum
)
self
.
setsig
(
signal
.
SIGUSR1
,
handler
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
monotonic
()
+
15.0
while
expected_sigs
<
N
:
# Hopefully the SIGALRM will be received somewhere during
# initial processing of SIGUSR1.
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
2
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
def
tearDownModule
():
support
.
reap_children
()
if
__name__
==
"__main__"
:
unittest
.
main
()
src/greentest/3.8/test_signal.py
0 → 100644
View file @
b7ba2f41
import
errno
import
os
import
random
import
signal
import
socket
import
statistics
import
subprocess
import
sys
import
threading
import
time
import
unittest
from
test
import
support
from
test.support.script_helper
import
assert_python_ok
,
spawn_python
try
:
import
_testcapi
except
ImportError
:
_testcapi
=
None
class
GenericTests
(
unittest
.
TestCase
):
def
test_enums
(
self
):
for
name
in
dir
(
signal
):
sig
=
getattr
(
signal
,
name
)
if
name
in
{
'SIG_DFL'
,
'SIG_IGN'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Handlers
)
elif
name
in
{
'SIG_BLOCK'
,
'SIG_UNBLOCK'
,
'SIG_SETMASK'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Sigmasks
)
elif
name
.
startswith
(
'SIG'
)
and
not
name
.
startswith
(
'SIG_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
elif
name
.
startswith
(
'CTRL_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
self
.
assertEqual
(
sys
.
platform
,
"win32"
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
PosixTests
(
unittest
.
TestCase
):
def
trivial_signal_handler
(
self
,
*
args
):
pass
def
test_out_of_range_signal_number_raises_error
(
self
):
self
.
assertRaises
(
ValueError
,
signal
.
getsignal
,
4242
)
self
.
assertRaises
(
ValueError
,
signal
.
signal
,
4242
,
self
.
trivial_signal_handler
)
self
.
assertRaises
(
ValueError
,
signal
.
strsignal
,
4242
)
def
test_setting_signal_handler_to_none_raises_error
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
signal
,
signal
.
SIGUSR1
,
None
)
def
test_getsignal
(
self
):
hup
=
signal
.
signal
(
signal
.
SIGHUP
,
self
.
trivial_signal_handler
)
self
.
assertIsInstance
(
hup
,
signal
.
Handlers
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
self
.
trivial_signal_handler
)
signal
.
signal
(
signal
.
SIGHUP
,
hup
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
hup
)
def
test_strsignal
(
self
):
self
.
assertIn
(
"Interrupt"
,
signal
.
strsignal
(
signal
.
SIGINT
))
self
.
assertIn
(
"Terminated"
,
signal
.
strsignal
(
signal
.
SIGTERM
))
self
.
assertIn
(
"Hangup"
,
signal
.
strsignal
(
signal
.
SIGHUP
))
# Issue 3864, unknown if this affects earlier versions of freebsd also
def
test_interprocess_signal
(
self
):
dirname
=
os
.
path
.
dirname
(
__file__
)
script
=
os
.
path
.
join
(
dirname
,
'signalinterproctester.py'
)
assert_python_ok
(
script
)
def
test_valid_signals
(
self
):
s
=
signal
.
valid_signals
()
self
.
assertIsInstance
(
s
,
set
)
self
.
assertIn
(
signal
.
Signals
.
SIGINT
,
s
)
self
.
assertIn
(
signal
.
Signals
.
SIGALRM
,
s
)
self
.
assertNotIn
(
0
,
s
)
self
.
assertNotIn
(
signal
.
NSIG
,
s
)
self
.
assertLess
(
len
(
s
),
signal
.
NSIG
)
@
unittest
.
skipUnless
(
sys
.
executable
,
"sys.executable required."
)
def
test_keyboard_interrupt_exit_code
(
self
):
"""KeyboardInterrupt triggers exit via SIGINT."""
process
=
subprocess
.
run
(
[
sys
.
executable
,
"-c"
,
"import os, signal, time
\
n
"
"os.kill(os.getpid(), signal.SIGINT)
\
n
"
"for _ in range(999): time.sleep(0.01)"
],
stderr
=
subprocess
.
PIPE
)
self
.
assertIn
(
b"KeyboardInterrupt"
,
process
.
stderr
)
self
.
assertEqual
(
process
.
returncode
,
-
signal
.
SIGINT
)
# Caveat: The exit code is insufficient to guarantee we actually died
# via a signal. POSIX shells do more than look at the 8 bit value.
# Writing an automation friendly test of an interactive shell
# to confirm that our process died via a SIGINT proved too complex.
@
unittest
.
skipUnless
(
sys
.
platform
==
"win32"
,
"Windows specific"
)
class
WindowsSignalTests
(
unittest
.
TestCase
):
def
test_valid_signals
(
self
):
s
=
signal
.
valid_signals
()
self
.
assertIsInstance
(
s
,
set
)
self
.
assertGreaterEqual
(
len
(
s
),
6
)
self
.
assertIn
(
signal
.
Signals
.
SIGINT
,
s
)
self
.
assertNotIn
(
0
,
s
)
self
.
assertNotIn
(
signal
.
NSIG
,
s
)
self
.
assertLess
(
len
(
s
),
signal
.
NSIG
)
def
test_issue9324
(
self
):
# Updated for issue #10003, adding SIGBREAK
handler
=
lambda
x
,
y
:
None
checked
=
set
()
for
sig
in
(
signal
.
SIGABRT
,
signal
.
SIGBREAK
,
signal
.
SIGFPE
,
signal
.
SIGILL
,
signal
.
SIGINT
,
signal
.
SIGSEGV
,
signal
.
SIGTERM
):
# Set and then reset a handler for signals that work on windows.
# Issue #18396, only for signals without a C-level handler.
if
signal
.
getsignal
(
sig
)
is
not
None
:
signal
.
signal
(
sig
,
signal
.
signal
(
sig
,
handler
))
checked
.
add
(
sig
)
# Issue #18396: Ensure the above loop at least tested *something*
self
.
assertTrue
(
checked
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
-
1
,
handler
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
7
,
handler
)
@
unittest
.
skipUnless
(
sys
.
executable
,
"sys.executable required."
)
def
test_keyboard_interrupt_exit_code
(
self
):
"""KeyboardInterrupt triggers an exit using STATUS_CONTROL_C_EXIT."""
# We don't test via os.kill(os.getpid(), signal.CTRL_C_EVENT) here
# as that requires setting up a console control handler in a child
# in its own process group. Doable, but quite complicated. (see
# @eryksun on https://github.com/python/cpython/pull/11862)
process
=
subprocess
.
run
(
[
sys
.
executable
,
"-c"
,
"raise KeyboardInterrupt"
],
stderr
=
subprocess
.
PIPE
)
self
.
assertIn
(
b"KeyboardInterrupt"
,
process
.
stderr
)
STATUS_CONTROL_C_EXIT
=
0xC000013A
self
.
assertEqual
(
process
.
returncode
,
STATUS_CONTROL_C_EXIT
)
class
WakeupFDTests
(
unittest
.
TestCase
):
def
test_invalid_call
(
self
):
# First parameter is positional-only
with
self
.
assertRaises
(
TypeError
):
signal
.
set_wakeup_fd
(
signum
=
signal
.
SIGINT
)
# warn_on_full_buffer is a keyword-only parameter
with
self
.
assertRaises
(
TypeError
):
signal
.
set_wakeup_fd
(
signal
.
SIGINT
,
False
)
def
test_invalid_fd
(
self
):
fd
=
support
.
make_bad_fd
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_invalid_socket
(
self
):
sock
=
socket
.
socket
()
fd
=
sock
.
fileno
()
sock
.
close
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_set_wakeup_fd_result
(
self
):
r1
,
w1
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r1
)
self
.
addCleanup
(
os
.
close
,
w1
)
r2
,
w2
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r2
)
self
.
addCleanup
(
os
.
close
,
w2
)
if
hasattr
(
os
,
'set_blocking'
):
os
.
set_blocking
(
w1
,
False
)
os
.
set_blocking
(
w2
,
False
)
signal
.
set_wakeup_fd
(
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
w2
),
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
w2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
def
test_set_wakeup_fd_socket_result
(
self
):
sock1
=
socket
.
socket
()
self
.
addCleanup
(
sock1
.
close
)
sock1
.
setblocking
(
False
)
fd1
=
sock1
.
fileno
()
sock2
=
socket
.
socket
()
self
.
addCleanup
(
sock2
.
close
)
sock2
.
setblocking
(
False
)
fd2
=
sock2
.
fileno
()
signal
.
set_wakeup_fd
(
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
fd2
),
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
fd2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
# On Windows, files are always blocking and Windows does not provide a
# function to test if a socket is in non-blocking mode.
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"tests specific to POSIX"
)
def
test_set_wakeup_fd_blocking
(
self
):
rfd
,
wfd
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
rfd
)
self
.
addCleanup
(
os
.
close
,
wfd
)
# fd must be non-blocking
os
.
set_blocking
(
wfd
,
True
)
with
self
.
assertRaises
(
ValueError
)
as
cm
:
signal
.
set_wakeup_fd
(
wfd
)
self
.
assertEqual
(
str
(
cm
.
exception
),
"the fd %s must be in non-blocking mode"
%
wfd
)
# non-blocking is ok
os
.
set_blocking
(
wfd
,
False
)
signal
.
set_wakeup_fd
(
wfd
)
signal
.
set_wakeup_fd
(
-
1
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
WakeupSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
check_wakeup
(
self
,
test_body
,
*
signals
,
ordered
=
True
):
# use a subprocess to have only one thread
code
=
"""if 1:
import _testcapi
import os
import signal
import struct
signals = {!r}
def handler(signum, frame):
pass
def check_signum(signals):
data = os.read(read, len(signals)+1)
raised = struct.unpack('%uB' % len(data), data)
if not {!r}:
raised = set(raised)
signals = set(signals)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
{}
signal.signal(signal.SIGALRM, handler)
read, write = os.pipe()
os.set_blocking(write, False)
signal.set_wakeup_fd(write)
test()
check_signum(signals)
os.close(read)
os.close(write)
"""
.
format
(
tuple
(
map
(
int
,
signals
)),
ordered
,
test_body
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_wakeup_write_error
(
self
):
# Issue #16105: write() errors in the C signal handler should not
# pass silently.
# Use a subprocess to have only one thread.
code
=
"""if 1:
import _testcapi
import errno
import os
import signal
import sys
from test.support import captured_stderr
def handler(signum, frame):
1/0
signal.signal(signal.SIGALRM, handler)
r, w = os.pipe()
os.set_blocking(r, False)
# Set wakeup_fd a read-only file descriptor to trigger the error
signal.set_wakeup_fd(r)
try:
with captured_stderr() as err:
signal.raise_signal(signal.SIGALRM)
except ZeroDivisionError:
# An ignored exception should have been printed out on stderr
err = err.getvalue()
if ('Exception ignored when trying to write to the signal wakeup fd'
not in err):
raise AssertionError(err)
if ('OSError: [Errno %d]' % errno.EBADF) not in err:
raise AssertionError(err)
else:
raise AssertionError("ZeroDivisionError not raised")
os.close(r)
os.close(w)
"""
r
,
w
=
os
.
pipe
()
try
:
os
.
write
(
r
,
b'x'
)
except
OSError
:
pass
else
:
self
.
skipTest
(
"OS doesn't report write() error on the read end of a pipe"
)
finally
:
os
.
close
(
r
)
os
.
close
(
w
)
assert_python_ok
(
'-c'
,
code
)
def
test_wakeup_fd_early
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
# We attempt to get a signal during the sleep,
# before select is called
try:
select.select([], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
before_time = time.monotonic()
select.select([read], [], [], TIMEOUT_FULL)
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_wakeup_fd_during
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
before_time = time.monotonic()
# We attempt to get a signal during the select call
try:
select.select([read], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_signum
(
self
):
self
.
check_wakeup
(
"""def test():
signal.signal(signal.SIGUSR1, handler)
signal.raise_signal(signal.SIGUSR1)
signal.raise_signal(signal.SIGALRM)
"""
,
signal
.
SIGUSR1
,
signal
.
SIGALRM
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pending
(
self
):
self
.
check_wakeup
(
"""def test():
signum1 = signal.SIGUSR1
signum2 = signal.SIGUSR2
signal.signal(signum1, handler)
signal.signal(signum2, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, (signum1, signum2))
signal.raise_signal(signum1)
signal.raise_signal(signum2)
# Unblocking the 2 signals calls the C signal handler twice
signal.pthread_sigmask(signal.SIG_UNBLOCK, (signum1, signum2))
"""
,
signal
.
SIGUSR1
,
signal
.
SIGUSR2
,
ordered
=
False
)
@
unittest
.
skipUnless
(
hasattr
(
socket
,
'socketpair'
),
'need socket.socketpair'
)
class
WakeupSocketSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_socket
(
self
):
# use a subprocess to have only one thread
code
=
"""if 1:
import signal
import socket
import struct
import _testcapi
signum = signal.SIGINT
signals = (signum,)
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
signal.raise_signal(signum)
data = read.recv(1)
if not data:
raise Exception("no signum written")
raised = struct.unpack('B', data)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
read.close()
write.close()
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_send_error
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
read.setblocking(False)
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
# Close sockets: send() will fail
read.close()
write.close()
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if ('Exception ignored when trying to {action} to the signal wakeup fd'
not in err):
raise AssertionError(err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_warn_on_full_buffer
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
# This handler will be called, but we intentionally won't read from
# the wakeup fd.
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
# Fill the socketpair buffer
if sys.platform == 'win32':
# bpo-34130: On Windows, sometimes non-blocking send fails to fill
# the full socketpair buffer, so use a timeout of 50 ms instead.
write.settimeout(0.050)
else:
write.setblocking(False)
# Start with large chunk size to reduce the
# number of send needed to fill the buffer.
written = 0
for chunk_size in (2 ** 16, 2 ** 8, 1):
chunk = b"x" * chunk_size
try:
while True:
write.send(chunk)
written += chunk_size
except (BlockingIOError, socket.timeout):
pass
print(f"%s bytes written into the socketpair" % written, flush=True)
write.setblocking(False)
try:
write.send(b"x")
except BlockingIOError:
# The socketpair buffer seems full
pass
else:
raise AssertionError("%s bytes failed to fill the socketpair "
"buffer" % written)
# By default, we get a warning when a signal arrives
msg = ('Exception ignored when trying to {action} '
'to the signal wakeup fd')
signal.set_wakeup_fd(write.fileno())
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("first set_wakeup_fd() test failed, "
"stderr: %r" % err)
# And also if warn_on_full_buffer=True
signal.set_wakeup_fd(write.fileno(), warn_on_full_buffer=True)
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("set_wakeup_fd(warn_on_full_buffer=True) "
"test failed, stderr: %r" % err)
# But not if warn_on_full_buffer=False
signal.set_wakeup_fd(write.fileno(), warn_on_full_buffer=False)
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if err != "":
raise AssertionError("set_wakeup_fd(warn_on_full_buffer=False) "
"test failed, stderr: %r" % err)
# And then check the default again, to make sure warn_on_full_buffer
# settings don't leak across calls.
signal.set_wakeup_fd(write.fileno())
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("second set_wakeup_fd() test failed, "
"stderr: %r" % err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
SiginterruptTest
(
unittest
.
TestCase
):
def
readpipe_interrupted
(
self
,
interrupt
):
"""Perform a read during which a signal will arrive. Return True if the
read is interrupted by the signal and raises an exception. Return False
if it returns normally.
"""
# use a subprocess to have only one thread, to have a timeout on the
# blocking read and to not touch signal handling in this process
code
=
"""if 1:
import errno
import os
import signal
import sys
interrupt = %r
r, w = os.pipe()
def handler(signum, frame):
1 / 0
signal.signal(signal.SIGALRM, handler)
if interrupt is not None:
signal.siginterrupt(signal.SIGALRM, interrupt)
print("ready")
sys.stdout.flush()
# run the test twice
try:
for loop in range(2):
# send a SIGALRM in a second (during the read)
signal.alarm(1)
try:
# blocking call: read from a pipe without data
os.read(r, 1)
except ZeroDivisionError:
pass
else:
sys.exit(2)
sys.exit(3)
finally:
os.close(r)
os.close(w)
"""
%
(
interrupt
,)
with
spawn_python
(
'-c'
,
code
)
as
process
:
try
:
# wait until the child process is loaded and has started
first_line
=
process
.
stdout
.
readline
()
stdout
,
stderr
=
process
.
communicate
(
timeout
=
5.0
)
except
subprocess
.
TimeoutExpired
:
process
.
kill
()
return
False
else
:
stdout
=
first_line
+
stdout
exitcode
=
process
.
wait
()
if
exitcode
not
in
(
2
,
3
):
raise
Exception
(
"Child error (exit code %s): %r"
%
(
exitcode
,
stdout
))
return
(
exitcode
==
3
)
def
test_without_siginterrupt
(
self
):
# If a signal handler is installed and siginterrupt is not called
# at all, when that signal arrives, it interrupts a syscall that's in
# progress.
interrupted
=
self
.
readpipe_interrupted
(
None
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_on
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a true value for the second argument, when that signal arrives, it
# interrupts a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
True
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_off
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a false value for the second argument, when that signal arrives, it
# does not interrupt a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
False
)
self
.
assertFalse
(
interrupted
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
ItimerTest
(
unittest
.
TestCase
):
def
setUp
(
self
):
self
.
hndl_called
=
False
self
.
hndl_count
=
0
self
.
itimer
=
None
self
.
old_alarm
=
signal
.
signal
(
signal
.
SIGALRM
,
self
.
sig_alrm
)
def
tearDown
(
self
):
signal
.
signal
(
signal
.
SIGALRM
,
self
.
old_alarm
)
if
self
.
itimer
is
not
None
:
# test_itimer_exc doesn't change this attr
# just ensure that itimer is stopped
signal
.
setitimer
(
self
.
itimer
,
0
)
def
sig_alrm
(
self
,
*
args
):
self
.
hndl_called
=
True
def
sig_vtalrm
(
self
,
*
args
):
self
.
hndl_called
=
True
if
self
.
hndl_count
>
3
:
# it shouldn't be here, because it should have been disabled.
raise
signal
.
ItimerError
(
"setitimer didn't disable ITIMER_VIRTUAL "
"timer."
)
elif
self
.
hndl_count
==
3
:
# disable ITIMER_VIRTUAL, this function shouldn't be called anymore
signal
.
setitimer
(
signal
.
ITIMER_VIRTUAL
,
0
)
self
.
hndl_count
+=
1
def
sig_prof
(
self
,
*
args
):
self
.
hndl_called
=
True
signal
.
setitimer
(
signal
.
ITIMER_PROF
,
0
)
def
test_itimer_exc
(
self
):
# XXX I'm assuming -1 is an invalid itimer, but maybe some platform
# defines it ?
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
-
1
,
0
)
# Negative times are treated as zero on some platforms.
if
0
:
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
-
1
)
def
test_itimer_real
(
self
):
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1.0
)
signal
.
pause
()
self
.
assertEqual
(
self
.
hndl_called
,
True
)
# Issue 3864, unknown if this affects earlier versions of freebsd also
@
unittest
.
skipIf
(
sys
.
platform
in
(
'netbsd5'
,),
'itimer not reliable (does not mix well with threading) on some BSDs.'
)
def
test_itimer_virtual
(
self
):
self
.
itimer
=
signal
.
ITIMER_VIRTUAL
signal
.
signal
(
signal
.
SIGVTALRM
,
self
.
sig_vtalrm
)
signal
.
setitimer
(
self
.
itimer
,
0.3
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# use up some virtual time by doing real work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_vtalrm handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# virtual itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_itimer_prof
(
self
):
self
.
itimer
=
signal
.
ITIMER_PROF
signal
.
signal
(
signal
.
SIGPROF
,
self
.
sig_prof
)
signal
.
setitimer
(
self
.
itimer
,
0.2
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# do some work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_prof handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# profiling itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_setitimer_tiny
(
self
):
# bpo-30807: C setitimer() takes a microsecond-resolution interval.
# Check that float -> timeval conversion doesn't round
# the interval down to zero, which would disable the timer.
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1e-6
)
time
.
sleep
(
1
)
self
.
assertEqual
(
self
.
hndl_called
,
True
)
class
PendingSignalsTests
(
unittest
.
TestCase
):
"""
Test pthread_sigmask(), pthread_kill(), sigpending() and sigwait()
functions.
"""
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending_empty
(
self
):
self
.
assertEqual
(
signal
.
sigpending
(),
set
())
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending
(
self
):
code
=
"""if 1:
import os
import signal
def handler(signum, frame):
1/0
signum = signal.SIGUSR1
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
os.kill(os.getpid(), signum)
pending = signal.sigpending()
for sig in pending:
assert isinstance(sig, signal.Signals), repr(pending)
if pending != {signum}:
raise Exception('%s != {%s}' % (pending, signum))
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill
(
self
):
code
=
"""if 1:
import signal
import threading
import sys
signum = signal.SIGUSR1
def handler(signum, frame):
1/0
signal.signal(signum, handler)
tid = threading.get_ident()
try:
signal.pthread_kill(tid, signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
wait_helper
(
self
,
blocked
,
test
):
"""
test: body of the "def test(signum):" function.
blocked: number of the blocked signal
"""
code
=
'''if 1:
import signal
import sys
from signal import Signals
def handler(signum, frame):
1/0
%s
blocked = %s
signum = signal.SIGALRM
# child: block and wait the signal
try:
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [blocked])
# Do the tests
test(signum)
# The handler must not be called on unblock
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [blocked])
except ZeroDivisionError:
print("the signal handler has been called",
file=sys.stderr)
sys.exit(1)
except BaseException as err:
print("error: {}".format(err), file=sys.stderr)
sys.stderr.flush()
sys.exit(1)
'''
%
(
test
.
strip
(),
blocked
)
# sig*wait* must be called with the signal blocked: since the current
# process might have several threads running, use a subprocess to have
# a single thread.
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
def
test_sigwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
received = signal.sigwait([signum])
assert isinstance(received, signal.Signals), received
if received != signum:
raise Exception('received %s, not %s' % (received, signum))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwaitinfo'
),
'need signal.sigwaitinfo()'
)
def
test_sigwaitinfo
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigwaitinfo([signum])
if info.si_signo != signum:
raise Exception("info.si_signo != %s" % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigtimedwait([signum], 10.1000)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_poll
(
self
):
# check that polling with sigtimedwait works
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
import os
os.kill(os.getpid(), signum)
info = signal.sigtimedwait([signum], 0)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_timeout
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
received = signal.sigtimedwait([signum], 1.0)
if received is not None:
raise Exception("received=%r" % (received,))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_negative_timeout
(
self
):
signum
=
signal
.
SIGALRM
self
.
assertRaises
(
ValueError
,
signal
.
sigtimedwait
,
[
signum
],
-
1.0
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_sigwait_thread
(
self
):
# Check that calling sigwait() from a thread doesn't suspend the whole
# process. A new interpreter is spawned to avoid problems when mixing
# threads and fork(): only async-safe functions are allowed between
# fork() and exec().
assert_python_ok
(
"-c"
,
"""if True:
import os, threading, sys, time, signal
# the default handler terminates the process
signum = signal.SIGUSR1
def kill_later():
# wait until the main thread is waiting in sigwait()
time.sleep(1)
os.kill(os.getpid(), signum)
# the signal must be blocked by all the threads
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
killer = threading.Thread(target=kill_later)
killer.start()
received = signal.sigwait([signum])
if received != signum:
print("sigwait() received %s, not %s" % (received, signum),
file=sys.stderr)
sys.exit(1)
killer.join()
# unblock the signal, which should have been cleared by sigwait()
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
"""
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask_arguments
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
,
2
,
3
)
self
.
assertRaises
(
OSError
,
signal
.
pthread_sigmask
,
1700
,
[])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
signal
.
NSIG
])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
0
])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
1
<<
1000
])
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask_valid_signals
(
self
):
s
=
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
signal
.
valid_signals
())
self
.
addCleanup
(
signal
.
pthread_sigmask
,
signal
.
SIG_SETMASK
,
s
)
# Get current blocked set
s
=
signal
.
pthread_sigmask
(
signal
.
SIG_UNBLOCK
,
signal
.
valid_signals
())
self
.
assertLessEqual
(
s
,
signal
.
valid_signals
())
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask
(
self
):
code
=
"""if 1:
import signal
import os; import threading
def handler(signum, frame):
1/0
def kill(signum):
os.kill(os.getpid(), signum)
def check_mask(mask):
for sig in mask:
assert isinstance(sig, signal.Signals), repr(sig)
def read_sigmask():
sigmask = signal.pthread_sigmask(signal.SIG_BLOCK, [])
check_mask(sigmask)
return sigmask
signum = signal.SIGUSR1
# Install our signal handler
old_handler = signal.signal(signum, handler)
# Unblock SIGUSR1 (and copy the old mask) to test our signal handler
old_mask = signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
check_mask(old_mask)
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Block and then raise SIGUSR1. The signal is blocked: the signal
# handler is not called, and the signal is now pending
mask = signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
check_mask(mask)
kill(signum)
# Check the new mask
blocked = read_sigmask()
check_mask(blocked)
if signum not in blocked:
raise Exception("%s not in %s" % (signum, blocked))
if old_mask ^ blocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (old_mask, blocked, signum))
# Unblock SIGUSR1
try:
# unblock the pending signal calls immediately the signal handler
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Check the new mask
unblocked = read_sigmask()
if signum in unblocked:
raise Exception("%s in %s" % (signum, unblocked))
if blocked ^ unblocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (blocked, unblocked, signum))
if old_mask != unblocked:
raise Exception("%s != %s" % (old_mask, unblocked))
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill_main_thread
(
self
):
# Test that a signal can be sent to the main thread with pthread_kill()
# before any other thread has been created (see issue #12392).
code
=
"""if True:
import threading
import signal
import sys
def handler(signum, frame):
sys.exit(3)
signal.signal(signal.SIGUSR1, handler)
signal.pthread_kill(threading.get_ident(), signal.SIGUSR1)
sys.exit(2)
"""
with
spawn_python
(
'-c'
,
code
)
as
process
:
stdout
,
stderr
=
process
.
communicate
()
exitcode
=
process
.
wait
()
if
exitcode
!=
3
:
raise
Exception
(
"Child error (exit code %s): %s"
%
(
exitcode
,
stdout
))
class
StressTest
(
unittest
.
TestCase
):
"""
Stress signal delivery, especially when a signal arrives in
the middle of recomputing the signal state or executing
previously tripped signal handlers.
"""
def
setsig
(
self
,
signum
,
handler
):
old_handler
=
signal
.
signal
(
signum
,
handler
)
self
.
addCleanup
(
signal
.
signal
,
signum
,
old_handler
)
def
measure_itimer_resolution
(
self
):
N
=
20
times
=
[]
def
handler
(
signum
=
None
,
frame
=
None
):
if
len
(
times
)
<
N
:
times
.
append
(
time
.
perf_counter
())
# 1 µs is the smallest possible timer interval,
# we want to measure what the concrete duration
# will be on this platform
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
)
self
.
addCleanup
(
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
0
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
handler
()
while
len
(
times
)
<
N
:
time
.
sleep
(
1e-3
)
durations
=
[
times
[
i
+
1
]
-
times
[
i
]
for
i
in
range
(
len
(
times
)
-
1
)]
med
=
statistics
.
median
(
durations
)
if
support
.
verbose
:
print
(
"detected median itimer() resolution: %.6f s."
%
(
med
,))
return
med
def
decide_itimer_count
(
self
):
# Some systems have poor setitimer() resolution (for example
# measured around 20 ms. on FreeBSD 9), so decide on a reasonable
# number of sequential timers based on that.
reso
=
self
.
measure_itimer_resolution
()
if
reso
<=
1e-4
:
return
10000
elif
reso
<=
1e-2
:
return
100
else
:
self
.
skipTest
(
"detected itimer resolution (%.3f s.) too high "
"(> 10 ms.) on this platform (or system too busy)"
%
(
reso
,))
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_dependent
(
self
):
"""
This test uses dependent signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
first_handler
(
signum
,
frame
):
# 1e-6 is the minimum non-zero value for `setitimer()`.
# Choose a random delay so as to improve chances of
# triggering a race condition. Ideally the signal is received
# when inside critical signal-handling routines such as
# Py_MakePendingCalls().
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
def
second_handler
(
signum
=
None
,
frame
=
None
):
sigs
.
append
(
signum
)
# Here on Linux, SIGPROF > SIGALRM > SIGUSR1. By using both
# ascending and descending sequences (SIGUSR1 then SIGALRM,
# SIGPROF then SIGALRM), we maximize chances of hitting a bug.
self
.
setsig
(
signal
.
SIGPROF
,
first_handler
)
self
.
setsig
(
signal
.
SIGUSR1
,
first_handler
)
self
.
setsig
(
signal
.
SIGALRM
,
second_handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
monotonic
()
+
15.0
while
expected_sigs
<
N
:
os
.
kill
(
os
.
getpid
(),
signal
.
SIGPROF
)
expected_sigs
+=
1
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
1
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_simultaneous
(
self
):
"""
This test uses simultaneous signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
handler
(
signum
,
frame
):
sigs
.
append
(
signum
)
self
.
setsig
(
signal
.
SIGUSR1
,
handler
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
monotonic
()
+
15.0
while
expected_sigs
<
N
:
# Hopefully the SIGALRM will be received somewhere during
# initial processing of SIGUSR1.
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
2
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"SIGUSR1"
),
"test needs SIGUSR1"
)
def
test_stress_modifying_handlers
(
self
):
# bpo-43406: race condition between trip_signal() and signal.signal
signum
=
signal
.
SIGUSR1
num_sent_signals
=
0
num_received_signals
=
0
do_stop
=
False
def
custom_handler
(
signum
,
frame
):
nonlocal
num_received_signals
num_received_signals
+=
1
def
set_interrupts
():
nonlocal
num_sent_signals
while
not
do_stop
:
signal
.
raise_signal
(
signum
)
num_sent_signals
+=
1
def
cycle_handlers
():
while
num_sent_signals
<
100
:
for
i
in
range
(
20000
):
# Cycle between a Python-defined and a non-Python handler
for
handler
in
[
custom_handler
,
signal
.
SIG_IGN
]:
signal
.
signal
(
signum
,
handler
)
old_handler
=
signal
.
signal
(
signum
,
custom_handler
)
self
.
addCleanup
(
signal
.
signal
,
signum
,
old_handler
)
t
=
threading
.
Thread
(
target
=
set_interrupts
)
try
:
ignored
=
False
with
support
.
catch_unraisable_exception
()
as
cm
:
t
.
start
()
cycle_handlers
()
do_stop
=
True
t
.
join
()
if
cm
.
unraisable
is
not
None
:
# An unraisable exception may be printed out when
# a signal is ignored due to the aforementioned
# race condition, check it.
self
.
assertIsInstance
(
cm
.
unraisable
.
exc_value
,
OSError
)
self
.
assertIn
(
f"Signal
{
signum
}
ignored due to race condition"
,
str
(
cm
.
unraisable
.
exc_value
))
ignored
=
True
# bpo-43406: Even if it is unlikely, it's technically possible that
# all signals were ignored because of race conditions.
if
not
ignored
:
# Sanity check that some signals were received, but not all
self
.
assertGreater
(
num_received_signals
,
0
)
self
.
assertLess
(
num_received_signals
,
num_sent_signals
)
finally
:
do_stop
=
True
t
.
join
()
class
RaiseSignalTest
(
unittest
.
TestCase
):
def
test_sigint
(
self
):
with
self
.
assertRaises
(
KeyboardInterrupt
):
signal
.
raise_signal
(
signal
.
SIGINT
)
@
unittest
.
skipIf
(
sys
.
platform
!=
"win32"
,
"Windows specific test"
)
def
test_invalid_argument
(
self
):
try
:
SIGHUP
=
1
# not supported on win32
signal
.
raise_signal
(
SIGHUP
)
self
.
fail
(
"OSError (Invalid argument) expected"
)
except
OSError
as
e
:
if
e
.
errno
==
errno
.
EINVAL
:
pass
else
:
raise
def
test_handler
(
self
):
is_ok
=
False
def
handler
(
a
,
b
):
nonlocal
is_ok
is_ok
=
True
old_signal
=
signal
.
signal
(
signal
.
SIGINT
,
handler
)
self
.
addCleanup
(
signal
.
signal
,
signal
.
SIGINT
,
old_signal
)
signal
.
raise_signal
(
signal
.
SIGINT
)
self
.
assertTrue
(
is_ok
)
def
tearDownModule
():
support
.
reap_children
()
if
__name__
==
"__main__"
:
unittest
.
main
()
src/greentest/3.9/test_signal.py
0 → 100644
View file @
b7ba2f41
import
errno
import
os
import
random
import
signal
import
socket
import
statistics
import
subprocess
import
sys
import
threading
import
time
import
unittest
from
test
import
support
from
test.support.script_helper
import
assert_python_ok
,
spawn_python
try
:
import
_testcapi
except
ImportError
:
_testcapi
=
None
class
GenericTests
(
unittest
.
TestCase
):
def
test_enums
(
self
):
for
name
in
dir
(
signal
):
sig
=
getattr
(
signal
,
name
)
if
name
in
{
'SIG_DFL'
,
'SIG_IGN'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Handlers
)
elif
name
in
{
'SIG_BLOCK'
,
'SIG_UNBLOCK'
,
'SIG_SETMASK'
}:
self
.
assertIsInstance
(
sig
,
signal
.
Sigmasks
)
elif
name
.
startswith
(
'SIG'
)
and
not
name
.
startswith
(
'SIG_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
elif
name
.
startswith
(
'CTRL_'
):
self
.
assertIsInstance
(
sig
,
signal
.
Signals
)
self
.
assertEqual
(
sys
.
platform
,
"win32"
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
PosixTests
(
unittest
.
TestCase
):
def
trivial_signal_handler
(
self
,
*
args
):
pass
def
test_out_of_range_signal_number_raises_error
(
self
):
self
.
assertRaises
(
ValueError
,
signal
.
getsignal
,
4242
)
self
.
assertRaises
(
ValueError
,
signal
.
signal
,
4242
,
self
.
trivial_signal_handler
)
self
.
assertRaises
(
ValueError
,
signal
.
strsignal
,
4242
)
def
test_setting_signal_handler_to_none_raises_error
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
signal
,
signal
.
SIGUSR1
,
None
)
def
test_getsignal
(
self
):
hup
=
signal
.
signal
(
signal
.
SIGHUP
,
self
.
trivial_signal_handler
)
self
.
assertIsInstance
(
hup
,
signal
.
Handlers
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
self
.
trivial_signal_handler
)
signal
.
signal
(
signal
.
SIGHUP
,
hup
)
self
.
assertEqual
(
signal
.
getsignal
(
signal
.
SIGHUP
),
hup
)
def
test_strsignal
(
self
):
self
.
assertIn
(
"Interrupt"
,
signal
.
strsignal
(
signal
.
SIGINT
))
self
.
assertIn
(
"Terminated"
,
signal
.
strsignal
(
signal
.
SIGTERM
))
self
.
assertIn
(
"Hangup"
,
signal
.
strsignal
(
signal
.
SIGHUP
))
# Issue 3864, unknown if this affects earlier versions of freebsd also
def
test_interprocess_signal
(
self
):
dirname
=
os
.
path
.
dirname
(
__file__
)
script
=
os
.
path
.
join
(
dirname
,
'signalinterproctester.py'
)
assert_python_ok
(
script
)
def
test_valid_signals
(
self
):
s
=
signal
.
valid_signals
()
self
.
assertIsInstance
(
s
,
set
)
self
.
assertIn
(
signal
.
Signals
.
SIGINT
,
s
)
self
.
assertIn
(
signal
.
Signals
.
SIGALRM
,
s
)
self
.
assertNotIn
(
0
,
s
)
self
.
assertNotIn
(
signal
.
NSIG
,
s
)
self
.
assertLess
(
len
(
s
),
signal
.
NSIG
)
@
unittest
.
skipUnless
(
sys
.
executable
,
"sys.executable required."
)
def
test_keyboard_interrupt_exit_code
(
self
):
"""KeyboardInterrupt triggers exit via SIGINT."""
process
=
subprocess
.
run
(
[
sys
.
executable
,
"-c"
,
"import os, signal, time
\
n
"
"os.kill(os.getpid(), signal.SIGINT)
\
n
"
"for _ in range(999): time.sleep(0.01)"
],
stderr
=
subprocess
.
PIPE
)
self
.
assertIn
(
b"KeyboardInterrupt"
,
process
.
stderr
)
self
.
assertEqual
(
process
.
returncode
,
-
signal
.
SIGINT
)
# Caveat: The exit code is insufficient to guarantee we actually died
# via a signal. POSIX shells do more than look at the 8 bit value.
# Writing an automation friendly test of an interactive shell
# to confirm that our process died via a SIGINT proved too complex.
@
unittest
.
skipUnless
(
sys
.
platform
==
"win32"
,
"Windows specific"
)
class
WindowsSignalTests
(
unittest
.
TestCase
):
def
test_valid_signals
(
self
):
s
=
signal
.
valid_signals
()
self
.
assertIsInstance
(
s
,
set
)
self
.
assertGreaterEqual
(
len
(
s
),
6
)
self
.
assertIn
(
signal
.
Signals
.
SIGINT
,
s
)
self
.
assertNotIn
(
0
,
s
)
self
.
assertNotIn
(
signal
.
NSIG
,
s
)
self
.
assertLess
(
len
(
s
),
signal
.
NSIG
)
def
test_issue9324
(
self
):
# Updated for issue #10003, adding SIGBREAK
handler
=
lambda
x
,
y
:
None
checked
=
set
()
for
sig
in
(
signal
.
SIGABRT
,
signal
.
SIGBREAK
,
signal
.
SIGFPE
,
signal
.
SIGILL
,
signal
.
SIGINT
,
signal
.
SIGSEGV
,
signal
.
SIGTERM
):
# Set and then reset a handler for signals that work on windows.
# Issue #18396, only for signals without a C-level handler.
if
signal
.
getsignal
(
sig
)
is
not
None
:
signal
.
signal
(
sig
,
signal
.
signal
(
sig
,
handler
))
checked
.
add
(
sig
)
# Issue #18396: Ensure the above loop at least tested *something*
self
.
assertTrue
(
checked
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
-
1
,
handler
)
with
self
.
assertRaises
(
ValueError
):
signal
.
signal
(
7
,
handler
)
@
unittest
.
skipUnless
(
sys
.
executable
,
"sys.executable required."
)
def
test_keyboard_interrupt_exit_code
(
self
):
"""KeyboardInterrupt triggers an exit using STATUS_CONTROL_C_EXIT."""
# We don't test via os.kill(os.getpid(), signal.CTRL_C_EVENT) here
# as that requires setting up a console control handler in a child
# in its own process group. Doable, but quite complicated. (see
# @eryksun on https://github.com/python/cpython/pull/11862)
process
=
subprocess
.
run
(
[
sys
.
executable
,
"-c"
,
"raise KeyboardInterrupt"
],
stderr
=
subprocess
.
PIPE
)
self
.
assertIn
(
b"KeyboardInterrupt"
,
process
.
stderr
)
STATUS_CONTROL_C_EXIT
=
0xC000013A
self
.
assertEqual
(
process
.
returncode
,
STATUS_CONTROL_C_EXIT
)
class
WakeupFDTests
(
unittest
.
TestCase
):
def
test_invalid_call
(
self
):
# First parameter is positional-only
with
self
.
assertRaises
(
TypeError
):
signal
.
set_wakeup_fd
(
signum
=
signal
.
SIGINT
)
# warn_on_full_buffer is a keyword-only parameter
with
self
.
assertRaises
(
TypeError
):
signal
.
set_wakeup_fd
(
signal
.
SIGINT
,
False
)
def
test_invalid_fd
(
self
):
fd
=
support
.
make_bad_fd
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_invalid_socket
(
self
):
sock
=
socket
.
socket
()
fd
=
sock
.
fileno
()
sock
.
close
()
self
.
assertRaises
((
ValueError
,
OSError
),
signal
.
set_wakeup_fd
,
fd
)
def
test_set_wakeup_fd_result
(
self
):
r1
,
w1
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r1
)
self
.
addCleanup
(
os
.
close
,
w1
)
r2
,
w2
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
r2
)
self
.
addCleanup
(
os
.
close
,
w2
)
if
hasattr
(
os
,
'set_blocking'
):
os
.
set_blocking
(
w1
,
False
)
os
.
set_blocking
(
w2
,
False
)
signal
.
set_wakeup_fd
(
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
w2
),
w1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
w2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
def
test_set_wakeup_fd_socket_result
(
self
):
sock1
=
socket
.
socket
()
self
.
addCleanup
(
sock1
.
close
)
sock1
.
setblocking
(
False
)
fd1
=
sock1
.
fileno
()
sock2
=
socket
.
socket
()
self
.
addCleanup
(
sock2
.
close
)
sock2
.
setblocking
(
False
)
fd2
=
sock2
.
fileno
()
signal
.
set_wakeup_fd
(
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
fd2
),
fd1
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
fd2
)
self
.
assertEqual
(
signal
.
set_wakeup_fd
(
-
1
),
-
1
)
# On Windows, files are always blocking and Windows does not provide a
# function to test if a socket is in non-blocking mode.
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"tests specific to POSIX"
)
def
test_set_wakeup_fd_blocking
(
self
):
rfd
,
wfd
=
os
.
pipe
()
self
.
addCleanup
(
os
.
close
,
rfd
)
self
.
addCleanup
(
os
.
close
,
wfd
)
# fd must be non-blocking
os
.
set_blocking
(
wfd
,
True
)
with
self
.
assertRaises
(
ValueError
)
as
cm
:
signal
.
set_wakeup_fd
(
wfd
)
self
.
assertEqual
(
str
(
cm
.
exception
),
"the fd %s must be in non-blocking mode"
%
wfd
)
# non-blocking is ok
os
.
set_blocking
(
wfd
,
False
)
signal
.
set_wakeup_fd
(
wfd
)
signal
.
set_wakeup_fd
(
-
1
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
WakeupSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
check_wakeup
(
self
,
test_body
,
*
signals
,
ordered
=
True
):
# use a subprocess to have only one thread
code
=
"""if 1:
import _testcapi
import os
import signal
import struct
signals = {!r}
def handler(signum, frame):
pass
def check_signum(signals):
data = os.read(read, len(signals)+1)
raised = struct.unpack('%uB' % len(data), data)
if not {!r}:
raised = set(raised)
signals = set(signals)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
{}
signal.signal(signal.SIGALRM, handler)
read, write = os.pipe()
os.set_blocking(write, False)
signal.set_wakeup_fd(write)
test()
check_signum(signals)
os.close(read)
os.close(write)
"""
.
format
(
tuple
(
map
(
int
,
signals
)),
ordered
,
test_body
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_wakeup_write_error
(
self
):
# Issue #16105: write() errors in the C signal handler should not
# pass silently.
# Use a subprocess to have only one thread.
code
=
"""if 1:
import _testcapi
import errno
import os
import signal
import sys
from test.support import captured_stderr
def handler(signum, frame):
1/0
signal.signal(signal.SIGALRM, handler)
r, w = os.pipe()
os.set_blocking(r, False)
# Set wakeup_fd a read-only file descriptor to trigger the error
signal.set_wakeup_fd(r)
try:
with captured_stderr() as err:
signal.raise_signal(signal.SIGALRM)
except ZeroDivisionError:
# An ignored exception should have been printed out on stderr
err = err.getvalue()
if ('Exception ignored when trying to write to the signal wakeup fd'
not in err):
raise AssertionError(err)
if ('OSError: [Errno %d]' % errno.EBADF) not in err:
raise AssertionError(err)
else:
raise AssertionError("ZeroDivisionError not raised")
os.close(r)
os.close(w)
"""
r
,
w
=
os
.
pipe
()
try
:
os
.
write
(
r
,
b'x'
)
except
OSError
:
pass
else
:
self
.
skipTest
(
"OS doesn't report write() error on the read end of a pipe"
)
finally
:
os
.
close
(
r
)
os
.
close
(
w
)
assert_python_ok
(
'-c'
,
code
)
def
test_wakeup_fd_early
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
# We attempt to get a signal during the sleep,
# before select is called
try:
select.select([], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
before_time = time.monotonic()
select.select([read], [], [], TIMEOUT_FULL)
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_wakeup_fd_during
(
self
):
self
.
check_wakeup
(
"""def test():
import select
import time
TIMEOUT_FULL = 10
TIMEOUT_HALF = 5
class InterruptSelect(Exception):
pass
def handler(signum, frame):
raise InterruptSelect
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
before_time = time.monotonic()
# We attempt to get a signal during the select call
try:
select.select([read], [], [], TIMEOUT_FULL)
except InterruptSelect:
pass
else:
raise Exception("select() was not interrupted")
after_time = time.monotonic()
dt = after_time - before_time
if dt >= TIMEOUT_HALF:
raise Exception("%s >= %s" % (dt, TIMEOUT_HALF))
"""
,
signal
.
SIGALRM
)
def
test_signum
(
self
):
self
.
check_wakeup
(
"""def test():
signal.signal(signal.SIGUSR1, handler)
signal.raise_signal(signal.SIGUSR1)
signal.raise_signal(signal.SIGALRM)
"""
,
signal
.
SIGUSR1
,
signal
.
SIGALRM
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pending
(
self
):
self
.
check_wakeup
(
"""def test():
signum1 = signal.SIGUSR1
signum2 = signal.SIGUSR2
signal.signal(signum1, handler)
signal.signal(signum2, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, (signum1, signum2))
signal.raise_signal(signum1)
signal.raise_signal(signum2)
# Unblocking the 2 signals calls the C signal handler twice
signal.pthread_sigmask(signal.SIG_UNBLOCK, (signum1, signum2))
"""
,
signal
.
SIGUSR1
,
signal
.
SIGUSR2
,
ordered
=
False
)
@
unittest
.
skipUnless
(
hasattr
(
socket
,
'socketpair'
),
'need socket.socketpair'
)
class
WakeupSocketSignalTests
(
unittest
.
TestCase
):
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_socket
(
self
):
# use a subprocess to have only one thread
code
=
"""if 1:
import signal
import socket
import struct
import _testcapi
signum = signal.SIGINT
signals = (signum,)
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
signal.raise_signal(signum)
data = read.recv(1)
if not data:
raise Exception("no signum written")
raised = struct.unpack('B', data)
if raised != signals:
raise Exception("%r != %r" % (raised, signals))
read.close()
write.close()
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_send_error
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
read.setblocking(False)
write.setblocking(False)
signal.set_wakeup_fd(write.fileno())
# Close sockets: send() will fail
read.close()
write.close()
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if ('Exception ignored when trying to {action} to the signal wakeup fd'
not in err):
raise AssertionError(err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
_testcapi
is
None
,
'need _testcapi'
)
def
test_warn_on_full_buffer
(
self
):
# Use a subprocess to have only one thread.
if
os
.
name
==
'nt'
:
action
=
'send'
else
:
action
=
'write'
code
=
"""if 1:
import errno
import signal
import socket
import sys
import time
import _testcapi
from test.support import captured_stderr
signum = signal.SIGINT
# This handler will be called, but we intentionally won't read from
# the wakeup fd.
def handler(signum, frame):
pass
signal.signal(signum, handler)
read, write = socket.socketpair()
# Fill the socketpair buffer
if sys.platform == 'win32':
# bpo-34130: On Windows, sometimes non-blocking send fails to fill
# the full socketpair buffer, so use a timeout of 50 ms instead.
write.settimeout(0.050)
else:
write.setblocking(False)
# Start with large chunk size to reduce the
# number of send needed to fill the buffer.
written = 0
for chunk_size in (2 ** 16, 2 ** 8, 1):
chunk = b"x" * chunk_size
try:
while True:
write.send(chunk)
written += chunk_size
except (BlockingIOError, socket.timeout):
pass
print(f"%s bytes written into the socketpair" % written, flush=True)
write.setblocking(False)
try:
write.send(b"x")
except BlockingIOError:
# The socketpair buffer seems full
pass
else:
raise AssertionError("%s bytes failed to fill the socketpair "
"buffer" % written)
# By default, we get a warning when a signal arrives
msg = ('Exception ignored when trying to {action} '
'to the signal wakeup fd')
signal.set_wakeup_fd(write.fileno())
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("first set_wakeup_fd() test failed, "
"stderr: %r" % err)
# And also if warn_on_full_buffer=True
signal.set_wakeup_fd(write.fileno(), warn_on_full_buffer=True)
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("set_wakeup_fd(warn_on_full_buffer=True) "
"test failed, stderr: %r" % err)
# But not if warn_on_full_buffer=False
signal.set_wakeup_fd(write.fileno(), warn_on_full_buffer=False)
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if err != "":
raise AssertionError("set_wakeup_fd(warn_on_full_buffer=False) "
"test failed, stderr: %r" % err)
# And then check the default again, to make sure warn_on_full_buffer
# settings don't leak across calls.
signal.set_wakeup_fd(write.fileno())
with captured_stderr() as err:
signal.raise_signal(signum)
err = err.getvalue()
if msg not in err:
raise AssertionError("second set_wakeup_fd() test failed, "
"stderr: %r" % err)
"""
.
format
(
action
=
action
)
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
SiginterruptTest
(
unittest
.
TestCase
):
def
readpipe_interrupted
(
self
,
interrupt
):
"""Perform a read during which a signal will arrive. Return True if the
read is interrupted by the signal and raises an exception. Return False
if it returns normally.
"""
# use a subprocess to have only one thread, to have a timeout on the
# blocking read and to not touch signal handling in this process
code
=
"""if 1:
import errno
import os
import signal
import sys
interrupt = %r
r, w = os.pipe()
def handler(signum, frame):
1 / 0
signal.signal(signal.SIGALRM, handler)
if interrupt is not None:
signal.siginterrupt(signal.SIGALRM, interrupt)
print("ready")
sys.stdout.flush()
# run the test twice
try:
for loop in range(2):
# send a SIGALRM in a second (during the read)
signal.alarm(1)
try:
# blocking call: read from a pipe without data
os.read(r, 1)
except ZeroDivisionError:
pass
else:
sys.exit(2)
sys.exit(3)
finally:
os.close(r)
os.close(w)
"""
%
(
interrupt
,)
with
spawn_python
(
'-c'
,
code
)
as
process
:
try
:
# wait until the child process is loaded and has started
first_line
=
process
.
stdout
.
readline
()
stdout
,
stderr
=
process
.
communicate
(
timeout
=
support
.
SHORT_TIMEOUT
)
except
subprocess
.
TimeoutExpired
:
process
.
kill
()
return
False
else
:
stdout
=
first_line
+
stdout
exitcode
=
process
.
wait
()
if
exitcode
not
in
(
2
,
3
):
raise
Exception
(
"Child error (exit code %s): %r"
%
(
exitcode
,
stdout
))
return
(
exitcode
==
3
)
def
test_without_siginterrupt
(
self
):
# If a signal handler is installed and siginterrupt is not called
# at all, when that signal arrives, it interrupts a syscall that's in
# progress.
interrupted
=
self
.
readpipe_interrupted
(
None
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_on
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a true value for the second argument, when that signal arrives, it
# interrupts a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
True
)
self
.
assertTrue
(
interrupted
)
def
test_siginterrupt_off
(
self
):
# If a signal handler is installed and siginterrupt is called with
# a false value for the second argument, when that signal arrives, it
# does not interrupt a syscall that's in progress.
interrupted
=
self
.
readpipe_interrupted
(
False
)
self
.
assertFalse
(
interrupted
)
@
unittest
.
skipIf
(
sys
.
platform
==
"win32"
,
"Not valid on Windows"
)
class
ItimerTest
(
unittest
.
TestCase
):
def
setUp
(
self
):
self
.
hndl_called
=
False
self
.
hndl_count
=
0
self
.
itimer
=
None
self
.
old_alarm
=
signal
.
signal
(
signal
.
SIGALRM
,
self
.
sig_alrm
)
def
tearDown
(
self
):
signal
.
signal
(
signal
.
SIGALRM
,
self
.
old_alarm
)
if
self
.
itimer
is
not
None
:
# test_itimer_exc doesn't change this attr
# just ensure that itimer is stopped
signal
.
setitimer
(
self
.
itimer
,
0
)
def
sig_alrm
(
self
,
*
args
):
self
.
hndl_called
=
True
def
sig_vtalrm
(
self
,
*
args
):
self
.
hndl_called
=
True
if
self
.
hndl_count
>
3
:
# it shouldn't be here, because it should have been disabled.
raise
signal
.
ItimerError
(
"setitimer didn't disable ITIMER_VIRTUAL "
"timer."
)
elif
self
.
hndl_count
==
3
:
# disable ITIMER_VIRTUAL, this function shouldn't be called anymore
signal
.
setitimer
(
signal
.
ITIMER_VIRTUAL
,
0
)
self
.
hndl_count
+=
1
def
sig_prof
(
self
,
*
args
):
self
.
hndl_called
=
True
signal
.
setitimer
(
signal
.
ITIMER_PROF
,
0
)
def
test_itimer_exc
(
self
):
# XXX I'm assuming -1 is an invalid itimer, but maybe some platform
# defines it ?
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
-
1
,
0
)
# Negative times are treated as zero on some platforms.
if
0
:
self
.
assertRaises
(
signal
.
ItimerError
,
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
-
1
)
def
test_itimer_real
(
self
):
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1.0
)
signal
.
pause
()
self
.
assertEqual
(
self
.
hndl_called
,
True
)
# Issue 3864, unknown if this affects earlier versions of freebsd also
@
unittest
.
skipIf
(
sys
.
platform
in
(
'netbsd5'
,),
'itimer not reliable (does not mix well with threading) on some BSDs.'
)
def
test_itimer_virtual
(
self
):
self
.
itimer
=
signal
.
ITIMER_VIRTUAL
signal
.
signal
(
signal
.
SIGVTALRM
,
self
.
sig_vtalrm
)
signal
.
setitimer
(
self
.
itimer
,
0.3
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# use up some virtual time by doing real work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_vtalrm handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# virtual itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_itimer_prof
(
self
):
self
.
itimer
=
signal
.
ITIMER_PROF
signal
.
signal
(
signal
.
SIGPROF
,
self
.
sig_prof
)
signal
.
setitimer
(
self
.
itimer
,
0.2
,
0.2
)
start_time
=
time
.
monotonic
()
while
time
.
monotonic
()
-
start_time
<
60.0
:
# do some work
_
=
pow
(
12345
,
67890
,
10000019
)
if
signal
.
getitimer
(
self
.
itimer
)
==
(
0.0
,
0.0
):
break
# sig_prof handler stopped this itimer
else
:
# Issue 8424
self
.
skipTest
(
"timeout: likely cause: machine too slow or load too "
"high"
)
# profiling itimer should be (0.0, 0.0) now
self
.
assertEqual
(
signal
.
getitimer
(
self
.
itimer
),
(
0.0
,
0.0
))
# and the handler should have been called
self
.
assertEqual
(
self
.
hndl_called
,
True
)
def
test_setitimer_tiny
(
self
):
# bpo-30807: C setitimer() takes a microsecond-resolution interval.
# Check that float -> timeval conversion doesn't round
# the interval down to zero, which would disable the timer.
self
.
itimer
=
signal
.
ITIMER_REAL
signal
.
setitimer
(
self
.
itimer
,
1e-6
)
time
.
sleep
(
1
)
self
.
assertEqual
(
self
.
hndl_called
,
True
)
class
PendingSignalsTests
(
unittest
.
TestCase
):
"""
Test pthread_sigmask(), pthread_kill(), sigpending() and sigwait()
functions.
"""
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending_empty
(
self
):
self
.
assertEqual
(
signal
.
sigpending
(),
set
())
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigpending'
),
'need signal.sigpending()'
)
def
test_sigpending
(
self
):
code
=
"""if 1:
import os
import signal
def handler(signum, frame):
1/0
signum = signal.SIGUSR1
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
os.kill(os.getpid(), signum)
pending = signal.sigpending()
for sig in pending:
assert isinstance(sig, signal.Signals), repr(pending)
if pending != {signum}:
raise Exception('%s != {%s}' % (pending, signum))
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill
(
self
):
code
=
"""if 1:
import signal
import threading
import sys
signum = signal.SIGUSR1
def handler(signum, frame):
1/0
signal.signal(signum, handler)
tid = threading.get_ident()
try:
signal.pthread_kill(tid, signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
wait_helper
(
self
,
blocked
,
test
):
"""
test: body of the "def test(signum):" function.
blocked: number of the blocked signal
"""
code
=
'''if 1:
import signal
import sys
from signal import Signals
def handler(signum, frame):
1/0
%s
blocked = %s
signum = signal.SIGALRM
# child: block and wait the signal
try:
signal.signal(signum, handler)
signal.pthread_sigmask(signal.SIG_BLOCK, [blocked])
# Do the tests
test(signum)
# The handler must not be called on unblock
try:
signal.pthread_sigmask(signal.SIG_UNBLOCK, [blocked])
except ZeroDivisionError:
print("the signal handler has been called",
file=sys.stderr)
sys.exit(1)
except BaseException as err:
print("error: {}".format(err), file=sys.stderr)
sys.stderr.flush()
sys.exit(1)
'''
%
(
test
.
strip
(),
blocked
)
# sig*wait* must be called with the signal blocked: since the current
# process might have several threads running, use a subprocess to have
# a single thread.
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
def
test_sigwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
received = signal.sigwait([signum])
assert isinstance(received, signal.Signals), received
if received != signum:
raise Exception('received %s, not %s' % (received, signum))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwaitinfo'
),
'need signal.sigwaitinfo()'
)
def
test_sigwaitinfo
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigwaitinfo([signum])
if info.si_signo != signum:
raise Exception("info.si_signo != %s" % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
signal.alarm(1)
info = signal.sigtimedwait([signum], 10.1000)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_poll
(
self
):
# check that polling with sigtimedwait works
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
import os
os.kill(os.getpid(), signum)
info = signal.sigtimedwait([signum], 0)
if info.si_signo != signum:
raise Exception('info.si_signo != %s' % signum)
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_timeout
(
self
):
self
.
wait_helper
(
signal
.
SIGALRM
,
'''
def test(signum):
received = signal.sigtimedwait([signum], 1.0)
if received is not None:
raise Exception("received=%r" % (received,))
'''
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigtimedwait'
),
'need signal.sigtimedwait()'
)
def
test_sigtimedwait_negative_timeout
(
self
):
signum
=
signal
.
SIGALRM
self
.
assertRaises
(
ValueError
,
signal
.
sigtimedwait
,
[
signum
],
-
1.0
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'sigwait'
),
'need signal.sigwait()'
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_sigwait_thread
(
self
):
# Check that calling sigwait() from a thread doesn't suspend the whole
# process. A new interpreter is spawned to avoid problems when mixing
# threads and fork(): only async-safe functions are allowed between
# fork() and exec().
assert_python_ok
(
"-c"
,
"""if True:
import os, threading, sys, time, signal
# the default handler terminates the process
signum = signal.SIGUSR1
def kill_later():
# wait until the main thread is waiting in sigwait()
time.sleep(1)
os.kill(os.getpid(), signum)
# the signal must be blocked by all the threads
signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
killer = threading.Thread(target=kill_later)
killer.start()
received = signal.sigwait([signum])
if received != signum:
print("sigwait() received %s, not %s" % (received, signum),
file=sys.stderr)
sys.exit(1)
killer.join()
# unblock the signal, which should have been cleared by sigwait()
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
"""
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask_arguments
(
self
):
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
)
self
.
assertRaises
(
TypeError
,
signal
.
pthread_sigmask
,
1
,
2
,
3
)
self
.
assertRaises
(
OSError
,
signal
.
pthread_sigmask
,
1700
,
[])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
signal
.
NSIG
])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
0
])
with
self
.
assertRaises
(
ValueError
):
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
[
1
<<
1000
])
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask_valid_signals
(
self
):
s
=
signal
.
pthread_sigmask
(
signal
.
SIG_BLOCK
,
signal
.
valid_signals
())
self
.
addCleanup
(
signal
.
pthread_sigmask
,
signal
.
SIG_SETMASK
,
s
)
# Get current blocked set
s
=
signal
.
pthread_sigmask
(
signal
.
SIG_UNBLOCK
,
signal
.
valid_signals
())
self
.
assertLessEqual
(
s
,
signal
.
valid_signals
())
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_sigmask'
),
'need signal.pthread_sigmask()'
)
def
test_pthread_sigmask
(
self
):
code
=
"""if 1:
import signal
import os; import threading
def handler(signum, frame):
1/0
def kill(signum):
os.kill(os.getpid(), signum)
def check_mask(mask):
for sig in mask:
assert isinstance(sig, signal.Signals), repr(sig)
def read_sigmask():
sigmask = signal.pthread_sigmask(signal.SIG_BLOCK, [])
check_mask(sigmask)
return sigmask
signum = signal.SIGUSR1
# Install our signal handler
old_handler = signal.signal(signum, handler)
# Unblock SIGUSR1 (and copy the old mask) to test our signal handler
old_mask = signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
check_mask(old_mask)
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Block and then raise SIGUSR1. The signal is blocked: the signal
# handler is not called, and the signal is now pending
mask = signal.pthread_sigmask(signal.SIG_BLOCK, [signum])
check_mask(mask)
kill(signum)
# Check the new mask
blocked = read_sigmask()
check_mask(blocked)
if signum not in blocked:
raise Exception("%s not in %s" % (signum, blocked))
if old_mask ^ blocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (old_mask, blocked, signum))
# Unblock SIGUSR1
try:
# unblock the pending signal calls immediately the signal handler
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum])
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
try:
kill(signum)
except ZeroDivisionError:
pass
else:
raise Exception("ZeroDivisionError not raised")
# Check the new mask
unblocked = read_sigmask()
if signum in unblocked:
raise Exception("%s in %s" % (signum, unblocked))
if blocked ^ unblocked != {signum}:
raise Exception("%s ^ %s != {%s}" % (blocked, unblocked, signum))
if old_mask != unblocked:
raise Exception("%s != %s" % (old_mask, unblocked))
"""
assert_python_ok
(
'-c'
,
code
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
'pthread_kill'
),
'need signal.pthread_kill()'
)
def
test_pthread_kill_main_thread
(
self
):
# Test that a signal can be sent to the main thread with pthread_kill()
# before any other thread has been created (see issue #12392).
code
=
"""if True:
import threading
import signal
import sys
def handler(signum, frame):
sys.exit(3)
signal.signal(signal.SIGUSR1, handler)
signal.pthread_kill(threading.get_ident(), signal.SIGUSR1)
sys.exit(2)
"""
with
spawn_python
(
'-c'
,
code
)
as
process
:
stdout
,
stderr
=
process
.
communicate
()
exitcode
=
process
.
wait
()
if
exitcode
!=
3
:
raise
Exception
(
"Child error (exit code %s): %s"
%
(
exitcode
,
stdout
))
class
StressTest
(
unittest
.
TestCase
):
"""
Stress signal delivery, especially when a signal arrives in
the middle of recomputing the signal state or executing
previously tripped signal handlers.
"""
def
setsig
(
self
,
signum
,
handler
):
old_handler
=
signal
.
signal
(
signum
,
handler
)
self
.
addCleanup
(
signal
.
signal
,
signum
,
old_handler
)
def
measure_itimer_resolution
(
self
):
N
=
20
times
=
[]
def
handler
(
signum
=
None
,
frame
=
None
):
if
len
(
times
)
<
N
:
times
.
append
(
time
.
perf_counter
())
# 1 µs is the smallest possible timer interval,
# we want to measure what the concrete duration
# will be on this platform
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
)
self
.
addCleanup
(
signal
.
setitimer
,
signal
.
ITIMER_REAL
,
0
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
handler
()
while
len
(
times
)
<
N
:
time
.
sleep
(
1e-3
)
durations
=
[
times
[
i
+
1
]
-
times
[
i
]
for
i
in
range
(
len
(
times
)
-
1
)]
med
=
statistics
.
median
(
durations
)
if
support
.
verbose
:
print
(
"detected median itimer() resolution: %.6f s."
%
(
med
,))
return
med
def
decide_itimer_count
(
self
):
# Some systems have poor setitimer() resolution (for example
# measured around 20 ms. on FreeBSD 9), so decide on a reasonable
# number of sequential timers based on that.
reso
=
self
.
measure_itimer_resolution
()
if
reso
<=
1e-4
:
return
10000
elif
reso
<=
1e-2
:
return
100
else
:
self
.
skipTest
(
"detected itimer resolution (%.3f s.) too high "
"(> 10 ms.) on this platform (or system too busy)"
%
(
reso
,))
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_dependent
(
self
):
"""
This test uses dependent signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
first_handler
(
signum
,
frame
):
# 1e-6 is the minimum non-zero value for `setitimer()`.
# Choose a random delay so as to improve chances of
# triggering a race condition. Ideally the signal is received
# when inside critical signal-handling routines such as
# Py_MakePendingCalls().
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
def
second_handler
(
signum
=
None
,
frame
=
None
):
sigs
.
append
(
signum
)
# Here on Linux, SIGPROF > SIGALRM > SIGUSR1. By using both
# ascending and descending sequences (SIGUSR1 then SIGALRM,
# SIGPROF then SIGALRM), we maximize chances of hitting a bug.
self
.
setsig
(
signal
.
SIGPROF
,
first_handler
)
self
.
setsig
(
signal
.
SIGUSR1
,
first_handler
)
self
.
setsig
(
signal
.
SIGALRM
,
second_handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
monotonic
()
+
support
.
SHORT_TIMEOUT
while
expected_sigs
<
N
:
os
.
kill
(
os
.
getpid
(),
signal
.
SIGPROF
)
expected_sigs
+=
1
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
1
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"setitimer"
),
"test needs setitimer()"
)
def
test_stress_delivery_simultaneous
(
self
):
"""
This test uses simultaneous signal handlers.
"""
N
=
self
.
decide_itimer_count
()
sigs
=
[]
def
handler
(
signum
,
frame
):
sigs
.
append
(
signum
)
self
.
setsig
(
signal
.
SIGUSR1
,
handler
)
self
.
setsig
(
signal
.
SIGALRM
,
handler
)
# for ITIMER_REAL
expected_sigs
=
0
deadline
=
time
.
monotonic
()
+
support
.
SHORT_TIMEOUT
while
expected_sigs
<
N
:
# Hopefully the SIGALRM will be received somewhere during
# initial processing of SIGUSR1.
signal
.
setitimer
(
signal
.
ITIMER_REAL
,
1e-6
+
random
.
random
()
*
1e-5
)
os
.
kill
(
os
.
getpid
(),
signal
.
SIGUSR1
)
expected_sigs
+=
2
# Wait for handlers to run to avoid signal coalescing
while
len
(
sigs
)
<
expected_sigs
and
time
.
monotonic
()
<
deadline
:
time
.
sleep
(
1e-5
)
# All ITIMER_REAL signals should have been delivered to the
# Python handler
self
.
assertEqual
(
len
(
sigs
),
N
,
"Some signals were lost"
)
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"SIGUSR1"
),
"test needs SIGUSR1"
)
def
test_stress_modifying_handlers
(
self
):
# bpo-43406: race condition between trip_signal() and signal.signal
signum
=
signal
.
SIGUSR1
num_sent_signals
=
0
num_received_signals
=
0
do_stop
=
False
def
custom_handler
(
signum
,
frame
):
nonlocal
num_received_signals
num_received_signals
+=
1
def
set_interrupts
():
nonlocal
num_sent_signals
while
not
do_stop
:
signal
.
raise_signal
(
signum
)
num_sent_signals
+=
1
def
cycle_handlers
():
while
num_sent_signals
<
100
:
for
i
in
range
(
20000
):
# Cycle between a Python-defined and a non-Python handler
for
handler
in
[
custom_handler
,
signal
.
SIG_IGN
]:
signal
.
signal
(
signum
,
handler
)
old_handler
=
signal
.
signal
(
signum
,
custom_handler
)
self
.
addCleanup
(
signal
.
signal
,
signum
,
old_handler
)
t
=
threading
.
Thread
(
target
=
set_interrupts
)
try
:
ignored
=
False
with
support
.
catch_unraisable_exception
()
as
cm
:
t
.
start
()
cycle_handlers
()
do_stop
=
True
t
.
join
()
if
cm
.
unraisable
is
not
None
:
# An unraisable exception may be printed out when
# a signal is ignored due to the aforementioned
# race condition, check it.
self
.
assertIsInstance
(
cm
.
unraisable
.
exc_value
,
OSError
)
self
.
assertIn
(
f"Signal
{
signum
}
ignored due to race condition"
,
str
(
cm
.
unraisable
.
exc_value
))
ignored
=
True
# bpo-43406: Even if it is unlikely, it's technically possible that
# all signals were ignored because of race conditions.
if
not
ignored
:
# Sanity check that some signals were received, but not all
self
.
assertGreater
(
num_received_signals
,
0
)
self
.
assertLess
(
num_received_signals
,
num_sent_signals
)
finally
:
do_stop
=
True
t
.
join
()
class
RaiseSignalTest
(
unittest
.
TestCase
):
def
test_sigint
(
self
):
with
self
.
assertRaises
(
KeyboardInterrupt
):
signal
.
raise_signal
(
signal
.
SIGINT
)
@
unittest
.
skipIf
(
sys
.
platform
!=
"win32"
,
"Windows specific test"
)
def
test_invalid_argument
(
self
):
try
:
SIGHUP
=
1
# not supported on win32
signal
.
raise_signal
(
SIGHUP
)
self
.
fail
(
"OSError (Invalid argument) expected"
)
except
OSError
as
e
:
if
e
.
errno
==
errno
.
EINVAL
:
pass
else
:
raise
def
test_handler
(
self
):
is_ok
=
False
def
handler
(
a
,
b
):
nonlocal
is_ok
is_ok
=
True
old_signal
=
signal
.
signal
(
signal
.
SIGINT
,
handler
)
self
.
addCleanup
(
signal
.
signal
,
signal
.
SIGINT
,
old_signal
)
signal
.
raise_signal
(
signal
.
SIGINT
)
self
.
assertTrue
(
is_ok
)
class
PidfdSignalTest
(
unittest
.
TestCase
):
@
unittest
.
skipUnless
(
hasattr
(
signal
,
"pidfd_send_signal"
),
"pidfd support not built in"
,
)
def
test_pidfd_send_signal
(
self
):
with
self
.
assertRaises
(
OSError
)
as
cm
:
signal
.
pidfd_send_signal
(
0
,
signal
.
SIGINT
)
if
cm
.
exception
.
errno
==
errno
.
ENOSYS
:
self
.
skipTest
(
"kernel does not support pidfds"
)
elif
cm
.
exception
.
errno
==
errno
.
EPERM
:
self
.
skipTest
(
"Not enough privileges to use pidfs"
)
self
.
assertEqual
(
cm
.
exception
.
errno
,
errno
.
EBADF
)
my_pidfd
=
os
.
open
(
f'/proc/
{
os
.
getpid
()
}
'
,
os
.
O_DIRECTORY
)
self
.
addCleanup
(
os
.
close
,
my_pidfd
)
with
self
.
assertRaisesRegex
(
TypeError
,
"^siginfo must be None$"
):
signal
.
pidfd_send_signal
(
my_pidfd
,
signal
.
SIGINT
,
object
(),
0
)
with
self
.
assertRaises
(
KeyboardInterrupt
):
signal
.
pidfd_send_signal
(
my_pidfd
,
signal
.
SIGINT
)
def
tearDownModule
():
support
.
reap_children
()
if
__name__
==
"__main__"
:
unittest
.
main
()
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment