CasperSecurity
o
`�wi�{ � @ s� g d �Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl
m Z m
Z
ddlmZ ejZzddlZddlmZmZmZmZ W n eyb ejdkr^� dZY nw d Zd
Ze�� Z d Z d gZ e ed �r|d Z e d g7 Z ejdkr�d
Z e d
g7 Z efdd�Z dd� Z!dd� Z"dd� Z#dd� Z$G dd� d�Z%er�G dd� de%�Z&G d d � d e%�Z'G d d � d e(�Z)dPd d!�Z*ejdkr�dQd#d$�Z+ndQd%d$�Z+G d&d'� d'e(�Z,d(d)� Z-ejdkr�G d*d+� d+e(�Z.d,d-� Z/d.Z0d/Z1d0Z2d1Z3d2d3� Z4d4d5� Z5G d6d7� d7e(�Z6d8d9� Z7d:d;� Z8G d<d=� d=e)�Z9d>d?� Z:ejdk�r3d@dA� Z;ej<ej=hZ>dRdBdC�Z?nddl@Z@e e@dD��rAe@jAZBne@jCZBdRdEdC�Z?ejdk�rmdFdG� ZDdHdI� ZEe�Fe'eD� dJdK� ZGdLdM� ZHe�Fe&eG� dS dNdG� ZDdOdI� ZEe�Fe'eD� dS )S)�Client�Listener�Pipe�wait� N� )�util)�AuthenticationError�BufferTooShort)� reduction)�
WAIT_OBJECT_0�WAIT_ABANDONED_0� WAIT_TIMEOUT�INFINITE�win32i g 4@�AF_INET�AF_UNIX�AF_PIPEc C s t �� | S �N��time� monotonic)�timeout� r �1/usr/lib/python3.10/multiprocessing/connection.py�
_init_timeout; � r c C s t �� | kS r r )�tr r r �_check_timeout> r r c C sT | dkrdS | dkrt jdt�� d�S | dkr&t jdt�� tt�f dd�S td ��)
z?
Return an arbitrary free address for the given family
r )� localhostr r z listener-)�prefix�dirr z\\.\pipe\pyc-%d-%d-� zunrecognized family) �tempfile�mktempr � get_temp_dir�os�getpid�next�
_mmap_counter�
ValueError��familyr r r �arbitrary_addressE s ��r, c C sR t jdkr| dkrtd| ��t jdkr#| dkr%tt| �s'td| ��dS dS dS )zD
Checks if the family is valid for the current environment.
r r z Family %s is not recognized.r N)�sys�platformr) �hasattr�socketr* r r r �_validate_familyS s
�r1 c C sP t | �tkrdS t | �tu r| �d�rdS t | �tu s t�| �r"dS td| ��)z]
Return the types of the address
This can be 'AF_INET', 'AF_UNIX', or 'AF_PIPE'
r z\\r r z address type of %r unrecognized)�type�tuple�str�
startswithr � is_abstract_socket_namespacer) )�addressr r r � address_type_ s r8 c @ s� e Zd ZdZd+dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
dd� �Z e
dd� �Z e
dd� �Z
dd� Zdd� Zd,dd�Zd d � Zd-d d �Zd.d d!�Zd"d#� Zd/d%d&�Zd'd(� Zd)d*� ZdS )0�_ConnectionBaseNTc C s> |� � }|dk r td��|s|std��|| _|| _|| _d S )Nr zinvalid handlez6at least one of `readable` and `writable` must be True)� __index__r) �_handle� _readable� _writable)�self�handle�readable�writabler r r �__init__u s �
z_ConnectionBase.__init__c C s | j d ur | �� d S d S r �r; �_close�r> r r r �__del__� s
�z_ConnectionBase.__del__c C s | j d u r td��d S )Nzhandle is closed)r; �OSErrorrE r r r �
_check_closed� s
�z _ConnectionBase._check_closedc C � | j std��d S )Nzconnection is write-only)r< rG rE r r r �_check_readable� � �z _ConnectionBase._check_readablec C rI )Nzconnection is read-only)r= rG rE r r r �_check_writable� rK z _ConnectionBase._check_writablec C s$ | j r
d| _td��| �� td��)NFzbad message length)r= r< �closerG rE r r r �_bad_message_length� s
�z#_ConnectionBase._bad_message_lengthc C s
| j du S )z True if the connection is closedN�r; rE r r r �closed� s
z_ConnectionBase.closedc C � | j S )z"True if the connection is readable)r< rE r r r r@ � � z_ConnectionBase.readablec C rQ )z"True if the connection is writable)r= rE r r r rA � rR z_ConnectionBase.writablec C s | � � | jS )z+File descriptor or handle of the connection)rH r; rE r r r �fileno� s z_ConnectionBase.filenoc C s, | j durz
| �� W d| _ dS d| _ w dS )zClose the connectionNrC rE r r r rM � s
�z_ConnectionBase.closer c C s� | � � | �� t|�}|jdkrtt|��}t|�}|dk r#td��||k r+td��|du r4|| }n|dk r<td��|| |krFtd��| �|||| � � dS )z,Send the bytes data from a bytes-like objectr r zoffset is negativezbuffer length < offsetNzsize is negativez buffer length < offset + size)rH rL �
memoryview�itemsize�bytes�lenr) � _send_bytes)r> �buf�offset�size�m�nr r r �
send_bytes� s"
z_ConnectionBase.send_bytesc C s$ | � � | �� | �t�|�� dS )zSend a (picklable) objectN)rH rL rX �_ForkingPickler�dumps�r> �objr r r �send� s z_ConnectionBase.sendc C sJ | � � | �� |dur|dk rtd��| �|�}|du r!| �� |�� S )z7
Receive bytes data as a bytes object.
Nr znegative maxlength)rH rJ r) � _recv_bytesrN �getvalue)r> � maxlengthrY r r r �
recv_bytes� s
z_ConnectionBase.recv_bytesc C s� | � � | �� t|��K}|j}|t|� }|dk r td��||kr&td��| �� }|�� }||| k r:t|� � ��|�
d� |� ||| || | � � |W d � S 1 sZw Y dS )zq
Receive bytes data into a writeable bytes-like object.
Return the number of bytes read.
r znegative offsetzoffset too largeN) rH rJ rT rU rW r) rd �tellr re �seek�readinto)r> rY rZ r\ rU �bytesize�resultr[ r r r �recv_bytes_into� s&
�$�z _ConnectionBase.recv_bytes_intoc C s& | � � | �� | �� }t�|�� �S )z Receive a (picklable) object)rH rJ rd r_ �loads� getbuffer)r> rY r r r �recv� s z_ConnectionBase.recv� c C s | � � | �� | �|�S )z/Whether there is any input available to be read)rH rJ �_poll�r> r r r r �poll� s
z_ConnectionBase.pollc C � | S r r rE r r r � __enter__ � z_ConnectionBase.__enter__c C � | � � d S r �rM �r> �exc_type� exc_value�exc_tbr r r �__exit__ r z_ConnectionBase.__exit__)TT)r Nr )r )rq )�__name__�
__module__� __qualname__r; rB rF rH rJ rL rN �propertyrP r@ rA rS rM r^ rc rg rm rp rt rv r~ r r r r r9 r s0
r9 c @ sD e Zd ZdZdZejfdd�Zdd� Zddd �Z d
d � Z
d d
� Z dS )�PipeConnectionz�
Connection class based on a Windows named pipe.
Overlapped I/O is used, so the handles must have been created
with FILE_FLAG_OVERLAPPED.
Fc C � || j � d S r rO )r> � _CloseHandler r r rD � zPipeConnection._closec C s� t j| j|dd�\}}z(z|t jkr!t �|jgdt�}|tks!J �W n |�� � W |� d�\}}n|� d�\}}w |dksBJ �|t
|�ksJJ �d S )NT��
overlappedFr ) �_winapi� WriteFiler; �ERROR_IO_PENDING�WaitForMultipleObjects�eventr r �cancel�GetOverlappedResultrW )r> rY �ov�err�waitres�nwrittenr r r rX s
� �� zPipeConnection._send_bytesNc
C sh | j r
d| _ t�� S |d u rdnt|d�}z�tj| j|dd�\}}zFz|tjkr7t�|j gdt
�}|t ks7J �W n |� � � W |�
d�\}}|dkr[t�� }|�|�� � |W S |tjkrg| �||�W S n2|�
d�\}}|dkr�t�� }|�|�� � | Y W S |tjkr�| �||� Y W S w W td�� ty� } z |jtjkr�t�� d }~ww )NF� Tr� r z.shouldn't get here; expected KeyboardInterrupt)�_got_empty_message�io�BytesIO�minr� �ReadFiler; r� r� r� r r r� r� �writero �ERROR_MORE_DATA�_get_more_datarG �winerror�ERROR_BROKEN_PIPE�EOFError� RuntimeError) r> �maxsize�bsizer� r� r� �nread�f�er r r rd & sT
�
� ��
��
�� ��zPipeConnection._recv_bytesc C s. | j s
t�| j�d dkrdS tt| g|��S )Nr T)r� r� �
PeekNamedPiper; �boolr rs r r r rr F s zPipeConnection._pollc C s� |� � }t�� }|�|� t�| j�d }|dksJ �|d ur+t|�| |kr+| �� tj | j|dd�\}}|�
d�\}}|dksCJ �||ksIJ �|�|� � � |S )Nr r Tr� ) ro r� r� r� r� r� r; rW rN r� r� )r> r� r� rY r� �leftr� �rbytesr r r r� L s
z PipeConnection._get_more_datar ) r r� r� �__doc__r� r� � CloseHandlerD rX rd rr r� r r r r r� s
r� c @ s| e Zd ZdZerejfdd�ZejZ ej
Z n
e j
fdd�Ze jZ e jZ e fdd�Ze fdd�Zd d
� Zdd d
�Zdd� Zd S )�
Connectionzo
Connection class based on an arbitrary file descriptor (Unix only), or
a socket handle (Windows).
c C r� r rO �r> rD r r r rD c r� zConnection._closec C r� r rO r� r r r rD h r� c C s8 t |�} || j|�}||8 }|dkrd S ||d � }q)NTr )rW r; )r> rY r� � remainingr] r r r �_sendm s �zConnection._sendc C sf t �� }| j}|}|dkr1|||�}t|�}|dkr$||kr t�td��|�|� ||8 }|dks
|S )Nr z got end of file during message)r� r� r; rW r� rG r� )r> r[ �readrY r? r� �chunkr] r r r �_recvv s
�
zConnection._recvc C s� t |�}|dkr%t�dd�}t�d|�}| �|� | �|� | �|� d S t�d|�}|dkr;| �|� | �|� d S | �|| � d S )Ni����!i����!Qi @ )rW �struct�packr� )r> rY r] �
pre_header�headerr r r rX � s
zConnection._send_bytesNc C s^ | � d�}t�d|�� �\}|dkr | � d�}t�d|�� �\}|d ur*||kr*d S | � |�S )N� r� r� � r� )r� r� �unpackre )r> r� rY r[ r r r rd � s
zConnection._recv_bytesc C s t | g|�}t|�S r )r r� )r> r �rr r r rr � s zConnection._pollr )r r� r� r� r� �_multiprocessing� closesocketrD rc �_writerp �_readr% rM r� r� r� r� rX rd rr r r r r r� \ s
r� c @ sR e Zd ZdZddd�Zdd� Zdd � Zed
d � �Zed d
� �Z dd� Z
dd� Z dS )r z�
Returns a listener object.
This is a wrapper for a bound socket which is 'listening' for
connections, or for a Windows named pipe.
Nr c C sp |p |rt |�p t}|pt|�}t|� |dkr t||�| _nt|||�| _|d ur3t|t�s3t d��|| _
d S )Nr � authkey should be a byte string) r8 �default_familyr, r1 � PipeListener� _listener�SocketListener�
isinstancerV � TypeError�_authkey)r> r7 r+ �backlog�authkeyr r r rB � s �
zListener.__init__c C s> | j du r td��| j �� }| jr t|| j� t|| j� |S )zz
Accept a connection on the bound socket or named pipe of `self`.
Returns a `Connection` object.
Nzlistener is closed)r� rG �acceptr� �deliver_challenge�answer_challenge)r> �cr r r r� � s
zListener.acceptc C s$ | j }|durd| _ |�� dS dS )zA
Close the bound socket or named pipe of `self`.
N)r� rM )r> �listenerr r r rM � s
�zListener.closec C � | j jS r )r� �_addressrE r r r r7 � � zListener.addressc C r� r )r� �_last_acceptedrE r r r �
last_accepted� r� zListener.last_acceptedc C ru r r rE r r r rv � rw zListener.__enter__c C rx r ry rz r r r r~ � r zListener.__exit__)NNr N) r r� r� r� rB r� rM r� r7 r� rv r~ r r r r r � s
r c C sh |pt | �}t|� |dkrt| �}nt| �}|dur$t|t�s$td��|dur2t||� t||� |S )z=
Returns a connection to the address of a `Listener`
r Nr� ) r8 r1 �
PipeClient� SocketClientr� rV r� r� r� )r7 r+ r� r� r r r r � s
r Tc C sp | r"t �� \}}|�d� |�d� t|�� �}t|�� �}||fS t�� \}}t|dd�}t|dd�}||fS )�L
Returns pair of connection objects at either end of a pipe
TF�rA �r@ )r0 �
socketpair� setblockingr� �detachr% �pipe)�duplex�s1�s2�c1�c2�fd1�fd2r r r r s
� r c
C s� t d�}| rtj}tjtjB }tt}}n tj}tj}dt}}t�||tjB tj B tj
tj B tj B d||tj
tj�}t�||dtjtjtjtj�}t�|tj dd� tj|dd�}|�d�\} }
|
dksgJ �t|| d�} t|| d�} | | fS ) r� r r r NTr� r� r� )r, r� �PIPE_ACCESS_DUPLEX� GENERIC_READ�
GENERIC_WRITE�BUFSIZE�PIPE_ACCESS_INBOUND�CreateNamedPipe�FILE_FLAG_OVERLAPPED� FILE_FLAG_FIRST_PIPE_INSTANCE�PIPE_TYPE_MESSAGE�PIPE_READMODE_MESSAGE� PIPE_WAIT�NMPWAIT_WAIT_FOREVER�NULL�
CreateFile�
OPEN_EXISTING�SetNamedPipeHandleState�ConnectNamedPiper� r� )
r� r7 �openmode�access�obsize�ibsize�h1�h2r� �_r� r� r� r r r r s>
�
�
� �
� c @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) r� zO
Representation of a socket which is bound to an address and listening
r c C s� t � tt |��| _z)tjdkr| j�t jt jd� | j�d� | j� |� | j�
|� | j� � | _ W n
t
y? | j�� � w || _d | _|dkr]t�|�s]tj| tj|fdd�| _d S d | _d S )N�posixr Tr r ��args� exitpriority)r0 �getattr�_socketr% �name�
setsockopt�
SOL_SOCKET� SO_REUSEADDRr� �bind�listen� getsocknamer� rG rM �_familyr� r r6 �Finalize�unlink�_unlink)r> r7 r+ r� r r r rB G s*
�
� �
zSocketListener.__init__c C s& | j �� \}| _|�d� t|�� �S )NT)r r� r� r� r� r� �r> �sr r r r� ` s
zSocketListener.acceptc C sN z| j �� W | j}|d urd | _|� d S d S | j}|d ur&d | _|� w w r )r rM r )r> r
r r r rM e s
���zSocketListener.closeN)r )r r� r� r� rB r� rM r r r r r� C s
r� c C s\ t | �}t�tt|���}|�d� |�| � t|�� �W d � S 1 s'w Y dS )zO
Return a connection object connected to the socket given by `address`
TN)r8 r0 r r� �connectr� r� )r7 r+ r r r r r� o s
$�r� c @ s8 e Zd ZdZd dd�Zd
dd�Zdd � Zed
d � �ZdS )r� z0
Representation of a named pipe
Nc C sL || _ | jdd�g| _d | _t�d| j � tj| tj| j| j fdd�| _ d S )NT)�firstz listener created with address=%rr r� )
r� � _new_handle�
_handle_queuer� r � sub_debugr r� �_finalize_pipe_listenerrM )r> r7 r� r r r rB � s �zPipeListener.__init__Fc
C sH t jt jB }|r
|t jO }t �| j|t jt jB t jB t j t
t
t j t j �S r )
r� r� r� r� r� r� r� r� r� �PIPE_UNLIMITED_INSTANCESr� r� r� )r> r �flagsr r r r � s
��zPipeListener._new_handlec
C s� | j �| �� � | j �d�}z tj|dd�}W n ty3 } z|jtjkr&� W Y d }~t|�S d }~ww z+z t� |j
gdt �}W n
|� � t�
|� � W |�d�\}}|dks\J �t|�S |�d�\}}|dksmJ �w )Nr Tr� F)r �appendr �popr� r� rG r� �
ERROR_NO_DATAr� r� r r� r� r� r� )r> r? r� r� �resr� r� r r r r� � s2
���
�
� �zPipeListener.acceptc C s$ t �d|� | D ]}t�|� qd S )Nz closing listener with address=%r)r r r� r� )�queuer7 r? r r r r � s �z$PipeListener._finalize_pipe_listenerr )F) r r� r� r� rB r r� � staticmethodr r r r r r� s
r� c
C s� t � } zt�| d� t�| tjtjB dtjtjtjtj�}W n t y? } z|j
tj tj fvs4t
|�r5� W Y d}~nd}~ww nqt�|tjdd� t|�S )zU
Return a connection object connected to the pipe given by `address`
r �� r N)r r� �
WaitNamedPiper� r� r� r� r� r� rG r� �ERROR_SEM_TIMEOUT�ERROR_PIPE_BUSYr r� r� r� )r7 r �hr� r r r r� � s2
��� ����
�r� � s #CHALLENGE#s #WELCOME#s #FAILURE#c C s� dd l }t|t�std�t|����t�t�}| � t
| � |� ||d�� � }| �
d�}||kr7| � t� d S | � t� td��)Nr � Authkey must be bytes, not {0!s}�md5� zdigest received was wrong)�hmacr� rV r) �formatr2 r% �urandom�MESSAGE_LENGTHr^ � CHALLENGE�new�digestrg �WELCOME�FAILUREr ��
connectionr� r( �messager. �responser r r r� � s
�
r� c C s� dd l }t|t�std�t|����| �d�}|d tt�� tks'J d| ��|tt�d � }|� ||d��
� }| � |� | �d�}|t krJt
d��d S )Nr r% r' z message = %rr& zdigest sent was rejected)r( r� rV r) r) r2 rg rW r, r- r. r^ r/ r r1 r r r r� � s
�
�r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ConnectionWrapperc C s6 || _ || _|| _dD ]
}t||�}t| ||� q d S )N)rS rM rt rg r^ )�_conn�_dumps�_loadsr �setattr)r> �connr` rn �attrrb r r r rB � s
�zConnectionWrapper.__init__c C s | � |�}| j�|� d S r )r7 r6 r^ )r> rb r r r r rc s
zConnectionWrapper.sendc C s | j �� }| �|�S r )r6 rg r8 r r r r rp s
zConnectionWrapper.recvN)r r� r� rB rc rp r r r r r5 � s r5 c C s t �| fd d d d��d�S )Nr �utf-8)� xmlrpclibr` �encode)rb r r r �
_xml_dumps s r? c C s t �| �d��\\}}|S )Nr< )r= rn �decode)r rb �methodr r r �
_xml_loads s rB c @ s e Zd Zdd� ZdS )� XmlListenerc C s" dd l ma t�| �}t|tt�S �Nr )�
xmlrpc.client�clientr= r r� r5 r? rB ra r r r r� s
zXmlListener.acceptN)r r� r� r� r r r r rC s rC c O s" dd l ma tt| i |��tt�S rD )rE rF r= r5 r r? rB )r �kwdsr r r � XmlClient s rH c C s� t | �}g }|rZt�|d|�}|tkr |S t| kr$tt|� k r+n n|t8 }n t| kr9tt|� k rCn td��|t8 }ntd��|�|| � ||d d � }d}|s|S )NFzShould not get herer r ) �listr� r� r
r rW r r� r )�handlesr �L�readyr r r r �_exhaustive_wait% s$
�
��
rM c
s� |du rt }n|dk rd}nt|d d �}t| �} i �g }t� � t� }z�| D ]�}zt|d�}W n ty? |�|�� < Y q'w z t�|� dd�\}}W n t yi } zd|j
}}|t vr_� W Y d}~nd}~ww |tj krz|�
|� |�|j< q'|r�t�� dd� dkr�z |�d �\} }W n t y� } z |j
}W Y d}~nd}~ww |s�t|d
�r�d|_� �|� d}q't��� |�}W |D ]}|�� q�|D ]D}z |�d�\} }W n t y� } z|j
}|t vr� W Y d}~nd}~ww |tjk�r�|j }� �|� |dk�rt|d
��rd|_q�nU|D ]}|�� �q|D ]G}z |�d�\} }W n t �yC } z|j
}|t v�r9� W Y d}~nd}~ww |tjk�rb�|j }� �|� |dk�rbt|d
��rbd|_�q w � ��fd d �|D �� � fd
d�| D �S )��
Wait till an object in object_list is ready/readable.
Returns list of those objects in object_list which are ready/readable.
Nr r g �?rS T� )� rO Fr� c 3 s � | ]}� | V qd S r r )�.0r# )�waithandle_to_objr r � <genexpr>� s � zwait.<locals>.<genexpr>c s g | ]}|� v r|�qS r r )rQ �o)�
ready_objectsr r �
<listcomp>� s �wait.<locals>.<listcomp>)r �intrI �setr �AttributeErrorr: r� r� rG r� �
_ready_errorsr� r r� r- �getwindowsversionr� r/ r� �addrM �keysr� �ERROR_OPERATION_ABORTED�update)
� object_listr �ov_list�
ready_handlesrT rS r� r� r� r� r )rU rR r r ; s� � ���
��
���
���
���
��r � PollSelectorc C s� t � �E}| D ] }|�|tj� q|durt�� | } |�|�}|r0dd� |D �W d � S |durG|t�� }|dk rG|W d � S q1 sKw Y dS )rN NTc S s g | ]\}}|j �qS r )�fileobj)rQ �key�eventsr r r rV � s rW r )�
_WaitSelector�register� selectors�
EVENT_READr r �select)ra r �selectorrb �deadlinerL r r r r � s"
� ��c C sf | � � }t�|tjtj�� }ddlm} |�|�}t|| j | j
ffW d � S 1 s,w Y d S )Nr )�resource_sharer) rS r0 �fromfdr � SOCK_STREAMr! ro � DupSocket�rebuild_connectionr@ rA )r: r? r ro �dsr r r �reduce_connection� s
$�ru c C s | � � }t|� � ||�S r �r� r� )rt r@ rA �sockr r r rs � s rs c C sB | j rtjnd| jr
tjndB }t�| �� |�}t|| j | jffS rD ) r@ r� �FILE_GENERIC_READrA �FILE_GENERIC_WRITEr
� DupHandlerS �rebuild_pipe_connection)r: r� �dhr r r �reduce_pipe_connection� s
�r} c C � | � � }t|||�S r )r� r� )r| r@ rA r? r r r r{ � � r{ c C s t �| �� �}t|| j| jffS r )r
�DupFdrS rs r@ rA )r: �dfr r r ru � s c C r~ r rv )r� r@ rA �fdr r r rs � r )NN)Tr )I�__all__r� r% r- r0 r� r r" � itertoolsr� r! r r r �contextr
�ForkingPicklerr_ r� r r r
r � ImportErrorr. r� �CONNECTION_TIMEOUT�countr( r� �familiesr/ r r r, r1 r8 r9 r� r� �objectr r r r� r� r� r� r+ r, r/ r0 r� r� r5 r? rB rC rH rM r� �ERROR_NETNAME_DELETEDr[ r rj rd rh �SelectSelectorru rs ri r} r{ r r r r �<module> s�
�
PT
=
,,
8 P