DNS: add new module

CMakeLists.txt

@@ -340,6 +340,7 @@ set(LIBFASTFETCH_SRC
     src/modules/de/de.c
     src/modules/disk/disk.c
     src/modules/diskio/diskio.c
+    src/modules/dns/dns.c
     src/modules/editor/editor.c
     src/modules/font/font.c
     src/modules/gpu/gpu.c
@@ -417,6 +418,7 @@ if(LINUX)
         src/detection/cursor/cursor_linux.c
         src/detection/bluetooth/bluetooth_linux.c
         src/detection/disk/disk_linux.c
+        src/detection/dns/dns_linux.c
         src/detection/physicaldisk/physicaldisk_linux.c
         src/detection/physicalmemory/physicalmemory_linux.c
         src/detection/diskio/diskio_linux.c
@@ -487,6 +489,7 @@ elseif(ANDROID)
         src/detection/cursor/cursor_nosupport.c
         src/detection/cpuusage/cpuusage_linux.c
         src/detection/disk/disk_linux.c
+        src/detection/dns/dns_nosupport.c
         src/detection/physicaldisk/physicaldisk_linux.c
         src/detection/physicalmemory/physicalmemory_nosupport.c
         src/detection/diskio/diskio_linux.c
@@ -548,6 +551,7 @@ elseif(BSD)
         src/detection/cpuusage/cpuusage_bsd.c
         src/detection/cursor/cursor_linux.c
         src/detection/disk/disk_bsd.c
+        src/detection/dns/dns_linux.c
         src/detection/physicaldisk/physicaldisk_bsd.c
         src/detection/physicalmemory/physicalmemory_linux.c
         src/detection/diskio/diskio_bsd.c
@@ -619,6 +623,7 @@ elseif(APPLE)
         src/detection/cpuusage/cpuusage_apple.c
         src/detection/cursor/cursor_apple.m
         src/detection/disk/disk_bsd.c
+        src/detection/dns/dns_linux.c
         src/detection/physicaldisk/physicaldisk_apple.c
         src/detection/physicalmemory/physicalmemory_apple.m
         src/detection/diskio/diskio_apple.c
@@ -684,6 +689,7 @@ elseif(WIN32)
         src/detection/physicaldisk/physicaldisk_windows.c
         src/detection/diskio/diskio_windows.c
         src/detection/displayserver/displayserver_windows.c
+        src/detection/dns/dns_linux.c
         src/detection/font/font_windows.c
         src/detection/gpu/gpu_windows.c
         src/detection/host/host_windows.c

src/common/modules.c

@@ -34,6 +34,7 @@ static FFModuleBaseInfo* D[] = {
     (void*) &instance.config.modules.display,
     (void*) &instance.config.modules.disk,
     (void*) &instance.config.modules.diskIo,
+    (void*) &instance.config.modules.dns,
     NULL,
 };
 

src/detection/dns/dns.h

@@ -0,0 +1,5 @@
+#pragma once
+
+#include "fastfetch.h"
+
+const char* ffDetectDNS(FFlist* results /* list of FFstrbuf */);

src/detection/dns/dns_linux.c

@@ -0,0 +1,32 @@
+#include "detection/dns/dns.h"
+
+#include <arpa/inet.h>
+#include <resolv.h>
+
+const char* ffDetectDNS(FFlist* results)
+{
+    if (res_init() < 0)
+        return "res_init() failed";
+
+    for (int i = 0; i < _res.nscount; ++i)
+    {
+        struct sockaddr_in* addr = &_res.nsaddr_list[i];
+        char addressBuffer[INET_ADDRSTRLEN + 4];
+        inet_ntop(AF_INET, &addr->sin_addr, addressBuffer, INET_ADDRSTRLEN);
+
+        FFstrbuf* result = (FFstrbuf*) ffListAdd(results);
+        ffStrbufInitS(result, addressBuffer);
+    }
+
+    for (int i = 0; i < _res._u._ext.nscount; ++i)
+    {
+        struct sockaddr_in6* addr = _res._u._ext.nsaddrs[i];
+        char addressBuffer[INET6_ADDRSTRLEN + 4];
+        inet_ntop(AF_INET6, &addr->sin6_addr, addressBuffer, INET6_ADDRSTRLEN);
+
+        FFstrbuf* result = (FFstrbuf*) ffListAdd(results);
+        ffStrbufInitS(result, addressBuffer);
+    }
+
+    return NULL;
+}

src/detection/dns/dns_nosupport.c

@@ -0,0 +1,6 @@
+#include "detection/dns/dns.h"
+
+const char* ffDetectDNS(FF_MAYBE_UNUSED FFlist* results)
+{
+    return "Not supported on this platform";
+}

src/modules/dns/dns.c

@@ -0,0 +1,141 @@
+#include "common/printing.h"
+#include "common/jsonconfig.h"
+#include "detection/dns/dns.h"
+#include "modules/dns/dns.h"
+#include "util/stringUtils.h"
+
+#define FF_DNS_NUM_FORMAT_ARGS 1
+
+void ffPrintDNS(FFDNSOptions* options)
+{
+    FF_LIST_AUTO_DESTROY result = ffListCreate(sizeof(FFstrbuf));
+
+    const char* error = ffDetectDNS(&result);
+
+    if (error)
+    {
+        ffPrintError(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "%s", error);
+        return;
+    }
+
+    if (result.length == 0)
+    {
+        ffPrintError(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "NO DNS detected");
+        return;
+    }
+
+    FF_STRBUF_AUTO_DESTROY buf = ffStrbufCreate();
+    FF_LIST_FOR_EACH(FFstrbuf, item, result)
+    {
+        if (buf.length)
+            ffStrbufAppendC(&buf, ' ');
+        ffStrbufAppend(&buf, item);
+    }
+
+    if(options->moduleArgs.outputFormat.length == 0)
+    {
+        ffPrintLogoAndKey(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT);
+
+        ffStrbufPutTo(&buf, stdout);
+    }
+    else
+    {
+        FF_PRINT_FORMAT_CHECKED(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, FF_DNS_NUM_FORMAT_ARGS, ((FFformatarg[]) {
+            {FF_FORMAT_ARG_TYPE_STRBUF, &buf, "result"},
+        }));
+    }
+
+    FF_LIST_FOR_EACH(FFstrbuf, item, result)
+    {
+        ffStrbufDestroy(item);
+    }
+}
+
+bool ffParseDNSCommandOptions(FFDNSOptions* options, const char* key, const char* value)
+{
+    const char* subKey = ffOptionTestPrefix(key, FF_DNS_MODULE_NAME);
+    if (!subKey) return false;
+    if (ffOptionParseModuleArgs(key, subKey, value, &options->moduleArgs))
+        return true;
+
+    return false;
+}
+
+void ffParseDNSJsonObject(FFDNSOptions* options, yyjson_val* module)
+{
+    yyjson_val *key_, *val;
+    size_t idx, max;
+    yyjson_obj_foreach(module, idx, max, key_, val)
+    {
+        const char* key = yyjson_get_str(key_);
+        if(ffStrEqualsIgnCase(key, "type"))
+            continue;
+
+        if (ffJsonConfigParseModuleArgs(key, val, &options->moduleArgs))
+            continue;
+
+        ffPrintError(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "Unknown JSON key %s", key);
+    }
+}
+
+void ffGenerateDNSJsonConfig(FFDNSOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    __attribute__((__cleanup__(ffDestroyDNSOptions))) FFDNSOptions defaultOptions;
+    ffInitDNSOptions(&defaultOptions);
+
+    ffJsonConfigGenerateModuleArgsConfig(doc, module, &defaultOptions.moduleArgs, &options->moduleArgs);
+}
+
+void ffGenerateDNSJsonResult(FF_MAYBE_UNUSED FFDNSOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    FF_LIST_AUTO_DESTROY result = ffListCreate(sizeof(FFstrbuf));
+
+    const char* error = ffDetectDNS(&result);
+
+    if (error)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", error);
+        goto exit;
+    }
+
+    yyjson_mut_val* arr = yyjson_mut_obj_add_arr(doc, module, "result");
+
+    FF_LIST_FOR_EACH(FFstrbuf, item, result)
+    {
+        yyjson_mut_arr_add_strbuf(doc, arr, item);
+    }
+
+exit:
+    FF_LIST_FOR_EACH(FFstrbuf, item, result)
+    {
+        ffStrbufDestroy(item);
+    }
+}
+
+void ffPrintDNSHelpFormat(void)
+{
+    FF_PRINT_MODULE_FORMAT_HELP_CHECKED(FF_DNS_MODULE_NAME, "{1}", FF_DNS_NUM_FORMAT_ARGS, ((const char* []) {
+        "DNS result - result",
+    }));
+}
+
+void ffInitDNSOptions(FFDNSOptions* options)
+{
+    ffOptionInitModuleBaseInfo(
+        &options->moduleInfo,
+        FF_DNS_MODULE_NAME,
+        "Print configured DNS servers",
+        ffParseDNSCommandOptions,
+        ffParseDNSJsonObject,
+        ffPrintDNS,
+        ffGenerateDNSJsonResult,
+        ffPrintDNSHelpFormat,
+        ffGenerateDNSJsonConfig
+    );
+    ffOptionInitModuleArg(&options->moduleArgs);
+}
+
+void ffDestroyDNSOptions(FFDNSOptions* options)
+{
+    ffOptionDestroyModuleArg(&options->moduleArgs);
+}

