aboutsummaryrefslogtreecommitdiffstats
path: root/libmproxy
diff options
context:
space:
mode:
authorMarcelo Glezer <mg@tekii.com.ar>2015-01-05 18:12:38 -0300
committerMarcelo Glezer <mg@tekii.com.ar>2015-01-05 18:12:38 -0300
commit9695377b283f97c872e4ddbb27060883c3c32980 (patch)
tree2ffb9ccfd3afdc32f2eddd90f8610d9c5eb682d5 /libmproxy
parent812ea90dc5c7ccc6bb5729f74d7a85f826d9f3cb (diff)
downloadmitmproxy-9695377b283f97c872e4ddbb27060883c3c32980.tar.gz
mitmproxy-9695377b283f97c872e4ddbb27060883c3c32980.tar.bz2
mitmproxy-9695377b283f97c872e4ddbb27060883c3c32980.zip
first steps in scenario support
Diffstat (limited to 'libmproxy')
-rw-r--r--libmproxy/cmdline.py4
-rw-r--r--libmproxy/dump.py10
2 files changed, 8 insertions, 6 deletions
diff --git a/libmproxy/cmdline.py b/libmproxy/cmdline.py
index 185944da..e45cc54c 100644
--- a/libmproxy/cmdline.py
+++ b/libmproxy/cmdline.py
@@ -418,14 +418,14 @@ def common_options(parser):
group = parser.add_argument_group("Client Replay")
group.add_argument(
"-c", "--client-replay",
- action="store", dest="client_replay", default=None, metavar="PATH",
+ action="append", dest="client_replay", default=None, metavar="PATH",
help="Replay client requests from a saved file."
)
group = parser.add_argument_group("Server Replay")
group.add_argument(
"-S", "--server-replay",
- action="store", dest="server_replay", default=None, metavar="PATH",
+ action="append", dest="server_replay", default=None, metavar="PATH",
help="Replay server responses from a saved file."
)
group.add_argument(
diff --git a/libmproxy/dump.py b/libmproxy/dump.py
index 91ccb42b..c4a70a14 100644
--- a/libmproxy/dump.py
+++ b/libmproxy/dump.py
@@ -143,11 +143,13 @@ class DumpMaster(flow.FlowMaster):
if self.o.app:
self.start_app(self.o.app_host, self.o.app_port)
- def _readflow(self, path):
- path = os.path.expanduser(path)
+ def _readflow(self, paths):
try:
- f = file(path, "rb")
- flows = list(flow.FlowReader(f).stream())
+ flows = []
+ for path in paths:
+ path = os.path.expanduser(path)
+ f = file(path, "rb")
+ flows.extend(list(flow.FlowReader(f).stream()))
except (IOError, flow.FlowReadError), v:
raise DumpError(v.strerror)
return flows