mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-26 18:18:25 +00:00
33 lines
678 B
Python
33 lines
678 B
Python
|
import typing
|
||
|
|
|
||
|
|
from mitmproxy import command
|
||
|
|
from mitmproxy import ctx
|
||
|
|
from mitmproxy import flow
|
||
|
|
from mitmproxy import types
|
||
|
|
|
||
|
|
|
||
|
|
class MyAddon:
|
||
|
|
def __init__(self):
|
||
|
|
self.num = 0
|
||
|
|
|
||
|
|
@command.command("myaddon.histogram")
|
||
|
|
def histogram(
|
||
|
|
self,
|
||
|
|
flows: typing.Sequence[flow.Flow],
|
||
|
|
path: types.Path,
|
||
|
|
) -> None:
|
||
|
|
totals = {}
|
||
|
|
for f in flows:
|
||
|
|
totals[f.request.host] = totals.setdefault(f.request.host, 0) + 1
|
||
|
|
|
||
|
|
fp = open(path, "w+")
|
||
|
|
for cnt, dom in sorted([(v, k) for (k, v) in totals.items()]):
|
||
|
|
fp.write("%s: %s\n" % (cnt, dom))
|
||
|
|
|
||
|
|
ctx.log.alert("done")
|
||
|
|
|
||
|
|
|
||
|
|
addons = [
|
||
|
|
MyAddon()
|
||
|
|
]
|