万普插件库

我写的代码：

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;
import java.security.SecureRandom;

public class AesCmdTool {

    private static final String AES = "AES";
    private static final String SHA1PRNG = "SHA1PRNG";
    public static final String PRE_KEY = "DES@";

    /**
     * 将二进制转换成16进制
     */
    public static String parseByte2HexStr(byte[] buf) {
        StringBuilder sb = new StringBuilder();
        String hex = null;
        for (byte each : buf) {
            hex = Integer.toHexString(each & 0xFF);
            if (hex.length() == 1) {
                sb.append("0");
            }
            sb.append(hex.toUpperCase());
        }
        return sb.toString();
    }

    /**
     * 将16进制转换为二进制
     */
    public static byte[] parseHexStr2Byte(String hexStr) {
        if (hexStr == null) {
            return null;
        }

        int hexStrLen = hexStr.length();
        if (hexStrLen < 1) {
            return null;
        }

        int iLen = hexStrLen / 2;

        byte[] result = new byte[iLen];
        for (int i = 0; i < iLen; ++i) {
            int high = Integer.parseInt(hexStr.substring(i * 2, i * 2 + 1), 16);
            int low = Integer.parseInt(hexStr.substring(i * 2 + 1, i * 2 + 2), 16);
            result[i] = (byte) (high * 16 + low);
        }
        return result;
    }

    public static String encode(String secret, String strPwd) {
        byte[] encryptResult = encrypt(secret, strPwd);
        if (encryptResult == null) {
            return "";
        }
        return parseByte2HexStr(encryptResult);
    }


    public static String decode(String secret, String strPwd) {
        // 解密
        byte[] decryptFrom = parseHexStr2Byte(secret);
        byte[] decryptResult = decrypt(decryptFrom, strPwd);
        if (decryptResult == null) {
            return "";
        }
        return new String(decryptResult);
    }


    /**
     * 加密
     *
     * @param content  需要加密的内容
     * @param password 加密密码
     */
    private static byte[] encrypt(String content, String password) {
        try {
            KeyGenerator keyGen = KeyGenerator.getInstance(AES);
            SecureRandom secureRandom = SecureRandom.getInstance(SHA1PRNG);
            secureRandom.setSeed(password.getBytes());
            keyGen.init(128, secureRandom);
            SecretKey secretKey = keyGen.generateKey();
            byte[] enCodeFormat = secretKey.getEncoded();
            SecretKeySpec key = new SecretKeySpec(enCodeFormat, AES);
            Cipher cipher = Cipher.getInstance(AES);
            byte[] byteContent = content.getBytes(StandardCharsets.UTF_8);
            cipher.init(Cipher.ENCRYPT_MODE, key);
            return cipher.doFinal(byteContent);
        } catch (Exception e) {
            return null;
        }
    }

    private static byte[] decrypt(byte[] content, String password) {
        try {
            KeyGenerator keyGen = KeyGenerator.getInstance(AES);
            //防止linux下 随机生成key
            SecureRandom secureRandom = SecureRandom.getInstance(SHA1PRNG);
            secureRandom.setSeed(password.getBytes());
            keyGen.init(128, secureRandom);
            SecretKey secretKey = keyGen.generateKey();
            byte[] enCodeFormat = secretKey.getEncoded();
            SecretKeySpec key = new SecretKeySpec(enCodeFormat, AES);
            Cipher cipher = Cipher.getInstance(AES);
            cipher.init(Cipher.DECRYPT_MODE, key);
            return cipher.doFinal(content);
        } catch (Exception e) {
            return null;
        }
    }

    public static String textDecodeForPreText(String data, String password) {
        if (data == null) {
            return "";
        }

        if (data.startsWith(PRE_KEY)) {
            return decode(data.substring(PRE_KEY.length()), password);
        }

        return data;
    }

    public static void main(String[] args) {

        String one = "看代码网";
        String pwd = "workForLife";
        String enText = encode(one,pwd);
        String deText = decode(enText,pwd);
        System.out.println("en："+enText);
        System.out.println("de："+deText);
    }
}

