X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=tests%2Fregression%2Ftools%2Flive%2Flive_test.c;h=b428f1add176567f5972126581b0dac3fb25c2df;hb=a0377dfefe40662ba7d68617bce6ff467114136c;hp=bfbd842c0e1f2754e367e130b60afec7a36c9c53;hpb=b53d4e59bf0a980aa774dcccc27123986e77b8ff;p=lttng-tools.git

diff --git a/tests/regression/tools/live/live_test.c b/tests/regression/tools/live/live_test.c
index bfbd842c0..b428f1add 100644
--- a/tests/regression/tools/live/live_test.c
+++ b/tests/regression/tools/live/live_test.c
@@ -5,8 +5,6 @@
  *
  */
 
-#include <assert.h>
-#include <errno.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -31,6 +29,7 @@
 #include <bin/lttng-relayd/lttng-viewer-abi.h>
 #include <common/index/ctf-index.h>
 
+#include <common/compat/errno.h>
 #include <common/compat/endian.h>
 
 #define SESSION1 "test1"
@@ -41,6 +40,11 @@
 #define NUM_TESTS 11
 #define mmap_size 524288
 
+#ifdef HAVE_LIBLTTNG_UST_CTL
+#include <lttng/ust-sigbus.h>
+DEFINE_LTTNG_UST_SIGBUS_STATE();
+#endif
+
 static int control_sock;
 struct live_session *session;
 
@@ -73,7 +77,7 @@ ssize_t lttng_live_recv(int fd, void *buf, size_t len)
 	do {
 		ret = recv(fd, buf + copied, to_copy, 0);
 		if (ret > 0) {
-			assert(ret <= to_copy);
+			LTTNG_ASSERT(ret <= to_copy);
 			copied += ret;
 			to_copy -= ret;
 		}
@@ -247,7 +251,7 @@ int create_viewer_session(void)
 		diag("[error] Error sending cmd");
 		goto error;
 	}
-	assert(ret_len == sizeof(cmd));
+	LTTNG_ASSERT(ret_len == sizeof(cmd));
 
 	ret_len = lttng_live_recv(control_sock, &resp, sizeof(resp));
 	if (ret_len == 0) {
@@ -258,7 +262,7 @@ int create_viewer_session(void)
 		diag("[error] Error receiving create session reply");
 		goto error;
 	}
-	assert(ret_len == sizeof(resp));
+	LTTNG_ASSERT(ret_len == sizeof(resp));
 
 	if (be32toh(resp.status) != LTTNG_VIEWER_CREATE_SESSION_OK) {
 		diag("[error] Error creating viewer session");