C++使用openssl的EVP对文件进行AES-256-CBC加解密

1、背景

有项目需求,有对文件进行加密的功能,经过评估,最终决定使用AES-256-CBC加密。在C++中要实现这种加密有很多中方式和第三方库,由于运行环境的限制,可选择的库不多,最终决定使用openssl来进行。

关于AES加密的相关知识直接百度就可以百度到了,这里就不赘述了。

2、加密

XuFile.h

cpp 复制代码
//
// Created by zhengqiuxu on 2021/10/15.
//

#ifndef VIS_ADOS_I7_XUFILE_H
#define VIS_ADOS_I7_XUFILE_H


#include <vector>
#include <string>
#include <openssl/evp.h>
#include <openssl/aes.h>
#include <openssl/rand.h>
#include <iostream>
#include <openssl/err.h>

class XuFile {
public:
    /**
     * 用AES-256-CBC的方式加密一个文件
     * @param inputFile : 输入文件(源文件)地址
     * @param outputFile : 输出文件(加密后的文件)地址
     * @param key : 密钥
     * @param iv : 初始向量
     * @return  0:成功  其他:失败
     */
    int encryptFile_AES_256_CBC(const std::string &inputFile, const std::string &outputFile, const uint8_t *key, const uint8_t *iv);


private:



};


#endif //VIS_ADOS_I7_XUFILE_H

XuFile.cpp

cpp 复制代码
/**
 * 用AES-256-CBC的方式加密一个文件
 * @param inputFile : 输入文件(源文件)地址
 * @param outputFile : 输出文件(加密后的文件)地址
 * @param key : 密钥
 * @param iv : 初始向量
 * @return  0:成功  其他:失败
 */
int XuFile::encryptFile_AES_256_CBC(const std::string &inputFile, const std::string &outputFile, const uint8_t *key,
                                    const uint8_t *iv) {
    int ret = -1;

    try {
        /* 源文件的输入流 */
        std::ifstream srcIn(inputFile.c_str(), std::ios_base::in | std::ios_base::binary | std::ios_base::ate);
        /* 加密后的文件的输出流 */
        std::ofstream decOut(outputFile.c_str(), std::ios_base::out | std::ios_base::binary);
        /* 如果都打开了才能进行解密 */
        if (srcIn.is_open() && decOut.is_open()) {
            /* 初始化OpenSSL库 */
            OPENSSL_init_crypto(OPENSSL_INIT_ADD_ALL_CIPHERS \
 | OPENSSL_INIT_ADD_ALL_DIGESTS, NULL);
            OPENSSL_init_crypto(OPENSSL_INIT_LOAD_CRYPTO_STRINGS, NULL);
            /* 初始化下加密工具 */
            EVP_CIPHER_CTX *ctx = EVP_CIPHER_CTX_new();
            EVP_CIPHER_CTX_init(ctx);
            EVP_EncryptInit_ex(ctx, EVP_aes_256_cbc(), NULL, key, iv);
            EVP_CIPHER_CTX_set_padding(ctx, EVP_PADDING_PKCS7);
            /* 确定文件的大小 */
            uint64_t inputFileLen = srcIn.tellg();
            /* 重新将文件流指针置于文件开始的位置 */
            srcIn.seekg(0, std::ios_base::beg);
            /* 用来读数据的缓冲区 */
            char readBuf[8192] = {0x00};
            /* 用来进行加密然后写数据的缓冲区(比读取大32个字节是为了放填充数据) */
            uint8_t writeBuf[8192 + 32] = {0x00};
            /* 记录读了多少长度的变量 */
            uint64_t allReadLen = 0;
            /* 循环从流里读出文件 */
            while (allReadLen < inputFileLen) {
                /* 当前应该读多长的数据 */
                int curRead = sizeof(readBuf);
                if (inputFileLen - allReadLen < sizeof(readBuf)) {
                    curRead = inputFileLen - allReadLen;
                }
                /* 读出原文 */
                srcIn.read(readBuf, curRead);
                /* 记录一下长度 */
                allReadLen = allReadLen + curRead;

                /* 对原文一包包读出来进行加密 */
                int toEnBufLen = 0;
                int curEnLen = 0;
                if (!EVP_EncryptUpdate(ctx, writeBuf, &toEnBufLen, reinterpret_cast<const unsigned char *>(readBuf),
                                       curRead)) {
                    printf("EVP_EncryptUpdate failed!  err:%s  \n", ERR_error_string(ERR_get_error(), NULL));
                    break;
                }
                /* 如果是最后一包了,那么就调用一下结束,同时拿到填充出来的数据,只有调用了结束,才会加填充,这样加密一整个文件的时候就只是尾部有填充字节 */
                if (curRead < sizeof(readBuf)) {
                    if (!EVP_EncryptFinal_ex(ctx, writeBuf + toEnBufLen, &curEnLen)) {
                        printf("EVP_EncryptFinal_ex failed!  err:%s  \n", ERR_error_string(ERR_get_error(), NULL));
                        break;
                    }
                    toEnBufLen += curEnLen;
                }
//                    /* 将密文写入文件 */
                decOut.write(reinterpret_cast<const char *>(writeBuf), toEnBufLen);

//                    printf("allReadLen=%llu   inputFileLen=%llu  toEnBufLen=%d curEnLen=%d curRead=%d\n",allReadLen,inputFileLen,toEnBufLen,curEnLen,curRead);
            }
            /* 关闭流 */
            srcIn.close();
            decOut.close();
            /* 关闭加密工具 */
            EVP_CIPHER_CTX_reset(ctx);
            EVP_CIPHER_CTX_free(ctx);

            ret = 0;
        } else {
            printf("%s or %s can not open! \n", inputFile.c_str(), outputFile.c_str());
        }
    } catch (...) {
        printf("encryptFile_AES_256_CBC failed! err:%s  \n", strerror(errno));
    }
    return ret;

}

