From 2ecd89fc51676a98c25a80857584923aae9248a1 Mon Sep 17 00:00:00 2001 From: smill Date: Mon, 5 Sep 2016 10:49:39 +0000 Subject: Made it possible to modify the server_conn.connection attribute, using the serverconnect stub. --- netlib/tcp.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'netlib/tcp.py') diff --git a/netlib/tcp.py b/netlib/tcp.py index 1fd0164f..c3b8a407 100644 --- a/netlib/tcp.py +++ b/netlib/tcp.py @@ -729,10 +729,15 @@ class TCPClient(_Connection): def connect(self): try: - connection = socket.socket(self.address.family, socket.SOCK_STREAM) + if not self.connection: + connection = socket.socket(self.address.family, socket.SOCK_STREAM) + else: + connection = self.connection + if self.spoof_source_address: try: - connection.setsockopt(socket.SOL_IP, 19, 1) + if not connection.getsockopt(socket.SOL_IP, 19): + connection.setsockopt(socket.SOL_IP, 19, 1) except socket.error as e: raise exceptions.ProtocolException( "Failed to spoof the source address: " + e.strerror) -- cgit v1.2.3