Ultrix-3.1/src/cmd/awk/lib.c
/**********************************************************************
* Copyright (c) Digital Equipment Corporation 1984, 1985, 1986. *
* All Rights Reserved. *
* Reference "/usr/src/COPYRIGHT" for applicable restrictions. *
**********************************************************************/
static char Sccsid[] = "@(#)lib.c 3.0 4/21/86";
#include "stdio.h"
#include "awk.def"
#include "awk.h"
#include "ctype.h"
FILE *infile = NULL;
char *file;
#ifdef SMALL
#define RECSIZE (1 * 512)
#else
#define RECSIZE (5 * 512)
#endif
char record[RECSIZE];
char fields[RECSIZE];
#ifdef SMALL
#define MAXFLD 32
#else
#define MAXFLD 100
#endif
int donefld; /* 1 = implies rec broken into fields */
int donerec; /* 1 = record is valid (no flds have changed) */
int mustfld; /* 1 = NF seen, so always break*/
#define FINIT {0, NULL, 0.0, FLD|STR}
cell fldtab[MAXFLD] = { /*room for fields */
{ "$record", record, 0.0, STR|FLD},
FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
#ifdef SMALL
FINIT, FINIT, FINIT
#else
FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT
#endif
};
int maxfld = 0; /* last used field */
getrec()
{
register char *rr;
extern int svargc;
extern char **svargv;
register c, sep;
dprintf("**RS=%o, **FS=%o\n", **RS, **FS, NULL);
donefld = 0;
donerec = 1;
record[0] = 0;
while (svargc > 0) {
dprintf("svargc=%d, *svargv=%s\n", svargc, *svargv, NULL);
if (infile == NULL) { /* have to open a new file */
if (member('=', *svargv)) { /* it's a var=value argument */
setclvar(*svargv);
svargv++;
svargc--;
continue;
}
*FILENAME = file = *svargv;
dprintf("opening file %s\n", file, NULL, NULL);
if (*file == '-')
infile = stdin;
else if ((infile = fopen(file, "r")) == NULL)
error("can't open %s", FATAL, file);
}
if ((sep = **RS) == 0)
sep = '\n';
for (rr = record; ; ) {
for (; (c=getc(infile)) != sep && c != EOF; *rr++ = c)
;
if (**RS == sep || c == EOF)
break;
if ((c = getc(infile)) == '\n' || c == EOF) /* 2 in a row */
break;
*rr++ = '\n';
*rr++ = c;
}
if (rr > record+RECSIZE)
error("record `%.20s...' too long", FATAL, record);
*rr = 0;
if (mustfld)
fldbld();
if (c != EOF || rr > record) { /* normal record */
recloc->tval &= ~NUM;
recloc->tval |= STR;
++nrloc->fval;
nrloc->tval &= ~STR;
nrloc->tval |= NUM;
return(1);
}
/* EOF arrived on this file; set up next */
if (infile != stdin)
fclose(infile);
infile = NULL;
svargc--;
svargv++;
}
return(0); /* true end of file */
}
setclvar(s) /* set var=value from s */
char *s;
{
char *p;
cell *q;
for (p=s; *p != '='; p++)
;
*p++ = 0;
q = setsymtab(s, tostring(p), 0.0, STR, symtab);
setsval(q, p);
dprintf("command line set %s to |%s|\n", s, p, NULL);
}
fldbld()
{
register char *r, *fr, sep,*kjb;
int i, j;
r = record;
fr = fields;
i = 0; /* number of fields accumulated here */
if ((sep = **FS) == ' ')
for (i = 0; ; ) {
if(dbg){
printf("record = %s ",r);
for(kjb = r; *kjb && kjb;kjb++)
printf("%o ",(j = *kjb));
printf("\n");
}
while (*r == ' ' || *r == '\t' || *r == '\n')
r++;
if (*r == 0)
break;
i++;
if (i >= MAXFLD)
error("record `%.20s...' has too many fields", FATAL, record);
if (!(fldtab[i].tval&FLD))
xfree(fldtab[i].sval);
fldtab[i].sval = fr;
if(dbg)
printf("fldtab[%d].sval %s\n",i,fldtab[i].sval);
fldtab[i].tval = FLD | STR;
do
*fr++ = *r++;
while (*r != ' ' && *r != '\t' && *r != '\n' && *r != '\0');
*fr++ = 0;
}
else if (*r != 0) /* if 0, it's a null field */
for (;;) {
i++;
if (i >= MAXFLD)
error("record `%.20s...' has too many fields", FATAL, record);
if (!(fldtab[i].tval&FLD))
xfree(fldtab[i].sval);
fldtab[i].sval = fr;
if(dbg)
printf("fldtab[%d].sval %s\n",i,fldtab[i].sval);
fldtab[i].tval = FLD | STR;
while (*r != sep && *r != '\n' && *r != '\0') /* \n always a separator */
*fr++ = *r++;
*fr++ = 0;
if (*r++ == 0)
break;
}
*fr = 0;
for (j=MAXFLD-1; j>i; j--) { /* clean out junk from previous record */
if (!(fldtab[j].tval&FLD))
xfree(fldtab[j].sval);
fldtab[j].tval = STR | FLD;
fldtab[j].sval = NULL;
}
maxfld = i;
donefld = 1;
for(i=1; i<=maxfld; i++)
if(isnumber(fldtab[i].sval)) {
fldtab[i].fval = atof(fldtab[i].sval);
fldtab[i].tval |= NUM;
}
setfval(lookup("NF", symtab, 0), (awkfloat) maxfld);
if (dbg)
for (i = 0; i <= maxfld; i++)
printf("field %d: |%s|\n", i, fldtab[i].sval);
}
recbld()
{
int i;
register char *r, *p;
if (donefld == 0 || donerec == 1)
return;
r = record;
for (i = 1; i <= *NF; i++) {
p = getsval(&fldtab[i]);
while (*r++ = *p++)
;
*(r-1) = **OFS;
}
*(r-1) = '\0';
dprintf("in recbld FS=%o, recloc=%o\n", **FS, recloc, NULL);
recloc->tval = STR | FLD;
dprintf("in recbld FS=%o, recloc=%o\n", **FS, recloc, NULL);
if (r > record+RECSIZE)
error("built giant record `%.20s...'", FATAL, record);
dprintf("recbld = |%s|\n", record, NULL, NULL);
}
cell *fieldadr(n)
{
if (n >= MAXFLD)
error("trying to access field %d", FATAL, n);
return(&fldtab[n]);
}
int errorflag = 0;
yyerror(s) char *s; {
fprintf(stderr, "awk: %s near line %d\n", s, lineno);
errorflag = 2;
}
#ifdef ERRFILE
/*
* error routine that uses an external error file.
*/
char efilname[] = "/usr/lib/awk_strings";
int efil = -1;
error(index, type, a1, a2, a3, a4, a5)
int index;
{
char buf[100];
long lseek();
struct Nameblock *v;
int skrtn, rdrtn;
fprintf(stderr, "awk: ");
if (efil < 0 && (efil = open(efilname, 0)) < 0) {
fprintf(stderr, "cannot get error: (cannot open %s)", efilname);
goto out;
}
if ((skrtn = lseek(efil, (long)index, 0)) < 0) {
fprintf(stderr, "cannot get error: (bad seek into %s)", efilname);
goto out;
}
if ((rdrtn = read(efil, buf, 100)) <= 0) {
fprintf(stderr, "cannot get error: (bad read from %s)", efilname);
goto out;
}
fprintf(stderr, buf, a1, a2, a3, a4, a5);
out:
fprintf(stderr, "\n");
if (*NR > 0)
fprintf(stderr, " record number %g\n", *NR);
if (type == FATAL)
exit(2);
}
#else
/*
* used to be (f, s, ..., the order was changed so that
* mkstr could be run on it.
*/
error(s, f, a1, a2, a3, a4, a5, a6, a7) {
fprintf(stderr, "awk: ");
fprintf(stderr, s, a1, a2, a3, a4, a5, a6, a7);
fprintf(stderr, "\n");
if (*NR > 0)
fprintf(stderr, " record number %g\n", *NR);
if (f)
exit(2);
}
#endif
PUTS(s) char *s; {
dprintf("%s\n", s, NULL, NULL);
}
#define MAXEXPON 38 /* maximum exponenet for fp number */
isnumber(s)
register char *s;
{
register d1, d2;
int point;
char *es;
d1 = d2 = point = 0;
while (*s == ' ' || *s == '\t' || *s == '\n')
s++;
if (*s == '\0')
return(0); /* empty stuff isn't number */
if (*s == '+' || *s == '-')
s++;
if (!isdigit(*s) && *s != '.')
return(0);
if (isdigit(*s)) {
do {
d1++;
s++;
} while (isdigit(*s));
}
if(d1 >= MAXEXPON)
return(0); /* too many digits to convert */
if (*s == '.') {
point++;
s++;
}
if (isdigit(*s)) {
d2++;
do {
s++;
} while (isdigit(*s));
}
if (!(d1 || point && d2))
return(0);
if (*s == 'e' || *s == 'E') {
s++;
if (*s == '+' || *s == '-')
s++;
if (!isdigit(*s))
return(0);
es = s;
do {
s++;
} while (isdigit(*s));
if (s - es > 2)
return(0);
else if (s - es == 2 && 10 * (*es-'0') + *(es+1)-'0' >= MAXEXPON)
return(0);
}
while (*s == ' ' || *s == '\t' || *s == '\n')
s++;
if (*s == '\0')
return(1);
else
return(0);
}
/*
isnumber(s) char *s; {return(0);}
*/