Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

Add a finished flag to BaseHandler, and catch an extra OpenSSL exception.

Browse Source
This commit is contained in:
Aldo Cortesi 2012-06-25 11:34:10 +12:00
parent f3237503a7
commit 47f862ae27
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
netlib/tcp.py
View File

@ -20,7 +20,7 @@ class FileLike:
while len(result) < length:
try:
data = self.o.read(length)
except SSL.ZeroReturnError:
except (SSL.ZeroReturnError, SSL.SysCallError):
break
if not data:
break
@ -86,6 +86,7 @@ class BaseHandler:
self.client_address = client_address
self.server = server
self.finished = False
def convert_to_ssl(self, cert, key):
ctx = SSL.Context(SSL.SSLv23_METHOD)
@ -97,6 +98,7 @@ class BaseHandler:
self.wfile = FileLike(self.connection)
def finish(self):
self.finished = True
try:
if not getattr(self.wfile, "closed", False):
self.wfile.flush()