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 20110506"; 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 Awkfloat srand_seed = 1; 42 char *cmdname; /* gets argv[0] for error messages */ 43 extern FILE *yyin; /* lex input file */ 44 char *lexprog; /* points to program argument if it exists */ 45 extern int errorflag; /* non-zero if any syntax errors; set by yyerror */ 46 int 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 51 char *pfile[MAX_PFILE]; /* program filenames from -f's */ 52 int npfile = 0; /* number of filenames */ 53 int curpfile = 0; /* current filename */ 54 55 int safe = 0; /* 1 => "safe" mode */ 56 57 int 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' && --argc > 1 && isclvar((++argv)[1])) 125 setclvar(argv[1]); 126 else if (argv[1][2] != '\0') 127 setclvar(&argv[1][2]); 128 break; 129 case 'd': 130 dbg = atoi(&argv[1][2]); 131 if (dbg == 0) 132 dbg = 1; 133 printf("awk %s\n", version); 134 break; 135 default: 136 WARNING("unknown option %s ignored", argv[1]); 137 break; 138 } 139 argc--; 140 argv++; 141 } 142 /* argv[1] is now the first argument */ 143 if (npfile == 0) { /* no -f; first argument is program */ 144 if (argc <= 1) { 145 if (dbg) 146 exit(0); 147 FATAL("no program given"); 148 } 149 dprintf( ("program = |%s|\n", argv[1]) ); 150 lexprog = argv[1]; 151 argc--; 152 argv++; 153 } 154 recinit(recsize); 155 syminit(); 156 compile_time = 1; 157 argv[0] = cmdname; /* put prog name at front of arglist */ 158 dprintf( ("argc=%d, argv[0]=%s\n", argc, argv[0]) ); 159 arginit(argc, argv); 160 if (!safe) 161 envinit(environ); 162 yyparse(); 163 setlocale(LC_NUMERIC, ""); /* back to whatever it is locally */ 164 if (fs) 165 *FS = qstring(fs, '\0'); 166 dprintf( ("errorflag=%d\n", errorflag) ); 167 if (errorflag == 0) { 168 compile_time = 0; 169 run(winner); 170 } else 171 bracecheck(); 172 return(errorflag); 173 } 174 175 int pgetc(void) /* get 1 character from awk program */ 176 { 177 int c; 178 179 for (;;) { 180 if (yyin == NULL) { 181 if (curpfile >= npfile) 182 return EOF; 183 if (strcmp(pfile[curpfile], "-") == 0) 184 yyin = stdin; 185 else if ((yyin = fopen(pfile[curpfile], "r")) == NULL) 186 FATAL("can't open file %s", pfile[curpfile]); 187 lineno = 1; 188 } 189 if ((c = getc(yyin)) != EOF) 190 return c; 191 if (yyin != stdin) 192 fclose(yyin); 193 yyin = NULL; 194 curpfile++; 195 } 196 } 197 198 char *cursource(void) /* current source file name */ 199 { 200 if (npfile > 0) 201 return pfile[curpfile]; 202 else 203 return NULL; 204 } 205