o i{.@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).0xrrXC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\daphne/ws_protocol.py Gsz/WebSocketProtocol.onConnect..,r ) typepathraw_path root_pathheadersZ query_stringclientserver subprotocols)3factory server_classrZprotocol_connectedrequestZprotocol_to_acceptrtime socket_opened last_pingZ clean_headersritemsencodelowerrappend transportZgetPeerZgetHosthasattrstrrr client_addrZ server_addrZproxy_forwarded_address_headerrdictZproxy_forwarded_port_headerZproxy_forwarded_proto_headerZ client_schemedecodesplitrrZ maybeDeferredZcreate_applicationZ_raw_query_stringZapplication_deferredZ addCallbackapplicationCreateWorkedZ addErrbackapplicationCreateFailed Exceptionloggererror traceback format_excZDeferredhandshake_deferred)selfr!namevalueZpeerrrheaderrrr onConnects~              zWebSocketProtocol.onConnectcCsH||_|jddi|jdd|jj|jrdt|jndddS)zk Called when the background thread has successfully made the application instance. rzwebsocket.connectr Z connecting%s:%sNrr)application_queue put_nowaitr log_actionr!rr,tuple)r8r?rrrr0hsz)WebSocketProtocol.applicationCreateWorkedcCst||S)z9 Called when application creation fails. N)r3r4)r8Zfailurerrrr1|s z)WebSocketProtocol.applicationCreateFailedcCs@td|j|jdd|jj|jrdt|jndddS)Nz!WebSocket %s open and establishedr connectedr=r>)r3debugr,rrAr!rrBr8rrronOpenszWebSocketProtocol.onOpencCsf|jr td|jdStd|jt|_|r%|jd|ddS|jd|dddS)NzMuting incoming frame on %szWebSocket incoming frame on %szwebsocket.receive)rbytesutf8)rtext) mutedr3rDr,r"r$r?r@r.)r8payloadZisBinaryrrr 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 %sr?zwebsocket.disconnect)rcoder disconnectedr=Nr>) rprotocol_disconnectedr3rDr,rJr*r?r@rAr!rrB)r8ZwasCleanrMreasonrrronCloses 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.closerMrMzwebsocket.sendz4Socket has not been accepted, so cannot send over itrGrIzMGot invalid WebSocket reply message on %s - contains both bytes and text keysTF) ValueError serverAcceptgetstateSTATE_CONNECTING serverReject serverClose serverSend)r8messagerrr handle_replys.         zWebSocketProtocol.handle_replycCs2t|dr|jtddddS|jdddS)zF Called by the server when our application tracebacks r7izInternal server errorrMrPirSN)r*r7errbackrZsendCloseFrame)r8 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)r7callbackr3rDr,)r8rRrrrrUs 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 deniedr^z$WebSocket %s rejected by applicationr Zrejectedr=Nr>) r7r_rrrOr3rDr,rAr!rrBrErrrrYs  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 %srHN)rWrXrUr3rDr,Z sendMessager&)r8contentbinaryrrrr[s zWebSocketProtocol.serverSendcCs |durdn|}|j|ddS)zA Server-side channel message to close the socket NirS)Z sendClose)r8rMrrrrZszWebSocketProtocol.serverClosecCst|jS)z> Returns the time since the socket was opened N)r"r#rErrrdurationszWebSocketProtocol.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) rerZwebsocket_timeoutrZrWrXZwebsocket_connect_timeoutrYZ STATE_OPENr"r$Z ping_intervalZ _sendAutoPingrErrrcheck_timeoutss    z WebSocketProtocol.check_timeoutscCs tt|SN)hashidrErrr__hash__*s zWebSocketProtocol.__hash__cCst|t|kSrg)ri)r8otherrrr__eq__-szWebSocketProtocol.__eq__cCsd|jd|jdS)Nz)r,rrErrr__repr__0szWebSocketProtocol.__repr__rg)F)__name__ __module__ __qualname____doc__Zapplication_typerJr<r0r1rFrLrQr]rarUrYr[rZrerfrjrlrnrrrrr 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|dSrg)r r__init__)r8r argskwargsrrrrt=szWebSocketFactory.__init__cs<z t|}||_|WStytdtw)zh Builds protocol instances. We use this to inject the factory object into the protocol. zCannot build protocol: %sN)super buildProtocolrr2r3r4r5r6)r8addrprotocol __class__rrrxAs  zWebSocketFactory.buildProtocol) rorprqrrr rzrtrx __classcell__rrr{rrs4s rs)loggingr"r5 urllib.parserZautobahn.twisted.websocketrrrZtwisted.internetrutilsr getLoggerror3r rsrrrrs    $