X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=formats%2Fctf%2Fmetadata%2Fctf-visitor-generate-io-struct.c;h=de9986113de508f36bfe66ac5950c6786da728ef;hp=af75788aa92dedfd9794ea9b2f80473330e23cca;hb=b448902b7c2d8fa734e0d5cb1e6acbb009db97e4;hpb=f6714e20aa8136e794cb409c6f268fa070e80a52

diff --git a/formats/ctf/metadata/ctf-visitor-generate-io-struct.c b/formats/ctf/metadata/ctf-visitor-generate-io-struct.c
index af75788a..de998611 100644
--- a/formats/ctf/metadata/ctf-visitor-generate-io-struct.c
+++ b/formats/ctf/metadata/ctf-visitor-generate-io-struct.c
@@ -2433,6 +2433,175 @@ void clock_free(gpointer data)
 	g_free(clock);
 }
 
+static
+int ctf_callsite_declaration_visit(FILE *fd, int depth, struct ctf_node *node,
+		struct ctf_callsite *callsite, struct ctf_trace *trace)
+{
+	int ret = 0;
+
+	switch (node->type) {
+	case NODE_CTF_EXPRESSION:
+	{
+		char *left;
+
+		left = concatenate_unary_strings(&node->u.ctf_expression.left);
+		if (!strcmp(left, "name")) {
+			char *right;
+
+			if (CTF_CALLSITE_FIELD_IS_SET(callsite, name)) {
+				fprintf(fd, "[error] %s: name already declared in callsite declaration\n", __func__);
+				ret = -EPERM;
+				goto error;
+			}
+			right = concatenate_unary_strings(&node->u.ctf_expression.right);
+			if (!right) {
+				fprintf(fd, "[error] %s: unexpected unary expression for callsite name\n", __func__);
+				ret = -EINVAL;
+				goto error;
+			}
+			callsite->name = g_quark_from_string(right);
+			g_free(right);
+			CTF_CALLSITE_SET_FIELD(callsite, name);
+		} else if (!strcmp(left, "func")) {
+			char *right;
+
+			if (CTF_CALLSITE_FIELD_IS_SET(callsite, func)) {
+				fprintf(fd, "[error] %s: func already declared in callsite declaration\n", __func__);
+				ret = -EPERM;
+				goto error;
+			}
+			right = concatenate_unary_strings(&node->u.ctf_expression.right);
+			if (!right) {
+				fprintf(fd, "[error] %s: unexpected unary expression for callsite func\n", __func__);
+				ret = -EINVAL;
+				goto error;
+			}
+			callsite->func = right;
+			CTF_CALLSITE_SET_FIELD(callsite, func);
+		} else if (!strcmp(left, "file")) {
+			char *right;
+
+			if (CTF_CALLSITE_FIELD_IS_SET(callsite, file)) {
+				fprintf(fd, "[error] %s: file already declared in callsite declaration\n", __func__);
+				ret = -EPERM;
+				goto error;
+			}
+			right = concatenate_unary_strings(&node->u.ctf_expression.right);
+			if (!right) {
+				fprintf(fd, "[error] %s: unexpected unary expression for callsite file\n", __func__);
+				ret = -EINVAL;
+				goto error;
+			}
+			callsite->file = right;
+			CTF_CALLSITE_SET_FIELD(callsite, file);
+		} else if (!strcmp(left, "line")) {
+			if (CTF_CALLSITE_FIELD_IS_SET(callsite, line)) {
+				fprintf(fd, "[error] %s: line already declared in callsite declaration\n", __func__);
+				ret = -EPERM;
+				goto error;
+			}
+			ret = get_unary_unsigned(&node->u.ctf_expression.right, &callsite->line);
+			if (ret) {
+				fprintf(fd, "[error] %s: unexpected unary expression for callsite line\n", __func__);
+				ret = -EINVAL;
+				goto error;
+			}
+			CTF_CALLSITE_SET_FIELD(callsite, line);
+		} else if (!strcmp(left, "ip")) {
+			if (CTF_CALLSITE_FIELD_IS_SET(callsite, ip)) {
+				fprintf(fd, "[error] %s: ip already declared in callsite declaration\n", __func__);
+				ret = -EPERM;
+				goto error;
+			}
+			ret = get_unary_unsigned(&node->u.ctf_expression.right, &callsite->ip);
+			if (ret) {
+				fprintf(fd, "[error] %s: unexpected unary expression for callsite ip\n", __func__);
+				ret = -EINVAL;
+				goto error;
+			}
+			CTF_CALLSITE_SET_FIELD(callsite, ip);
+		} else {
+			fprintf(fd, "[warning] %s: attribute \"%s\" is unknown in callsite declaration.\n", __func__, left);
+		}
+
+error:
+		g_free(left);
+		break;
+	}
+	default:
+		return -EPERM;
+	/* TODO: declaration specifier should be added. */
+	}
+
+	return ret;
+}
+
+static
+int ctf_callsite_visit(FILE *fd, int depth, struct ctf_node *node, struct ctf_trace *trace)
+{
+	int ret = 0;
+	struct ctf_node *iter;
+	struct ctf_callsite *callsite;
+	struct ctf_callsite_dups *cs_dups;
+
+	callsite = g_new0(struct ctf_callsite, 1);
+	bt_list_for_each_entry(iter, &node->u.callsite.declaration_list, siblings) {
+		ret = ctf_callsite_declaration_visit(fd, depth + 1, iter, callsite, trace);
+		if (ret)
+			goto error;
+	}
+	if (!CTF_CALLSITE_FIELD_IS_SET(callsite, name)) {
+		ret = -EPERM;
+		fprintf(fd, "[error] %s: missing name field in callsite declaration\n", __func__);
+		goto error;
+	}
+	if (!CTF_CALLSITE_FIELD_IS_SET(callsite, func)) {
+		ret = -EPERM;
+		fprintf(fd, "[error] %s: missing func field in callsite declaration\n", __func__);
+		goto error;
+	}
+	if (!CTF_CALLSITE_FIELD_IS_SET(callsite, file)) {
+		ret = -EPERM;
+		fprintf(fd, "[error] %s: missing file field in callsite declaration\n", __func__);
+		goto error;
+	}
+	if (!CTF_CALLSITE_FIELD_IS_SET(callsite, line)) {
+		ret = -EPERM;
+		fprintf(fd, "[error] %s: missing line field in callsite declaration\n", __func__);
+		goto error;
+	}
+
+	cs_dups = g_hash_table_lookup(trace->callsites,
+		(gpointer) (unsigned long) callsite->name);
+	if (!cs_dups) {
+		cs_dups = g_new0(struct ctf_callsite_dups, 1);
+		BT_INIT_LIST_HEAD(&cs_dups->head);
+		g_hash_table_insert(trace->callsites,
+			(gpointer) (unsigned long) callsite->name, cs_dups);
+	}
+	bt_list_add_tail(&callsite->node, &cs_dups->head);
+	return 0;
+
+error:
+	g_free(callsite->func);
+	g_free(callsite->file);
+	g_free(callsite);
+	return ret;
+}
+
+static
+void callsite_free(gpointer data)
+{
+	struct ctf_callsite_dups *cs_dups = data;
+	struct ctf_callsite *callsite, *cs_n;
+
+	bt_list_for_each_entry_safe(callsite, cs_n, &cs_dups->head, node) {
+		g_free(callsite->func);
+		g_free(callsite->file);
+		g_free(callsite);
+	}
+}
+
 static
 int ctf_env_declaration_visit(FILE *fd, int depth, struct ctf_node *node,
 		struct ctf_trace *trace)