src/modules/dns/dns.h

@@ -0,0 +1,9 @@
+#pragma once
+
+#include "fastfetch.h"
+
+#define FF_DNS_MODULE_NAME "DNS"
+
+void ffPrintDNS(FFDNSOptions* options);
+void ffInitDNSOptions(FFDNSOptions* options);
+void ffDestroyDNSOptions(FFDNSOptions* options);

src/modules/dns/option.h

@@ -0,0 +1,11 @@
+#pragma once
+
+// This file will be included in "fastfetch.h", do NOT put unnecessary things here
+
+#include "common/option.h"
+
+typedef struct FFDNSOptions
+{
+    FFModuleBaseInfo moduleInfo;
+    FFModuleArgs moduleArgs;
+} FFDNSOptions;

src/modules/modules.h

@@ -23,6 +23,7 @@
 #include "modules/diskio/diskio.h"
 #include "modules/display/display.h"
 #include "modules/de/de.h"
+#include "modules/dns/dns.h"
 #include "modules/editor/editor.h"
 #include "modules/font/font.h"
 #include "modules/gamepad/gamepad.h"

src/modules/options.h

@@ -23,6 +23,7 @@
 #include "modules/disk/option.h"
 #include "modules/diskio/option.h"
 #include "modules/display/option.h"
