|
|
|
@ -53,7 +53,6 @@ CUdpSocket::~CUdpSocket()
|
|
|
|
bool CUdpSocket::Open(uint16 uiPort)
|
|
|
|
bool CUdpSocket::Open(uint16 uiPort)
|
|
|
|
{
|
|
|
|
{
|
|
|
|
bool open = false;
|
|
|
|
bool open = false;
|
|
|
|
int on = 1;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// create socket
|
|
|
|
// create socket
|
|
|
|
m_Socket = socket(PF_INET,SOCK_DGRAM,0);
|
|
|
|
m_Socket = socket(PF_INET,SOCK_DGRAM,0);
|
|
|
|
@ -68,7 +67,6 @@ bool CUdpSocket::Open(uint16 uiPort)
|
|
|
|
if ( bind(m_Socket, (struct sockaddr *)&m_SocketAddr, sizeof(struct sockaddr_in)) == 0 )
|
|
|
|
if ( bind(m_Socket, (struct sockaddr *)&m_SocketAddr, sizeof(struct sockaddr_in)) == 0 )
|
|
|
|
{
|
|
|
|
{
|
|
|
|
fcntl(m_Socket, F_SETFL, O_NONBLOCK);
|
|
|
|
fcntl(m_Socket, F_SETFL, O_NONBLOCK);
|
|
|
|
setsockopt(m_Socket, IPPROTO_IP, IP_PKTINFO, (char *)&on, sizeof(on));
|
|
|
|
|
|
|
|
open = true;
|
|
|
|
open = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
else
|
|
|
|
@ -102,33 +100,9 @@ int CUdpSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
|
|
|
|
int iRecvLen = -1;
|
|
|
|
int iRecvLen = -1;
|
|
|
|
struct timeval tv;
|
|
|
|
struct timeval tv;
|
|
|
|
|
|
|
|
|
|
|
|
struct msghdr Msg;
|
|
|
|
|
|
|
|
struct iovec Iov[1];
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
union {
|
|
|
|
|
|
|
|
struct cmsghdr cm;
|
|
|
|
|
|
|
|
unsigned char pktinfo_sizer[sizeof(struct cmsghdr) + sizeof(struct in_pktinfo)];
|
|
|
|
|
|
|
|
} Control;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// socket valid ?
|
|
|
|
// socket valid ?
|
|
|
|
if ( m_Socket != -1 )
|
|
|
|
if ( m_Socket != -1 )
|
|
|
|
{
|
|
|
|
{
|
|
|
|
// allocate buffer
|
|
|
|
|
|
|
|
Buffer->resize(UDP_BUFFER_LENMAX);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//prepare msghdr
|
|
|
|
|
|
|
|
bzero(&Msg, sizeof(Msg));
|
|
|
|
|
|
|
|
Iov[0].iov_base = Buffer->data();
|
|
|
|
|
|
|
|
Iov[0].iov_len = UDP_BUFFER_LENMAX;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
bzero(&Sin, sizeof(Sin));
|
|
|
|
|
|
|
|
Msg.msg_name = &Sin;
|
|
|
|
|
|
|
|
Msg.msg_namelen = sizeof(Sin);
|
|
|
|
|
|
|
|
Msg.msg_iov = Iov;
|
|
|
|
|
|
|
|
Msg.msg_iovlen = 1;
|
|
|
|
|
|
|
|
Msg.msg_control = &Control;
|
|
|
|
|
|
|
|
Msg.msg_controllen = sizeof(Control);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// control socket
|
|
|
|
// control socket
|
|
|
|
FD_ZERO(&FdSet);
|
|
|
|
FD_ZERO(&FdSet);
|
|
|
|
FD_SET(m_Socket, &FdSet);
|
|
|
|
FD_SET(m_Socket, &FdSet);
|
|
|
|
@ -136,8 +110,13 @@ int CUdpSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
|
|
|
|
tv.tv_usec = (timeout % 1000) * 1000;
|
|
|
|
tv.tv_usec = (timeout % 1000) * 1000;
|
|
|
|
select(m_Socket + 1, &FdSet, 0, 0, &tv);
|
|
|
|
select(m_Socket + 1, &FdSet, 0, 0, &tv);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// allocate buffer
|
|
|
|
|
|
|
|
Buffer->resize(UDP_BUFFER_LENMAX);
|
|
|
|
|
|
|
|
|
|
|
|
// read
|
|
|
|
// read
|
|
|
|
iRecvLen = (int)recvmsg(m_Socket, &Msg, 0);
|
|
|
|
iRecvLen = (int)recvfrom(m_Socket,
|
|
|
|
|
|
|
|
(void *)Buffer->data(), UDP_BUFFER_LENMAX,
|
|
|
|
|
|
|
|
0, (struct sockaddr *)&Sin, &uiFromLen);
|
|
|
|
|
|
|
|
|
|
|
|
// handle
|
|
|
|
// handle
|
|
|
|
if ( iRecvLen != -1 )
|
|
|
|
if ( iRecvLen != -1 )
|
|
|
|
@ -147,17 +126,6 @@ int CUdpSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
|
|
|
|
|
|
|
|
|
|
|
|
// get IP
|
|
|
|
// get IP
|
|
|
|
Ip->SetSockAddr(&Sin);
|
|
|
|
Ip->SetSockAddr(&Sin);
|
|
|
|
|
|
|
|
|
|
|
|
// get local IP
|
|
|
|
|
|
|
|
struct cmsghdr *Cmsg;
|
|
|
|
|
|
|
|
for (Cmsg = CMSG_FIRSTHDR(&Msg); Cmsg != NULL; Cmsg = CMSG_NXTHDR(&Msg, Cmsg))
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
if (Cmsg->cmsg_level == IPPROTO_IP && Cmsg->cmsg_type == IP_PKTINFO)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
struct in_pktinfo *PktInfo = (struct in_pktinfo *)CMSG_DATA(Cmsg);
|
|
|
|
|
|
|
|
m_LocalAddr.s_addr = PktInfo->ipi_spec_dst.s_addr;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|