1234219Sadrian/**************************************************************** 2234219SadrianCopyright (C) Lucent Technologies 1997 3234219SadrianAll Rights Reserved 4234219Sadrian 5234219SadrianPermission to use, copy, modify, and distribute this software and 6234219Sadrianits documentation for any purpose and without fee is hereby 7234219Sadriangranted, provided that the above copyright notice appear in all 8234219Sadriancopies and that both that the copyright notice and this 9234219Sadrianpermission notice and warranty disclaimer appear in supporting 10234219Sadriandocumentation, and that the name Lucent Technologies or any of 11234219Sadrianits entities not be used in advertising or publicity pertaining 12234219Sadrianto distribution of the software without specific, written prior 13234219Sadrianpermission. 14234219Sadrian 15234219SadrianLUCENT DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, 16291049SimpINCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. 17291049SimpIN NO EVENT SHALL LUCENT OR ANY OF ITS ENTITIES BE LIABLE FOR ANY 18234219SadrianSPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 19234219SadrianWHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER 20234219SadrianIN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, 21234219SadrianARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF 22263301SimpTHIS SOFTWARE. 23234219Sadrian****************************************************************/ 24263301Simp 25234220Sadrian#include <sys/cdefs.h> 26234235Sadrian__FBSDID("$FreeBSD$"); 27234235Sadrian 28234219Sadrianconst char *version = "version 20121220 (FreeBSD)"; 29234235Sadrian 30345344Skib#define DEBUG 31295945Ssobomax#include <stdio.h> 32295945Ssobomax#include <ctype.h> 33234235Sadrian#include <locale.h> 34234235Sadrian#include <stdlib.h> 35234235Sadrian#include <string.h> 36234235Sadrian#include <signal.h> 37234235Sadrian#include "awk.h" 38234232Sadrian#include "ytab.h" 39234232Sadrian 40234232Sadrianextern char **environ; 41234863Sadrianextern int nfields; 42234863Sadrian 43234863Sadrianint dbg = 0; 44234863SadrianAwkfloat srand_seed = 1; 45235290Sadrianchar *cmdname; /* gets argv[0] for error messages */ 46235290Sadrianextern FILE *yyin; /* lex input file */ 47235290Sadrianchar *lexprog; /* points to program argument if it exists */ 48extern int errorflag; /* non-zero if any syntax errors; set by yyerror */ 49int compile_time = 2; /* for error printing: */ 50 /* 2 = cmdline, 1 = compile, 0 = running */ 51 52#define MAX_PFILE 20 /* max number of -f's */ 53 54char *pfile[MAX_PFILE]; /* program filenames from -f's */ 55int npfile = 0; /* number of filenames */ 56int curpfile = 0; /* current filename */ 57 58int safe = 0; /* 1 => "safe" mode */ 59 60int main(int argc, char *argv[]) 61{ 62 const char *fs = NULL; 63 64 setlocale(LC_CTYPE, ""); 65 setlocale(LC_COLLATE, ""); 66 setlocale(LC_NUMERIC, "C"); /* for parsing cmdline & prog */ 67 cmdname = argv[0]; 68 if (argc == 1) { 69 fprintf(stderr, 70 "usage: %s [-F fs] [-v var=value] [-f progfile | 'prog'] [file ...]\n", 71 cmdname); 72 exit(1); 73 } 74 signal(SIGFPE, fpecatch); 75 76 srand_seed = 1; 77 srandom((unsigned long) srand_seed); 78 79 yyin = NULL; 80 symtab = makesymtab(NSYMTAB/NSYMTAB); 81 while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') { 82 if (strcmp(argv[1],"-version") == 0 || strcmp(argv[1],"--version") == 0) { 83 printf("awk %s\n", version); 84 exit(0); 85 break; 86 } 87 if (strncmp(argv[1], "--", 2) == 0) { /* explicit end of args */ 88 argc--; 89 argv++; 90 break; 91 } 92 switch (argv[1][1]) { 93 case 's': 94 if (strcmp(argv[1], "-safe") == 0) 95 safe = 1; 96 break; 97 case 'f': /* next argument is program filename */ 98 if (argv[1][2] != 0) { /* arg is -fsomething */ 99 if (npfile >= MAX_PFILE - 1) 100 FATAL("too many -f options"); 101 pfile[npfile++] = &argv[1][2]; 102 } else { /* arg is -f something */ 103 argc--; argv++; 104 if (argc <= 1) 105 FATAL("no program filename"); 106 if (npfile >= MAX_PFILE - 1) 107 FATAL("too many -f options"); 108 pfile[npfile++] = argv[1]; 109 } 110 break; 111 case 'F': /* set field separator */ 112 if (argv[1][2] != 0) { /* arg is -Fsomething */ 113 if (argv[1][2] == 't' && argv[1][3] == 0) /* wart: t=>\t */ 114 fs = "\t"; 115 else if (argv[1][2] != 0) 116 fs = &argv[1][2]; 117 } else { /* arg is -F something */ 118 argc--; argv++; 119 if (argc > 1 && argv[1][0] == 't' && argv[1][1] == 0) /* wart: t=>\t */ 120 fs = "\t"; 121 else if (argc > 1 && argv[1][0] != 0) 122 fs = &argv[1][0]; 123 } 124 if (fs == NULL || *fs == '\0') 125 WARNING("field separator FS is empty"); 126 break; 127 case 'v': /* -v a=1 to be done NOW. one -v for each */ 128 if (argv[1][2] != 0) { /* arg is -vsomething */ 129 if (isclvar(&argv[1][2])) 130 setclvar(&argv[1][2]); 131 else 132 FATAL("invalid -v option argument: %s", &argv[1][2]); 133 } else { /* arg is -v something */ 134 argc--; argv++; 135 if (argc <= 1) 136 FATAL("no variable name"); 137 if (isclvar(argv[1])) 138 setclvar(argv[1]); 139 else 140 FATAL("invalid -v option argument: %s", argv[1]); 141 } 142 break; 143 case 'd': 144 dbg = atoi(&argv[1][2]); 145 if (dbg == 0) 146 dbg = 1; 147 printf("awk %s\n", version); 148 break; 149 default: 150 WARNING("unknown option %s ignored", argv[1]); 151 break; 152 } 153 argc--; 154 argv++; 155 } 156 /* argv[1] is now the first argument */ 157 if (npfile == 0) { /* no -f; first argument is program */ 158 if (argc <= 1) { 159 if (dbg) 160 exit(0); 161 FATAL("no program given"); 162 } 163 dprintf( ("program = |%s|\n", argv[1]) ); 164 lexprog = argv[1]; 165 argc--; 166 argv++; 167 } 168 recinit(recsize); 169 syminit(); 170 compile_time = 1; 171 argv[0] = cmdname; /* put prog name at front of arglist */ 172 dprintf( ("argc=%d, argv[0]=%s\n", argc, argv[0]) ); 173 arginit(argc, argv); 174 if (!safe) 175 envinit(environ); 176 yyparse(); 177 setlocale(LC_NUMERIC, ""); /* back to whatever it is locally */ 178 if (fs) 179 *FS = qstring(fs, '\0'); 180 dprintf( ("errorflag=%d\n", errorflag) ); 181 if (errorflag == 0) { 182 compile_time = 0; 183 run(winner); 184 } else 185 bracecheck(); 186 return(errorflag); 187} 188 189int pgetc(void) /* get 1 character from awk program */ 190{ 191 int c; 192 193 for (;;) { 194 if (yyin == NULL) { 195 if (curpfile >= npfile) 196 return EOF; 197 if (strcmp(pfile[curpfile], "-") == 0) 198 yyin = stdin; 199 else if ((yyin = fopen(pfile[curpfile], "r")) == NULL) 200 FATAL("can't open file %s", pfile[curpfile]); 201 lineno = 1; 202 } 203 if ((c = getc(yyin)) != EOF) 204 return c; 205 if (yyin != stdin) 206 fclose(yyin); 207 yyin = NULL; 208 curpfile++; 209 } 210} 211 212char *cursource(void) /* current source file name */ 213{ 214 if (npfile > 0) 215 return pfile[curpfile]; 216 else 217 return NULL; 218} 219