diff options
Diffstat (limited to 'Lib/test/test_signal.py')
| -rw-r--r-- | Lib/test/test_signal.py | 494 |
1 files changed, 256 insertions, 238 deletions
diff --git a/Lib/test/test_signal.py b/Lib/test/test_signal.py index e5df000..e23b126 100644 --- a/Lib/test/test_signal.py +++ b/Lib/test/test_signal.py @@ -224,117 +224,115 @@ class WindowsSignalTests(unittest.TestCase): @unittest.skipIf(sys.platform == "win32", "Not valid on Windows") class WakeupSignalTests(unittest.TestCase): - TIMEOUT_FULL = 10 - TIMEOUT_HALF = 5 + def check_wakeup(self, test_body, *signals): + # use a subprocess to have only one thread + code = """if 1: + import fcntl + import os + import signal + import struct - def handler(self, signum, frame): - pass + signals = {!r} - def check_signum(self, *signals): - data = os.read(self.read, len(signals)+1) - raised = struct.unpack('%uB' % len(data), data) - # We don't care of the signal delivery order (it's not portable or - # reliable) - raised = set(raised) - signals = set(signals) - self.assertEqual(raised, signals) + def handler(signum, frame): + pass - def test_wakeup_fd_early(self): - import select - - signal.alarm(1) - before_time = time.time() - # We attempt to get a signal during the sleep, - # before select is called - time.sleep(self.TIMEOUT_FULL) - mid_time = time.time() - self.assertTrue(mid_time - before_time < self.TIMEOUT_HALF) - select.select([self.read], [], [], self.TIMEOUT_FULL) - after_time = time.time() - self.assertTrue(after_time - mid_time < self.TIMEOUT_HALF) - self.check_signum(signal.SIGALRM) + def check_signum(signals): + data = os.read(read, len(signals)+1) + raised = struct.unpack('%uB' % len(data), data) + # We don't care of the signal delivery order (it's not portable or + # reliable) + raised = set(raised) + signals = set(signals) + assert raised == signals, "%r != %r" % (raised, signals) - def test_wakeup_fd_during(self): - import select + {} - signal.alarm(1) - before_time = time.time() - # We attempt to get a signal during the select call - self.assertRaises(select.error, select.select, - [self.read], [], [], self.TIMEOUT_FULL) - after_time = time.time() - self.assertTrue(after_time - before_time < self.TIMEOUT_HALF) - self.check_signum(signal.SIGALRM) + signal.signal(signal.SIGALRM, handler) + read, write = os.pipe() + for fd in (read, write): + flags = fcntl.fcntl(fd, fcntl.F_GETFL, 0) + flags = flags | os.O_NONBLOCK + fcntl.fcntl(fd, fcntl.F_SETFL, flags) + signal.set_wakeup_fd(write) - def test_signum(self): - old_handler = signal.signal(signal.SIGUSR1, self.handler) - self.addCleanup(signal.signal, signal.SIGUSR1, old_handler) - os.kill(os.getpid(), signal.SIGUSR1) - os.kill(os.getpid(), signal.SIGALRM) - self.check_signum(signal.SIGUSR1, signal.SIGALRM) + test() + check_signum(signals) - @unittest.skipUnless(hasattr(signal, 'pthread_sigmask'), - 'need signal.pthread_sigmask()') - @unittest.skipUnless(hasattr(signal, 'pthread_kill'), - 'need signal.pthread_kill()') - def test_pending(self): - signum1 = signal.SIGUSR1 - signum2 = signal.SIGUSR2 - tid = threading.current_thread().ident + os.close(read) + os.close(write) + """.format(signals, test_body) - old_handler = signal.signal(signum1, self.handler) - self.addCleanup(signal.signal, signum1, old_handler) - old_handler = signal.signal(signum2, self.handler) - self.addCleanup(signal.signal, signum2, old_handler) + assert_python_ok('-c', code) - signal.pthread_sigmask(signal.SIG_BLOCK, (signum1, signum2)) - signal.pthread_kill(tid, signum1) - signal.pthread_kill(tid, signum2) - # Unblocking the 2 signals calls the C signal handler twice - signal.pthread_sigmask(signal.SIG_UNBLOCK, (signum1, signum2)) + def test_wakeup_fd_early(self): + self.check_wakeup("""def test(): + import select + import time - self.check_signum(signum1, signum2) + TIMEOUT_FULL = 10 + TIMEOUT_HALF = 5 - @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 + signal.alarm(1) + before_time = time.time() + # We attempt to get a signal during the sleep, + # before select is called + time.sleep(TIMEOUT_FULL) + mid_time = time.time() + dt = mid_time - before_time + assert dt < TIMEOUT_HALF, dt + select.select([read], [], [], TIMEOUT_FULL) + after_time = time.time() + dt = after_time - mid_time + assert dt < TIMEOUT_HALF, dt + """, signal.SIGALRM) - def handler(signum, frame): - sys.exit(3) + def test_wakeup_fd_during(self): + self.check_wakeup("""def test(): + import select + import time + TIMEOUT_FULL = 10 + TIMEOUT_HALF = 5 + + signal.alarm(1) + before_time = time.time() + # We attempt to get a signal during the select call + try: + select.select([read], [], [], TIMEOUT_FULL) + except select.error: + pass + else: + raise Exception("select.error not raised") + after_time = time.time() + dt = after_time - before_time + assert dt < TIMEOUT_HALF, dt + """, signal.SIGALRM) + + def test_signum(self): + self.check_wakeup("""def test(): signal.signal(signal.SIGUSR1, handler) - signal.pthread_kill(threading.get_ident(), signal.SIGUSR1) - sys.exit(1) - """ + os.kill(os.getpid(), signal.SIGUSR1) + os.kill(os.getpid(), signal.SIGALRM) + """, signal.SIGUSR1, signal.SIGALRM) - 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)) + @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 - def setUp(self): - import fcntl + signal.signal(signum1, handler) + signal.signal(signum2, handler) - self.alrm = signal.signal(signal.SIGALRM, self.handler) - self.read, self.write = os.pipe() - flags = fcntl.fcntl(self.write, fcntl.F_GETFL, 0) - flags = flags | os.O_NONBLOCK - fcntl.fcntl(self.write, fcntl.F_SETFL, flags) - self.old_wakeup = signal.set_wakeup_fd(self.write) + signal.pthread_sigmask(signal.SIG_BLOCK, (signum1, signum2)) + os.kill(os.getpid(), signum1) + os.kill(os.getpid(), signum2) + # Unblocking the 2 signals calls the C signal handler twice + signal.pthread_sigmask(signal.SIG_UNBLOCK, (signum1, signum2)) + """, signal.SIGUSR1, signal.SIGUSR2) - def tearDown(self): - signal.set_wakeup_fd(self.old_wakeup) - os.close(self.read) - os.close(self.write) - signal.signal(signal.SIGALRM, self.alrm) @unittest.skipIf(sys.platform == "win32", "Not valid on Windows") class SiginterruptTest(unittest.TestCase): @@ -519,60 +517,6 @@ class PendingSignalsTests(unittest.TestCase): Test pthread_sigmask(), pthread_kill(), sigpending() and sigwait() functions. """ - def setUp(self): - self.has_pthread_kill = hasattr(signal, 'pthread_kill') - - def handler(self, signum, frame): - 1/0 - - def read_sigmask(self): - return signal.pthread_sigmask(signal.SIG_BLOCK, []) - - def can_test_blocked_signals(self, skip): - """ - Check if a blocked signal can be raised to the main thread without - calling its signal handler. We need pthread_kill() or exactly one - thread (the main thread). - - Return True if it's possible. Otherwise, return False and print a - warning if skip is False, or raise a SkipTest exception if skip is - True. - """ - if self.has_pthread_kill: - return True - - # The fault handler timeout thread masks all signals. If the main - # thread masks also SIGUSR1, all threads mask this signal. In this - # case, if we send SIGUSR1 to the process, the signal is pending in the - # main or the faulthandler timeout thread. Unblock SIGUSR1 in the main - # thread calls the signal handler only if the signal is pending for the - # main thread. Stop the faulthandler timeout thread to workaround this - # problem. - import faulthandler - faulthandler.cancel_dump_tracebacks_later() - - # Issue #11998: The _tkinter module loads the Tcl library which - # creates a thread waiting events in select(). This thread receives - # signals blocked by all other threads. We cannot test blocked - # signals - if '_tkinter' in sys.modules: - message = ("_tkinter is loaded and pthread_kill() is missing, " - "cannot test blocked signals (issue #11998)") - if skip: - self.skipTest(message) - else: - print("WARNING: %s" % message) - return False - return True - - def kill(self, signum): - if self.has_pthread_kill: - tid = threading.get_ident() - signal.pthread_kill(tid, signum) - else: - pid = os.getpid() - os.kill(pid, signum) - @unittest.skipUnless(hasattr(signal, 'sigpending'), 'need signal.sigpending()') def test_sigpending_empty(self): @@ -583,70 +527,103 @@ class PendingSignalsTests(unittest.TestCase): @unittest.skipUnless(hasattr(signal, 'sigpending'), 'need signal.sigpending()') def test_sigpending(self): - self.can_test_blocked_signals(True) + code = """if 1: + import os + import signal - signum = signal.SIGUSR1 - old_handler = signal.signal(signum, self.handler) - self.addCleanup(signal.signal, signum, old_handler) + def handler(signum, frame): + 1/0 - signal.pthread_sigmask(signal.SIG_BLOCK, [signum]) - self.kill(signum) - self.assertEqual(signal.sigpending(), {signum}) - with self.assertRaises(ZeroDivisionError): - signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum]) + signum = signal.SIGUSR1 + signal.signal(signum, handler) + + signal.pthread_sigmask(signal.SIG_BLOCK, [signum]) + os.kill(os.getpid(), signum) + pending = signal.sigpending() + assert pending == {signum}, '%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): - signum = signal.SIGUSR1 - current = threading.get_ident() + code = """if 1: + import signal + import threading + import sys - old_handler = signal.signal(signum, self.handler) - self.addCleanup(signal.signal, signum, old_handler) + signum = signal.SIGUSR1 + + def handler(signum, frame): + 1/0 - with self.assertRaises(ZeroDivisionError): - signal.pthread_kill(current, signum) + 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, test, blocked): + def wait_helper(self, blocked, test): """ test: body of the "def test(signum):" function. blocked: number of the blocked signal """ - code = ''' -import signal -import sys + code = '''if 1: + import signal + import sys -def handler(signum, frame): - 1/0 + def handler(signum, frame): + 1/0 -def test(signum): -%s + %s -blocked = %s -signum = signal.SIGALRM + blocked = %s + signum = signal.SIGALRM -# child: block and wait the signal -try: - signal.signal(signum, handler) - signal.pthread_sigmask(signal.SIG_BLOCK, [blocked]) + # child: block and wait the signal + try: + signal.signal(signum, handler) + signal.pthread_sigmask(signal.SIG_BLOCK, [blocked]) - # Do the tests - test(signum) + # 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, blocked) + # 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 @@ -656,61 +633,56 @@ except BaseException as err: @unittest.skipUnless(hasattr(signal, 'sigwait'), 'need signal.sigwait()') def test_sigwait(self): - test = ''' + self.wait_helper(signal.SIGALRM, ''' + def test(signum): signal.alarm(1) received = signal.sigwait([signum]) assert received == signum , 'received %s, not %s' % (received, signum) - ''' - - self.wait_helper(test, signal.SIGALRM) + ''') @unittest.skipUnless(hasattr(signal, 'sigwaitinfo'), 'need signal.sigwaitinfo()') def test_sigwaitinfo(self): - test = ''' + self.wait_helper(signal.SIGALRM, ''' + def test(signum): signal.alarm(1) info = signal.sigwaitinfo([signum]) assert info.si_signo == signum, "info.si_signo != %s" % signum - ''' - - self.wait_helper(test, signal.SIGALRM) + ''') @unittest.skipUnless(hasattr(signal, 'sigtimedwait'), 'need signal.sigtimedwait()') def test_sigtimedwait(self): - test = ''' + self.wait_helper(signal.SIGALRM, ''' + def test(signum): signal.alarm(1) info = signal.sigtimedwait([signum], (10, 1000)) assert info.si_signo == signum, 'info.si_signo != %s' % signum - ''' - - self.wait_helper(test, signal.SIGALRM) + ''') @unittest.skipUnless(hasattr(signal, 'sigtimedwait'), 'need signal.sigtimedwait()') # issue #12303: sigtimedwait() takes 30 seconds on FreeBSD 6 (kernel bug) @unittest.skipIf(sys.platform =='freebsd6', - 'sigtimedwait() with a null timeout doens\'t work on FreeBSD 6') + "sigtimedwait() with a null timeout doens't work on FreeBSD 6") def test_sigtimedwait_poll(self): # check that polling with sigtimedwait works - test = ''' + self.wait_helper(signal.SIGALRM, ''' + def test(signum): import os os.kill(os.getpid(), signum) info = signal.sigtimedwait([signum], (0, 0)) assert info.si_signo == signum, 'info.si_signo != %s' % signum - ''' - - self.wait_helper(test, signal.SIGALRM) + ''') @unittest.skipUnless(hasattr(signal, 'sigtimedwait'), 'need signal.sigtimedwait()') def test_sigtimedwait_timeout(self): - test = ''' + self.wait_helper(signal.SIGALRM, ''' + def test(signum): received = signal.sigtimedwait([signum], (1, 0)) assert received is None, "received=%r" % (received,) - ''' - - self.wait_helper(test, signal.SIGALRM) + ''') @unittest.skipUnless(hasattr(signal, 'sigtimedwait'), 'need signal.sigtimedwait()') @@ -723,7 +695,8 @@ except BaseException as err: @unittest.skipUnless(hasattr(signal, 'sigwaitinfo'), 'need signal.sigwaitinfo()') def test_sigwaitinfo_interrupted(self): - test = ''' + self.wait_helper(signal.SIGUSR1, ''' + def test(signum): import errno hndl_called = True @@ -741,9 +714,7 @@ except BaseException as err: raise Exception("Expected EINTR to be raised by sigwaitinfo") else: raise Exception("Expected EINTR to be raised by sigwaitinfo") - ''' - - self.wait_helper(test, signal.SIGUSR1) + ''') @unittest.skipUnless(hasattr(signal, 'sigwait'), 'need signal.sigwait()') @@ -791,46 +762,93 @@ except BaseException as err: @unittest.skipUnless(hasattr(signal, 'pthread_sigmask'), 'need signal.pthread_sigmask()') def test_pthread_sigmask(self): - test_blocked_signals = self.can_test_blocked_signals(False) + code = """if 1: + import signal + import os; import threading + + def handler(signum, frame): + 1/0 + + def kill(signum): + os.kill(os.getpid(), signum) + + def read_sigmask(): + return signal.pthread_sigmask(signal.SIG_BLOCK, []) + signum = signal.SIGUSR1 # Install our signal handler - old_handler = signal.signal(signum, self.handler) - self.addCleanup(signal.signal, signum, old_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]) - self.addCleanup(signal.pthread_sigmask, signal.SIG_SETMASK, old_mask) - with self.assertRaises(ZeroDivisionError): - self.kill(signum) + 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 signal.pthread_sigmask(signal.SIG_BLOCK, [signum]) - if test_blocked_signals: - self.kill(signum) + kill(signum) # Check the new mask - blocked = self.read_sigmask() - self.assertIn(signum, blocked) - self.assertEqual(old_mask ^ blocked, {signum}) + blocked = read_sigmask() + assert signum in blocked, "%s not in %s" % (signum, blocked) + assert old_mask ^ blocked == {signum}, "%s ^ %s != {%s}" % (old_mask, blocked, signum) # Unblock SIGUSR1 - if test_blocked_signals: - with self.assertRaises(ZeroDivisionError): - # unblock the pending signal calls immediatly the signal handler - signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum]) - else: + try: + # unblock the pending signal calls immediatly the signal handler signal.pthread_sigmask(signal.SIG_UNBLOCK, [signum]) - with self.assertRaises(ZeroDivisionError): - self.kill(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 = self.read_sigmask() - self.assertNotIn(signum, unblocked) - self.assertEqual(blocked ^ unblocked, {signum}) - self.assertSequenceEqual(old_mask, unblocked) - # Finally, restore the previous signal handler and the signal mask + unblocked = read_sigmask() + assert signum not in unblocked, "%s in %s" % (signum, unblocked) + assert blocked ^ unblocked == {signum}, "%s ^ %s != {%s}" % (blocked, unblocked, signum) + assert old_mask == unblocked, "%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)) def test_main(): |