U R¹Öh‚<ã@sFddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z ddl Z ddlZ ddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*ddl+m,Z,ddl-m.Z.m/Z/m0Z0m1Z1m2Z2ddl3m4Z4Gdd„dƒZ5Gdd„deƒZ6dS)é)Ú annotationsN)ÚAnyÚLiteralÚOptionalÚSequenceÚcast)Úunquote)ÚHeaders)ÚConnectionClosed)ÚServerExtensionFactory)ÚServerPerMessageDeflateFactory)Ú HTTPResponse)ÚWebSocketServerProtocol)Ú Subprotocol) ÚASGI3ApplicationÚ ASGISendEventÚWebSocketAcceptEventÚWebSocketCloseEventÚWebSocketConnectEventÚWebSocketDisconnectEventÚWebSocketReceiveEventÚWebSocketResponseBodyEventÚWebSocketResponseStartEventÚWebSocketScopeÚWebSocketSendEvent)ÚConfig)ÚTRACE_LOG_LEVEL)ÚClientDisconnectedÚget_local_addrÚget_path_with_query_stringÚget_remote_addrÚis_ssl)Ú ServerStatec@s>eZdZdZdddœdd„Zdddœdd„Zd d œd d „Zd S)ÚServerFrÚNone)ÚwsÚreturncCsdS©N©©Úselfr%r(r(úR./venv/lib/python3.8/site-packages/uvicorn/protocols/websockets/websockets_impl.pyÚregister/szServer.registercCsdSr'r(r)r(r(r+Ú unregister2szServer.unregisterÚbool©r&cCs|j Sr')Úclosing©r*r(r(r+Ú is_serving5szServer.is_servingN)Ú__name__Ú __module__Ú __qualname__r0r,r-r2r(r(r(r+r#,sr#csòeZdZUded<ded<d8dddd d œ‡fd d „ Zd ddœ‡fdd„ Zdddœ‡fdd„ Zddœdd„Zdddœdd„Zdddd œd!d"„Z dd#d$d%œd&d'„Z ddœd(d)„Z d*dd+d,œd-d.„Z ddœd/d0„Z d1dd2œd3d4„Zd5dœd6d7„Z‡ZS)9ÚWebSocketProtocolzlist[tuple[str, str]]Ú extra_headersz+logging.Logger | logging.LoggerAdapter[Any]ÚloggerNrr"zdict[str, Any]z asyncio.AbstractEventLoop | None)ÚconfigÚ server_stateÚ app_stateÚ_loopc s|js| ¡||_tt|jƒ|_|p,t ¡|_ |j |_ ||_ |j |_ |j |_ d|_d|_d|_d|_|t ¡|_t ¡|_t ¡|_d|_d|_d|_d|_tƒ|_g}|jjrÀ| tƒ¡tƒj |j!|j|jj"|jj#|jj$|jj%|t& 'd¡dd|_(dd„|j)Dƒ|_*dS)NFz uvicorn.error)Ú ws_handlerÚ ws_serverZmax_sizeZ max_queueZ ping_intervalZ ping_timeoutÚ extensionsr8cSs$g|]\}}| d¡| d¡f‘qS©úlatin-1©Údecode©Ú.0ÚnameÚvaluer(r(r+Ú rsz.WebSocketProtocol.__init__..)+ZloadedÚloadr9rrZ loaded_appÚappÚasyncioZget_event_loopÚloopÚ root_pathr;Ú connectionsÚtasksÚ transportÚserverÚclientÚschemeÚEventÚhandshake_started_eventÚhandshake_completed_eventÚ closed_eventÚinitial_responseÚ connect_sentÚ lost_connection_before_handshakeÚaccepted_subprotocolr#r>Zws_per_message_deflateÚappendr ÚsuperÚ__init__r=Z ws_max_sizeZ ws_max_queueZws_ping_intervalZws_ping_timeoutÚloggingZ getLoggerZ server_headerZdefault_headersr7)r*r9r:r;r<r?©Ú __class__r(r+r^=sN    ø ÿzWebSocketProtocol.__init__zasyncio.Transportr$)rPr&csx|j |¡||_t|ƒ|_t|ƒ|_t|ƒr2dnd|_|j   t ¡rh|jrTd|jnd}|j   t d|¡t ƒ |¡dS)NZwssr%ú%s:%d - Úz%sWebSocket connection made)rNÚaddrPrrQr rRr!rSr8Ú isEnabledForrÚlogr]Úconnection_made)r*rPÚprefixr`r(r+rgvs    z!WebSocketProtocol.connection_madezException | None)Úexcr&csv|j |¡|j t¡r<|jr(d|jnd}|j td|¡|j ¡ |_ |j  ¡t ƒ  |¡|dkrr|j  ¡dS)Nrbrcz%sWebSocket connection lost)rNÚremover8rerrRrfrVÚis_setrZÚsetr]Úconnection_lostrPÚclose)r*rirhr`r(r+rm…s    z!WebSocketProtocol.connection_lostr/cCs4d|j_|j ¡r| d¡n| ¡|j ¡dS)NTéô)r>r0rVrkZfail_connectionÚsend_500_responserPrnr1r(r(r+Úshutdown’s   zWebSocketProtocol.shutdownzasyncio.Task[None])Útaskr&cCs|j |¡dSr')rOÚdiscard)r*rrr(r(r+Úon_task_completešsz"WebSocketProtocol.on_task_completeÚstrr zHTTPResponse | None)ÚpathÚrequest_headersr&c Ãs | d¡\}}}tjj |¡g}| d¡D]}| dd„| d¡Dƒ¡q,dd„| ¡Dƒ}t |ƒ}|j |} |j   d¡|  d¡} d|j j d d œd |j|j|j|j | | |  d¡|||j ¡d iid œ|_|j | ¡¡} |  |j¡|j | ¡|j ¡IdH|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)rEÚtokenr(r(r+rH¬sz5WebSocketProtocol.process_request..ú,cSs(g|] \}}| d¡|jdddf‘qS)ÚasciiÚsurrogateescape)Úerrors)ÚencoderDr(r(r+rH®sÿr|Z websocketz2.4)ÚversionZ spec_versionz1.1zwebsocket.http.response)ÚtypeZasgiZ http_versionrSrQrRrMrvZraw_pathÚ query_stringÚheadersÚ subprotocolsÚstater?N)Ú partitionÚ websocketsÚlegacyZ handshakeZ check_requestZget_allÚextendÚsplitZ raw_itemsrrMrr9Z asgi_versionrSrQrRr;ÚcopyÚscoperLZ create_taskÚrun_asgiZadd_done_callbackrtrOrdrUÚwaitrX) r*rvrwZ path_portionÚ_r‚r„ÚheaderZ asgi_headersÚ full_pathZ full_raw_pathrrr(r(r+Úprocess_requests> þ  ò  z!WebSocketProtocol.process_requestzSequence[Subprotocol] | NonezSubprotocol | None)rƒÚavailable_subprotocolsr&cCs|jS)zž We override the standard 'process_subprotocol' behavior here so that we return whatever subprotocol is sent in the 'accept' message. )r[)r*rƒr“r(r(r+Úprocess_subprotocolÌsz%WebSocketProtocol.process_subprotocolcCsHd}ddtt|ƒƒ d¡ddd|g}|j d |¡¡|j ¡dS)NsInternal Server ErrorsMHTTP/1.1 500 Internal Server Error content-type: text/plain; charset=utf-8 scontent-length: r|s sconnection: close ó)ruÚlenrrPÚwriteÚjoinrUrl)r*ÚmsgZcontentr(r(r+rpÕsûz#WebSocketProtocol.send_500_responserr)Úprotocolrvr&cÃs|j ¡| ¡IdHdS)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)rVrlZ wait_closed)r*ršrvr(r(r+r=ãs zWebSocketProtocol.ws_handlercÃsz| |j|j|j¡IdH}Wn~tk rF|j ¡|j ¡Yn¾t k rœ|j ¡|j   d¡|j   ¡s~| ¡n|j ¡IdH|j ¡YnhX|j ¡|j   ¡sÒ|j  d¡| ¡|j ¡n2|dk r|j  d|¡|j ¡IdH|j ¡dS)zr Wrapper around the ASGI callable, handling exceptions and unexpected termination states. NzException in ASGI application z1ASGI callable returned without sending handshake.z4ASGI callable should return None, but returned '%s'.)rJrŒÚ asgi_receiveÚ asgi_sendrrWrlrPrnÚ BaseExceptionr8Z exceptionrUrkrprVrŽÚerror)r*Úresultr(r(r+rìs*          zWebSocketProtocol.run_asgir)Úmessager&c Ãsô|d}|j ¡sf|dkrŒtd|ƒ}|j d|jdt|jƒ¡d|_ttt |  d¡ƒ|_ d|kr€|j   dd „|dDƒ¡|j ¡nÖ|d kràtd |ƒ}|j d |jdt|jƒ¡tjjgd f|_|j ¡|j ¡n‚|dkrRtd|ƒ}|j d|jdt|jƒ|d¡t |d¡}dd„|  dg¡Dƒ}||d f|_|j ¡nd}t||ƒ‚nŠ|j ¡sl|jdkrl|j ¡IdHz°|dkrÜtd|ƒ}|  d¡}|  d¡}|dkrÆ|n|}| |¡IdHn`|d kr,td |ƒ}|  dd¡} |  dd¡p d} | | | ¡IdH|j ¡nd}t||ƒ‚Wn*tk rh} z t| ‚W5d} ~ XYnXn„|jdk rà|dkrÎtd|ƒ}|jd |d!} |jdd …| f|_|  d"d#¡sÞ|j ¡nd$}t||ƒ‚nd%}t||ƒ‚dS)&Nrzwebsocket.acceptrz%s - "WebSocket %s" [accepted]rRZ subprotocolrƒcss&|]\}}| d¡| d¡fVqdS)rANrBrDr(r(r+Ú sÿz.WebSocketProtocol.asgi_send..zwebsocket.closerz%s - "WebSocket %s" 403r•zwebsocket.http.response.startrz%s - "WebSocket %s" %dÚstatuscSs$g|]\}}| d¡| d¡f‘qSr@rBrDr(r(r+rH4sz/WebSocketProtocol.asgi_send..zmExpected ASGI message 'websocket.accept', 'websocket.close', or 'websocket.http.response.start' but got '%s'.zwebsocket.sendrÚbytesÚtextÚcodeièÚreasonrczJExpected ASGI message 'websocket.send' or 'websocket.close', but got '%s'.zwebsocket.http.response.bodyréÚbodyZ more_bodyFzBExpected ASGI message 'websocket.http.response.body' but got '%s'.z\Unexpected ASGI message '%s', after sending 'websocket.close' or response already completed.)rUrkrr8ÚinforŒrrXrrÚgetr[r7r‰rlÚhttpZ HTTPStatusZ FORBIDDENrWÚ RuntimeErrorrVrŽÚsendrnr r) r*r Z message_typer¢rƒr™Ú bytes_dataZ text_dataÚdatar¥r¦rir¨r(r(r+rœ s’  ý ü  ý    ü ÿ  ÿ            zWebSocketProtocol.asgi_sendzHWebSocketDisconnectEvent | WebSocketConnectEvent | WebSocketReceiveEventcÃsÂ|jsd|_ddiS|j ¡IdH|jr4dddœS|j ¡rHdddœSz| ¡IdH}WnHtk r¢|j ¡|j j rŠdddœYSd|j p”d|j d œYSXt |tƒr¸d |d œSd |d œS) NTrzwebsocket.connectzwebsocket.disconnectiî)rr¥iíro)rr¥r¦zwebsocket.receive)rr¤)rr£)rYrVrŽrZrWrkZrecvr rlr>r0Z close_codeZ close_reasonÚ isinstanceru)r*r¯r(r(r+r›hs$      zWebSocketProtocol.asgi_receive)N)r3r4r5Ú__annotations__r^rgrmrqrtr’r”rpr=rrœr›Ú __classcell__r(r(r`r+r69s û9 /  _r6)7Z __future__rrKr«r_ÚtypingrrrrrZ urllib.parserr‡Zwebsockets.legacy.handshakeZwebsockets.datastructuresr Zwebsockets.exceptionsr Zwebsockets.extensions.baser Z(websockets.extensions.permessage_deflater Zwebsockets.legacy.serverr Zwebsockets.serverrZwebsockets.typingrZuvicorn._typesrrrrrrrrrrrZuvicorn.configrZuvicorn.loggingrZuvicorn.protocols.utilsrrrr r!Zuvicorn.serverr"r#r6r(r(r(r+Ús*         4