2015-08-08 14:08:57 +00:00
|
|
|
from __future__ import (absolute_import, print_function, division)
|
|
|
|
|
|
|
|
from ..exceptions import ProtocolException
|
|
|
|
from .. import platform
|
|
|
|
from .layer import Layer, ServerConnectionMixin
|
|
|
|
|
|
|
|
|
|
|
|
class TransparentProxy(Layer, ServerConnectionMixin):
|
|
|
|
|
|
|
|
def __init__(self, ctx):
|
|
|
|
super(TransparentProxy, self).__init__(ctx)
|
|
|
|
self.resolver = platform.resolver()
|
|
|
|
|
|
|
|
def __call__(self):
|
|
|
|
try:
|
2015-08-16 21:25:02 +00:00
|
|
|
self.server_conn.address = self.resolver.original_addr(self.client_conn.connection)
|
2015-08-08 14:08:57 +00:00
|
|
|
except Exception as e:
|
|
|
|
raise ProtocolException("Transparent mode failure: %s" % repr(e), e)
|
|
|
|
|
2015-08-11 18:27:34 +00:00
|
|
|
layer = self.ctx.next_layer(self)
|
2015-08-08 14:08:57 +00:00
|
|
|
for message in layer():
|
|
|
|
if not self._handle_server_message(message):
|
|
|
|
yield message
|