151 lines
4.7 KiB
C++
151 lines
4.7 KiB
C++
#include <openssl/evp.h>
|
||
#include <openssl/bio.h>
|
||
#include <openssl/buffer.h>
|
||
#include <iostream>
|
||
#include <vector>
|
||
#include <string>
|
||
#include <stdexcept>
|
||
#include <iomanip>
|
||
#include <sstream>
|
||
#include "Crypto_Basic.hpp"
|
||
|
||
namespace qing {
|
||
class Base64 : public qing::Crypto_Basic {
|
||
public:
|
||
/* 默认构造函数 */
|
||
Base64() = default;
|
||
|
||
/* 编码：将字符串编码为Base64 */
|
||
std::string encode(const std::string& input_text) {
|
||
return base64_encode(input_text);
|
||
}
|
||
|
||
/* 编码：将字节向量编码为Base64 */
|
||
std::string encode(const std::vector<unsigned char>& input_bytes) {
|
||
return base64_encode(input_bytes);
|
||
}
|
||
|
||
/* 解码：将Base64字符串解码为原始字符串 */
|
||
std::string decode_to_string(const std::string& base64_text) {
|
||
std::vector<unsigned char> decoded = base64_decode_to_bytes(base64_text);
|
||
return std::string(decoded.begin(), decoded.end());
|
||
}
|
||
|
||
/* 解码：将Base64字符串解码为字节向量 */
|
||
std::vector<unsigned char> decode_to_bytes(const std::string& base64_text) {
|
||
return base64_decode_to_bytes(base64_text);
|
||
}
|
||
|
||
/* 静态方法：直接编码字符串（工具函数） */
|
||
static std::string base64_encode(const std::string& input_text) {
|
||
std::vector<unsigned char> input(input_text.begin(), input_text.end());
|
||
return base64_encode(input);
|
||
}
|
||
|
||
/* 静态方法：直接编码字节向量（工具函数） */
|
||
static std::string base64_encode(const std::vector<unsigned char>& input_bytes) {
|
||
// 创建BIO链：内存BIO -> Base64 BIO
|
||
BIO* bio = BIO_new(BIO_s_mem());
|
||
CRYPTO_CHECK(bio != nullptr, "无法创建内存BIO");
|
||
|
||
BIO* b64 = BIO_new(BIO_f_base64());
|
||
CRYPTO_CHECK(b64 != nullptr, "无法创建Base64 BIO");
|
||
|
||
// 将Base64 BIO连接到内存BIO
|
||
bio = BIO_push(b64, bio);
|
||
|
||
// 设置不添加换行符（默认会添加换行）
|
||
BIO_set_flags(bio, BIO_FLAGS_BASE64_NO_NL);
|
||
|
||
// 写入数据
|
||
int written = BIO_write(bio, input_bytes.data(), input_bytes.size());
|
||
CRYPTO_CHECK(written == (int)input_bytes.size(), "Base64编码写入失败");
|
||
|
||
// 刷新缓冲区
|
||
CRYPTO_CHECK(BIO_flush(bio) == 1, "Base64编码刷新失败");
|
||
|
||
// 读取编码结果
|
||
BUF_MEM* bptr;
|
||
BIO_get_mem_ptr(bio, &bptr);
|
||
|
||
// 复制结果到字符串（不包括空终止符）
|
||
std::string result(bptr->data, bptr->length);
|
||
|
||
// 清理BIO
|
||
BIO_free_all(bio);
|
||
|
||
return result;
|
||
}
|
||
|
||
/* 静态方法：直接解码为字符串（工具函数） */
|
||
static std::string base64_decode_to_string(const std::string& base64_text) {
|
||
std::vector<unsigned char> decoded = base64_decode_to_bytes(base64_text);
|
||
return std::string(decoded.begin(), decoded.end());
|
||
}
|
||
|
||
/* 静态方法：直接解码为字节向量（工具函数） */
|
||
static std::vector<unsigned char> base64_decode_to_bytes(const std::string& base64_text) {
|
||
// 创建BIO链：Base64 BIO -> 内存BIO
|
||
BIO* bio = BIO_new_mem_buf(base64_text.c_str(), -1);
|
||
CRYPTO_CHECK(bio != nullptr, "无法创建内存BIO");
|
||
|
||
BIO* b64 = BIO_new(BIO_f_base64());
|
||
CRYPTO_CHECK(b64 != nullptr, "无法创建Base64 BIO");
|
||
|
||
// 将Base64 BIO连接到内存BIO
|
||
bio = BIO_push(b64, bio);
|
||
|
||
// 设置不添加换行符（解码时也要匹配）
|
||
BIO_set_flags(bio, BIO_FLAGS_BASE64_NO_NL);
|
||
|
||
// 准备接收缓冲区
|
||
std::vector<unsigned char> result;
|
||
char buffer[4096];
|
||
int bytes_read;
|
||
|
||
// 循环读取解码数据
|
||
while ((bytes_read = BIO_read(bio, buffer, sizeof(buffer))) > 0) {
|
||
result.insert(result.end(), buffer, buffer + bytes_read);
|
||
}
|
||
|
||
// 检查是否解码成功（BIO_read返回0表示结束，负值表示错误）
|
||
CRYPTO_CHECK(bytes_read >= 0, "Base64解码失败");
|
||
|
||
// 清理BIO
|
||
BIO_free_all(bio);
|
||
|
||
return result;
|
||
}
|
||
|
||
/* 验证Base64格式是否有效 */
|
||
static bool is_valid_base64(const std::string& base64_text) {
|
||
// 检查是否只包含Base64字符（A-Z, a-z, 0-9, +, /, =）
|
||
if (base64_text.empty()) return false;
|
||
|
||
for (char c : base64_text) {
|
||
if (!((c >= 'A' && c <= 'Z') ||
|
||
(c >= 'a' && c <= 'z') ||
|
||
(c >= '0' && c <= '9') ||
|
||
c == '+' || c == '/' || c == '=')) {
|
||
return false;
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/* 打印编码结果（调试用） */
|
||
static void print_encoded(const std::string& encoded_text) {
|
||
std::cerr << "Base64 Encoded: " << encoded_text << std::endl;
|
||
}
|
||
|
||
/* 打印解码结果（调试用） */
|
||
static void print_decoded(const std::vector<unsigned char>& decoded_bytes) {
|
||
std::cerr << "Decoded bytes: ";
|
||
for (unsigned char byte : decoded_bytes) {
|
||
std::cerr << std::setw(2) << static_cast<int>(byte);
|
||
}
|
||
std::cerr << std::endl;
|
||
}
|
||
};
|
||
}
|