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