MINOR: counters: provide a generic function to retrieve a stkctr for sc* and src.
This function aims at simplifying the prefetching of the table and entry
when using any of the session counters fetches. The principle is that the
src_* variant produces a stkctr that is used instead of the one from the
session. That way we can call the same function from all session counter
fetch functions and always have a single function to support sc[0-9]_/src_.
diff --git a/src/session.c b/src/session.c
index aa42c31..35643eb 100644
--- a/src/session.c
+++ b/src/session.c
@@ -2578,6 +2578,33 @@
/* All supported ACL keywords must be declared here. */
/************************************************************************/
+/* Returns a pointer to an stkctr depending on the fetch keyword name.
+ * It is designed to be called as sc[0-9]_* or src_* exclusively.
+ * sc[0-9]_* will return a pointer to the respective field in the
+ * session <l4>. src_* will fill a locally allocated structure with
+ * the table and entry corresponding to what is specified with src_*.
+ * NULL may be returned if the designated stkctr is not tracked.
+ */
+static struct stkctr *
+smp_fetch_sc_stkctr(struct session *l4, const struct arg *args, const char *kw)
+{
+ static struct stkctr stkctr;
+ unsigned char num = kw[2];
+
+ if (num - '0' <= 9) { /* sc[0-9]_* variant */
+ return l4->stkctr[num - '0'].entry ? &l4->stkctr[num - '0'] : NULL;
+ }
+ else { /* src_* variant, arg[0] = table */
+ struct stktable_key *key = addr_to_stktable_key(&l4->si[0].conn->addr.from);
+
+ if (!key)
+ return NULL;
+ stkctr.table = &args->data.prx->table;
+ stkctr.entry = stktable_lookup_key(stkctr.table, key);
+ return &stkctr;
+ }
+}
+
/* set return a boolean indicating if the requested session counter is
* currently being tracked or not.
* Supports being called as "sc[0-9]_tracked" only.