diff options
author | Sascha Hauer <s.hauer@pengutronix.de> | 2007-09-24 17:00:49 +0200 |
---|---|---|
committer | Sascha Hauer <s.hauer@pengutronix.de> | 2007-09-24 17:00:49 +0200 |
commit | ca5ae599d690f3870b10797f57a7ca046b0b9e44 (patch) | |
tree | a6ad2da0782e95526c89a05ba04428a9134a00fc /lib/recursive_action.c | |
parent | ddfa0c174eae92db9576499c4f4eed66765373c8 (diff) | |
download | barebox-ca5ae599d690f3870b10797f57a7ca046b0b9e44.tar.gz barebox-ca5ae599d690f3870b10797f57a7ca046b0b9e44.tar.xz |
add recursive_action() and concat_subpath_file() from busybox
Diffstat (limited to 'lib/recursive_action.c')
-rw-r--r-- | lib/recursive_action.c | 130 |
1 files changed, 130 insertions, 0 deletions
diff --git a/lib/recursive_action.c b/lib/recursive_action.c new file mode 100644 index 0000000000..91b2d21b84 --- /dev/null +++ b/lib/recursive_action.c @@ -0,0 +1,130 @@ +/* vi: set sw=8 ts=8: */ +/* + * Utility routines. + * + * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org> + * + * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. + */ +#ifdef __U_BOOT__ + +#include <common.h> +#include <fs.h> +#include <linux/stat.h> +#include <malloc.h> +#include <libbb.h> + +#endif + +/* + * Walk down all the directories under the specified + * location, and do something (something specified + * by the fileAction and dirAction function pointers). + * + * Unfortunately, while nftw(3) could replace this and reduce + * code size a bit, nftw() wasn't supported before GNU libc 2.1, + * and so isn't sufficiently portable to take over since glibc2.1 + * is so stinking huge. + */ + +static int true_action(const char *fileName, struct stat *statbuf, + void* userData, int depth) +{ + return 1; +} + +/* fileAction return value of 0 on any file in directory will make + * recursive_action() return 0, but it doesn't stop directory traversal + * (fileAction/dirAction will be called on each file). + * + * if !depthFirst, dirAction return value of 0 (FALSE) or 2 (SKIP) + * prevents recursion into that directory, instead + * recursive_action() returns 0 (if FALSE) or 1 (if SKIP). + * + * followLinks=0/1 differs mainly in handling of links to dirs. + * 0: lstat(statbuf). Calls fileAction on link name even if points to dir. + * 1: stat(statbuf). Calls dirAction and optionally recurse on link to dir. + */ + +int recursive_action(const char *fileName, + unsigned flags, + int (*fileAction)(const char *fileName, struct stat *statbuf, void* userData, int depth), + int (*dirAction)(const char *fileName, struct stat *statbuf, void* userData, int depth), + void* userData, + const unsigned depth) +{ + struct stat statbuf; + int status; + DIR *dir; + struct dirent *next; + + if (!fileAction) fileAction = true_action; + if (!dirAction) dirAction = true_action; + status = stat(fileName, &statbuf); + + if (status < 0) { +#ifdef DEBUG_RECURS_ACTION + bb_error_msg("status=%d followLinks=%d TRUE=%d", + status, flags & ACTION_FOLLOWLINKS, TRUE); +#endif + goto done_nak_warn; + } + + /* If S_ISLNK(m), then we know that !S_ISDIR(m). + * Then we can skip checking first part: if it is true, then + * (!dir) is also true! */ + if ( /* (!(flags & ACTION_FOLLOWLINKS) && S_ISLNK(statbuf.st_mode)) || */ + !S_ISDIR(statbuf.st_mode) + ) { + return fileAction(fileName, &statbuf, userData, depth); + } + + /* It's a directory (or a link to one, and followLinks is set) */ + + if (!(flags & ACTION_RECURSE)) { + return dirAction(fileName, &statbuf, userData, depth); + } + + if (!(flags & ACTION_DEPTHFIRST)) { + status = dirAction(fileName, &statbuf, userData, depth); + if (!status) { + goto done_nak_warn; + } + if (status == 2) + return 1; + } + + dir = opendir(fileName); + if (!dir) { + /* findutils-4.1.20 reports this */ + /* (i.e. it doesn't silently return with exit code 1) */ + /* To trigger: "find -exec rm -rf {} \;" */ + goto done_nak_warn; + } + status = 1; + while ((next = readdir(dir)) != NULL) { + char *nextFile; + + nextFile = concat_subpath_file(fileName, next->d_name); + if (nextFile == NULL) + continue; + /* now descend into it, forcing recursion. */ + if (!recursive_action(nextFile, flags | ACTION_RECURSE, + fileAction, dirAction, userData, depth+1)) { + status = 0; + } + free(nextFile); + } + closedir(dir); + if ((flags & ACTION_DEPTHFIRST) && + !dirAction(fileName, &statbuf, userData, depth)) { + goto done_nak_warn; + } + + if (!status) + return 0; + return 1; +done_nak_warn: + printf("%s", fileName); + return 0; +} |