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 +++++++++++++++++ test/test_examples.py | 2 +- 2 files changed, 18 insertions(+), 1 deletion(-) create mode 100644 examples/flowwriter.py 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 diff --git a/test/test_examples.py b/test/test_examples.py index daf4b902..fd901b5d 100644 --- a/test/test_examples.py +++ b/test/test_examples.py @@ -11,7 +11,7 @@ def test_load_scripts(): tmaster = tservers.TestMaster(config.ProxyConfig()) for f in scripts: - if "har_extractor" in f: + if "har_extractor" in f or "flowwriter" in f: f += " -" if "iframe_injector" in f: f += " foo" # one argument required -- cgit v1.2.3