o KK&h@sddlmZddlZddlZddlmZmZddlmZm Z m Z ddl m Z m Z ddlmZmZddlmZmZgd ZdddZdddZdddZdddZdS)) annotationsN)HeadersMultipleValuesError) InvalidHeaderInvalidHeaderValueInvalidUpgrade)parse_connection parse_upgrade)ConnectionOptionUpgradeProtocol) accept_key generate_key) build_request check_requestbuild_responsecheck_responseheadersrreturnstrcCs*t}d|d<d|d<||d<d|d<|S)z Build a handshake request to send to the server. Update request headers passed in argument. Args: headers: Handshake request headers. Returns: ``key`` that must be passed to :func:`check_response`. websocketUpgrade ConnectionSec-WebSocket-Key13Sec-WebSocket-Version)rrkeyrN/var/www/html/venv/lib/python3.10/site-packages/websockets/legacy/handshake.pyrs  rc Cstdd|dDg}tdd|Dstdd|tdd|dDg}t|d kr9|d d ksAtdd|z|d }Wn"tyX}ztd |d }~wt yi}ztd d|d }~wwz t j | dd}Wnt jy}ztd ||d }~wwt|dkrtd |z|d}Wn"ty}ztd|d }~wt y}ztdd|d }~ww|dkrtd||S)ab Check a handshake request received from the client. This function doesn't verify that the request is an HTTP/1.1 or higher GET request and doesn't perform ``Host`` and ``Origin`` checks. These controls are usually performed earlier in the HTTP request handling code. They're the responsibility of the caller. Args: headers: Handshake request headers. Returns: ``key`` that must be passed to :func:`build_response`. Raises: InvalidHandshake: If the handshake request is invalid. Then, the server must return a 400 Bad Request error. cSg|]}t|qSrr .0valuerrr :z!check_request..rcs|] }|dkVqdSupgradeNlowerr"rrr =z check_request.., cSr rr r"rrrr%Ar&rrrrNmultiple valuesT)validaterr)sumget_allanyrjoinlenr+KeyErrorrrbase64 b64decodeencodebinasciiErrorr)r connectionr)s_w_keyexcraw_key s_w_versionrrrr%sN          rrNonecCs d|d<d|d<t||d<dS)z Build a handshake response to send to the client. Update response headers passed in argument. Args: headers: Handshake response headers. key: Returned by :func:`check_request`. rrrSec-WebSocket-AcceptN)acceptrrrrres rc Cstdd|dDg}tdd|Dstdd|tdd|dDg}t|d kr9|d d ksAtdd |z|d }Wn"tyX}ztd |d}~wt yi}ztd d|d}~ww|t |krut d |dS)a Check a handshake response received from the server. This function doesn't verify that the response is an HTTP/1.1 or higher response with a 101 status code. These controls are the responsibility of the caller. Args: headers: Handshake response headers. key: Returned by :func:`build_request`. Raises: InvalidHandshake: If the handshake response is invalid. cSr rr!r"rrrr%r&z"check_response..rcsr'r(r*r"rrrr,r-z!check_response.. cSr rr/r"rrrr%r&rr0rrr.rENr1) r4r5r6rr7r8r+r9rrrFr)rrr?r) s_w_acceptrArrrrus,     r)rrrr)rrrrrrD) __future__rr:r=datastructuresrr exceptionsrrrrr r typingr r utilsr rFr__all__rrrrrrrrs    @