c6aaf7aab4b1f267dfa704c176b4572d01593567
[people/mcb30/busybox.git] / findutils / find.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini find implementation for busybox
4  *
5  * Copyright (C) 1999-2004 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 static const char msg_req_arg[] = "option `%s' requires an argument";
38 static 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_MMIN
57 static char mmin_char;
58 static int mmin_mins;
59 #endif
60
61 #ifdef CONFIG_FEATURE_FIND_XDEV
62 static dev_t *xdev_dev;
63 static int xdev_count = 0;
64 #endif
65
66 #ifdef CONFIG_FEATURE_FIND_NEWER
67 static time_t newer_mtime;
68 #endif
69
70 #ifdef CONFIG_FEATURE_FIND_INUM
71 static ino_t inode_num;
72 #endif
73
74 #ifdef CONFIG_FEATURE_FIND_EXEC
75 static char **exec_str;
76 static int num_matches;
77 static int exec_opt;
78 #endif
79
80 static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
81 {
82         if (pattern != NULL) {
83                 const char *tmp = strrchr(fileName, '/');
84
85                 if (tmp == NULL)
86                         tmp = fileName;
87                 else
88                         tmp++;
89                 if (!(fnmatch(pattern, tmp, FNM_PERIOD) == 0))
90                         goto no_match;
91         }
92 #ifdef CONFIG_FEATURE_FIND_TYPE
93         if (type_mask != 0) {
94                 if (!((statbuf->st_mode & S_IFMT) == type_mask))
95                         goto no_match;
96         }
97 #endif
98 #ifdef CONFIG_FEATURE_FIND_PERM
99         if (perm_mask != 0) {
100                 if (!((isdigit(perm_char) && (statbuf->st_mode & 07777) == perm_mask) ||
101                          (perm_char == '-' && (statbuf->st_mode & perm_mask) == perm_mask) ||
102                          (perm_char == '+' && (statbuf->st_mode & perm_mask) != 0)))
103                         goto no_match;
104         }
105 #endif
106 #ifdef CONFIG_FEATURE_FIND_MTIME
107         if (mtime_char != 0) {
108                 time_t file_age = time(NULL) - statbuf->st_mtime;
109                 time_t mtime_secs = mtime_days * 24 * 60 * 60;
110                 if (!((isdigit(mtime_char) && file_age >= mtime_secs &&
111                                                 file_age < mtime_secs + 24 * 60 * 60) ||
112                                 (mtime_char == '+' && file_age >= mtime_secs + 24 * 60 * 60) ||
113                                 (mtime_char == '-' && file_age < mtime_secs)))
114                         goto no_match;
115         }
116 #endif
117 #ifdef CONFIG_FEATURE_FIND_MMIN
118         if (mmin_char != 0) {
119                 time_t file_age = time(NULL) - statbuf->st_mtime;
120                 time_t mmin_secs = mmin_mins * 60;
121                 if (!((isdigit(mmin_char) && file_age >= mmin_secs &&
122                                                 file_age < mmin_secs + 60) ||
123                                 (mmin_char == '+' && file_age >= mmin_secs + 60) ||
124                                 (mmin_char == '-' && file_age < mmin_secs)))
125                         goto no_match;
126         }
127 #endif
128 #ifdef CONFIG_FEATURE_FIND_XDEV
129         if (xdev_count) {
130                 int i;
131                 for (i=0; i<xdev_count; i++) {
132                         if (xdev_dev[i] == statbuf-> st_dev)
133                                 break;
134                 }
135                 if (i == xdev_count) {
136                         if(S_ISDIR(statbuf->st_mode))
137                                 return SKIP;
138                         else
139                                 goto no_match;
140                 }
141         }
142 #endif
143 #ifdef CONFIG_FEATURE_FIND_NEWER
144         if (newer_mtime != 0) {
145                 time_t file_age = newer_mtime - statbuf->st_mtime;
146                 if (file_age >= 0)
147                         goto no_match;
148         }
149 #endif
150 #ifdef CONFIG_FEATURE_FIND_INUM
151         if (inode_num != 0) {
152                 if (!(statbuf->st_ino == inode_num))
153                         goto no_match;
154         }
155 #endif
156 #ifdef CONFIG_FEATURE_FIND_EXEC
157         if (exec_opt) {
158                 int i;
159                 char *cmd_string = "";
160                 for (i = 0; i < num_matches; i++)
161                         cmd_string = bb_xasprintf("%s%s%s", cmd_string, exec_str[i], fileName);
162                 cmd_string = bb_xasprintf("%s%s", cmd_string, exec_str[num_matches]);
163                 system(cmd_string);
164                 goto no_match;
165         }
166 #endif
167
168         puts(fileName);
169 no_match:
170         return (TRUE);
171 }
172
173 #ifdef CONFIG_FEATURE_FIND_TYPE
174 static int find_type(char *type)
175 {
176         int mask = 0;
177
178         switch (type[0]) {
179                 case 'b':
180                         mask = S_IFBLK;
181                         break;
182                 case 'c':
183                         mask = S_IFCHR;
184                         break;
185                 case 'd':
186                         mask = S_IFDIR;
187                         break;
188                 case 'p':
189                         mask = S_IFIFO;
190                         break;
191                 case 'f':
192                         mask = S_IFREG;
193                         break;
194                 case 'l':
195                         mask = S_IFLNK;
196                         break;
197                 case 's':
198                         mask = S_IFSOCK;
199                         break;
200         }
201
202         if (mask == 0 || type[1] != '\0')
203                 bb_error_msg_and_die(msg_invalid_arg, type, "-type");
204
205         return mask;
206 }
207 #endif
208
209 int find_main(int argc, char **argv)
210 {
211         int dereference = FALSE;
212         int i, firstopt, status = EXIT_SUCCESS;
213
214         for (firstopt = 1; firstopt < argc; firstopt++) {
215                 if (argv[firstopt][0] == '-')
216                         break;
217         }
218
219         /* Parse any options */
220         for (i = firstopt; i < argc; i++) {
221                 if (strcmp(argv[i], "-follow") == 0)
222                         dereference = TRUE;
223                 else if (strcmp(argv[i], "-print") == 0) {
224                         ;
225                         }
226                 else if (strcmp(argv[i], "-name") == 0) {
227                         if (++i == argc)
228                                 bb_error_msg_and_die(msg_req_arg, "-name");
229                         pattern = argv[i];
230 #ifdef CONFIG_FEATURE_FIND_TYPE
231                 } else if (strcmp(argv[i], "-type") == 0) {
232                         if (++i == argc)
233                                 bb_error_msg_and_die(msg_req_arg, "-type");
234                         type_mask = find_type(argv[i]);
235 #endif
236 #ifdef CONFIG_FEATURE_FIND_PERM
237                 } else if (strcmp(argv[i], "-perm") == 0) {
238                         char *end;
239                         if (++i == argc)
240                                 bb_error_msg_and_die(msg_req_arg, "-perm");
241                         perm_mask = strtol(argv[i], &end, 8);
242                         if ((end[0] != '\0') || (perm_mask > 07777))
243                                 bb_error_msg_and_die(msg_invalid_arg, argv[i], "-perm");
244                         if ((perm_char = argv[i][0]) == '-')
245                                 perm_mask = -perm_mask;
246 #endif
247 #ifdef CONFIG_FEATURE_FIND_MTIME
248                 } else if (strcmp(argv[i], "-mtime") == 0) {
249                         char *end;
250                         if (++i == argc)
251                                 bb_error_msg_and_die(msg_req_arg, "-mtime");
252                         mtime_days = strtol(argv[i], &end, 10);
253                         if (end[0] != '\0')
254                                 bb_error_msg_and_die(msg_invalid_arg, argv[i], "-mtime");
255                         if ((mtime_char = argv[i][0]) == '-')
256                                 mtime_days = -mtime_days;
257 #endif
258 #ifdef CONFIG_FEATURE_FIND_MMIN
259                 } else if (strcmp(argv[i], "-mmin") == 0) {
260                         char *end;
261                         if (++i == argc)
262                                 bb_error_msg_and_die(msg_req_arg, "-mmin");
263                         mmin_mins = strtol(argv[i], &end, 10);
264                         if (end[0] != '\0')
265                                 bb_error_msg_and_die(msg_invalid_arg, argv[i], "-mmin");
266                         if ((mmin_char = argv[i][0]) == '-')
267                                 mmin_mins = -mmin_mins;
268 #endif
269 #ifdef CONFIG_FEATURE_FIND_XDEV
270                 } else if (strcmp(argv[i], "-xdev") == 0) {
271                         struct stat stbuf;
272
273                         xdev_count = ( firstopt - 1 ) ? ( firstopt - 1 ) : 1;
274                         xdev_dev = xmalloc ( xdev_count * sizeof( dev_t ));
275
276                         if ( firstopt == 1 ) {
277                                 xstat ( ".", &stbuf );
278                                 xdev_dev [0] = stbuf. st_dev;
279                         }
280                         else {
281
282                                 for (i = 1; i < firstopt; i++) {
283                                         xstat ( argv [i], &stbuf );
284                                         xdev_dev [i-1] = stbuf. st_dev;
285                                 }
286                         }
287 #endif
288 #ifdef CONFIG_FEATURE_FIND_NEWER
289                 } else if (strcmp(argv[i], "-newer") == 0) {
290                         struct stat stat_newer;
291                         if (++i == argc)
292                                 bb_error_msg_and_die(msg_req_arg, "-newer");
293                         xstat (argv[i], &stat_newer);
294                         newer_mtime = stat_newer.st_mtime;
295 #endif
296 #ifdef CONFIG_FEATURE_FIND_INUM
297                 } else if (strcmp(argv[i], "-inum") == 0) {
298                         char *end;
299                         if (++i == argc)
300                                 bb_error_msg_and_die(msg_req_arg, "-inum");
301                         inode_num = strtol(argv[i], &end, 10);
302                         if (end[0] != '\0')
303                                 bb_error_msg_and_die(msg_invalid_arg, argv[i], "-inum");
304 #endif
305 #ifdef CONFIG_FEATURE_FIND_EXEC
306                 } else if (strcmp(argv[i], "-exec") == 0) {
307                         int b_pos;
308                         char *cmd_string = "";
309
310                         while (i++) {
311                                 if (i == argc)
312                                         bb_error_msg_and_die(msg_req_arg, "-exec");
313                                 if (*argv[i] == ';')
314                                         break;
315                                 cmd_string = bb_xasprintf("%s %s", cmd_string, argv[i]);
316                         }
317
318                         if (*cmd_string == 0)
319                                 bb_error_msg_and_die(msg_req_arg, "-exec");
320                         cmd_string++;
321                         exec_str = xmalloc(sizeof(char *));
322
323                         while ((b_pos = strstr(cmd_string, "{}") - cmd_string), (b_pos >= 0)) {
324                                 num_matches++;
325                                 exec_str = xrealloc(exec_str, (num_matches + 1) * sizeof(char *));
326                                 exec_str[num_matches - 1] = bb_xstrndup(cmd_string, b_pos);
327                                 cmd_string += b_pos + 2;
328                         }
329                         exec_str[num_matches] = bb_xstrdup(cmd_string);
330                         exec_opt = 1;
331 #endif
332                 } else
333                         bb_show_usage();
334         }
335
336         if (firstopt == 1) {
337                 if (! recursive_action(".", TRUE, dereference, FALSE, fileAction,
338                                         fileAction, NULL))
339                         status = EXIT_FAILURE;
340         } else {
341                 for (i = 1; i < firstopt; i++) {
342                         if (! recursive_action(argv[i], TRUE, dereference, FALSE, fileAction,
343                                                 fileAction, NULL))
344                                 status = EXIT_FAILURE;
345                 }
346         }
347
348         return status;
349 }