mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-26 18:18:25 +00:00
9af8f4bb31
This commit is largely based on work by Thiago Arrais (@thiagoarrais) and Shane Bradfield (@l33tLumberjack). I wasn't really able to get their PR reasonably merged onto the latest master, so I reapplied their changes manually here and did some further improvements on that.
30 lines
822 B
Python
30 lines
822 B
Python
# 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])
|