aboutsummaryrefslogtreecommitdiffstats
path: root/libmproxy/protocol2/reverse_proxy.py
blob: cb6d1d78670787a7a174805f8b619ba359a66eea (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from __future__ import (absolute_import, print_function, division)

from .layer import Layer, ServerConnectionMixin
from .tls import TlsLayer


class ReverseProxy(Layer, ServerConnectionMixin):

    def __init__(self, ctx, server_address, client_tls, server_tls):
        super(ReverseProxy, self).__init__(ctx)
        self.server_address = server_address
        self._client_tls = client_tls
        self._server_tls = server_tls

    def __call__(self):
        layer = TlsLayer(self, self._client_tls, self._server_tls)
        for message in layer():
            if not self._handle_server_message(message):
                yield message