U @f @sddlZddlZddlZddlZddlmZddlmZddl Z ddl Z ddl Z ddl Z ddl Z zddlmZmZWn,ek rddlmZddlmZYnXddlmZddlmZdd lmZdd lmZd D]:Zzeeiid gZWnek rdZYqXqqejejejhZd Z ddgZ!dZ"e#e$e%dde$e%dde$e%dde$e%ddZ&Gddde'Z(GdddZ)GdddZ*Gddde'Z+Gddde'Z,Gd d!d!e-Z.Gd"d#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 iicseZdZdfdd ZZS) BadRequest400 Bad RequestNcs$tt|||_||_||_dSN)super Exception__init__statusbodyheaders)selfrrr __class__L/var/www/html/chatgpt/venv/lib/python3.8/site-packages/eventlet/websocket.pyr9szBadRequest.__init__)rNN__name__ __module__ __qualname__r __classcell__rrrrr8src@sbeZdZdZefddZedddZdd Zd d Z d d Z ddZ ddZ ddZ ddZdS)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. cCs(||_d|_d|_g|_d|_||_dS)NT)handlerprotocol_versionsupport_legacy_versionssupported_protocolsorigin_checkermax_frame_length)rr#r(rrrrUs zWebSocketWSGI.__init__NFcs&fdd}|dkr|S||S)Ncs"|}|_|_r|_|Sr)r%r'r&)r#instclsr'r%r&rr decoratorcs z+WebSocketWSGI.configured..decoratorr)r+r#r&r'r%r,rr*r configured]szWebSocketWSGI.configuredc Cs@dd|dddD}d|kr<|dddksL|d d ggSz0d |krb||}n|jrt||}ntWnVtk r}z8|j}|jpd }|j pg}||d g||gWYSd}~XYnXd |kr|d z| |Wn4t k r(}zt |t krW5d}~XYnX|ddtj_gS)NcSsg|] }|qSrstrip).0partrrr osz*WebSocketWSGI.__call__..ZHTTP_CONNECTIONr ,upgradeZ HTTP_UPGRADEZ websocketr) ConnectioncloseHTTP_SEC_WEBSOCKET_VERSIONzeventlet.set_idleT)getlowersplit_handle_hybi_requestr%_handle_legacy_requestrrrrr#OSErrorr ACCEPTABLE_CLIENT_ERRORS_send_closing_framerZ WSGI_LOCALZalready_handled) renvironZstart_responseZhttp_connection_partswserrrrrr__call__ns@        zWebSocketWSGI.__call__c Csd|kr|d}nd|kr(|d}ntdd|krNd|_d|krTtnd|_|jdkr||d}||d}d|d _|d d}td |||}t | }d }| d d krd}d|| d| d| df} | d} | dk r| d| 7} |jdkr8d| d d| d} nP|jdkrd| d d| dd d| d|} nt d|| t|||jS)Neventlet.inputgunicorn.socket8No eventlet.input or gunicorn.socket present in environ.ZHTTP_SEC_WEBSOCKET_KEY1LZHTTP_SEC_WEBSOCKET_KEY2Kz wsgi.inputz>IIrBzwsgi.url_schemehttpswssz %s://%s%s%s HTTP_HOST SCRIPT_NAME PATH_INFOZ 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_lengthreadstructpackrdigestr9encode ValueErrorsendallr ) rrAsockkey1key2Zkey3keyresponseschemelocationqshandshake_replyrrrr=sx             z$WebSocketWSGI._handle_legacy_requestcCs|dkr dSi}|dD]}|d}i}|ddD]V}|d}t|dkrhd||d<q8|dd||d<q8||dg|q|S)Nr3;=Tr")r;lenr/r: setdefaultappend)rheaderresextpartsconfigr1Zkey_valrrr_parse_extension_headers   ( z%WebSocketWSGI._parse_extension_headercCs|sdS|d}|dkrdS|D]}|dd|ddd}ttjd}|d}|dk r|dkrp||d<n0t|d||d<d |dkrdksq"q"|d }|dk r|dkr||d <n0t|d ||d <d |d krdksq"q"|SdS) Npermessage-deflateserver_no_context_takeoverFclient_no_context_takeover)rvrwserver_max_window_bitsTrJclient_max_window_bits)r9minzlib MAX_WBITSint)r extensionsdeflatersZ want_configZ max_wbitsZmwbrrr_negotiate_permessage_deflates8        z+WebSocketWSGI._negotiate_permessage_deflatecCs|sdSg}|D]p\}}|g}|D]D\}}|dkr@q.|dkrX||q.|d|t|fq.|d|qd|S)NFTz%s=%ss; s, )itemsr\rnstrjoin)rparsed_extensionsrrnamersZ ext_partsrbvaluerrr_format_extension_header s z&WebSocketWSGI._format_extension_headercCsd|kr|d}nd|kr(|d}ntd|d}|dkrNtddgdt||_d |krft|d |jd kr|d d nd }|jdk r||d|stdd|dd}d}|rdd|dDD]}||jkr|}qq|d }t t | t } dddd| g} |r8| d| i} ||d} || } | dk rh| | d<|| }|dk r| d||d| dt|||j|| |jdS) NrErFrGr7)8Z13z426 Upgrade Required)zSec-WebSocket-Versionz8, 13)rrZHTTP_SEC_WEBSOCKET_KEYrQrJZHTTP_SEC_WEBSOCKET_ORIGINr rMz 403 Forbidden)rrScss|]}|VqdSrr.r0irrr 6sz5WebSocketWSGI._handle_hybi_request..r3s HTTP/1.1 101 Switching ProtocolssUpgrade: websocketsConnection: UpgradesSec-WebSocket-Accept: sSec-WebSocket-Protocol: ZHTTP_SEC_WEBSOCKET_EXTENSIONSrusSec-WebSocket-Extensions: s rR)protocolrr()rUrrr~r$r9r'r;r&base64 b64encoderr\ PROTOCOL_GUIDr[rnrtrrr^rRFC6455WebSocketr()rrAr_Z hybi_versionorigin protocolsZnegotiated_protocolprbrcrgrrrZ formatted_extrrrr<sh           z"WebSocketWSGI._handle_hybi_requestcCsBd}d}|D](}|tjkr$||7}q |dkr |d7}q t||S)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 ri)stringdigitsr~)rroutspacescharrrrrVUs   zWebSocketWSGI._extract_number)NNNF)rr r!__doc__DEFAULT_MAX_FRAME_LENGTHr classmethodr-rDr=rtrrr<rVrrrrr @s  *>&9c@sLeZdZdZdddZddZddZd d Zd d ZdddZ ddZ dS)r a{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. rHcCs|dtj|_dj|d|dd|dd|d|dd |_||_|d |_|d |_|d|_ ||_ ||_ d |_ d |_ t|_t|_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}rMrNr rOZ REMOTE_ADDRZ REMOTE_PORT)ZshostZspathZcaddrZcportrQZHTTP_WEBSOCKET_PROTOCOLFr8N)r9sysstderrlogformat log_contextr rrpathrAversionwebsocket_closed_buf collectionsdeque_msgsr Semaphore _sendlock)rr_rArrrrrys"    zWebSocket.__init__cCs<t|tr|d}nt|ts,t|}d|d}|S)zwPack the message inside ``00`` and ``FF`` As per the dataframing section (5.3) for the websocket spec utf-8) isinstancerr\bytesrmessagepackedrrr _pack_messages      zWebSocket._pack_messagecCsg}d}|j}|r|d}|dkrb|d}|dkr6q||d|dd||dd}q|dkr|ddkstd |d |_qqtd |q||_|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.rrrirreplaceNz Unexpected closing handshake: %rTz6Don't understand how to parse this type of message: %r)rfindrndecodeAssertionErrorrr])rmsgsZend_idxbufZ frame_typerrr_parse_messagess$ zWebSocket._parse_messagescCs6||}|jz|j|W5|jXdS)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)rracquirereleaser r^rrrrsends   zWebSocket.sendcCsV|jsL|jrdS|jd}|dkr(dS|j|7_|}|j|q|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.Nir8)rrr recvrrextendpopleft)rdeltarrrrwaits zWebSocket.waitFcCsF|jdkrB|jsBz|jdWntk r:|s6YnXd|_dS)z3Sends the closing frame to the client, if required.rHsTN)rrr r^r>)rignore_send_errorsrrrr@szWebSocket._send_closing_framec Csxzfz|d|jdWnFtk rb}z(|jtjkrR|jdj |j |dW5d}~XYnXW5|jXdS)cForcibly close the websocket; generally it is preferable to return from the handler method.T {ctx} socket shutdown error: {e}ctxrCN) r r6r@shutdownr>errnoENOTCONNrwriterr)rrCrrrr6s  .zWebSocket.closeN)rH)F) rr r!rrrrrrr@r6rrrrr ds   c@s eZdZdS)ConnectionClosedErrorNrr r!rrrrrsrcseZdZfddZZS)FailedConnectionErrorcst||||_||_dSr)rrrr)rrrrrrrszFailedConnectionError.__init__rrrrrrsrc@s eZdZdS) ProtocolErrorNrrrrrrsrcseZdZddddeffdd ZGdddZdd Zd d Zd d ZGdddZ e d&ddZ ddZ ddZ d'ddZd(ddZddZddZd d!Zd)d"d#Zd*d$d%ZZS)+r NFcsLt|||||_||_||_|p,i|_d|_d|_||_ d|_ dSr) rr _iter_framesiteratorclientrr _deflate_enc _deflate_decr(_remote_close_data)rr_rArrrrr(rrrrs  zRFC6455WebSocket.__init__c@s&eZdZddZddZd ddZdS) zRFC6455WebSocket.UTF8DecodercCs,trt|_nd|_td}||_dS)Nutf8) utf8validatorZ Utf8Validator validatorcodecsgetincrementaldecoderdecoder)rZ decoderclassrrrrs   z%RFC6455WebSocket.UTF8Decoder.__init__cCs|jr|j|jdSr)rresetr)rrrrrs z"RFC6455WebSocket.UTF8Decoder.resetFcCs4|jr&|j|\}}}}|s&td|j||S)NzData is not valid unicode)rvalidater]rr)rdatafinalZvalidZeocpZc_iZt_irrrr!s z#RFC6455WebSocket.UTF8Decoder.decodeN)F)rr r!rrrrrrr UTF8Decodersrcs\jddkrdSfdd}jr4dndr@|SjdkrR|_jSdS)Nrucs(ttjtjjrdndtj S)Nrzry)r| compressobjZ_DEFAULT_COMPRESSIONDEFLATEDr9rr}roptionsrrr_make-s  z;RFC6455WebSocket._get_permessage_deflate_enc.._makerwrv)rr9rr)rrrrr_get_permessage_deflate_enc(s  z,RFC6455WebSocket._get_permessage_deflate_enccs`jddks|sdSfdd}jr8dndrD|SjdkrV|_jSdS)Nrucs tjrdndtj S)Nryrz)r| decompressobjr9rr}rrrrrAsz;RFC6455WebSocket._get_permessage_deflate_dec.._makervrw)rr9rr)rrsv1rrrr_get_permessage_deflate_dec<s   z,RFC6455WebSocket._get_permessage_deflate_deccCs<d}t||kr8|j|t|}|s.t||}q|S)Nr8)rlr rr)rnumbytesrdrrr _get_bytesOs  zRFC6455WebSocket._get_bytesc@s(eZdZd ddZd ddZddZdS) zRFC6455WebSocket.MessageNcCs(||_g|_d|_||_||_||_dS)NF)rrfinishedopcode decompressorr()rrr(rrrrrrYs z!RFC6455WebSocket.Message.__init__FcCs||_|j|dSr)rrrn)rrrrrrpushaszRFC6455WebSocket.Message.pushcCsdd|j}|jd@sJ|jrJ|j|d|j}|jjrJtdd|j|j r`|j ||j }|S)Nr8rJz;Incoming compressed frame exceeds length limit of {} bytes.) rrrr decompressr(unconsumed_tailrrrrr)rrrrrgetvaluees  z!RFC6455WebSocket.Message.getvalue)NN)F)rr r!rrrrrrrMessageXs  rrcs4|dkrt}t|}dfdd|DS)Nr8c3s,|]$}t||dAfVqdS)N)rrrmaskoffsetrrrwsz/RFC6455WebSocket._apply_mask..)rlranger)rrlengthrZcntrrr _apply_maskrszRFC6455WebSocket._apply_maskc Cs|dkr||_|sd}nxt|dkrtd|d}|r@|tkrJtddz||ddd }Wqtt fk rtdd YqXnd}|j |d fd t n,|d kr|j |ddn|dkrn tdddS)NrJrri!HrzUnexpected close status code.Tz)Close message data should be valid UTF-8.r  close_data  control_codezUnknown control frame received.) rrlrY unpack_fromVALID_CLOSE_STATUSrrrUnicodeDecodeErrorr]r6rr)rrrrrrr_handle_control_frameys:   z&RFC6455WebSocket._handle_control_frameccsd}z`|j|d}|jd@r0||j|q|rD||k rDtd|}|jr|}d}|VqWnjtk rt\}}}|j |j |j fdYn6t k rYdSt k r|j ddYnXdS)N)rrJzUnexpected message change.r)izInternal Server Error) _recv_framerrr RuntimeErrorrrrexc_infor6rrrr)rZfragmented_messagerrZexc_typexc_valexc_tbrrrrs0     zRFC6455WebSocket._iter_framesc CsL|j}|d}td|\}}|d?dk}|d?d@}|d@}|rZ|rZd|jkrZtdd|d @} | d krttdd |d @d k} | s|jstdd |d@} | d@r|stdd| dkrtddn&| r|rtddn| s|stdd| dkr td|dd} n| dkr*td|dd} | |jkrJtdd| |j| r`td|d} d} |rt| d@r| dkr|nd}| |}|j | |j||d}| s|j d|dn| | krH|j | | }|stt|}| r|j|| || d}| |} z|j ||dWn$ttfk rBtd d!YnXq|S)"Nrz!BBrirrurzhRSV1, RSV2, RSV3: MUST be 0 unless an extension is negotiated that defines meanings for non-zero values.rx)rrirrJrr zUnknown opcode received.z9A client MUST mask all frames that it sends to the serverrJz&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.~rrz!Qrz=Incoming frame of {} bytes is above length limit of {} bytes.!BBBB)rrr8)r)rrrzText data must be valid utf-8)rrYunpackrrrr(rrrrrr rrrlrrr])rrrroabrZrsv123rrmaskedrrZreceivedrrrdlenrrrrs            zRFC6455WebSocket._recv_frameTcsd}t|tr|d}d}d}|}|p.dd@} |r~|r~| s~||}||tj7}|dddksnt|dd}d}t |} | sd}|r|d krt d |s|st d | d krt d t d|dB} n0|rdn|rdnd|B} t d| |rdndB} |rdnd} | dkr0t d| dB| } n.| d krNt d| dB| } nt d| | B} |rt tfddtdD}t||| }t j d|}nd}d| | ||fS)NFrTrrJr@)rJrr zUnknown control opcode.z#Control frame cannot be a fragment.rz$Control frame data too large (>125).z!Brririz!BQrz!BHrcsg|]}dqS)rJ) getrandbits)r0_Zrandrrr20sz2RFC6455WebSocket._pack_message..rrr8)r)rrr\rcompressflushr| Z_SYNC_FLUSHrrlrrYrZrtimerrrr)rrrZ continuationrr Zis_textZ compress_bitZ compressorZis_control_framerrorZ lengthdatarZmaskdatarr$rrsN         zRFC6455WebSocket._pack_messagecCs|jD] }|SdSr)r)rrrrrr8s zRFC6455WebSocket.waitcCs,|jz|j|W5|jXdSr)rrrr r^)rframerrr_send<s zRFC6455WebSocket._sendcKs&|j|d<|j|f|}||dS)Nr)rrr*)rrkwpayloadrrrrCs zRFC6455WebSocket.sendcCs|jdkr~|js~|dk rF|\}}t|tr4|d}td||}nd}z|j|ddWntk rv|srYnXd|_dS)N)rJrrrr rJr T) rrrrr\rYrZrr>)rrrrmsgrrrrr@Hs  z$RFC6455WebSocket._send_closing_framec Cs~zlz |j|dd|jtjWnFtk rh}z(|jtjkrX|j dj |j |dW5d}~XYnXW5|jXdS)rT)rrrrN) r r6r@rSHUT_WRr>rrrrrr)rrrCrrrr6Zs .zRFC6455WebSocket.close)Nr)N)FFTN)FN)N)rr r!rrrrrrr staticmethodrrrrrrr*rr@r6r"rrrrrs.    M < r)0rrrrrandomrr r SocketErrorrrYrr(r|hashlibrr ImportErrorrZeventletrrZeventlet.greenZeventlet.supportr _mod __import__r ECONNRESETEPIPE ESHUTDOWNr?r__all__rsetlistrr rrr r rrr]rrrrrrs`             &