/*

* Copyright (c) 2000 - 2001 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: 

*

*/

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

**  File: wbxml_parse_buffer.c

**  Description:

*

* A buffer-based implementation of the "parser_reader interface".

* To create a parser that gets wbxml from some other source (such

* as reading from a stream), these are the only functions that should

* have to be ported. wbxml_parse.c contains implementation of the

* parsing logic, independent of how wbxml is read/stored (with a few

* small exceptions).

*

* The parser reader is responsible for safely getting various primitive

* data types and in-line strings from the wbxml NW_Byte code. The reader

* also returns a pointer to the string table (which in this case is

* just a pointer into the wbxml buffer.)

*

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

#include "cxml_internal.h"

#include <xml/cxml/nw_wbxml_reader.h>

#include <xml/cxml/nw_wbxml_dictionary.h>

#include <xml/cxml/nw_wbxml_event.h>

#include <xml/cxml/nw_wbxml_token.h>

#include <xml/cxml/nw_wbxml_opaque.h>

#include "nw_wbxml_parsei.h"

#include <xml/cxml/nw_string_char.h>

#include <xml/cxml/nw_tinydom.h>

/*

 * Initialize the parser

 */

EXPORT_C void

NW_WBXML_Parser_newInPlace (NW_WBXML_Parser_t * parser)

{

  NW_ASSERT(parser != NULL);

  parser->handler = NULL;

  parser->flags = 0;

  parser->state = NW_WBXML_PARSER_S_START;

  parser->tag_code_page = 0;

  parser->attribute_code_page = 0;

  parser->cp_registry.storage = NULL;

  parser->cp_registry.current = 0;

  parser->cp_registry.count = 0;

  parser->dictionary = 0;

  parser->doc = NULL;

  parser->p = NULL;

  parser->offset = 0;

  parser->left = 0;

  parser->status = NW_STAT_SUCCESS;

  parser->recursiveCallCnt = 0;

  parser->ext_t_not_table_index = 0;

}

void

NW_WBXML_Parser_reset (NW_WBXML_Parser_t * parser)

{

  NW_ASSERT(parser != NULL);

  parser->handler = NULL;

  parser->state = NW_WBXML_PARSER_S_START;

  parser->tag_code_page = 0;

  parser->attribute_code_page = 0;

  parser->p = NULL;

  parser->offset = 0;

  parser->left = 0;

  parser->status = NW_STAT_SUCCESS;

  parser->recursiveCallCnt = 0;

}

void

NW_WBXML_Parser_delete (NW_WBXML_Parser_t * parser)

{

  if (parser->cp_registry.storage != NULL) {

    NW_Mem_Free(parser->cp_registry.storage);

  }

}

/*

* Safely advance the parser through bytecode.

* The functions which read values from the parser don't advance

* the parser since this may not be necessary or safe (e.g. when reading

* the last data item.).

*

* "Count" may equal "left" so zero is a possible "left" value.

*/

NW_Int32

NW_WBXML_Parser_advance (NW_WBXML_Parser_t * parser,

                         NW_Int32 count)

{

  NW_ASSERT(parser != NULL);

  if ((count > (NW_Int32) parser->left)

      || (((NW_Int32)parser->offset + count) < 0)){

    return -1;

  }

  parser->p += count;

  parser->offset = (NW_Uint32) ((NW_Int32)parser->offset +  count);

  parser->left = (NW_Uint32) ((NW_Int32) parser->left - count);

  return count;

}

/* Get the current offset */

NW_Uint32

NW_WBXML_Parser_getOffset(NW_WBXML_Parser_t * parser)

{

  NW_ASSERT(parser != NULL);

  return parser->offset;

}

/* Safely set the parser to a new offset */

NW_Int32

NW_WBXML_Parser_setOffset(NW_WBXML_Parser_t * parser,

                          NW_Uint32 offset)

{

  NW_Int32 count =

    (NW_Int32)offset - (NW_Int32)NW_WBXML_Parser_getOffset(parser);

  NW_ASSERT(parser != NULL);

  /* If the code page registry exists and is inited, reset code pages */

  if((parser->cp_registry.count > 0)

    && ((parser->flags & NW_WBXML_REGISTRY_INIT) == NW_WBXML_REGISTRY_INIT)){

    NW_WBXML_CPRegistry_getCodePages(&(parser->cp_registry),

                                     offset,

                                     &(parser->tag_code_page),

                                     &(parser->attribute_code_page));

  }

  /*

   * Try to "advance" the parser to the new position. This will fail

   * if the offset is not in the parser's current buffer.

   */

  return NW_WBXML_Parser_advance(parser, count);

}

