[PATCH v6 2/2] xmllite/writer: Handle surrogate pairs

[David Kahurani]

From: David Kahurani

Surrogates have to be written in pairs. Also, handle related errors.

Signed-off-by: David Kahurani 
---
 dlls/xmllite/reader.c          | 10 -----
 dlls/xmllite/tests/writer.c    | 20 +++++++++
 dlls/xmllite/writer.c          | 75 +++++++++++++++++++++++++++++-----
 dlls/xmllite/xmllite_private.h |  9 ++++
 4 files changed, 93 insertions(+), 21 deletions(-)

diff --git a/dlls/xmllite/reader.c b/dlls/xmllite/reader.c
index d747e36e623..f238e2af79d 100644
--- a/dlls/xmllite/reader.c
+++ b/dlls/xmllite/reader.c
@@ -1466,16 +1466,6 @@ static HRESULT reader_parse_comment(xmlreader *reader)
     return S_OK;
 }

-/* [2] Char ::= #x9 | #xA | #xD | [#x20-#xD7FF] | [#xE000-#xFFFD] | [#x10000-#x10FFFF] */
-static inline BOOL is_char(WCHAR ch)
-{
-    return (ch == '\t') || (ch == '\r') || (ch == '\n') ||
-           (ch >= 0x20 && ch <= 0xd7ff) ||
-           (ch >= 0xd800 && ch <= 0xdbff) || /* high surrogate */
-           (ch >= 0xdc00 && ch <= 0xdfff) || /* low surrogate */
-           (ch >= 0xe000 && ch <= 0xfffd);
-}
-
 /* [13] PubidChar ::= #x20 | #xD | #xA | [a-zA-Z0-9] | [-'()+,./:=?;!*#@$_%] */
 BOOL is_pubchar(WCHAR ch)
 {
diff --git a/dlls/xmllite/tests/writer.c b/dlls/xmllite/tests/writer.c
index c4ab079373c..deb8efa98d8 100644
--- a/dlls/xmllite/tests/writer.c
+++ b/dlls/xmllite/tests/writer.c
@@ -1339,6 +1339,7 @@ static void test_WriteRaw(void)
     IXmlWriter *writer;
     IStream *stream;
     HRESULT hr;
+    static const WCHAR surrogates[] = {0xdc00, 0xd800, '\0'};

hr = CreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
     ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
@@ -1351,6 +1352,9 @@ static void test_WriteRaw(void)

stream = writer_set_output(writer);

+    hr = IXmlWriter_WriteRaw(writer, surrogates);
+    ok(hr == WR_E_INVALIDSURROGATEPAIR, "Unexpected hr %#lx.\n", hr);
+
     hr = IXmlWriter_WriteRaw(writer, NULL);
     ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);

@@ -1888,6 +1892,7 @@ static void test_WriteString(void)
     IXmlWriter *writer;
     IStream *stream;
     HRESULT hr;
+    static const WCHAR surrogates[] = {0xd800, 0xdc00, 'x', 'y', '\0'};

hr = CreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
     ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
@@ -1905,6 +1910,21 @@ static void test_WriteString(void)

stream = writer_set_output(writer);

+    hr = IXmlWriter_WriteStartElement(writer, NULL, L"sub", NULL);
+    ok(hr == S_OK, "Unexpected hr #%lx.\n", hr);
+
+    hr = IXmlWriter_WriteString(writer, surrogates);
+    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
+
+    hr = IXmlWriter_Flush(writer);
+    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
+
+    CHECK_OUTPUT(stream,
+        "<sub>\U00010000xy");
+    IStream_Release(stream);
+
+    stream = writer_set_output(writer);
+
     hr = IXmlWriter_WriteStartElement(writer, NULL, L"b", NULL);
     ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);

diff --git a/dlls/xmllite/writer.c b/dlls/xmllite/writer.c
index ec6a05d73a7..7ab2b66aaf9 100644
--- a/dlls/xmllite/writer.c
+++ b/dlls/xmllite/writer.c
@@ -27,6 +27,7 @@
 #include "winbase.h"
 #include "objbase.h"
 #include "xmllite.h"
+#include "winnls.h"
 #include "xmllite_private.h"
 #include "initguid.h"

@@ -1684,9 +1685,43 @@ static HRESULT WINAPI xmlwriter_WriteQualifiedName(IXmlWriter *iface, LPCWSTR pw
     return E_NOTIMPL;
 }

