X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=libtracectl%2Ftracectl.c;h=1725bcab727361f330d0e6127fa260d1f37b9037;hb=a584bc4edb73e3f4c0023c42f2fdbc816fb6eab0;hp=ab62adfd720aa416125b1f4ce2580f18e37ef1cc;hpb=cf49034025d46568a4db4aab3f09dfe912349bb8;p=ust.git

diff --git a/libtracectl/tracectl.c b/libtracectl/tracectl.c
index ab62adf..1725bca 100644
--- a/libtracectl/tracectl.c
+++ b/libtracectl/tracectl.c
@@ -5,8 +5,11 @@
 #include <sys/socket.h>
 #include <sys/un.h>
 #include <sched.h>
+#include <fcntl.h>
 
 #include "marker.h"
+#include "tracer.h"
+#include "usterr.h"
 
 #define UNIX_PATH_MAX 108
 
@@ -17,13 +20,15 @@
 
 #define DBG(fmt, args...) fprintf(stderr, fmt "\n", ## args)
 #define WARN(fmt, args...) fprintf(stderr, "usertrace: WARNING: " fmt "\n", ## args)
-#define ERR(fmt, args...) fprintf(stderr, "usertrace: ERROR: " fmt "\n", ## args)
+#define ERR(fmt, args...) fprintf(stderr, "usertrace: ERROR: " fmt "\n", ## args); fflush(stderr)
 #define PERROR(call) perror("usertrace: ERROR: " call)
 
 #define MAX_MSG_SIZE (100)
 #define MSG_NOTIF 1
 #define MSG_REGISTER_NOTIF 2
 
+char consumer_stack[10000];
+
 struct tracecmd { /* no padding */
 	uint32_t size;
 	uint16_t command;
@@ -47,6 +52,126 @@ pid_t mypid;
 char mysocketfile[UNIX_PATH_MAX] = "";
 int pfd = -1;
 
+struct consumer_channel {
+	int fd;
+	struct ltt_channel_struct *chan;
+};
+
+int consumer(void *arg)
+{
+	int result;
+	int fd;
+	char str[] = "Hello, this is the consumer.\n";
+	struct ltt_trace_struct *trace;
+	struct consumer_channel *consumer_channels;
+	int i;
+	char trace_name[] = "auto";
+
+	ltt_lock_traces();
+	trace = _ltt_trace_find(trace_name);
+	ltt_unlock_traces();
+
+	if(trace == NULL) {
+		CPRINTF("cannot find trace!");
+		return 1;
+	}
+
+	consumer_channels = (struct consumer_channel *) malloc(trace->nr_channels * sizeof(struct consumer_channel));
+	if(consumer_channels == NULL) {
+		ERR("malloc returned NULL");
+		return 1;
+	}
+
+	CPRINTF("opening trace files");
+	for(i=0; i<trace->nr_channels; i++) {
+		char tmp[100];
+		struct ltt_channel_struct *chan = &trace->channels[i];
+
+		consumer_channels[i].chan = chan;
+
+		snprintf(tmp, sizeof(tmp), "trace/%s", chan->channel_name);
+		result = consumer_channels[i].fd = open(tmp, O_WRONLY | O_CREAT | O_TRUNC, 00644);
+		if(result == -1) {
+			perror("open");
+			return -1;
+		}
+		CPRINTF("\topened trace file %s", tmp);
+		
+	}
+	CPRINTF("done opening trace files");
+
+	for(;;) {
+		/*wait*/
+
+		for(i=0; i<trace->nr_channels; i++) {
+			struct rchan *rchan = consumer_channels[i].chan->trans_channel_data;
+			struct rchan_buf *rbuf = rchan->buf;
+			struct ltt_channel_buf_struct *lttbuf = consumer_channels[i].chan->buf;
+			long consumed_old;
+
+			result = ltt_do_get_subbuf(rbuf, lttbuf, &consumed_old);
+			if(result < 0) {
+				CPRINTF("ltt_do_get_subbuf: error: %s", strerror(-result));
+			}
+			else {
+				CPRINTF("success!");
+
+				result = write(consumer_channels[i].fd, rbuf->buf_data + (consumed_old & (2 * 4096-1)), 4096);
+				ltt_do_put_subbuf(rbuf, lttbuf, consumed_old);
+			}
+		}
+
+		sleep(1);
+	}
+
+//	CPRINTF("consumer: got a trace: %s with %d channels\n", trace_name, trace->nr_channels);
+//
+//	struct ltt_channel_struct *chan = &trace->channels[0];
+//
+//	CPRINTF("channel 1 (%s) active=%u", chan->channel_name, chan->active & 1);
+
+//	struct rchan *rchan = chan->trans_channel_data;
+//	struct rchan_buf *rbuf = rchan->buf;
+//	struct ltt_channel_buf_struct *lttbuf = chan->buf;
+//	long consumed_old;
+//
+//	result = fd = open("trace.out", O_WRONLY | O_CREAT | O_TRUNC, 00644);
+//	if(result == -1) {
+//		perror("open");
+//		return -1;
+//	}
+
+//	for(;;) {
+//		write(STDOUT_FILENO, str, sizeof(str));
+//
+//		result = ltt_do_get_subbuf(rbuf, lttbuf, &consumed_old);
+//		if(result < 0) {
+//			CPRINTF("ltt_do_get_subbuf: error: %s", strerror(-result));
+//		}
+//		else {
+//			CPRINTF("success!");
+//
+//			result = write(fd, rbuf->buf_data + (consumed_old & (2 * 4096-1)), 4096);
+//			ltt_do_put_subbuf(rbuf, lttbuf, consumed_old);
+//		}
+//
+//		//CPRINTF("There seems to be %ld bytes available", SUBBUF_TRUNC(local_read(&lttbuf->offset), rbuf->chan) - consumed_old);
+//		CPRINTF("Commit count %ld", local_read(&lttbuf->commit_count[0]));
+//
+//
+//		sleep(1);
+//	}
+}
+
+void start_consumer(void)
+{
+	int result;
+
+	result = clone(consumer, consumer_stack+sizeof(consumer_stack)-1, CLONE_FS | CLONE_FILES | CLONE_VM | CLONE_SIGHAND | CLONE_THREAD, NULL);
+	if(result == -1) {
+		perror("clone");
+	}
+}
 
 static void print_markers(void)
 {
@@ -310,6 +435,8 @@ static void __attribute__((constructor(101))) init0()
 	}
 }
 
+static void fini(void);
+
 static void __attribute__((constructor(1000))) init()
 {
 	int result;
@@ -359,6 +486,7 @@ static void __attribute__((constructor(1000))) init()
 			ERR("ltt_trace_start failed");
 			return;
 		}
+		start_consumer();
 	}
 
 	/* Must create socket before signal handler to prevent races
@@ -386,5 +514,24 @@ static void __attribute__((constructor(1000))) init()
 
 static void __attribute__((destructor)) fini()
 {
+	int result;
+
+	/* if trace running, finish it */
+
+	DBG("destructor stopping traces");
+
+	result = ltt_trace_stop("auto");
+	if(result == -1) {
+		ERR("ltt_trace_stop error");
+	}
+
+	result = ltt_trace_destroy("auto");
+	if(result == -1) {
+		ERR("ltt_trace_destroy error");
+	}
+
+	/* FIXME: wait for the consumer to be done */
+	sleep(10);
+
 	destroy_socket();
 }