123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721 |
- /*
- * This file is part of the UCB release of Plan 9. It is subject to the license
- * terms in the LICENSE file found in the top-level directory of this
- * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No
- * part of the UCB release of Plan 9, including this file, may be copied,
- * modified, propagated, or distributed except according to the terms contained
- * in the LICENSE file.
- */
- /****************************************************************
- Copyright (C) Lucent Technologies 1997
- All Rights Reserved
- Permission to use, copy, modify, and distribute this software and
- its documentation for any purpose and without fee is hereby
- granted, provided that the above copyright notice appear in all
- copies and that both that the copyright notice and this
- permission notice and warranty disclaimer appear in supporting
- documentation, and that the name Lucent Technologies or any of
- its entities not be used in advertising or publicity pertaining
- to distribution of the software without specific, written prior
- permission.
- LUCENT DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
- INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS.
- IN NO EVENT SHALL LUCENT OR ANY OF ITS ENTITIES BE LIABLE FOR ANY
- SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
- WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER
- IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,
- ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF
- THIS SOFTWARE.
- ****************************************************************/
- #define DEBUG
- #include <stdio.h>
- #include <string.h>
- #include <ctype.h>
- #include <errno.h>
- #include <stdlib.h>
- #include <stdarg.h>
- #include "awk.h"
- #include "y.tab.h"
- FILE *infile = NULL;
- char *file = "";
- char *record;
- int recsize = RECSIZE;
- char *fields;
- int fieldssize = RECSIZE;
- Cell **fldtab; /* pointers to Cells */
- char inputFS[100] = " ";
- #define MAXFLD 200
- int nfields = MAXFLD; /* last allocated slot for $i */
- int donefld; /* 1 = implies rec broken into fields */
- int donerec; /* 1 = record is valid (no flds have changed) */
- int lastfld = 0; /* last used field */
- int argno = 1; /* current input argument number */
- extern Awkfloat *ARGC;
- static Cell dollar0 = { OCELL, CFLD, NULL, "", 0.0, REC|STR|DONTFREE };
- static Cell dollar1 = { OCELL, CFLD, NULL, "", 0.0, FLD|STR|DONTFREE };
- void recinit(unsigned int n)
- {
- record = (char *) malloc(n);
- fields = (char *) malloc(n);
- fldtab = (Cell **) malloc((nfields+1) * sizeof(Cell *));
- if (record == NULL || fields == NULL || fldtab == NULL)
- FATAL("out of space for $0 and fields");
- fldtab[0] = (Cell *) malloc(sizeof (Cell));
- *fldtab[0] = dollar0;
- fldtab[0]->sval = record;
- fldtab[0]->nval = tostring("0");
- makefields(1, nfields);
- }
- void makefields(int n1, int n2) /* create $n1..$n2 inclusive */
- {
- char temp[50];
- int i;
- for (i = n1; i <= n2; i++) {
- fldtab[i] = (Cell *) malloc(sizeof (struct Cell));
- if (fldtab[i] == NULL)
- FATAL("out of space in makefields %d", i);
- *fldtab[i] = dollar1;
- sprintf(temp, "%d", i);
- fldtab[i]->nval = tostring(temp);
- }
- }
- void initgetrec(void)
- {
- int i;
- char *p;
- for (i = 1; i < *ARGC; i++) {
- if (!isclvar(p = getargv(i))) { /* find 1st real filename */
- setsval(lookup("FILENAME", symtab), getargv(i));
- return;
- }
- setclvar(p); /* a commandline assignment before filename */
- argno++;
- }
- infile = stdin; /* no filenames, so use stdin */
- }
- int getrec(char **pbuf, int *pbufsize, int isrecord) /* get next input record */
- { /* note: cares whether buf == record */
- int c;
- static int firsttime = 1;
- char *buf = *pbuf;
- int bufsize = *pbufsize;
- if (firsttime) {
- firsttime = 0;
- initgetrec();
- }
- dprintf( ("RS=<%s>, FS=<%s>, ARGC=%g, FILENAME=%s\n",
- *RS, *FS, *ARGC, *FILENAME) );
- if (isrecord) {
- donefld = 0;
- donerec = 1;
- }
- buf[0] = 0;
- while (argno < *ARGC || infile == stdin) {
- dprintf( ("argno=%d, file=|%s|\n", argno, file) );
- if (infile == NULL) { /* have to open a new file */
- file = getargv(argno);
- if (*file == '\0') { /* it's been zapped */
- argno++;
- continue;
- }
- if (isclvar(file)) { /* a var=value arg */
- setclvar(file);
- argno++;
- continue;
- }
- *FILENAME = file;
- dprintf( ("opening file %s\n", file) );
- if (*file == '-' && *(file+1) == '\0')
- infile = stdin;
- else if ((infile = fopen(file, "r")) == NULL)
- FATAL("can't open file %s", file);
- setfval(fnrloc, 0.0);
- }
- c = readrec(&buf, &bufsize, infile);
- if (c != 0 || buf[0] != '\0') { /* normal record */
- if (isrecord) {
- if (freeable(fldtab[0]))
- xfree(fldtab[0]->sval);
- fldtab[0]->sval = buf; /* buf == record */
- fldtab[0]->tval = REC | STR | DONTFREE;
- if (is_number(fldtab[0]->sval)) {
- fldtab[0]->fval = atof(fldtab[0]->sval);
- fldtab[0]->tval |= NUM;
- }
- }
- setfval(nrloc, nrloc->fval+1);
- setfval(fnrloc, fnrloc->fval+1);
- *pbuf = buf;
- *pbufsize = bufsize;
- return 1;
- }
- /* EOF arrived on this file; set up next */
- if (infile != stdin)
- fclose(infile);
- infile = NULL;
- argno++;
- }
- *pbuf = buf;
- *pbufsize = bufsize;
- return 0; /* true end of file */
- }
- void nextfile(void)
- {
- if (infile != stdin)
- fclose(infile);
- infile = NULL;
- argno++;
- }
- int readrec(char **pbuf, int *pbufsize, FILE *inf) /* read one record into buf */
- {
- int sep, c;
- char *rr, *buf = *pbuf;
- int bufsize = *pbufsize;
- if (strlen(*FS) >= sizeof(inputFS))
- FATAL("field separator %.10s... is too long", *FS);
- strcpy(inputFS, *FS); /* for subsequent field splitting */
- if ((sep = **RS) == 0) {
- sep = '\n';
- while ((c=getc(inf)) == '\n' && c != EOF) /* skip leading \n's */
- ;
- if (c != EOF)
- ungetc(c, inf);
- }
- for (rr = buf; ; ) {
- for (; (c=getc(inf)) != sep && c != EOF; ) {
- if (rr-buf+1 > bufsize)
- if (!adjbuf(&buf, &bufsize, 1+rr-buf, recsize, &rr, "readrec 1"))
- FATAL("input record `%.30s...' too long", buf);
- *rr++ = c;
- }
- if (**RS == sep || c == EOF)
- break;
- if ((c = getc(inf)) == '\n' || c == EOF) /* 2 in a row */
- break;
- if (!adjbuf(&buf, &bufsize, 2+rr-buf, recsize, &rr, "readrec 2"))
- FATAL("input record `%.30s...' too long", buf);
- *rr++ = '\n';
- *rr++ = c;
- }
- if (!adjbuf(&buf, &bufsize, 1+rr-buf, recsize, &rr, "readrec 3"))
- FATAL("input record `%.30s...' too long", buf);
- *rr = 0;
- dprintf( ("readrec saw <%s>, returns %d\n", buf, c == EOF && rr == buf ? 0 : 1) );
- *pbuf = buf;
- *pbufsize = bufsize;
- return c == EOF && rr == buf ? 0 : 1;
- }
- char *getargv(int n) /* get ARGV[n] */
- {
- Cell *x;
- char *s, temp[50];
- extern Array *ARGVtab;
- sprintf(temp, "%d", n);
- x = setsymtab(temp, "", 0.0, STR, ARGVtab);
- s = getsval(x);
- dprintf( ("getargv(%d) returns |%s|\n", n, s) );
- return s;
- }
- void setclvar(char *s) /* set var=value from s */
- {
- char *p;
- Cell *q;
- for (p=s; *p != '='; p++)
- ;
- *p++ = 0;
- p = qstring(p, '\0');
- q = setsymtab(s, p, 0.0, STR, symtab);
- setsval(q, p);
- if (is_number(q->sval)) {
- q->fval = atof(q->sval);
- q->tval |= NUM;
- }
- dprintf( ("command line set %s to |%s|\n", s, p) );
- }
- void fldbld(void) /* create fields from current record */
- {
- /* this relies on having fields[] the same length as $0 */
- /* the fields are all stored in this one array with \0's */
- char *r, *fr, sep;
- Cell *p;
- int i, j, n;
- if (donefld)
- return;
- if (!isstr(fldtab[0]))
- getsval(fldtab[0]);
- r = fldtab[0]->sval;
- n = strlen(r);
- if (n > fieldssize) {
- xfree(fields);
- if ((fields = (char *) malloc(n+1)) == NULL)
- FATAL("out of space for fields in fldbld %d", n);
- fieldssize = n;
- }
- fr = fields;
- i = 0; /* number of fields accumulated here */
- if (strlen(inputFS) > 1) { /* it's a regular expression */
- i = refldbld(r, inputFS);
- } else if ((sep = *inputFS) == ' ') { /* default whitespace */
- for (i = 0; ; ) {
- while (*r == ' ' || *r == '\t' || *r == '\n')
- r++;
- if (*r == 0)
- break;
- i++;
- if (i > nfields)
- growfldtab(i);
- if (freeable(fldtab[i]))
- xfree(fldtab[i]->sval);
- fldtab[i]->sval = fr;
- fldtab[i]->tval = FLD | STR | DONTFREE;
- do
- *fr++ = *r++;
- while (*r != ' ' && *r != '\t' && *r != '\n' && *r != '\0');
- *fr++ = 0;
- }
- *fr = 0;
- } else if ((sep = *inputFS) == 0) { /* new: FS="" => 1 char/field */
- for (i = 0; *r != 0; r++) {
- char buf[2];
- i++;
- if (i > nfields)
- growfldtab(i);
- if (freeable(fldtab[i]))
- xfree(fldtab[i]->sval);
- buf[0] = *r;
- buf[1] = 0;
- fldtab[i]->sval = tostring(buf);
- fldtab[i]->tval = FLD | STR;
- }
- *fr = 0;
- } else if (*r != 0) { /* if 0, it's a null field */
- for (;;) {
- i++;
- if (i > nfields)
- growfldtab(i);
- if (freeable(fldtab[i]))
- xfree(fldtab[i]->sval);
- fldtab[i]->sval = fr;
- fldtab[i]->tval = FLD | STR | DONTFREE;
- while (*r != sep && *r != '\n' && *r != '\0') /* \n is always a separator */
- *fr++ = *r++;
- *fr++ = 0;
- if (*r++ == 0)
- break;
- }
- *fr = 0;
- }
- if (i > nfields)
- FATAL("record `%.30s...' has too many fields; can't happen", r);
- cleanfld(i+1, lastfld); /* clean out junk from previous record */
- lastfld = i;
- donefld = 1;
- for (j = 1; j <= lastfld; j++) {
- p = fldtab[j];
- if(is_number(p->sval)) {
- p->fval = atof(p->sval);
- p->tval |= NUM;
- }
- }
- setfval(nfloc, (Awkfloat) lastfld);
- if (dbg) {
- for (j = 0; j <= lastfld; j++) {
- p = fldtab[j];
- printf("field %d (%s): |%s|\n", j, p->nval, p->sval);
- }
- }
- }
- void cleanfld(int n1, int n2) /* clean out fields n1 .. n2 inclusive */
- { /* nvals remain intact */
- Cell *p;
- int i;
- for (i = n1; i <= n2; i++) {
- p = fldtab[i];
- if (freeable(p))
- xfree(p->sval);
- p->sval = "";
- p->tval = FLD | STR | DONTFREE;
- }
- }
- void newfld(int n) /* add field n after end of existing lastfld */
- {
- if (n > nfields)
- growfldtab(n);
- cleanfld(lastfld+1, n);
- lastfld = n;
- setfval(nfloc, (Awkfloat) n);
- }
- Cell *fieldadr(int n) /* get nth field */
- {
- if (n < 0)
- FATAL("trying to access field %d", n);
- if (n > nfields) /* fields after NF are empty */
- growfldtab(n); /* but does not increase NF */
- return(fldtab[n]);
- }
- void growfldtab(int n) /* make new fields up to at least $n */
- {
- int nf = 2 * nfields;
- if (n > nf)
- nf = n;
- fldtab = (Cell **) realloc(fldtab, (nf+1) * (sizeof (struct Cell *)));
- if (fldtab == NULL)
- FATAL("out of space creating %d fields", nf);
- makefields(nfields+1, nf);
- nfields = nf;
- }
- int refldbld(char *rec, char *fs) /* build fields from reg expr in FS */
- {
- /* this relies on having fields[] the same length as $0 */
- /* the fields are all stored in this one array with \0's */
- char *fr;
- void *p;
- int i, tempstat, n;
- n = strlen(rec);
- if (n > fieldssize) {
- xfree(fields);
- if ((fields = (char *) malloc(n+1)) == NULL)
- FATAL("out of space for fields in refldbld %d", n);
- fieldssize = n;
- }
- fr = fields;
- *fr = '\0';
- if (*rec == '\0')
- return 0;
- p = compre(fs);
- dprintf( ("into refldbld, rec = <%s>, pat = <%s>\n", rec, fs) );
- for (i = 1; ; i++) {
- if (i > nfields)
- growfldtab(i);
- if (freeable(fldtab[i]))
- xfree(fldtab[i]->sval);
- fldtab[i]->tval = FLD | STR | DONTFREE;
- fldtab[i]->sval = fr;
- dprintf( ("refldbld: i=%d\n", i) );
- if (nematch(p, rec, rec)) {
- dprintf( ("match %s (%d chars)\n", patbeg, patlen) );
- strncpy(fr, rec, patbeg-rec);
- fr += patbeg - rec + 1;
- *(fr-1) = '\0';
- rec = patbeg + patlen;
- } else {
- dprintf( ("no match %s\n", rec) );
- strcpy(fr, rec);
- break;
- }
- }
- return i;
- }
- void recbld(void) /* create $0 from $1..$NF if necessary */
- {
- int i;
- char *r, *p;
- if (donerec == 1)
- return;
- r = record;
- for (i = 1; i <= *NF; i++) {
- p = getsval(fldtab[i]);
- if (!adjbuf(&record, &recsize, 1+strlen(p)+r-record, recsize, &r, "recbld 1"))
- FATAL("created $0 `%.30s...' too long", record);
- while ((*r = *p++) != 0)
- r++;
- if (i < *NF) {
- if (!adjbuf(&record, &recsize, 2+strlen(*OFS)+r-record, recsize, &r, "recbld 2"))
- FATAL("created $0 `%.30s...' too long", record);
- for (p = *OFS; (*r = *p++) != 0; )
- r++;
- }
- }
- if (!adjbuf(&record, &recsize, 2+r-record, recsize, &r, "recbld 3"))
- FATAL("built giant record `%.30s...'", record);
- *r = '\0';
- dprintf( ("in recbld inputFS=%s, fldtab[0]=%p\n", inputFS, fldtab[0]) );
- if (freeable(fldtab[0]))
- xfree(fldtab[0]->sval);
- fldtab[0]->tval = REC | STR | DONTFREE;
- fldtab[0]->sval = record;
- dprintf( ("in recbld inputFS=%s, fldtab[0]=%p\n", inputFS, fldtab[0]) );
- dprintf( ("recbld = |%s|\n", record) );
- donerec = 1;
- }
- int errorflag = 0;
- void yyerror(char *s)
- {
- SYNTAX(s);
- }
- void SYNTAX(char *fmt, ...)
- {
- extern char *cmdname, *curfname;
- static int been_here = 0;
- va_list varg;
- if (been_here++ > 2)
- return;
- fprintf(stderr, "%s: ", cmdname);
- va_start(varg, fmt);
- vfprintf(stderr, fmt, varg);
- va_end(varg);
- if(compile_time == 1 && cursource() != NULL)
- fprintf(stderr, " at %s:%d", cursource(), lineno);
- else
- fprintf(stderr, " at line %d", lineno);
- if (curfname != NULL)
- fprintf(stderr, " in function %s", curfname);
- fprintf(stderr, "\n");
- errorflag = 2;
- eprint();
- }
- void fpecatch(int n)
- {
- FATAL("floating point exception %d", n);
- }
- extern int bracecnt, brackcnt, parencnt;
- void bracecheck(void)
- {
- int c;
- static int beenhere = 0;
- if (beenhere++)
- return;
- while ((c = input()) != EOF && c != '\0')
- bclass(c);
- bcheck2(bracecnt, '{', '}');
- bcheck2(brackcnt, '[', ']');
- bcheck2(parencnt, '(', ')');
- }
- void bcheck2(int n, int c1, int c2)
- {
- if (n == 1)
- fprintf(stderr, "\tmissing %c\n", c2);
- else if (n > 1)
- fprintf(stderr, "\t%d missing %c's\n", n, c2);
- else if (n == -1)
- fprintf(stderr, "\textra %c\n", c2);
- else if (n < -1)
- fprintf(stderr, "\t%d extra %c's\n", -n, c2);
- }
- void FATAL(char *fmt, ...)
- {
- extern char *cmdname;
- va_list varg;
- fflush(stdout);
- fprintf(stderr, "%s: ", cmdname);
- va_start(varg, fmt);
- vfprintf(stderr, fmt, varg);
- va_end(varg);
- error();
- if (dbg > 1) /* core dump if serious debugging on */
- abort();
- exit(2);
- }
- void WARNING(char *fmt, ...)
- {
- extern char *cmdname;
- va_list varg;
- fflush(stdout);
- fprintf(stderr, "%s: ", cmdname);
- va_start(varg, fmt);
- vfprintf(stderr, fmt, varg);
- va_end(varg);
- error();
- }
- void error()
- {
- extern Node *curnode;
- int line;
- fprintf(stderr, "\n");
- if (compile_time != 2 && NR && *NR > 0) {
- if (strcmp(*FILENAME, "-") != 0)
- fprintf(stderr, " input record %s:%d", *FILENAME, (int) (*FNR));
- else
- fprintf(stderr, " input record number %d", (int) (*FNR));
- fprintf(stderr, "\n");
- }
- if (compile_time != 2 && curnode)
- line = curnode->lineno;
- else if (compile_time != 2 && lineno)
- line = lineno;
- else
- line = -1;
- if (compile_time == 1 && cursource() != NULL){
- if(line >= 0)
- fprintf(stderr, " source %s:%d", cursource(), line);
- else
- fprintf(stderr, " source file %s", cursource());
- }else if(line >= 0)
- fprintf(stderr, " source line %d", line);
- fprintf(stderr, "\n");
- eprint();
- }
- void eprint(void) /* try to print context around error */
- {
- char *p, *q;
- int c;
- static int been_here = 0;
- extern char ebuf[], *ep;
- if (compile_time == 2 || compile_time == 0 || been_here++ > 0)
- return;
- p = ep - 1;
- if (p > ebuf && *p == '\n')
- p--;
- for ( ; p > ebuf && *p != '\n' && *p != '\0'; p--)
- ;
- while (*p == '\n')
- p++;
- fprintf(stderr, " context is\n\t");
- for (q=ep-1; q>=p && *q!=' ' && *q!='\t' && *q!='\n'; q--)
- ;
- for ( ; p < q; p++)
- if (*p)
- putc(*p, stderr);
- fprintf(stderr, " >>> ");
- for ( ; p < ep; p++)
- if (*p)
- putc(*p, stderr);
- fprintf(stderr, " <<< ");
- if (*ep)
- while ((c = input()) != '\n' && c != '\0' && c != EOF) {
- putc(c, stderr);
- bclass(c);
- }
- putc('\n', stderr);
- ep = ebuf;
- }
- void bclass(int c)
- {
- switch (c) {
- case '{': bracecnt++; break;
- case '}': bracecnt--; break;
- case '[': brackcnt++; break;
- case ']': brackcnt--; break;
- case '(': parencnt++; break;
- case ')': parencnt--; break;
- }
- }
- double errcheck(double x, char *s)
- {
- if (errno == EDOM) {
- errno = 0;
- WARNING("%s argument out of domain", s);
- x = 1;
- } else if (errno == ERANGE) {
- errno = 0;
- WARNING("%s result out of range", s);
- x = 1;
- }
- return x;
- }
- int isclvar(char *s) /* is s of form var=something ? */
- {
- char *os = s;
- if (!isalpha(*s) && *s != '_')
- return 0;
- for ( ; *s; s++)
- if (!(isalnum(*s) || *s == '_'))
- break;
- return *s == '=' && s > os && *(s+1) != '=';
- }
- /* strtod is supposed to be a proper test of what's a valid number */
- #include <math.h>
- int is_number(char *s)
- {
- double r;
- char *ep;
- /*
- * fast could-it-be-a-number check before calling strtod,
- * which takes a surprisingly long time to reject non-numbers.
- */
- switch (*s) {
- case '0': case '1': case '2': case '3': case '4':
- case '5': case '6': case '7': case '8': case '9':
- case '\t':
- case '\n':
- case '\v':
- case '\f':
- case '\r':
- case ' ':
- case '-':
- case '+':
- case '.':
- case 'n': /* nans */
- case 'N':
- case 'i': /* infs */
- case 'I':
- break;
- default:
- return 0; /* can't be a number */
- }
- errno = 0;
- r = strtod(s, &ep);
- if (ep == s || r == HUGE_VAL || errno == ERANGE)
- return 0;
- while (*ep == ' ' || *ep == '\t' || *ep == '\n')
- ep++;
- if (*ep == '\0')
- return 1;
- else
- return 0;
- }
|