1// RUN: %clang_profgen -DCHECK_SYMBOLS -O3 -o %t.symbols %s
2// RUN: llvm-nm %t.symbols | FileCheck %s --check-prefix=CHECK-SYMBOLS
3// RUN: %clang_profgen -O3 -o %t %s
4// RUN: %run %t %t.profraw
5// RUN: llvm-profdata merge -o %t.profdata %t.profraw
6// RUN: %clang_profuse=%t.profdata -o - -S -emit-llvm %s | FileCheck %s
7
8#include <stdint.h>
9#include <stdlib.h>
10
11#ifndef CHECK_SYMBOLS
12#include <stdio.h>
13#endif
14
15int __llvm_profile_runtime = 0;
16uint64_t __llvm_profile_get_size_for_buffer(void);
17int __llvm_profile_write_buffer(char *);
18int write_buffer(uint64_t, const char *);
19int main(int argc, const char *argv[]) {
20  // CHECK-LABEL: define i32 @main(
21  // CHECK: br i1 %{{.*}}, label %{{.*}}, label %{{.*}}, !prof !1
22  if (argc < 2)
23    return 1;
24
25  const uint64_t MaxSize = 10000;
26  static char Buffer[MaxSize];
27
28  uint64_t Size = __llvm_profile_get_size_for_buffer();
29  if (Size > MaxSize)
30    return 1;
31  int Write = __llvm_profile_write_buffer(Buffer);
32  if (__llvm_profile_write_buffer(Buffer))
33    return Write;
34
35#ifdef CHECK_SYMBOLS
36  // Don't write it out.  Since we're checking the symbols, we don't have libc
37  // available.
38  return 0;
39#else
40  // Actually write it out so we can FileCheck the output.
41  FILE *File = fopen(argv[1], "w");
42  if (!File)
43    return 1;
44  if (fwrite(Buffer, 1, Size, File) != Size)
45    return 1;
46  return fclose(File);
47#endif
48}
49// CHECK: !1 = metadata !{metadata !"branch_weights", i32 1, i32 2}
50
51// CHECK-SYMBOLS-NOT: ___cxx_global_var_init
52// CHECK-SYMBOLS-NOT: ___llvm_profile_register_write_file_atexit
53// CHECK-SYMBOLS-NOT: ___llvm_profile_set_filename
54// CHECK-SYMBOLS-NOT: ___llvm_profile_write_file
55// CHECK-SYMBOLS-NOT: _fdopen
56// CHECK-SYMBOLS-NOT: _fopen
57// CHECK-SYMBOLS-NOT: _fwrite
58// CHECK-SYMBOLS-NOT: _getenv
59// CHECK-SYMBOLS-NOT: _malloc
60// CHECK-SYMBOLS-NOT: _open
61