Handle ip change on onion connection correctly

src/Connection/Connection.py

@@ -162,8 +162,8 @@ def messageLoop(self):
         self.connected = True
         buff_len = 0
         req_len = 0
-
         unpacker_bytes = 0
+
         try:
             while not self.closed:
                 buff = self.sock.recv(64 * 1024)
@@ -330,7 +330,11 @@ def setHandshake(self, handshake):
             self.port = handshake["fileserver_port"]  # Set peer fileserver port
 
         if handshake.get("onion") and not self.ip.endswith(".onion"):  # Set incoming connection's onion address
+            if self.server.ips.get(self.ip) == self:
+                del self.server.ips[self.ip]
             self.ip = handshake["onion"] + ".onion"
+            self.log("Changing ip to %s" % self.ip)
+            self.server.ips[self.ip] = self
             self.updateName()
 
         # Check if we can encrypt the connection