X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=include%2Flttng%2Fust-compiler.h;h=c938361d496ed84e1b924f0e69da489b8e3893f9;hb=05bfa3dc3a6e6b2ece3686a5f384b6645c2a5010;hp=eb201bdc93c791f142d2cbc0aa7a86fd90d2472d;hpb=d0cd72be0eacd90e56470e9cb954fa129997037e;p=lttng-ust.git

diff --git a/include/lttng/ust-compiler.h b/include/lttng/ust-compiler.h
index eb201bdc..c938361d 100644
--- a/include/lttng/ust-compiler.h
+++ b/include/lttng/ust-compiler.h
@@ -86,4 +86,56 @@
     typedef char lttng_ust_static_assert_##c_identifier_msg[2*!!(predicate)-1]
 #endif
 
+/*
+ * Wrap constructor and destructor functions to invoke them as functions with
+ * the constructor/destructor GNU C attributes when building as C, or as the
+ * constructor/destructor of a variable defined within an anonymous namespace
+ * when building as C++.
+ */
+#ifdef __cplusplus
+#define LTTNG_UST_DECLARE_CONSTRUCTOR_DESTRUCTOR(name, constructor_func,        \
+						 destructor_func, ...)          \
+namespace lttng {                                                               \
+namespace ust {                                                                 \
+namespace details {                                                             \
+class LTTNG_UST__TP_COMBINE_TOKENS(lttng_ust_constructor_destructor_,           \
+				   name) {                                      \
+public:                                                                         \
+	LTTNG_UST__TP_COMBINE_TOKENS(lttng_ust_constructor_destructor_,         \
+				     name)() __VA_ARGS__        \
+	{                                                                       \
+		constructor_func();                                             \
+	}                                                                       \
+	~LTTNG_UST__TP_COMBINE_TOKENS(lttng_ust_constructor_destructor_,        \
+                                      name)() __VA_ARGS__       \
+	{                                                                       \
+		destructor_func();                                              \
+	}                                                                       \
+};                                                                              \
+}                                                                               \
+}                                                                               \
+}                                                                               \
+                                                                                \
+namespace {                                                                     \
+const lttng::ust::details::LTTNG_UST__TP_COMBINE_TOKENS(                        \
+	lttng_ust_constructor_destructor_, name)                                \
+		LTTNG_UST__TP_COMBINE_TOKENS(name, registration_instance);      \
+}
+#else /* __cplusplus */
+#define LTTNG_UST_DECLARE_CONSTRUCTOR_DESTRUCTOR(name, constructor_func,             \
+						 destructor_func, ...)               \
+	static void LTTNG_UST__TP_COMBINE_TOKENS(lttng_ust_constructor_, name)(void) \
+		__attribute__((constructor)) __VA_ARGS__;            \
+	static void LTTNG_UST__TP_COMBINE_TOKENS(lttng_ust_constructor_, name)(void) \
+	{                                                                            \
+		constructor_func();                                                  \
+	}                                                                            \
+	static void LTTNG_UST__TP_COMBINE_TOKENS(lttng_ust_destructor_, name)(void)  \
+		__attribute__((destructor)) __VA_ARGS__;             \
+	static void LTTNG_UST__TP_COMBINE_TOKENS(lttng_ust_destructor_, name)(void)  \
+	{                                                                            \
+		destructor_func();                                                   \
+	}
+#endif
+
 #endif /* _LTTNG_UST_COMPILER_H */