o >h{.@s~ddlZddlZddlZddlmZddlmZmZmZddl m Z ddl m Z e eZGdddeZGd d d eZdS) N)unquote)ConnectionDenyWebSocketServerFactoryWebSocketServerProtocol)defer)parse_x_forwarded_forc@seZdZdZdZdZddZddZdd Zd d Z d d Z ddZ ddZ ddZ d'ddZddZd(ddZd'ddZddZdd Zd!d"Zd#d$Zd%d&ZdS))WebSocketProtocolzj Protocol which supports WebSockets and forwards incoming messages to the websocket channels. websocketFcCs|jj|_|j|||_d|_|jj|_t|_t|_ zg|_ |j D]'\}}| d}d|vr7q)|dkrCt||_q)|j || dfq)|j}|j}t|drxt|drxt|j|jg|_t|j|jg|_nd|_d|_|jjrtt|j |jj|jj|jj|j\|_|_g}|j D]\}}|dkrdd t| d!d D}q|j" d|_"t#$|jj%|d t|j" d|j"|j|j |j&|j|j|d |_'|j'dur|j'(|j)|j'*|j+Wnt,yt-.t/0wt#1|_2|j2S) Nascii_sdaphne-root-pathlatin1hostportssec-websocket-protocolcSsg|]}|qS)strip).0xrrr/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/daphne/ws_protocol.py Gsz/WebSocketProtocol.onConnect..,r ) typepathraw_path root_pathheaders query_stringclientserver subprotocols)3factory server_classrprotocol_connectedrequestprotocol_to_acceptrtime socket_opened last_ping clean_headersritemsencodelowerrappend transportgetPeergetHosthasattrstrrr client_addr server_addrproxy_forwarded_address_headerrdictproxy_forwarded_port_headerproxy_forwarded_proto_header client_schemedecodesplitrr maybeDeferredcreate_application_raw_query_stringapplication_deferred addCallbackapplicationCreateWorked addErrbackapplicationCreateFailed Exceptionloggererror traceback format_excDeferredhandshake_deferred)selfr#namevaluepeerrrheaderrrr onConnects~              zWebSocketProtocol.onConnectcCsH||_|jddi|jdd|jj|jrdt|jndddS)zk Called when the background thread has successfully made the application instance. rzwebsocket.connectr connecting%s:%sNrr)application_queue put_nowaitr log_actionr#rr2tuple)rJrSrrrr@hsz)WebSocketProtocol.applicationCreateWorkedcCst||S)z9 Called when application creation fails. )rDrE)rJfailurerrrrB|s z)WebSocketProtocol.applicationCreateFailedcCs@td|j|jdd|jj|jrdt|jndddS)Nz!WebSocket %s open and establishedr connectedrQrR)rDdebugr2rrUr#rrVrJrrronOpenszWebSocketProtocol.onOpencCsf|jr td|jdStd|jt|_|r%|jd|ddS|jd|dddS)NzMuting incoming frame on %szWebSocket incoming frame on %szwebsocket.receive)rbytesutf8)rtext) mutedrDrYr2r%r'rSrTr9)rJpayloadisBinaryrrr onMessages zWebSocketProtocol.onMessagecCsn|j|td|j|jst|dr|jd|d|j dd|j j |jr0dt |jndd dS) z8 Called when Twisted closes the socket. zWebSocket closed for %srSzwebsocket.disconnect)rcoder disconnectedrQNrR) rprotocol_disconnectedrDrYr2r_r0rSrTrUr#rrV)rJwasCleanrcreasonrrronCloses zWebSocketProtocol.onClosecCsd|vrtd|ddkr||dddS|ddkr7|j|jkr+|dS|j|ddddS|ddkrx|j|jkrGtd |d drZ|d drZtd |f|d drh||d d |d drz||d ddSdSdS)NrzMessage has no type definedzwebsocket.accept subprotocolzwebsocket.closercrczwebsocket.sendz4Socket has not been accepted, so cannot send over itr\r^zMGot invalid WebSocket reply message on %s - contains both bytes and text keysTF) ValueError serverAcceptgetstateSTATE_CONNECTING serverReject serverClose serverSend)rJmessagerrr handle_replys.         zWebSocketProtocol.handle_replycCs2t|dr|jtddddS|jdddS)zF Called by the server when our application tracebacks rIizInternal server errorrcrgirjN)r0rIerrbackrsendCloseFrame)rJ exceptionrrrhandle_exceptions  z"WebSocketProtocol.handle_exceptionNcCs"|j||`td|jdS)zO Called when we get a message saying to accept the connection. z$WebSocket %s accepted by applicationN)rIcallbackrDrYr2)rJrirrrrls zWebSocketProtocol.serverAcceptcCsd|jtddd|`|j|td|j|jdd|j j |jr+dt |jndd dS) zO Called when we get a message saying to reject the connection. iz Access deniedruz$WebSocket %s rejected by applicationr rejectedrQNrR) rIrvrrrerDrYr2rUr#rrVrZrrrrps  zWebSocketProtocol.serverRejectcCsL|j|jkr |td|j|r|||dS||d|dS)z@ Server-side channel message to send a message. z&Sent WebSocket packet to client for %sr]N)rnrorlrDrYr2 sendMessager*)rJcontentbinaryrrrrrs zWebSocketProtocol.serverSendcCs |durdn|}|j|ddS)zA Server-side channel message to close the socket Nirj) sendClose)rJrcrrrrqszWebSocketProtocol.serverClosecCst|jS)z> Returns the time since the socket was opened )r%r&rZrrrdurationszWebSocketProtocol.durationcCs||jjkr|jjdkr||j|jkr(||jjkr&|dSdS|j|jkrDt |j |jj krF| t |_ dSdSdS)zK Called periodically to see if we should timeout something rN) rrwebsocket_timeoutrqrnrowebsocket_connect_timeoutrp STATE_OPENr%r' ping_interval _sendAutoPingrZrrrcheck_timeoutss    z WebSocketProtocol.check_timeoutscCs tt|SN)hashidrZrrr__hash__*s zWebSocketProtocol.__hash__cCst|t|kSr)r)rJotherrrr__eq__-szWebSocketProtocol.__eq__cCsd|jd|jdS)Nz)r2rrZrrr__repr__0szWebSocketProtocol.__repr__r)F)__name__ __module__ __qualname____doc__application_typer_rOr@rBr[rbrhrtryrlrprrrqrrrrrrrrrr s*K      r cs,eZdZdZeZddZfddZZS)WebSocketFactoryz Factory subclass that remembers what the "main" factory is, so WebSocket protocols can access it to get reply ID info. cOs"||_tj|g|Ri|dSr)r!r__init__)rJr!argskwargsrrrr=szWebSocketFactory.__init__cs<z t|}||_|WStytdtw)zh Builds protocol instances. We use this to inject the factory object into the protocol. zCannot build protocol: %s)super buildProtocolr rCrDrErFrG)rJaddrprotocol __class__rrrAs  zWebSocketFactory.buildProtocol) rrrrr rrr __classcell__rrrrr4s r)loggingr%rF urllib.parserautobahn.twisted.websocketrrrtwisted.internetrutilsr getLoggerrrDr rrrrrs    $