gen/ftw.c

   1 /*      $OpenBSD: ftw.c,v 1.2 2003/07/21 21:15:32 millert Exp $ */
   2
   3 /*
   4  * Copyright (c) 2003 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
  23 #if defined(LIBC_SCCS) && !defined(lint)
  24 static const char rcsid[] = "$OpenBSD: ftw.c,v 1.2 2003/07/21 21:15:32 millert Exp $";
  25 #endif /* LIBC_SCCS and not lint */
  26
  27 #include <sys/types.h>
  28 #include <sys/stat.h>
  29 #include <errno.h>
  30 #include <fts.h>
  31 #include <ftw.h>
  32 #include <limits.h>
  33
  34 int
  35 ftw(const char *path, int (*fn)(const char *, const struct stat *, int),
  36     int nfds)
  37 {
  38         const char *paths[2];
  39         FTSENT *cur;
  40         FTS *ftsp;
  41         int fnflag, error, sverrno;
  42
  43         /* XXX - nfds is currently unused */
  44         if (nfds < 1 || nfds > OPEN_MAX) {
  45                 errno = EINVAL;
  46                 return (-1);
  47         }
  48
  49         paths[0] = path;
  50         paths[1] = NULL;
  51         ftsp = fts_open((char * const *)paths, FTS_COMFOLLOW | FTS_NOCHDIR,
  52             NULL);
  53         if (ftsp == NULL)
  54                 return (-1);
  55         error = 0;
  56         while ((cur = fts_read(ftsp)) != NULL) {
  57                 switch (cur->fts_info) {
  58                 case FTS_D:
  59                         fnflag = FTW_D;
  60                         break;
  61                 case FTS_DNR:
  62                         fnflag = FTW_DNR;
  63                         break;
  64                 case FTS_DP:
  65                         /* we only visit in preorder */
  66                         continue;
  67                 case FTS_F:
  68                 case FTS_DEFAULT:
  69                         fnflag = FTW_F;
  70                         break;
  71                 case FTS_NS:
  72                 case FTS_NSOK:
  73                 case FTS_SLNONE:
  74                         fnflag = FTW_NS;
  75                         break;
  76                 case FTS_SL:
  77                         fnflag = FTW_SL;
  78                         break;
  79                 case FTS_DC:
  80                         errno = ELOOP;
  81                         /* FALLTHROUGH */
  82                 default:
  83                         error = -1;
  84                         goto done;
  85                 }
  86                 error = fn(cur->fts_path, cur->fts_statp, fnflag);
  87                 if (error != 0)
  88                         break;
  89         }
  90 done:
  91         sverrno = errno;
  92         (void) fts_close(ftsp);
  93         errno = sverrno;
  94         return (error);
  95 }