ust_safe_snprintf should check print format

[ust.git] / include / ust / ust_trace.h

diff --git a/include/ust/ust_trace.h b/include/ust/ust_trace.h
index e1b02576d6ef93b88c45bca1cdcdcf071e355d09..cedd47fdf5ab8bec5a5038dee3b6f998bd677a66 100644 (file)
--- a/include/ust/ust_trace.h
+++ b/include/ust/ust_trace.h
@@ -23,6 +23,8 @@
  * to a printf
  */
 
+#include <stdio.h>
+
 /*
  * Stage 1. Create a struct and a printf calling function
  * that is connected to the tracepoint at load time.
@@ -66,10 +68,23 @@
                                                                        \
                printf(print);                                          \
        }                                                               \
+       static inline int register_event_##name(void *data)             \
+       {                                                               \
+               return register_trace_##name(trace_printf_##name, data); \
+       }                                                               \
+       static inline int unregister_event_##name(void *data)           \
+       {                                                               \
+               return unregister_trace_##name(trace_printf_##name, data); \
+       }                                                               \
+       struct trace_event __event_##name                               \
+       __attribute__((section("__trace_events"), aligned(32))) = {     \
+               __tpstrtab_##name,                                      \
+               register_event_##name,                                  \
+               unregister_event_##name                                 \
+       };                                                              \
        static void __attribute__((constructor)) init_##name()          \
        {                                                               \
                void *dummy;                                            \
-               printf("connecting tracepoint " #name "\n");            \
                register_trace_##name(trace_printf_##name, dummy);      \
        }