1 #define PETSC_DESIRE_FEATURE_TEST_MACROS /* for lstat() */ 2 #include <petscsys.h> 3 #include <sys/stat.h> 4 #if defined(PETSC_HAVE_DIRECT_H) 5 #include <direct.h> 6 #endif 7 #if defined(PETSC_HAVE_IO_H) 8 #include <io.h> 9 #endif 10 #if defined(PETSC_HAVE_STDINT_H) 11 #include <stdint.h> 12 #endif 13 #if defined(PETSC_HAVE_UNISTD_H) /* for mkdtemp */ 14 #include <unistd.h> 15 #endif 16 17 PetscErrorCode PetscPathJoin(const char dname[], const char fname[], size_t n, char fullname[]) 18 { 19 size_t l1, l2; 20 PetscFunctionBegin; 21 PetscCall(PetscStrlen(dname, &l1)); 22 PetscCall(PetscStrlen(fname, &l2)); 23 PetscCheck((l1 + l2 + 2) <= n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Path length is greater than buffer size"); 24 PetscCall(PetscStrncpy(fullname, dname, n)); 25 PetscCall(PetscStrlcat(fullname, "/", n)); 26 PetscCall(PetscStrlcat(fullname, fname, n)); 27 PetscFunctionReturn(0); 28 } 29 30 PetscErrorCode PetscMkdir(const char dir[]) 31 { 32 int err; 33 PetscBool flg; 34 35 PetscFunctionBegin; 36 PetscCall(PetscTestDirectory(dir, 'w', &flg)); 37 if (flg) PetscFunctionReturn(0); 38 #if defined(PETSC_HAVE__MKDIR) && defined(PETSC_HAVE_DIRECT_H) 39 err = _mkdir(dir); 40 #else 41 err = mkdir(dir, S_IRWXU | S_IRGRP | S_IXGRP); 42 #endif 43 PetscCheck(!err, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Could not create dir: %s", dir); 44 PetscFunctionReturn(0); 45 } 46 47 #if defined(PETSC_USING_DARWIN) && defined(PETSC_HAVE_FSTATAT) 48 /* 49 Apple's mkdtemp() crashes under Valgrind so this replaces it with a version that does not crash under valgrind 50 */ 51 #include "apple_fdir.h" 52 #endif 53 54 /*@C 55 PetscMkdtemp - Create a folder with a unique name given a filename template. 56 57 Not Collective 58 59 Input Parameters: 60 . dir - file name template, the last six characters must be 'XXXXXX', and they will be modified upon return 61 62 Level: developer 63 64 .seealso: `PetscMkdir()` 65 @*/ 66 PetscErrorCode PetscMkdtemp(char dir[]) 67 { 68 PetscFunctionBegin; 69 #if defined(PETSC_HAVE_WINDOWS_H) && defined(PETSC_HAVE_IO_H) && defined(PETSC_HAVE__MKDIR) && defined(PETSC_HAVE_DIRECT_H) 70 { 71 int err = 1; 72 char name[PETSC_MAX_PATH_LEN]; 73 PetscInt i = 0, max_retry = 26; 74 size_t len; 75 76 while (err && i < max_retry) { 77 PetscCall(PetscStrncpy(name, dir, sizeof(name))); 78 PetscCall(PetscStrlen(name, &len)); 79 err = _mktemp_s(name, len + 1); 80 PetscCheck(!err, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Could not generate a unique name using the template: %s", dir); 81 err = _mkdir(name); 82 i++; 83 } 84 PetscCheck(!err, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Exceeds maximum retry time when creating temporary dir using the template: %s", dir); 85 PetscCall(PetscStrncpy(dir, name, len + 1)); 86 } 87 #else 88 dir = mkdtemp(dir); 89 PetscCheck(dir, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Could not create temporary dir"); 90 #endif 91 PetscFunctionReturn(0); 92 } 93 94 #if defined(PETSC_HAVE_DIRECT_H) 95 PetscErrorCode PetscRMTree(const char dir[]) 96 { 97 struct _finddata_t data; 98 char loc[PETSC_MAX_PATH_LEN]; 99 PetscBool flg1, flg2; 100 #if defined(PETSC_HAVE_STDINT_H) 101 intptr_t handle; 102 #else 103 long handle; 104 #endif 105 106 PetscFunctionBegin; 107 PetscCall(PetscPathJoin(dir, "*", PETSC_MAX_PATH_LEN, loc)); 108 handle = _findfirst(loc, &data); 109 if (handle == -1) { 110 PetscBool flg; 111 PetscCall(PetscTestDirectory(loc, 'r', &flg)); 112 PetscCheck(!flg, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Cannot access directory to delete: %s", dir); 113 PetscCall(PetscTestFile(loc, 'r', &flg)); 114 PetscCheck(!flg, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Specified path is a file - not a dir: %s", dir); 115 PetscFunctionReturn(0); /* perhaps the dir was not yet created */ 116 } 117 while (_findnext(handle, &data) != -1) { 118 PetscCall(PetscStrcmp(data.name, ".", &flg1)); 119 PetscCall(PetscStrcmp(data.name, "..", &flg2)); 120 if (flg1 || flg2) continue; 121 PetscCall(PetscPathJoin(dir, data.name, PETSC_MAX_PATH_LEN, loc)); 122 if (data.attrib & _A_SUBDIR) { 123 PetscCall(PetscRMTree(loc)); 124 } else { 125 PetscCheck(!remove(loc), PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Could not delete file: %s", loc); 126 } 127 } 128 _findclose(handle); 129 PetscCheck(!_rmdir(dir), PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Could not delete dir: %s", dir); 130 PetscFunctionReturn(0); 131 } 132 #else 133 #include <dirent.h> 134 #include <unistd.h> 135 PetscErrorCode PetscRMTree(const char dir[]) 136 { 137 struct dirent *data; 138 char loc[PETSC_MAX_PATH_LEN]; 139 PetscBool flg1, flg2; 140 DIR *dirp; 141 struct stat statbuf; 142 143 PetscFunctionBegin; 144 dirp = opendir(dir); 145 if (!dirp) { 146 PetscBool flg; 147 PetscCall(PetscTestDirectory(dir, 'r', &flg)); 148 PetscCheck(!flg, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Cannot access directory to delete: %s", dir); 149 PetscCall(PetscTestFile(dir, 'r', &flg)); 150 PetscCheck(!flg, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Specified path is a file - not a dir: %s", dir); 151 PetscFunctionReturn(0); /* perhaps the dir was not yet created */ 152 } 153 while ((data = readdir(dirp))) { 154 PetscCall(PetscStrcmp(data->d_name, ".", &flg1)); 155 PetscCall(PetscStrcmp(data->d_name, "..", &flg2)); 156 if (flg1 || flg2) continue; 157 PetscCall(PetscPathJoin(dir, data->d_name, PETSC_MAX_PATH_LEN, loc)); 158 PetscCheck(lstat(loc, &statbuf) >= 0, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "cannot run lstat() on: %s", loc); 159 if (S_ISDIR(statbuf.st_mode)) { 160 PetscCall(PetscRMTree(loc)); 161 } else { 162 PetscCheck(!unlink(loc), PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Could not delete file: %s", loc); 163 } 164 } 165 closedir(dirp); 166 PetscCheck(!rmdir(dir), PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Could not delete dir: %s", dir); 167 PetscFunctionReturn(0); 168 } 169 #endif 170