271 lines
6.8 KiB
C
271 lines
6.8 KiB
C
|
/******************************************************************************
|
||
|
|
Copyright (C) 2016 B. Lee <bl4@postpile.net>
|
||
|
|
|
||
|
|
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 2 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 "net-if.h"
|
||
|
|
#include <util/platform.h>
|
||
|
|
#include <util/dstr.h>
|
||
|
|
|
||
|
|
#define do_log(level, format, ...) \
|
||
|
|
blog(level, "[net if] " format, ##__VA_ARGS__)
|
||
|
|
|
||
|
|
#define warn(format, ...) do_log(LOG_WARNING, format, ##__VA_ARGS__)
|
||
|
|
#define info(format, ...) do_log(LOG_INFO, format, ##__VA_ARGS__)
|
||
|
|
#define debug(format, ...) do_log(LOG_DEBUG, format, ##__VA_ARGS__)
|
||
|
|
|
||
|
|
static inline void netif_saddr_data_push_back(struct netif_saddr_data *sd,
|
||
|
|
const char *ip, const char *adapter)
|
||
|
|
{
|
||
|
|
struct netif_saddr_item item;
|
||
|
|
struct dstr full_name = {0};
|
||
|
|
char *ip_dup = bstrdup(ip);
|
||
|
|
|
||
|
|
if (adapter && *adapter)
|
||
|
|
dstr_printf(&full_name, "[%s] %s", adapter, ip);
|
||
|
|
else
|
||
|
|
dstr_copy(&full_name, ip);
|
||
|
|
|
||
|
|
item.name = full_name.array;
|
||
|
|
item.addr = ip_dup;
|
||
|
|
|
||
|
|
da_push_back(sd->addrs, &item);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void netif_convert_to_string(char *dest,
|
||
|
|
struct sockaddr_storage *byte_address)
|
||
|
|
{
|
||
|
|
int family = byte_address->ss_family;
|
||
|
|
char temp_char[INET6_ADDRSTRLEN] = {0};
|
||
|
|
|
||
|
|
#ifndef _WIN32
|
||
|
|
if (family == AF_INET)
|
||
|
|
inet_ntop(family, &(((struct sockaddr_in*)byte_address)->sin_addr),
|
||
|
|
temp_char, INET6_ADDRSTRLEN);
|
||
|
|
else if (family == AF_INET6)
|
||
|
|
inet_ntop(family, &(((struct sockaddr_in*)byte_address)->sin_addr),
|
||
|
|
temp_char, INET6_ADDRSTRLEN);
|
||
|
|
#else
|
||
|
|
if (family == AF_INET)
|
||
|
|
InetNtopA(family, &(((SOCKADDR_IN *)byte_address)->sin_addr),
|
||
|
|
temp_char, INET6_ADDRSTRLEN);
|
||
|
|
else if (family == AF_INET6)
|
||
|
|
InetNtopA(family, &(((SOCKADDR_IN6 *)byte_address)->sin6_addr),
|
||
|
|
temp_char, INET6_ADDRSTRLEN);
|
||
|
|
#endif
|
||
|
|
strncpy(dest, temp_char, INET6_ADDRSTRLEN);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void netif_push(struct sockaddr *copy_source,
|
||
|
|
struct netif_saddr_data *saddr_d,
|
||
|
|
const char *adapter)
|
||
|
|
{
|
||
|
|
char temp_char[INET6_ADDRSTRLEN] = {0};
|
||
|
|
struct sockaddr_storage sa = {0};
|
||
|
|
|
||
|
|
if (copy_source->sa_family == AF_INET)
|
||
|
|
memcpy(&sa, copy_source, sizeof(struct sockaddr_in));
|
||
|
|
else if (copy_source->sa_family == AF_INET6)
|
||
|
|
memcpy(&sa, copy_source, sizeof(struct sockaddr_in6));
|
||
|
|
|
||
|
|
netif_convert_to_string(temp_char, &sa);
|
||
|
|
netif_saddr_data_push_back(saddr_d, temp_char, adapter);
|
||
|
|
}
|
||
|
|
|
||
|
|
void netif_log_saddrs(struct netif_saddr_data *sd)
|
||
|
|
{
|
||
|
|
for(size_t i = 0; i < sd->addrs.num; i++)
|
||
|
|
info("\t\t%s", sd->addrs.array[i].name);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool netif_str_to_addr(struct sockaddr_storage *out, int *addr_len,
|
||
|
|
const char *addr)
|
||
|
|
{
|
||
|
|
bool ipv6;
|
||
|
|
|
||
|
|
memset(out, 0, sizeof(*out));
|
||
|
|
*addr_len = 0;
|
||
|
|
|
||
|
|
if (!addr)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
ipv6 = (strchr(addr, ':') != NULL);
|
||
|
|
out->ss_family = ipv6 ? AF_INET6 : AF_INET;
|
||
|
|
*addr_len = sizeof(*out);
|
||
|
|
|
||
|
|
#ifdef _WIN32
|
||
|
|
int ret = WSAStringToAddressA((LPSTR)addr, out->ss_family, NULL,
|
||
|
|
(LPSOCKADDR)out, addr_len);
|
||
|
|
if (ret == SOCKET_ERROR)
|
||
|
|
warn("Could not parse address, error code: %d", GetLastError());
|
||
|
|
return ret != SOCKET_ERROR;
|
||
|
|
#else
|
||
|
|
struct sockaddr_in *sin = (struct sockaddr_in *)out;
|
||
|
|
if (inet_pton(out->ss_family, addr, &sin->sin_addr)) {
|
||
|
|
*addr_len = ipv6 ?
|
||
|
|
sizeof(struct sockaddr_in6) :
|
||
|
|
sizeof(struct sockaddr_in);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
return false;
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifndef _WIN32
|
||
|
|
static inline bool is_loopback(struct ifaddrs *ifa)
|
||
|
|
{
|
||
|
|
const char *n = ifa->ifa_name;
|
||
|
|
return n && (strcmp(n, "lo") == 0 || strcmp(n, "lo0") == 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline void netif_get_addrs_nix(struct netif_saddr_data *ifaddrs)
|
||
|
|
{
|
||
|
|
struct ifaddrs *ifaddr, *ifa;
|
||
|
|
unsigned int family, s;
|
||
|
|
char host[NI_MAXHOST];
|
||
|
|
|
||
|
|
if (getifaddrs(&ifaddr) == -1) {
|
||
|
|
warn("getifaddrs() failed");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
|
||
|
|
if (ifa->ifa_addr == NULL || is_loopback(ifa))
|
||
|
|
continue;
|
||
|
|
|
||
|
|
family = ifa->ifa_addr->sa_family;
|
||
|
|
|
||
|
|
if ((family == AF_INET) || (family == AF_INET6)) {
|
||
|
|
s = getnameinfo(ifa->ifa_addr,
|
||
|
|
(family == AF_INET) ?
|
||
|
|
sizeof(struct sockaddr_in) :
|
||
|
|
sizeof(struct sockaddr_in6),
|
||
|
|
host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
|
||
|
|
if (s != 0) {
|
||
|
|
warn("getnameinfo() failed: %s",
|
||
|
|
gai_strerror(s));
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
netif_push(ifa->ifa_addr, ifaddrs, ifa->ifa_name);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
freeifaddrs(ifaddr);
|
||
|
|
}
|
||
|
|
|
||
|
|
#else
|
||
|
|
|
||
|
|
static inline PIP_ADAPTER_ADDRESSES get_adapters(void)
|
||
|
|
{
|
||
|
|
PIP_ADAPTER_ADDRESSES adapter = NULL;
|
||
|
|
unsigned long ret = 0;
|
||
|
|
unsigned long out_buf_len = 4096;
|
||
|
|
unsigned long flags =
|
||
|
|
GAA_FLAG_SKIP_ANYCAST |
|
||
|
|
GAA_FLAG_SKIP_MULTICAST |
|
||
|
|
GAA_FLAG_SKIP_DNS_SERVER;
|
||
|
|
const int max_tries = 3;
|
||
|
|
int i = 0;
|
||
|
|
|
||
|
|
do {
|
||
|
|
adapter = (IP_ADAPTER_ADDRESSES*)bmalloc(out_buf_len);
|
||
|
|
if (!adapter)
|
||
|
|
return NULL;
|
||
|
|
|
||
|
|
ret = GetAdaptersAddresses(AF_UNSPEC, flags, NULL, adapter,
|
||
|
|
&out_buf_len);
|
||
|
|
if (ret == ERROR_BUFFER_OVERFLOW) {
|
||
|
|
bfree(adapter);
|
||
|
|
adapter = NULL;
|
||
|
|
} else {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
i++;
|
||
|
|
} while ((ret == ERROR_BUFFER_OVERFLOW) && (i < max_tries));
|
||
|
|
|
||
|
|
if (ret != NO_ERROR && ret != ERROR_NO_DATA) {
|
||
|
|
LPSTR msg_buf = NULL;
|
||
|
|
|
||
|
|
bfree(adapter);
|
||
|
|
adapter = NULL;
|
||
|
|
|
||
|
|
FormatMessageA(
|
||
|
|
FORMAT_MESSAGE_ALLOCATE_BUFFER |
|
||
|
|
FORMAT_MESSAGE_FROM_SYSTEM |
|
||
|
|
FORMAT_MESSAGE_IGNORE_INSERTS,
|
||
|
|
NULL, ret,
|
||
|
|
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
||
|
|
msg_buf, 0, NULL);
|
||
|
|
if (msg_buf) {
|
||
|
|
warn("Call to GetAdaptersAddresses failed: %s (%d)",
|
||
|
|
msg_buf, ret);
|
||
|
|
LocalFree(msg_buf);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return adapter;
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline void netif_get_addrs_win32(struct netif_saddr_data *ifaddrs)
|
||
|
|
{
|
||
|
|
PIP_ADAPTER_ADDRESSES adapter = get_adapters();
|
||
|
|
PIP_ADAPTER_UNICAST_ADDRESS unicast = NULL;
|
||
|
|
PIP_ADAPTER_ADDRESSES cur_adap = NULL;
|
||
|
|
SOCKET_ADDRESS socket_addr;
|
||
|
|
int family;
|
||
|
|
|
||
|
|
if (!adapter)
|
||
|
|
return;
|
||
|
|
|
||
|
|
for (cur_adap = adapter; !!cur_adap; cur_adap = cur_adap->Next) {
|
||
|
|
char *adap_name = NULL;
|
||
|
|
|
||
|
|
if (cur_adap->OperStatus != IfOperStatusUp ||
|
||
|
|
cur_adap->IfType == IF_TYPE_SOFTWARE_LOOPBACK)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
os_wcs_to_utf8_ptr(cur_adap->FriendlyName, 0, &adap_name);
|
||
|
|
|
||
|
|
unicast = cur_adap->FirstUnicastAddress;
|
||
|
|
|
||
|
|
for (; !!unicast; unicast = unicast->Next) {
|
||
|
|
socket_addr = unicast->Address;
|
||
|
|
|
||
|
|
family = socket_addr.lpSockaddr->sa_family;
|
||
|
|
if (family == AF_INET || family == AF_INET6)
|
||
|
|
netif_push(socket_addr.lpSockaddr, ifaddrs,
|
||
|
|
adap_name);
|
||
|
|
}
|
||
|
|
|
||
|
|
bfree(adap_name);
|
||
|
|
}
|
||
|
|
|
||
|
|
bfree(adapter);
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
void netif_get_addrs(struct netif_saddr_data *ifaddrs)
|
||
|
|
{
|
||
|
|
da_init(ifaddrs->addrs);
|
||
|
|
|
||
|
|
#ifdef _WIN32
|
||
|
|
netif_get_addrs_win32(ifaddrs);
|
||
|
|
#else
|
||
|
|
netif_get_addrs_nix(ifaddrs);
|
||
|
|
#endif
|
||
|
|
}
|