cmd/awk/tran.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 2006 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

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

#pragma ident   "%Z%%M% %I% %E% SMI"

#define DEBUG
#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <string.h>
#include "awk.h"
#include "y.tab.h"

#define FULLTAB 2   /* rehash when table gets this x full */
#define GROWTAB 4   /* grow table by this factor */

Array   *symtab;    /* main symbol table */

uchar   **FS;       /* initial field sep */
uchar   **RS;       /* initial record sep */
uchar   **OFS;      /* output field sep */
uchar   **ORS;      /* output record sep */
uchar   **OFMT;     /* output format for numbers */
Awkfloat *NF;       /* number of fields in current record */
Awkfloat *NR;       /* number of current record */
Awkfloat *FNR;      /* number of current record in current file */
uchar   **FILENAME; /* current filename argument */
Awkfloat *ARGC;     /* number of arguments from command line */
uchar   **SUBSEP;   /* subscript separator for a[i,j,k]; default \034 */
Awkfloat *RSTART;   /* start of re matched with ~; origin 1 (!) */
Awkfloat *RLENGTH;  /* length of same */

Cell    *recloc;    /* location of record */
Cell    *nrloc;     /* NR */
Cell    *nfloc;     /* NF */
Cell    *fnrloc;    /* FNR */
Array   *ARGVtab;   /* symbol table containing ARGV[...] */
Array   *ENVtab;    /* symbol table containing ENVIRON[...] */
Cell    *rstartloc; /* RSTART */
Cell    *rlengthloc;    /* RLENGTH */
Cell    *symtabloc; /* SYMTAB */

Cell    *nullloc;
Node    *nullnode;  /* zero&null, converted into a node for comparisons */

static  void    rehash(Array *);

void
syminit(void)
{
    init_buf(&record, &record_size, LINE_INCR);

    /* initialize $0 */
    recloc = getfld(0);
    recloc->nval = (uchar *)"$0";
    recloc->sval = record;
    recloc->tval = REC|STR|DONTFREE;

    symtab = makesymtab(NSYMTAB);
    (void) setsymtab((uchar *)"0", (uchar *)"0", 0.0,
        NUM|STR|CON|DONTFREE, symtab);
    /* this is used for if(x)... tests: */
    nullloc = setsymtab((uchar *)"$zero&null", (uchar *)"", 0.0,
        NUM|STR|CON|DONTFREE, symtab);
    nullnode = valtonode(nullloc, CCON);
    FS = &setsymtab((uchar *)"FS", (uchar *)" ", 0.0,
        STR|DONTFREE, symtab)->sval;
    RS = &setsymtab((uchar *)"RS", (uchar *)"\n", 0.0,
        STR|DONTFREE, symtab)->sval;
    OFS = &setsymtab((uchar *)"OFS", (uchar *)" ", 0.0,
        STR|DONTFREE, symtab)->sval;
    ORS = &setsymtab((uchar *)"ORS", (uchar *)"\n", 0.0,
        STR|DONTFREE, symtab)->sval;
    OFMT = &setsymtab((uchar *)"OFMT", (uchar *)"%.6g", 0.0,
        STR|DONTFREE, symtab)->sval;
    FILENAME = &setsymtab((uchar *)"FILENAME", (uchar *)"-", 0.0,
        STR|DONTFREE, symtab)->sval;
    nfloc = setsymtab((uchar *)"NF", (uchar *)"", 0.0, NUM, symtab);
    NF = &nfloc->fval;
    nrloc = setsymtab((uchar *)"NR", (uchar *)"", 0.0, NUM, symtab);
    NR = &nrloc->fval;
    fnrloc = setsymtab((uchar *)"FNR", (uchar *)"", 0.0, NUM, symtab);
    FNR = &fnrloc->fval;
    SUBSEP = &setsymtab((uchar *)"SUBSEP", (uchar *)"\034", 0.0,
        STR|DONTFREE, symtab)->sval;
    rstartloc = setsymtab((uchar *)"RSTART", (uchar *)"", 0.0,
        NUM, symtab);
    RSTART = &rstartloc->fval;
    rlengthloc = setsymtab((uchar *)"RLENGTH", (uchar *)"", 0.0,
        NUM, symtab);
    RLENGTH = &rlengthloc->fval;
    symtabloc = setsymtab((uchar *)"SYMTAB", (uchar *)"", 0.0, ARR, symtab);
    symtabloc->sval = (uchar *)symtab;
}

