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 void TextCodecUTF16::registerEncodingNames(EncodingNameRegistrar registrar)
  38 {
  39     registrar("UTF-16LE", "UTF-16LE");
  40     registrar("UTF-16BE", "UTF-16BE");
  41
  42     registrar("ISO-10646-UCS-2", "UTF-16LE");
  43     registrar("UCS-2", "UTF-16LE");
  44     registrar("UTF-16", "UTF-16LE");
  45     registrar("Unicode", "UTF-16LE");
  46     registrar("csUnicode", "UTF-16LE");
  47     registrar("unicodeFEFF", "UTF-16LE");
  48
  49     registrar("unicodeFFFE", "UTF-16BE");
  50 }
  51
  52 static auto_ptr<TextCodec> newStreamingTextDecoderUTF16LE(const TextEncoding&, const void*)
  53 {
  54     return auto_ptr<TextCodec>(new TextCodecUTF16(true));
  55 }
  56
  57 static auto_ptr<TextCodec> newStreamingTextDecoderUTF16BE(const TextEncoding&, const void*)
  58 {
  59     return auto_ptr<TextCodec>(new TextCodecUTF16(false));
  60 }
  61
  62 void TextCodecUTF16::registerCodecs(TextCodecRegistrar registrar)
  63 {
  64     registrar("UTF-16LE", newStreamingTextDecoderUTF16LE, 0);
  65     registrar("UTF-16BE", newStreamingTextDecoderUTF16BE, 0);
  66 }
  67
  68 String TextCodecUTF16::decode(const char* bytes, size_t length, bool, bool stopOnError, bool& sawError)
  69 {
  70     if (!length)
  71         return String();
  72
  73     const unsigned char* p = reinterpret_cast<const unsigned char*>(bytes);
  74     size_t numBytes = length + m_haveBufferedByte;
  75     size_t numChars = numBytes / 2;
  76
  77     StringBuffer buffer(numChars);
  78     UChar* q = buffer.characters();
  79
  80     if (m_haveBufferedByte) {
  81         UChar c;
  82         if (m_littleEndian)
  83             c = m_bufferedByte | (p[0] << 8);
  84         else
  85             c = (m_bufferedByte << 8) | p[0];
  86         *q++ = c;
  87         m_haveBufferedByte = false;
  88         p += 1;
  89         numChars -= 1;
  90     }
  91
  92     if (m_littleEndian)
  93         for (size_t i = 0; i < numChars; ++i) {
  94             UChar c = p[0] | (p[1] << 8);
  95             p += 2;
  96             *q++ = c;
  97         }
  98     else
  99         for (size_t i = 0; i < numChars; ++i) {
 100             UChar c = (p[0] << 8) | p[1];
 101             p += 2;
 102             *q++ = c;
 103         }
 104
 105     if (numBytes & 1) {
 106         ASSERT(!m_haveBufferedByte);
 107         m_haveBufferedByte = true;
 108         m_bufferedByte = p[0];
 109     }
 110
 111     buffer.shrink(q - buffer.characters());
 112
 113     return String::adopt(buffer);
 114 }
 115
 116 CString TextCodecUTF16::encode(const UChar* characters, size_t length, UnencodableHandling)
 117 {
 118     char* bytes;
 119     CString string = CString::newUninitialized(length * 2, bytes);
 120
 121     // FIXME: CString is not a reasonable data structure for encoded UTF-16, which will have
 122     // null characters inside it. Perhaps the result of encode should not be a CString?
 123     if (m_littleEndian)
 124         for (size_t i = 0; i < length; ++i) {
 125             UChar c = characters[i];
 126             bytes[i * 2] = c;
 127             bytes[i * 2 + 1] = c >> 8;
 128         }
 129     else
 130         for (size_t i = 0; i < length; ++i) {
 131             UChar c = characters[i];
 132             bytes[i * 2] = c >> 8;
 133             bytes[i * 2 + 1] = c;
 134         }
 135
 136     return string;
 137 }
 138
 139 } // namespace WebCore