Annotation of src/usr.bin/grep/grep.c, Revision 1.19
1.19 ! deraadt 1: /* $OpenBSD: grep.c,v 1.18 2003/07/10 16:49:12 millert Exp $ */
1.6 deraadt 2:
1.1 deraadt 3: /*-
1.3 deraadt 4: * Copyright (c) 1999 James Howard and Dag-Erling Coïdan Smørgrav
5: * All rights reserved.
1.1 deraadt 6: *
7: * Redistribution and use in source and binary forms, with or without
8: * modification, are permitted provided that the following conditions
9: * are met:
10: * 1. Redistributions of source code must retain the above copyright
11: * notice, this list of conditions and the following disclaimer.
12: * 2. Redistributions in binary form must reproduce the above copyright
13: * notice, this list of conditions and the following disclaimer in the
14: * documentation and/or other materials provided with the distribution.
15: *
16: * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17: * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19: * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22: * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23: * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26: * SUCH DAMAGE.
27: */
28:
1.3 deraadt 29: #include <sys/types.h>
1.9 tedu 30: #include <sys/limits.h>
1.3 deraadt 31: #include <sys/stat.h>
1.1 deraadt 32:
1.18 millert 33: #include <ctype.h>
1.3 deraadt 34: #include <err.h>
35: #include <errno.h>
36: #include <getopt.h>
37: #include <regex.h>
1.1 deraadt 38: #include <stdio.h>
39: #include <stdlib.h>
1.3 deraadt 40: #include <string.h>
1.1 deraadt 41: #include <unistd.h>
42:
1.3 deraadt 43: #include "grep.h"
44:
45: /* Flags passed to regcomp() and regexec() */
46: int cflags;
47: int eflags = REG_STARTEND;
48:
49: int matchall; /* shortcut */
50: int patterns, pattern_sz;
51: char **pattern;
52: regex_t *r_pattern;
1.13 tedu 53: fastgrep_t *fg_pattern;
1.3 deraadt 54:
55: /* For regex errors */
56: char re_error[RE_ERROR_BUF + 1];
57:
58: /* Command-line flags */
59: int Aflag; /* -A x: print x lines trailing each match */
60: int Bflag; /* -B x: print x lines leading each match */
61: int Eflag; /* -E: interpret pattern as extended regexp */
62: int Fflag; /* -F: interpret pattern as list of fixed strings */
63: int Gflag; /* -G: interpret pattern as basic regexp */
64: int Hflag; /* -H: if -R, follow explicitly listed symlinks */
65: int Lflag; /* -L: only show names of files with no matches */
66: int Pflag; /* -P: if -R, no symlinks are followed */
67: int Rflag; /* -R: recursively search directory trees */
68: int Sflag; /* -S: if -R, follow all symlinks */
69: int Vflag; /* -V: display version information */
1.5 deraadt 70: #ifndef NOZ
1.3 deraadt 71: int Zflag; /* -Z: decompress input before processing */
1.5 deraadt 72: #endif
1.3 deraadt 73: int bflag; /* -b: show block numbers for each match */
74: int cflag; /* -c: only show a count of matching lines */
75: int hflag; /* -h: don't print filename headers */
76: int iflag; /* -i: ignore case */
77: int lflag; /* -l: only show names of files with matches */
78: int nflag; /* -n: show line numbers in front of matching lines */
79: int oflag; /* -o: always print file name */
80: int qflag; /* -q: quiet mode (don't output anything) */
81: int sflag; /* -s: silent mode (ignore errors) */
82: int vflag; /* -v: only show non-matching lines */
83: int wflag; /* -w: pattern must start and end on word boundaries */
84: int xflag; /* -x: pattern must match entire line */
85:
1.9 tedu 86: int binbehave = BIN_FILE_BIN;
87:
88: enum {
89: BIN_OPT = CHAR_MAX + 1,
90: HELP_OPT,
91: MMAP_OPT
92: };
93:
1.3 deraadt 94: /* Housekeeping */
1.10 tedu 95: int first; /* flag whether or not this is our first match */
1.3 deraadt 96: int tail; /* lines left to print */
97: int lead; /* number of lines in leading context queue */
1.13 tedu 98: int boleol; /* At least one pattern has a bol or eol */
1.17 millert 99: size_t maxPatternLen; /* Longest length of all patterns */
1.3 deraadt 100:
1.8 tedu 101: extern char *__progname;
1.3 deraadt 102:
103: static void
104: usage(void)
105: {
1.5 deraadt 106: fprintf(stderr,
107: #ifdef NOZ
1.9 tedu 108: "usage: %s [-[AB] num] [-CEFGHILPRSUVabchilnoqsvwx]"
1.5 deraadt 109: #else
1.9 tedu 110: "usage: %s [-[AB] num] [-CEFGHILPRSUVZabchilnoqsvwx]"
1.5 deraadt 111: #endif
1.8 tedu 112: " [-e pattern] [-f file] [file ...]\n", __progname);
1.3 deraadt 113: exit(2);
114: }
115:
1.5 deraadt 116: #ifdef NOZ
1.9 tedu 117: static char *optstr = "0123456789A:B:CEFGHILPSRUVabce:f:hilnoqrsuvwxy";
1.5 deraadt 118: #else
1.9 tedu 119: static char *optstr = "0123456789A:B:CEFGHILPSRUVZabce:f:hilnoqrsuvwxy";
1.5 deraadt 120: #endif
1.3 deraadt 121:
1.5 deraadt 122: struct option long_options[] =
1.3 deraadt 123: {
1.19 ! deraadt 124: {"binary-files", required_argument, NULL, BIN_OPT},
! 125: {"help", no_argument, NULL, HELP_OPT},
! 126: {"mmap", no_argument, NULL, MMAP_OPT},
! 127: {"after-context", required_argument, NULL, 'A'},
! 128: {"before-context", required_argument, NULL, 'B'},
! 129: {"context", optional_argument, NULL, 'C'},
! 130: {"devices", required_argument, NULL, 'D'},
! 131: {"extended-regexp", no_argument, NULL, 'E'},
! 132: {"fixed-strings", no_argument, NULL, 'F'},
! 133: {"basic-regexp", no_argument, NULL, 'G'},
! 134: {"binary", no_argument, NULL, 'U'},
! 135: {"version", no_argument, NULL, 'V'},
! 136: {"text", no_argument, NULL, 'a'},
! 137: {"byte-offset", no_argument, NULL, 'b'},
! 138: {"count", no_argument, NULL, 'c'},
! 139: {"regexp", required_argument, NULL, 'e'},
! 140: {"file", required_argument, NULL, 'f'},
! 141: {"no-filename", no_argument, NULL, 'h'},
! 142: {"ignore-case", no_argument, NULL, 'i'},
! 143: {"files-without-match", no_argument, NULL, 'L'},
! 144: {"files-with-matches", no_argument, NULL, 'l'},
! 145: {"line-number", no_argument, NULL, 'n'},
! 146: {"quiet", no_argument, NULL, 'q'},
! 147: {"silent", no_argument, NULL, 'q'},
! 148: {"recursive", no_argument, NULL, 'r'},
! 149: {"no-messages", no_argument, NULL, 's'},
! 150: {"revert-match", no_argument, NULL, 'v'},
! 151: {"word-regexp", no_argument, NULL, 'w'},
! 152: {"line-regexp", no_argument, NULL, 'x'},
! 153: {"unix-byte-offsets", no_argument, NULL, 'u'},
1.5 deraadt 154: #ifndef NOZ
1.19 ! deraadt 155: {"decompress", no_argument, NULL, 'Z'},
1.5 deraadt 156: #endif
1.19 ! deraadt 157: {NULL, no_argument, NULL, 0}
1.3 deraadt 158: };
159:
1.1 deraadt 160:
1.3 deraadt 161: static void
162: add_pattern(char *pat, size_t len)
163: {
164: if (len == 0 || matchall) {
165: matchall = 1;
166: return;
167: }
168: if (patterns == pattern_sz) {
169: pattern_sz *= 2;
1.13 tedu 170: pattern = grep_realloc(pattern, ++pattern_sz * sizeof(int));
1.3 deraadt 171: }
1.11 tedu 172: if (pat[len - 1] == '\n')
1.3 deraadt 173: --len;
1.11 tedu 174: pattern[patterns] = grep_malloc(len + 1);
1.15 tedu 175: /* pat may not be NUL-terminated */
176: memcpy(pattern[patterns], pat, len);
1.14 tedu 177: pattern[patterns][len] = '\0';
1.3 deraadt 178: ++patterns;
1.13 tedu 179:
180: if (len > maxPatternLen)
181: maxPatternLen = len;
1.3 deraadt 182: }
1.1 deraadt 183:
1.3 deraadt 184: static void
185: read_patterns(char *fn)
186: {
187: FILE *f;
188: char *line;
189: size_t len;
190: int nl;
191:
192: if ((f = fopen(fn, "r")) == NULL)
193: err(1, "%s", fn);
194: nl = 0;
195: while ((line = fgetln(f, &len)) != NULL) {
196: if (*line == '\n') {
197: ++nl;
198: continue;
199: }
200: if (nl) {
201: matchall = 1;
202: break;
203: }
204: nl = 0;
205: add_pattern(line, len);
206: }
207: if (ferror(f))
208: err(1, "%s", fn);
209: fclose(f);
210: }
1.1 deraadt 211:
1.13 tedu 212: static void
213: free_patterns(void)
214: {
215: int i;
216:
217: for (i = 0; i < patterns; i++) {
218: if (fg_pattern[i].pattern)
219: free(fg_pattern[i].pattern);
220: else
221: regfree(&r_pattern[i]);
222: free(pattern[i]);
223: }
224:
225: free(fg_pattern);
226: free(r_pattern);
227: free(pattern);
228: }
229:
1.1 deraadt 230: int
1.3 deraadt 231: main(int argc, char *argv[])
1.1 deraadt 232: {
1.18 millert 233: int c, lastc, prevoptind, i;
1.3 deraadt 234:
1.16 tedu 235: switch (__progname[0]) {
236: case 'e':
237: Eflag++;
238: break;
239: case 'f':
240: Fflag++;
241: break;
242: case 'g':
243: Gflag++;
244: break;
245: #ifndef NOZ
246: case 'z':
247: Zflag++;
248: switch(__progname[1]) {
249: case 'e':
250: Eflag++;
251: break;
252: case 'f':
253: Fflag++;
254: break;
255: case 'g':
256: Gflag++;
257: break;
258: }
259: break;
260: #endif
261: }
262:
1.18 millert 263: lastc = '\0';
264: prevoptind = 0;
1.5 deraadt 265: while ((c = getopt_long(argc, argv, optstr,
1.17 millert 266: long_options, NULL)) != -1) {
1.3 deraadt 267: switch (c) {
268: case '0': case '1': case '2': case '3': case '4':
269: case '5': case '6': case '7': case '8': case '9':
1.18 millert 270: if (optind == prevoptind && isdigit(lastc))
271: Aflag = Bflag = (Aflag * 10) + (c - '0');
1.3 deraadt 272: else
1.18 millert 273: Aflag = Bflag = c - '0';
1.3 deraadt 274: break;
275: case 'A':
1.17 millert 276: Aflag = strtol(optarg, NULL, 10);
1.3 deraadt 277: break;
278: case 'B':
1.17 millert 279: Bflag = strtol(optarg, NULL, 10);
1.3 deraadt 280: break;
281: case 'C':
1.5 deraadt 282: if (optarg == NULL)
1.3 deraadt 283: Aflag = Bflag = 2;
284: else
1.17 millert 285: Aflag = Bflag = strtol(optarg, NULL, 10);
1.1 deraadt 286: break;
287: case 'E':
1.16 tedu 288: Fflag = Gflag = 0;
1.3 deraadt 289: Eflag++;
1.1 deraadt 290: break;
291: case 'F':
1.16 tedu 292: Eflag = Gflag = 0;
1.3 deraadt 293: Fflag++;
294: break;
295: case 'G':
1.16 tedu 296: Eflag = Fflag = 0;
1.3 deraadt 297: Gflag++;
1.1 deraadt 298: break;
299: case 'H':
1.3 deraadt 300: Hflag++;
1.1 deraadt 301: break;
1.9 tedu 302: case 'I':
303: binbehave = BIN_FILE_SKIP;
304: break;
1.1 deraadt 305: case 'L':
1.3 deraadt 306: lflag = 0;
307: Lflag = qflag = 1;
1.1 deraadt 308: break;
309: case 'P':
1.3 deraadt 310: Pflag++;
311: break;
312: case 'S':
313: Sflag++;
1.1 deraadt 314: break;
315: case 'R':
1.3 deraadt 316: case 'r':
317: Rflag++;
318: oflag++;
319: break;
320: case 'U':
1.9 tedu 321: binbehave = BIN_FILE_BIN;
1.3 deraadt 322: break;
323: case 'V':
324: fprintf(stderr, "grep version %u.%u\n", VER_MAJ, VER_MIN);
1.9 tedu 325: exit(0);
1.1 deraadt 326: break;
1.5 deraadt 327: #ifndef NOZ
1.3 deraadt 328: case 'Z':
329: Zflag++;
1.1 deraadt 330: break;
1.5 deraadt 331: #endif
1.1 deraadt 332: case 'a':
1.9 tedu 333: binbehave = BIN_FILE_TEXT;
1.1 deraadt 334: break;
335: case 'b':
1.3 deraadt 336: bflag = 1;
1.1 deraadt 337: break;
338: case 'c':
1.3 deraadt 339: cflag = 1;
1.1 deraadt 340: break;
341: case 'e':
1.3 deraadt 342: add_pattern(optarg, strlen(optarg));
1.1 deraadt 343: break;
344: case 'f':
1.3 deraadt 345: read_patterns(optarg);
1.1 deraadt 346: break;
347: case 'h':
1.3 deraadt 348: oflag = 0;
349: hflag = 1;
1.1 deraadt 350: break;
351: case 'i':
1.3 deraadt 352: case 'y':
1.13 tedu 353: iflag = 1;
1.1 deraadt 354: cflags |= REG_ICASE;
355: break;
356: case 'l':
1.3 deraadt 357: Lflag = 0;
358: lflag = qflag = 1;
1.1 deraadt 359: break;
360: case 'n':
1.3 deraadt 361: nflag = 1;
362: break;
363: case 'o':
364: hflag = 0;
365: oflag = 1;
1.1 deraadt 366: break;
367: case 'q':
1.3 deraadt 368: qflag = 1;
1.1 deraadt 369: break;
370: case 's':
1.3 deraadt 371: sflag = 1;
1.1 deraadt 372: break;
373: case 'v':
1.3 deraadt 374: vflag = 1;
1.1 deraadt 375: break;
376: case 'w':
1.3 deraadt 377: wflag = 1;
1.1 deraadt 378: break;
379: case 'x':
1.3 deraadt 380: xflag = 1;
1.1 deraadt 381: break;
1.9 tedu 382: case BIN_OPT:
383: if (strcmp("binary", optarg) == 0)
384: binbehave = BIN_FILE_BIN;
385: else if (strcmp("without-match", optarg) == 0)
386: binbehave = BIN_FILE_SKIP;
387: else if (strcmp("text", optarg) == 0)
388: binbehave = BIN_FILE_TEXT;
389: else
390: errx(2, "Unknown binary-files option");
391: break;
392: case 'u':
393: case MMAP_OPT:
394: /* default, compatibility */
395: break;
396: case HELP_OPT:
1.1 deraadt 397: default:
398: usage();
399: }
1.18 millert 400: lastc = c;
401: prevoptind = optind;
1.1 deraadt 402: }
403:
1.3 deraadt 404: argc -= optind;
405: argv += optind;
406:
407: if (argc == 0 && patterns == 0)
1.1 deraadt 408: usage();
409:
1.3 deraadt 410: if (patterns == 0) {
411: add_pattern(*argv, strlen(*argv));
412: --argc;
413: ++argv;
414: }
1.5 deraadt 415:
1.16 tedu 416: if (Eflag)
417: cflags |= REG_EXTENDED;
418: else if (Fflag)
419: cflags |= REG_NOSPEC;
1.13 tedu 420: fg_pattern = grep_malloc(patterns * sizeof(*fg_pattern));
1.3 deraadt 421: r_pattern = grep_malloc(patterns * sizeof(regex_t));
422: for (i = 0; i < patterns; ++i) {
1.13 tedu 423: /* Check if cheating is allowed */
424: if (fastcomp(&fg_pattern[i], pattern[i])) {
425: /* Fall back to full regex library */
426: if ((c = regcomp(&r_pattern[i], pattern[i], cflags))) {
427: regerror(c, &r_pattern[i], re_error,
428: RE_ERROR_BUF);
429: errx(1, "%s", re_error);
430: }
1.1 deraadt 431: }
432: }
433:
1.3 deraadt 434: if ((argc == 0 || argc == 1) && !oflag)
435: hflag = 1;
1.1 deraadt 436:
1.3 deraadt 437: if (argc == 0)
438: exit(!procfile(NULL));
1.5 deraadt 439:
1.3 deraadt 440: if (Rflag)
441: c = grep_tree(argv);
1.1 deraadt 442: else
1.3 deraadt 443: for (c = 0; argc--; ++argv)
444: c += procfile(*argv);
1.13 tedu 445:
446: free_patterns();
1.1 deraadt 447:
1.3 deraadt 448: exit(!c);
1.1 deraadt 449: }