@@ -2665,6 +2834,8 @@ int ctf_visitor_construct_metadata(FILE *fd, int depth, struct ctf_node *node,
 	trace->byte_order = byte_order;
 	trace->clocks = g_hash_table_new_full(g_direct_hash, g_direct_equal,
 				NULL, clock_free);
+	trace->callsites = g_hash_table_new_full(g_direct_hash, g_direct_equal,
+				NULL, callsite_free);
 
 retry:
 	trace->root_declaration_scope = new_declaration_scope(NULL);
@@ -2714,6 +2885,14 @@ retry:
 				goto error;
 			}
 		}
+		bt_list_for_each_entry(iter, &node->u.root.callsite, siblings) {
+			ret = ctf_callsite_visit(fd, depth + 1, iter,
+					      trace);
+			if (ret) {
+				fprintf(fd, "[error] %s: callsite declaration error\n", __func__);
+				goto error;
+			}
+		}
 		if (!trace->streams) {
 			fprintf(fd, "[error] %s: missing trace declaration\n", __func__);
 			ret = -EINVAL;
@@ -2755,6 +2934,7 @@ retry:
 
 error:
 	free_declaration_scope(trace->root_declaration_scope);
+	g_hash_table_destroy(trace->callsites);
 	g_hash_table_destroy(trace->clocks);
 	return ret;
 }