281 lines
		
	
	
		
			6.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			281 lines
		
	
	
		
			6.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/* vi: set sw=4 ts=4: */
 | 
						|
/*
 | 
						|
 * Mini find implementation for busybox
 | 
						|
 *
 | 
						|
 * Copyright (C) 1999-2003 by Erik Andersen <andersen@codepoet.org>
 | 
						|
 *
 | 
						|
 * Reworked by David Douthitt <n9ubh@callsign.net> and
 | 
						|
 *  Matt Kraai <kraai@alumni.carnegiemellon.edu>.
 | 
						|
 *
 | 
						|
 * This program is free software; you can redistribute it and/or modify
 | 
						|
 * it under the terms of the GNU General Public License as published by
 | 
						|
 * the Free Software Foundation; either version 2 of the License, or
 | 
						|
 * (at your option) any later version.
 | 
						|
 *
 | 
						|
 * This program is distributed in the hope that it will be useful,
 | 
						|
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 | 
						|
 * General Public License for more details.
 | 
						|
 *
 | 
						|
 * You should have received a copy of the GNU General Public License
 | 
						|
 * along with this program; if not, write to the Free Software
 | 
						|
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 | 
						|
 *
 | 
						|
 */
 | 
						|
 | 
						|
#include <stdio.h>
 | 
						|
#include <unistd.h>
 | 
						|
#include <dirent.h>
 | 
						|
#include <string.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#include <fnmatch.h>
 | 
						|
#include <time.h>
 | 
						|
#include <ctype.h>
 | 
						|
#include "busybox.h"
 | 
						|
 | 
						|
//XXX just found out about libbb/messages.c . maybe move stuff there ? - ghoz
 | 
						|
const char msg_req_arg[] = "option `%s' requires an argument";
 | 
						|
const char msg_invalid_arg[] = "invalid argument `%s' to `%s'";
 | 
						|
 | 
						|
static char *pattern;
 | 
						|
 | 
						|
#ifdef CONFIG_FEATURE_FIND_TYPE
 | 
						|
static int type_mask = 0;
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef CONFIG_FEATURE_FIND_PERM
 | 
						|
static char perm_char = 0;
 | 
						|
static int perm_mask = 0;
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef CONFIG_FEATURE_FIND_MTIME
 | 
						|
static char mtime_char;
 | 
						|
static int mtime_days;
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef CONFIG_FEATURE_FIND_XDEV
 | 
						|
static dev_t *xdev_dev;
 | 
						|
static int xdev_count = 0;
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef CONFIG_FEATURE_FIND_NEWER
 | 
						|
time_t newer_mtime;
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef CONFIG_FEATURE_FIND_INUM
 | 
						|
static ino_t inode_num;
 | 
						|
#endif
 | 
						|
 | 
						|
