[BACK]Return to grep.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / grep

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: }