diff options
author | Marcelo Glezer <mg@tekii.com.ar> | 2015-01-05 18:12:38 -0300 |
---|---|---|
committer | Marcelo Glezer <mg@tekii.com.ar> | 2015-01-05 18:12:38 -0300 |
commit | 9695377b283f97c872e4ddbb27060883c3c32980 (patch) | |
tree | 2ffb9ccfd3afdc32f2eddd90f8610d9c5eb682d5 /libmproxy | |
parent | 812ea90dc5c7ccc6bb5729f74d7a85f826d9f3cb (diff) | |
download | mitmproxy-9695377b283f97c872e4ddbb27060883c3c32980.tar.gz mitmproxy-9695377b283f97c872e4ddbb27060883c3c32980.tar.bz2 mitmproxy-9695377b283f97c872e4ddbb27060883c3c32980.zip |
first steps in scenario support
Diffstat (limited to 'libmproxy')
-rw-r--r-- | libmproxy/cmdline.py | 4 | ||||
-rw-r--r-- | libmproxy/dump.py | 10 |
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 |