/*

 * Copyright (c) 2005, 2008 Sun Microsystems, Inc. All Rights Reserved.

 * Use is subject to license terms.

 *

 *	Copyright (c) 1984 AT&T

 *	  All Rights Reserved  	

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *  http://www.apache.org/licenses/LICENSE-2.0. 

 * 

 * Unless required by applicable law or agreed to in writing, software 

 * distributed under the License is distributed on an "AS IS" BASIS, 

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express 

 * or implied. 

 * See the License for the specific language governing permissions and

 * limitations under the License. 

 */

#include "apr.h"

#include "apr_strings.h"

#include "libsed.h"

#include "sed.h"

#include "regexp.h"

#define CCEOF 22

static int fcomp(sed_commands_t *commands, apr_file_t *fin);

static char *compsub(sed_commands_t *commands,

                     sed_comp_args *compargs, char *rhsbuf);

static int rline(sed_commands_t *commands, apr_file_t *fin,

                 char *lbuf, char *lbend);

static char *address(sed_commands_t *commands, char *expbuf,

                     apr_status_t* status);

static char *text(sed_commands_t *commands, char *textbuf, char *endbuf);

static sed_label_t *search(sed_commands_t *commands);

static char *ycomp(sed_commands_t *commands, char *expbuf);

static char *comple(sed_commands_t *commands, sed_comp_args *compargs,

                    char *x1, char *ep, char *x3, char x4);

static sed_reptr_t *alloc_reptr(sed_commands_t *commands);

static int check_finalized(const sed_commands_t *commands);

void command_errf(sed_commands_t *commands, const char *fmt, ...)

{

    if (commands->errfn && commands->pool) {

        va_list args;

        const char* error;

        va_start(args, fmt);

        error = apr_pvsprintf(commands->pool, fmt, args);

        commands->errfn(commands->data, error);

        va_end(args);

    }

}

/*

 * sed_init_commands

 */

apr_status_t sed_init_commands(sed_commands_t *commands, sed_err_fn_t *errfn, void *data,

                               apr_pool_t *p)

{

    memset(commands, 0, sizeof(*commands));

    commands->errfn = errfn;

    commands->data = data;

    commands->labtab = commands->ltab;

    commands->lab = commands->labtab + 1;

    commands->pool = p;

    commands->respace = apr_pcalloc(p, RESIZE);

    if (commands->respace == NULL) {

        command_errf(commands, SEDERR_OOMMES);

        return APR_EGENERAL;

    }

    commands->rep = alloc_reptr(commands);

    if (commands->rep == NULL)

        return APR_EGENERAL;

    commands->rep->ad1 = commands->respace;

    commands->reend = &commands->respace[RESIZE - 1];

    commands->labend = &commands->labtab[SED_LABSIZE];

    commands->canbefinal = 1;

    return APR_SUCCESS;

}

/*

 * sed_destroy_commands

 */

void sed_destroy_commands(sed_commands_t *commands)

{

}

/*

 * sed_compile_string

 */

apr_status_t sed_compile_string(sed_commands_t *commands, const char *s)

{

    apr_status_t rv;

    commands->earg = s;

    commands->eflag = 1;

    rv = fcomp(commands, NULL);

    if (rv == APR_SUCCESS)

        commands->canbefinal = check_finalized(commands);

    commands->eflag = 0;

    return (rv != 0 ? APR_EGENERAL : APR_SUCCESS);

}

/*

 * sed_compile_file

 */

apr_status_t sed_compile_file(sed_commands_t *commands, apr_file_t *fin)

{

    apr_status_t rv = fcomp(commands, fin);

    return (rv != 0 ? APR_EGENERAL : APR_SUCCESS);

}

/*

 * sed_get_finalize_error

 */

char* sed_get_finalize_error(const sed_commands_t *commands, apr_pool_t* pool)

{

    const sed_label_t *lab;

    if (commands->depth) {

        return SEDERR_TMOMES;

    }

    /* Empty branch chain is not a issue */

    for (lab = commands->labtab + 1; lab < commands->lab; lab++) {

        char *error;

        if (lab->address == 0) {

            error = apr_psprintf(pool, SEDERR_ULMES, lab->asc);

            return error;

        }

        if (lab->chain) {

            return SEDERR_INTERNAL;

        }

    }

    return NULL;

}

/*

 * sed_canbe_finalized

 */

int sed_canbe_finalized(const sed_commands_t *commands)

{

    return commands->canbefinal;

}

/*

 * check_finalized

 */

static int check_finalized(const sed_commands_t *commands)

{

    const sed_label_t *lab;

    if (commands->depth) {

        return 0;

    }

    /* Empty branch chain is not a issue */

    for (lab = commands->labtab + 1; lab < commands->lab; lab++) {

        if (lab->address == 0 || (lab->chain)) {

            return 0;

        }

    }

    return 1;

}

