X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;ds=sidebyside;f=formats%2Fctf%2Fctf.c;h=50cc93de1d866b83a3302767871707ff3c4e5799;hb=312623540c466defab45503fbe0ce7ec79dcce85;hp=a418a6d507d4f1f33051495bdd3c95ada513f5ae;hpb=58b0b8835f84f2a028328e02eb5044640880d1e5;p=babeltrace.git

diff --git a/formats/ctf/ctf.c b/formats/ctf/ctf.c
index a418a6d5..50cc93de 100644
--- a/formats/ctf/ctf.c
+++ b/formats/ctf/ctf.c
@@ -79,6 +79,144 @@ struct format ctf_format = {
 	.close_trace = ctf_close_trace,
 };
 
+static
+int ctf_read_event(struct stream_pos *ppos, struct ctf_stream *stream_class)
+{
+	struct ctf_stream_pos *pos =
+		container_of(ppos, struct ctf_stream_pos, parent);
+	struct ctf_event *event_class;
+	uint64_t id = 0;
+	int len_index;
+	int ret;
+
+	if (pos->offset == EOF)
+		return EOF;
+
+	/* Read event header */
+	if (stream_class->event_header) {
+		ret = generic_rw(ppos, &stream_class->event_header->p);
+		if (ret)
+			goto error;
+		/* lookup event id */
+		len_index = struct_declaration_lookup_field_index(stream_class->event_header_decl,
+				g_quark_from_static_string("id"));
+		if (len_index >= 0) {
+			struct definition_integer *defint;
+			struct definition *field;
+
+			field = struct_definition_get_field_from_index(stream_class->event_header, len_index);
+			assert(field->declaration->id == CTF_TYPE_INTEGER);
+			defint = container_of(field, struct definition_integer, p);
+			assert(defint->declaration->signedness == FALSE);
+			id = defint->value._unsigned;	/* set id */
+		}
+	}
+
+	/* Read stream-declared event context */
+	if (stream_class->event_context) {
+		ret = generic_rw(ppos, &stream_class->event_context->p);
+		if (ret)
+			goto error;
+	}
+
+	if (id >= stream_class->events_by_id->len) {
+		fprintf(stdout, "[error] Event id %" PRIu64 " is outside range.\n", id);
+		return -EINVAL;
+	}
+	event_class = g_ptr_array_index(stream_class->events_by_id, id);
+	if (!event_class) {
+		fprintf(stdout, "[error] Event id %" PRIu64 " is unknown.\n", id);
+		return -EINVAL;
+	}
+
+	/* Read event-declared event context */
+	if (event_class->context) {
+		ret = generic_rw(ppos, &event_class->context->p);
+		if (ret)
+			goto error;
+	}
+
+	/* Read event payload */
+	if (event_class->fields) {
+		ret = generic_rw(ppos, &event_class->fields->p);
+		if (ret)
+			goto error;
+	}
+
+	return 0;
+
+error:
+	fprintf(stdout, "[error] Unexpected end of stream. Either the trace data stream is corrupted or metadata description does not match data layout.\n");
+	return ret;
+}
+
+static
+int ctf_write_event(struct stream_pos *pos, struct ctf_stream *stream_class)
+{
+	struct ctf_event *event_class;
+	uint64_t id = 0;
+	int len_index;
+	int ret;
+
+	/* print event header */
+	if (stream_class->event_header) {
+		/* lookup event id */
+		len_index = struct_declaration_lookup_field_index(stream_class->event_header_decl,
+				g_quark_from_static_string("id"));
+		if (len_index >= 0) {
+			struct definition_integer *defint;
+			struct definition *field;
+
+			field = struct_definition_get_field_from_index(stream_class->event_header, len_index);
+			assert(field->declaration->id == CTF_TYPE_INTEGER);
+			defint = container_of(field, struct definition_integer, p);
+			assert(defint->declaration->signedness == FALSE);
+			id = defint->value._unsigned;	/* set id */
+		}
+
+		ret = generic_rw(pos, &stream_class->event_header->p);
+		if (ret)
+			goto error;
+	}
+
+	/* print stream-declared event context */
+	if (stream_class->event_context) {
+		ret = generic_rw(pos, &stream_class->event_context->p);
+		if (ret)
+			goto error;
+	}
+
+	if (id >= stream_class->events_by_id->len) {
+		fprintf(stdout, "[error] Event id %" PRIu64 " is outside range.\n", id);
+		return -EINVAL;
+	}
+	event_class = g_ptr_array_index(stream_class->events_by_id, id);
+	if (!event_class) {
+		fprintf(stdout, "[error] Event id %" PRIu64 " is unknown.\n", id);
+		return -EINVAL;
+	}
+
+	/* print event-declared event context */
+	if (event_class->context) {
+		ret = generic_rw(pos, &event_class->context->p);
+		if (ret)
+			goto error;
+	}
+
+	/* Read and print event payload */
+	if (event_class->fields) {
+		ret = generic_rw(pos, &event_class->fields->p);
+		if (ret)
+			goto error;
+	}
+
+	return 0;
+
+error:
+	fprintf(stdout, "[error] Unexpected end of stream. Either the trace data stream is corrupted or metadata description does not match data layout.\n");
+	return ret;
+}
+
 void ctf_init_pos(struct ctf_stream_pos *pos, int fd, int open_flags)
 {
 	pos->fd = fd;
@@ -100,12 +238,13 @@ void ctf_init_pos(struct ctf_stream_pos *pos, int fd, int open_flags)
 		pos->prot = PROT_READ;
 		pos->flags = MAP_PRIVATE;
 		pos->parent.rw_table = read_dispatch_table;
+		pos->parent.event_cb = ctf_read_event;
 		break;
-	case O_WRONLY:
 	case O_RDWR:
 		pos->prot = PROT_WRITE;	/* Write has priority */
 		pos->flags = MAP_SHARED;
 		pos->parent.rw_table = write_dispatch_table;
+		pos->parent.event_cb = ctf_write_event;
 		if (fd >= 0)
 			ctf_move_pos_slow(pos, 0, SEEK_SET);	/* position for write */
 		break;
@@ -138,9 +277,6 @@ void ctf_move_pos_slow(struct ctf_stream_pos *pos, size_t offset, int whence)
 	off_t off;
 	struct packet_index *index;
 
-	/* Only allow random seek in read mode */
-	assert(pos->prot != PROT_WRITE || whence == SEEK_CUR);
-
 	if (pos->prot == PROT_WRITE && pos->content_size_loc)
 		*pos->content_size_loc = pos->offset;
 
@@ -160,18 +296,23 @@ void ctf_move_pos_slow(struct ctf_stream_pos *pos, size_t offset, int whence)
 	 * except to get exactly at the beginning of the next packet.
 	 */
 	if (pos->prot == PROT_WRITE) {
-		/* The writer will add padding */
-		assert(pos->offset + offset == pos->packet_size);
-
-		/*
-		 * Don't increment for initial stream move (only condition where
-		 * pos->offset can be 0.
-		 */
-		if (pos->offset)
+		switch (whence) {
+		case SEEK_CUR:
+			/* The writer will add padding */
+			assert(pos->offset + offset == pos->packet_size);
 			pos->mmap_offset += WRITE_PACKET_LEN / CHAR_BIT;
+			break;
+		case SEEK_SET:
+			assert(offset == 0);	/* only seek supported for now */
+			pos->cur_index = 0;
+			break;
+		default:
+			assert(0);
+		}
 		pos->content_size = -1U;	/* Unknown at this point */
 		pos->packet_size = WRITE_PACKET_LEN;
-		off = posix_fallocate(pos->fd, pos->mmap_offset, pos->packet_size / CHAR_BIT);
+		off = posix_fallocate(pos->fd, pos->mmap_offset,
+				      pos->packet_size / CHAR_BIT);
 		assert(off >= 0);
 		pos->offset = 0;
 	} else {
@@ -189,7 +330,7 @@ void ctf_move_pos_slow(struct ctf_stream_pos *pos, size_t offset, int whence)
 			assert(0);
 		}
 		if (pos->cur_index >= pos->packet_index->len) {
-			pos->offset = -EOF;
+			pos->offset = EOF;
 			return;
 		}
 		index = &g_array_index(pos->packet_index, struct packet_index,
@@ -330,14 +471,14 @@ int create_stream_packet_index(struct ctf_trace *td,
 			len_index = struct_declaration_lookup_field_index(td->packet_header->declaration, g_quark_from_static_string("magic"));
 			if (len_index >= 0) {
 				struct definition_integer *defint;
-				struct field *field;
+				struct definition *field;
 
 				field = struct_definition_get_field_from_index(td->packet_header, len_index);
-				assert(field->definition->declaration->id == CTF_TYPE_INTEGER);
-				defint = container_of(field->definition, struct definition_integer, p);
+				assert(field->declaration->id == CTF_TYPE_INTEGER);
+				defint = container_of(field, struct definition_integer, p);
 				assert(defint->declaration->signedness == FALSE);
 				if (defint->value._unsigned != CTF_MAGIC) {
-					fprintf(stdout, "[error] Invalid magic number %" PRIX64 " at packet %u (file offset %zd).\n",
+					fprintf(stdout, "[error] Invalid magic number 0x%" PRIX64 " at packet %u (file offset %zd).\n",
 							defint->value._unsigned,
 							file_stream->pos.packet_index->len,
 							(ssize_t) pos->mmap_offset);
@@ -349,13 +490,13 @@ int create_stream_packet_index(struct ctf_trace *td,
 			len_index = struct_declaration_lookup_field_index(td->packet_header->declaration, g_quark_from_static_string("trace_uuid"));
 			if (len_index >= 0) {
 				struct definition_array *defarray;
-				struct field *field;
+				struct definition *field;
 				uint64_t i;
 				uint8_t uuidval[UUID_LEN];
 
 				field = struct_definition_get_field_from_index(td->packet_header, len_index);
-				assert(field->definition->declaration->id == CTF_TYPE_ARRAY);
-				defarray = container_of(field->definition, struct definition_array, p);
+				assert(field->declaration->id == CTF_TYPE_ARRAY);
+				defarray = container_of(field, struct definition_array, p);
 				assert(array_len(defarray) == UUID_LEN);
 				assert(defarray->declaration->elem->id == CTF_TYPE_INTEGER);
 
@@ -379,11 +520,11 @@ int create_stream_packet_index(struct ctf_trace *td,
 			len_index = struct_declaration_lookup_field_index(td->packet_header->declaration, g_quark_from_static_string("stream_id"));
 			if (len_index >= 0) {
 				struct definition_integer *defint;
-				struct field *field;
+				struct definition *field;
 
 				field = struct_definition_get_field_from_index(td->packet_header, len_index);
-				assert(field->definition->declaration->id == CTF_TYPE_INTEGER);
-				defint = container_of(field->definition, struct definition_integer, p);
+				assert(field->declaration->id == CTF_TYPE_INTEGER);
+				defint = container_of(field, struct definition_integer, p);
 				assert(defint->declaration->signedness == FALSE);
 				stream_id = defint->value._unsigned;
 			}
@@ -417,11 +558,11 @@ int create_stream_packet_index(struct ctf_trace *td,
 			len_index = struct_declaration_lookup_field_index(stream->packet_context->declaration, g_quark_from_static_string("content_size"));
 			if (len_index >= 0) {
 				struct definition_integer *defint;
-				struct field *field;
+				struct definition *field;
 
 				field = struct_definition_get_field_from_index(stream->packet_context, len_index);
-				assert(field->definition->declaration->id == CTF_TYPE_INTEGER);
-				defint = container_of(field->definition, struct definition_integer, p);
+				assert(field->declaration->id == CTF_TYPE_INTEGER);
+				defint = container_of(field, struct definition_integer, p);
 				assert(defint->declaration->signedness == FALSE);
 				packet_index.content_size = defint->value._unsigned;
 			} else {
@@ -433,11 +574,11 @@ int create_stream_packet_index(struct ctf_trace *td,
 			len_index = struct_declaration_lookup_field_index(stream->packet_context->declaration, g_quark_from_static_string("packet_size"));
 			if (len_index >= 0) {
 				struct definition_integer *defint;
-				struct field *field;
+				struct definition *field;
 
 				field = struct_definition_get_field_from_index(stream->packet_context, len_index);
-				assert(field->definition->declaration->id == CTF_TYPE_INTEGER);
-				defint = container_of(field->definition, struct definition_integer, p);
+				assert(field->declaration->id == CTF_TYPE_INTEGER);
+				defint = container_of(field, struct definition_integer, p);
 				assert(defint->declaration->signedness == FALSE);
 				packet_index.packet_size = defint->value._unsigned;
 			} else {
@@ -564,7 +705,8 @@ int ctf_open_trace_read(struct ctf_trace *td, const char *path, int flags)
 		}
 		if (!diriter)
 			break;
-		if (!strcmp(diriter->d_name, ".")
+		/* Ignore hidden files, ., .. and metadata. */
+		if (!strncmp(diriter->d_name, ".", 1)
 				|| !strcmp(diriter->d_name, "..")
 				|| !strcmp(diriter->d_name, "metadata"))
 			continue;
@@ -606,7 +748,7 @@ struct trace_descriptor *ctf_open_trace(const char *path, int flags)
 		if (ret)
 			goto error;
 		break;
-	case O_WRONLY:
+	case O_RDWR:
 		fprintf(stdout, "[error] Opening CTF traces for output is not supported yet.\n");
 		goto error;
 	default: