LocalIp: add new module

2026-10-29 08:13:44 +01:00 · 2021-10-17 12:42:15 +08:00
parent 0b3f96610b
commit 38e2f8cf3e
7 changed files with 116 additions and 2 deletions
@@ -0,0 +1,75 @@
+#include "fastfetch.h"
+
+#define FF_LOCALIP_MODULE_NAME "Local Ip"
+#define FF_LOCALIP_NUM_FORMAT_ARGS 1
+
+#include <sys/types.h>
+#include <ifaddrs.h>
+#include <netinet/in.h>
+#include <string.h>
+#include <arpa/inet.h>
+
+static void printValue(FFinstance* instance, const char* ifaName, const char* addressBuffer)
+{
+    FF_STRBUF_CREATE(key);
+
+    if (instance->config.localIpKey.length == 0) {
+        ffStrbufSetF(&key, FF_LOCALIP_MODULE_NAME " (%s)", ifaName);
+    } else {
+        ffParseFormatString(&key, &instance->config.localIpKey, NULL, 1, (FFformatarg[]){
+            {FF_FORMAT_ARG_TYPE_STRING, ifaName}
+        });
+    }
+
+    if (instance->config.localIpFormat.length == 0) {
+        ffPrintLogoAndKey(instance, FF_LOCALIP_MODULE_NAME, 0, &key);
+        puts(addressBuffer);
+    } else {
+        ffPrintFormatString(instance, FF_LOCALIP_MODULE_NAME, 0, &key, &instance->config.localIpFormat, NULL, FF_LOCALIP_NUM_FORMAT_ARGS, (FFformatarg[]){
+            {FF_FORMAT_ARG_TYPE_STRING, addressBuffer}
+        });
+    }
+
+    ffStrbufDestroy(&key);
+}
+
+void ffPrintLocalIp(FFinstance* instance)
+{
+    struct ifaddrs* ifAddrStruct = NULL;
+    int ret = getifaddrs(&ifAddrStruct);
+    if (ret < 0) {
+        ffPrintError(instance, FF_LOCALIP_MODULE_NAME, 0, &instance->config.localIpKey, &instance->config.localIpFormat, FF_LOCALIP_NUM_FORMAT_ARGS, "getifaddrs(&ifAddrStruct) < 0 (%i)", ret);
+        return;
+    }
+
+    for (struct ifaddrs* ifa = ifAddrStruct; ifa != NULL; ifa = ifa->ifa_next) {
+        if (!ifa->ifa_addr)
+            continue;
+
+        // loop back
+        if (strcmp(ifa->ifa_name, "lo") == 0 && !instance->config.localIpShowLoop)
+            continue;
+
+        if (ifa->ifa_addr->sa_family == AF_INET) { // check it is IP4
+            // is a valid IP4 Address
+            if (!instance->config.localIpShowIpV4)
+                continue;
+
+            void* tmpAddrPtr=&((struct sockaddr_in *)ifa->ifa_addr)->sin_addr;
+            char addressBuffer[INET_ADDRSTRLEN];
+            inet_ntop(AF_INET, tmpAddrPtr, addressBuffer, INET_ADDRSTRLEN);
+            printValue(instance, ifa->ifa_name, addressBuffer);
+        } else if (ifa->ifa_addr->sa_family == AF_INET6) { // check it is IP6
+            // is a valid IP6 Address
+            if (!instance->config.localIpShowIpV6)
+                continue;
+
+            void* tmpAddrPtr=&((struct sockaddr_in6 *)ifa->ifa_addr)->sin6_addr;
+            char addressBuffer[INET6_ADDRSTRLEN];
+            inet_ntop(AF_INET6, tmpAddrPtr, addressBuffer, INET6_ADDRSTRLEN);
+            printValue(instance, ifa->ifa_name, addressBuffer);
+        }
+    }
+
+    if (ifAddrStruct) freeifaddrs(ifAddrStruct);
+}