/
opt
/
gsutil
/
gslib
/
__pycache__
/
up file
home
a +(Wg� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZ dd lmZ dd lmZ er�ddlZi Zi Zdd d �Zdd� Zdd� Zdd� Zdd� Zdd� Z dd� Z dS )zSignal handling functions.� )�absolute_import)�print_function)�division)�unicode_literalsN)�metrics)�ControlCException)�UTF8)� IS_WINDOWSFc C s: | t � vrtd|| f ��|r(|t| < nt| �|� dS )a� Registers 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_handlers�append)� signal_num�handler�is_final_handler� r �!/opt/gsutil/gslib/sig_handling.py�RegisterSignalHandler, s � r c C s: | t v r t | D ]}|| |� q| tv r6t| | |� dS )z�Global 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)r r )r �cur_stack_framer r r r �_SignalHandlerO s r c C s$ t � D ]} g t| < t�| t� qdS )zbInitializes global signal handling. Sets up global signal handler for each signal we handle. N)r r �signalr )r r r r �InitializeSignalHandling_ s r c C s t jt jg} ts | �t j� | S )zCReturns terminating signals that can be caught on this OS platform.)r �SIGINT�SIGTERMr r �SIGQUIT)Zsignalsr r r r j s r c C sn z\t rLdtjdd� kr$dk rLn n$tjj}|�dd| �}|�|d� nt� | t j � W n Y n0 dS )z�Make 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. )� r Nr )r � � r ) r �sys� version_info�ctypes�windll�kernel32Z OpenProcessZTerminateProcess�os�killr �SIGKILL)�pidr# �handler r r � KillProcesss s &r) c C s� | t jkr�t�� �tj�r�d�t�t� � ��}d| t � dd|�f }zt j �|�t�� W q� ttfy� } zt j �|� W Y d}~q�d}~0 0 nt j �d| � tjt� d� t�� tt�� � dS )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)r r �logging� getLogger� isEnabledFor�DEBUG�join� traceback� format_list� extract_stack�re�subr �stderr�write�encoder �UnicodeDecodeError� TypeErrorr � LogFatalErrorr �Shutdownr) r$ �getpid)r r � stack_trace�err�er r r � MultithreadedMainSignalHandler� s �$rA c C s t t�� � dS )z)Final signal handler for child processes.N)r) r$ r= )r r r r r �ChildProcessSignalHandler� s rB )F) �__doc__� __future__r r r r r, r$ r4 r r r1 �gslibr �gslib.exceptionr �gslib.utils.constantsr �gslib.utils.system_utilr r! r r r r r r r) rA rB r r r r �<module> s2 #