/*-

 * Copyright (c) 2011 Varnish Software AS

 * All rights reserved.

 *

 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>

 *

 * SPDX-License-Identifier: BSD-2-Clause

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND

 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE

 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL

 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS

 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)

 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY

 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF

 * SUCH DAMAGE.

 *

 * VEP Vinyl Esi Parsing

 */

#include "config.h"

#include "cache_int.h"

#include "cache_filter.h"

#include "cache_vgz.h"

#include "cache_esi.h"

#include "vct.h"

#include "vend.h"

#include "vgz.h"

//#define Debug(fmt, ...) printf(fmt, __VA_ARGS__)

#define Debug(fmt, ...) /**/

struct vep_state;

enum dowhat {DO_ATTR, DO_TAG};

typedef void dostuff_f(struct vep_state *, enum dowhat);

struct vep_match {

  const char  *match;

  const char  * const *state;

};

enum vep_mark { VERBATIM = 0, SKIP };

struct vep_state {

  unsigned    magic;

#define VEP_MAGIC   0x55cb9b82

  // flags from bereq

  struct vep_flags  flags;

  struct vsb    *vsb;

  const char    *url;

  struct vfp_ctx    *vc;

  int     dogzip;

  vep_callback_t    *cb;

  void      *cb_priv;

  /* Internal Counter for default call-back function */

  ssize_t     cb_x;

  /* parser state */

  const char    *state;

  unsigned    startup;

  unsigned    esi_found;

  unsigned    endtag;

  unsigned    emptytag;

  unsigned    canattr;

  unsigned    remove;

  ssize_t     o_wait;

  ssize_t     o_pending;

  ssize_t     o_total;

  uint32_t    crc;

  ssize_t     o_crc;

  uint32_t    crcp;

  ssize_t     o_last;

  const char    *hack_p;

  const char    *ver_p;

  const char    *until;

  const char    *until_p;

  const char    *until_s;

  int     in_esi_tag;

  const char    *esicmt;

  const char    *esicmt_p;

  struct vep_match  *attr;

  struct vsb    *attr_vsb;

  int     attr_delim;

  struct vep_match  *match;

  struct vep_match  *match_hit;

  char      tag[8];

  int     tag_i;

  dostuff_f   *dostuff;

  struct vsb    *include_src;

  unsigned    include_continue;

  unsigned    nm_skip;

  unsigned    nm_verbatim;

  unsigned    nm_pending;

  enum vep_mark   last_mark;

};

/*---------------------------------------------------------------------*/

static const char * const VEP_START =   "[Start]";

static const char * const VEP_BOM =   "[BOM]";

static const char * const VEP_TESTXML =   "[TestXml]";

static const char * const VEP_NOTXML =    "[NotXml]";

static const char * const VEP_NEXTTAG =   "[NxtTag]";

static const char * const VEP_NOTMYTAG =  "[NotMyTag]";

static const char * const VEP_STARTTAG =  "[StartTag]";

static const char * const VEP_COMMENTESI =  "[CommentESI]";

static const char * const VEP_COMMENT =   "[Comment]";

static const char * const VEP_CDATA =   "[CDATA]";

static const char * const VEP_ESITAG =    "[ESITag]";

static const char * const VEP_ESIENDTAG = "[/ESITag]";

static const char * const VEP_ESIREMOVE = "[ESI:Remove]";

static const char * const VEP_ESIINCLUDE =  "[ESI:Include]";

static const char * const VEP_ESICOMMENT =  "[ESI:Comment]";

static const char * const VEP_ESIBOGON =  "[ESI:Bogon]";

static const char * const VEP_INTAG =   "[InTag]";

static const char * const VEP_TAGERROR =  "[TagError]";

static const char * const VEP_ATTR =    "[Attribute]";

static const char * const VEP_SKIPATTR =  "[SkipAttribute]";

static const char * const VEP_ATTRDELIM = "[AttrDelim]";

static const char * const VEP_ATTRGETVAL =  "[AttrGetValue]";

static const char * const VEP_ATTRVAL =   "[AttrValue]";

static const char * const VEP_UNTIL =   "[Until]";

