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 __init__(self, filter_id, package_name, description): self.report = {} self.filter_id = filter_id self.package_name = package_name self.description = description config = proxy.ProxyConfig(port=8080,mode="transparent") server = ProxyServer(config) controller.Master.__init__(self, server) flow_dump_file = open(self.get_package_name()+"_network_traffic", "wb") self.network_flow = FlowWriter(flow_dump_file) self.should_exit = None self.extra_analyzers = [InsecureTransmissionAnalyzer(self),ZIPPathTraversalAnalyzer(self)] logging.debug("Init analyzer")