compiler-rt/test/profile/instrprof-merge-entry-cover.c

   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
  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;
  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 */
  38 __attribute__((noinline)) void bar(int M) { g += M; }
  39
  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]