如何用C++实现Linux下的文件加密 - 问答

在Linux下使用C++实现文件加密，可以采用多种方法。以下介绍一种基于**AES（高级加密标准）**的对称加密方法，并使用OpenSSL库来简化加密过程。AES是一种广泛使用的对称加密算法，安全性和效率都较高。

步骤概述

安装OpenSSL开发库
编写C++代码实现加密和解密
编译并运行程序

1. 安装OpenSSL开发库

在开始编写代码之前，需要确保系统中安装了OpenSSL的开发库。可以使用以下命令进行安装：

对于基于Debian的系统（如Ubuntu）：

sudo apt-get update
sudo apt-get install libssl-dev

对于基于Red Hat的系统（如CentOS）：

sudo yum install openssl-devel

2. 编写C++代码

下面是一个使用OpenSSL库实现AES加密和解密的示例程序：

#include <iostream>
#include <fstream>
#include <string>
#include <openssl/aes.h>
#include <openssl/rand.h>
#include <cstring>

// 加密函数
bool aes_encrypt(const std::string &input_file, const std::string &output_file, const std::string &password) {
    // 读取密码生成密钥和IV
    unsigned char key[AES_KEYLEN_256];
    unsigned char iv[AES_BLOCK_SIZE];

    // 使用PBKDF2从密码派生密钥
    if (!PKCS5_PBKDF2_HMAC(password.c_str(), password.size(),
                          nullptr, 0,
                          10000, AES_KEYLEN_256,
                          EVP_sha256(), key)) {
        std::cerr << "密钥派生失败" << std::endl;
        return false;
    }

    // 生成随机IV
    if (!RAND_bytes(iv, sizeof(iv))) {
        std::cerr << "生成IV失败" << std::endl;
        return false;
    }

    // 打开输入文件
    std::ifstream infile(input_file, std::ios::binary);
    if (!infile) {
        std::cerr << "无法打开输入文件: " << input_file << std::endl;
        return false;
    }

    // 打开输出文件
    std::ofstream outfile(output_file, std::ios::binary);
    if (!outfile) {
        std::cerr << "无法打开输出文件: " << output_file << std::endl;
        return false;
    }

    // 写入IV到输出文件
    outfile.write(reinterpret_cast<const char*>(iv), sizeof(iv));

    AES_KEY enc_key;
    AES_set_encrypt_key(key, AES_KEYLEN_256, &enc_key);

    // 加密缓冲区
    const int buffer_size = 4096;
    unsigned char in_buf[buffer_size];
    unsigned char out_buf[buffer_size + AES_BLOCK_SIZE];

    while (infile.good()) {
        infile.read(reinterpret_cast<char*>(in_buf), sizeof(in_buf));
        std::streamsize bytes_read = infile.gcount();

        if (bytes_read > 0) {
            AES_encrypt(in_buf, out_buf, &enc_key);

            // 写入加密数据
            outfile.write(reinterpret_cast<const char*>(out_buf), sizeof(out_buf));
        }
    }

    infile.close();
    outfile.close();
    return true;
}

// 解密函数
bool aes_decrypt(const std::string &input_file, const std::string &output_file, const std::string &password) {
    // 读取IV
    unsigned char iv[AES_BLOCK_SIZE];

    // 打开输入文件
    std::ifstream infile(input_file, std::ios::binary);
    if (!infile) {
        std::cerr << "无法打开输入文件: " << input_file << std::endl;
        return false;
    }

    // 打开输出文件
    std::ofstream outfile(output_file, std::ios::binary);
    if (!outfile) {
        std::cerr << "无法打开输出文件: " << output_file << std::endl;
        return false;
    }

    // 读取IV
    infile.read(reinterpret_cast<char*>(iv), sizeof(iv));

    // 读取密码生成密钥
    unsigned char key[AES_KEYLEN_256];
    if (!PKCS5_PBKDF2_HMAC(password.c_str(), password.size(),
                          nullptr, 0,
                          10000, AES_KEYLEN_256,
                          EVP_sha256(), key)) {
        std::cerr << "密钥派生失败" << std::endl;
        return false;
    }

    AES_KEY dec_key;
    AES_set_decrypt_key(key, AES_KEYLEN_256, &dec_key);

    // 解密缓冲区
    const int buffer_size = 4096;
    unsigned char in_buf[buffer_size + AES_BLOCK_SIZE];
    unsigned char out_buf[buffer_size];

    while (infile.good()) {
        infile.read(reinterpret_cast<char*>(in_buf), sizeof(in_buf));
        std::streamsize bytes_read = infile.gcount();

        if (bytes_read > 0) {
            AES_decrypt(in_buf, out_buf, &dec_key);

            // 写入解密数据
            outfile.write(reinterpret_cast<const char*>(out_buf), bytes_read);
        }
    }

    infile.close();
    outfile.close();
    return true;
}

