[unix-history] / usr / src / old / awk / lib.c

/*-
 * Copyright (c) 1991 The Regents of the University of California.
 * All rights reserved.
 *
 * %sccs.include.proprietary.c%
 */

#ifndef lint
static char sccsid[] = "@(#)lib.c	4.11 (Berkeley) %G%";
#endif /* not lint */

#ifdef __STDC__
#include <stdarg.h>
#else
#include <varargs.h>
#endif
#include "stdio.h"
#include "awk.def"
#include "awk.h"
#include "ctype.h"

int	error __P((int, const char *, ...));

extern FILE	*yyin;	/* lex input file */
extern char	*lexprog;	/* points to program argument if it exists */
FILE	*infile	= NULL;
char	*file;
#define	RECSIZE	(5 * 512)
char	record[RECSIZE];
char	fields[RECSIZE];
char	EMPTY[] = "";

#define	MAXFLD	100
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	{EMPTY, EMPTY, 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, 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, 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,
};
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 == '-') {
				if (yyin == stdin && ! lexprog)
					error(FATAL, "standard input already used for reading commands");
				else
					infile = stdin;
			}
			else if ((infile = fopen(file, "r")) == NULL)
				error(FATAL, "can't open %s", 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(FATAL, "record `%.20s...' too long", 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;
	int i, j;

	r = record;
	fr = fields;
	i = 0;	/* number of fields accumulated here */
	if ((sep = **FS) == ' ')
		for (i = 0; ; ) {
			while (*r == ' ' || *r == '\t' || *r == '\n')
				r++;
			if (*r == 0)
				break;
			i++;
			if (i >= MAXFLD)
				error(FATAL, "record `%.20s...' has too many fields", record);
			if (!(fldtab[i].tval&FLD))
				strfree(fldtab[i].sval);
			fldtab[i].sval = fr;
			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(FATAL, "record `%.20s...' has too many fields", record);
			if (!(fldtab[i].tval&FLD))
				strfree(fldtab[i].sval);
			fldtab[i].sval = fr;
			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))
			strfree(fldtab[j].sval);
		fldtab[j].tval = STR | FLD;
		fldtab[j].sval = EMPTY;
	}
	maxfld = i;
	donefld = 1;
	for(i=1; i<=maxfld; i++)
		if(isanumber(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(FATAL, "built giant record `%.20s...'", record);
	dprintf("recbld = |%s|\n", record, NULL, NULL);
}

cell *fieldadr(n)
{
	if (n >= MAXFLD)
		error(FATAL, "trying to access field %d", n);
	return(&fldtab[n]);
}

int	errorflag	= 0;

yyerror(s) char *s; {
	fprintf(stderr, "awk: %s near line %d\n", s, lineno);
	errorflag = 2;
}

int
#ifdef __STDC__
error(int isfatal, const char *fmt, ...)
#else
error(isfatal, fmt, va_alist)
	int isfatal;
	char *fmt;
	va_dcl
#endif
{
	va_list ap;

#ifdef __STDC__
	va_start(ap, fmt);
#else
	va_start(ap);
#endif
	(void)fprintf(stderr, "awk: ");
	(void)vfprintf(stderr, fmt, ap);
	va_end(ap);
	(void)fprintf(stderr, "\n");
	if (NR && *NR > 0)
		(void)fprintf(stderr, " record number %g\n", *NR);
	if (isfatal)
		exit(2);
}

PUTS(s) char *s; {
	dprintf("%s\n", s, NULL, NULL);
}

#define	MAXEXPON	38	/* maximum exponenet for fp number */

isanumber(s)
register char *s;
{
	register d1, d2;
	int point;
	char *es;

	if (s == NULL) 
		return (0);
	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);
}
/*
isanumber(s) char *s; {return(0);}
*/
Commit	Line	Data
2791ff57 KB	1	/*-
	2	* Copyright (c) 1991 The Regents of the University of California.
	3	* All rights reserved.
	4	*
	5	* %sccs.include.proprietary.c%
	6	*/
	7