void
arginit(int ac, uchar *av[])
{
    Cell *cp;
    int i;
    uchar temp[11];

    /* first make FILENAME first real argument */
    for (i = 1; i < ac; i++) {
        if (!isclvar(av[i])) {
            (void) setsval(lookup((uchar *)"FILENAME", symtab),
                av[i]);
            break;
        }
    }
    ARGC = &setsymtab((uchar *)"ARGC", (uchar *)"", (Awkfloat)ac,
        NUM, symtab)->fval;
    cp = setsymtab((uchar *)"ARGV", (uchar *)"", 0.0, ARR, symtab);
    ARGVtab = makesymtab(NSYMTAB);  /* could be (int) ARGC as well */
    cp->sval = (uchar *) ARGVtab;
    for (i = 0; i < ac; i++) {
        (void) sprintf((char *)temp, "%d", i);
        if (is_number(*av)) {
            (void) setsymtab(temp, *av, atof((const char *)*av),
                STR|NUM, ARGVtab);
        } else {
            (void) setsymtab(temp, *av, 0.0, STR, ARGVtab);
        }
        av++;
    }
}

void
envinit(uchar *envp[])
{
    Cell *cp;
    uchar *p;

    cp = setsymtab((uchar *)"ENVIRON", (uchar *)"", 0.0, ARR, symtab);
    ENVtab = makesymtab(NSYMTAB);
    cp->sval = (uchar *) ENVtab;
    for (; *envp; envp++) {
        if ((p = (uchar *)strchr((char *)*envp, '=')) == NULL)
            continue;
        *p++ = 0;   /* split into two strings at = */
        if (is_number(p)) {
            (void) setsymtab(*envp, p, atof((const char *)p),
                STR|NUM, ENVtab);
        } else {
            (void) setsymtab(*envp, p, 0.0, STR, ENVtab);
        }
        /* restore in case env is passed down to a shell */
        p[-1] = '=';
    }
}

Array *
makesymtab(int n)
{
    Array *ap;
    Cell **tp;

    ap = (Array *)malloc(sizeof (Array));
    tp = (Cell **)calloc(n, sizeof (Cell *));
    if (ap == NULL || tp == NULL)
        ERROR "out of space in makesymtab" FATAL;
    ap->nelem = 0;
    ap->size = n;
    ap->tab = tp;
    return (ap);
}

void
freesymtab(Cell *ap)    /* free symbol table */
{
    Cell *cp, *next;
    Array *tp;
    int i;

    if (!isarr(ap))
        return;
    /*LINTED align*/
    tp = (Array *)ap->sval;
    if (tp == NULL)
        return;
    for (i = 0; i < tp->size; i++) {
        for (cp = tp->tab[i]; cp != NULL; cp = next) {
            next = cp->cnext;
            xfree(cp->nval);
            if (freeable(cp))
                xfree(cp->sval);
            free(cp);
        }
    }
    free(tp->tab);
    free(tp);
}

void
freeelem(Cell *ap, uchar *s)        /* free elem s from ap (i.e., ap["s"] */
{
    Array *tp;
    Cell *p, *prev = NULL;
    int h;

    /*LINTED align*/
    tp = (Array *)ap->sval;
    h = hash(s, tp->size);
    for (p = tp->tab[h]; p != NULL; prev = p, p = p->cnext)
        if (strcmp((char *)s, (char *)p->nval) == 0) {
            if (prev == NULL)   /* 1st one */
                tp->tab[h] = p->cnext;
            else            /* middle somewhere */
                prev->cnext = p->cnext;
            if (freeable(p))
                xfree(p->sval);
            free(p->nval);
            free(p);
            tp->nelem--;
            return;
        }
}

