o
    f{                     @   s  g d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
mZ ddl
mZmZ ddlmZ ejZzddlZddlmZmZmZmZ W n eyb   ejdkr^ dZY nw d	Zd
Ze ZdZdgZeedr|dZedg7 ZejdkrdZedg7 ZefddZ dd Z!dd Z"dd Z#dd Z$G dd dZ%erG dd de%Z&G dd de%Z'G dd de(Z)dPd d!Z*ejdkrdQd#d$Z+ndQd%d$Z+G d&d' d'e(Z,d(d) Z-ejdkrG 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:ejdkr3d@dA Z;ej<ej=hZ>dRdBdCZ?nddl@Z@ee@dDrAe@jAZBne@jCZBdRdEdCZ?ejdkrmdFdG ZDdHdI ZEeFe'eD dJdK ZGdLdM ZHeFe&eG dS dNdG ZDdOdI ZEeFe'eD dS )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  | kS r   r   )tr   r   r   _check_timeout>   r   r   c                 C   sT   | dkrdS | dkrt jdt dS | dkr&t jdt ttf ddS td	)
z?
    Return an arbitrary free address for the given family
    r   )	localhostr   r   z	listener-)prefixdirr   z\\.\pipe\pyc-%d-%d- zunrecognized family)	tempfilemktempr   get_temp_dirosgetpidnext_mmap_counter
ValueErrorfamilyr   r   r   arbitrary_addressE   s   r,   c                 C   sR   t jdkr| dkrtd|  t jdkr#| dkr%tt| s'td|  dS dS dS )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 | tkrdS t | tu r| drdS t | tu s t| r"dS td|  )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 Zd ZdZd+ddZdd Zdd Zd	d
 Zdd Zdd Z	e
dd Ze
dd Ze
dd Zdd Zdd Zd,ddZdd Zd-ddZd.d d!Zd"d# Zd/d%d&Zd'd( Zd)d* ZdS )0_ConnectionBaseNTc                 C   s>   |  }|dk rtd|s|std|| _|| _|| _d S )Nr   zinvalid 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 ur|   d S d S r   r;   _closer>   r   r   r   __del__   s   
z_ConnectionBase.__del__c                 C   s   | j d u r	tdd S )Nzhandle is closed)r;   OSErrorrE   r   r   r   _check_closed   s   
z_ConnectionBase._check_closedc                 C      | j stdd S )Nzconnection is write-only)r<   rG   rE   r   r   r   _check_readable      z_ConnectionBase._check_readablec                 C   rI   )Nzconnection is read-only)r=   rG   rE   r   r   r   _check_writable   rK   z_ConnectionBase._check_writablec                 C   s$   | j r
d| _td|   td)NFzbad message length)r=   r<   closerG   rE   r   r   r   _bad_message_length   s
   z#_ConnectionBase._bad_message_lengthc                 C   s
   | j du 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   |    | jS )z+File descriptor or handle of the connection)rH   r;   rE   r   r   r   fileno   s   z_ConnectionBase.filenoc                 C   s,   | j durz
|   W d| _ dS d| _ w dS )zClose the connectionNrC   rE   r   r   r   rM      s
   

z_ConnectionBase.closer   c                 C   s   |    |   t|}|jdkrtt|}t|}|dk r#td||k r+td|du r4|| }n|dk r<td|| |krFtd| ||||   dS )z,Send the bytes data from a bytes-like objectr   r   zoffset is negativezbuffer length < offsetNzsize 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| dS )zSend a (picklable) objectN)rH   rL   rX   _ForkingPicklerdumpsr>   objr   r   r   send   s   z_ConnectionBase.sendc                 C   sJ   |    |   |dur|dk rtd| |}|du r!|   | S )z7
        Receive bytes data as a bytes object.
        Nr   znegative 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| }|dk rtd||kr&td|  }| }||| k r:t|	 |
d |||| || |   |W  d   S 1 sZw   Y  dS )zq
        Receive bytes data into a writeable bytes-like object.
        Return the number of bytes read.
        r   znegative offsetzoffset 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 Zd ZdZdZejfddZdd Zddd	Z	d
d Z
dd ZdS )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        zPipeConnection._closec              	   C   s   t j| j|dd\}}z(z|t jkr!t |jgdt}|tks!J W n   |   W |	d\}}n|	d\}}w |dksBJ |t
|ksJJ d S )NT
overlappedFr   )_winapi	WriteFiler;   ERROR_IO_PENDINGWaitForMultipleObjectseventr   r   cancelGetOverlappedResultrW   )r>   rY   overrwaitresnwrittenr   r   r   rX     s   

 zPipeConnection._send_bytesNc           	   
   C   sh  | j r
