Mr.X


3


�]�iVy�@sddddgZddlZddlZddlZddlZddlZddlZddlZddlZddl    Z    ddl
mZddl
mZm
Z
dd    lmZejZy$ddlZdd
lmZmZmZmZWn$ek
r�ejdkrdZYnXdZd
ZdZej�ZdZdgZe ed��rdZedg7Zejdk�rdZedg7Zefdd�Z!dd�Z"dd�Z#dd�Z$dd�Z%Gdd�d�Z&e�rnGdd�de&�Z'Gd d!�d!e&�Z(Gd"d�de)�Z*dPd#d�Z+ejdk�r�dQd%d�Z,n
dRd&d�Z,Gd'd(�d(e)�Z-d)d*�Z.ejdk�r�Gd+d,�d,e)�Z/d-d.�Z0d/Z1d0Z2d1Z3d2Z4d3d4�Z5d5d6�Z6Gd7d8�d8e)�Z7d9d:�Z8d;d<�Z9Gd=d>�d>e*�Z:d?d@�Z;ejdk�rzdAdB�Z<ej=ej>hZ?dSdCd�Z@n,ddlAZAe eAdD��r�eAjBZCneAjDZCdTdEd�Z@ejdk�r�dFdG�ZEdHdI�ZFejGe(eE�dJdK�ZHdLdM�ZIejGe'eH�ndNdG�ZEdOdI�ZFejGe(eE�dS)U�Client�Listener�Pipe�wait�N�)�util)�AuthenticationError�BufferTooShort)�    reduction)�
WAIT_OBJECT_0�WAIT_ABANDONED_0�WAIT_TIMEOUT�INFINITE�win32i g4@Zsha256�AF_INET�AF_UNIX�AF_PIPEcCstj�|S)N)�time�    monotonic)�timeout�r�2/usr/lib64/python3.6/multiprocessing/connection.py�
_init_timeout?srcCstj�|kS)N)rr)�trrr�_check_timeoutBsrcCsX|dkrdS|dkr&tjdtj�d�S|dkrLtjdtj�tt�fd    d�Std
��dS)Nr�    localhostrrz    listener-)�prefix�dirrz\\.\pipe\pyc-%d-%d-�zunrecognized family)rr)    �tempfileZmktemprZget_temp_dir�os�getpid�next�
_mmap_counter�
ValueError)�familyrrr�arbitrary_addressIsr&cCsJtjdkr|dkrtd|��tjdkrF|dkrFtt|�sFtd|��dS)NrrzFamily %s is not recognized.r)�sys�platformr$�hasattr�socket)r%rrr�_validate_familyWs

r+cCsJt|�tkrdSt|�tkr*|jd�r*dSt|�tkr:dStd|��dS)Nrz\\rrzaddress type of %r unrecognized)�type�tuple�str�
startswithr$)�addressrrr�address_typecsr1c@s�eZdZdZd+dd�Zdd�Zdd�Zd    d
�Zdd�Zd
d�Z    e
dd��Ze
dd��Ze
dd��Z
dd�Zdd�Zd,dd�Zdd�Zd-dd�Zd.d d!�Zd"d#�Zd/d%d&�Zd'd(�Zd)d*�ZdS)0�_ConnectionBaseNTcCsB|j�}|dkrtd��|r,|r,td��||_||_||_dS)Nrzinvalid handlez6at least one of `readable` and `writable` must be True)�    __index__r$�_handle�    _readable�    _writable)�self�handle�readable�writablerrr�__init__ysz_ConnectionBase.__init__cCs|jdk    r|j�dS)N)r4�_close)r7rrr�__del__�s
z_ConnectionBase.__del__cCs|jdkrtd��dS)Nzhandle is closed)r4�OSError)r7rrr�
_check_closed�s
z_ConnectionBase._check_closedcCs|jstd��dS)Nzconnection is write-only)r5r>)r7rrr�_check_readable�sz_ConnectionBase._check_readablecCs|jstd��dS)Nzconnection is read-only)r6r>)r7rrr�_check_writable�sz_ConnectionBase._check_writablecCs"|jrd|_n|j�td��dS)NFzbad message length)r6r5�closer>)r7rrr�_bad_message_length�sz#_ConnectionBase._bad_message_lengthcCs
|jdkS)N)r4)r7rrr�closed�sz_ConnectionBase.closedcCs|jS)N)r5)r7rrrr9�sz_ConnectionBase.readablecCs|jS)N)r6)r7rrrr:�sz_ConnectionBase.writablecCs|j�|jS)N)r?r4)r7rrr�fileno�sz_ConnectionBase.filenocCs$|jdk    r z|j�Wdd|_XdS)N)r4r<)r7rrrrB�s
z_ConnectionBase.closercCs�|j�|j�t|�}|jdkr.tt|��}t|�}|dkrFtd��||krVtd��|dkrh||}n&|dkrztd��n|||kr�td��|j||||��dS)Nrrzoffset is negativezbuffer length < offsetzsize is negativezbuffer length < offset + size)r?rA�
memoryview�itemsize�bytes�lenr$�_send_bytes)r7�buf�offset�size�m�nrrr�
send_bytes�s"


