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