static const char * const VEP_MATCHBUF =  "[MatchBuf]";

static const char * const VEP_MATCH =   "[Match]";

/*---------------------------------------------------------------------*/

static struct vep_match vep_match_starttag[] = {

  { "!--esi", &VEP_COMMENTESI },

  { "!---->", &VEP_NEXTTAG },

  { "!--",  &VEP_COMMENT },

  { "/esi:",  &VEP_ESIENDTAG },

  { "esi:", &VEP_ESITAG },

  { "![CDATA[", &VEP_CDATA },

  { NULL,   &VEP_NOTMYTAG }

};

/*---------------------------------------------------------------------*/

static struct vep_match vep_match_esi[] = {

  { "include",  &VEP_ESIINCLUDE },

  { "remove", &VEP_ESIREMOVE },

  { "comment",  &VEP_ESICOMMENT },

  { NULL,   &VEP_ESIBOGON }

};

/*---------------------------------------------------------------------*/

static struct vep_match vep_match_attr_include[] = {

  { "src=", &VEP_ATTRGETVAL },

  { "onerror=", &VEP_ATTRGETVAL },

  { NULL,   &VEP_SKIPATTR }

};

/*---------------------------------------------------------------------*/

static struct vep_match vep_match_bom[] = {

  { "\xeb\xbb\xbf", &VEP_START },

  { NULL,     &VEP_BOM }

};

/*--------------------------------------------------------------------

 * Report a parsing error

 */

static void

vep_error(const struct vep_state *vep, const char *p)

{

  VSC_C_main->esi_errors++;

  VSLb(vep->vc->wrk->vsl, SLT_ESI_xmlerror, "ERR: after %zd %s",

       vep->o_last, p);

}

/*--------------------------------------------------------------------

 * Report a parsing warning

 */

static void

vep_warn(const struct vep_state *vep, const char *p)

{

  VSC_C_main->esi_warnings++;

  VSLb(vep->vc->wrk->vsl, SLT_ESI_xmlerror, "WARN: after %zd %s",

       vep->o_last, p);

}

/*---------------------------------------------------------------------

 * return match or NULL if more input needed.

 */

static struct vep_match *

vep_match(const struct vep_state *vep, const char *b, const char *e)

{

  struct vep_match *vm;

  const char *q, *r;

  AN(vep->match);

  for (vm = vep->match; vm->match != NULL; vm++) {

    assert(vstrlen(vm->match) <= sizeof (vep->tag));

    r = b;

    for (q = vm->match; *q != '\0' && r < e; q++, r++)

      if (*q != *r)

        break;

    if (*q == '\0')

      break;

    if (r == e)

      return (NULL);

  }

  return (vm);

}

/*---------------------------------------------------------------------

 *

 */

static void

vep_emit_len(const struct vep_state *vep, ssize_t l, int m8, int m16, int m64)

{

  uint8_t buf[9];

  assert(l > 0);

  if (l < 256) {

    buf[0] = (uint8_t)m8;

    buf[1] = (uint8_t)l;

    assert((ssize_t)buf[1] == l);

    VSB_bcat(vep->vsb, buf, 2);

  } else if (l < 65536) {

    buf[0] = (uint8_t)m16;

    vbe16enc(buf + 1, (uint16_t)l);

    assert((ssize_t)vbe16dec(buf + 1) == l);

    VSB_bcat(vep->vsb, buf, 3);

  } else {

    buf[0] = (uint8_t)m64;

    vbe64enc(buf + 1, l);

    assert((ssize_t)vbe64dec(buf + 1) == l);

    VSB_bcat(vep->vsb, buf, 9);

  }

}

static void

vep_emit_skip(const struct vep_state *vep, ssize_t l)

{

  vep_emit_len(vep, l, VEC_S1, VEC_S2, VEC_S8);

}

static void

vep_emit_verbatim(const struct vep_state *vep, ssize_t l, ssize_t l_crc)

