3
\N @ sh d Z ddlZddlZddlZddlZddlZddlZddlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZ ee e
eeeehZye W n ek
r i ZY nX 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!krdG 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 s> y
t j| S tttfk
r8 | tkr0t| S d| S X d S )NzUnknown error %s)osstrerror
ValueError
OverflowError NameErrorr )err r /usr/lib64/python3.6/asyncore.py _strerrorD s
r c @ s e Zd ZdS )ExitNowN)__name__
__module__ __qualname__r r r r r L s r c C s: y | j W n( tk
r" Y n | j Y nX d S )N)handle_read_event_reraised_exceptions handle_error)objr r r readQ s r c C s: y | j W n( tk
r" Y n | j Y nX d S )N)handle_write_eventr r )r r r r writeY s r" c C s: y | j W n( tk
r" Y n | j Y nX d S )N)handle_expt_eventr r )r r r r
_exceptiona s r$ c C s yX|t j@ r| j |t j@ r&| j |t j@ r8| j |t jt jB t j B @ rV| j
W nh t k
r } z&|j d t
kr| j n| j
W Y d d }~X n( tk
r Y n | j Y nX d S )Nr )selectPOLLINr POLLOUTr! POLLPRIr# ZPOLLHUPZPOLLERRZPOLLNVAL handle_closeOSErrorargs
_DISCONNECTEDr r )r flagser r r readwritei s"
r/ c C sN |d kr t }|rJg }g }g }x^t|j D ]N\}}|j }|j }|rR|j| |rh|j rh|j| |sp|r,|j| q,W g | ko| ko|kn rtj| d S t j |||| \}}}x(|D ] }|j
|}|d krqt | qW x*|D ]"}|j
|}|d krqt | qW x.|D ]&}|j
|}|d kr<q t
| q W d S )N)
socket_maplistitemsreadablewritableappend acceptingtimeZsleepr% getr r" r$ ) timeoutmaprwr. fdr Zis_rZis_wr r r poll} sD
"
r? c C s |d kr t }| d k r t| d } tj }|rx^t|j D ]N\}}d}|j r^|tjtjB O }|j rx|j
rx|tj O }|r:|j || q:W |j| }x.|D ]&\}}|j
|}|d krqt|| qW d S )Ni r )r1 intr% r? r2 r3 r4 r& r( r5 r7 r' registerr9 r/ )r: r; Zpollsterr> r r- r< r r r poll2 s(
rB >@Fc C sj |d kr t }|r ttdr t}nt}|d krBx8|r>|| | q.W n$x"|rd|dkrd|| | |d }qDW d S )Nr? r )r1 hasattrr% rB r? )r: Zuse_pollr; countZpoll_funr r r loop s
rG c @ s6 e Zd ZdZdZdZdZdZdZe dhZ
dAddZ dd Z e Z
dBdd ZdCd
d Zejejfd d
ZdDddZdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Z d 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 krt | _n|| _d | _|r|jd | j|| d| _y|j | _W q tk
r } z,|j d t
t fkrvd| _n | j | W Y d d }~X qX nd | _
d S )Nr TF)r1 _map_fileno setblocking
set_socket connectedZ getpeernameaddrr* r+ r r del_channelsocket)selfsockr; r r r r __init__ s
zdispatcher.__init__c C s | j jd | j j g}| jr.| jr.|jd n| jr>|jd | jd k ry|jd| j W n$ tk
r |jt| j Y nX ddj |t
| f S )N.Z listeningrN z%s:%dz <%s at %#x> ) __class__r r r7 rO r6 rN TypeErrorreprjoinid)rR Zstatusr r r __repr__ s
zdispatcher.__repr__c C s |d kr| j }| || j< d S )N)rJ rK )rR r; r r r add_channel s zdispatcher.add_channelc C s, | j }|d kr| j}||kr"||= d | _ d S )N)rK rJ )rR r; r> r r r rP s zdispatcher.del_channelc C s. ||f| _ tj||}|jd | j| d S )Nr )Zfamily_and_typerQ rL rM )rR ZfamilytyperS r r r
create_socket s
zdispatcher.create_socketc C s || _ |j | _| j| d S )N)rQ filenorK r] )rR rS r; r r r rM s
zdispatcher.set_socketc C sD y*| j jt jt j| j jt jt jdB W n tk
r> Y nX d S )NrD )rQ Z
setsockopt
SOL_SOCKETZ SO_REUSEADDR
getsockoptr* )rR r r r set_reuse_addr& s
zdispatcher.set_reuse_addrc C s dS )NTr )rR r r r r4 7 s zdispatcher.readablec C s dS )NTr )rR r r r r5 : s zdispatcher.writablec C s( d| _ tjdkr |dkr d}| jj|S )NTnt )r7 r namerQ listen)rR Znumr r r rg A s zdispatcher.listenc C s || _ | jj|S )N)rO rQ bind)rR rO r r r rh G s zdispatcher.bindc C sp d| _ d| _| jj|}|tttfks8|tkrBtj dkrB|| _
d S |dt fkr^|| _
| j nt
|t| d S )NFTrd r )rN
connectingrQ Z
connect_exr r r r r rf rO r handle_connect_eventr* r )rR Zaddressr r r r connectK s
zdispatcher.connectc C sn y| j j \}}W nN tk
r& d S tk
r` } z |jd tttfkrNd S W Y d d }~X n
X ||fS d S )Nr )rQ acceptrX r* r+ r r r
)rR ZconnrO whyr r r rl Y s zdispatcher.acceptc C sf y| j j|}|S tk
r` } z4|jd tkr4dS |jd tkrN| j dS W Y d d }~X nX d S )Nr )rQ sendr* r+ r r, r) )rR dataresultrm r r r rn g s zdispatcher.sendc C sh y$| j j|}|s | j dS |S W n> tk
rb } z"|jd tkrP| j dS W Y d d }~X nX d S )N r )rQ recvr) r* r+ r, )rR buffer_sizero rm r r r rr t s zdispatcher.recvc C sn d| _ d| _d| _| j | jd k rjy| jj W n6 tk
rh } z|jd tt fkrX W Y d d }~X nX d S )NFr )
rN r7 ri rP rQ closer* r+ r r
)rR rm r r r rt s
zdispatcher.closec C s t jjdt| d S )Nzlog: %s
)sysstderrr" str)rR messager r r log s zdispatcher.loginfoc C s || j krtd||f d S )Nz%s: %s)ignore_log_typesprint)rR rx r^ r r r log_info s
zdispatcher.log_infoc C s: | j r| j n&| js.| jr$| j | j n| j d S )N)r7
handle_acceptrN ri rj handle_read)rR r r r r s
z dispatcher.handle_read_eventc C s@ | j jt jt j}|dkr(t|t|| j d| _d| _d S )Nr TF) rQ rb ra SO_ERRORr* r handle_connectrN ri )rR r r r r rj s z dispatcher.handle_connect_eventc C s* | j r
d S | js | jr | j | j d S )N)r7 rN ri rj handle_write)rR r r r r! s z dispatcher.handle_write_eventc C s0 | j jt jt j}|dkr$| j n| j d S )Nr )rQ rb ra r r) handle_expt)rR r r r r r# s
z dispatcher.handle_expt_eventc C sX t \}}}}y t| }W n dt| }Y nX | jd||||f d | j d S )Nz)<__repr__(self) failed for object at %0x>z:uncaptured python exception, closing channel %s (%s:%s %s)error)compact_tracebackrY r[ r} r) )rR ZniltvtbinfoZ self_reprr r r r s zdispatcher.handle_errorc C s | j dd d S )Nz!unhandled incoming priority eventrI )r} )rR r r r r s zdispatcher.handle_exptc C s | j dd d S )Nzunhandled read eventrI )r} )rR r r r r s zdispatcher.handle_readc C s | j dd d S )Nzunhandled write eventrI )r} )rR r r r r s zdispatcher.handle_writec C s | j dd d S )Nzunhandled connect eventrI )r} )rR r r r r s zdispatcher.handle_connectc C s | j }|d k r| j| d S )N)rl handle_accepted)rR Zpairr r r r~ s zdispatcher.handle_acceptc C s |j | jdd d S )Nzunhandled accepted eventrI )rt r} )rR rS rO r r r r s zdispatcher.handle_acceptedc C s | j dd | j d S )Nzunhandled close eventrI )r} rt )rR r r r r) s zdispatcher.handle_close)NN)N)N)N)rz )-r r r debugrN r7 ri closingrO frozensetr{ rT r\ __str__r] rP rQ ZAF_INETZ SOCK_STREAMr_ rM rc r4 r5 rg rh rk rl rn rr rt ry r} r rj r! r# r r r r r r~ r r) r r r r rH sL
rH c @ s6 e Zd Zd ddZdd Zdd Zdd Zd
d ZdS )
dispatcher_with_sendNc C s t j| || d| _d S )Nrq )rH rT
out_buffer)rR rS r; r r r rT s z dispatcher_with_send.__init__c C s. d}t j| | jd d }| j|d | _d S )Nr i )rH rn r )rR Znum_sentr r r
initiate_send s z"dispatcher_with_send.initiate_sendc C s | j d S )N)r )rR r r r r s z!dispatcher_with_send.handle_writec C s | j pt| jS )N)rN lenr )rR r r r r5 s z dispatcher_with_send.writablec C s0 | j r| jdt| | j| | _| j d S )Nz
sending %s)r r} rY r r )rR ro r r r rn s zdispatcher_with_send.send)NN)r r r rT r r r5 rn r r r r r s
r c C s t j \} }}g }|s tdx0|rN|j|jjj|jjjt|j f |j
}q W ~|d \}}}dj dd |D }|||f| ||fS )Nztraceback does not existrD rV c S s g | ] }d | qS )z
[%s|%s|%s]r ).0xr r r