JNI/JCE: add AES and DESede SecretKeyFactory and SecretKey support to wolfJCE

Author: cconlon

scripts/infer.sh

@@ -100,6 +100,7 @@ infer --fail-on-issue run -- javac \
     src/main/java/com/wolfssl/provider/jce/WolfCryptProvider.java \
     src/main/java/com/wolfssl/provider/jce/WolfCryptPssParameters.java \
     src/main/java/com/wolfssl/provider/jce/WolfCryptRandom.java \
+    src/main/java/com/wolfssl/provider/jce/WolfCryptSecretKey.java \
     src/main/java/com/wolfssl/provider/jce/WolfCryptSecretKeyFactory.java \
     src/main/java/com/wolfssl/provider/jce/WolfCryptSignature.java \
     src/main/java/com/wolfssl/provider/jce/WolfCryptUtil.java \

src/main/java/com/wolfssl/provider/jce/WolfCryptProvider.java

@@ -484,6 +484,14 @@ private void registerServices() {
                     "com.wolfssl.provider.jce.WolfCryptSecretKeyFactory$wcPBKDF2WithHmacSHA3_512");
             }
         }
+        if (FeatureDetect.AesEnabled()) {
+            put("SecretKeyFactory.AES",
+                "com.wolfssl.provider.jce.WolfCryptSecretKeyFactory$wcAES");
+        }
+        if (FeatureDetect.Des3Enabled() && !Fips.enabled) {
+            put("SecretKeyFactory.DESede",
+                "com.wolfssl.provider.jce.WolfCryptSecretKeyFactory$wcDESede");
+        }
 
         /* KeyFactory */
         if (FeatureDetect.EccEnabled()) {

src/main/java/com/wolfssl/provider/jce/WolfCryptSecretKey.java

@@ -0,0 +1,234 @@
+/* WolfCryptSecretKey.java
+ *
+ * Copyright (C) 2006-2025 wolfSSL Inc.
+ *
+ * This file is part of wolfSSL.
+ *
+ * wolfSSL is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * wolfSSL is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA
+ */
+
+package com.wolfssl.provider.jce;
+
+import java.util.Arrays;
+import java.security.InvalidKeyException;
+import javax.crypto.SecretKey;
+
+import com.wolfssl.wolfcrypt.Aes;
+import com.wolfssl.wolfcrypt.Des3;
+
+/**
+ * wolfCrypt SecretKey implementation for symmetric algorithms.
+ *
+ * Supports AES and 3DES/DESede algorithms with key size validation.
+ */
+public class WolfCryptSecretKey implements SecretKey {
+
+    private static final long serialVersionUID = 1L;
+
+    /** Encoded key byte array */
+    private byte[] encoded = null;
+
+    /** Key algorithm name */
+    private String algorithm = null;
+
+    /** Has object been destroyed or not */
+    private boolean destroyed = false;
+
+    /* Lock around destroyed */
+    private transient final Object destroyedLock = new Object();
+
+    /**
+     * Create new WolfCryptSecretKey object.
+     *
+     * @param algorithm key algorithm name ("AES", "DESede")
+     * @param encoded encoded key byte array
+     *
+     * @throws InvalidKeyException if algorithm is null/empty,
+     *       encoded is null/zero length, or key size is invalid
+     */
+    public WolfCryptSecretKey(String algorithm, byte[] encoded)
+        throws InvalidKeyException {
+
+        if (algorithm == null || algorithm.isEmpty()) {
+            throw new InvalidKeyException(
+                "Algorithm String cannot be null or empty");
+        }
+
+        if (encoded == null || encoded.length == 0) {
+            throw new InvalidKeyException(
+                "Encoded key cannot be null or zero length");
+        }
+
+        /* Validate key size based on algorithm. Sanitize maching wolfCrypt
+         * aes.c and des3.c input sizes. */
+        if (algorithm.equalsIgnoreCase("AES")) {
+            if (encoded.length != Aes.KEY_SIZE_128 &&
+                encoded.length != Aes.KEY_SIZE_192 &&
+                encoded.length != Aes.KEY_SIZE_256) {
+                throw new InvalidKeyException(
+                    "AES key must be 16, 24, or 32 bytes, got: " +
+                    encoded.length);
+            }
+        }
+        else if (algorithm.equalsIgnoreCase("DESede") ||
+                 algorithm.equalsIgnoreCase("TripleDES")) {
+            if (encoded.length != Des3.KEY_SIZE) {
+                throw new InvalidKeyException(
+                    "DESede key must be 24 bytes, got: " + encoded.length);
+            }
+        }
+        else {
+            throw new InvalidKeyException(
+                "Unsupported algorithm: " + algorithm);
+        }
+
+        this.algorithm = algorithm;
+        this.encoded = encoded.clone();
+    }
+
+    /**
+     * Check if this object has been destroyed with destroy().
+     *
+     * @throws IllegalStateException if object has been destroyed
+     */
+    private synchronized void checkDestroyed()
+        throws IllegalStateException {
+
+        synchronized (destroyedLock) {
+            if (this.destroyed == true) {
+                throw new IllegalStateException(
+                    "SecretKey has been destroyed");
+            }
+        }
+    }
+
+    /**
+     * Return algorithm String representing this SecretKey.
+     *
+     * @return algorithm string matching this object
+     *
+     * @throws IllegalStateException if object has been destroyed
+     */
+    @Override
+    public synchronized String getAlgorithm() {
+
+        checkDestroyed();
+
+        return this.algorithm;
+    }
+
+    /**
+     * Return encoding format for this SecretKey.
+     *
+     * @return encoding format string, will be "RAW" for this object
+     *
+     * @throws IllegalStateException if object has been destroyed
+     */
+    @Override
+    public synchronized String getFormat() {
+
+        checkDestroyed();
+
+        return "RAW";
+    }
+
+    /**
+     * Return encoded byte array of this SecretKey.
+     *
+     * @return encoded byte array
+     *
+     * @throws IllegalStateException if object has been destroyed
+     */
+    @Override
+    public synchronized byte[] getEncoded() {
+
+        checkDestroyed();
+
+        return this.encoded.clone();
+    }
+
+    /**
+     * Destroy this object.
+     *
+     * Zeroize key bytes contained in this object and mark it as unusable.
+     */
+    public synchronized void destroy() {
+        synchronized (destroyedLock) {
+            if (this.encoded != null) {
+                Arrays.fill(this.encoded, (byte)0);
+                this.encoded = null;
+            }
+            this.algorithm = null;
+            this.destroyed = true;
+        }
+    }
+
+    /**
+     * Return if this object has been destroyed.
+     *
+     * Object can be destroyed by calling destroy(), which will zeroize
+     * internal buffers for this object.
+     *
+     * @return true if object has been destroyed, otherwise false
+     */
+    public synchronized boolean isDestroyed() {
+        synchronized (destroyedLock) {
+            if (this.destroyed) {
+                return true;
+            }
+            return false;
+        }
+    }
+
+    @Override
+    public synchronized int hashCode() {
+        checkDestroyed();
+        return Arrays.hashCode(encoded);
+    }
+
+    @Override
+    public synchronized boolean equals(Object obj) {
+
+        SecretKey sKey = null;
+
+        synchronized (destroyedLock) {
+            checkDestroyed();
+
+            if (obj == this) {
+                return true;
+            }
+
+            if (!(obj instanceof SecretKey)) {
+                return false;
+            }
+            sKey = (SecretKey)obj;
+
+            if (!Arrays.equals(sKey.getEncoded(), getEncoded())) {
+                return false;
+            }
+
+            if (!sKey.getAlgorithm().equals(getAlgorithm())) {
+                return false;
+            }
+
+            if (!sKey.getFormat().equals(getFormat())) {
+                return false;
+            }
+
+            return true;
+        }
+    }
+}
+