wangdongpo/model2mem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(优化) 优化文件读写与加密逻辑

Browse Source
This commit is contained in:
wangdongpo
2025-01-21 15:02:34 +08:00
parent 9fda3625c2
commit bbdd6a02b2
1 changed files with 27 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
src/model2mem.cpp
View File

@@ -7,77 +7,67 @@ using namespace std;
// 加密密钥 // 加密密钥
string ENCRYPTION_KEY = ""; string ENCRYPTION_KEY = "";
static int write_file(const std::string & fname, std::vector<char>& buf){ static int write_file(const std::string& fname, std::vector<char>& buf) {
std::ofstream fs(fname, std::ios::binary | std::ios::out); std::ofstream fs(fname, std::ios::binary | std::ios::out);
if(!fs.good()){ if (!fs.good()) {
std::cerr<<fname<<" does not exist"<<std::endl; std::cerr << fname << " does not exist" << std::endl;
return -1; return -1;
} }
// 加密数据 // 加密数据
size_t keyIndex = 0; size_t keyIndex = 0;
for(char &c : buf) { if (ENCRYPTION_KEY.length() > 0) {
c ^= ENCRYPTION_KEY[keyIndex]; for (char& c : buf) {
keyIndex = (keyIndex + 1) % ENCRYPTION_KEY.length(); c ^= ENCRYPTION_KEY[keyIndex];
fs << (int)((unsigned char)c) << ","; keyIndex = (keyIndex + 1) % ENCRYPTION_KEY.length();
fs << (int)((unsigned char)c) << ",";
}
}
else {
for (char& c : buf) {
fs << (int)((unsigned char)c) << ",";
}
} }
fs.close(); fs.close();
return 0; return 0;
} }
static int load_file(const std::string & fname, std::vector<char>& buf){ static int load_file(const std::string& fname, std::vector<char>& buf) {
std::ifstream fs(fname, std::ios::binary | std::ios::in); std::ifstream fs(fname, std::ios::binary | std::ios::in);
if(!fs.good()){ if (!fs.good()) {
std::cerr<<fname<<" does not exist"<<std::endl; std::cerr << fname << " does not exist" << std::endl;
return -1; return -1;
} }
fs.seekg(0, std::ios::end); fs.seekg(0, std::ios::end);
int fsize=fs.tellg(); int fsize = fs.tellg();
fs.seekg(0, std::ios::beg); fs.seekg(0, std::ios::beg);
buf.resize(fsize); buf.resize(fsize);
fs.read(buf.data(),fsize); fs.read(buf.data(), fsize);
fs.close(); fs.close();
// 解密数据
size_t keyIndex = 0;
for(char &c : buf) {
c ^= ENCRYPTION_KEY[keyIndex];
keyIndex = (keyIndex + 1) % ENCRYPTION_KEY.length();
}
return 0; return 0;
} }
int main(int argc, char** argv){ int main(int argc, char** argv) {
if(argc < 3) { if (argc < 3) {
cout << "model2mem [input] [output] <pwd>" << endl; cout << "model2mem [input] [output] <pwd>" << endl;
return -1; return -1;
} }
vector<char> content; vector<char> content;
load_file(argv[1], content); load_file(argv[1], content);
if(content.size() <= 0) { if (content.size() <= 0) {
cout << "do not read any content" << endl; cout << "do not read any content" << endl;
return -1; return -1;
} }
// 如果提供了密码，则进行加密操作 // 如果提供了密码，则进行加密操作
if(argc == 4) { if (argc == 4) {
ENCRYPTION_KEY = argv[3]; ENCRYPTION_KEY = argv[3];
write_file(argv[2], content);
} else {
// 否则，直接写入文件
std::ofstream fs(argv[2], std::ios::binary | std::ios::out);
if(!fs.good())
{
std::cerr<<argv[2]<<" does not exist"<<std::endl;
return -1;
}
for(char c : content) {
fs << (int)((unsigned char)c) << ",";
}
fs.close();
} }
else {
ENCRYPTION_KEY = "";
}
write_file(argv[2], content);
return 0; return 0;
} }