libbb/recursive_action.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
   6  *
   7  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
   8  */
   9
  10 #include <stdio.h>
  11 #include <string.h>
  12 #include <dirent.h>
  13 #include <sys/stat.h>
  14 #include <stdlib.h>     /* free() */
  15 #include "libbb.h"
  16
  17 #undef DEBUG_RECURS_ACTION
  18
  19
  20 /*
  21  * Walk down all the directories under the specified
  22  * location, and do something (something specified
  23  * by the fileAction and dirAction function pointers).
  24  *
  25  * Unfortunately, while nftw(3) could replace this and reduce
  26  * code size a bit, nftw() wasn't supported before GNU libc 2.1,
  27  * and so isn't sufficiently portable to take over since glibc2.1
  28  * is so stinking huge.
  29  */
  30 int recursive_action(const char *fileName,
  31                                         int recurse, int followLinks, int depthFirst,
  32                                         int (*fileAction) (const char *fileName,
  33                                                                            struct stat * statbuf,
  34                                                                            void* userData),
  35                                         int (*dirAction) (const char *fileName,
  36                                                                           struct stat * statbuf,
  37                                                                           void* userData),
  38                                         void* userData)
  39 {
  40         int status;
  41         struct stat statbuf;
  42         struct dirent *next;
  43
  44         if (followLinks)
  45                 status = stat(fileName, &statbuf);
  46         else
  47                 status = lstat(fileName, &statbuf);
  48
  49         if (status < 0) {
  50 #ifdef DEBUG_RECURS_ACTION
  51                 bb_error_msg("status=%d followLinks=%d TRUE=%d",
  52                                 status, followLinks, TRUE);
  53 #endif
  54                 bb_perror_msg("%s", fileName);
  55                 return FALSE;
  56         }
  57
  58         if (! followLinks && (S_ISLNK(statbuf.st_mode))) {
  59                 if (fileAction == NULL)
  60                         return TRUE;
  61                 else
  62                         return fileAction(fileName, &statbuf, userData);
  63         }
  64
  65         if (! recurse) {
  66                 if (S_ISDIR(statbuf.st_mode)) {
  67                         if (dirAction != NULL)
  68                                 return (dirAction(fileName, &statbuf, userData));
  69                         else
  70                                 return TRUE;
  71                 }
  72         }
  73
  74         if (S_ISDIR(statbuf.st_mode)) {
  75                 DIR *dir;
  76
  77                 if (dirAction != NULL && ! depthFirst) {
  78                         status = dirAction(fileName, &statbuf, userData);
  79                         if (! status) {
  80                                 bb_perror_msg("%s", fileName);
  81                                 return FALSE;
  82                         } else if (status == SKIP)
  83                                 return TRUE;
  84                 }
  85                 dir = bb_opendir(fileName);
  86                 if (!dir) {
  87                         return FALSE;
  88                 }
  89                 status = TRUE;
  90                 while ((next = readdir(dir)) != NULL) {
  91                         char *nextFile;
  92
  93                         nextFile = concat_subpath_file(fileName, next->d_name);
  94                         if(nextFile == NULL)
  95                                 continue;
  96                         if (! recursive_action(nextFile, TRUE, followLinks, depthFirst,
  97                                                 fileAction, dirAction, userData)) {
  98                                 status = FALSE;
  99                         }
 100                         free(nextFile);
 101                 }
 102                 closedir(dir);
 103                 if (dirAction != NULL && depthFirst) {
 104                         if (! dirAction(fileName, &statbuf, userData)) {
 105                                 bb_perror_msg("%s", fileName);
 106                                 return FALSE;
 107                         }
 108                 }
 109                 if (! status)
 110                         return FALSE;
 111         } else {
 112                 if (fileAction == NULL)
 113                         return TRUE;
 114                 else
 115                         return fileAction(fileName, &statbuf, userData);
 116         }
 117         return TRUE;
 118 }