X-Git-Url: https://git.liburcu.org/?a=blobdiff_plain;f=hello%2Fhello.c;h=d271a453149b4e74b05ee7242f185b408b645d18;hb=4db647c5217c9350040c781cdca7ad4a09384337;hp=1c97db939e9a5740846e9015c6be74d098ea77b6;hpb=ba6459ba344b54a4f7775e431447be79fd7790f8;p=ust.git diff --git a/hello/hello.c b/hello/hello.c index 1c97db9..d271a45 100644 --- a/hello/hello.c +++ b/hello/hello.c @@ -1,11 +1,157 @@ #include #include #include +#include +#include +#include +#include #include "../libmarkers/marker.h" #include "usterr.h" #include "tracer.h" #include "marker-control.h" +#include "relay.h" + + +char consumer_stack[10000]; + +#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; + 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; inr_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; inr_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, @@ -14,23 +160,6 @@ void probe(const struct marker *mdata, printf("In probe\n"); } -//ust// void try_map() -//ust// { -//ust// char *m; -//ust// -//ust// /* maybe add MAP_LOCKED */ -//ust// m = mmap(NULL, 4096, PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_POPULATE , -1, 0); -//ust// if(m == (char*)-1) { -//ust// perror("mmap"); -//ust// return; -//ust// } -//ust// -//ust// printf("The mapping is at %p.\n", m); -//ust// strcpy(m, "Hello, Mapping!"); -//ust// } - -//sig_atomic_t must_quit; - void inthandler(int sig) { printf("in handler\n"); @@ -63,145 +192,28 @@ int init_int_handler(void) return 0; } -//ust// DEFINE_MUTEX(probes_mutex); -//ust// -//ust// static LIST_HEAD(probes_registered_list); -//ust// -//ust// int ltt_marker_connect(const char *channel, const char *mname, -//ust// const char *pname) -//ust// -//ust// { -//ust// int ret; -//ust// struct ltt_active_marker *pdata; -//ust// struct ltt_available_probe *probe; -//ust// -//ust// ltt_lock_traces(); -//ust// mutex_lock(&probes_mutex); -//ust// probe = get_probe_from_name(pname); -//ust// if (!probe) { -//ust// ret = -ENOENT; -//ust// goto end; -//ust// } -//ust// pdata = marker_get_private_data(channel, mname, probe->probe_func, 0); -//ust// if (pdata && !IS_ERR(pdata)) { -//ust// ret = -EEXIST; -//ust// goto end; -//ust// } -//ust// pdata = kmem_cache_zalloc(markers_loaded_cachep, GFP_KERNEL); -//ust// if (!pdata) { -//ust// ret = -ENOMEM; -//ust// goto end; -//ust// } -//ust// pdata->probe = probe; -//ust// /* -//ust// * ID has priority over channel in case of conflict. -//ust// */ -//ust// ret = marker_probe_register(channel, mname, NULL, -//ust// probe->probe_func, pdata); -//ust// if (ret) -//ust// kmem_cache_free(markers_loaded_cachep, pdata); -//ust// else -//ust// list_add(&pdata->node, &markers_loaded_list); -//ust// end: -//ust// mutex_unlock(&probes_mutex); -//ust// ltt_unlock_traces(); -//ust// return ret; -//ust// } -//ust// -//ust// -//ust// int ltt_probe_register(struct ltt_available_probe *pdata) -//ust// { -//ust// int ret = 0; -//ust// int comparison; -//ust// struct ltt_available_probe *iter; -//ust// -//ust// mutex_lock(&probes_mutex); -//ust// list_for_each_entry_reverse(iter, &probes_registered_list, node) { -//ust// comparison = strcmp(pdata->name, iter->name); -//ust// if (!comparison) { -//ust// ret = -EBUSY; -//ust// goto end; -//ust// } else if (comparison > 0) { -//ust// /* We belong to the location right after iter. */ -//ust// list_add(&pdata->node, &iter->node); -//ust// goto end; -//ust// } -//ust// } -//ust// /* Should be added at the head of the list */ -//ust// list_add(&pdata->node, &probes_registered_list); -//ust// end: -//ust// mutex_unlock(&probes_mutex); -//ust// return ret; -//ust// } -//ust// -//ust// -//ust// struct ltt_available_probe default_probe = { -//ust// .name = "default", -//ust// .format = NULL, -//ust// .probe_func = ltt_vtrace, -//ust// .callbacks[0] = ltt_serialize_data, -//ust// }; - int main() { int result; + int i; init_int_handler(); - init_ustrelay_transport(); - - printf("page size is %d\n", sysconf(_SC_PAGE_SIZE)); - - char trace_name[] = "theusttrace"; - char trace_type[] = "ustrelay"; - - marker_control_init(); - - marker_probe_register("abc", "testmark", "", probe, NULL); - marker_probe_register("metadata", "core_marker_id", "channel %s name %s event_id %hu int #1u%zu long #1u%zu pointer #1u%zu size_t #1u%zu alignment #1u%u", probe, NULL); -//ust// result = ltt_probe_register(&default_probe); -//ust// if(result) -//ust// ERR("ltt_probe_register"); - - result = ltt_marker_connect("abc", "testmark2", "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"); - for(;;) { - trace_mark(abc, testmark, "", MARK_NOARGS); - trace_mark(abc, testmark2, "", MARK_NOARGS); - sleep(1); + sleep(1); + for(i=0; i<50; i++) { + trace_mark(foo, bar, "%s", "FOOBAZ"); + usleep(100000); } + ltt_trace_stop("auto"); + ltt_trace_destroy("auto"); + scanf("%*s"); return 0; } + +MARKER_LIB