Add error handling to Warp (#1068)

* add error handling

* Update warp/backend.go

Co-authored-by: Ceyhun Onur <[email protected]>
Signed-off-by: minghinmatthewlam <[email protected]>

* Update warp/backend.go

Co-authored-by: Ceyhun Onur <[email protected]>
Signed-off-by: minghinmatthewlam <[email protected]>

---------

Signed-off-by: minghinmatthewlam <[email protected]>
Co-authored-by: Ceyhun Onur <[email protected]>

Author: minghinmatthewlam

warp/aggregator/signature_getter.go

@@ -105,6 +105,9 @@ func (s *NetworkSignatureGetter) GetSignature(ctx context.Context, nodeID ids.No
 		if _, err := message.Codec.Unmarshal(signatureRes, &response); err != nil {
 			return nil, fmt.Errorf("failed to unmarshal signature res: %w", err)
 		}
+		if response.Signature == [bls.SignatureLen]byte{} {
+			return nil, fmt.Errorf("received empty signature response")
+		}
 		blsSignature, err := bls.SignatureFromBytes(response.Signature[:])
 		if err != nil {
 			return nil, fmt.Errorf("failed to parse signature from res: %w", err)

warp/backend.go

@@ -94,6 +94,14 @@ func (b *backend) initOffChainMessages(offchainMessages [][]byte) error {
 			return fmt.Errorf("%w at index %d: %w", errParsingOffChainMessage, i, err)
 		}
 
+		if unsignedMsg.NetworkID != b.networkID {
+			return fmt.Errorf("%w at index %d", avalancheWarp.ErrWrongNetworkID, i)
+		}
+
+		if unsignedMsg.SourceChainID != b.sourceChainID {
+			return fmt.Errorf("%w at index %d", avalancheWarp.ErrWrongSourceChainID, i)
+		}
+
 		_, err = payload.ParseAddressedCall(unsignedMsg.Payload)
 		if err != nil {
 			return fmt.Errorf("%w at index %d as AddressedCall: %w", errParsingOffChainMessage, i, err)