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_DIR "/var/APM"
|
||||||
#define STORAGE_FILE STORAGE_DIR "/" SKETCHNAME ".stg"
|
#define STORAGE_FILE STORAGE_DIR "/" SKETCHNAME ".stg"
|
||||||
|
|
||||||
|
extern const char *get_ipv4_broadcast(void);
|
||||||
|
|
||||||
// time since startup in microseconds
|
// time since startup in microseconds
|
||||||
static uint64_t micros64()
|
static uint64_t micros64()
|
||||||
{
|
{
|
||||||
|
@ -82,6 +84,7 @@ static void get_storage(void)
|
||||||
*/
|
*/
|
||||||
static void socket_check(void)
|
static void socket_check(void)
|
||||||
{
|
{
|
||||||
|
static const char *bcast = NULL;
|
||||||
uint8_t buf[300];
|
uint8_t buf[300];
|
||||||
ssize_t ret = sock.recv(buf, sizeof(buf), 0);
|
ssize_t ret = sock.recv(buf, sizeof(buf), 0);
|
||||||
if (ret > 0) {
|
if (ret > 0) {
|
||||||
|
@ -99,9 +102,16 @@ static void socket_check(void)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
uint32_t nbytes;
|
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 (ardupilot_socket_check(buf, sizeof(buf), &nbytes) == 0) {
|
||||||
if (!connected) {
|
if (!connected) {
|
||||||
sock.sendto(buf, nbytes, "255.255.255.255", 14550);
|
sock.sendto(buf, nbytes, bcast, 14550);
|
||||||
} else {
|
} else {
|
||||||
sock.send(buf, nbytes);
|
sock.send(buf, nbytes);
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue