[test/profile] Add test coverage for __llvm_profile_write_buffer_internal

Reviewed By: davidxl, MaskRay

Differential Revision: https://reviews.llvm.org/D97697

diff --git a/compiler-rt/test/profile/instrprof-write-buffer-internal.c b/compiler-rt/test/profile/instrprof-write-buffer-internal.c
new file mode 100644 (file)
index 0000000..7003e63
--- /dev/null
+++ b/compiler-rt/test/profile/instrprof-write-buffer-internal.c
@@ -0,0 +1,63 @@
+// RUN: rm -f %t.buf.profraw %t.profraw
+// RUN: %clang_profgen -o %t %s
+// RUN: env LLVM_PROFILE_FILE=%t.profraw %run %t %t.buf.profraw
+// RUN: llvm-profdata show %t.buf.profraw | FileCheck %s -check-prefix=WRITE-BUFFER
+// RUN: not llvm-profdata show %t.profraw 2>&1 | FileCheck %s -check-prefix=ALREADY-DUMPED
+
+// WRITE-BUFFER: Instrumentation level: Front-end
+// WRITE-BUFFER: Total functions: 1
+// WRITE-BUFFER: Maximum function count: 1
+// WRITE-BUFFER: Maximum internal block count: 0
+
+// ALREADY-DUMPED: error: {{.*}} Empty raw profile file
+
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+const void *__llvm_profile_begin_data(void);
+const void *__llvm_profile_end_data(void);
+const char *__llvm_profile_begin_names(void);
+const char *__llvm_profile_end_names(void);
+uint64_t *__llvm_profile_begin_counters(void);
+uint64_t *__llvm_profile_end_counters(void);
+
+uint64_t __llvm_profile_get_size_for_buffer_internal(
+    const void *DataBegin, const void *DataEnd,
+    const uint64_t *CountersBegin, const uint64_t *CountersEnd,
+    const char *NamesBegin, const char *NamesEnd);
+
+int __llvm_profile_write_buffer_internal(
+    char *Buffer, const void *DataBegin,
+    const void *DataEnd, const uint64_t *CountersBegin,
+    const uint64_t *CountersEnd, const char *NamesBegin,
+    const char *NamesEnd);
+
+void __llvm_profile_set_dumped(void);
+
+int main(int argc, const char *argv[]) {
+  uint64_t bufsize = __llvm_profile_get_size_for_buffer_internal(
+      __llvm_profile_begin_data(), __llvm_profile_end_data(),
+      __llvm_profile_begin_counters(), __llvm_profile_end_counters(),
+      __llvm_profile_begin_names(), __llvm_profile_end_names());
+
+  char *buf = malloc(bufsize);
+  int ret = __llvm_profile_write_buffer_internal(buf,
+                 __llvm_profile_begin_data(), __llvm_profile_end_data(),
+                  __llvm_profile_begin_counters(), __llvm_profile_end_counters(),
+                  __llvm_profile_begin_names(), __llvm_profile_end_names());
+
+  if (ret != 0) {
+    fprintf(stderr, "failed to write buffer");
+    return ret;
+  }
+
+  FILE *f = fopen(argv[1], "w");
+  fwrite(buf, bufsize, 1, f);
+  fclose(f);
+  free(buf);
+
+  __llvm_profile_set_dumped();
+
+  return 0;
+}