Cell *
setsymtab(uchar *n, uchar *s, Awkfloat f, unsigned int t, Array *tp)
{
    register int h;
    register Cell *p;

    if (n != NULL && (p = lookup(n, tp)) != NULL) {
        dprintf(("setsymtab found %p: n=%s", (void *)p, p->nval));
        dprintf((" s=\"%s\" f=%g t=%p\n",
            p->sval, p->fval, (void *)p->tval));
        return (p);
    }
    p = (Cell *)malloc(sizeof (Cell));
    if (p == NULL)
        ERROR "symbol table overflow at %s", n FATAL;
    p->nval = tostring(n);
    p->sval = s ? tostring(s) : tostring((uchar *)"");
    p->fval = f;
    p->tval = t;
    p->csub = 0;

    tp->nelem++;
    if (tp->nelem > FULLTAB * tp->size)
        rehash(tp);
    h = hash(n, tp->size);
    p->cnext = tp->tab[h];
    tp->tab[h] = p;
    dprintf(("setsymtab set %p: n=%s", (void *)p, p->nval));
    dprintf((" s=\"%s\" f=%g t=%p\n", p->sval, p->fval, (void *)p->tval));
    return (p);
}

int
hash(uchar *s, int n)   /* form hash value for string s */
{
    register unsigned hashval;

    for (hashval = 0; *s != '\0'; s++)
        hashval = (*s + 31 * hashval);
    return (hashval % n);
}

static void
rehash(Array *tp)   /* rehash items in small table into big one */
{
    int i, nh, nsz;
    Cell *cp, *op, **np;

    nsz = GROWTAB * tp->size;
    np = (Cell **)calloc(nsz, sizeof (Cell *));
    if (np == NULL)
        ERROR "out of space in rehash" FATAL;
    for (i = 0; i < tp->size; i++) {
        for (cp = tp->tab[i]; cp; cp = op) {
            op = cp->cnext;
            nh = hash(cp->nval, nsz);
            cp->cnext = np[nh];
            np[nh] = cp;
        }
    }
    free(tp->tab);
    tp->tab = np;
    tp->size = nsz;
}

Cell *
lookup(uchar *s, Array *tp) /* look for s in tp */
{
    register Cell *p;
    int h;

    h = hash(s, tp->size);
    for (p = tp->tab[h]; p != NULL; p = p->cnext) {
        if (strcmp((char *)s, (char *)p->nval) == 0)
            return (p); /* found it */
    }
    return (NULL);          /* not found */
}

Awkfloat
setfval(Cell *vp, Awkfloat f)
{
    int i;

    if ((vp->tval & (NUM | STR)) == 0)
        funnyvar(vp, "assign to");
    if (vp->tval & FLD) {
        donerec = 0;    /* mark $0 invalid */
        i = fldidx(vp);
        if (i > *NF)
            newfld(i);
        dprintf(("setting field %d to %g\n", i, f));
    } else if (vp->tval & REC) {
        donefld = 0;    /* mark $1... invalid */
        donerec = 1;
    }
    vp->tval &= ~STR;   /* mark string invalid */
    vp->tval |= NUM;    /* mark number ok */
    dprintf(("setfval %p: %s = %g, t=%p\n", (void *)vp,
        vp->nval ? vp->nval : (unsigned char *)"NULL",
        f, (void *)vp->tval));
    return (vp->fval = f);
}

void
funnyvar(Cell *vp, char *rw)
{
    if (vp->tval & ARR)
        ERROR "can't %s %s; it's an array name.", rw, vp->nval FATAL;
    if (vp->tval & FCN)
        ERROR "can't %s %s; it's a function.", rw, vp->nval FATAL;
    ERROR "funny variable %o: n=%s s=\"%s\" f=%g t=%o",
        vp, vp->nval, vp->sval, vp->fval, vp->tval CONT;
}

uchar *
setsval(Cell *vp, uchar *s)
{
    int i;

    if ((vp->tval & (NUM | STR)) == 0)
        funnyvar(vp, "assign to");
    if (vp->tval & FLD) {
        donerec = 0;    /* mark $0 invalid */
        i = fldidx(vp);
        if (i > *NF)
            newfld(i);
        dprintf(("setting field %d to %s\n", i, s));
    } else if (vp->tval & REC) {
        donefld = 0;    /* mark $1... invalid */
        donerec = 1;
    }
    vp->tval &= ~NUM;
    vp->tval |= STR;
    if (freeable(vp))
        xfree(vp->sval);
    vp->tval &= ~DONTFREE;
    dprintf(("setsval %p: %s = \"%s\", t=%p\n",
        (void *)vp,
        vp->nval ? (char *)vp->nval : "",
        s,
        (void *)(vp->tval ? (char *)vp->tval : "")));
    return (vp->sval = tostring(s));
}

