mirror of
https://github.com/fastfetch-cli/fastfetch.git
synced 2026-09-12 09:58:03 +02:00
李通洲
125 lines
4.2 KiB
C
125 lines
4.2 KiB
C
#include "common/netif.h"
|
|
#include "common/io.h"
|
|
#include "common/mallocHelper.h"
|
|
|
|
#include <net/if.h>
|
|
#include <net/if_dl.h>
|
|
#include <net/route.h>
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/sysctl.h>
|
|
|
|
#define ROUNDUP2(a, n) ((a) > 0 ? (1 + (((a) - 1U) | ((n) - 1))) : (n))
|
|
|
|
#if __DragonFly__
|
|
// https://github.com/DragonFlyBSD/DragonFlyBSD/blob/cf0aa2f1e47a3f0a6055fe427563cb3f3e627064/sys/net/route.h#L315C9-L315C19
|
|
#define ROUNDUP(a) ROUNDUP2((a), sizeof(long))
|
|
#elif __FreeBSD__
|
|
// https://github.com/freebsd/freebsd-src/blob/e4c0ecba44b20ebb2e4d80978c2cb6d16b730cb9/sys/net/route.h#L368C9-L368C16
|
|
#define ROUNDUP(a) ROUNDUP2((a), sizeof(long))
|
|
#elif __NetBSD__
|
|
// https://github.com/NetBSD/src/blob/29beb637d057520c0ed37ac2cde966f7cc0cadf4/sys/net/route.h#L330
|
|
#define ROUNDUP(a) ROUNDUP2((a), sizeof(uint64_t))
|
|
#elif __OpenBSD__
|
|
// https://github.com/openbsd/src/blob/ca647cfa4ec3ccb8360714bc0ebc32a394f7fb6a/regress/sys/netinet/bindconnect/bindconnect.c#L250
|
|
#define ROUNDUP(a) ROUNDUP2((a), sizeof(long))
|
|
#else
|
|
#error unknown platform
|
|
#endif
|
|
|
|
static struct sockaddr*
|
|
get_rt_address(struct rt_msghdr* rtm, int desired) {
|
|
struct sockaddr* sa = (struct sockaddr*) (rtm + 1);
|
|
|
|
for (int i = 0; i < RTAX_MAX; i++) {
|
|
if (rtm->rtm_addrs & (1 << i)) {
|
|
if ((1 << i) == desired) {
|
|
return sa;
|
|
}
|
|
sa = (struct sockaddr*) (ROUNDUP(sa->sa_len) + (char*) sa);
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
bool ffNetifGetDefaultRouteImplV4(FFNetifDefaultRouteResult* result) {
|
|
int mib[6] = { CTL_NET, PF_ROUTE, 0, AF_INET, NET_RT_FLAGS, RTF_GATEWAY };
|
|
size_t needed;
|
|
|
|
if (sysctl(mib, 6, nullptr, &needed, nullptr, 0) < 0 || needed == 0) {
|
|
return false;
|
|
}
|
|
|
|
FF_AUTO_FREE char* buf = malloc(needed);
|
|
|
|
if (sysctl(mib, 6, buf, &needed, nullptr, 0) < 0) {
|
|
return false;
|
|
}
|
|
|
|
char* lim = buf + needed;
|
|
struct rt_msghdr* rtm;
|
|
for (char* next = buf; next < lim; next += rtm->rtm_msglen) {
|
|
rtm = (struct rt_msghdr*) next;
|
|
struct sockaddr* sa = (struct sockaddr*) (rtm + 1);
|
|
|
|
if ((rtm->rtm_flags & RTF_GATEWAY) && !(rtm->rtm_flags & RTF_REJECT) && (sa->sa_family == AF_INET)) {
|
|
struct sockaddr_dl* sdl = (struct sockaddr_dl*) get_rt_address(rtm, RTA_IFP);
|
|
if (sdl && sdl->sdl_family == AF_LINK) {
|
|
if (sdl->sdl_nlen > IF_NAMESIZE) {
|
|
continue;
|
|
}
|
|
memcpy(result->ifName, sdl->sdl_data, sdl->sdl_nlen);
|
|
result->ifName[sdl->sdl_nlen] = '\0';
|
|
result->ifIndex = sdl->sdl_index;
|
|
|
|
// Get the preferred source address
|
|
struct sockaddr_in* src = (struct sockaddr_in*) get_rt_address(rtm, RTA_IFA);
|
|
if (src && src->sin_family == AF_INET) {
|
|
result->preferredSourceAddrV4 = src->sin_addr.s_addr;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool ffNetifGetDefaultRouteImplV6(FFNetifDefaultRouteResult* result) {
|
|
int mib[6] = { CTL_NET, PF_ROUTE, 0, AF_INET6, NET_RT_FLAGS, RTF_GATEWAY };
|
|
size_t needed;
|
|
|
|
if (sysctl(mib, 6, nullptr, &needed, nullptr, 0) < 0 || needed == 0) {
|
|
return false;
|
|
}
|
|
|
|
FF_AUTO_FREE char* buf = malloc(needed);
|
|
|
|
if (sysctl(mib, 6, buf, &needed, nullptr, 0) < 0) {
|
|
return false;
|
|
}
|
|
|
|
char* lim = buf + needed;
|
|
struct rt_msghdr* rtm;
|
|
for (char* next = buf; next < lim; next += rtm->rtm_msglen) {
|
|
rtm = (struct rt_msghdr*) next;
|
|
struct sockaddr* sa = (struct sockaddr*) (rtm + 1);
|
|
|
|
if ((rtm->rtm_flags & RTF_GATEWAY) && !(rtm->rtm_flags & RTF_REJECT) && (sa->sa_family == AF_INET6)) {
|
|
struct sockaddr_dl* sdl = (struct sockaddr_dl*) get_rt_address(rtm, RTA_IFP);
|
|
if (sdl && sdl->sdl_family == AF_LINK) {
|
|
if (sdl->sdl_nlen > IF_NAMESIZE) {
|
|
continue;
|
|
}
|
|
memcpy(result->ifName, sdl->sdl_data, sdl->sdl_nlen);
|
|
result->ifName[sdl->sdl_nlen] = '\0';
|
|
result->ifIndex = sdl->sdl_index;
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|