aboutsummaryrefslogtreecommitdiffstats
path: root/libmproxy/protocol2/reverse_proxy.py
blob: dfffd2f2597b5147721355972278053a51d27948 (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 .ssl import SslLayer


class ReverseProxy(Layer, ServerConnectionMixin):

    def __init__(self, ctx, server_address, client_ssl, server_ssl):
        super(ReverseProxy, self).__init__(ctx)
        self.server_address = server_address
        self.client_ssl = client_ssl
        self.server_ssl = server_ssl

    def __call__(self):
        layer = SslLayer(self, self.client_ssl, self.server_ssl)
        for message in layer():
            if not self._handle_server_message(message):
                yield message