[XMc@sdZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZejejBZd e jfd YZd Zd d gZdS( s An epoll() based implementation of the twisted main loop. To install the event loop (and you should do this before any connections, listeners or connectors are added):: from twisted.internet import epollreactor epollreactor.install() iN(t implements(t IReactorFDSet(t_epoll(tlog(t posixbaseterror(tCONNECTION_DONEtCONNECTION_LOSTt EPollReactorcBseZdZeedZdZdZdZdZ dZ dZ dZ d Z d Zd ZeZd ZRS( sX A reactor that uses epoll(4). @ivar _poller: A L{poll} which will be used to check for I/O readiness. @ivar _selectables: A dictionary mapping integer file descriptors to instances of L{FileDescriptor} which have been registered with the reactor. All L{FileDescriptors} which are currently receiving read or write readiness notifications will be present as values in this dictionary. @ivar _reads: A dictionary mapping integer file descriptors to arbitrary values (this is essentially a set). Keys in this dictionary will be registered with C{_poller} for read readiness notifications which will be dispatched to the corresponding L{FileDescriptor} instances in C{_selectables}. @ivar _writes: A dictionary mapping integer file descriptors to arbitrary values (this is essentially a set). Keys in this dictionary will be registered with C{_poller} for write readiness notifications which will be dispatched to the corresponding L{FileDescriptor} instances in C{_selectables}. cCsAtjd|_i|_i|_i|_tjj|dS(sm Initialize epoll object, file descriptor tracking dictionaries, and the base class. iN( Rtepollt_pollert_readst_writest _selectablesRtPosixReactorBaset__init__(tself((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyR7s    c Csz|j}||krvtj}|} ||krI| |O} tj}n|jj||| d||<||| read/write for example). iN(tfilenoRtCTL_ADDtCTL_MODR t_control( Rtxertprimarytothert selectablesteventt antieventtfdtcmdtflags((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt_addEs       cCs/|j||j|j|jtjtjdS(sR Add a FileDescriptor for notification of data available to read. N(RR R R RtINtOUT(Rtreader((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt addReader`scCs/|j||j|j|jtjtjdS(sS Add a FileDescriptor for notification of data available to write. N(RR R R RR R(Rtwriter((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt addWritergsc Cs|j}|dkrLx1|jD]\}}||kr%Pq%q%WdSn||krtj} |} ||kr|} tj} n||=||=|jj| || ndS(s Private method for removing a descriptor from the event loop. It does the inverse job of _add, and also add a check in case of the fd has gone away. iN(RtitemsRtCTL_DELRR R( RRRRRRRRtfdesRR((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt_removens       cCs/|j||j|j|jtjtjdS(sQ Remove a Selectable for notification of data available to read. N(R(R R R RRR (RR!((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt removeReaderscCs/|j||j|j|jtjtjdS(sR Remove a Selectable for notification of data available to write. N(R(R R R RR R(RR#((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt removeWriterscCsJ|jg|jD]}|j|^qg|jD]}|j|^q0S(sD Remove all selectables, and return a list of them. (t _removeAllR R R (RR((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt removeAlls cCs!g|jD]}|j|^q S(N(R R (RR((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt getReadersscCs!g|jD]}|j|^q S(N(R R (RR((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt getWritersscCs|dkrd}nt|d}y"|jjt|j|}Wn,tk ru}|jtjkrodSnX|j }xR|D]J\}}y|j|}Wnt k rqXt j |||||qWdS(s1 Poll the poller for new events. iiN( tNonetintR twaittlenR tIOErrorterrnotEINTRt_doReadOrWritetKeyErrorRtcallWithLogger(Rttimeouttlterrt_drdwRRt selectable((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pytdoPolls   "  cCsd}t}|t@rK|tj@ rK||jkrBt}t}qt}nyy|tj@rp|j }t}n| r|tj @r|j }t}n|j |krt jd}t}nWn!tjtjd}nX|r|j|||ndS(si fd is available for read or write, make the work and raise errors if necessary. sFiledescriptor went awayiN(R/tFalset_POLL_DISCONNECTEDRRR tTrueRRtdoReadR tdoWriteRRtConnectionFdescWentAwayRR;tsystexc_infot_disconnectSelectable(RR=RRtwhytinRead((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyR6s.          (t__name__t __module__t__doc__RRRRR"R$R(R)R*R,R-R.R>t doIterationR6(((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyRs           cCs't}ddlm}||dS(s& Install the epoll() reactor. i(tinstallReactorN(Rttwisted.internet.mainRN(tpRN((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pytinstalls RQ(RLRER4tzope.interfaceRttwisted.internet.interfacesRttwisted.pythonRRttwisted.internetRRRORRtHUPtERRR@RRRQt__all__(((sA/usr/lib/python2.7/dist-packages/twisted/internet/epollreactor.pyt s