add reverseproxy mode, fix bugs

1 files changed, 19 insertions, 0 deletions

diff --git a/libmproxy/protocol2/reverse_proxy.py b/libmproxy/protocol2/reverse_proxy.py
new file mode 100644
index 00000000..dfffd2f2
--- /dev/null
+++ b/libmproxy/protocol2/reverse_proxy.py
@@ -0,0 +1,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