lang/nawk/files/main.c

   1 /* $NetBSD$ */
   2
   3 /****************************************************************
   4 Copyright (C) Lucent Technologies 1997
   5 All Rights Reserved
   6
   7 Permission to use, copy, modify, and distribute this software and
   8 its documentation for any purpose and without fee is hereby
   9 granted, provided that the above copyright notice appear in all
  10 copies and that both that the copyright notice and this
  11 permission notice and warranty disclaimer appear in supporting
  12 documentation, and that the name Lucent Technologies or any of
  13 its entities not be used in advertising or publicity pertaining
  14 to distribution of the software without specific, written prior
  15 permission.
  16
  17 LUCENT DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
  18 INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS.
  19 IN NO EVENT SHALL LUCENT OR ANY OF ITS ENTITIES BE LIABLE FOR ANY
  20 SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  21 WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER
  22 IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,
  23 ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF
  24 THIS SOFTWARE.
  25 ****************************************************************/
  26
  27 const char      *version = "version 20050424";
  28
  29 #define DEBUG
  30 #include <stdio.h>
  31 #include <ctype.h>
  32 #include <locale.h>
  33 #include <stdlib.h>
  34 #include <string.h>
  35 #include <signal.h>
  36 #include "awk.h"
  37 #include "ytab.h"
  38
  39 extern  char    **environ;
  40 extern  int     nfields;
  41
  42 int     dbg     = 0;
  43 char    *cmdname;       /* gets argv[0] for error messages */
  44 extern  FILE    *yyin;  /* lex input file */
  45 char    *lexprog;       /* points to program argument if it exists */
  46 extern  int errorflag;  /* non-zero if any syntax errors; set by yyerror */
  47 int     compile_time = 2;       /* for error printing: */
  48                                 /* 2 = cmdline, 1 = compile, 0 = running */
  49
  50 #define MAX_PFILE       20      /* max number of -f's */
  51
  52 char    *pfile[MAX_PFILE];      /* program filenames from -f's */
  53 int     npfile = 0;     /* number of filenames */
  54 int     curpfile = 0;   /* current filename */
  55
  56 int     safe    = 0;    /* 1 => "safe" mode */
  57
  58 int main(int argc, char *argv[])
  59 {
  60         const char *fs = NULL;
  61
  62         setlocale(LC_CTYPE, "");
  63         setlocale(LC_NUMERIC, "C"); /* for parsing cmdline & prog */
  64         cmdname = argv[0];
  65         if (argc == 1) {
  66                 fprintf(stderr, "Usage: %s [-f programfile | 'program'] [-Ffieldsep] [-v var=value] [files]\n", cmdname);
  67                 exit(1);
  68         }
  69         signal(SIGFPE, fpecatch);
  70         yyin = NULL;
  71         symtab = makesymtab(NSYMTAB);
  72         while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') {
  73                 if (strcmp(argv[1], "--") == 0) {       /* explicit end of args */
  74                         argc--;
  75                         argv++;
  76                         break;
  77                 }
  78                 switch (argv[1][1]) {
  79                 case 's':
  80                         if (strcmp(argv[1], "-safe") == 0)
  81                                 safe = 1;
  82                         break;
  83                 case 'f':       /* next argument is program filename */
  84                         argc--;
  85                         argv++;
  86                         if (argc <= 1)
  87                                 FATAL("no program filename");
  88                         if (npfile >= MAX_PFILE - 1)
  89                                 FATAL("too many -f options");
  90                         pfile[npfile++] = argv[1];
  91                         break;
  92                 case 'F':       /* set field separator */
  93                         if (argv[1][2] != 0) {  /* arg is -Fsomething */
  94                                 if (argv[1][2] == 't' && argv[1][3] == 0)       /* wart: t=>\t */
  95                                         fs = "\t";
  96                                 else if (argv[1][2] != 0)
  97                                         fs = &argv[1][2];
  98                         } else {                /* arg is -F something */
  99                                 argc--; argv++;
 100                                 if (argc > 1 && argv[1][0] == 't' && argv[1][1] == 0)   /* wart: t=>\t */
 101                                         fs = "\t";
 102                                 else if (argc > 1 && argv[1][0] != 0)
 103                                         fs = &argv[1][0];
 104                         }
 105                         if (fs == NULL || *fs == '\0')
 106                                 WARNING("field separator FS is empty");
 107                         break;
 108                 case 'v':       /* -v a=1 to be done NOW.  one -v for each */
 109                         if (argv[1][2] == '\0' && --argc > 1 && isclvar((++argv)[1]))
 110                                 setclvar(argv[1]);
 111                         break;
 112                 case 'm':       /* more memory: -mr=record, -mf=fields */
 113                                 /* no longer supported */
 114                         WARNING("obsolete option %s ignored", argv[1]);
 115                         break;
 116                 case 'd':
 117                         dbg = atoi(&argv[1][2]);
 118                         if (dbg == 0)
 119                                 dbg = 1;
 120                         printf("awk %s\n", version);
 121                         break;
 122                 case 'V':       /* added for exptools "standard" */
 123                         printf("awk %s\n", version);
 124                         exit(0);
 125                         break;
 126                 default:
 127                         WARNING("unknown option %s ignored", argv[1]);
 128                         break;
 129                 }
 130                 argc--;
 131                 argv++;
 132         }
 133         /* argv[1] is now the first argument */
 134         if (npfile == 0) {      /* no -f; first argument is program */
 135                 if (argc <= 1) {
 136                         if (dbg)
 137                                 exit(0);
 138                         FATAL("no program given");
 139                 }
 140                    dprintf( ("program = |%s|\n", argv[1]) );
 141                 lexprog = argv[1];
 142                 argc--;
 143                 argv++;
 144         }
 145         recinit(recsize);
 146         syminit();
 147         compile_time = 1;
 148         argv[0] = cmdname;      /* put prog name at front of arglist */
 149            dprintf( ("argc=%d, argv[0]=%s\n", argc, argv[0]) );
 150         arginit(argc, argv);
 151         if (!safe)
 152                 envinit(environ);
 153         yyparse();
 154         setlocale(LC_NUMERIC, ""); /* back to whatever it is locally */
 155         if (fs)
 156                 *FS = qstring(fs, '\0');
 157            dprintf( ("errorflag=%d\n", errorflag) );
 158         if (errorflag == 0) {
 159                 compile_time = 0;
 160                 run(winner);
 161         } else
 162                 bracecheck();
 163         return(errorflag);
 164 }
 165
 166 int pgetc(void)         /* get 1 character from awk program */
 167 {
 168         int c;
 169
 170         for (;;) {
 171                 if (yyin == NULL) {
 172                         if (curpfile >= npfile)
 173                                 return EOF;
 174                         if (strcmp(pfile[curpfile], "-") == 0)
 175                                 yyin = stdin;
 176                         else if ((yyin = fopen(pfile[curpfile], "r")) == NULL)
 177                                 FATAL("can't open file %s", pfile[curpfile]);
 178                         lineno = 1;
 179                 }
 180                 if ((c = getc(yyin)) != EOF)
 181                         return c;
 182                 if (yyin != stdin)
 183                         fclose(yyin);
 184                 yyin = NULL;
 185                 curpfile++;
 186         }
 187 }
 188
 189 char *cursource(void)   /* current source file name */
 190 {
 191         if (npfile > 0)
 192                 return pfile[curpfile];
 193         else
 194                 return NULL;
 195 }