U
e5d@, � @ s@ d dl Z d dlmZ G dd� de j�ZG dd� d�Zdd� ZdS ) � N)�dequec @ s� e Zd ZdZdZdZdZd'dd�Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� ZdS )(�
async_chati r zlatin-1Nc C s( d| _ g | _t� | _tj�| ||� d S �N� )� ac_in_buffer�incomingr �
producer_fifo�asyncore�
dispatcher�__init__)�selfZsock�map� r � /usr/lib64/python3.8/asynchat.pyr C s zasync_chat.__init__c C s t d��d S �Nz must be implemented in subclass��NotImplementedError�r �datar r r �collect_incoming_dataQ s z async_chat.collect_incoming_datac C s | j �|� d S �N)r �appendr r r r �_collect_incoming_dataT s z!async_chat._collect_incoming_datac C s d� | j�}| jd d �= |S r )�joinr )r �dr r r � _get_dataW s zasync_chat._get_datac C s t d��d S r r �r r r r �found_terminator\ s zasync_chat.found_terminatorc C sB t |t�r | jr t|| j�}nt |t�r8|dk r8td��|| _d S )Nr z-the number of received bytes must be positive)�
isinstance�str� use_encoding�bytes�encoding�int�
ValueError�
terminator)r Ztermr r r �set_terminator_ s
zasync_chat.set_terminatorc C s | j S r )r% r r r r �get_terminatorj s zasync_chat.get_terminatorc
C s� z| � | j�}W nD tk
r& Y d S tk
rT } z| �� W Y �d S d }~X Y nX t|t�rr| jrrtt| j �}| j
| | _
| j
�r�t | j
�}| � � }|s�| �
| j
� d| _
q~t|t��r |}||k r�| �
| j
� d| _
| j| | _n2| �
| j
d |� � | j
|d � | _
d| _| �� q~t |�}| j
�|�}|dk�rv|dk�rX| �
| j
d |� � | j
|| d � | _
| �� q~t| j
|�}|�r�||k�r�| �
| j
d | � � | j
| d � | _
�q�q~| �
| j
� d| _
q~d S )Nr r �����)Zrecv�ac_in_buffer_size�BlockingIOError�OSError� handle_errorr r r r! r" r �lenr' r r# r% r �find�find_prefix_at_end)r r ZwhyZlbr% �nZterminator_len�indexr r r � handle_readr sR
zasync_chat.handle_readc C s | � � d S r )�
initiate_sendr r r r � handle_write� s zasync_chat.handle_writec C s | � � d S r )�closer r r r � handle_close� s zasync_chat.handle_closec C sx t |tttf�s tdt|���| j}t|�|kr`tdt|�|�D ] }| j �
|||| � � q@n | j �
|� | � � d S )Nz#data argument must be byte-ish (%r)r ) r r! � bytearray�
memoryview� TypeError�type�ac_out_buffer_sizer- �ranger r r3 )r r Zsabs�ir r r �push� s � zasync_chat.pushc C s | j �|� | �� d S r )r r r3 )r Zproducerr r r �push_with_producer� s z async_chat.push_with_producerc C s dS �N� r r r r r �readable� s zasync_chat.readablec C s | j p | j S r )r � connectedr r r r �writable� s zasync_chat.writablec C s | j �d � d S r )r r r r r r �close_when_done� s zasync_chat.close_when_donec C s | j �r| j�r| j d }|s:| j d= |d kr:| �� d S | j}z|d |� }W n: tk
r� |�� }|rz| j �|� n| j d= Y q Y nX t|t�r�| j r�t
|| j �}z| � |�}W n t
k
r� | �� Y d S X |�r|t|�k s�|t|�k �r
||d � | j d< n| j d= d S d S )Nr )r rC r6 r; r9 �more�
appendleftr r r r! r" �sendr+ r, r- )r �firstZobsr Znum_sentr r r r3 � s8
zasync_chat.initiate_sendc C s d| _ | jd d �= | j�� d S r )r r r �clearr r r r �discard_buffers s zasync_chat.discard_buffers)NN)�__name__�
__module__� __qualname__r) r; r r" r r r r r r& r' r2 r4 r6 r> r? rB rD rE r3 rK r r r r r 4 s*
H (r c @ s e Zd Zddd�Zdd� ZdS )�simple_producer� c C s || _ || _d S r )r � buffer_size)r r rQ r r r r s zsimple_producer.__init__c C sJ t | j�| jkr6| jd | j� }| j| jd � | _|S | j}d| _|S d S r )r- r rQ )r �resultr r r rF s zsimple_producer.moreN)rP )rL rM rN r rF r r r r rO s
rO c C s0 t |�d }|r,| �|d |� �s,|d8 }q |S r@ )r- �endswith)ZhaystackZneedle�lr r r r/ / s
r/ )r � collectionsr r
r rO r/ r r r r �0 s
\