diff options
Diffstat (limited to 'mitmproxy/addons/dumper.py')
| -rw-r--r-- | mitmproxy/addons/dumper.py | 26 |
1 files changed, 25 insertions, 1 deletions
diff --git a/mitmproxy/addons/dumper.py b/mitmproxy/addons/dumper.py index 89a9eab8..29f60cfe 100644 --- a/mitmproxy/addons/dumper.py +++ b/mitmproxy/addons/dumper.py @@ -223,6 +223,29 @@ class Dumper: if self.match(f): self.echo_flow(f) + def websocket_error(self, f): + self.echo( + "Error in WebSocket connection to {}: {}".format( + repr(f.server_conn.address), f.error + ), + fg="red" + ) + + def websocket_message(self, f): + if self.match(f): + message = f.messages[-1] + self.echo(message.info) + if self.flow_detail >= 3: + self._echo_message(message) + + def websocket_end(self, f): + if self.match(f): + self.echo("WebSocket connection closed by {}: {} {}, {}".format( + f.close_sender, + f.close_code, + f.close_message, + f.close_reason)) + def tcp_error(self, f): self.echo( "Error in TCP connection to {}: {}".format( @@ -240,4 +263,5 @@ class Dumper: server=repr(f.server_conn.address), direction=direction, )) - self._echo_message(message) + if self.flow_detail >= 3: + self._echo_message(message) |