cache signatures only in acp118 handler (#3474)

network/p2p/acp118/handler.go

@@ -39,26 +39,26 @@ func NewHandler(verifier Verifier, signer warp.Signer) *Handler {
 }
 
 // NewCachedHandler returns an instance of Handler that caches successful
-// requests.
+// signatures.
 func NewCachedHandler(
 	cacher cache.Cacher[ids.ID, []byte],
 	verifier Verifier,
 	signer warp.Signer,
 ) *Handler {
 	return &Handler{
-		cacher:   cacher,
-		verifier: verifier,
-		signer:   signer,
+		signatureCache: cacher,
+		verifier:       verifier,
+		signer:         signer,
 	}
 }
 
 // Handler signs warp messages
 type Handler struct {
 	p2p.NoOpHandler
 
-	cacher   cache.Cacher[ids.ID, []byte]
-	verifier Verifier
-	signer   warp.Signer
+	signatureCache cache.Cacher[ids.ID, []byte]
+	verifier       Verifier
+	signer         warp.Signer
 }
 
 func (h *Handler) AppRequest(
@@ -84,8 +84,8 @@ func (h *Handler) AppRequest(
 	}
 
 	msgID := msg.ID()
-	if responseBytes, ok := h.cacher.Get(msgID); ok {
-		return responseBytes, nil
+	if signatureBytes, ok := h.signatureCache.Get(msgID); ok {
+		return signatureToResponse(signatureBytes)
 	}
 
 	if err := h.verifier.Verify(ctx, msg, request.Justification); err != nil {
@@ -100,6 +100,11 @@ func (h *Handler) AppRequest(
 		}
 	}
 
+	h.signatureCache.Put(msgID, signature)
+	return signatureToResponse(signature)
+}
+
+func signatureToResponse(signature []byte) ([]byte, *common.AppError) {
 	response := &sdk.SignatureResponse{
 		Signature: signature,
 	}
@@ -111,7 +116,5 @@ func (h *Handler) AppRequest(
 			Message: fmt.Sprintf("failed to marshal response: %s", err),
 		}
 	}
-
-	h.cacher.Put(msgID, responseBytes)
 	return responseBytes, nil
 }

network/p2p/acp118/handler_test.go

@@ -125,6 +125,12 @@ func TestHandler(t *testing.T) {
 				require.NoError(err)
 
 				require.True(bls.Verify(pk, signature, request.Message))
+
+				// Ensure the cache is populated with correct signature
+				sig, ok := tt.cacher.Get(unsignedMessage.ID())
+				if ok {
+					require.Equal(sig, response.Signature)
+				}
 			}
 
 			for _, expectedErr = range tt.expectedErrs {