X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Fstream.cpp;h=9f7bfcb15bb513229118302765c8c8f2655f7f91;hb=ff0a1009ba5fdd3d76f0faa9dcf2f8d4745b6082;hp=b57c061c571b41fea32997d5c778eabee233764c;hpb=7586404633a22e219a7ffd7a0f2624b5bc56e37b;p=lttng-tools.git

diff --git a/src/bin/lttng-relayd/stream.cpp b/src/bin/lttng-relayd/stream.cpp
index b57c061c5..9f7bfcb15 100644
--- a/src/bin/lttng-relayd/stream.cpp
+++ b/src/bin/lttng-relayd/stream.cpp
@@ -10,18 +10,18 @@
 
 #define _LGPL_SOURCE
 #include <algorithm>
-#include <common/common.h>
-#include <common/defaults.h>
-#include <common/fs-handle.h>
-#include <common/sessiond-comm/relayd.h>
-#include <common/utils.h>
+#include <common/common.hpp>
+#include <common/defaults.hpp>
+#include <common/fs-handle.hpp>
+#include <common/sessiond-comm/relayd.hpp>
+#include <common/utils.hpp>
 #include <sys/stat.h>
 #include <urcu/rculist.h>
 
-#include "lttng-relayd.h"
-#include "index.h"
-#include "stream.h"
-#include "viewer-stream.h"
+#include "lttng-relayd.hpp"
+#include "index.hpp"
+#include "stream.hpp"
+#include "viewer-stream.hpp"
 
 #include <sys/types.h>
 #include <fcntl.h>
@@ -31,6 +31,8 @@
 /* Should be called with RCU read-side lock held. */
 bool stream_get(struct relay_stream *stream)
 {
+	ASSERT_RCU_READ_LOCKED();
+
 	return urcu_ref_get_unless_zero(&stream->ref);
 }
 
@@ -578,7 +580,7 @@ struct relay_stream *stream_create(struct ctf_trace *trace,
 	bool acquired_reference = false;
 	struct lttng_trace_chunk *current_trace_chunk;
 
-	stream = (relay_stream *) zmalloc(sizeof(struct relay_stream));
+	stream = zmalloc<relay_stream>();
 	if (stream == NULL) {
 		PERROR("relay stream zmalloc");
 		goto error_no_alloc;