让豆包帮我优化，提供入参可输入的改进

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;
import java.security.SecureRandom;
import java.util.Arrays;

public class AesCmdTool {

    private static final String AES = "AES";
    private static final String DEFAULT_ALGORITHM = "AES/ECB/PKCS5Padding";
    private static final String USAGE = "用法: java AesCmdTool -e|-d -k <密码> [-m <算法>] <内容>";

    public static void main(String[] args) {
        if (args.length < 4) {
            System.err.println(USAGE);
            System.exit(1);
        }

        try {
            boolean isEncrypt = false;
            String password = null;
            String algorithm = DEFAULT_ALGORITHM;
            String content = null;

            for (int i = 0; i < args.length; i++) {
                switch (args[i]) {
                    case "-e":
                        isEncrypt = true;
                        break;
                    case "-d":
                        isEncrypt = false;
                        break;
                    case "-k":
                        password = args[++i];
                        break;
                    case "-m":
                        algorithm = args[++i];
                        break;
                    default:
                        content = args[i];
                        break;
                }
            }

            if (password == null || content == null) {
                System.err.println(USAGE);
                System.exit(1);
            }

            String result = isEncrypt ? 
                encrypt(content, password, algorithm) : 
                decrypt(content, password, algorithm);
                
            System.out.println(result);
            
        } catch (Exception e) {
            System.err.println("操作失败: " + e.getMessage());
            e.printStackTrace();
        }
    }

    public static String encrypt(String plaintext, String password, String algorithm) {
        try {
            SecretKeySpec key = generateKey(password);
            Cipher cipher = Cipher.getInstance(algorithm);
            cipher.init(Cipher.ENCRYPT_MODE, key);
            byte[] encryptedBytes = cipher.doFinal(plaintext.getBytes(StandardCharsets.UTF_8));
            return parseByte2HexStr(encryptedBytes);
        } catch (Exception e) {
            throw new RuntimeException("加密失败", e);
        }
    }

    public static String decrypt(String ciphertext, String password, String algorithm) {
        try {
            SecretKeySpec key = generateKey(password);
            Cipher cipher = Cipher.getInstance(algorithm);
            cipher.init(Cipher.DECRYPT_MODE, key);
            byte[] decryptedBytes = cipher.doFinal(parseHexStr2Byte(ciphertext));
            return new String(decryptedBytes, StandardCharsets.UTF_8);
        } catch (Exception e) {
            throw new RuntimeException("解密失败", e);
        }
    }

    private static SecretKeySpec generateKey(String password) throws Exception {
        KeyGenerator keyGen = KeyGenerator.getInstance(AES);
        SecureRandom secureRandom = SecureRandom.getInstanceStrong();
        secureRandom.setSeed(password.getBytes(StandardCharsets.UTF_8));
        keyGen.init(128, secureRandom);
        SecretKey secretKey = keyGen.generateKey();
        return new SecretKeySpec(secretKey.getEncoded(), AES);
    }

    public static String parseByte2HexStr(byte[] buf) {
        StringBuilder sb = new StringBuilder();
        for (byte b : buf) {
            String hex = Integer.toHexString(b & 0xFF);
            if (hex.length() == 1) {
                sb.append('0');
            }
            sb.append(hex);
        }
        return sb.toString().toUpperCase();
    }

    public static byte[] parseHexStr2Byte(String hexStr) {
        if (hexStr == null || hexStr.isEmpty()) {
            return new byte[0];
        }
        
        int len = hexStr.length();
        byte[] data = new byte[len / 2];
        for (int i = 0; i < len; i += 2) {
            data[i / 2] = (byte) ((Character.digit(hexStr.charAt(i), 16) << 4)
                                 + Character.digit(hexStr.charAt(i+1), 16));
        }
        return data;
    }
}

但豆包提供的代码，我竟然运行不出内容