CasperSecurity
o
`�wi,O � @ sx d Z ddlZddlZddlZddlZddlZddlZddlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZ ejdedd� ee e
eeeeh�Zze W n eyY i ZY nw dd� ZG d d
� d
e �Z e e e fZ d d � Z!d
d� Z"dd� Z#dd� Z$d)dd�Z%d)dd�Z&e&Z'd*dd�Z(G d d � d �Z)G d d � d e)�Z*d d!� Z+d+d"d#�Z,ej-d$kr�G d%d&� d&�Z.G d'd(� d(e)�Z/dS dS ),a� Basic infrastructure for asynchronous socket service clients and servers.
There are only two ways to have a program on a single processor do "more
than one thing at a time". Multi-threaded programming is the simplest and
most popular way to do it, but there is another very different technique,
that lets you have nearly all the advantages of multi-threading, without
actually using multiple threads. it's really only practical if your program
is largely I/O bound. If your program is CPU bound, then pre-emptive
scheduled threads are probably what you really need. Network servers are
rarely CPU-bound, however.
If your operating system supports the select() system call in its I/O
library (and nearly all do), then you can use it to juggle multiple
communication channels at once; doing other work while your I/O is taking
place in the "background." Although this strategy can seem strange and
complex, especially at first, it is in many ways easier to understand and
control than multi-threaded programming. The module documented here solves
many of the difficult problems for you, making the task of building
sophisticated high-performance network servers and clients a snap.
� N)
�EALREADY� EINPROGRESS� EWOULDBLOCK�
ECONNRESET�EINVAL�ENOTCONN� ESHUTDOWN�EISCONN�EBADF� ECONNABORTED�EPIPE�EAGAIN� errorcodezlThe asyncore module is deprecated and will be removed in Python 3.12. The recommended replacement is asyncio� )�
stacklevelc
C sB zt �| �W S tttfy | tv rt| Y S d| Y S w )NzUnknown error %s)�os�strerror�
ValueError�
OverflowError� NameErrorr )�err� r � /usr/lib/python3.10/asyncore.py� _strerrorK s �r c @ s e Zd ZdS )�ExitNowN)�__name__�
__module__� __qualname__r r r r r S s r c C �2 z| � � W d S ty � | �� Y d S �N)�handle_read_event�_reraised_exceptions� handle_error��objr r r �readX � r% c C r r )�handle_write_eventr! r" r# r r r �write` r& r( c C r r )�handle_expt_eventr! r" r# r r r �
_exceptionh r& r* c
C s� z0|t j@ r
| �� |t j@ r| �� |t j@ r | �� |t jt jB t j B @ r.| �
� W d S W d S t yX } z |j t
vrB| �� n | �
� W Y d }~d S W Y d }~d S d }~w ty_ � | �� Y d S r )�select�POLLINr �POLLOUTr' �POLLPRIr) �POLLHUP�POLLERR�POLLNVAL� handle_close�OSError�errno�
_DISCONNECTEDr" r! )r$ �flags�er r r � readwritep s(
�
�� r8 � c C s6 |d u rt }|r�g }g }g }t|�� �D ]&\}}|�� }|�� }|r'|�|� |r1|js1|�|� |s5|r:|�|� qg | krJ| krJ|krSn nt�| � d S t � |||| �\}}}|D ]}|�
|�}|d u rlq`t |� q`|D ]}|�
|�}|d u rqst |� qs|D ]}|�
|�}|d u r�q�t
|� q�d S d S r )�
socket_map�list�items�readable�writable�append� accepting�time�sleepr+ �getr% r( r* ) �timeout�map�r�wr7 �fdr$ �is_r�is_wr r r �poll� sF
�"
� rK c C s� |d u rt }| d urt| d �} t�� }|r^t|�� �D ]&\}}d}|�� r.|tjtjB O }|� � r:|j
s:|tj O }|rB|� ||� q |�| �}|D ]\}}|�
|�}|d u rXqJt||� qJd S d S )Ni� r )r: �intr+ rK r; r< r= r, r. r> r@ r- �registerrC r8 )rD rE �pollsterrH r$ r6 rF r r r �poll2� s.
�
� rO � >@Fc C s� |d u rt }|rttd�rt}nt}|d u r#|r!|| |� |sd S d S |r:|dkr>|| |� |d }|r<|dks)d S d S d S d S )NrK r � )r: �hasattrr+ rO rK )rD �use_pollrE �count�poll_funr r r �loop� s
�
�rV c @ s2 e Zd ZdZdZdZdZdZdZe dh�Z
dAdd�Z dd� Z dBdd �Z
dBd
d �Zejejfd d
�ZdBdd�Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Z d"d#� Z d$d%� Z dCd'd(�Z d)d*� Z d+d,� Z!d-d.� Z"d/d0� Z#d1d2� Z$d3d4� Z%d5d6� Z&d7d8� Z'd9d:� Z(d;d<� Z)d=d>� Z*d?d@� Z+dS )D�
dispatcherFN�warningc
C s� |d u rt | _n|| _d | _|rK|�d� | �||� d| _z|�� | _W d S tyJ } z|j t
t fv r9d| _n| � |� � W Y d }~d S d }~ww d | _
d S )NFT)r: �_map�_fileno� setblocking�
set_socket� connected� getpeername�addrr3 r4 r r � del_channel�socket)�self�sockrE r r r r �__init__� s&
���
zdispatcher.__init__c C s� | j jd | j j g}| jr| jr|�d� n| jr |�d� | jd ur@z
|�d| j � W n ty? |�t| j�� Y nw dd� |�t
| �f S )N�.� listeningr] z%s:%dz <%s at %#x>� ) � __class__r r r@ r_ r? r] � TypeError�repr�join�id)rb �statusr r r �__repr__ s
�zdispatcher.__repr__c C s |d u r| j }| || j< d S r )rY rZ )rb rE r r r � add_channel s zdispatcher.add_channelc C s, | j }|d u r
| j}||v r||= d | _ d S r )rZ rY )rb rE rH r r r r` s
zdispatcher.del_channelc C s. ||f| _ t�||�}|�d� | �|� d S �NF)�family_and_typera r[ r\ )rb �family�typerc r r r �
create_socket s
zdispatcher.create_socketc C s || _ |�� | _| �|� d S r )ra �filenorZ ro �rb rc rE r r r r\ % s
zdispatcher.set_socketc C sB z| j �t jt j| j �t jt j�dB � W d S ty Y d S w )NrQ )ra �
setsockopt�
SOL_SOCKET� SO_REUSEADDR�
getsockoptr3 �rb r r r �set_reuse_addr* s
��
� �zdispatcher.set_reuse_addrc C � dS �NTr r{ r r r r= ; � zdispatcher.readablec C r} r~ r r{ r r r r> > r zdispatcher.writablec C s( d| _ tjdkr|dkrd}| j�|�S )NT�nt� )r@ r �namera �listen)rb �numr r r r� E s zdispatcher.listenc C s || _ | j�|�S r )r_ ra �bind)rb r_ r r r r� K s zdispatcher.bindc C sn d| _ d| _| j�|�}|tttfv s |tkr!tj dkr!|| _
d S |dt fv r0|| _
| � � d S t
|t| ��)NFTr� r )r] �
connectingra �
connect_exr r r r r r� r_ r �handle_connect_eventr3 r )rb �addressr r r r �connectO s zdispatcher.connectc
C sd z | j �� \}}W ||fS ty Y d S ty1 } z|jtttfv r,W Y d }~d S � d }~ww r )ra �acceptri r3 r4 r r r
)rb �connr_ �whyr r r r� ] s ���zdispatcher.acceptc
C sf z | j �|�}|W S ty2 } z |jtkr W Y d }~dS |jtv r-| �� W Y d }~dS � d }~ww �Nr )ra �sendr3 r4 r r5 r2 )rb �data�resultr� r r r r� k s
��zdispatcher.sendc
C s` z| j �|�}|s| �� W dS |W S ty/ } z|jtv r*| �� W Y d }~dS � d }~ww �N� )ra �recvr2 r3 r4 r5 )rb � buffer_sizer� r� r r r r� x s
��zdispatcher.recvc
C sp d| _ d| _d| _| �� | jd ur6z| j�� W d S ty5 } z|jtt fvr*� W Y d }~d S d }~ww d S rp )
r] r@ r� r` ra �closer3 r4 r r
)rb r� r r r r� � s
����zdispatcher.closec C s t j�dt|� � d S )Nzlog: %s
)�sys�stderrr( �str)rb �messager r r �log� s zdispatcher.log�infoc C s" || j vrtd||f � d S d S )Nz%s: %s)�ignore_log_types�print)rb r� rs r r r �log_info� s
�zdispatcher.log_infoc C s> | j r | �� d S | js| jr| �� | �� d S | �� d S r )r@ �
handle_acceptr] r� r� � handle_readr{ r r r r � s z dispatcher.handle_read_eventc C s@ | j �t jt j�}|dkrt|t|���| �� d| _d| _d S )Nr TF) ra rz rx �SO_ERRORr3 r �handle_connectr] r� �rb r r r r r� � s
z dispatcher.handle_connect_eventc C s* | j rd S | js| jr| �� | �� d S r )r@ r] r� r� � handle_writer{ r r r r' � s z dispatcher.handle_write_eventc C s2 | j �t jt j�}|dkr| �� d S | �� d S r� )ra rz rx r� r2 � handle_exptr� r r r r) � s z dispatcher.handle_expt_eventc C sT t � \}}}}zt| �}W n
dt| � }Y | �d||||f d� | �� d S )Nz)<__repr__(self) failed for object at %0x>z:uncaptured python exception, closing channel %s (%s:%s %s)�error)�compact_tracebackrj rl r� r2 )rb �nil�t�v�tbinfo� self_reprr r r r" � s �� zdispatcher.handle_errorc C � | � dd� d S )Nz!unhandled incoming priority eventrX �r� r{ r r r r� � � zdispatcher.handle_exptc C r� )Nzunhandled read eventrX r� r{ r r r r� � r� zdispatcher.handle_readc C r� )Nzunhandled write eventrX r� r{ r r r r� � r� zdispatcher.handle_writec C r� )Nzunhandled connect eventrX r� r{ r r r r� � r� zdispatcher.handle_connectc C s" | � � }|d ur| j|� d S d S r )r� �handle_accepted)rb �pairr r r r� � s �zdispatcher.handle_acceptc C s |� � | �dd� d S )Nzunhandled accepted eventrX )r� r� )rb rc r_ r r r r� � s zdispatcher.handle_acceptedc C s | � dd� | �� d S )Nzunhandled close eventrX )r� r� r{ r r r r2 � s zdispatcher.handle_close�NNr )r� ),r r r �debugr] r@ r� �closingr_ � frozensetr� rd rn ro r` ra �AF_INET� SOCK_STREAMrt r\ r| r= r> r� r� r� r� r� r� r� r� r� r r� r' r) r" r� r� r� r� r� r� r2 r r r r rW � sL
rW c @ s6 e Zd Zd dd�Zdd� Zdd� Zdd � Zd
d � ZdS )
�dispatcher_with_sendNc C s t �| ||� d| _d S r� )rW rd �
out_bufferrv r r r rd s
z dispatcher_with_send.__init__c C s. d}t �| | jd d� �}| j|d � | _d S )Nr i )rW r� r� )rb �num_sentr r r �
initiate_send s z"dispatcher_with_send.initiate_sendc C s | � � d S r )r� r{ r r r r� s z!dispatcher_with_send.handle_writec C s | j pt| j�S r )r] �lenr� r{ r r r r> s z dispatcher_with_send.writablec C s0 | j r | �dt|� � | j| | _| �� d S )Nz
sending %s)r� r� rj r� r� )rb r� r r r r� s zdispatcher_with_send.sendr� )r r r rd r� r� r> r� r r r r r� s
r� c C s� t �� \} }}g }|std��|r'|�|jjj|jjjt|j �f� |j
}|s~|d \}}}d� dd� |D ��}|||f| ||fS )Nztraceback does not exist���rg c S s g | ]}d | �qS )z
[%s|%s|%s]r )�.0�xr r r �
<listcomp>- s z%compact_traceback.<locals>.<listcomp>) r� �exc_info�AssertionErrorr? �tb_frame�f_code� co_filename�co_namer� � tb_lineno�tb_nextrk )r� r� �tbr� �file�function�liner� r r r r� s �� r� c C s� | d u rt } t| �� �D ]1}z|�� W q ty/ } z|jtkr"n|s%� W Y d }~q d }~w ty6 � |s<� Y q | �� d S r ) r: r; �valuesr� r3 r4 r
r! �clear)rE �
ignore_allr� r r r � close_all0 s$
� � r� �posixc @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zdd
d �ZeZeZ d d
� Z
dd� Z d S )� file_wrapperc C s t �|�| _d S r )r �duprH �rb rH r r r rd U r� zfile_wrapper.__init__c C s* | j dkrtjd| t| d� | �� d S )Nr zunclosed file %r)�source)rH �warnings�warn�ResourceWarningr� r{ r r r �__del__X s
� zfile_wrapper.__del__c G � t j| jg|�R � S r )r r% rH �rb �argsr r r r� ^ � zfile_wrapper.recvc G r� r )r r( rH r� r r r r� a r� zfile_wrapper.sendNc C s$ |t jkr|t jkr|sdS td��)Nr z-Only asyncore specific behaviour implemented.)ra rx r� �NotImplementedError)rb �level�optname�buflenr r r rz d s
�zfile_wrapper.getsockoptc C s( | j dk rd S | j }d| _ t�|� d S )Nr r� )rH r r� r� r r r r� o s
zfile_wrapper.closec C s | j S r )rH r{ r r r ru v s zfile_wrapper.filenor ) r r r rd r� r� r� rz r% r( r� ru r r r r r� P s
r� c @ s e Zd Zddd�Zdd� ZdS )�file_dispatcherNc C sN t �| d |� d| _z|�� }W n ty Y nw | �|� t�|d� d S )NTF)rW rd r] ru �AttributeError�set_filer � set_blocking)rb rH rE r r r rd { s �
zfile_dispatcher.__init__c C s" t |�| _| j�� | _| �� d S r )r� ra ru rZ ro r� r r r r� � s
zfile_dispatcher.set_filer )r r r rd r� r r r r r� y s
r� )r9 N)rP FNNrp )0�__doc__r+ ra r� rA r� r r4 r r r r r r r r r
r r r
r r� �DeprecationWarningr� r5 r: r r � Exceptionr �KeyboardInterrupt�
SystemExitr! r% r( r* r8 rK rO �poll3rV rW r� r� r� r� r� r� r r r r �<module> sT <� � �
'
*
)