3
\ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZddlmZ ddlmZ dddgZ G dd � d e
�Z e � Z e j
Z
e jZe jZe jZd
d � ZdS ) � N� )�spawn)�util�ensure_running�register�
unregisterc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�SemaphoreTrackerc C s t j� | _d | _d | _d S )N)� threadingZLock�_lock�_fd�_pid)�self� r �9/usr/lib64/python3.6/multiprocessing/semaphore_tracker.py�__init__ s
zSemaphoreTracker.__init__c C s | j � | jS )N)r r )r
r r r �getfd" s zSemaphoreTracker.getfdc C s | j �� | jdk rRtj| jtj�\}}|s0dS tj| j� d| _d| _tjd� g }y|j t
j j � � W n t
k
r~ Y nX d}tj� \}}zlyB|j |� tj� }|gtj� }|d|| g7 }tj|||�}W n tj|� � Y nX || _|| _W dtj|� X W dQ R X dS )z�Make sure that semaphore tracker process is running.
This can be run from any process. Usually a child process will use
the semaphore created by its parent.NzWsemaphore_tracker: process died unexpectedly, relaunching. Some semaphores might leak.z;from multiprocessing.semaphore_tracker import main;main(%d)z-c)r
r �os�waitpid�WNOHANG�closer �warnings�warn�append�sys�stderr�fileno� Exception�piper Zget_executabler Z _args_from_interpreter_flagsZspawnv_passfds) r
�pidZstatusZ fds_to_pass�cmd�r�wZexe�argsr r r r & s8
z SemaphoreTracker.ensure_runningc C s | j d|� dS )z2Register name of semaphore with semaphore tracker.ZREGISTERN)�_send)r
�namer r r r Q s zSemaphoreTracker.registerc C s | j d|� dS )z4Unregister name of semaphore with semaphore tracker.Z
UNREGISTERN)r# )r
r$ r r r r U s zSemaphoreTracker.unregisterc C sP | j � dj||�jd�}t|�dkr.td��tj| j|�}|t|�ksLt�d S )Nz{0}:{1}
�asciii z
name too long) r �format�encode�len�
ValueErrorr �writer �AssertionError)r
r r$ �msg�nbytesr r r r# Y s zSemaphoreTracker._sendN) �__name__�
__module__� __qualname__r r r r r r# r r r r r s +r c - C s� t j t jt j� t j t jt j� x6tjtjfD ]&}y |j� W q. tk
rR Y q.X q.W t � }z�t
| d���}x�|D ]�}yJ|j � j d�\}}|dkr�|j
|� n |dkr�|j|� n td| ��W qr tk
r� ytjtj� � W n Y nX Y qrX qrW W dQ R X W d|�r<ytjdt|� � W n tk
�r: Y nX xf|D ]^}zT|jd�}ytj|� W n6 tk
�r� } ztjd ||f � W Y dd}~X nX W dX �qBW X dS )
zRun semaphore tracker.�rb� :s REGISTERs
UNREGISTERzunrecognized command %rNzRsemaphore_tracker: There appear to be %d leaked semaphores to clean up at shutdownr% zsemaphore_tracker: %r: %s)�signal�SIGINT�SIG_IGN�SIGTERMr �stdin�stdoutr r �set�open�strip�split�add�remove� RuntimeError�
excepthook�exc_infor r r( �decode�_multiprocessingZ
sem_unlink)�fd�f�cache�liner r$ �er r r �maink sJ
(rI )r r3 r r r rC � r r �__all__�objectr Z_semaphore_trackerr r r r rI r r r r �<module> s
I