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