MockAlternativeKeyProvider Class — netty Architecture

Source Code

handler/src/test/java/io/netty/handler/ssl/MockAlternativeKeyProvider.java lines 37–269

public final class MockAlternativeKeyProvider extends Provider {

    private static final String PROVIDER_NAME = "MockAlternativeKeyProvider";
    private static final double PROVIDER_VERSION = 1.0;
    private static final String PROVIDER_INFO = "Mock provider simulating alternative key providers";

    // Track signature operations for test verification
    private static final AtomicInteger signatureOperations = new AtomicInteger(0);

    MockAlternativeKeyProvider() {
        super(PROVIDER_NAME, PROVIDER_VERSION, PROVIDER_INFO);

        // Register RSA signature algorithms
        put("Signature.RSASSA-PSS", MockRsaPssSignature.class.getName());
        put("Signature.SHA1withRSA", MockSha1WithRsaSignature.class.getName());
        put("Signature.SHA256withRSA", MockSha256WithRsaSignature.class.getName());
        put("Signature.SHA384withRSA", MockSha384WithRsaSignature.class.getName());
        put("Signature.SHA512withRSA", MockSha512WithRsaSignature.class.getName());
        put("Signature.MD5withRSA", MockMd5WithRsaSignature.class.getName());

        // Register ECDSA signature algorithms
        put("Signature.SHA1withECDSA", MockSha1WithEcdsaSignature.class.getName());
        put("Signature.SHA256withECDSA", MockSha256WithEcdsaSignature.class.getName());
        put("Signature.SHA384withECDSA", MockSha384WithEcdsaSignature.class.getName());
        put("Signature.SHA512withECDSA", MockSha512WithEcdsaSignature.class.getName());
    }

    static PrivateKey wrapPrivateKey(PrivateKey realKey) {
        return new AlternativePrivateKeyWrapper(realKey);
    }

    /**
     * Reset the signature operation counter for test isolation.
     */
    static void resetSignatureOperationCount() {
        signatureOperations.set(0);
    }

    /**
     * Get the number of signature operations performed by this provider.
     */
    static int getSignatureOperationCount() {
        return signatureOperations.get();
    }

    private static final class AlternativePrivateKeyWrapper implements PrivateKey {
        private final PrivateKey delegate;

        AlternativePrivateKeyWrapper(PrivateKey delegate) {
            this.delegate = delegate;
        }

        @Override
        public String getAlgorithm() {
            return delegate.getAlgorithm();
        }

        @Override
        public String getFormat() {
            // Alternative key providers typically don't support standard formats
            return null;
        }

        @Override
        public byte[] getEncoded() {
            // This is the key behavior: alternative key providers return null
            // because the private key material is not directly accessible
            return null;
        }

        // Provide access to the real key for signature operations
        PrivateKey getDelegate() {
            return delegate;
        }
    }

    // Abstract base class for all mock signatures
    public abstract static class MockSignature extends SignatureSpi {
        protected final String algorithm;
        protected final String providerName;
        protected final Signature realSignature;