diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2018-03-31 19:50:36 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@corte.si> | 2018-04-01 09:46:32 +1200 |
commit | b6d943cfa3a226651e705ff45aa7154010ea29ba (patch) | |
tree | bc8aabd32dba9fefcace1bb0d943f8a71f50ba54 | |
parent | 54170ee6572e8ba38b94a4e51f3c8e832e5f9ac7 (diff) | |
download | mitmproxy-b6d943cfa3a226651e705ff45aa7154010ea29ba.tar.gz mitmproxy-b6d943cfa3a226651e705ff45aa7154010ea29ba.tar.bz2 mitmproxy-b6d943cfa3a226651e705ff45aa7154010ea29ba.zip |
asyncio: fix client replay
-rw-r--r-- | mitmproxy/master.py | 8 | ||||
-rw-r--r-- | mitmproxy/proxy/protocol/http_replay.py | 18 |
2 files changed, 3 insertions, 23 deletions
diff --git a/mitmproxy/master.py b/mitmproxy/master.py index a23be0a7..c2d2b6a0 100644 --- a/mitmproxy/master.py +++ b/mitmproxy/master.py @@ -202,13 +202,7 @@ class Master: host = f.request.headers.pop(":authority") f.request.headers.insert(0, "host", host) - rt = http_replay.RequestReplayThread( - self.options, - f, - asyncio.get_event_loop(), - self.event_queue, - self.should_exit - ) + rt = http_replay.RequestReplayThread(self.options, f, self.server.channel) rt.start() # pragma: no cover if block: rt.join() diff --git a/mitmproxy/proxy/protocol/http_replay.py b/mitmproxy/proxy/protocol/http_replay.py index 8682f50e..b2cca2b1 100644 --- a/mitmproxy/proxy/protocol/http_replay.py +++ b/mitmproxy/proxy/protocol/http_replay.py @@ -1,8 +1,3 @@ -import asyncio -import queue -import threading -import typing - from mitmproxy import log from mitmproxy import controller from mitmproxy import exceptions @@ -26,21 +21,12 @@ class RequestReplayThread(basethread.BaseThread): self, opts: options.Options, f: http.HTTPFlow, - loop: asyncio.AbstractEventLoop, - event_queue: typing.Optional[queue.Queue], - should_exit: threading.Event + channel: controller.Channel, ) -> None: - """ - event_queue can be a queue or None, if no scripthooks should be - processed. - """ self.options = opts self.f = f f.live = True - if event_queue: - self.channel = controller.Channel(loop, event_queue) - else: - self.channel = None + self.channel = channel super().__init__( "RequestReplay (%s)" % f.request.url ) |