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