mitmproxy/libmproxy/web/__init__.py

148 lines
3.5 KiB
Python
Raw Normal View History

from __future__ import absolute_import, print_function
2014-09-13 23:30:00 +00:00
import tornado.ioloop
import tornado.httpserver
from .. import controller, flow
from . import app
2014-09-13 23:30:00 +00:00
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)
2014-12-09 17:55:16 +00:00
app.ClientConnection.broadcast(
type="flows",
cmd="add",
data=f.get_state(short=True)
)
def _update(self, f):
super(WebFlowView, self)._update(f)
2014-12-09 17:55:16 +00:00
app.ClientConnection.broadcast(
type="flows",
cmd="update",
data=f.get_state(short=True)
)
def _remove(self, f):
super(WebFlowView, self)._remove(f)
2014-12-09 17:55:16 +00:00
app.ClientConnection.broadcast(
type="flows",
cmd="remove",
data=f.get_state(short=True)
)
def _recalculate(self, flows):
super(WebFlowView, self)._recalculate(flows)
2014-12-09 17:55:16 +00:00
app.ClientConnection.broadcast(
type="flows",
cmd="reset"
)
2014-09-13 23:30:00 +00:00
class WebState(flow.State):
def __init__(self):
super(WebState, self).__init__()
self.view._close()
self.view = WebFlowView(self.flows)
2014-09-13 23:30:00 +00:00
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",
2014-09-14 00:22:28 +00:00
"wdebug",
"wport",
"wiface",
2014-09-13 23:30:00 +00:00
]
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):
2014-09-14 00:22:28 +00:00
self.options = options
super(WebMaster, self).__init__(server, WebState())
self.app = app.Application(self.state, self.options.wdebug)
2014-09-13 23:30:00 +00:00
2014-09-17 15:30:19 +00:00
self.last_log_id = 0
2014-09-13 23:30:00 +00:00
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()
2014-09-14 00:22:28 +00:00
http_server = tornado.httpserver.HTTPServer(self.app)
2014-09-14 00:22:28 +00:00
http_server.listen(self.options.wport)
2014-09-13 23:30:00 +00:00
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)
2014-09-13 23:30:00 +00:00
if f:
f.reply()
return f
def handle_response(self, f):
super(WebMaster, self).handle_response(f)
2014-09-13 23:30:00 +00:00
if f:
f.reply()
return f
def handle_log(self, l):
2014-09-17 15:30:19 +00:00
self.last_log_id += 1
app.ClientConnection.broadcast(
2014-12-09 17:55:16 +00:00
type="add_event",
data={
2014-09-17 15:30:19 +00:00
"id": self.last_log_id,
"message": l.msg,
"level": l.level
}
)
self.add_event(l.msg, l.level)
l.reply()