Fredrik Fornwall
8 years ago
3 changed files with 613 additions and 5 deletions
@ -0,0 +1,607 @@ |
|||||
|
/*
|
||||
|
* Copyright (c) 1995, 1999 |
||||
|
* Berkeley Software Design, Inc. All rights reserved. |
||||
|
* |
||||
|
* Redistribution and use in source and binary forms, with or without |
||||
|
* modification, are permitted provided that the following conditions |
||||
|
* are met: |
||||
|
* 1. Redistributions of source code must retain the above copyright |
||||
|
* notice, this list of conditions and the following disclaimer. |
||||
|
* |
||||
|
* THIS SOFTWARE IS PROVIDED BY Berkeley Software Design, Inc. ``AS IS'' AND |
||||
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
||||
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
||||
|
* ARE DISCLAIMED. IN NO EVENT SHALL Berkeley Software Design, Inc. BE LIABLE |
||||
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
||||
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
||||
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
||||
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
||||
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
||||
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
||||
|
* SUCH DAMAGE. |
||||
|
* |
||||
|
* BSDI ifaddrs.h,v 2.5 2000/02/23 14:51:59 dab Exp |
||||
|
*/ |
||||
|
/*
|
||||
|
Copyright (c) 2013, Kenneth MacKay |
||||
|
All rights reserved. |
||||
|
|
||||
|
Redistribution and use in source and binary forms, with or without modification, |
||||
|
are permitted provided that the following conditions are met: |
||||
|
* Redistributions of source code must retain the above copyright notice, this |
||||
|
list of conditions and the following disclaimer. |
||||
|
* Redistributions in binary form must reproduce the above copyright notice, |
||||
|
this list of conditions and the following disclaimer in the documentation |
||||
|
and/or other materials provided with the distribution. |
||||
|
|
||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND |
||||
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
||||
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
||||
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR |
||||
|
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
||||
|
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
||||
|
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON |
||||
|
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
||||
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
*/ |
||||
|
|
||||
|
#ifndef _IFADDRS_H_ |
||||
|
#define _IFADDRS_H_ |
||||
|
|
||||
|
struct ifaddrs { |
||||
|
struct ifaddrs *ifa_next; |
||||
|
char *ifa_name; |
||||
|
unsigned int ifa_flags; |
||||
|
struct sockaddr *ifa_addr; |
||||
|
struct sockaddr *ifa_netmask; |
||||
|
struct sockaddr *ifa_dstaddr; |
||||
|
void *ifa_data; |
||||
|
}; |
||||
|
|
||||
|
/*
|
||||
|
* This may have been defined in <net/if.h>. Note that if <net/if.h> is |
||||
|
* to be included it must be included before this header file. |
||||
|
*/ |
||||
|
#ifndef ifa_broadaddr |
||||
|
#define ifa_broadaddr ifa_dstaddr /* broadcast address interface */ |
||||
|
#endif |
||||
|
|
||||
|
#include <sys/cdefs.h> |
||||
|
|
||||
|
__BEGIN_DECLS |
||||
|
extern int getifaddrs(struct ifaddrs **ifap); |
||||
|
extern void freeifaddrs(struct ifaddrs *ifa); |
||||
|
__END_DECLS |
||||
|
|
||||
|
#endif |
||||
|
|
||||
|
|
||||
|
#include <string.h> |
||||
|
#include <stdlib.h> |
||||
|
#include <stddef.h> |
||||
|
#include <errno.h> |
||||
|
#include <unistd.h> |
||||
|
#include <sys/socket.h> |
||||
|
#include <netpacket/packet.h> |
||||
|
#include <net/if_arp.h> |
||||
|
#include <netinet/in.h> |
||||
|
#include <linux/netlink.h> |
||||
|
#include <linux/rtnetlink.h> |
||||
|
|
||||
|
typedef struct NetlinkList { |
||||
|
struct NetlinkList *m_next; |
||||
|
struct nlmsghdr *m_data; |
||||
|
unsigned int m_size; |
||||
|
} NetlinkList; |
||||
|
|
||||
|
static int netlink_socket(void) |
||||
|
{ |
||||
|
int l_socket = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE); |
||||
|
if (l_socket < 0) return -1; |
||||
|
|
||||
|
struct sockaddr_nl l_addr; |
||||
|
memset(&l_addr, 0, sizeof(l_addr)); |
||||
|
l_addr.nl_family = AF_NETLINK; |
||||
|
if(bind(l_socket, (struct sockaddr *)&l_addr, sizeof(l_addr)) < 0) { |
||||
|
close(l_socket); |
||||
|
return -1; |
||||
|
} |
||||
|
|
||||
|
return l_socket; |
||||
|
} |
||||
|
|
||||
|
static int netlink_send(int p_socket, int p_request) |
||||
|
{ |
||||
|
struct { |
||||
|
struct nlmsghdr m_hdr; |
||||
|
struct rtgenmsg m_msg; |
||||
|
} l_data; |
||||
|
|
||||
|
memset(&l_data, 0, sizeof(l_data)); |
||||
|
|
||||
|
l_data.m_hdr.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtgenmsg)); |
||||
|
l_data.m_hdr.nlmsg_type = p_request; |
||||
|
l_data.m_hdr.nlmsg_flags = NLM_F_ROOT | NLM_F_MATCH | NLM_F_REQUEST; |
||||
|
l_data.m_hdr.nlmsg_pid = 0; |
||||
|
l_data.m_hdr.nlmsg_seq = p_socket; |
||||
|
l_data.m_msg.rtgen_family = AF_UNSPEC; |
||||
|
|
||||
|
struct sockaddr_nl l_addr; |
||||
|
memset(&l_addr, 0, sizeof(l_addr)); |
||||
|
l_addr.nl_family = AF_NETLINK; |
||||
|
return (sendto(p_socket, &l_data.m_hdr, l_data.m_hdr.nlmsg_len, 0, (struct sockaddr *)&l_addr, sizeof(l_addr))); |
||||
|
} |
||||
|
|
||||
|
static int netlink_recv(int p_socket, void *p_buffer, size_t p_len) |
||||
|
{ |
||||
|
struct msghdr l_msg; |
||||
|
struct iovec l_iov = { p_buffer, p_len }; |
||||
|
struct sockaddr_nl l_addr; |
||||
|
|
||||
|
for (;;) { |
||||
|
l_msg.msg_name = (void *)&l_addr; |
||||
|
l_msg.msg_namelen = sizeof(l_addr); |
||||
|
l_msg.msg_iov = &l_iov; |
||||
|
l_msg.msg_iovlen = 1; |
||||
|
l_msg.msg_control = NULL; |
||||
|
l_msg.msg_controllen = 0; |
||||
|
l_msg.msg_flags = 0; |
||||
|
int l_result = recvmsg(p_socket, &l_msg, 0); |
||||
|
|
||||
|
if (l_result < 0) { |
||||
|
if (errno == EINTR) { |
||||
|
continue; |
||||
|
} |
||||
|
return -2; |
||||
|
} |
||||
|
|
||||
|
if (l_msg.msg_flags & MSG_TRUNC) { |
||||
|
/* Buffer was too small. */ |
||||
|
return -1; |
||||
|
} |
||||
|
return l_result; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
static struct nlmsghdr *getNetlinkResponse(int p_socket, int *p_size, int *p_done) |
||||
|
{ |
||||
|
size_t l_size = 4096; |
||||
|
void *l_buffer = NULL; |
||||
|
|
||||
|
for(;;) { |
||||
|
free(l_buffer); |
||||
|
l_buffer = malloc(l_size); |
||||
|
if (l_buffer == NULL) return NULL; |
||||
|
|
||||
|
int l_read = netlink_recv(p_socket, l_buffer, l_size); |
||||
|
*p_size = l_read; |
||||
|
if (l_read == -2) { |
||||
|
free(l_buffer); |
||||
|
return NULL; |
||||
|
} |
||||
|
if (l_read >= 0) { |
||||
|
pid_t l_pid = getpid(); |
||||
|
struct nlmsghdr *l_hdr; |
||||
|
for(l_hdr = (struct nlmsghdr *)l_buffer; NLMSG_OK(l_hdr, (unsigned int)l_read); l_hdr = (struct nlmsghdr *)NLMSG_NEXT(l_hdr, l_read)) { |
||||
|
if((pid_t)l_hdr->nlmsg_pid != l_pid || (int)l_hdr->nlmsg_seq != p_socket) continue; |
||||
|
|
||||
|
if(l_hdr->nlmsg_type == NLMSG_DONE) { |
||||
|
*p_done = 1; |
||||
|
break; |
||||
|
} |
||||
|
|
||||
|
if(l_hdr->nlmsg_type == NLMSG_ERROR) { |
||||
|
free(l_buffer); |
||||
|
return NULL; |
||||
|
} |
||||
|
} |
||||
|
return (struct nlmsghdr*)l_buffer; |
||||
|
} |
||||
|
|
||||
|
l_size *= 2; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
static NetlinkList *newListItem(struct nlmsghdr *p_data, unsigned int p_size) |
||||
|
{ |
||||
|
NetlinkList *l_item = (NetlinkList*)malloc(sizeof(NetlinkList)); |
||||
|
if (l_item == NULL) return NULL; |
||||
|
|
||||
|
l_item->m_next = NULL; |
||||
|
l_item->m_data = p_data; |
||||
|
l_item->m_size = p_size; |
||||
|
return l_item; |
||||
|
} |
||||
|
|
||||
|
static void freeResultList(NetlinkList *p_list) |
||||
|
{ |
||||
|
NetlinkList *l_cur; |
||||
|
while (p_list) { |
||||
|
l_cur = p_list; |
||||
|
p_list = p_list->m_next; |
||||
|
free(l_cur->m_data); |
||||
|
free(l_cur); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
static NetlinkList *getResultList(int p_socket, int p_request) |
||||
|
{ |
||||
|
if (netlink_send(p_socket, p_request) < 0) return NULL; |
||||
|
|
||||
|
NetlinkList *l_list = NULL; |
||||
|
NetlinkList *l_end = NULL; |
||||
|
int l_size; |
||||
|
int l_done = 0; |
||||
|
while (!l_done) { |
||||
|
struct nlmsghdr *l_hdr = getNetlinkResponse(p_socket, &l_size, &l_done); |
||||
|
if (!l_hdr) { |
||||
|
freeResultList(l_list); |
||||
|
return NULL; |
||||
|
} |
||||
|
|
||||
|
NetlinkList *l_item = newListItem(l_hdr, l_size); |
||||
|
if (!l_item) { |
||||
|
freeResultList(l_list); |
||||
|
return NULL; |
||||
|
} |
||||
|
if (!l_list) { |
||||
|
l_list = l_item; |
||||
|
} else { |
||||
|
l_end->m_next = l_item; |
||||
|
} |
||||
|
l_end = l_item; |
||||
|
} |
||||
|
return l_list; |
||||
|
} |
||||
|
|
||||
|
static size_t maxSize(size_t a, size_t b) |
||||
|
{ |
||||
|
return (a > b ? a : b); |
||||
|
} |
||||
|
|
||||
|
static size_t calcAddrLen(sa_family_t p_family, int p_dataSize) |
||||
|
{ |
||||
|
switch (p_family) { |
||||
|
case AF_INET: |
||||
|
return sizeof(struct sockaddr_in); |
||||
|
case AF_INET6: |
||||
|
return sizeof(struct sockaddr_in6); |
||||
|
case AF_PACKET: |
||||
|
return maxSize(sizeof(struct sockaddr_ll), offsetof(struct sockaddr_ll, sll_addr) + p_dataSize); |
||||
|
default: |
||||
|
return maxSize(sizeof(struct sockaddr), offsetof(struct sockaddr, sa_data) + p_dataSize); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
static void makeSockaddr(sa_family_t p_family, struct sockaddr *p_dest, void *p_data, size_t p_size) |
||||
|
{ |
||||
|
switch (p_family) { |
||||
|
case AF_INET: |
||||
|
memcpy(&((struct sockaddr_in*)p_dest)->sin_addr, p_data, p_size); |
||||
|
break; |
||||
|
case AF_INET6: |
||||
|
memcpy(&((struct sockaddr_in6*)p_dest)->sin6_addr, p_data, p_size); |
||||
|
break; |
||||
|
case AF_PACKET: |
||||
|
memcpy(((struct sockaddr_ll*)p_dest)->sll_addr, p_data, p_size); |
||||
|
((struct sockaddr_ll*)p_dest)->sll_halen = p_size; |
||||
|
break; |
||||
|
default: |
||||
|
memcpy(p_dest->sa_data, p_data, p_size); |
||||
|
break; |
||||
|
} |
||||
|
p_dest->sa_family = p_family; |
||||
|
} |
||||
|
|
||||
|
static void addToEnd(struct ifaddrs **p_resultList, struct ifaddrs *p_entry) |
||||
|
{ |
||||
|
if (!*p_resultList) { |
||||
|
*p_resultList = p_entry; |
||||
|
} else { |
||||
|
struct ifaddrs *l_cur = *p_resultList; |
||||
|
while(l_cur->ifa_next) { |
||||
|
l_cur = l_cur->ifa_next; |
||||
|
} |
||||
|
l_cur->ifa_next = p_entry; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
static int interpretLink(struct nlmsghdr *p_hdr, struct ifaddrs **p_resultList) |
||||
|
{ |
||||
|
struct ifinfomsg *l_info = (struct ifinfomsg *)NLMSG_DATA(p_hdr); |
||||
|
|
||||
|
size_t l_nameSize = 0; |
||||
|
size_t l_addrSize = 0; |
||||
|
size_t l_dataSize = 0; |
||||
|
|
||||
|
size_t l_rtaSize = NLMSG_PAYLOAD(p_hdr, sizeof(struct ifinfomsg)); |
||||
|
struct rtattr *l_rta; |
||||
|
for (l_rta = IFLA_RTA(l_info); RTA_OK(l_rta, l_rtaSize); l_rta = RTA_NEXT(l_rta, l_rtaSize)) { |
||||
|
void *l_rtaData = RTA_DATA(l_rta); |
||||
|
size_t l_rtaDataSize = RTA_PAYLOAD(l_rta); |
||||
|
switch (l_rta->rta_type) { |
||||
|
case IFLA_ADDRESS: |
||||
|
case IFLA_BROADCAST: |
||||
|
l_addrSize += NLMSG_ALIGN(calcAddrLen(AF_PACKET, l_rtaDataSize)); |
||||
|
break; |
||||
|
case IFLA_IFNAME: |
||||
|
l_nameSize += NLMSG_ALIGN(l_rtaSize + 1); |
||||
|
break; |
||||
|
case IFLA_STATS: |
||||
|
l_dataSize += NLMSG_ALIGN(l_rtaSize); |
||||
|
break; |
||||
|
default: |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
struct ifaddrs *l_entry = (struct ifaddrs*)malloc(sizeof(struct ifaddrs) + sizeof(int) + l_nameSize + l_addrSize + l_dataSize); |
||||
|
if (l_entry == NULL) return -1; |
||||
|
memset(l_entry, 0, sizeof(struct ifaddrs)); |
||||
|
l_entry->ifa_name = (char*)""; |
||||
|
|
||||
|
char *l_index = ((char *)l_entry) + sizeof(struct ifaddrs); |
||||
|
char *l_name = l_index + sizeof(int); |
||||
|
char *l_addr = l_name + l_nameSize; |
||||
|
char *l_data = l_addr + l_addrSize; |
||||
|
|
||||
|
/* Save the interface index so we can look it up when handling the addresses. */ |
||||
|
memcpy(l_index, &l_info->ifi_index, sizeof(int)); |
||||
|
|
||||
|
l_entry->ifa_flags = l_info->ifi_flags; |
||||
|
|
||||
|
l_rtaSize = NLMSG_PAYLOAD(p_hdr, sizeof(struct ifinfomsg)); |
||||
|
for (l_rta = IFLA_RTA(l_info); RTA_OK(l_rta, l_rtaSize); l_rta = RTA_NEXT(l_rta, l_rtaSize)) { |
||||
|
void *l_rtaData = RTA_DATA(l_rta); |
||||
|
size_t l_rtaDataSize = RTA_PAYLOAD(l_rta); |
||||
|
switch (l_rta->rta_type) { |
||||
|
case IFLA_ADDRESS: |
||||
|
case IFLA_BROADCAST: |
||||
|
{ |
||||
|
size_t l_addrLen = calcAddrLen(AF_PACKET, l_rtaDataSize); |
||||
|
makeSockaddr(AF_PACKET, (struct sockaddr *)l_addr, l_rtaData, l_rtaDataSize); |
||||
|
((struct sockaddr_ll *)l_addr)->sll_ifindex = l_info->ifi_index; |
||||
|
((struct sockaddr_ll *)l_addr)->sll_hatype = l_info->ifi_type; |
||||
|
if (l_rta->rta_type == IFLA_ADDRESS) { |
||||
|
l_entry->ifa_addr = (struct sockaddr *)l_addr; |
||||
|
} else { |
||||
|
l_entry->ifa_broadaddr = (struct sockaddr *)l_addr; |
||||
|
} |
||||
|
l_addr += NLMSG_ALIGN(l_addrLen); |
||||
|
break; |
||||
|
} |
||||
|
case IFLA_IFNAME: |
||||
|
strncpy(l_name, (char*)l_rtaData, l_rtaDataSize); |
||||
|
l_name[l_rtaDataSize] = '\0'; |
||||
|
l_entry->ifa_name = l_name; |
||||
|
break; |
||||
|
case IFLA_STATS: |
||||
|
memcpy(l_data, l_rtaData, l_rtaDataSize); |
||||
|
l_entry->ifa_data = l_data; |
||||
|
break; |
||||
|
default: |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
addToEnd(p_resultList, l_entry); |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
static struct ifaddrs *findInterface(int p_index, struct ifaddrs **p_links, int p_numLinks) |
||||
|
{ |
||||
|
int l_num = 0; |
||||
|
struct ifaddrs *l_cur = *p_links; |
||||
|
while (l_cur && l_num < p_numLinks) { |
||||
|
char *l_indexPtr = ((char *)l_cur) + sizeof(struct ifaddrs); |
||||
|
int l_index; |
||||
|
memcpy(&l_index, l_indexPtr, sizeof(int)); |
||||
|
if(l_index == p_index) return l_cur; |
||||
|
|
||||
|
l_cur = l_cur->ifa_next; |
||||
|
++l_num; |
||||
|
} |
||||
|
return NULL; |
||||
|
} |
||||
|
|
||||
|
static int interpretAddr(struct nlmsghdr *p_hdr, struct ifaddrs **p_resultList, int p_numLinks) |
||||
|
{ |
||||
|
struct ifaddrmsg *l_info = (struct ifaddrmsg *)NLMSG_DATA(p_hdr); |
||||
|
struct ifaddrs *l_interface = findInterface(l_info->ifa_index, p_resultList, p_numLinks); |
||||
|
|
||||
|
if (l_info->ifa_family == AF_PACKET) return 0; |
||||
|
|
||||
|
size_t l_nameSize = 0; |
||||
|
size_t l_addrSize = 0; |
||||
|
int l_addedNetmask = 0; |
||||
|
|
||||
|
size_t l_rtaSize = NLMSG_PAYLOAD(p_hdr, sizeof(struct ifaddrmsg)); |
||||
|
struct rtattr *l_rta; |
||||
|
for (l_rta = IFA_RTA(l_info); RTA_OK(l_rta, l_rtaSize); l_rta = RTA_NEXT(l_rta, l_rtaSize)) { |
||||
|
void *l_rtaData = RTA_DATA(l_rta); |
||||
|
size_t l_rtaDataSize = RTA_PAYLOAD(l_rta); |
||||
|
|
||||
|
switch (l_rta->rta_type) { |
||||
|
case IFA_ADDRESS: |
||||
|
case IFA_LOCAL: |
||||
|
if ((l_info->ifa_family == AF_INET || l_info->ifa_family == AF_INET6) && !l_addedNetmask) { |
||||
|
/* Make room for netmask. */ |
||||
|
l_addrSize += NLMSG_ALIGN(calcAddrLen(l_info->ifa_family, l_rtaDataSize)); |
||||
|
l_addedNetmask = 1; |
||||
|
} |
||||
|
case IFA_BROADCAST: |
||||
|
l_addrSize += NLMSG_ALIGN(calcAddrLen(l_info->ifa_family, l_rtaDataSize)); |
||||
|
break; |
||||
|
case IFA_LABEL: |
||||
|
l_nameSize += NLMSG_ALIGN(l_rtaSize + 1); |
||||
|
break; |
||||
|
default: |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
struct ifaddrs *l_entry = (struct ifaddrs*)malloc(sizeof(struct ifaddrs) + l_nameSize + l_addrSize); |
||||
|
if (l_entry == NULL) return -1; |
||||
|
memset(l_entry, 0, sizeof(struct ifaddrs)); |
||||
|
l_entry->ifa_name = (l_interface ? l_interface->ifa_name : (char*)""); |
||||
|
|
||||
|
char *l_name = ((char *)l_entry) + sizeof(struct ifaddrs); |
||||
|
char *l_addr = l_name + l_nameSize; |
||||
|
|
||||
|
l_entry->ifa_flags = l_info->ifa_flags; |
||||
|
if (l_interface) { |
||||
|
l_entry->ifa_flags |= l_interface->ifa_flags; |
||||
|
} |
||||
|
|
||||
|
l_rtaSize = NLMSG_PAYLOAD(p_hdr, sizeof(struct ifaddrmsg)); |
||||
|
for (l_rta = IFA_RTA(l_info); RTA_OK(l_rta, l_rtaSize); l_rta = RTA_NEXT(l_rta, l_rtaSize)) { |
||||
|
void *l_rtaData = RTA_DATA(l_rta); |
||||
|
size_t l_rtaDataSize = RTA_PAYLOAD(l_rta); |
||||
|
switch (l_rta->rta_type) { |
||||
|
case IFA_ADDRESS: |
||||
|
case IFA_BROADCAST: |
||||
|
case IFA_LOCAL: |
||||
|
{ |
||||
|
size_t l_addrLen = calcAddrLen(l_info->ifa_family, l_rtaDataSize); |
||||
|
makeSockaddr(l_info->ifa_family, (struct sockaddr *)l_addr, l_rtaData, l_rtaDataSize); |
||||
|
if (l_info->ifa_family == AF_INET6) { |
||||
|
if (IN6_IS_ADDR_LINKLOCAL((struct in6_addr *)l_rtaData) || IN6_IS_ADDR_MC_LINKLOCAL((struct in6_addr *)l_rtaData)) { |
||||
|
((struct sockaddr_in6 *)l_addr)->sin6_scope_id = l_info->ifa_index; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if (l_rta->rta_type == IFA_ADDRESS) { |
||||
|
/* Apparently in a point-to-point network IFA_ADDRESS contains
|
||||
|
the dest address and IFA_LOCAL contains the local address. */ |
||||
|
if(l_entry->ifa_addr) { |
||||
|
l_entry->ifa_dstaddr = (struct sockaddr *)l_addr; |
||||
|
} else { |
||||
|
l_entry->ifa_addr = (struct sockaddr *)l_addr; |
||||
|
} |
||||
|
} else if (l_rta->rta_type == IFA_LOCAL) { |
||||
|
if(l_entry->ifa_addr) { |
||||
|
l_entry->ifa_dstaddr = l_entry->ifa_addr; |
||||
|
} |
||||
|
l_entry->ifa_addr = (struct sockaddr *)l_addr; |
||||
|
} else { |
||||
|
l_entry->ifa_broadaddr = (struct sockaddr *)l_addr; |
||||
|
} |
||||
|
l_addr += NLMSG_ALIGN(l_addrLen); |
||||
|
break; |
||||
|
} |
||||
|
case IFA_LABEL: |
||||
|
strncpy(l_name, (char*)l_rtaData, l_rtaDataSize); |
||||
|
l_name[l_rtaDataSize] = '\0'; |
||||
|
l_entry->ifa_name = l_name; |
||||
|
break; |
||||
|
default: |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if (l_entry->ifa_addr && (l_entry->ifa_addr->sa_family == AF_INET || l_entry->ifa_addr->sa_family == AF_INET6)) { |
||||
|
unsigned l_maxPrefix = (l_entry->ifa_addr->sa_family == AF_INET ? 32 : 128); |
||||
|
unsigned l_prefix = (l_info->ifa_prefixlen > l_maxPrefix ? l_maxPrefix : l_info->ifa_prefixlen); |
||||
|
char l_mask[16] = {0}; |
||||
|
unsigned i; |
||||
|
for (i=0; i<(l_prefix/8); ++i) { |
||||
|
l_mask[i] = 0xff; |
||||
|
} |
||||
|
if (l_prefix % 8) { |
||||
|
l_mask[i] = 0xff << (8 - (l_prefix % 8)); |
||||
|
} |
||||
|
|
||||
|
makeSockaddr(l_entry->ifa_addr->sa_family, (struct sockaddr *)l_addr, l_mask, l_maxPrefix / 8); |
||||
|
l_entry->ifa_netmask = (struct sockaddr *)l_addr; |
||||
|
} |
||||
|
|
||||
|
addToEnd(p_resultList, l_entry); |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
static int interpretLinks(int p_socket, NetlinkList *p_netlinkList, struct ifaddrs **p_resultList) |
||||
|
{ |
||||
|
int l_numLinks = 0; |
||||
|
pid_t l_pid = getpid(); |
||||
|
for(; p_netlinkList; p_netlinkList = p_netlinkList->m_next) { |
||||
|
unsigned int l_nlsize = p_netlinkList->m_size; |
||||
|
struct nlmsghdr *l_hdr; |
||||
|
for(l_hdr = p_netlinkList->m_data; NLMSG_OK(l_hdr, l_nlsize); l_hdr = NLMSG_NEXT(l_hdr, l_nlsize)) |
||||
|
{ |
||||
|
if ((pid_t)l_hdr->nlmsg_pid != l_pid || (int)l_hdr->nlmsg_seq != p_socket) continue; |
||||
|
|
||||
|
if (l_hdr->nlmsg_type == NLMSG_DONE) break; |
||||
|
|
||||
|
if (l_hdr->nlmsg_type == RTM_NEWLINK) { |
||||
|
if(interpretLink(l_hdr, p_resultList) == -1) return -1; |
||||
|
++l_numLinks; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
return l_numLinks; |
||||
|
} |
||||
|
|
||||
|
static int interpretAddrs(int p_socket, NetlinkList *p_netlinkList, struct ifaddrs **p_resultList, int p_numLinks) |
||||
|
{ |
||||
|
pid_t l_pid = getpid(); |
||||
|
for (; p_netlinkList; p_netlinkList = p_netlinkList->m_next) { |
||||
|
unsigned int l_nlsize = p_netlinkList->m_size; |
||||
|
struct nlmsghdr *l_hdr; |
||||
|
for (l_hdr = p_netlinkList->m_data; NLMSG_OK(l_hdr, l_nlsize); l_hdr = NLMSG_NEXT(l_hdr, l_nlsize)) { |
||||
|
if ((pid_t)l_hdr->nlmsg_pid != l_pid || (int)l_hdr->nlmsg_seq != p_socket) continue; |
||||
|
|
||||
|
if (l_hdr->nlmsg_type == NLMSG_DONE) break; |
||||
|
|
||||
|
if (l_hdr->nlmsg_type == RTM_NEWADDR) { |
||||
|
if (interpretAddr(l_hdr, p_resultList, p_numLinks) == -1) return -1; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
int getifaddrs(struct ifaddrs **ifap) |
||||
|
{ |
||||
|
if (!ifap) return -1; |
||||
|
*ifap = NULL; |
||||
|
|
||||
|
int l_socket = netlink_socket(); |
||||
|
if (l_socket < 0) { |
||||
|
return -1; |
||||
|
} |
||||
|
|
||||
|
NetlinkList *l_linkResults = getResultList(l_socket, RTM_GETLINK); |
||||
|
if (!l_linkResults) { |
||||
|
close(l_socket); |
||||
|
return -1; |
||||
|
} |
||||
|
|
||||
|
NetlinkList *l_addrResults = getResultList(l_socket, RTM_GETADDR); |
||||
|
if (!l_addrResults) { |
||||
|
close(l_socket); |
||||
|
freeResultList(l_linkResults); |
||||
|
return -1; |
||||
|
} |
||||
|
|
||||
|
int l_result = 0; |
||||
|
int l_numLinks = interpretLinks(l_socket, l_linkResults, ifap); |
||||
|
if (l_numLinks == -1 || interpretAddrs(l_socket, l_addrResults, ifap, l_numLinks) == -1) { |
||||
|
l_result = -1; |
||||
|
} |
||||
|
|
||||
|
freeResultList(l_linkResults); |
||||
|
freeResultList(l_addrResults); |
||||
|
close(l_socket); |
||||
|
return l_result; |
||||
|
} |
||||
|
|
||||
|
void freeifaddrs(struct ifaddrs *ifa) |
||||
|
{ |
||||
|
struct ifaddrs *l_cur; |
||||
|
while (ifa) { |
||||
|
l_cur = ifa; |
||||
|
ifa = ifa->ifa_next; |
||||
|
free(l_cur); |
||||
|
} |
||||
|
} |
Loading…
Reference in new issue