80 lines
1.8 KiB
C++
80 lines
1.8 KiB
C++
|
||
#pragma once
|
||
|
||
#include <string.h>
|
||
#include "_osdef.h"
|
||
#include "CRC16.h"
|
||
|
||
class Encrypt
|
||
{
|
||
public:
|
||
Encrypt()
|
||
{
|
||
Reset();
|
||
}
|
||
|
||
//重置
|
||
inline void Reset()
|
||
{
|
||
memset(this,0,sizeof(*this));
|
||
}
|
||
void SetSelfSalt(unsigned int salt); //设置自己的随机数
|
||
|
||
/*
|
||
* Comments: 设置目标的key
|
||
* Parameter: unsigned int salt:目标的salt
|
||
* Parameter: bool isClient:自身是作为客户端还是服务器，使用了不对称的key生成算法
|
||
* @Return void:
|
||
*/
|
||
void SetTargetSalt(unsigned int salt,bool isClient=false); //设置通信方的随机数,同时会生成key
|
||
|
||
unsigned int GetSelfSalt(){return m_nSelfSalt;} //获取自身的随机数
|
||
|
||
unsigned int GetTargetSalt() {return m_nTargetSalt; } //获取目标的随机数
|
||
|
||
bool Encode( char *pInBuff,size_t len,char *pOutBuff ); //加密数据
|
||
|
||
bool Decode( char *pInBuff,size_t len,char *pOutBuff ); //解密数据
|
||
|
||
inline bool IsKeyCrcCorrect( WORD nKeyCRC) //
|
||
{
|
||
return (GetKeyCRC()== nKeyCRC);
|
||
}
|
||
|
||
inline static UINT_PTR CRC16(LPCVOID lpBuffer, size_t len) //计算CRC-16的Checksum
|
||
{
|
||
return CRC16Calc((unsigned char *)lpBuffer,len);
|
||
}
|
||
inline unsigned int GetKey()
|
||
{
|
||
return m_nKey;
|
||
}
|
||
WORD GetKeyCRC() //获取key的crc值
|
||
{
|
||
return (CRC16(m_sKeybuff,4) & 0xffff);
|
||
}
|
||
unsigned int GenSalt(); // 产生salt
|
||
|
||
//增长checksum
|
||
inline void IncreaseCheckSum()
|
||
{
|
||
m_wCheckSum += 1;
|
||
}
|
||
|
||
inline WORD GetCheckSum()
|
||
{
|
||
return m_wCheckSum;
|
||
}
|
||
|
||
private:
|
||
void GenKey(bool isClient=false); //产生密钥
|
||
|
||
|
||
private:
|
||
unsigned int m_nSelfSalt; //自己产生的随机数
|
||
unsigned int m_nTargetSalt; //通信方的随机数
|
||
unsigned int m_nKey; //密钥
|
||
unsigned char m_sKeybuff[4] ; //密钥的buff，添加这个为了方便计算
|
||
WORD m_wCheckSum; //一个数，自增长的,用于防止重复发包
|
||
};
|