!ni ddlZddlZddlZddlZddlmZddlmZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZmZddlmZddlmZdd lmZdd lmZd D]Z eeiid gZnej8ej:ej<hZd 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/y#e$rddlmZddlmZYwxYw#e$rdZYwxYw)$N)Random)error)md5sha1)r)sha) semaphore)wsgi)socket) get_errno)zwsaccel.utf8validatorzautobahn.utf8validatori WebSocketWSGI WebSockets$258EAFA5-E914-47DA-95CA-C5AB0DC85B11iii iic eZdZdfd ZxZS) BadRequestcTtt| ||_||_||_yN)super Exception__init__statusbodyheaders)selfrrr __class__s R/home/homepc/tiktok-worker/venv/lib/python3.12/site-packages/eventlet/websocket.pyrzBadRequest.__init__9s% i')   )400 Bad RequestNN__name__ __module__ __qualname__r __classcell__rs@rrr8s rrc^eZdZdZefdZe d dZdZdZ dZ dZ d Z d Z d Zy) 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. cX||_d|_d|_g|_d|_||_y)NT)handlerprotocol_versionsupport_legacy_versionssupported_protocolsorigin_checkermax_frame_length)rr(r-s rrzWebSocketWSGI.__init__Us1 $'+$#% " 0rNc2fd}||S||S)NcF|}|_|_r|_|Sr)r*r,r+)r(instclsr,r*r+s r decoratorz+WebSocketWSGI.configured..decoratorcs,w(Kr)r1r(r+r,r*r2s` ``` r configuredzWebSocketWSGI.configured]s   ? !!rc|jddjjdDcgc]}|j}}d|vr#|jddjdk(s |ddggS d |vr|j |}n(|j r|j |}n td |vr |d  |j||jd d t j"_gScc}w#t$rF}|j}|jxsd }|jxsg} ||dg| z|gcYd}~Sd}~wwxYw#t$r}t|tvrYd}~d}~wwxYw) NHTTP_CONNECTIONr ,upgrade HTTP_UPGRADE websocketr) ConnectioncloseHTTP_SEC_WEBSOCKET_VERSIONrzeventlet.set_idleT)getlowersplitstrip_handle_hybi_requestr*_handle_legacy_requestrrrrr(OSErrorr ACCEPTABLE_CLIENT_ERRORS_send_closing_framer WSGI_LOCALalready_handled) renvironstart_responseparthttp_connection_partswserrrs r__call__zWebSocketWSGI.__call__ns $5r:@@BHHM!O JJL!O!O22 NB/557;F ,/F.G HI +w6..w7--009 l" ' ) (G' ( *  LL  t$+/' O!O  XXF66=SDiio2G 636@ B6M   |#;;< s;D?>D E E;EEE F%E<<FcZd|vr|dj}nd|vr|d}n tdd|vrd|_d|vrtd|_|jdk(ry|j |d}|j |d}d|d _|d j d}tjd |||z}t|j}d }|jd d k(rd}|d|jd|jd|jd} |jd} | | d| zz } |jdk(r:d|jdjzdz| jzdz} n{|jdk(rad|jdjzdz|jddjzdz| jzdzz} n td|j| t|||jS)Neventlet.inputgunicorn.socket8No eventlet.input or gunicorn.socket present in environ.HTTP_SEC_WEBSOCKET_KEY1LHTTP_SEC_WEBSOCKET_KEY2Kz wsgi.inputz>IIrMzwsgi.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_lengthreadstructpackrdigestr>encode ValueErrorsendallr) rrIsockkey1key2key3keyresponseschemelocationqshandshake_replys rrCz$WebSocketWSGI._handle_legacy_requestsy w &+,779D ' ),-DVW W $ /$&D !(7 l"$&D !  B &''0I(JKD''0I(JKD45GL ! 0<(--a0D++eT4047C3x(H ;;( )W 4F  KK $ KK & KK $   [[ ( > b H  B &&)0 M(B(I(I(KLO((+3//*; <?J J  " "b (*-4KK ,F,M,M,OPS,,  99ELLN O R, , /7oo.? @ C ## BC C _%w(=(=>>rc"|yi}|jdD]}|jd}i}|ddD]}|jd}t|dk(r%d||djj<G|djjdj||djj<|j |djjgj ||S)Nr7;=Tr")r@lenrAr? setdefaultappend)rheaderresextpartsconfigrKkey_vals r_parse_extension_headerz%WebSocketWSGI._parse_extension_headers ><<$ HCIIcNEFab  _**S/w<1$9=F71:++-33569@9I9I9K9Q9QRU9V9\9\9^F71:++-3356  _ NN58>>+113R 8 ? ? G H rc|sy|jd}|y|D]}|jdd|jddd}ttjd}|jd}|9|dur||d<n/t |jd||d<d |dcxkrdksn|jd }|<|dur ||d <|cSt |jd ||d <d |d cxkrdksn|cSy) Npermessage-deflateserver_no_context_takeoverFclient_no_context_takeover)rrserver_max_window_bitsTrXclient_max_window_bits)r>minzlib MAX_WBITSint)r extensionsdeflater want_config max_wbitsmwbs r_negotiate_permessage_deflatez+WebSocketWSGI._negotiate_permessage_deflates6..!56 ? F /5jj9UW\.].4jj9UW\.]KDNNB/I**56C$;d]$$SZZ\2$$c%j&A%I%I%KL  M LLI. / 0zz%  rc`d|vr|dj}nd|vr|d}n td|d}|dvrtddgt||_d |vr t|j d |jd kr|j d d nd }|j -|j |j d|s td|j dd}d}|r/d|jdDD]}||jvs|}n|d }tjt|jtzj} dddd| zg} |r"| jd|jzi} |j!|j d} |j#| } | | | d<|j%| }|| jd|z|j'dj)| dzt+|||j|| |j,S)NrQrRrSr=)813z426 Upgrade Required)zSec-WebSocket-Versionz8, 13)rrHTTP_SEC_WEBSOCKET_KEYr`rXHTTP_SEC_WEBSOCKET_ORIGINr r[z 403 Forbidden)rrbc3<K|]}|jywr)rA).0is r z5WebSocketWSGI._handle_hybi_request..6s>Aaggi>sr7s HTTP/1.1 101 Switching ProtocolssUpgrade: websocketsConnection: UpgradesSec-WebSocket-Accept: sSec-WebSocket-Protocol: HTTP_SEC_WEBSOCKET_EXTENSIONSrsSec-WebSocket-Extensions: s ra)protocolrr-)rdrrrr)r>r,r@r+base64 b64encoderrk PROTOCOL_GUIDrjrrrrrmrRFC6455WebSocketr-)rrIrn hybi_versionorigin protocolsnegotiated_protocolprrrsrwrrr formatted_exts rrBz"WebSocketWSGI._handle_hybi_requests[ w &+,779D ' ),-DVW W;< } ,$:&H%IK K #L 1 #7 2,  %%*[[4b 9025    *&&w{{;'?H 88KK =tD " >)=> 000*+'  ./##D )E$F$M$M$OP>014x?A   " "#>ATA[A[A]#] ^11'++>]2^_ 44Z@  6= 2 3556GH  $  " "#@=#P Q W\\/2[@Agt/D/D)<+<151F1FH Hrczd}d}|D]%}|tjvr||z }|dk(s!|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 rz)stringdigitsr)rroutspaceschars rrezWebSocketWSGI._extract_numberUsS  Dv}}$t !    3x6!!r)NNNF)r!r"r#__doc__DEFAULT_MAX_FRAME_LENGTHr classmethodr4rOrCrrrrBrer3rrr r @sZ(2J1'+"&+0 "" (T<?| $L! 7Hr "rc>eZdZdZd dZdZdZdZdZd dZ dZ y ) 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. cP|jdtj|_dj |jd|jdd|jddz|jd|jd |_||_|jd |_|jd |_|jd|_ ||_ ||_ d |_ d |_ tj|_t#j$|_y)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 r] REMOTE_ADDR REMOTE_PORT)shostspathcaddrcportr`HTTP_WEBSOCKET_PROTOCOLFrN)r>sysstderrlogformat log_contextr rrpathrIversionwebsocket_closed_buf collectionsdeque_msgsr Semaphore _sendlock)rrnrIrs rrzWebSocket.__init__ys;;}cjj9JQQ++k*++mR07;;{B3OO++m,GKK 4NR   kk-0  $=> KK ,   %  &&( ",,.rct|tr|jd}n)t|tst|j}d|zdz}|S)zwPack the message inside ``00`` and ``FF`` As per the dataframing section (5.3) for the websocket spec utf-8) isinstancerrkbytesrmessagepackeds r _pack_messagezWebSocket._pack_messagesL gs #nnW-GGU+'l))+G7"W, rc4g}d}|j}|r~|d}|dk(rD|jd}|dk(rn]|j|d|jdd||dzd}n-|dk(r|ddk(s Jd |zd |_nt d |z|r~||_|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.rrrzrreplaceNz Unexpected closing handshake: %rTz6Don't understand how to parse this type of message: %r)rfindrdecoderrl)rmsgsend_idxbuf frame_types r_parse_messageszWebSocket._parse_messagessiiQJQ((7+b= C'N11'9EF'A+,'s"1v{L$F$LL{(,% !Y\_!_``   rc|j|}|jj |jj ||jj y#|jj wxYw)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 rmreleasers rsendzWebSocket.sends^ ##G,   % KK   ' NN " " $DNN " " $s A##A?cD|js{|jry|jjd}|dk(ry|xj|z c_|j }|jj ||js{|jjS)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**$$KK$$T*E| II I'')D JJ  d #**zz!!##rc|jdk(r1|js$ |jjdd|_yyy#t$r|sYwxYw)z3Sends the closing frame to the client, if required.rUsTN)rrr rmrD)rignore_send_errorss rrFzWebSocket._send_closing_frames] <<2 d&;&;  ##K0 %)D !'<  ** sA AAc |jd|jjd|jjy#t$r]}|jtj k7r6|j jdj|j|Yd}~|d}~wwxYw#|jjwxYw)cForcibly close the websocket; generally it is preferable to return from the handler method.T {ctx} socket shutdown error: {e}ctxrNN) rFr shutdownrDerrnoENOTCONNrwriterrr<)rrNs rr<zWebSocket.closes  $ $T * KK  & KK     eww%..(AHHTM]M]abHcd e KK   s*,A B/AB*%B2*B//B22CN)rUF) r!r"r#rrrrrrrFr<r3rrrrds*(/0 : %$( )  rc eZdZy)ConnectionClosedErrorNr!r"r#r3rrrrrrceZdZfdZxZS)FailedConnectionErrorcBt|||||_||_yr)rrrr)rrrrs rrzFailedConnectionError.__init__s  )  rr r%s@rrrs rrc eZdZy) ProtocolErrorNrr3rrr r rrr ceZdZddddeffd ZGddZdZdZd ZGd d Z e dd Z d Z dZ ddZ ddZdZdZdZddZddZxZS)r NFct|||||j|_||_||_|xsi|_d|_d|_||_ d|_ yr) rr _iter_framesiteratorclientrr _deflate_enc _deflate_decr-_remote_close_data) rrnrIrrrrr-rs rrzRFC6455WebSocket.__init__s^ w0))+    $*   0"&rc eZdZdZdZddZy)RFC6455WebSocket.UTF8Decoderctrtj|_nd|_tjd}||_y)Nutf8) utf8validator Utf8Validator validatorcodecsgetincrementaldecoderdecoder)r decoderclasss rrz%RFC6455WebSocket.UTF8Decoder.__init__s5!.!!%!77?L'>DLrc|jr|jj|jjyr)rresetr)rs rr z"RFC6455WebSocket.UTF8Decoder.resets)~~$$& LL   rc|jr-|jj|\}}}}|s td|jj ||S)NzData is not valid unicode)rvalidaterlrr)rdatafinalvalideocpc_it_is rrz#RFC6455WebSocket.UTF8Decoder.decode!sL~~(,(?(?(E%tS#$%@AA<<&&tU3 3rNr)r!r"r#rr rr3rr UTF8Decoderrs * !  4rr)cjjdyfd}jjrdndr|Sj |_jS)Nrctjtjtjj j rdndtj  S)Nrr)r compressobjZ_DEFAULT_COMPRESSIONDEFLATEDr>rroptionsrsr_makez;RFC6455WebSocket._get_permessage_deflate_enc.._make-sL##D$>$> %,[[T[[1I6N15&A%AB Brrr)rr>rr)rr1r0s` @r_get_permessage_deflate_encz,RFC6455WebSocket._get_permessage_deflate_enc(sk//%%&:; ? B ;;t{{38 :7N  ($)G!$$ $rcjjd|syfd}jjrdndr|Sj |_jS)Nrctjjjrdndtj S)Nrr)r decompressobjr>rrr/srr1z;RFC6455WebSocket._get_permessage_deflate_dec.._makeAs=%%w{{t{{3K8P37>>(C'CD Drrr)rr>rr)rrsv1r1r0s` @r_get_permessage_deflate_decz,RFC6455WebSocket._get_permessage_deflate_dec<sm//%%&:; ?$ D ;;t{{38 :7N  ($)G!$$ $rcd}t||krG|jj|t|z }|s t||z}t||krG|S)Nr)r}r rr)rnumbytesr#ds r _get_byteszRFC6455WebSocket._get_bytesOsZ$i("   CI!56A+--!8D $i("  rc"eZdZddZddZdZy)RFC6455WebSocket.MessageNcX||_g|_d|_||_||_||_y)NF)rr#finishedopcode decompressorr-)rr@r-rrAs rrz!RFC6455WebSocket.Message.__init__Ys."DLDI!DM DK ,D $4D !rcH||_|jj|yr)r?r#r)rr#r$s rpushzRFC6455WebSocket.Message.pushas!DM II  T "rcdj|j}|jdzsp|jrd|jj |dz|j }|jj r%tddj|j |jr&|jj||j}|S)NrrXz;Incoming compressed frame exceeds length limit of {} bytes.) rr#r@rA decompressr-unconsumed_tailrrrrr?)rr#s rgetvaluez!RFC6455WebSocket.Message.getvaluees88DII&D;;?t'8'8((33D;N4NPTPePef$$44/U\\]a]r]rsuu||||**4?Kr)NNr)r!r"r#rrCrIr3rrMessager=Xs 5 # rrJcp| t}t|}djfd|DS)Nrc3TK|]}t||zdzz f!yw)N)r)rrr#maskoffsets rrz/RFC6455WebSocket._apply_mask..ws0PqtAwvzQ.>)??ABPs%()r}ranger)r#rNlengthrOcnts`` ` r _apply_maskzRFC6455WebSocket._apply_maskrs/ >YFFmxxPCPPPrc|dk(r||_|sd}ndt|dkDrTtjd|d}|r|tvr t dd |j j|ddd }nd}|j|d f t|d k(r|j|dy|dk(ryt dd#ttf$r t dd wxYw)NrXrrz!HrzUnexpected close status code.Tz)Close message data should be valid UTF-8.r  close_data  control_codezUnknown control frame received.) rr}rh unpack_fromVALID_CLOSE_STATUSrr)rUnicodeDecodeErrorrlr<rr)rr@r#rs r_handle_control_framez&RFC6455WebSocket._handle_control_frameys Q;&*D #TQ++D$7:/A!A/799E++-44T!"XtDD  JJ62,J /') ) q[ IIdI - s] '79 9+J7E/CEEEs #B??Cc#Kd} |j|}|jdzr+|j|j|jM|r||ur t d|}|j r|j}d}|#t $rCtj\}}}|j|j|jfYyt$rYyt$r|jdwxYww)N)rrXzUnexpected message change.rX)izInternal Server Error) _recv_framer@rarI RuntimeErrorr?rrexc_infor<rrrr)rfragmented_messagerr#exc_typexc_valexc_tbs rrzRFC6455WebSocket._iter_framess! **3E*F>>A%..(8(8(:<%'9K*K&'CDD%,"##-668D)-&J% E'*||~ $GWf JJ7>>7??"CJ D$    JJ"AJ B  s0C=BB A C:C=C:C=C::C=c|j}|d}tjd|\}}|dz dk(}|dz dz}|dz}|r|rd|jvr t dd|d z} | d vr t dd |d zd k(} | s|j s t dd |dz} | dzr|s t dd| dkDr,t dd| r|r t dd| s|s t dd| dk(r tjd|dd} n$| dk(rtjd|dd} | |j kDr&t ddj| |j | rtjd|d} d} |r| dzrG| dk(r|jnd}|j|}|j| |j ||}| s|jd||S| | krk|jj| | z }|s tt|}| r|j!| || }| |z} |j||| | krk|S#t"t$f$r t d d!wxYw)"NrWz!BBrzrMrrVzhRSV1, RSV2, RSV3: MUST be 0 unless an extension is negotiated that defines meanings for non-zero values.r)rrzrWrXrZr[zUnknown opcode received.z9A client MUST mask all frames that it sends to the serverrXz&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.~rUrz!QrFz=Incoming frame of {} bytes is above length limit of {} bytes.!BBBB)rrAr)r$)rQrOrzText data must be valid utf-8)r;rhunpackrrrr-rr)r7rJrCr rrr}rSr`rl)rrrrabr?rsv123r6r@maskedrQrNreceivedrrAr:dlens rrczRFC6455WebSocket._recv_framesa}}UF+16Q;a!z ,DOOC+MNNR - -'.HI ISCdkk'/GH HS A:+D3>??|+%&&''( (''( ( S=]]4a1!4F s]]]4a1!4F D)) )'.m.t.t--//0 0 ==$q'2D&1*,2aKd&&(TG;;DALll64+@+@'`llmG LLHL -V#KK$$Vh%67/111v((Dh(OA#d??LL(L3V#+J7?/=???s ;II2cd}t|tr|jd}d}d}|j}|xsddz} |rH|rF| sD|j |}||j t jz }|dddk(sJ|dd}d}t|} | sd}|rH|d vr td |s|s td | d kDr td tjd|dz} n*|rdn|rdnd|z} tjd| |rdndz} |rdnd} | dkDrtjd| dz| } n9| d kDrtjd| dz| } ntjd| | z} |rrttj}tdDcgc]}|jd}}t j#||| }tjdg|}nd}dj%| | ||fScc}w)NFrTrrXrE@)rXrZr[zUnknown control opcode.z#Control frame cannot be a fragment.rnz$Control frame data too large (>125).z!BrlrzrWiz!BQrmz!BHrorMrpr)rrrkr2compressflushr Z_SYNC_FLUSHr}r rhrirtimerP getrandbitsrrSr)rrru continuationr$r]is_text compress_bit compressoris_control_framerQrr@ lengthdatarand_rNmaskdatas rrzRFC6455WebSocket._pack_messages  gs #nnW-GG 557 )-A2 z*: ))'2G z''(9(9: :G23<#66 66crlG!LWF ;.#$=>>5#$IJJ|#$JKK[[|f'<=F&Q7a\-QF[[v5a'HIF%V1 E>UJ,?J $))+&D16q:AD$$Q':D:&227D&IG{{72T2HHxxXw?@@ ;sG8c,|jD]}|cSyr)r)rrs rrzRFC6455WebSocket.wait8s AH rc|jj |jj||jj y#|jj wxYwr)rrr rmr)rframes r_sendzRFC6455WebSocket._send<sJ   % KK   & NN " " $DNN " " $s AA.c j|j|d<|j|fi|}|j|y)Nru)rrr)rrkwpayloads rrzRFC6455WebSocket.sendCs2{{8 $$$$W33 7rc|jdvrm|js`|@|\}}t|tr|j d}t j d||z}nd} |j|dd|_yyy#t$r|sYwxYw)N)rXr rrUr rXr\T) rrrrrkrhrirrD)rrrYrmsgr#s rrFz$RFC6455WebSocket._send_closing_frameHs <<7 "4+@+@%( c3'**W-C{{4036  $Q / %)D !,A " ** s A== B  B c |j|d|jjtj|jjy#t$r]}|j t j k7r6|jjdj|j|Yd}~|d}~wwxYw#|jjwxYw)rT)rYrrrN) rFr rSHUT_WRrDrrrrrrr<)rrYrNs rr<zRFC6455WebSocket.closeZs  $ $ t $ T KK  0 KK     eww%..(AHHTM]M]abHcd e KK   s*rs{  '  ! !& @ D"4R"6   ",,ekk5??K" K (7 tT tT  tT   tT  a"a"H O O d I I J _ y_ k  $  s#D# D:#D76D7:EE