Review changes

pymobiledevice3/tcp_forwarder.py

@@ -55,7 +55,7 @@ def start(self, address='0.0.0.0'):
             # as synchronous blocking
             readable, writable, exceptional = select.select(self.inputs, [], self.inputs, self.TIMEOUT)
             if self.stopped.is_set():
-                self.logger.info("Closing since stopped is set")
+                self.logger.debug("Closing since stopped is set")
                 break
 
             closed_sockets = set()
@@ -95,18 +95,20 @@ def _handle_close_or_error(self, from_sock):
 
     def _handle_data(self, from_sock, closed_sockets):
         self.logger.debug("Handling data from %s", from_sock)
+        data = None
         try:
             data = from_sock.recv(1024)
         except OSError:
-            self.logger.debug("oserror when reading from_sock")
-            self._handle_close_or_error(from_sock)
-            closed_sockets.add(from_sock)
-            closed_sockets.add(self.connections[from_sock])
-            return
+            # Socket closing is handled in another if block
+            pass
 
-        if len(data) == 0:
-            # no data means socket was closed
-            self.logger.info("No data was read from the socket")
+        if data is None or len(data) == 0:
+            if data is None:
+                # data is none means we had an error reading from socket
+                self.logger.debug("oserror when reading from_sock")
+            else:
+                # Empty data means socket was closed
+                self.logger.info("No data was read from the socket")
             self._handle_close_or_error(from_sock)
             closed_sockets.add(from_sock)
             closed_sockets.add(self.connections[from_sock])