xref: /illumos-gate/usr/src/cmd/awk/main.c (revision 7c478bd9)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License, Version 1.0 only
6  * (the "License").  You may not use this file except in compliance
7  * with the License.
8  *
9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10  * or http://www.opensolaris.org/os/licensing.
11  * See the License for the specific language governing permissions
12  * and limitations under the License.
13  *
14  * When distributing Covered Code, include this CDDL HEADER in each
15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16  * If applicable, add the following below this CDDL HEADER, with the
17  * fields enclosed by brackets "[]" replaced with your own identifying
18  * information: Portions Copyright [yyyy] [name of copyright owner]
19  *
20  * CDDL HEADER END
21  */
22 /*
23  * Copyright (c) 1999 by Sun Microsystems, Inc.
24  * All rights reserved.
25  */
26 /*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
27 /*	  All Rights Reserved  	*/
28 
29 
30 #pragma ident	"%Z%%M%	%I%	%E% SMI"	/* SVr4.0 2.12	*/
31 
32 #include <stdio.h>
33 #include <ctype.h>
34 #include <signal.h>
35 #include <locale.h>
36 #include <libintl.h>
37 #include <stdarg.h>
38 #include <errno.h>
39 #include <values.h>
40 #include <langinfo.h>
41 #include "awk.h"
42 #include "y.tab.h"
43 
44 char	*version = "version Oct 11, 1989";
45 
46 int	dbg	= 0;
47 uchar	*cmdname;	/* gets argv[0] for error messages */
48 extern	FILE	*yyin;	/* lex input file */
49 uchar	*lexprog;	/* points to program argument if it exists */
50 extern	int errorflag;	/* non-zero if any syntax errors; set by yyerror */
51 int	compile_time = 2;	/* for error printing: */
52 				/* 2 = cmdline, 1 = compile, 0 = running */
53 
54 uchar	*pfile[20];	/* program filenames from -f's */
55 int	npfile = 0;	/* number of filenames */
56 int	curpfile = 0;	/* current filename */
57 char	radixpoint = '.';
58 
59 main(argc, argv, envp)
60 	int argc;
61 	uchar *argv[], *envp[];
62 {
63 	uchar *fs = NULL;
64 	char	*nl_radix;
65 	extern void fpecatch();
66 	/*
67 	 * At this point, numbers are still scanned as in
68 	 * the POSIX locale.
69 	 * (POSIX.2, volume 2, P867, L4742-4757)
70 	 */
71 	(void) setlocale(LC_ALL, "");
72 	(void) setlocale(LC_NUMERIC, "C");
73 #if !defined(TEXT_DOMAIN)	/* Should be defined by cc -D */
74 #define	TEXT_DOMAIN	"SYS_TEST"	/* Use this only if it weren't */
75 #endif
76 	(void) textdomain(TEXT_DOMAIN);
77 	cmdname = argv[0];
78 	if (argc == 1) {
79 		fprintf(stderr, gettext(
80 			"Usage: %s [-f programfile | 'program'] [-Ffieldsep] "
81 			"[-v var=value] [files]\n"), cmdname);
82 		exit(1);
83 	}
84 	signal(SIGFPE, fpecatch);
85 	yyin = NULL;
86 	syminit();
87 	while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') {
88 		if (strcmp(argv[1], "--") == 0) {
89 			/* explicit end of args */
90 			argc--;
91 			argv++;
92 			break;
93 		}
94 		switch (argv[1][1]) {
95 		case 'f':	/* next argument is program filename */
96 			argc--;
97 			argv++;
98 			if (argc <= 1)
99 				ERROR "no program filename" FATAL;
100 			pfile[npfile++] = argv[1];
101 			break;
102 		case 'F':	/* set field separator */
103 			if (argv[1][2] != 0) {	/* arg is -Fsomething */
104 				/* wart: t=>\t */
105 				if (argv[1][2] == 't' && argv[1][3] == 0)
106 					fs = (uchar *) "\t";
107 				else if (argv[1][2] != 0)
108 					fs = &argv[1][2];
109 			} else {		/* arg is -F something */
110 				argc--; argv++;
111 				if (argc > 1) {
112 					/* wart: t=>\t */
113 					if (argv[1][0] == 't' &&
114 						argv[1][1] == 0)
115 						fs = (uchar *) "\t";
116 					else if (argv[1][0] != 0)
117 						fs = &argv[1][0];
118 				}
119 			}
120 			if (fs == NULL || *fs == '\0')
121 				ERROR "field separator FS is empty" WARNING;
122 			break;
123 		case 'v':	/* -v a=1 to be done NOW.  one -v for each */
124 			if (argv[1][2] == '\0' && --argc > 1 &&
125 				isclvar((++argv)[1]))
126 				setclvar(argv[1]);
127 			break;
128 		case 'd':
129 			dbg = atoi(&argv[1][2]);
130 			if (dbg == 0)
131 				dbg = 1;
132 			printf("awk %s\n", version);
133 			break;
134 		default:
135 			ERROR "unknown option %s ignored", argv[1] WARNING;
136 			break;
137 		}
138 		argc--;
139 		argv++;
140 	}
141 	/* argv[1] is now the first argument */
142 	if (npfile == 0) {	/* no -f; first argument is program */
143 		if (argc <= 1)
144 			ERROR "no program given" FATAL;
145 		dprintf(("program = |%s|\n", argv[1]));
146 		lexprog = argv[1];
147 		argc--;
148 		argv++;
149 	}
150 	compile_time = 1;
151 	argv[0] = cmdname;	/* put prog name at front of arglist */
152 	dprintf(("argc=%d, argv[0]=%s\n", argc, argv[0]));
153 	arginit(argc, argv);
154 	envinit(envp);
155 	yyparse();
156 	if (fs)
157 		*FS = tostring(qstring(fs, '\0'));
158 	dprintf(("errorflag=%d\n", errorflag));
159 	/*
160 	 * done parsing, so now activate the LC_NUMERIC
161 	 */
162 	(void) setlocale(LC_ALL, "");
163 	nl_radix = nl_langinfo(RADIXCHAR);
164 	if (nl_radix)
165 		radixpoint = *nl_radix;
166 
167 	if (errorflag == 0) {
168 		compile_time = 0;
169 		run(winner);
170 	} else
171 		bracecheck();
172 	exit(errorflag);
173 }
174 
175 pgetc()		/* get program character */
176 {
177 	int c;
178 
179 	for (;;) {
180 		if (yyin == NULL) {
181 			if (curpfile >= npfile)
182 				return (EOF);
183 			yyin = (strcmp((char *) pfile[curpfile], "-") == 0) ?
184 				stdin : fopen((char *) pfile[curpfile], "r");
185 			if (yyin == NULL)
186 				ERROR "can't open file %s",
187 					pfile[curpfile] FATAL;
188 		}
189 		if ((c = getc(yyin)) != EOF)
190 			return (c);
191 		yyin = NULL;
192 		curpfile++;
193 	}
194 }
195