Annotation of src/usr.bin/grep/grep.c, Revision 1.7
1.7 ! deraadt 1: /* $OpenBSD: grep.c,v 1.6 2003/06/22 22:38:50 deraadt 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>
30: #include <sys/stat.h>
1.1 deraadt 31:
1.3 deraadt 32: #include <err.h>
33: #include <errno.h>
34: #include <getopt.h>
35: #include <regex.h>
1.1 deraadt 36: #include <stdio.h>
37: #include <stdlib.h>
1.3 deraadt 38: #include <string.h>
1.1 deraadt 39: #include <unistd.h>
40:
1.3 deraadt 41: #include "grep.h"
42:
43: /* Flags passed to regcomp() and regexec() */
44: int cflags;
45: int eflags = REG_STARTEND;
46:
47: int matchall; /* shortcut */
48: int patterns, pattern_sz;
49: char **pattern;
50: regex_t *r_pattern;
51:
52: /* For regex errors */
53: char re_error[RE_ERROR_BUF + 1];
54:
55: /* Command-line flags */
56: int Aflag; /* -A x: print x lines trailing each match */
57: int Bflag; /* -B x: print x lines leading each match */
58: int Eflag; /* -E: interpret pattern as extended regexp */
59: int Fflag; /* -F: interpret pattern as list of fixed strings */
60: int Gflag; /* -G: interpret pattern as basic regexp */
61: int Hflag; /* -H: if -R, follow explicitly listed symlinks */
62: int Lflag; /* -L: only show names of files with no matches */
63: int Pflag; /* -P: if -R, no symlinks are followed */
64: int Rflag; /* -R: recursively search directory trees */
65: int Sflag; /* -S: if -R, follow all symlinks */
66: int Vflag; /* -V: display version information */
1.5 deraadt 67: #ifndef NOZ
1.3 deraadt 68: int Zflag; /* -Z: decompress input before processing */
1.5 deraadt 69: #endif
1.3 deraadt 70: int aflag; /* -a: only search ascii files */
71: int bflag; /* -b: show block numbers for each match */
72: int cflag; /* -c: only show a count of matching lines */
73: int hflag; /* -h: don't print filename headers */
74: int iflag; /* -i: ignore case */
75: int lflag; /* -l: only show names of files with matches */
76: int nflag; /* -n: show line numbers in front of matching lines */
77: int oflag; /* -o: always print file name */
78: int qflag; /* -q: quiet mode (don't output anything) */
79: int sflag; /* -s: silent mode (ignore errors) */
80: int vflag; /* -v: only show non-matching lines */
81: int wflag; /* -w: pattern must start and end on word boundaries */
82: int xflag; /* -x: pattern must match entire line */
83:
84: /* Housekeeping */
85: int first; /* flag whether or not this is our fist match */
86: int tail; /* lines left to print */
87: int lead; /* number of lines in leading context queue */
88:
89: char *progname;
90:
91: static void
92: usage(void)
93: {
1.5 deraadt 94: fprintf(stderr,
95: #ifdef NOZ
96: "usage: %s [-[AB] num] [-CEFGHLPRSVabchilnoqsvwx]"
97: #else
98: "usage: %s [-[AB] num] [-CEFGHLPRSVZabchilnoqsvwx]"
99: #endif
1.7 ! deraadt 100: " [-e pattern] [-f file] [file ...]\n", progname);
1.3 deraadt 101: exit(2);
102: }
103:
1.5 deraadt 104: #ifdef NOZ
105: static char *optstr = "0123456789A:B:CEFGHLPSRUVabce:f:hilnoqrsuvwxy";
106: #else
1.3 deraadt 107: static char *optstr = "0123456789A:B:CEFGHLPSRUVZabce:f:hilnoqrsuvwxy";
1.5 deraadt 108: #endif
1.3 deraadt 109:
1.5 deraadt 110: struct option long_options[] =
1.3 deraadt 111: {
112: {"basic-regexp", no_argument, NULL, 'G'},
113: {"extended-regexp", no_argument, NULL, 'E'},
114: {"fixed-strings", no_argument, NULL, 'F'},
115: {"after-context", required_argument, NULL, 'A'},
116: {"before-context", required_argument, NULL, 'B'},
117: {"context", optional_argument, NULL, 'C'},
118: {"version", no_argument, NULL, 'V'},
119: {"byte-offset", no_argument, NULL, 'b'},
120: {"count", no_argument, NULL, 'c'},
121: {"regexp", required_argument, NULL, 'e'},
122: {"file", required_argument, NULL, 'f'},
123: {"no-filename", no_argument, NULL, 'h'},
124: {"ignore-case", no_argument, NULL, 'i'},
125: {"files-without-match", no_argument, NULL, 'L'},
126: {"files-with-matches", no_argument, NULL, 'l'},
127: {"line-number", no_argument, NULL, 'n'},
128: {"quiet", no_argument, NULL, 'q'},
129: {"silent", no_argument, NULL, 'q'},
130: {"recursive", no_argument, NULL, 'r'},
131: {"no-messages", no_argument, NULL, 's'},
132: {"text", no_argument, NULL, 'a'},
133: {"revert-match", no_argument, NULL, 'v'},
134: {"word-regexp", no_argument, NULL, 'w'},
135: {"line-regexp", no_argument, NULL, 'x'},
136: {"binary", no_argument, NULL, 'U'},
137: {"unix-byte-offsets", no_argument, NULL, 'u'},
1.5 deraadt 138: #ifndef NOZ
1.3 deraadt 139: {"decompress", no_argument, NULL, 'Z'},
1.5 deraadt 140: #endif
1.3 deraadt 141: {NULL, no_argument, NULL, 0}
142: };
143:
1.1 deraadt 144:
1.3 deraadt 145: static void
146: add_pattern(char *pat, size_t len)
147: {
148: if (len == 0 || matchall) {
149: matchall = 1;
150: return;
151: }
152: if (patterns == pattern_sz) {
153: pattern_sz *= 2;
154: pattern = grep_realloc(pattern, ++pattern_sz);
155: }
156: if (pat[len-1] == '\n')
157: --len;
158: pattern[patterns] = grep_malloc(len+1);
159: strncpy(pattern[patterns], pat, len);
160: pattern[patterns][len] = '\0';
161: ++patterns;
162: }
1.1 deraadt 163:
1.3 deraadt 164: static void
165: read_patterns(char *fn)
166: {
167: FILE *f;
168: char *line;
169: size_t len;
170: int nl;
171:
172: if ((f = fopen(fn, "r")) == NULL)
173: err(1, "%s", fn);
174: nl = 0;
175: while ((line = fgetln(f, &len)) != NULL) {
176: if (*line == '\n') {
177: ++nl;
178: continue;
179: }
180: if (nl) {
181: matchall = 1;
182: break;
183: }
184: nl = 0;
185: add_pattern(line, len);
186: }
187: if (ferror(f))
188: err(1, "%s", fn);
189: fclose(f);
190: }
1.1 deraadt 191:
192: int
1.3 deraadt 193: main(int argc, char *argv[])
1.1 deraadt 194: {
1.3 deraadt 195: char *tmp;
196: int c, i;
197:
198: if ((progname = strrchr(*argv, '/')) != NULL)
199: ++progname;
200: else
201: progname = *argv;
202:
1.5 deraadt 203: while ((c = getopt_long(argc, argv, optstr,
1.3 deraadt 204: long_options, (int *)NULL)) != -1) {
205: switch (c) {
206: case '0': case '1': case '2': case '3': case '4':
207: case '5': case '6': case '7': case '8': case '9':
208: tmp = argv[optind - 1];
209: if (tmp[0] == '-' && tmp[1] == c && !tmp[2])
210: Aflag = Bflag = strtol(++tmp, (char **)NULL, 10);
211: else
212: Aflag = Bflag = strtol(argv[optind] + 1, (char **)NULL, 10);
213: break;
214: case 'A':
215: Aflag = strtol(optarg, (char **)NULL, 10);
216: break;
217: case 'B':
218: Bflag = strtol(optarg, (char **)NULL, 10);
219: break;
220: case 'C':
1.5 deraadt 221: if (optarg == NULL)
1.3 deraadt 222: Aflag = Bflag = 2;
223: else
224: Aflag = Bflag = strtol(optarg, (char **)NULL, 10);
1.1 deraadt 225: break;
226: case 'E':
1.3 deraadt 227: Eflag++;
1.1 deraadt 228: break;
229: case 'F':
1.3 deraadt 230: Fflag++;
231: break;
232: case 'G':
233: Gflag++;
1.1 deraadt 234: break;
235: case 'H':
1.3 deraadt 236: Hflag++;
1.1 deraadt 237: break;
238: case 'L':
1.3 deraadt 239: lflag = 0;
240: Lflag = qflag = 1;
1.1 deraadt 241: break;
242: case 'P':
1.3 deraadt 243: Pflag++;
244: break;
245: case 'S':
246: Sflag++;
1.1 deraadt 247: break;
248: case 'R':
1.3 deraadt 249: case 'r':
250: Rflag++;
251: oflag++;
252: break;
253: case 'U':
254: case 'u':
255: /* these are here for compatability */
256: break;
257: case 'V':
258: fprintf(stderr, "grep version %u.%u\n", VER_MAJ, VER_MIN);
259: fprintf(stderr, argv[0]);
260: usage();
1.1 deraadt 261: break;
1.5 deraadt 262: #ifndef NOZ
1.3 deraadt 263: case 'Z':
264: Zflag++;
1.1 deraadt 265: break;
1.5 deraadt 266: #endif
1.1 deraadt 267: case 'a':
1.3 deraadt 268: aflag = 1;
1.1 deraadt 269: break;
270: case 'b':
1.3 deraadt 271: bflag = 1;
1.1 deraadt 272: break;
273: case 'c':
1.3 deraadt 274: cflag = 1;
1.1 deraadt 275: break;
276: case 'e':
1.3 deraadt 277: add_pattern(optarg, strlen(optarg));
1.1 deraadt 278: break;
279: case 'f':
1.3 deraadt 280: read_patterns(optarg);
1.1 deraadt 281: break;
282: case 'h':
1.3 deraadt 283: oflag = 0;
284: hflag = 1;
1.1 deraadt 285: break;
286: case 'i':
1.3 deraadt 287: case 'y':
1.1 deraadt 288: cflags |= REG_ICASE;
289: break;
290: case 'l':
1.3 deraadt 291: Lflag = 0;
292: lflag = qflag = 1;
1.1 deraadt 293: break;
294: case 'n':
1.3 deraadt 295: nflag = 1;
296: break;
297: case 'o':
298: hflag = 0;
299: oflag = 1;
1.1 deraadt 300: break;
301: case 'q':
1.3 deraadt 302: qflag = 1;
1.1 deraadt 303: break;
304: case 's':
1.3 deraadt 305: sflag = 1;
1.1 deraadt 306: break;
307: case 'v':
1.3 deraadt 308: vflag = 1;
1.1 deraadt 309: break;
310: case 'w':
1.3 deraadt 311: wflag = 1;
1.1 deraadt 312: break;
313: case 'x':
1.3 deraadt 314: xflag = 1;
1.1 deraadt 315: break;
316: default:
317: usage();
318: }
319: }
320:
1.3 deraadt 321: argc -= optind;
322: argv += optind;
323:
324: if (argc == 0 && patterns == 0)
1.1 deraadt 325: usage();
326:
1.3 deraadt 327: if (patterns == 0) {
328: add_pattern(*argv, strlen(*argv));
329: --argc;
330: ++argv;
331: }
1.5 deraadt 332:
1.3 deraadt 333: switch (*progname) {
334: case 'e':
335: Eflag++;
336: break;
337: case 'f':
338: Fflag++;
339: break;
340: case 'g':
341: Gflag++;
342: break;
1.5 deraadt 343: #ifndef NOZ
1.3 deraadt 344: case 'z':
345: Zflag++;
1.4 deraadt 346: switch(progname[1]) {
347: case 'e':
348: Eflag++;
349: break;
350: case 'f':
351: Fflag++;
352: break;
353: case 'g':
354: Gflag++;
355: break;
356: }
1.3 deraadt 357: break;
1.5 deraadt 358: #endif
1.3 deraadt 359: }
360:
361: cflags |= Eflag ? REG_EXTENDED : REG_BASIC;
362: r_pattern = grep_malloc(patterns * sizeof(regex_t));
363: for (i = 0; i < patterns; ++i) {
364: if ((c = regcomp(&r_pattern[i], pattern[i], cflags))) {
365: regerror(c, &r_pattern[i], re_error, RE_ERROR_BUF);
366: errx(1, "%s", re_error);
1.1 deraadt 367: }
368: }
369:
1.3 deraadt 370: if ((argc == 0 || argc == 1) && !oflag)
371: hflag = 1;
1.1 deraadt 372:
1.3 deraadt 373: if (argc == 0)
374: exit(!procfile(NULL));
1.5 deraadt 375:
1.3 deraadt 376: if (Rflag)
377: c = grep_tree(argv);
1.1 deraadt 378: else
1.3 deraadt 379: for (c = 0; argc--; ++argv)
380: c += procfile(*argv);
1.1 deraadt 381:
1.3 deraadt 382: exit(!c);
1.1 deraadt 383: }