X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=formats%2Fctf%2Fctf.c;h=08834e9744318e944ae97ad006c43f8ca79b8bb0;hp=a63eb60b496c1ca85c9b2621ec7a3985d9c4635d;hb=5385cf15115d777de4a220a164c101da0e4c3bac;hpb=1ae19169d9cb823765444d22cdb05cd2ed3f162f

diff --git a/formats/ctf/ctf.c b/formats/ctf/ctf.c
index a63eb60b..08834e97 100644
--- a/formats/ctf/ctf.c
+++ b/formats/ctf/ctf.c
@@ -107,7 +107,7 @@ void ctf_init_pos(struct ctf_stream_pos *pos, int fd, int open_flags)
 		pos->flags = MAP_SHARED;
 		pos->parent.rw_table = write_dispatch_table;
 		if (fd >= 0)
-			ctf_move_pos_slow(pos, 0);	/* position for write */
+			ctf_move_pos_slow(pos, 0, SEEK_SET);	/* position for write */
 		break;
 	default:
 		assert(0);
@@ -132,12 +132,14 @@ void ctf_fini_pos(struct ctf_stream_pos *pos)
 	(void) g_array_free(pos->packet_index, TRUE);
 }
 
-void ctf_move_pos_slow(struct ctf_stream_pos *pos, size_t offset)
+void ctf_move_pos_slow(struct ctf_stream_pos *pos, size_t offset, int whence)
 {
 	int ret;
 	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;
@@ -150,6 +152,7 @@ void ctf_move_pos_slow(struct ctf_stream_pos *pos, size_t offset)
 				strerror(errno));
 			assert(0);
 		}
+		pos->base = NULL;
 	}
 
 	/*
@@ -170,16 +173,25 @@ void ctf_move_pos_slow(struct ctf_stream_pos *pos, size_t offset)
 		pos->packet_size = WRITE_PACKET_LEN;
 		off = posix_fallocate(pos->fd, pos->mmap_offset, pos->packet_size / CHAR_BIT);
 		assert(off >= 0);
+		pos->offset = 0;
 	} else {
-		/* The reader will expect us to skip padding */
-		assert(pos->offset + offset == pos->content_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 reader will expect us to skip padding */
+			assert(pos->offset + offset == pos->content_size);
 			++pos->cur_index;
+			break;
+		case SEEK_SET:
+			assert(offset == 0);	/* only seek supported for now */
+			pos->cur_index = 0;
+			break;
+		default:
+			assert(0);
+		}
+		if (pos->cur_index >= pos->packet_index->len) {
+			pos->offset = -EOF;
+			return;
+		}
 		index = &g_array_index(pos->packet_index, struct packet_index,
 				       pos->cur_index);
 		pos->mmap_offset = index->offset;
@@ -187,11 +199,16 @@ void ctf_move_pos_slow(struct ctf_stream_pos *pos, size_t offset)
 		/* Lookup context/packet size in index */
 		pos->content_size = index->content_size;
 		pos->packet_size = index->packet_size;
+		pos->offset = index->data_offset;
 	}
 	/* map new base. Need mapping length from header. */
 	pos->base = mmap(NULL, pos->packet_size / CHAR_BIT, pos->prot,
 			 pos->flags, pos->fd, pos->mmap_offset);
-	pos->offset = 0;
+	if (pos->base == MAP_FAILED) {
+		fprintf(stdout, "[error] mmap error %s.\n",
+			strerror(errno));
+		assert(0);
+	}
 }
 
 /*
@@ -431,6 +448,8 @@ int create_stream_packet_index(struct ctf_trace *td,
 			/* Use content size if non-zero, else file size */
 			packet_index.packet_size = packet_index.content_size ? : filestats.st_size * CHAR_BIT;
 		}
+		/* Save position after header and context */
+		packet_index.data_offset = pos->offset;
 
 		/* add index to packet array */
 		g_array_append_val(file_stream->pos.packet_index, packet_index);
@@ -438,6 +457,9 @@ int create_stream_packet_index(struct ctf_trace *td,
 		pos->mmap_offset += packet_index.packet_size / CHAR_BIT;
 	}
 
+	/* Move pos back to beginning of file */
+	ctf_move_pos_slow(pos, 0, SEEK_SET);	/* position for write */
+
 	return 0;
 }
 
@@ -551,30 +573,6 @@ error:
 	return ret;
 }
 
-static
-int ctf_open_trace_write(struct ctf_trace *td, const char *path, int flags)
-{
-	int ret;
-
-	ret = mkdir(path, S_IRWXU|S_IRWXG);
-	if (ret)
-		return ret;
-
-	/* Open trace directory */
-	td->dir = opendir(path);
-	if (!td->dir) {
-		fprintf(stdout, "[error] 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 ctf_trace *td;
@@ -584,6 +582,10 @@ struct trace_descriptor *ctf_open_trace(const char *path, int flags)
 
 	switch (flags & O_ACCMODE) {
 	case O_RDONLY:
+		if (!path) {
+			fprintf(stdout, "[error] Path missing for input CTF trace.\n");
+			goto error;
+		}
 		ret = ctf_open_trace_read(td, path, flags);
 		if (ret)
 			goto error;
@@ -591,12 +593,6 @@ struct trace_descriptor *ctf_open_trace(const char *path, int flags)
 	case O_WRONLY:
 		fprintf(stdout, "[error] Opening CTF traces for output is not supported yet.\n");
 		goto error;
-#if 0
-		ret = ctf_open_trace_write(td, path, flags);
-		if (ret)
-			goto error;
-#endif //0
-		break;
 	default:
 		fprintf(stdout, "[error] Incorrect open flags.\n");
 		goto error;