X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=formats%2Fctf%2Fctf.c;h=9b2f0fbf65dc195b426433de7596f6c18850aba2;hp=580133cc86c873fef4225e0bf3fbd65ff2019bb8;hb=ff00cad2a14ea6dad073761b3dbb4aaa259723d0;hpb=c054553dac076f91196b372fa19efaf2adc4e4f9

diff --git a/formats/ctf/ctf.c b/formats/ctf/ctf.c
index 580133cc..9b2f0fbf 100644
--- a/formats/ctf/ctf.c
+++ b/formats/ctf/ctf.c
@@ -18,8 +18,30 @@
 
 #include <babeltrace/format.h>
 #include <babeltrace/ctf/types.h>
+#include <babeltrace/ctf/metadata.h>
+#include <babeltrace/babeltrace.h>
+#include <errno.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <dirent.h>
+#include <glib.h>
+#include <unistd.h>
+#include <stdlib.h>
 
-void __attribute__((constructor)) ctf_init(void);
+
+#include "metadata/ctf-scanner.h"
+#include "metadata/ctf-parser.h"
+#include "metadata/ctf-ast.h"
+
+extern int yydebug;
+
+struct trace_descriptor {
+	struct ctf_trace ctf_trace;
+};
+
+struct trace_descriptor *ctf_open_trace(const char *path, int flags);
+void ctf_close_trace(struct trace_descriptor *descriptor);
 
 static struct format ctf_format = {
 	.uint_read = ctf_uint_read,
@@ -43,9 +65,210 @@ static struct format ctf_format = {
 	.array_end = ctf_array_end,
 	.sequence_begin = ctf_sequence_begin,
 	.sequence_end = ctf_sequence_end,
+	.open_trace = ctf_open_trace,
+	.close_trace = ctf_close_trace,
 };
 
-void ctf_init(void)
+/*
+ * TODO: for now, we treat the metadata file as a simple text file
+ * (without any header nor packets nor padding).
+ */
+static
+int ctf_open_trace_metadata_read(struct trace_descriptor *td)
+{
+	struct ctf_scanner *scanner;
+	FILE *fp;
+	int ret = 0;
+
+	td->ctf_trace.metadata.fd = openat(td->ctf_trace.dirfd,
+			"metadata", O_RDONLY);
+	if (td->ctf_trace.metadata.fd < 0) {
+		fprintf(stdout, "Unable to open metadata.\n");
+		return td->ctf_trace.metadata.fd;
+	}
+
+	if (babeltrace_debug)
+		yydebug = 1;
+
+	fp = fdopen(td->ctf_trace.metadata.fd, "r");
+	if (!fp) {
+		fprintf(stdout, "Unable to open metadata stream.\n");
+		ret = -errno;
+		goto end_stream;
+	}
+
+	scanner = ctf_scanner_alloc(fp);
+	if (!scanner) {
+		fprintf(stdout, "Error allocating scanner\n");
+		ret = -ENOMEM;
+		goto end_scanner_alloc;
+	}
+	ret = ctf_scanner_append_ast(scanner);
+	if (ret) {
+		fprintf(stdout, "Error creating AST\n");
+		goto end;
+	}
+
+	if (babeltrace_debug) {
+		ret = ctf_visitor_print_xml(stdout, 0, &scanner->ast->root);
+		if (ret) {
+			fprintf(stdout, "Error visiting AST for XML output\n");
+			goto end;
+		}
+	}
+
+	ret = ctf_visitor_semantic_check(stdout, 0, &scanner->ast->root);
+	if (ret) {
+		fprintf(stdout, "Error in CTF semantic validation %d\n", ret);
+		goto end;
+	}
+	ret = ctf_visitor_construct_metadata(stdout, 0, &scanner->ast->root,
+			&td->ctf_trace, BYTE_ORDER);
+	if (ret) {
+		fprintf(stdout, "Error in CTF metadata constructor %d\n", ret);
+		goto end;
+	}
+end:
+	ctf_scanner_free(scanner);
+end_scanner_alloc:
+	fclose(fp);
+end_stream:
+	close(td->ctf_trace.metadata.fd);
+	return ret;
+}
+
+static
+int ctf_open_trace_read(struct trace_descriptor *td, const char *path, int flags)
+{
+	int ret;
+	struct dirent *dirent;
+	struct dirent *diriter;
+	size_t dirent_len;
+
+	td->ctf_trace.flags = flags;
+
+	/* Open trace directory */
+	td->ctf_trace.dir = opendir(path);
+	if (!td->ctf_trace.dir) {
+		fprintf(stdout, "Unable to open trace directory.\n");
+		ret = -ENOENT;
+		goto error;
+	}
+
+	td->ctf_trace.dirfd = open(path, 0);
+	if (td->ctf_trace.dirfd < 0) {
+		fprintf(stdout, "Unable to open trace directory file descriptor.\n");
+		ret = -ENOENT;
+		goto error_dirfd;
+	}
+	/*
+	 * Keep the metadata file separate.
+	 */
+
+	ret = ctf_open_trace_metadata_read(td);
+	if (ret) {
+		goto error_metadata;
+	}
+
+	/*
+	 * Open each stream: for each file, try to open, check magic
+	 * number, and get the stream ID to add to the right location in
+	 * the stream array.
+	 */
+
+	dirent_len = offsetof(struct dirent, d_name) +
+			fpathconf(td->ctf_trace.dirfd, _PC_NAME_MAX) + 1;
+
+	dirent = malloc(dirent_len);
+
+	for (;;) {
+		ret = readdir_r(td->ctf_trace.dir, dirent, &diriter);
+		if (ret) {
+			fprintf(stdout, "Readdir error.\n");
+			goto readdir_error;
+			
+		}
+		if (!diriter)
+			break;
+		if (!strcmp(diriter->d_name, ".")
+				|| !strcmp(diriter->d_name, "..")
+				|| !strcmp(diriter->d_name, "metadata"))
+			continue;
+	}
+
+	free(dirent);
+	return 0;
+
+readdir_error:
+	free(dirent);
+error_metadata:
+	close(td->ctf_trace.dirfd);
+error_dirfd:
+	closedir(td->ctf_trace.dir);
+error:
+	return ret;
+}
+
+static
+int ctf_open_trace_write(struct trace_descriptor *td, const char *path, int flags)
+{
+	int ret;
+
+	ret = mkdir(path, S_IRWXU|S_IRWXG);
+	if (ret)
+		return ret;
+
+	/* Open trace directory */
+	td->ctf_trace.dir = opendir(path);
+	if (!td->ctf_trace.dir) {
+		fprintf(stdout, "Unable to open trace directory.\n");
+		ret = -ENOENT;
+		goto error;
+	}
+	
+
+	return 0;
+
+error:
+	return ret;
+}
+
+struct trace_descriptor *ctf_open_trace(const char *path, int flags)
+{
+	struct trace_descriptor *td;
+	int ret;
+
+	td = g_new0(struct trace_descriptor, 1);
+
+	switch (flags) {
+	case O_RDONLY:
+		ret = ctf_open_trace_read(td, path, flags);
+		if (ret)
+			goto error;
+		break;
+	case O_WRONLY:
+		ret = ctf_open_trace_write(td, path, flags);
+		if (ret)
+			goto error;
+		break;
+	default:
+		fprintf(stdout, "Incorrect open flags.\n");
+		goto error;
+	}
+
+	return td;
+error:
+	g_free(td);
+	return NULL;
+}
+
+void ctf_close_trace(struct trace_descriptor *td)
+{
+	closedir(td->ctf_trace.dir);
+	g_free(td);
+}
+
+void __attribute__((constructor)) ctf_init(void)
 {
 	int ret;