diff --git a/examples/iframe_injector b/examples/iframe_injector
deleted file mode 100755
index 8b1e02f15..000000000
--- a/examples/iframe_injector
+++ /dev/null
@@ -1,50 +0,0 @@
-#!/usr/bin/env python
-"""
- Zap encoding in requests and inject iframe after body tag in html responses.
- Usage:
- iframe_injector http://someurl/somefile.html
-"""
-from libmproxy import controller, proxy
-import os
-import sys
-
-
-class InjectingMaster(controller.Master):
- def __init__(self, server, iframe_url):
- controller.Master.__init__(self, server)
- self._iframe_url = iframe_url
-
- def run(self):
- try:
- return controller.Master.run(self)
- except KeyboardInterrupt:
- self.shutdown()
-
- def handle_request(self, msg):
- if 'Accept-Encoding' in msg.headers:
- msg.headers["Accept-Encoding"] = 'none'
- msg.reply()
-
- def handle_response(self, msg):
- if msg.content:
- c = msg.replace('', '' % self._iframe_url)
- if c > 0:
- print 'Iframe injected!'
- msg.reply()
-
-
-def main(argv):
- if len(argv) != 2:
- print "Usage: %s IFRAME_URL" % argv[0]
- sys.exit(1)
- iframe_url = argv[1]
- config = proxy.ProxyConfig(
- cacert = os.path.expanduser("~/.mitmproxy/mitmproxy-ca.pem")
- )
- server = proxy.ProxyServer(config, 8080)
- print 'Starting proxy...'
- m = InjectingMaster(server, iframe_url)
- m.run()
-
-if __name__ == '__main__':
- main(sys.argv)
diff --git a/examples/iframe_injector.py b/examples/iframe_injector.py
new file mode 100644
index 000000000..7042dbab7
--- /dev/null
+++ b/examples/iframe_injector.py
@@ -0,0 +1,18 @@
+# Usage: mitmdump -s "iframe_injector.py url"
+# (this script works best with --anticache)
+from libmproxy.protocol.http import decoded
+
+
+def start(ctx, argv):
+ if len(argv) != 2:
+ raise ValueError('Usage: -s "iframe_injector.py url"')
+ ctx.iframe_url = argv[1]
+
+
+def handle_response(ctx, flow):
+ with decoded(flow.response): # Remove content encoding (gzip, ...)
+ c = flow.response.replace(
+ '',
+ '' % ctx.iframe_url)
+ if c > 0:
+ ctx.log("Iframe injected!")
\ No newline at end of file
diff --git a/test/test_examples.py b/test/test_examples.py
index d18b58625..d557080ef 100644
--- a/test/test_examples.py
+++ b/test/test_examples.py
@@ -12,6 +12,8 @@ def test_load_scripts():
tmaster = tservers.TestMaster(config.ProxyConfig())
for f in scripts:
+ if "iframe_injector" in f:
+ f += " foo" # one argument required
if "modify_response_body" in f:
f += " foo bar" # two arguments required
script.Script(f, tmaster) # Loads the script file.
\ No newline at end of file