169 lines
3.8 KiB
C++
169 lines
3.8 KiB
C++
#include "Udp.h"
|
|
|
|
//#define NET_DEBUG DEBUG
|
|
#define NET_DEBUG 0
|
|
|
|
UdpSocket::UdpSocket(TinyIP* tip) : Socket(tip)
|
|
{
|
|
Type = IP_UDP;
|
|
Port = 0;
|
|
|
|
// 累加端口
|
|
static ushort g_udp_port = 1024;
|
|
if(g_udp_port < 1024) g_udp_port = 1024;
|
|
BindPort = g_udp_port++;
|
|
}
|
|
|
|
string UdpSocket::ToString()
|
|
{
|
|
static char name[10];
|
|
sprintf(name, "UDP_%d", Port);
|
|
return name;
|
|
}
|
|
|
|
bool UdpSocket::OnOpen()
|
|
{
|
|
if(Port != 0) BindPort = Port;
|
|
|
|
if(Port)
|
|
debug_printf("Udp::Open %d 过滤 %d\r\n", BindPort, Port);
|
|
else
|
|
debug_printf("Udp::Open %d 监听所有端口UDP数据包\r\n", BindPort);
|
|
|
|
Enable = true;
|
|
return Enable;
|
|
}
|
|
|
|
void UdpSocket::OnClose()
|
|
{
|
|
debug_printf("Udp::Close %d\r\n", BindPort);
|
|
Enable = false;
|
|
}
|
|
|
|
bool UdpSocket::Process(IP_HEADER& ip, Stream& ms)
|
|
{
|
|
UDP_HEADER* udp = ms.Retrieve<UDP_HEADER>();
|
|
if(!udp) return false;
|
|
|
|
ushort port = __REV16(udp->DestPort);
|
|
ushort remotePort = __REV16(udp->SrcPort);
|
|
|
|
// 仅处理本连接的IP和端口
|
|
if(Port != 0 && port != Port) return false;
|
|
|
|
CurRemote.Port = remotePort;
|
|
CurRemote.Address = ip.SrcIP;
|
|
CurLocal.Port = port;
|
|
CurLocal.Address = ip.DestIP;
|
|
|
|
#if NET_DEBUG
|
|
byte* data = udp->Next();
|
|
uint len = ms.Remain();
|
|
uint plen = __REV16(udp->Length);
|
|
assert_param2(len + sizeof(UDP_HEADER) == plen, "UDP数据包标称长度与实际不符");
|
|
#endif
|
|
|
|
OnProcess(ip, *udp, ms);
|
|
|
|
return true;
|
|
}
|
|
|
|
void UdpSocket::OnProcess(IP_HEADER& ip, UDP_HEADER& udp, Stream& ms)
|
|
{
|
|
byte* data = ms.Current();
|
|
//uint len = ms.Remain();
|
|
// 计算标称的数据长度
|
|
uint len = __REV16(udp.Length) - sizeof(UDP_HEADER);
|
|
assert_param2(len <= ms.Remain(), "UDP数据包不完整");
|
|
|
|
// 触发ITransport接口事件
|
|
uint len2 = OnReceive(data, len);
|
|
// 如果有返回，说明有数据要回复出去
|
|
//if(len2) Write(data, len2);
|
|
if(len2)
|
|
{
|
|
ByteArray bs(data, len2);
|
|
Send(bs, CurRemote.Address, CurRemote.Port);
|
|
}
|
|
|
|
if(OnReceived)
|
|
{
|
|
|
|
// 返回值指示是否向对方发送数据包
|
|
bool rs = OnReceived(*this, udp, CurRemote, ms);
|
|
if(rs && ms.Remain() > 0) Send(udp, len, CurRemote.Address, CurRemote.Port, false);
|
|
}
|
|
else
|
|
{
|
|
#if NET_DEBUG
|
|
debug_printf("UDP ");
|
|
CurRemote.Show();
|
|
debug_printf(" => ");
|
|
CurLocal.Show();
|
|
debug_printf(" Payload=%d udp_len=%d \r\n", len, __REV16(udp.Length));
|
|
|
|
Sys.ShowHex(data, len);
|
|
debug_printf(" \r\n");
|
|
#endif
|
|
}
|
|
}
|
|
|
|
void UdpSocket::Send(UDP_HEADER& udp, uint len, IPAddress& ip, ushort port, bool checksum)
|
|
{
|
|
uint tlen = sizeof(UDP_HEADER) + len;
|
|
udp.SrcPort = __REV16(BindPort);
|
|
udp.DestPort = __REV16(port);
|
|
udp.Length = __REV16(tlen);
|
|
|
|
// 网络序是大端
|
|
udp.Checksum = 0;
|
|
if(checksum) udp.Checksum = __REV16(Tip->CheckSum(&ip, (byte*)&udp, tlen, 1));
|
|
|
|
#if NET_DEBUG
|
|
debug_printf("SendUdp: len=%d(0x%x) %d => ", tlen, tlen, __REV16(udp.SrcPort));
|
|
ip.Show();
|
|
debug_printf(":%d ", port);
|
|
if(tlen > 0) Sys.ShowHex(udp.Next(), tlen > 64 ? 64 : tlen, false);
|
|
debug_printf("\r\n");
|
|
#else
|
|
Sys.Sleep(1);
|
|
#endif
|
|
|
|
Tip->SendIP(IP_UDP, ip, (byte*)&udp, tlen);
|
|
}
|
|
|
|
// 发送UDP数据到目标地址
|
|
void UdpSocket::Send(ByteArray& bs, IPAddress& ip, ushort port)
|
|
{
|
|
if(ip.IsAny()) ip = Remote.Address;
|
|
if(!port) port = Remote.Port;
|
|
|
|
byte buf[sizeof(ETH_HEADER) + sizeof(IP_HEADER) + sizeof(UDP_HEADER) + 256];
|
|
Stream ms(buf, ArrayLength(buf));
|
|
ms.Seek(sizeof(ETH_HEADER) + sizeof(IP_HEADER));
|
|
|
|
UDP_HEADER* udp = ms.Retrieve<UDP_HEADER>();
|
|
udp->Init(true);
|
|
|
|
// 复制数据，确保数据不会溢出
|
|
ms.Write(bs);
|
|
|
|
// 发送的时候采用LocalPort
|
|
//LocalPort = BindPort;
|
|
|
|
Send(*udp, bs.Length(), ip, port, true);
|
|
}
|
|
|
|
bool UdpSocket::OnWrite(const byte* buf, uint len)
|
|
{
|
|
ByteArray bs(buf, len);
|
|
Send(bs);
|
|
return len;
|
|
}
|
|
|
|
uint UdpSocket::OnRead(byte* buf, uint len)
|
|
{
|
|
// 暂时不支持
|
|
return 0;
|
|
}
|