From 41958dcd304a887b90d9f0b83a995663e1861c4a Mon Sep 17 00:00:00 2001
From: Chris Gianelloni <wolf31o2@blinklabs.io>
Date: Thu, 14 Nov 2024 16:55:13 -0500
Subject: [PATCH] fix: guard against a nil roll forward message

Signed-off-by: Chris Gianelloni <wolf31o2@blinklabs.io>
---
 protocol/chainsync/server.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/protocol/chainsync/server.go b/protocol/chainsync/server.go
index 671ef586..54957f05 100644
--- 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)
 	}
 }