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