X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=tests%2Fregression%2Fust%2Flibc-wrapper%2Fprog.c;h=4eb7d48a5925a212df2f971fdcc8ad9e47646873;hb=9f35c1ad255b65e97d196033a1d7ca6e123a8006;hp=3068257f88323bad164cfaf69b65ec29eda8b258;hpb=5836cdc2ccafa85a1813d5740133e8b652cfa7e3;p=lttng-tools.git

diff --git a/tests/regression/ust/libc-wrapper/prog.c b/tests/regression/ust/libc-wrapper/prog.c
index 3068257f8..4eb7d48a5 100644
--- a/tests/regression/ust/libc-wrapper/prog.c
+++ b/tests/regression/ust/libc-wrapper/prog.c
@@ -23,7 +23,7 @@
 
 #define N_ITER 1000
 
-int main()
+int main(int argc, char **argv)
 {
 	int i;
 	const char teststr[] = "Hello World! 1234567890abc";
@@ -31,6 +31,9 @@ int main()
 
 	for (i = 0; i < N_ITER; i++) {
 		ptrs[i] = malloc(i+1000);
+		if (!ptrs[i]) {
+			exit(EXIT_FAILURE);
+		}
 
 		memcpy(ptrs[i], teststr, sizeof(teststr));