lib/libc/gen/nftw.c

   1 /*      $OpenBSD: nftw.c,v 1.4 2004/07/07 16:05:23 millert Exp $        */
   2
   3 /*
   4  * Copyright (c) 2003, 2004 Todd C. Miller <Todd.Miller@courtesan.com>
   5  *
   6  * Permission to use, copy, modify, and distribute this software for any
   7  * purpose with or without fee is hereby granted, provided that the above
   8  * copyright notice and this permission notice appear in all copies.
   9  *
  10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  17  *
  18  * Sponsored in part by the Defense Advanced Research Projects
  19  * Agency (DARPA) and Air Force Research Laboratory, Air Force
  20  * Materiel Command, USAF, under agreement number F39502-99-1-0512.
  21  *
  22  * $FreeBSD: src/lib/libc/gen/nftw.c,v 1.1 2004/08/24 13:00:55 tjr Exp $
  23  */
  24
  25 #include <sys/types.h>
  26 #include <sys/stat.h>
  27 #include <errno.h>
  28 #include <fts.h>
  29 #include <ftw.h>
  30 #include <limits.h>
  31
  32 int
  33 nftw(const char *path, int (*fn)(const char *, const struct stat *, int,
  34      struct FTW *), int nfds, int ftwflags)
  35 {
  36         char * const paths[2] = { (char *)path, NULL };
  37         struct FTW ftw;
  38         FTSENT *cur;
  39         FTS *ftsp;
  40         int error = 0, ftsflags, fnflag, postorder, sverrno;
  41
  42         /* XXX - nfds is currently unused */
  43         if (nfds < 1 || nfds > OPEN_MAX) {
  44                 errno = EINVAL;
  45                 return (-1);
  46         }
  47
  48         ftsflags = FTS_COMFOLLOW;
  49         if (!(ftwflags & FTW_CHDIR))
  50                 ftsflags |= FTS_NOCHDIR;
  51         if (ftwflags & FTW_MOUNT)
  52                 ftsflags |= FTS_XDEV;
  53         if (ftwflags & FTW_PHYS)
  54                 ftsflags |= FTS_PHYSICAL;
  55         else
  56                 ftsflags |= FTS_LOGICAL;
  57         postorder = (ftwflags & FTW_DEPTH) != 0;
  58         ftsp = fts_open(paths, ftsflags, NULL);
  59         if (ftsp == NULL)
  60                 return (-1);
  61         while ((cur = fts_read(ftsp)) != NULL) {
  62                 switch (cur->fts_info) {
  63                 case FTS_D:
  64                         if (postorder)
  65                                 continue;
  66                         fnflag = FTW_D;
  67                         break;
  68                 case FTS_DNR:
  69                         fnflag = FTW_DNR;
  70                         break;
  71                 case FTS_DP:
  72                         if (!postorder)
  73                                 continue;
  74                         fnflag = FTW_DP;
  75                         break;
  76                 case FTS_F:
  77                 case FTS_DEFAULT:
  78                         fnflag = FTW_F;
  79                         break;
  80                 case FTS_NS:
  81                 case FTS_NSOK:
  82                         fnflag = FTW_NS;
  83                         break;
  84                 case FTS_SL:
  85                         fnflag = FTW_SL;
  86                         break;
  87                 case FTS_SLNONE:
  88                         fnflag = FTW_SLN;
  89                         break;
  90                 case FTS_DC:
  91                         errno = ELOOP;
  92                         /* FALLTHROUGH */
  93                 default:
  94                         error = -1;
  95                         goto done;
  96                 }
  97                 ftw.base = cur->fts_pathlen - cur->fts_namelen;
  98                 ftw.level = cur->fts_level;
  99                 error = fn(cur->fts_path, cur->fts_statp, fnflag, &ftw);
 100                 if (error != 0)
 101                         break;
 102         }
 103 done:
 104         sverrno = errno;
 105         if (fts_close(ftsp) != 0 && error == 0)
 106                 error = -1;
 107         else
 108                 errno = sverrno;
 109         return (error);
 110 }