X-Git-Url: http://drtracing.org/?a=blobdiff_plain;f=plugins%2Fctf%2Ffs%2Ffs.c;h=b406ee96deee4e494626c9ad4542a4c9383491b5;hb=33f93973c6b1fb17a8a1c8556d48300b4901ff4c;hp=b7b049569981c27379c80b3e6d37ebd23482215f;hpb=c1081aa6b31fb4ff1fa9f0d46f06f1df89bc8dc4;p=babeltrace.git

diff --git a/plugins/ctf/fs/fs.c b/plugins/ctf/fs/fs.c
index b7b04956..b406ee96 100644
--- a/plugins/ctf/fs/fs.c
+++ b/plugins/ctf/fs/fs.c
@@ -45,6 +45,7 @@
 #define PRINT_ERR_STREAM	ctf_fs->error_fp
 #define PRINT_PREFIX		"ctf-fs"
 #include "print.h"
+#define METADATA_TEXT_SIG	"/* CTF 1.8"
 
 BT_HIDDEN
 bool ctf_fs_debug;
@@ -735,3 +736,129 @@ error:
         ctf_fs_destroy_data(ctf_fs);
 	return ret;
 }
+
+BT_HIDDEN
+struct bt_value *ctf_fs_query_info(struct bt_component_class *comp_class,
+		const char *action, struct bt_value *params)
+{
+	struct bt_value *results = NULL;
+	struct bt_value *path_value = NULL;
+	char *metadata_text = NULL;
+	FILE *metadata_fp = NULL;
+	GString *g_metadata_text = NULL;
+
+	if (strcmp(action, "get-metadata-info") == 0) {
+		int ret;
+		int bo;
+		const char *path;
+		bool is_packetized;
+
+		results = bt_value_map_create();
+		if (!results) {
+			goto error;
+		}
+
+		if (!bt_value_is_map(params)) {
+			fprintf(stderr,
+				"Query info parameters is not a map value object\n");
+			goto error;
+		}
+
+		path_value = bt_value_map_get(params, "path");
+		ret = bt_value_string_get(path_value, &path);
+		if (ret) {
+			fprintf(stderr,
+				"Cannot get `path` string parameter\n");
+			goto error;
+		}
+
+		assert(path);
+		metadata_fp = ctf_fs_metadata_open_file(path);
+		if (!metadata_fp) {
+			fprintf(stderr,
+				"Cannot open trace at path `%s`\n", path);
+			goto error;
+		}
+
+		is_packetized = ctf_metadata_is_packetized(metadata_fp, &bo);
+
+		if (is_packetized) {
+			ret = ctf_metadata_packetized_file_to_buf(NULL,
+				metadata_fp, (uint8_t **) &metadata_text, bo);
+			if (ret) {
+				fprintf(stderr,
+					"Cannot decode packetized metadata file\n");
+				goto error;
+			}
+		} else {
+			long filesize;
+
+			fseek(metadata_fp, 0, SEEK_END);
+			filesize = ftell(metadata_fp);
+			rewind(metadata_fp);
+			metadata_text = malloc(filesize + 1);
+			if (!metadata_text) {
+				fprintf(stderr,
+					"Cannot allocate buffer for metadata text\n");
+				goto error;
+			}
+
+			if (fread(metadata_text, filesize, 1, metadata_fp) !=
+					1) {
+				fprintf(stderr,
+					"Cannot read metadata file\n");
+				goto error;
+			}
+
+			metadata_text[filesize] = '\0';
+		}
+
+		g_metadata_text = g_string_new(NULL);
+		if (!g_metadata_text) {
+			goto error;
+		}
+
+		if (strncmp(metadata_text, METADATA_TEXT_SIG,
+				sizeof(METADATA_TEXT_SIG) - 1) != 0) {
+			g_string_assign(g_metadata_text, METADATA_TEXT_SIG);
+			g_string_append(g_metadata_text, " */\n\n");
+		}
+
+		g_string_append(g_metadata_text, metadata_text);
+
+		ret = bt_value_map_insert_string(results, "text",
+			g_metadata_text->str);
+		if (ret) {
+			fprintf(stderr, "Cannot insert metadata text into results\n");
+			goto error;
+		}
+
+		ret = bt_value_map_insert_bool(results, "is-packetized",
+			is_packetized);
+		if (ret) {
+			fprintf(stderr, "Cannot insert is packetized into results\n");
+			goto error;
+		}
+	} else {
+		fprintf(stderr, "Unknown query info action `%s`\n", action);
+		goto error;
+	}
+
+	goto end;
+
+error:
+	BT_PUT(results);
+
+end:
+	bt_put(path_value);
+	free(metadata_text);
+
+	if (g_metadata_text) {
+		g_string_free(g_metadata_text, TRUE);
+	}
+
+	if (metadata_fp) {
+		fclose(metadata_fp);
+	}
+	return results;
+}