findutils/find.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini find implementation for busybox
   4  *
   5  * Copyright (C) 1999-2003 by Erik Andersen <andersen@codepoet.org>
   6  *
   7  * Reworked by David Douthitt <n9ubh@callsign.net> and
   8  *  Matt Kraai <kraai@alumni.carnegiemellon.edu>.
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18  * General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  23  *
  24  */
  25
  26 #include <stdio.h>
  27 #include <unistd.h>
  28 #include <dirent.h>
  29 #include <string.h>
  30 #include <stdlib.h>
  31 #include <fnmatch.h>
  32 #include <time.h>
  33 #include <ctype.h>
  34 #include "busybox.h"
  35
  36 //XXX just found out about libbb/messages.c . maybe move stuff there ? - ghoz
  37 const char msg_req_arg[] = "option `%s' requires an argument";
  38 const char msg_invalid_arg[] = "invalid argument `%s' to `%s'";
  39
  40 static char *pattern;
  41
  42 #ifdef CONFIG_FEATURE_FIND_TYPE
  43 static int type_mask = 0;
  44 #endif
  45
  46 #ifdef CONFIG_FEATURE_FIND_PERM
  47 static char perm_char = 0;
  48 static int perm_mask = 0;
  49 #endif
  50
  51 #ifdef CONFIG_FEATURE_FIND_MTIME
  52 static char mtime_char;
  53 static int mtime_days;
  54 #endif
  55
  56 #ifdef CONFIG_FEATURE_FIND_XDEV
  57 static dev_t *xdev_dev;
  58 static int xdev_count = 0;
  59 #endif
  60
  61 #ifdef CONFIG_FEATURE_FIND_NEWER
  62 time_t newer_mtime;
  63 #endif
  64
  65 #ifdef CONFIG_FEATURE_FIND_INUM
  66 static ino_t inode_num;
  67 #endif
  68
  69 static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
  70 {
  71         if (pattern != NULL) {
  72                 const char *tmp = strrchr(fileName, '/');
  73
  74                 if (tmp == NULL)
  75                         tmp = fileName;
  76                 else
  77                         tmp++;
  78                 if (!(fnmatch(pattern, tmp, FNM_PERIOD) == 0))
  79                         goto no_match;
  80         }
  81 #ifdef CONFIG_FEATURE_FIND_TYPE
  82         if (type_mask != 0) {
  83                 if (!((statbuf->st_mode & S_IFMT) == type_mask))
  84                         goto no_match;
  85         }
  86 #endif
  87 #ifdef CONFIG_FEATURE_FIND_PERM
  88         if (perm_mask != 0) {
  89                 if (!((isdigit(perm_char) && (statbuf->st_mode & 07777) == perm_mask) ||
  90                          (perm_char == '-' && (statbuf->st_mode & perm_mask) == perm_mask) ||
  91                          (perm_char == '+' && (statbuf->st_mode & perm_mask) != 0)))
  92                         goto no_match;
  93         }
  94 #endif
  95 #ifdef CONFIG_FEATURE_FIND_MTIME
  96         if (mtime_char != 0) {
  97                 time_t file_age = time(NULL) - statbuf->st_mtime;
  98                 time_t mtime_secs = mtime_days * 24 * 60 * 60;
  99                 if (!((isdigit(mtime_char) && file_age >= mtime_secs &&
 100                                                 file_age < mtime_secs + 24 * 60 * 60) ||
 101                                 (mtime_char == '+' && file_age >= mtime_secs + 24 * 60 * 60) ||
 102                                 (mtime_char == '-' && file_age < mtime_secs)))
 103                         goto no_match;
 104         }
 105 #endif
 106 #ifdef CONFIG_FEATURE_FIND_XDEV
 107         if (xdev_count) {
 108                 int i;
 109                 for (i=0; i<xdev_count; i++) {
 110                         if (xdev_dev[i] == statbuf-> st_dev)
 111                                 break;
 112                 }
 113                 if (i == xdev_count) {
 114                         if(S_ISDIR(statbuf->st_mode))
 115                                 return SKIP;
 116                         else
 117                                 goto no_match;
 118                 }
 119         }
 120 #endif
 121 #ifdef CONFIG_FEATURE_FIND_NEWER
 122         if (newer_mtime != 0) {
 123                 time_t file_age = newer_mtime - statbuf->st_mtime;
 124                 if (file_age >= 0)
 125                         goto no_match;
 126         }
 127 #endif
 128 #ifdef CONFIG_FEATURE_FIND_INUM
 129         if (inode_num != 0) {
 130                 if (!(statbuf->st_ino == inode_num))
 131                         goto no_match;
 132         }
 133 #endif
 134         puts(fileName);
 135 no_match:
 136         return (TRUE);
 137 }
 138
 139 #ifdef CONFIG_FEATURE_FIND_TYPE
 140 static int find_type(char *type)
 141 {
 142         int mask = 0;
 143
 144         switch (type[0]) {
 145                 case 'b':
 146                         mask = S_IFBLK;
 147                         break;
 148                 case 'c':
 149                         mask = S_IFCHR;
 150                         break;
 151                 case 'd':
 152                         mask = S_IFDIR;
 153                         break;
 154                 case 'p':
 155                         mask = S_IFIFO;
 156                         break;
 157                 case 'f':
 158                         mask = S_IFREG;
 159                         break;
 160                 case 'l':
 161                         mask = S_IFLNK;
 162                         break;
 163                 case 's':
 164                         mask = S_IFSOCK;
 165                         break;
 166         }
 167
 168         if (mask == 0 || type[1] != '\0')
 169                 bb_error_msg_and_die(msg_invalid_arg, type, "-type");
 170
 171         return mask;
 172 }
 173 #endif
 174
 175 int find_main(int argc, char **argv)
 176 {
 177         int dereference = FALSE;
 178         int i, firstopt, status = EXIT_SUCCESS;
 179
 180         for (firstopt = 1; firstopt < argc; firstopt++) {
 181                 if (argv[firstopt][0] == '-')
 182                         break;
 183         }
 184
 185         /* Parse any options */
 186         for (i = firstopt; i < argc; i++) {
 187                 if (strcmp(argv[i], "-follow") == 0)
 188                         dereference = TRUE;
 189                 else if (strcmp(argv[i], "-print") == 0) {
 190                         ;
 191                         }
 192                 else if (strcmp(argv[i], "-name") == 0) {
 193                         if (++i == argc)
 194                                 bb_error_msg_and_die(msg_req_arg, "-name");
 195                         pattern = argv[i];
 196 #ifdef CONFIG_FEATURE_FIND_TYPE
 197                 } else if (strcmp(argv[i], "-type") == 0) {
 198                         if (++i == argc)
 199                                 bb_error_msg_and_die(msg_req_arg, "-type");
 200                         type_mask = find_type(argv[i]);
 201 #endif
 202 #ifdef CONFIG_FEATURE_FIND_PERM
 203                 } else if (strcmp(argv[i], "-perm") == 0) {
 204                         char *end;
 205                         if (++i == argc)
 206                                 bb_error_msg_and_die(msg_req_arg, "-perm");
 207                         perm_mask = strtol(argv[i], &end, 8);
 208                         if ((end[0] != '\0') || (perm_mask > 07777))
 209                                 bb_error_msg_and_die(msg_invalid_arg, argv[i], "-perm");
 210                         if ((perm_char = argv[i][0]) == '-')
 211                                 perm_mask = -perm_mask;
 212 #endif
 213 #ifdef CONFIG_FEATURE_FIND_MTIME
 214                 } else if (strcmp(argv[i], "-mtime") == 0) {
 215                         char *end;
 216                         if (++i == argc)
 217                                 bb_error_msg_and_die(msg_req_arg, "-mtime");
 218                         mtime_days = strtol(argv[i], &end, 10);
 219                         if (end[0] != '\0')
 220                                 bb_error_msg_and_die(msg_invalid_arg, argv[i], "-mtime");
 221                         if ((mtime_char = argv[i][0]) == '-')
 222                                 mtime_days = -mtime_days;
 223 #endif
 224 #ifdef CONFIG_FEATURE_FIND_XDEV
 225                 } else if (strcmp(argv[i], "-xdev") == 0) {
 226                         struct stat stbuf;
 227
 228                         xdev_count = ( firstopt - 1 ) ? ( firstopt - 1 ) : 1;
 229                         xdev_dev = xmalloc ( xdev_count * sizeof( dev_t ));
 230
 231                         if ( firstopt == 1 ) {
 232                                 if ( stat ( ".", &stbuf ) < 0 )
 233                                         bb_error_msg_and_die("could not stat '.'" );
 234                                 xdev_dev [0] = stbuf. st_dev;
 235                         }
 236                         else {
 237
 238                                 for (i = 1; i < firstopt; i++) {
 239                                         if ( stat ( argv [i], &stbuf ) < 0 )
 240                                                 bb_error_msg_and_die("could not stat '%s'", argv [i] );
 241                                         xdev_dev [i-1] = stbuf. st_dev;
 242                                 }
 243                         }
 244 #endif
 245 #ifdef CONFIG_FEATURE_FIND_NEWER
 246                 } else if (strcmp(argv[i], "-newer") == 0) {
 247                         struct stat stat_newer;
 248                         if (++i == argc)
 249                                 bb_error_msg_and_die(msg_req_arg, "-newer");
 250                     if (stat (argv[i], &stat_newer) != 0)
 251                                 bb_error_msg_and_die("file %s not found", argv[i]);
 252                         newer_mtime = stat_newer.st_mtime;
 253 #endif
 254 #ifdef CONFIG_FEATURE_FIND_INUM
 255                 } else if (strcmp(argv[i], "-inum") == 0) {
 256                         char *end;
 257                         if (++i == argc)
 258                                 bb_error_msg_and_die(msg_req_arg, "-inum");
 259                         inode_num = strtol(argv[i], &end, 10);
 260                         if (end[0] != '\0')
 261                                 bb_error_msg_and_die(msg_invalid_arg, argv[i], "-inum");
 262 #endif
 263                 } else
 264                         bb_show_usage();
 265         }
 266
 267         if (firstopt == 1) {
 268                 if (! recursive_action(".", TRUE, dereference, FALSE, fileAction,
 269                                         fileAction, NULL))
 270                         status = EXIT_FAILURE;
 271         } else {
 272                 for (i = 1; i < firstopt; i++) {
 273                         if (! recursive_action(argv[i], TRUE, dereference, FALSE, fileAction,
 274                                                 fileAction, NULL))
 275                                 status = EXIT_FAILURE;
 276                 }
 277         }
 278
 279         return status;
 280 }