/* Check if there is more NW_Byte code to read */

NW_Int32

NW_WBXML_Parser_hasMoreBytecode (NW_WBXML_Parser_t * parser)

{

  NW_ASSERT(parser != NULL);

  return (parser->left > 0);

}

/*

 * Read one unsigned NW_Byte from the current parser position.

 */

NW_Uint8

NW_WBXML_Parser_readUint8 (NW_WBXML_Parser_t * parser)

{

  NW_ASSERT(parser != NULL);

  return (NW_Uint8) (*(parser->p));

}

/*

 * Safely read an unsigned_32 from the current parser position

 */

NW_Int32

NW_WBXML_Parser_readMbUint32 (NW_WBXML_Parser_t * parser,

                              NW_Uint32 * val)

{

  NW_Uint32 i, space, next;

  NW_ASSERT(parser != NULL);

  space = parser->left;

  for (i = 0, *val = 0;; i++, *val <<= 7)

  {

    if (i > 4 || i == space)

    {

      return -1;

    }

    next = (parser->p)[i];

    if (!(next & 0x80))

    {

      *val |= next;

      break;

    }

    else

      *val |= (next &= 0x7F);

  }

  return (NW_Int32)(i + 1);

}

/*

 * Safely get the length of an inline string at current parser position

 */

NW_Int32 NW_WBXML_Parser_getInlineStrLen(NW_WBXML_Parser_t *parser,

                                         NW_WBXML_Document_t *doc)

{

  NW_Uint32 len = 0;

  NW_ASSERT(parser != NULL);

  if (!NW_String_valid(parser->p, parser->left, doc->charset)) {

    return -1;

  }

  if(NW_String_charBuffGetLength (parser->p, doc->charset, &len) < 0){

    return -1;

  }

  return (NW_Int32)len;

}

/*

 * Safely read an inline string at the current parser position

 */

EXPORT_C NW_Status_t

NW_WBXML_Parser_getStringInline (NW_WBXML_Parser_t * parser,

                                 NW_WBXML_Document_t * doc,

                                 NW_String_t *string)

{

  NW_ASSERT(parser != NULL);

  NW_ASSERT(doc != NULL);

  NW_ASSERT(string != NULL);

  if (!NW_String_valid (parser->p, parser->left, doc->charset))

  {

    return NW_STAT_WBXML_ERROR_BYTECODE;

  }

  if (NW_String_initialize (string, parser->p, doc->charset)

      != NW_STAT_SUCCESS){

    return NW_STAT_WBXML_ERROR_BYTECODE;

  }

  return NW_STAT_SUCCESS;

}

/*

 * Safely get an opaque at the current parser position

 */

EXPORT_C NW_Status_t

NW_WBXML_Parser_getOpaque (NW_WBXML_Parser_t * parser,

                           NW_Uint32 length,

                           NW_WBXML_Opaque_t *opaque)

{

  NW_ASSERT(parser != NULL);

  NW_ASSERT(length != 0);

  NW_ASSERT(opaque != NULL);

  if (length > parser->left){

    return NW_STAT_WBXML_ERROR_BYTECODE;

  }

  return NW_WBXML_Opaque_construct (opaque, parser->p, length);

}

/*

 * Read in the string table

 */

NW_Status_t

NW_WBXML_Parser_readStringTable (NW_WBXML_Parser_t * parser,

                                 NW_WBXML_Document_t * doc,

                                 NW_Byte **table)

