aboutsummaryrefslogtreecommitdiffstats
path: root/examples/flowwriter.py
blob: 2c253fbb2c1ed7c54e078aad8c4f6d32dc510fe8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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)