105 lines
3.8 KiB
C
105 lines
3.8 KiB
C
|
#ifndef HGL_UDPSOCKET_INCLUDE
|
|||
|
|
#define HGL_UDPSOCKET_INCLUDE
|
|||
|
|
|
|||
|
|
#include<hgl/type/DataType.h>
|
|||
|
|
#include<hgl/network/Socket.h>
|
|||
|
|
namespace hgl
|
|||
|
|
{
|
|||
|
|
namespace network
|
|||
|
|
{
|
|||
|
|
/**
|
|||
|
|
* 这个类提供使用UDP协议的通信，但它并不提供可靠数据传输的支持。
|
|||
|
|
*/
|
|||
|
|
class UDPSocket:public Socket ///UDP通信类
|
|||
|
|
{
|
|||
|
|
IPAddress *bind_addr;
|
|||
|
|
IPAddress *tar_addr;
|
|||
|
|
|
|||
|
|
public: //事件函数
|
|||
|
|
|
|||
|
|
virtual int ProcRecv(int=-1){return -1;}
|
|||
|
|
virtual int ProcSend(int,int &left_bytes){return -1;}
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
UDPSocket(); ///<本类构造函数
|
|||
|
|
virtual ~UDPSocket(); ///<本类析构函数
|
|||
|
|
|
|||
|
|
virtual bool Create(const IPAddress *); ///<创建一个udp,并绑定一个IP地址与指定端口
|
|||
|
|
// virtual bool Create(int family); ///<创建一个udp
|
|||
|
|
|
|||
|
|
uint GetBindPort()const{return bind_addr->GetPort();} ///<取得绑定端口
|
|||
|
|
|
|||
|
|
bool SetSendAddr(const IPAddress *); ///<设定发送地址
|
|||
|
|
|
|||
|
|
int SendPacket(const void *,int); ///<发送数据包
|
|||
|
|
int SendPacket(IPAddress *,const void *,int); ///<向指定地址发送数据包
|
|||
|
|
int RecvPacket(void *,int,IPAddress *); ///<接收数据包
|
|||
|
|
};//class UDPSocket
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* UDPLite协议封装使用
|
|||
|
|
*/
|
|||
|
|
class UDPLiteSocket:public UDPSocket
|
|||
|
|
{
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
virtual ~UDPLiteSocket()=default;
|
|||
|
|
|
|||
|
|
bool Create(const IPAddress *)override; ///<创建一个udp lite,并绑定一个IP地址与指定端口
|
|||
|
|
// bool Create(int family)override; ///<创建一个udp lite
|
|||
|
|
|
|||
|
|
void SetChecksumCoverage(int send_val=20,int recv_val=20); ///<设定UDPLite检验位长度,最小20
|
|||
|
|
};//class UDPLiteSocket
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 使用回呼事件机制的UDPSocket
|
|||
|
|
*/
|
|||
|
|
template<typename BASE> class _UDPSocketCB:public BASE
|
|||
|
|
{
|
|||
|
|
public: //事件函数
|
|||
|
|
|
|||
|
|
DefEvent(void, OnDisconnect, (BASE *));
|
|||
|
|
DefEvent(int, OnRecv, (BASE *,int));
|
|||
|
|
DefEvent(int, OnSend, (BASE *,int,int &));
|
|||
|
|
|
|||
|
|
virtual void ClearEvent()
|
|||
|
|
{
|
|||
|
|
OnDisconnect=nullptr;
|
|||
|
|
OnRecv =nullptr;
|
|||
|
|
OnSend =nullptr;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
_UDPSocketCB(){ClearEvent();} ///<本类构造函数
|
|||
|
|
virtual ~_UDPSocketCB()=default;
|
|||
|
|
|
|||
|
|
virtual void ProcDisconnect()
|
|||
|
|
{
|
|||
|
|
SafeCallEvent(OnDisconnect,(this));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
virtual int ProcRecv(int size)
|
|||
|
|
{
|
|||
|
|
if(OnRecv==nullptr)return(-1);
|
|||
|
|
|
|||
|
|
return OnRecv(this,size);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
virtual int ProcSend(int size,int &left_bytes)
|
|||
|
|
{
|
|||
|
|
if(OnSend==nullptr)return(-1);
|
|||
|
|
|
|||
|
|
return OnSend(this,size,left_bytes);
|
|||
|
|
}
|
|||
|
|
};//class _UDPSocketCB
|
|||
|
|
|
|||
|
|
using UDPSocketCB =_UDPSocketCB<UDPSocket>;
|
|||
|
|
using UDPLiteSocketCB =_UDPSocketCB<UDPLiteSocket>;
|
|||
|
|
}//namespace network
|
|||
|
|
|
|||
|
|
using namespace network;
|
|||
|
|
}//namespace hgl
|
|||
|
|
#endif//HGL_UDPSOCKET_INCLUDE
|