123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812 |
- /*
- * Copyright (C) 2010 iptelorg GmbH
- *
- * Permission to use, copy, modify, and distribute this software for any
- * purpose with or without fee is hereby granted, provided that the above
- * copyright notice and this permission notice appear in all copies.
- *
- * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
- * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
- * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
- * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
- * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
- * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
- */
- /**
- * @brief counters/stats
- * @file
- * @ingroup: core
- */
- /*
- * History:
- * --------
- * 2010-08-06 initial version (andrei)
- * 2010-08-24 counters can be used (inc,add) before prefork_init (andrei)
- */
- #include "counters.h"
- #include "str_hash.h"
- #include "str.h"
- #include "compiler_opt.h"
- #include "mem/mem.h"
- #include "mem/shm_mem.h"
- #define CNT_HASH_SIZE 64
- /* group hash size (rpc use) */
- #define GRP_HASH_SIZE 16
- /* initial sorted groups array size (rpc use) */
- #define GRP_SORTED_SIZE 16
- /* intial counter id 2 record array size */
- #define CNT_ID2RECORD_SIZE 64
- #define CACHELINE_PAD 128
- /* leave space for one flag */
- #define MAX_COUNTER_ID 32767
- /* size (number of entries) of the temporary array used for keeping stats
- pre-prefork init. Note: if more counters are registered then this size,
- the array will be dynamically increased (doubled each time). The value
- here is meant only to optimize startup/memory fragmentation. */
- #define PREINIT_CNTS_VALS_SIZE 128
- struct counter_record {
- str group;
- str name;
- counter_handle_t h;
- unsigned short flags;
- void* cbk_param;
- counter_cbk_f cbk;
- struct counter_record* grp_next; /* next in group */
- str doc;
- };
- struct grp_record {
- str group;
- struct counter_record* first;
- };
- /** hash table mapping a counter name to an id */
- static struct str_hash_table cnts_hash_table;
- /** array maping id 2 record */
- struct counter_record** cnt_id2record;
- static int cnt_id2record_size;
- /** hash table for groups (maps a group name to a counter list) */
- static struct str_hash_table grp_hash_table;
- /** array of groups, sorted */
- static struct grp_record** grp_sorted;
- static int grp_sorted_max_size;
- static int grp_sorted_crt_size;
- static int grp_no; /* number of groups */
- /** counters array. a[proc_no][counter_id] =>
- _cnst_vals[proc_no*cnts_no+counter_id] */
- counter_array_t* _cnts_vals = 0;
- int _cnts_row_len; /* number of elements per row */
- static int cnts_no; /* number of registered counters */
- static int cnts_max_rows; /* set to 0 if not yet fully init */
- int counters_initialized(void)
- {
- if (unlikely(_cnts_vals == 0)) {
- /* not init yet */
- return 0;
- }
- return 1;
- }
- /** init the coutner hash table(s).
- * @return 0 on success, -1 on error.
- */
- int init_counters()
- {
- if (str_hash_alloc(&cnts_hash_table, CNT_HASH_SIZE) < 0)
- goto error;
- str_hash_init(&cnts_hash_table);
- if (str_hash_alloc(&grp_hash_table, GRP_HASH_SIZE) < 0)
- goto error;
- str_hash_init(&grp_hash_table);
- cnts_no = 1; /* start at 1 (0 used only for invalid counters) */
- cnts_max_rows = 0; /* 0 initially, !=0 after full init
- (counters_prefork_init()) */
- grp_no = 0;
- cnt_id2record_size = CNT_ID2RECORD_SIZE;
- cnt_id2record = pkg_malloc(sizeof(*cnt_id2record) * cnt_id2record_size);
- if (cnt_id2record == 0)
- goto error;
- memset(cnt_id2record, 0, sizeof(*cnt_id2record) * cnt_id2record_size);
- grp_sorted_max_size = GRP_SORTED_SIZE;
- grp_sorted_crt_size = 0;
- grp_sorted = pkg_malloc(sizeof(*grp_sorted) * grp_sorted_max_size);
- if (grp_sorted == 0)
- goto error;
- memset(grp_sorted, 0, sizeof(*grp_sorted) * grp_sorted_max_size);
- return 0;
- error:
- destroy_counters();
- return -1;
- }
- void destroy_counters()
- {
- int r;
- struct str_hash_entry* e;
- struct str_hash_entry* bak;
- if (_cnts_vals) {
- if (cnts_max_rows)
- /* fully init => it is in shm */
- shm_free(_cnts_vals);
- else
- /* partially init (before prefork) => pkg */
- pkg_free(_cnts_vals);
- _cnts_vals = 0;
- }
- if (cnts_hash_table.table) {
- for (r=0; r< cnts_hash_table.size; r++) {
- clist_foreach_safe(&cnts_hash_table.table[r], e, bak, next) {
- pkg_free(e);
- }
- }
- pkg_free(cnts_hash_table.table);
- }
- if (grp_hash_table.table) {
- for (r=0; r< grp_hash_table.size; r++) {
- clist_foreach_safe(&grp_hash_table.table[r], e, bak, next) {
- pkg_free(e);
- }
- }
- pkg_free(grp_hash_table.table);
- }
- if (cnt_id2record)
- pkg_free(cnt_id2record);
- if (grp_sorted)
- pkg_free(grp_sorted);
- cnts_hash_table.table = 0;
- cnts_hash_table.size = 0;
- cnt_id2record = 0;
- grp_sorted = 0;
- grp_hash_table.table = 0;
- grp_hash_table.size = 0;
- grp_sorted_crt_size = 0;
- grp_sorted_max_size = 0;
- cnts_no = 0;
- _cnts_row_len = 0;
- cnts_max_rows = 0;
- grp_no = 0;
- }
- /** complete counter intialization, when the number of processes is known.
- * shm must be available.
- * @return 0 on success, < 0 on error
- */
- int counters_prefork_init(int max_process_no)
- {
- counter_array_t* old;
- int size, row_size;
- counter_handle_t h;
- /* round cnts_no so that cnts_no * sizeof(counter) it's a CACHELINE_PAD
- multiple */
- /* round-up row_size to a CACHELINE_PAD multiple if needed */
- row_size = ((sizeof(*_cnts_vals) * cnts_no - 1) / CACHELINE_PAD + 1) *
- CACHELINE_PAD;
- /* round-up the resulted row_siue to a sizeof(*_cnts_vals) multiple */
- row_size = ((row_size -1) / sizeof(*_cnts_vals) + 1) *
- sizeof(*_cnts_vals);
- /* get updated cnts_no (row length) */
- _cnts_row_len = row_size / sizeof(*_cnts_vals);
- size = max_process_no * row_size;
- /* replace the temporary pre-fork pkg array (with only 1 row) with
- the final shm version (with max_process_no rows) */
- old = _cnts_vals;
- _cnts_vals = shm_malloc(size);
- if (_cnts_vals == 0)
- return -1;
- memset(_cnts_vals, 0, size);
- cnts_max_rows = max_process_no;
- /* copy prefork values into the newly shm array */
- if (old) {
- for (h.id = 0; h.id < cnts_no; h.id++)
- counter_pprocess_val(process_no, h) = old[h.id].v;
- pkg_free(old);
- }
- return 0;
- }
- /** adds new group to the group hash table (no checks, internal version).
- * @return pointer to new group record on success, 0 on error.
- */
- static struct grp_record* grp_hash_add(str* group)
- {
- struct str_hash_entry* g;
- struct grp_record* grp_rec;
- struct grp_record** r;
- /* grp_rec copied at &g->u.data */
- g = pkg_malloc(sizeof(struct str_hash_entry) - sizeof(g->u.data) +
- sizeof(*grp_rec) + group->len + 1);
- if (g == 0)
- goto error;
- grp_rec = (struct grp_record*)&g->u.data[0];
- grp_rec->group.s = (char*)(grp_rec + 1);
- grp_rec->group.len = group->len;
- grp_rec->first = 0;
- memcpy(grp_rec->group.s, group->s, group->len + 1);
- g->key = grp_rec->group;
- g->flags = 0;
- /* insert group into the sorted group array */
- if (grp_sorted_max_size <= grp_sorted_crt_size) {
- /* must increase the array */
- r = pkg_realloc(grp_sorted, 2 * grp_sorted_max_size *
- sizeof(*grp_sorted));
- if (r == 0)
- goto error;
- grp_sorted= r;
- grp_sorted_max_size *= 2;
- memset(&grp_sorted[grp_sorted_crt_size], 0,
- (grp_sorted_max_size - grp_sorted_crt_size) *
- sizeof(*grp_sorted));
- }
- for (r = grp_sorted; r < (grp_sorted + grp_sorted_crt_size); r++)
- if (strcmp(grp_rec->group.s, (*r)->group.s) < 0)
- break;
- if (r != (grp_sorted + grp_sorted_crt_size))
- memmove(r+1, r, (int)(long)((char*)(grp_sorted + grp_sorted_crt_size) -
- (char*)r));
- grp_sorted_crt_size++;
- *r = grp_rec;
- /* insert into the hash only on success */
- str_hash_add(&grp_hash_table, g);
- return grp_rec;
- error:
- if (g)
- pkg_free(g);
- return 0;
- }
- /** lookup a group into the group hash (internal version).
- * @return pointer to grp_record on success, 0 on failure (not found).
- */
- static struct grp_record* grp_hash_lookup(str* group)
- {
- struct str_hash_entry* e;
- e = str_hash_get(&grp_hash_table, group->s, group->len);
- return e?(struct grp_record*)&e->u.data[0]:0;
- }
- /** lookup a group and if not found create a new group record.
- * @return pointer to grp_record on succes, 0 on failure ( not found and
- * failed to create new group record).
- */
- static struct grp_record* grp_hash_get_create(str* group)
- {
- struct grp_record* ret;
- ret = grp_hash_lookup(group);
- if (ret)
- return ret;
- return grp_hash_add(group);
- }
- /** adds new counter to the hash table (no checks, internal version).
- * @return pointer to new record on success, 0 on error.
- */
- static struct counter_record* cnt_hash_add(
- str* group, str* name,
- int flags, counter_cbk_f cbk,
- void* param, const char* doc)
- {
- struct str_hash_entry* e;
- struct counter_record* cnt_rec;
- struct grp_record* grp_rec;
- struct counter_record** p;
- counter_array_t* v;
- int doc_len;
- int n;
-
- e = 0;
- if (cnts_no >= MAX_COUNTER_ID)
- /* too many counters */
- goto error;
- grp_rec = grp_hash_get_create(group);
- if (grp_rec == 0)
- /* non existing group an no new one could be created */
- goto error;
- doc_len = doc?strlen(doc):0;
- /* cnt_rec copied at &e->u.data[0] */
- e = pkg_malloc(sizeof(struct str_hash_entry) - sizeof(e->u.data) +
- sizeof(*cnt_rec) + name->len + 1 + group->len + 1 +
- doc_len + 1);
- if (e == 0)
- goto error;
- cnt_rec = (struct counter_record*)&e->u.data[0];
- cnt_rec->group.s = (char*)(cnt_rec + 1);
- cnt_rec->group.len = group->len;
- cnt_rec->name.s = cnt_rec->group.s + group->len + 1;
- cnt_rec->name.len = name->len;
- cnt_rec->doc.s = cnt_rec->name.s + name->len +1;
- cnt_rec->doc.len = doc_len;
- cnt_rec->h.id = cnts_no++;
- cnt_rec->flags = flags;
- cnt_rec->cbk_param = param;
- cnt_rec->cbk = cbk;
- cnt_rec->grp_next = 0;
- memcpy(cnt_rec->group.s, group->s, group->len + 1);
- memcpy(cnt_rec->name.s, name->s, name->len + 1);
- if (doc)
- memcpy(cnt_rec->doc.s, doc, doc_len + 1);
- else
- cnt_rec->doc.s[0] = 0;
- e->key = cnt_rec->name;
- e->flags = 0;
- /* check to see if it fits in the prefork tmp. vals array.
- This array contains only one "row", is allocated in pkg and
- is used only until counters_prefork_init() (after that the
- array is replaced with a shm version with all the needed rows).
- */
- if (cnt_rec->h.id >= _cnts_row_len || _cnts_vals == 0) {
- /* array to small or not yet allocated => reallocate/allocate it
- (min size PREINIT_CNTS_VALS_SIZE, max MAX_COUNTER_ID)
- */
- n = (cnt_rec->h.id < PREINIT_CNTS_VALS_SIZE) ?
- PREINIT_CNTS_VALS_SIZE :
- ((2 * (cnt_rec->h.id + (cnt_rec->h.id == 0)) < MAX_COUNTER_ID)?
- (2 * (cnt_rec->h.id + (cnt_rec->h.id == 0))) :
- MAX_COUNTER_ID + 1);
- v = pkg_realloc(_cnts_vals, n * sizeof(*_cnts_vals));
- if (v == 0)
- /* realloc/malloc error */
- goto error;
- _cnts_vals = v;
- /* zero newly allocated memory */
- memset(&_cnts_vals[_cnts_row_len], 0,
- (n - _cnts_row_len) * sizeof(*_cnts_vals));
- _cnts_row_len = n; /* record new length */
- }
- /* add a pointer to it in the records array */
- if (cnt_id2record_size <= cnt_rec->h.id) {
- /* must increase the array */
- p = pkg_realloc(cnt_id2record,
- 2 * cnt_id2record_size * sizeof(*cnt_id2record));
- if (p == 0)
- goto error;
- cnt_id2record = p;
- cnt_id2record_size *= 2;
- memset(&cnt_id2record[cnt_rec->h.id], 0,
- (cnt_id2record_size - cnt_rec->h.id) * sizeof(*cnt_id2record));
- }
- cnt_id2record[cnt_rec->h.id] = cnt_rec;
- /* add into the hash */
- str_hash_add(&cnts_hash_table, e);
- /* insert it sorted in the per group list */
- for (p = &grp_rec->first; *p; p = &((*p)->grp_next))
- if (strcmp(cnt_rec->name.s, (*p)->name.s) < 0)
- break;
- cnt_rec->grp_next = *p;
- *p = cnt_rec;
- return cnt_rec;
- error:
- if (e)
- pkg_free(e);
- return 0;
- }
- /** lookup a (group, name) pair into the cnts hash (internal version).
- * @param group - counter group name. If "" the first matching counter with
- * the given name will be returned (k compat).
- * @param name
- * @return pointer to counter_record on success, 0 on failure (not found).
- */
- static struct counter_record* cnt_hash_lookup(str* group, str* name)
- {
- struct str_hash_entry* e;
- struct str_hash_entry* first;
- struct counter_record* cnt_rec;
- e = str_hash_get(&cnts_hash_table, name->s, name->len);
- /* fast path */
- if (likely(e)) {
- cnt_rec = (struct counter_record*)&e->u.data[0];
- if (likely( group->len == 0 ||
- (cnt_rec->group.len == group->len &&
- memcmp(cnt_rec->group.s, group->s, group->len) == 0)))
- return cnt_rec;
- } else
- return 0;
- /* search between records with same name, but different groups */
- first = e;
- do {
- cnt_rec = (struct counter_record*)&e->u.data[0];
- if (cnt_rec->group.len == group->len &&
- cnt_rec->name.len == name->len &&
- memcmp(cnt_rec->group.s, group->s, group->len) == 0 &&
- memcmp(cnt_rec->name.s, name->s, name->len) == 0)
- /* found */
- return cnt_rec;
- e = e->next;
- } while(e != first);
- return 0;
- }
- /** lookup a counter and if not found create a new counter record.
- * @return pointer to counter_record on succes, 0 on failure ( not found and
- * failed to create new group record).
- */
- static struct counter_record* cnt_hash_get_create(
- str* group, str* name,
- int flags,
- counter_cbk_f cbk,
- void* param, const char* doc)
- {
- struct counter_record* ret;
- ret = cnt_hash_lookup(group, name);
- if (ret)
- return ret;
- return cnt_hash_add(group, name, flags, cbk, param, doc);
- }
- /** register a new counter.
- * Can be called only before forking (e.g. from mod_init() or
- * init_child(PROC_INIT)).
- * @param handle - result parameter, it will be filled with the counter
- * handle on success (can be null if not needed).
- * @param group - group name
- * @param name - counter name (group.name must be unique).
- * @param flags - counter flags: one of CNT_F_*.
- * @param cbk - read callback function (if set it will be called each time
- * someone will call counter_get()).
- * @param cbk_param - callback param.
- * @param doc - description/documentation string.
- * @param reg_flags - register flags: 1 - don't fail if counter already
- * registered (act like counter_lookup(handle, group, name).
- * @return 0 on succes, < 0 on error (-1 not init or malloc error, -2 already
- * registered (and register_flags & 1 == 0).
- */
- int counter_register( counter_handle_t* handle, const char* group,
- const char* name, int flags,
- counter_cbk_f cbk, void* cbk_param,
- const char* doc,
- int reg_flags)
- {
- str grp;
- str n;
- struct counter_record* cnt_rec;
- if (unlikely(cnts_max_rows)) {
- /* too late */
- BUG("late attempt to register counter: %s.%s\n", group, name);
- goto error;
- }
- n.s = (char*)name;
- n.len = strlen(name);
- if (unlikely(group == 0 || *group == 0)) {
- BUG("attempt to register counter %s without a group\n", name);
- goto error;
- }
- grp.s = (char*)group;
- grp.len = strlen(group);
- cnt_rec = cnt_hash_lookup(&grp, &n);
- if (cnt_rec) {
- if (reg_flags & 1)
- goto found;
- else {
- if (handle) handle->id = 0;
- return -2;
- }
- } else
- cnt_rec = cnt_hash_get_create(&grp, &n, flags, cbk, cbk_param, doc);
- if (unlikely(cnt_rec == 0))
- goto error;
- found:
- if (handle) *handle = cnt_rec->h;
- return 0;
- error:
- if (handle) handle->id = 0;
- return -1;
- }
- /** fill in the handle of an existing counter (str parameters).
- * @param handle - filled with the corresp. handle on success.
- * @param group - counter group name. If "" the first matching
- * counter with the given name will be returned
- * (k compat).
- * @param name - counter name.
- * @return 0 on success, < 0 on error
- */
- int counter_lookup_str(counter_handle_t* handle, str* group, str* name)
- {
- struct counter_record* cnt_rec;
- cnt_rec = cnt_hash_lookup(group, name);
- if (likely(cnt_rec)) {
- *handle = cnt_rec->h;
- return 0;
- }
- handle->id = 0;
- return -1;
- }
- /** fill in the handle of an existing counter (asciiz parameters).
- * @param handle - filled with the corresp. handle on success.
- * @param group - counter group name. If 0 or "" the first matching
- * counter with the given name will be returned
- * (k compat).
- * @param name - counter name.
- * @return 0 on success, < 0 on error
- */
- int counter_lookup(counter_handle_t* handle,
- const char* group, const char* name)
- {
- str grp;
- str n;
- n.s = (char*)name;
- n.len = strlen(name);
- grp.s = (char*)group;
- grp.len = group?strlen(group):0;
- return counter_lookup_str(handle, &grp, &n);
- }
- /** register all the counters declared in a null-terminated array.
- * @param group - counters group.
- * @param defs - null terminated array containing counters definitions.
- * @return 0 on success, < 0 on error ( - (counter_number+1))
- */
- int counter_register_array(const char* group, counter_def_t* defs)
- {
- int r;
-
- for (r=0; defs[r].name; r++)
- if (counter_register( defs[r].handle,
- group, defs[r].name, defs[r].flags,
- defs[r].get_cbk, defs[r].get_cbk_param,
- defs[r].descr, 0) <0)
- return -(r+1); /* return - (idx of bad counter + 1) */
- return 0;
- }
- /** get the value of the counter, bypassing callbacks.
- * @param handle - counter handle obtained using counter_lookup() or
- * counter_register().
- * @return counter value.
- */
- counter_val_t counter_get_raw_val(counter_handle_t handle)
- {
- int r;
- counter_val_t ret;
- if (unlikely(_cnts_vals == 0)) {
- /* not init yet */
- BUG("counters not fully initialized yet\n");
- return 0;
- }
- if (unlikely(handle.id >= cnts_no || (short)handle.id < 0)) {
- BUG("invalid counter id %d (max %d)\n", handle.id, cnts_no - 1);
- return 0;
- }
- ret = 0;
- for (r = 0; r < cnts_max_rows; r++)
- ret += counter_pprocess_val(r, handle);
- return ret;
- }
- /** get the value of the counter, using the callbacks (if defined).
- * @param handle - counter handle obtained using counter_lookup() or
- * counter_register().
- * @return counter value. */
- counter_val_t counter_get_val(counter_handle_t handle)
- {
- struct counter_record* cnt_rec;
- if (unlikely(_cnts_vals == 0 || cnt_id2record == 0)) {
- /* not init yet */
- BUG("counters not fully initialized yet\n");
- return 0;
- }
- cnt_rec = cnt_id2record[handle.id];
- if (unlikely(cnt_rec->cbk))
- return cnt_rec->cbk(handle, cnt_rec->cbk_param);
- return counter_get_raw_val(handle);
- }
- /** reset the counter.
- * Reset a counter, unless it has the CNT_F_NO_RESET flag set.
- * @param handle - counter handle obtained using counter_lookup() or
- * counter_register().
- * Note: it's racy.
- */
- void counter_reset(counter_handle_t handle)
- {
- int r;
- if (unlikely(_cnts_vals == 0 || cnt_id2record == 0)) {
- /* not init yet */
- BUG("counters not fully initialized yet\n");
- return;
- }
- if (unlikely(handle.id >= cnts_no)) {
- BUG("invalid counter id %d (max %d)\n", handle.id, cnts_no - 1);
- return;
- }
- if (unlikely(cnt_id2record[handle.id]->flags & CNT_F_NO_RESET))
- return;
- for (r=0; r < cnts_max_rows; r++)
- counter_pprocess_val(r, handle) = 0;
- return;
- }
- /** return the name for counter handle.
- * @param handle - counter handle obtained using counter_lookup() or
- * counter_register().
- * @return asciiz pointer on success, 0 on error.
- */
- char* counter_get_name(counter_handle_t handle)
- {
- if (unlikely(_cnts_vals == 0 || cnt_id2record == 0)) {
- /* not init yet */
- BUG("counters not fully initialized yet\n");
- goto error;
- }
- if (unlikely(handle.id >= cnts_no)) {
- BUG("invalid counter id %d (max %d)\n", handle.id, cnts_no - 1);
- goto error;
- }
- return cnt_id2record[handle.id]->name.s;
- error:
- return 0;
- }
- /** return the group name for counter handle.
- * @param handle - counter handle obtained using counter_lookup() or
- * counter_register().
- * @return asciiz pointer on success, 0 on error.
- */
- char* counter_get_group(counter_handle_t handle)
- {
- if (unlikely(_cnts_vals == 0 || cnt_id2record == 0)) {
- /* not init yet */
- BUG("counters not fully initialized yet\n");
- goto error;
- }
- if (unlikely(handle.id >= cnts_no)) {
- BUG("invalid counter id %d (max %d)\n", handle.id, cnts_no - 1);
- goto error;
- }
- return cnt_id2record[handle.id]->group.s;
- error:
- return 0;
- }
- /** return the description (doc) string for a given counter.
- * @param handle - counter handle obtained using counter_lookup() or
- * counter_register().
- * @return asciiz pointer on success, 0 on error.
- */
- char* counter_get_doc(counter_handle_t handle)
- {
- if (unlikely(_cnts_vals == 0 || cnt_id2record == 0)) {
- /* not init yet */
- BUG("counters not fully initialized yet\n");
- goto error;
- }
- if (unlikely(handle.id >= cnts_no)) {
- BUG("invalid counter id %d (max %d)\n", handle.id, cnts_no - 1);
- goto error;
- }
- return cnt_id2record[handle.id]->doc.s;
- error:
- return 0;
- }
- /** iterate on all the counter group names.
- * @param cbk - pointer to a callback function that will be called for each
- * group name.
- * @param p - parameter that will be passed to the callback function
- * (along the group name).
- */
- void counter_iterate_grp_names(void (*cbk)(void* p, str* grp_name), void* p)
- {
- int r;
- for (r=0; r < grp_sorted_crt_size; r++)
- cbk(p, &grp_sorted[r]->group);
- }
- /** iterate on all the variable names in a specified group.
- * @param group - group name.
- * @param cbk - pointer to a callback function that will be called for each
- * variable name.
- * @param p - parameter that will be passed to the callback function
- * (along the variable name).
- */
- void counter_iterate_grp_var_names( const char* group,
- void (*cbk)(void* p, str* var_name),
- void* p)
- {
- struct counter_record* r;
- struct grp_record* g;
- str grp;
-
- grp.s = (char*)group;
- grp.len = strlen(group);
- g = grp_hash_lookup(&grp);
- if (g)
- for (r = g->first; r; r = r->grp_next)
- cbk(p, &r->name);
- }
- /** iterate on all the variable names and handles in a specified group.
- * @param group - group name.
- * @param cbk - pointer to a callback function that will be called for each
- * [variable name, variable handle] pair.
- * @param p - parameter that will be passed to the callback function
- * (along the group name, variable name and variable handle).
- */
- void counter_iterate_grp_vars(const char* group,
- void (*cbk)(void* p, str* g, str* n,
- counter_handle_t h),
- void *p)
- {
- struct counter_record* r;
- struct grp_record* g;
- str grp;
-
- grp.s = (char*)group;
- grp.len = strlen(group);
- g = grp_hash_lookup(&grp);
- if (g)
- for (r = g->first; r; r = r->grp_next)
- cbk(p, &r->group, &r->name, r->h);
- }
- /* vi: set ts=4 sw=4 tw=79:ai:cindent: */
|