dZddlmZddlmZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZdd lmZdd lmZerddlZiZiZdd Zd Zd ZdZdZdZdZy)zSignal handling functions.)absolute_import)print_function)division)unicode_literalsN)metrics)ControlCException)UTF8) IS_WINDOWSc|tvrtd||fz|r |t|<yt|j |y)aRegisters a handler for signal signal_num. Unlike calling signal.signal(): - This function can be called from any thread (and will cause the handler to be run by the main thread when the signal is received). - Handlers are cumulative: When a given signal is received, all registered handlers will be executed (with the exception that only the last handler to register with is_final_handler=True will be called). Handlers should make no ordering assumptions, other than that the last handler to register with is_final_handler=True will be called after all the other handlers. Args: signal_num: The signal number with which to associate handler. handler: The handler. is_final_handler: Bool indicator whether handler should be called last among all the handlers for this signal_num. The last handler to register this way survives; other handlers registered with is_final_handler=True will not be called when the signal is received. Raises: RuntimeError: if attempt is made to register a signal_num not in GetCaughtSignals. zPAttempt to register handler (%s) for signal %d, which is not in GetCaughtSignalsN)GetCaughtSignals RuntimeError_final_signal_handlers_non_final_signal_handlersappend) signal_numhandleris_final_handlers %platform/gsutil/gslib/sig_handling.pyRegisterSignalHandlerr,sS4')) 47> 6KL MM)0:&z*11':cr|tvrt|D] }||| |tvrt|||yy)zGlobal signal handler. When a signal is caught we execute each registered handler for that signal. Args: signal_num: Signal that was caught. cur_stack_frame: Unused. N)rr)rcur_stack_framers r_SignalHandlerrOsC---j9 j/*:)):&z?C*rchtD]%}gt|<tj|t'y)zbInitializes global signal handling. Sets up global signal handler for each signal we handle. N)r rsignalr)rs rInitializeSignalHandlingr_s* %&j-/z* MM*n-'rctjtjg}ts|j tj |S)zCReturns terminating signals that can be caught on this OS platform.)rSIGINTSIGTERMr rSIGQUIT)signalss rr r js, ]]FNN +'  NN6>>" .rc& tr_dtjddcxkrdkrCnn@tjj }|j dd|}|j|dytj|tjy#YyxYw)zMake best effort to kill the given process. We ignore all exceptions so a caller looping through a list of processes will continue attempting to kill each, even if one encounters a problem. Args: pid: The process ID. )rNr#)r#r) r sys version_infoctypeswindllkernel32 OpenProcessTerminateProcessoskillrSIGKILL)pidr*handles r KillProcessr2ssr  v!1!1"1!5>>''h##Aq#.f*ggc6>>" sA$B '$B Bc|tjk(rtjj tj rdj tjtj}d|tjdd|fz} tjj|jt n"tjjd|zt'j(t+t'j,t/t1j2y#t"t$f$r)}tjj|Yd}~d}~wwxYw)z5Final signal handler for multi-threaded main process.z@DEBUG: Caught CTRL-C (signal %d) - Exception stack trace: %sz\nz Nz$Caught CTRL-C (signal %d) - exiting ) exception)rrlogging getLogger isEnabledForDEBUGjoin traceback format_list extract_stackresubr&stderrwriteencoder UnicodeDecodeError TypeErrorr LogFatalErrorrShutdownr2r-getpid)rr stack_traceerres rMultithreadedMainSignalHandlerrKs6== '' 6GGI11)2I2I2KLMk#RVVE8[%IJKc D)* jj>KL "3"56  biik!) ,  s2D==E5 E00E5c>ttjy)z)Final signal handler for child processes.N)r2r-rG)rrs rChildProcessSignalHandlerrMs biikr)F)__doc__ __future__rrrrr6r-r>rr&r;gslibrgslib.exceptionrgslib.utils.constantsr gslib.utils.system_utilr r(rrrrrr r2rKrMrrrUso!&%' -&.   ;F D . ,,r