X-Git-Url: http://git.efficios.com/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Fevent-expr-to-bytecode.c;fp=src%2Fcommon%2Fevent-expr-to-bytecode.c;h=05eccb7eaef7e29bcd5120ff01aa3dd6813022e5;hp=0000000000000000000000000000000000000000;hb=2463b7879c00298daa79744cdaae82ac061a4ed8;hpb=3a4595c2469472dee1656cde5f8882c2123efd3c

diff --git a/src/common/event-expr-to-bytecode.c b/src/common/event-expr-to-bytecode.c
new file mode 100644
index 000000000..05eccb7ea
--- /dev/null
+++ b/src/common/event-expr-to-bytecode.c
@@ -0,0 +1,201 @@
+/*
+ * Copyright 2020 EfficiOS, Inc.
+ *
+ * SPDX-License-Identifier: LGPL-2.1-only
+ *
+ */
+
+#include "event-expr-to-bytecode.h"
+
+#include <stdio.h>
+#include <lttng/event-expr.h>
+#include <common/bytecode/bytecode.h>
+
+
+static
+int event_expr_to_bytecode_recursive(const struct lttng_event_expr *expr,
+		struct lttng_bytecode_alloc **bytecode,
+		struct lttng_bytecode_alloc **bytecode_reloc)
+{
+	enum lttng_event_expr_status event_expr_status;
+	int status;
+
+	switch (lttng_event_expr_get_type(expr)) {
+	case LTTNG_EVENT_EXPR_TYPE_EVENT_PAYLOAD_FIELD:
+	{
+		const char *name;
+
+		status = bytecode_push_get_payload_root(bytecode);
+		if (status) {
+			goto end;
+		}
+
+		name = lttng_event_expr_event_payload_field_get_name(expr);
+		if (!name) {
+			status = -1;
+			goto end;
+		}
+
+		status = bytecode_push_get_symbol(bytecode, bytecode_reloc, name);
+		if (status) {
+			goto end;
+		}
+
+		break;
+	}
+
+	case LTTNG_EVENT_EXPR_TYPE_CHANNEL_CONTEXT_FIELD:
+	{
+		const char *name;
+
+		status = bytecode_push_get_context_root(bytecode);
+		if (status) {
+			goto end;
+		}
+
+		name = lttng_event_expr_channel_context_field_get_name(expr);
+		if (!name) {
+			status = -1;
+			goto end;
+		}
+
+		status = bytecode_push_get_symbol(bytecode, bytecode_reloc, name);
+		if (status) {
+			goto end;
+		}
+
+		break;
+	}
+
+	case LTTNG_EVENT_EXPR_TYPE_APP_SPECIFIC_CONTEXT_FIELD:
+	{
+		const char *provider_name, *type_name;
+		char *name = NULL;
+		int ret;
+
+		status = bytecode_push_get_app_context_root(bytecode);
+		if (status) {
+			goto end;
+		}
+
+		provider_name = lttng_event_expr_app_specific_context_field_get_provider_name(expr);
+		if (!provider_name) {
+			status = -1;
+			goto end;
+		}
+
+		type_name = lttng_event_expr_app_specific_context_field_get_type_name(expr);
+		if (!type_name) {
+			status = -1;
+			goto end;
+		}
+
+		/* Reconstitute the app context field name from its two parts.  */
+		ret = asprintf(&name, "%s:%s", provider_name, type_name);
+		if (ret < 0) {
+			status = -1;
+			goto end;
+		}
+
+		status = bytecode_push_get_symbol(bytecode, bytecode_reloc, name);
+		free(name);
+		if (status) {
+			goto end;
+		}
+
+		break;
+	}
+
+	case LTTNG_EVENT_EXPR_TYPE_ARRAY_FIELD_ELEMENT:
+	{
+		const struct lttng_event_expr *parent;
+		unsigned int index;
+
+		parent = lttng_event_expr_array_field_element_get_parent_expr(expr);
+		if (!parent) {
+			status = -1;
+			goto end;
+		}
+
+		status = event_expr_to_bytecode_recursive(parent, bytecode, bytecode_reloc);
+		if (status) {
+			goto end;
+		}
+
+		event_expr_status = lttng_event_expr_array_field_element_get_index(
+			expr, &index);
+		if (event_expr_status != LTTNG_EVENT_EXPR_STATUS_OK) {
+			status = -1;
+			goto end;
+		}
+
+		status = bytecode_push_get_index_u64(bytecode, index);
+		if (status) {
+			goto end;
+		}
+
+		break;
+	}
+
+	default:
+		abort();
+	}
+
+	status = 0;
+end:
+	return status;
+}
+
+LTTNG_HIDDEN
+int lttng_event_expr_to_bytecode(const struct lttng_event_expr *expr,
+		struct lttng_bytecode **bytecode_out)
+{
+	struct lttng_bytecode_alloc *bytecode = NULL;
+	struct lttng_bytecode_alloc *bytecode_reloc = NULL;
+	struct return_op ret_insn;
+	int status;
+
+	status = bytecode_init(&bytecode);
+	if (status) {
+		goto end;
+	}
+
+	status = bytecode_init(&bytecode_reloc);
+	if (status) {
+		goto end;
+	}
+
+	status = event_expr_to_bytecode_recursive (expr, &bytecode, &bytecode_reloc);
+	if (status) {
+		goto end;
+	}
+
+	ret_insn.op = BYTECODE_OP_RETURN;
+	bytecode_push(&bytecode, &ret_insn, 1, sizeof(ret_insn));
+
+	/* Append symbol table to bytecode. */
+	bytecode->b.reloc_table_offset = bytecode_get_len(&bytecode->b);
+	status = bytecode_push(&bytecode, bytecode_reloc->b.data,
+		1, bytecode_get_len(&bytecode_reloc->b));
+	if (status) {
+		goto end;
+	}
+
+	/* Copy the `lttng_bytecode` out of the `lttng_bytecode_alloc`.  */
+	*bytecode_out = bytecode_copy(&bytecode->b);
+	if (!*bytecode_out) {
+		status = -1;
+		goto end;
+	}
+
+end:
+	if (bytecode) {
+		free(bytecode);
+	}
+
+	if (bytecode_reloc) {
+		free(bytecode_reloc);
+	}
+
+	return status;
+}