mitmproxy/libmproxy/proxy/modes/http_proxy.py

27 lines
725 B
Python
Raw Normal View History

from __future__ import (absolute_import, print_function, division)
2015-08-30 13:27:29 +00:00
from ...protocol import Layer, ServerConnectionMixin
2015-08-14 08:41:11 +00:00
class HttpProxy(Layer, ServerConnectionMixin):
2015-08-11 18:27:34 +00:00
def __call__(self):
layer = self.ctx.next_layer(self)
2015-08-18 13:59:44 +00:00
try:
layer()
finally:
if self.server_conn:
self._disconnect()
2015-08-29 23:21:58 +00:00
2015-08-11 18:27:34 +00:00
class HttpUpstreamProxy(Layer, ServerConnectionMixin):
def __init__(self, ctx, server_address):
2015-08-16 21:25:02 +00:00
super(HttpUpstreamProxy, self).__init__(ctx, server_address=server_address)
def __call__(self):
layer = self.ctx.next_layer(self)
2015-08-18 13:59:44 +00:00
try:
layer()
finally:
if self.server_conn:
2015-08-29 23:21:58 +00:00
self._disconnect()