Awkfloat
r_getfval(Cell *vp)
{
    if ((vp->tval & (NUM | STR)) == 0)
        funnyvar(vp, "read value of");
    if ((vp->tval & FLD) && donefld == 0)
        fldbld();
    else if ((vp->tval & REC) && donerec == 0)
        recbld();
    if (!isnum(vp)) {   /* not a number */
        vp->fval = atof((const char *)vp->sval);    /* best guess */
        if (is_number(vp->sval) && !(vp->tval&CON))
            vp->tval |= NUM;    /* make NUM only sparingly */
    }
    dprintf(("getfval %p: %s = %g, t=%p\n",
        (void *)vp, vp->nval, vp->fval, (void *)vp->tval));
    return (vp->fval);
}

uchar *
r_getsval(Cell *vp)
{
    uchar s[256];

    if ((vp->tval & (NUM | STR)) == 0)
        funnyvar(vp, "read value of");
    if ((vp->tval & FLD) && donefld == 0)
        fldbld();
    else if ((vp->tval & REC) && donerec == 0)
        recbld();
    if ((vp->tval & STR) == 0) {
        if (!(vp->tval&DONTFREE))
            xfree(vp->sval);
        if ((long long)vp->fval == vp->fval) {
            (void) snprintf((char *)s, sizeof (s),
                "%.20g", vp->fval);
        } else {
            /*LINTED*/
            (void) snprintf((char *)s, sizeof (s),
                (char *)*OFMT, vp->fval);
        }
        vp->sval = tostring(s);
        vp->tval &= ~DONTFREE;
        vp->tval |= STR;
    }
    dprintf(("getsval %p: %s = \"%s\", t=%p\n",
        (void *)vp,
        vp->nval ? (char *)vp->nval : "",
        vp->sval ? (char *)vp->sval : "",
        (void *)vp->tval));
    return (vp->sval);
}

uchar *
tostring(uchar *s)
{
    register uchar *p;

    p = (uchar *)malloc(strlen((char *)s)+1);
    if (p == NULL)
        ERROR "out of space in tostring on %s", s FATAL;
    (void) strcpy((char *)p, (char *)s);
    return (p);
}

uchar *
qstring(uchar *s, int delim)    /* collect string up to delim */
{
    uchar *cbuf, *ret;
    int c, n;
    size_t  cbufsz, cnt;

    init_buf(&cbuf, &cbufsz, LINE_INCR);

    for (cnt = 0; (c = *s) != delim; s++) {
        if (c == '\n') {
            ERROR "newline in string %.10s...", cbuf SYNTAX;
        } else if (c != '\\') {
            expand_buf(&cbuf, &cbufsz, cnt);
            cbuf[cnt++] = c;
        } else {    /* \something */
            expand_buf(&cbuf, &cbufsz, cnt);
            switch (c = *++s) {
            case '\\':  cbuf[cnt++] = '\\'; break;
            case 'n':   cbuf[cnt++] = '\n'; break;
            case 't':   cbuf[cnt++] = '\t'; break;
            case 'b':   cbuf[cnt++] = '\b'; break;
            case 'f':   cbuf[cnt++] = '\f'; break;
            case 'r':   cbuf[cnt++] = '\r'; break;
            default:
                if (!isdigit(c)) {
                    cbuf[cnt++] = c;
                    break;
                }
                n = c - '0';
                if (isdigit(s[1])) {
                    n = 8 * n + *++s - '0';
                    if (isdigit(s[1]))
                        n = 8 * n + *++s - '0';
                }
                cbuf[cnt++] = n;
                break;
            }
        }
    }
    cbuf[cnt] = '\0';
    ret = tostring(cbuf);
    free(cbuf);
    return (ret);
}