add method

hutool-crypto/src/main/java/cn/hutool/crypto/KeyUtil.java

@@ -5,7 +5,6 @@ import cn.hutool.core.io.IoUtil;
 import cn.hutool.core.lang.Assert;
 import cn.hutool.core.util.ArrayUtil;
 import cn.hutool.core.util.CharUtil;
-import cn.hutool.core.util.CharsetUtil;
 import cn.hutool.core.util.RandomUtil;
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.crypto.asymmetric.AsymmetricAlgorithm;
@@ -108,22 +107,43 @@ public class KeyUtil {
 	}
 
 	/**
-	 * 生成 {@link SecretKey}，仅用于对称加密和摘要算法密钥生成
+	 * 生成 {@link SecretKey}，仅用于对称加密和摘要算法密钥生成<br>
+	 * 当指定keySize&lt;0时，AES默认长度为128，其它算法不指定。
 	 *
 	 * @param algorithm 算法，支持PBE算法
-	 * @param keySize   密钥长度
+	 * @param keySize   密钥长度，&lt;0表示不设定密钥长度，即使用默认长度
 	 * @return {@link SecretKey}
 	 * @since 3.1.2
 	 */
 	public static SecretKey generateKey(String algorithm, int keySize) {
+		return generateKey(algorithm, keySize, null);
+	}
+
+	/**
+	 * 生成 {@link SecretKey}，仅用于对称加密和摘要算法密钥生成<br>
+	 * 当指定keySize&lt;0时，AES默认长度为128，其它算法不指定。
+	 *
+	 * @param algorithm 算法，支持PBE算法
+	 * @param keySize   密钥长度，&lt;0表示不设定密钥长度，即使用默认长度
+	 * @param random 随机数生成器，null表示默认
+	 * @return {@link SecretKey}
+	 * @since 5.5.2
+	 */
+	public static SecretKey generateKey(String algorithm, int keySize, SecureRandom random) {
 		algorithm = getMainAlgorithm(algorithm);
 
 		final KeyGenerator keyGenerator = getKeyGenerator(algorithm);
-		if (keySize > 0) {
-			keyGenerator.init(keySize);
-		} else if (SymmetricAlgorithm.AES.getValue().equals(algorithm)) {
+		if (keySize <= 0 && SymmetricAlgorithm.AES.getValue().equals(algorithm)) {
 			// 对于AES的密钥，除非指定，否则强制使用128位
-			keyGenerator.init(128);
+			keySize = 128;
+		}
+
+		if(keySize > 0){
+			if (null == random) {
+				keyGenerator.init(keySize);
+			} else {
+				keyGenerator.init(keySize, random);
+			}
 		}
 		return keyGenerator.generateKey();
 	}
@@ -140,7 +160,7 @@ public class KeyUtil {
 		SecretKey secretKey;
 		if (algorithm.startsWith("PBE")) {
 			// PBE密钥
-			secretKey = generatePBEKey(algorithm, (null == key) ? null : StrUtil.str(key, CharsetUtil.CHARSET_UTF_8).toCharArray());
+			secretKey = generatePBEKey(algorithm, (null == key) ? null : StrUtil.utf8Str(key).toCharArray());
 		} else if (algorithm.startsWith("DES")) {
 			// DES密钥
 			secretKey = generateDESKey(algorithm, key);

hutool-crypto/src/main/java/cn/hutool/crypto/symmetric/AES.java

@@ -2,9 +2,9 @@ package cn.hutool.crypto.symmetric;
 
 import cn.hutool.core.util.ArrayUtil;
 import cn.hutool.core.util.StrUtil;
+import cn.hutool.crypto.KeyUtil;
 import cn.hutool.crypto.Mode;
 import cn.hutool.crypto.Padding;
-import cn.hutool.crypto.SecureUtil;
 
 import javax.crypto.SecretKey;
 import javax.crypto.spec.IvParameterSpec;
@@ -49,6 +49,16 @@ public class AES extends SymmetricCrypto {
 		super(SymmetricAlgorithm.AES, key);
 	}
 
+	/**
+	 * 构造，使用默认的AES/ECB/PKCS5Padding
+	 *
+	 * @param key 密钥
+	 * @since 5.5.2
+	 */
+	public AES(SecretKey key) {
+		super(SymmetricAlgorithm.AES, key);
+	}
+
 	/**
 	 * 构造，使用随机密钥
 	 *
@@ -152,7 +162,7 @@ public class AES extends SymmetricCrypto {
 	 */
 	public AES(String mode, String padding, byte[] key, byte[] iv) {
 		this(mode, padding,//
-				SecureUtil.generateKey(SymmetricAlgorithm.AES.getValue(), key),//
+				KeyUtil.generateKey(SymmetricAlgorithm.AES.getValue(), key),//
 				ArrayUtil.isEmpty(iv) ? null : new IvParameterSpec(iv));
 	}
 

hutool-crypto/src/main/java/cn/hutool/crypto/symmetric/SymmetricCrypto.java

@@ -134,7 +134,7 @@ public class SymmetricCrypto implements Serializable {
 	 *
 	 * @param algorithm 算法
 	 * @param key       密钥，如果为<code>null</code>自动生成一个key
-	 * @return {@link SymmetricCrypto}的子对象，即子对象自身
+	 * @return SymmetricCrypto的子对象，即子对象自身
 	 */
 	public SymmetricCrypto init(String algorithm, SecretKey key) {
 		Assert.notBlank(algorithm, "'algorithm' must be not blank !");

hutool-crypto/src/test/java/cn/hutool/crypto/test/asymmetric/SignTest.java

@@ -1,13 +1,12 @@
 package cn.hutool.crypto.test.asymmetric;
 
 import cn.hutool.core.map.MapUtil;
-import org.junit.Assert;
-import org.junit.Test;
-
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.crypto.SecureUtil;
 import cn.hutool.crypto.asymmetric.Sign;
 import cn.hutool.crypto.asymmetric.SignAlgorithm;
+import org.junit.Assert;
+import org.junit.Test;
 
 import java.util.HashMap;
 import java.util.Map;

hutool-crypto/src/test/java/cn/hutool/crypto/test/symmetric/AESTest.java

@@ -2,12 +2,17 @@ package cn.hutool.crypto.test.symmetric;
 
 import cn.hutool.core.codec.Base64;
 import cn.hutool.core.util.HexUtil;
+import cn.hutool.core.util.RandomUtil;
+import cn.hutool.crypto.KeyUtil;
 import cn.hutool.crypto.Mode;
 import cn.hutool.crypto.Padding;
 import cn.hutool.crypto.symmetric.AES;
 import org.junit.Assert;
 import org.junit.Test;
 
+import javax.crypto.SecretKey;
+import java.security.SecureRandom;
+
 public class AESTest {
 
 	@Test
@@ -95,4 +100,17 @@ public class AESTest {
 		Assert.assertEquals("16c5", aes.decryptStr(Base64.decode("ecIQ0+MEkyz56mqciHxtfA==")));
 		// ------------------------------------------------------------------------
 	}
+
+	@Test
+	public void aesWithSha1PrngTest() {
+		final SecureRandom random = RandomUtil.getSecureRandom("123456".getBytes());
+		final SecretKey secretKey = KeyUtil.generateKey("AES", 128, random);
+
+		String content = "12sdfsdfs你好啊！";
+		AES aes = new AES(secretKey);
+		final String result1 = aes.encryptBase64(content);
+
+		final String decryptStr = aes.decryptStr(result1);
+		Assert.assertEquals(content, decryptStr);
+	}
 }