Skip to content

Commit a2f49b9

Browse files
committed
fix(protocols): guard against empty leiosfetch messages
Signed-off-by: Chris Gianelloni <wolf31o2@blinklabs.io>
1 parent b93f5f3 commit a2f49b9

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

protocol/leiosfetch/server.go

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -104,8 +104,10 @@ func (s *Server) handleBlockRequest(msg protocol.Message) error {
104104
if err != nil {
105105
return err
106106
}
107-
if err := s.SendMessage(resp); err != nil {
108-
return err
107+
if resp != nil {
108+
if err := s.SendMessage(resp); err != nil {
109+
return err
110+
}
109111
}
110112
return nil
111113
}
@@ -133,8 +135,10 @@ func (s *Server) handleBlockTxsRequest(msg protocol.Message) error {
133135
if err != nil {
134136
return err
135137
}
136-
if err := s.SendMessage(resp); err != nil {
137-
return err
138+
if resp != nil {
139+
if err := s.SendMessage(resp); err != nil {
140+
return err
141+
}
138142
}
139143
return nil
140144
}
@@ -160,8 +164,10 @@ func (s *Server) handleVotesRequest(msg protocol.Message) error {
160164
if err != nil {
161165
return err
162166
}
163-
if err := s.SendMessage(resp); err != nil {
164-
return err
167+
if resp != nil {
168+
if err := s.SendMessage(resp); err != nil {
169+
return err
170+
}
165171
}
166172
return nil
167173
}

0 commit comments

Comments
 (0)