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

Feature: Add encryption option for client building

Browse Source
This commit is contained in:
yuanyuanxiang
2025-07-13 03:55:33 +08:00
parent db6b55d59b
commit 41295ca03b
12 changed files with 122 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
client/IOCPClient.h
View File

@@ -15,16 +15,63 @@
#include "zstd/zstd.h"
#include "domain_pool.h"
#include "common/mask.h"
#include "common/header.h"
#define NO_AES
#include "common/encrypt.h"
#define MAX_RECV_BUFFER 1024*32
#define MAX_SEND_BUFFER 1024*32
#define FLAG_LENGTH 5
#define HDR_LENGTH 13
enum { S_STOP = 0, S_RUN, S_END };
typedef int (*DataProcessCB)(void* userData, PBYTE szBuffer, ULONG ulLength);
class ProtocolEncoder {
public:
virtual ~ProtocolEncoder(){}
virtual HeaderFlag GetHead() const {
return "Shine";
}
virtual int GetHeadLen() const {
return 13;
}
virtual int GetFlagLen() const {
return 5;
}
virtual void Encode(unsigned char* data, int len, unsigned char* param = 0) {}
virtual void Decode(unsigned char* data, int len, unsigned char* param = 0) {}
};
class HellEncoder : public ProtocolEncoder {
private:
EncFun m_HeaderEnc;
Encoder *m_BodyEnc;
public:
HellEncoder(EncFun head, Encoder *body) {
m_HeaderEnc = head;
m_BodyEnc = body;
}
~HellEncoder() {
SAFE_DELETE(m_BodyEnc);
}
virtual HeaderFlag GetHead() const override {
return ::GetHead(m_HeaderEnc);
}
virtual int GetHeadLen() const override {
return 16;
}
virtual int GetFlagLen() const override {
return 8;
}
virtual void Encode(unsigned char* data, int len, unsigned char* param = 0) override {
return m_BodyEnc->Encode(data, len, param);
}
virtual void Decode(unsigned char* data, int len, unsigned char* param = 0) override {
return m_BodyEnc->Decode(data, len, param);
}
};
class IOCPManager {
public:
virtual ~IOCPManager() {}
@@ -57,7 +104,7 @@ public:
class IOCPClient
{
public:
IOCPClient(State& bExit, bool exit_while_disconnect = false, int mask=0);
IOCPClient(State& bExit, bool exit_while_disconnect = false, int mask=0, int encoder=0);
virtual ~IOCPClient();
int SendLoginInfo(const LOGIN_INFOR& logInfo) {
@@ -114,7 +161,6 @@ protected:
protected:
sockaddr_in m_ServerAddr;
char m_szPacketFlag[FLAG_LENGTH + 3];
SOCKET m_sClientSocket;
CBuffer m_CompressedBuffer;
BOOL m_bWorkThread;
@@ -130,6 +176,7 @@ protected:
State& g_bExit; // ȫ<><C8AB>״̬<D7B4><CCAC>
void* m_Manager; // <20>û<EFBFBD><C3BB><EFBFBD><EFBFBD><EFBFBD>
DataProcessCB m_DataProcess; // <20><><EFBFBD><EFBFBD><EFBFBD>û<EFBFBD><C3BB><EFBFBD><EFBFBD><EFBFBD>
ProtocolEncoder* m_Encoder; // <20><><EFBFBD><EFBFBD>
DomainPool m_Domain;
std::string m_sCurIP;
int m_nHostPort;