mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-26 18:18:25 +00:00
e6eeab6094
- Addons now nest, which means that addons can manage addons. This has a number of salutary effects - the scripts addon no longer has to poke into the global addons list, we no longer have to replace/remove/boot-outof parent addons when we load scripts, and this paves the way for making our top-level tools into addons themselves. - All addon calls are now wrapped in a safe execution environment where exceptions are caught, and output to stdout/stderr are intercepted and turned into logs. - We no longer support script arguments in sys.argv - creating an option properly is the only way to pass arguments. This means that all scripts are always directly controllable from interctive tooling, and that arguments are type-checked. For now, I've disabled testing of the har dump example - it needs to be moved to the new argument handling, and become a class addon. I'll address that in a separate patch.
32 lines
841 B
Python
32 lines
841 B
Python
# (this script works best with --anticache)
|
|
from bs4 import BeautifulSoup
|
|
|
|
|
|
class Injector:
|
|
def __init__(self):
|
|
self.iframe_url = None
|
|
|
|
def load(self, loader):
|
|
loader.add_option(
|
|
"iframe", str, "", "IFrame to inject"
|
|
)
|
|
|
|
def configure(self, options, updated):
|
|
self.iframe_url = options.iframe
|
|
|
|
def response(self, flow):
|
|
if self.iframe_url:
|
|
html = BeautifulSoup(flow.response.content, "html.parser")
|
|
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")
|
|
|
|
|
|
addons = [Injector()]
|