o  jH@s:dZgdZddlmZddlmZmZddlmZm Z m Z m Z ddl m Z ddlmZmZddlmZdd lmZdd lmZdd lmZGd d d eZGdddeZdZddZddZd%ddZd&ddZ GdddeZ!ee!GdddeZ"Gdd d e"Z#Gd!d"d"e Z$eee$Gd#d$d$e%Z&dS)'zS The WebSockets protocol (RFC 6455), provided as a resource which wraps a factory. )WebSocketsResourceIWebSocketsProtocolIWebSocketsResourceWebSocketsProtocolWebSocketsProtocolWrapper)sha1)packunpack) implementer Interface providedBydirectlyProvides)log)Flags FlagConstant)Protocol) IProtocol) IResource) NOT_DONE_YETc@seZdZdZdS) _WSExceptionzQ Internal exception for control flow inside the WebSockets frame parser. N)__name__ __module__ __qualname____doc__rrV/var/www/html/LicenciasMiG/venv/lib/python3.10/site-packages/spyne/util/_twisted_ws.pyrsrc@s@eZdZdZedZedZedZedZedZ edZ dS) CONTROLSz# Control frame specifiers. r N) rrrrrCONTINUETEXTBINARYCLOSEPINGPONGrrrrr&s rz$258EAFA5-E914-47DA-95CA-C5AB0DC85B11cCstd|tfdS)z Create an B{accept} response for a given key. @type key: C{str} @param key: The key to respond to. @rtype: C{str} @return: An encoded response. %s%sbase64)r_WS_GUIDdigestencodestrip)keyrrr _makeAccept8s r/cCsNdd|D}t|}t|D]\}}tt|||dA||<qd|S)a Mask or unmask a buffer of bytes with a masking key. @type buf: C{str} @param buf: A buffer of bytes. @type key: C{str} @param key: The masking key. Must be exactly four bytes. @rtype: C{str} @return: A masked buffer of bytes. cSsg|]}t|qSr)ord).0irrr Ssz_mask..)list enumeratechrr0join)bufr.r2charrrr_maskFs   r<Nc Cst|}|dur d}nd}|dkrdt|dBtd|f}n|dkr1dt|d Btd |f}nt||B}|rQ}~>Hrz%s%s%s)lenr8rvaluer<) r:opcodefinmask bufferLength lengthMasklengthheaderframerrr _makeFrame[s" rMTc csFd}d|} t||dkrnt||}|d@r"td||d@}|d@}zt|}Wn ty>td |wt||d }|d@}|sS|rStd |d M}d} |d kr|t||dkrfn||d|d}td|d}| d7} n"|d krt||dkrn~||d|d}td|d}| d7} |rt||| dkrn\||| || d} | d7} t||| |krnA||| || |} |rt| | } |tj krt| dkrtd| ddd| ddf} nd} || t |fV|| |7}q t||kr||dg|dd<dSg|dd<dS)z Parse frames in a highly compliant manner. @param frameBuffer: A buffer of bytes. @type frameBuffer: C{list} @param needMask: If C{True}, refuse any frame which is not masked. @type needMask: C{bool} rr5TrpzReserved flag in frame (%d)r=zUnknown opcode %d in framerzReceived data not maskedr>rAr4rBr!r?rN)izNo reason given) r9rCr0rr lookupByValue ValueErrorr r<r%bool) frameBufferneedMaskstartpayloadrKrFrErJmaskedoffsetr.datarrr _parseFramessl         $ RrZc@s(eZdZdZddZddZddZdS) rzR A protocol which understands the WebSockets interface. @since: 13.1 cCdS)z Send a frame. NrrErYrFrrr sendFramezIWebSocketsProtocol.sendFramecCr[)z4 Callback when a frame is received. Nrr\rrr frameReceivedr^z!IWebSocketsProtocol.frameReceivedcCr[)zC Close the connection sending a close frame first. NrrrrrloseConnectionr^z"IWebSocketsProtocol.loseConnectionN)rrrrr]r_r`rrrrrs  rc@sHeZdZdZdZdZddZddZdd Zd d Z d d Z ddZ dS)rz @since: 13.1 FNcCstd|jg|_dS)zH Log the new connection and initialize the buffer list. zOpening connection with %sN)rmsg transportgetPeer_bufferselfrrrconnectionMades z!WebSocketsProtocol.connectionMadecCst|jD]C}|\}}}|tjtjtjfvr||||q|tjkr8|\}}t d||f|j dS|tj krH|j t|tjdqdS)zX Find frames in incoming data and pass them to the underlying protocol. zClosing connection: %r (%d)NT)rZrdrr"r#r$r_r%rrarbr`r&writerMr')rfrLrErYrFreasontextrrrrZs    zWebSocketsProtocol._parseFramescCst)z( Callback to implement. )NotImplementedError)rfrErYrFrrrr_0sz WebSocketsProtocol.frameReceivedcCst|||}|j|dS)zA Build a frame packet and send it over the wire. N)rMrbrh)rfrErYrFpacketrrrr]7s zWebSocketsProtocol.sendFramecCsB|j|z|WdSty t|jYdSw)zI Append the data to the buffer list and parse the whole. N)rdappendrZrrerrrbr`rfrYrrr dataReceived?s  zWebSocketsProtocol.dataReceivedcCs8|jstdtjd}|j|d|_|jdSdS)aw Close the connection. This includes telling the other side we're closing the connection. If the other side didn't signal that the connection is being closed, then we might not see their last message, but since their last message should, according to the spec, be a simple acknowledgement, it shouldn't be a problem. r5TN)_disconnectingrMrr%rbrhr`)rfrLrrrr`Ls   z!WebSocketsProtocol.loseConnection) rrrrrqrdrgrZr_r]rpr`rrrrrs rc@sVeZdZdZejfddZddZddZdd Z d d Z d d Z ddZ ddZ dS)rz{ A protocol wrapper which provides L{IWebSocketsProtocol} by making messages as data frames. @since: 13.1 cCs||_||_dSN)wrappedProtocol defaultOpcode)rfrsrtrrr__init__is z"WebSocketsProtocolWrapper.__init__cCs*t|t|t|||j|dS)z Upon connection, provides the transport interface, and forwards ourself as the transport to C{self.wrappedProtocol}. N)r r rmakeConnectionrs)rfrbrrrrvns z(WebSocketsProtocolWrapper.makeConnectioncCst|g|_dS)z2 Initialize the list of messages. N)rrg _messagesrerrrrgxs  z(WebSocketsProtocolWrapper.connectionMadecCs||j|ddS)zS Write to the websocket protocol, transforming C{data} in a frame. TN)r]rtrorrrrhszWebSocketsProtocolWrapper.writecCs|D]}||qdS)z> Send all chunks from C{data} using C{write}. N)rh)rfrYchunkrrr writeSequences z'WebSocketsProtocolWrapper.writeSequencecCs t|j|S)zT Forward all non-local attributes and methods to C{self.transport}. )getattrrb)rfnamerrr __getattr__s z%WebSocketsProtocolWrapper.__getattr__cCs>|j||rd|j}g|jdd<|j|dSdS)z FOr each frame received, accumulate the data (ignoring the opcode), and forwarding the messages if C{fin} is set. r5N)rwrmr9rsrp)rfrErYrFcontentrrrr_s  z'WebSocketsProtocolWrapper.frameReceivedcCs|j|dS)zG Forward C{connectionLost} to C{self.wrappedProtocol}. N)rsconnectionLost)rfrirrrr~sz(WebSocketsProtocolWrapper.connectionLostN)rrrrrr#rurvrgrhryr|r_r~rrrrras  rc@seZdZdZddZdS)rz2 A WebSockets resource. @since: 13.1 cCr[)a Build a protocol instance for the given protocol options and request. The returned protocol is plugged to the HTTP transport, and the returned protocol name, if specified, is used as I{Sec-WebSocket-Protocol} value. If the protocol provides L{IWebSocketsProtocol}, it will be connected directly, otherwise it will be wrapped by L{WebSocketsProtocolWrapper}. @param protocolNames: The asked protocols from the client. @type protocolNames: C{list} of C{str} @param request: The connecting client request. @type request: L{IRequest} @return: A tuple of (protocol, matched protocol name or C{None}). @rtype: C{tuple} Nr) protocolNamesrequestrrrlookupProtocolr^z"IWebSocketsResource.lookupProtocolN)rrrrrrrrrrs rc@s<eZdZdZdZddZddZddZd d Zd d Z d S)ra. A resource for serving a protocol through WebSockets. This class wraps a factory and connects it to WebSockets clients. Each connecting client will be connected to a new protocol of the factory. Due to unresolved questions of logistics, this resource cannot have children. @param factory: The factory producing either L{IWebSocketsProtocol} or L{IProtocol} providers, which will be used by the default C{lookupProtocol} implementation. @type factory: L{twisted.internet.protocol.Factory} @since: 13.1 TcCs ||_dSrr)_factory)rffactoryrrrrus zWebSocketsResource.__init__cCtd)z Reject attempts to retrieve a child resource. All path segments beyond the one which refers to this resource are handled by the WebSocket connection. z5Cannot get IResource children from WebSocketsResource RuntimeError)rfr{rrrrgetChildWithDefaultz&WebSocketsResource.getChildWithDefaultcCr)z Reject attempts to add a child resource to this resource. The WebSocket connection handles all path segments beneath this resource, so L{IResource} children can never be found. z6Cannot put IResource children under WebSocketsResourcer)rfpathchildrrrputChildrzWebSocketsResource.putChildcCs|j|j}|dfS)a  Build a protocol instance for the given protocol names and request. This default implementation ignores the protocol names and just return a protocol instance built by C{self._factory}. @param protocolNames: The asked protocols from the client. @type protocolNames: C{list} of C{str} @param request: The connecting client request. @type request: L{Request} @return: A tuple of (protocol, C{None}). @rtype: C{tuple} N)r buildProtocolrbrc)rfrrprotocolrrrrsz!WebSocketsResource.lookupProtocolc Csd|_d}|jdkrd}td|j|d}|dus!d|vr+d}td|d|d }|dus:d |vrDd}td |d |d }|durWd}td |d |d}|dkrpd}|ddtd|d|ry|ddS|jd}| ||\}} |s|ddS|d|dd|d d|dt || r|d| | d|j d} |_ t |st|}|r|| j_n|| _|| tS)a] Render a request. We're not actually rendering a request. We are secretly going to handle a WebSockets connection instead. @param request: The connecting client request. @type request: L{Request} @return: a string if the request fails, otherwise C{NOT_DONE_YET}. NFGETTzrequest.methodUpgrade websocketzrequest.getHeader("Upgrade") Connectionupgradezrequest.getHeader("Connection")zSec-WebSocket-Keyz&request.getHeader("Sec-WebSocket-Key")zSec-WebSocket-Version13z*request.getHeader("Sec-WebSocket-Version")ir5zSec-WebSocket-Protocolie WebSocketzSec-WebSocket-Accept)defaultContentTypemethodprint getHeaderlower setHeadersetResponseCoderequestHeaders getRawHeadersrr/rhrbrr risSecurerrsrvr) rfrfailedr connectionr.versionaskedProtocolsr protocolNamerbrrrrenders^                  zWebSocketsResource.renderN) rrrrisLeafrurrrrrrrrrs  rrr)T)'r__all__hashlibrstructrr zope.interfacer r r r twisted.pythonrtwisted.python.constantsrrtwisted.internet.protocolrtwisted.internet.interfacesrtwisted.web.resourcertwisted.web.serverr Exceptionrrr*r/r<rMrZrrrrobjectrrrrrs2        *gZI