main.c (5282B)
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 char *version = "version 19990602"; 26 27 #define DEBUG 28 #include <stdio.h> 29 #include <ctype.h> 30 #include <stdlib.h> 31 #include <string.h> 32 #include <signal.h> 33 #include "awk.h" 34 #include "y.tab.h" 35 36 extern char **environ; 37 extern int nfields; 38 39 int dbg = 0; 40 char *cmdname; /* gets argv[0] for error messages */ 41 extern FILE *yyin; /* lex input file */ 42 char *lexprog; /* points to program argument if it exists */ 43 extern int errorflag; /* non-zero if any syntax errors; set by yyerror */ 44 int compile_time = 2; /* for error printing: */ 45 /* 2 = cmdline, 1 = compile, 0 = running */ 46 47 char *pfile[20]; /* program filenames from -f's */ 48 int npfile = 0; /* number of filenames */ 49 int curpfile = 0; /* current filename */ 50 51 int safe = 0; /* 1 => "safe" mode */ 52 53 int main(int argc, char *argv[]) 54 { 55 char *fs = NULL, *marg; 56 int temp; 57 58 cmdname = argv[0]; 59 if (argc == 1) { 60 fprintf(stderr, "Usage: %s [-F fieldsep] [-mf n] [-mr n] [-v var=value] [-f programfile | 'program'] [file ...]\n", cmdname); 61 exit(1); 62 } 63 signal(SIGFPE, fpecatch); 64 yyin = NULL; 65 symtab = makesymtab(NSYMTAB); 66 while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') { 67 if (strcmp(argv[1], "--") == 0) { /* explicit end of args */ 68 argc--; 69 argv++; 70 break; 71 } 72 switch (argv[1][1]) { 73 case 's': 74 if (strcmp(argv[1], "-safe") == 0) 75 safe = 1; 76 break; 77 case 'f': /* next argument is program filename */ 78 argc--; 79 argv++; 80 if (argc <= 1) 81 FATAL("no program filename"); 82 pfile[npfile++] = argv[1]; 83 break; 84 case 'F': /* set field separator */ 85 if (argv[1][2] != 0) { /* arg is -Fsomething */ 86 if (argv[1][2] == 't' && argv[1][3] == 0) /* wart: t=>\t */ 87 fs = "\t"; 88 else if (argv[1][2] != 0) 89 fs = &argv[1][2]; 90 } else { /* arg is -F something */ 91 argc--; argv++; 92 if (argc > 1 && argv[1][0] == 't' && argv[1][1] == 0) /* wart: t=>\t */ 93 fs = "\t"; 94 else if (argc > 1 && argv[1][0] != 0) 95 fs = &argv[1][0]; 96 } 97 if (fs == NULL || *fs == '\0') 98 WARNING("field separator FS is empty"); 99 break; 100 case 'v': /* -v a=1 to be done NOW. one -v for each */ 101 if (argv[1][2] == '\0' && --argc > 1 && isclvar((++argv)[1])) 102 setclvar(argv[1]); 103 break; 104 case 'm': /* more memory: -mr=record, -mf=fields */ 105 /* no longer needed */ 106 marg = argv[1]; 107 if (argv[1][3]) 108 temp = atoi(&argv[1][3]); 109 else { 110 argv++; argc--; 111 temp = atoi(&argv[1][0]); 112 } 113 switch (marg[2]) { 114 case 'r': recsize = temp; break; 115 case 'f': nfields = temp; break; 116 default: FATAL("unknown option %s\n", marg); 117 } 118 break; 119 case 'd': 120 dbg = atoi(&argv[1][2]); 121 if (dbg == 0) 122 dbg = 1; 123 printf("awk %s\n", version); 124 break; 125 case 'V': /* added for exptools "standard" */ 126 printf("awk %s\n", version); 127 exit(0); 128 break; 129 default: 130 WARNING("unknown option %s ignored", argv[1]); 131 break; 132 } 133 argc--; 134 argv++; 135 } 136 /* argv[1] is now the first argument */ 137 if (npfile == 0) { /* no -f; first argument is program */ 138 if (argc <= 1) { 139 if (dbg) 140 exit(0); 141 FATAL("no program given"); 142 } 143 dprintf( ("program = |%s|\n", argv[1]) ); 144 lexprog = argv[1]; 145 argc--; 146 argv++; 147 } 148 recinit(recsize); 149 syminit(); 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 if (!safe) 155 envinit(environ); 156 yyparse(); 157 if (fs) 158 *FS = qstring(fs, '\0'); 159 dprintf( ("errorflag=%d\n", errorflag) ); 160 if (errorflag == 0) { 161 compile_time = 0; 162 run(winner); 163 } else 164 bracecheck(); 165 return(errorflag); 166 } 167 168 int pgetc(void) /* get 1 character from awk program */ 169 { 170 int c; 171 172 for (;;) { 173 if (yyin == NULL) { 174 if (curpfile >= npfile) 175 return EOF; 176 if (strcmp(pfile[curpfile], "-") == 0) 177 yyin = stdin; 178 else if ((yyin = fopen(pfile[curpfile], "r")) == NULL) 179 FATAL("can't open file %s", pfile[curpfile]); 180 lineno = 1; 181 } 182 if ((c = getc(yyin)) != EOF) 183 return c; 184 if (yyin != stdin) 185 fclose(yyin); 186 yyin = NULL; 187 curpfile++; 188 } 189 } 190 191 char *cursource(void) /* current source file name */ 192 { 193 if (npfile > 0) 194 return pfile[curpfile]; 195 else 196 return NULL; 197 } 198