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