From 98a7aaca182ce6b879329bbaeb06efca284c6220 Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Fri, 5 Aug 2011 10:23:32 +1200 Subject: Documentation. --- examples/add_header.py | 8 ++++++++ examples/simple_script | 6 ------ examples/stickycookies | 35 +++++++++++++++++++++++++++++++++++ examples/stickycookies.py | 34 ---------------------------------- 4 files changed, 43 insertions(+), 40 deletions(-) create mode 100644 examples/add_header.py delete mode 100755 examples/simple_script create mode 100755 examples/stickycookies delete mode 100644 examples/stickycookies.py (limited to 'examples') diff --git a/examples/add_header.py b/examples/add_header.py new file mode 100644 index 00000000..dab33c8c --- /dev/null +++ b/examples/add_header.py @@ -0,0 +1,8 @@ +#!/usr/bin/env python +""" + This script adds a new header to all responses. +""" +from libmproxy import script + +def response(ctx, f): + f.response.headers["newheader"] = ["foo"] diff --git a/examples/simple_script b/examples/simple_script deleted file mode 100755 index de562163..00000000 --- a/examples/simple_script +++ /dev/null @@ -1,6 +0,0 @@ -#!/usr/bin/env python -from libmproxy import script - -def response(ctx, f): - ctx.log("processing a response") - f.response.headers["newheader"] = ["foo"] diff --git a/examples/stickycookies b/examples/stickycookies new file mode 100755 index 00000000..1ae0e113 --- /dev/null +++ b/examples/stickycookies @@ -0,0 +1,35 @@ +#!/usr/bin/env python +from libmproxy import controller, proxy + +class StickyMaster(controller.Master): + def __init__(self, server): + controller.Master.__init__(self, server) + self.stickyhosts = {} + + def run(self): + try: + return controller.Master.run(self) + except KeyboardInterrupt: + self.shutdown() + + def handle_request(self, msg): + hid = (msg.host, msg.port) + if msg.headers["cookie"]: + self.stickyhosts[hid] = msg.headers["cookie"] + elif hid in self.stickyhosts: + msg.headers["cookie"] = self.stickyhosts[hid] + msg._ack() + + def handle_response(self, msg): + hid = (msg.request.host, msg.request.port) + if msg.headers["set-cookie"]: + self.stickyhosts[hid] = f.response.headers["set-cookie"] + msg._ack() + + +ssl_config = proxy.SSLConfig( + "~/.mitmproxy/cert.pem" +) +server = proxy.ProxyServer(ssl_config, 8080) +m = StickyMaster(server) +m.run() diff --git a/examples/stickycookies.py b/examples/stickycookies.py deleted file mode 100644 index 682fa226..00000000 --- a/examples/stickycookies.py +++ /dev/null @@ -1,34 +0,0 @@ -from libmproxy import controller, proxy - -class StickyMaster(controller.Master): - def __init__(self, server): - controller.Master.__init__(self, server) - self.stickyhosts = {} - - def run(self): - try: - return controller.Master.run(self) - except KeyboardInterrupt: - self.shutdown() - - def handle_request(self, msg): - hid = (msg.host, msg.port) - if msg.headers["cookie"]: - self.stickyhosts[hid] = msg.headers["cookie"] - elif hid in self.stickyhosts: - msg.headers["cookie"] = self.stickyhosts[hid] - msg._ack() - - def handle_response(self, msg): - hid = (msg.request.host, msg.request.port) - if msg.headers["set-cookie"]: - self.stickyhosts[hid] = f.response.headers["set-cookie"] - msg._ack() - - -ssl_config = proxy.SSLConfig( - "~/.mitmproxy/cert.pem" -) -server = proxy.ProxyServer(ssl_config, 8080) -m = StickyMaster(server) -m.run() -- cgit v1.2.3