107 lines
3.1 KiB
C
107 lines
3.1 KiB
C
|
#pragma once
|
|||
|
|
#include <list>
|
|||
|
|
|
|||
|
|
#define SEND_PACKET_MAX 1024*1024*8
|
|||
|
|
#define DEFAULT_TAG 0xCA0FFFFF
|
|||
|
|
|
|||
|
|
enum WebSocketOpcode { //操作码定义类型
|
|||
|
|
OPCODE_MID = 0x0,//标识一个中间数据包
|
|||
|
|
OPCODE_TXT = 0x1,//标识一个text类型数据包
|
|||
|
|
OPCODE_BIN = 0x2,//标识一个binary类型数据包
|
|||
|
|
//0x3 - 7：保留
|
|||
|
|
OPCODE_CLR = 0x8,//标识一个断开连接类型数据包
|
|||
|
|
OPCODE_PIN = 0x9,//标识一个ping类型数据包
|
|||
|
|
OPCODE_PON = 0xA,//表示一个pong类型数据包
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
#pragma pack(push,1)
|
|||
|
|
struct WebSocketHead {
|
|||
|
|
uint8_t fin : 1;//标识是否为此消息的最后一个数据包
|
|||
|
|
uint8_t rsv1 : 1;//保留位1
|
|||
|
|
uint8_t rsv2 : 1;//保留位2
|
|||
|
|
uint8_t rsv3 : 1;//保留位3
|
|||
|
|
uint8_t opcode : 4;//操作码
|
|||
|
|
|
|||
|
|
uint8_t mask : 1; //是否需要掩码
|
|||
|
|
uint8_t len : 7;//长度
|
|||
|
|
union {
|
|||
|
|
uint16_t v16;//长度为126时
|
|||
|
|
uint32_t v32;//长度为127时
|
|||
|
|
} ex_len;
|
|||
|
|
uint8_t mkey[4];
|
|||
|
|
uint8_t rh : 1;//head读取完成
|
|||
|
|
uint8_t rl : 1;//len读取完成
|
|||
|
|
uint8_t rk : 1;//mkey读取完成
|
|||
|
|
uint8_t rs : 5;//扩展保留
|
|||
|
|
WebSocketHead(void) { reset(); }
|
|||
|
|
void reset(void) { memset(this,0,sizeof(WebSocketHead)); }
|
|||
|
|
inline uint32_t GetLen(void) {
|
|||
|
|
if (len == 126) {
|
|||
|
|
return ex_len.v16;
|
|||
|
|
} else if (len == 127) {
|
|||
|
|
return ex_len.v32;
|
|||
|
|
}
|
|||
|
|
return len;
|
|||
|
|
}
|
|||
|
|
inline uint8_t GetLenNeedByte(void) {
|
|||
|
|
if (len == 126) {
|
|||
|
|
return 2;
|
|||
|
|
} else if (len == 127) {
|
|||
|
|
return 8;
|
|||
|
|
}
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
#pragma pack(pop)
|
|||
|
|
|
|||
|
|
enum ConnStatusDef {
|
|||
|
|
CS_CONNED,//连接上
|
|||
|
|
CS_COMMUNICATE,//通信状态
|
|||
|
|
};
|
|||
|
|
typedef std::list<DataPacket*> DataPacketList;
|
|||
|
|
class Client {
|
|||
|
|
public:
|
|||
|
|
Client(GameClientMgr *cli_mgr, SOCKET s, sockaddr_in* addr);
|
|||
|
|
~Client();
|
|||
|
|
|
|||
|
|
//设置阻塞模式。true表示阻塞，false表示非阻塞，返回值为socket错误号，0表示成功
|
|||
|
|
int SetBlockMode(const bool block);
|
|||
|
|
void SetSessionId(const uint16_t sid) { net_id_.index_ = sid; }
|
|||
|
|
uint16_t GetSessionId(void) { return net_id_.index_; }
|
|||
|
|
NetId &GetNetId(void) { return net_id_; }
|
|||
|
|
void RecvData(void);
|
|||
|
|
bool SendData(void);
|
|||
|
|
void WriteEvent(void);
|
|||
|
|
void Flush(const char *buf, size_t len);
|
|||
|
|
void SendNewSession(void);
|
|||
|
|
void SendSessionClose(void);
|
|||
|
|
void Close(void);
|
|||
|
|
bool OnRecv(const char* buf, size_t size);
|
|||
|
|
void FlushWsPack(const char* buf, size_t size);
|
|||
|
|
void OnGameWorldRecv(const char* buf, size_t size);
|
|||
|
|
inline SOCKET GetFd(void) { return fd_; }
|
|||
|
|
inline bool IsClose(void) { return is_close_; }
|
|||
|
|
void PacketHandle(void);
|
|||
|
|
uint16_t GetSKey() { return skey_; }
|
|||
|
|
protected:
|
|||
|
|
void ShakeHandsHandle(const char* buf, int buflen);
|
|||
|
|
private:
|
|||
|
|
inline bool MergePacketList(std::list<DataPacket*>::iterator &dpit);
|
|||
|
|
bool FindHttpParam(const char* param, const char* buf);
|
|||
|
|
|
|||
|
|
bool shake_hands_;//是否已经握手
|
|||
|
|
SOCKET fd_; //套接字
|
|||
|
|
bool is_close_; //是否已经关闭连接
|
|||
|
|
sockaddr_in remote_addr_;//远程地址
|
|||
|
|
DataPacketList recv_list_;//收到的包列表
|
|||
|
|
DataPacket* send_packet_;//需要发包的缓冲区
|
|||
|
|
DataPacketList gw_recv_list_;//从游戏服收到的包列表
|
|||
|
|
WebSocketHead ws_head_;//包头
|
|||
|
|
NetId net_id_;
|
|||
|
|
uint8_t conn_status_;//连接状态
|
|||
|
|
uint16_t skey_;//服务端KEY
|
|||
|
|
static BaseAllocator alloc;
|
|||
|
|
GameClientMgr* cli_mgr_;
|
|||
|
|
char real_ip[64];
|
|||
|
|
};
|