Version 2.2.3

[lttng-ust.git] / liblttng-ust / lttng-context.c

diff --git a/liblttng-ust/lttng-context.c b/liblttng-ust/lttng-context.c
index 16894ec1032b722407ce578d43f47051a62bbc60..ecee23445787bce42397e96c200537b1385017ef 100644 (file)
--- a/liblttng-ust/lttng-context.c
+++ b/liblttng-ust/lttng-context.c
@@ -23,10 +23,21 @@
 
 #include <lttng/ust-events.h>
 #include <lttng/ust-tracer.h>
+#include <usterr-signal-safe.h>
 #include <helper.h>
 #include <string.h>
 #include <assert.h>
 
+/*
+ * The filter implementation requires that two consecutive "get" for the
+ * same context performed by the same thread return the same result.
+ */
+
+/*
+ * Static array of contexts, for $ctx filters.
+ */
+struct lttng_ctx *lttng_static_ctx;
+
 int lttng_find_context(struct lttng_ctx *ctx, const char *name)
 {
        unsigned int i;
@@ -41,6 +52,22 @@ int lttng_find_context(struct lttng_ctx *ctx, const char *name)
        return 0;
 }
 
+int lttng_get_context_index(struct lttng_ctx *ctx, const char *name)
+{
+       unsigned int i;
+
+       if (!ctx)
+               return -1;
+       for (i = 0; i < ctx->nr_fields; i++) {
+               /* Skip allocated (but non-initialized) contexts */
+               if (!ctx->fields[i].event_field.name)
+                       continue;
+               if (!strcmp(ctx->fields[i].event_field.name, name))
+                       return i;
+       }
+       return -1;
+}
+
 /*
  * Note: as we append context information, the pointer location may change.
  */
@@ -99,3 +126,31 @@ void lttng_destroy_context(struct lttng_ctx *ctx)
        free(ctx->fields);
        free(ctx);
 }
+
+void lttng_context_init(void)
+{
+       int ret;
+
+       ret = lttng_add_pthread_id_to_ctx(&lttng_static_ctx);
+       if (ret) {
+               WARN("Cannot add context lttng_add_pthread_id_to_ctx");
+       }
+       ret = lttng_add_vtid_to_ctx(&lttng_static_ctx);
+       if (ret) {
+               WARN("Cannot add context lttng_add_vtid_to_ctx");
+       }
+       ret = lttng_add_vpid_to_ctx(&lttng_static_ctx);
+       if (ret) {
+               WARN("Cannot add context lttng_add_vpid_to_ctx");
+       }
+       ret = lttng_add_procname_to_ctx(&lttng_static_ctx);
+       if (ret) {
+               WARN("Cannot add context lttng_add_procname_to_ctx");
+       }
+}
+
+void lttng_context_exit(void)
+{
+       lttng_destroy_context(lttng_static_ctx);
+       lttng_static_ctx = NULL;
+}