1 files changed, 110 insertions, 0 deletions
diff --git a/libmproxy/cmdline.py b/libmproxy/cmdline.py
new file mode 100644
index 00000000..789cfaec
--- /dev/null
+++ b/libmproxy/cmdline.py
@@ -0,0 +1,110 @@
+import proxy
+import optparse
+
+
+def get_common_options(options):
+    stickycookie = None
+    if options.stickycookie_all:
+        stickycookie = ".*"
+    elif options.stickycookie_filt:
+        stickycookie = stickycookie_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,
+        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(
+        "-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(
+        "-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)