check for sScanf error

DropMorePackets · Feb 16, 2024 · 601cedf · 601cedf
1 parent 0d2674e
commit 601cedf
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/peers/conn.go b/peers/conn.go
@@ -198,12 +198,18 @@ func (h *Handshake) ReadFrom(r io.Reader) (n int64, err error) {
 
 	scanner.Scan()
 	_, err = fmt.Sscanf(scanner.Text(), "%s %s", &h.ProtocolIdentifier, &h.Version)
+	if err != nil {
+		return -1, err
+	}
 
 	scanner.Scan()
 	h.RemotePeer = scanner.Text()
 
 	scanner.Scan()
 	_, err = fmt.Sscanf(scanner.Text(), "%s %d %d", &h.LocalPeerIdentifier, &h.ProcessID, &h.RelativeProcessID)
+	if err != nil {
+		return -1, err
+	}
 
 	//TODO: find out how many bytes where read.
 	return -1, scanner.Err()