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 �� | jd k rRtj| jtj�\}}|s0d S 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 d tj|� X W d Q R X d S )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|� d S )NZREGISTER)�_send)r
�namer r r r Q s zSemaphoreTracker.registerc C s | j d|� d S )NZ
UNREGISTER)r# )r
r$ r r r r U s zSemaphoreTracker.unregisterc C s@ | j � dj||�jd�}t|�dkr.td��tj| j|�}d S )Nz{0}:{1}
�asciii z
name too long)r �format�encode�len�
ValueErrorr �writer )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 d Q 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 d d }~X nX W d X �qBW X d S ) N�rb� :s REGISTERs
UNREGISTERzunrecognized command %rzRsemaphore_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
(rH )r r2 r r r rB � r r �__all__�objectr Z_semaphore_trackerr r r r rH r r r r �<module> s
I