compiler-rt/test/profile/ContinuousSyncMode/online-merging-windows.c

   1 // REQUIRES: target={{.*windows-msvc.*}}
   2 // REQUIRES: lld-available
   3
   4 // Fails on Windows on Arm for unknown reasons.
   5 // UNSUPPORTED: target=aarch64-pc-windows-msvc
   6
   7 // Test the online merging mode (%m) along with continuous mode (%c).
   8 //
   9 // Split files & cd into a temporary directory.
  10 // RUN: rm -rf %t.dir && split-file %s %t.dir && cd %t.dir
  11 //
  12 // Create two DLLs and a driver program that uses them.
  13 // RUN: %clang_pgogen foo.c -mllvm -instrprof-atomic-counter-update-all=1 -mllvm -runtime-counter-relocation=true -fuse-ld=lld -Wl,-dll -o %t.dir/foo.dll
  14 // RUN: %clang_pgogen bar.c -mllvm -instrprof-atomic-counter-update-all=1 -mllvm -runtime-counter-relocation=true -fuse-ld=lld -Wl,-dll -o %t.dir/bar.dll
  15 // RUN: %clang_pgogen main.c -o main.exe %t.dir/foo.lib %t.dir/bar.lib -mllvm -instrprof-atomic-counter-update-all=1 -mllvm -runtime-counter-relocation=true -fuse-ld=lld
  16 //
  17 // === Round 1 ===
  18 // Test merging+continuous mode without any file contention.
  19 //
  20 // RUN: env LLVM_PROFILE_FILE="%t.dir/profdir/%m%c.profraw" %run %t.dir/main.exe nospawn
  21 // RUN: llvm-profdata merge -o %t.profdata %t.dir/profdir
  22 // RUN: llvm-profdata show --counts --all-functions %t.profdata | FileCheck %s -check-prefix=ROUND1
  23
  24 // ROUND1-LABEL: Counters:
  25 // ROUND1-DAG:   foo:
  26 // ROUND1-DAG:     Hash: 0x{{.*}}
  27 // ROUND1-DAG:     Counters: 1
  28 // ROUND1-DAG:     Block counts: [1]
  29 // ROUND1-DAG:   bar:
  30 // ROUND1-DAG:     Hash: 0x{{.*}}
  31 // ROUND1-DAG:     Counters: 1
  32 // ROUND1-DAG:     Block counts: [1]
  33 // ROUND1-DAG:   main:
  34 // ROUND1-DAG:     Hash: 0x{{.*}}
  35 // ROUND1-LABEL: Instrumentation level: IR
  36 //
  37 // === Round 2 ===
  38 // Test merging+continuous mode with some file contention.
  39 //
  40 // RUN: env LLVM_PROFILE_FILE="%t.dir/profdir/%m%c.profraw" %run %t.dir/main.exe spawn
  41 // RUN: llvm-profdata merge -o %t.profdata %t.dir/profdir
  42 // RUN: llvm-profdata show --counts --all-functions %t.profdata | FileCheck %s -check-prefix=ROUND2
  43
  44 // ROUND2-LABEL: Counters:
  45 // ROUND2-DAG:   foo:
  46 // ROUND2-DAG:     Hash: 0x{{.*}}
  47 // ROUND2-DAG:     Counters: 1
  48 // ROUND2-DAG:     Block counts: [97]
  49 // ROUND2-DAG:   bar:
  50 // ROUND2-DAG:     Hash: 0x{{.*}}
  51 // ROUND2-DAG:     Counters: 1
  52 // ROUND2-DAG:     Block counts: [97]
  53 // ROUND2-DAG:   main:
  54 // ROUND2-DAG:     Hash: 0x{{.*}}
  55 // ROUND2-LABEL: Instrumentation level: IR
  56
  57 //--- foo.c
  58 __declspec(dllexport) void foo(void) {}
  59
  60 //--- bar.c
  61 __declspec(dllexport) void bar(void) {}
  62
  63 //--- main.c
  64 #include <stdio.h>
  65 #include <string.h>
  66 #include <windows.h>
  67
  68
  69 const int num_child_procs_to_spawn = 32;
  70
  71 extern int __llvm_profile_is_continuous_mode_enabled(void);
  72 extern char *__llvm_profile_get_filename(void);
  73
  74 __declspec(dllimport) void foo(void);
  75 __declspec(dllimport) void bar(void);
  76
  77 // Change to "#define" for debug output.
  78 #undef DEBUG_TEST
  79
  80 #ifdef DEBUG_TEST
  81 #  define DEBUG(...) fprintf(stderr, __VA_ARGS__);
  82 #else
  83 #  define DEBUG(...)
  84 #endif
  85
  86 int main(int argc, char *const argv[]) {
  87   if (argc < 2) {
  88     DEBUG("Requires at least one argument.\n");
  89     return 1;
  90   }
  91   if (strcmp(argv[1], "nospawn") == 0) {
  92     DEBUG(
  93         "Hello from child (pid = %lu, cont-mode-enabled = %d, profile = %s).\n",
  94         GetCurrentProcessId(), __llvm_profile_is_continuous_mode_enabled(),
  95         __llvm_profile_get_filename());
  96
  97     foo();
  98     bar();
  99     return 0;
 100   } else if (strcmp(argv[1], "spawn") == 0) {
 101     // This is the start of Round 2.
 102     // Expect Counts[dsoX] = 1, as this was the state at the end of Round 1.
 103     int I;
 104     HANDLE child_pids[num_child_procs_to_spawn];
 105     for (I = 0; I < num_child_procs_to_spawn; ++I) {
 106       foo(); // Counts[dsoX] += 2 * num_child_procs_to_spawn
 107       bar();
 108
 109       DEBUG("Spawning child with argv = {%s, %s, NULL} and envp = {%s, NULL}\n",
 110             child_argv[0], child_argv[1], child_envp[0]);
 111
 112       // Start the child process.
 113       STARTUPINFO si;
 114       ZeroMemory(&si, sizeof(si));
 115       PROCESS_INFORMATION pi;
 116       ZeroMemory(&pi, sizeof(pi));
 117       if (!CreateProcess(NULL,               // No module name (use command line)
 118                          "main.exe nospawn", // Command line
 119                          NULL,               // Process handle not inheritable
 120                          NULL,               // Thread handle not inheritable
 121                          FALSE,              // Set handle inheritance to FALSE
 122                          0,                  // No creation flags
 123                          NULL,               // Use parent's environment block
 124                          NULL,               // Use parent's starting directory
 125                          &si,                // Pointer to STARTUPINFO structure
 126                          &pi)                // Pointer to PROCESS_INFORMATION structure
 127       ) {
 128         fprintf(stderr, "Child %d could not be spawned: %lu\n", I,
 129                 GetLastError());
 130         return 1;
 131       }
 132       child_pids[I] = pi.hProcess;
 133
 134       DEBUG("Spawned child %d (pid = %zu).\n", I, pi.dwProcessId);
 135     }
 136     for (I = 0; I < num_child_procs_to_spawn; ++I) {
 137       foo(); // Counts[dsoX] += num_child_procs_to_spawn
 138       bar();
 139
 140       DWORD exit_code;
 141       WaitForSingleObject(child_pids[I], INFINITE);
 142       if (!GetExitCodeProcess(child_pids[I], &exit_code)) {
 143         fprintf(stderr, "Failed to get exit code of child %d.\n", I);
 144         return 1;
 145       }
 146       if (exit_code != 0) {
 147         fprintf(stderr, "Child %d did not exit with code 0.\n", I);
 148         return 1;
 149       }
 150     }
 151
 152     // At the end of Round 2, we have:
 153     // Counts[dsoX] = 1 + (2 * num_child_procs_to_spawn) + num_child_procs_to_spawn
 154     //              = 97
 155
 156     return 0;
 157   }
 158
 159   return 1;
 160 }