mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2025-02-02 00:05:27 +00:00
c726519e73
This allows us to replay an HTTP Authorization header, in the same way as we replay cookies using stickycookies. This lets us conveniently get at HTTP Basic Auth protected resources through the proxy, but is not enough to do the same for HTTP Digest auth. We'll put that on the todo list.
133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
import proxy
|
|
import optparse
|
|
|
|
|
|
def get_common_options(options):
|
|
stickycookie, stickyauth = None, None
|
|
if options.stickycookie_all:
|
|
stickycookie = ".*"
|
|
elif options.stickycookie_filt:
|
|
stickycookie = options.stickycookie_filt
|
|
|
|
if options.stickyauth_all:
|
|
stickyauth = ".*"
|
|
elif options.stickyauth_filt:
|
|
stickyauth = options.stickyauth_filt
|
|
|
|
return dict(
|
|
verbosity = options.verbose,
|
|
wfile = options.wfile,
|
|
request_script = options.request_script,
|
|
response_script = options.response_script,
|
|
server_replay = options.server_replay,
|
|
kill = options.kill,
|
|
rheaders = options.rheaders,
|
|
client_replay = options.client_replay,
|
|
stickycookie = stickycookie,
|
|
stickyauth = stickyauth,
|
|
anticache = options.anticache,
|
|
refresh_server_playback = not options.norefresh,
|
|
)
|
|
|
|
|
|
def common_options(parser):
|
|
parser.add_option(
|
|
"-a",
|
|
action="store", type = "str", dest="addr", default='',
|
|
help = "Address to bind proxy to (defaults to all interfaces)"
|
|
)
|
|
parser.add_option(
|
|
"--confdir",
|
|
action="store", type = "str", dest="confdir", default='~/.mitmproxy',
|
|
help = "Configuration directory. (~/.mitmproxy)"
|
|
)
|
|
parser.add_option(
|
|
"-p",
|
|
action="store", type = "int", dest="port", default=8080,
|
|
help = "Proxy service port."
|
|
)
|
|
parser.add_option(
|
|
"-q",
|
|
action="store_true", dest="quiet",
|
|
help="Quiet."
|
|
)
|
|
parser.add_option(
|
|
"--anticache",
|
|
action="store_true", dest="anticache", default=False,
|
|
help="Strip out request headers that might cause the server to return 304-not-modified."
|
|
)
|
|
parser.add_option(
|
|
"--reqscript",
|
|
action="store", dest="request_script", default=None,
|
|
help="Script to run when a request is recieved."
|
|
)
|
|
parser.add_option(
|
|
"--respscript",
|
|
action="store", dest="response_script", default=None,
|
|
help="Script to run when a response is recieved."
|
|
)
|
|
parser.add_option(
|
|
"-t",
|
|
action="store_true", dest="stickycookie_all", default=None,
|
|
help="Set sticky cookie for all requests."
|
|
)
|
|
parser.add_option(
|
|
"-T",
|
|
action="store", dest="stickycookie_filt", default=None, metavar="FILTER",
|
|
help="Set sticky cookie filter. Matched against requests."
|
|
)
|
|
parser.add_option(
|
|
"-u",
|
|
action="store_true", dest="stickyauth_all", default=None,
|
|
help="Set sticky auth for all requests."
|
|
)
|
|
parser.add_option(
|
|
"-U",
|
|
action="store", dest="stickyauth_filt", default=None, metavar="FILTER",
|
|
help="Set sticky auth filter. Matched against requests."
|
|
)
|
|
parser.add_option(
|
|
"-v",
|
|
action="count", dest="verbose", default=1,
|
|
help="Increase verbosity. Can be passed multiple times."
|
|
)
|
|
parser.add_option(
|
|
"-w",
|
|
action="store", dest="wfile", default=None,
|
|
help="Write flows to file."
|
|
)
|
|
group = optparse.OptionGroup(parser, "Client Replay")
|
|
group.add_option(
|
|
"-c",
|
|
action="store", dest="client_replay", default=None, metavar="PATH",
|
|
help="Replay client requests from a saved file."
|
|
)
|
|
parser.add_option_group(group)
|
|
|
|
group = optparse.OptionGroup(parser, "Server Replay")
|
|
group.add_option(
|
|
"-s",
|
|
action="store", dest="server_replay", default=None, metavar="PATH",
|
|
help="Replay server responses from a saved file."
|
|
)
|
|
group.add_option(
|
|
"-k",
|
|
action="store_true", dest="kill", default=False,
|
|
help="Kill extra requests during replay."
|
|
)
|
|
group.add_option(
|
|
"--rheader",
|
|
action="append", dest="rheaders", type="str",
|
|
help="Request headers to be considered during replay. "
|
|
"Can be passed multiple times."
|
|
)
|
|
group.add_option(
|
|
"--norefresh",
|
|
action="store_true", dest="norefresh", default=False,
|
|
help= "Disable response refresh, "
|
|
"which updates times in cookies and headers for replayed responses."
|
|
)
|
|
parser.add_option_group(group)
|
|
|
|
proxy.certificate_option_group(parser)
|