diff options
| author | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2015-07-30 13:52:50 +0200 |
|---|---|---|
| committer | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2015-07-30 19:53:29 +0200 |
| commit | 4f38c6b90e239d192863dee271e267b498c72206 (patch) | |
| tree | f41642541100a191cc2a92ce3d5a293697f7dae0 /libmproxy/proxy | |
| parent | a5d9e1f44dbe0fc6fee174b1953806f9b148b5ad (diff) | |
| download | mitmproxy-4f38c6b90e239d192863dee271e267b498c72206.tar.gz mitmproxy-4f38c6b90e239d192863dee271e267b498c72206.tar.bz2 mitmproxy-4f38c6b90e239d192863dee271e267b498c72206.zip | |
attach application protocol to connection
Diffstat (limited to 'libmproxy/proxy')
| -rw-r--r-- | libmproxy/proxy/connection.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/libmproxy/proxy/connection.py b/libmproxy/proxy/connection.py index 54b3688e..a0bf2af9 100644 --- a/libmproxy/proxy/connection.py +++ b/libmproxy/proxy/connection.py @@ -23,6 +23,7 @@ class ClientConnection(tcp.BaseHandler, stateobject.StateObject): self.timestamp_start = utils.timestamp() self.timestamp_end = None self.timestamp_ssl_setup = None + self.protocol = None def __repr__(self): return "<ClientConnection: {ssl}{host}:{port}>".format( @@ -58,6 +59,8 @@ class ClientConnection(tcp.BaseHandler, stateobject.StateObject): return copy.copy(self) def send(self, message): + if isinstance(message, list): + message = b''.join(message) self.wfile.write(message) self.wfile.flush() @@ -93,6 +96,7 @@ class ServerConnection(tcp.TCPClient, stateobject.StateObject): self.timestamp_end = None self.timestamp_tcp_setup = None self.timestamp_ssl_setup = None + self.protocol = None def __repr__(self): if self.ssl_established and self.sni: @@ -157,6 +161,8 @@ class ServerConnection(tcp.TCPClient, stateobject.StateObject): self.timestamp_tcp_setup = utils.timestamp() def send(self, message): + if isinstance(message, list): + message = b''.join(message) self.wfile.write(message) self.wfile.flush() |