static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
 | 
						|
{
 | 
						|
	if (pattern != NULL) {
 | 
						|
		const char *tmp = strrchr(fileName, '/');
 | 
						|
 | 
						|
		if (tmp == NULL)
 | 
						|
			tmp = fileName;
 | 
						|
		else
 | 
						|
			tmp++;
 | 
						|
		if (!(fnmatch(pattern, tmp, FNM_PERIOD) == 0))
 | 
						|
			goto no_match;
 | 
						|
	}
 | 
						|
#ifdef CONFIG_FEATURE_FIND_TYPE
 | 
						|
	if (type_mask != 0) {
 | 
						|
		if (!((statbuf->st_mode & S_IFMT) == type_mask))
 | 
						|
			goto no_match;
 | 
						|
	}
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_PERM
 | 
						|
	if (perm_mask != 0) {
 | 
						|
		if (!((isdigit(perm_char) && (statbuf->st_mode & 07777) == perm_mask) ||
 | 
						|
			 (perm_char == '-' && (statbuf->st_mode & perm_mask) == perm_mask) ||
 | 
						|
			 (perm_char == '+' && (statbuf->st_mode & perm_mask) != 0)))
 | 
						|
			goto no_match;
 | 
						|
	}
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_MTIME
 | 
						|
	if (mtime_char != 0) {
 | 
						|
		time_t file_age = time(NULL) - statbuf->st_mtime;
 | 
						|
		time_t mtime_secs = mtime_days * 24 * 60 * 60;
 | 
						|
		if (!((isdigit(mtime_char) && file_age >= mtime_secs &&
 | 
						|
						file_age < mtime_secs + 24 * 60 * 60) ||
 | 
						|
				(mtime_char == '+' && file_age >= mtime_secs + 24 * 60 * 60) || 
 | 
						|
				(mtime_char == '-' && file_age < mtime_secs)))
 | 
						|
			goto no_match;
 | 
						|
	}
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_XDEV
 | 
						|
	if (xdev_count) {
 | 
						|
		int i;
 | 
						|
		for (i=0; i<xdev_count; i++) {
 | 
						|
			if (xdev_dev[i] == statbuf-> st_dev)
 | 
						|
				break;
 | 
						|
		}
 | 
						|
		if (i == xdev_count) {
 | 
						|
			if(S_ISDIR(statbuf->st_mode))
 | 
						|
				return SKIP;
 | 
						|
			else
 | 
						|
				goto no_match;
 | 
						|
		}
 | 
						|
	}
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_NEWER
 | 
						|
	if (newer_mtime != 0) {
 | 
						|
		time_t file_age = newer_mtime - statbuf->st_mtime;
 | 
						|
		if (file_age >= 0)
 | 
						|
			goto no_match;
 | 
						|
	}
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_INUM
 | 
						|
	if (inode_num != 0) {
 | 
						|
		if (!(statbuf->st_ino == inode_num))
 | 
						|
			goto no_match;
 | 
						|
	}
 | 
						|
#endif
 | 
						|
	puts(fileName);
 | 
						|
no_match:
 | 
						|
	return (TRUE);
 | 
						|
}
 | 
						|
 | 
						|
#ifdef CONFIG_FEATURE_FIND_TYPE
 | 
						|
