2 * Written by Josh Dybnis and released to the public domain, as explained at
3 * http://creativecommons.org/licenses/publicdomain
9 #include "runtime_local.h"
13 #define LWT_BUFFER_SCALE 16
14 #define LWT_BUFFER_SIZE (1 << LWT_BUFFER_SCALE)
15 #define LWT_BUFFER_MASK (LWT_BUFFER_SIZE - 1)
17 typedef struct lwt_record {
24 typedef struct lwt_buffer {
29 lwt_buffer_t *lwt_buf_[MAX_NUM_THREADS] = {};
30 uint64_t flag_mask_ = 0;
31 static const char *flags_ = "";
33 void lwt_thread_init (int thread_id)
35 assert(thread_id < MAX_NUM_THREADS);
36 if (lwt_buf_[thread_id] == NULL) {
37 lwt_buf_[thread_id] = (lwt_buffer_t *)nbd_malloc(sizeof(lwt_buffer_t) + sizeof(lwt_record_t) * LWT_BUFFER_SIZE);
38 memset(lwt_buf_[thread_id], 0, sizeof(lwt_buffer_t));
42 void lwt_set_trace_level (const char *flags)
44 assert(strlen(flags) % 2 == 0); // a well formed <flags> should be an even number of characters long
47 for (i = 0; flags[i]; i+=2) {
48 flag_mask_ |= 1 << (flags[i] - 'A');
52 static inline void dump_record (FILE *file, int thread_id, lwt_record_t *r, uint64_t offset)
54 // print the record if its trace category is enabled at a high enough level
55 int flag = (size_t)r->format >> 56;
56 int level = ((size_t)r->format >> 48) & 0xFF;
57 const char *f = strchr(flags_, flag);
58 if (f != NULL && level <= f[1]) {
59 char s[3] = {flag, level, '\0'};
60 fprintf(file, "%09llu %d %s ", ((uint64_t)r->timestamp - offset) >> 6, thread_id, s);
61 const char *format = (const char *)(((uint64_t)r->format << 16) >> 16); // strip out the embedded flags
62 fprintf(file, format, r->value1, r->value2);
67 static void dump_buffer (FILE *file, int thread_id, uint64_t offset)
69 lwt_buffer_t *tb = lwt_buf_[thread_id];
71 if (tb->head > LWT_BUFFER_SIZE) {
72 for (int i = tb->head & LWT_BUFFER_MASK; i < LWT_BUFFER_SIZE; ++i) {
73 dump_record(file, thread_id, tb->x + i, offset);
77 for (int i = 0; i < (tb->head & LWT_BUFFER_MASK); ++i) {
78 dump_record(file, thread_id, tb->x + i, offset);
82 void lwt_dump (const char *file_name)
84 uint64_t offset = (uint64_t)-1;
86 for (int i = 0; i < MAX_NUM_THREADS; ++i) {
87 if (lwt_buf_[i] != NULL && lwt_buf_[i]->head != 0) {
88 uint64_t x = lwt_buf_[i]->x[0].timestamp;
92 if (lwt_buf_[i]->head > LWT_BUFFER_SIZE)
94 x = lwt_buf_[i]->x[lwt_buf_[i]->head & LWT_BUFFER_MASK].timestamp;
102 if (offset != (uint64_t)-1) {
103 FILE *file = fopen(file_name, "w");
105 for (int i = 0; i < MAX_NUM_THREADS; ++i) {
106 if (lwt_buf_[i] != NULL) {
107 dump_buffer(file, i, offset);
115 void lwt_trace_i (const char *format, size_t value1, size_t value2) {
116 LOCALIZE_THREAD_LOCAL(tid_, int);
117 lwt_buffer_t *tb = lwt_buf_[tid_];
120 __asm__ __volatile__("rdtsc" : "=a" (l), "=d" (u));
121 uint64_t timestamp = ((uint64_t)u << 32) | l;
122 lwt_record_t temp = { timestamp, format, value1, value2 };
124 tb->x[tb->head++ & LWT_BUFFER_MASK] = temp;