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 20210724"; 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 36 extern char **environ; 37 extern int nfields; 38 39 int dbg = 0; 40 Awkfloat srand_seed = 1; 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 enum compile_states compile_time = ERROR_PRINTING; 46 47 static char **pfile; /* program filenames from -f's */ 48 static size_t maxpfile; /* max program filename */ 49 static size_t npfile; /* number of filenames */ 50 static size_t curpfile; /* current filename */ 51 52 bool safe = false; /* true => "safe" mode */ 53 54 static noreturn void fpecatch(int n 55 #ifdef SA_SIGINFO 56 , siginfo_t *si, void *uc 57 #endif 58 ) 59 { 60 #ifdef SA_SIGINFO 61 static const char *emsg[] = { 62 [0] = "Unknown error", 63 [FPE_INTDIV] = "Integer divide by zero", 64 [FPE_INTOVF] = "Integer overflow", 65 [FPE_FLTDIV] = "Floating point divide by zero", 66 [FPE_FLTOVF] = "Floating point overflow", 67 [FPE_FLTUND] = "Floating point underflow", 68 [FPE_FLTRES] = "Floating point inexact result", 69 [FPE_FLTINV] = "Invalid Floating point operation", 70 [FPE_FLTSUB] = "Subscript out of range", 71 }; 72 #endif 73 FATAL("floating point exception" 74 #ifdef SA_SIGINFO 75 ": %s", (size_t)si->si_code < sizeof(emsg) / sizeof(emsg[0]) && 76 emsg[si->si_code] ? emsg[si->si_code] : emsg[0] 77 #endif 78 ); 79 } 80 81 /* Can this work with recursive calls? I don't think so. 82 void segvcatch(int n) 83 { 84 FATAL("segfault. Do you have an unbounded recursive call?", n); 85 } 86 */ 87 88 static const char * 89 setfs(char *p) 90 { 91 /* wart: t=>\t */ 92 if (p[0] == 't' && p[1] == '\0') { 93 WARNING("-Ft to imply tab separator is deprecated behavior."); 94 return "\t"; 95 } 96 return p; 97 } 98 99 static char * 100 getarg(int *argc, char ***argv, const char *msg) 101 { 102 if ((*argv)[1][2] != '\0') { /* arg is -fsomething */ 103 return &(*argv)[1][2]; 104 } else { /* arg is -f something */ 105 (*argc)--; (*argv)++; 106 if (*argc <= 1) 107 FATAL("%s", msg); 108 return (*argv)[1]; 109 } 110 } 111 112 int main(int argc, char *argv[]) 113 { 114 const char *fs = NULL; 115 char *fn, *vn; 116 117 setlocale(LC_CTYPE, ""); 118 setlocale(LC_NUMERIC, "C"); /* for parsing cmdline & prog */ 119 cmdname = argv[0]; 120 if (argc == 1) { 121 fprintf(stderr, 122 "usage: %s [-F fs] [-v var=value] [-f progfile | 'prog'] [file ...]\n", 123 cmdname); 124 exit(1); 125 } 126 #ifdef SA_SIGINFO 127 { 128 struct sigaction sa; 129 sa.sa_sigaction = fpecatch; 130 sa.sa_flags = SA_SIGINFO; 131 sigemptyset(&sa.sa_mask); 132 (void)sigaction(SIGFPE, &sa, NULL); 133 } 134 #else 135 (void)signal(SIGFPE, fpecatch); 136 #endif 137 /*signal(SIGSEGV, segvcatch); experiment */ 138 139 /* Set and keep track of the random seed */ 140 srand_seed = 1; 141 srandom((unsigned long) srand_seed); 142 143 yyin = NULL; 144 symtab = makesymtab(NSYMTAB/NSYMTAB); 145 while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') { 146 if (strcmp(argv[1], "-version") == 0 || strcmp(argv[1], "--version") == 0) { 147 printf("awk %s\n", version); 148 return 0; 149 } 150 if (strcmp(argv[1], "--") == 0) { /* explicit end of args */ 151 argc--; 152 argv++; 153 break; 154 } 155 switch (argv[1][1]) { 156 case 's': 157 if (strcmp(argv[1], "-safe") == 0) 158 safe = true; 159 break; 160 case 'f': /* next argument is program filename */ 161 fn = getarg(&argc, &argv, "no program filename"); 162 if (npfile >= maxpfile) { 163 maxpfile += 20; 164 pfile = (char **) realloc(pfile, maxpfile * sizeof(*pfile)); 165 if (pfile == NULL) 166 FATAL("error allocating space for -f options"); 167 } 168 pfile[npfile++] = fn; 169 break; 170 case 'F': /* set field separator */ 171 fs = setfs(getarg(&argc, &argv, "no field separator")); 172 break; 173 case 'v': /* -v a=1 to be done NOW. one -v for each */ 174 vn = getarg(&argc, &argv, "no variable name"); 175 if (isclvar(vn)) 176 setclvar(vn); 177 else 178 FATAL("invalid -v option argument: %s", vn); 179 break; 180 case 'd': 181 dbg = atoi(&argv[1][2]); 182 if (dbg == 0) 183 dbg = 1; 184 printf("awk %s\n", version); 185 break; 186 default: 187 WARNING("unknown option %s ignored", argv[1]); 188 break; 189 } 190 argc--; 191 argv++; 192 } 193 /* argv[1] is now the first argument */ 194 if (npfile == 0) { /* no -f; first argument is program */ 195 if (argc <= 1) { 196 if (dbg) 197 exit(0); 198 FATAL("no program given"); 199 } 200 DPRINTF("program = |%s|\n", argv[1]); 201 lexprog = argv[1]; 202 argc--; 203 argv++; 204 } 205 recinit(recsize); 206 syminit(); 207 compile_time = COMPILING; 208 argv[0] = cmdname; /* put prog name at front of arglist */ 209 DPRINTF("argc=%d, argv[0]=%s\n", argc, argv[0]); 210 arginit(argc, argv); 211 if (!safe) 212 envinit(environ); 213 yyparse(); 214 #if 0 215 // Doing this would comply with POSIX, but is not compatible with 216 // other awks and with what most users expect. So comment it out. 217 setlocale(LC_NUMERIC, ""); /* back to whatever it is locally */ 218 #endif 219 if (fs) 220 *FS = qstring(fs, '\0'); 221 DPRINTF("errorflag=%d\n", errorflag); 222 if (errorflag == 0) { 223 compile_time = RUNNING; 224 run(winner); 225 } else 226 bracecheck(); 227 return(errorflag); 228 } 229 230 int pgetc(void) /* get 1 character from awk program */ 231 { 232 int c; 233 234 for (;;) { 235 if (yyin == NULL) { 236 if (curpfile >= npfile) 237 return EOF; 238 if (strcmp(pfile[curpfile], "-") == 0) 239 yyin = stdin; 240 else if ((yyin = fopen(pfile[curpfile], "r")) == NULL) 241 FATAL("can't open file %s", pfile[curpfile]); 242 lineno = 1; 243 } 244 if ((c = getc(yyin)) != EOF) 245 return c; 246 if (yyin != stdin) 247 fclose(yyin); 248 yyin = NULL; 249 curpfile++; 250 } 251 } 252 253 char *cursource(void) /* current source file name */ 254 { 255 if (npfile > 0) 256 return pfile[curpfile < npfile ? curpfile : curpfile - 1]; 257 else 258 return NULL; 259 } 260