static int find_type(char *type)
 | 
						|
{
 | 
						|
	int mask = 0;
 | 
						|
 | 
						|
	switch (type[0]) {
 | 
						|
		case 'b':
 | 
						|
			mask = S_IFBLK;
 | 
						|
			break;
 | 
						|
		case 'c':
 | 
						|
			mask = S_IFCHR;
 | 
						|
			break;
 | 
						|
		case 'd':
 | 
						|
			mask = S_IFDIR;
 | 
						|
			break;
 | 
						|
		case 'p':
 | 
						|
			mask = S_IFIFO;
 | 
						|
			break;
 | 
						|
		case 'f':
 | 
						|
			mask = S_IFREG;
 | 
						|
			break;
 | 
						|
		case 'l':
 | 
						|
			mask = S_IFLNK;
 | 
						|
			break;
 | 
						|
		case 's':
 | 
						|
			mask = S_IFSOCK;
 | 
						|
			break;
 | 
						|
	}
 | 
						|
 | 
						|
	if (mask == 0 || type[1] != '\0')
 | 
						|
		bb_error_msg_and_die(msg_invalid_arg, type, "-type");
 | 
						|
 | 
						|
	return mask;
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
int find_main(int argc, char **argv)
 | 
						|
{
 | 
						|
	int dereference = FALSE;
 | 
						|
	int i, firstopt, status = EXIT_SUCCESS;
 | 
						|
 | 
						|
	for (firstopt = 1; firstopt < argc; firstopt++) {
 | 
						|
		if (argv[firstopt][0] == '-')
 | 
						|
			break;
 | 
						|
	}
 | 
						|
 | 
						|
	/* Parse any options */
 | 
						|
	for (i = firstopt; i < argc; i++) {
 | 
						|
		if (strcmp(argv[i], "-follow") == 0)
 | 
						|
			dereference = TRUE;
 | 
						|
		else if (strcmp(argv[i], "-print") == 0) {
 | 
						|
			;
 | 
						|
			}
 | 
						|
		else if (strcmp(argv[i], "-name") == 0) {
 | 
						|
			if (++i == argc)
 | 
						|
				bb_error_msg_and_die(msg_req_arg, "-name");
 | 
						|
			pattern = argv[i];
 | 
						|
#ifdef CONFIG_FEATURE_FIND_TYPE
 | 
						|
		} else if (strcmp(argv[i], "-type") == 0) {
 | 
						|
			if (++i == argc)
 | 
						|
				bb_error_msg_and_die(msg_req_arg, "-type");
 | 
						|
			type_mask = find_type(argv[i]);
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_PERM
 | 
						|
		} else if (strcmp(argv[i], "-perm") == 0) {
 | 
						|
			char *end;
 | 
						|
			if (++i == argc)
 | 
						|
				bb_error_msg_and_die(msg_req_arg, "-perm");
 | 
						|
			perm_mask = strtol(argv[i], &end, 8);
 | 
						|
			if ((end[0] != '\0') || (perm_mask > 07777))
 | 
						|
				bb_error_msg_and_die(msg_invalid_arg, argv[i], "-perm");
 | 
						|
			if ((perm_char = argv[i][0]) == '-')
 | 
						|
				perm_mask = -perm_mask;
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_MTIME
 | 
						|
		} else if (strcmp(argv[i], "-mtime") == 0) {
 | 
						|
			char *end;
 | 
						|
			if (++i == argc)
 | 
						|
				bb_error_msg_and_die(msg_req_arg, "-mtime");
 | 
						|
			mtime_days = strtol(argv[i], &end, 10);
 | 
						|
			if (end[0] != '\0')
 | 
						|
				bb_error_msg_and_die(msg_invalid_arg, argv[i], "-mtime");
 | 
						|
			if ((mtime_char = argv[i][0]) == '-')
 | 
						|
				mtime_days = -mtime_days;
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_XDEV
 | 
						|
		} else if (strcmp(argv[i], "-xdev") == 0) {
 | 
						|
			struct stat stbuf;
 | 
						|
 | 
						|
			xdev_count = ( firstopt - 1 ) ? ( firstopt - 1 ) : 1;
 | 
						|
			xdev_dev = xmalloc ( xdev_count * sizeof( dev_t ));
 | 
						|
 | 
						|
			if ( firstopt == 1 ) {
 | 
						|
				if ( stat ( ".", &stbuf ) < 0 )
 | 
						|
					bb_error_msg_and_die("could not stat '.'" );
 | 
						|
				xdev_dev [0] = stbuf. st_dev;
 | 
						|
			}
 | 
						|
			else {
 | 
						|
			
 | 
						|
				for (i = 1; i < firstopt; i++) {
 | 
						|
					if ( stat ( argv [i], &stbuf ) < 0 )
 | 
						|
						bb_error_msg_and_die("could not stat '%s'", argv [i] );
 | 
						|
					xdev_dev [i-1] = stbuf. st_dev;
 | 
						|
				}
 | 
						|
			}						
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_NEWER
 | 
						|
		} else if (strcmp(argv[i], "-newer") == 0) {
 | 
						|
			struct stat stat_newer;
 | 
						|
			if (++i == argc)
 | 
						|
				bb_error_msg_and_die(msg_req_arg, "-newer");
 | 
						|
		    if (stat (argv[i], &stat_newer) != 0)
 | 
						|
				bb_error_msg_and_die("file %s not found", argv[i]);
 | 
						|
			newer_mtime = stat_newer.st_mtime;
 | 
						|
#endif
 | 
						|
#ifdef CONFIG_FEATURE_FIND_INUM
 | 
						|
		} else if (strcmp(argv[i], "-inum") == 0) {
 | 
						|
			char *end;
 | 
						|
			if (++i == argc)
 | 
						|
				bb_error_msg_and_die(msg_req_arg, "-inum");
 | 
						|
			inode_num = strtol(argv[i], &end, 10);
 | 
						|
			if (end[0] != '\0')
 | 
						|
				bb_error_msg_and_die(msg_invalid_arg, argv[i], "-inum");
 | 
						|
#endif
 | 
						|
		} else
 | 
						|
			bb_show_usage();
 | 
						|
	}
 | 
						|
 | 
						|
	if (firstopt == 1) {
 | 
						|
		if (! recursive_action(".", TRUE, dereference, FALSE, fileAction,
 | 
						|
					fileAction, NULL))
 | 
						|
			status = EXIT_FAILURE;
 | 
						|
	} else {
 | 
						|
		for (i = 1; i < firstopt; i++) {
 | 
						|
			if (! recursive_action(argv[i], TRUE, dereference, FALSE, fileAction,
 | 
						|
						fileAction, NULL))
 | 
						|
				status = EXIT_FAILURE;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return status;
 | 
						|
}
 |