a
iN @ sf d Z ddlZddlZddlZddlZddlZddlZddlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZ ee e
eeeehZze W n ey i ZY n0 dd ZG dd deZee e fZ dd Z d
d Z d d
Z!dd Z"d&ddZ#d'ddZ$e$Z%d(ddZ&G dd dZ'G dd d e'Z(d d Z)d)d d Z*ej+d!krbG d"d# d#Z,G d$d% d%e'Z-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
ECONNRESETEINVALENOTCONN ESHUTDOWNEISCONNEBADF ECONNABORTEDEPIPEEAGAIN errorcodec
C sF z t | W S tttfy@ | tv r4t| Y S d| Y S 0 d S )NzUnknown error %s)osstrerror
ValueError
OverflowError NameErrorr )err r /usr/lib64/python3.9/asyncore.py _strerrorD s r c @ s e Zd ZdS )ExitNowN)__name__
__module__ __qualname__r r r r r L s r c C s8 z | W n& ty Y n | Y n0 d S N)handle_read_event_reraised_exceptions handle_errorobjr r r readQ s r" c C s8 z | W n& ty Y n | Y n0 d S r )handle_write_eventr r r r r r writeY s r$ c C s8 z | W n& ty Y n | Y n0 d S r )handle_expt_eventr r r r r r
_exceptiona s r& c
C s zX|t j@ r| |t j@ r&| |t j@ r8| |t jt jB t j B @ rV|
W nh t y } z,|j d t
vr| n|
W Y d }~n.d }~0 ty Y n | Y n0 d S Nr )selectPOLLINr POLLOUTr# POLLPRIr% POLLHUPPOLLERRPOLLNVAL handle_closeOSErrorargs
_DISCONNECTEDr r )r! flagser r r readwritei s"
r5 c C s< |d u r t }|r8g }g }g }t| D ]L\}}| }| }|rP|| |rd|jsd|| |sl|r*|| q*g | kr| kr|krn nt| d S t |||| \}}}|D ] }|
|}|d u rqt | q|D ]"}|
|}|d u rqt | q|D ]&}|
|}|d u r,qt
| qd S r )
socket_maplistitemsreadablewritableappend acceptingtimesleepr( getr" r$ r& ) timeoutmaprwr4 fdr! Zis_rZis_wr r r poll} s@
"
rF c C s |d u r t }| d ur t| d } t }|rt| D ]L\}}d}| r\|tjtjB O }| rt|j
st|tj O }|r8| || q8|| }|D ]&\}}|
|}|d u rqt|| qd S )Ni r )r7 intr( rF r8 r9 r: r) r+ r; r= r* registerr@ r5 )rA rB ZpollsterrE r! r3 rC r r r poll2 s(
rI >@Fc C sb |d u r t }|r ttdr t}nt}|d u r>|r^|| | q,n |r^|dkr^|| | |d }q>d S )NrF r )r7 hasattrr( rI rF )rA Zuse_pollrB countZpoll_funr r r loop s
rN c @ s2 e Zd ZdZdZdZdZdZdZe dhZ
dAddZ dd Z dBdd Z
dCd
d Zejejfd d
ZdDddZdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Z d"d# Z d$d% Z dEd'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 )F
dispatcherFNwarningc
C s |d u rt | _n|| _d | _|r|d | || d| _z| | _W q ty } z2|j d t
t fv rtd| _n | | W Y d }~qd }~0 0 nd | _
d S )NFTr )r7 _map_fileno setblocking
set_socket connected getpeernameaddrr0 r1 r r del_channelsocket)selfsockrB r r r r __init__ s
zdispatcher.__init__c C s | j jd | j j g}| jr.| jr.|d n| jr>|d | jd urz|d| j W n" ty~ |t| j Y n0 dd |t
| f S )N.Z listeningrU z%s:%dz <%s at %#x> ) __class__r r r= rW r< rU TypeErrorreprjoinid)rZ statusr r r __repr__ s
zdispatcher.__repr__c C s |d u r| j }| || j< d S r )rQ rR )rZ rB r r r add_channel s zdispatcher.add_channelc C s, | j }|d u r| j}||v r"||= d | _ d S r )rR rQ )rZ rB rE r r r rX s zdispatcher.del_channelc C s. ||f| _ t||}|d | | d S )NF)Zfamily_and_typerY rS rT )rZ familytyper[ r r r
create_socket s
zdispatcher.create_socketc C s || _ | | _| | d S r )rY filenorR rf rZ r[ rB r r r rT s
zdispatcher.set_socketc C sB z*| j t jt j| j t jt jdB W n ty< Y n0 d S )NrK )rY
setsockopt
SOL_SOCKET SO_REUSEADDR
getsockoptr0 rZ r r r set_reuse_addr# s
zdispatcher.set_reuse_addrc C s dS NTr rp r r r r: 4 s zdispatcher.readablec C s dS rr r rp r r r r; 7 s zdispatcher.writablec C s( d| _ tjdkr |dkr d}| j|S )NTnt )r= r namerY listen)rZ numr r r rv > s zdispatcher.listenc C s || _ | j|S r )rW rY bind)rZ rW r r r rx D s zdispatcher.bindc C sp d| _ d| _| j|}|tttfv s8|tkrBtj dkrB|| _
d S |dt fv r^|| _
| nt
|t| d S )NFTrs r )rU
connectingrY
connect_exr r r r r ru rW r handle_connect_eventr0 r )rZ addressr r r r connectH s
zdispatcher.connectc
C sz z| j \}}W nZ ty& Y d S tyl } z0|jd tttfv rVW Y d }~d S W Y d }~nd }~0 0 ||fS d S r' )rY acceptr` r0 r1 r r r
)rZ connrW whyr r r r~ V s zdispatcher.acceptc
C s~ z| j |}|W S tyx } zN|jd tkr>W Y d }~dS |jd tv rb| W Y d }~dS W Y d }~n
d }~0 0 d S r' )rY sendr0 r1 r r2 r/ )rZ dataresultr r r r r d s zdispatcher.sendc
C sx z(| j |}|s | W dS |W S W nJ tyr } z2|jd tv r\| W Y d }~dS W Y d }~n
d }~0 0 d S )N r )rY recvr/ r0 r1 r2 )rZ Z buffer_sizer r r r r r q s
zdispatcher.recvc
C sp d| _ d| _d| _| | jd urlz| j W n8 tyj } z |jd tt fvrV W Y d }~n
d }~0 0 d S )NFr )
rU r= ry rX rY closer0 r1 r r
)rZ r r r r r s
zdispatcher.closec C s t jdt| d S )Nzlog: %s
)sysstderrr$ str)rZ messager r r log s zdispatcher.loginfoc C s || j vrtd||f d S )Nz%s: %s)ignore_log_typesprint)rZ r rh r r r log_info s
zdispatcher.log_infoc C s: | j r| n&| js.| jr$| | n| d S r )r=
handle_acceptrU ry r{ handle_readrp r r r r s
z dispatcher.handle_read_eventc C s@ | j t jt j}|dkr(t|t|| d| _d| _d S )Nr TF) rY ro rm SO_ERRORr0 r handle_connectrU ry rZ r r r r r{ s z dispatcher.handle_connect_eventc C s* | j r
d S | js | jr | | d S r )r= rU ry r{ handle_writerp r r r r# s z dispatcher.handle_write_eventc C s0 | j t jt j}|dkr$| n| d S r' )rY ro rm r r/ handle_exptr r r r r% s
z dispatcher.handle_expt_eventc C sX t \}}}}z t| }W n dt| }Y n0 | 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_tracebackra rc r r/ )rZ niltvtbinfoZ self_reprr r r r s zdispatcher.handle_errorc C s | dd d S )Nz!unhandled incoming priority eventrP r rp r r r r s zdispatcher.handle_exptc C s | dd d S )Nzunhandled read eventrP r rp r r r r s zdispatcher.handle_readc C s | dd d S )Nzunhandled write eventrP r rp r r r r s zdispatcher.handle_writec C s | dd d S )Nzunhandled connect eventrP r rp r r r r s zdispatcher.handle_connectc C s | }|d ur| j| d S r )r~ handle_accepted)rZ Zpairr r r r s zdispatcher.handle_acceptc C s | | dd d S )Nzunhandled accepted eventrP )r r )rZ r[ rW r r r r s zdispatcher.handle_acceptedc C s | dd | d S )Nzunhandled close eventrP )r r rp r r r r/ s zdispatcher.handle_close)NN)N)N)N)r ),r r r debugrU r= ry closingrW frozensetr r\ re rf rX rY AF_INET SOCK_STREAMri rT rq r: r; rv rx r} r~ r r r r r r r{ r# r% r r r r r r r r/ r r r r rO sJ
rO c @ s6 e Zd Zd ddZdd Zdd Zdd Zd
d ZdS )
dispatcher_with_sendNc C s t | || d| _d S )Nr )rO r\
out_bufferrk r r r r\ s z dispatcher_with_send.__init__c C s. d}t | | jd d }| j|d | _d S )Nr i )rO r r )rZ Znum_sentr r r
initiate_send s z"dispatcher_with_send.initiate_sendc C s | d S r )r rp r r r r s z!dispatcher_with_send.handle_writec C s | j pt| jS r )rU lenr rp r r r r; s z dispatcher_with_send.writablec C s0 | j r| dt| | j| | _| d S )Nz
sending %s)r r ra r r )rZ r r r r r s zdispatcher_with_send.send)NN)r r r r\ r r r; r r r r r r s
r c C s t \} }}g }|s td|rL||jjj|jjjt|j f |j
}q ~|d \}}}d dd |D }|||f| ||fS )Nztraceback does not existr^ c S s g | ] }d | qS )z
[%s|%s|%s]r ).0xr r r