o 4ήc4@sRddlZddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlZddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd l m!Z!m"Z"m#Z#m$Z$dd l%m&Z&ej'dkr|ddl(m)Z)nddlm)Z)erddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2GdddZ3GdddeZ4dS)N) TYPE_CHECKINGAnyListOptionalSequenceTupleUnioncast)unquote)Headers)ConnectionClosed)ServerPerMessageDeflateFactory) HTTPResponse)WebSocketServerProtocol) Subprotocol)Config)TRACE_LOG_LEVEL)get_local_addrget_path_with_query_stringget_remote_addris_ssl) ServerState))Literal) ASGISendEventWebSocketAcceptEventWebSocketCloseEventWebSocketConnectEventWebSocketDisconnectEventWebSocketReceiveEventWebSocketScopeWebSocketSendEventc@sBeZdZdZdeddfddZdeddfddZdefd d ZdS) ServerFwsreturnNcCdSNselfr$r(r(S/tmp/pip-target-vg8gfxp4/lib/python/uvicorn/protocols/websockets/websockets_impl.pyregister/zServer.registercCr&r'r(r)r(r(r+ unregister2r-zServer.unregistercCs|j Sr')closingr*r(r(r+ is_serving5szServer.is_serving) __name__ __module__ __qualname__r/rr,r.boolr1r(r(r(r+r#,s r#cseZdZUeeeefed< d)dedede e j ffdd Z de j d dffd d Zd e ed dffd d Zd*ddZde jd dfddZdeded e efddZdede eed e efddZd*ddZdeded efddZd*d d!Zd+d$d%Zd e d&fd'd(Z!Z"S),WebSocketProtocol extra_headersNconfig server_state_loopc s|js|||_|j|_|pt|_|j|_|j |_ |j |_ d|_ d|_ d|_ d|_d|_t|_t|_t|_d|_d|_d|_d|_d|_t|_g}|jjr^|ttj|j |j|jj!|jj"|jj#|t$%ddd|_&dd|j'D|_(dS)NFz uvicorn.error) ws_handler ws_servermax_size ping_interval ping_timeout extensionsloggercS$g|]\}}|d|dfqS)latin-1decode.0namevaluer(r(r+ oz.WebSocketProtocol.__init__..))loadedloadr8 loaded_appappasyncioget_event_looploop root_path connectionstasks transportserverclientschemescopeEventhandshake_started_eventhandshake_completed_event closed_eventinitial_response connect_sent lost_connection_before_handshakeaccepted_subprotocoltransfer_data_taskr#r<ws_per_message_deflateappendr super__init__r; ws_max_sizews_ping_intervalws_ping_timeoutlogging getLogger server_headerdefault_headersr7)r*r8r9r:r@ __class__r(r+rg<sL      zWebSocketProtocol.__init__rVr%csx|j|||_t||_t||_t|rdnd|_|j t r4|jr*d|jnd}|j t d|t |dS)Nwssr$%s:%d - z%sWebSocket connection made)rTaddrVrrWrrXrrYrA isEnabledForrlogrfconnection_made)r*rVprefixror(r+rwts    z!WebSocketProtocol.connection_madeexccsz|j||jtr|jrd|jnd}|jtd||j |_ |j t ||dur;|j dSdS)Nrrrsz%sWebSocket connection lost)rTremoverArurrXrvr]is_setrasetrfconnection_lostrVclose)r*ryrxror(r+r}s     z!WebSocketProtocol.connection_lostcCsd|j_|jdS)NT)r<r/rVr~r0r(r(r+shutdownszWebSocketProtocol.shutdowntaskcCs|j|dSr')rUdiscard)r*rr(r(r+on_task_completesz"WebSocketProtocol.on_task_completepathheadersc s|d\}}}tjj|g}|dD]}|dd|dDqdd|D}d|j j dd d |j |j |j |jt||d |d ||d |_|j|} | |j|j| |jId H|jS)a( This hook is called to determine if the websocket should return an HTTP response and close. Our behavior here is to start the ASGI application, and then wait for either `accept` or `close` in order to determine if we should close the connection. ?zSec-WebSocket-ProtocolcSsg|]}|qSr()strip)rGtokenr(r(r+rJsz5WebSocketProtocol.process_request..,cSrB)ascii)encoderFr(r(r+rJrK websocketz2.3)version spec_versionz1.1r) typeasgi http_versionrYrWrXrSrraw_path query_stringr subprotocolsN) partition websocketslegacy handshake check_requestget_allextendsplit raw_itemsr8 asgi_versionrYrWrXrSr rrZrR create_taskrun_asgiadd_done_callbackrrUrtr\waitr_) r*rr path_portion_rrheader asgi_headersrr(r(r+process_requests6    z!WebSocketProtocol.process_requestavailable_subprotocolscCs|jS)z We override the standard 'process_subprotocol' behavior here so that we return whatever subprotocol is sent in the 'accept' message. )rb)r*rrr(r(r+process_subprotocolsz%WebSocketProtocol.process_subprotocolcCsHd}ddtt|dddd|g}|jd||jdS)NsInternal Server ErrorsMHTTP/1.1 500 Internal Server Error content-type: text/plain; charset=utf-8 scontent-length: rs sconnection: close )strlenrrVwritejoinr\r|)r*msgcontentr(r(r+send_500_responsesz#WebSocketProtocol.send_500_responseprotocolcs |j|jIdHdS)z This is the main handler function for the 'websockets' implementation to call into. We just wait for close then return, and instead allow 'send' and 'receive' events to drive the flow. N)r]r|r^r)r*rrr(r(r+r;s zWebSocketProtocol.ws_handlerc s z||j|j|jIdH}Wn9tyI}z-|jd}|jj||d|j s1| n|j IdH|jWYd}~dSd}~ww|j|j sgd}|j|| |jdS|durd}|j|||j IdH|jdSdS)zr Wrapper around the ASGI callable, handling exceptions and unexpected termination states. NzException in ASGI application )exc_infoz1ASGI callable returned without sending handshake.z4ASGI callable should return None, but returned '%s'.)rOrZ asgi_receive asgi_send BaseExceptionr^r|rAerrorr\r{rr]rrVr~)r*resultryrr(r(r+rs2     zWebSocketProtocol.run_asgimessagerc s|d}|jsz|dkrGtd|}|jd|jdt|jd|_ttt | d|_ d|vr@|j dd |dD|jdS|d krrtd |}|jd |jdt|jtjjgd f|_|j|jdSd}t|||js|jIdH|dkrtd|}| d}| d}|dur|n|}||IdHdS|d krtd |}| dd}| ddpd}|||IdH|jdSd}t||d}t||)Nrzwebsocket.acceptrz%s - "WebSocket %s" [accepted]rX subprotocolrcss(|]\}}|d|dfVqdS)rCNrDrFr(r(r+ s  z.WebSocketProtocol.asgi_send..zwebsocket.closerz%s - "WebSocket %s" 403rzLExpected ASGI message 'websocket.accept' or 'websocket.close', but got '%s'.zwebsocket.sendr"bytestextcodeireasonrszJExpected ASGI message 'websocket.send' or 'websocket.close', but got '%s'.z>Unexpected ASGI message '%s', after sending 'websocket.close'.)r\r{r rAinforZrr_rrgetrbr7rr|http HTTPStatus FORBIDDENr^ RuntimeErrorr]rsendr~) r*r message_typer bytes_data text_datadatarrr(r(r+rsf              zWebSocketProtocol.asgi_send)rrr c s|js d|_ddiS|jIdH|jrdddS|jr%dddSz |IdH}Wn,tyZ}z |j|j j rJdddWYd}~Sd|j dWYd}~Sd}~wwdd i}t |t rj||d <|S||d <|S) NTrzwebsocket.connectzwebsocket.disconnecti)rriizwebsocket.receiverr)r`r]rrar^r{recvr r|r<r/r isinstancer)r*rryrr(r(r+rKs2     zWebSocketProtocol.asgi_receiver')r%N)rrr%N)#r2r3r4rrr__annotations__rrrrPAbstractEventLooprg Transportrw Exceptionr}rTaskrr rrrrrrrrr;rrrr __classcell__r(r(ror+r69s^ 8  ,   Fr6)5rPrrksystypingrrrrrrrr urllib.parser rwebsockets.datastructuresr websockets.exceptionsr (websockets.extensions.permessage_deflater websockets.legacy.serverrwebsockets.serverrwebsockets.typingruvicorn.configruvicorn.loggingruvicorn.protocols.utilsrrrruvicorn.serverr version_infotyping_extensionsrasgiref.typingrrrrrr r!r"r#r6r(r(r(r+s0(            (