Patch from Denis Vlasenko to add xstat() and use it.
[oweals/busybox.git] / findutils / find.c
index 6346d454f4a08fcdad7c1a6d45eb908016ebe2f2..c6aaf7aab4b1f267dfa704c176b4572d01593567 100644 (file)
@@ -2,9 +2,10 @@
 /*
  * Mini find implementation for busybox
  *
+ * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
  *
- * Copyright (C) 1999 by Lineo, inc.
- * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.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
  *
  */
 
-#include "internal.h"
-#include "regexp.h"
 #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
+static const char msg_req_arg[] = "option `%s' requires an argument";
+static const char msg_invalid_arg[] = "invalid argument `%s' to `%s'";
 
-static char *pattern = NULL;
-static char *directory = ".";
-static int dereferenceFlag = FALSE;
+static char *pattern;
 
-static const char find_usage[] = "find [PATH...] [EXPRESSION]\n\n"
-       "Search for files in a directory hierarchy.  The default PATH is\n"
-       "the current directory; default EXPRESSION is '-print'\n\n"
-       "\nEXPRESSION may consist of:\n"
-       "\t-follow\n\t\tDereference symbolic links.\n"
-       "\t-name PATTERN\n\t\tFile name (with leading directories removed) matches PATTERN.\n"
-       "\t-print\n\t\tprint the full file name followed by a newline to stdout.\n\n"
-#if defined BB_REGEXP
-       "This version of find matches full regular expresions.\n";
-#else
-       "This version of find matches strings (not regular expresions).\n";
+#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_MMIN
+static char mmin_char;
+static int mmin_mins;
+#endif
+
+#ifdef CONFIG_FEATURE_FIND_XDEV
+static dev_t *xdev_dev;
+static int xdev_count = 0;
+#endif
+
+#ifdef CONFIG_FEATURE_FIND_NEWER
+static time_t newer_mtime;
+#endif
 
+#ifdef CONFIG_FEATURE_FIND_INUM
+static ino_t inode_num;
+#endif
+
+#ifdef CONFIG_FEATURE_FIND_EXEC
+static char **exec_str;
+static int num_matches;
+static int exec_opt;
+#endif
 
-static int fileAction(const char *fileName, struct stat *statbuf)
+static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
 {
-       if (pattern == NULL)
-               fprintf(stdout, "%s\n", fileName);
-       else {
-               char *tmp = strrchr(fileName, '/');
+       if (pattern != NULL) {
+               const char *tmp = strrchr(fileName, '/');
 
                if (tmp == NULL)
-                       tmp = (char *) fileName;
+                       tmp = fileName;
                else
                        tmp++;
-               if (check_wildcard_match(tmp, pattern) == TRUE)
-                       fprintf(stdout, "%s\n", fileName);
+               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_MMIN
+       if (mmin_char != 0) {
+               time_t file_age = time(NULL) - statbuf->st_mtime;
+               time_t mmin_secs = mmin_mins * 60;
+               if (!((isdigit(mmin_char) && file_age >= mmin_secs &&
+                                               file_age < mmin_secs + 60) ||
+                               (mmin_char == '+' && file_age >= mmin_secs + 60) ||
+                               (mmin_char == '-' && file_age < mmin_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
+#ifdef CONFIG_FEATURE_FIND_EXEC
+       if (exec_opt) {
+               int i;
+               char *cmd_string = "";
+               for (i = 0; i < num_matches; i++)
+                       cmd_string = bb_xasprintf("%s%s%s", cmd_string, exec_str[i], fileName);
+               cmd_string = bb_xasprintf("%s%s", cmd_string, exec_str[num_matches]);
+               system(cmd_string);
+               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)
 {
-       /* peel off the "find" */
-       argc--;
-       argv++;
-
-       if (argc > 0 && **argv != '-') {
-               directory = *argv;
-               argc--;
-               argv++;
+       int dereference = FALSE;
+       int i, firstopt, status = EXIT_SUCCESS;
+
+       for (firstopt = 1; firstopt < argc; firstopt++) {
+               if (argv[firstopt][0] == '-')
+                       break;
        }
 
        /* Parse any options */
-       while (argc > 0 && **argv == '-') {
-               int stopit = FALSE;
-
-               while (*++(*argv) && stopit == FALSE)
-                       switch (**argv) {
-                       case 'f':
-                               if (strcmp(*argv, "follow") == 0) {
-                                       argc--;
-                                       argv++;
-                                       dereferenceFlag = TRUE;
-                               }
-                               break;
-                       case 'n':
-                               if (strcmp(*argv, "name") == 0) {
-                                       if (argc-- > 1) {
-                                               pattern = *(++argv);
-                                               stopit = TRUE;
-                                       } else {
-                                               usage(find_usage);
-                                       }
+       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_MMIN
+               } else if (strcmp(argv[i], "-mmin") == 0) {
+                       char *end;
+                       if (++i == argc)
+                               bb_error_msg_and_die(msg_req_arg, "-mmin");
+                       mmin_mins = strtol(argv[i], &end, 10);
+                       if (end[0] != '\0')
+                               bb_error_msg_and_die(msg_invalid_arg, argv[i], "-mmin");
+                       if ((mmin_char = argv[i][0]) == '-')
+                               mmin_mins = -mmin_mins;
+#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 ) {
+                               xstat ( ".", &stbuf );
+                               xdev_dev [0] = stbuf. st_dev;
+                       }
+                       else {
+
+                               for (i = 1; i < firstopt; i++) {
+                                       xstat ( argv [i], &stbuf );
+                                       xdev_dev [i-1] = stbuf. st_dev;
                                }
-                               break;
-                       case '-':
-                               /* Ignore all long options */
-                               break;
-                       default:
-                               usage(find_usage);
                        }
-               if (argc-- > 1)
-                       argv++;
-               if (**argv != '-')
-                       break;
-               else
-                       break;
+#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");
+                       xstat (argv[i], &stat_newer);
+                       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
+#ifdef CONFIG_FEATURE_FIND_EXEC
+               } else if (strcmp(argv[i], "-exec") == 0) {
+                       int b_pos;
+                       char *cmd_string = "";
+
+                       while (i++) {
+                               if (i == argc)
+                                       bb_error_msg_and_die(msg_req_arg, "-exec");
+                               if (*argv[i] == ';')
+                                       break;
+                               cmd_string = bb_xasprintf("%s %s", cmd_string, argv[i]);
+                       }
+
+                       if (*cmd_string == 0)
+                               bb_error_msg_and_die(msg_req_arg, "-exec");
+                       cmd_string++;
+                       exec_str = xmalloc(sizeof(char *));
+
+                       while ((b_pos = strstr(cmd_string, "{}") - cmd_string), (b_pos >= 0)) {
+                               num_matches++;
+                               exec_str = xrealloc(exec_str, (num_matches + 1) * sizeof(char *));
+                               exec_str[num_matches - 1] = bb_xstrndup(cmd_string, b_pos);
+                               cmd_string += b_pos + 2;
+                       }
+                       exec_str[num_matches] = bb_xstrdup(cmd_string);
+                       exec_opt = 1;
+#endif
+               } else
+                       bb_show_usage();
        }
 
-       if (recursiveAction(directory, TRUE, FALSE, FALSE,
-                                               fileAction, fileAction) == FALSE) {
-               exit(FALSE);
+       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;
+               }
        }
 
-       exit(TRUE);
+       return status;
 }