#include <Windows.h>
#include <iostream>
#include <fstream>
#include <string>// 加解密函数，使用XOR运算
void XORCrypt(char* data, int size, const std::string& key) {int keyLength = key.length();for (int i = 0; i < size; ++i) {data[i] = data[i] ^ key[i % keyLength]; // 使用循环的方式重复密钥}
}// 读取文件并进行XOR加密
void EncryptFile(const std::string& inputFileName, const std::string& outputFileName, const std::string& key) {// 打开输入文件std::ifstream inputFile(inputFileName, std::ios::binary);if (!inputFile.is_open()) {std::cerr << "Error: Unable to open input file." << std::endl;return;}// 获取文件大小inputFile.seekg(0, std::ios::end);int fileSize = inputFile.tellg();inputFile.seekg(0, std::ios::beg);// 分配内存来存储文件内容char* buffer = new char[fileSize];if (!buffer) {std::cerr << "Error: Memory allocation failed." << std::endl;inputFile.close();return;}// 读取文件内容到缓冲区inputFile.read(buffer, fileSize);inputFile.close();// 对缓冲区中的数据进行加密XORCrypt(buffer, fileSize, key);// 打开输出文件std::ofstream outputFile(outputFileName, std::ios::binary);if (!outputFile.is_open()) {std::cerr << "Error: Unable to open output file." << std::endl;delete[] buffer;return;}// 写入加密后的数据到输出文件outputFile.write(buffer, fileSize);outputFile.close();// 释放缓冲区内存delete[] buffer;std::cout << "Encryption completed successfully." << std::endl;
}int main() {std::string inputFileName; // 输入文件名std::string outputFileName; // 输出文件名std::string key; // 加解密密钥// 获取输入文件名、输出文件名和密钥std::cout << "Enter input file name: ";std::getline(std::cin, inputFileName);std::cout << "Enter output file name: ";std::getline(std::cin, outputFileName);std::cout << "Enter encryption key: ";std::getline(std::cin, key);// 加密文件EncryptFile(inputFileName, outputFileName, key);system("pause");return 0;
}