int main(int argc, char *argv[]) {
    if (argc != 4) {
        std::cerr << "用法: " << argv[0] << " encrypt|decrypt 输入文件 输出文件 密码" << std::endl;
        return 1;
    }

    std::string mode = argv[1];
    std::string input_file = argv[2];
    std::string output_file = argv[3];
    std::string password = argv[4];

    if (mode == "encrypt") {
        if (aes_encrypt(input_file, output_file, password)) {
            std::cout << "加密成功！" << std::endl;
        } else {
            std::cerr << "加密失败！" << std::endl;
            return 1;
        }
    }
    else if (mode == "decrypt") {
        if (aes_decrypt(input_file, output_file, password)) {
            std::cout << "解密成功！" << std::endl;
        } else {
            std::cerr << "解密失败！" << std::endl;
            return 1;
        }
    }
    else {
        std::cerr << "未知模式。使用encrypt或decrypt。" << std::endl;
        return 1;
    }

    return 0;
}

代码说明

密钥派生：使用PKCS5_PBKDF2_HMAC函数从用户输入的密码派生出一个256位的AES密钥。这增加了密码的安全性，防止暴力破解。
初始化向量（IV）：使用RAND_bytes生成一个随机的IV，确保每次加密的结果不同，即使明文相同。
加密过程：
- 打开输入文件和输出文件。
- 将IV写入输出文件的开头，以便解密时使用。
- 使用AES加密算法对数据进行分块加密，并将加密后的数据写入输出文件。
解密过程：
- 打开加密文件和输出文件。
- 从加密文件中读取IV。
- 使用相同的密码派生出密钥。
- 使用AES解密算法对数据进行分块解密，并将解密后的数据写入输出文件。
主函数：通过命令行参数指定操作模式（encrypt或decrypt）、输入文件、输出文件和密码。

3. 编译程序

使用g++编译器，并链接OpenSSL库：

g++ -o aes_encrypt aes_encrypt.cpp -lcrypto

4. 使用示例

加密文件

./aes_encrypt encrypt plaintext.txt encrypted.bin mypassword

解密文件

./aes_encrypt decrypt encrypted.bin decrypted.txt mypassword

注意事项

密码管理：在实际应用中，避免将密码硬编码在源代码中。可以考虑通过命令行参数、配置文件或安全存储机制来管理密码。
错误处理：示例代码中对一些可能的错误进行了基本处理，但在生产环境中应添加更完善的错误检查和日志记录。
安全性：AES-256提供了很高的安全性，但整个加密方案的安全性还取决于密钥管理和密码的强度。确保使用复杂且难以猜测的密码，并妥善保护密钥。
依赖库：本示例依赖于OpenSSL库，确保在编译和运行环境中已正确安装并配置。

扩展

除了AES，OpenSSL还支持其他加密算法和模式（如CBC、GCM等）。根据需求选择合适的算法和模式。例如，GCM模式提供了认证加密（AEAD），可以同时保证数据的机密性和完整性。

以下是使用GCM模式的简要示例：

// 设置加密模式为GCM
AES_KEY enc_key;
AES_set_encrypt_key(key, AES_KEYLEN_256, &enc_key);
int len = AES_KEYLEN_256 / 8;
int tag_len = 16; // GCM标签长度

// 加密
AES_GCM_encrypt(in_buf, out_buf, bytes_read,
               &enc_key, iv,
               tag_len,
               out_buf + bytes_read, // 输出附加数据（AAD）
               sizeof(out_buf) - bytes_read - tag_len,
               nullptr, 0, // 无需额外的nonce，因为IV已经用作nonce
               nullptr);

// 解密
AES_GCM_decrypt(in_buf, out_buf, &dec_key, iv,
               tag_len,
               out_buf + bytes_read, // 输出附加数据（AAD）
               sizeof(out_buf) - bytes_read - tag_len,
               in_buf + bytes_read,
               sizeof(in_buf) - bytes_read - tag_len,
               nullptr, 0,
               nullptr);

使用GCM模式时，需要额外处理认证标签（Tag），以确保数据的完整性和真实性。

总结

通过以上步骤，您可以使用C++和OpenSSL库在Linux环境下实现文件的AES加密和解密。根据具体需求选择合适的加密算法和模式，并注意密钥管理和密码安全，以确保加密数据的安全性。

0 赞

0 踩