Add a dlopen test for toggle traces

Author: thilinarmtb

integration_tests/setup_suite.bash

@@ -3,7 +3,10 @@
 setup_suite() {
   export MPIRUN=${MPIRUN:-mpirun}
 
+  # Set the path to find iprof, babeltrace_thapi, etc.
   export PATH=$(pkg-config --variable=bindir thapi):${PATH}
+  # We need this for the toggle_api/toggle_dlopen test.
+  export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:$(pkg-config --variable=libdir thapi)
 
   missing_tools=()
 

integration_tests/toggle.bats

@@ -10,10 +10,9 @@ get_unique_jobid() {
 }
 
 @test "toggle_api" {
-  rm -rf toggle_traces 2>/dev/null
-
   cc ${THAPI_INCFLAGS} ./integration_tests/toggle.c -o toggle ${THAPI_LDFLAGS}
 
+  rm -rf toggle_traces 2>/dev/null
   iprof --trace-output toggle_traces --no-analysis -- ./toggle
   dir=$(ls -d -1 ./toggle_traces/*/)
 
@@ -33,6 +32,21 @@ get_unique_jobid() {
   auto_stop_count=$(babeltrace_thapi -c $dir | grep lttng_ust_toggle:auto_stop | wc -l)
   [ "$auto_stop_count" -eq 1 ]
 
+  cc ./integration_tests/toggle_dlopen.c -o toggle_dlopen -ldl
+
+  rm -rf toggle_traces 2>/dev/null
+  iprof --trace-output toggle_traces --no-analysis -- ./toggle_dlopen
+  dir=$(ls -d -1 ./toggle_traces/*/)
+
+  # Check expected trace counts.
+  start_count=$(babeltrace_thapi -c $dir | grep lttng_ust_toggle:start | wc -l)
+  [ "$start_count" -eq 2 ]
+
+  stop_count=$(babeltrace_thapi -c $dir | grep lttng_ust_toggle:stop | wc -l)
+  [ "$stop_count" -eq 2 ]
+
+  auto_stop_count=$(babeltrace_thapi -c $dir | grep lttng_ust_toggle:auto_stop | wc -l)
+  [ "$auto_stop_count" -eq 2 ]
 }
 
 toggle_count_base() {

integration_tests/toggle.c

@@ -1,6 +1,7 @@
 #include <thapi.h>
 
-int main(int argc, char *argv[]) {
+int main(void) {
   thapi_start();
   thapi_stop();
+	return 0;
 }

integration_tests/toggle_dlopen.c

@@ -0,0 +1,35 @@
+#include <dlfcn.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#define check_error(ptr_)                                                      \
+  {                                                                            \
+    void *ptr = (void *)ptr_;                                                  \
+    if (!ptr) {                                                                \
+      printf("%s:%d -- %s\n", __FILE__, __LINE__, dlerror());                  \
+      return 1;                                                                \
+    }                                                                          \
+  }
+
+int main(void) {
+  dlerror();
+
+  for (int i = 0; i < 2; i++) {
+    void *thapi = dlopen("libThapi.so", RTLD_NOW | RTLD_LOCAL);
+    check_error(thapi);
+
+    void (*start)(void) = (void (*)(void))dlsym(thapi, "thapi_start");
+    check_error(start);
+
+    void (*stop)(void) = (void (*)(void))dlsym(thapi, "thapi_stop");
+    check_error(stop);
+
+    (*start)(), (*stop)();
+
+    dlclose(thapi);
+  }
+
+  return 0;
+}
+
+#undef check_error