cmd/awk/main.c

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*  Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
/*    All Rights Reserved   */

#include <stdio.h>
#include <ctype.h>
#include <signal.h>
#include <locale.h>
#include <libintl.h>
#include <stdarg.h>
#include <errno.h>
#include <values.h>
#include <langinfo.h>
#include "awk.h"
#include "y.tab.h"

char    *version = "version Oct 11, 1989";

int dbg = 0;
uchar   *cmdname;   /* gets argv[0] for error messages */
uchar   *lexprog;   /* points to program argument if it exists */
int compile_time = 2;   /* for error printing: */
                /* 2 = cmdline, 1 = compile, 0 = running */
char    radixpoint = '.';

static uchar    **pfile = NULL; /* program filenames from -f's */
static int  npfile = 0; /* number of filenames */
static int  curpfile = 0;   /* current filename */

int
main(int argc, char *argv[], char *envp[])
{
    uchar *fs = NULL;
    char    *nl_radix;
    /*
     * At this point, numbers are still scanned as in
     * the POSIX locale.
     * (POSIX.2, volume 2, P867, L4742-4757)
     */
    (void) setlocale(LC_ALL, "");
    (void) setlocale(LC_NUMERIC, "C");
#if !defined(TEXT_DOMAIN)   /* Should be defined by cc -D */
#define TEXT_DOMAIN "SYS_TEST"  /* Use this only if it weren't */
#endif
    (void) textdomain(TEXT_DOMAIN);
    cmdname = (uchar *)argv[0];
    if (argc == 1) {
        (void) fprintf(stderr, gettext(
            "Usage: %s [-f programfile | 'program'] [-Ffieldsep] "
            "[-v var=value] [files]\n"), cmdname);
        exit(1);
    }
    (void) signal(SIGFPE, fpecatch);
    yyin = NULL;
    syminit();
    while (argc > 1 && argv[1][0] == '-' && argv[1][1] != '\0') {
        if (strcmp(argv[1], "--") == 0) {
            /* explicit end of args */
            argc--;
            argv++;
            break;
        }
        switch (argv[1][1]) {
        case 'f':   /* next argument is program filename */
            argc--;
            argv++;
            if (argc <= 1)
                ERROR "no program filename" FATAL;
            pfile = realloc(pfile, sizeof (uchar *) * (npfile + 1));
            if (pfile == NULL)
                ERROR "out of space in main" FATAL;
            pfile[npfile++] = (uchar *)argv[1];
            break;
        case 'F':   /* set field separator */
            if (argv[1][2] != 0) {  /* arg is -Fsomething */
                /* wart: t=>\t */
                if (argv[1][2] == 't' && argv[1][3] == 0)
                    fs = (uchar *) "\t";
                else if (argv[1][2] != 0)
                    fs = (uchar *)&argv[1][2];
            } else {        /* arg is -F something */
                argc--; argv++;
                if (argc > 1) {
                    /* wart: t=>\t */
                    if (argv[1][0] == 't' &&
                        argv[1][1] == 0)
                        fs = (uchar *) "\t";
                    else if (argv[1][0] != 0)
                        fs = (uchar *)&argv[1][0];
                }
            }
            if (fs == NULL || *fs == '\0')
                ERROR "field separator FS is empty" WARNING;
            break;
        case 'v':   /* -v a=1 to be done NOW.  one -v for each */
            if (argv[1][2] == '\0' && --argc > 1 &&
                isclvar((uchar *)(++argv)[1]))
                setclvar((uchar *)argv[1]);
            break;
        case 'd':
            dbg = atoi(&argv[1][2]);
            if (dbg == 0)
                dbg = 1;
            (void) printf("awk %s\n", version);
            break;
        default:
            ERROR "unknown option %s ignored", argv[1] WARNING;
            break;
        }
        argc--;
        argv++;
    }
    /* argv[1] is now the first argument */
    if (npfile == 0) {  /* no -f; first argument is program */
        if (argc <= 1) {
            if (dbg)
                exit(0);
            ERROR "no program given" FATAL;
        }
        dprintf(("program = |%s|\n", argv[1]));
        lexprog = (uchar *)argv[1];
        argc--;
        argv++;
    }
    compile_time = 1;
    argv[0] = (char *)cmdname;  /* put prog name at front of arglist */
    dprintf(("argc=%d, argv[0]=%s\n", argc, argv[0]));
    arginit(argc, (uchar **)argv);
    envinit((uchar **)envp);
    (void) yyparse();
    if (fs)
        *FS = qstring(fs, '\0');
    dprintf(("errorflag=%d\n", errorflag));
    /*
     * done parsing, so now activate the LC_NUMERIC
     */
    (void) setlocale(LC_ALL, "");
    nl_radix = nl_langinfo(RADIXCHAR);
    if (nl_radix)
        radixpoint = *nl_radix;

    if (errorflag == 0) {
        compile_time = 0;
        run(winner);
    } else
        bracecheck();
    return (errorflag);
}

int
pgetc(void)     /* get program character */
{
    int c;

    for (;;) {
        if (yyin == NULL) {
            if (curpfile >= npfile)
                return (EOF);
            yyin = (strcmp((char *)pfile[curpfile], "-") == 0) ?
                stdin : fopen((char *)pfile[curpfile], "r");
            if (yyin == NULL) {
                ERROR "can't open file %s",
                    pfile[curpfile] FATAL;
            }
        }
        if ((c = getc(yyin)) != EOF)
            return (c);
        (void) fclose(yyin);
        yyin = NULL;
        curpfile++;
    }
}