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 20100523";
  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;
  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
 105         setlocale(LC_ALL, "");
 106         setlocale(LC_NUMERIC, "C"); /* for parsing cmdline & prog */
 107         cmdname = argv[0];
 108         if (argc == 1) {
 109                 fprintf(stderr,
 110                   "usage: %s [-F fs] [-v var=value] [-f progfile | 'prog'] [file ...]\n",
 111                   cmdname);
 112                 exit(1);
 113         }
 114
 115 #ifdef SA_SIGINFO
 116         {
 117                 struct sigaction sa;
 118                 sa.sa_sigaction = fpecatch;
 119                 sa.sa_flags = SA_SIGINFO;
 120                 sigemptyset(&sa.sa_mask);
 121                 (void)sigaction(SIGFPE, &sa, NULL);
 122         }
 123 #else
 124         (void)signal(SIGFPE, fpecatch);
 125 #endif
 126         /* Set and keep track of the random seed */
 127         srand_seed = 1;
 128         srand(srand_seed);
 129
 130         yyin = NULL;
 131         symtab = makesymtab(NSYMTAB/NSYMTAB);
 132         while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') {
 133                 if (strcmp(argv[1],"-version") == 0 || strcmp(argv[1],"--version") == 0) {
 134                         printf("awk %s\n", version);
 135                         exit(0);
 136                         break;
 137                 }
 138                 if (strncmp(argv[1], "--", 2) == 0) {   /* explicit end of args */
 139                         argc--;
 140                         argv++;
 141                         break;
 142                 }
 143                 switch (argv[1][1]) {
 144                 case 's':
 145                         if (strcmp(argv[1], "-safe") == 0)
 146                                 safe = 1;
 147                         break;
 148                 case 'f':       /* next argument is program filename */
 149                         argc--;
 150                         argv++;
 151                         if (argc <= 1)
 152                                 FATAL("no program filename");
 153                         if (npfile >= maxpfile) {
 154                                 maxpfile += 20;
 155                                 pfile = realloc(pfile,
 156                                     maxpfile * sizeof(*pfile));
 157                                 if (pfile == NULL)
 158                                         FATAL("error allocating space for "
 159                                             "-f options");
 160                         }
 161                         pfile[npfile++] = argv[1];
 162                         break;
 163                 case 'F':       /* set field separator */
 164                         if (argv[1][2] != 0) {  /* arg is -Fsomething */
 165                                 fs = setfs(argv[1] + 2);
 166                         } else {                /* arg is -F something */
 167                                 argc--; argv++;
 168                                 if (argc > 1)
 169                                         fs = setfs(argv[1]);
 170                         }
 171                         if (fs == NULL || *fs == '\0')
 172                                 WARNING("field separator FS is empty");
 173                         break;
 174                 case 'v':       /* -v a=1 to be done NOW.  one -v for each */
 175                         if (argv[1][2] == '\0' && --argc > 1 && isclvar((++argv)[1]))
 176                                 setclvar(argv[1]);
 177                         else if (argv[1][2] != '\0')
 178                                 setclvar(&argv[1][2]);
 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 = 1;
 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 (fs)
 215                 *FS = qstring(fs, '\0');
 216            dprintf( ("errorflag=%d\n", errorflag) );
 217         if (errorflag == 0) {
 218                 compile_time = 0;
 219                 run(winner);
 220         } else
 221                 bracecheck();
 222         return(errorflag);
 223 }
 224
 225 int pgetc(void)         /* get 1 character from awk program */
 226 {
 227         int c;
 228
 229         for (;;) {
 230                 if (yyin == NULL) {
 231                         if (curpfile >= npfile)
 232                                 return EOF;
 233                         if (strcmp(pfile[curpfile], "-") == 0)
 234                                 yyin = stdin;
 235                         else if ((yyin = fopen(pfile[curpfile], "r")) == NULL)
 236                                 FATAL("can't open file %s", pfile[curpfile]);
 237                         lineno = 1;
 238                 }
 239                 if ((c = getc(yyin)) != EOF)
 240                         return c;
 241                 if (yyin != stdin)
 242                         fclose(yyin);
 243                 yyin = NULL;
 244                 curpfile++;
 245         }
 246 }
 247
 248 char *cursource(void)   /* current source file name */
 249 {
 250         if (npfile > 0)
 251                 return pfile[curpfile];
 252         else
 253                 return NULL;
 254 }