getrec(uchar *buf) /* get next input record from whatever source */
{ /* note: tests whether buf == record */
int c;
static int firsttime = 1;
if (firsttime) {
firsttime = 0;
initgetrec();
}
dprintf( ("RS=<%s>, FS=<%s>, ARGC=%g, FILENAME=%s\n",
*RS, *FS, *ARGC, *FILENAME) );
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((char *)file, "r")) == NULL)
ERROR "can't open file %s", file FATAL;
setfval(fnrloc, 0.0);
}
c = readrec(buf, recsize, infile);
if (c != 0 || buf[0] != '\0') { /* normal record */
if (buf == record) {
if (!(recloc->tval & DONTFREE))
xfree(recloc->sval);
recloc->sval = record;
recloc->tval = REC | STR | DONTFREE;
if (isnumber(recloc->sval)) {
recloc->fval = atof(recloc->sval);
recloc->tval |= NUM;
}
}
setfval(nrloc, nrloc->fval+1);
setfval(fnrloc, fnrloc->fval+1);
return 1;
}
/* EOF arrived on this file; set up next */
if (infile != stdin)
fclose(infile);
infile = NULL;
argno++;
}
return 0; /* true end of file */
}
void initgetrec(void)
{
extern unsigned char **start_delayed, **after_delayed;
unsigned char **pp;
int i;
unsigned char *p;
/* first handle delayed name=val arguments */
for (pp = start_delayed; pp != after_delayed; pp++)
setclvar(*pp);
for (i = 1; i < *ARGC; i++) {
if (!isclvar(p = getargv(i))) /* find 1st real filename */
return;
setclvar(p); /* a commandline assignment before filename */
argno++;
}
infile = stdin; /* no filenames, so use stdin */
/* *FILENAME = file = (unsigned char*) "-"; */
}
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(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) /* normal record */
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 */
}
void initgetrec(void)
{
int i;
uchar *p;
for (i = 1; i < *ARGC; i++) {
if (!isclvar(p = getargv(i))) /* find 1st real filename */
return;
setclvar(p); /* a commandline assignment before filename */
argno++;
}
infile = stdin; /* no filenames, so use stdin */
/* *FILENAME = file = (uchar*) "-"; */
}
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 */
}
void initgetrec(void)
{
int i;
char *p;
for (i = 1; i < *ARGC; i++) {
p = getargv(i); /* find 1st real filename */
if (p == NULL || *p == '\0') { /* deleted or zapped */
argno++;
continue;
}
if (!isclvar(p)) {
setsval(lookup("FILENAME", symtab), p);
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 */
}
int main( int argc, char *argv[] )
{
const char *fs = NULL;
char **eargv = NULL;
int eargc;
setlocale( LC_CTYPE, "" );
setlocale( LC_NUMERIC, "C" ); /* for parsing cmdline & prog */
cmdname = argv[0];
if( argc == 1 ) {
fprintf( stderr,
"usage: %s [-F fs] [-v var=value] [-f progfile | 'prog'] [file ...]\n",
cmdname );
exit( 1 );
}
signal( SIGFPE, fpecatch );
srand_seed = 1;
srand( (unsigned)srand_seed );
yyin = NULL;
symtab = makesymtab( NSYMTAB/NSYMTAB );
if( argc == 2 && argv[1][0] == '@' && argv[1][1] != '\0' ) {
const char *env;
env = getenv( &argv[1][1] );
if( env != NULL ) {
eargc = ParseEnvVar( env, NULL, NULL ); // count parameters.
eargv = malloc( eargc * sizeof( char * ) + strlen( env ) + 1 + eargc );
ParseEnvVar( env, eargv, (char *)( eargv + eargc ) );
argc = eargc;
argv = eargv;
}
}
while( argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0' ) {
if( strcmp( argv[1], "-version" ) == 0 || strcmp( argv[1], "--version" ) == 0 ) {
printf( "awk %s\n", version );
exit( 0 );
break;
}
if( strncmp( argv[1], "--", 2 ) == 0 ) { /* explicit end of args */
argc--;
argv++;
break;
}
switch( argv[1][1] ) {
case 's':
if( strcmp( argv[1], "-safe" ) == 0 )
safe = true;
break;
case 'f': /* next argument is program filename */
if( argv[1][2] != '\0' ) { /* arg is -fsomething */
if( npfile >= MAX_PFILE - 1 ) {
FATAL( "too many -f options" );
}
pfile[npfile++] = unquote( &argv[1][2] );
} else { /* arg is -f something */
argc--; argv++;
if( argc <= 1 ) {
FATAL( "no program filename" );
}
if( npfile >= MAX_PFILE - 1 ) {
FATAL( "too many -f options" );
}
pfile[npfile++] = unquote( argv[1] );
}
break;
case 'F': /* set field separator */
if( argv[1][2] != '\0' ) { /* arg is -Fsomething */
if( argv[1][2] == 't' && argv[1][3] == '\0' ) { /* wart: t=>\t */
fs = "\t";
} else if( argv[1][2] != '\0' ) {
fs = &argv[1][2];
}
} else { /* arg is -F something */
argc--; argv++;
if( argc > 1 && argv[1][0] == 't' && argv[1][1] == '\0' ) { /* wart: t=>\t */
fs = "\t";
} else if( argc > 1 && argv[1][0] != '\0' ) {
fs = &argv[1][0];
}
}
if( fs == NULL || *fs == '\0' )
WARNING( "field separator FS is empty" );
break;
case 'v': /* -v a=1 to be done NOW. one -v for each */
if( argv[1][2] != '\0' ) { /* arg is -vsomething */
char *p;
p = unquote( &argv[1][2] );
if( isclvar( p ) ) {
setclvar( p );
} else {
FATAL( "invalid -v option argument: %s", p );
}
} else { /* arg is -v something */
char *p;
argc--; argv++;
if( argc <= 1 ) {
FATAL( "no variable name" );
}
p = unquote( argv[1] );
if( isclvar( p ) ) {
setclvar( p );
} else {
FATAL( "invalid -v option argument: %s", p );
}
}
break;
case 'd':
dbg = atoi( &argv[1][2] );
if( dbg == 0 )
dbg = 1;
printf( "awk %s\n", version );
break;
default:
WARNING( "unknown option %s ignored", argv[1] );
break;
}
argc--;
argv++;
}
/* argv[1] is now the first argument */
if( npfile == 0 ) { /* no -f; first argument is program */
char *p;
if( argc <= 1 ) {
if( dbg )
exit( 0 );
FATAL( "no program given" );
}
p = unquote( argv[1] );
dprintf(( "program = |%s|\n", p ));
lexprog = p;
argc--;
argv++;
}
for( eargc = 1; eargc < argc; ++eargc ) {
argv[eargc] = unquote( argv[eargc] );
}
recinit( recsize );
syminit();
compile_time = 1;
argv[0] = cmdname; /* put prog name at front of arglist */
dprintf(( "argc=%d, argv[0]=%s\n", argc, argv[0] ));
arginit( argc, argv );
if( !safe )
envinit( environ );
yyparse();
setlocale( LC_NUMERIC, "" ); /* back to whatever it is locally */
if( fs )
*FS = qstring( fs, '\0' );
dprintf(( "errorflag=%d\n", errorflag ));
if( errorflag == 0 ) {
compile_time = 0;
run( winner );
} else {
bracecheck();
}
return( errorflag );
}
int main(int argc, const char *argv[])
{
const char *fs = NULL, *marg;
int temp;
__progname = argv[0];
_wildcard(&argc, &argv);
setlocale(LC_ALL, "");
cmdname = __progname;
if (argc == 1) {
fprintf(stderr, "Usage: %s [-f programfile | 'program'] [-Ffieldsep] [-v var=value] [-safe] [-mrn] [-mfn] [files]\n", cmdname);
exit(1);
}
signal(SIGFPE, fpecatch);
yyin = NULL;
symtab = makesymtab(NSYMTAB);
while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') {
if (strcmp(argv[1], "--") == 0) { /* explicit end of args */
argc--;
argv++;
break;
}
switch (argv[1][1]) {
case 's':
if (strcmp(argv[1], "-safe") == 0)
safe = 1;
break;
case 'f': /* next argument is program filename */
argc--;
argv++;
if (argc <= 1)
ERROR "no program filename" FATAL;
pfile[npfile++] = argv[1];
break;
case 'F': /* set field separator */
if (argv[1][2] != 0) { /* arg is -Fsomething */
if (argv[1][2] == 't' && argv[1][3] == 0) /* wart: t=>\t */
fs = "\t";
else if (argv[1][2] != 0)
fs = &argv[1][2];
} else { /* arg is -F something */
argc--; argv++;
if (argc > 1 && argv[1][0] == 't' && argv[1][1] == 0) /* wart: t=>\t */
fs = "\t";
else if (argc > 1 && argv[1][0] != 0)
fs = &argv[1][0];
}
if (fs == NULL || *fs == '\0')
ERROR "field separator FS is empty" WARNING;
break;
case 'v': /* -v a=1 to be done NOW. one -v for each */
if (argv[1][2] == '\0' && --argc > 1 && isclvar((++argv)[1]))
setclvar((char*)argv[1]);
break;
case 'm': /* more memory: -mr=record, -mf=fields */
/* no longer needed */
marg = argv[1];
if (argv[1][3])
temp = atoi(&argv[1][3]);
else {
argv++; argc--;
temp = atoi(&argv[1][0]);
}
switch (marg[2]) {
case 'r': recsize = temp; break;
case 'f': nfields = temp; break;
default: ERROR "unknown option %s\n", marg FATAL;
}
break;
case 'd':
dbg = atoi(&argv[1][2]);
if (dbg == 0)
dbg = 1;
printf("awk %s\n", version);
break;
case 'V': /* added for exptools "standard" */
printf("awk %s\n", version);
exit(0);
break;
default:
ERROR "unknown option %s ignored", argv[1] WARNING;
break;
}
argc--;
argv++;
}
/* argv[1] is now the first argument */
if (npfile == 0) { /* no -f; first argument is program */
if (argc <= 1) {
if (dbg)
exit(0);
ERROR "no program given" FATAL;
}
dprintf( ("program = |%s|\n", argv[1]) );
lexprog = argv[1];
argc--;
argv++;
}
recinit(recsize);
syminit();
compile_time = 1;
argv[0] = cmdname; /* put prog name at front of arglist */
dprintf( ("argc=%d, argv[0]=%s\n", argc, argv[0]) );
arginit(argc, argv);
if (!safe)
envinit(environ);
yyparse();
if (fs)
*FS = qstring(fs, '\0');
dprintf( ("errorflag=%d\n", errorflag) );
if (errorflag == 0) {
compile_time = 0;
run(winner);
} else
bracecheck();
return(errorflag);
}
int main(int argc, char *argv[])
{
const char *fs = NULL;
setlocale(LC_CTYPE, "");
setlocale(LC_NUMERIC, "C"); /* for parsing cmdline & prog */
cmdname = argv[0];
if (argc == 1) {
fprintf(stderr,
"usage: %s [-F fs] [-v var=value] [-f progfile | 'prog'] [file ...]\n",
cmdname);
exit(1);
}
signal(SIGFPE, fpecatch);
yyin = NULL;
symtab = makesymtab(NSYMTAB/NSYMTAB);
while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') {
if (strcmp(argv[1],"-version") == 0 || strcmp(argv[1],"--version") == 0) {
printf("awk %s\n", version);
exit(0);
break;
}
if (strncmp(argv[1], "--", 2) == 0) { /* explicit end of args */
argc--;
argv++;
break;
}
switch (argv[1][1]) {
case 's':
if (strcmp(argv[1], "-safe") == 0)
safe = 1;
break;
case 'f': /* next argument is program filename */
argc--;
argv++;
if (argc <= 1)
FATAL("no program filename");
if (npfile >= MAX_PFILE - 1)
FATAL("too many -f options");
pfile[npfile++] = argv[1];
break;
case 'F': /* set field separator */
if (argv[1][2] != 0) { /* arg is -Fsomething */
if (argv[1][2] == 't' && argv[1][3] == 0) /* wart: t=>\t */
fs = "\t";
else if (argv[1][2] != 0)
fs = &argv[1][2];
} else { /* arg is -F something */
argc--; argv++;
if (argc > 1 && argv[1][0] == 't' && argv[1][1] == 0) /* wart: t=>\t */
fs = "\t";
else if (argc > 1 && argv[1][0] != 0)
fs = &argv[1][0];
}
if (fs == NULL || *fs == '\0')
WARNING("field separator FS is empty");
break;
case 'v': /* -v a=1 to be done NOW. one -v for each */
if (argv[1][2] == '\0' && --argc > 1 && isclvar((++argv)[1]))
setclvar(argv[1]);
break;
case 'd':
dbg = atoi(&argv[1][2]);
if (dbg == 0)
dbg = 1;
printf("awk %s\n", version);
break;
default:
WARNING("unknown option %s ignored", argv[1]);
break;
}
argc--;
argv++;
}
/* argv[1] is now the first argument */
if (npfile == 0) { /* no -f; first argument is program */
if (argc <= 1) {
if (dbg)
exit(0);
FATAL("no program given");
}
dprintf( ("program = |%s|\n", argv[1]) );
lexprog = argv[1];
argc--;
argv++;
}
recinit(recsize);
syminit();
compile_time = 1;
argv[0] = cmdname; /* put prog name at front of arglist */
dprintf( ("argc=%d, argv[0]=%s\n", argc, argv[0]) );
arginit(argc, argv);
if (!safe)
envinit(environ);
yyparse();
setlocale(LC_NUMERIC, ""); /* back to whatever it is locally */
if (fs)
*FS = qstring(fs, '\0');
dprintf( ("errorflag=%d\n", errorflag) );
if (errorflag == 0) {
compile_time = 0;
run(winner);
} else
bracecheck();
return(errorflag);
}
int
getrec(void)
{
register char *rr;
extern int svargc;
extern char **svargv;
register int c, sep, k, m, n;
wchar_t wc;
dprintf("**RS=%o, **FS=%o\n", **RS, **FS);
donefld = 0;
donerec = 1;
record[0] = 0;
while (svargc > 0) {
dprintf("svargc=%d, *svargv=%s\n", svargc, *svargv);
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);
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);
}
next(wc, *RS, n);
if ((sep = **RS) == 0)
sep = '\n';
for (rr = record; ; ) {
cont: for (; (c=getc(infile)) != sep && c != EOF; *rr++ = c) {
if (rr >= record+RECSIZE-n-3) {
size_t diff = rr - record;
growrec();
rr = &record[diff];
}
}
if (c != EOF) {
/*
* Note: This code does not restrict occurences
* of the multibyte sequence in RS to the start
* of an input character.
*/
for (m = 1; m < n; m++) {
if ((c = getc(infile)) == EOF ||
c != (*RS)[m]) {
for (k = 0; k < m; k++)
*rr++ = (*RS)[k];
if (c == EOF)
break;
*rr++ = c;
goto cont;
}
}
}
if (**RS == sep || c == EOF)
break;
if ((c = getc(infile)) == '\n' || c == EOF) /* 2 in a row */
break;
if (rr >= record+RECSIZE-n-3) {
size_t diff = rr - record;
growrec();
rr = &record[diff];
}
*rr++ = '\n';
*rr++ = c;
}
*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 */
}
int getrec(unsigned char **buf, int *bufsize)
{
int c, saved;
static int firsttime = 1;
if (firsttime) {
firsttime = 0;
initgetrec();
}
dprintf( ("RS=<%s>, FS=<%s>, ARGC=%d, FILENAME=%s\n",
*RS ? *RS : tostring(""),
*FS ? *FS : tostring(""),
(int) *ARGC,
*FILENAME ? *FILENAME : tostring("")) );
donefld = 0;
donerec = 1;
if (*bufsize == 0) {
if ((*buf = malloc(*bufsize = CHUNK)) == NULL)
error(MM_ERROR, outofspace, "getrec");
**buf = '\0';
}
saved = (*buf)[0];
(*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((char *)file, "r")) == NULL)
error(MM_ERROR, badopen, file, strerror(errno));
setfval(fnrloc, 0.0);
}
c = readrec(buf, bufsize, infile);
if (c != 0 || (*buf)[0] != '\0') { /* normal record */
if (*buf == record) {
if (!(recloc->tval & DONTFREE))
xfree(recloc->sval);
recloc->sval = record;
recloc->tval = REC | STR | DONTFREE;
(void)is2number(0, recloc);
}
setfval(nrloc, nrloc->fval+1);
setfval(fnrloc, fnrloc->fval+1);
return 1;
}
/* EOF arrived on this file; set up next */
if (infile != stdin)
fclose(infile);
infile = NULL;
argno++;
}
/*
* POSIX.2 requires that NF stick with its last value
* at the start of the END code. The most straightforward
* way to do this is to restore the contents of record
* [==buf when called from program()] so that getnf() will
* recompute the same NF value unless something strange
* occurs. This has the side effect of $0...$NF *also*
* having sticky values into END, but that seems to match
* the spirit of POSIX.2's rule for NF.
*/
if (posix)
(*buf)[0] = saved;
return 0; /* true end of file */
}
