o Òó·ikOã@sêdZddlmZddlZddlZddlZddlmZmZm Z ddl m Z m Z ddl mZmZmZmZmZddlmZmZddlmZmZdd lmZdd lmZmZdd lm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(ddl)m*Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5zddl6m7Z8Wn e9y›dZ:Ynwe8Z:er¦ddl;miZ?e e&ej@ejAƒGdd„dejBe1e,ƒƒZCe:durËe eCejDƒGdd„de-e.eCƒZEGdd„deCƒZFGdd„de+ƒZ*e ejGƒGdd„de1eƒƒZHdS)z TCP support for IOCP reactor é)Ú annotationsN)Ú TYPE_CHECKINGÚOptionalÚUnion)ÚclassImplementsÚ implementer)ÚaddressÚdeferÚerrorÚ interfacesÚmain)Ú _LogOwnerÚ isIPv6Address)Ú IPv4AddressÚ IPv6Address)Ú IProtocol)ÚabstractÚ iocpsupport)ÚERROR_CONNECTION_REFUSEDÚERROR_IO_PENDINGÚERROR_NETWORK_UNREACHABLEÚSO_UPDATE_ACCEPT_CONTEXTÚSO_UPDATE_CONNECT_CONTEXT)ÚIReadWriteHandle)ÚProtocol)Ú ConnectorÚ_AbortingMixinÚ_BaseBaseClientÚ_BaseTCPClientÚ _getsocknameÚ _resolveIPv6Ú _SocketCloser)ÚfailureÚlogÚreflect)ÚstartTLS)Ú IOCPReactorc@sÊeZdZdZdZd/dd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„ZedurFd0dd „Zd!d"„Zd#d$„Zd/d%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„ZdS)1Ú ConnectionzÞ @ivar TLS: C{False} to indicate the connection is in normal TCP mode, C{True} to indicate that TLS has been started and that operations must be routed through the L{TLSMemoryBIOProtocol} instance. FNcCs&tj ||¡||_|j|_||_dS©N)rÚ FileHandleÚ__init__ÚsocketÚfilenoÚ getFileHandleÚprotocol)ÚselfÚsockÚprotoÚreactor©r3úfC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/internet/iocpreactor/tcp.pyr*Fs zConnection.__init__cCó|jSr()r+©r/r3r3r4Ú getHandleLszConnection.getHandlecCs@t|tƒrnt|tƒrt|ƒ}ntdt|ƒƒ‚|j |¡dS)z` @param rbuffer: Data received. @type rbuffer: L{bytes} or L{bytearray} z%data must be bytes or bytearray, not N)Ú isinstanceÚbytesÚ bytearrayÚ TypeErrorÚtyper.Ú dataReceived)r/Zrbufferr3r3r4r=Os   zConnection.dataReceivedcCst | ¡||¡Sr()Ú_iocpÚrecvr-)r/ZbufflistÚevtr3r3r4ÚreadFromHandle^szConnection.readFromHandlecCs(t|ƒ}t | ¡|d|j… ¡|¡S)zŽ Send C{buff} to current file handle using C{_iocp.send}. The buffer sent is limited to a size of C{self.SEND_LIMIT}. rN)Ú memoryviewr>Úsendr-Z SEND_LIMITÚtobytes)r/Úbuffr@Z writeViewr3r3r4Ú writeToHandleasÿzConnection.writeToHandlecCsxz|j d¡Wn tyYnwt |jd¡}|r:z| ¡WdSty9t  ¡}t   ¡|  |¡YdSwdS)Né) r+ÚshutdownÚOSErrorr ÚIHalfCloseableProtocolr.ZwriteConnectionLostÚ BaseExceptionr"ÚFailurer#ÚerrÚconnectionLost)r/ÚpÚfr3r3r4Ú_closeWriteConnectionks ÿ ýýz Connection._closeWriteConnectioncCsZt |jd¡}|r&z| ¡WdSty%t ¡| t  ¡¡YdSw| |¡dSr() r rJr.ÚreadConnectionLostrKr#rMrNr"rL)r/ÚreasonrOr3r3r4rRys þzConnection.readConnectionLostcCsX|jrdStj ||¡|dup| tj¡ }| |¡|j}|`|` |` | |¡dSr() Ú disconnectedrr)rNÚcheckr ZConnectionAbortedÚ _closeSocketr.r+r-)r/rSZisCleanr.r3r3r4rN„s zConnection.connectionLostcCr5)zN Return the prefix to log with when I own the logging thread. N)Úlogstrr6r3r3r4Ú logPrefixózConnection.logPrefixcCót|j tjtj¡ƒSr()Úboolr+Ú getsockoptÚ IPPROTO_TCPÚ TCP_NODELAYr6r3r3r4Ú getTcpNoDelay–ózConnection.getTcpNoDelaycCó|j tjtj|¡dSr()r+Ú setsockoptr]r^©r/Úenabledr3r3r4Ú setTcpNoDelay™ózConnection.setTcpNoDelaycCrZr()r[r+r\Ú SOL_SOCKETÚ SO_KEEPALIVEr6r3r3r4ÚgetTcpKeepAliveœr`zConnection.getTcpKeepAlivecCrar()r+rbrgrhrcr3r3r4ÚsetTcpKeepAliveŸrfzConnection.setTcpKeepAliveTcCst|||tjƒdS)z9 @see: L{ITLSTransport.startTLS} N)Ú _startTLSrr))r/ZcontextFactoryÚnormalr3r3r4r%¤szConnection.startTLScCó2|jrdS|jr|j |¡dStj ||¡dS)zò Write some data, either directly to the underlying handle or, if TLS has been started, to the L{TLSMemoryBIOProtocol} for it to encrypt and send. @see: L{twisted.internet.interfaces.ITransport.write} N)rTÚTLSr.Úwriterr))r/Údatar3r3r4roªó zConnection.writecCrm)zú Write some data, either directly to the underlying handle or, if TLS has been started, to the L{TLSMemoryBIOProtocol} for it to encrypt and send. @see: L{twisted.internet.interfaces.ITransport.writeSequence} N)rTrnr.Ú writeSequencerr))r/Ziovecr3r3r4rr¹rqzConnection.writeSequencecCs:|jr|jr|js|j ¡dSdSdStj ||¡dS)z® Close the underlying handle or, if TLS has been started, first shut it down. @see: L{twisted.internet.interfaces.ITransport.loseConnection} N)rnÚ connectedÚ disconnectingr.ÚloseConnectionrr))r/rSr3r3r4ruÈs  ÿzConnection.loseConnectioncCs,|jr |j ||¡dStj |||¡dS)zc Register a producer. If TLS is enabled, the TLS connection handles this. N)rnr.ÚregisterProducerrr))r/ZproducerZ streamingr3r3r4rvÕszConnection.registerProducercCs$|jr |j ¡dStj |¡dS)ze Unregister a producer. If TLS is enabled, the TLS connection handles this. N)rnr.ÚunregisterProducerrr)r6r3r3r4rwäszConnection.unregisterProducercCódSr(r3r6r3r3r4ÚgetHostïózConnection.getHostcCrxr(r3r6r3r3r4ÚgetPeerórzzConnection.getPeerr()T)Ú__name__Ú __module__Ú __qualname__Ú__doc__rnr*r7r=rArFrQrRrNrXr_rerirjrkr%rorrrurvrwryr{r3r3r3r4r'<s2        r'c@sTeZdZdZejZejZdZ e Z dd„Z dd„Z dd„Zd d „Zd d „Zd d„ZdS)ÚClientzº @ivar _tlsClientDefault: Always C{True}, indicating that this is a client connection, and by default when TLS is negotiated this class will act as a TLS client. TcCs*|durd}||_t ||||||¡dS)N)Úr)r2rr*)r/ÚhostÚportÚ bindAddressÚ connectorr2r3r3r4r* szClient.__init__cCs|j |j|j¡S)zd Create a socket registered with the IOCP reactor. @see: L{_BaseTCPClient} N)r2Ú createSocketÚ addressFamilyÚ socketTyper6r3r3r4ÚcreateInternetSocketszClient.createInternetSocketcCs |`|`dS)z– Clean up potentially circular references to the socket and to its C{getFileHandle} method. @see: L{_BaseBaseClient} N)r+r-r6r3r3r4Ú_collectSocketDetailss zClient._collectSocketDetailscCs|j |¡dS)z^ Remove the active handle from the reactor. @see: L{_BaseBaseClient} N)r2ÚremoveActiveHandler6r3r3r4Ú_stopReadingAndWriting!szClient._stopReadingAndWritingc Cs¶|rt ||¡}| t |tj |d¡f¡¡dS|j tj t t   d|j  ¡¡¡|j | ¡¡|_d|_| |j¡}|d|_|jdurOtƒ|_| ¡dS|j |¡| ¡dS)Nz Unknown errorÚPTz,client)ÚconnectExErrorsÚgetZfailIfNotConnectedr ZgetConnectErrorÚerrnoÚ errorcoder+rbrgrÚstructÚpackr,r…Ú buildProtocolr{r.rsÚ _getLogPrefixrWrruÚmakeConnectionÚ startReading)r/Úrcrpr@rXr3r3r4Ú cbConnect)s& ÿý      zClient.cbConnectcCsnt|dƒsdStjs J‚|j |¡t |j|¡}t |j  ¡|j |¡}|r3|t kr5| |d|¡dSdSdS)Nr…r) Úhasattrr>Zhave_connectexr2ÚaddActiveHandleÚEventr™Úconnectr+r,Z realAddressr)r/r@r˜r3r3r4Ú doConnectGs    ÿzClient.doConnectN)r|r}r~rr+ÚAF_INETr‡Ú SOCK_STREAMrˆÚ_tlsClientDefaultr'Z_commonConnectionr*r‰rŠrŒr™ržr3r3r3r4r€üs  r€c@s8eZdZdZdZddd„Zddd„Zdd„Zdd„ZdS)ÚServeraV Serverside socket-stream connection class. I am a serverside network connection transport; a socket which came from an accept() on a server. @ivar _tlsClientDefault: Always C{False}, indicating that this is a server connection, and by default when TLS is negotiated this class will act as a TLS server. Fr0ú socket.socketr.rÚ clientAddrúUnion[IPv4Address, IPv6Address]Ú serverAddrÚ sessionnoÚintr2r&cCsvt ||||¡||_||_||_| |j¡}|›d|›d|jj›|_d  |jj j |j|jj ¡|_ d|_| ¡dS)a  Server(sock, protocol, client, server, sessionno) Initialize me with a socket, a protocol, a descriptor for my peer (a tuple of host, port describing the other end of the connection), an instance of Port, and a session number. ú,z<{} #{} on {}>TN)r'r*r¦r¤r§r•r.r‚rWÚformatÚ __class__r|rƒÚrepstrrsr—)r/r0r.r¤r¦r§r2rXr3r3r4r*cs ý zServer.__init__ÚreturnÚstrcCr5)z= A string representation of this connection. N)r¬r6r3r3r4Ú__repr__rYzServer.__repr__cCr5)zW Returns an IPv4Address. This indicates the server's address. N)r¦r6r3r3r4ry‡ózServer.getHostcCr5)zW Returns an IPv4Address. This indicates the client's address. N)r¤r6r3r3r4r{r°zServer.getPeerN) r0r£r.rr¤r¥r¦r¥r§r¨r2r&©r­r®) r|r}r~rr¡r*r¯ryr{r3r3r3r4r¢Us   r¢c@seZdZdd„ZdS)rcCst|j|j|j||jƒSr()r€r‚rƒr„r2r6r3r3r4Ú_makeTransport™rfzConnector._makeTransportN)r|r}r~r²r3r3r3r4r˜s rc@s¬eZdZUdZdZdZejZej Z e j Z dZdZded<dZd!d d „Zd"d d„Zdd„Ze ej¡fdd„ZeZdd„Zdd„Zdd„Zdd„Zdd„Zdd„Z dd „Z!dS)#ÚPortFrNz Optional[int]Ú_realPortNumberÚTCPé2rcCs>||_||_||_||_||_t|ƒrtj|_t j |_ dSdSr() rƒÚfactoryÚbacklogÚ interfacer2rr+ÚAF_INET6r‡rrÚ _addressType)r/rƒr·r¸r¹r2r3r3r4r*±s þz Port.__init__r­r®cCs2|jdurd |j|jj|j¡Sd |j|jj¡S)Nz<{} of {} on {}>z<{} of {} (not listening)>)r´rªr«r·r6r3r3r4r¯»s ýþz Port.__repr__c Csøz$|j |j|j¡}|jtjkrt|j|jƒ}n|j|jf}|  |¡Wnt y9}z t   |j|j|¡‚d}~wwt  | ¡¡|_| ¡d|_t d| |j¡|jf¡|j ¡| |j¡d|_d|_|j |¡||_|jj|_| ¡dS)NrGz%s starting on %sTF)r2r†r‡rˆr+rºr r¹rƒÚbindrIr ZCannotListenErrorr>Z maxAddrLenr,ÚaddrLenÚ getsocknamer´r#Úmsgr•r·ZdoStartÚlistenr¸rsrTr›r-ÚdoAccept)r/ZsktÚaddrÚler3r3r4ÚstartListeningÈs2  €ÿÿÿ     zPort.startListeningcCs2d|_|jrt ¡|_|j d|j|¡|jSdS)zæ Stop accepting connections on this port. This will shut down my socket and call self.connectionLost(). It returns a deferred which will fire successfully when the port is actually closed. TrN)rtrsr ZDeferredÚdeferredr2Z callLaterrN)r/ZconnDoner3r3r4ruès  ýzPort.loseConnectioncCs t d|j›d|j›d¡dS)z. Log message for closing port ú(z Port z Closed)N)r#r¿Ú_typer´r6r3r3r4Ú_logConnectionLostMsgøs zPort._logConnectionLostMsgcCs²| ¡d|_d}t|dƒr|j}|`d|_|j |¡d|_| d¡|` |` z|j   ¡Wnt yHd|_|durG| t ¡¡YdS‚wd|_|durW| d¡dSdS)z' Cleans up the socket. NrÅTF)rÈr´ršrÅrTr2r‹rsrVr+r-r·ZdoStoprKrtZerrbackr"rLÚcallback)r/rSÚdr3r3r4rNþs0    ûÿzPort.connectionLostcCst |jj¡S)zK Returns the name of my class, to prefix log entries with. N)r$Zqualr·r«r6r3r3r4rXszPort.logPrefixcCs|jdgt|jƒ¢RŽS)zf Returns an IPv4Address or IPv6Address. This indicates the server's address. rµN)r»rr+r6r3r3r4ry#sz Port.getHostcCs,| ||¡|js|js| ¡dSdSdSr()Ú handleAcceptrtrTrÁ)r/r˜rpr@r3r3r4ÚcbAccept+s   ÿz Port.cbAcceptc Csh|js|jrdS|rt dtj |d¡|f¡dS|j t j t t   d|j  ¡¡¡t |j ¡|j¡\}}}||jks>J‚d|dvrYt|d d¡dƒ}|d|dd|f}d|dvrtt|d d¡dƒ}|d|dd|f}|j |jdg|¢RŽ¡}|durŒ|j ¡d S|j}|d|_t|j||jdg|¢RŽ|jdg|¢RŽ||jƒ} | | ¡d S) NFz*Could not accept new connection -- %s (%s)z unknown errorrú%rrGrµT)rtrTr#r¿rr‘rÚnewsktrbr+rgrr’r“r,r>Zget_accept_addrsrEr‡r¨Úsplitr·r”r»Úcloser§r¢r2r–) r/r˜r@ÚfamilyZlAddrZrAddrZscoper.ÚsÚ transportr3r3r4rË0sL ÿÿý    õ ú zPort.handleAcceptcCs|t |j|¡}td|jdƒ|_}|j |j|j ¡|_ }t  |j   ¡|  ¡||¡}|r:|tkr<| ||¡dSdSdS)Néé)r>rœrÌr:r½rEr2r†r‡rˆrÎÚacceptr+r,rrË)r/r@rErÎr˜r3r3r4rÁcs ÿ ÿz Port.doAccept)r¶rNr±)"r|r}r~rsrTrtr+rŸr‡r rˆrrr»r§r´Ú__annotations__rÇr*r¯rÄr"rLr ZCONNECTION_DONEruÚ stopListeningrÈrNrXryrÌrËrÁr3r3r3r4r³s,       3r³)IrÚ __future__rrr+r’ÚtypingrrrZzope.interfacerrZtwisted.internetrr r r r Ztwisted.internet.abstractr rZtwisted.internet.addressrrZtwisted.internet.interfacesrZtwisted.internet.iocpreactorrrr>Z"twisted.internet.iocpreactor.constrrrrrZ'twisted.internet.iocpreactor.interfacesrZtwisted.internet.protocolrZtwisted.internet.tcprZ TCPConnectorrrrrr r!Ztwisted.pythonr"r#r$Ztwisted.internet._newtlsr%Z __startTLSÚ ImportErrorrkZ$twisted.internet.iocpreactor.reactorr&ZWSAECONNREFUSEDZWSAENETUNREACHrŽZ ITCPTransportZ ISystemHandler)r'Z ITLSTransportr€r¢ZIListeningPortr³r3r3r3r4ÚsL    $  ÿ þ< YC