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: /*@
     32:   PetscMkdir - Create a directory

     34:   Not Collective

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

     39:   Level: advanced

     41: .seealso: `PetscMkdtemp()`, `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: /*@
     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: /*@
    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