d| _ t S |d u rdnt|d}ztj| j|dd\}}zFz|tjkr7t|j	gdt
}|tks7J W n   |   W |d\}}|dkr[t }||  |W S |tjkrg| ||W S n2|d\}}|dkrt }||  |     Y W S |tjkr| ||     Y W S w W td ty } z	|jtjkrt d }~ww )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   





zPipeConnection._recv_bytesc                 C   s.   | j st| jd dkrdS tt| g|S )Nr   T)r   r   PeekNamedPiper;   boolr   rs   r   r   r   rr   F  s   zPipeConnection._pollc                 C   s   |  }t }|| t| jd }|dksJ |d ur+t|| |kr+|   tj	| j|dd\}}|
d\}}|dksCJ ||ksIJ ||   |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 Zd ZdZerejfddZejZ	ej
ZnejfddZejZ	ejZe	fddZefddZd	d
 ZdddZdd Zd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   zConnection._closec                 C   r   r   rO   r   r   r   r   rD   h  r   c                 C   s8   t |}	 || j|}||8 }|dkrd S ||d  }q)NTr   )rW   r;   )r>   rY   r   	remainingr]   r   r   r   _sendm  s   zConnection._sendc                 C   sf   t  }| j}|}|dkr1|||}t|}|dkr$||kr ttd|| ||8 }|dks|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   


zConnection._recvc                 C   s   t |}|dkr%tdd}td|}| | | | | | d S td|}|dkr;| | | | d S | ||  d S )Ni!i!Qi @  )rW   structpackr   )r>   rY   r]   
pre_headerheaderr   r   r   rX     s   


zConnection._send_bytesNc                 C   s^   |  d}td| \}|dkr |  d}td| \}|d ur*||kr*d S |  |S )N   r   r      r   )r   r   unpackre   )r>   r   rY   r[   r   r   r   rd     s   


zConnection._recv_bytesc                 C   s   t | g|}t|S r   )r   r   )r>   r   rr   r   r   rr     s   zConnection._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 Zd ZdZdddZdd Zdd	 Zed
d Zedd Z	dd Z
dd ZdS )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	|rt |p	t}|pt|}t| |dkrt||| _nt|||| _|d ur3t|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   
zListener.__init__c                 C   s>   | j du r	td| j  }| jrt|| j t|| j |S )zz
        Accept a connection on the bound socket or named pipe of `self`.

        Returns a `Connection` object.
        Nzlistener is closed)r   rG   acceptr   deliver_challengeanswer_challenge)r>   cr   r   r   r     s   

zListener.acceptc                 C   s$   | j }|durd| _ |  dS dS )zA
        Close the bound socket or named pipe of `self`.
        N)r   rM   )r>   listenerr   r   r   rM     s
   zListener.closec                 C      | j jS r   )r   _addressrE   r   r   r   r7        zListener.addressc                 C   r   r   )r   _last_acceptedrE   r   r   r   last_accepted  r   zListener.last_acceptedc                 C   ru   r   r   rE   r   r   r   rv     rw   zListener.__enter__c                 C   rx   r   ry   rz   r   r   r   r~     r   zListener.__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   |pt | }t| |dkrt| }nt| }|dur$t|ts$td|dur2t|| 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| }||fS t \}}t|dd}t|dd}||fS )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}| rtj}tjtjB }tt}}ntj}tj}dt}}t||tjB tj	B tj
tjB tjB d||tjtj}t||dtjtjtjtj}t|tjdd tj|dd}|d\}	}
|
dksgJ t|| d}t|| d}||fS )	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 Zd ZdZd
ddZdd Zdd Zd	S )r   zO
    Representation of a socket which is bound to an address and listening
    r   c                 C   s   t  tt || _z)tjdkr| jt jt jd | jd | j	| | j
| | j | _W n ty?   | j   w || _d | _|dkr]t|s]tj| tj|fdd| _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*   



