applying fix for:
[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 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 static 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 }