kylefang/mitmproxy
1
0
Fork 0
mirror of https://github.com/zhigang1992/mitmproxy.git synced 2026-04-28 20:24:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't create fresh FileLike objects when converting to SSL

Browse Source
This commit is contained in:
Aldo Cortesi
2012-09-24 10:47:41 +12:00
parent 1c80c2fdd7
commit 8a6cca530c
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
netlib/tcp.py
View File

@@ -44,6 +44,9 @@ class FileLike:
def __init__(self, o):
self.o = o
def set_descriptor(self, o):
self.o = o
def __getattr__(self, attr):
return getattr(self.o, attr)
@@ -140,8 +143,8 @@ class TCPClient:
except SSL.Error, v:
raise NetLibError("SSL handshake error: %s"%str(v))
self.cert = certutils.SSLCert(self.connection.get_peer_certificate())
self.rfile = FileLike(self.connection)
self.wfile = FileLike(self.connection)
self.rfile.set_descriptor(self.connection)
self.wfile.set_descriptor(self.connection)
def connect(self):
try:
@@ -209,8 +212,8 @@ class BaseHandler:
self.connection.do_handshake()
except SSL.Error, v:
raise NetLibError("SSL handshake error: %s"%str(v))
self.rfile = FileLike(self.connection)
self.wfile = FileLike(self.connection)
self.rfile.set_descriptor(self.connection)
self.wfile.set_descriptor(self.connection)
def finish(self):
self.finished = True