X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=ltt-usertrace%2Fltt-usertrace-fast.c;h=1364d3bb6a0a15c1fa0f15620ffccf2b104ba116;hb=4a89deea80bc8f79cefc4a3a5219fd55f37f1416;hp=a0544b67d57b407e7ddc985fa317346ae3a3d432;hpb=fcbe1ea8ecb990b6b2c12afc929c079b3639a4c9;p=lttv.git

diff --git a/ltt-usertrace/ltt-usertrace-fast.c b/ltt-usertrace/ltt-usertrace-fast.c
index a0544b67..1364d3bb 100644
--- a/ltt-usertrace/ltt-usertrace-fast.c
+++ b/ltt-usertrace/ltt-usertrace-fast.c
@@ -109,6 +109,9 @@ static void ltt_usertrace_fast_cleanup(void *arg)
 static pid_t traced_pid = 0;
 static pid_t traced_tid = 0;
 static int parent_exited = 0;
+static int fd_process = -1;
+static char outfile_name[PATH_MAX];
+static char identifier_name[PATH_MAX];
 
 /* signal handling */
 static void handler_sigusr1(int signo)
@@ -284,6 +287,45 @@ static void flush_buffer(struct ltt_buf *ltt_buf, enum force_switch_mode mode)
 
 }
 
+
+static int open_output_files(void)
+{
+	int ret;
+	int fd;
+	/* Open output files */
+	umask(00000);
+	ret = mkdir(LTT_USERTRACE_ROOT, 0777);
+	if(ret < 0 && errno != EEXIST) {
+		perror("LTT Error in creating output (mkdir)");
+		exit(-1);
+	}
+	ret = chdir(LTT_USERTRACE_ROOT);
+	if(ret < 0) {
+		perror("LTT Error in creating output (chdir)");
+		exit(-1);
+	}
+	snprintf(identifier_name, PATH_MAX-1,	"%lu.%lu.%llu",
+			traced_tid, traced_pid, get_cycles());
+	snprintf(outfile_name, PATH_MAX-1,	"process-%s", identifier_name);
+
+#ifndef LTT_NULL_OUTPUT_TEST
+	fd = creat(outfile_name, 0644);
+#else
+	/* NULL test */
+	ret = symlink("/dev/null", outfile_name);
+	if(ret < 0) {
+		perror("error in symlink");
+		exit(-1);
+	}
+	fd = open(outfile_name, O_WRONLY);
+	if(fd_process < 0) {
+		perror("Error in open");
+		exit(-1);
+	}
+#endif //LTT_NULL_OUTPUT_TEST
+	return fd;
+}
+
 static inline int ltt_buffer_get(struct ltt_buf *ltt_buf,
 		unsigned int *offset)
 {
@@ -320,6 +362,8 @@ static inline int ltt_buffer_put(struct ltt_buf *ltt_buf,
 		 * It can also happen if this is a buffer we never got. */
 		return -EIO;
 	} else {
+		if(traced_pid == 0 || parent_exited) return 0;
+
 		ret = sem_post(&ltt_buf->writer_sem);
 		if(ret < 0) {
 			printf("error in sem_post");
@@ -339,6 +383,9 @@ static int read_subbuffer(struct ltt_buf *ltt_buf, int fd)
 		if(err != -EAGAIN) dbg_printf("LTT Reserving sub buffer failed\n");
 		goto get_error;
 	}
+	if(fd_process == -1) {
+		fd_process = fd = open_output_files();
+	}
 
 	err = TEMP_FAILURE_RETRY(write(fd,
 				ltt_buf->start 
@@ -378,9 +425,6 @@ static void ltt_usertrace_fast_daemon(struct ltt_trace_info *shared_trace_info,
 {
 	struct sigaction act;
 	int ret;
-	int fd_process;
-	char outfile_name[PATH_MAX];
-	char identifier_name[PATH_MAX];
 
 	traced_pid = l_traced_pid;
 	traced_tid = l_traced_tid;
@@ -408,44 +452,11 @@ static void ltt_usertrace_fast_daemon(struct ltt_trace_info *shared_trace_info,
 
 	alarm(3);
 
-	/* Open output files */
-	umask(00000);
-	ret = mkdir(LTT_USERTRACE_ROOT, 0777);
-	if(ret < 0 && errno != EEXIST) {
-		perror("LTT Error in creating output (mkdir)");
-		exit(-1);
-	}
-	ret = chdir(LTT_USERTRACE_ROOT);
-	if(ret < 0) {
-		perror("LTT Error in creating output (chdir)");
-		exit(-1);
-	}
-	snprintf(identifier_name, PATH_MAX-1,	"%lu.%lu.%llu",
-			traced_tid, traced_pid, get_cycles());
-	snprintf(outfile_name, PATH_MAX-1,	"process-%s", identifier_name);
-
-	/* Wait for the first signal before creating files */
-	ret = sigsuspend(&oldset);
-	if(ret != -1) {
-		perror("LTT Error in sigsuspend\n");
-	}
-	if((traced_pid == 0) || parent_exited) goto dead_parent;
-
-#ifndef LTT_NULL_OUTPUT_TEST
-	fd_process = creat(outfile_name, 0644);
-#else
-	/* NULL test */
-	ret = symlink("/dev/null", outfile_name);
-	if(ret < 0) {
-		perror("error in symlink");
-	}
-	fd_process = open(outfile_name, O_WRONLY);
-	if(fd_process < 0) {
-		perror("Error in open");
-	}
-#endif //LTT_NULL_OUTPUT_TEST
-	
 	while(1) {
+		ret = sigsuspend(&oldset);
+		if(ret != -1) {
+			perror("LTT Error in sigsuspend\n");
+		}
 		if(traced_pid == 0) break; /* parent died */
 		if(parent_exited) break;
 		dbg_printf("LTT Doing a buffer switch read. pid is : %lu\n", getpid());
@@ -453,13 +464,7 @@ static void ltt_usertrace_fast_daemon(struct ltt_trace_info *shared_trace_info,
 		do {
 			ret = read_subbuffer(&shared_trace_info->channel.process, fd_process);
 		} while(ret == 0);
-
-		ret = sigsuspend(&oldset);
-		if(ret != -1) {
-			perror("LTT Error in sigsuspend\n");
-		}
 	}
-dead_parent:
 	/* The parent thread is dead and we have finished with the buffer */
 
 	/* Buffer force switch (flush). Using FLUSH instead of ACTIVE because we know
@@ -469,8 +474,8 @@ dead_parent:
 		ret = read_subbuffer(&shared_trace_info->channel.process, fd_process);
 	} while(ret == 0);
 
-
-	close(fd_process);
+	if(fd_process != -1)
+		close(fd_process);
 	
 	ret = sem_destroy(&shared_trace_info->channel.process.writer_sem);
 	if(ret < 0) {