3、解密

XuFile.h

cpp 复制代码
//
// Created by zhengqiuxu on 2021/10/15.
//

#ifndef VIS_ADOS_I7_XUFILE_H
#define VIS_ADOS_I7_XUFILE_H


#include <vector>
#include <string>
#include <openssl/evp.h>
#include <openssl/aes.h>
#include <openssl/rand.h>
#include <iostream>
#include <openssl/err.h>


class XuFile {
public:
    /**
     * 用AES-256-CBC的方式解密一个文件
     * @param inputFile : 输入文件(源文件)地址
     * @param outputFile : 输出文件(解密后的文件)地址
     * @param key : 密钥
     * @param iv : 初始向量
     * @return  0:成功  其他:失败
     */
    int decryptFile_AES_256_CBC(const std::string& inputFile, const std::string& outputFile, const uint8_t *key, const uint8_t *iv);


private:



};


#endif //VIS_ADOS_I7_XUFILE_H

XuFile.cpp

cpp 复制代码
/**
 * 用AES-256-CBC的方式解密一个文件
 * @param inputFile : 输入文件(源文件)地址
 * @param outputFile : 输出文件(解密后的文件)地址
 * @param key : 密钥
 * @param iv : 初始向量
 * @return  0:成功  其他:失败
 */
int XuFile::decryptFile_AES_256_CBC(const std::string &inputFile, const std::string &outputFile, const uint8_t *key, const uint8_t *iv) {
    int ret = -1;
    try {
            /* 源文件的输入流 */
            std::ifstream srcIn(inputFile.c_str(),std::ios_base::in | std::ios_base::binary | std::ios_base::ate);
            /* 解密后的文件的输出流 */
            std::ofstream decOut(outputFile.c_str(),std::ios_base::out | std::ios_base::binary | std::ios_base::trunc);
            /* 如果都打开了才能进行解密 */
            if(srcIn.is_open() && decOut.is_open()){
                /* 初始化OpenSSL库 */
                OPENSSL_init_crypto(OPENSSL_INIT_ADD_ALL_CIPHERS \
             | OPENSSL_INIT_ADD_ALL_DIGESTS, NULL);
                OPENSSL_init_crypto(OPENSSL_INIT_LOAD_CRYPTO_STRINGS, NULL);
                /* 初始化下解密工具 */
                EVP_CIPHER_CTX* ctx = EVP_CIPHER_CTX_new();
                EVP_CIPHER_CTX_init(ctx);
                int rr = EVP_DecryptInit_ex(ctx, EVP_aes_256_cbc(), NULL, key, iv);

                EVP_CIPHER_CTX_set_padding(ctx,EVP_PADDING_PKCS7);
                /* 确定文件的大小 */
                uint64_t inputFileLen = srcIn.tellg();
                /* 重新将文件流指针置于文件开始的位置 */
                srcIn.seekg(0, std::ios_base::beg);
                /* 用来读数据的缓冲区 */
                char readBuf[8192] = {0x00};
                /* 用来进行解密然后写数据的缓冲区(比读取大32个字节是为了放填充数据) */
                uint8_t writeBuf[8192+32] = {0x00};
                /* 记录读了多少长度的变量 */
                uint64_t allReadLen = 0;
                /* 循环从流里读出文件 */
                while (allReadLen < inputFileLen){
                    /* 当前应该读多长的数据 */
                    int curRead = sizeof(readBuf);
                    if(inputFileLen - allReadLen < sizeof(readBuf)){
                        curRead = inputFileLen - allReadLen;
                    }
                    /* 读出密文 */
                    srcIn.read(readBuf, curRead);
                    /* 记录一下长度 */
                    allReadLen = allReadLen + curRead;
                    /* 对密文一包包读取数据的数据进行解密 */
                    int toEnBufLen = 0;
                    int curEnLen = 0;
                    if(!EVP_DecryptUpdate(ctx, writeBuf, &toEnBufLen,reinterpret_cast<const unsigned char *>(readBuf), curRead)){
                        printf("EVP_DecryptUpdate failed!  err:%s  \n", ERR_error_string(ERR_get_error(),NULL));
                        break;
                    }
                    /* 如果是最后一包了,那么就调用一下结束 */
                    if(curRead < sizeof(readBuf)){
                        if(!EVP_DecryptFinal(ctx, writeBuf + toEnBufLen, &curEnLen)){
                            printf("EVP_DecryptFinal failed!  err:%s  \n", ERR_error_string(ERR_get_error(),NULL));
                            break;
                        }
                        toEnBufLen += curEnLen;
                    }

//                    /* 将解密后的数据写入文件 */
                    decOut.write(reinterpret_cast<const char *>(writeBuf), toEnBufLen);

                    printf("decryptFile_AES_256_CBC allReadLen=%llu   inputFileLen=%llu  toEnBufLen=%d curEnLen=%d curRead=%d\n",allReadLen,inputFileLen,toEnBufLen,curEnLen,curRead);
                }
                /* 关闭流 */
                srcIn.close();
                decOut.close();
                /* 关闭解密工具 */
                EVP_CIPHER_CTX_reset(ctx);
                EVP_CIPHER_CTX_free(ctx);

                ret = 0;
            }else{
                printf("%s or %s can not open! \n", inputFile.c_str(),outputFile.c_str());
            }

    } catch (...) {
        printf("decryptFile_AES_256_CBC failed! errstr:%s  \n",  ERR_error_string(ERR_get_error(),NULL));

    }
    return ret;

}

