aboutsummaryrefslogtreecommitdiffstats
path: root/libmproxy/proxy/message.py
blob: a667123c9c6bf1171603dc37562cc1844d3fb482 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""
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