WASEProxy.py 文件源码

def mitm_request(self, data):
    # Initialize ES connection and index
    res = connections.create_connection(hosts=[args.elasticsearch])
    idx = Index(args.index)
    idx.doc_type(DocHTTPRequestResponse)
    try:
        DocHTTPRequestResponse.init()
        idx.create()
    except:
        pass

        r = HTTPRequest(data)

        # determine url
        if self.is_connect:
            scheme = "https"
        else:
            scheme = "http"
        url = scheme + "://" + self.hostname
        if scheme == "http" and int(self.port) != 80 or scheme == "https" and int(self.port) != 443:
            url += ":" + str(self.port)
        url += self.path

        if args.verbose:
            print(url)

        self.doc = DocHTTPRequestResponse(host=self.hostname, port=int(self.port), protocol=scheme)
        self.doc.meta.index = args.index
        self.doc.request.url = url
        self.doc.request.requestline = r.requestline
        self.doc.request.method = r.command
        self.doc.host = self.hostname
        self.doc.port = int(self.port)
        self.doc.protocol = scheme

        return data