1 // RUN: %clang_profgen -O2 -o %t %s
2 // RUN: %run %t %t.profraw 1 1
3 // RUN: llvm-profdata show --all-functions --counts %t.profraw | FileCheck %s
4
5 #include <stdint.h>
6 #include <stdio.h>
7 #include <stdlib.h>
8
9 int __llvm_profile_runtime = 0;
10 uint64_t __llvm_profile_get_size_for_buffer(void);
11 int __llvm_profile_write_buffer(char *);
12 void __llvm_profile_reset_counters(void);
13 void __llvm_profile_merge_from_buffer(const char *, uint64_t);
14
dumpBuffer(const char * FileN,const char * Buffer,uint64_t Size)15 int dumpBuffer(const char *FileN, const char *Buffer, uint64_t Size) {
16 FILE *File = fopen(FileN, "w");
17 if (!File)
18 return 1;
19 if (fwrite(Buffer, 1, Size, File) != Size)
20 return 1;
21 return fclose(File);
22 }
23
24 int g = 0;
foo(char c)25 void foo(char c) {
26 if (c == '1')
27 g++;
28 else
29 g--;
30 }
31
32 /* This function is not profiled */
bar(int M)33 void bar(int M) { g += M; }
34
main(int argc,const char * argv[])35 int main(int argc, const char *argv[]) {
36 int i;
37 if (argc < 4)
38 return 1;
39
40 const uint64_t MaxSize = 10000;
41 static char Buffer[MaxSize];
42
43 uint64_t Size = __llvm_profile_get_size_for_buffer();
44 if (Size > MaxSize)
45 return 1;
46
47 /* Start profiling. */
48 __llvm_profile_reset_counters();
49 foo(argv[2][0]);
50 /* End profiling by freezing counters. */
51 if (__llvm_profile_write_buffer(Buffer))
52 return 1;
53
54 /* Its profile will be discarded. */
55 for (i = 0; i < 10; i++)
56 bar(1);
57
58 /* Start profiling again and merge in previously
59 saved counters in buffer. */
60 __llvm_profile_reset_counters();
61 __llvm_profile_merge_from_buffer(Buffer, Size);
62 foo(argv[3][0]);
63 /* End profiling */
64 if (__llvm_profile_write_buffer(Buffer))
65 return 1;
66
67 /* Its profile will be discarded. */
68 bar(2);
69
70 /* Now it is time to dump the profile to file. */
71 return dumpBuffer(argv[1], Buffer, Size);
72 }
73
74 // Not profiled
75 // CHECK-LABEL: dumpBuffer:
76 // CHECK: Counters: 3
77 // CHECK-NEXT: Function count: 0
78 // CHECK-NEXT: Block counts: [0, 0]
79
80 // Profiled with entry count == 2
81 // CHECK-LABEL: foo:
82 // CHECK: Counters: 2
83 // CHECK-NEXT: Function count: 2
84 // CHECK-NEXT: Block counts: [2]
85
86 // Not profiled
87 // CHECK-LABEL: bar:
88 // CHECK: Counters: 1
89 // CHECK-NEXT Function count: 0
90 // CHECK-NEXT Block counts: []
91
92 // Not profiled
93 // CHECK-LABEL: main:
94 // CHECK: Counters: 6
95 // CHECK-NEXT: Function count: 0
96 // CHECK-NEXT: Block counts: [0, 0, 0, 0, 0]
97