external/historical/nawk/dist/main.c

   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 20121220";
  26
  27 #if HAVE_NBTOOL_CONFIG_H
  28 #include "nbtool_config.h"
  29 #endif
  30
  31 #define DEBUG
  32 #include <stdio.h>
  33 #include <ctype.h>
  34 #include <locale.h>
  35 #include <stdlib.h>
  36 #include <string.h>
  37 #include <signal.h>
  38 #include "awk.h"
  39 #include "awkgram.h"
  40
  41 extern  char    **environ;
  42 extern  int     nfields;
  43
  44 int     dbg     = 0;
  45 unsigned int srand_seed = 1;
  46 char    *cmdname;       /* gets argv[0] for error messages */
  47 extern  FILE    *yyin;  /* lex input file */
  48 char    *lexprog;       /* points to program argument if it exists */
  49 extern  int errorflag;  /* non-zero if any syntax errors; set by yyerror */
  50 int     compile_time = 2;       /* for error printing: */
  51                                 /* 2 = cmdline, 1 = compile, 0 = running */
  52
  53 static char     **pfile = NULL; /* program filenames from -f's */
  54 static size_t   maxpfile = 0;   /* max program filenames */
  55 static size_t   npfile = 0;     /* number of filenames */
  56 static size_t   curpfile = 0;   /* current filename */
  57
  58 int     safe    = 0;    /* 1 => "safe" mode */
  59
  60 static char *
  61 setfs(char *p)
  62 {
  63 #ifdef notdef
  64         /* wart: t=>\t */
  65         if (p[0] == 't' && p[1] == 0)
  66                 return "\t";
  67         else
  68 #endif
  69         if (p[0] != 0)
  70                 return p;
  71         return NULL;
  72 }
  73
  74 __dead static void fpecatch(int n
  75 #ifdef SA_SIGINFO
  76         , siginfo_t *si, void *uc
  77 #endif
  78 )
  79 {
  80 #ifdef SA_SIGINFO
  81         static const char *emsg[] = {
  82             "Unknown error",
  83             "Integer divide by zero",
  84             "Integer overflow",
  85             "Floating point divide by zero",
  86             "Floating point overflow",
  87             "Floating point underflow",
  88             "Floating point inexact result",
  89             "Invalid Floating point operation",
  90             "Subscript out of range",
  91         };
  92 #endif
  93         FATAL("floating point exception"
  94 #ifdef SA_SIGINFO
  95             ": %s\n", emsg[si->si_code >= 1 && si->si_code <= 8 ?
  96             si->si_code : 0]
  97 #endif
  98             );
  99 }
 100
 101 int main(int argc, char *argv[])
 102 {
 103         const char *fs = NULL;
 104         char *fn;
 105
 106         setlocale(LC_ALL, "");
 107         setlocale(LC_NUMERIC, "C"); /* for parsing cmdline & prog */
 108         cmdname = argv[0];
 109         if (argc == 1) {
 110                 fprintf(stderr,
 111                   "usage: %s [-F fs] [-v var=value] [-f progfile | 'prog'] [file ...]\n",
 112                   cmdname);
 113                 exit(1);
 114         }
 115
 116 #ifdef SA_SIGINFO
 117         {
 118                 struct sigaction sa;
 119                 sa.sa_sigaction = fpecatch;
 120                 sa.sa_flags = SA_SIGINFO;
 121                 sigemptyset(&sa.sa_mask);
 122                 (void)sigaction(SIGFPE, &sa, NULL);
 123         }
 124 #else
 125         (void)signal(SIGFPE, fpecatch);
 126 #endif
 127         /* Set and keep track of the random seed */
 128         srand_seed = 1;
 129         srand(srand_seed);
 130
 131         yyin = NULL;
 132         symtab = makesymtab(NSYMTAB/NSYMTAB);
 133         while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') {
 134                 if (strcmp(argv[1],"-version") == 0 || strcmp(argv[1],"--version") == 0) {
 135                         printf("awk %s\n", version);
 136                         exit(0);
 137                         break;
 138                 }
 139                 if (strncmp(argv[1], "--", 2) == 0) {   /* explicit end of args */
 140                         argc--;
 141                         argv++;
 142                         break;
 143                 }
 144                 switch (argv[1][1]) {
 145                 case 's':
 146                         if (strcmp(argv[1], "-safe") == 0)
 147                                 safe = 1;
 148                         break;
 149                 case 'f':       /* next argument is program filename */
 150                         if (argv[1][2] != 0) /* arg is -fsomething */
 151                                 fn = &argv[1][2];
 152                         else {
 153                                 argc--;
 154                                 argv++;
 155                                 if (argc <= 1)
 156                                         FATAL("no program filename");
 157                                 fn = argv[1];
 158                         }
 159                         if (npfile >= maxpfile) {
 160                                 maxpfile += 20;
 161                                 pfile = realloc(pfile,
 162                                     maxpfile * sizeof(*pfile));
 163                                 if (pfile == NULL)
 164                                         FATAL("error allocating space for "
 165                                             "-f options");
 166                         }
 167                         pfile[npfile++] = fn;
 168                         break;
 169                 case 'F':       /* set field separator */
 170                         if (argv[1][2] != 0) {  /* arg is -Fsomething */
 171                                 fs = setfs(argv[1] + 2);
 172                         } else {                /* arg is -F something */
 173                                 argc--; argv++;
 174                                 if (argc > 1)
 175                                         fs = setfs(argv[1]);
 176                         }
 177                         if (fs == NULL || *fs == '\0')
 178                                 WARNING("field separator FS is empty");
 179                         break;
 180                 case 'v':       /* -v a=1 to be done NOW.  one -v for each */
 181                         if (argv[1][2] != 0) {  /* arg is -vsomething */
 182                                 if (isclvar(&argv[1][2]))
 183                                         setclvar(&argv[1][2]);
 184                                 else
 185                                         FATAL("invalid -v option argument: %s", &argv[1][2]);
 186                         } else {                /* arg is -v something */
 187                                 argc--; argv++;
 188                                 if (argc <= 1)
 189                                         FATAL("no variable name");
 190                                 if (isclvar(argv[1]))
 191                                         setclvar(argv[1]);
 192                                 else
 193                                         FATAL("invalid -v option argument: %s", argv[1]);
 194                         }
 195                         break;
 196                 case 'd':
 197                         dbg = atoi(&argv[1][2]);
 198                         if (dbg == 0)
 199                                 dbg = 1;
 200                         printf("awk %s\n", version);
 201                         break;
 202                 default:
 203                         WARNING("unknown option %s ignored", argv[1]);
 204                         break;
 205                 }
 206                 argc--;
 207                 argv++;
 208         }
 209         /* argv[1] is now the first argument */
 210         if (npfile == 0) {      /* no -f; first argument is program */
 211                 if (argc <= 1) {
 212                         if (dbg)
 213                                 exit(0);
 214                         FATAL("no program given");
 215                 }
 216                    dprintf( ("program = |%s|\n", argv[1]) );
 217                 lexprog = argv[1];
 218                 argc--;
 219                 argv++;
 220         }
 221         recinit(recsize);
 222         syminit();
 223         compile_time = 1;
 224         argv[0] = cmdname;      /* put prog name at front of arglist */
 225            dprintf( ("argc=%d, argv[0]=%s\n", argc, argv[0]) );
 226         arginit(argc, argv);
 227         if (!safe)
 228                 envinit(environ);
 229         yyparse();
 230         if (fs)
 231                 *FS = qstring(fs, '\0');
 232            dprintf( ("errorflag=%d\n", errorflag) );
 233         if (errorflag == 0) {
 234                 compile_time = 0;
 235                 run(winner);
 236         } else
 237                 bracecheck();
 238         return(errorflag);
 239 }
 240
 241 int pgetc(void)         /* get 1 character from awk program */
 242 {
 243         int c;
 244
 245         for (;;) {
 246                 if (yyin == NULL) {
 247                         if (curpfile >= npfile)
 248                                 return EOF;
 249                         if (strcmp(pfile[curpfile], "-") == 0)
 250                                 yyin = stdin;
 251                         else if ((yyin = fopen(pfile[curpfile], "r")) == NULL)
 252                                 FATAL("can't open file %s", pfile[curpfile]);
 253                         lineno = 1;
 254                 }
 255                 if ((c = getc(yyin)) != EOF)
 256                         return c;
 257                 if (yyin != stdin)
 258                         fclose(yyin);
 259                 yyin = NULL;
 260                 curpfile++;
 261         }
 262 }
 263
 264 char *cursource(void)   /* current source file name */
 265 {
 266         if (npfile > 0)
 267                 return pfile[curpfile];
 268         else
 269                 return NULL;
 270 }