admin/SimpleRemoter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feature: Encrypt for server address

Browse Source
This commit is contained in:
yuanyuanxiang
2025-05-27 04:15:35 +08:00
parent 2c4e8528b8
commit 26470b6ad3
3 changed files with 61 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
common/commands.h
View File

@@ -58,8 +58,7 @@ typedef void* LPVOID, * HANDLE;
#define CancelIo(p) close(reinterpret_cast<intptr_t>(p))
#endif
#include <string>
#include <vector>
#include "ip_enc.h"
#include <time.h>
#include <unordered_map>
@@ -352,10 +351,16 @@ public:
int FlagLen() const {
return strlen(szFlag);
}
const char* ServerIP()const {
const char* ServerIP(){
if (bEncrypt) {
Decrypt();
}
return szServerIP;
}
int ServerPort()const {
int ServerPort() {
if (bEncrypt) {
Decrypt();
}
return atoi(szPort);
}
int ClientType()const {
@@ -372,8 +377,24 @@ public:
return modified;
}
bool IsValid()const {
return strlen(szServerIP) != 0 && atoi(szPort) > 0;
void Encrypt() {
if (!bEncrypt){
bEncrypt = true;
StreamCipher cipher(0x12345678);
cipher.process((uint8_t*)szServerIP, sizeof(szServerIP));
cipher.process((uint8_t*)szPort, sizeof(szPort));
}
}
void Decrypt() {
if (bEncrypt) {
bEncrypt = false;
StreamCipher cipher(0x12345678);
cipher.process((uint8_t*)szServerIP, sizeof(szServerIP));
cipher.process((uint8_t*)szPort, sizeof(szPort));
}
}
bool IsValid() {
return strlen(ServerIP()) != 0 && ServerPort() > 0;
}
int Size() const {
return sizeof(CONNECT_ADDRESS);