{

  uint8_t buf[4];

  vep_emit_len(vep, l, VEC_V1, VEC_V2, VEC_V8);

  if (vep->dogzip) {

    vep_emit_len(vep, l_crc, VEC_C1, VEC_C2, VEC_C8);

    vbe32enc(buf, vep->crc);

    VSB_bcat(vep->vsb, buf, sizeof buf);

  }

}

static void

vep_emit_common(struct vep_state *vep, ssize_t l, enum vep_mark mark)

{

  assert(l >= 0);

  if (l == 0)

    return;

  assert(mark == SKIP || mark == VERBATIM);

  if (mark == SKIP)

    vep_emit_skip(vep, l);

  else

    vep_emit_verbatim(vep, l, vep->o_crc);

  vep->crc = crc32(0L, Z_NULL, 0);

  vep->o_crc = 0;

  vep->o_total += l;

}

/*---------------------------------------------------------------------

 *

 */

static void

vep_mark_common(struct vep_state *vep, const char *p, enum vep_mark mark)

{

  ssize_t l, lcb;

  assert(mark == SKIP || mark == VERBATIM);

  /* The NO-OP case, no data, no pending data & no change of mode */

  if (vep->last_mark == mark && p == vep->ver_p && vep->o_pending == 0)

    return;

  /*

   * If we changed mode, emit whatever the opposite mode

   * assembled before the pending bytes.

   */

  if (vep->last_mark != mark && (vep->o_wait > 0 || vep->startup)) {

    lcb = vep->cb(vep->vc, vep->cb_priv, 0,

        mark == VERBATIM ? VGZ_RESET : VGZ_ALIGN);

    vep_emit_common(vep, lcb - vep->o_last, vep->last_mark);

    vep->o_last = lcb;

    vep->o_wait = 0;

  }

  /* Transfer pending bytes CRC into active mode CRC */

  if (vep->o_pending) {

    (void)vep->cb(vep->vc, vep->cb_priv, vep->o_pending,

         VGZ_NORMAL);

    if (vep->o_crc == 0) {

      vep->crc = vep->crcp;

      vep->o_crc = vep->o_pending;

    } else {

      vep->crc = crc32_combine(vep->crc,

          vep->crcp, vep->o_pending);

      vep->o_crc += vep->o_pending;

    }

    vep->crcp = crc32(0L, Z_NULL, 0);

    vep->o_wait += vep->o_pending;

    vep->o_pending = 0;

  }

  /* * Process this bit of input */

  AN(vep->ver_p);

  l = p - vep->ver_p;

  assert(l >= 0);

  vep->crc = crc32(vep->crc, (const void*)vep->ver_p, l);

  vep->o_crc += l;

  vep->ver_p = p;

  vep->o_wait += l;

  vep->last_mark = mark;

  (void)vep->cb(vep->vc, vep->cb_priv, l, VGZ_NORMAL);

}

static void

vep_mark_verbatim(struct vep_state *vep, const char *p)

{

  vep_mark_common(vep, p, VERBATIM);

  vep->nm_verbatim++;

}

static void

vep_mark_skip(struct vep_state *vep, const char *p)

{

  vep_mark_common(vep, p, SKIP);

  vep->nm_skip++;

}

static void

vep_mark_pending(struct vep_state *vep, const char *p)

{

  ssize_t l;

  AN(vep->ver_p);

  l = p - vep->ver_p;

  assert(l > 0);

  vep->crcp = crc32(vep->crcp, (const void *)vep->ver_p, l);

  vep->ver_p = p;

  vep->o_pending += l;

  vep->nm_pending++;

}

/*---------------------------------------------------------------------

 */

static void v_matchproto_()

vep_do_comment(struct vep_state *vep, enum dowhat what)

{

  Debug("DO_COMMENT(%d)\n", what);

  assert(what == DO_TAG);

  if (!vep->emptytag)

    vep_error(vep, "ESI 1.0 <esi:comment> needs final '/'");

}

/*---------------------------------------------------------------------

 */

static void v_matchproto_()

vep_do_remove(struct vep_state *vep, enum dowhat what)

