mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
132 lines
3.2 KiB
Python
132 lines
3.2 KiB
Python
from __future__ import absolute_import, print_function
|
|
import tornado.ioloop
|
|
import tornado.httpserver
|
|
from .. import controller, flow
|
|
from . import app
|
|
|
|
|
|
class Stop(Exception):
|
|
pass
|
|
|
|
|
|
class WebFlowView(flow.FlowView):
|
|
def __init__(self, store):
|
|
super(WebFlowView, self).__init__(store, None)
|
|
|
|
def _add(self, f):
|
|
super(WebFlowView, self)._add(f)
|
|
app.FlowUpdates.broadcast("add", f.get_state(short=True))
|
|
|
|
def _update(self, f):
|
|
super(WebFlowView, self)._update(f)
|
|
app.FlowUpdates.broadcast("update", f.get_state(short=True))
|
|
|
|
def _remove(self, f):
|
|
super(WebFlowView, self)._remove(f)
|
|
app.FlowUpdates.broadcast("remove", f.get_state(short=True))
|
|
|
|
def _recalculate(self, flows):
|
|
super(WebFlowView, self)._recalculate(flows)
|
|
app.FlowUpdates.broadcast("reset", None)
|
|
|
|
|
|
class WebState(flow.State):
|
|
def __init__(self):
|
|
super(WebState, self).__init__()
|
|
self.view._close()
|
|
self.view = WebFlowView(self.flows)
|
|
|
|
|
|
class Options(object):
|
|
attributes = [
|
|
"app",
|
|
"app_domain",
|
|
"app_ip",
|
|
"anticache",
|
|
"anticomp",
|
|
"client_replay",
|
|
"eventlog",
|
|
"keepserving",
|
|
"kill",
|
|
"intercept",
|
|
"no_server",
|
|
"refresh_server_playback",
|
|
"rfile",
|
|
"scripts",
|
|
"showhost",
|
|
"replacements",
|
|
"rheaders",
|
|
"setheaders",
|
|
"server_replay",
|
|
"stickycookie",
|
|
"stickyauth",
|
|
"stream_large_bodies",
|
|
"verbosity",
|
|
"wfile",
|
|
"nopop",
|
|
|
|
"wdebug",
|
|
"wport",
|
|
"wiface",
|
|
]
|
|
|
|
def __init__(self, **kwargs):
|
|
for k, v in kwargs.items():
|
|
setattr(self, k, v)
|
|
for i in self.attributes:
|
|
if not hasattr(self, i):
|
|
setattr(self, i, None)
|
|
|
|
|
|
class WebMaster(flow.FlowMaster):
|
|
def __init__(self, server, options):
|
|
self.options = options
|
|
super(WebMaster, self).__init__(server, WebState())
|
|
self.app = app.Application(self.state, self.options.wdebug)
|
|
|
|
self.last_log_id = 0
|
|
|
|
def tick(self):
|
|
flow.FlowMaster.tick(self, self.masterq, timeout=0)
|
|
|
|
def run(self): # pragma: no cover
|
|
self.server.start_slave(
|
|
controller.Slave,
|
|
controller.Channel(self.masterq, self.should_exit)
|
|
)
|
|
iol = tornado.ioloop.IOLoop.instance()
|
|
|
|
http_server = tornado.httpserver.HTTPServer(self.app)
|
|
http_server.listen(self.options.wport)
|
|
|
|
tornado.ioloop.PeriodicCallback(self.tick, 5).start()
|
|
try:
|
|
iol.start()
|
|
except (Stop, KeyboardInterrupt):
|
|
self.shutdown()
|
|
|
|
def handle_request(self, f):
|
|
super(WebMaster, self).handle_request(f)
|
|
if f:
|
|
f.reply()
|
|
return f
|
|
|
|
def handle_response(self, f):
|
|
super(WebMaster, self).handle_response(f)
|
|
if f:
|
|
f.reply()
|
|
return f
|
|
|
|
def handle_log(self, l):
|
|
self.last_log_id += 1
|
|
app.ClientConnection.broadcast(
|
|
"add_event", {
|
|
"id": self.last_log_id,
|
|
"message": l.msg,
|
|
"level": l.level
|
|
}
|
|
)
|
|
self.add_event(l.msg, l.level)
|
|
l.reply()
|
|
|