Actual source code: fdir.c

  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

 17: static PetscErrorCode PetscPathJoin(const char dname[], const char fname[], size_t n, char fullname[])
 18: {
 19:   size_t l1, l2;

 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: }

 31: /*@C
 32:   PetscMkdir - Create a directory

 34:   Not Collective

 36:   Input Parameter:
 37: . dir - the directory name

 39:   Level: advanced

 41: .seealso: `PetscMktemp()`, `PetscRMTree()`
 42: @*/
 43: PetscErrorCode PetscMkdir(const char dir[])
 44: {
 45:   int       err;
 46:   PetscBool flg;

 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: }

 60: /*@C
 61:   PetscMkdtemp - Create a directory with a unique name given a name template.

 63:   Input Parameter:
 64: . dir - file name template, the last six characters must be 'XXXXXX', and they will be modified upon return

 66:   Level: advanced

 68: .seealso: `PetscMkdir()`, `PetscRMTree()`
 69: @*/
 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;

 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: }

 98: #if defined(PETSC_HAVE_DIRECT_H)
 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

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>

141: /*@C
142:   PetscRMTree - delete a directory and all of its children

144:   Input Parameter:
145: . dir - the name of the directory

147:   Level: advanced

149: .seealso: `PetscMkdtemp()`, `PetscMkdir()`
150: @*/
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;

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