MockSignature Class — netty Architecture

Source Code

handler/src/test/java/io/netty/handler/ssl/MockAlternativeKeyProvider.java lines 114–206

    public abstract static class MockSignature extends SignatureSpi {
        protected final String algorithm;
        protected final String providerName;
        protected final Signature realSignature;

        protected MockSignature(String algorithm, String providerName)
                throws NoSuchProviderException, NoSuchAlgorithmException {
            this.algorithm = algorithm;
            this.providerName = providerName;
            this.realSignature = Signature.getInstance(algorithm, providerName);
        }

        @Override
        protected void engineInitVerify(PublicKey publicKey) throws InvalidKeyException {
            try {
                realSignature.initVerify(publicKey);
            } catch (Exception e) {
                throw new InvalidKeyException("Failed to initialize signature", e);
            }
        }

        @Override
        protected void engineInitSign(PrivateKey privateKey) throws InvalidKeyException {
            engineInitSign(privateKey, null);
        }

        @Override
        protected void engineInitSign(PrivateKey privateKey, SecureRandom random) throws InvalidKeyException {
            try {
                // Extract the real key if it's wrapped
                if (privateKey instanceof AlternativePrivateKeyWrapper) {
                    privateKey = ((AlternativePrivateKeyWrapper) privateKey).getDelegate();
                    realSignature.initSign(privateKey, random);
                } else {
                    throw new InvalidKeyException("Unrecognized key type: " + privateKey.getClass().getName());
                }
            } catch (Exception e) {
                throw new InvalidKeyException("Failed to initialize signature", e);
            }
        }

        @Override
        protected void engineUpdate(byte b) throws SignatureException {
            realSignature.update(b);
        }

        @Override
        protected void engineUpdate(byte[] b, int off, int len) throws SignatureException {
            realSignature.update(b, off, len);
        }

        @Override
        protected byte[] engineSign() throws SignatureException {
            // Track signature operations
            signatureOperations.incrementAndGet();
            return realSignature.sign();
        }

        @Override
        protected boolean engineVerify(byte[] sigBytes) throws SignatureException {
            return realSignature.verify(sigBytes);
        }

        @Override
        protected void engineSetParameter(String param, Object value) {
            try {
                realSignature.setParameter(param, value);
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }

        @Override
        protected void engineSetParameter(AlgorithmParameterSpec params)
                throws InvalidAlgorithmParameterException {
            try {
                realSignature.setParameter(params);
            } catch (InvalidAlgorithmParameterException e) {
                throw e;
            } catch (Exception e) {
                throw new InvalidAlgorithmParameterException("Failed to set parameter", e);