o
0i{ @ 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 ejdkrd
Z e d
g7 Z efddZ dd Z!dd Z"dd Z#dd Z$G dd dZ%erG dd de%Z&G d d d e%Z'G d d d e(Z)dPd d!Z*ejdkrdQd#d$Z+ndQd%d$Z+G d&d' d'e(Z,d(d) Z-ejdkrG d*d+ d+e(Z.d,d- Z/d.Z0d/Z1d0Z2d1Z3d2d3 Z4d4d5 Z5G d6d7 d7e(Z6d8d9 Z7d:d; Z8G dd? Z:ejdkr3d@dA Z;ejdRdBdCZ?nddl@Z@e e@dDrAe@jAZBne@jCZBdRdEdCZ?ejdkrmdFdG ZDdHdI ZEeFe'eD dJdK ZGdLdM ZHeFe&eG dS dNdG ZDdOdI ZEeFe'eD dS )S)ClientListenerPipewait N )util)AuthenticationErrorBufferTooShort) reduction)
WAIT_OBJECT_0WAIT_ABANDONED_0 WAIT_TIMEOUTINFINITEwin32i g 4@AF_INETAF_UNIXAF_PIPEc C s t | S Ntime monotonic)timeout r A/opt/alt/python310/lib64/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 dS | dkr&t jdt ttf ddS td )
z?
Return an arbitrary free address for the given family
r )Z localhostr r z listener-)prefixdirr z\\.\pipe\pyc-%d-%d- zunrecognized family) tempfileZmktempr Z get_temp_dirosgetpidnext
_mmap_counter
ValueErrorfamilyr 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)sysplatformr& hasattrsocketr' r r r _validate_familyS s
r. c C sP t | tkrdS t | tu r| drdS 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)typetuplestr
startswithr is_abstract_socket_namespacer& )addressr r r address_type_ s r5 c @ s e Zd ZdZd+ddZdd Zdd Zd d
Zd d Zd
d Z e
dd Z e
dd Z e
dd Z
dd Zdd Zd,ddZd 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)selfhandlereadablewritabler r r __init__u s
z_ConnectionBase.__init__c C s | j d ur | d S d S r r8 _closer; r r r __del__ s
z_ConnectionBase.__del__c C s | j d u r tdd S )Nzhandle is closed)r8 OSErrorrB r r r
_check_closed s
z _ConnectionBase._check_closedc C | j stdd S )Nzconnection is write-only)r9 rD rB r r r _check_readable z _ConnectionBase._check_readablec C rF )Nzconnection is read-only)r: rD rB r r r _check_writable rH z _ConnectionBase._check_writablec C s$ | j r
d| _td| td)NFzbad message length)r: r9 closerD rB r r r _bad_message_length s
z#_ConnectionBase._bad_message_lengthc C s
| j du S )z True if the connection is closedNr8 rB r r r closed s
z_ConnectionBase.closedc C | j S )z"True if the connection is readable)r9 rB r r r r= z_ConnectionBase.readablec C rN )z"True if the connection is writable)r: rB r r r r> rO z_ConnectionBase.writablec C s | | jS )z+File descriptor or handle of the connection)rE r8 rB r r r fileno s z_ConnectionBase.filenoc C s, | j durz
| W d| _ dS d| _ w dS )zClose the connectionNr@ rB r r r rJ s
z_ConnectionBase.closer c C s | | t|}|jdkrtt|}t|}|dk r#td||k r+td|du r4|| }n|dk r rP rJ r[ r` rc rh rk ro rq rx r r r r r6 r s0
r6 c @ sD e Zd ZdZdZejfddZdd 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 rL )r; Z _CloseHandler r r rA 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 ) _winapiZ WriteFiler8 ERROR_IO_PENDINGWaitForMultipleObjectseventr r cancelGetOverlappedResultrT )r; rV overrwaitresZnwrittenr r r rU s
zPipeConnection._send_bytesNc
C sh | j r
d| _ t S |d u rdnt|d}ztj| j|dd\}}zFz|tjkr7t|j gdt
}|t ks7J W n | W |
d\}}|dkr[t }|| |W S |tjkrg| ||W S n2|
d\}}|dkrt }|| | Y W S |tjkr| || Y W S w W td ty } z |jtjkrt d }~ww )NF Tr r z.shouldn't get here; expected KeyboardInterrupt)_got_empty_messageioBytesIOminr ReadFiler8 r r r r r r r writerj ZERROR_MORE_DATA_get_more_datarD winerrorERROR_BROKEN_PIPEEOFError RuntimeError) r; maxsizeZbsizer r r Znreadfer r r ra & sT
zPipeConnection._recv_bytesc C s. | j s
t| jd dkrdS tt| g|S )Nr T)r r
PeekNamedPiper8 boolr rn r r r rm F s zPipeConnection._pollc C s | }t }|| t| jd }|dksJ |d ur+t|| |kr+| tj | j|dd\}}|
d\}}|dksCJ ||ksIJ || |S )Nr r Tr ) rj r r r r r r8 rT rK r r )r; r r rV r leftr Zrbytesr r r r L s
z PipeConnection._get_more_datar ) ry rz r{ __doc__r r CloseHandlerA rU ra rm r r r r r r} s
r} c @ s| e Zd ZdZerejfddZejZ ej
Z n
e j
fddZe jZ e jZ e fddZe fddZd 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 rL r; rA r r r rA c r zConnection._closec C r~ r rL r r r r rA h r c C s8 t |} || j|}||8 }|dkrd S ||d }q)NTr )rT r8 )r; rV r remainingrZ r r r _sendm s zConnection._sendc C sf t }| j}|}|dkr1|||}t|}|dkr$||kr ttd|| ||8 }|dks
|S )Nr z got end of file during message)r r r8 rT r rD r )r; rX readrV r< r chunkrZ r r r _recvv s
zConnection._recvc C s t |}|dkr%tdd}td|}| | | | | | d S td|}|dkr;| | | | d S | || d S )Ni!i!Qi @ )rT structZpackr )r; rV rZ Z
pre_headerheaderr r r rU s
zConnection._send_bytesNc C s^ | d}td| \}|dkr | d}td| \}|d ur*||kr*d S | |S )N r r r )r r Zunpackrb )r; r rV rX r r r ra s
zConnection._recv_bytesc C s t | g|}t|S r )r r )r; r rr r r rm s zConnection._pollr )ry rz r{ r r _multiprocessingZ closesocketrA r` Z_writerk Z_readr" rJ r r r r rU ra rm r r r r r \ s
r c @ sR e Zd ZdZdddZdd 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|ts3t d|| _
d S )Nr authkey should be a byte string) r5 default_familyr) r. PipeListener _listenerSocketListener
isinstancerS TypeError_authkey)r; r4 r( backlogauthkeyr r r r? 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 rD acceptr deliver_challengeanswer_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 rJ )r; Zlistenerr r r rJ s
zListener.closec C | j jS r )r _addressrB r r r r4 zListener.addressc C r r )r _last_acceptedrB r r r
last_accepted r zListener.last_acceptedc C rp r r rB r r r rq rr zListener.__enter__c C rs r rt ru r r r rx r zListener.__exit__)NNr N) ry rz r{ r r? r rJ r| r4 r rq rx r r r r r s
r c C sh |pt | }t| |dkrt| }nt| }|dur$t|ts$td|dur2t|| t|| |S )z=
Returns a connection to the address of a `Listener`
r Nr ) r5 r.
PipeClient SocketClientr rS r r r )r4 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
TFr> r= )r- Z
socketpair setblockingr detachr" pipe)duplexs1s2c1c2Zfd1Zfd2r 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_WRITEBUFSIZEZPIPE_ACCESS_INBOUNDCreateNamedPipeFILE_FLAG_OVERLAPPED FILE_FLAG_FIRST_PIPE_INSTANCEPIPE_TYPE_MESSAGEPIPE_READMODE_MESSAGE PIPE_WAITNMPWAIT_WAIT_FOREVERNULL
CreateFile
OPEN_EXISTINGSetNamedPipeHandleStateConnectNamedPiper r} )
r r4 ZopenmodeaccessZobsizeZibsizeZh1Zh2r _r r r r r r r s>
c @ s* e Zd ZdZd
ddZdd 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| jt jt jd | jd | j | | j
| | j | _ W n
t
y? | j w || _d | _|dkr]t|s]tj| tj|fdd| _d S d | _d S )Nposixr Tr r argsZ exitpriority)r- getattr_socketr" nameZ
setsockoptZ
SOL_SOCKETZ SO_REUSEADDRr ZbindZlistenZ getsocknamer rD rJ Z_familyr r r3 Finalizeunlink_unlink)r; r4 r( r r r r r? 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 rJ r )r; r r r r rJ e s
zSocketListener.closeN)r )ry rz r{ r r? r rJ r r r r r C s
r c C s\ t | }ttt|}|d || t| W d S 1 s'w Y dS )zO
Return a connection object connected to the socket given by `address`
TN)r5 r- r r Zconnectr r )r4 r( r r r r r o s
$r c @ s8 e Zd ZdZd ddZd
ddZdd Zed
d ZdS )r z0
Representation of a named pipe
Nc C sL || _ | jddg| _d | _td| 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_listenerrJ )r; r4 r r r r r? 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 ZPIPE_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 rD r Z
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 )queuer4 r< r r r r s z$PipeListener._finalize_pipe_listenerr )F) ry rz r{ r r? 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 Z
WaitNamedPiper r r r r r rD r ZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYr r r r} )r4 r hr r r r r s2
r s #CHALLENGE#s #WELCOME#s #FAILURE#c C s dd l }t|tstdt|tt}| t
| | ||d }|
d}||kr7| t d S | t td)Nr Authkey must be bytes, not {0!s}md5 zdigest received was wrong)hmacr rS r& formatr/ r" urandomMESSAGE_LENGTHr[ CHALLENGEnewdigestrc WELCOMEFAILUREr Z
connectionr r messager Zresponser r r r s
r c C s dd l }t|tstdt|| d}|d tt tks'J d| |ttd }| ||d
}| | | d}|t krJt
dd S )Nr r r z message = %rr zdigest sent was rejected)r r rS r& r r/ rc rT r r r r[ r r r r r r r s
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ConnectionWrapperc C s6 || _ || _|| _dD ]
}t||}t| || q d S )N)rP rJ ro rc r[ )_conn_dumps_loadsr setattr)r; connr] ri attrr_ r r r r? s
zConnectionWrapper.__init__c C s | |}| j| d S r )r
r r[ )r; r_ r r r r r` s
zConnectionWrapper.sendc C s | j }| |S r )r rc r r r r r rk s
zConnectionWrapper.recvN)ry rz r{ r? r` rk r r r r r s r c C s t | fd d d ddS )Nr utf-8) xmlrpclibr] encode)r_ r r r
_xml_dumps s r c C s t | d\\}}|S )Nr )r ri decode)r r_ methodr r r
_xml_loads s r c @ s e Zd Zdd ZdS ) XmlListenerc C s" dd l ma t| }t|ttS Nr )
xmlrpc.clientclientr r r r r r r^ r r r r s
zXmlListener.acceptN)ry rz r{ r r r r r r s r c O s" dd l ma tt| i |ttS r )r r r r r r r )r kwdsr r r XmlClient s r 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 rT r r r )Zhandlesr Lreadyr r r r _exhaustive_wait% s$
r 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'|rt dd dkrz |d \} }W n t y } z |j
}W Y d}~nd}~ww |st|d
rd|_ | d}q't |}W |D ]}| q|D ]D}z |d\} }W n t y } z|j
}|t vr W Y d}~nd}~ww |tjkr|j } | |dkrt|d
rd|_qnU|D ]}| q|D ]G}z |d\} }W n t yC } z|j
}|t vr9 W Y d}~nd}~ww |tjkrb|j } | |dkrbt|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 ?rP T ) r! Fr c 3 s | ]} | V qd S r r ).0r )waithandle_to_objr r s zwait..c s g | ]}| v r|qS r r )r# o)
ready_objectsr r
s wait..)r intr setr AttributeErrorr7 r r rD r
_ready_errorsr r r r* Zgetwindowsversionr r, r addr keysr ZERROR_OPERATION_ABORTEDupdate)
object_listr Zov_listZ
ready_handlesr& rP r r r r r )r' r$ 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 )r NTc S s g | ]\}}|j qS r )Zfileobj)r# keyZeventsr r r r( s r) r )
_WaitSelectorregister selectorsZ
EVENT_READr r Zselect)r1 r Zselectorr_ Zdeadliner 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) rP r- Zfromfdr Z SOCK_STREAMr r7 Z DupSocketrebuild_connectionr= r> )r
r< r r7 dsr r r reduce_connection s
$r: c C s | }t| ||S r r r )r9 r= r> Zsockr r r r8 s r8 c C sB | j rtjnd| jr
tjndB }t| |}t|| j | jffS r ) r= r ZFILE_GENERIC_READr> ZFILE_GENERIC_WRITEr
Z DupHandlerP rebuild_pipe_connection)r
r dhr r r reduce_pipe_connection s
r> c C | }t|||S r )r r} )r= r= r> r< r r r r< r< c C s t | }t|| j| jffS r )r
ZDupFdrP r8 r= r> )r
dfr r r r: s c C r? r r; )rA r= r> fdr r r r8 r@ )NN)Tr )I__all__r r" r* r- r r r! itertoolsr r r r r contextr
ZForkingPicklerr\ r r r r
r ImportErrorr+ r ZCONNECTION_TIMEOUTcountr% r Zfamiliesr, r r r) r. r5 r6 r} r objectr r r r r r r r r r r r r r r r r r r r ZERROR_NETNAME_DELETEDr- r r6 r2 r4 ZSelectSelectorr: r8 r5 r> r< r r r r s
PT
=
,,
8 P