From d002371d30e4b0ab7d1d23023236a9446d4c2396 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Mon, 7 Sep 2015 13:51:46 +0200 Subject: expose `next_layer` to inline scripts --- libmproxy/proxy/root_context.py | 5 ++++- libmproxy/proxy/server.py | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) (limited to 'libmproxy/proxy') diff --git a/libmproxy/proxy/root_context.py b/libmproxy/proxy/root_context.py index 87a540c0..dccdf023 100644 --- a/libmproxy/proxy/root_context.py +++ b/libmproxy/proxy/root_context.py @@ -40,7 +40,10 @@ class RootContext(object): Returns: The next layer """ + layer = self._next_layer(top_layer) + return self.channel.ask("next_layer", layer) + def _next_layer(self, top_layer): # 1. Check for --ignore. if self.config.check_ignore(top_layer.server_conn.address): return RawTCPLayer(top_layer, logging=False) @@ -119,4 +122,4 @@ class RootContext(object): class Log(object): def __init__(self, msg, level="info"): self.msg = msg - self.level = level \ No newline at end of file + self.level = level diff --git a/libmproxy/proxy/server.py b/libmproxy/proxy/server.py index 5d067b45..c12bbbfa 100644 --- a/libmproxy/proxy/server.py +++ b/libmproxy/proxy/server.py @@ -137,4 +137,4 @@ class ConnectionHandler(object): def log(self, msg, level): msg = "{}: {}".format(repr(self.client_conn.address), msg) - self.channel.tell("log", Log(msg, level)) \ No newline at end of file + self.channel.tell("log", Log(msg, level)) -- cgit v1.2.3