From 9af8f4bb31c94a25780a4189bffa406906249626 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Mon, 21 Nov 2016 02:16:20 +0100 Subject: organize examples This commit is largely based on work by Thiago Arrais (@thiagoarrais) and Shane Bradfield (@l33tLumberjack). I wasn't really able to get their PR reasonably merged onto the latest master, so I reapplied their changes manually here and did some further improvements on that. --- examples/simple/modify_body_inject_iframe.py | 29 ++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 examples/simple/modify_body_inject_iframe.py (limited to 'examples/simple/modify_body_inject_iframe.py') diff --git a/examples/simple/modify_body_inject_iframe.py b/examples/simple/modify_body_inject_iframe.py new file mode 100644 index 00000000..33d18bbd --- /dev/null +++ b/examples/simple/modify_body_inject_iframe.py @@ -0,0 +1,29 @@ +# Usage: mitmdump -s "iframe_injector.py url" +# (this script works best with --anticache) +import sys +from bs4 import BeautifulSoup + + +class Injector: + def __init__(self, iframe_url): + self.iframe_url = iframe_url + + def response(self, flow): + if flow.request.host in self.iframe_url: + return + html = BeautifulSoup(flow.response.content, "lxml") + if html.body: + iframe = html.new_tag( + "iframe", + src=self.iframe_url, + frameborder=0, + height=0, + width=0) + html.body.insert(0, iframe) + flow.response.content = str(html).encode("utf8") + + +def start(): + if len(sys.argv) != 2: + raise ValueError('Usage: -s "iframe_injector.py url"') + return Injector(sys.argv[1]) -- cgit v1.2.3