xref: /aosp_15_r20/external/compiler-rt/test/asan/TestCases/Posix/coverage-module-unloaded.cc (revision 7c3d14c8b49c529e04be81a3ce6f5cc23712e4c6)
1*7c3d14c8STreehugger Robot // Check that unloading a module doesn't break coverage dumping for remaining
2*7c3d14c8STreehugger Robot // modules.
3*7c3d14c8STreehugger Robot // RUN: %clangxx_asan -fsanitize-coverage=func -DSHARED %s -shared -o %dynamiclib1 -fPIC
4*7c3d14c8STreehugger Robot // RUN: %clangxx_asan -fsanitize-coverage=func -DSHARED %s -shared -o %dynamiclib2 -fPIC
5*7c3d14c8STreehugger Robot // RUN: %clangxx_asan -fsanitize-coverage=func %s %libdl -o %t
6*7c3d14c8STreehugger Robot // RUN: mkdir -p %T/coverage-module-unloaded && cd %T/coverage-module-unloaded
7*7c3d14c8STreehugger Robot // RUN: %env_asan_opts=coverage=1:verbosity=1 %run %t %dynamiclib1 %dynamiclib2 2>&1        | FileCheck %s
8*7c3d14c8STreehugger Robot // RUN: %env_asan_opts=coverage=1:verbosity=1 %run %t %dynamiclib1 %dynamiclib2 foo 2>&1    | FileCheck %s
9*7c3d14c8STreehugger Robot // RUN: rm -r %T/coverage-module-unloaded
10*7c3d14c8STreehugger Robot //
11*7c3d14c8STreehugger Robot // https://code.google.com/p/address-sanitizer/issues/detail?id=263
12*7c3d14c8STreehugger Robot // XFAIL: android
13*7c3d14c8STreehugger Robot 
14*7c3d14c8STreehugger Robot #include <assert.h>
15*7c3d14c8STreehugger Robot #include <dlfcn.h>
16*7c3d14c8STreehugger Robot #include <stdio.h>
17*7c3d14c8STreehugger Robot #include <unistd.h>
18*7c3d14c8STreehugger Robot 
19*7c3d14c8STreehugger Robot #ifdef SHARED
20*7c3d14c8STreehugger Robot extern "C" {
bar()21*7c3d14c8STreehugger Robot void bar() { printf("bar\n"); }
22*7c3d14c8STreehugger Robot }
23*7c3d14c8STreehugger Robot #else
24*7c3d14c8STreehugger Robot 
main(int argc,char ** argv)25*7c3d14c8STreehugger Robot int main(int argc, char **argv) {
26*7c3d14c8STreehugger Robot   fprintf(stderr, "PID: %d\n", getpid());
27*7c3d14c8STreehugger Robot   assert(argc > 2);
28*7c3d14c8STreehugger Robot   void *handle1 = dlopen(argv[1], RTLD_LAZY);  // %dynamiclib1
29*7c3d14c8STreehugger Robot   assert(handle1);
30*7c3d14c8STreehugger Robot   void (*bar1)() = (void (*)())dlsym(handle1, "bar");
31*7c3d14c8STreehugger Robot   assert(bar1);
32*7c3d14c8STreehugger Robot   bar1();
33*7c3d14c8STreehugger Robot   void *handle2 = dlopen(argv[2], RTLD_LAZY);  // %dynamiclib2
34*7c3d14c8STreehugger Robot   assert(handle2);
35*7c3d14c8STreehugger Robot   void (*bar2)() = (void (*)())dlsym(handle2, "bar");
36*7c3d14c8STreehugger Robot   assert(bar2);
37*7c3d14c8STreehugger Robot   bar2();
38*7c3d14c8STreehugger Robot 
39*7c3d14c8STreehugger Robot   // It matters whether the unloaded module has a higher or lower address range
40*7c3d14c8STreehugger Robot   // than the remaining one. Make sure to test both cases.
41*7c3d14c8STreehugger Robot   if (argc < 2)
42*7c3d14c8STreehugger Robot     dlclose(bar1 < bar2 ? handle1 : handle2);
43*7c3d14c8STreehugger Robot   else
44*7c3d14c8STreehugger Robot     dlclose(bar1 < bar2 ? handle2 : handle1);
45*7c3d14c8STreehugger Robot   return 0;
46*7c3d14c8STreehugger Robot }
47*7c3d14c8STreehugger Robot #endif
48*7c3d14c8STreehugger Robot 
49*7c3d14c8STreehugger Robot // CHECK: PID: [[PID:[0-9]+]]
50*7c3d14c8STreehugger Robot // CHECK: [[PID]].sancov: 1 PCs written
51*7c3d14c8STreehugger Robot // CHECK: coverage-module-unloaded{{.*}}1.[[PID]]
52*7c3d14c8STreehugger Robot // CHECK: coverage-module-unloaded{{.*}}2.[[PID]]
53*7c3d14c8STreehugger Robot // Even though we've unloaded one of the libs we still dump the coverage file
54*7c3d14c8STreehugger Robot // for that lib (although the data will be inaccurate, if at all useful)
55