blob: b319e22cec7d999e256ad769a62cdc38f5deff3e [file] [log] [blame]
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +00001/* vi: set sw=4 ts=4: */
Eric Andersencc8ed391999-10-05 16:24:54 +00002/*
Mark Whitleyd3721892000-06-28 22:00:26 +00003 * Mini grep implementation for busybox using libc regex.
Eric Andersenc4996011999-10-20 22:08:37 +00004 *
Eric Andersenbdfd0d72001-10-24 05:00:29 +00005 * Copyright (C) 1999,2000,2001 by Lineo, inc. and Mark Whitley
Eric Andersenc7bda1c2004-03-15 08:29:22 +00006 * Copyright (C) 1999,2000,2001 by Mark Whitley <markw@codepoet.org>
Eric Andersencc8ed391999-10-05 16:24:54 +00007 *
Bernhard Reutner-Fischer7fee0c42006-09-13 16:39:19 +00008 * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
Eric Andersencc8ed391999-10-05 16:24:54 +00009 */
Denis Vlasenko1d426652008-03-17 09:09:09 +000010/* BB_AUDIT SUSv3 defects - unsupported option -x "match whole line only". */
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +000011/* BB_AUDIT GNU defects - always acts as -a. */
12/* http://www.opengroup.org/onlinepubs/007904975/utilities/grep.html */
Eric Andersen8876fb22003-06-20 09:01:58 +000013/*
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +000014 * 2004,2006 (C) Vladimir Oleynik <dzo@simtreas.ru> -
Eric Andersen7f164cd2004-05-26 09:46:41 +000015 * correction "-e pattern1 -e pattern2" logic and more optimizations.
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +000016 * precompiled regex
Denis Vlasenko51937532006-09-29 20:58:53 +000017 */
18/*
19 * (C) 2006 Jac Goudsmit added -o option
20 */
Eric Andersencc8ed391999-10-05 16:24:54 +000021
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000022#include "libbb.h"
"Vladimir N. Oleynik"23f62fc2005-09-14 16:59:11 +000023#include "xregex.h"
Mark Whitleyd3721892000-06-28 22:00:26 +000024
Mark Whitleyd3721892000-06-28 22:00:26 +000025/* options */
Denis Vlasenko385304d2007-02-25 02:38:20 +000026#define OPTSTR_GREP \
Denis Vlasenko4652daa2007-07-15 12:39:08 +000027 "lnqvscFiHhe:f:Lorm:" \
Denis Vlasenko385304d2007-02-25 02:38:20 +000028 USE_FEATURE_GREP_CONTEXT("A:B:C:") \
29 USE_FEATURE_GREP_EGREP_ALIAS("E") \
30 USE_DESKTOP("w") \
Denis Vlasenkof8ea0f32007-02-25 02:38:54 +000031 "aI"
32/* ignored: -a "assume all files to be text" */
33/* ignored: -I "assume binary files have no matches" */
Denis Vlasenko385304d2007-02-25 02:38:20 +000034
35enum {
Denis Vlasenko4652daa2007-07-15 12:39:08 +000036 OPTBIT_l, /* list matched file names only */
37 OPTBIT_n, /* print line# */
38 OPTBIT_q, /* quiet - exit(0) of first match */
39 OPTBIT_v, /* invert the match, to select non-matching lines */
40 OPTBIT_s, /* suppress errors about file open errors */
41 OPTBIT_c, /* count matches per file (suppresses normal output) */
42 OPTBIT_F, /* literal match */
43 OPTBIT_i, /* case-insensitive */
44 OPTBIT_H, /* force filename display */
45 OPTBIT_h, /* inhibit filename display */
46 OPTBIT_e, /* -e PATTERN */
47 OPTBIT_f, /* -f FILE_WITH_PATTERNS */
48 OPTBIT_L, /* list unmatched file names only */
49 OPTBIT_o, /* show only matching parts of lines */
50 OPTBIT_r, /* recurse dirs */
51 OPTBIT_m, /* -m MAX_MATCHES */
52 USE_FEATURE_GREP_CONTEXT( OPTBIT_A ,) /* -A NUM: after-match context */
53 USE_FEATURE_GREP_CONTEXT( OPTBIT_B ,) /* -B NUM: before-match context */
54 USE_FEATURE_GREP_CONTEXT( OPTBIT_C ,) /* -C NUM: -A and -B combined */
55 USE_FEATURE_GREP_EGREP_ALIAS(OPTBIT_E ,) /* extended regexp */
56 USE_DESKTOP( OPTBIT_w ,) /* whole word match */
Denis Vlasenko385304d2007-02-25 02:38:20 +000057 OPT_l = 1 << OPTBIT_l,
58 OPT_n = 1 << OPTBIT_n,
59 OPT_q = 1 << OPTBIT_q,
60 OPT_v = 1 << OPTBIT_v,
61 OPT_s = 1 << OPTBIT_s,
62 OPT_c = 1 << OPTBIT_c,
63 OPT_F = 1 << OPTBIT_F,
64 OPT_i = 1 << OPTBIT_i,
65 OPT_H = 1 << OPTBIT_H,
66 OPT_h = 1 << OPTBIT_h,
67 OPT_e = 1 << OPTBIT_e,
68 OPT_f = 1 << OPTBIT_f,
69 OPT_L = 1 << OPTBIT_L,
70 OPT_o = 1 << OPTBIT_o,
71 OPT_r = 1 << OPTBIT_r,
Denis Vlasenko4652daa2007-07-15 12:39:08 +000072 OPT_m = 1 << OPTBIT_m,
Denis Vlasenko9acfed22007-06-08 15:41:27 +000073 OPT_A = USE_FEATURE_GREP_CONTEXT( (1 << OPTBIT_A)) + 0,
74 OPT_B = USE_FEATURE_GREP_CONTEXT( (1 << OPTBIT_B)) + 0,
75 OPT_C = USE_FEATURE_GREP_CONTEXT( (1 << OPTBIT_C)) + 0,
76 OPT_E = USE_FEATURE_GREP_EGREP_ALIAS((1 << OPTBIT_E)) + 0,
77 OPT_w = USE_DESKTOP( (1 << OPTBIT_w)) + 0,
Denis Vlasenko385304d2007-02-25 02:38:20 +000078};
79
Denis Vlasenko9acfed22007-06-08 15:41:27 +000080#define PRINT_FILES_WITH_MATCHES (option_mask32 & OPT_l)
81#define PRINT_LINE_NUM (option_mask32 & OPT_n)
82#define BE_QUIET (option_mask32 & OPT_q)
83#define SUPPRESS_ERR_MSGS (option_mask32 & OPT_s)
84#define PRINT_MATCH_COUNTS (option_mask32 & OPT_c)
85#define FGREP_FLAG (option_mask32 & OPT_F)
Denis Vlasenko385304d2007-02-25 02:38:20 +000086#define PRINT_FILES_WITHOUT_MATCHES (option_mask32 & OPT_L)
Eric Andersen8876fb22003-06-20 09:01:58 +000087
Denis Vlasenko04ea11b2007-09-10 12:18:32 +000088struct globals {
89 int max_matches;
90 int reflags;
91 smalluint invert_search;
92 smalluint print_filename;
93 smalluint open_errors;
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +000094#if ENABLE_FEATURE_GREP_CONTEXT
Denis Vlasenko04ea11b2007-09-10 12:18:32 +000095 smalluint did_print_line;
96 int lines_before;
97 int lines_after;
98 char **before_buf;
99 int last_line_printed;
100#endif
101 /* globals used internally */
102 llist_t *pattern_head; /* growable list of patterns to match */
103 const char *cur_file; /* the current file we are reading */
104};
105#define G (*(struct globals*)&bb_common_bufsiz1)
106#define INIT_G() \
107 do { \
108 struct G_sizecheck { \
109 char G_sizecheck[sizeof(G) > COMMON_BUFSIZE ? -1 : 1]; \
110 }; \
111 } while (0)
112#define max_matches (G.max_matches )
113#define reflags (G.reflags )
114#define invert_search (G.invert_search )
115#define print_filename (G.print_filename )
116#define open_errors (G.open_errors )
117#define did_print_line (G.did_print_line )
118#define lines_before (G.lines_before )
119#define lines_after (G.lines_after )
120#define before_buf (G.before_buf )
121#define last_line_printed (G.last_line_printed )
122#define pattern_head (G.pattern_head )
123#define cur_file (G.cur_file )
124
Mark Whitleyd3721892000-06-28 22:00:26 +0000125
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000126typedef struct grep_list_data_t {
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000127 char *pattern;
128 regex_t preg;
129#define PATTERN_MEM_A 1
130#define COMPILED 2
131 int flg_mem_alocated_compiled;
132} grep_list_data_t;
Mark Whitleyfa43e542001-05-24 18:36:18 +0000133
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000134
Mark Whitley2fd52982001-02-09 00:41:10 +0000135static void print_line(const char *line, int linenum, char decoration)
136{
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000137#if ENABLE_FEATURE_GREP_CONTEXT
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000138 /* Happens when we go to next file, immediately hit match
139 * and try to print prev context... from prev file! Don't do it */
140 if (linenum < 1)
141 return;
Eric Andersenaff114c2004-04-14 17:51:38 +0000142 /* possibly print the little '--' separator */
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000143 if ((lines_before || lines_after) && did_print_line &&
144 last_line_printed != linenum - 1) {
Mark Whitley2fd52982001-02-09 00:41:10 +0000145 puts("--");
146 }
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000147 /* guard against printing "--" before first line of first file */
148 did_print_line = 1;
Mark Whitley2fd52982001-02-09 00:41:10 +0000149 last_line_printed = linenum;
150#endif
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000151 if (print_filename)
Mark Whitley2fd52982001-02-09 00:41:10 +0000152 printf("%s%c", cur_file, decoration);
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000153 if (PRINT_LINE_NUM)
Mark Whitley2fd52982001-02-09 00:41:10 +0000154 printf("%i%c", linenum, decoration);
Denis Vlasenko51937532006-09-29 20:58:53 +0000155 /* Emulate weird GNU grep behavior with -ov */
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000156 if ((option_mask32 & (OPT_v|OPT_o)) != (OPT_v|OPT_o))
Denis Vlasenko51937532006-09-29 20:58:53 +0000157 puts(line);
Mark Whitley2fd52982001-02-09 00:41:10 +0000158}
Mark Whitleyd3721892000-06-28 22:00:26 +0000159
Eric Andersen8876fb22003-06-20 09:01:58 +0000160static int grep_file(FILE *file)
Mark Whitleyd3721892000-06-28 22:00:26 +0000161{
Eric Andersen8876fb22003-06-20 09:01:58 +0000162 char *line;
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000163 smalluint found;
Mark Whitleyd3721892000-06-28 22:00:26 +0000164 int linenum = 0;
Matt Kraaideb95f62000-08-06 15:25:53 +0000165 int nmatches = 0;
Denis Vlasenko51937532006-09-29 20:58:53 +0000166 regmatch_t regmatch;
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000167#if ENABLE_FEATURE_GREP_CONTEXT
Mark Whitley2fd52982001-02-09 00:41:10 +0000168 int print_n_lines_after = 0;
169 int curpos = 0; /* track where we are in the circular 'before' buffer */
170 int idx = 0; /* used for iteration through the circular buffer */
Denis Vlasenko4652daa2007-07-15 12:39:08 +0000171#else
172 enum { print_n_lines_after = 0 };
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000173#endif /* ENABLE_FEATURE_GREP_CONTEXT */
Mark Whitleyd3721892000-06-28 22:00:26 +0000174
Denis Vlasenko8ee649a2008-03-26 20:04:27 +0000175 while ((line = xmalloc_fgetline(file)) != NULL) {
Glenn L McGrath26df70a2003-04-27 01:50:57 +0000176 llist_t *pattern_ptr = pattern_head;
Denis Vlasenko68102362007-11-04 00:46:03 +0000177 grep_list_data_t *gl = gl; /* for gcc */
Glenn L McGrath26df70a2003-04-27 01:50:57 +0000178
Mark Whitleyd3721892000-06-28 22:00:26 +0000179 linenum++;
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000180 found = 0;
Glenn L McGrath26df70a2003-04-27 01:50:57 +0000181 while (pattern_ptr) {
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000182 gl = (grep_list_data_t *)pattern_ptr->data;
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000183 if (FGREP_FLAG) {
Denis Vlasenkoac074b32007-09-10 12:23:27 +0000184 found |= (strstr(line, gl->pattern) != NULL);
Glenn L McGrath26df70a2003-04-27 01:50:57 +0000185 } else {
Denis Vlasenko6c30db82006-09-29 21:04:12 +0000186 if (!(gl->flg_mem_alocated_compiled & COMPILED)) {
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000187 gl->flg_mem_alocated_compiled |= COMPILED;
188 xregcomp(&(gl->preg), gl->pattern, reflags);
189 }
Denis Vlasenko51937532006-09-29 20:58:53 +0000190 regmatch.rm_so = 0;
191 regmatch.rm_eo = 0;
Denis Vlasenko385304d2007-02-25 02:38:20 +0000192 if (regexec(&(gl->preg), line, 1, &regmatch, 0) == 0) {
193 if (!(option_mask32 & OPT_w))
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000194 found = 1;
Denis Vlasenko385304d2007-02-25 02:38:20 +0000195 else {
196 char c = ' ';
197 if (regmatch.rm_so)
198 c = line[regmatch.rm_so - 1];
199 if (!isalnum(c) && c != '_') {
200 c = line[regmatch.rm_eo];
201 if (!c || (!isalnum(c) && c != '_'))
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000202 found = 1;
Denis Vlasenko385304d2007-02-25 02:38:20 +0000203 }
204 }
205 }
Glenn L McGrath26df70a2003-04-27 01:50:57 +0000206 }
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000207 /* If it's non-inverted search, we can stop
208 * at first match */
209 if (found && !invert_search)
210 goto do_found;
Eric Andersen8876fb22003-06-20 09:01:58 +0000211 pattern_ptr = pattern_ptr->link;
212 } /* while (pattern_ptr) */
Mark Whitleyd3721892000-06-28 22:00:26 +0000213
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000214 if (found ^ invert_search) {
215 do_found:
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000216 /* keep track of matches */
217 nmatches++;
Mark Whitley1d9d4112001-05-21 21:13:00 +0000218
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000219 /* quiet/print (non)matching file names only? */
220 if (option_mask32 & (OPT_q|OPT_l|OPT_L)) {
221 free(line); /* we don't need line anymore */
222 if (BE_QUIET) {
223 /* manpage says about -q:
224 * "exit immediately with zero status
225 * if any match is found,
226 * even if errors were detected" */
227 exit(0);
228 }
229 /* if we're just printing filenames, we stop after the first match */
230 if (PRINT_FILES_WITH_MATCHES) {
231 puts(cur_file);
Bernhard Reutner-Fischera985d302008-02-11 11:44:38 +0000232 /* fall through to "return 1" */
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000233 }
234 /* OPT_L aka PRINT_FILES_WITHOUT_MATCHES: return early */
235 return 1; /* one match */
236 }
Mark Whitley1d9d4112001-05-21 21:13:00 +0000237
Denis Vlasenko4652daa2007-07-15 12:39:08 +0000238#if ENABLE_FEATURE_GREP_CONTEXT
239 /* Were we printing context and saw next (unwanted) match? */
240 if ((option_mask32 & OPT_m) && nmatches > max_matches)
241 break;
242#endif
243
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000244 /* print the matched line */
245 if (PRINT_MATCH_COUNTS == 0) {
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000246#if ENABLE_FEATURE_GREP_CONTEXT
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000247 int prevpos = (curpos == 0) ? lines_before - 1 : curpos - 1;
Mark Whitley2fd52982001-02-09 00:41:10 +0000248
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000249 /* if we were told to print 'before' lines and there is at least
250 * one line in the circular buffer, print them */
251 if (lines_before && before_buf[prevpos] != NULL) {
252 int first_buf_entry_line_num = linenum - lines_before;
Mark Whitley2fd52982001-02-09 00:41:10 +0000253
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000254 /* advance to the first entry in the circular buffer, and
255 * figure out the line number is of the first line in the
256 * buffer */
257 idx = curpos;
258 while (before_buf[idx] == NULL) {
259 idx = (idx + 1) % lines_before;
260 first_buf_entry_line_num++;
Mark Whitley2fd52982001-02-09 00:41:10 +0000261 }
262
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000263 /* now print each line in the buffer, clearing them as we go */
264 while (before_buf[idx] != NULL) {
265 print_line(before_buf[idx], first_buf_entry_line_num, '-');
266 free(before_buf[idx]);
267 before_buf[idx] = NULL;
268 idx = (idx + 1) % lines_before;
269 first_buf_entry_line_num++;
270 }
271 }
272
273 /* make a note that we need to print 'after' lines */
274 print_n_lines_after = lines_after;
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000275#endif
Denis Vlasenko385304d2007-02-25 02:38:20 +0000276 if (option_mask32 & OPT_o) {
Denis Vlasenko68102362007-11-04 00:46:03 +0000277 if (FGREP_FLAG) {
278 /* -Fo just prints the pattern
279 * (unless -v: -Fov doesnt print anything at all) */
280 if (found)
281 print_line(gl->pattern, linenum, ':');
282 } else {
283 line[regmatch.rm_eo] = '\0';
284 print_line(line + regmatch.rm_so, linenum, ':');
285 }
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000286 } else {
287 print_line(line, linenum, ':');
Mark Whitley2fd52982001-02-09 00:41:10 +0000288 }
Matt Kraai0810f722001-01-04 15:11:52 +0000289 }
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000290 }
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000291#if ENABLE_FEATURE_GREP_CONTEXT
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000292 else { /* no match */
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000293 /* if we need to print some context lines after the last match, do so */
Denis Vlasenko4652daa2007-07-15 12:39:08 +0000294 if (print_n_lines_after) {
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000295 print_line(line, linenum, '-');
296 print_n_lines_after--;
297 } else if (lines_before) {
298 /* Add the line to the circular 'before' buffer */
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000299 free(before_buf[curpos]);
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000300 before_buf[curpos] = line;
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000301 curpos = (curpos + 1) % lines_before;
Denis Vlasenko1d426652008-03-17 09:09:09 +0000302 /* avoid free(line) - we took the line */
Denis Vlasenko4652daa2007-07-15 12:39:08 +0000303 line = NULL;
Mark Whitley2fd52982001-02-09 00:41:10 +0000304 }
Denis Vlasenko4222ae42007-02-25 02:37:49 +0000305 }
Mark Whitley2fd52982001-02-09 00:41:10 +0000306
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000307#endif /* ENABLE_FEATURE_GREP_CONTEXT */
Mark Whitleyd3721892000-06-28 22:00:26 +0000308 free(line);
Denis Vlasenko4652daa2007-07-15 12:39:08 +0000309
310 /* Did we print all context after last requested match? */
311 if ((option_mask32 & OPT_m)
312 && !print_n_lines_after && nmatches == max_matches)
313 break;
Mark Whitleyd3721892000-06-28 22:00:26 +0000314 }
Mark Whitley8f122432000-07-18 18:37:01 +0000315
Mark Whitley35e59be2001-05-14 19:40:32 +0000316 /* special-case file post-processing for options where we don't print line
Mark Whitley1d9d4112001-05-21 21:13:00 +0000317 * matches, just filenames and possibly match counts */
Mark Whitley35e59be2001-05-14 19:40:32 +0000318
Mark Whitley1d9d4112001-05-21 21:13:00 +0000319 /* grep -c: print [filename:]count, even if count is zero */
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000320 if (PRINT_MATCH_COUNTS) {
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000321 if (print_filename)
Mark Whitley1d9d4112001-05-21 21:13:00 +0000322 printf("%s:", cur_file);
Denis Vlasenko92758142006-10-03 19:56:34 +0000323 printf("%d\n", nmatches);
Mark Whitley35e59be2001-05-14 19:40:32 +0000324 }
Mark Whitley1d9d4112001-05-21 21:13:00 +0000325
Denis Vlasenko8d5aa872007-07-15 12:38:18 +0000326 /* grep -L: print just the filename */
327 if (PRINT_FILES_WITHOUT_MATCHES) {
328 /* nmatches is zero, no need to check it:
329 * we return 1 early if we detected a match
330 * and PRINT_FILES_WITHOUT_MATCHES is set */
Eric Andersendec7f812004-05-26 11:47:55 +0000331 puts(cur_file);
332 }
333
Eric Andersen8876fb22003-06-20 09:01:58 +0000334 return nmatches;
Mark Whitleyd3721892000-06-28 22:00:26 +0000335}
Eric Andersencc8ed391999-10-05 16:24:54 +0000336
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000337#if ENABLE_FEATURE_CLEAN_UP
338#define new_grep_list_data(p, m) add_grep_list_data(p, m)
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000339static char *add_grep_list_data(char *pattern, int flg_used_mem)
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000340#else
341#define new_grep_list_data(p, m) add_grep_list_data(p)
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000342static char *add_grep_list_data(char *pattern)
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000343#endif
344{
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000345 grep_list_data_t *gl = xzalloc(sizeof(*gl));
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000346 gl->pattern = pattern;
347#if ENABLE_FEATURE_CLEAN_UP
348 gl->flg_mem_alocated_compiled = flg_used_mem;
349#else
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000350 /*gl->flg_mem_alocated_compiled = 0;*/
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000351#endif
352 return (char *)gl;
353}
354
Eric Andersen8876fb22003-06-20 09:01:58 +0000355static void load_regexes_from_file(llist_t *fopt)
Mark Whitleyfa43e542001-05-24 18:36:18 +0000356{
357 char *line;
Eric Andersen8876fb22003-06-20 09:01:58 +0000358 FILE *f;
359
Denis Vlasenko6c30db82006-09-29 21:04:12 +0000360 while (fopt) {
Eric Andersen8876fb22003-06-20 09:01:58 +0000361 llist_t *cur = fopt;
362 char *ffile = cur->data;
363
364 fopt = cur->link;
365 free(cur);
Rob Landleyd921b2e2006-08-03 15:41:12 +0000366 f = xfopen(ffile, "r");
Denis Vlasenko8ee649a2008-03-26 20:04:27 +0000367 while ((line = xmalloc_fgetline(f)) != NULL) {
Rob Landley8bb50782006-05-26 23:44:51 +0000368 llist_add_to(&pattern_head,
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000369 new_grep_list_data(line, PATTERN_MEM_A));
Eric Andersen31c27a92004-10-08 08:10:57 +0000370 }
Mark Whitleyfa43e542001-05-24 18:36:18 +0000371 }
372}
Mark Whitleyfa43e542001-05-24 18:36:18 +0000373
Denis Vlasenko1d426652008-03-17 09:09:09 +0000374static int file_action_grep(const char *filename,
375 struct stat *statbuf ATTRIBUTE_UNUSED,
376 void* matched,
377 int depth ATTRIBUTE_UNUSED)
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000378{
379 FILE *file = fopen(filename, "r");
380 if (file == NULL) {
381 if (!SUPPRESS_ERR_MSGS)
Denis Vlasenko8e8200a2008-01-24 01:30:36 +0000382 bb_simple_perror_msg(filename);
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000383 open_errors = 1;
384 return 0;
385 }
386 cur_file = filename;
387 *(int*)matched += grep_file(file);
Denis Vlasenkobf392162006-10-15 18:38:01 +0000388 fclose(file);
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000389 return 1;
390}
391
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000392static int grep_dir(const char *dir)
393{
394 int matched = 0;
395 recursive_action(dir,
Denis Vlasenkobbd695d2007-04-08 10:52:28 +0000396 /* recurse=yes */ ACTION_RECURSE |
397 /* followLinks=no */
398 /* depthFirst=yes */ ACTION_DEPTHFIRST,
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000399 /* fileAction= */ file_action_grep,
400 /* dirAction= */ NULL,
Denis Vlasenko8c35d652006-10-27 23:42:25 +0000401 /* userData= */ &matched,
402 /* depth= */ 0);
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000403 return matched;
404}
405
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000406int grep_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Rob Landleydfba7412006-03-06 20:47:33 +0000407int grep_main(int argc, char **argv)
Eric Andersencc8ed391999-10-05 16:24:54 +0000408{
Eric Andersen8876fb22003-06-20 09:01:58 +0000409 FILE *file;
410 int matched;
Eric Andersen31c27a92004-10-08 08:10:57 +0000411 llist_t *fopt = NULL;
Matt Kraai999623e2001-10-29 15:49:03 +0000412
Mark Whitleyd3721892000-06-28 22:00:26 +0000413 /* do normal option parsing */
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000414#if ENABLE_FEATURE_GREP_CONTEXT
Denis Vlasenko1d426652008-03-17 09:09:09 +0000415 int Copt;
Eric Andersen8876fb22003-06-20 09:01:58 +0000416
Denis Vlasenko1d426652008-03-17 09:09:09 +0000417 /* -H unsets -h; -C unsets -A,-B; -e,-f are lists;
418 * -m,-A,-B,-C have numeric param */
419 opt_complementary = "H-h:C-AB:e::f::m+:A+:B+:C+";
Denis Vlasenkofe7cd642007-08-18 15:32:12 +0000420 getopt32(argv,
Denis Vlasenko385304d2007-02-25 02:38:20 +0000421 OPTSTR_GREP,
Denis Vlasenko1d426652008-03-17 09:09:09 +0000422 &pattern_head, &fopt, &max_matches,
423 &lines_after, &lines_before, &Copt);
Eric Andersen8876fb22003-06-20 09:01:58 +0000424
Denis Vlasenko385304d2007-02-25 02:38:20 +0000425 if (option_mask32 & OPT_C) {
Denis Vlasenko2295b492006-10-22 11:42:51 +0000426 /* -C unsets prev -A and -B, but following -A or -B
427 may override it */
Denis Vlasenko385304d2007-02-25 02:38:20 +0000428 if (!(option_mask32 & OPT_A)) /* not overridden */
Denis Vlasenko1d426652008-03-17 09:09:09 +0000429 lines_after = Copt;
Denis Vlasenko385304d2007-02-25 02:38:20 +0000430 if (!(option_mask32 & OPT_B)) /* not overridden */
Denis Vlasenko1d426652008-03-17 09:09:09 +0000431 lines_before = Copt;
432 //option_mask32 |= OPT_A|OPT_B; /* for parser */
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000433 }
Denis Vlasenko2295b492006-10-22 11:42:51 +0000434 /* sanity checks */
Denis Vlasenko385304d2007-02-25 02:38:20 +0000435 if (option_mask32 & (OPT_c|OPT_q|OPT_l|OPT_L)) {
436 option_mask32 &= ~OPT_n;
Eric Andersen8876fb22003-06-20 09:01:58 +0000437 lines_before = 0;
438 lines_after = 0;
Denis Vlasenko6c30db82006-09-29 21:04:12 +0000439 } else if (lines_before > 0)
Denis Vlasenko4cccc032006-12-22 18:37:07 +0000440 before_buf = xzalloc(lines_before * sizeof(char *));
Eric Andersen8876fb22003-06-20 09:01:58 +0000441#else
442 /* with auto sanity checks */
Denis Vlasenko1d426652008-03-17 09:09:09 +0000443 /* -H unsets -h; -c,-q or -l unset -n; -e,-f are lists; -m N */
444 opt_complementary = "H-h:c-n:q-n:l-n:e::f::m+";
Denis Vlasenkofe7cd642007-08-18 15:32:12 +0000445 getopt32(argv, OPTSTR_GREP,
Denis Vlasenko1d426652008-03-17 09:09:09 +0000446 &pattern_head, &fopt, &max_matches);
Eric Andersen8876fb22003-06-20 09:01:58 +0000447#endif
Denis Vlasenko385304d2007-02-25 02:38:20 +0000448 invert_search = ((option_mask32 & OPT_v) != 0); /* 0 | 1 */
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000449
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000450 if (pattern_head != NULL) {
Bernhard Reutner-Fischerfebe3c42007-04-04 20:52:03 +0000451 /* convert char **argv to grep_list_data_t */
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000452 llist_t *cur;
453
Denis Vlasenko6c30db82006-09-29 21:04:12 +0000454 for (cur = pattern_head; cur; cur = cur->link)
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000455 cur->data = new_grep_list_data(cur->data, 0);
456 }
Denis Vlasenko385304d2007-02-25 02:38:20 +0000457 if (option_mask32 & OPT_f)
Eric Andersen8876fb22003-06-20 09:01:58 +0000458 load_regexes_from_file(fopt);
459
Denis Vlasenko8f8f2682006-10-03 21:00:43 +0000460 if (ENABLE_FEATURE_GREP_FGREP_ALIAS && applet_name[0] == 'f')
Denis Vlasenko385304d2007-02-25 02:38:20 +0000461 option_mask32 |= OPT_F;
Mike Frysinger15ca5862005-07-31 22:41:05 +0000462
Denis Vlasenko385304d2007-02-25 02:38:20 +0000463 if (!(option_mask32 & (OPT_o | OPT_w)))
Denis Vlasenko51937532006-09-29 20:58:53 +0000464 reflags = REG_NOSUB;
465
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000466 if (ENABLE_FEATURE_GREP_EGREP_ALIAS
467 && (applet_name[0] == 'e' || (option_mask32 & OPT_E))
468 ) {
Denis Vlasenko51937532006-09-29 20:58:53 +0000469 reflags |= REG_EXTENDED;
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000470 }
Eric Andersen8876fb22003-06-20 09:01:58 +0000471
Denis Vlasenko385304d2007-02-25 02:38:20 +0000472 if (option_mask32 & OPT_i)
Eric Andersen8876fb22003-06-20 09:01:58 +0000473 reflags |= REG_ICASE;
474
475 argv += optind;
476 argc -= optind;
Eric Andersen053b1462000-06-13 06:24:53 +0000477
Denis Vlasenko1d426652008-03-17 09:09:09 +0000478 /* if we didn't get a pattern from -e and no command file was specified,
479 * first parameter should be the pattern. no pattern, no worky */
Glenn L McGrath26df70a2003-04-27 01:50:57 +0000480 if (pattern_head == NULL) {
Denis Vlasenko385304d2007-02-25 02:38:20 +0000481 char *pattern;
Eric Andersen8876fb22003-06-20 09:01:58 +0000482 if (*argv == NULL)
Manuel Novoa III cad53642003-03-19 09:13:01 +0000483 bb_show_usage();
Denis Vlasenko385304d2007-02-25 02:38:20 +0000484 pattern = new_grep_list_data(*argv++, 0);
485 llist_add_to(&pattern_head, pattern);
486 argc--;
Mark Whitleyfa43e542001-05-24 18:36:18 +0000487 }
Mark Whitleyd3721892000-06-28 22:00:26 +0000488
Mark Whitleyfa43e542001-05-24 18:36:18 +0000489 /* argv[(optind)..(argc-1)] should be names of file to grep through. If
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000490 * there is more than one file to grep, we will print the filenames. */
Denis Vlasenko2295b492006-10-22 11:42:51 +0000491 if (argc > 1)
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000492 print_filename = 1;
Denis Vlasenko2295b492006-10-22 11:42:51 +0000493 /* -H / -h of course override */
Denis Vlasenko385304d2007-02-25 02:38:20 +0000494 if (option_mask32 & OPT_H)
Denis Vlasenko2295b492006-10-22 11:42:51 +0000495 print_filename = 1;
Denis Vlasenko385304d2007-02-25 02:38:20 +0000496 if (option_mask32 & OPT_h)
Denis Vlasenko2295b492006-10-22 11:42:51 +0000497 print_filename = 0;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000498
Mark Whitley2e1148b2000-06-28 22:59:30 +0000499 /* If no files were specified, or '-' was specified, take input from
500 * stdin. Otherwise, we grep through all the files specified. */
Eric Andersen8876fb22003-06-20 09:01:58 +0000501 matched = 0;
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000502 do {
Eric Andersen8876fb22003-06-20 09:01:58 +0000503 cur_file = *argv++;
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000504 file = stdin;
Denis Vlasenko1d426652008-03-17 09:09:09 +0000505 if (!cur_file || LONE_DASH(cur_file)) {
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000506 cur_file = "(standard input)";
Eric Andersen8876fb22003-06-20 09:01:58 +0000507 } else {
Denis Vlasenko385304d2007-02-25 02:38:20 +0000508 if (option_mask32 & OPT_r) {
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000509 struct stat st;
510 if (stat(cur_file, &st) == 0 && S_ISDIR(st.st_mode)) {
Denis Vlasenko385304d2007-02-25 02:38:20 +0000511 if (!(option_mask32 & OPT_h))
Denis Vlasenko3544ae62006-10-14 14:51:59 +0000512 print_filename = 1;
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000513 matched += grep_dir(cur_file);
514 goto grep_done;
515 }
Mark Whitley2ef880b2000-07-18 21:02:06 +0000516 }
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000517 /* else: fopen(dir) will succeed, but reading won't */
518 file = fopen(cur_file, "r");
519 if (file == NULL) {
520 if (!SUPPRESS_ERR_MSGS)
Denis Vlasenko0c97c9d2007-10-01 11:58:38 +0000521 bb_simple_perror_msg(cur_file);
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000522 open_errors = 1;
523 continue;
524 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000525 }
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000526 matched += grep_file(file);
Denis Vlasenkoddec5af2006-10-26 23:25:17 +0000527 fclose_if_not_stdin(file);
Denis Vlasenko4652daa2007-07-15 12:39:08 +0000528 grep_done: ;
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000529 } while (--argc > 0);
Eric Andersen8876fb22003-06-20 09:01:58 +0000530
Eric Andersen8876fb22003-06-20 09:01:58 +0000531 /* destroy all the elments in the pattern list */
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000532 if (ENABLE_FEATURE_CLEAN_UP) {
533 while (pattern_head) {
534 llist_t *pattern_head_ptr = pattern_head;
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000535 grep_list_data_t *gl = (grep_list_data_t *)pattern_head_ptr->data;
Eric Andersen8876fb22003-06-20 09:01:58 +0000536
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000537 pattern_head = pattern_head->link;
Denis Vlasenko6c30db82006-09-29 21:04:12 +0000538 if ((gl->flg_mem_alocated_compiled & PATTERN_MEM_A))
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000539 free(gl->pattern);
Denis Vlasenko6c30db82006-09-29 21:04:12 +0000540 if ((gl->flg_mem_alocated_compiled & COMPILED))
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000541 regfree(&(gl->preg));
Mike Frysinger3a62a732007-04-12 18:29:27 +0000542 free(gl);
"Vladimir N. Oleynik"716bbe92006-02-28 10:10:19 +0000543 free(pattern_head_ptr);
544 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000545 }
Bernhard Reutner-Fischeraaf0e232005-09-23 15:38:49 +0000546 /* 0 = success, 1 = failed, 2 = error */
Denis Vlasenko3a6755f2006-10-14 14:24:30 +0000547 if (open_errors)
"Vladimir N. Oleynik"bf449742005-09-23 13:50:24 +0000548 return 2;
Denis Vlasenko04ea11b2007-09-10 12:18:32 +0000549 return !matched; /* invert return value: 0 = success, 1 = failed */
Eric Andersencc8ed391999-10-05 16:24:54 +0000550}