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

   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(FileN, "w");
  21   if (!File)
  22     return 1;
  23   if (fwrite(Buffer, 1, Size, File) != Size)
  24     return 1;
  25   return fclose(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(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(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(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(2);
  73
  74   /* Now it is time to dump the profile to file.  */
  75   return dumpBuffer(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]