#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;
}