def run(self):
while True:
try:
Master.run(self)
except KeyboardInterrupt:
self.shutdown()
sys.exit(0)
class ProxyStart():
def __init__(self, option, obj):
self.opts = options.Options()
for i in option:
self.__addOption(i)
self.cf = config.ProxyConfig(self.opts)
self.server = ProxyServer(config=self.cf)
self.master = Master(opts=self.opts)
self.master.server = self.server
self.master.addons.add(obj)
def __addOption(self, *args):
self.opts.add_option(args[0][0], args[0][1], args[0][2], args[0][3])
def run(self):
self.master.run()
class MitmProxy:
"""Run and manage a mitmproxy server instance.
DEPRECATED.
"""
def __init__(self, host, port, options):
self.options = options
# Used to stored captured requests
self.storage = RequestStorage(
base_dir=options.pop('request_storage_base_dir', None)
)
# Used to modify requests/responses passing through the server
# DEPRECATED. Will be superceded by request/response interceptors.
self.modifier = RequestModifier()
# The scope of requests we're interested in capturing.
self.scopes = []
self.request_interceptor = None
self.response_interceptor = None
self._event_loop = asyncio.new_event_loop()
asyncio.set_event_loop(self._event_loop)
# mitmproxy specific options
mitmproxy_opts = Options(
confdir=options.pop('mitm_confdir', DEFAULT_CONFDIR),
listen_host=host,
listen_port=port,
)
# Create an instance of the mitmproxy server
self._master = Master(mitmproxy_opts)
self._master.server = ProxyServer(ProxyConfig(mitmproxy_opts))
self._master.addons.add(*addons.default_addons())
self._master.addons.add(SendToLogger())
self._master.addons.add(MitmProxyRequestHandler(self))
# Update the options now all addons have been added
mitmproxy_opts.update(
ssl_insecure=options.get('verify_ssl', True),
upstream_cert=DEFAULT_UPSTREAM_CERT,
stream_websockets=DEFAULT_STREAM_WEBSOCKETS,
**self._get_upstream_proxy_args(),
)
# Options that are prefixed mitm_ are passed through to mitmproxy
mitmproxy_opts.update(**{k[5:]: v for k, v in options.items() if k.startswith('mitm_')})
def serve_forever(self):
"""Run the server."""
asyncio.set_event_loop(self._event_loop)
self._master.run()
def address(self):
"""Get a tuple of the address and port the mitmproxy server
is listening on.
"""
return self._master.server.address
def shutdown(self):
"""Shutdown the server and perform any cleanup."""
try:
# Wait for any active requests to finish. This reduces the
# probability of seeing shutdown errors in the console.
self._master.server.wait_for_silence()
except Timeout:
pass
self._master.shutdown()
self.storage.cleanup()
def _get_upstream_proxy_args(self):
proxy_config = get_upstream_proxy(self.options)
http_proxy = proxy_config.get('http')
https_proxy = proxy_config.get('https')
conf = None
if http_proxy and https_proxy:
if http_proxy.hostport != https_proxy.hostport:
# We only support a single upstream mitmproxy server
raise ValueError('Cannot specify both http AND https '
'mitmproxy settings with mitmproxy backend')
conf = https_proxy
elif http_proxy:
conf = http_proxy
elif https_proxy:
conf = https_proxy
args = {}
if conf:
scheme, username, password, hostport = conf
args['mode'] = 'upstream:{}://{}'.format(scheme, hostport)
if username:
args['upstream_auth'] = '{}:{}'.format(username, password)
return args