1// REQUIRES: continuous-mode
2// UNSUPPORTED: powerpc-{{.*}}
3
4// Test the online merging mode (%m) along with continuous mode (%c).
5//
6// Create & cd into a temporary directory.
7// RUN: rm -rf %t.dir && mkdir -p %t.dir && cd %t.dir
8//
9// Create two DSOs and a driver program that uses them.
10// RUN: echo "void dso1(void) {}" > dso1.c
11// RUN: echo "void dso2(void) {}" > dso2.c
12// RUN: %clang_pgogen=%t.dir/profdir -fprofile-continuous -fprofile-update=atomic %shared_lib_flag -o %t.dir/dso1.dylib dso1.c
13// RUN: %clang_pgogen=%t.dir/profdir -fprofile-continuous -fprofile-update=atomic %shared_lib_flag -o %t.dir/dso2.dylib dso2.c
14// RUN: %clang_pgogen=%t.dir/profdir -fprofile-continuous -fprofile-update=atomic -o main.exe %s %t.dir/dso1.dylib %t.dir/dso2.dylib
15//
16// === Round 1 ===
17// Test merging+continuous mode without any file contention.
18//
19// RUN: %run %t.dir/main.exe nospawn
20// RUN: llvm-profdata merge -o %t.profdata %t.dir/profdir
21// RUN: llvm-profdata show --counts --all-functions %t.profdata | FileCheck %s -check-prefix=ROUND1
22
23// ROUND1-LABEL: Counters:
24// ROUND1-DAG: dso1:
25// ROUND1-DAG: Hash: 0x{{.*}}
26// ROUND1-DAG: Counters: 1
27// ROUND1-DAG: Block counts: [1]
28// ROUND1-DAG: dso2:
29// ROUND1-DAG: Hash: 0x{{.*}}
30// ROUND1-DAG: Counters: 1
31// ROUND1-DAG: Block counts: [1]
32// ROUND1-DAG: main:
33// ROUND1-DAG: Hash: 0x{{.*}}
34// ROUND1-LABEL: Instrumentation level: IR
35// ROUND1-NEXT: Functions shown: 3
36// ROUND1-NEXT: Total functions: 3
37// ROUND1-NEXT: Maximum function count: 1
38// ROUND1-NEXT: Maximum internal block count: 1
39//
40// === Round 2 ===
41// Test merging+continuous mode with some file contention.
42//
43// RUN: %run %t.dir/main.exe spawn 'LLVM_PROFILE_FILE=%t.dir/profdir/%m%c.profraw'
44// RUN: llvm-profdata merge -o %t.profdata %t.dir/profdir
45// RUN: llvm-profdata show --counts --all-functions %t.profdata | FileCheck %s -check-prefix=ROUND2
46
47// ROUND2-LABEL: Counters:
48// ROUND2-DAG: dso1:
49// ROUND2-DAG: Hash: 0x{{.*}}
50// ROUND2-DAG: Counters: 1
51// ROUND2-DAG: Block counts: [97]
52// ROUND2-DAG: dso2:
53// ROUND2-DAG: Hash: 0x{{.*}}
54// ROUND2-DAG: Counters: 1
55// ROUND2-DAG: Block counts: [97]
56// ROUND2-DAG: main:
57// ROUND2-DAG: Hash: 0x{{.*}}
58// ROUND2-LABEL: Instrumentation level: IR
59// ROUND2-NEXT: Functions shown: 3
60// ROUND2-NEXT: Total functions: 3
61// ROUND2-NEXT: Maximum function count: 97
62// ROUND2-NEXT: Maximum internal block count: 33
63
64#include <spawn.h>
65#include <sys/wait.h>
66#include <sys/errno.h>
67#include <unistd.h>
68#include <string.h>
69#include <stdio.h>
70
71const int num_child_procs_to_spawn = 32;
72
73extern int __llvm_profile_is_continuous_mode_enabled(void);
74extern char *__llvm_profile_get_filename(void);
75
76void dso1(void);
77void dso2(void);
78
79// Change to "#define" for debug output.
80#undef DEBUG_TEST
81
82#ifdef DEBUG_TEST
83#define DEBUG(...) fprintf(stderr, __VA_ARGS__);
84#else
85#define DEBUG(...)
86#endif
87
88int main(int argc, char *const argv[]) {
89 if (strcmp(s1: argv[1], s2: "nospawn") == 0) {
90 DEBUG("Hello from child (pid = %d, cont-mode-enabled = %d, profile = %s).\n",
91 getpid(), __llvm_profile_is_continuous_mode_enabled(), __llvm_profile_get_filename());
92
93 dso1();
94 dso2();
95 return 0;
96 } else if (strcmp(s1: argv[1], s2: "spawn") == 0) {
97 // This is the start of Round 2.
98 // Expect Counts[dsoX] = 1, as this was the state at the end of Round 1.
99
100 int I;
101 pid_t child_pids[num_child_procs_to_spawn];
102 char *const child_argv[] = {argv[0], "nospawn", NULL};
103 char *const child_envp[] = {argv[2], NULL};
104 for (I = 0; I < num_child_procs_to_spawn; ++I) {
105 dso1(); // Counts[dsoX] += 2 * num_child_procs_to_spawn
106 dso2();
107
108 DEBUG("Spawning child with argv = {%s, %s, NULL} and envp = {%s, NULL}\n",
109 child_argv[0], child_argv[1], child_envp[0]);
110
111 int ret = posix_spawn(pid: &child_pids[I], path: argv[0], NULL, NULL, argv: child_argv,
112 envp: child_envp);
113 if (ret != 0) {
114 fprintf(stderr, format: "Child %d could not be spawned: ret = %d, msg = %s\n",
115 I, ret, strerror(errnum: ret));
116 return 1;
117 }
118
119 DEBUG("Spawned child %d (pid = %d).\n", I, child_pids[I]);
120 }
121 for (I = 0; I < num_child_procs_to_spawn; ++I) {
122 dso1(); // Counts[dsoX] += num_child_procs_to_spawn
123 dso2();
124
125 int status;
126 waitpid(pid: child_pids[I], stat_loc: &status, options: 0);
127 if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
128 fprintf(stderr, format: "Child %d did not exit with code 0.\n", I);
129 return 1;
130 }
131 }
132
133 // At the end of Round 2, we have:
134 // Counts[dsoX] = 1 + (2 * num_child_procs_to_spawn) + num_child_procs_to_spawn
135 // = 97
136
137 return 0;
138 }
139
140 return 1;
141}
142

source code of compiler-rt/test/profile/ContinuousSyncMode/online-merging.c