diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2012-06-26 23:52:35 +1200 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2012-06-26 23:52:35 +1200 |
| commit | abe335e57dd2871a6ea6cfe2559f9b29ae0c33bb (patch) | |
| tree | 9e6148b27a345198c50d2a83ade9d0298ed7a23b /netlib/tcp.py | |
| parent | 658c9c0446591e41d6ebdb223c62c00342b83206 (diff) | |
| download | mitmproxy-abe335e57dd2871a6ea6cfe2559f9b29ae0c33bb.tar.gz mitmproxy-abe335e57dd2871a6ea6cfe2559f9b29ae0c33bb.tar.bz2 mitmproxy-abe335e57dd2871a6ea6cfe2559f9b29ae0c33bb.zip | |
Add a flag to track SSL connection establishment.
Diffstat (limited to 'netlib/tcp.py')
| -rw-r--r-- | netlib/tcp.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/netlib/tcp.py b/netlib/tcp.py index aa923fdd..9b1fc65e 100644 --- a/netlib/tcp.py +++ b/netlib/tcp.py @@ -52,6 +52,7 @@ class TCPClient: self.host, self.port = host, port self.connection, self.rfile, self.wfile = None, None, None self.cert = None + self.ssl_established = False def convert_to_ssl(self, clientcert=None, sni=None): context = SSL.Context(SSL.SSLv23_METHOD) @@ -68,6 +69,7 @@ class TCPClient: self.cert = self.connection.get_peer_certificate() self.rfile = FileLike(self.connection) self.wfile = FileLike(self.connection) + self.ssl_established = True def connect(self): try: @@ -94,6 +96,7 @@ class BaseHandler: self.client_address = client_address self.server = server self.finished = False + self.ssl_established = False def convert_to_ssl(self, cert, key): ctx = SSL.Context(SSL.SSLv23_METHOD) @@ -109,6 +112,7 @@ class BaseHandler: raise NetLibError("SSL handshake error: %s"%str(v)) self.rfile = FileLike(self.connection) self.wfile = FileLike(self.connection) + self.ssl_established = True def finish(self): self.finished = True |