def __init__(self, server, options, sessionFactory): FlowMaster.__init__(self, server, flow.State()) self.sessionFactory = sessionFactory self.o = options self.flows = FlowCollection() self.anticache = options.anticache self.anticomp = options.anticomp if options.stickycookie: self.set_stickycookie(options.stickycookie) if options.stickyauth: self.set_stickyauth(options.stickyauth) if options.wfile: path = os.path.expanduser(options.wfile) directory = os.path.split(path)[0] if not os.path.exists(directory): os.makedirs(directory) try: f = file(path, "wb") self.fwriter = flow.FlowWriter(f) except IOError, v: raise Exception(v.strerror)
def __init__(self, server, options, sessionFactory): FlowMaster.__init__(self, server, flow.State()) self.sessionFactory = sessionFactory self.o = options self.flows = FlowCollection() self.anticache = options.anticache self.anticomp = options.anticomp if options.stickycookie: self.set_stickycookie(options.stickycookie) if options.stickyauth: self.set_stickyauth(options.stickyauth) if options.wfile: path = os.path.abspath(os.path.expanduser(options.wfile)) directory = os.path.split(path)[0] if not os.path.exists(directory): os.makedirs(directory) try: f = file(path, "wb") self.fwriter = flow.FlowWriter(f) except IOError, v: raise Exception(v.strerror)
def handle_request(self, request): flow = FlowMaster.handle_request(self, request) if flow: request._ack() #print "request to "+request.host #self.sessionFactory.write(request.host) return flow
def handle_request(self, request): flow = FlowMaster.handle_request(self, request) if flow: request.reply() #print "request to "+request.host #self.sessionFactory.write(request.host) return flow
def handle_response(self, response): flow = FlowMaster.handle_response(self, response) if flow: response._ack() flowId = self.flows.addFlow(flow) if self.o.wfile: self.fwriter.add(flow) if self.o.dumpdir: self.dirdumper.add(flow) #print "response from "+flow.request.host self.sessionFactory.onNewFlow(self.flows.getFlowsSerialized()[flowId]) return flow
def handle_response(self, response): flow = FlowMaster.handle_response(self, response) if flow: response.reply() flowId = self.flows.addFlow(flow) if self.o.wfile: self.fwriter.add(flow) if self.o.dumpdir: self.dirdumper.add(flow) #print "response from "+flow.request.host self.sessionFactory.onNewFlow( self.flows.getFlowsSerialized()[flowId]) return flow
def shutdown(self): if(self.o.wfile): self.fwriter.fo.close() return FlowMaster.shutdown(self)
def shutdown(self): if (self.o.wfile): self.fwriter.fo.close() return FlowMaster.shutdown(self)