zSocketListener.__init__c                 C   s&   | j  \}| _|d t| S )NT)r  r   r   r   r   r   r>   sr   r   r   r   `  s   
zSocketListener.acceptc                 C   sN   z| j   W | j}|d urd | _|  d S d S | j}|d ur&d | _|  w w r   )r  rM   r  )r>   r  r   r   r   rM   e  s   
zSocketListener.closeN)r   )r   r   r   r   rB   r   rM   r   r   r   r   r   C  s
    
r   c                 C   s\   t | }ttt|}|d ||  t| W  d   S 1 s'w   Y  dS )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 Zd ZdZdddZdddZdd	 Zed
d ZdS )r   z0
        Representation of a named pipe
        Nc                 C   sL   || _ | jddg| _d | _td| j  tj| tj| j| j fdd| _	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   zPipeListener.__init__Fc              
   C   sH   t jt jB }|r|t jO }t | j|t jt jB t jB 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   

zPipeListener._new_handlec              
   C   s   | j |   | j d}z	tj|dd}W n ty3 } z|jtjkr& W Y d }~t|S d }~ww z+zt	|j
gdt}W n   |  t|  W |d\}}|dks\J t|S |d\}}|dksmJ 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   


zPipeListener.acceptc                 C   s$   t d| | D ]}t| qd 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  }	 zt| d t| tjtjB dtjtjtjtj}W n  t	y? } z|j
tjtjfvs4t|r5 W Y d}~nd}~ww nqt|tjdd 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   dd l }t|tstdt|tt}| 	t
|  |||d }| d}||kr7| 	t d S | 	t td)Nr    Authkey must be bytes, not {0!s}md5   zdigest 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   dd l }t|tstdt|| d}|d tt tks'J d| |ttd  }|	||d
 }| | | d}|tkrJtdd S )Nr   r%  r'  zmessage = %rr&  zdigest 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 Zd Zdd Zdd Zdd ZdS )ConnectionWrapperc                 C   s6   || _ || _|| _dD ]}t||}t| || qd S )N)rS   rM   rt   rg   r^   )_conn_dumps_loadsr  setattr)r>   connr`   rn   attrrb   r   r   r   rB     s   
zConnectionWrapper.__init__c                 C   s   |  |}| j| d S r   )r7  r6  r^   )r>   rb   r  r   r   r   rc     s   
zConnectionWrapper.sendc                 C   s   | j  }| |S r   )r6  rg   r8  r  r   r   r   rp     s   

zConnectionWrapper.recvN)r   r   r   rB   rc   rp   r   r   r   r   r5    s    r5  c                 C   s   t | fd 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 Zd Zdd ZdS )XmlListenerc                 C   s"   dd l ma t| }t|ttS Nr   )xmlrpc.clientclientr=  r   r   r5  r?  rB  ra   r   r   r   r     s   
zXmlListener.acceptN)r   r   r   r   r   r   r   r   rC    s    rC  c                  O   s"   dd l ma tt| i |ttS rD  )rE  rF  r=  r5  r   r?  rB  )r   kwdsr   r   r   	XmlClient  s   rH  c                 C   s   t | }g }|rZt|d|}|tkr	 |S t|  kr$tt| k r+n n|t8 }nt|  kr9tt| k rCn td|t8 }ntd|||  ||d d  }d}|s|S )NFzShould not get herer   r   )	listr   r   r   r   rW   r   r   r  )handlesr   Lreadyr  r   r   r   _exhaustive_wait%  s$   
 

rM  c           
         s  |du rt }n|dk rd}nt|d d }t| } i g }t  t }z| D ]}zt|d}W n ty?   || < Y q'w zt| dd\}}W n t	yi } zd|j
}}|tvr_ W Y d}~nd}~ww |tjkrz|| ||j< q'|rt dd dkrz	|d	\}	}W n t	y } z	|j
}W Y d}~nd}~ww |st|d
rd|_ | d}q't |}W |D ]}|  q|D ]D}z	|d\}	}W n t	y } z|j
}|tvr W Y d}~nd}~ww |tjkr|j } | |dkrt|d
rd|_qnU|D ]}|  q|D ]G}z	|d\}	}W n t	yC } z|j
}|tvr9 W Y d}~nd}~ww |tjkrb|j } | |dkrbt|d
rbd|_qw  fdd|D   fd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  qd S r   r   ).0r#  )waithandle_to_objr   r   	<genexpr>  s    zwait.<locals>.<genexpr>c                    s   g | ]}| v r|qS 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 ]	}||tj q|durt | }	 ||}|r0dd |D W  d   S |durG|t  }|dk rG|W  d   S q1 sKw   Y  dS )rN  NTc                 S   s   g | ]\}}|j qS 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|tjtj}ddlm} ||}t|| j	| j
ffW  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 rtjnd| jrtjndB }t|  |}t|| j | jffS 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| jffS 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   	
	



 PT
=

,,
8	P
