add recursive_action() and concat_subpath_file() from busybox

3 files changed, 146 insertions, 0 deletions

diff --git a/lib/Makefile b/lib/Makefile
index 32dafafe9f..2ee3d20c18 100644
--- a/lib/Makefile
+++ b/lib/Makefile
@@ -12,6 +12,8 @@ obj-y			+= getopt.o
 obj-y 			+= readkey.o
 obj-y			+= kfifo.o
 obj-y			+= libbb.o
+obj-y			+= libgen.o
+obj-y			+= recursive_action.o
 obj-$(CONFIG_BZLIB)	+= bzlib.o bzlib_crctable.o bzlib_decompress.o bzlib_huffman.o bzlib_randtable.o
 obj-$(CONFIG_ZLIB)	+= zlib.o gunzip.o
 obj-$(CONFIG_CRC32)	+= crc32.o
diff --git a/lib/libbb.c b/lib/libbb.c
index d2cefd6653..daa484c24b 100644
--- a/lib/libbb.c
+++ b/lib/libbb.c
@@ -35,6 +35,19 @@ char *concat_path_file(const char *path, const char *filename)
 	return str;
 }
 
+/*
+ * This function make special for recursive actions with usage
+ * concat_path_file(path, filename)
+ * and skipping "." and ".." directory entries
+ */
+
+char *concat_subpath_file(const char *path, const char *f)
+{
+	if (f && DOT_OR_DOTDOT(f))
+		return NULL;
+	return concat_path_file(path, f);
+}
+
 /* check if path points to an executable file;
  * return 1 if found;
  * return 0 otherwise;
@@ -85,3 +98,4 @@ char* last_char_is(const char *s, int c)
 	}
 	return NULL;
 }
+
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;
+}