Merge pull request #1768 from mhils/organize-examples

Organize Examples

1 files changed, 29 insertions, 0 deletions

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])