2015-08-08 14:08:57 +00:00
|
|
|
from __future__ import (absolute_import, print_function, division)
|
|
|
|
|
2015-08-30 13:27:29 +00:00
|
|
|
from ... import platform
|
|
|
|
from ...exceptions import ProtocolException
|
|
|
|
from ...protocol import Layer, ServerConnectionMixin
|
2015-08-08 14:08:57 +00:00
|
|
|
|
|
|
|
|
|
|
|
class TransparentProxy(Layer, ServerConnectionMixin):
|
2016-01-27 09:12:18 +00:00
|
|
|
|
2015-08-08 14:08:57 +00:00
|
|
|
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:
|
2015-09-11 00:17:04 +00:00
|
|
|
raise ProtocolException("Transparent mode failure: %s" % repr(e))
|
2015-08-08 14:08:57 +00:00
|
|
|
|
2015-08-11 18:27:34 +00:00
|
|
|
layer = self.ctx.next_layer(self)
|
2015-08-18 13:59:44 +00:00
|
|
|
try:
|
|
|
|
layer()
|
|
|
|
finally:
|
|
|
|
if self.server_conn:
|
2015-09-03 15:01:25 +00:00
|
|
|
self.disconnect()
|