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
11void __llvm_profile_set_dir_mode(unsigned Mode);
12unsigned __llvm_profile_get_dir_mode(void);
13void __llvm_profile_recursive_mkdir(char *Path);
14
15static 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(s: TestDir);
20 if (!Dir)
21 return -1;
22 __llvm_profile_set_dir_mode(Mode);
23 __llvm_profile_recursive_mkdir(Path: 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(mask: 0) & Mode;
33 struct stat DirSt;
34 if (stat(file: Dir, buf: &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(format: "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(ptr: Dir);
45 return Ret;
46}
47
48int 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

source code of compiler-rt/test/profile/instrprof-set-dir-mode.c