mitmproxy/libmproxy/protocol2/http_proxy.py
2015-08-16 15:19:11 +02:00

25 lines
781 B
Python

from __future__ import (absolute_import, print_function, division)
from .layer import Layer, ServerConnectionMixin
from .http import Http1Layer, HttpLayer
class HttpProxy(Layer, ServerConnectionMixin):
def __call__(self):
layer = Http1Layer(self, "regular")
for message in layer():
if not self._handle_server_message(message):
yield message
class HttpUpstreamProxy(Layer, ServerConnectionMixin):
def __init__(self, ctx, server_address):
super(HttpUpstreamProxy, self).__init__(ctx)
self.server_address = server_address
def __call__(self):
layer = Http1Layer(self, "upstream")
for message in layer():
if not self._handle_server_message(message):
yield message