Annotation of researchv10no/cmd/fgrep.c, revision 1.1.1.1

1.1       root        1: static char *sccsid = "@(#)fgrep.c     4.1 (Berkeley) 10/1/80";
                      2: /*
                      3:  * fgrep -- print all lines containing any of a set of keywords
                      4:  *
                      5:  *     status returns:
                      6:  *             0 - ok, and some matches
                      7:  *             1 - ok, but no matches
                      8:  *             2 - some error
                      9:  */
                     10: /*#define      DOSTATS         /* define this to gather stats */
                     11: #ifdef DOSTATS
                     12: char *statsexpr, *statspat;
                     13: long nlines, nbytes, nfail, ntrans0;
                     14: char statsflags[1024], *statsfptr = statsflags;
                     15: dostats();
                     16: #endif
                     17: 
                     18: #include "stdio.h"
                     19: # include "ctype.h"
                     20: 
                     21: #define        MAXSIZ 6000
                     22: #define QSIZE 400
                     23: struct words {
                     24:        char    inp;
                     25:        char    out;
                     26:        struct  words *nst;
                     27:        struct  words *link;
                     28:        struct  words *fail;
                     29: } w[MAXSIZ], *smax, *q;
                     30: 
                     31: long   lnum;
                     32: int    bflag, cflag, fflag, lflag, nflag, vflag, xflag, iflag;
                     33: int    hflag   = 1;
                     34: int    sflag;
                     35: int    retcode = 0;
                     36: int    nfile;
                     37: long   blkno;
                     38: int    nsucc;
                     39: long   tln;
                     40: FILE   *wordf;
                     41: char   *argptr;
                     42: 
                     43: /*
                     44:  * preserve the interactive feel with stdio
                     45:  */
                     46: #define        Read(f, b, s)   (fflush(stdout), read(f, b, s))
                     47: 
                     48: main(argc, argv)
                     49: char **argv;
                     50: {
                     51:        while (--argc > 0 && (++argv)[0][0]=='-'){
                     52: #ifdef DOSTATS
                     53:                *statsfptr++ = argv[0][1];
                     54: #endif
                     55:                switch (argv[0][1]) {
                     56: 
                     57:                case 's':
                     58:                        sflag++;
                     59:                        continue;
                     60: 
                     61:                case 'h':
                     62:                        hflag = 0;
                     63:                        continue;
                     64: 
                     65:                case 'b':
                     66:                        bflag++;
                     67:                        continue;
                     68: 
                     69:                case 'c':
                     70:                        cflag++;
                     71:                        continue;
                     72: 
                     73:                case 'e':
                     74:                        argc--;
                     75:                        argv++;
                     76:                        goto out;
                     77: 
                     78:                case 'f':
                     79:                        fflag++;
                     80:                        continue;
                     81: 
                     82:                case 'l':
                     83:                        lflag++;
                     84:                        continue;
                     85: 
                     86:                case 'n':
                     87:                        nflag++;
                     88:                        continue;
                     89: 
                     90:                case 'v':
                     91:                        vflag++;
                     92:                        continue;
                     93: 
                     94:                case 'x':
                     95:                        xflag++;
                     96:                        continue;
                     97: 
                     98:                case 'i':
                     99:                        iflag++;
                    100:                        continue;
                    101:                default:
                    102:                        fprintf(stderr, "fgrep: unknown flag\n");
                    103:                        continue;
                    104:                }
                    105:        }
                    106: out:
                    107:        if (argc<=0)
                    108:                exit(2);
                    109:        if (fflag) {
                    110:                wordf = fopen(*argv, "r");
                    111:                if (wordf==NULL) {
                    112:                        fprintf(stderr, "fgrep: can't open %s\n", *argv);
                    113:                        exit(2);
                    114:                }
                    115:        }
                    116:        else argptr = *argv;
                    117:        argc--;
                    118:        argv++;
                    119: 
                    120: #ifdef DOSTATS
                    121:        statsexpr = statspat = (char *)malloc(MAXSIZ);
                    122:        onexit(dostats);
                    123: #endif
                    124:        cgotofn();
                    125:        cfail();
                    126:        nfile = argc;
                    127:        if (argc<=0) {
                    128:                if (lflag) exit(1);
                    129:                execute((char *)NULL);
                    130:        }
                    131:        else while (--argc >= 0) {
                    132:                execute(*argv);
                    133:                argv++;
                    134:        }
                    135:        exit(retcode != 0 ? retcode : nsucc == 0);
                    136: }
                    137: 
                    138: # define ccomp(a,b) (iflag ? lca(a)==lca(b) : a==b)
                    139: # define lca(x) (isupper(x) ? ((x)+'a'-'A') : x)
                    140: execute(file)
                    141: char *file;
                    142: {
                    143:        register struct words *c;
                    144:        register ccount;
                    145:        register char ch;
                    146:        register char *p;
                    147:        char buf[2*BUFSIZ];
                    148:        int f;
                    149:        int failed;
                    150:        char *nlp;
                    151:        if (file) {
                    152:                if ((f = open(file, 0)) < 0) {
                    153:                        fprintf(stderr, "fgrep: can't open %s\n", file);
                    154:                        retcode = 2;
                    155:                        return;
                    156:                }
                    157:        }
                    158:        else f = 0;
                    159:        ccount = 0;
                    160:        failed = 0;
                    161:        lnum = 1;
                    162:        tln = 0;
                    163:        blkno = 0;
                    164:        p = buf;
                    165:        nlp = p;
                    166:        c = w;
                    167:        for (;;) {
                    168:                if (--ccount <= 0) {
                    169:                        if (p == &buf[2*BUFSIZ]) p = buf;
                    170:                        if (p > &buf[BUFSIZ]) {
                    171:                                if ((ccount = Read(f, p, &buf[2*BUFSIZ] - p)) <= 0) break;
                    172:                        }
                    173:                        else if ((ccount = Read(f, p, BUFSIZ)) <= 0) break;
                    174:                        blkno += ccount;
                    175:                }
                    176:                nstate:
                    177:                        if (ccomp(c->inp, *p)) {
                    178: #ifdef DOSTATS
                    179:                                if(c == w) ntrans0++;
                    180: #endif
                    181:                                c = c->nst;
                    182:                        }
                    183:                        else if (c->link != 0) {
                    184:                                c = c->link;
                    185:                                goto nstate;
                    186:                        }
                    187:                        else {
                    188: #ifdef DOSTATS
                    189:                                nfail++;
                    190: #endif
                    191:                                c = c->fail;
                    192:                                failed = 1;
                    193:                                if (c==0) {
                    194:                                        c = w;
                    195:                                        istate:
                    196:                                        if (ccomp(c->inp ,  *p)) {
                    197:                                                c = c->nst;
                    198:                                        }
                    199:                                        else if (c->link != 0) {
                    200:                                                c = c->link;
                    201:                                                goto istate;
                    202:                                        }
                    203:                                }
                    204:                                else goto nstate;
                    205:                        }
                    206:                if (c->out) {
                    207:                        while (*p++ != '\n') {
                    208:                                if (--ccount <= 0) {
                    209:                                        if (p == &buf[2*BUFSIZ]) p = buf;
                    210:                                        if (p > &buf[BUFSIZ]) {
                    211:                                                if ((ccount = Read(f, p, &buf[2*BUFSIZ] - p)) <= 0) break;
                    212:                                        }
                    213:                                        else if ((ccount = Read(f, p, BUFSIZ)) <= 0) break;
                    214:                                        blkno += ccount;
                    215:                                }
                    216:                        }
                    217:                        if ( (vflag && (failed == 0 || xflag == 0)) || (vflag == 0 && xflag && failed) )
                    218:                                goto nomatch;
                    219:        succeed:        nsucc = 1;
                    220:                        if (cflag) tln++;
                    221:                        else if (sflag)
                    222:                                ;       /* ugh */
                    223:                        else if (lflag) {
                    224:                                printf("%s\n", file);
                    225:                                close(f);
                    226:                                return;
                    227:                        }
                    228:                        else {
                    229:                                if (nfile > 1 && hflag) printf("%s:", file);
                    230:                                if (bflag) printf("%ld:", (blkno-ccount-1)/1024);
                    231:                                if (nflag) printf("%ld:", lnum);
                    232:                                if (p <= nlp) {
                    233:                                        while (nlp < &buf[2*BUFSIZ]) putchar(*nlp++);
                    234:                                        nlp = buf;
                    235:                                }
                    236:                                while (nlp < p) putchar(*nlp++);
                    237:                        }
                    238:        nomatch:        lnum++;
                    239:                        nlp = p;
                    240:                        c = w;
                    241:                        failed = 0;
                    242:                        continue;
                    243:                }
                    244:                if (*p++ == '\n')
                    245:                        if (vflag) goto succeed;
                    246:                        else {
                    247:                                lnum++;
                    248:                                nlp = p;
                    249:                                c = w;
                    250:                                failed = 0;
                    251:                        }
                    252:        }
                    253:        close(f);
                    254:        if (cflag) {
                    255:                if (nfile > 1)
                    256:                        printf("%s:", file);
                    257:                printf("%ld\n", tln);
                    258:        }
                    259: #ifdef DOSTATS
                    260:        nbytes += blkno;
                    261:        nlines += lnum-1;
                    262: #endif
                    263: }
                    264: 
                    265: static int mailfd = -1;
                    266: 
                    267: getargc()
                    268: {
                    269:        register c;
                    270:        if (wordf)
                    271:                c = getc(wordf);
                    272:        else if ((c = *argptr++) == '\0')
                    273:                c = EOF;
                    274: #ifdef DOSTATS
                    275:        if(c != EOF) *statsexpr++ = c;
                    276: #endif
                    277:        return(c);
                    278: }
                    279: 
                    280: cgotofn() {
                    281:        register c;
                    282:        register struct words *s;
                    283: 
                    284:        s = smax = w;
                    285: nword: for(;;) {
                    286:                c = getargc();
                    287:                if (c==EOF)
                    288:                        return;
                    289:                if (c == '\n') {
                    290:                        if (xflag) {
                    291:                                for(;;) {
                    292:                                        if (s->inp == c) {
                    293:                                                s = s->nst;
                    294:                                                break;
                    295:                                        }
                    296:                                        if (s->inp == 0) goto nenter;
                    297:                                        if (s->link == 0) {
                    298:                                                if (smax >= &w[MAXSIZ -1]) overflo();
                    299:                                                s->link = ++smax;
                    300:                                                s = smax;
                    301:                                                goto nenter;
                    302:                                        }
                    303:                                        s = s->link;
                    304:                                }
                    305:                        }
                    306:                        s->out = 1;
                    307:                        s = w;
                    308:                } else {
                    309:                loop:   if (s->inp == c) {
                    310:                                s = s->nst;
                    311:                                continue;
                    312:                        }
                    313:                        if (s->inp == 0) goto enter;
                    314:                        if (s->link == 0) {
                    315:                                if (smax >= &w[MAXSIZ - 1]) overflo();
                    316:                                s->link = ++smax;
                    317:                                s = smax;
                    318:                                goto enter;
                    319:                        }
                    320:                        s = s->link;
                    321:                        goto loop;
                    322:                }
                    323:        }
                    324: 
                    325:        enter:
                    326:        do {
                    327:                s->inp = c;
                    328:                if (smax >= &w[MAXSIZ - 1]) overflo();
                    329:                s->nst = ++smax;
                    330:                s = smax;
                    331:        } while ((c = getargc()) != '\n' && c!=EOF);
                    332:        if (xflag) {
                    333:        nenter: s->inp = '\n';
                    334:                if (smax >= &w[MAXSIZ -1]) overflo();
                    335:                s->nst = ++smax;
                    336:        }
                    337:        smax->out = 1;
                    338:        s = w;
                    339:        if (c != EOF)
                    340:                goto nword;
                    341: }
                    342: 
                    343: overflo() {
                    344:        fprintf(stderr, "wordlist too large\n");
                    345:        exit(2);
                    346: }
                    347: cfail() {
                    348:        struct words *queue[QSIZE];
                    349:        struct words **front, **rear;
                    350:        struct words *state;
                    351:        register char c;
                    352:        register struct words *s;
                    353:        s = w;
                    354:        front = rear = queue;
                    355: init:  if ((s->inp) != 0) {
                    356:                *rear++ = s->nst;
                    357:                if (rear >= &queue[QSIZE - 1]) overflo();
                    358:        }
                    359:        if ((s = s->link) != 0) {
                    360:                goto init;
                    361:        }
                    362: 
                    363:        while (rear!=front) {
                    364:                s = *front;     /* s is next state in queue */
                    365:                if (front == &queue[QSIZE-1])
                    366:                        front = queue;
                    367:                else front++;
                    368:        cloop:  if ((c = s->inp) != 0) {        /* g(s,c) != 0 */
                    369:                        *rear = (q = s->nst);   /* add q = g(s,c) to queue */
                    370:                        if (front < rear)
                    371:                                if (rear >= &queue[QSIZE-1])
                    372:                                        if (front == queue) overflo();
                    373:                                        else rear = queue;
                    374:                                else rear++;
                    375:                        else
                    376:                                if (++rear == front) overflo();
                    377:                        state = s->fail;        /* state = f(s) */
                    378:                floop:  if (state == 0)
                    379:                                state = w;
                    380:                        if (state->inp == c)
                    381:                                do {    /* f(q) = g(state,c) for all q links */
                    382:                                        q->fail = state->nst;
                    383:                                        if ((state->nst)->out == 1) q->out = 1;
                    384:                                } while ((q = q->link) != 0);
                    385:                        else if (state->link != 0) {
                    386:                                state = state->link;
                    387:                                goto floop;
                    388:                        }
                    389:                        else if ((state = state->fail) != 0)
                    390:                                goto floop;     /* state = f(state) */
                    391:                }
                    392:                if ((s = s->link) != 0)
                    393:                        goto cloop;
                    394:        }
                    395: }
                    396: 
                    397: #ifdef DOSTATS
                    398: #include       <errno.h>
                    399: #define                NAME            "/tmp/grepdata"
                    400: dostats()
                    401: {
                    402:        umask(0);
                    403:        mailfd = open(NAME, 1);
                    404:        if((mailfd < 0) && (errno != ECONC)){
                    405:                umask(0);
                    406:                mailfd = creat(NAME, 03666);
                    407:        }
                    408:        if(mailfd >= 0){
                    409:                Finit(mailfd, (char *)0);
                    410:                Fseek(mailfd, 0L, 2);
                    411:                *statsexpr = 0;
                    412:                *statsfptr = 0;
                    413:                Fprint(mailfd, "\321fgrep:%s:%d:%d:%d:%d: %s\n", statsflags, nlines, nbytes, ntrans0, nfail, statspat);
                    414:        }
                    415:        Fflush(mailfd);
                    416: }
                    417: #endif

unix.superglobalmegacorp.com

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