a �DOg9,�@s@ddlZddlmZGdd�dej�ZGdd�d�Zdd�ZdS) �N)�dequec@s�eZdZdZdZdZdZd'dd�Zdd�Zd d �Z d d �Z d d�Z dd�Z dd�Z dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�ZdS)(� async_chatirzlatin-1NcCs(d|_g|_t�|_tj�|||�dS�N�)� ac_in_buffer�incomingr� producer_fifo�asyncore� dispatcher�__init__)�self�sock�map�r� /usr/lib64/python3.9/asynchat.pyr Cszasync_chat.__init__cCs td��dS�Nzmust be implemented in subclass��NotImplementedError�r �datarrr�collect_incoming_dataQsz async_chat.collect_incoming_datacCs|j�|�dS�N)r�appendrrrr�_collect_incoming_dataTsz!async_chat._collect_incoming_datacCsd�|j�}|jdd�=|Sr)�joinr)r �drrr� _get_dataWs  zasync_chat._get_datacCs td��dSrr�r rrr�found_terminator\szasync_chat.found_terminatorcCsBt|t�r|jrt||j�}nt|t�r8|dkr8td��||_dS)Nrz-the number of received bytes must be positive)� isinstance�str� use_encoding�bytes�encoding�int� ValueError� terminator)r Ztermrrr�set_terminator_s zasync_chat.set_terminatorcCs|jSr)r&rrrr�get_terminatorjszasync_chat.get_terminatorcCs�z|�|j�}Wn.ty$YdSty>|��YdS0t|t�r\|jr\tt|j �}|j ||_ |j �r�t |j �}|� �}|s�|� |j �d|_ qht|t��r|}||kr�|� |j �d|_ |j||_n2|� |j d|��|j |d�|_ d|_|��qht |�}|j �|�}|dk�r`|dk�rB|� |j d|��|j ||d�|_ |��qht|j |�}|�r�||k�r�|� |j d| ��|j | d�|_ �q�qh|� |j �d|_ qhdS)Nrr�����)�recv�ac_in_buffer_size�BlockingIOError�OSError� handle_errorrr r!r"r#r�lenr(rr$r&r�find�find_prefix_at_end)r rZlbr&�nZterminator_len�indexrrr� handle_readrsR                zasync_chat.handle_readcCs |��dSr)� initiate_sendrrrr� handle_write�szasync_chat.handle_writecCs |��dSr)�closerrrr� handle_close�szasync_chat.handle_closecCsxt|tttf�stdt|���|j}t|�|kr`tdt|�|�D]}|j � ||||��q@n |j � |�|� �dS)Nz#data argument must be byte-ish (%r)r) rr"� bytearray� memoryview� TypeError�type�ac_out_buffer_sizer/�rangerrr5)r rZsabs�irrr�push�s�  zasync_chat.pushcCs|j�|�|��dSr)rrr5)r Zproducerrrr�push_with_producer�s zasync_chat.push_with_producercCsdS�N�rrrrr�readable�szasync_chat.readablecCs|jp |j Sr)r� connectedrrrr�writable�szasync_chat.writablecCs|j�d�dSr)rrrrrr�close_when_done�szasync_chat.close_when_donecCs|j�r|j�r|jd}|s:|jd=|dur:|��dS|j}z|d|�}Wn8ty�|��}|rx|j�|�n|jd=YqYn0t|t�r�|j r�t ||j �}z|� |�}Wnt y�|��YdS0|�r|t|�ks�|t|�k�r||d�|jd<n|jd=dSdS)Nr)rrEr8r=r;�more� appendleftrr r!r"r#�sendr-r.r/)r �firstZobsrZnum_sentrrrr5�s8     zasync_chat.initiate_sendcCs d|_|jdd�=|j��dSr)rrr�clearrrrr�discard_bufferss zasync_chat.discard_buffers)NN)�__name__� __module__� __qualname__r+r=r!r#r rrrrr'r(r4r6r8r@rArDrFrGr5rMrrrrr4s*  H (rc@seZdZddd�Zdd�ZdS)�simple_producer�cCs||_||_dSr)r� buffer_size)r rrSrrrr szsimple_producer.__init__cCsJt|j�|jkr6|jd|j�}|j|jd�|_|S|j}d|_|SdSr)r/rrS)r �resultrrrrHszsimple_producer.moreN)rR)rNrOrPr rHrrrrrQs rQcCs0t|�d}|r,|�|d|��s,|d8}q |SrB)r/�endswith)ZhaystackZneedle�lrrrr1/s  r1)r � collectionsrr rrQr1rrrr�0s  \