3
2ax @ s d dddgZ ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl
m Z m
Z
dd lmZ ejZy$ddlZdd
lmZmZmZmZ W n$ ek
r ejd kr dZY nX d Zd
Zej Z dZ dgZ e edrdZ e dg7 Z ejd krdZ e dg7 Z efddZ dd Z!dd Z"dd Z#dd Z$G dd d Z%erhG d d d e%Z&G d d d e%Z'G d!d de(Z)dOd"d Z*ejd krdPd$dZ+n
dQd%dZ+G d&d' d'e(Z,d(d) Z-ejd krG d*d+ d+e(Z.d,d- Z/d.Z0d/Z1d0Z2d1Z3d2d3 Z4d4d5 Z5G d6d7 d7e(Z6d8d9 Z7d:d; Z8G dd? Z:ejd krtd@dA Z;ejdRdBdZ?n,ddl@Z@e e@dCre@jAZBne@jCZBdSdDdZ?ejd krdEdF ZDdGdH ZEejFe'eD dIdJ ZGdKdL ZHejFe&eG n dMdF ZDdNdH ZEejFe'eD dS )TClientListenerPipewait N )util)AuthenticationErrorBufferTooShort) reduction)
WAIT_OBJECT_0WAIT_ABANDONED_0 WAIT_TIMEOUTINFINITEwin32i g 4@AF_INETAF_UNIXAF_PIPEc C s t j | S )N)time monotonic)timeout r //opt/alt/python36/lib64/python3.6/connection.py
_init_timeout; s r c C s t j | kS )N)r r )tr r r _check_timeout> s r c C sX | dkr d S | dkr&t jdtj dS | dkrLt jdtj ttf d dS td
d S ) Nr localhostr r z listener-)prefixdirr z\\.\pipe\pyc-%d-%d- zunrecognized family)r r ) tempfileZmktempr Z get_temp_dirosgetpidnext
_mmap_counter
ValueError)familyr r r arbitrary_addressE s r& c C sJ t jdkr | dkr td| t jdkrF| dkrFtt| sFtd| d S )Nr r z Family %s is not recognized.r )sysplatformr$ hasattrsocket)r% r r r _validate_familyS s
r+ c C sJ t | tkrdS t | tkr*| jdr*dS t | tkr:dS td| d S )Nr z\\r r z address type of %r unrecognized)typetuplestr
startswithr$ )addressr r r address_type_ s r1 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 sB |j }|dk rtd| r,| r,td|| _|| _|| _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 k r| j d S )N)r4 _close)r7 r r r __del__ s
z_ConnectionBase.__del__c C s | j d krtdd S )Nzhandle is closed)r4 OSError)r7 r r r
_check_closed s
z _ConnectionBase._check_closedc C s | j stdd S )Nzconnection is write-only)r5 r> )r7 r r r _check_readable s z _ConnectionBase._check_readablec C s | j stdd S )Nzconnection is read-only)r6 r> )r7 r r r _check_writable s z _ConnectionBase._check_writablec C s" | j rd| _n| j tdd S )NFzbad message length)r6 r5 closer> )r7 r r r _bad_message_length s z#_ConnectionBase._bad_message_lengthc C s
| j d kS )N)r4 )r7 r r r closed s z_ConnectionBase.closedc C s | j S )N)r5 )r7 r r r r9 s z_ConnectionBase.readablec C s | j S )N)r6 )r7 r r r r: s z_ConnectionBase.writablec C s | j | jS )N)r? r4 )r7 r r r fileno s z_ConnectionBase.filenoc C s$ | j d k r z | j W d d | _ X d S )N)r4 r< )r7 r r r rB s
z_ConnectionBase.closer c C s | j | j t|}|jdkr.tt|}t|}|dk rFtd||k rVtd|d krh|| }n&|dk rztdn|| |krtd| j|||| d S )Nr r zoffset is negativezbuffer length < offsetzsize is negativez buffer length < offset + size)r? rA
memoryviewitemsizebyteslenr$ _send_bytes)r7 bufoffsetsizemnr r r
send_bytes s"
z_ConnectionBase.send_bytesc C s$ | j | j | jtj| d S )N)r? rA rJ _ForkingPicklerdumps)r7 objr r r send s z_ConnectionBase.sendc C sJ | j | j |d k r(|dk r(td| j|}|d krB| j |j S )Nr znegative maxlength)r? r@ r$ _recv_bytesrC getvalue)r7 Z maxlengthrK r r r
recv_bytes s
z_ConnectionBase.recv_bytesc C s | j | j t|}|j}|t| }|dk r>tdn||krNtd| j }|j }||| k rvt|j |j
d |j ||| || | |S Q R X d S )Nr znegative offsetzoffset too large) r? r@ rF rG rI r$ rU tellr rV seekreadinto)r7 rK rL rN rG ZbytesizeresultrM r r r recv_bytes_into s"
z _ConnectionBase.recv_bytes_intoc C s& | j | j | j }tj|j S )N)r? r@ rU rQ loads getbuffer)r7 rK r r r recv s z_ConnectionBase.recv c C s | j | j | j|S )N)r? r@ _poll)r7 r r r r poll s z_ConnectionBase.pollc C s | S )Nr )r7 r r r __enter__ s z_ConnectionBase.__enter__c C s | j d S )N)rB )r7 exc_type exc_valueexc_tbr r r __exit__ s z_ConnectionBase.__exit__)TT)r N)N)r )r` )__name__
__module__ __qualname__r4 r; r= r? r@ rA rC propertyrD r9 r: rE rB rP rT rW r\ r_ rb rc rg r r r r r2 r s(
r2 c @ s@ e Zd ZdZejfddZdd Zd
ddZd d
Z d d Z
dS )PipeConnectionFc C s || j d S )N)r4 )r7 Z _CloseHandler r r r< s zPipeConnection._closec C sh t j| j|dd\}}z winerrorERROR_BROKEN_PIPEEOFError RuntimeError) r7 maxsizeZbsizert ru rv Znreadfer r r rU & s6
zPipeConnection._recv_bytesc C s. | j stj| jd dkr dS tt| g|S )Nr T)rx rn
PeekNamedPiper4 boolr )r7 r r r r ra F s zPipeConnection._pollc C s |j }tj }|j| tj| jd }|d k rJt|| |krJ| j tj | j|dd\}}|j
d\}}|j|j |S )Nr T)rm ) r^ ry rz r} rn r r4 rI rC r| rs )r7 rt r rK r leftru Zrbytesr r r r~ L s
z PipeConnection._get_more_data)N) rh ri rj rx rn CloseHandler< rJ rU ra r~ r r r r rl s
rl c @ sx e Zd Zer(ejfddZejZej Z
ne j fddZe j
Ze jZ
efddZe
fddZdd Zdd d Zd
d Zd
S )
Connectionc C s || j d S )N)r4 )r7 r< r r r r< c s zConnection._closec C s || j d S )N)r4 )r7 r< r r r r< h s c C s< t |}x.|| j|}||8 }|dkr(P ||d }q
W d S )Nr )rI r4 )r7 rK r} remainingrO r r r _sendm s zConnection._sendc C sf t j }| j}|}xN|dkr`|||}t|}|dkrL||krDtntd|j| ||8 }qW |S )Nr z got end of file during message)ry rz r4 rI r r> r} )r7 rM readrK r8 r chunkrO r r r _recvv s
zConnection._recvc C sD t |}tjd|}|dkr2| j| | j| n| j|| d S )Nz!ii @ )rI structpackr )r7 rK rO headerr r r rJ s
zConnection._send_bytesNc C s: | j d}tjd|j \}|d k r0||kr0d S | j |S )N z!i)r r unpackrV )r7 r rK rM r r r rU s
zConnection._recv_bytesc C s t | g|}t|S )N)r r )r7 r rr r r ra s zConnection._poll)N)rh ri rj rn _multiprocessingZ closesocketr< rT Z_writer_ Z_readr rB r} r r r rJ rU ra r r r r r \ s
r c @ sN e Zd ZdddZdd Zdd Zed d
Zed d
Zd d
Z dd Z
dS )r Nr c C sr |p|rt |pt}|p t|}t| |dkr>t||| _nt|||| _|d k rht|t rht d|| _
d S )Nr z authkey should be a byte string) r1 default_familyr& r+ PipeListener _listenerSocketListener
isinstancerH TypeError_authkey)r7 r0 r% backlogauthkeyr r r r; s zListener.__init__c C s> | j d krtd| j j }| jr:t|| j t|| j |S )Nzlistener is closed)r r> acceptr deliver_challengeanswer_challenge)r7 cr r r r s
zListener.acceptc C s | j }|d k r d | _ |j d S )N)r rB )r7 Zlistenerr r r rB s zListener.closec C s | j jS )N)r _address)r7 r r r s zListener.c C s | j jS )N)r _last_accepted)r7 r r r r s c C s | S )Nr )r7 r r r rc s zListener.__enter__c C s | j d S )N)rB )r7 rd re rf r r r rg s zListener.__exit__)NNr N) rh ri rj r; r rB rk r0 Z
last_acceptedrc rg r r r r r s
c C sj |p
t | }t| |dkr&t| }nt| }|d k rJt|t rJtd|d k rft|| t|| |S )Nr z authkey should be a byte string) r1 r+
PipeClient SocketClientr rH r r r )r0 r% r r r r r r s
Tc C sj | r>t j \}}|jd |jd t|j }t|j }n$tj \}}t|dd}t|dd}||fS )NTF)r: )r9 )r* Z
socketpair setblockingr detachr pipe)duplexs1s2c1c2Zfd1Zfd2r r r r s
c
C s t d}| r*tj}tjtjB }tt }}ntj}tj}dt }}tj||tjB tj B tj
tj B tj B d||tj
tj}tj||dtjtjtjtj}tj|tj d d tj|dd}|jd\} }
t|| d} t|| d} | | fS )Nr r r T)rm )r: )r9 )r& rn 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_EXISTINGSetNamedPipeHandleStateConnectNamedPipers rl )
r r0 ZopenmodeaccessZobsizeZibsizeZh1Zh2rm _ru r r r r r r s2
c @ s& e Zd Zd ddZdd Zdd ZdS )
r r c C s t j tt || _yRtjdkr2| jjt jt jd | jjd | jj | | jj
| | jj | _ W n t
k
r | jj Y nX || _d | _|dkrtj| tj|fdd| _nd | _d S )Nposixr Tr r )args exitpriority)r* getattr_socketr nameZ
setsockoptZ
SOL_SOCKETZ SO_REUSEADDRr ZbindZlistenZ getsocknamer r> rB Z_familyr r Finalizeunlink_unlink)r7 r0 r% r r r r r; 8 s$
zSocketListener.__init__c C s& | j j \}| _|jd t|j S )NT)r r r r r r )r7 sr r r r P s
zSocketListener.acceptc C s0 z| j j W d | j}|d k r*d | _| X d S )N)r rB r )r7 r r r r rB U s zSocketListener.closeN)r )rh ri rj r; r rB r r r r r 4 s
r c C sD t | }tjtt|"}|jd |j| t|j S Q R X d S )NT)r1 r* r r Zconnectr r )r0 r% r r r r r _ s
r c @ s4 e Zd Zd ddZd ddZdd Zed d
ZdS )
r Nc C sL || _ | jddg| _d | _tjd| j tj| tj| j| j fdd| _ d S )NT)firstz listener created with address=%rr )r r )
r _new_handle
_handle_queuer r sub_debugr r _finalize_pipe_listenerrB )r7 r0 r r r r r; s s zPipeListener.__init__Fc C sH t jt jB }|r|t jO }t j| j|t jt jB t jB t j t
t
t j t j S )N)
rn r r r r r r r r ZPIPE_UNLIMITED_INSTANCESr r r )r7 r flagsr r r r ~ s
zPipeListener._new_handlec C s | j j| j | j jd}ytj|dd}W n0 tk
r^ } z|jtjkrN W Y d d }~X nPX z r Z
ERROR_NO_DATArp rq r rr r rs rl )r7 r8 rt r resr ru r r r r s"
zPipeListener.acceptc C s( t jd| x| D ]}tj| qW d S )Nz closing listener with address=%r)r r rn r )Zqueuer0 r8 r r r r s
z$PipeListener._finalize_pipe_listener)N)F)rh ri rj r; r r staticmethodr r r r r r o s
r c C s t }x~y6tj| d tj| tjtjB dtjtjtjtj}W n> t k
r| } z"|j
tj tj fksjt
|rl W Y d d }~X qX P qW tj|tjd d t|S )Ni r )r rn Z
WaitNamedPiper r r r r r r> r ZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYr r r rl )r0 r hr r r r r s"
r s #CHALLENGE#s #WELCOME#s #FAILURE#c C sf dd l }tjt}| jt| |j||dj }| jd}||krP| jt n| jt
t dd S )Nr md5 zdigest received was wrong) hmacr urandomMESSAGE_LENGTHrP CHALLENGEnewdigestrW WELCOMEFAILUREr )
connectionr r messager responser r r r s
r c C s\ dd l }| jd}|ttd }|j||dj }| j| | jd}|tkrXtdd S )Nr r r zdigest sent was rejected) r rW rI r r r rP r r )r r r r r r r r r r s
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ConnectionWrapperc C s: || _ || _|| _x"dD ]}t||}t| || qW d S )NrE rB rb rW rP )rE rB rb rW rP )_conn_dumps_loadsr setattr)r7 connrR r] attrrS r r r r; s
zConnectionWrapper.__init__c C s | j |}| jj| d S )N)r r rP )r7 rS r r r r rT s
zConnectionWrapper.sendc C s | j j }| j|S )N)r rW r )r7 r r r r r_ s
zConnectionWrapper.recvN)rh ri rj r; rT r_ r r r r r s r c C s t j| fd d d djdS )Nr zutf-8) xmlrpclibrR encode)rS r r r
_xml_dumps s r c C s t j| jd\\}}|S )Nzutf-8)r r] decode)r rS methodr r r
_xml_loads s r c @ s e Zd Zdd ZdS ) XmlListenerc C s dd l jatj| }t|ttS )Nr )
xmlrpc.clientclientr r r r r r )r7 rS r r r r s
zXmlListener.acceptN)rh ri rj r r r r r r s r c O s dd l jatt| |ttS )Nr )r r r r r r r )r kwdsr r r XmlClient s
r c C s t | }g }x|rtj|d|}|tkr,P n\t| koFtt| k n rV|t8 }n2t| koptt| k n r|t8 }ntd|j|| ||d d }d}qW |S )NFzShould not get herer r ) listrn rp r
r rI r r r )Zhandlesr Lreadyr r r r _exhaustive_wait s
r c
. sn |d krt }n |dk r d}nt|d d }t| } i g }t t }zHx2| D ](}yt|d}W n tk
r ||j < Y qTX ytj| dd\}}W n8 t k
r } z d |j
}}|t krʂ W Y d d }~X nX |tj kr|j
| ||j< qT|rptj d d dkrpy|jd\} }W n* t k
rT } z |j
}W Y d d }~X nX | rpt|d rpd|_ j| d}qTW tj |}W d x|D ]}|j qW x|D ]}y|jd\} }W n6 t k
r } z|j
}|t kr W Y d d }~X nX |tjkr|j } j| |dkrt|d rd|_qW X jfd
d |D fd d
| D S )Nr i g ?rE T Frx c 3 s | ]} | V qd S )Nr ).0r )waithandle_to_objr r r s zwait..c s g | ]}| kr|qS r r )r o)
ready_objectsr r
s s zwait..)r r )r intr setr AttributeErrorr3 rn r| r> r
_ready_errorsro r rq r' Zgetwindowsversionrs r) rx addr keysrr ZERROR_OPERATION_ABORTEDupdate)
object_listr Zov_listZ
ready_handlesr rE rt ru r r r )r r r r ' sh
PollSelectorc C s t x}x| D ]}|j|tj qW |d k r8tj | }x@|j|}|rVdd |D S |d k r:|tj }|dk r:|S q:W W d Q R X d S )Nc S s g | ]\}}|j qS r )Zfileobj)r keyZeventsr r r r s zwait..r )
_WaitSelectorregister selectorsZ
EVENT_READr r Zselect)r r ZselectorrS Zdeadliner r r r r s
c C sN | j }tj|tjtj*}ddlm} |j|}t|| j | j
ffS Q R X d S )Nr )resource_sharer) rE r* Zfromfdr Z SOCK_STREAMr r Z DupSocketrebuild_connectionr9 r: )r r8 r r dsr r r reduce_connection s
r c C s | j }t|j ||S )N)r r )r r9 r: Zsockr r r r s r c C sB | j r tjnd| jrtjndB }tj| j |}t|| j | jffS )Nr ) r9 rn ZFILE_GENERIC_READr: ZFILE_GENERIC_WRITEr
Z DupHandlerE rebuild_pipe_connection)r r dhr r r reduce_pipe_connection s r c C s | j }t|||S )N)r rl )r r9 r: r8 r r r r s r c C s t j| j }t|| j| jffS )N)r
ZDupFdrE r r9 r: )r dfr r r r s c C s | j }t|||S )N)r r )r r9 r: fdr r r r s )NN)T)T)N)N)I__all__ry r r' r* r r r itertoolsr r r r r contextr
ZForkingPicklerrQ rn r r r
r ImportErrorr( r ZCONNECTION_TIMEOUTcountr# r Zfamiliesr) r r r& r+ r1 r2 rl 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 r r r ZSelectSelectorr r r r r r r r r
s
PJ8
,+ 8
P