+static HRESULT write_raw_char(xmlwriteroutput *output, LPCWSTR *context,
+		              unsigned *length)
+{
+    const WCHAR *data = *context;
+    HRESULT hr;
+
+    if (!is_char(*data)) return WC_E_XMLCHARACTER;
+
+    if ((*length - 1) && IS_HIGH_SURROGATE(*data))
+    {
+        if (!IS_LOW_SURROGATE(*(data + 1)))
+            return WR_E_INVALIDSURROGATEPAIR;
+
+        hr = write_output_buffer(output, data, 2);
+        (*context) += 2;
+        (*length) -= 2;
+    }
+    else if (IS_LOW_SURROGATE(*data))
+	return WR_E_INVALIDSURROGATEPAIR;
+    else
+    {
+        /* no surrogate or string end */
+        if (IS_HIGH_SURROGATE(*data))
+            return WR_E_INVALIDSURROGATEPAIR;
+
+        hr = write_output_buffer(output, data, 1);
+        (*context)++;
+        (*length)--;
+    }
+
+    return hr;
+}
+
 static HRESULT WINAPI xmlwriter_WriteRaw(IXmlWriter *iface, LPCWSTR data)
 {
     xmlwriter *This = impl_from_IXmlWriter(iface);
+    unsigned length;

TRACE("%p %s\n", This, debugstr_w(data));

@@ -1713,7 +1748,13 @@ static HRESULT WINAPI xmlwriter_WriteRaw(IXmlWriter *iface, LPCWSTR data)
         return WR_E_INVALIDACTION;
     }

-    write_output_buffer(This->output, data, -1);
+    length = lstrlenW(data);
+    while (*data && length)
+    {
+        HRESULT hr = write_raw_char(This->output, &data, &length);
+        if (FAILED(hr)) return hr;
+    }
+
     return S_OK;
 }

@@ -1834,32 +1875,44 @@ static HRESULT WINAPI xmlwriter_WriteStartElement(IXmlWriter *iface, LPCWSTR pre
     return S_OK;
 }

-static void write_escaped_string(xmlwriteroutput *output, const WCHAR *string)
+static HRESULT write_escaped_string(xmlwriteroutput *output, const WCHAR *string, unsigned len)
 {
-    while (*string)
+    HRESULT hr;
+
+    while (*string && len)
     {
         switch (*string)
         {
         case '<':
-            write_output_buffer(output, L"&lt;", 4);
+            hr = write_output_buffer(output, L"&lt;", 4);
+            string++;
+            len--;
             break;
         case '&':
-            write_output_buffer(output, L"&amp;", 5);
+            hr = write_output_buffer(output, L"&amp;", 5);
+            string++;
+            len--;
             break;
         case '>':
-            write_output_buffer(output, L"&gt;", 4);
+            hr = write_output_buffer(output, L"&gt;", 4);
+            string++;
+            len--;
             break;
         default:
-            write_output_buffer(output, string, 1);
+            hr = write_raw_char(output, &string, &len);
         }
-
-        string++;
+        if (FAILED(hr)) return hr;
     }
+
+    if (len) return WC_E_XMLCHARACTER;
+
+    return S_OK;
 }

static HRESULT WINAPI xmlwriter_WriteString(IXmlWriter *iface, const WCHAR *string)
 {
     xmlwriter *This = impl_from_IXmlWriter(iface);
+    unsigned length;

TRACE("%p %s\n", This, debugstr_w(string));

@@ -1884,8 +1937,8 @@ static HRESULT WINAPI xmlwriter_WriteString(IXmlWriter *iface, const WCHAR *stri
     }

This->textnode = 1;
-    write_escaped_string(This->output, string);
-    return S_OK;
+    length = lstrlenW(string);
+    return write_escaped_string(This->output, string, length);
 }

static HRESULT WINAPI xmlwriter_WriteSurrogateCharEntity(IXmlWriter *iface, WCHAR wchLow, WCHAR wchHigh)
diff --git a/dlls/xmllite/xmllite_private.h b/dlls/xmllite/xmllite_private.h
index bd53fca575f..45993d20567 100644
--- a/dlls/xmllite/xmllite_private.h
+++ b/dlls/xmllite/xmllite_private.h
@@ -68,4 +68,13 @@ static inline BOOL is_wchar_space(WCHAR ch)
     return ch == ' ' || ch == '\t' || ch == '\r' || ch == '\n';
 }

+/* [2] Char ::= #x9 | #xA | #xD | [#x20-#xD7FF] | [#xE000-#xFFFD] | [#x10000-#x10FFFF] */
+static inline BOOL is_char(WCHAR ch)
+{
+    return (ch == '\t') || (ch == '\r') || (ch == '\n') ||
+           (ch >= 0x20 && ch <= 0xd7ff) ||
+           (ch >= 0xd800 && ch <= 0xdbff) || /* high surrogate */
+           (ch >= 0xdc00 && ch <= 0xdfff) || /* low surrogate */
+           (ch >= 0xe000 && ch <= 0xfffd);
+}
 #endif /* __XMLLITE_PRIVATE__ */