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)utilensure_runningregister
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 )zMake 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 oswaitpidWNOHANGcloser warningswarnappendsysstderrfileno Exceptionpiper Zget_executabler Z _args_from_interpreter_flagsZspawnv_passfds) r
pidZstatusZ fds_to_passcmdrwZexeargsr 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 s@ | j dj||jd}t|dkr.tdtj| j|}d S )Nz{0}:{1}
asciii z
name too long)r formatencodelen
ValueErrorr writer )r
r r$ msgnbytesr 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 }zt
| 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 s
I