From 969a846d9574d13da064c8231dd146e58b0edfe6 Mon Sep 17 00:00:00 2001 From: Leo Galambos Date: Thu, 1 Jan 2026 11:42:13 +0100 Subject: [PATCH] fix: proxy object might be returned Signed-off-by: Leo Galambos --- .../core/alg/mldsa/MldsaLargeDataTest.java | 21 ++----------------- 1 file changed, 2 insertions(+), 19 deletions(-) diff --git a/lib/src/test/java/zeroecho/core/alg/mldsa/MldsaLargeDataTest.java b/lib/src/test/java/zeroecho/core/alg/mldsa/MldsaLargeDataTest.java index 5ba348f..686c9c8 100644 --- a/lib/src/test/java/zeroecho/core/alg/mldsa/MldsaLargeDataTest.java +++ b/lib/src/test/java/zeroecho/core/alg/mldsa/MldsaLargeDataTest.java @@ -120,16 +120,7 @@ public final class MldsaLargeDataTest { KeyPair kp = CryptoAlgorithms.keyPair("ML-DSA", spec); - SignatureContext verifierCtx = CryptoAlgorithms.create("ML-DSA", KeyUsage.VERIFY, kp.getPublic()); - if (!(verifierCtx instanceof MldsaSignatureContext mldsaVerifier)) { - try { - verifierCtx.close(); - } catch (Exception ignore) { - } - throw new AssertionError( - "VERIFY context must be MldsaSignatureContext, got: " + verifierCtx.getClass().getName()); - } - + SignatureContext mldsaVerifier = CryptoAlgorithms.create("ML-DSA", KeyUsage.VERIFY, kp.getPublic()); int expectedSigLen = mldsaVerifier.tagLength(); System.out.println(INDENT + " expectedSigLen=" + expectedSigLen); @@ -188,15 +179,7 @@ public final class MldsaLargeDataTest { byte[] badSig = Arrays.copyOf(signature, signature.length); badSig[0] = (byte) (badSig[0] ^ 0x01); - SignatureContext badVerifierCtx = CryptoAlgorithms.create("ML-DSA", KeyUsage.VERIFY, kp.getPublic()); - if (!(badVerifierCtx instanceof MldsaSignatureContext badVerifier)) { - try { - badVerifierCtx.close(); - } catch (Exception ignore) { - } - throw new AssertionError("VERIFY context must be MldsaSignatureContext (negative), got: " - + badVerifierCtx.getClass().getName()); - } + SignatureContext badVerifier = CryptoAlgorithms.create("ML-DSA", KeyUsage.VERIFY, kp.getPublic()); try { badVerifier.setExpectedTag(badSig);