Actual source code: ftest.c
petsc-3.7.1 2016-05-15
Report Typos and Errors

      2: #include <petscsys.h>
      3: #include <errno.h>
      4: #if defined(PETSC_HAVE_PWD_H)
      5: #include <pwd.h>
      6: #endif
      7: #include <ctype.h>
      8: #include <sys/stat.h>
      9: #if defined(PETSC_HAVE_UNISTD_H)
     10: #include <unistd.h>
     11: #endif
     12: #if defined(PETSC_HAVE_SYS_UTSNAME_H)
     13: #include <sys/utsname.h>
     14: #endif
     15: #if defined(PETSC_HAVE_IO_H)
     16: #include <io.h>
     17: #endif
     18: #if defined(PETSC_HAVE_SYS_SYSTEMINFO_H)
     19: #include <sys/systeminfo.h>
     20: #endif

     22: #if defined(PETSC_HAVE__ACCESS) || defined(PETSC_HAVE_ACCESS)

     26: static PetscErrorCode PetscTestOwnership(const char fname[], char mode, uid_t fuid, gid_t fgid, int fmode, PetscBool  *flg)
     27: {
     28:   int            m = R_OK;

     32:   if (mode == 'r') m = R_OK;
     33:   else if (mode == 'w') m = W_OK;
     34:   else if (mode == 'x') m = X_OK;
     35:   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG, "Mode must be one of r, w, or x");
     36: #if defined(PETSC_HAVE_ACCESS)
     37:   if (!access(fname, m)) {
     38:     PetscInfo1(NULL,"System call access() succeeded on file %s\n",fname);
     39:     *flg = PETSC_TRUE;
     40:   } else {
     41:     PetscInfo1(NULL,"System call access() failed on file %s\n",fname);
     42:     *flg = PETSC_FALSE;
     43:   }
     44: #else
     45:   if (m == X_OK) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP, "Unable to check execute permission for file %s", fname);
     46:   if (!_access(fname, m)) *flg = PETSC_TRUE;
     47: #endif
     48:   return(0);
     49: }

     51: #else  /* PETSC_HAVE_ACCESS or PETSC_HAVE__ACCESS */

     55: static PetscErrorCode PetscTestOwnership(const char fname[], char mode, uid_t fuid, gid_t fgid, int fmode, PetscBool  *flg)
     56: {
     57:   uid_t          uid;
     58:   gid_t          *gid = NULL;
     59:   int            numGroups;
     60:   int            rbit = S_IROTH;
     61:   int            wbit = S_IWOTH;
     62:   int            ebit = S_IXOTH;

     66:   /* Get the number of supplementary group IDs */
     67: #if !defined(PETSC_MISSING_GETGROUPS)
     68:   numGroups = getgroups(0, gid); if (numGroups < 0) SETERRQ(PETSC_COMM_SELF,numGroups, "Unable to count supplementary group IDs");
     69:   PetscMalloc1(numGroups+1, &gid);
     70: #else
     71:   numGroups = 0;
     72: #endif

     74:   /* Get the (effective) user and group of the caller */
     75:   uid    = geteuid();
     76:   gid[0] = getegid();

     78:   /* Get supplementary group IDs */
     79: #if !defined(PETSC_MISSING_GETGROUPS)
     80:   getgroups(numGroups, gid+1); if (ierr < 0) SETERRQ(PETSC_COMM_SELF,ierr, "Unable to obtain supplementary group IDs");
     81: #endif

     83:   /* Test for accessibility */
     84:   if (fuid == uid) {
     85:     rbit = S_IRUSR;
     86:     wbit = S_IWUSR;
     87:     ebit = S_IXUSR;
     88:   } else {
     89:     int g;

     91:     for (g = 0; g <= numGroups; g++) {
     92:       if (fgid == gid[g]) {
     93:         rbit = S_IRGRP;
     94:         wbit = S_IWGRP;
     95:         ebit = S_IXGRP;
     96:         break;
     97:       }
     98:     }
     99:   }
    100:   PetscFree(gid);

    102:   if (mode == 'r') {
    103:     if (fmode & rbit) *flg = PETSC_TRUE;
    104:   } else if (mode == 'w') {
    105:     if (fmode & wbit) *flg = PETSC_TRUE;
    106:   } else if (mode == 'x') {
    107:     if (fmode & ebit) *flg = PETSC_TRUE;
    108:   }
    109:   return(0);
    110: }

    112: #endif /* PETSC_HAVE_ACCESS */

    116: static PetscErrorCode PetscGetFileStat(const char fname[], uid_t *fileUid, gid_t *fileGid, int *fileMode,PetscBool  *exists)
    117: {
    118:   struct stat    statbuf;

    122:   *fileMode = 0;
    123:   *exists = PETSC_FALSE;
    124: #if defined(PETSC_HAVE_STAT_NO_CONST)
    125:   stat((char*) fname, &statbuf);
    126: #else
    127:   stat(fname, &statbuf);
    128: #endif
    129:   if (ierr) {
    130: #if defined(EOVERFLOW)
    131:     if (errno == EOVERFLOW) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"EOVERFLOW in stat(), configure PETSc --with-large-file-io=1 to support files larger than 2GiB");
    132: #endif
    133:     PetscInfo1(NULL,"System call stat() failed on file %s\n",fname);
    134:     *exists = PETSC_FALSE;
    135:   } else {
    136:     PetscInfo1(NULL,"System call stat() succeeded on file %s\n",fname);
    137:     *exists   = PETSC_TRUE;
    138:     *fileUid  = statbuf.st_uid;
    139:     *fileGid  = statbuf.st_gid;
    140:     *fileMode = statbuf.st_mode;
    141:   }
    142:   return(0);
    143: }

    147: PetscErrorCode  PetscTestFile(const char fname[], char mode, PetscBool  *flg)
    148: {
    149:   uid_t          fuid;
    150:   gid_t          fgid;
    151:   int            fmode;
    153:   PetscBool      exists;

    156:   *flg = PETSC_FALSE;
    157:   if (!fname) return(0);

    159:   PetscGetFileStat(fname, &fuid, &fgid, &fmode,&exists);
    160:   if (!exists) return(0);
    161:   /* Except for systems that have this broken stat macros (rare), this is the correct way to check for a regular file */
    162:   if (!S_ISREG(fmode)) return(0);

    164:   PetscTestOwnership(fname, mode, fuid, fgid, fmode, flg);
    165:   return(0);
    166: }

    170: PetscErrorCode  PetscTestDirectory(const char fname[],char mode,PetscBool  *flg)
    171: {
    172:   uid_t          fuid;
    173:   gid_t          fgid;
    174:   int            fmode;
    176:   PetscBool      exists;

    179:   *flg = PETSC_FALSE;
    180:   if (!fname) return(0);

    182:   PetscGetFileStat(fname, &fuid, &fgid, &fmode,&exists);
    183:   if (!exists) return(0);
    184:   /* Except for systems that have this broken stat macros (rare), this
    185:      is the correct way to check for a directory */
    186:   if (!S_ISDIR(fmode)) return(0);

    188:   PetscTestOwnership(fname, mode, fuid, fgid, fmode, flg);
    189:   return(0);
    190: }

    194: PetscErrorCode  PetscLs(MPI_Comm comm,const char libname[],char found[],size_t tlen,PetscBool  *flg)
    195: {
    197:   size_t         len;
    198:   char           *f,program[PETSC_MAX_PATH_LEN];
    199:   FILE           *fp;

    202:   PetscStrcpy(program,"ls ");
    203:   PetscStrcat(program,libname);
    204: #if defined(PETSC_HAVE_POPEN)
    205:   PetscPOpen(comm,NULL,program,"r",&fp);
    206: #else
    207:   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine");
    208: #endif
    209:   f = fgets(found,tlen,fp);
    210:   if (f) *flg = PETSC_TRUE;
    211:   else *flg = PETSC_FALSE;
    212:   while (f) {
    213:     PetscStrlen(found,&len);
    214:     f    = fgets(found+len,tlen-len,fp);
    215:   }
    216:   if (*flg) {PetscInfo2(0,"ls on %s gives \n%s\n",libname,found);}
    217: #if defined(PETSC_HAVE_POPEN)
    218:   PetscPClose(comm,fp,NULL);
    219: #else
    220:   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine");
    221: #endif
    222:   return(0);
    223: }