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