llvm-project

1
// UNSUPPORTED: target={{.*windows.*}}
2
// RUN: %clang_pgogen -o %t.bin %s -DTESTPATH=\"%t.dir\"
3
// RUN: rm -rf %t.dir
4
// RUN: %run %t.bin
5

6
#include <stdio.h>
7
#include <stdlib.h>
8
#include <string.h>
9
#include <sys/stat.h>
10

11
void __llvm_profile_set_dir_mode(unsigned Mode);
12
unsigned __llvm_profile_get_dir_mode(void);
13
void __llvm_profile_recursive_mkdir(char *Path);
14

15
static int test(unsigned Mode, const char *TestDir) {
16
  int Ret = 0;
17

18
  /* Create a dir and set the mode accordingly. */
19
  char *Dir = strdup(TestDir);
20
  if (!Dir)
21
    return -1;
22
  __llvm_profile_set_dir_mode(Mode);
23
  __llvm_profile_recursive_mkdir(Dir);
24

25
  if (Mode != __llvm_profile_get_dir_mode())
26
    Ret = -1;
27
  else {
28
    // From 'man mkdir':
29
    // "If the parent directory has the set-group-ID bit set, then so will the
30
    // newly created directory."  So we mask off S_ISGID below; this test cannot
31
    // control its parent directory.
32
    const unsigned Expected = ~umask(0) & Mode;
33
    struct stat DirSt;
34
    if (stat(Dir, &DirSt) == -1)
35
      Ret = -1;
36
    // AIX has some extended definition of high order bits for st_mode, avoid trying to comparing those by masking them off.
37
    else if (((DirSt.st_mode & ~S_ISGID) & 0xFFFF) != Expected) {
38
      printf("Modes do not match: Expected %o but found %o (%s)\n", Expected,
39
             DirSt.st_mode, Dir);
40
      Ret = -1;
41
    }
42
  }
43

44
  free(Dir);
45
  return Ret;
46
}
47

48
int main(void) {
49
  if (test(S_IFDIR | 0777, TESTPATH "/foo/bar/baz/") ||
50
      test(S_IFDIR | 0666, TESTPATH "/foo/bar/qux/"))
51
    return -1;
52
  return 0;
53
}
54