src/compiler-rt/test/asan/TestCases/Windows/report_globals_reload_dll.cc

   1 // Make sure we can handle reloading the same DLL multiple times.
   2 // RUN: %clang_cl_asan -LD -O0 -DDLL %s -Fe%t.dll
   3 // RUN: %clang_cl_asan -O0 -DEXE %s -Fe%te.exe
   4 // RUN: %env_asan_opts=report_globals=1 %run %te.exe %t.dll 2>&1 | FileCheck %s
   5
   6 #include <windows.h>
   7 #include <stdio.h>
   8 #include <string.h>
   9
  10 extern "C" {
  11 #if defined(EXE)
  12 int main(int argc, char **argv) {
  13   if (argc != 2) {
  14     printf("Usage: %s [client].dll\n", argv[0]);
  15     return 101;
  16   }
  17   const char *dll_name = argv[1];
  18
  19 // CHECK: time to load DLL
  20   printf("time to load DLL\n");
  21   fflush(0);
  22
  23 // CHECK: in DLL(reason=1)
  24 // CHECK: in DLL(reason=0)
  25 // CHECK: in DLL(reason=1)
  26 // CHECK: in DLL(reason=0)
  27 // CHECK: in DLL(reason=1)
  28 // CHECK: in DLL(reason=0)
  29   for (int i = 0; i < 30; ++i) {
  30     HMODULE dll = LoadLibrary(dll_name);
  31     if (dll == NULL)
  32       return 3;
  33
  34     if (!FreeLibrary(dll))
  35       return 4;
  36   }
  37
  38 // CHECK: All OK!
  39   printf("All OK!\n");
  40   fflush(0);
  41 }
  42 #elif defined(DLL)
  43 BOOL WINAPI DllMain(HMODULE, DWORD reason, LPVOID) {
  44   printf("in DLL(reason=%d)\n", (int)reason);
  45   fflush(0);
  46   return TRUE;
  47 }
  48 #else
  49 # error oops!
  50 #endif
  51 }