z_ConnectionBase.send_bytescCs$|j�|j�|jtj|��dS)N)r?rArJ�_ForkingPickler�dumps)r7�objrrr�send�sz_ConnectionBase.sendcCsJ|j�|j�|dk    r(|dkr(td��|j|�}|dkrB|j�|j�S)Nrznegative maxlength)r?r@r$�_recv_bytesrC�getvalue)r7Z    maxlengthrKrrr�
recv_bytes�s
z_ConnectionBase.recv_bytescCs�|j�|j�t|���}|j}|t|�}|dkr>td��n||krNtd��|j�}|j�}|||krvt|j    ���|j
d�|j||||||��|SQRXdS)Nrznegative offsetzoffset too large)r?r@rFrGrIr$rU�tellr    rV�seek�readinto)r7rKrLrNrGZbytesize�resultrMrrr�recv_bytes_into�s"


z_ConnectionBase.recv_bytes_intocCs&|j�|j�|j�}tj|j��S)N)r?r@rUrQ�loads�    getbuffer)r7rKrrr�recv�sz_ConnectionBase.recv�cCs|j�|j�|j|�S)N)r?r@�_poll)r7rrrr�pollsz_ConnectionBase.pollcCs|S)Nr)r7rrr�    __enter__sz_ConnectionBase.__enter__cCs|j�dS)N)rB)r7�exc_type�    exc_value�exc_tbrrr�__exit__
sz_ConnectionBase.__exit__)TT)rN)N)r)r`)�__name__�
__module__�__qualname__r4r;r=r?r@rArC�propertyrDr9r:rErBrPrTrWr\r_rbrcrgrrrrr2vs(






r2c@s@eZdZdZejfdd�Zdd�Zd
dd�Zd    d
�Z    dd�Z
dS)�PipeConnectionFcCs||j�dS)N)r4)r7Z_CloseHandlerrrr<szPipeConnection._closecCshtj|j|dd�\}}z<y |tjkr6tj|jgdt�}Wn|j��YnXWd|jd�\}}XdS)NT)�
overlappedF)    �_winapiZ    WriteFiler4�ERROR_IO_PENDING�WaitForMultipleObjects�eventr�cancel�GetOverlappedResult)r7rK�ov�err�waitresZnwrittenrrrrJs
zPipeConnection._send_bytesNc    Cs|jrd|_tj�S|dkr dnt|d�}y�tj|j|dd�\}}z<y |tjkrbtj|j    gdt
�}Wn|j��YnXWd|jd�\}}|dkr�tj�}|j
|j��|S|tjkr�|j||�SXWn8tk
�r}z|jtjkr�t�n�WYdd}~XnXtd��dS)NF�T)rmrz.shouldn't get here; expected KeyboardInterrupt)�_got_empty_message�io�BytesIO�minrn�ReadFiler4rorprqrrrrs�writer^ZERROR_MORE_DATA�_get_more_datar>�winerror�ERROR_BROKEN_PIPE�EOFError�RuntimeError)    r7�maxsizeZbsizertrurvZnread�f�errrrU*s6


zPipeConnection._recv_bytescCs.|jstj|j�ddkrdStt|g|��S)NrT)rxrn�
PeekNamedPiper4�boolr)r7rrrrraJszPipeConnection._pollcCs�|j�}tj�}|j|�tj|j�d}|dk    rJt|�||krJ|j�tj    |j|dd�\}}|j
d�\}}|j|j��|S)NrT)rm)r^ryrzr}rnr�r4rIrCr|rs)r7rtr�rKr��leftruZrbytesrrrr~Ps
zPipeConnection._get_more_data)N)rhrirjrxrn�CloseHandler<rJrUrar~rrrrrls
 rlc@sxeZdZer(ejfdd�ZejZej    Z
nejfdd�Zej
ZejZ
efdd�Ze
fdd�Zdd    �Zddd�Zd
d�Zd
S)�
ConnectioncCs||j�dS)N)r4)r7r<rrrr<gszConnection._closecCs||j�dS)N)r4)r7r<rrrr<lscCs<t|�}x.||j|�}||8}|dkr(P||d�}q
WdS)Nr)rIr4)r7rKr}�    remainingrOrrr�_sendqszConnection._sendcCsftj�}|j}|}xN|dkr`|||�}t|�}|dkrL||krDt�ntd��|j|�||8}qW|S)Nrzgot end of file during message)ryrzr4rIr�r>r})r7rM�readrKr8r��chunkrOrrr�_recvzs


