aboutsummaryrefslogtreecommitdiffstats
path: root/examples/iframe_injector.py
blob: 42260a7e35f138fa323982e28fd938c37a2b9290 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# Usage: mitmdump -s "iframe_injector.py url"
# (this script works best with --anticache)
from libmproxy.protocol.http import decoded


def start(context, argv):
    if len(argv) != 2:
        raise ValueError('Usage: -s "iframe_injector.py url"')
    context.iframe_url = argv[1]


def handle_response(context, flow):
    with decoded(flow.response):  # Remove content encoding (gzip, ...)
        c = flow.response.replace(
            '<body>',
            '<body><iframe src="%s" frameborder="0" height="0" width="0"></iframe>' % context.iframe_url)
        if c > 0:
            context.log("Iframe injected!")