-#include <stdio.h>
-#include <unistd.h>
-#include <sys/mman.h>
-#include <stdarg.h>
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <fcntl.h>
-
-#include "../libmarkers/marker.h"
-#include "usterr.h"
-#include "tracer.h"
-#include "marker-control.h"
-#include "relay.h"
-
-
-char consumer_stack[10000];
-char trace_name[] = "theusttrace";
-char trace_type[] = "ustrelay";
-
-#define CPRINTF(fmt, args...) safe_printf(fmt "\n", ## args)
-
-int safe_printf(const char *fmt, ...)
-{
- static char buf[500];
- va_list ap;
- int n;
-
- va_start(ap, fmt);
-
- n = vsnprintf(buf, sizeof(buf), fmt, ap);
-
- write(STDOUT_FILENO, buf, n);
-
- va_end(ap);
-}
-
-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;
-
- 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(<tbuf->offset), rbuf->chan) - consumed_old);
-// CPRINTF("Commit count %ld", local_read(<tbuf->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");
- }
-}
-
-void probe(const struct marker *mdata,
- void *probe_private, void *call_private,
- const char *fmt, va_list *args)
-{
- printf("In probe\n");
-}
-
-void inthandler(int sig)
-{
- printf("in handler\n");
- exit(0);
-}
-
-int init_int_handler(void)
-{
- int result;
- struct sigaction act;
-
- result = sigemptyset(&act.sa_mask);
- if(result == -1) {
- PERROR("sigemptyset");
- return -1;
- }
-
- act.sa_handler = inthandler;
- act.sa_flags = SA_RESTART;
-
- /* Only defer ourselves. Also, try to restart interrupted
- * syscalls to disturb the traced program as little as possible.
- */
- result = sigaction(SIGINT, &act, NULL);
- if(result == -1) {
- PERROR("sigaction");
- return -1;
- }
-
- return 0;
-}
-
-int main()
-{
- int result;
- int i;
-
- init_int_handler();
-
- result = ltt_marker_connect("foo", "bar", "default");
- if(result)
- ERR("ltt_marker_connect");
-
- result = ltt_trace_setup(trace_name);
- if(result < 0) {
- ERR("ltt_trace_setup failed");
- return 1;
- }
-
- result = ltt_trace_set_type(trace_name, trace_type);
- if(result < 0) {
- ERR("ltt_trace_set_type failed");
- return 1;
- }
-
- result = ltt_trace_alloc(trace_name);
- if(result < 0) {
- ERR("ltt_trace_alloc failed");
- return 1;
- }
-
- result = ltt_trace_start(trace_name);
- if(result < 0) {
- ERR("ltt_trace_start failed");
- return 1;
- }
-
- start_consumer();
- printf("Hello, World!\n");
-
- sleep(1);
- for(i=0; i<50; i++) {
- trace_mark(foo, bar, "%s", "FOOBAZ");
- usleep(100000);
- }
-
- ltt_trace_stop(trace_name);
- ltt_trace_destroy(trace_name);
-
- scanf("%*s");
-
- return 0;
-}
-
-MARKER_LIB