o [hd@sddlmZddlZddlZddlmZddlmZmZddl m Z m Z m Z ddl mZmZddlmZmZdd lmZmZgd ZdddZdddZdddZdddZdS)) annotationsN)List)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: str: ``key`` that must be passed to :func:`check_response`. websocketUpgrade ConnectionSec-WebSocket-Key13Sec-WebSocket-Version)rrkeyr/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/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)ad 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: str: ``key`` that must be passed to :func:`build_response`. Raises: InvalidHandshake: if the handshake request is invalid; then the server must return 400 Bad Request error. cSg|]}t|qSrr .0valuerrr ;z!check_request..rcs|] }|dkVqdSupgradeNlowerr#rrr >z check_request.., cSr!rr r#rrr r&Br'rrrrNz,more than one Sec-WebSocket-Key header foundT)validaterz0more than one Sec-WebSocket-Version header foundr)sumget_allanyr joinlenr,KeyErrorrrbase64 b64decodeencodebinasciiErrorr)r connectionr*Zs_w_keyexcZraw_keyZ s_w_versionrrr r&s^        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)acceptrrrr rjs 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#rrr r&r'z"check_response..rcsr(r)r+r#rrr r-r.z!check_response.. cSr!rr0r#rrr r&r'rr1rrr/rBNz/more than one Sec-WebSocket-Accept header found) r4r5r6r r7r8r,r9rrrCr)rrr?r*Z s_w_acceptr@rrr rzs4    r)rrrr)rrrrrrA) __future__rr:r=typingrZdatastructuresrr exceptionsrrr rr r r r utilsrrCr__all__rrrrrrrr s     D