o _cm.@sddlmZddlmZmZmZmZddlmZm Z m Z m Z m Z ddl mZmZddlmZmZmZmZmZmZddlmZmZddlmZmZmZmZmZm Z m!Z!m"Z"dd l#m$Z$m%Z%dd lm&Z&m'Z'm(Z(m)Z)m*Z*m+Z+dd l,m-Z-dd l.m/Z/m0Z0dd l1TdgZ2Gddde Z3dS)) annotations) GeneratorListOptionalSequence)CLIENT CONNECTINGOPEN ConnectionState)HeadersMultipleValuesError)InvalidHandshake InvalidHeaderInvalidHeaderValue InvalidStatusInvalidUpgradeNegotiationError)ClientExtensionFactory Extension)build_authorization_basicbuild_extension build_hostbuild_subprotocolparse_connectionparse_extensionparse_subprotocol parse_upgrade)RequestResponse)ConnectionOptionExtensionHeader LoggerLikeOrigin SubprotocolUpgradeProtocol) WebSocketURI) accept_key generate_key)*ClientConnectioncspeZdZdZdddeddfd+fdd Zd,ddZd-ddZd.d d!Zd/d#d$Z d0d&d'Z d1fd)d* Z Z S)2r+a Sans-I/O implementation of a WebSocket client connection. Args: wsuri: URI of the WebSocket server, parsed with :func:`~websockets.uri.parse_uri`. origin: value of the ``Origin`` header. This is useful when connecting to a server that validates the ``Origin`` header to defend against Cross-Site WebSocket Hijacking attacks. extensions: list of supported extensions, in order in which they should be tried. subprotocols: list of supported subprotocols, in order of decreasing preference. state: initial state of the WebSocket connection. max_size: maximum size of incoming messages in bytes; :obj:`None` to disable the limit. logger: logger for this connection; defaults to ``logging.getLogger("websockets.client")``; see the :doc:`logging guide <../topics/logging>` for details. Niwsurir'originOptional[Origin] extensions*Optional[Sequence[ClientExtensionFactory]] subprotocolsOptional[Sequence[Subprotocol]]stater max_size Optional[int]loggerOptional[LoggerLike]cs8tjt|||d||_||_||_||_t|_dS)N)Zsider3r4r6) super__init__rr,r-available_extensionsavailable_subprotocolsr)key)selfr,r-r/r1r3r4r6 __class__zs z,ClientConnection.connect..Sec-WebSocket-ExtensionsSec-WebSocket-Protocol)r rr,hostportsecureZ user_inforr-r<r:rr;rr resource_name)r=headersZextensions_headerZprotocol_headerr@r@rAconnect[s.       zClientConnection.connectresponser Nonec Cs4|jdkr t||j}tdd|dDg}tdd|Ds/td|r,d|dtd d|d Dg}t|d krJ|d  d ksWtd |rTd|dz|d}Wn"t yn}zt d|d}~wt y}zt dd|d}~ww|t |jkrtd||||_|||_dS)z Check a handshake response. Args: request: WebSocket handshake response received from the server. Raises: InvalidHandshake: if the handshake response is invalid. ecSg|]}t|qSr@)rrHvaluer@r@rArJz5ClientConnection.process_response..r css|] }|dkVqdS)upgradeN)lowerrWr@r@rA sz4ClientConnection.process_response.., NcSrVr@)rrWr@r@rArJrYrFrrrEzSec-WebSocket-Acceptz/more than one Sec-WebSocket-Accept header found) status_coderrQsumget_allanyrjoinlenr[KeyErrorrrr(r<rprocess_extensionsr/process_subprotocol subprotocol)r=rSrQ connectionrZZ s_w_acceptexcr@r@rAprocess_responsesF     z!ClientConnection.process_responserQr List[Extension]c Csg}|d}|rQ|jdurtdtdd|Dg}|D]2\}}|jD] }|j|kr-q%z|||}Wn ty>Yq%w||n td|d|q|S)a{ Handle the Sec-WebSocket-Extensions HTTP response header. Check that each extension is supported, as well as its parameters. :rfc:`6455` leaves the rules up to the specification of each extension. To provide this level of flexibility, for each extension accepted by the server, we check for a match with each extension available in the client configuration. If no match is found, an exception is raised. If several variants of the same extension are accepted by the server, it may be configured several times, which won't make sense in general. Extensions must implement their own requirements. For this purpose, the list of previously accepted extensions is provided. Other requirements, for example related to mandatory extensions or the order of extensions, may be implemented by overriding this method. Args: headers: WebSocket handshake response headers. Returns: List[Extension]: List of accepted extensions. Raises: InvalidHandshake: to abort the handshake. rKNzno extensions supportedcSrVr@)rrH header_valuer@r@rArJrYz7ClientConnection.process_extensions..zUnsupported extension: name = z , params = )r`r:rr_rGZprocess_response_paramsrappend) r=rQZaccepted_extensionsr/Zparsed_extensionsrGZresponse_paramsrI extensionr@r@rAres<        z#ClientConnection.process_extensionsOptional[Subprotocol]cCsd}|d}|r>|jdurtdtdd|Dg}t|dkr.d|}td||d }||jvr>td ||S) a: Handle the Sec-WebSocket-Protocol HTTP response header. If provided, check that it contains exactly one supported subprotocol. Args: headers: WebSocket handshake response headers. Returns: Optional[Subprotocol]: Subprotocol, if one was selected. NrLzno subprotocols supportedcSrVr@)rrlr@r@rArJrYz8ClientConnection.process_subprotocol..rr]zmultiple subprotocols: rzunsupported subprotocol: )r`r;rr_rcrbr)r=rQrgr1Zparsed_subprotocolsZsubprotocols_displayr@r@rArfs     z$ClientConnection.process_subprotocolrequestcCsN|jr|jd|j|jD] \}}|jd||q|j|dS)z Send a handshake request to the server. Args: request: WebSocket handshake request event. z> GET %s HTTP/1.1z> %s: %sN)debugr6pathrQ raw_itemsZwritesrn serialize)r=rqr<rXr@r@rA send_request,s zClientConnection.send_requestGenerator[None, None, None]c #s.|jturt|jj|jj|jjEdH}|jrI|j |j }}|j d|||j D] \}}|j d||q-|jdurI|j dt|jzr@rArx;s8    zClientConnection.parse)r,r'r-r.r/r0r1r2r3r r4r5r6r7)rBr)rSr rBrT)rQr rBrk)rQr rBrp)rqrrBrT)rBrw) __name__ __module__ __qualname____doc__r r9rRrjrerfrvrx __classcell__r@r@r>rAr+.s  , 4 L %N)4 __future__rtypingrrrrrhrr r r r Zdatastructuresr r exceptionsrrrrrrr/rrrQrrrrrrrrZhttp11rr r!r"r#r$r%r&urir'utilsr(r)Z legacy.client__all__r+r@r@r@rAs  (