b803c326	8	#ifndef lint
89616477	9	static char sccsid[] = "@(#)lib.c 4.11 (Berkeley) %G%";
2791ff57	10	#endif /* not lint */
b9bbf539	11
66d1b859	12	#ifdef __STDC__
2404822a	13	#include <stdarg.h>
66d1b859 CT	14	#else
	15	#include <varargs.h>
	16	#endif
b9bbf539 KM	17	#include "stdio.h"
	18	#include "awk.def"
	19	#include "awk.h"
	20	#include "ctype.h"
	21
66d1b859 CT	22	int error __P((int, const char *, ...));
66d1b859 CT	23
26758489	24	extern FILE yyin; / lex input file */
e068ed21	25	extern char lexprog; / points to program argument if it exists */
b9bbf539 KM	26	FILE *infile = NULL;
	27	char *file;
	28	#define RECSIZE (5 * 512)
	29	char record[RECSIZE];
	30	char fields[RECSIZE];
c48090b0	31	char EMPTY[] = "";
b9bbf539 KM	32
	33	#define MAXFLD 100
	34	int donefld; /* 1 = implies rec broken into fields */
	35	int donerec; /* 1 = record is valid (no flds have changed) */
	36	int mustfld; /* 1 = NF seen, so always break*/
	37
c48090b0	38	#define FINIT {EMPTY, EMPTY, 0.0, FLD\|STR}
b9bbf539 KM	39	cell fldtab[MAXFLD] = { /room for fields /
b9bbf539 KM	40	{ "$record", record, 0.0, STR\|FLD},
83b07d63 KB	41	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
	42	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
	43	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
	44	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
	45	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
	46	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
	47	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
	48	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
	49	FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT,
b9bbf539 KM	50	};
	51	int maxfld = 0; /* last used field */
	52
	53
	54	getrec()
	55	{
	56	register char *rr;
	57	extern int svargc;
	58	extern char **svargv;
	59	register c, sep;
	60
	61	dprintf("RS=%o, FS=%o\n", RS, FS, NULL);
	62	donefld = 0;
	63	donerec = 1;
	64	record[0] = 0;
	65	while (svargc > 0) {
	66	dprintf("svargc=%d, svargv=%s\n", svargc, svargv, NULL);
	67	if (infile == NULL) { /* have to open a new file */
	68	if (member('=', svargv)) { / it's a var=value argument */
	69	setclvar(*svargv);
	70	svargv++;
	71	svargc--;
	72	continue;
	73	}
	74	FILENAME = file = svargv;
	75	dprintf("opening file %s\n", file, NULL, NULL);
26758489	76	if (*file == '-') {
e068ed21	77	if (yyin == stdin && ! lexprog)
26758489 KB	78	error(FATAL, "standard input already used for reading commands");
	79	else
	80	infile = stdin;
	81	}
b9bbf539 KM	82	else if ((infile = fopen(file, "r")) == NULL)
	83	error(FATAL, "can't open %s", file);
	84	}
	85	if ((sep = **RS) == 0)
	86	sep = '\n';
	87	for (rr = record; ; ) {
	88	for (; (c=getc(infile)) != sep && c != EOF; *rr++ = c)
	89	;
	90	if (**RS == sep \|\| c == EOF)
	91	break;
	92	if ((c = getc(infile)) == '\n' \|\| c == EOF) /* 2 in a row */
	93	break;
	94	*rr++ = '\n';
	95	*rr++ = c;
	96	}
	97	if (rr > record+RECSIZE)
	98	error(FATAL, "record `%.20s...' too long", record);
	99	*rr = 0;
	100	if (mustfld)
	101	fldbld();
	102	if (c != EOF \|\| rr > record) { /* normal record */
	103	recloc->tval &= ~NUM;
	104	recloc->tval \|= STR;
	105	++nrloc->fval;
	106	nrloc->tval &= ~STR;
	107	nrloc->tval \|= NUM;
	108	return(1);
	109	}
	110	/* EOF arrived on this file; set up next */
	111	if (infile != stdin)
	112	fclose(infile);
	113	infile = NULL;
	114	svargc--;
	115	svargv++;
	116	}
	117	return(0); /* true end of file */
	118	}
	119
	120	setclvar(s) /* set var=value from s */
	121	char *s;
	122	{
	123	char *p;
	124	cell *q;
	125
	126	for (p=s; *p != '='; p++)
	127	;
	128	*p++ = 0;
	129	q = setsymtab(s, tostring(p), 0.0, STR, symtab);
	130	setsval(q, p);
	131	dprintf("command line set %s to \|%s\|\n", s, p, NULL);
	132	}
	133
	134	fldbld()
	135	{
	136	register char r, fr, sep;
	137	int i, j;
	138
	139	r = record;
	140	fr = fields;
	141	i = 0; /* number of fields accumulated here */
	142	if ((sep = **FS) == ' ')
	143	for (i = 0; ; ) {
	144	while (r == ' ' \|\| r == '\t' \|\| *r == '\n')
	145	r++;
146	if (*r == 0)
147	break;
148	i++;
149	if (i >= MAXFLD)
150	error(FATAL, "record `%.20s...' has too many fields", record);
151	if (!(fldtab[i].tval&FLD))
c48090b0	152	strfree(fldtab[i].sval);
b9bbf539 KM	153	fldtab[i].sval = fr;
	154	fldtab[i].tval = FLD \| STR;
	155	do
	156	fr++ = r++;
	157	while (r != ' ' && r != '\t' && r != '\n' && r != '\0');
	158	*fr++ = 0;
	159	}
	160	else if (r != 0) / if 0, it's a null field */
	161	for (;;) {
	162	i++;
	163	if (i >= MAXFLD)
	164	error(FATAL, "record `%.20s...' has too many fields", record);
	165	if (!(fldtab[i].tval&FLD))
c48090b0	166	strfree(fldtab[i].sval);
b9bbf539 KM	167	fldtab[i].sval = fr;
	168	fldtab[i].tval = FLD \| STR;
	169	while (r != sep && r != '\n' && r != '\0') / \n always a separator */
	170	fr++ = r++;
	171	*fr++ = 0;
	172	if (*r++ == 0)
	173	break;
	174	}
	175	*fr = 0;
	176	for (j=MAXFLD-1; j>i; j--) { /* clean out junk from previous record */
	177	if (!(fldtab[j].tval&FLD))
c48090b0	178	strfree(fldtab[j].sval);
b9bbf539	179	fldtab[j].tval = STR \| FLD;
c48090b0	180	fldtab[j].sval = EMPTY;
b9bbf539 KM	181	}
	182	maxfld = i;
	183	donefld = 1;
	184	for(i=1; i<=maxfld; i++)
89616477	185	if(isanumber(fldtab[i].sval)) {
b9bbf539 KM	186	fldtab[i].fval = atof(fldtab[i].sval);
	187	fldtab[i].tval \|= NUM;
	188	}
	189	setfval(lookup("NF", symtab, 0), (awkfloat) maxfld);
	190	if (dbg)
	191	for (i = 0; i <= maxfld; i++)
	192	printf("field %d: \|%s\|\n", i, fldtab[i].sval);
	193	}
	194
	195	recbld()
	196	{
	197	int i;
	198	register char r, p;
	199
	200	if (donefld == 0 \|\| donerec == 1)
	201	return;
	202	r = record;
	203	for (i = 1; i <= *NF; i++) {
	204	p = getsval(&fldtab[i]);
	205	while (r++ = p++)
	206	;
	207	(r-1) = *OFS;
	208	}
	209	*(r-1) = '\0';
	210	dprintf("in recbld FS=%o, recloc=%o\n", **FS, recloc, NULL);
	211	recloc->tval = STR \| FLD;
	212	dprintf("in recbld FS=%o, recloc=%o\n", **FS, recloc, NULL);
	213	if (r > record+RECSIZE)
	214	error(FATAL, "built giant record `%.20s...'", record);
	215	dprintf("recbld = \|%s\|\n", record, NULL, NULL);
	216	}
	217
	218	cell *fieldadr(n)
	219	{
	220	if (n >= MAXFLD)
	221	error(FATAL, "trying to access field %d", n);
	222	return(&fldtab[n]);
	223	}
	224
	225	int errorflag = 0;
	226
	227	yyerror(s) char *s; {
	228	fprintf(stderr, "awk: %s near line %d\n", s, lineno);
	229	errorflag = 2;
	230	}
	231
