llvm-project

1
// RUN: %clang_pgogen -O2 -o %t.0 %s
2
// RUN: %clang_pgogen=%/t.d1 -O2 -o %t.1 %s
3
// RUN: %clang_pgogen=%/t.d1/%:t.d2 -O2 -o %t.2 %s
4
//
5
// RUN: %run %t.0  ""
6
// RUN: env LLVM_PROFILE_FILE=%/t.d1/default.profraw %run %t.0  %/t.d1
7
// RUN: env LLVM_PROFILE_FILE=%/t.d1/%:t.d2/default.profraw %run %t.0 %/t.d1/%:t.d2
8
// RUN: %run %t.1 %/t.d1
9
// RUN: %run %t.2 %/t.d1/%:t.d2
10
// RUN: %run %t.2 %/t.d1/%:t.d2 %/t.d1/%:t.d2/%:t.d3/blah.profraw %/t.d1/%:t.d2/%:t.d3/
11

12
#include <string.h>
13

14
const char *__llvm_profile_get_path_prefix();
15
void __llvm_profile_set_filename(const char*);
16

17
int main(int argc, const char *argv[]) {
18
  const char *expected;
19
  const char *prefix;
20
  if (argc < 2)
21
    return 1;
22

23
  expected = argv[1];
24
  prefix = __llvm_profile_get_path_prefix();
25

26
  // The last character should be a trailing slash. Ignore it in the comparison
27
  // since it could be '/' or '\\'.
28
  int slashpos = strlen(prefix);
29
  if (slashpos > 0) {
30
    --slashpos;
31
    if (prefix[slashpos] != '/' && prefix[slashpos] != '\\')
32
      return 1;
33
  }
34

35
  if (strncmp(prefix, expected, slashpos))
36
    return 1;
37

38
  if (argc == 4) {
39
    __llvm_profile_set_filename(argv[2]);
40
    prefix = __llvm_profile_get_path_prefix();
41
    expected = argv[3];
42
    if (strcmp(prefix, expected))
43
      return 1;
44
  }
45

46
  return 0;
47
}
48