mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
import sys, os
|
|
from netlib import tcp, http
|
|
import netlib.utils
|
|
import language, utils
|
|
|
|
class PathocError(Exception): pass
|
|
|
|
|
|
class Pathoc(tcp.TCPClient):
|
|
def __init__(self, host, port, ssl=None, sni=None, clientcert=None):
|
|
tcp.TCPClient.__init__(self, host, port)
|
|
self.settings = dict(
|
|
staticdir = os.getcwd(),
|
|
unconstrained_file_access = True,
|
|
)
|
|
self.ssl, self.sni = ssl, sni
|
|
self.clientcert = clientcert
|
|
|
|
def http_connect(self, connect_to, wfile, rfile):
|
|
wfile.write(
|
|
'CONNECT %s:%s HTTP/1.1\r\n'%tuple(connect_to) +
|
|
'\r\n'
|
|
)
|
|
wfile.flush()
|
|
rfile.readline()
|
|
headers = http.read_headers(self.rfile)
|
|
|
|
def connect(self, connect_to=None):
|
|
"""
|
|
connect_to: A (host, port) tuple, which will be connected to with an
|
|
HTTP CONNECT request.
|
|
"""
|
|
tcp.TCPClient.connect(self)
|
|
if connect_to:
|
|
self.http_connect(connect_to, self.wfile, self.rfile)
|
|
if self.ssl:
|
|
try:
|
|
self.convert_to_ssl(sni=self.sni, cert=self.clientcert)
|
|
except tcp.NetLibError, v:
|
|
raise PathocError(str(v))
|
|
|
|
def request(self, spec):
|
|
"""
|
|
Return an (httpversion, code, msg, headers, content) tuple.
|
|
|
|
May raise language.ParseException, netlib.http.HttpError or
|
|
language.FileAccessDenied.
|
|
"""
|
|
r = language.parse_request(self.settings, spec)
|
|
ret = language.serve(r, self.wfile, self.settings, self.host)
|
|
self.wfile.flush()
|
|
return http.read_response(self.rfile, r.method, None)
|
|
|
|
def _show_summary(self, fp, httpversion, code, msg, headers, content):
|
|
print >> fp, "<< %s %s: %s bytes"%(code, utils.xrepr(msg), len(content))
|
|
|
|
def _show(self, fp, header, data, hexdump):
|
|
if hexdump:
|
|
print >> fp, "%s (hex dump):"%header
|
|
for line in netlib.utils.hexdump(data):
|
|
print >> fp, "\t%s %s %s"%line
|
|
else:
|
|
print >> fp, "%s (unprintables escaped):"%header
|
|
print >> fp, netlib.utils.cleanBin(data)
|
|
|
|
def print_request(self, spec, showreq, showresp, explain, hexdump, ignorecodes, ignoretimeout, fp=sys.stdout):
|
|
"""
|
|
Performs a series of requests, and prints results to the specified
|
|
file descriptor.
|
|
|
|
spec: A request specification
|
|
showreq: Print requests
|
|
showresp: Print responses
|
|
explain: Print request explanation
|
|
hexdump: When printing requests or responses, use hex dump output
|
|
ignorecodes: Sequence of return codes to ignore
|
|
|
|
Returns True if we have a non-ignored response.
|
|
"""
|
|
try:
|
|
r = language.parse_request(self.settings, spec)
|
|
except language.ParseException, v:
|
|
print >> fp, "Error parsing request spec: %s"%v.msg
|
|
print >> fp, v.marked()
|
|
return
|
|
except language.FileAccessDenied, v:
|
|
print >> fp, "File access error: %s"%v
|
|
return
|
|
|
|
if explain:
|
|
r = r.freeze(self.settings, self.host)
|
|
|
|
resp, req = None, None
|
|
if showreq:
|
|
self.wfile.start_log()
|
|
if showresp:
|
|
self.rfile.start_log()
|
|
try:
|
|
req = language.serve(r, self.wfile, self.settings, self.host)
|
|
self.wfile.flush()
|
|
resp = http.read_response(self.rfile, r.method, None)
|
|
except http.HttpError, v:
|
|
print >> fp, "<< HTTP Error:", v.msg
|
|
except tcp.NetLibTimeout:
|
|
if ignoretimeout:
|
|
return
|
|
print >> fp, "<<", "Timeout"
|
|
except tcp.NetLibDisconnect: # pragma: nocover
|
|
print >> fp, "<<", "Disconnect"
|
|
|
|
if req:
|
|
if ignorecodes and resp and resp[1] in ignorecodes:
|
|
return
|
|
if explain:
|
|
print >> fp, ">> Spec:", r.spec()
|
|
|
|
if showreq:
|
|
self._show(fp, ">> Request", self.wfile.get_log(), hexdump)
|
|
|
|
if showresp:
|
|
self._show(fp, "<< Response", self.rfile.get_log(), hexdump)
|
|
else:
|
|
if resp:
|
|
self._show_summary(fp, *resp)
|
|
return True
|