Made it possible to modify the server_conn.connection attribute,

using the serverconnect stub.

1 files changed, 7 insertions, 2 deletions

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)