mirror of https://github.com/ArduPilot/ardupilot
HAL_QURT: automatically find broadcast address for UDP network
This commit is contained in:
parent
a0af5515b8
commit
68cf1b6956
|
@ -0,0 +1,187 @@
|
|||
/*
|
||||
get system network addresses
|
||||
|
||||
based on code from Samba
|
||||
|
||||
Copyright (C) Andrew Tridgell 1998
|
||||
Copyright (C) Jeremy Allison 2007
|
||||
Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
|
||||
|
||||
This program 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.
|
||||
|
||||
This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <unistd.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/time.h>
|
||||
#include <ifaddrs.h>
|
||||
#include <net/if.h>
|
||||
#include <sys/ioctl.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
|
||||
void freeifaddrs(struct ifaddrs *ifp)
|
||||
{
|
||||
if (ifp != NULL) {
|
||||
free(ifp->ifa_name);
|
||||
free(ifp->ifa_addr);
|
||||
free(ifp->ifa_netmask);
|
||||
free(ifp->ifa_dstaddr);
|
||||
freeifaddrs(ifp->ifa_next);
|
||||
free(ifp);
|
||||
}
|
||||
}
|
||||
|
||||
static struct sockaddr *sockaddr_dup(struct sockaddr *sa)
|
||||
{
|
||||
struct sockaddr *ret;
|
||||
socklen_t socklen;
|
||||
socklen = sizeof(struct sockaddr_storage);
|
||||
ret = (struct sockaddr *)calloc(1, socklen);
|
||||
if (ret == NULL)
|
||||
return NULL;
|
||||
memcpy(ret, sa, socklen);
|
||||
return ret;
|
||||
}
|
||||
|
||||
/* this works for Linux 2.2, Solaris 2.5, SunOS4, HPUX 10.20, OSF1
|
||||
V4.0, Ultrix 4.4, SCO Unix 3.2, IRIX 6.4 and FreeBSD 3.2.
|
||||
|
||||
It probably also works on any BSD style system. */
|
||||
|
||||
int getifaddrs(struct ifaddrs **ifap)
|
||||
{
|
||||
struct ifconf ifc;
|
||||
char buff[8192];
|
||||
int fd, i, n;
|
||||
struct ifreq *ifr=NULL;
|
||||
struct ifaddrs *curif;
|
||||
struct ifaddrs *lastif = NULL;
|
||||
|
||||
*ifap = NULL;
|
||||
|
||||
if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
ifc.ifc_len = sizeof(buff);
|
||||
ifc.ifc_buf = buff;
|
||||
|
||||
if (ioctl(fd, SIOCGIFCONF, &ifc) != 0) {
|
||||
close(fd);
|
||||
return -1;
|
||||
}
|
||||
|
||||
ifr = ifc.ifc_req;
|
||||
|
||||
n = ifc.ifc_len / sizeof(struct ifreq);
|
||||
|
||||
/* Loop through interfaces, looking for given IP address */
|
||||
for (i=n-1; i>=0; i--) {
|
||||
if (ioctl(fd, SIOCGIFFLAGS, &ifr[i]) == -1) {
|
||||
freeifaddrs(*ifap);
|
||||
close(fd);
|
||||
return -1;
|
||||
}
|
||||
|
||||
curif = (struct ifaddrs *)calloc(1, sizeof(struct ifaddrs));
|
||||
if (curif == NULL) {
|
||||
freeifaddrs(*ifap);
|
||||
close(fd);
|
||||
return -1;
|
||||
}
|
||||
curif->ifa_name = strdup(ifr[i].ifr_name);
|
||||
if (curif->ifa_name == NULL) {
|
||||
free(curif);
|
||||
freeifaddrs(*ifap);
|
||||
close(fd);
|
||||
return -1;
|
||||
}
|
||||
curif->ifa_flags = ifr[i].ifr_flags;
|
||||
curif->ifa_dstaddr = NULL;
|
||||
curif->ifa_data = NULL;
|
||||
curif->ifa_next = NULL;
|
||||
|
||||
curif->ifa_addr = NULL;
|
||||
if (ioctl(fd, SIOCGIFADDR, &ifr[i]) != -1) {
|
||||
curif->ifa_addr = sockaddr_dup(&ifr[i].ifr_addr);
|
||||
if (curif->ifa_addr == NULL) {
|
||||
free(curif->ifa_name);
|
||||
free(curif);
|
||||
freeifaddrs(*ifap);
|
||||
close(fd);
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
curif->ifa_netmask = NULL;
|
||||
if (ioctl(fd, SIOCGIFNETMASK, &ifr[i]) != -1) {
|
||||
curif->ifa_netmask = sockaddr_dup(&ifr[i].ifr_addr);
|
||||
if (curif->ifa_netmask == NULL) {
|
||||
if (curif->ifa_addr != NULL) {
|
||||
free(curif->ifa_addr);
|
||||
}
|
||||
free(curif->ifa_name);
|
||||
free(curif);
|
||||
freeifaddrs(*ifap);
|
||||
close(fd);
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
if (lastif == NULL) {
|
||||
*ifap = curif;
|
||||
} else {
|
||||
lastif->ifa_next = curif;
|
||||
}
|
||||
lastif = curif;
|
||||
}
|
||||
|
||||
close(fd);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
const char *get_ipv4_broadcast(void)
|
||||
{
|
||||
struct ifaddrs *ifap = NULL;
|
||||
if (getifaddrs(&ifap) != 0) {
|
||||
return NULL;
|
||||
}
|
||||
struct ifaddrs *ia;
|
||||
for (ia=ifap; ia; ia=ia->ifa_next) {
|
||||
struct sockaddr_in *sin = (struct sockaddr_in *)ia->ifa_addr;
|
||||
struct sockaddr_in *nmask = (struct sockaddr_in *)ia->ifa_netmask;
|
||||
struct in_addr bcast;
|
||||
if (strcmp(ia->ifa_name, "lo") == 0) {
|
||||
continue;
|
||||
}
|
||||
bcast.s_addr = sin->sin_addr.s_addr | ~nmask->sin_addr.s_addr;
|
||||
const char *ret = inet_ntoa(bcast);
|
||||
freeifaddrs(ifap);
|
||||
return ret;
|
||||
}
|
||||
freeifaddrs(ifap);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
#ifdef MAIN_PROGRAM
|
||||
int main(void)
|
||||
{
|
||||
printf("%s\n", get_ipv4_broadcast());
|
||||
return 0;
|
||||
}
|
||||
#endif
|
|
@ -23,6 +23,8 @@ static uint64_t start_time;
|
|||
#define STORAGE_DIR "/var/APM"
|
||||
#define STORAGE_FILE STORAGE_DIR "/" SKETCHNAME ".stg"
|
||||
|
||||
extern const char *get_ipv4_broadcast(void);
|
||||
|
||||
// time since startup in microseconds
|
||||
static uint64_t micros64()
|
||||
{
|
||||
|
@ -82,6 +84,7 @@ static void get_storage(void)
|
|||
*/
|
||||
static void socket_check(void)
|
||||
{
|
||||
static const char *bcast = NULL;
|
||||
uint8_t buf[300];
|
||||
ssize_t ret = sock.recv(buf, sizeof(buf), 0);
|
||||
if (ret > 0) {
|
||||
|
@ -99,9 +102,16 @@ static void socket_check(void)
|
|||
}
|
||||
}
|
||||
uint32_t nbytes;
|
||||
if (bcast == NULL) {
|
||||
bcast = get_ipv4_broadcast();
|
||||
if (bcast == NULL) {
|
||||
bcast = "255.255.255.255";
|
||||
}
|
||||
printf("Broadcasting to %s\n", bcast);
|
||||
}
|
||||
if (ardupilot_socket_check(buf, sizeof(buf), &nbytes) == 0) {
|
||||
if (!connected) {
|
||||
sock.sendto(buf, nbytes, "255.255.255.255", 14550);
|
||||
sock.sendto(buf, nbytes, bcast, 14550);
|
||||
} else {
|
||||
sock.send(buf, nbytes);
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue