src/third_party/skia/src/ports/SkXMLPullParser_expat.cpp

   1
   2 /*
   3  * Copyright 2006 The Android Open Source Project
   4  *
   5  * Use of this source code is governed by a BSD-style license that can be
   6  * found in the LICENSE file.
   7  */
   8
   9
  10 #include "SkXMLParser.h"
  11 #include "SkChunkAlloc.h"
  12 #include "SkString.h"
  13 #include "SkStream.h"
  14
  15 #include "expat.h"
  16
  17 static inline char* dupstr(SkChunkAlloc& chunk, const char src[], size_t len)
  18 {
  19     SkASSERT(src);
  20     char*   dst = (char*)chunk.alloc(len + 1, SkChunkAlloc::kThrow_AllocFailType);
  21
  22     memcpy(dst, src, len);
  23     dst[len] = 0;
  24     return dst;
  25 }
  26
  27 static inline int count_pairs(const char** p)
  28 {
  29     const char** start = p;
  30     while (*p)
  31     {
  32         SkASSERT(p[1] != NULL);
  33         p += 2;
  34     }
  35     return (p - start) >> 1;
  36 }
  37
  38 struct Data {
  39     Data() : fAlloc(2048), fState(NORMAL) {}
  40
  41     XML_Parser              fParser;
  42     SkXMLPullParser::Curr*  fCurr;
  43     SkChunkAlloc            fAlloc;
  44
  45     enum State {
  46         NORMAL,
  47         MISSED_START_TAG,
  48         RETURN_END_TAG
  49     };
  50     State fState;
  51     const char* fEndTag;    // if state is RETURN_END_TAG
  52 };
  53
  54 static void XMLCALL start_proc(void *data, const char *el, const char **attr)
  55 {
  56     Data*                   p = (Data*)data;
  57     SkXMLPullParser::Curr*  c = p->fCurr;
  58     SkChunkAlloc&           alloc = p->fAlloc;
  59
  60     c->fName = dupstr(alloc, el, strlen(el));
  61
  62     int n = count_pairs(attr);
  63     SkXMLPullParser::AttrInfo* info = (SkXMLPullParser::AttrInfo*)alloc.alloc(n * sizeof(SkXMLPullParser::AttrInfo),
  64                                                                               SkChunkAlloc::kThrow_AllocFailType);
  65     c->fAttrInfoCount = n;
  66     c->fAttrInfos = info;
  67
  68     for (int i = 0; i < n; i++)
  69     {
  70         info[i].fName = dupstr(alloc, attr[0], strlen(attr[0]));
  71         info[i].fValue = dupstr(alloc, attr[1], strlen(attr[1]));
  72         attr += 2;
  73     }
  74
  75     c->fEventType = SkXMLPullParser::START_TAG;
  76     XML_StopParser(p->fParser, true);
  77 }
  78
  79 static void XMLCALL end_proc(void *data, const char *el)
  80 {
  81     Data*                   p = (Data*)data;
  82     SkXMLPullParser::Curr*  c = p->fCurr;
  83
  84     if (c->fEventType == SkXMLPullParser::START_TAG)
  85     {
  86         /*  if we get here, we were called with a start_tag immediately
  87             followed by this end_tag. The caller will only see the end_tag,
  88             so we set a flag to notify them of the missed start_tag
  89         */
  90         p->fState = Data::MISSED_START_TAG;
  91
  92         SkASSERT(c->fName != NULL);
  93         SkASSERT(strcmp(c->fName, el) == 0);
  94     }
  95     else
  96         c->fName = dupstr(p->fAlloc, el, strlen(el));
  97
  98     c->fEventType = SkXMLPullParser::END_TAG;
  99     XML_StopParser(p->fParser, true);
 100 }
 101
 102 #include <ctype.h>
 103
 104 static bool isws(const char s[])
 105 {
 106     for (; *s; s++)
 107         if (!isspace(*s))
 108             return false;
 109     return true;
 110 }
 111
 112 static void XMLCALL text_proc(void* data, const char* text, int len)
 113 {
 114     Data*                   p = (Data*)data;
 115     SkXMLPullParser::Curr*  c = p->fCurr;
 116
 117     c->fName = dupstr(p->fAlloc, text, len);
 118     c->fIsWhitespace = isws(c->fName);
 119
 120     c->fEventType = SkXMLPullParser::TEXT;
 121     XML_StopParser(p->fParser, true);
 122 }
 123
 124 //////////////////////////////////////////////////////////////////////////
 125
 126 struct SkXMLPullParser::Impl {
 127     Data    fData;
 128     void*   fBuffer;
 129     size_t  fBufferLen;
 130 };
 131
 132 static void reportError(XML_Parser parser)
 133 {
 134     XML_Error code = XML_GetErrorCode(parser);
 135     int lineNumber = XML_GetCurrentLineNumber(parser);
 136     const char* msg = XML_ErrorString(code);
 137
 138     printf("-------- XML error [%d] on line %d, %s\n", code, lineNumber, msg);
 139 }
 140
 141 bool SkXMLPullParser::onInit()
 142 {
 143     fImpl = new Impl;
 144
 145     XML_Parser p = XML_ParserCreate(NULL);
 146     SkASSERT(p);
 147
 148     fImpl->fData.fParser = p;
 149     fImpl->fData.fCurr = &fCurr;
 150
 151     XML_SetElementHandler(p, start_proc, end_proc);
 152     XML_SetCharacterDataHandler(p, text_proc);
 153     XML_SetUserData(p, &fImpl->fData);
 154
 155     size_t len = fStream->getLength();
 156     fImpl->fBufferLen = len;
 157     fImpl->fBuffer = sk_malloc_throw(len);
 158     fStream->rewind();
 159     size_t  len2 = fStream->read(fImpl->fBuffer, len);
 160     return len2 == len;
 161 }
 162
 163 void SkXMLPullParser::onExit()
 164 {
 165     sk_free(fImpl->fBuffer);
 166     XML_ParserFree(fImpl->fData.fParser);
 167     delete fImpl;
 168     fImpl = NULL;
 169 }
 170
 171 SkXMLPullParser::EventType SkXMLPullParser::onNextToken()
 172 {
 173     if (Data::RETURN_END_TAG == fImpl->fData.fState)
 174     {
 175         fImpl->fData.fState = Data::NORMAL;
 176         fCurr.fName = fImpl->fData.fEndTag; // restore name from (below) save
 177         return SkXMLPullParser::END_TAG;
 178     }
 179
 180     fImpl->fData.fAlloc.reset();
 181
 182     XML_Parser p = fImpl->fData.fParser;
 183     XML_Status status;
 184
 185     status = XML_ResumeParser(p);
 186
 187 CHECK_STATUS:
 188     switch (status) {
 189     case XML_STATUS_OK:
 190         return SkXMLPullParser::END_DOCUMENT;
 191
 192     case XML_STATUS_ERROR:
 193         if (XML_GetErrorCode(p) != XML_ERROR_NOT_SUSPENDED)
 194         {
 195             reportError(p);
 196             return SkXMLPullParser::ERROR;
 197         }
 198         status = XML_Parse(p, (const char*)fImpl->fBuffer, fImpl->fBufferLen, true);
 199         goto CHECK_STATUS;
 200
 201     case XML_STATUS_SUSPENDED:
 202         if (Data::MISSED_START_TAG == fImpl->fData.fState)
 203         {
 204             // return a start_tag, and clear the flag so we return end_tag next
 205             SkASSERT(SkXMLPullParser::END_TAG == fCurr.fEventType);
 206             fImpl->fData.fState = Data::RETURN_END_TAG;
 207             fImpl->fData.fEndTag = fCurr.fName;  // save this pointer
 208             return SkXMLPullParser::START_TAG;
 209         }
 210         break;
 211     }
 212     return fCurr.fEventType;
 213 }