+#include "modules/dns/option.h"
 #include "modules/editor/option.h"
 #include "modules/font/option.h"
 #include "modules/host/option.h"

src/options/modules.c

@@ -24,6 +24,7 @@ void ffOptionsInitModules(FFOptionsModules* options)
     ffInitDiskOptions(&options->disk);
     ffInitDiskIOOptions(&options->diskIo);
     ffInitDisplayOptions(&options->display);
+    ffInitDNSOptions(&options->dns);
     ffInitEditorOptions(&options->editor);
     ffInitFontOptions(&options->font);
     ffInitGPUOptions(&options->gpu);
@@ -94,6 +95,7 @@ void ffOptionsDestroyModules(FFOptionsModules* options)
     ffDestroyDiskOptions(&options->disk);
     ffDestroyDiskIOOptions(&options->diskIo);
     ffDestroyDisplayOptions(&options->display);
+    ffDestroyDNSOptions(&options->dns);
     ffDestroyEditorOptions(&options->editor);
     ffDestroyFontOptions(&options->font);
     ffDestroyGPUOptions(&options->gpu);

src/options/modules.h

@@ -25,6 +25,7 @@ typedef struct FFOptionsModules
     FFDiskOptions disk;
     FFDiskIOOptions diskIo;
     FFDisplayOptions display;
+    FFDNSOptions dns;
     FFEditorOptions editor;
     FFFontOptions font;
     FFGPUOptions gpu;