66d1b859 CT	232	int
	233	#ifdef __STDC__
	234	error(int isfatal, const char *fmt, ...)
	235	#else
	236	error(isfatal, fmt, va_alist)
2404822a KB	237	int isfatal;
2404822a KB	238	char *fmt;
66d1b859 CT	239	va_dcl
66d1b859 CT	240	#endif
2404822a KB	241	{
	242	va_list ap;
	243
66d1b859	244	#ifdef __STDC__
2404822a	245	va_start(ap, fmt);
66d1b859 CT	246	#else
	247	va_start(ap);
	248	#endif
2404822a KB	249	(void)fprintf(stderr, "awk: ");
	250	(void)vfprintf(stderr, fmt, ap);
	251	va_end(ap);
	252	(void)fprintf(stderr, "\n");
8820cba9	253	if (NR && *NR > 0)
2404822a KB	254	(void)fprintf(stderr, " record number %g\n", *NR);
2404822a KB	255	if (isfatal)
b9bbf539 KM	256	exit(2);
	257	}
	258
	259	PUTS(s) char *s; {
	260	dprintf("%s\n", s, NULL, NULL);
	261	}
	262
	263	#define MAXEXPON 38 /* maximum exponenet for fp number */
	264
89616477	265	isanumber(s)
b9bbf539 KM	266	register char *s;
	267	{
	268	register d1, d2;
	269	int point;
	270	char *es;
	271
c48090b0 SL	272	if (s == NULL)
c48090b0 SL	273	return (0);
b9bbf539 KM	274	d1 = d2 = point = 0;
	275	while (s == ' ' \|\| s == '\t' \|\| *s == '\n')
	276	s++;
	277	if (*s == '\0')
	278	return(0); /* empty stuff isn't number */
	279	if (s == '+' \|\| s == '-')
	280	s++;
	281	if (!isdigit(s) && s != '.')
	282	return(0);
	283	if (isdigit(*s)) {
	284	do {
	285	d1++;
	286	s++;
	287	} while (isdigit(*s));
	288	}
	289	if(d1 >= MAXEXPON)
	290	return(0); /* too many digits to convert */
	291	if (*s == '.') {
	292	point++;
	293	s++;
	294	}
	295	if (isdigit(*s)) {
	296	d2++;
	297	do {
	298	s++;
	299	} while (isdigit(*s));
	300	}
	301	if (!(d1 \|\| point && d2))
	302	return(0);
	303	if (s == 'e' \|\| s == 'E') {
	304	s++;
	305	if (s == '+' \|\| s == '-')
	306	s++;
	307	if (!isdigit(*s))
	308	return(0);
	309	es = s;
	310	do {
	311	s++;
	312	} while (isdigit(*s));
	313	if (s - es > 2)
	314	return(0);
	315	else if (s - es == 2 && 10 * (es-'0') + (es+1)-'0' >= MAXEXPON)
	316	return(0);
	317	}
	318	while (s == ' ' \|\| s == '\t' \|\| *s == '\n')
	319	s++;
	320	if (*s == '\0')
	321	return(1);
	322	else
	323	return(0);
	324	}
	325	/*
89616477	326	isanumber(s) char *s; {return(0);}
b9bbf539	327	*/