o iv@@sdZddlmZddlZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZddlmZdd lmZmZmZdd lmZdd lmZmZmZmZmZmZmZmZm Z!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9m:Z:m;Z;mZ>ddl?m@Z@mAZAddlBmCZCddlDmEZEm Z GdddejFZGeGdddeHZIGdddejFZJe dd eJd!ZKe e eefe.gdfZLe3ZMGd"d#d#ejFeKZNGd$d%d%ejFeKZOed d&Gd'd(d(ZPedXd/d0ZQee+eGd1d2d2e eKZReReJZSeQjTdYd4d5ZUeQjTdZd7d8ZVeQjTd[d:d;ZWeQjTd\d=d>ZXeQjTd]d@dAZYeQjTd^dCdDZZeQjTd_dFdGZ[eQjTd`dIdJZ\dadNdOZ]GdPdQdQeCZ^dbdVdWZ_dS)czD Private implementation module for public L{twisted.web.websocket}. ) annotationsN) dataclassfield)singledispatch)CallableGenericTypeVarUnion) implementer)URL) ConnectionConnectionType WSConnection)ConnectionState) AcceptConnection BytesMessageCloseConnectionEventPingPongRejectConnection RejectDataRequest TextMessage) H11Handshake)RemoteProtocolError)Deferred) IConsumer IProtocol IPushProducer IReactorTCP ITransport)Factory)Logger)Failure) BAD_REQUEST)URIBrowserLikePolicyForHTTPSResponse_StandardEndpointFactory)Headers)IAgentEndpointFactoryIPolicyForHTTPS)Resource) NOT_DONE_YETrc@sPeZdZdZdddZdd d ZddddZd d!ddZd"ddZd#ddZ dS)$WebSocketTransportz9 The transport that can send websocket messages. textstrreturnNonecCdS)z& Send a text message. N)selfr0r5r5aC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/web/_websocket_impl.pysendTextMessageAz"WebSocketTransport.sendTextMessagedatabytescCr4)z' Send a bytes message. Nr5r6r:r5r5r7sendBytesMessageFr9z#WebSocketTransport.sendBytesMessagecodeintcCr4)z0 Drop the websocket connection. Nr5)r6r?r5r5r7loseConnectionKr9z!WebSocketTransport.loseConnectionpayloadcCr4)a Send a websocket Ping request to measure latency. @note: Per U{Mozilla's documentation }, multiple 'ping' requests may be coalesced into a single 'pong', and unsolicited 'pong' requests must be ignored, so we do not return a L{deferred } here; pongs are delivered separately. Nr5r6rCr5r5r7pingPr9zWebSocketTransport.pingproducerrcCr4)zF Attach the given L{IPushProducer} to this transport. Nr5r6rFr5r5r7attachProducer\r9z!WebSocketTransport.attachProducercCr4)zT Detach a previously attached L{IPushProducer} from this transport. Nr5r6r5r5r7detachProducerar9z!WebSocketTransport.detachProducerNr0r1r2r3r:r;r2r3)r>)r?r@r2r3rBrCr;r2r3rFrr2r3r2r3) __name__ __module__ __qualname____doc__r8r=rArErHrJr5r5r5r7r/<s     r/c@seZdZUdZded<dS)ConnectionRejectedz A websocket connection was rejected by an HTTP response. @ivar response: The HTTP response that describes the rejection. r(responseN)rQrRrSrT__annotations__r5r5r5r7rUgs  rUc@sLeZdZdZdddZddd Zdd d ZdddZdddZdddZ dS) WebSocketProtocolac An object that conforms to L{WebSocketProtocol} can receive all the events from a websocket connection. @note: While this is I{sort of} like a L{byte-stream protocol }, the interface is distinct in a few ways; in particular, we have discrete C{negotiationStarted} and C{negotiationFinished} methods, representing those events in the websocket handshaking process, and no C{connectionMade}; similarly, since websockets can natively support both text and bytes messages, rather than fragmentable segments of a byte stream, we have C{textMessageReceived} and C{bytesMessageReceived} but no C{dataReceived}. Finally, this is a L{typing.Protocol} and not a L{zope.interface.Interface}, since it does not predate the L{typing} module. transportr/r2r3cCr4)z An underlying transport (e.g.: a TCP connection) has been established; negotiation of the websocket transport has begun. Nr5r6rYr5r5r7negotiationStartedr9z$WebSocketProtocol.negotiationStartedcCr4)zn Negotiation is complete: a bidirectional websocket channel is now fully established. Nr5rIr5r5r7negotiationFinishedr9z%WebSocketProtocol.negotiationFinishedrCr;cCr4)a A Pong message was received. @note: Per U{the standard }:: A Pong frame sent in response to a Ping frame must have identical "Application data" as found in the message body of the Ping frame being replied to. If an endpoint receives a Ping frame and has not yet sent Pong frame(s) in response to previous Ping frame(s), the endpoint MAY elect to send a Pong frame for only the most recently processed Ping frame. Given that some Pong frames may be dropped, this event should only be used in concert with the transport's L{.ping } method for its intended purpose, to measure latency and connection durability, not to transport application data. Nr5rDr5r5r7 pongReceivedr9zWebSocketProtocol.pongReceivedmessager1cCr4)z< A text message was received from the peer. Nr5)r6r^r5r5r7textMessageReceivedr9z%WebSocketProtocol.textMessageReceivedr:cCr4)z= A bytes message was received from the peer. Nr5r<r5r5r7bytesMessageReceivedr9z&WebSocketProtocol.bytesMessageReceivedreasonr$cCr4)z4 The websocket connection was lost. Nr5r6rar5r5r7connectionLostr9z WebSocketProtocol.connectionLostN)rYr/r2r3rPrN)r^r1r2r3rLrar$r2r3) rQrRrSrTr[r\r]r_r`rcr5r5r5r7rXrs     rX_WSPT) covariantboundc@eZdZdZd ddZdS) WebSocketServerFactoryz A L{WebSocketServerFactory} is a factory for a particular kind of L{WebSocketProtocol} that implements server-side websocket listeners via L{WebSocketResource}. requestrr2recCr4)a/ To conform to L{WebSocketServerFactory}, you must implement a C{buildProtocol} method which takes a L{Request } and returns a L{WebSocketProtocol}. @return: a L{WebSocketProtocol} that will handle the inbound connection. Nr5)r6rjr5r5r7 buildProtocolr9z$WebSocketServerFactory.buildProtocolN)rjrr2rerQrRrSrTrkr5r5r5r7riric@rh) WebSocketClientFactoryz A L{WebSocketClientFactory} is a factory for a particular kind of L{WebSocketProtocol} that implements client-side websocket listeners via L{WebSocketClientEndpoint}. urlr1r2recCr4)a# To conform to L{WebSocketServerFactory}, you must implement a C{buildProtocol} method which takes a string representing an URL and returns a L{WebSocketProtocol}. @return: a L{WebSocketProtocol} that will handle the outgoing connection. Nr5)r6ror5r5r7rkr9z$WebSocketClientFactory.buildProtocolN)ror1r2rerlr5r5r5r7rnrmrn)frozenc@sHeZdZUdZded< ded< eeddfdddZdddZdS)WebSocketClientEndpointz A L{WebSocketClientEndpoint} describes an URL to connect to and a way of connecting to that URL, that can connect a L{WebSocketClientFactory} to that URL. r+endpointFactoryr1roNreactorr tlsPolicyr,connectTimeout int | None bindAddress bytes | Noner2cCst||||}t||S)a) Construct a L{WebSocketClientEndpoint} from a reactor and a URL. @param reactor: The reactor to use for the TCP connection. @param url: a string describing an URL where a websocket server lives. @param tlsPolicy: The TLS policy to use for HTTPS connections. @param connectTimeout: The number of seconds for the TCP-level connection timeout. @param bindAddress: The bind address to use for the TCP client connections. @return: the newly constructed endpoint. N)r)rq)clsrsrortrurwrrr5r5r7news zWebSocketClientEndpoint.newprotocolFactoryWebSocketClientFactory[_WSP]recsXjtjd}d fdd |tfd d Id H}|j Id HS)ae Make an outgoing connection to this L{WebSocketClientEndpoint}'s HTTPS connection. @param protocolFactory: The constructor for the protocol. @return: A coroutine (that yields L{Deferred}s) that completes with the connected L{WebSocketProtocol} once the websocket connection is established. zutf-8wscWSConnection | Connectiontr!r2r3cs<tj}t|jdddd}||t|j|dS)N)schemehostport) r ZfromTextror1replacewritesend WSRequestr)r}rhtargetrIr5r7clientBootstrap s z8WebSocketClientEndpoint.connect..clientBootstrapcstttjjSN)_WebSocketWireProtocolrr ZCLIENTrkror5rr{r6r5r7's  z1WebSocketClientEndpoint.connect..Nr}r~rr!r2r3) rrZendpointForURIr&Z fromBytesroencodeconnectProtocolFactoryZ forProtocol_done)r6r{Zendpoint connectedr5rr7rs  zWebSocketClientEndpoint.connect) rsr ror1rtr,rurvrwrxr2rq)r{r|r2re) rQrRrSrTrW classmethodr'rzrr5r5r5r7rqs rqeventrprotoAnyWSWPr2r3cCr4)z, Handle a websocket protocol event. Nr5rrr5r5r7 _handleEvent1r9rc@seZdZUded<ded<ded<eddZd ed <eddZd ed <d Zded<d8ddZd9ddZ d:ddZ d;ddZ dd&d'Zd?d@d,d-ZdAd0d1Zd9d2d3Zd9d4d5Zd9d6d7Zd S)Brr~_wsconn _Bootstrap _bootstrapre_wspF)initr!rYzDeferred[_WSP]rNzResponse | None_rejectResponser2r3cCs||_t|_|dSr)rYrrconnectionMaderZr5r5r7makeConnectionGs z%_WebSocketWireProtocol.makeConnectioncCs |j|||j|jdSr)rr[rrrYrIr5r5r7rLs z%_WebSocketWireProtocol.connectionMader:r;cCs*|j||jD]}t||q dSr)rZ receive_dataeventsr)r6r:rr5r5r7 dataReceivedQs  z#_WebSocketWireProtocol.dataReceivedrar$cCs0|j||jdur|j|d|_dSdSr)rrcrZ_bodyDataFinishedrbr5r5r7rcVs    z%_WebSocketWireProtocol.connectionLostr0r1cC,|j}|dus J||jt|dSr)rYrrrr)r6r0rr5r5r7r8] z&_WebSocketWireProtocol.sendTextMessagecCrr)rYrrrr)r6r:rr5r5r7r=brz'_WebSocketWireProtocol.sendBytesMessagerBrCcCrr)rYrrrr)r6rCrr5r5r7rEgrz_WebSocketWireProtocol.pingr>rr?r@cCs6|j}|dus J||jt|||dSr)rYrrrrrA)r6r?rarr5r5r7rAls  z%_WebSocketWireProtocol.loseConnectionrFrcCst|j|ddS)NT)rrYZregisterProducerrGr5r5r7rHrsz%_WebSocketWireProtocol.attachProducercCst|jdSr)rrYZunregisterProducerrIr5r5r7rJusz%_WebSocketWireProtocol.detachProducercCs$|j}|`||j|jdSr)rcallbackrr\r6doner5r5r7_completeConnectionxs z*_WebSocketWireProtocol._completeConnectioncCs,|jdusJ|j}|`|t|jdSr)rrZerrbackrUrr5r5r7_rejectConnection~sz(_WebSocketWireProtocol._rejectConnection)rYr!r2r3rPrLrdrKrMrN)r>r)r?r@rar1r2r3rO)rQrRrSrWrrYrrrrrrcr8r=rErArHrJrrr5r5r5r7r8s&             rrcCs |dSr)rrr5r5r7_handle_acceptConnections rrcCsDt}|jD] \}}|||qtd|jd||j|_|dS)Nz1.1r)r*headersZ addRawHeaderr( status_coderYrr)rrhdrkvr5r5r7_handle_rejectConnections  rrcCs8|jdus Jd|j|j|jr|jdSdS)Nz7response should never be None when receiving RejectData)rZ_bodyDataReceivedr:Z body_finishedrYrArr5r5r7_handle_rejectDatasrrcC|j|jdSr)rr_r:rr5r5r7_handle_textMessagerrcCrr)rr`r:rr5r5r7_handle_bytesMessagerrrcCs|j|j|dSr)rYrrrrVrr5r5r7 _handle_pingsrrcCrr)rr]rCrr5r5r7 _handle_pongrrrcCsB|jdusJ|jjtjkr|j|j||jdSr) rYrstaterZCLOSEDrrrVrArr5r5r7_handle_closeConnectionsrrjrr;cCs|t|dddS)Nz content-typez text/plains$websocket protocol negotiation error)ZsetResponseCoder%Z setHeader)rjr5r5r7_negotiationErrors  rcs,eZdZdZd fdd Zdd d ZZS)WebSocketResourcez A L{WebSocketResource} is a L{Resource} that presents a websocket listener. You can install it into any twisted web server resource hierarchy. factory)WebSocketServerFactory[WebSocketProtocol]r2r3cst||_dS)a# Create a L{WebSocketResource} that will respond to incoming connections with the given L{WebSocketServerFactory}. @param factory: The factory that will be used to respond to inbound websocket connections on appropriately formatted GET requests. N)super__init__r)r6r __class__r5r7rs  zWebSocketResource.__init__rjr bytes | intc Csttj}|j}dd|D}z |||jWnty8}ztj d||dt |WYd}~Sd}~ww|j |}|dusGJd| t}|j}|jj} | dus\Jd|dusdJdt|t|} |j| | ||tS) zv This implementation of the C{GET} HTTP method will respond to inbound websocket connections. cSs"g|] \}}|D]}||fqqSr5r5).0hkeyZhvalsvalr5r5r7 s"z0WebSocketResource.render_GET..z{request} failed with {rpe})rjrpeNz"connection not accepted by twistedzchannel transport not connectedz"connection not accepted by wsproto)rr ZSERVERZrequestHeadersZgetAllRawHeadersZinitiate_upgrade_connectionpathr_logerrorrrrkrr connectionZchannelrYrserverBootstrapZ_protocolUpgradeForWebsocketsrr\r.) r6rjZ handshakerawZ simpleHeadersrZwsprotZtoSendZwsconrZ wireProtor5r5r7 render_GETs,       zWebSocketResource.render_GET)rrr2r3)rjrr2r)rQrRrSrTrr __classcell__r5r5rr7rs rr}r~rr!cCr4)zC The server requires no bootstrapping, so this is a no-op. Nr5)r}rr5r5r7rr9r)rrrrr2r3)rrrrr2r3)rrrrr2r3)rrrrr2r3)rrrrr2r3)rrrrr2r3)rrrrr2r3)rrrrr2r3)rrrrr2r3)rjrr2r;r)`rT __future__rtyping dataclassesrr functoolsrrrrr Zzope.interfacer Z hyperlinkr Zwsprotor r rZwsproto.connectionrZwsproto.eventsrrrrrrrrrrrZwsproto.handshakerZwsproto.utilitiesrZtwisted.internet.deferrZtwisted.internet.interfacesrrrr r!Ztwisted.internet.protocolr"rZtwisted.loggerr#Ztwisted.python.failurer$Ztwisted.web._responsesr%Ztwisted.web.clientr&r'r(r)Ztwisted.web.http_headersr*Ztwisted.web.iwebr+r,Ztwisted.web.resourcer-Ztwisted.web.serverr.Protocolr/ ExceptionrUrXrerrrirnrqrrrregisterrrrrrrrrrrrr5r5r5r7sx     0        + EP K         ,