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 "libbb.h"
  11
  12 #undef DEBUG_RECURS_ACTION
  13
  14 /*
  15  * Walk down all the directories under the specified
  16  * location, and do something (something specified
  17  * by the fileAction and dirAction function pointers).
  18  *
  19  * Unfortunately, while nftw(3) could replace this and reduce
  20  * code size a bit, nftw() wasn't supported before GNU libc 2.1,
  21  * and so isn't sufficiently portable to take over since glibc2.1
  22  * is so stinking huge.
  23  */
  24
  25 static int true_action(const char *fileName, struct stat *statbuf,
  26                                                 void* userData, int depth)
  27 {
  28         return TRUE;
  29 }
  30
  31 /* fileAction return value of 0 on any file in directory will make
  32  * recursive_action() return 0, but it doesn't stop directory traversal
  33  * (fileAction/dirAction will be called on each file).
  34  *
  35  * if !depthFirst, dirAction return value of 0 (FALSE) or 2 (SKIP)
  36  * prevents recursion into that directory, instead
  37  * recursive_action() returns 0 (if FALSE) or 1 (if SKIP).
  38  *
  39  * followLinks=0/1 differs mainly in handling of links to dirs.
  40  * 0: lstat(statbuf). Calls fileAction on link name even if points to dir.
  41  * 1: stat(statbuf). Calls dirAction and optionally recurse on link to dir.
  42  */
  43
  44 int recursive_action(const char *fileName,
  45                 unsigned flags,
  46                 int (*fileAction)(const char *fileName, struct stat *statbuf, void* userData, int depth),
  47                 int (*dirAction)(const char *fileName, struct stat *statbuf, void* userData, int depth),
  48                 void* userData,
  49                 const unsigned depth)
  50 {
  51         struct stat statbuf;
  52         int status;
  53         DIR *dir;
  54         struct dirent *next;
  55
  56         if (!fileAction) fileAction = true_action;
  57         if (!dirAction) dirAction = true_action;
  58         status = (flags & action_followLinks ? stat : lstat)(fileName, &statbuf);
  59
  60         if (status < 0) {
  61 #ifdef DEBUG_RECURS_ACTION
  62                 bb_error_msg("status=%d followLinks=%d TRUE=%d",
  63                                 status, flags & action_followLinks, TRUE);
  64 #endif
  65                 goto done_nak_warn;
  66         }
  67
  68         /* If S_ISLNK(m), then we know that !S_ISDIR(m).
  69          * Then we can skip checking first part: if it is true, then
  70          * (!dir) is also true! */
  71         if ( /* (!(flags & action_followLinks) && S_ISLNK(statbuf.st_mode)) || */
  72          !S_ISDIR(statbuf.st_mode)
  73         ) {
  74                 return fileAction(fileName, &statbuf, userData, depth);
  75         }
  76
  77         /* It's a directory (or a link to one, and followLinks is set) */
  78
  79         if (!(flags & action_recurse)) {
  80                 return dirAction(fileName, &statbuf, userData, depth);
  81         }
  82
  83         if (!(flags & action_depthFirst)) {
  84                 status = dirAction(fileName, &statbuf, userData, depth);
  85                 if (!status) {
  86                         goto done_nak_warn;
  87                 }
  88                 if (status == SKIP)
  89                         return TRUE;
  90         }
  91
  92         dir = opendir(fileName);
  93         if (!dir) {
  94                 /* findutils-4.1.20 reports this */
  95                 /* (i.e. it doesn't silently return with exit code 1) */
  96                 /* To trigger: "find -exec rm -rf {} \;" */
  97                 goto done_nak_warn;
  98         }
  99         status = TRUE;
 100         while ((next = readdir(dir)) != NULL) {
 101                 char *nextFile;
 102
 103                 nextFile = concat_subpath_file(fileName, next->d_name);
 104                 if (nextFile == NULL)
 105                         continue;
 106                 /* now descend into it, forcing recursion. */
 107                 if (!recursive_action(nextFile, flags | action_recurse,
 108                                 fileAction, dirAction, userData, depth+1)) {
 109                         status = FALSE;
 110                 }
 111                 free(nextFile);
 112         }
 113         closedir(dir);
 114         if (flags & action_depthFirst &&
 115                 !dirAction(fileName, &statbuf, userData, depth)) {
 116                         goto done_nak_warn;
 117         }
 118
 119         if (!status)
 120                 return FALSE;
 121         return TRUE;
 122 done_nak_warn:
 123         bb_perror_msg("%s", fileName);
 124         return FALSE;
 125 }