LocalIP: add support for Windows

src/detection/localip/localip.h

@@ -0,0 +1,17 @@
+#pragma once
+
+#ifndef FF_INCLUDED_detection_localip_localip
+#define FF_INCLUDED_detection_localip_localip
+
+#include "fastfetch.h"
+
+typedef struct FFLocalIpResult
+{
+    FFstrbuf name;
+    FFstrbuf addr;
+    bool ipv6;
+} FFLocalIpResult;
+
+const char* ffDetectLocalIps(const FFinstance* instance, FFlist* results);
+
+#endif

src/detection/localip/localip_linux.c

@@ -0,0 +1,66 @@
+#include "localip.h"
+
+#include <string.h>
+#include <ctype.h>
+#include <ifaddrs.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <net/if.h>
+
+#ifdef __FreeBSD__
+    #include <sys/socket.h> // FreeBSD needs this for AF_INET
+#endif
+
+static void addNewIp(FFlist* list, const char* name, const char* addr, bool ipv6)
+{
+    FFLocalIpResult* ip = (FFLocalIpResult*) ffListAdd(list);
+    ffStrbufInitS(&ip->name, name);
+    ffStrbufInitS(&ip->addr, addr);
+    ip->ipv6 = ipv6;
+}
+
+const char* ffDetectLocalIps(const FFinstance* instance, FFlist* results)
+{
+    struct ifaddrs* ifAddrStruct = NULL;
+    if(getifaddrs(&ifAddrStruct) < 0)
+        return "getifaddrs(&ifAddrStruct) failed";
+
+    for (struct ifaddrs* ifa = ifAddrStruct; ifa; ifa = ifa->ifa_next)
+    {
+        if (!ifa->ifa_addr || !(ifa->ifa_flags & IFF_RUNNING))
+            continue;
+
+        // loop back
+        if (strncmp(ifa->ifa_name, "lo", 2) == 0 && (ifa->ifa_name[2] == '\0' || isdigit(ifa->ifa_name[2])) && !instance->config.localIpShowLoop)
+            continue;
+
+        if (instance->config.localIpNamePrefix.length && strncmp(ifa->ifa_name, instance->config.localIpNamePrefix.chars, instance->config.localIpNamePrefix.length) != 0)
+            continue;
+
+        if (ifa->ifa_addr->sa_family == AF_INET)
+        {
+            // IPv4
+            if (!instance->config.localIpShowIpV4)
+                continue;
+
+            struct sockaddr_in* ipv4 = (struct sockaddr_in*) ifa->ifa_addr;
+            char addressBuffer[INET_ADDRSTRLEN];
+            inet_ntop(AF_INET, &ipv4->sin_addr, addressBuffer, INET_ADDRSTRLEN);
+            addNewIp(results, ifa->ifa_name, addressBuffer, false);
+        }
+        else if (ifa->ifa_addr->sa_family == AF_INET6)
+        {
+            // IPv6
+            if (!instance->config.localIpShowIpV6)
+                continue;
+
+            struct sockaddr_in6* ipv6 = (struct sockaddr_in6 *)ifa->ifa_addr;
+            char addressBuffer[INET6_ADDRSTRLEN];
+            inet_ntop(AF_INET6, &ipv6->sin6_addr, addressBuffer, INET6_ADDRSTRLEN);
+            addNewIp(results, ifa->ifa_name, addressBuffer, true);
+        }
+    }
+
+    if (ifAddrStruct) freeifaddrs(ifAddrStruct);
+    return NULL;
+}

src/detection/localip/localip_windows.c

@@ -0,0 +1,89 @@
+#include <winsock2.h>
+#include <ws2tcpip.h>
+#include <iphlpapi.h>
+#include <wchar.h>
+
+#include "localip.h"
+
+static void addNewIp(FFlist* list, const wchar_t* name, const char* addr, bool ipv6)
+{
+    FFLocalIpResult* ip = (FFLocalIpResult*) ffListAdd(list);
+
+    int len = (int)wcslen(name);
+    int size_needed = WideCharToMultiByte(CP_UTF8, 0, name, len, NULL, 0, NULL, NULL);
+    ffStrbufInitA(&ip->name, (uint32_t)size_needed + 1);
+    WideCharToMultiByte(CP_UTF8, 0, name, len, ip->name.chars, size_needed, NULL, NULL);
+    ip->name.length = (uint32_t)size_needed;
+    ip->name.chars[size_needed] = '\0';
+
+    ffStrbufInitS(&ip->addr, addr);
+    ip->ipv6 = ipv6;
+}
+
+const char* ffDetectLocalIps(const FFinstance* instance, FFlist* results)
+{
+    IP_ADAPTER_ADDRESSES* adapter_addresses = NULL;
+
+    // Start with a 16 KB buffer and resize if needed -
+    // multiple attempts in case interfaces change while
+    // we are in the middle of querying them.
+    DWORD adapter_addresses_buffer_size = 16 * 1024;
+    for (int attempts = 0; attempts != 3; ++attempts)
+    {
+        adapter_addresses = (IP_ADAPTER_ADDRESSES*)realloc(adapter_addresses, adapter_addresses_buffer_size);
+        assert(adapter_addresses);
+
+        DWORD error = GetAdaptersAddresses(
+            AF_UNSPEC,
+            GAA_FLAG_SKIP_ANYCAST |
+            GAA_FLAG_SKIP_MULTICAST |
+            GAA_FLAG_SKIP_DNS_SERVER |
+            GAA_FLAG_SKIP_FRIENDLY_NAME,
+            NULL,
+            adapter_addresses,
+            &adapter_addresses_buffer_size);
+
+        if (error == ERROR_SUCCESS)
+            break;
+        else if (ERROR_BUFFER_OVERFLOW == error)
+            continue;
+        else
+            return "GetAdaptersAddresses() failed";
+    }
+
+    // Iterate through all of the adapters
+    for (IP_ADAPTER_ADDRESSES* adapter = adapter_addresses; adapter; adapter = adapter->Next)
+    {
+        if (adapter->IfType == IF_TYPE_SOFTWARE_LOOPBACK && !instance->config.localIpShowLoop)
+            continue;
+
+        for (IP_ADAPTER_UNICAST_ADDRESS* ifa = adapter->FirstUnicastAddress; ifa; ifa = ifa->Next)
+        {
+            if (ifa->Address.lpSockaddr->sa_family == AF_INET)
+            {
+                // IPv4
+                if (!instance->config.localIpShowIpV4)
+                    continue;
+
+                SOCKADDR_IN* ipv4 = (SOCKADDR_IN*) ifa->Address.lpSockaddr;
+                char addressBuffer[INET_ADDRSTRLEN];
+                inet_ntop(AF_INET, &ipv4->sin_addr, addressBuffer, INET_ADDRSTRLEN);
+                addNewIp(results, adapter->FriendlyName, addressBuffer, false);
+            }
+            else if (ifa->Address.lpSockaddr->sa_family == AF_INET6)
+            {
+                // IPv6
+                if (!instance->config.localIpShowIpV6)
+                    continue;
+
+                SOCKADDR_IN6* ipv6 = (SOCKADDR_IN6*) ifa->Address.lpSockaddr;
+                char addressBuffer[INET6_ADDRSTRLEN];
+                inet_ntop(AF_INET6, &ipv6->sin6_addr, addressBuffer, INET6_ADDRSTRLEN);
+                addNewIp(results, adapter->FriendlyName, addressBuffer, false);
+            }
+        }
+    }
+
+    free(adapter_addresses);
+    return NULL;
+}