mitmproxy/libmproxy/proxy/message.py

"""
This module contains all valid messages layers can send to the underlying layers.
"""


class _Message(object):
    def __eq__(self, other):
        # Allow message == Connect checks.
        # FIXME: make Connect == message work.
        if isinstance(self, other):
            return True
        return self is other


class Connect(_Message):
    """
    Connect to the server
    """


class Reconnect(_Message):
    """
    Re-establish the server connection
    """


class ChangeServer(_Message):
    """
    Change the upstream server.
    """

    def __init__(self, address, server_ssl, sni, depth=1):
        self.address = address
        self.server_ssl = server_ssl
        self.sni = sni

        # upstream proxy scenario: you may want to change either the final target or the upstream proxy.
        # We can express this neatly as the "nth-server-providing-layer"
        # ServerConnection could get a `via` attribute.
        self.depth = depth