X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=lttng-events.c;h=c129767469db48ebc3880496a0bf2fa0e7249dea;hb=b63af41cb645dcf8513e997714e47fafcb4492b4;hp=d8cf337de93bf75d92b81754a7325fc0e2d0d1ee;hpb=30913c473110ba9969813c05d4db017868772c4d;p=lttng-modules.git

diff --git a/lttng-events.c b/lttng-events.c
index d8cf337d..c1297674 100644
--- a/lttng-events.c
+++ b/lttng-events.c
@@ -44,6 +44,9 @@
 #include "lttng-events.h"
 #include "lttng-tracer.h"
 #include "lttng-abi-old.h"
+#include "wrapper/vzalloc.h"
+#include "wrapper/ringbuffer/backend.h"
+#include "wrapper/ringbuffer/frontend.h"
 
 #define METADATA_CACHE_DEFAULT_SIZE 4096
 
@@ -98,7 +101,7 @@ struct lttng_session *lttng_session_create(void)
 			GFP_KERNEL);
 	if (!metadata_cache)
 		goto err_free_session;
-	metadata_cache->data = vzalloc(METADATA_CACHE_DEFAULT_SIZE);
+	metadata_cache->data = lttng_vzalloc(METADATA_CACHE_DEFAULT_SIZE);
 	if (!metadata_cache->data)
 		goto err_free_cache;
 	metadata_cache->cache_alloc = METADATA_CACHE_DEFAULT_SIZE;
@@ -185,6 +188,10 @@ int lttng_session_enable(struct lttng_session *session)
 			chan->header_type = 2;	/* large */
 	}
 
+	/* Clear each stream's quiescent state. */
+	list_for_each_entry(chan, &session->chan, list)
+		lib_ring_buffer_clear_quiescent_channel(chan->chan);
+
 	ACCESS_ONCE(session->active) = 1;
 	ACCESS_ONCE(session->been_active) = 1;
 	ret = _lttng_session_metadata_statedump(session);
@@ -203,6 +210,7 @@ end:
 int lttng_session_disable(struct lttng_session *session)
 {
 	int ret = 0;
+	struct lttng_channel *chan;
 
 	mutex_lock(&sessions_mutex);
 	if (!session->active) {
@@ -210,6 +218,11 @@ int lttng_session_disable(struct lttng_session *session)
 		goto end;
 	}
 	ACCESS_ONCE(session->active) = 0;
+
+	/* Set each stream's quiescent state. */
+	list_for_each_entry(chan, &session->chan, list)
+		lib_ring_buffer_set_quiescent_channel(chan->chan);
+
 end:
 	mutex_unlock(&sessions_mutex);
 	return ret;
@@ -688,7 +701,7 @@ int lttng_metadata_printf(struct lttng_session *session,
 		tmp_cache_alloc_size = max_t(unsigned int,
 				session->metadata_cache->cache_alloc + len,
 				session->metadata_cache->cache_alloc << 1);
-		tmp_cache_realloc = vzalloc(tmp_cache_alloc_size);
+		tmp_cache_realloc = lttng_vzalloc(tmp_cache_alloc_size);
 		if (!tmp_cache_realloc)
 			goto err;
 		if (session->metadata_cache->data) {