xmllite: Support resuming for start tag.

dlls/xmllite/reader.c

@@ -62,13 +62,15 @@ typedef enum
 {
     XmlReadResumeState_Initial,
     XmlReadResumeState_PITarget,
-    XmlReadResumeState_PIBody
+    XmlReadResumeState_PIBody,
+    XmlReadResumeState_STag
 } XmlReaderResumeState;
 
 /* saved pointer index to resume from particular input position */
 typedef enum
 {
-    XmlReadResume_Name, /* PITarget */
+    XmlReadResume_Name,  /* PITarget, name for NCName, prefix for QName */
+    XmlReadResume_Local, /* local for QName */
     XmlReadResume_Body,  /* PI body, comment text */
     XmlReadResume_Last
 } XmlReaderResume;
@@ -1525,44 +1527,101 @@ static HRESULT reader_parse_dtd(xmlreader *reader)
     return S_OK;
 }
 
+/* [11 NS] LocalPart ::= NCName */
+static HRESULT reader_parse_local(xmlreader *reader, strval *local)
+{
+    WCHAR *ptr, *start;
+
+    if (reader->resume[XmlReadResume_Local])
+    {
+        start = reader->resume[XmlReadResume_Local];
+        ptr = reader_get_cur(reader);
+    }
+    else
+    {
+        ptr = start = reader_get_cur(reader);
+    }
+
+    while (is_ncnamechar(*ptr))
+    {
+        reader_skipn(reader, 1);
+        ptr = reader_get_cur(reader);
+    }
+
+    if (is_reader_pending(reader))
+    {
+         reader->resume[XmlReadResume_Local] = start;
+         return E_PENDING;
+    }
+    else
+         reader->resume[XmlReadResume_Local] = NULL;
+
+    local->str = start;
+    local->len = ptr-start;
+
+    return S_OK;
+}
+
 /* [7 NS]  QName ::= PrefixedName | UnprefixedName
    [8 NS]  PrefixedName ::= Prefix ':' LocalPart
    [9 NS]  UnprefixedName ::= LocalPart
-   [10 NS] Prefix ::= NCName
-   [11 NS] LocalPart ::= NCName */
+   [10 NS] Prefix ::= NCName */
 static HRESULT reader_parse_qname(xmlreader *reader, strval *prefix, strval *local, strval *qname)
 {
-    WCHAR *ptr, *start = reader_get_cur(reader);
+    WCHAR *ptr, *start;
+    HRESULT hr;
 
-    ptr = start;
+    if (reader->resume[XmlReadResume_Name])
+    {
+        start = reader->resume[XmlReadResume_Name];
+        ptr = reader_get_cur(reader);
+    }
+    else
+    {
+        ptr = start = reader_get_cur(reader);
+        reader->resume[XmlReadResume_Name] = start;
         if (!is_ncnamechar(*ptr)) return NC_E_QNAMECHARACTER;
+    }
+
+    if (reader->resume[XmlReadResume_Local])
+    {
+        hr = reader_parse_local(reader, local);
+        if (FAILED(hr)) return hr;
 
+        prefix->str = reader->resume[XmlReadResume_Name];
+        prefix->len = local->str - prefix->str - 1;
+    }
+    else
+    {
+        /* skip prefix part */
         while (is_ncnamechar(*ptr))
         {
             reader_skipn(reader, 1);
             ptr = reader_get_cur(reader);
         }
 
+        if (is_reader_pending(reader)) return E_PENDING;
+
         /* got a qualified name */
         if (*ptr == ':')
         {
             prefix->str = start;
             prefix->len = ptr-start;
 
+            /* skip ':' */
             reader_skipn(reader, 1);
-        start = ptr = reader_get_cur(reader);
-
-        while (is_ncnamechar(*ptr))
-        {
-            reader_skipn(reader, 1);
-            ptr = reader_get_cur(reader);
-        }
+            hr = reader_parse_local(reader, local);
+            if (FAILED(hr)) return hr;
         }
         else
         {
+            local->str = reader->resume[XmlReadResume_Name];
+            local->len = ptr-local->str;
+
             prefix->str = NULL;
             prefix->len = 0;
         }
+    }
 
     local->str = start;
     local->len = ptr-start;
@@ -1576,6 +1635,9 @@ static HRESULT reader_parse_qname(xmlreader *reader, strval *prefix, strval *loc
     /* count ':' too */
     qname->len = (prefix->len ? prefix->len + 1 : 0) + local->len;
 
+    reader->resume[XmlReadResume_Name] = NULL;
+    reader->resume[XmlReadResume_Local] = NULL;
+
     return S_OK;
 }
 
@@ -1586,9 +1648,6 @@ static HRESULT reader_parse_stag(xmlreader *reader, strval *prefix, strval *loca
     static const WCHAR endW[] = {'/','>',0};
     HRESULT hr;
 
-    /* skip '<' */
-    reader_skipn(reader, 1);
-
     hr = reader_parse_qname(reader, prefix, local, qname);
     if (FAILED(hr)) return hr;
 
@@ -1617,16 +1676,25 @@ static HRESULT reader_parse_stag(xmlreader *reader, strval *prefix, strval *loca
 /* [39] element ::= EmptyElemTag | STag content ETag */
 static HRESULT reader_parse_element(xmlreader *reader)
 {
-    strval qname, prefix, local;
     HRESULT hr;
-    int empty;
 
+    switch (reader->resumestate)
+    {
+    case XmlReadResumeState_Initial:
         /* check if we are really on element */
         if (reader_cmp(reader, ltW)) return S_FALSE;
+
+        /* skip '<' */
+        reader_skipn(reader, 1);
+
         reader_shrink(reader);
+        reader->resumestate = XmlReadResumeState_STag;
+    case XmlReadResumeState_STag:
+    {
+        strval qname, prefix, local;
+        int empty = 0;
 
         /* this handles empty elements too */
-    empty = 0;
         hr = reader_parse_stag(reader, &prefix, &local, &qname, &empty);
         if (FAILED(hr)) return hr;
 
@@ -1641,8 +1709,14 @@ static HRESULT reader_parse_element(xmlreader *reader)
             reader->instate = XmlReadInState_Content;
 
         reader->nodetype = XmlNodeType_Element;
+        reader->resumestate = XmlReadResumeState_Initial;
         reader_set_strvalue(reader, StringValue_LocalName, &local);
         reader_set_strvalue(reader, StringValue_QualifiedName, &qname);
+        break;
+    }
+    default:
+        hr = E_FAIL;
+    }
 
     return hr;
 }

dlls/xmllite/tests/reader.c

@@ -994,7 +994,7 @@ static void test_read_full(void)
         hr = IXmlReader_Read(reader, &type);
         i++;
     }
-    ok(test->types[i] == XmlNodeType_None, "incomplete sequence\n");
+    ok(test->types[i] == XmlNodeType_None, "incomplete sequence, got %d\n", test->types[i]);
 
     IStream_Release(stream);
     IXmlReader_Release(reader);