a zep@sddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl m Z ddlZddlZddlmZz ddlZWneydZYn0dZejdkZdd d Zd d ZeddZGdddejZedkredS))contextmanagerN)support) script_helper is_android)skip_if_sanitizer)dedentg?ntcCsL|}|d|7}|d|7}d|kr", line %s in func z& File "", line %s in r ^ $)Zlineno1Zlineno2header min_countregexr r ,/usr/lib/python3.9/test/test_faulthandler.pyexpected_tracebacks   rcCsttd|S)Nz(raising SIGSEGV on Android is unreliable)unittestskipIfr)testr r rskip_segfault_on_android!s rc cs.t}z|VWt|n t|0dSN)tempfilemktemprunlinkfilenamer r rtemporary_filename&src@s2eZdZdddZdddddddddZdd d Zd d Zee j d dddZ e ddZddZddZee j dkdddZeedudeeedde ddZeedudeeed d!e d"d#Zd$d%Zd&d'Zee j d(d)eeed* d+d,d-Ze d.d/Zeddd0d1e d2d3Zee j dkd4eddd0d1e d5d6Ze d7d8Z e d9d:Z!d;d<Z"d=d>Z#d?d@Z$dAdBZ%dddCdDdEZ&dFdGZ'dHdIZ(ee j dkd4dJdKZ)dLdMZ*dNdOZ+dPdQZ,dRdSZ-ddddCdUdVZ.dWdXZ/dYdZZ0d[d\Z1d]d^Z2ee j dkd4d_d`Z3dadbZ4eeedc ddddedfZ5dgdhZ6didjZ7dkdlZ8ee j dkd4dmdnZ9dodpZ:dqdrZ;eee?dwdxdyZ@ee?dwdzd{ZAee?dwd|d}ZBee?dwd~dZCddZDdS)FaultHandlerTestsNc Csdt|}g}|dur"||tXtjd||d}|$|\}}|}Wdn1sj0YWdn1s0Y| dd}|r| |dt |d} | }Wdn1s0Y| dd}nj|durX| |dt |t jdt |ddd } | }Wdn1sB0Y| dd}||fS) N-c)pass_fdsasciibackslashreplacerbrF)closefd)rstripappendrZSuppressCrashReportrZ spawn_python communicatewaitdecode assertEqualopenreadoslseekSEEK_SET splitlines) selfcoderfdr processoutputstderrexitcodefpr r r get_output/s,    D   &  ( zFaultHandlerTests.get_outputTF)r all_threads other_regexr4know_current_threadpy_fatal_errorcCs|r|rd} qd} nd} d} | r(|d7}t| j||| d} |rP| d|7} |j|||d\} } d | } || | || d dS) NzCurrent thread 0x[0-9a-f]+zThread 0x[0-9a-f]+ZStackz (?m)^{fatal_error} {header} \(most recent call first\): File "", line {lineno} in z" Python runtime state: initialized)lineno fatal_errorr|rr4r r)rformatr&r:join assertRegexassertNotEqual)r2r3 line_numberr@rr;r<r4r=r>rrr6r8r r r check_errorPs&     zFaultHandlerTests.check_errorcKs2|rd||f}d|}|j|||fi|dS)Nz%s: %szFatal Python error: %srH)r2r3rG name_regexfunckwr@r r rcheck_fatal_errorts z#FaultHandlerTests.check_fatal_errorcKs"d|}|j|||fi|dS)NzWindows fatal exception: %srI)r2r3rGrJrLr@r r rcheck_windows_exceptionzsz)FaultHandlerTests.check_windows_exceptionZaixz5the first page of memory is a mapped read-only on AIXcCs&ts|dddn|ddddS)Nz import faulthandler faulthandler.enable() faulthandler._read_null() z4(?:Segmentation fault|Bus error|Illegal instruction)access violation) MS_WINDOWSrMrNr2r r rtest_read_null~s z FaultHandlerTests.test_read_nullcCs|ddddS)Nzs import faulthandler faulthandler.enable() faulthandler._sigsegv() rOSegmentation faultrMrRr r r test_sigsegvszFaultHandlerTests.test_sigsegvcCs|jddddddddS)Nz import faulthandler faulthandler.enable() faulthandler._fatal_error_c_thread() rOz in new threadFZfaulthandler_fatal_error_threadT)r=rKr>rUrRr r rtest_fatal_error_c_threadsz+FaultHandlerTests.test_fatal_error_c_threadcCs|ddddS)Nzs import faulthandler faulthandler.enable() faulthandler._sigabrt() rOZAbortedrUrRr r r test_sigabrtszFaultHandlerTests.test_sigabrtwin32z"SIGFPE cannot be caught on WindowscCs|ddddS)Nzr import faulthandler faulthandler.enable() faulthandler._sigfpe() rOzFloating point exceptionrUrRr r r test_sigfpeszFaultHandlerTests.test_sigfpezneed _testcapiSIGBUSzneed signal.SIGBUScCs|ddddS)Nz import faulthandler import signal faulthandler.enable() signal.raise_signal(signal.SIGBUS) z Bus errorrUrRr r r test_sigbusszFaultHandlerTests.test_sigbusSIGILLzneed signal.SIGILLcCs|ddddS)Nz import faulthandler import signal faulthandler.enable() signal.raise_signal(signal.SIGILL) r\zIllegal instructionrUrRr r r test_sigillszFaultHandlerTests.test_sigillcCs|jdddddddS)Nz[ import faulthandler faulthandler._fatal_error(b'xyz') xyzfaulthandler_fatal_error_pyTrKr>rUrRr r rtest_fatal_errors z"FaultHandlerTests.test_fatal_errorcCs|jdddddddS)Nza import faulthandler faulthandler._fatal_error(b'xyz', True) r`rarbTrcrUrRr r rtest_fatal_error_without_gils z.FaultHandlerTests.test_fatal_error_without_gilZopenbsdzVIssue #12868: sigaltstack() doesn't work on OpenBSD if Python is compiled with pthreadZ_stack_overflowz#need faulthandler._stack_overflow()cCs|jddddddS)Nzz import faulthandler faulthandler.enable() faulthandler._stack_overflow() rOz (?:Segmentation fault|Bus error)z unable to raise a stack overflow)r<rUrRr r rtest_stack_overflows z%FaultHandlerTests.test_stack_overflowcCs|ddddS)Nzw import faulthandler faulthandler.enable() faulthandler._sigsegv(True) rOrTrUrRr r rtest_gil_releasedsz#FaultHandlerTests.test_gil_releasedz0sanitizer builds change crashing process output.)ZmemoryZubreasoncCsHt.}|jdjt|ddd|dWdn1s:0YdS)Nz import faulthandler output = open({filename}, 'wb') faulthandler.enable(output) faulthandler._sigsegv() rrT)rrMrCreprr2rr r rtest_enable_filesz"FaultHandlerTests.test_enable_filez.subprocess doesn't support pass_fds on WindowscCsLtd.}|}|jd|dd|dWdn1s>0YdS)Nwb+z import faulthandler import sys faulthandler.enable(%s) faulthandler._sigsegv() rirTr4)r TemporaryFilefilenorM)r2r9r4r r rtest_enable_fds z FaultHandlerTests.test_enable_fdcCs|jddddddS)Nz import faulthandler faulthandler.enable(all_threads=False) faulthandler._sigsegv() rOrTFr;rUrRr r rtest_enable_single_thread)s z+FaultHandlerTests.test_enable_single_threadcCsHd}d}||\}}d|}|||vd||f||ddS)Nz import faulthandler faulthandler.enable() faulthandler.disable() faulthandler._sigsegv() zFatal Python errorr z%r is present in %rr)r:rD assertTruerF)r2r3Z not_expectedr7r8r r r test_disable4s   zFaultHandlerTests.test_disablecCstj}zztjt_t}zFt|tt|tW|rVtqxtn|rntnt0W|t_n|t_0dSr) sysr7 __stderr__ faulthandler is_enabledenablertdisableZ assertFalse)r2Z orig_stderrZ was_enabledr r rtest_is_enabledCs     z!FaultHandlerTests.test_is_enabledcCs0d}tjdd|f}t|}||ddS)N5import faulthandler; print(faulthandler.is_enabled())-ErFalse)rv executable subprocess check_outputr+rstrip)r2r3argsr6r r rtest_disabled_by_defaultYs z*FaultHandlerTests.test_disabled_by_defaultcCs`d}tdtjtjjrdndddd|f}tj}|ddt j ||d}| | d dS) Nr}r~r#z-XrxrPYTHONFAULTHANDLERenvTrue) filterrvrflagsignore_environmentr.environcopypoprrr+rr2r3rrr6r r rtest_sys_xoptionsas  z#FaultHandlerTests.test_sys_xoptionscCsd}tjd|f}ttj}d|d<d|d<tj||d}||dttj}d|d<d|d<tj||d}||d dS) Nr}rr#rZ PYTHONDEVMODErr1r) rvrdictr.rrrr+rrr r r test_env_varms   zFaultHandlerTests.test_env_varrBcCsld}|j||d}|rd}n|dur*d}nd}dd|dd g}||||\}}|||||d dS) Na[ import faulthandler filename = {filename!r} fd = {fd} def funcB(): if filename: with open(filename, "wb") as fp: faulthandler.dump_traceback(fp, all_threads=False) elif fd is not None: faulthandler.dump_traceback(fd, all_threads=False) else: faulthandler.dump_traceback(all_threads=False) def funcA(): funcB() funcA() rB Stack (most recent call first):z# File "", line %s in funcBz# File "", line 17 in funcAz& File "", line 19 in rrCr:r+)r2rr4r3r?expectedtracer8r r rcheck_dump_tracebacks$ z&FaultHandlerTests.check_dump_tracebackcCs |dSr)rrRr r rtest_dump_tracebacksz%FaultHandlerTests.test_dump_tracebackcCs6t}|j|dWdn1s(0YdSNr)rrrkr r rtest_dump_traceback_filesz*FaultHandlerTests.test_dump_traceback_filecCs>td }|j|dWdn1s00YdSNrmrn)rrorrpr2r9r r rtest_dump_traceback_fds z(FaultHandlerTests.test_dump_traceback_fdcCsdd}d|d}d|d}d}|j|d}dd|d g}||\}}|||||d dS) Nix2z...z import faulthandler def {func_name}(): faulthandler.dump_traceback(all_threads=False) {func_name}() ) func_namerz File "", line 4 in %sz% File "", line 6 in rr)r2maxlenrZ truncatedr3rrr8r r r test_truncates   zFaultHandlerTests.test_truncatecCspd}|jt|d}|||\}}d|}|r8d}nd}d}t|j|d}|||||ddS) Na import faulthandler from threading import Thread, Event import time def dump(): if {filename}: with open({filename}, "wb") as fp: faulthandler.dump_traceback(fp, all_threads=True) else: faulthandler.dump_traceback(all_threads=True) class Waiter(Thread): # avoid blocking if the main thread raises an exception. daemon = True def __init__(self): Thread.__init__(self) self.running = Event() self.stop = Event() def run(self): self.running.set() self.stop.wait() waiter = Waiter() waiter.start() waiter.running.wait() dump() waiter.stop.set() waiter.join() rr  a ^Thread 0x[0-9a-f]+ \(most recent call first\): (?: File ".*threading.py", line [0-9]+ in [_a-z]+ ){{1,3}} File "", line 23 in run File ".*threading.py", line [0-9]+ in _bootstrap_inner File ".*threading.py", line [0-9]+ in _bootstrap Current thread 0x[0-9a-f]+ \(most recent call first\): File "", line {lineno} in dump File "", line 28 in $ )r?r)rCrjr:rDrr&rEr+)r2rr3r6r8r?rr r rcheck_dump_traceback_threadss    z.FaultHandlerTests.check_dump_traceback_threadscCs|ddSr)rrRr r rtest_dump_traceback_threads sz-FaultHandlerTests.test_dump_traceback_threadscCs4t}||Wdn1s&0YdSr)rrrkr r r test_dump_traceback_threads_filesz2FaultHandlerTests.test_dump_traceback_threads_filer c Csttjtd}d}|jt|||||d}|||\}} d|}|s~|} |rX| d9} d|} tdd| | d } ||| n | |d | | d dS) N)Zsecondsa import faulthandler import time import sys timeout = {timeout} repeat = {repeat} cancel = {cancel} loops = {loops} filename = {filename!r} fd = {fd} def func(timeout, repeat, cancel, file, loops): for loop in range(loops): faulthandler.dump_traceback_later(timeout, repeat=repeat, file=file) if cancel: faulthandler.cancel_dump_traceback_later() time.sleep(timeout * 5) faulthandler.cancel_dump_traceback_later() if filename: file = open(filename, "wb") elif fd is not None: file = sys.stderr.fileno() else: file = None func(timeout, repeat, cancel, file, loops) if filename: file.close() )timeoutrepeatcancelloopsrr4r r`zATimeout \(%s\)!\nThread 0x[0-9a-f]+ \(most recent call first\):\n)rr#r) strdatetimeZ timedeltaTIMEOUTrCr:rDrrEr+) r2rrrrr4Z timeout_strr3rr8countrrr r rcheck_dump_traceback_laters*   z,FaultHandlerTests.check_dump_traceback_latercCs |dSrrrRr r rtest_dump_traceback_laterRsz+FaultHandlerTests.test_dump_traceback_latercCs|jdddS)NT)rrrRr r r test_dump_traceback_later_repeatUsz2FaultHandlerTests.test_dump_traceback_later_repeatcCs|jdddS)NT)rrrRr r r test_dump_traceback_later_cancelXsz2FaultHandlerTests.test_dump_traceback_later_cancelcCs6t}|j|dWdn1s(0YdSr)rrrkr r rtest_dump_traceback_later_file[sz0FaultHandlerTests.test_dump_traceback_later_filecCs>td }|j|dWdn1s00YdSr)rrorrprr r rtest_dump_traceback_later_fd_s z.FaultHandlerTests.test_dump_traceback_later_fdcCs|jdddS)Nr`)rrrRr r rtest_dump_traceback_later_twiceesz1FaultHandlerTests.test_dump_traceback_later_twiceregisterzneed faulthandler.registerc Cstj}d}|j||||||d}|||\}} d|}|sf|rHd} nd} tdd| } ||| n ||d|r|| d n || d dS) Nax import faulthandler import os import signal import sys all_threads = {all_threads} signum = {signum} unregister = {unregister} chain = {chain} filename = {filename!r} fd = {fd} def func(signum): os.kill(os.getpid(), signum) def handler(signum, frame): handler.called = True handler.called = False if filename: file = open(filename, "wb") elif fd is not None: file = sys.stderr.fileno() else: file = None if chain: signal.signal(signum, handler) faulthandler.register(signum, file=file, all_threads=all_threads, chain={chain}) if unregister: faulthandler.unregister(signum) func(signum) if chain and not handler.called: if file is not None: output = file else: output = sys.stderr print("Error: signal handler not called!", file=output) exitcode = 1 else: exitcode = 0 if filename: file.close() sys.exit(exitcode) )r;signum unregisterchainrr4r z8Current thread 0x[0-9a-f]+ \(most recent call first\):\nz#Stack \(most recent call first\):\nr r#r) signalSIGUSR1rCr:rDrrEr+rF) r2rr;rrr4rr3rr8rr r rcheck_registerhs, .   z FaultHandlerTests.check_registercCs |dSrrrRr r r test_registerszFaultHandlerTests.test_registercCs|jdddS)NT)rrrRr r rtest_unregistersz!FaultHandlerTests.test_unregistercCs6t}|j|dWdn1s(0YdSr)rrrkr r rtest_register_filesz$FaultHandlerTests.test_register_filecCs>td }|j|dWdn1s00YdSr)rrorrprr r rtest_register_fds z"FaultHandlerTests.test_register_fdcCs|jdddS)NTrrrrRr r rtest_register_threadssz'FaultHandlerTests.test_register_threadscCs|jdddS)NT)rrrRr r rtest_register_chainsz%FaultHandlerTests.test_register_chainccsftj}zRdt_|t}dVWdn1s40Y|t|jdW|t_n|t_0dS)Nzsys.stderr is None)rvr7Z assertRaises RuntimeErrorr+r exception)r2r7cmr r rcheck_stderr_nones $z#FaultHandlerTests.check_stderr_nonecCs|tWdn1s&0Y|tWdn1sV0Y|tdWdn1s0Yttdr|ttjWdn1s0YdS)NgMbP?r) rrxrzZdump_tracebackZdump_traceback_laterhasattrrrrrRr r rtest_stderr_Nones & & (  z"FaultHandlerTests.test_stderr_Nonezspecific to WindowscCs(dD]\}}|d|dd|qdS)N))ZEXCEPTION_ACCESS_VIOLATIONrP)ZEXCEPTION_INT_DIVIDE_BY_ZEROzint divide by zero)ZEXCEPTION_STACK_OVERFLOWzstack overflowz import faulthandler faulthandler.enable() faulthandler._raise_exception(faulthandler._) rO)rN)r2excnamer r rtest_raise_exceptions z&FaultHandlerTests.test_raise_exceptioncCsHdD]>}d|d}t|}||\}}||g|||qdS)N)lcs@lRC@z import faulthandler faulthandler.enable() faulthandler._raise_exception(z) rr:r+)r2Zexc_coder3r6r8r r rtest_ignore_exceptions z'FaultHandlerTests.test_ignore_exceptioncCsFdD]<}|d|dd\}}||g||||d@fqdS)N)rixV4i@i@ipiz{ import faulthandler faulthandler.enable() faulthandler._raise_exception(0xrri)r:r+ZassertIn)r2rr6r8r r rtest_raise_nonfatal_exceptions  z/FaultHandlerTests.test_raise_nonfatal_exceptioncCs2td}||\}}||g||ddS)Nz import faulthandler faulthandler.enable() faulthandler.disable() code = faulthandler._EXCEPTION_ACCESS_VIOLATION faulthandler._raise_exception(code) lrr2r3r6r8r r r test_disable_windows_exc_handler$s z2FaultHandlerTests.test_disable_windows_exc_handlercCs2td}||\}}||g||ddS)Nz` import faulthandler faulthandler.cancel_dump_traceback_later() rrrr r r.test_cancel_later_without_dump_traceback_later1s z@FaultHandlerTests.test_cancel_later_without_dump_traceback_later)NN)N)FFr )FFFFN)E__name__ __module__ __qualname__r:rHrMrNrrrvplatform startswithrSrrVrWrXrZ _testcapiZ skipUnlessrrr]r_rdrerxrfrgrrlrqrsrur|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrQrrrrrr r r rr.s " $                  .  ; >   Q           r__main__)r ) contextlibrrrxr.rrrvrrZ test.supportrrrrrtextwraprr ImportErrorrrrQrrrZTestCaserrmainr r r rs>