fix: guard against a nil roll forward message (#790)

Signed-off-by: Chris Gianelloni <wolf31o2@blinklabs.io>
blinklabs-io · Nov 14, 2024 · a307899 · a307899
1 parent 7e0a43b
commit a307899
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/protocol/chainsync/server.go b/protocol/chainsync/server.go
@@ -135,6 +135,17 @@ func (s *Server) RollForward(blockType uint, blockData []byte, tip Tip) error {
 			blockData,
 			tip,
 		)
+		if msg == nil {
+			s.Protocol.Logger().
+				Error(
+					"failed to create roll forward message",
+					"component", "network",
+					"protocol", ProtocolName,
+					"role", "server",
+					"connection_id", s.callbackContext.ConnectionId.String(),
+				)
+			return fmt.Errorf("failed to create roll forward message")
+		}
 		return s.SendMessage(msg)
 	}
 }