i ddlZddlZddlZddlZddlmZddlmZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZmZn#e$rddlmZddlmZYnwxYwddlmZddlmZdd lmZdd lmZd D] Z eeiid gZn #e$rdZYwxYwejejejhZd Z ddgZ!dZ"e#e$e%dde$e%ddze$e%ddze$e%ddzZ&Gdde'Z(GddZ)GddZ*Gdde'Z+Gdde'Z,Gd d!e-Z.Gd"d#e*Z/dS)$N)Random)error)md5sha1)r)sha) semaphore)wsgi)socket) get_errno)zwsaccel.utf8validatorzautobahn.utf8validatori WebSocketWSGI WebSockets$258EAFA5-E914-47DA-95CA-C5AB0DC85B11iii iic eZdZdfd ZxZS) BadRequest400 Bad RequestNctt|||_||_||_dSN)super Exception__init__statusbodyheaders)selfrrr __class__s =/usr/local/lib/python3.11/dist-packages/eventlet/websocket.pyrzBadRequest.__init__9s8 i'')))   )rNN__name__ __module__ __qualname__r __classcell__rs@rrr8s=rrcfeZdZdZefdZe d dZdZdZ dZ d Z d Z d Z d ZdS)r aWraps a websocket handler function in a WSGI application. Use it like this:: @websocket.WebSocketWSGI def my_handler(ws): from_browser = ws.wait() ws.send("from server") The single argument to the function will be an instance of :class:`WebSocket`. To close the socket, simply return from the function. Note that the server will log the websocket request at the time of closure. An optional argument max_frame_length can be given, which will set the maximum incoming *uncompressed* payload length of a frame. By default, this is set to 8MiB. Note that excessive values here might create a DOS attack vector. cZ||_d|_d|_g|_d|_||_dS)NT)handlerprotocol_versionsupport_legacy_versionssupported_protocolsorigin_checkermax_frame_length)rr(r-s rrzWebSocketWSGI.__init__Us7 $'+$#% " 0rNFc8fd}||S||S)NcL|}|_|_r|_|Sr)r*r,r+)r(instclsr,r*r+s r decoratorz+WebSocketWSGI.configured..decoratorcs53w<(Kr)r1r(r+r,r*r2s` ``` r configuredzWebSocketWSGI.configured]sK          ? y!!!rc d|dddD}d|vr,|dddks|dd ggS d |vr||}n+|jr||}nt nC#t $r6}|j}|jpd }|j pg}||d g|z|gcYd}~Sd}~wwxYwd |vr|d  | |n.#t$r!}t|tvrYd}~nd}~wwxYw|d d tj_gS)Nc6g|]}|Sr3strip).0parts r z*WebSocketWSGI.__call__..os6!O!O!O JJLL!O!O!OrHTTP_CONNECTIONr ,upgrade HTTP_UPGRADE websocketr) ConnectioncloseHTTP_SEC_WEBSOCKET_VERSIONrzeventlet.set_idleT)getlowersplit_handle_hybi_requestr*_handle_legacy_requestrrrrr(OSErrorr ACCEPTABLE_CLIENT_ERRORS_send_closing_framer WSGI_LOCALalready_handled) renvironstart_responsehttp_connection_partswserrrs r__call__zWebSocketWSGI.__call__ns!O!O $5r::@@BBHHMM!O!O!O222 NB//5577;FF N,/F.G H H HI +w66..w77- #0099 ll"   XF6=SDio2G N636@ B B B6MMMMMM   ' ) ) (G' ( * * *  LL       ||#;;;<;;;;  t$$$+/' s7AC D +DD D $D:: E%E  E%cd|vr|d}nd|vr |d}ntdd|vrd|_d|vrtnd|_|jdkr||d}||d}d|d _|d d}tjd |||z}t| }d }| d d krd}|d| d| d| d} | d} | | d| zz } |jdkrFd| d zdz| zdz} n|jdkrud| d zdz| dd zdz| zdz|z} ntd|| t|||jS)Neventlet.inputgunicorn.socket8No eventlet.input or gunicorn.socket present in environ.HTTP_SEC_WEBSOCKET_KEY1LHTTP_SEC_WEBSOCKET_KEY2Kz wsgi.inputz>IIrQzwsgi.url_schemehttpswssz:// HTTP_HOST SCRIPT_NAME PATH_INFO QUERY_STRING?sgHTTP/1.1 101 Web Socket Protocol Handshake Upgrade: WebSocket Connection: Upgrade WebSocket-Origin: HTTP_ORIGINs WebSocket-Location:  sjHTTP/1.1 101 WebSocket Protocol Handshake Upgrade: WebSocket Connection: Upgrade Sec-WebSocket-Origin: s Sec-WebSocket-Protocol: HTTP_SEC_WEBSOCKET_PROTOCOLdefaults Sec-WebSocket-Location: z#Unknown WebSocket protocol version.) get_socketrr)r_extract_numbercontent_lengthreadstructpackrdigestrDencode ValueErrorsendallr) rrNsockkey1key2key3keyresponseschemelocationqshandshake_replys rrHz$WebSocketWSGI._handle_legacy_requests w & &+,7799DD ' ) ),-DDVWW W $ / /$&D !(77 ll"8%'D !  B & &''0I(JKKD''0I(JKKD45GL ! 0<(--a00D+eT40047C3xx((H ;;( ) )W 4 4F FF KK $ $ $ KK & & & KK $ $ $   [[ ( ( > b H  B & &&)0 M(B(B(I(I(K(KLO((+3//*;*; <?J J O "b ( (*-4KK ,F,F,M,M,O,OPS,,  99EELLNN O R, , /7oo.?.? @ C ## OBCC C _%%%w(=>>>rc|dSi}|dD]>}|d}i}|ddD]}|d}t|dkr0d||d<Z|dd||d<||dg|@|S)Nr=;=Tr")rFlenr8rE setdefaultappend)rheaderresextpartsconfigr:key_vals r_parse_extension_headerz%WebSocketWSGI._parse_extension_headers< >4<<$$ H HCIIcNNEFabb  _ _**S//w<<1$$9=F71:++--3355669@9I9I9K9K9Q9QRU9V9V9\9\9^9^F71:++--335566 NN58>>++1133R 8 8 ? ? G G G G rcV|sdS|d}|dS|D]}|dd|ddd}ttjd}|d}|F|dur||d<n>d]]$$SZZ\\2222$$c%jjj&A%I%I%K%KLLLL LLI.. / / / /zz%   rcd|vr|d}nd|vr |d}ntd|d}|dvrtddgt||_d |vrt|d |jd kr|d d nd }|j9||d|std|dd}d}|r/d|dDD]}||jvr|}n|d }tj t| tz} dddd| zg} |r*| d| zi} ||d} || } | | | d<|| }|| d|z|d| dzt+|||j|| |jS)NrUrVrWrC)813z426 Upgrade Required)zSec-WebSocket-Versionz8, 13)rrHTTP_SEC_WEBSOCKET_KEYrdr\HTTP_SEC_WEBSOCKET_ORIGINr r_z 403 Forbidden)rrfc3>K|]}|VdSrr7)r9is r z5WebSocketWSGI._handle_hybi_request..6s*>>Aaggii>>>>>>rr=s HTTP/1.1 101 Switching ProtocolssUpgrade: websocketsConnection: UpgradesSec-WebSocket-Accept: sSec-WebSocket-Protocol: HTTP_SEC_WEBSOCKET_EXTENSIONSrsSec-WebSocket-Extensions: s re)protocolrr-)rhrrrr)rDr,rFr+base64 b64encoderro PROTOCOL_GUIDrnrrrrrqrRFC6455WebSocketr-)rrNrr hybi_versionorigin protocolsnegotiated_protocolprvrwr{rrr formatted_exts rrGz"WebSocketWSGI._handle_hybi_requests w & &+,7799DD ' ) ),-DDVWW W;< } , ,$:&H%IKKK K #L 1 1 #7 2 2,,  %**[[4b 9 9 90255   *&&w{{;'?'?HH 9 8888KK =tDD "  >>)=)=>>>  000*+'E1./#D )E$F$F$M$M$O$OPP>014x?A  _  " "#>ATA[A[A]A]#] ^ ^ ^11'++>]2^2^__ 44Z@@  6= 2 3556GHH  $  " "#@=#P Q Q Q W\\/22[@AAAgt/D)<+<151FHHH Hrcvd}d}|D]!}|tjvr||z }|dkr|dz }"t||zS)z Utility function which, given a string like 'g98sd 5[]221@1', will return 9852211. Used to parse the Sec-WebSocket-Key headers. r r r~)stringdigitsr)rroutspaceschars rrizWebSocketWSGI._extract_numberUsY   Dv}$$t ! 3xx6!!r)NNNF)r!r"r#__doc__DEFAULT_MAX_FRAME_LENGTHr classmethodr4rSrHrrrrGrir3rrr r @s(2J1111'+"&+0 """[" (((T<?<?<?| $$$L!!! 7H7H7Hr " " " " "rc@eZdZdZd dZdZdZdZdZd d Z d Z d S)ra{A websocket object that handles the details of serialization/deserialization to the socket. The primary way to interact with a :class:`WebSocket` object is to call :meth:`send` and :meth:`wait` in order to pass messages back and forth with the browser. Also available are the following properties: path The path value of the request. This is the same as the WSGI PATH_INFO variable, but more convenient. protocol The value of the Websocket-Protocol header. origin The value of the 'Origin' header. environ The full WSGI environment for this request. rYc|dtj|_d|d|dd|ddz|d|d |_||_|d |_|d |_|d|_ ||_ ||_ d |_ d |_ tj|_t#j|_dS)z :param socket: The eventlet socket :type socket: :class:`eventlet.greenio.GreenSocket` :param environ: The wsgi environment :param version: The WebSocket spec version to follow (default is 76) z wsgi.errorsz-server={shost}/{spath} client={caddr}:{cport}r_r`r ra REMOTE_ADDR REMOTE_PORT)shostspathcaddrcportrdHTTP_WEBSOCKET_PROTOCOLFrN)rDsysstderrlogformat log_contextr rrpathrNversionwebsocket_closed_buf collectionsdeque_msgsr Semaphore _sendlock)rrrrNrs rrzWebSocket.__init__ys ;;}cj99JQQ++k**++mR007;;{B3O3OO++m,,GKK 4N4NR    kk-00  $=>> KK ,,   %  &(( ",..rct|tr|d}n6t|ts!t|}d|zdz}|S)zwPack the message inside ``00`` and ``FF`` As per the dataframing section (5.3) for the websocket spec utf-8) isinstancerrobytesrmessagepackeds r _pack_messagezWebSocket._pack_messagesf gs # # ,nnW--GGGU++ ,'ll))++G7"W, rcvg}d}|j}|r|d}|dkr[|d}|dkrnz||d|dd||dzd}n9|dkr!|ddks Jd |zd |_nt d |z|||_|S) a1 Parses for messages in the buffer *buf*. It is assumed that the buffer contains the start character for a message, but that it may contain only part of the rest of the message. Returns an array of messages, and the buffer remainder that didn't contain any full messages.rrr~rreplaceNz Unexpected closing handshake: %rTz6Don't understand how to parse this type of message: %r)rfindrdecoderrp)rmsgsend_idxbuf frame_types r_parse_messageszWebSocket._parse_messagessi aQJQ((7++b== C' N11'9EEFFF'A+,,'s""1v{{{$F$L{{{(,% !Y\_!_``` a   rc||}|j |j||jdS#|jwxYw)zSend a message to the browser. *message* should be convertable to a string; unicode objects should be encodable as utf-8. Raises socket.error with errno of 32 (broken pipe) if the socket has already been closed by the client.N)rracquirer rqreleasers rsendzWebSocket.sends| ##G,,     % K   ' ' ' N " " $ $ $ $ $DN " " $ $ $ $s A%%Bc"|jsp|jrdS|jd}|dkrdS|xj|z c_|}|j||jp|jS)a"Waits for and deserializes messages. Returns a single message; the oldest not yet processed. If the client has already closed the connection, returns None. This is different from normal socket behavior because the empty string is a valid websocket message.Nir)rrr recvrrextendpopleft)rdeltars rwaitzWebSocket.waits* $$ tK$$T**E||t II II''))D J  d # # #* $z!!###rFc|jdkr?|js: |jdn#t$r|sYnwxYwd|_dSdSdS)z3Sends the closing frame to the client, if required.rYsTN)rrr rqrI)rignore_send_errorss rrKzWebSocket._send_closing_frames <2  d&;   ##K0000   *  %)D ! ! !    s / ??c |d|jdn`#t$rS}|jtjkr4|jd|j |Yd}~nd}~wwxYw|j dS#|j wxYw)cForcibly close the websocket; generally it is preferable to return from the handler method.T {ctx} socket shutdown error: {e}ctxrRN) rKr shutdownrIerrnoENOTCONNrwriterrrB)rrRs rrBzWebSocket.closes  $ $T * * * K  & & & & e e ew%.((AHHTM]abHccddd e K       DK      s,/2B- BA B B- BB--CN)rYF) r!r"r#rrrrrrrKrBr3rrrrds(////0   : % % %$$$( ) ) ) )      rceZdZdS)ConnectionClosedErrorNr!r"r#r3rrrrDrrceZdZfdZxZS)FailedConnectionErrorcht||||_||_dSr)rrrr)rrrrs rrzFailedConnectionError.__init__s/ )))  rr r%s@rr r s8rr ceZdZdS) ProtocolErrorNrr3rrr r rrr ceZdZddddeffd ZGddZdZdZd ZGd d Z e dd Z dZ dZ ddZ ddZdZdZdZddZddZxZS)r NFct|||||_||_||_|pi|_d|_d|_||_ d|_ dSr) rr _iter_framesiteratorclientrr _deflate_enc _deflate_decr-_remote_close_data) rrrrNrrrrr-rs rrzRFC6455WebSocket.__init__sr w000))++    $*   0"&rc"eZdZdZdZddZdS)RFC6455WebSocket.UTF8Decoderctrt|_nd|_tjd}||_dS)Nutf8) utf8validator Utf8Validator validatorcodecsgetincrementaldecoderdecoder)r decoderclasss rrz%RFC6455WebSocket.UTF8Decoder.__init__sG &!.!!>!%!7??L'<>>DLLLrcx|jr|j|jdSr)rresetr )rs rr#z"RFC6455WebSocket.UTF8Decoder.resets:~ '$$&&& L   rFc|jr0|j|\}}}}|std|j||S)NzData is not valid unicode)rvalidaterpr r)rdatafinalvalideocpc_it_is rrz#RFC6455WebSocket.UTF8Decoder.decode!s[~ B(,(?(?(E(E%tS#B$%@AAA<&&tU33 3rNr)r!r"r#rr#rr3rr UTF8DecoderrsF * * * ! ! !  4 4 4 4 4 4rr,cjddSfd}jrdndr |Sj|_jS)Nrctjtjtjjrdndtj S)Nrr)r compressobjZ_DEFAULT_COMPRESSIONDEFLATEDrDrroptionsrsr_makez;RFC6455WebSocket._get_permessage_deflate_enc.._make-sU#D$> %,[[T[2O1I1I6N15&A&A%ABB Brrr)rrDrr)rr4r3s` @r_get_permessage_deflate_encz,RFC6455WebSocket._get_permessage_deflate_enc(s/%%&:;; ?4 B B B B B B ;;t{9338 : : %577N ($)EGG!$ $rcjd|sdSfd}jrdndr |Sj|_jS)Nrc|tjjrdndtj S)Nrr)r decompressobjrDrrr2srr4z;RFC6455WebSocket._get_permessage_deflate_dec.._makeAsK%w{{t{4Q3K3K8P37>(C(C'CDD Drrr)rrDrr)rrsv1r4r3s` @r_get_permessage_deflate_decz,RFC6455WebSocket._get_permessage_deflate_dec<s/%%&:;; ?$?4 D D D D D D ;;t{9338 : : %577N ($)EGG!$ $rcd}t||krR|j|t|z }|st||z}t||kR|S)Nr)rr rr)rnumbytesr&ds r _get_byteszRFC6455WebSocket._get_bytesOso$ii(""   CII!566A .+---!8D $ii(""  rc$eZdZddZddZdZdS)RFC6455WebSocket.MessageNcZ||_g|_d|_||_||_||_dS)NF)r r&finishedopcode decompressorr-)rrCr-r rDs rrz!RFC6455WebSocket.Message.__init__Ys4"DLDI!DM DK ,D $4D ! ! !rFcH||_|j|dSr)rBr&r)rr&r's rpushzRFC6455WebSocket.Message.pushas%!DM I  T " " " " "rcXd|j}|jdzs^|jrW|j|dz|j}|jjr(tdd|j|j r |j ||j }|S)Nrr\z;Incoming compressed frame exceeds length limit of {} bytes.) rr&rCrD decompressr-unconsumed_tailr rr rrB)rr&s rgetvaluez!RFC6455WebSocket.Message.getvaluees88DI&&D;? ut'8 u(33D;N4NPTPeff$4u/U\\]a]rssuuu| @|**4??Kr)NNr)r!r"r#rrFrLr3rrMessager@XsK 5 5 5 5 # # # #     rrMrc|t}t|}dfd|DS)Nrc3bK|])}t||zdzz fV*dS)N)r)r9rr&maskoffsets rrz/RFC6455WebSocket._apply_mask..wsEPPqtAwvzQ.>)??ABBPPPPPPr)rranger)r&rQlengthrRcnts`` ` r _apply_maskzRFC6455WebSocket._apply_maskrsM >YYFFmmxxPPPPPPCPPPPPPrc$|dkr||_|sd}nt|dkrtjd|d}|r |tvrt dd ||ddd }n'#ttf$rt dd wxYwd}| |d f t|d kr| |ddS|dkrdSt dd)Nr\rr~!HrzUnexpected close status code.Tz)Close message data should be valid UTF-8.r  close_data  control_codezUnknown control frame received.) rrrl unpack_fromVALID_CLOSE_STATUSr r,rUnicodeDecodeErrorrprBrr)rrCr&rs r_handle_control_framez&RFC6455WebSocket._handle_control_frameysY Q;;&*D # TQ+D$77:9/A!A!A/7999E++--44T!""XtDDDD*J7EEE/CEEEE  JJ62,J / / /')) ) q[[ IIdI - - - - - s]] D'799 9s 0B"B0c#Kd} ||}|jdzr.||j|O|r||urt d|}|jr|}d}|V#t $r=tj\}}}| |j |j fYdSt$rYdSt$r| dwxYw)NT)rr\zUnexpected message change.r[)izInternal Server Error) _recv_framerCrdrL RuntimeErrorrBr rexc_inforBrrrr)rfragmented_messagerr&exc_typexc_valexc_tbs rrzRFC6455WebSocket._iter_framess^!  **3E*FF>A%..(8(8(:(:<<<%E'9K*K*K&'CDDD%,"#-6688D)-&JJJ % E E E'*|~~ $GWf JJ7>7?"CJ D D D D D D$    FF    JJ"AJ B B B  sBBAD D !Dcn|j}|d}tjd|\}}|dz dk}|dz dz}|dz}|r|rd|jvrt dd|d z} | d vrt dd |d zd k} | s|jst dd |dz} | dzr)|st dd| dkrt ddn(| r|rt dd| s|st dd| dkr%tjd|dd} n*| dkr$tjd|dd} | |jkr)t dd| |j| rtjd|d} d} |r| dzrO| dkr|nd}| |}| | |j||}| s| d|n| | kr|j | | z }|stt|}| r||| || }| |z} | ||n%#t"t$f$rt d d!wxYw| | k|S)"NrZz!BBr~rPrrYzhRSV1, RSV2, RSV3: MUST be 0 unless an extension is negotiated that defines meanings for non-zero values.r)rr~rZr\r]r^zUnknown opcode received.z9A client MUST mask all frames that it sends to the serverr\z&Control frames must not be fragmented.}zBAll control frames MUST have a payload length of 125 bytes or lessz=Received a non-continuation opcode within fragmented message.zAReceived continuation opcode with no previous fragments received.~rXrz!QrIz=Incoming frame of {} bytes is above length limit of {} bytes.!BBBB)r rDr)r')rTrRrzText data must be valid utf-8)r>rlunpackrr rr-rr,r:rMrFr rrrrVrcrp)rrrrabrBrsv123r9rCmaskedrTrQreceivedr rDr=dlens rrfzRFC6455WebSocket._recv_framesa}UF++16Q;a!z  N N,DOCC+MNNNR - - -'.HII ISC Hdk H'/GHH HS A: ( ?+D3>???||+%&&&  ( (''(( ( ( (''(( ( S==]4a11!4FF s]]]4a11!4F D) ) )'.m.t.t-////00 0  3=$$q''22D n&1* n,2aKKd&&(((TG;;DAALll64+@'`llmmG ? LLHL - - - -V##K$$Vh%6772/1111vvP((Dh(OOA#d??LL(L3333*J7???/=????V##s 2J "J,Tc:d}t|tr|d}d}d}|}|pddz} |rW|rU| sS||}||t jz }|dddksJ|dd}d}t|} | sd}|rT|d vrtd |s|std | d krtd tj d|dz} n)|rdn|rdnd|z} tj d| |rdndz} |rdnd} | dkrtj d| dz| } n8| d krtj d| dz| } ntj d| | z} |rjttj fdtdD}t||| }tj dg|R}nd}d| | ||fS)NFrTrr\rH@)r\r]r^zUnknown control opcode.z#Control frame cannot be a fragment.rqz$Control frame data too large (>125).z!Bror~rZiz!BQrpz!BHrrc:g|]}dS)r\) getrandbits)r9_rands rr;z2RFC6455WebSocket._pack_message..0s':::AD$$Q'':::rrPrsr)rrror5compressflushr Z_SYNC_FLUSHrr rlrmrtimerSrrVr)rrrx continuationr'r`is_text compress_bit compressoris_control_framerTrrC lengthdatarQmaskdatars @rrzRFC6455WebSocket._pack_messages gs # # nnW--GG 5577 )-A2  "z "*: " ))'22G z''(9:: :G233<#66666crclG!LW F  J;..#$=>>> K5 K#$IJJJ||#$JKKK[|f'<==FF&RQQ7.Aaa\-QF[v51Ga'HIIF%,VV1 E>>UJ,??J  $)++&&D::::q:::D&227D&IIG{72T222HHHxxXw?@@@rc |jD]}|cSdSr)r)rrs rrzRFC6455WebSocket.wait8s&  AHHH  rc|j |j||jdS#|jwxYwr)rrr rqr)rframes r_sendzRFC6455WebSocket._send<se     % K   & & & N " " $ $ $ $ $DN " " $ $ $ $s AA+c `|j|d<|j|fi|}||dS)Nrx)rrr)rrkwpayloads rrzRFC6455WebSocket.sendCs?{8 $$$W3333 7rc,|jdvr|js|H|\}}t|tr|d}t jd||z}nd} ||dn#t$r|sYnwxYwd|_dSdSdS)N)r\rrrXr r\r_T) rrrrrorlrmrrI)rrr\rmsgr&s rrKz$RFC6455WebSocket._send_closing_frameHs <7 " "4+@ "%( c3''.**W--C{40036  $Q ////   *  %)D ! ! ! # " " "sA66 BBc ||d|jtjn`#t$rS}|jt jkr4|jd |j |Yd}~nd}~wwxYw|j dS#|j wxYw)rT)r\rrrN) rKr rSHUT_WRrIrrrrrrrB)rr\rRs rrBzRFC6455WebSocket.closeZs  $ $ t $ T T T K  0 0 0 0 e e ew%.((AHHTM]abHccddd e K       DK      s-;>B9 BA BB9BB99C)Nrr)FFTN)FN)r!r"r#rrr,r5r:r>rM staticmethodrVrdrrfrrrrrKrBr$r%s@rrrs.04Z^": ' ' ' ' ' '44444444*%%%(%%%&4QQQ\Q 999:2KKKKZ-2CG:A:A:A:Ax%%% ))))$         rr)0rrrrrandomrr r SocketErrorrrlrrrhashlibrr ImportErrorreventletrr eventlet.greeneventlet.supportr _mod __import__r ECONNRESETEPIPE ESHUTDOWNrJr__all__rsetlistrSrbrrr rrr rpr rr3rrrs  ''''''  !!!!!!!!!    !!!!!!&&&&&& @  D" 4R"66     ",ek5?K" K (7 SDtT  DtT   DtT    DtT    a"a"a"a"a"a"a"a"H O O O O O O O O d     I   I     J   _ _ _ _ _ y_ _ _ _ _ s!;AA/BB  B