diff options
Diffstat (limited to 'examples/flowwriter.py')
-rw-r--r-- | examples/flowwriter.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/examples/flowwriter.py b/examples/flowwriter.py new file mode 100644 index 00000000..2c253fbb --- /dev/null +++ b/examples/flowwriter.py @@ -0,0 +1,17 @@ +from libmproxy.flow import FlowWriter +import random +import sys + +def start(context, argv): + if len(argv) != 2: + raise ValueError('Usage: -s "flowriter.py filename"') + + if argv[1] == "-": + f = sys.stdout + else: + f = open(argv[1], "wb") + context.flow_writer = FlowWriter(f) + +def response(context, flow): + if random.choice([True, False]): + context.flow_writer.add(flow)
\ No newline at end of file |