rename com.imyeyu.server to com.imyeyu.api

src/main/java/com/imyeyu/api/util/AES.java

@@ -0,0 +1,84 @@
+package com.imyeyu.api.util;
+
+import org.springframework.stereotype.Component;
+
+import javax.crypto.Cipher;
+import javax.crypto.KeyGenerator;
+import javax.crypto.SecretKey;
+import javax.crypto.spec.SecretKeySpec;
+
+/**
+ * AES 加解密
+ *
+ * @author 夜雨
+ * @version 2021-08-11 00:52
+ */
+@Component
+public class AES {
+
+	/**
+	 * 生产密钥
+	 *
+	 * @return 密钥
+	 * @throws Exception 生产异常
+	 */
+	public byte[] initKey() throws Exception {
+		KeyGenerator keyGen = KeyGenerator.getInstance("AES");
+		keyGen.init(256);
+		return keyGen.generateKey().getEncoded();
+	}
+
+	/**
+	 * 加密字符串
+	 *
+	 * @param data 待加密字符串
+	 * @param key  密钥
+	 * @return 加密结果
+	 * @throws Exception 加密异常
+	 */
+	public byte[] encrypt(String data, byte[] key) throws Exception {
+		return encrypt(data.getBytes(), key);
+	}
+
+	/**
+	 * 加密
+	 *
+	 * @param data 待加密字节数据
+	 * @param key  密钥
+	 * @return 加密结果
+	 * @throws Exception 加密异常
+	 */
+	public byte[] encrypt(byte[] data, byte[] key) throws Exception {
+		SecretKey secretKey = new SecretKeySpec(key, "AES");
+		Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
+		cipher.init(Cipher.ENCRYPT_MODE, secretKey); // 密钥
+		return cipher.doFinal(data); // 加密返回
+	}
+
+	/**
+	 * 解密字符串
+	 *
+	 * @param data 待解密字节数据
+	 * @param key  密钥
+	 * @return 解密结果
+	 * @throws Exception 解密异常
+	 */
+	public byte[] decrypt(String data, byte[] key) throws Exception {
+		return decrypt(data.getBytes(), key);
+	}
+
+	/**
+	 * 解密
+	 *
+	 * @param data 待解密字节数据
+	 * @param key  密钥
+	 * @return 解密结果
+	 * @throws Exception 解密异常
+	 */
+	public byte[] decrypt(byte[] data, byte[] key) throws Exception {
+		SecretKey secretKey = new SecretKeySpec(key, "AES");
+		Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
+		cipher.init(Cipher.DECRYPT_MODE, secretKey);
+		return cipher.doFinal(data);
+	}
+}