59 lines
1.2 KiB
C++
59 lines
1.2 KiB
C++
#ifndef _ENCRYPT_H_
|
||
#define _ENCRYPT_H_
|
||
|
||
//#include <Windows.h>
|
||
#include "wrand.h"
|
||
class Encrypt
|
||
{
|
||
public:
|
||
Encrypt()
|
||
{
|
||
m_nSelfSalt = 0;
|
||
m_nTargetSalt = 0;
|
||
m_nKey = 0;
|
||
m_nSelfSalt = 0;
|
||
|
||
}
|
||
void SetSelfSalt(int salt); //设置自己的随机数
|
||
void SetTargetSalt(int salt); //设置通信方的随机数,同时会生成key
|
||
|
||
int GetSelfSalt()
|
||
{
|
||
return m_nSelfSalt; //获取自身的随机数
|
||
}
|
||
|
||
bool Encode(char* pInBuff, size_t len, char* pOutBuff); //加密数据
|
||
|
||
bool Decode(char* pInBuff, size_t len, char* pOutBuff); //解密数据
|
||
|
||
inline bool IsKeyCrcCorrect(unsigned short nKeyCRC) //
|
||
{
|
||
return (GetKeyCRC() == nKeyCRC);
|
||
}
|
||
|
||
inline static unsigned short CRC16(const char* lpBuffer, size_t len) //计算CRC-16的Checksum
|
||
{
|
||
return (unsigned short)CRC16Calc((unsigned char*)lpBuffer, len);
|
||
}
|
||
inline int GetKey()
|
||
{
|
||
return m_nKey;
|
||
}
|
||
unsigned int GetKeyCRC() //获取key的crc值
|
||
{
|
||
return (CRC16((const char*)m_sKeybuff, 4));
|
||
}
|
||
private:
|
||
void GenKey(); //产生密钥
|
||
|
||
int GenSalt(); // 产生salt
|
||
private:
|
||
int m_nSelfSalt; //自己产生的随机数
|
||
int m_nTargetSalt; //通信方的随机数
|
||
int m_nKey; //密钥
|
||
unsigned char m_sKeybuff[4] ; //密钥的buff，添加这个为了方便计算
|
||
};
|
||
|
||
|
||
#endif
|