E gddlZddlmZddlmZmZddlmZGddZGddeZGd d Z dS) N) controller) connections exceptions)configc"eZdZdZfdZxZS)_LayerCodeCompletionz` Dummy class that provides type hinting in PyCharm, which simplifies development a lot. c <tjdi| dS)NT)super__init__r client_conn server_connchannelctx)self mixin_args __class__s k/var/www/sysmax/venv/lib/python3.11/site-packages/seleniumwire/thirdparty/mitmproxy/server/protocol/base.pyr z_LayerCodeCompletion.__init__s*&&:&&&  F)__name__ __module__ __qualname____doc__r __classcell__rs@rrrsK M M M M M M M M Mrrc.eZdZdZfdZdZdZxZS)Layerz^ Base class for all layers. All other protocol layers should inherit from this class. c J||_ tjdi|dS)a? Each layer usually passes itself to its child layers as a context. Properties of the context are transparently mapped to the layer, so that the following works: .. code-block:: python root_layer = Layer(None) root_layer.client_conn = 42 sub_layer = Layer(root_layer) print(sub_layer.client_conn) # 42 The root layer is passed a :py:class:`seleniumwire.thirdparty.mitmproxy.server.RootContext` object, which provides access to :py:attr:`.client_conn `, :py:attr:`.next_layer ` and other basic attributes. Args: ctx: The (read-only) parent layer / context. Nr )rr r )rrrrs rr zLayer.__init__ s6* &&:&&&&&rct)aLogic of the layer. Returns: Once the protocol has finished without exceptions. Raises: ~seleniumwire.thirdparty.mitmproxy.exceptions.ProtocolException: if an exception occurs. No other exceptions must be raised. )NotImplementedError)rs r__call__zLayer.__call__=s"###rc,t|j|S)z[ Attributes not present on the current layer are looked up on the context. )getattrr)rnames r __getattr__zLayer.__getattr__Istx&&&r)rrrrr r!r%rrs@rrrs`''''': $ $ $'''''''rrcBeZdZdZd fd ZdZdZdZdZdZ xZ S) ServerConnectionMixinaI Mixin that provides a layer with the capabilities to manage a server connection. The server address can be passed in the constructor or set by calling :py:meth:`set_server`. Subclasses are responsible for calling :py:meth:`disconnect` before returning. Recommended Usage: .. code-block:: python class MyLayer(Layer, ServerConnectionMixin): def __call__(self): try: # Do something. finally: if self.server_conn.connected(): self.disconnect() Nct|||_|dSN)r r (_ServerConnectionMixin__make_server_connr*_ServerConnectionMixin__check_self_connect)rserver_addressrs rr zServerConnectionMixin.__init__dsG 22>BB !!#####rcN|jj}|rgd}|jjjr$||jjj|d|jjjko |d|v}|r6tjd t|dSdS)z We try to protect the mitmproxy from _accidentally_ connecting to itself, e.g. because of a failed transparent lookup or an invalid configuration. ) localhostz 127.0.0.1z::1rzFInvalid server address: {} The mitmproxy shall not connect to itself.N) raddressroptions listen_hostappend listen_portrProtocolExceptionformatrepr)rr0forbidden_hosts self_connects r__check_self_connectz*ServerConnectionMixin.__check_self_connectks "*  ???O{". H&&t{':'FGGG dk1==. o-   2AAGW AVAV    rc|jjjrB|jjjdkr-t j||jjjddfdSt j||jjjdf|jjjS)NrT) rr1spoof_source_addressupstream_bind_addressrServerConnectionrr r0)rr,s r__make_server_connz(ServerConnectionMixin.__make_server_conns ;  3  8K8aeg8g8g/!5!=a!@! DdLL L/!4!JA N #8 rc |jr||d|d|dd||j_|dS)zc Sets a new server address. If there is an existing connection, it will be closed. zSet new server address: {}:{}rr/debugN)r connected disconnectlogr6r0r+rr0s r set_serverz ServerConnectionMixin.set_servers|   % % ' '  OO    077 GAJOOQXYYY#*  !!#####rcT|ddt|jjg|jj}|j|j|jd|j|||_dS)z Deletes (and closes) an existing server connection. Must not be called if there is no existing connection. serverdisconnectrBN) rEr7rr0finishclosertellr*rFs rrDz ServerConnectionMixin.disconnects #WtD4D4L/M/M.NOOO"* !!!     ,d.>???227;;rc |jjstjd |j|ddt |jjg|jd|jdS#tj $r}|j j j r$|t |dnJ|t |d|tjdtjdt |jjt!|d}~wwxYw)a Establishes a server connection. Must not be called if there is an existing connection. Raises: ~seleniumwire.thirdparty.mitmproxy.exceptions.ProtocolException: if the connection could not be established. z2Cannot connect to server, no server address given. serverconnectrBerrorz"Server connection to {} failed: {}N)rr0rr5connectrEr7rask TcpExceptionrr1suppress_connection_errors traceback format_excr6str)res rrPzServerConnectionMixin.connectsN' e./cdd d    $ $ & & & HH_gT5E5M0N0N/O P P P L  _d.> ? ? ? ? ?&   {"= :a'****a'***-//999.4;;)122CFF  sA(B E+C E&&E+r)) rrrrr r+r*rGrDrPrrs@rr'r'Ps$$$$$$$,$$$ < < <rr') rT!seleniumwire.thirdparty.mitmproxyrrr(seleniumwire.thirdparty.mitmproxy.serverrrrr'r rrrZs888888EEEEEEEE;;;;;;MMMMMMMM$3'3'3'3'3' 3'3'3'lkkkkkkkkkkr