| // UNSUPPORTED: windows |
| // RUN: %clang_pgogen -o %t.bin %s -DTESTPATH=\"%t.dir\" |
| // RUN: rm -rf %t.dir |
| // RUN: %run %t.bin |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <sys/stat.h> |
| |
| void __llvm_profile_set_dir_mode(unsigned Mode); |
| unsigned __llvm_profile_get_dir_mode(void); |
| void __llvm_profile_recursive_mkdir(char *Path); |
| |
| static int test(unsigned Mode, const char *TestDir) { |
| int Ret = 0; |
| |
| /* Create a dir and set the mode accordingly. */ |
| char *Dir = strdup(TestDir); |
| if (!Dir) |
| return -1; |
| __llvm_profile_set_dir_mode(Mode); |
| __llvm_profile_recursive_mkdir(Dir); |
| |
| if (Mode != __llvm_profile_get_dir_mode()) |
| Ret = -1; |
| else { |
| // From 'man mkdir': |
| // "If the parent directory has the set-group-ID bit set, then so will the |
| // newly created directory." So we mask off S_ISGID below; this test cannot |
| // control its parent directory. |
| const unsigned Expected = ~umask(0) & Mode; |
| struct stat DirSt; |
| if (stat(Dir, &DirSt) == -1) |
| Ret = -1; |
| // AIX has some extended definition of high order bits for st_mode, avoid trying to comparing those by masking them off. |
| else if (((DirSt.st_mode & ~S_ISGID) & 0xFFFF) != Expected) { |
| printf("Modes do not match: Expected %o but found %o (%s)\n", Expected, |
| DirSt.st_mode, Dir); |
| Ret = -1; |
| } |
| } |
| |
| free(Dir); |
| return Ret; |
| } |
| |
| int main(void) { |
| if (test(S_IFDIR | 0777, TESTPATH "/foo/bar/baz/") || |
| test(S_IFDIR | 0666, TESTPATH "/foo/bar/qux/")) |
| return -1; |
| return 0; |
| } |