1148bce560c1a7fe425eec8238ed667e20ae9f87
[WebKit-https.git] / WebCore / platform / text / TextCodecUTF16.cpp
1 /*
2  * Copyright (C) 2004, 2006, 2008 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #include "config.h"
27 #include "TextCodecUTF16.h"
28
29 #include "CString.h"
30 #include "PlatformString.h"
31 #include "StringBuffer.h"
32
33 using std::auto_ptr;
34
35 namespace WebCore {
36
37 const UChar BOM = 0xFEFF;
38
39 void TextCodecUTF16::registerEncodingNames(EncodingNameRegistrar registrar)
40 {
41     registrar("UTF-16LE", "UTF-16LE");
42     registrar("UTF-16BE", "UTF-16BE");
43
44     registrar("ISO-10646-UCS-2", "UTF-16LE");
45     registrar("UCS-2", "UTF-16LE");
46     registrar("UTF-16", "UTF-16LE");
47     registrar("Unicode", "UTF-16LE");
48     registrar("csUnicode", "UTF-16LE");
49     registrar("unicodeFEFF", "UTF-16LE");
50
51     registrar("unicodeFFFE", "UTF-16BE");
52 }
53
54 static auto_ptr<TextCodec> newStreamingTextDecoderUTF16LE(const TextEncoding&, const void*)
55 {
56     return auto_ptr<TextCodec>(new TextCodecUTF16(true));
57 }
58
59 static auto_ptr<TextCodec> newStreamingTextDecoderUTF16BE(const TextEncoding&, const void*)
60 {
61     return auto_ptr<TextCodec>(new TextCodecUTF16(false));
62 }
63
64 void TextCodecUTF16::registerCodecs(TextCodecRegistrar registrar)
65 {
66     registrar("UTF-16LE", newStreamingTextDecoderUTF16LE, 0);
67     registrar("UTF-16BE", newStreamingTextDecoderUTF16BE, 0);
68 }
69
70 String TextCodecUTF16::decode(const char* bytes, size_t length, bool, bool stopOnError, bool& sawError)
71 {
72     if (!length)
73         return String();
74
75     const unsigned char* p = reinterpret_cast<const unsigned char*>(bytes);
76     size_t numBytes = length + m_haveBufferedByte;
77     size_t numChars = numBytes / 2;
78
79     StringBuffer buffer(numChars);
80     UChar* q = buffer.characters();
81
82     if (m_haveBufferedByte) {
83         UChar c;
84         if (m_littleEndian)
85             c = m_bufferedByte | (p[0] << 8);
86         else
87             c = (m_bufferedByte << 8) | p[0];
88         if (c != BOM)
89             *q++ = c;
90         m_haveBufferedByte = false;
91         p += 1;
92         numChars -= 1;
93     }
94
95     if (m_littleEndian)
96         for (size_t i = 0; i < numChars; ++i) {
97             UChar c = p[0] | (p[1] << 8);
98             p += 2;
99             if (c != BOM)
100                 *q++ = c;
101         }
102     else
103         for (size_t i = 0; i < numChars; ++i) {
104             UChar c = (p[0] << 8) | p[1];
105             p += 2;
106             if (c != BOM)
107                 *q++ = c;
108         }
109
110     if (numBytes & 1) {
111         ASSERT(!m_haveBufferedByte);
112         m_haveBufferedByte = true;
113         m_bufferedByte = p[0];
114     }
115
116     buffer.shrink(q - buffer.characters());
117
118     return String::adopt(buffer);
119 }
120
121 CString TextCodecUTF16::encode(const UChar* characters, size_t length, UnencodableHandling)
122 {
123     char* bytes;
124     CString string = CString::newUninitialized(length * 2, bytes);
125
126     // FIXME: CString is not a reasonable data structure for encoded UTF-16, which will have
127     // null characters inside it. Perhaps the result of encode should not be a CString?
128     if (m_littleEndian)
129         for (size_t i = 0; i < length; ++i) {
130             UChar c = characters[i];
131             bytes[i * 2] = c;
132             bytes[i * 2 + 1] = c >> 8;
133         }
134     else
135         for (size_t i = 0; i < length; ++i) {
136             UChar c = characters[i];
137             bytes[i * 2] = c >> 8;
138             bytes[i * 2 + 1] = c;
139         }
140
141     return string;
142 }
143
144 } // namespace WebCore