python3 AES加密解密

参考博客：https://blog.csdn.net/weixin_42068117/article/details/80084034 

　　工作中开发人员用的是Java，但是写mock用的是Python，所以Java的加密解密算法转Python遇到了不少坑。下面以AES算法为例说明一下。

Java加密：

/**
     * aes加密-128位
     * 
     */
    public static  String AesEncrypt(String content ,String key){
        if (StringUtils.isEmpty(key) || key.length() != 16) {
            throw new RuntimeException("密钥长度为16位");
        }
        try {
            String iv = key;
            Cipher cipher = Cipher.getInstance("AES/CBC/NoPadding");
            int blockSize = cipher.getBlockSize();
            byte[] dataBytes = content.getBytes("utf-8");
            int plaintextLength = dataBytes.length;
            if (plaintextLength % blockSize != 0) {
                plaintextLength = plaintextLength + (blockSize - (plaintextLength % blockSize));
            }
            byte[] plaintext = new byte[plaintextLength];
            System.arraycopy(dataBytes, 0, plaintext, 0, dataBytes.length);
            SecretKeySpec keyspec = new SecretKeySpec(key.getBytes(), "AES");
            IvParameterSpec ivspec = new IvParameterSpec(iv.getBytes());
            cipher.init(Cipher.ENCRYPT_MODE, keyspec, ivspec);
            byte[] encrypted = cipher.doFinal(plaintext);
            return byte2Hex(encrypted);

        } catch (Exception e) {
            throw new RuntimeException("aes加密发生错误", e);
        }
    }

Java解密：

    // ==Aes加解密==================================================================
    /**
     * aes解密-128位
     */
    public static String AesDecrypt(String encryptContent, String password) {
        if (StringUtils.isEmpty(password) || password.length() != 16) {
            throw new RuntimeException("密钥长度为16位");
        }
        try {
            String key = password;
            String iv = password;
            byte[] encrypted1 = hex2Bytes(encryptContent);
            Cipher cipher = Cipher.getInstance("AES/CBC/NoPadding");
            SecretKeySpec keyspec = new SecretKeySpec(key.getBytes(), "AES");
            IvParameterSpec ivspec = new IvParameterSpec(iv.getBytes());
            cipher.init(Cipher.DECRYPT_MODE, keyspec, ivspec);
            byte[] original = cipher.doFinal(encrypted1);
            return new String(original,"UTF-8").trim();
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("aes解密发生错误", e);
        }
    }

对应Python加密：

def aes_encrypt(data, key):
    """aes加密函数，如果data不是16的倍数【加密文本data必须为16的倍数！】，那就补足为16的倍数
    :param key:
    :param data:
    """
    cipher = AES.new(key, AES.MODE_CBC, key)  # 设置AES加密模式 此处设置为CBC模式
    block_size = AES.block_size
    # 判断data是不是16的倍数，如果不是用b'\0'补足
    if len(data) % block_size != 0:
        add = block_size - (len(data) % block_size)
    else:
        add = 0
    data += b'\0' * add
    encrypted = cipher.encrypt(data)  # aes加密
    result = binascii.b2a_hex(encrypted)  # b2a_hex encode  将二进制转换成16进制
    return result

Python解密：

def aes_decode(data, key):
    """aes解密
    :param key:
    :param data:
    """
    cipher = AES.new(key, AES.MODE_CBC, key)
    result2 = binascii.a2b_hex(data)  # 十六进制还原成二进制
    decrypted = cipher.decrypt(result2)
    return decrypted.rstrip(b'\0')  # 解密完成后将加密时添加的多余字符'\0'删除

　　从Java加密的代码可以看出使用的是AES的CBC、128位模式，对应的Python加密也需要相同的模式，这里模式一定不要搞错，否则加密的结果就不一样了。

　　加密的时候遇到一个比较坑的问题，就是Python加密时用的方法AES.new有三个参数，第一个参数是密钥，第二个是模式，但是第三个参数就有点蒙了，之前在网上搜的结果是需要设置一个16位的字符，可以设置为b'0000000000000000'。但是加密后结果不对，这个参数不可以随便设置，最后又回过头来看了看Java的代码，发现第三个参数和第一个参数设置的值是一样的。。。

　　还有个地方需要注意一下，就是Java算法中最后将字符串转换成16进制了，所以转换成Python的时候也需要转换成16进制。

　　解密的时候算法同上理。

　　最后像这种Java转Python的算法，一定要看清楚Java的每一步算法的处理方法，否则就坑大了。

以上