Gen-ust-events: use options instead of arguments

[lttng-tools.git] / tests / regression / ust / libc-wrapper / prog.c

diff --git a/tests/regression/ust/libc-wrapper/prog.c b/tests/regression/ust/libc-wrapper/prog.c
index 3068257f88323bad164cfaf69b65ec29eda8b258..4eb7d48a5925a212df2f971fdcc8ad9e47646873 100644 (file)
--- 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));