aboutsummaryrefslogtreecommitdiffstats
path: root/examples/iframe_injector.py
blob: 352c3c24a26054ae97d7c632afa05d9e8ea13277 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# Usage: mitmdump -s "iframe_injector.py url"
# (this script works best with --anticache)
import sys
from bs4 import BeautifulSoup

iframe_url = None


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


def response(flow):
    if flow.request.host in iframe_url:
        return
    html = BeautifulSoup(flow.response.content, "lxml")
    if html.body:
        iframe = html.new_tag(
            "iframe",
            src=iframe_url,
            frameborder=0,
            height=0,
            width=0)
        html.body.insert(0, iframe)
        flow.response.content = str(html).encode("utf8")