← Back
Editing: connection.cpython-310.opt-2.pyc
o 0�i�{ � @ 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 ejdkr�d Z e d g7 Z efdd�Z dd� Z!dd� Z"dd� Z#dd� Z$G dd� d�Z%er�G dd� de%�Z&G d d � d e%�Z'G d d � d e(�Z)dPd d!�Z*ejdkr�dQd#d$�Z+ndQd%d$�Z+G d&d'� d'e(�Z,d(d)� Z-ejdkr�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:ejdk�r3d@dA� Z;ej<ej=hZ>dRdBdC�Z?nddl@Z@e e@dD��rAe@jAZBne@jCZBdRdEdC�Z?ejdk�rmdFdG� ZDdHdI� ZEe�Fe'eD� dJdK� ZGdLdM� ZHe�Fe&eG� dS dNdG� ZDdOdI� ZEe�Fe'eD� dS )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 �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 sV | dkrdS | dkrt jdt�� d�S | dkr't jdt�� tt�f dd�S td ��) 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 sT t jdkr| dkrtd| ��t jdkr$| dkr&tt| �s(td| ��d S d S d S )Nr r z Family %s is not recognized.r )�sys�platformr& �hasattr�socketr' r r r �_validate_familyS s �r. c C sR t | �tkr dS t | �tu r| �d�rdS t | �tu s!t�| �r#dS td| ��)Nr z\\r r z address type of %r unrecognized)�type�tuple�str� startswithr � is_abstract_socket_namespacer& )�addressr r r � address_type_ s r5 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 r td��|s|std��|| _|| _|| _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 ur | �� d S d S r �r8 �_close�r; r r r �__del__� s �z_ConnectionBase.__del__c C s | j d u r td��d S )Nzhandle is closed)r8 �OSErrorrB r r r � _check_closed� s �z _ConnectionBase._check_closedc C � | j std��d 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 d u S r �r8 rB r r r �closed� s z_ConnectionBase.closedc C � | j S r )r9 rB r r r r= � � z_ConnectionBase.readablec C rN r )r: rB r r r r> � rO z_ConnectionBase.writablec C s | � � | jS r )rE r8 rB r r r �fileno� s z_ConnectionBase.filenoc C s. | j d urz | �� W d | _ d S d | _ w d S r r@ rB r r r rJ � s �z_ConnectionBase.closer c C s� | � � | �� t|�}|jdkrtt|��}t|�}|dk r$td��||k r,td��|d u r5|| }n|dk r=td��|| |krGtd��| �|||| � � d S )Nr r zoffset is negativezbuffer length < offsetzsize is negativez buffer length < offset + size)rE rI � 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 r )rE rI rU �_ForkingPickler�dumps�r; �objr r r �send� s z_ConnectionBase.sendc C sL | � � | �� |d ur|dk rtd��| �|�}|d u r"| �� |�� S )Nr znegative maxlength)rE rG r& � _recv_bytesrK �getvalue)r; Z maxlengthrV r r r � recv_bytes� s z_ConnectionBase.recv_bytesc C s� | � � | �� t|��K}|j}|t|� }|dk r td��||kr'td��| �� }|�� }||| k r;t|� � ��|� d� |� ||| || | � � |W d � S 1 s[w Y d S )Nr znegative offsetzoffset too large) rE rG rQ rR rT r& ra �tellr rb �seek�readinto)r; rV rW rY rR Zbytesize�resultrX r r r �recv_bytes_into� s( �$�z _ConnectionBase.recv_bytes_intoc C s( | � � | �� | �� }t�|�� �S r )rE rG ra r\ �loads� getbuffer)r; rV r r r �recv� s z_ConnectionBase.recv� c C s | � � | �� | �|�S r )rE rG �_poll�r; r r r r �poll� s z_ConnectionBase.pollc C � | S r r rB r r r � __enter__ � z_ConnectionBase.__enter__c C � | � � d S r �rJ �r; �exc_type� exc_valueZexc_tbr r r �__exit__ r z_ConnectionBase.__exit__)TT)r Nr )r )rl )�__name__� __module__� __qualname__r8 r? rC rE rG rI rK �propertyrM r= r> rP rJ r[ r` rc rh rk ro rq rx r r r r r6 r s0 r6 c @ sB e Zd Z dZejfdd�Zdd� Zd dd�Zd d � Z d d � Z dS )�PipeConnectionFc C � || j � d S r rL )r; Z _CloseHandler r r rA � zPipeConnection._closec C sn t j| j|dd�\}}z#z|t jkrt �|jgdt�}W n |�� � W |�d�\}}d S |�d�\}}w )NT�� overlappedF) �_winapiZ WriteFiler8 �ERROR_IO_PENDING�WaitForMultipleObjects�eventr �cancel�GetOverlappedResult)r; rV �ov�err�waitresZnwrittenr r r rU s ���"zPipeConnection._send_bytesNc C s\ | j r d| _ t�� S |d u rdnt|d�}z�tj| j|dd�\}}z@z|tjkr1t�|j gdt �}W n |� � � W |� d�\}}|dkrUt�� }|� |�� � |W S |tjkra| �||�W S n2|� d�\}}|dkr�t�� }|� |�� � | Y W S |tjkr�| �||� Y W S w W td�� ty� } z |jtjkr�t�� d }~ww )NF� Tr� r z.shouldn't get here; expected KeyboardInterrupt)�_got_empty_message�io�BytesIO�minr� �ReadFiler8 r� r� r� r r� r� �writerj ZERROR_MORE_DATA�_get_more_datarD �winerror�ERROR_BROKEN_PIPE�EOFError� RuntimeError) r; �maxsizeZbsizer� r� r� Znread�f�er r r ra & sR � ��� �� �� ��zPipeConnection._recv_bytesc C s. | j s t�| j�d dkrdS tt| g|��S )Nr T)r� r� � PeekNamedPiper8 �boolr rn r r r rm F s zPipeConnection._pollc C s� |� � }t�� }|�|� t�| j�d }|d ur%t|�| |kr%| �� tj | j|dd�\}}|� d�\}}|�|� � � |S )Nr 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{ r� r� � CloseHandlerA rU ra rm r� r r r r r} s r} c @ sz e Zd Z erejfdd�ZejZej Z n e j fdd�Ze j Ze jZ efdd�Ze fdd�Zdd � Zdd d �Zd d� Zd S )� Connectionc 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 t�td��|�|� ||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%t�dd�}t�d|�}| �|� | �|� | �|� d S t�d|�}|dkr;| �|� | �|� d S | �|| � d S )Ni����!i����!Qi @ )rT �structZpackr� )r; rV rZ Z pre_header�headerr r r rU � s zConnection._send_bytesNc C s^ | � d�}t�d|�� �\}|dkr | � d�}t�d|�� �\}|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� �_multiprocessingZ closesocketrA r` Z_writerk Z_readr" rJ r� r� r� r� rU ra rm r r r r r� \ s r� c @ sP e Zd Z ddd�Zdd� Zdd� Zed d � �Zed d � �Zd d� Z dd� Z dS )r Nr c C sp |p |rt |�p t}|pt|�}t|� |dkr t||�| _nt|||�| _|d ur3t|t�s3t d��|| _ d S �Nr z authkey should be a byte string) r5 �default_familyr) r. � PipeListener� _listener�SocketListener� isinstancerS � TypeError�_authkey)r; r4 r( �backlog�authkeyr r r r? � s � zListener.__init__c C s@ | j d u r td��| j �� }| jr t|| j� t|| j� |S )Nzlistener is closed)r� rD �acceptr� �deliver_challenge�answer_challenge)r; �cr r r r� � s zListener.acceptc C s& | j }|d urd | _ |�� d S d S r )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� rJ r| r4 r� rq rx r r r r r � s r c C sj |pt | �}t|� |dkrt| �}nt| �}|d ur%t|t�s%td��|d ur3t||� t||� |S r� ) r5 r. � PipeClient� SocketClientr� rS r� r� r� )r4 r( r� r� r r r r � s r Tc C sr | r#t �� \}}|�d� |�d� t|�� �}t|�� �}||fS t�� \}}t|dd�}t|dd�}||fS )NTF�r> �r= )r- Z socketpair� setblockingr� �detachr" �pipe)�duplex�s1�s2�c1�c2Zfd1Zfd2r 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 d d � tj|dd�}|�d�\} } t|| d�} t|| d�} | | fS )Nr r r Tr� r� r� )r) r� �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�ConnectNamedPiper� r} ) r� r4 Zopenmode�accessZobsizeZibsizeZh1Zh2r� �_r� r� r� r r r r s> � � � � � c @ s( e Zd Z d dd�Zdd� Zdd� ZdS ) r� r c C s� t � tt |��| _z)tjdkr| j�t jt jd� | j�d� | j� |� | j� |� | j� � | _ W n t y? | j�� � w || _d | _|dkr]t�|�s]tj| tj|fdd�| _d S d | _d S )N�posixr Tr r ��argsZ exitpriority)r- �getattr�_socketr" �nameZ setsockoptZ SOL_SOCKETZ SO_REUSEADDRr� ZbindZlistenZ getsocknamer� rD rJ Z_familyr� r r3 �Finalize�unlink�_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� rJ r r r r r� C s r� c C s^ t | �}t�tt|���}|�d� |�| � t|�� �W d � S 1 s(w Y d S r� )r5 r- r� r� Zconnectr� r� )r4 r( r� r r r r� o s $�r� c @ s6 e Zd Z d 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� � _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�\}}t|�S |�d�\}}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� � s0 ��� � ��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� � 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 fvs5t |�r6� W Y d }~nd }~ww nqt�|tjd d � t|�S )Nr �� r )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� � s4 ��� ���� �r� � s #CHALLENGE#s #WELCOME#s #FAILURE#c C s� dd l }t|t�std�t|����t�t�}| � 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" �urandom�MESSAGE_LENGTHr[ � CHALLENGE�new�digestrc �WELCOME�FAILUREr �Z connectionr� r� �messager Zresponser r r r� � s � r� c C sx dd l }t|t�std�t|����| �d�}|tt�d � }|� ||d�� � }| � |� | �d�}|t kr:t d��d S )Nr r� r� r� 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 d��d�S )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|tt�S �Nr )� xmlrpc.client�clientr 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 |��tt�S 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 �L�readyr� r r r �_exhaustive_wait% s$ � �� r c s� |d u 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 yj } zd |j }}|t vr`� W Y d }~nd }~ww |tj kr{|� |� |�|j< q(|r�t�� d d� dkr�z |�d�\} }W n t y� } z |j }W Y d }~nd }~ww |s�t|d �r�d|_� �|� d}q(t��� |�}W |D ]}|�� q�|D ]D}z |�d�\} }W n t y� } z|j }|t vr� W Y d }~nd }~ww |tjk�r�|j }� �|� |dk�rt|d ��rd|_q�nU|D ]}|�� �q|D ]G}z |�d�\} }W n t �yD } z|j }|t v�r:� W Y d }~nd }~ww |tjk�rc�|j }� �|� |dk�rct|d ��rcd|_�q w � ��fd d �|D �� � fd d �| D �S )Nr r� g �?rP T� )� r Fr� c 3 s � | ]}� | V qd S r r )�.0r� )�waithandle_to_objr r � <genexpr>� s � zwait.<locals>.<genexpr>c s g | ]}|� v r|�qS r r )r! �o)� ready_objectsr r � <listcomp>� s �wait.<locals>.<listcomp>)r �intr �setr� �AttributeErrorr7 r� r� rD r� � _ready_errorsr� r� r� r* Zgetwindowsversionr� r, r� �addr �keysr� ZERROR_OPERATION_ABORTED�update) � 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|d urt�� | } |�|�}|r1dd� |D �W d � S |d urH|t�� }|dk rH|W d � S q 1 sLw Y d S )NTc S s g | ]\}}|j �qS r )Zfileobj)r! �keyZeventsr r r r& � s r' r )� _WaitSelector�register� selectorsZ EVENT_READr r Zselect)r/ 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 r5 Z DupSocket�rebuild_connectionr= r> )r r< r� r5 �dsr r r �reduce_connection� s $�r8 c C s | � � }t|� � ||�S r �r� r� )r7 r= r> Zsockr r r r6 � s r6 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 r6 r= r> )r �dfr r r r8 � s c C r= r r9 )r? r= r> �fdr r r r6 � 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_TIMEOUT�countr% 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 r4 r0 r2 ZSelectSelectorr8 r6 r3 r< r: r r r r �<module> s� � PT = ,, 8 P
Save File
Cancel