D-STAR G3: fix for IPv6-supported CIp, CUdpSocket class and *BSD support

D-STAR G3 support strongly depends on IPv4, no IPv6 support.
pull/162/head
SASANO Takayoshi 6 years ago
parent 8c83138793
commit b7e3e77d13

@ -33,6 +33,9 @@
#include <netinet/ip.h> #include <netinet/ip.h>
#include <netinet/ip_icmp.h> #include <netinet/ip_icmp.h>
#if !defined(ICMP_DEST_UNREACH) && defined(ICMP_UNREACH)
#define ICMP_DEST_UNREACH ICMP_UNREACH
#endif
//////////////////////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////////////////////
// operation // operation
@ -157,9 +160,10 @@ void CG3Protocol::PresenceTask(void)
if ( m_PresenceSocket.Receive(&Buffer, &ReqIp, 20) != -1 ) if ( m_PresenceSocket.Receive(&Buffer, &ReqIp, 20) != -1 )
{ {
CIp Ip(ReqIp); CIp Ip(ReqIp);
Ip.GetSockAddr()->sin_port = htons(G3_DV_PORT); socklen_t len;
struct sockaddr_in *sin = (struct sockaddr_in *)Ip.GetSockAddr(len);
sin->sin_port = htons(G3_DV_PORT);
if (Buffer.size() == 32) if (Buffer.size() == 32)
{ {

@ -47,6 +47,9 @@ public:
void SetSockAddr(struct sockaddr_storage *, socklen_t); void SetSockAddr(struct sockaddr_storage *, socklen_t);
struct sockaddr_storage *GetSockAddr(socklen_t &); struct sockaddr_storage *GetSockAddr(socklen_t &);
// converter (IPv6 not supported)
uint32 GetAddr(void) const { return ((struct sockaddr_in *)&m_Addr)->sin_addr.s_addr; }
// operator // operator
bool operator ==(const CIp &) const; bool operator ==(const CIp &) const;
operator const char *() const; operator const char *() const;

@ -113,7 +113,7 @@ int CRawSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
Buffer->resize(iRecvLen); Buffer->resize(iRecvLen);
// get IP // get IP
Ip->SetSockAddr(&Sin); Ip->SetSockAddr((struct sockaddr_storage *)&Sin, sizeof(Sin));
} }
} }
@ -148,7 +148,7 @@ int CRawSocket::IcmpReceive(CBuffer *Buffer, CIp *Ip, int timeout)
Sin.sin_family = AF_INET; Sin.sin_family = AF_INET;
Sin.sin_addr.s_addr = remote_iph->ip_dst.s_addr; Sin.sin_addr.s_addr = remote_iph->ip_dst.s_addr;
Ip->SetSockAddr(&Sin); Ip->SetSockAddr((struct sockaddr_storage *)&Sin, sizeof(Sin));
} }
} }

