mirror of
https://github.com/LX3JL/xlxd.git
synced 2026-04-07 23:43:43 +00:00
Update and cleanup
- created raw socket class - moved udp recv via msg to a derived class - option file read optimized
This commit is contained in:
parent
66bd9aca8a
commit
73648794f0
6 changed files with 447 additions and 58 deletions
102
src/cudpmsgsocket.cpp
Normal file
102
src/cudpmsgsocket.cpp
Normal file
|
|
@ -0,0 +1,102 @@
|
|||
//
|
||||
// cudpmsgsocket.cpp
|
||||
// xlxd
|
||||
//
|
||||
// Created by Marius Petrescu (YO2LOJ) on 22/02/2020.
|
||||
// Copyright © 2020 Marius Petrescu (YO2LOJ). All rights reserved.
|
||||
//
|
||||
// ----------------------------------------------------------------------------
|
||||
// This file is part of xlxd.
|
||||
//
|
||||
// xlxd is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// xlxd is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with Foobar. If not, see <http://www.gnu.org/licenses/>.
|
||||
// ----------------------------------------------------------------------------
|
||||
|
||||
#include "main.h"
|
||||
#include <string.h>
|
||||
#include "cudpmsgsocket.h"
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////////////
|
||||
// read
|
||||
|
||||
int CUdpMsgSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
|
||||
{
|
||||
struct sockaddr_in Sin;
|
||||
fd_set FdSet;
|
||||
unsigned int uiFromLen = sizeof(struct sockaddr_in);
|
||||
int iRecvLen = -1;
|
||||
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 ?
|
||||
if ( m_Socket != -1 )
|
||||
{
|
||||
// allocate buffer
|
||||
Buffer->resize(UDP_MSG_BUFFER_LENMAX);
|
||||
|
||||
//prepare msghdr
|
||||
bzero(&Msg, sizeof(Msg));
|
||||
Iov[0].iov_base = Buffer->data();
|
||||
Iov[0].iov_len = UDP_MSG_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
|
||||
FD_ZERO(&FdSet);
|
||||
FD_SET(m_Socket, &FdSet);
|
||||
tv.tv_sec = timeout / 1000;
|
||||
tv.tv_usec = (timeout % 1000) * 1000;
|
||||
select(m_Socket + 1, &FdSet, 0, 0, &tv);
|
||||
|
||||
// read
|
||||
iRecvLen = (int)recvmsg(m_Socket, &Msg, 0);
|
||||
|
||||
// handle
|
||||
if ( iRecvLen != -1 )
|
||||
{
|
||||
// adjust buffer size
|
||||
Buffer->resize(iRecvLen);
|
||||
|
||||
// get IP
|
||||
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;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// done
|
||||
return iRecvLen;
|
||||
}
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue