diff options
Diffstat (limited to 'mitmproxy/examples/iframe_injector.py')
-rw-r--r-- | mitmproxy/examples/iframe_injector.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/mitmproxy/examples/iframe_injector.py b/mitmproxy/examples/iframe_injector.py new file mode 100644 index 00000000..29de9b63 --- /dev/null +++ b/mitmproxy/examples/iframe_injector.py @@ -0,0 +1,27 @@ +# Usage: mitmdump -s "iframe_injector.py url" +# (this script works best with --anticache) +from bs4 import BeautifulSoup +from libmproxy.models import decoded + + +def start(context, argv): + if len(argv) != 2: + raise ValueError('Usage: -s "iframe_injector.py url"') + context.iframe_url = argv[1] + + +def response(context, flow): + if flow.request.host in context.iframe_url: + return + with decoded(flow.response): # Remove content encoding (gzip, ...) + html = BeautifulSoup(flow.response.content) + if html.body: + iframe = html.new_tag( + "iframe", + src=context.iframe_url, + frameborder=0, + height=0, + width=0) + html.body.insert(0, iframe) + flow.response.content = str(html) + context.log("Iframe inserted.") |