U 9ܟj{.@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]N\}}| d}d|krpqT|dkrt||_qT|j || dfqT|j}|j}t|drt|drt|j|jg|_t|j|jg|_n d|_d|_|jjr4tt|j |jj|jj|jj|j\|_|_g}|j D]2\}}|dkr>dd t| d!d D}q>|j" d|_"t#$|jj%|d t|j" d|j"|j|j |j&|j|j|d |_'|j'dk r|j'(|j)|j'*|j+Wn&t,k rt-.t/0YnXt#1|_2|j2S) Nascii_sdaphne-root-pathlatin1hostportssec-websocket-protocolcSsg|] }|qS)strip).0xrr6/tmp/pip-unpacked-wheel-tk6c4rm9/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 onConnectsx                zWebSocketProtocol.onConnectcCsH||_|jddi|jdd|jj|jr:dt|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. )r3r4)r8Zfailurerrrr1|s z)WebSocketProtocol.applicationCreateFailedcCs@td|j|jdd|jj|jr2dt|jndddS)Nz!WebSocket %s open and establishedr connectedr=r>)r3debugr,rrAr!rrBr8rrronOpenszWebSocketProtocol.onOpencCsd|jrtd|jdStd|jt|_|rH|jd|dn|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|js) rprotocol_disconnectedr3rDr,rJr*r?r@rAr!rrB)r8ZwasCleanrMreasonrrronCloses zWebSocketProtocol.onClosecCsd|krtd|ddkr0||ddn|ddkrh|j|jkrR|q|j|dddn~|ddkr|j|jkrtd |d dr|d drtd |f|d dr||d d |d dr||d ddS)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_replycCs0t|dr |jtdddn |jdddS)zF Called by the server when our application tracebacks r7izInternal server errorrMrOirRN)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,)r8rQrrrrTs zWebSocketProtocol.serverAcceptcCsd|jtddd|`|j|td|j|jdd|j j |jrVdt |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^rrrNr3rDr,rAr!rrBrErrrrXs  zWebSocketProtocol.serverRejectcCsJ|j|jkr|td|j|r4|||n||d|dS)z@ Server-side channel message to send a message. z&Sent WebSocket packet to client for %srHN)rVrWrTr3rDr,Z sendMessager&)r8contentbinaryrrrrZs  zWebSocketProtocol.serverSendcCs |dkr dn|}|j|ddS)zA Server-side channel message to close the socket NirR)Z sendClose)r8rMrrrrYszWebSocketProtocol.serverClosecCst|jS)z> Returns the time since the socket was opened )r"r#rErrrdurationszWebSocketProtocol.durationcCs||jjkr$|jjdkr$||j|jkrJ||jjkr~|n4|j|jkr~t |j |jj kr~| t |_ dS)zK Called periodically to see if we should timeout something rN) rdrZwebsocket_timeoutrYrVrWZwebsocket_connect_timeoutrXZ STATE_OPENr"r$Z ping_intervalZ _sendAutoPingrErrrcheck_timeoutss    z WebSocketProtocol.check_timeoutscCs tt|SN)hashidrErrr__hash__*szWebSocketProtocol.__hash__cCst|t|kSrf)rh)r8otherrrr__eq__-szWebSocketProtocol.__eq__cCsd|jd|jdS)Nz)r,rrErrr__repr__0szWebSocketProtocol.__repr__)N)F)N)__name__ __module__ __qualname____doc__Zapplication_typerJr<r0r1rFrLrPr\r`rTrXrZrYrdrerirkrmrrrrr 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|f||dSrf)r r__init__)r8r argskwargsrrrrs=szWebSocketFactory.__init__csFzt|}||_|WStk r@tdtYnXdS)zh Builds protocol instances. We use this to inject the factory object into the protocol. zCannot build protocol: %sN)super buildProtocolrr2r3r4r5r6)r8addrprotocol __class__rrrwAs zWebSocketFactory.buildProtocol) rnrorprqr ryrsrw __classcell__rrrzrrr4srr)loggingr"r5 urllib.parserZautobahn.twisted.websocketrrrZtwisted.internetrutilsr getLoggerrnr3r rrrrrrs    $