in_windows_exporter_metrics: Support TCPv4/TCPv6 metrics

Signed-off-by: Hiroshi Hatake <[email protected]>

Author: cosmo0920

plugins/in_windows_exporter_metrics/CMakeLists.txt

@@ -19,6 +19,7 @@ set(src
   we_wmi_memory.c
   we_wmi_paging_file.c
   we_wmi_process.c
+  we_wmi_tcp.c
   )
 
 set(libs

plugins/in_windows_exporter_metrics/we.c

@@ -45,6 +45,7 @@
 #include "we_wmi_memory.h"
 #include "we_wmi_paging_file.h"
 #include "we_wmi_process.h"
+#include "we_wmi_tcp.h"
 
 static int we_timer_cpu_metrics_cb(struct flb_input_instance *ins,
                                    struct flb_config *config, void *in_context)
@@ -186,6 +187,16 @@ static int we_timer_wmi_process_metrics_cb(struct flb_input_instance *ins,
     return 0;
 }
 
+static int we_timer_wmi_tcp_metrics_cb(struct flb_input_instance *ins,
+                                       struct flb_config *config, void *in_context)
+{
+    struct flb_ne *ctx = in_context;
+
+    we_wmi_tcp_update(ctx);
+
+    return 0;
+}
+
 struct flb_we_callback {
     char *name;
     void (*func)(char *, void *, void *);
@@ -336,6 +347,13 @@ static void we_wmi_process_update_cb(char *name, void *p1, void *p2)
     we_wmi_process_update(ctx);
 }
 
+static void we_wmi_tcp_update_cb(char *name, void *p1, void *p2)
+{
+    struct flb_we *ctx = p1;
+
+    we_wmi_tcp_update(ctx);
+}
+
 static int we_update_cb(struct flb_we *ctx, char *name)
 {
     int ret;
@@ -363,6 +381,7 @@ struct flb_we_callback ne_callbacks[] = {
     { "memory", we_wmi_memory_update_cb },
     { "paging_file", we_wmi_paging_file_update_cb },
     { "process", we_wmi_process_update_cb },
+    { "tcp", we_wmi_tcp_update_cb },
     { 0 }
 };
 
@@ -401,6 +420,7 @@ static int in_we_init(struct flb_input_instance *in,
     ctx->coll_wmi_memory_fd = -1;
     ctx->coll_wmi_paging_file_fd = -1;
     ctx->coll_wmi_process_fd = -1;
+    ctx->coll_wmi_tcp_fd = -1;
 
     ctx->callback = flb_callback_create(in->name);
     if (!ctx->callback) {
@@ -806,6 +826,31 @@ static int in_we_init(struct flb_input_instance *in,
                         return -1;
                     }
                 }
+                else if (strncmp(entry->str, "tcp", 3) == 0) {
+                    if (ctx->wmi_tcp_scrape_interval == 0) {
+                        flb_plg_debug(ctx->ins, "enabled metrics %s", entry->str);
+                        metric_idx = 14;
+                    }
+                    else {
+                        /* Create the process collector */
+                        ret = flb_input_set_collector_time(in,
+                                                           we_timer_wmi_tcp_metrics_cb,
+                                                           ctx->wmi_tcp_scrape_interval, 0,
+                                                           config);
+                        if (ret == -1) {
+                            flb_plg_error(ctx->ins,
+                                          "could not set tcp collector for Windows Exporter Metrics plugin");
+                            return -1;
+                        }
+                        ctx->coll_wmi_tcp_fd = ret;
+                    }
+
+                    /* Initialize tcp metric collectors */
+                    ret = we_wmi_tcp_init(ctx);
+                    if (ret) {
+                        return -1;
+                    }
+                }
                 else {
                     flb_plg_warn(ctx->ins, "Unknown metrics: %s", entry->str);
                     metric_idx = -1;
@@ -891,6 +936,9 @@ static int in_we_exit(void *data, struct flb_config *config)
                 else if (strncmp(entry->str, "process", 7) == 0) {
                     we_wmi_process_exit(ctx);
                 }
+                else if (strncmp(entry->str, "tcp", 3) == 0) {
+                    we_wmi_tcp_exit(ctx);
+                }
                 else {
                     flb_plg_warn(ctx->ins, "Unknown metrics: %s", entry->str);
                 }
@@ -943,6 +991,9 @@ static int in_we_exit(void *data, struct flb_config *config)
     if (ctx->coll_wmi_process_fd != -1) {
         we_wmi_process_exit(ctx);
     }
+    if (ctx->coll_wmi_tcp_fd != -1) {
+        we_wmi_tcp_exit(ctx);
+    }
 
     flb_we_config_destroy(ctx);
 
@@ -998,6 +1049,9 @@ static void in_we_pause(void *data, struct flb_config *config)
     if (ctx->coll_wmi_process_fd != -1) {
         flb_input_collector_pause(ctx->coll_wmi_process_fd, ctx->ins);
     }
+    if (ctx->coll_wmi_tcp_fd != -1) {
+        flb_input_collector_pause(ctx->coll_wmi_tcp_fd, ctx->ins);
+    }
 }
 
 static void in_we_resume(void *data, struct flb_config *config)
@@ -1049,6 +1103,12 @@ static void in_we_resume(void *data, struct flb_config *config)
     if (ctx->coll_wmi_paging_file_fd != -1) {
         flb_input_collector_resume(ctx->coll_wmi_paging_file_fd, ctx->ins);
     }
+    if (ctx->coll_wmi_process_fd != -1) {
+        flb_input_collector_resume(ctx->coll_wmi_process_fd, ctx->ins);
+    }
+    if (ctx->coll_wmi_tcp_fd != -1) {
+        flb_input_collector_resume(ctx->coll_wmi_tcp_fd, ctx->ins);
+    }
 }
 
 /* Configuration properties map */
@@ -1143,9 +1203,15 @@ static struct flb_config_map config_map[] = {
      "scrape interval to collect process metrics from the node."
     },
 
+    {
+     FLB_CONFIG_MAP_TIME, "collector.tcp.scrape_interval", "0",
+     0, FLB_TRUE, offsetof(struct flb_we, wmi_tcp_scrape_interval),
+     "scrape interval to collect tcp metrics from the node."
+    },
+
     {
      FLB_CONFIG_MAP_CLIST, "metrics",
-     "cpu,cpu_info,os,net,logical_disk,cs,cache,thermalzone,logon,system,service",
+     "cpu,cpu_info,os,net,logical_disk,cs,cache,thermalzone,logon,system,service,tcp",
      0, FLB_TRUE, offsetof(struct flb_we, metrics),
      "Comma separated list of keys to enable metrics."
     },

plugins/in_windows_exporter_metrics/we.h

@@ -229,6 +229,22 @@ struct we_wmi_process_counters {
     int                    operational;
 };
 
+struct we_wmi_tcp_counters {
+    int                          operational;
+    struct wmi_query_spec       *v4_info;
+    struct wmi_query_spec       *v6_info;
+    
+    struct cmt_counter          *connection_failures;
+    struct cmt_gauge            *connections_active;
+    struct cmt_counter          *connections_established;
+    struct cmt_counter          *connections_passive;
+    struct cmt_counter          *connections_reset;
+    struct cmt_gauge            *segments_per_sec;
+    struct cmt_gauge            *segments_received_per_sec;
+    struct cmt_gauge            *segments_retransmitted_per_sec;
+    struct cmt_gauge            *segments_sent_per_sec;
+};
+
 struct we_os_counters {
     struct cmt_gauge *info;
     struct cmt_gauge *users;
@@ -303,6 +319,7 @@ struct flb_we {
     int wmi_memory_scrape_interval;
     int wmi_paging_file_scrape_interval;
     int wmi_process_scrape_interval;
+    int wmi_tcp_scrape_interval;
 
     int coll_cpu_fd;                                    /* collector fd (cpu)    */
     int coll_net_fd;                                    /* collector fd (net)  */
@@ -318,6 +335,7 @@ struct flb_we {
     int coll_wmi_memory_fd;                             /* collector fd (wmi_memory)    */
     int coll_wmi_paging_file_fd;                        /* collector fd (wmi_paging_file) */
     int coll_wmi_process_fd;                            /* collector fd (wmi_process) */
+    int coll_wmi_tcp_fd;                                /* collector fd (wmi_tcp) */
 
     /*
      * Metrics Contexts
@@ -338,6 +356,7 @@ struct flb_we {
     struct we_wmi_memory_counters *wmi_memory;
     struct we_wmi_paging_file_counters *wmi_paging_file;
     struct we_wmi_process_counters *wmi_process;
+    struct we_wmi_tcp_counters *wmi_tcp;
 };
 
 typedef int (*collector_cb)(struct flb_we *);