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