{

  Debug("DO_REMOVE(%d, end %d empty %d remove %d)\n",

      what, vep->endtag, vep->emptytag, vep->remove);

  assert(what == DO_TAG);

  if (vep->emptytag)

    vep_error(vep, "ESI 1.0 <esi:remove/> not legal");

  else if (vep->remove && !vep->endtag)

    vep_error(vep, "ESI 1.0 <esi:remove> already open");

  else if (!vep->remove && vep->endtag)

    vep_error(vep, "ESI 1.0 <esi:remove> not open");

  else

    vep->remove = !vep->endtag;

}

/*---------------------------------------------------------------------

 */

static void

include_attr_src(struct vep_state *vep)

{

  const char *p;

  if (vep->include_src != NULL) {

    vep_error(vep,

        "ESI 1.0 <esi:include> "

        "has multiple src= attributes");

    vep->state = VEP_TAGERROR;

    VSB_destroy(&vep->attr_vsb);

    VSB_destroy(&vep->include_src);

    return;

  }

  for (p = VSB_data(vep->attr_vsb); *p != '\0'; p++)

    if (vct_islws(*p))

      break;

  if (*p != '\0') {

    vep_error(vep,

        "ESI 1.0 <esi:include> "

        "has whitespace in src= attribute");

    vep->state = VEP_TAGERROR;

    VSB_destroy(&vep->attr_vsb);

    if (vep->include_src != NULL)

      VSB_destroy(&vep->include_src);

    return;

  }

  vep->include_src = vep->attr_vsb;

  vep->attr_vsb = NULL;

}

static void

include_attr_onerror(struct vep_state *vep)

{

  vep->include_continue = !vstrcmp("continue", VSB_data(vep->attr_vsb));

  VSB_destroy(&vep->attr_vsb);

}

static void v_matchproto_()

vep_do_include(struct vep_state *vep, enum dowhat what)

