CasperSecurity
o
`�wi�. � @ s� g d �Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z ddl Z ddl
mZ ddl
mZ ejjZddlmZmZmZmZmZ G dd � d e�Ze� ZG d
d � d e�Z G d d
� d
e�Z dS ))�Queue� SimpleQueue�
JoinableQueue� N)�Empty�Full� )�
connection)�context)�debug�info�Finalize�register_after_fork�
is_exitingc @ s� e Zd Zd-dd�Zdd� Zdd� Zdd � Zd.d d �Zd/dd�Zd/dd�Z dd� Z
dd� Z dd� Z dd� Z
dd � Zd d � Zd d � Zd!d"� Zd#d$� Zed%d&� �Zed'd(� �Zed)d*� �Zed+d,� �ZdS )0r r c C s� |dkr
ddl m} || _tjdd�\| _| _|�� | _t �
� | _ t j
dkr*d | _n|�� | _|�|�| _d| _| �� t j
dkrIt| tj� d S d S )Nr r )�
SEM_VALUE_MAXF��duplex�win32)� synchronizer �_maxsizer �Pipe�_reader�_writer�Lock�_rlock�os�getpid�_opid�sys�platform�_wlock�BoundedSemaphore�_sem�
_ignore_epipe�_resetr
r � _after_fork��self�maxsize�ctx� r) �-/usr/lib/python3.10/multiprocessing/queues.py�__init__% s
�zQueue.__init__c C s. t �| � | j| j| j| j| j| j| j| j fS �N)
r �assert_spawningr" r r r r r r! r �r& r) r) r* � __getstate__9 s
�zQueue.__getstate__c C s0 |\| _ | _| _| _| _| _| _| _| �� d S r, ) r" r r r r r r! r r# �r& �stater) r) r* � __setstate__> s � zQueue.__setstate__c C s t d� | jdd� d S )NzQueue._after_fork()T)�
after_fork)r
r# r. r) r) r* r$ C s zQueue._after_forkFc C sj |r| j �� nt�t�� �| _ t�� | _d | _d | _ d| _
d| _ d | _ | j
j| _| jj| _| jj| _d S �NF)� _notempty�_at_fork_reinit� threading� Conditionr � collections�deque�_buffer�_thread� _jointhread�_joincancelled�_closed�_closer �
send_bytes� _send_bytesr �
recv_bytes� _recv_bytes�poll�_poll)r& r3 r) r) r* r# G s
z Queue._resetTNc C s| | j r td| �d���| j�||�st�| j� | jd u r!| �� | j� |� | j�
� W d � d S 1 s7w Y d S �NzQueue z
is closed) r? �
ValueErrorr! �acquirer r5 r< �
_start_threadr; �append�notify�r& �obj�block�timeoutr) r) r* �putV s
"�z Queue.putc C s� | j r td| �d���|r.|d u r.| j� | �� }W d � n1 s#w Y | j�� n>|r6t�� | }| j�||�s?t �z&|rP|t�� }| �
|�sOt �n| �
� sVt �| �� }| j�� W | j�� n| j�� w t � |�S rG )
r? rH r rD r! �release�time� monotonicrI r rF �_ForkingPickler�loads)r& rO rP �res�deadliner) r) r* �getb s.
�
�
z Queue.getc C s | j | jj�� S r, )r r! �_semlock�
_get_valuer. r) r) r* �qsize| s z Queue.qsizec C �
| � � S r, �rF r. r) r) r* �empty� �
z Queue.emptyc C s | j j�� S r, )r! rZ �_is_zeror. r) r) r* �full� � z
Queue.fullc C s
| � d�S r4 )rY r. r) r) r* �
get_nowait� r` zQueue.get_nowaitc C s | � |d�S r4 )rQ �r& rN r) r) r* �
put_nowait� rc zQueue.put_nowaitc C s$ d| _ | j}|rd | _|� d S d S )NT)r? r@ )r& �closer) r) r* rg � s
�z Queue.closec C s2 t d� | jsJ d�| ���| jr| �� d S d S )NzQueue.join_thread()zQueue {0!r} not closed)r
r? �formatr= r. r) r) r* � join_thread� s
�zQueue.join_threadc C s4 t d� d| _z| j�� W d S ty Y d S w )NzQueue.cancel_join_thread()T)r
r> r= �cancel�AttributeErrorr. r) r) r* �cancel_join_thread� s �zQueue.cancel_join_threadc C s� t d� | j�� tjtj| j| j| j| j | j
j | j j | j
| j| jf dd�| _d| j_t d� | j�� t d� | jsKt| jtjt�| j�gdd�| _t| tj| j| jgd d�| _d S )
NzQueue._start_thread()�QueueFeederThread)�target�args�nameTzdoing self._thread.start()z ... done self._thread.start()���)� exitpriority�
)r
r; �clearr7 �Threadr �_feedr5 rB r r rg r r" �_on_queue_feeder_errorr! r< �daemon�startr> r �_finalize_join�weakref�refr= �_finalize_closer@ r. r) r) r* rJ � s4
��
�
�zQueue._start_threadc C s6 t d� | � }|d ur|�� t d� d S t d� d S )Nzjoining queue threadz... queue thread joinedz ... queue thread already dead)r
�join)�twr�threadr) r) r* rz � s zQueue._finalize_joinc C sD t d� |� | �t� |�� W d � d S 1 sw Y d S )Nz telling queue thread to quit)r
rK � _sentinelrL )�buffer�notemptyr) r) r* r} � s
"�zQueue._finalize_closec
C sf t d� |j} |j}
|j} | j} t}
tjdkr |j}|j}nd } zT| � z
| s+| � W |
� n|
� w z6 | � }||
u rKt d� |� |� W W d S t� |�}|d u rY||� n|� z ||� W |� n|� w q6 t
ys Y nw W n< t y� } z0|r�t |dd�t
jkr�W Y d }~d S t� r�td|� W Y d }~d S |�� |||� W Y d }~nd }~ww q!)Nz$starting thread to feed data to piper r z%feeder thread got sentinel -- exiting�errnor zerror in queue thread: %s)r
rI rR �wait�popleftr� r r rU �dumps�
IndexError� Exception�getattrr� �EPIPEr r )r� r� rA � writelock� reader_close� writer_close� ignore_epipe�onerror� queue_sem�nacquire�nrelease�nwait�bpopleft�sentinel�wacquire�wreleaserN �er) r) r* rv � sb
�
� �
���z Queue._feedc C s ddl }|�� dS )z�
Private API hook called when feeding data in the background thread
raises an exception. For overriding by concurrent.futures.
r N)� traceback� print_exc)r� rN r� r) r) r* rw s z Queue._on_queue_feeder_error�r )F�TN)�__name__�
__module__� __qualname__r+ r/ r2 r$ r# rQ rY r\ r_ rb rd rf rg ri rl rJ � staticmethodrz r} rv rw r) r) r) r* r # s2
!
>r c @ s@ e Zd Zddd�Zdd� Zdd� Zdd
d �Zd d
� Zdd� Zd S )r r c C s* t j| ||d� |�d�| _|�� | _d S )N)r( r )r r+ � Semaphore�_unfinished_tasksr8 �_condr% r) r) r* r+ ( s zJoinableQueue.__init__c C s t �| �| j| jf S r, )r r/ r� r� r. r) r) r* r/ - s zJoinableQueue.__getstate__c C s, t �| |d d� � |dd � \| _| _d S )N���)r r2 r� r� r0 r) r) r* r2 0 s zJoinableQueue.__setstate__TNc C s� | j r td| �d���| j�||�st�| j�= | j�! | jd u r%| �� | j �
|� | j � � | j�
� W d � n1 s?w Y W d � d S W d � d S 1 sWw Y d S rG )r? rH r! rI r r5 r� r< rJ r; rK r� rR rL rM r) r) r* rQ 4 s
P�zJoinableQueue.putc C sf | j �& | j�d�std��| jj�� r!| j �� W d � d S W d � d S 1 s,w Y d S )NFz!task_done() called too many times)r� r� rI rH rZ ra �
notify_allr. r) r) r* � task_doneA s �"�zJoinableQueue.task_donec C sR | j � | jj�� s| j �� W d � d S W d � d S 1 s"w Y d S r, )r� r� rZ ra r� r. r) r) r* r~ H s �"�zJoinableQueue.joinr� r� ) r� r� r� r+ r/ r2 rQ r� r~ r) r) r) r* r & s
r c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
e j �Z
dS )r c C sJ t jdd�\| _| _|�� | _| jj| _tj dkr d | _
d S |�� | _
d S )NFr r ) r r r r r r rE rF r r r )r& r( r) r) r* r+ S s
zSimpleQueue.__init__c C s | j �� | j�� d S r, )r rg r r. r) r) r* rg \ s
zSimpleQueue.closec C r] r, r^ r. r) r) r* r_ ` r` zSimpleQueue.emptyc C s t �| � | j| j| j| jfS r, )r r- r r r r r. r) r) r* r/ c s
zSimpleQueue.__getstate__c C s" |\| _ | _| _| _| j j| _d S r, )r r r r rE rF r0 r) r) r* r2 g s zSimpleQueue.__setstate__c C s: | j �
| j�� }W d � n1 sw Y t�|�S r, )r r rC rU rV )r& rW r) r) r* rY k s �
zSimpleQueue.getc C s\ t �|�}| jd u r| j�|� d S | j� | j�|� W d � d S 1 s'w Y d S r, )rU r� r r rA re r) r) r* rQ q s
"�zSimpleQueue.putN)r� r� r� r+ rg r_ r/ r2 rY rQ � classmethod�types� GenericAlias�__class_getitem__r) r) r) r* r Q s
r ) �__all__r r r7 r9 rS r� r{ r� �queuer r �_multiprocessing� r r � reduction�ForkingPicklerrU �utilr
r r r
r �objectr r� r r r) r) r) r* �<module> s( z
+