fix(safe-client): check safe client for nil (#154)

0xPolygon · Feb 5, 2025 · 2fe6df8 · 2fe6df8
1 parent 59e340f
commit 2fe6df8
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/datastreamer/streamserver.go b/datastreamer/streamserver.go
@@ -336,9 +336,17 @@ func (s *StreamServer) handleConnection(conn net.Conn) {
 			return
 		}
 
+		// Check if the client is nil
+		safeClient := s.getSafeClient(clientID)
+		if safeClient == nil {
+			log.Errorf("Client %s is nil", clientID)
+			s.killClient(clientID)
+			return
+		}
+
 		// Manage the requested command
 		log.Debugf("Command %d[%s] received from %s", command, StrCommand[Command(command)], clientID)
-		err = s.processCommand(Command(command), s.getSafeClient(clientID))
+		err = s.processCommand(Command(command), safeClient)
 		if err != nil {
 			log.Errorf("Error processing command %d[%s] from %s: %v", command, StrCommand[Command(command)], clientID, err)
 		}