4、其他

开发过程中遇到过几个问题,由于一开始不熟悉耽误了时间,现在记录下。

1、加密的时候由于是从流里面一段段数据读出来的,所以每次调用完EVP_EncryptUpdate都会去调用一下EVP_EncryptFinal_ex,导致每一段数据后面都被添加了填充数据,所以后面改成所有的数据加密完后再调用EVP_EncryptFinal_ex获取填充数据。

2、使用C++加密后再使用Java解密,用的是javax.crypto.Cipher这个类。在key或者IV错误的情况,它可能有各种提示,比如说:填充数据错误、块错误等,但是原因都是key或者IV错误,所以解密失败第一时间还是要检查key跟IV对不对。

3、加密的时候可以选择PKCS7填充或者PKCS5填充,但是解密的时候要使用PKCS7,因为PKCS7兼容PKCS5。

相关推荐
快乐的划水a2 小时前
组合模式及优化
c++·设计模式·组合模式
星星火柴9363 小时前
关于“双指针法“的总结
数据结构·c++·笔记·学习·算法
艾莉丝努力练剑4 小时前
【洛谷刷题】用C语言和C++做一些入门题,练习洛谷IDE模式:分支机构(一)
c语言·开发语言·数据结构·c++·学习·算法
阿巴~阿巴~6 小时前
深入解析C++ STL链表(List)模拟实现
开发语言·c++·链表·stl·list
旺小仔.7 小时前
双指针和codetop复习
数据结构·c++·算法
jingfeng5147 小时前
C++ STL-string类底层实现
前端·c++·算法
郝学胜-神的一滴7 小时前
基于C++的词法分析器:使用正则表达式的实现
开发语言·c++·程序人生·正则表达式·stl
努力努力再努力wz9 小时前
【c++深入系列】:万字详解模版(下)
java·c++·redis
瓦特what?10 小时前
关于C++的#include的超超超详细讲解
java·开发语言·数据结构·c++·算法·信息可视化·数据挖掘
祁同伟.11 小时前
【C++】动态内存管理
开发语言·c++