Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 10:16:27 +00:00
Code Issues Projects Releases Wiki Activity

minor cleanup

Browse Source
This commit is contained in:
Maximilian Hils 2016-09-22 01:58:01 -07:00
parent ec588f8822
commit e5b79a6d72
3 changed files with 11 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mitmproxy/protocol/base.py
View File

@ -163,8 +163,11 @@ class ServerConnectionMixin(object):
self.server_conn.close()
self.channel.tell("serverdisconnect", self.server_conn)
self.server_conn = models.ServerConnection(address,
(self.server_conn.source_address.host, 0), self.config.options.spoof_source_address)
self.server_conn = models.ServerConnection(
address,
(self.server_conn.source_address.host, 0),
self.config.options.spoof_source_address
)
def connect(self):
"""

4
netlib/exceptions.py
View File

@ -58,7 +58,3 @@ class InvalidCertificateException(TlsException):
class Timeout(TcpException):
pass
class ProtocolException(NetlibException):
pass

17
netlib/tcp.py
View File

@ -484,14 +484,12 @@ class _Connection(object):
if not isinstance(self.connection, SSL.Connection):
if not getattr(self.wfile, "closed", False):
try:
if self.wfile:
self.wfile.flush()
self.wfile.close()
self.wfile.flush()
self.wfile.close()
except exceptions.TcpDisconnect:
pass
if self.rfile:
self.rfile.close()
self.rfile.close()
else:
try:
self.connection.shutdown()
@ -731,11 +729,7 @@ class TCPClient(_Connection):
def connect(self):
try:
# Allow the socket to be manipulated by using the server_conn stub.
if not self.connection:
connection = socket.socket(self.address.family, socket.SOCK_STREAM)
else:
connection = self.connection
connection = socket.socket(self.address.family, socket.SOCK_STREAM)
if self.spoof_source_address:
try:
@ -744,7 +738,8 @@ class TCPClient(_Connection):
connection.setsockopt(socket.SOL_IP, 19, 1)
except socket.error as e:
raise exceptions.TcpException(
"Failed to spoof the source address: " + e.strerror)
"Failed to spoof the source address: " + e.strerror
)
if self.source_address:
connection.bind(self.source_address())
connection.connect(self.address())