From c0a318566ae0fb5a264bfaa130f3ff3a09ea5bf5 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Sun, 12 Apr 2015 03:47:58 +0200 Subject: add flowwriter example --- examples/flowwriter.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 examples/flowwriter.py (limited to 'examples') 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 -- cgit v1.2.3