Annotation of researchv9/cmd/grep.c, revision 1.1

1.1     ! root        1: /*
        !             2:  * grep -- print lines matching (or not matching) a pattern
        !             3:  *
        !             4:  *     status returns:
        !             5:  *             0 - ok, and some matches
        !             6:  *             1 - ok, but no matches
        !             7:  *             2 - some error
        !             8:  */
        !             9: 
        !            10: #include <stdio.h>
        !            11: #include <ctype.h>
        !            12: #include       <fio.h>
        !            13: 
        !            14: #define        CBRA    1
        !            15: #define        CCHR    2
        !            16: #define        CDOT    4
        !            17: #define        CCL     6
        !            18: #define        NCCL    8
        !            19: #define        CDOL    10
        !            20: #define        CEOF    11
        !            21: #define        CKET    12
        !            22: #define        CBACK   18
        !            23: 
        !            24: #define        STAR    01
        !            25: 
        !            26: #define        LBSIZE  512
        !            27: #define        ESIZE   256
        !            28: #define        NBRA    9
        !            29: 
        !            30: char   expbuf[ESIZE];
        !            31: long   lnum;
        !            32: char   *linebuf;
        !            33: char   ybuf[sizeof(Fbuffer)+1];
        !            34: int    bflag;
        !            35: int    lflag;
        !            36: int    nflag;
        !            37: int    cflag;
        !            38: int    vflag;
        !            39: int    nfile;
        !            40: int    hflag   = 1;
        !            41: int    sflag;
        !            42: int    iflag;
        !            43: int    retcode = 0;
        !            44: int    circf;
        !            45: long   tln;
        !            46: int    nsucc;
        !            47: char   *braslist[NBRA];
        !            48: char   *braelist[NBRA];
        !            49: char   bittab[] = {
        !            50:        1,
        !            51:        2,
        !            52:        4,
        !            53:        8,
        !            54:        16,
        !            55:        32,
        !            56:        64,
        !            57:        128
        !            58: };
        !            59: 
        !            60: main(argc, argv)
        !            61: char **argv;
        !            62: {
        !            63:        while (--argc > 0 && (++argv)[0][0]=='-')
        !            64:                switch (argv[0][1]) {
        !            65: 
        !            66:                case 'i':
        !            67:                        iflag++;
        !            68:                        continue;
        !            69: 
        !            70:                case 'h':
        !            71:                        hflag = 0;
        !            72:                        continue;
        !            73: 
        !            74:                case 's':
        !            75:                        sflag++;
        !            76:                        continue;
        !            77: 
        !            78:                case 'v':
        !            79:                        vflag++;
        !            80:                        continue;
        !            81: 
        !            82:                case 'b':
        !            83:                        bflag++;
        !            84:                        continue;
        !            85: 
        !            86:                case 'l':
        !            87:                        lflag++;
        !            88:                        continue;
        !            89: 
        !            90:                case 'c':
        !            91:                        cflag++;
        !            92:                        continue;
        !            93: 
        !            94:                case 'n':
        !            95:                        nflag++;
        !            96:                        continue;
        !            97: 
        !            98:                case 'e':
        !            99:                        --argc;
        !           100:                        ++argv;
        !           101:                        goto out;
        !           102: 
        !           103:                default:
        !           104:                        errexit("grep: unknown flag %c\n", argv[0][1]);
        !           105:                        continue;
        !           106:                }
        !           107: out:
        !           108:        if (argc<=0)
        !           109:                exit(2);
        !           110:        if (iflag) {
        !           111:                register char *p;
        !           112:                for ( p = *argv; *p; p++ )
        !           113:                        *p = tolower(*p);
        !           114:        }
        !           115:        compile(*argv);
        !           116:        nfile = --argc;
        !           117:        if (argc<=0) {
        !           118:                if (lflag) {
        !           119:                        fprintf(stderr, "panic: lflag\n");
        !           120:                        exit(1);
        !           121:                }
        !           122:                execute("/dev/stdin");
        !           123:        } else while (--argc >= 0) {
        !           124:                argv++;
        !           125:                execute(*argv);
        !           126:        }
        !           127:        exit (retcode != 0 ? retcode : nsucc == 0);
        !           128: }
        !           129: 
        !           130: compile(astr)
        !           131: char *astr;
        !           132: {
        !           133:        register c;
        !           134:        register char *ep, *sp;
        !           135:        char *cstart;
        !           136:        char *lastep;
        !           137:        int cclcnt;
        !           138:        char bracket[NBRA], *bracketp;
        !           139:        int closed;
        !           140:        char numbra;
        !           141:        char neg;
        !           142: 
        !           143:        ep = expbuf;
        !           144:        sp = astr;
        !           145:        lastep = 0;
        !           146:        bracketp = bracket;
        !           147:        closed = numbra = 0;
        !           148:        if (*sp == '^') {
        !           149:                circf++;
        !           150:                sp++;
        !           151:        }
        !           152:        for (;;) {
        !           153:                if (ep >= &expbuf[ESIZE])
        !           154:                        goto cerror;
        !           155:                if ((c = *sp++) != '*')
        !           156:                        lastep = ep;
        !           157:                switch (c) {
        !           158: 
        !           159:                case '\0':
        !           160:                        *ep++ = CEOF;
        !           161:                        return;
        !           162: 
        !           163:                case '.':
        !           164:                        *ep++ = CDOT;
        !           165:                        continue;
        !           166: 
        !           167:                case '*':
        !           168:                        if (lastep==0 || *lastep==CBRA || *lastep==CKET)
        !           169:                                goto defchar;
        !           170:                        *lastep |= STAR;
        !           171:                        continue;
        !           172: 
        !           173:                case '$':
        !           174:                        if (*sp != '\0')
        !           175:                                goto defchar;
        !           176:                        *ep++ = CDOL;
        !           177:                        continue;
        !           178: 
        !           179:                case '[':
        !           180:                        if(&ep[17] >= &expbuf[ESIZE])
        !           181:                                goto cerror;
        !           182:                        *ep++ = CCL;
        !           183:                        neg = 0;
        !           184:                        if((c = *sp++) == '^') {
        !           185:                                neg = 1;
        !           186:                                c = *sp++;
        !           187:                        }
        !           188:                        cstart = sp;
        !           189:                        do {
        !           190:                                if (c=='\0')
        !           191:                                        goto cerror;
        !           192:                                if (c=='-' && sp>cstart && *sp!=']') {
        !           193:                                        for (c = sp[-2]; c<*sp; c++)
        !           194:                                                ep[c>>3] |= bittab[c&07];
        !           195:                                        sp++;
        !           196:                                }
        !           197:                                ep[c>>3] |= bittab[c&07];
        !           198:                        } while((c = *sp++) != ']');
        !           199:                        if(neg) {
        !           200:                                for(cclcnt = 0; cclcnt < 16; cclcnt++)
        !           201:                                        ep[cclcnt] ^= -1;
        !           202:                                ep[0] &= 0376;
        !           203:                        }
        !           204: 
        !           205:                        ep += 16;
        !           206: 
        !           207:                        continue;
        !           208: 
        !           209:                case '\\':
        !           210:                        if((c = *sp++) == '(') {
        !           211:                                if(numbra >= NBRA) {
        !           212:                                        goto cerror;
        !           213:                                }
        !           214:                                *bracketp++ = numbra;
        !           215:                                *ep++ = CBRA;
        !           216:                                *ep++ = numbra++;
        !           217:                                continue;
        !           218:                        }
        !           219:                        if(c == ')') {
        !           220:                                if(bracketp <= bracket) {
        !           221:                                        goto cerror;
        !           222:                                }
        !           223:                                *ep++ = CKET;
        !           224:                                *ep++ = *--bracketp;
        !           225:                                closed++;
        !           226:                                continue;
        !           227:                        }
        !           228: 
        !           229:                        if(c >= '1' && c <= '9') {
        !           230:                                if((c -= '1') >= closed)
        !           231:                                        goto cerror;
        !           232:                                *ep++ = CBACK;
        !           233:                                *ep++ = c;
        !           234:                                continue;
        !           235:                        }
        !           236: 
        !           237:                defchar:
        !           238:                default:
        !           239:                        *ep++ = CCHR;
        !           240:                        *ep++ = c;
        !           241:                }
        !           242:        }
        !           243:     cerror:
        !           244:        errexit("grep: RE error\n", (char *)NULL);
        !           245: }
        !           246: 
        !           247: execute(file)
        !           248: char *file;
        !           249: {
        !           250:        register char *p1, *p2;
        !           251:        register c;
        !           252:        register fd;
        !           253: 
        !           254:        if ((fd = open(file, 0)) < 0) {
        !           255:                fprintf(stderr, "grep: can't open %s\n", file);
        !           256:                retcode = 2;
        !           257:                return;
        !           258:        }
        !           259:        Finit(fd, (char *)0);
        !           260:        lnum = 0;
        !           261:        tln = 0;
        !           262:        for (;;) {
        !           263:                if (tln && lflag) {
        !           264:                        close(fd);
        !           265:                        return;
        !           266:                }
        !           267:                lnum++;
        !           268:                if((p1 = linebuf = Frdline(fd)) == 0) {
        !           269:                        if (cflag) {
        !           270:                                if (nfile>1)
        !           271:                                        printf("%s:", file);
        !           272:                                printf("%ld\n", tln);
        !           273:                        }
        !           274:                        close(fd);
        !           275:                        return;
        !           276:                }
        !           277:                if (iflag) {
        !           278:                        char *s = p1;
        !           279:                        char *t = ybuf;
        !           280:                        do {
        !           281:                                *t++ = tolower(*s);
        !           282:                        } while (*s++);
        !           283:                        p1 = ybuf;
        !           284:                }
        !           285:                p2 = expbuf;
        !           286:                if (circf) {
        !           287:                        if (advance(p1, p2))
        !           288:                                goto found;
        !           289:                        goto nfound;
        !           290:                }
        !           291:                /* fast check for first character */
        !           292:                if (*p2==CCHR) {
        !           293:                        c = p2[1];
        !           294:                        do {
        !           295:                                if (*p1!=c)
        !           296:                                        continue;
        !           297:                                if (advance(p1, p2))
        !           298:                                        goto found;
        !           299:                        } while (*p1++);
        !           300:                        goto nfound;
        !           301:                }
        !           302:                /* regular algorithm */
        !           303:                do {
        !           304:                        if (advance(p1, p2))
        !           305:                                goto found;
        !           306:                } while (*p1++);
        !           307:        nfound:
        !           308:                if (vflag)
        !           309:                        succeed(file, fd);
        !           310:                continue;
        !           311:        found:
        !           312:                if (vflag==0)
        !           313:                        succeed(file, fd);
        !           314:        }
        !           315: }
        !           316: 
        !           317: advance(lp, ep)
        !           318: register char *lp, *ep;
        !           319: {
        !           320:        register char *curlp;
        !           321:        char c;
        !           322:        char *bbeg;
        !           323:        int ct;
        !           324: 
        !           325:        for (;;) switch (*ep++) {
        !           326: 
        !           327:        case CCHR:
        !           328:                if (*ep++ == *lp++)
        !           329:                        continue;
        !           330:                return(0);
        !           331: 
        !           332:        case CDOT:
        !           333:                if (*lp++)
        !           334:                        continue;
        !           335:                return(0);
        !           336: 
        !           337:        case CDOL:
        !           338:                if (*lp==0)
        !           339:                        continue;
        !           340:                return(0);
        !           341: 
        !           342:        case CEOF:
        !           343:                return(1);
        !           344: 
        !           345:        case CCL:
        !           346:                c = *lp++ & 0177;
        !           347:                if(ep[c>>3] & bittab[c & 07]) {
        !           348:                        ep += 16;
        !           349:                        continue;
        !           350:                }
        !           351:                return(0);
        !           352:        case CBRA:
        !           353:                braslist[*ep++] = lp;
        !           354:                continue;
        !           355: 
        !           356:        case CKET:
        !           357:                braelist[*ep++] = lp;
        !           358:                continue;
        !           359: 
        !           360:        case CBACK:
        !           361:                bbeg = braslist[*ep];
        !           362:                if (braelist[*ep]==0)
        !           363:                        return(0);
        !           364:                ct = braelist[*ep++] - bbeg;
        !           365:                if(ecmp(bbeg, lp, ct)) {
        !           366:                        lp += ct;
        !           367:                        continue;
        !           368:                }
        !           369:                return(0);
        !           370: 
        !           371:        case CBACK|STAR:
        !           372:                bbeg = braslist[*ep];
        !           373:                if (braelist[*ep]==0)
        !           374:                        return(0);
        !           375:                ct = braelist[*ep++] - bbeg;
        !           376:                curlp = lp;
        !           377:                while(ecmp(bbeg, lp, ct))
        !           378:                        lp += ct;
        !           379:                while(lp >= curlp) {
        !           380:                        if(advance(lp, ep))     return(1);
        !           381:                        lp -= ct;
        !           382:                }
        !           383:                return(0);
        !           384: 
        !           385: 
        !           386:        case CDOT|STAR:
        !           387:                curlp = lp;
        !           388:                while (*lp++);
        !           389:                goto star;
        !           390: 
        !           391:        case CCHR|STAR:
        !           392:                curlp = lp;
        !           393:                while (*lp++ == *ep);
        !           394:                ep++;
        !           395:                goto star;
        !           396: 
        !           397:        case CCL|STAR:
        !           398:                curlp = lp;
        !           399:                do {
        !           400:                        c = *lp++ & 0177;
        !           401:                } while(ep[c>>3] & bittab[c & 07]);
        !           402:                ep += 16;
        !           403:                goto star;
        !           404: 
        !           405:        star:
        !           406:                if(--lp == curlp) {
        !           407:                        continue;
        !           408:                }
        !           409: 
        !           410:                if(*ep == CCHR) {
        !           411:                        c = ep[1];
        !           412:                        do {
        !           413:                                if(*lp != c)
        !           414:                                        continue;
        !           415:                                if(advance(lp, ep))
        !           416:                                        return(1);
        !           417:                        } while(lp-- > curlp);
        !           418:                        return(0);
        !           419:                }
        !           420: 
        !           421:                do {
        !           422:                        if (advance(lp, ep))
        !           423:                                return(1);
        !           424:                } while (lp-- > curlp);
        !           425:                return(0);
        !           426: 
        !           427:        default:
        !           428:                errexit("grep RE botch\n", (char *)NULL);
        !           429:        }
        !           430: }
        !           431: 
        !           432: succeed(f, fd)
        !           433: char *f;
        !           434: {
        !           435:        long ftell();
        !           436: 
        !           437:        nsucc = 1;
        !           438:        if (sflag)
        !           439:                return;
        !           440:        if (cflag) {
        !           441:                tln++;
        !           442:                return;
        !           443:        }
        !           444:        if (lflag) {
        !           445:                printf("%s\n", f);
        !           446:                tln++;
        !           447:                return;
        !           448:        }
        !           449:        if (nfile > 1 && hflag)
        !           450:                printf("%s:", f);
        !           451:        if (bflag)
        !           452:                printf("%ld:", (FIOSEEK(fd)-1)/1024);
        !           453:        if (nflag)
        !           454:                printf("%ld:", lnum);
        !           455:        printf("%s\n", linebuf);
        !           456: }
        !           457: 
        !           458: ecmp(a, b, count)
        !           459: char   *a, *b;
        !           460: {
        !           461:        register cc = count;
        !           462:        while(cc--)
        !           463:                if(*a++ != *b++)        return(0);
        !           464:        return(1);
        !           465: }
        !           466: 
        !           467: errexit(s, f)
        !           468: char *s, *f;
        !           469: {
        !           470:        fprintf(stderr, s, f);
        !           471:        exit(2);
        !           472: }

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.