Home | History | Annotate | Download | only in nawk-20071023
      1 /****************************************************************
      2 Copyright (C) Lucent Technologies 1997
      3 All Rights Reserved
      4 
      5 Permission to use, copy, modify, and distribute this software and
      6 its documentation for any purpose and without fee is hereby
      7 granted, provided that the above copyright notice appear in all
      8 copies and that both that the copyright notice and this
      9 permission notice and warranty disclaimer appear in supporting
     10 documentation, and that the name Lucent Technologies or any of
     11 its entities not be used in advertising or publicity pertaining
     12 to distribution of the software without specific, written prior
     13 permission.
     14 
     15 LUCENT DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
     16 INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS.
     17 IN NO EVENT SHALL LUCENT OR ANY OF ITS ENTITIES BE LIABLE FOR ANY
     18 SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
     19 WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER
     20 IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,
     21 ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF
     22 THIS SOFTWARE.
     23 ****************************************************************/
     24 
     25 const char	*version = "version 20070501";
     26 
     27 #define DEBUG
     28 #include <stdio.h>
     29 #include <ctype.h>
     30 #include <locale.h>
     31 #include <stdlib.h>
     32 #include <string.h>
     33 #include <signal.h>
     34 #include "awk.h"
     35 #include "ytab.h"
     36 
     37 extern	char	**environ;
     38 extern	int	nfields;
     39 
     40 int	dbg	= 0;
     41 char	*cmdname;	/* gets argv[0] for error messages */
     42 extern	FILE	*yyin;	/* lex input file */
     43 char	*lexprog;	/* points to program argument if it exists */
     44 extern	int errorflag;	/* non-zero if any syntax errors; set by yyerror */
     45 int	compile_time = 2;	/* for error printing: */
     46 				/* 2 = cmdline, 1 = compile, 0 = running */
     47 
     48 #define	MAX_PFILE	20	/* max number of -f's */
     49 
     50 char	*pfile[MAX_PFILE];	/* program filenames from -f's */
     51 int	npfile = 0;	/* number of filenames */
     52 int	curpfile = 0;	/* current filename */
     53 
     54 int	safe	= 0;	/* 1 => "safe" mode */
     55 
     56 int main(int argc, char *argv[])
     57 {
     58 	const char *fs = NULL;
     59 
     60 	setlocale(LC_CTYPE, "");
     61 	setlocale(LC_NUMERIC, "C"); /* for parsing cmdline & prog */
     62 	cmdname = argv[0];
     63 	if (argc == 1) {
     64 		fprintf(stderr,
     65 		  "usage: %s [-F fs] [-v var=value] [-f progfile | 'prog'] [file ...]\n",
     66 		  cmdname);
     67 		exit(1);
     68 	}
     69 	signal(SIGFPE, fpecatch);
     70 	yyin = NULL;
     71 	symtab = makesymtab(NSYMTAB/NSYMTAB);
     72 	while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') {
     73 		if (strcmp(argv[1],"-version") == 0 || strcmp(argv[1],"--version") == 0) {
     74 			printf("awk %s\n", version);
     75 			exit(0);
     76 			break;
     77 		}
     78 		if (strncmp(argv[1], "--", 2) == 0) {	/* explicit end of args */
     79 			argc--;
     80 			argv++;
     81 			break;
     82 		}
     83 		switch (argv[1][1]) {
     84 		case 's':
     85 			if (strcmp(argv[1], "-safe") == 0)
     86 				safe = 1;
     87 			break;
     88 		case 'f':	/* next argument is program filename */
     89 			argc--;
     90 			argv++;
     91 			if (argc <= 1)
     92 				FATAL("no program filename");
     93 			if (npfile >= MAX_PFILE - 1)
     94 				FATAL("too many -f options");
     95 			pfile[npfile++] = argv[1];
     96 			break;
     97 		case 'F':	/* set field separator */
     98 			if (argv[1][2] != 0) {	/* arg is -Fsomething */
     99 				if (argv[1][2] == 't' && argv[1][3] == 0)	/* wart: t=>\t */
    100 					fs = "\t";
    101 				else if (argv[1][2] != 0)
    102 					fs = &argv[1][2];
    103 			} else {		/* arg is -F something */
    104 				argc--; argv++;
    105 				if (argc > 1 && argv[1][0] == 't' && argv[1][1] == 0)	/* wart: t=>\t */
    106 					fs = "\t";
    107 				else if (argc > 1 && argv[1][0] != 0)
    108 					fs = &argv[1][0];
    109 			}
    110 			if (fs == NULL || *fs == '\0')
    111 				WARNING("field separator FS is empty");
    112 			break;
    113 		case 'v':	/* -v a=1 to be done NOW.  one -v for each */
    114 			if (argv[1][2] == '\0' && --argc > 1 && isclvar((++argv)[1]))
    115 				setclvar(argv[1]);
    116 			break;
    117 		case 'd':
    118 			dbg = atoi(&argv[1][2]);
    119 			if (dbg == 0)
    120 				dbg = 1;
    121 			printf("awk %s\n", version);
    122 			break;
    123 		default:
    124 			WARNING("unknown option %s ignored", argv[1]);
    125 			break;
    126 		}
    127 		argc--;
    128 		argv++;
    129 	}
    130 	/* argv[1] is now the first argument */
    131 	if (npfile == 0) {	/* no -f; first argument is program */
    132 		if (argc <= 1) {
    133 			if (dbg)
    134 				exit(0);
    135 			FATAL("no program given");
    136 		}
    137 		   dprintf( ("program = |%s|\n", argv[1]) );
    138 		lexprog = argv[1];
    139 		argc--;
    140 		argv++;
    141 	}
    142 	recinit(recsize);
    143 	syminit();
    144 	compile_time = 1;
    145 	argv[0] = cmdname;	/* put prog name at front of arglist */
    146 	   dprintf( ("argc=%d, argv[0]=%s\n", argc, argv[0]) );
    147 	arginit(argc, argv);
    148 	if (!safe)
    149 		envinit(environ);
    150 	yyparse();
    151 	setlocale(LC_NUMERIC, ""); /* back to whatever it is locally */
    152 	if (fs)
    153 		*FS = qstring(fs, '\0');
    154 	   dprintf( ("errorflag=%d\n", errorflag) );
    155 	if (errorflag == 0) {
    156 		compile_time = 0;
    157 		run(winner);
    158 	} else
    159 		bracecheck();
    160 	return(errorflag);
    161 }
    162 
    163 int pgetc(void)		/* get 1 character from awk program */
    164 {
    165 	int c;
    166 
    167 	for (;;) {
    168 		if (yyin == NULL) {
    169 			if (curpfile >= npfile)
    170 				return EOF;
    171 			if (strcmp(pfile[curpfile], "-") == 0)
    172 				yyin = stdin;
    173 			else if ((yyin = fopen(pfile[curpfile], "r")) == NULL)
    174 				FATAL("can't open file %s", pfile[curpfile]);
    175 			lineno = 1;
    176 		}
    177 		if ((c = getc(yyin)) != EOF)
    178 			return c;
    179 		if (yyin != stdin)
    180 			fclose(yyin);
    181 		yyin = NULL;
    182 		curpfile++;
    183 	}
    184 }
    185 
    186 char *cursource(void)	/* current source file name */
    187 {
    188 	if (npfile > 0)
    189 		return pfile[curpfile];
    190 	else
    191 		return NULL;
    192 }
    193