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 25const char *version = "version 20121220"; 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; 41Awkfloat srand_seed = 1; 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); 71 72 srand_seed = 1; 73 srand(srand_seed); 74 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 */ 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 } 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 */ 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 } 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} 215