2015-04-12 01:47:58 +00:00
|
|
|
import random
|
|
|
|
|
import sys
|
|
|
|
|
|
2016-02-16 19:49:10 +00:00
|
|
|
from mitmproxy.flow import FlowWriter
|
2015-04-12 01:50:14 +00:00
|
|
|
|
2016-07-22 23:57:31 +00:00
|
|
|
|
|
|
|
|
class Writer:
|
|
|
|
|
def __init__(self, path):
|
|
|
|
|
if path == "-":
|
|
|
|
|
f = sys.stdout
|
|
|
|
|
else:
|
|
|
|
|
f = open(path, "wb")
|
|
|
|
|
self.w = FlowWriter(f)
|
|
|
|
|
|
|
|
|
|
def response(self, flow):
|
|
|
|
|
if random.choice([True, False]):
|
|
|
|
|
self.w.add(flow)
|
2015-04-12 01:50:14 +00:00
|
|
|
|
2016-07-08 01:37:33 +00:00
|
|
|
|
|
|
|
|
def start():
|
2016-06-14 01:17:09 +00:00
|
|
|
if len(sys.argv) != 2:
|
2015-04-12 01:47:58 +00:00
|
|
|
raise ValueError('Usage: -s "flowriter.py filename"')
|
2016-07-22 23:57:31 +00:00
|
|
|
return Writer(sys.argv[1])
|