{

  NW_Int32 skip;

  NW_ASSERT(parser != NULL);

  NW_ASSERT(doc != NULL);

  NW_ASSERT(table != NULL);

  *table = NULL;

  /* Read the string table length into the document header */

  skip = NW_WBXML_Parser_readMbUint32 (parser, &(doc->strtbl.length));

  if (skip < 0) {

    return NW_STAT_WBXML_ERROR_BYTECODE;

  }

  /* Try to advance over the length bytes to beginning of table */

  if (NW_WBXML_Parser_advance (parser, skip) < 0){

    return NW_STAT_WBXML_ERROR_BYTECODE;

  }

  /* Set table to current parser location*/

  *table = parser->p;

  /* Try to advance to end of table */

  if (NW_WBXML_Parser_advance (parser, (NW_Int32) doc->strtbl.length) < 0){

    return NW_STAT_WBXML_ERROR_BYTECODE;

  }

  return NW_STAT_SUCCESS;

}

/* Reset the parser to the start of a buffer. The buffsize is

 * critical here since it is used in all safety tests by the

 * parser reader to make sure the parser doesn't read past the

 * end of the buffer.

 */

NW_Status_t

NW_WBXML_Parser_resetBuffer(NW_WBXML_Parser_t *parser,

                            NW_Byte *buff,

                            NW_Uint32 buffsize)

{

  NW_ASSERT(parser != NULL);

  parser->offset = 0;

  parser->p = buff;

  if (buffsize == 0 || buff == NULL) {

    parser->left = 0;

  }

  else {

    parser->left = buffsize;

  }

  parser->tag_code_page = 0;

  parser->attribute_code_page = 0;

  parser->state = NW_WBXML_PARSER_S_HEADER;

  parser->recursiveCallCnt = 0;

  return NW_STAT_SUCCESS;

}

/*

 * This method sets the parser to a buffer which may contain only

 * wbxml fragments and no header. Used by writable DOM only.

 *

 */

NW_Status_t

NW_WBXML_Parser_setBuffer (NW_WBXML_Parser_t * parser,

                           NW_WBXML_Document_t * doc,

                           NW_Byte * buff,

                           NW_Uint32 buffsize)

{

  NW_ASSERT(parser != NULL);

  NW_ASSERT(doc != NULL);

  NW_ASSERT(buff != NULL);

  NW_ASSERT(buffsize > 0);

  /* Find dictionary from document */

  if (doc->publicid > 0){

    parser->dictionary =

      NW_WBXML_Dictionary_getIndexByPublicId (doc->publicid);

  }

  else if (doc->doc_type){

    parser->dictionary =

      NW_WBXML_Dictionary_getIndexByDocType (doc->doc_type, doc->charset);

  }

  else{

    NW_ASSERT(0); /* Doc not correctly initialized */

  }

  /* Set the buffer */

  NW_WBXML_Parser_resetBuffer(parser, buff, buffsize);

  return NW_STAT_SUCCESS;

}

/*

 * Parse a document from a buffer

 */

EXPORT_C NW_Status_t

NW_WBXML_Parser_parseBuffer (NW_WBXML_Parser_t * parser,

                             NW_WBXML_Document_t * doc,

                             NW_Byte * buff,

                             NW_Uint32 buffsize)

{

  NW_Status_t status;

  NW_ASSERT(parser != NULL);

  NW_ASSERT(doc != NULL);

  NW_ASSERT(buff != NULL);

  NW_ASSERT(buffsize > 0);

  if (parser->p == NULL)

  {

    parser->p = buff;

    parser->left = buffsize;

    /* First parser the document header */

    if ((status = NW_WBXML_Parser_docHeaderParse (parser, doc))

      != NW_STAT_SUCCESS)

      return status;

    /* Now get the dictionary from the document */

    if (doc->publicid > 0){

      parser->dictionary =

        NW_WBXML_Dictionary_getIndexByPublicId (doc->publicid);

    }

    else if (doc->doc_type){

      parser->dictionary =

        NW_WBXML_Dictionary_getIndexByDocType (doc->doc_type, doc->charset);

    }

    /* If a dictionary could not be attained try using the default public id */

    if (parser->dictionary == 0){

      doc->publicid = doc->default_public_id;

      parser->dictionary =

        NW_WBXML_Dictionary_getIndexByPublicId (doc->publicid);

    }

    /* Make the StartDocument callback */

    if (parser->handler && parser->handler->StartDocument_CB)

    {

      status = (*(parser->handler->StartDocument_CB)) (parser, doc,

        parser->context);

      if (status != NW_STAT_SUCCESS)

        return status;

    }

  }

  /* Parse the document body */

  return NW_WBXML_Parser_bodyParse (parser);

}