/*

 * dechain

 */

static void dechain(sed_label_t *lpt, sed_reptr_t *address)

{

    sed_reptr_t *rep;

    if ((lpt == NULL) || (lpt->chain == NULL) || (address == NULL))

        return;

    rep = lpt->chain;

    while (rep->lb1) {

        sed_reptr_t *next;

        next = rep->lb1;

        rep->lb1 = address;

        rep = next;

    }

    rep->lb1 = address;

    lpt->chain = NULL;

}

/*

 * fcomp

 */

static int fcomp(sed_commands_t *commands, apr_file_t *fin)

{

    char *p, *op, *tp;

    sed_reptr_t *pt, *pt1;

    int i, ii;

    sed_label_t *lpt;

    char fnamebuf[APR_PATH_MAX];

    apr_status_t status;

    sed_comp_args compargs;

    op = commands->lastre;

    if (!commands->linebuf) {

        commands->linebuf = apr_pcalloc(commands->pool, LBSIZE + 1);

    }

    if (rline(commands, fin, commands->linebuf,

              (commands->linebuf + LBSIZE + 1)) < 0)

        return 0;

    if (*commands->linebuf == '#') {

        if (commands->linebuf[1] == 'n')

            commands->nflag = 1;

    }

    else {

        commands->cp = commands->linebuf;

        goto comploop;

    }

    for (;;) {

        if (rline(commands, fin, commands->linebuf,

                  (commands->linebuf + LBSIZE + 1)) < 0)

            break;

        commands->cp = commands->linebuf;

comploop:

        while (*commands->cp == ' ' || *commands->cp == '\t')

            commands->cp++;

        if (*commands->cp == '\0' || *commands->cp == '#')

            continue;

        if (*commands->cp == ';') {

            commands->cp++;

            goto comploop;

        }

        p = address(commands, commands->rep->ad1, &status);

        if (status != APR_SUCCESS) {

            command_errf(commands, SEDERR_CGMES, commands->linebuf);

            return -1;

        }

        if (p == commands->rep->ad1) {

            if (op)

                commands->rep->ad1 = op;

            else {

                command_errf(commands, SEDERR_NRMES);

                return -1;

            }

        } else if (p == 0) {

            p = commands->rep->ad1;

            commands->rep->ad1 = 0;

        } else {

            op = commands->rep->ad1;

            if (*commands->cp == ',' || *commands->cp == ';') {

                commands->cp++;

                commands->rep->ad2 = p;

                p = address(commands, commands->rep->ad2, &status);

                if ((status != APR_SUCCESS) || (p == 0)) {

                    command_errf(commands, SEDERR_CGMES, commands->linebuf);

                    return -1;

                }

                if (p == commands->rep->ad2)

                    commands->rep->ad2 = op;

                else

                    op = commands->rep->ad2;

            } else

                commands->rep->ad2 = 0;

        }

        if(p > &commands->respace[RESIZE-1]) {

            command_errf(commands, SEDERR_TMMES);

            return -1;

        }

        while (*commands->cp == ' ' || *commands->cp == '\t')

            commands->cp++;

swit:

        switch(*commands->cp++) {

        default:

            command_errf(commands, SEDERR_UCMES, commands->linebuf);

            return -1;

        case '!':

            commands->rep->negfl = 1;

            goto swit;

        case '{':

            commands->rep->command = BCOM;

            commands->rep->negfl = !(commands->rep->negfl);

            commands->cmpend[commands->depth++] = &commands->rep->lb1;

            commands->rep = alloc_reptr(commands);

            commands->rep->ad1 = p;

            if (*commands->cp == '\0')

                continue;

            goto comploop;

        case '}':

            if (commands->rep->ad1) {

                command_errf(commands, SEDERR_AD0MES, commands->linebuf);

                return -1;

            }

            if (--commands->depth < 0) {

                command_errf(commands, SEDERR_TMCMES);

                return -1;

            }

            *commands->cmpend[commands->depth] = commands->rep;

            commands->rep->ad1 = p;

            continue;

        case '=':

            commands->rep->command = EQCOM;

            if (commands->rep->ad2) {

                command_errf(commands, SEDERR_AD1MES, commands->linebuf);

                return -1;

            }

            break;

        case ':':

            if (commands->rep->ad1) {

                command_errf(commands, SEDERR_AD0MES, commands->linebuf);

                return -1;

            }

            while (*commands->cp++ == ' ');

            commands->cp--;

            tp = commands->lab->asc;

            while ((*tp++ = *commands->cp++)) {

                if (tp >= &(commands->lab->asc[8])) {

                    command_errf(commands, SEDERR_LTLMES, commands->linebuf);

                    return -1;

                }

            }

            *--tp = '\0';

            if ((lpt = search(commands)) != NULL) {

                if (lpt->address) {

                    command_errf(commands, SEDERR_DLMES, commands->linebuf);

                    return -1;

                }

                dechain(lpt, commands->rep);

            } else {

                commands->lab->chain = 0;

                lpt = commands->lab;

                if (++commands->lab >= commands->labend) {

                    command_errf(commands, SEDERR_TMLMES, commands->linebuf);

                    return -1;

                }

            }

            lpt->address = commands->rep;

            commands->rep->ad1 = p;

            continue;

        case 'a':

            commands->rep->command = ACOM;

            if (commands->rep->ad2) {

                command_errf(commands, SEDERR_AD1MES, commands->linebuf);

                return -1;

            }

            if (*commands->cp == '\\')

                commands->cp++;

            if (*commands->cp++ != '\n') {

                command_errf(commands, SEDERR_CGMES, commands->linebuf);

                return -1;

            }

            commands->rep->re1 = p;

            p = text(commands, commands->rep->re1, commands->reend);

            if (p == NULL)

                return -1;

            break;

        case 'c':

            commands->rep->command = CCOM;

            if (*commands->cp == '\\') commands->cp++;

            if (*commands->cp++ != ('\n')) {

                command_errf(commands, SEDERR_CGMES, commands->linebuf);

                return -1;

            }

            commands->rep->re1 = p;

            p = text(commands, commands->rep->re1, commands->reend);

            if (p == NULL)

                return -1;

            break;

        case 'i':

            commands->rep->command = ICOM;

            if (commands->rep->ad2) {

                command_errf(commands, SEDERR_AD1MES, commands->linebuf);

                return -1;

            }

            if (*commands->cp == '\\') commands->cp++;

            if (*commands->cp++ != ('\n')) {

                command_errf(commands, SEDERR_CGMES, commands->linebuf);

                return -1;

            }

            commands->rep->re1 = p;

            p = text(commands, commands->rep->re1, commands->reend);

            if (p == NULL)

                return -1;

            break;

        case 'g':

            commands->rep->command = GCOM;

            break;

        case 'G':

            commands->rep->command = CGCOM;

            break;

        case 'h':

            commands->rep->command = HCOM;

            break;

        case 'H':

            commands->rep->command = CHCOM;

            break;

        case 't':

            commands->rep->command = TCOM;

            goto jtcommon;

        case 'b':

            commands->rep->command = BCOM;

jtcommon:

            while (*commands->cp++ == ' ');

            commands->cp--;

            if (*commands->cp == '\0') {

                if ((pt = commands->labtab->chain) != NULL) {

                    while ((pt1 = pt->lb1) != NULL)

                        pt = pt1;

                    pt->lb1 = commands->rep;

                } else

                    commands->labtab->chain = commands->rep;

                break;

            }

            tp = commands->lab->asc;

            while ((*tp++ = *commands->cp++))

                if (tp >= &(commands->lab->asc[8])) {

                    command_errf(commands, SEDERR_LTLMES, commands->linebuf);

                    return -1;

                }

            commands->cp--;

            *--tp = '\0';

            if ((lpt = search(commands)) != NULL) {

                if (lpt->address) {

                    commands->rep->lb1 = lpt->address;

                } else {

                    pt = lpt->chain;

                    while ((pt1 = pt->lb1) != NULL)

                        pt = pt1;

                    pt->lb1 = commands->rep;

                }

            } else {

                commands->lab->chain = commands->rep;

                commands->lab->address = 0;

                if (++commands->lab >= commands->labend) {

                    command_errf(commands, SEDERR_TMLMES, commands->linebuf);

                    return -1;

                }

            }

            break;

        case 'n':

            commands->rep->command = NCOM;

            break;

        case 'N':

            commands->rep->command = CNCOM;

            break;

        case 'p':

            commands->rep->command = PCOM;

            break;

        case 'P':

            commands->rep->command = CPCOM;

            break;

        case 'r':

            commands->rep->command = RCOM;

            if (commands->rep->ad2) {

                command_errf(commands, SEDERR_AD1MES, commands->linebuf);

                return -1;

            }

            if (*commands->cp++ != ' ') {

                command_errf(commands, SEDERR_CGMES, commands->linebuf);

                return -1;

            }

            commands->rep->re1 = p;

            p = text(commands, commands->rep->re1, commands->reend);

            if (p == NULL)

                return -1;

            break;

        case 'd':

            commands->rep->command = DCOM;

            break;

        case 'D':

            commands->rep->command = CDCOM;

            commands->rep->lb1 = commands->ptrspace;

            break;

        case 'q':

            commands->rep->command = QCOM;

            if (commands->rep->ad2) {

                command_errf(commands, SEDERR_AD1MES, commands->linebuf);

                return -1;

            }

            break;

        case 'l':

            commands->rep->command = LCOM;

            break;

        case 's':