{

  const char *p, *q, *h;

  ssize_t l;

  char incl;

  Debug("DO_INCLUDE(%d)\n", what);

  if (what == DO_ATTR) {

    Debug("ATTR (%s) (%s)\n", vep->match_hit->match,

      VSB_data(vep->attr_vsb));

    if (!vstrcmp("src=", vep->match_hit->match)) {

      include_attr_src(vep);

      return;

    }

    if (!vstrcmp("onerror=", vep->match_hit->match)) {

      include_attr_onerror(vep);

      return;

    }

    WRONG("Unhandled <esi:include> attribute");

  }

  assert(what == DO_TAG);

  if (!vep->emptytag)

    vep_warn(vep, "ESI 1.0 <esi:include> lacks final '/'");

  if (vep->include_src == NULL) {

    vep_error(vep, "ESI 1.0 <esi:include> lacks src attr");

    return;

  }

  /*

   * Strictly speaking, we ought to spit out any piled up skip before

   * emitting the VEC for the include, but objectively that makes no

   * difference and robs us of a chance to collapse another skip into

   * this on so we don't do that.

   * However, we cannot tolerate any verbatim stuff piling up.

   * The mark_skip() before calling dostuff should have taken

   * care of that.  Make sure.

   */

  assert(vep->o_wait == 0 || vep->last_mark == SKIP);

  /* XXX: what if it contains NUL bytes ?? */

  p = VSB_data(vep->include_src);

  l = VSB_len(vep->include_src);

  h = 0;

  incl = vep->include_continue ? VEC_IC : VEC_IA;

  if (l > 7 && !vmemcmp(p, "http://", 7)) {

    h = p + 7;

    p = strchr(h, '/');

    if (p == NULL) {

      vep_error(vep,

          "ESI 1.0 <esi:include> invalid src= URL");

      vep->state = VEP_TAGERROR;

      AZ(vep->attr_vsb);

      VSB_destroy(&vep->include_src);

      return;

    }

    Debug("HOST <%.*s> PATH <%s>\n", (int)(p-h),h, p);

    VSB_printf(vep->vsb, "%c", incl);

    VSB_printf(vep->vsb, "Host: %.*s%c", (int)(p-h), h, 0);

  } else if (l > 8 && !vmemcmp(p, "https://", 8)) {

    if (!vep->flags.esi_ignore_https) {

      vep_warn(vep,

          "ESI 1.0 <esi:include> with https:// ignored");

      vep->state = VEP_TAGERROR;

      AZ(vep->attr_vsb);

      VSB_destroy(&vep->include_src);

      return;

    }

    vep_warn(vep,

        "ESI 1.0 <esi:include> https:// treated as http://");

    h = p + 8;

    p = strchr(h, '/');

    if (p == NULL) {

      vep_error(vep,

          "ESI 1.0 <esi:include> invalid src= URL");

      vep->state = VEP_TAGERROR;

      AZ(vep->attr_vsb);

      VSB_destroy(&vep->include_src);

      return;

    }

    VSB_printf(vep->vsb, "%c", incl);

    VSB_printf(vep->vsb, "Host: %.*s%c", (int)(p-h), h, 0);

  } else if (*p == '/') {

    VSB_printf(vep->vsb, "%c", incl);

    VSB_printf(vep->vsb, "%c", 0);

  } else {

    VSB_printf(vep->vsb, "%c", incl);

    VSB_printf(vep->vsb, "%c", 0);

    /* Look for the last / before a '?' */

    h = NULL;

    for (q = vep->url; *q && *q != '?'; q++)

      if (*q == '/')

        h = q;

    if (h == NULL)

      h = q + 1;

    Debug("INCL:: [%.*s]/[%s]\n",

        (int)(h - vep->url), vep->url, p);

    VSB_printf(vep->vsb, "%.*s/", (int)(h - vep->url), vep->url);

  }

  l -= (p - VSB_data(vep->include_src));

  for (q = p; *q != '\0'; ) {

    if (*q == '&') {

#define R(w,f,r)              \

      if (q + w <= p + l && !vmemcmp(q, f, w)) { \

        VSB_printf(vep->vsb, "%c", r);  \

        q += w;       \

        continue;     \

      }

      R(6, "'", '\'');

      R(6, """, '"');

      R(4, "&lt;", '<');

      R(4, ">", '>');

      R(5, "&", '&');

    }

    VSB_printf(vep->vsb, "%c", *q++);

  }

#undef R

  VSB_printf(vep->vsb, "%c", 0);

  VSB_destroy(&vep->include_src);

  vep->include_continue = 0;

}

/*---------------------------------------------------------------------

 * Lex/Parse object for ESI instructions

 *

 * This function is called with the input object piecemeal so do not

 * assume that we have more than one char available at at time, but

 * optimize for getting huge chunks.

 *

 * NB: At the bottom of this source-file, there is a dot-diagram matching

 * NB: the state-machine.  Please maintain it along with the code.

 */

void

VEP_Parse(struct vep_state *vep, const char *p, size_t l)

{

  const char *e;

  struct vep_match *vm;

  int i;

  CHECK_OBJ_NOTNULL(vep, VEP_MAGIC);

  assert(l > 0);

  if (vep->startup) {

    /*

     * We must force the GZIP header out as a SKIP string,

     * otherwise an object starting with <esi:include would

     * have its GZIP header appear after the included object

     * (e000026.vtc)

     */

    vep->ver_p = "";

    vep->last_mark = SKIP;

    vep_mark_common(vep, vep->ver_p, VERBATIM);

    vep->startup = 0;

    AZ(vep->hack_p);

    vep->hack_p = p;

  }

  vep->ver_p = p;

  e = p + l;

  while (p < e) {

    AN(vep->state);

    Debug("EP %s %d (%.*s) [%.*s]\n",

        vep->state,

        vep->remove,

        vep->tag_i, vep->tag,

        (e - p) > 10 ? 10 : (int)(e-p), p);

    assert(p >= vep->ver_p);

    /******************************************************

     * SECTION A

     */

    if (vep->state == VEP_START) {

      if (vep->flags.esi_remove_bom &&

          *p == (char)0xeb) {

        vep->match = vep_match_bom;

        vep->state = VEP_MATCH;

      } else

        vep->state = VEP_BOM;

    } else if (vep->state == VEP_BOM) {

      vep_mark_skip(vep, p);

      if (vep->flags.esi_disable_xml_check)

        vep->state = VEP_NEXTTAG;

      else

        vep->state = VEP_TESTXML;

    } else if (vep->state == VEP_TESTXML) {

      /*

       * If the first non-whitespace char is different

       * from '<' we assume this is not XML.

       */

      while (p < e && vct_islws(*p))

        p++;

      vep_mark_verbatim(vep, p);

      if (p < e && *p == '<') {

        p++;

        vep->state = VEP_STARTTAG;

      } else if (p < e && *p == (char)0xeb) {

        VSLb(vep->vc->wrk->vsl, SLT_ESI_xmlerror,

            "WARN: No ESI processing, "

            "first char not '<' but BOM."

            " (See feature esi_remove_bom)"

        );

        vep->state = VEP_NOTXML;

      } else if (p < e) {

        VSLb(vep->vc->wrk->vsl, SLT_ESI_xmlerror,

            "WARN: No ESI processing, "

            "first char not '<'."

            " (See feature esi_disable_xml_check)"

        );

        vep->state = VEP_NOTXML;

      }

    } else if (vep->state == VEP_NOTXML) {

      /*

       * This is not recognized as XML, just skip thru

       * vfp_esi_end() will handle the rest

       */

      p = e;

      vep_mark_verbatim(vep, p);

    /******************************************************

     * SECTION B

     */

    } else if (vep->state == VEP_NOTMYTAG) {

      if (vep->flags.esi_ignore_other_elements) {

        p++;

        vep->state = VEP_NEXTTAG;

      } else {

        vep->tag_i = 0;

        while (p < e) {

          if (*p++ == '>') {

            vep->state = VEP_NEXTTAG;

            break;

          }

        }

      }

      if (p == e && !vep->remove)

        vep_mark_verbatim(vep, p);

    } else if (vep->state == VEP_NEXTTAG) {

      /*

       * Hunt for start of next tag and keep an eye

       * out for end of EsiCmt if armed.

       */

      vep->emptytag = 0;

      vep->attr = NULL;

      vep->dostuff = NULL;

      while (p < e && *p != '<') {

        if (vep->esicmt_p == NULL) {

          p++;

          continue;

        }

        if (*p != *vep->esicmt_p) {

          p++;

          vep->esicmt_p = vep->esicmt;

          continue;

        }

        if (!vep->remove && vep->esicmt_p == vep->esicmt)

          vep_mark_verbatim(vep, p);

        p++;

        if (*++vep->esicmt_p == '\0') {

          vep->esi_found = 1;

          vep->esicmt = NULL;

          vep->esicmt_p = NULL;

          /*

           * The end of the esicmt

           * should not be emitted.

           * But the stuff before should

           */

          vep_mark_skip(vep, p);

        }

      }

      if (p < e) {

        if (!vep->remove)

          vep_mark_verbatim(vep, p);

        assert(*p == '<');

        p++;

        vep->state = VEP_STARTTAG;

      } else if (vep->esicmt_p == vep->esicmt && !vep->remove)

        vep_mark_verbatim(vep, p);

    /******************************************************

     * SECTION C

     */

    } else if (vep->state == VEP_STARTTAG) {

      /* Start of tag, set up match table */

      vep->endtag = 0;

      vep->match = vep_match_starttag;

      vep->state = VEP_MATCH;

    } else if (vep->state == VEP_COMMENT) {

      vep->esicmt_p = vep->esicmt = NULL;

      vep->until_p = vep->until = "-->";

      vep->until_s = VEP_NEXTTAG;

      vep->state = VEP_UNTIL;

    } else if (vep->state == VEP_COMMENTESI) {

      if (vep->remove)

        vep_error(vep,

            "ESI 1.0 Nested <!--esi"

            " element in <esi:remove>");

      vep->esicmt_p = vep->esicmt = "-->";

      vep->state = VEP_NEXTTAG;

      vep_mark_skip(vep, p);

    } else if (vep->state == VEP_CDATA) {

      /*

       * Easy: just look for the end of CDATA

       */

      vep->until_p = vep->until = "]]>";

      vep->until_s = VEP_NEXTTAG;

      vep->state = VEP_UNTIL;

    } else if (vep->state == VEP_ESIENDTAG) {

      vep->endtag = 1;

      vep->state = VEP_ESITAG;

    } else if (vep->state == VEP_ESITAG) {

      vep->in_esi_tag = 1;

      vep->esi_found = 1;

      vep_mark_skip(vep, p);

      vep->match = vep_match_esi;

      vep->state = VEP_MATCH;

    } else if (vep->state == VEP_ESIINCLUDE) {

      if (vep->remove) {

        vep_error(vep,

            "ESI 1.0 <esi:include> element"

            " nested in <esi:remove>");

        vep->state = VEP_TAGERROR;

      } else if (vep->endtag) {

        vep_error(vep,

            "ESI 1.0 </esi:include> illegal end-tag");

        vep->state = VEP_TAGERROR;

      } else {

        vep->dostuff = vep_do_include;

        vep->state = VEP_INTAG;

        vep->attr = vep_match_attr_include;

      }

    } else if (vep->state == VEP_ESIREMOVE) {

      vep->dostuff = vep_do_remove;

      vep->state = VEP_INTAG;

    } else if (vep->state == VEP_ESICOMMENT) {

      if (vep->remove) {

        vep_error(vep,

            "ESI 1.0 <esi:comment> element"

            " nested in <esi:remove>");

        vep->state = VEP_TAGERROR;

      } else if (vep->endtag) {

        vep_error(vep,

            "ESI 1.0 </esi:comment> illegal end-tag");

        vep->state = VEP_TAGERROR;

      } else {

        vep->dostuff = vep_do_comment;

        vep->state = VEP_INTAG;

      }

    } else if (vep->state == VEP_ESIBOGON) {

      vep_error(vep,

          "ESI 1.0 <esi:bogus> element");

      vep->state = VEP_TAGERROR;

    /******************************************************

     * SECTION D

     */

    } else if (vep->state == VEP_INTAG) {

      vep->tag_i = 0;

      while (p < e && vct_islws(*p) && !vep->emptytag) {

        p++;

        vep->canattr = 1;

      }

      if (p < e && *p == '/' && !vep->emptytag) {

        p++;

        vep->emptytag = 1;

        vep->canattr = 0;

      }

      if (p < e && *p == '>') {

        p++;

        AN(vep->dostuff);

        vep_mark_skip(vep, p);

        vep->dostuff(vep, DO_TAG);

        vep->in_esi_tag = 0;

        vep->state = VEP_NEXTTAG;

      } else if (p < e && vep->emptytag) {

        vep_error(vep,

            "XML 1.0 '>' does not follow '/' in tag");

        vep->state = VEP_TAGERROR;

      } else if (p < e && vep->canattr &&

          vct_isxmlnamestart(*p)) {

        vep->state = VEP_ATTR;

      } else if (p < e) {

        vep_error(vep,

            "XML 1.0 Illegal attribute start char");

        vep->state = VEP_TAGERROR;

      }

    } else if (vep->state == VEP_TAGERROR) {

      while (p < e && *p != '>')

        p++;

      if (p < e) {

        p++;

        vep_mark_skip(vep, p);

        vep->in_esi_tag = 0;

        vep->state = VEP_NEXTTAG;

        if (vep->attr_vsb)

          VSB_destroy(&vep->attr_vsb);

      }

    /******************************************************

     * SECTION E

     */

    } else if (vep->state == VEP_ATTR) {

      AZ(vep->attr_delim);

      if (vep->attr == NULL) {

        p++;

        AZ(vep->attr_vsb);

        vep->state = VEP_SKIPATTR;

      } else {

        vep->match = vep->attr;

        vep->state = VEP_MATCH;

      }

    } else if (vep->state == VEP_SKIPATTR) {

      while (p < e && vct_isxmlname(*p))

        p++;

      if (p < e && *p == '=') {

        p++;

        vep->state = VEP_ATTRDELIM;

      } else if (p < e && *p == '>') {

        vep->state = VEP_INTAG;

      } else if (p < e && *p == '/') {

        vep->state = VEP_INTAG;

      } else if (p < e && vct_issp(*p)) {

        vep->state = VEP_INTAG;

      } else if (p < e) {

        vep_error(vep,

            "XML 1.0 Illegal attr char");

        vep->state = VEP_TAGERROR;

      }

    } else if (vep->state == VEP_ATTRGETVAL) {

      AZ(vep->attr_vsb);

      vep->attr_vsb = VSB_new_auto();

      vep->state = VEP_ATTRDELIM;

    } else if (vep->state == VEP_ATTRDELIM) {

      AZ(vep->attr_delim);

      if (*p == '"' || *p == '\'') {

        vep->attr_delim = *p++;

        vep->state = VEP_ATTRVAL;

      } else if (!vct_issp(*p)) {

        vep->attr_delim = ' ';

        vep->state = VEP_ATTRVAL;

      } else {

        vep_error(vep,

            "XML 1.0 Illegal attribute delimiter");

        vep->state = VEP_TAGERROR;

      }

    } else if (vep->state == VEP_ATTRVAL) {

      while (p < e && *p != '>' && *p != vep->attr_delim &&

         (vep->attr_delim != ' ' || !vct_issp(*p))) {

        if (vep->attr_vsb != NULL)

          VSB_putc(vep->attr_vsb, *p);

        p++;

      }

      if (p < e && *p == '>') {

        vep_error(vep,

            "XML 1.0 Missing end attribute delimiter");

        vep->state = VEP_TAGERROR;

        vep->attr_delim = 0;

        if (vep->attr_vsb != NULL) {

          AZ(VSB_finish(vep->attr_vsb));

          VSB_destroy(&vep->attr_vsb);

        }

      } else if (p < e) {

        vep->attr_delim = 0;

        p++;

        vep->state = VEP_INTAG;

        if (vep->attr_vsb != NULL) {

          AZ(VSB_finish(vep->attr_vsb));

          AN(vep->dostuff);

          vep->dostuff(vep, DO_ATTR);

          vep->attr_vsb = NULL;

        }

      }

    /******************************************************

     * Utility Section

     */

    } else if (vep->state == VEP_MATCH) {

      /*

       * Match against a table

       */

      vm = vep_match(vep, p, e);

      vep->match_hit = vm;

      if (vm != NULL) {

        if (vm->match != NULL)

          p += vstrlen(vm->match);

        vep->state = *vm->state;

        vep->match = NULL;

        vep->tag_i = 0;

      } else {

        assert(p + sizeof(vep->tag) >= e);

        vmemcpy(vep->tag, p, e - p);

        vep->tag_i = e - p;

        vep->state = VEP_MATCHBUF;

        p = e;

      }

    } else if (vep->state == VEP_MATCHBUF) {

      /*

       * Match against a table while split over input

       * sections.

       */

      AN(vep->match);

      i = sizeof(vep->tag) - vep->tag_i;

      if (i > e - p)

        i = e - p;

      vmemcpy(vep->tag + vep->tag_i, p, i);

      vm = vep_match(vep, vep->tag,

          vep->tag + vep->tag_i + i);

      Debug("MB (%.*s) tag_i %d i %d = vm %p match %s\n",

          vep->tag_i + i, vep->tag,

          vep->tag_i,

          i,

          vm,

          vm ? vm->match : "(nil)");

      if (vm == NULL) {

        vep->tag_i += i;

        p += i;

        assert(p == e);

      } else {

        vep->match_hit = vm;

        vep->state = *vm->state;

        if (vm->match != NULL) {

          i = vstrlen(vm->match);

          if (i > vep->tag_i)

            p += i - vep->tag_i;

        }

        vep->match = NULL;

        vep->tag_i = 0;

      }

    } else if (vep->state == VEP_UNTIL) {

      /*

       * Skip until we see magic string

       */

      while (p < e) {

        if (*p++ != *vep->until_p++) {

          vep->until_p = vep->until;

        } else if (*vep->until_p == '\0') {

          vep->state = vep->until_s;

          break;

        }

      }

      if (p == e && !vep->remove)

        vep_mark_verbatim(vep, p);

    } else {

      Debug("*** Unknown state %s\n", vep->state);

      WRONG("WRONG ESI PARSER STATE");

    }

  }

  /*