#include <vppinfra/linux/sysfs.h>
#include <vlib/vlib.h>
#include <vlib/unix/unix.h>
+#include <vpp/stats/stat_segment.h>
#define VLIB_BUFFER_DEFAULT_BUFFERS_PER_NUMA 16384
#define VLIB_BUFFER_DEFAULT_BUFFERS_PER_NUMA_UNPRIV 8192
bm->default_data_size = VLIB_BUFFER_DEFAULT_DATA_SIZE;
}
+static u32
+buffer_get_cached (vlib_buffer_pool_t * bp)
+{
+ u32 cached = 0;
+ vlib_buffer_pool_thread_t *bpt;
+
+ /* *INDENT-OFF* */
+ vec_foreach (bpt, bp->threads)
+ cached += vec_len (bpt->cached_buffers);
+ /* *INDENT-ON* */
+
+ return cached;
+}
+
+static vlib_buffer_pool_t *
+buffer_get_by_name (vlib_buffer_main_t * bm, char *name)
+{
+ vlib_buffer_pool_t *bp;
+ vec_foreach (bp, bm->buffer_pools)
+ {
+ if (!strcmp ((char *) bp->name, name))
+ return bp;
+ }
+
+ return 0;
+}
+
+static void
+buffer_gauges_update_used_fn (stat_segment_directory_entry_t * e)
+{
+ vlib_main_t *vm = vlib_get_main ();
+ vlib_buffer_pool_t *bp;
+
+ bp = buffer_get_by_name (vm->buffer_main,
+ &e->name[sizeof ("/buffer/used/") - 1]);
+ if (!bp)
+ return;
+
+ e->value = bp->n_buffers - vec_len (bp->buffers) - buffer_get_cached (bp);
+}
+
+static void
+buffer_gauges_update_available_fn (stat_segment_directory_entry_t * e)
+{
+ vlib_main_t *vm = vlib_get_main ();
+ vlib_buffer_pool_t *bp;
+
+ bp = buffer_get_by_name (vm->buffer_main,
+ &e->name[sizeof ("/buffer/available/") - 1]);
+ if (!bp)
+ return;
+
+ e->value = vec_len (bp->buffers);
+}
+
+static void
+buffer_gauges_update_cached_fn (stat_segment_directory_entry_t * e)
+{
+ vlib_main_t *vm = vlib_get_main ();
+ vlib_buffer_pool_t *bp;
+
+ bp = buffer_get_by_name (vm->buffer_main,
+ &e->name[sizeof ("/buffer/cached/") - 1]);
+ if (!bp)
+ return;
+
+ e->value = buffer_get_cached (bp);
+}
+
clib_error_t *
-vlib_buffer_main_init (struct vlib_main_t *vm)
+vlib_buffer_main_init (struct vlib_main_t * vm)
{
vlib_buffer_main_t *bm;
clib_error_t *err;
clib_bitmap_t *bmp = 0;
u32 numa_node;
+ vlib_buffer_pool_t *bp;
+ u8 *name;
vlib_buffer_main_alloc (vm);
bm->n_numa_nodes = clib_bitmap_last_set (bmp) + 1;
+ vec_foreach (bp, bm->buffer_pools)
+ {
+ name = format (0, "/buffer/cached/%s%c", bp->name, 0);
+ stat_segment_register_gauge (name, buffer_gauges_update_cached_fn);
+ vec_free (name);
+ name = format (0, "/buffer/used/%s%c", bp->name, 0);
+ stat_segment_register_gauge (name, buffer_gauges_update_used_fn);
+ vec_free (name);
+ name = format (0, "/buffer/available/%s%c", bp->name, 0);
+ stat_segment_register_gauge (name, buffer_gauges_update_available_fn);
+ vec_free (name);
+ }
+
+
done:
vec_free (bmp);
return err;
stat_segment_main_t *sm = &stat_segment_main;
stat_segment_shared_header_t *shared_header;
stat_segment_directory_entry_t *ep;
-
- f64 *scalar_data;
- u8 *name;
void *oldheap;
- u32 *lock;
- int rv;
ssize_t memory_size;
-
-
int mfd;
char *mem_name = "stat_segment_test";
void *memaddr;
/* Save the vector offset in the shared segment, for clients */
shared_header->directory_offset =
stat_segment_offset (shared_header, sm->directory_vector);
+ sm->gauges_fns = 0;
clib_mem_set_heap (oldheap);
if (sm->node_counters_enabled)
update_node_counters (sm);
+ for (i = 0; i < vec_len (sm->gauges_fns); i++)
+ {
+ if (sm->gauges_fns[i])
+ sm->gauges_fns[i] (&sm->directory_vector[i]);
+ }
+
/* Heartbeat, so clients detect we're still here */
sm->directory_vector[STAT_COUNTER_HEARTBEAT].value++;
}
return 0;
}
+clib_error_t *
+stat_segment_register_gauge (u8 * name, stat_segment_update_fn update_fn)
+{
+ stat_segment_main_t *sm = &stat_segment_main;
+ stat_segment_shared_header_t *shared_header = sm->shared_header;
+ void *oldheap;
+ stat_segment_directory_entry_t e;
+ u32 index;
+
+ ASSERT (shared_header);
+
+ oldheap = vlib_stats_push_heap ();
+ vlib_stat_segment_lock ();
+
+ memset (&e, 0, sizeof (e));
+ e.type = STAT_DIR_TYPE_SCALAR_INDEX;
+
+ memcpy (e.name, name, vec_len (name));
+ index = vec_len (sm->directory_vector);
+ vec_add1 (sm->directory_vector, e);
+
+ shared_header->directory_offset =
+ stat_segment_offset (shared_header, sm->directory_vector);
+
+ vlib_stat_segment_unlock ();
+ clib_mem_set_heap (oldheap);
+
+ vec_validate (sm->gauges_fns, index);
+ sm->gauges_fns[index] = update_fn;
+
+ return NULL;
+}
+
static clib_error_t *
statseg_config (vlib_main_t * vm, unformat_input_t * input)
{