stats: refactor

Type: refactor
Change-Id: Ifd533a095d979dc55bfbe5fac7e0b7510a4d900c
Signed-off-by: Damjan Marion <damarion@cisco.com>

1 files changed, 95 insertions, 0 deletions

diff --git a/src/vnet/interface/stats.c b/src/vnet/interface/stats.c
new file mode 100644
index 00000000000..55e49eb1bc2
--- /dev/null
+++ b/src/vnet/interface/stats.c
@@ -0,0 +1,95 @@
+/* SPDX-License-Identifier: Apache-2.0
+ * Copyright(c) 2022 Cisco Systems, Inc.
+ */
+
+#include <vlib/vlib.h>
+#include <vlib/unix/unix.h>
+#include <vlib/stats/stats.h>
+#include <vnet/vnet.h>
+#include <vnet/devices/devices.h> /* vnet_get_aggregate_rx_packets */
+#include <vnet/interface.h>
+
+static u32 if_names_stats_entry_index = ~0;
+static u32 **dir_entry_indices = 0;
+
+static struct
+{
+  char *prefix, *name;
+  u32 index;
+} if_counters[] = {
+#define _(e, n, p) { .prefix = #p, .name = #n },
+  foreach_simple_interface_counter_name foreach_combined_interface_counter_name
+#undef _
+};
+
+static clib_error_t *
+statseg_sw_interface_add_del (vnet_main_t *vnm, u32 sw_if_index, u32 is_add)
+{
+  vlib_stats_segment_t *sm = vlib_stats_get_segment ();
+  vlib_stats_entry_t *e;
+  void *oldheap;
+
+  if (if_names_stats_entry_index == ~0)
+    {
+      if_names_stats_entry_index = vlib_stats_add_string_vector ("/if/names");
+
+      for (int i = 0; i < ARRAY_LEN (if_counters); i++)
+	if_counters[i].index = vlib_stats_find_entry_index (
+	  "/%s/%s", if_counters[i].prefix, if_counters[i].name);
+    }
+
+  e = sm->directory_vector + if_names_stats_entry_index;
+
+  vec_validate (dir_entry_indices, sw_if_index);
+
+  vlib_stats_segment_lock ();
+
+  if (is_add)
+    {
+      vnet_sw_interface_t *si, *si_sup;
+      vnet_hw_interface_t *hi_sup;
+      u8 *s;
+
+      si = vnet_get_sw_interface (vnm, sw_if_index);
+      si_sup = vnet_get_sup_sw_interface (vnm, si->sw_if_index);
+      ASSERT (si_sup->type == VNET_SW_INTERFACE_TYPE_HARDWARE);
+      hi_sup = vnet_get_hw_interface (vnm, si_sup->hw_if_index);
+
+      oldheap = clib_mem_set_heap (sm->heap);
+      s = format (0, "%v", hi_sup->name);
+      if (si->type != VNET_SW_INTERFACE_TYPE_HARDWARE)
+	s = format (s, ".%d", si->sub.id);
+      s = format (s, "%c", 0);
+
+      vec_validate (e->string_vector, sw_if_index);
+
+      ASSERT (e->string_vector[sw_if_index] == 0);
+      e->string_vector[sw_if_index] = s;
+      clib_mem_set_heap (oldheap);
+
+      s = format (0, "/interfaces/%U", format_vlib_stats_symlink, s);
+      for (u32 index, i = 0; i < ARRAY_LEN (if_counters); i++)
+	{
+	  index = vlib_stats_add_symlink (if_counters[i].index, sw_if_index,
+					  "%v/%s", s, if_counters[i].name);
+	  ASSERT (index != ~0);
+	  vec_add1 (dir_entry_indices[sw_if_index], index);
+	}
+      vec_free (s);
+    }
+  else
+    {
+      oldheap = clib_mem_set_heap (sm->heap);
+      vec_free (e->string_vector[sw_if_index]);
+      clib_mem_set_heap (oldheap);
+      for (u32 i = 0; i < vec_len (dir_entry_indices[sw_if_index]); i++)
+	vlib_stats_remove_entry (dir_entry_indices[sw_if_index][i]);
+      vec_free (dir_entry_indices[sw_if_index]);
+    }
+
+  vlib_stats_segment_unlock ();
+
+  return 0;
+}
+
+VNET_SW_INTERFACE_ADD_DEL_FUNCTION (statseg_sw_interface_add_del);