Edit File: connection.cpython-37.opt-2.pyc
B YI~dBz � @ 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 Ze�� Z dZ dgZ e ed�r�dZ e dg7 Z ejd k�rdZ e dg7 Z efdd�Z dd� Z!dd� Z"dd� Z#dd� Z$G dd � d �Z%e�rhG d d � d e%�Z&G d d � d e%�Z'G d!d� de(�Z)dOd"d �Z*ejd k�r�dPd$d�Z+n dQd%d�Z+G d&d'� d'e(�Z,d(d)� Z-ejd k�r�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:ejd k�rtd@dA� Z;ej<ej=hZ>dRdBd�Z?n,ddl@Z@e e@dC��r�e@jAZBne@jCZBdSdDd�Z?ejd k�r�dEdF� ZDdGdH� ZEe�Fe'eD� dIdJ� ZGdKdL� ZHe�Fe&eG� n dMdF� ZDdNdH� ZEe�Fe'eD� dS )T�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 �?/opt/alt/python37/lib64/python3.7/multiprocessing/connection.py� _init_timeout; s r c C s t �� | kS )N)r r )�tr r r �_check_timeout> s r c C sX | dkr dS | dkr&t jdt�� d�S | dkrLt jdt�� tt�f dd�S td ��d S ) Nr )Z localhostr r z listener-)�prefix�dirr z\\.\pipe\pyc-%d-%d-� zunrecognized family) �tempfileZmktempr Z get_temp_dir�os�getpid�next� _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 )�sys�platformr# �hasattr�socket)r$ r r r �_validate_familyS s r* c C sT t | �tkrdS t | �tkr*| �d�r*dS t | �tks@t�| �rDdS td| ��d S )Nr z\\r r z address type of %r unrecognized)�type�tuple�str� startswithr � is_abstract_socket_namespacer# )�addressr r r � address_type_ s r1 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 rtd��|s(|s(td��|| _|| _|| _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 k r| �� d S )N)r4 �_close)r7 r r r �__del__� s z_ConnectionBase.__del__c C s | j d krtd��d S )Nzhandle is closed)r4 �OSError)r7 r r r � _check_closed� s z _ConnectionBase._check_closedc C s | j std��d S )Nzconnection is write-only)r5 r> )r7 r r r �_check_readable� s z _ConnectionBase._check_readablec C s | j std��d S )Nzconnection is read-only)r6 r> )r7 r r r �_check_writable� s z _ConnectionBase._check_writablec C s" | j rd| _n| �� td��d 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 | � � | jS )N)r? r4 )r7 r r r �fileno� s z_ConnectionBase.filenoc C s$ | j d k r z | �� W d d | _ X d S )N)r4 r< )r7 r r r rB � s z_ConnectionBase.closer c C s� | � � | �� t|�}|jdkr.tt|��}t|�}|dk rFtd��||k rVtd��|d krh|| }n&|dk rztd��n|| |kr�td��| �|||| � � d S )Nr r zoffset is negativezbuffer length < offsetzsize is negativez buffer length < offset + size)r? rA � memoryview�itemsize�bytes�lenr# � _send_bytes)r7 �buf�offset�size�m�nr r r � send_bytes� s" z_ConnectionBase.send_bytesc C s$ | � � | �� | �t�|�� d S )N)r? rA rJ �_ForkingPickler�dumps)r7 �objr r r �send� s z_ConnectionBase.sendc C sJ | � � | �� |d k r(|dk r(td��| �|�}|d krB| �� |�� 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� | � � | �� t|���}|j}|t|� }|dk r>td��n||krNtd��| �� }|�� }||| k rvt|� � ��|� d� |� ||| || | � � |S Q R X d S )Nr znegative offsetzoffset too large) r? r@ rF rG rI r# rU �tellr rV �seek�readinto)r7 rK rL rN rG Zbytesize�resultrM r r r �recv_bytes_into� s" z _ConnectionBase.recv_bytes_intoc C s& | � � | �� | �� }t�|�� �S )N)r? r@ rU rQ �loads� getbuffer)r7 rK r r r �recv� s z_ConnectionBase.recv� c C s | � � | �� | �|�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 | � � d S )N)rB )r7 �exc_type� exc_value�exc_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 ZdZejfdd�Zdd� Zd dd�Zd 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<y |t jkr6t �|jgdt�}W n |�� � Y nX W d |�d�\}}X d S )NT)� overlappedF) �_winapiZ WriteFiler4 �ERROR_IO_PENDING�WaitForMultipleObjects�eventr �cancel�GetOverlappedResult)r7 rK �ov�err�waitresZnwrittenr r r rJ s zPipeConnection._send_bytesNc C s | j rd| _ t�� S |d kr dnt|d�}y�tj| j|dd�\}}z<y |tjkrbt�|j gdt �}W n |� � � Y nX W d |� d�\}}|dkr�t�� }|� |�� � |S |tjkr�| �||�S X W n8 tk �r } z|jtjkr�t�n� W d d }~X Y nX td��d S )NF� T)rm r z.shouldn't get here; expected KeyboardInterrupt)�_got_empty_message�io�BytesIO�minrn �ReadFiler4 ro rp rq r rr rs �writer^ ZERROR_MORE_DATA�_get_more_datar> �winerror�ERROR_BROKEN_PIPE�EOFError� RuntimeError) r7 �maxsizeZbsizert ru rv Znread�f�er r r rU & s6 zPipeConnection._recv_bytesc C s. | j st�| j�d dkr dS tt| g|��S )Nr T)rx rn � PeekNamedPiper4 �boolr )r7 r r r r ra F s zPipeConnection._pollc C s� |� � }t�� }|�|� t�| j�d }|d k rJt|�| |krJ| �� tj | j|dd�\}}|� d�\}}|�|� � � |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(ejfdd�ZejZej Z ne j fdd�Ze j Ze jZ efdd�Ze fdd�Zdd � 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}|}xN|dkr`|||�}t|�}|dkrL||krDt�ntd��|�|� ||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 |�}t�d|�}|dkr2| �|� | �|� n| �|| � d S )Nz!ii @ )rI �struct�packr� )r7 rK rO �headerr r r rJ � s zConnection._send_bytesNc C s: | � d�}t�d|�� �\}|d k r0||kr0d S | � |�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 Zddd�Zdd� Zdd� Zed d � �Zed d � �Zd d� Z dd� Z dS )r Nr c C sp |p|rt |�pt}|p t|�}t|� |dkr>t||�| _nt|||�| _|d k rft|t�sft d��|| _ d S )Nr z authkey should be a byte string) r1 �default_familyr% r* � PipeListener� _listener�SocketListener� isinstancerH � TypeError�_authkey)r7 r0 r$ �backlog�authkeyr r r r; � s zListener.__init__c C s> | j d krtd��| j �� }| jr:t|| j� t|| j� |S )Nzlistener is closed)r� r> �acceptr� �deliver_challenge�answer_challenge)r7 �cr r r r� � s zListener.acceptc C s | j }|d k r d | _ |�� d S )N)r� rB )r7 Zlistenerr r r rB � s zListener.closec C s | j jS )N)r� �_address)r7 r r r r0 � s zListener.addressc C s | j jS )N)r� �_last_accepted)r7 r r r � last_accepted� s zListener.last_acceptedc C s | S )Nr )r7 r r r rc � s zListener.__enter__c C s | � � 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 r� rc rg r r r r r � s c C sh |p t | �}t|� |dkr&t| �}nt| �}|d k rHt|t�sHtd��|d k rdt||� 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 �� \}}|�d� |�d� t|�� �}t|�� �}n$t�� \}}t|dd�}t|dd�}||fS )NTF)r: )r9 )r) Z socketpair� setblockingr� �detachr �pipe)�duplex�s1�s2�c1�c2Zfd1Zfd2r r r r � s c C s� t d�}| r*tj}tjtjB }tt }}ntj}tj}dt }}t�||tjB tj B tj tj B tj B d||tj tj�}t�||dtjtjtjtj�}t�|tj d d � tj|dd�}|�d�\} } t|| d�} t|| d�} | | fS )Nr r r T)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�ConnectNamedPipers rl ) r� r0 Zopenmode�accessZobsizeZibsizeZh1Zh2rm �_ru r� r� r r r r s2 c @ s& e Zd Zd dd�Zdd� Zdd� ZdS ) r� r c C s� t � tt |��| _yRtjdkr2| j�t jt jd� | j�d� | j� |� | j� |� | j� � | _ W n t k r� | j�� � Y nX || _d | _|dkr�t�|�s�tj| tj|fdd�| _nd | _d S )N�posixr Tr r )�args� exitpriority)r) �getattr�_socketr �nameZ setsockoptZ SOL_SOCKETZ SO_REUSEADDRr� ZbindZlistenZ getsocknamer� r> rB Z_familyr� r r/ �Finalize�unlink�_unlink)r7 r0 r$ r� r r r r; = s$ zSocketListener.__init__c C s& | j �� \}| _|�d� t|�� �S )NT)r� r� r� r� r� r� )r7 �sr r r r� V s zSocketListener.acceptc C s0 z| j �� W d | j}|d k r*d | _|� X d S )N)r� rB r� )r7 r� r r r rB [ s zSocketListener.closeN)r )rh ri rj r; r� rB r r r r r� 9 s r� c C sD t | �}t�tt|���"}|�d� |�| � t|�� �S Q R X d S )NT)r1 r) r� r� Zconnectr� r� )r0 r$ r� r r r r� e s r� c @ s4 e Zd Zd dd�Zd dd�Zdd� Zed d � �ZdS ) r� 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� ) r� � _new_handle� _handle_queuer� r � sub_debugr� r� �_finalize_pipe_listenerrB )r7 r0 r� r r r r; y 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 )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 �d�}ytj|dd�}W n0 tk r^ } z|jtjkrN� W d d }~X Y nPX z<yt� |j gdt �}W n |� � t� |� � Y nX W d |�d�\}}X t|�S )Nr T)rm F)r� �appendr� �poprn r� r> 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 �d|� x| D ]}t�|� 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� u s r� c C s� t � }x~y6t�| d� t�| tjtjB dtjtjtjtj�}W n> t k r| } z |j tj tj fksjt |�rl� W d d }~X Y qX P qW � t�|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 s� dd l }t|t�s$td�t|����t�t�}| � t | � |� ||d�� � }| � d�}||krl| � t� n| � t� td��d S )Nr z Authkey must be bytes, not {0!s}�md5� zdigest received was wrong)�hmacr� rH r# �formatr+ r �urandom�MESSAGE_LENGTHrP � CHALLENGE�new�digestrW �WELCOME�FAILUREr )� connectionr� r� �messager� �responser r r r� � s r� c C sx dd l }t|t�s$td�t|����| �d�}|tt�d � }|� ||d�� � }| � |� | �d�}|t krtt d��d S )Nr z Authkey must be bytes, not {0!s}r� r� zdigest sent was rejected)r� r� rH r# r� 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 )N)rE rB rb rW rP )�_conn�_dumps�_loadsr� �setattr)r7 �connrR r] �attrrS r r r r; � s zConnectionWrapper.__init__c C s | � |�}| j�|� d S )N)r� r� rP )r7 rS r� r r r rT � s zConnectionWrapper.sendc C s | 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 �| fd d d d��d�S )Nr zutf-8)� xmlrpclibrR �encode)rS r r r � _xml_dumps s r� c C s t �| �d��\\}}|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 ma t�| �}t|tt�S )Nr )� xmlrpc.client�clientr� 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 ma tt| |�tt�S )Nr )r� r� r� r� r r� r� )r� �kwdsr r r � XmlClient s r� c C s� t | �}g }x�|r�t�|d|�}|tkr,P n\t| krHtt|� k rVn n |t8 }n2t| krrtt|� k r�n n |t8 }ntd��|�|| � ||d d � }d}qW |S )NFzShould not get herer r ) �listrn rp r r rI r r� r� )Zhandlesr �L�readyr� r r r �_exhaustive_wait s r� c sl |d krt }n |dk r d}nt|d d �}t| �} i �g }t� � t� }�zF�x0| D �]&}yt|d�}W n tk r� |�|�� < Y qTX yt�|� dd�\}}W n8 t k r� } zd |j }}|t krʂ W d d }~X Y nX |tj kr�|� |� |�|j< qT|�rnt�� d d� dk�rny|�d�\} }W n* t k �rT } z |j }W d d }~X Y nX |�snt|d ��rnd|_� �|� d}qTW t��� |�}W d x|D ]}|�� �q�W x�|D ]�}y|�d�\} }W n6 t k �r� } z|j }|t k�r� W d d }~X Y nX |tjk�r��|j }� �|� |dk�r�t|d ��r�d|_�q�W X � ��fd d �|D �� � fd d �| D �S )Nr i� g �?rE T� )� r� Frx c 3 s | ]}� | V qd S )Nr )�.0r� )�waithandle_to_objr r � <genexpr>| s zwait.<locals>.<genexpr>c s g | ]}|� kr|�qS r r )r� �o)� ready_objectsr r � <listcomp>} s zwait.<locals>.<listcomp>)r �intr� �setr� �AttributeErrorr3 rn r| r> r � _ready_errorsro r� rq r& Zgetwindowsversionrs r( rx �addr� �keysrr ZERROR_OPERATION_ABORTED�update) � object_listr Zov_listZ ready_handlesr rE rt ru r� r� r )r r r r 1 sh � PollSelectorc C s� t � �x}x| D ]}|�|tj� qW |d k r8t�� | }x@|�|�}|rVdd� |D �S |d k r:|t�� }|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.<locals>.<listcomp>r )� _WaitSelector�register� selectorsZ EVENT_READr r Zselect)r r ZselectorrS Zdeadliner� r r r r � s c C sN | � � }t�|tjtj��*}ddlm} |�|�}t|| j | j ffS Q R X d S )Nr )�resource_sharer) rE r) Zfromfdr Z SOCK_STREAMr r Z DupSocket�rebuild_connectionr9 r: )r� r8 r� r �dsr r r �reduce_connection� s r c C s | � � }t|� � ||�S )N)r� r� )r r9 r: Zsockr r r r � s r c C sB | j r tjnd| jrtjndB }t�| �� |�}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 | � � }t|||�S )N)r� rl )r r9 r: r8 r r r r � s r c C s t �| �� �}t|| j| jffS )N)r ZDupFdrE r r9 r: )r� �dfr r r r � s c C s | � � }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_TIMEOUT�countr" 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 �<module> s� PJ= ,, 8 P
Back to File Manager