Actual source code: ftest.c

      1: #define PETSC_DLL

     3:  #include petsc.h
     4:  #include petscsys.h
      5: #if defined(PETSC_HAVE_PWD_H)
      6: #include <pwd.h>
      7: #endif
      8: #include <ctype.h>
      9: #include <sys/types.h>
     10: #include <sys/stat.h>
     11: #if defined(PETSC_HAVE_UNISTD_H)
     12: #include <unistd.h>
     13: #endif
     14: #if defined(PETSC_HAVE_STDLIB_H)
     15: #include <stdlib.h>
     16: #endif
     17: #if defined(PETSC_HAVE_SYS_UTSNAME_H)
     18: #include <sys/utsname.h>
     19: #endif
     20: #if defined(PETSC_HAVE_IO_H)
     21: #include <io.h>
     22: #endif
     23: #if defined(PETSC_HAVE_SYS_SYSTEMINFO_H)
     24: #include <sys/systeminfo.h>
     25: #endif
     26: #include "petscfix.h"

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

     32: static PetscErrorCode PetscTestOwnership(const char fname[], char mode, uid_t fuid, gid_t fgid, int fmode, PetscTruth *flg)
     33: {
     34:   int m = R_OK;
     35:
     37:   if (mode == 'r') m = R_OK;
     38:   else if (mode == 'w') m = W_OK;
     39:   else if (mode == 'x') m = X_OK;
     40:   else SETERRQ(PETSC_ERR_ARG_WRONG, "Mode must be one of r, w, or x");
     41: #if defined(PETSC_HAVE_ACCESS)
     42:   if(!access(fname, m))  *flg = PETSC_TRUE;
     43: #else
     44:   if (m == X_OK) SETERRQ1(PETSC_ERR_SUP, "Unable to check execute permission for file %s", fname);
     45:   if(!_access(fname, m)) *flg = PETSC_TRUE;
     46: #endif
     47:   return(0);
     48: }

     50: #else  /* PETSC_HAVE_ACCESS or PETSC_HAVE__ACCESS */

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

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

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

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

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

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

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

    111: #endif /* PETSC_HAVE_ACCESS */

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

    121: #if defined(PETSC_HAVE_STAT_NO_CONST)
    122:   stat((char*) fname, &statbuf);
    123: #else
    124:   stat(fname, &statbuf);
    125: #endif
    126:   if (ierr) {
    127:     *exists = PETSC_FALSE;
    128:   } else {
    129:     *exists = PETSC_TRUE;
    130:     *fileUid  = statbuf.st_uid;
    131:     *fileGid  = statbuf.st_gid;
    132:     *fileMode = statbuf.st_mode;
    133:   }
    134:   return(0);
    135: }

    139: PetscErrorCode PETSC_DLLEXPORT PetscTestFile(const char fname[], char mode, PetscTruth *flg)
    140: {
    141:   uid_t          fuid;
    142:   gid_t          fgid;
    143:   int            fmode;
    145:   PetscTruth     exists;

    148:   *flg = PETSC_FALSE;
    149:   if (!fname) return(0);

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

    157:   PetscTestOwnership(fname, mode, fuid, fgid, fmode, flg);
    158:   return(0);
    159: }

    163: PetscErrorCode PETSC_DLLEXPORT PetscTestDirectory(const char fname[],char mode,PetscTruth *flg)
    164: {
    165:   uid_t          fuid;
    166:   gid_t          fgid;
    167:   int            fmode;
    169:   PetscTruth     exists;

    172:   *flg = PETSC_FALSE;
    173:   if (!fname) return(0);

    175:   PetscGetFileStat(fname, &fuid, &fgid, &fmode,&exists);
    176:   if (!exists) return(0);
    177:   /* Except for systems that have this broken stat macros (rare), this
    178:      is the correct way to check for a directory */
    179:   if (!S_ISDIR(fmode)) return(0);

    181:   PetscTestOwnership(fname, mode, fuid, fgid, fmode, flg);
    182:   return(0);
    183: }

    187: PetscErrorCode PETSC_DLLEXPORT PetscLs(MPI_Comm comm,const char libname[],char *found,size_t tlen,PetscTruth *flg)
    188: {
    190:   size_t         len;
    191:   char           *f,program[PETSC_MAX_PATH_LEN];
    192:   FILE           *fp;

    195:   PetscStrcpy(program,"ls ");
    196:   PetscStrcat(program,libname);
    197: #if defined(PETSC_HAVE_POPEN)
    198:   PetscPOpen(comm,PETSC_NULL,program,"r",&fp);
    199: #else
    200:   SETERRQ(PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine");
    201: #endif
    202:   f      = fgets(found,tlen,fp);
    203:   if (f) *flg = PETSC_TRUE; else *flg = PETSC_FALSE;
    204:   while (f) {
    205:     PetscStrlen(found,&len);
    206:     f     = fgets(found+len,tlen-len,fp);
    207:   }
    208:   if (*flg) {PetscLogInfo((0,"PetscLS:ls on %s gives \n%s\n",libname,found));}
    209: #if defined(PETSC_HAVE_POPEN)
    210:   PetscPClose(comm,fp);
    211: #else
    212:   SETERRQ(PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine");
    213: #endif
    214:   return(0);
    215: }