zConnection._recvcCsDt|�}tjd|�}|dkr2|j|�|j|�n|j||�dS)Nz!ii@)rI�structZpackr�)r7rKrO�headerrrrrJ�s
zConnection._send_bytesNcCs:|jd�}tjd|j��\}|dk    r0||kr0dS|j|�S)N�z!i)r�r�ZunpackrV)r7r�rKrMrrrrU�s

zConnection._recv_bytescCst|g|�}t|�S)N)rr�)r7r�rrrrra�szConnection._poll)N)rhrirjrn�_multiprocessingZclosesocketr<rTZ_writer_Z_readr rBr}r�r�r�rJrUrarrrrr�`s    
r�c@sNeZdZddd�Zdd�Zdd�Zed    d
��Zedd
��Zdd
�Z    dd�Z
dS)rNrcCsr|p|rt|�pt}|pt|�}t|�|dkr>t||�|_nt|||�|_|dk    rht|t�rht    d��||_
dS)Nrzauthkey should be a byte string)r1�default_familyr&r+�PipeListener�    _listener�SocketListener�
isinstancerH�    TypeError�_authkey)r7r0r%�backlog�authkeyrrrr;�szListener.__init__cCs>|jdkrtd��|jj�}|jr:t||j�t||j�|S)Nzlistener is closed)r�r>�acceptr��deliver_challenge�answer_challenge)r7�crrrr��s

zListener.acceptcCs |j}|dk    rd|_|j�dS)N)r�rB)r7ZlistenerrrrrB�szListener.closecCs|jjS)N)r��_address)r7rrr�<lambda>�szListener.<lambda>cCs|jjS)N)r��_last_accepted)r7rrrr��scCs|S)Nr)r7rrrrc�szListener.__enter__cCs|j�dS)N)rB)r7rdrerfrrrrg�szListener.__exit__)NNrN)rhrirjr;r�rBrkr0Z
last_acceptedrcrgrrrrr�s
    cCsj|p
t|�}t|�|dkr&t|�}nt|�}|dk    rJt|t�rJtd��|dk    rft||�t||�|S)Nrzauthkey should be a byte string)    r1r+�
PipeClient�SocketClientr�rHr�r�r�)r0r%r�r�rrrr�s


TcCsj|r>tj�\}}|jd�|jd�t|j��}t|j��}n$tj�\}}t|dd�}t|dd�}||fS)NTF)r:)r9)r*Z
socketpair�setblockingr��detachr �pipe)�duplex�s1�s2�c1�c2Zfd1Zfd2rrrr�s

c
    Cs�td�}|r*tj}tjtjB}tt}}ntj}tj}dt}}tj||tjBtj    Btj
tjBtjBd||tj
tj�}tj||dtjtjtjtj�}tj|tjdd�tj|dd�}|jd�\}    }
t||d�}t||d�}||fS)NrrrT)rm)r:)r9)r&rn�PIPE_ACCESS_DUPLEX�GENERIC_READ�
GENERIC_WRITE�BUFSIZEZPIPE_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�ConnectNamedPipersrl)
r�r0Zopenmode�accessZobsizeZibsizeZh1Zh2rm�_rur�r�rrrrs2



c@s&eZdZd    dd�Zdd�Zdd�ZdS)
r�rcCs�tjtt|��|_yRtjdkr2|jjtjtjd�|jjd�|jj    |�|jj
|�|jj�|_Wn t
k
r�|jj��YnX||_d|_|dkr�tj|tj|fdd�|_nd|_dS)N�posixrTrr)�args�exitpriority)r*�getattr�_socketr �nameZ
setsockoptZ
SOL_SOCKETZSO_REUSEADDRr�ZbindZlistenZgetsocknamer�r>rBZ_familyr�r�Finalize�unlink�_unlink)r7r0r%r�rrrr;<s$



zSocketListener.__init__cCs&|jj�\}|_|jd�t|j��S)NT)r�r�r�r�r�r�)r7�srrrr�Ts
zSocketListener.acceptcCs0z|jj�Wd|j}|dk    r*d|_|�XdS)N)r�rBr�)r7r�rrrrBYszSocketListener.closeN)r)rhrirjr;r�rBrrrrr�8s
r�c    CsDt|�}tjtt|���"}|jd�|j|�t|j��SQRXdS)NT)r1r*r�r�Zconnectr�r�)r0r%r�rrrr�cs


r�c@s4eZdZddd�Zddd�Zdd�Zed    d
��ZdS)
r�NcCsL||_|jdd�g|_d|_tjd|j�tj|tj|j|jfdd�|_    dS)NT)�firstz listener created with address=%rr)r�r�)
r��_new_handle�
_handle_queuer�r�    sub_debugr�r��_finalize_pipe_listenerrB)r7r0r�rrrr;wszPipeListener.__init__Fc    CsHtjtjB}|r|tjO}tj|j|tjtjBtjBtj    t
t
tjtj�S)N)
rnr�r�r�r�r�r�r�r�ZPIPE_UNLIMITED_INSTANCESr�r�r�)r7r��flagsrrrr��s

zPipeListener._new_handlecCs�|jj|j��|jjd�}ytj|dd�}Wn0tk
r^}z|jtjkrN�WYdd}~XnPXz<ytj    |j
gdt�}Wn |j�tj
|��YnXWd|jd�\}}Xt|�S)NrT)rmF)r��appendr��poprnr�r>rZ
ERROR_NO_DATArprqrrrr�rsrl)r7r8rtr��resr�rurrrr��s"
zPipeListener.acceptcCs(tjd|�x|D]}tj|�qWdS)Nz closing listener with address=%r)rr�rnr�)Zqueuer0r8rrrr��s
z$PipeListener._finalize_pipe_listener)N)F)rhrirjr;r�r��staticmethodr�rrrrr�ss

r�cCs�t�}x~y6tj|d�tj|tjtjBdtjtjtjtj�}Wn>t    k
r|}z"|j
tjtjfksjt
|�rl�WYdd}~XqXPqW�tj|tjdd�t|�S)Ni�r)rrnZ
WaitNamedPiper�r�r�r�r�r�r>rZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYrr�r�rl)r0r�hr�rrrr��s"
r��s#CHALLENGE#s    #WELCOME#s    #FAILURE#cCsfddl}tjt�}|jt|�|j||t�j�}|j    d�}||krP|jt
�n|jt�td��dS)Nr�zdigest received was wrong)
�hmacr �urandom�MESSAGE_LENGTHrP�    CHALLENGE�new�HMAC_DIGEST_NAME�digestrW�WELCOME�FAILUREr)�
connectionr�r��messager��responserrrr��s


r�cCs\ddl}|jd�}|tt�d�}|j||t�j�}|j|�|jd�}|tkrXt    d��dS)Nrr�zdigest sent was rejected)
r�rWrIr�r�r�r�rPr�r)r�r�r�r�r�r�rrrr��s


r�c@s$eZdZdd�Zdd�Zdd�ZdS)�ConnectionWrappercCs:||_||_||_x"dD]}t||�}t|||�qWdS)NrErBrbrWrP)rErBrbrWrP)�_conn�_dumps�_loadsr��setattr)r7�connrRr]�attrrSrrrr;�s

zConnectionWrapper.__init__cCs|j|�}|jj|�dS)N)r�r�rP)r7rSr�rrrrT�s
zConnectionWrapper.sendcCs|jj�}|j|�S)N)r�rWr�)r7r�rrrr_�s
zConnectionWrapper.recvN)rhrirjr;rTr_rrrrr��sr�cCstj|fdddd�jd�S)Nrzutf-8)�    xmlrpclibrR�encode)rSrrr�
_xml_dumps�sr�cCstj|jd��\\}}|S)Nzutf-8)r�r]�decode)r�rS�methodrrr�
_xml_loads�sr�c@seZdZdd�ZdS)�XmlListenercCs ddljatj|�}t|tt�S)Nr)�
xmlrpc.client�clientr�rr�r�r�r�)r7rSrrrr�s

zXmlListener.acceptN)rhrirjr�rrrrr�sr�cOsddljatt||�tt�S)Nr)r�r�r�r�rr�r�)r��kwdsrrr�    XmlClient
s
r�cCs�t|�}g}x�|r�tj|d|�}|tkr,Pn\t|koFtt|�knrV|t8}n2t|koptt|�knr�|t8}ntd��|j||�||dd�}d}qW|S)NFzShould not get hererr)    �listrnrpr
rrIrr�r�)Zhandlesr�L�readyr�rrr�_exhaustive_waits 
 
r�c
.sn|dkrt}n|dkrd}nt|dd�}t|�}i�g}t��t�}�zH�x2|D�](}yt|d�}Wn tk
r�|�|j�<YqTXytj|�dd�\}}Wn8t    k
r�}zd|j
}}|tkrʂWYdd}~XnX|tjkr�|j
|�|�|j<qT|�rptj�dd�dk�rpy|jd�\}    }Wn*t    k
�rT}z|j
}WYdd}~XnX|�rpt|d    ��rpd|_�j|�d}qTWt�j�|�}Wdx|D]}|j��q�Wx�|D]�}y|jd�\}    }Wn6t    k
�r�}z|j
}|tk�r�WYdd}~XnX|tjk�r��|j}�j|�|dk�r�t|d    ��r�d|_�q�WX�j�fd
d�|D���fdd
�|D�S)Nri�g�?rET��Frxc3s|]}�|VqdS)Nr)�.0r�)�waithandle_to_objrr�    <genexpr>vszwait.<locals>.<genexpr>csg|]}|�kr|�qSrr)r��o)�
ready_objectsrr�
<listcomp>wszwait.<locals>.<listcomp>)r�r�)r�intr��setr��AttributeErrorr3rnr|r>r�
_ready_errorsror�rqr'Zgetwindowsversionrsr)rx�addr��keysrrZERROR_OPERATION_ABORTED�update)
�object_listrZov_listZ
ready_handlesr�rErtrur�r�r)rr�rr+sh








�PollSelectorcCs�t��x}x|D]}|j|tj�qW|dk    r8tj�|}x@|j|�}|rVdd�|D�S|dk    r:|tj�}|dkr:|Sq:WWdQRXdS)NcSsg|]\}}|j�qSr)Zfileobj)r��keyZeventsrrrr�szwait.<locals>.<listcomp>r)�
_WaitSelector�register�    selectorsZ
EVENT_READrrZselect)r    rZselectorrSZdeadliner�rrrr�s

cCsN|j�}tj|tjtj��*}ddlm}|j|�}t||j    |j
ffSQRXdS)Nr)�resource_sharer)rEr*ZfromfdrZSOCK_STREAMrrZ    DupSocket�rebuild_connectionr9r:)r�r8r�r�dsrrr�reduce_connection�s

rcCs|j�}t|j�||�S)N)r�r�)rr9r:Zsockrrrr�srcCsB|jrtjnd|jrtjndB}tj|j�|�}t||j|jffS)Nr)    r9rnZFILE_GENERIC_READr:ZFILE_GENERIC_WRITEr
Z    DupHandlerE�rebuild_pipe_connection)r�r��dhrrr�reduce_pipe_connection�srcCs|j�}t|||�S)N)r�rl)rr9r:r8rrrr�srcCs tj|j��}t||j|jffS)N)r
ZDupFdrErr9r:)r��dfrrrr�scCs|j�}t|||�S)N)r�r�)rr9r:�fdrrrr�s)NN)T)T)N)N)J�__all__ryr r'r*r�rr�    itertoolsr�rrrr    �contextr
ZForkingPicklerrQrnrrr
r�ImportErrorr(r�ZCONNECTION_TIMEOUTr��countr#r�Zfamiliesr)rrr&r+r1r2rlr��objectrrrr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�ZERROR_NETNAME_DELETEDrrrr
rZSelectSelectorrrr
rrrrrr�<module>
s�



PJ8

,+8
    P