@ -26,17 +26,42 @@
#include <string.h> #include <string.h>
#include "cudpmsgsocket.h" #include "cudpmsgsocket.h"
// both multiple socket and IPv6 are not supported
#define MsgSocket m_Socket[0]
#define MsgIp m_Ip[0]
//////////////////////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////////////////////
// open // open
bool CUdpMsgSocket::Open(uint16 uiPort) bool CUdpMsgSocket::Open(uint16 uiPort)
{ {
bool ret; int on = 1, err = -1;
int on = 1; struct sockaddr_storage *ss;
socklen_t ss_len;
ret = CUdpSocket::Open(uiPort); if ( !CUdpSocket::Open(uiPort) )
setsockopt(m_Socket, IPPROTO_IP, IP_PKTINFO, (char *)&on, sizeof(on)); {
return false;
}
return ret; ss = MsgIp.GetSockAddr(ss_len);
if ( ss->ss_family == AF_INET )
{
#if defined(IP_PKTINFO)
err = setsockopt(MsgSocket, IPPROTO_IP, IP_PKTINFO, (char *)&on, sizeof(on));
#elif defined(IP_RECVDSTADDR)
err = setsockopt(MsgSocket, IPPROTO_IP, IP_RECVDSTADDR, (char *)&on, sizeof(on));
#endif
}
if ( err < 0 )
{
CUdpSocket::Close();
return false;
}
else
{
return true;
}
} }
@ -55,12 +80,16 @@ int CUdpMsgSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
struct iovec Iov[1]; struct iovec Iov[1];
union { union {
struct cmsghdr cm; struct cmsghdr cm;
unsigned char pktinfo_sizer[sizeof(struct cmsghdr) + sizeof(struct in_pktinfo)]; #if defined(IP_PKTINFO)
unsigned char pktinfo_sizer[sizeof(struct cmsghdr) + sizeof(struct in_pktinfo)];
#elif defined(IP_RECVDSTADDR)
unsigned char pktinfo_sizer[sizeof(struct cmsghdr) + sizeof(struct sockaddr_in)];
#endif
} Control; } Control;
// socket valid ? // socket valid ?
if ( m_Socket != -1 ) if ( MsgSocket != -1 )
{ {
// allocate buffer // allocate buffer
Buffer->resize(UDP_MSG_BUFFER_LENMAX); Buffer->resize(UDP_MSG_BUFFER_LENMAX);
@ -80,13 +109,13 @@ int CUdpMsgSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
// control socket // control socket
FD_ZERO(&FdSet); FD_ZERO(&FdSet);
FD_SET(m_Socket, &FdSet); FD_SET(MsgSocket, &FdSet);
tv.tv_sec = timeout / 1000; tv.tv_sec = timeout / 1000;
tv.tv_usec = (timeout % 1000) * 1000; tv.tv_usec = (timeout % 1000) * 1000;
select(m_Socket + 1, &FdSet, 0, 0, &tv); select(MsgSocket + 1, &FdSet, 0, 0, &tv);
// read // read
iRecvLen = (int)recvmsg(m_Socket, &Msg, 0); iRecvLen = (int)recvmsg(MsgSocket, &Msg, 0);
// handle // handle
if ( iRecvLen != -1 ) if ( iRecvLen != -1 )
@ -95,17 +124,25 @@ int CUdpMsgSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
Buffer->resize(iRecvLen); Buffer->resize(iRecvLen);
// get IP // get IP
Ip->SetSockAddr(&Sin); Ip->SetSockAddr((struct sockaddr_storage *)&Sin, sizeof(Sin));
// get local IP // get local IP
struct cmsghdr *Cmsg; struct cmsghdr *Cmsg;
for (Cmsg = CMSG_FIRSTHDR(&Msg); Cmsg != NULL; Cmsg = CMSG_NXTHDR(&Msg, Cmsg)) for (Cmsg = CMSG_FIRSTHDR(&Msg); Cmsg != NULL; Cmsg = CMSG_NXTHDR(&Msg, Cmsg))
{ {
#if defined(IP_PKTINFO)
if (Cmsg->cmsg_level == IPPROTO_IP && Cmsg->cmsg_type == IP_PKTINFO) if (Cmsg->cmsg_level == IPPROTO_IP && Cmsg->cmsg_type == IP_PKTINFO)
{ {
struct in_pktinfo *PktInfo = (struct in_pktinfo *)CMSG_DATA(Cmsg); struct in_pktinfo *PktInfo = (struct in_pktinfo *)CMSG_DATA(Cmsg);
m_LocalAddr.s_addr = PktInfo->ipi_spec_dst.s_addr; m_LocalAddr.s_addr = PktInfo->ipi_spec_dst.s_addr;
} }
#elif defined(IP_RECVDSTADDR)
if (Cmsg->cmsg_level == IPPROTO_IP && Cmsg->cmsg_type == IP_RECVDSTADDR)
{
struct sockaddr_in *DestAddr = (struct sockaddr_in *)CMSG_DATA(Cmsg);
m_LocalAddr.s_addr = DestAddr->sin_addr.s_addr;
}
#endif
} }
} }
} }
@ -113,4 +150,3 @@ int CUdpMsgSocket::Receive(CBuffer *Buffer, CIp *Ip, int timeout)
// done // done
return iRecvLen; return iRecvLen;
} }

Loading…
Cancel
Save

Powered by TurnKey Linux.