CasperSecurity
o
`�wiZ- � @ s g d �Z ddlZddlZddlZddlZddlZddlmZ ddlmZ ddlm Z z
ddlm
Z
m Z W n e y? e d��w e
ed ��\ZZej
jZG d
d � d e�Z
G d d
� d
e
�ZG dd� de�ZG dd� de
�ZG dd� de
�ZG dd� de�ZG dd� de�ZG dd� dej�ZdS ))�Lock�RLock� Semaphore�BoundedSemaphore� Condition�Event� N� )�context)�process)�util)�SemLock�
sem_unlinkz�This platform lacks a functioning sem_open implementation, therefore, the required synchronization primitives needed will not function, see issue 3770.� c @ s\ e Zd Ze�� Zdd� Zedd� �Zdd� Z dd� Z
d d
� Z d d � Z d
d� Z
edd� �ZdS )r c C s� |d u r t j�� }|�� }tjdkp|dk}td�D ] }zt�|||| � � |� }| _
W n t y5 Y qw nt d��t �
d|j � | �� tjdkrWdd� } t �| | � | j
jd urydd lm}
|
| j
jd
� t j| tj| j
jfd d � d S d S )
N�win32�fork�d z cannot find name for semaphorez created semlock with handle %sc S s | j �� d S �N)�_semlock� _after_fork)�obj� r �2/usr/lib/python3.10/multiprocessing/synchronize.pyr G s z%SemLock.__init__.<locals>._after_forkr )�register� semaphorer )� exitpriority)r �_default_context� get_context�get_start_method�sys�platform�range�_multiprocessingr �
_make_namer �FileExistsErrorr �debug�handle�
_make_methods�register_after_fork�name�resource_trackerr �Finalize�_cleanup) �self�kind�value�maxvalue�ctxr( �
unlink_now�i�slr r r r r �__init__2 s6
� �
��zSemLock.__init__c C s" ddl m} t| � || d� d S )Nr )�
unregisterr )r) r5 r
)r( r5 r r r r+ T s zSemLock._cleanupc C � | j j| _| j j| _d S r )r �acquire�release�r, r r r r& Z �
zSemLock._make_methodsc C �
| j �� S r )r � __enter__r9 r r r r< ^ �
zSemLock.__enter__c G � | j j|� S r )r �__exit__�r, �argsr r r r? a � zSemLock.__exit__c C sD t �| � | j}tjdkrt �� �|j�}n|j}||j|j |j
fS )Nr ) r �assert_spawningr r r �get_spawning_popen�duplicate_for_childr% r- r/ r( )r, r3 �hr r r � __getstate__d s
zSemLock.__getstate__c C s, t jj|� | _t�d|d � | �� d S )Nz recreated blocker with handle %rr )r! r �_rebuildr r r$ r& �r, �stater r r � __setstate__m s zSemLock.__setstate__c C s dt �� jd ttj�f S )Nz%s-%s� semprefix)r
�current_process�_config�nextr �_randr r r r r" r s �zSemLock._make_nameN)�__name__�
__module__� __qualname__�tempfile�_RandomNameSequencerP r4 � staticmethodr+ r& r< r? rG rK r" r r r r r . s "
r c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
r r c C s t j| t|t|d� d S �N�r0 )r r4 � SEMAPHORE�
SEM_VALUE_MAX�r, r. r0 r r r r4 } � zSemaphore.__init__c C r; r )r �
_get_valuer9 r r r � get_value� r= zSemaphore.get_valuec C s6 z| j �� }W n ty d}Y nw d| jj|f S )N�unknownz<%s(value=%s)>)r r] � Exception� __class__rQ �r, r. r r r �__repr__� s �zSemaphore.__repr__N�r )rQ rR rS r4 r^ rc r r r r r { s
r c @ s e Zd Zddd�Zdd� ZdS )r r c C s t j| t|||d� d S rW �r r4 rY r[ r r r r4 � r\ zBoundedSemaphore.__init__c C s< z| j �� }W n ty d}Y nw d| jj|| j jf S )Nr_ z<%s(value=%s, maxvalue=%s)>)r r] r` ra rQ r/ rb r r r rc � s ��zBoundedSemaphore.__repr__Nrd �rQ rR rS r4 rc r r r r r � s
r c @ � e Zd Zdd� Zdd� ZdS )r c C � t j| tdd|d� d S �Nr rX re �r, r0 r r r r4 � r\ z
Lock.__init__c C s� z3| j �� r t�� j}t�� jdkr|dt�� j 7 }n| j �� dkr&d}n | j �� dkr0d}nd}W n t y> d}Y nw d | j
j |f S )
N�
MainThread�|r �Noner �SomeOtherThread�SomeOtherProcessr_ z<%s(owner=%s)>) r �_is_miner
rM r( � threading�current_threadr] �_countr` ra rQ )r, r( r r r rc � s
�� �z
Lock.__repr__Nrf r r r r r � � r c @ rg )r c C rh ri )r r4 �RECURSIVE_MUTEXrj r r r r4 � r\ zRLock.__init__c C s� z>| j �� r!t�� j}t�� jdkr|dt�� j 7 }| j �� }n | j �� dkr-d\}}n| j �� dkr9d\}}nd\}}W n
t yK d\}}Y nw d | j
j ||f S )
Nrk rl r )rm r r )rn �nonzero)ro rv )r_ r_ � <%s(%s, %s)>) r rp r
rM r( rq rr rs r] r` ra rQ )r, r( �countr r r rc � s
� �zRLock.__repr__Nrf r r r r r � rt r c @ sl e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z ddd�Z
ddd�Z dd� Z ddd�Z
dS )r Nc C s> |p|� � | _|�d�| _|�d�| _|�d�| _| �� d S �Nr )r �_lockr �_sleeping_count� _woken_count�_wait_semaphorer& )r, �lockr0 r r r r4 � s
zCondition.__init__c C s t �| � | j| j| j| jfS r )r rC rz r{ r| r} r9 r r r rG � s
�zCondition.__getstate__c C s |\| _ | _| _| _| �� d S r )rz r{ r| r} r& rI r r r rK � s
� zCondition.__setstate__c C r; r )rz r< r9 r r r r< � r= zCondition.__enter__c G r> r )rz r? r@ r r r r? � rB zCondition.__exit__c C r6 r )rz r7 r8 r9 r r r r& � r: zCondition._make_methodsc C sH z| j j�� | jj�� }W n ty d}Y nw d| jj| j|f S )Nr_ rw )r{ r r] r| r` ra rQ rz )r, � num_waitersr r r rc � s
� �zCondition.__repr__c C s� | j j�� s
J d��| j�� | j j�� }t|�D ]}| j �� qz| j�d|�W | j �� t|�D ]}| j �� q2S | j �� t|�D ]}| j �� qDw )Nz,must acquire() condition before using wait()T)
rz r rp r{ r8 rs r r} r7 r| )r, �timeoutrx r2 r r r �wait� s" �
�
� �zCondition.waitr c C s� | j j�� s
J d��| j�d�rJ d��| j�d�r,| j�d�}|s&J d��| j�d�sd}||k rK| j�d�rK| j�� |d7 }||k rK| j�d�s8|rht|�D ]}| j�� qQ| j�d�rj | j�d�s_d S d S d S )Nzlock is not ownedFz<notify: Should not have been able to acquire _wait_semaphorez>notify: Bug in sleeping_count.acquire- res should not be Falser r ) rz r rp r} r7 r| r{ r8 r )r, �n�res�sleepersr2 r r r �notify s. �� �
� ��zCondition.notifyc C s | j tjd� d S )N)r� )r� r �maxsizer9 r r r �
notify_all( s zCondition.notify_allc C sj |� }|r|S |d urt �� | }nd }d }|s3|d ur)|t �� }|dkr) |S | �|� |� }|r|S ry )�time� monotonicr� )r, � predicater� �result�endtime�waittimer r r �wait_for+ s"
��zCondition.wait_forr rd )rQ rR rS r4 rG rK r< r? r& rc r� r� r� r� r r r r r � s
r c @ s6 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
d �Zd S )
r c C s |� |�� �| _|�d�| _d S ry )r r �_condr �_flagrj r r r r4 C s zEvent.__init__c C sV | j � | j�d�r| j�� W d � dS W d � dS 1 s$w Y d S �NFT)r� r� r7 r8 r9 r r r �is_setG s
�$�z Event.is_setc C sL | j � | j�d� | j�� | j �� W d � d S 1 s w Y d S �NF)r� r� r7 r8 r� r9 r r r �setN s
"�z Event.setc C s8 | j � | j�d� W d � d S 1 sw Y d S r� )r� r� r7 r9 r r r �clearT s "�z Event.clearNc C sz | j �0 | j�d�r| j�� n| j �|� | j�d�r*| j�� W d � dS W d � dS 1 s6w Y d S r� )r� r� r7 r8 r� )r, r� r r r r� X s
� $�z
Event.waitr )rQ rR rS r4 r� r� r� r� r r r r r A s r c @ sZ e Zd Zddd�Zdd� Zdd� Zedd � �Zejd
d � �Zed d � �Z e jd
d � �Z dS )�BarrierNc C sR dd l }ddlm} ||�d�d �}|�� }| �|||||f� d| _d| _d S )Nr r )�
BufferWrapperr2 r )�struct�heapr� �calcsizer rK �_staters ) r, �parties�actionr� r0 r� r� �wrapper�condr r r r4 j s
zBarrier.__init__c C s. |\| _ | _| _| _| _| j�� �d�| _d S )Nr2 )�_parties�_action�_timeoutr� �_wrapper�create_memoryview�cast�_arrayrI r r r rK s s �zBarrier.__setstate__c C s | j | j| j| j| jfS r )r� r� r� r� r� r9 r r r rG x s �zBarrier.__getstate__c C �
| j d S ry �r� r9 r r r r� | �
zBarrier._statec C � || j d< d S ry r� rb r r r r� � � c C r� �Nr r� r9 r r r rs � r� zBarrier._countc C r� r� r� rb r r r rs � r� )NN)
rQ rR rS r4 rK rG �propertyr� �setterrs r r r r r� h s
r� )�__all__rq r rT r! r� � r r
r r r
� ImportError�listr ru rY rZ �objectr r r r r r r� r r r r �<module> s0 �
Mo'