| /* |
| * Copyright (C) 2004, 2006, 2007, 2008, 2011 Apple Inc. All rights reserved. |
| * Copyright (C) 2006 Alexey Proskuryakov <ap@nypop.com> |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY |
| * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
| * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR |
| * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
| * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
| * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
| * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY |
| * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "wtf/text/TextCodecICU.h" |
| |
| #include "wtf/Assertions.h" |
| #include "wtf/PtrUtil.h" |
| #include "wtf/StringExtras.h" |
| #include "wtf/Threading.h" |
| #include "wtf/WTFThreadData.h" |
| #include "wtf/text/CString.h" |
| #include "wtf/text/CharacterNames.h" |
| #include "wtf/text/StringBuilder.h" |
| #include <memory> |
| #include <unicode/ucnv.h> |
| #include <unicode/ucnv_cb.h> |
| |
| namespace WTF { |
| |
| const size_t ConversionBufferSize = 16384; |
| |
| ICUConverterWrapper::~ICUConverterWrapper() { |
| if (converter) |
| ucnv_close(converter); |
| } |
| |
| static UConverter*& cachedConverterICU() { |
| return wtfThreadData().cachedConverterICU().converter; |
| } |
| |
| std::unique_ptr<TextCodec> TextCodecICU::create(const TextEncoding& encoding, |
| const void*) { |
| return wrapUnique(new TextCodecICU(encoding)); |
| } |
| |
| void TextCodecICU::registerEncodingNames(EncodingNameRegistrar registrar) { |
| // We register Hebrew with logical ordering using a separate name. |
| // Otherwise, this would share the same canonical name as the |
| // visual ordering case, and then TextEncoding could not tell them |
| // apart; ICU treats these names as synonyms. |
| registrar("ISO-8859-8-I", "ISO-8859-8-I"); |
| |
| int32_t numEncodings = ucnv_countAvailable(); |
| for (int32_t i = 0; i < numEncodings; ++i) { |
| const char* name = ucnv_getAvailableName(i); |
| UErrorCode error = U_ZERO_ERROR; |
| #if !defined(USING_SYSTEM_ICU) |
| const char* primaryStandard = "HTML"; |
| const char* secondaryStandard = "MIME"; |
| #else |
| const char* primaryStandard = "MIME"; |
| const char* secondaryStandard = "IANA"; |
| #endif |
| const char* standardName = |
| ucnv_getStandardName(name, primaryStandard, &error); |
| if (U_FAILURE(error) || !standardName) { |
| error = U_ZERO_ERROR; |
| // Try IANA to pick up 'windows-12xx' and other names |
| // which are not preferred MIME names but are widely used. |
| standardName = ucnv_getStandardName(name, secondaryStandard, &error); |
| if (U_FAILURE(error) || !standardName) |
| continue; |
| } |
| |
| // A number of these aliases are handled in Chrome's copy of ICU, but |
| // Chromium can be compiled with the system ICU. |
| |
| // 1. Treat GB2312 encoding as GBK (its more modern superset), to match other |
| // browsers. |
| // 2. On the Web, GB2312 is encoded as EUC-CN or HZ, while ICU provides a native |
| // encoding for encoding GB_2312-80 and several others. So, we need to |
| // override this behavior, too. |
| #if defined(USING_SYSTEM_ICU) |
| if (!strcmp(standardName, "GB2312") || !strcmp(standardName, "GB_2312-80")) |
| standardName = "GBK"; |
| // Similarly, EUC-KR encodings all map to an extended version, but |
| // per HTML5, the canonical name still should be EUC-KR. |
| else if (!strcmp(standardName, "EUC-KR") || |
| !strcmp(standardName, "KSC_5601") || |
| !strcmp(standardName, "cp1363")) |
| standardName = "EUC-KR"; |
| // And so on. |
| else if (!strcasecmp(standardName, "iso-8859-9")) |
| // This name is returned in different case by ICU 3.2 and 3.6. |
| standardName = "windows-1254"; |
| else if (!strcmp(standardName, "TIS-620")) |
| standardName = "windows-874"; |
| #endif |
| |
| registrar(standardName, standardName); |
| |
| uint16_t numAliases = ucnv_countAliases(name, &error); |
| ASSERT(U_SUCCESS(error)); |
| if (U_SUCCESS(error)) |
| for (uint16_t j = 0; j < numAliases; ++j) { |
| error = U_ZERO_ERROR; |
| const char* alias = ucnv_getAlias(name, j, &error); |
| ASSERT(U_SUCCESS(error)); |
| if (U_SUCCESS(error) && alias != standardName) |
| registrar(alias, standardName); |
| } |
| } |
| |
| // These two entries have to be added here because ICU's converter table |
| // cannot have both ISO-8859-8-I and ISO-8859-8. |
| registrar("csISO88598I", "ISO-8859-8-I"); |
| registrar("logical", "ISO-8859-8-I"); |
| |
| #if defined(USING_SYSTEM_ICU) |
| // Additional alias for MacCyrillic not present in ICU. |
| registrar("maccyrillic", "x-mac-cyrillic"); |
| |
| // Additional aliases that historically were present in the encoding |
| // table in WebKit on Macintosh that don't seem to be present in ICU. |
| // Perhaps we can prove these are not used on the web and remove them. |
| // Or perhaps we can get them added to ICU. |
| registrar("x-mac-roman", "macintosh"); |
| registrar("x-mac-ukrainian", "x-mac-cyrillic"); |
| registrar("cn-big5", "Big5"); |
| registrar("x-x-big5", "Big5"); |
| registrar("cn-gb", "GBK"); |
| registrar("csgb231280", "GBK"); |
| registrar("x-euc-cn", "GBK"); |
| registrar("x-gbk", "GBK"); |
| registrar("koi", "KOI8-R"); |
| registrar("visual", "ISO-8859-8"); |
| registrar("winarabic", "windows-1256"); |
| registrar("winbaltic", "windows-1257"); |
| registrar("wincyrillic", "windows-1251"); |
| registrar("iso-8859-11", "windows-874"); |
| registrar("iso8859-11", "windows-874"); |
| registrar("dos-874", "windows-874"); |
| registrar("wingreek", "windows-1253"); |
| registrar("winhebrew", "windows-1255"); |
| registrar("winlatin2", "windows-1250"); |
| registrar("winturkish", "windows-1254"); |
| registrar("winvietnamese", "windows-1258"); |
| registrar("x-cp1250", "windows-1250"); |
| registrar("x-cp1251", "windows-1251"); |
| registrar("x-euc", "EUC-JP"); |
| registrar("x-windows-949", "EUC-KR"); |
| registrar("KSC5601", "EUC-KR"); |
| registrar("x-uhc", "EUC-KR"); |
| registrar("shift-jis", "Shift_JIS"); |
| |
| // Alternative spelling of ISO encoding names. |
| registrar("ISO8859-1", "ISO-8859-1"); |
| registrar("ISO8859-2", "ISO-8859-2"); |
| registrar("ISO8859-3", "ISO-8859-3"); |
| registrar("ISO8859-4", "ISO-8859-4"); |
| registrar("ISO8859-5", "ISO-8859-5"); |
| registrar("ISO8859-6", "ISO-8859-6"); |
| registrar("ISO8859-7", "ISO-8859-7"); |
| registrar("ISO8859-8", "ISO-8859-8"); |
| registrar("ISO8859-8-I", "ISO-8859-8-I"); |
| registrar("ISO8859-9", "ISO-8859-9"); |
| registrar("ISO8859-10", "ISO-8859-10"); |
| registrar("ISO8859-13", "ISO-8859-13"); |
| registrar("ISO8859-14", "ISO-8859-14"); |
| registrar("ISO8859-15", "ISO-8859-15"); |
| // No need to have an entry for ISO8859-16. ISO-8859-16 has just one label |
| // listed in WHATWG Encoding Living Standard, http://encoding.spec.whatwg.org/ |
| |
| // Additional aliases present in the WHATWG Encoding Standard |
| // and Firefox (as of Oct 2014), but not in the upstream ICU. |
| // Three entries for windows-1252 need not be listed here because |
| // TextCodecLatin1 registers them. |
| registrar("csiso58gb231280", "GBK"); |
| registrar("csiso88596e", "ISO-8859-6"); |
| registrar("csiso88596i", "ISO-8859-6"); |
| registrar("csiso88598e", "ISO-8859-8"); |
| registrar("gb_2312", "GBK"); |
| registrar("iso88592", "ISO-8859-2"); |
| registrar("iso88593", "ISO-8859-3"); |
| registrar("iso88594", "ISO-8859-4"); |
| registrar("iso88595", "ISO-8859-5"); |
| registrar("iso88596", "ISO-8859-6"); |
| registrar("iso88597", "ISO-8859-7"); |
| registrar("iso88598", "ISO-8859-8"); |
| registrar("iso88599", "windows-1254"); |
| registrar("iso885910", "ISO-8859-10"); |
| registrar("iso885911", "windows-874"); |
| registrar("iso885913", "ISO-8859-13"); |
| registrar("iso885914", "ISO-8859-14"); |
| registrar("iso885915", "ISO-8859-15"); |
| registrar("iso_8859-2", "ISO-8859-2"); |
| registrar("iso_8859-3", "ISO-8859-3"); |
| registrar("iso_8859-4", "ISO-8859-4"); |
| registrar("iso_8859-5", "ISO-8859-5"); |
| registrar("iso_8859-6", "ISO-8859-6"); |
| registrar("iso_8859-7", "ISO-8859-7"); |
| registrar("iso_8859-8", "ISO-8859-8"); |
| registrar("iso_8859-9", "windows-1254"); |
| registrar("iso_8859-15", "ISO-8859-15"); |
| registrar("koi8_r", "KOI8-R"); |
| registrar("x-cp1253", "windows-1253"); |
| registrar("x-cp1254", "windows-1254"); |
| registrar("x-cp1255", "windows-1255"); |
| registrar("x-cp1256", "windows-1256"); |
| registrar("x-cp1257", "windows-1257"); |
| registrar("x-cp1258", "windows-1258"); |
| #endif |
| } |
| |
| void TextCodecICU::registerCodecs(TextCodecRegistrar registrar) { |
| // See comment above in registerEncodingNames. |
| registrar("ISO-8859-8-I", create, 0); |
| |
| int32_t numEncodings = ucnv_countAvailable(); |
| for (int32_t i = 0; i < numEncodings; ++i) { |
| const char* name = ucnv_getAvailableName(i); |
| UErrorCode error = U_ZERO_ERROR; |
| const char* standardName = ucnv_getStandardName(name, "MIME", &error); |
| if (!U_SUCCESS(error) || !standardName) { |
| error = U_ZERO_ERROR; |
| standardName = ucnv_getStandardName(name, "IANA", &error); |
| if (!U_SUCCESS(error) || !standardName) |
| continue; |
| } |
| registrar(standardName, create, 0); |
| } |
| } |
| |
| TextCodecICU::TextCodecICU(const TextEncoding& encoding) |
| : m_encoding(encoding), |
| m_converterICU(0) |
| #if defined(USING_SYSTEM_ICU) |
| , |
| m_needsGBKFallbacks(false) |
| #endif |
| { |
| } |
| |
| TextCodecICU::~TextCodecICU() { |
| releaseICUConverter(); |
| } |
| |
| void TextCodecICU::releaseICUConverter() const { |
| if (m_converterICU) { |
| UConverter*& cachedConverter = cachedConverterICU(); |
| if (cachedConverter) |
| ucnv_close(cachedConverter); |
| cachedConverter = m_converterICU; |
| m_converterICU = 0; |
| } |
| } |
| |
| void TextCodecICU::createICUConverter() const { |
| ASSERT(!m_converterICU); |
| |
| #if defined(USING_SYSTEM_ICU) |
| const char* name = m_encoding.name(); |
| m_needsGBKFallbacks = |
| name[0] == 'G' && name[1] == 'B' && name[2] == 'K' && !name[3]; |
| #endif |
| |
| UErrorCode err; |
| |
| UConverter*& cachedConverter = cachedConverterICU(); |
| if (cachedConverter) { |
| err = U_ZERO_ERROR; |
| const char* cachedName = ucnv_getName(cachedConverter, &err); |
| if (U_SUCCESS(err) && m_encoding == cachedName) { |
| m_converterICU = cachedConverter; |
| cachedConverter = 0; |
| return; |
| } |
| } |
| |
| err = U_ZERO_ERROR; |
| m_converterICU = ucnv_open(m_encoding.name(), &err); |
| DLOG_IF(ERROR, err == U_AMBIGUOUS_ALIAS_WARNING) |
| << "ICU ambiguous alias warning for encoding: " << m_encoding.name(); |
| if (m_converterICU) |
| ucnv_setFallback(m_converterICU, TRUE); |
| } |
| |
| int TextCodecICU::decodeToBuffer(UChar* target, |
| UChar* targetLimit, |
| const char*& source, |
| const char* sourceLimit, |
| int32_t* offsets, |
| bool flush, |
| UErrorCode& err) { |
| UChar* targetStart = target; |
| err = U_ZERO_ERROR; |
| ucnv_toUnicode(m_converterICU, &target, targetLimit, &source, sourceLimit, |
| offsets, flush, &err); |
| return target - targetStart; |
| } |
| |
| class ErrorCallbackSetter final { |
| STACK_ALLOCATED(); |
| |
| public: |
| ErrorCallbackSetter(UConverter* converter, bool stopOnError) |
| : m_converter(converter), m_shouldStopOnEncodingErrors(stopOnError) { |
| if (m_shouldStopOnEncodingErrors) { |
| UErrorCode err = U_ZERO_ERROR; |
| ucnv_setToUCallBack(m_converter, UCNV_TO_U_CALLBACK_STOP, 0, |
| &m_savedAction, &m_savedContext, &err); |
| DCHECK_EQ(err, U_ZERO_ERROR); |
| } |
| } |
| ~ErrorCallbackSetter() { |
| if (m_shouldStopOnEncodingErrors) { |
| UErrorCode err = U_ZERO_ERROR; |
| const void* oldContext; |
| UConverterToUCallback oldAction; |
| ucnv_setToUCallBack(m_converter, m_savedAction, m_savedContext, |
| &oldAction, &oldContext, &err); |
| DCHECK_EQ(oldAction, UCNV_TO_U_CALLBACK_STOP); |
| DCHECK(!oldContext); |
| DCHECK_EQ(err, U_ZERO_ERROR); |
| } |
| } |
| |
| private: |
| UConverter* m_converter; |
| bool m_shouldStopOnEncodingErrors; |
| const void* m_savedContext; |
| UConverterToUCallback m_savedAction; |
| }; |
| |
| String TextCodecICU::decode(const char* bytes, |
| size_t length, |
| FlushBehavior flush, |
| bool stopOnError, |
| bool& sawError) { |
| // Get a converter for the passed-in encoding. |
| if (!m_converterICU) { |
| createICUConverter(); |
| ASSERT(m_converterICU); |
| if (!m_converterICU) { |
| DLOG(ERROR) |
| << "error creating ICU encoder even though encoding was in table"; |
| return String(); |
| } |
| } |
| |
| ErrorCallbackSetter callbackSetter(m_converterICU, stopOnError); |
| |
| StringBuilder result; |
| |
| UChar buffer[ConversionBufferSize]; |
| UChar* bufferLimit = buffer + ConversionBufferSize; |
| const char* source = reinterpret_cast<const char*>(bytes); |
| const char* sourceLimit = source + length; |
| int32_t* offsets = nullptr; |
| UErrorCode err = U_ZERO_ERROR; |
| |
| do { |
| int ucharsDecoded = decodeToBuffer(buffer, bufferLimit, source, sourceLimit, |
| offsets, flush != DoNotFlush, err); |
| result.append(buffer, ucharsDecoded); |
| } while (err == U_BUFFER_OVERFLOW_ERROR); |
| |
| if (U_FAILURE(err)) { |
| // flush the converter so it can be reused, and not be bothered by this |
| // error. |
| do { |
| decodeToBuffer(buffer, bufferLimit, source, sourceLimit, offsets, true, |
| err); |
| } while (source < sourceLimit); |
| sawError = true; |
| } |
| |
| #if !defined(USING_SYSTEM_ICU) |
| // Chrome's copy of ICU does not have the issue described below. |
| return result.toString(); |
| #else |
| String resultString = result.toString(); |
| |
| // <http://bugs.webkit.org/show_bug.cgi?id=17014> |
| // Simplified Chinese pages use the code A3A0 to mean "full-width space", but |
| // ICU decodes it as U+E5E5. |
| if (!strcmp(m_encoding.name(), "GBK")) { |
| if (!strcasecmp(m_encoding.name(), "gb18030")) |
| resultString.replace(0xE5E5, ideographicSpaceCharacter); |
| // Make GBK compliant to the encoding spec and align with GB18030 |
| resultString.replace(0x01F9, 0xE7C8); |
| // FIXME: Once https://www.w3.org/Bugs/Public/show_bug.cgi?id=28740#c3 |
| // is resolved, add U+1E3F => 0xE7C7. |
| } |
| |
| return resultString; |
| #endif |
| } |
| |
| #if defined(USING_SYSTEM_ICU) |
| // U+01F9 and U+1E3F have to be mapped to xA8xBF and xA8xBC per the encoding |
| // spec, but ICU converter does not have them. |
| static UChar fallbackForGBK(UChar32 character) { |
| switch (character) { |
| case 0x01F9: |
| return 0xE7C8; // mapped to xA8xBF by ICU. |
| case 0x1E3F: |
| return 0xE7C7; // mapped to xA8xBC by ICU. |
| } |
| return 0; |
| } |
| #endif |
| |
| // Generic helper for writing escaped entities using the specfied |
| // UnencodableHandling. |
| static void formatEscapedEntityCallback(const void* context, |
| UConverterFromUnicodeArgs* fromUArgs, |
| const UChar* codeUnits, |
| int32_t length, |
| UChar32 codePoint, |
| UConverterCallbackReason reason, |
| UErrorCode* err, |
| UnencodableHandling handling) { |
| if (reason == UCNV_UNASSIGNED) { |
| *err = U_ZERO_ERROR; |
| |
| UnencodableReplacementArray entity; |
| int entityLen = |
| TextCodec::getUnencodableReplacement(codePoint, handling, entity); |
| ucnv_cbFromUWriteBytes(fromUArgs, entity, entityLen, 0, err); |
| } else { |
| UCNV_FROM_U_CALLBACK_ESCAPE(context, fromUArgs, codeUnits, length, |
| codePoint, reason, err); |
| } |
| } |
| |
| static void numericEntityCallback(const void* context, |
| UConverterFromUnicodeArgs* fromUArgs, |
| const UChar* codeUnits, |
| int32_t length, |
| UChar32 codePoint, |
| UConverterCallbackReason reason, |
| UErrorCode* err) { |
| formatEscapedEntityCallback(context, fromUArgs, codeUnits, length, codePoint, |
| reason, err, EntitiesForUnencodables); |
| } |
| |
| // Invalid character handler when writing escaped entities in CSS encoding for |
| // unrepresentable characters. See the declaration of TextCodec::encode for |
| // more. |
| static void cssEscapedEntityCallback(const void* context, |
| UConverterFromUnicodeArgs* fromUArgs, |
| const UChar* codeUnits, |
| int32_t length, |
| UChar32 codePoint, |
| UConverterCallbackReason reason, |
| UErrorCode* err) { |
| formatEscapedEntityCallback(context, fromUArgs, codeUnits, length, codePoint, |
| reason, err, CSSEncodedEntitiesForUnencodables); |
| } |
| |
| // Invalid character handler when writing escaped entities in HTML/XML encoding |
| // for unrepresentable characters. See the declaration of TextCodec::encode for |
| // more. |
| static void urlEscapedEntityCallback(const void* context, |
| UConverterFromUnicodeArgs* fromUArgs, |
| const UChar* codeUnits, |
| int32_t length, |
| UChar32 codePoint, |
| UConverterCallbackReason reason, |
| UErrorCode* err) { |
| formatEscapedEntityCallback(context, fromUArgs, codeUnits, length, codePoint, |
| reason, err, URLEncodedEntitiesForUnencodables); |
| } |
| |
| #if defined(USING_SYSTEM_ICU) |
| // Substitutes special GBK characters, escaping all other unassigned entities. |
| static void gbkCallbackEscape(const void* context, |
| UConverterFromUnicodeArgs* fromUArgs, |
| const UChar* codeUnits, |
| int32_t length, |
| UChar32 codePoint, |
| UConverterCallbackReason reason, |
| UErrorCode* err) { |
| UChar outChar; |
| if (reason == UCNV_UNASSIGNED && (outChar = fallbackForGBK(codePoint))) { |
| const UChar* source = &outChar; |
| *err = U_ZERO_ERROR; |
| ucnv_cbFromUWriteUChars(fromUArgs, &source, source + 1, 0, err); |
| return; |
| } |
| numericEntityCallback(context, fromUArgs, codeUnits, length, codePoint, |
| reason, err); |
| } |
| |
| // Combines both gbkCssEscapedEntityCallback and GBK character substitution. |
| static void gbkCssEscapedEntityCallack(const void* context, |
| UConverterFromUnicodeArgs* fromUArgs, |
| const UChar* codeUnits, |
| int32_t length, |
| UChar32 codePoint, |
| UConverterCallbackReason reason, |
| UErrorCode* err) { |
| if (reason == UCNV_UNASSIGNED) { |
| if (UChar outChar = fallbackForGBK(codePoint)) { |
| const UChar* source = &outChar; |
| *err = U_ZERO_ERROR; |
| ucnv_cbFromUWriteUChars(fromUArgs, &source, source + 1, 0, err); |
| return; |
| } |
| cssEscapedEntityCallback(context, fromUArgs, codeUnits, length, codePoint, |
| reason, err); |
| return; |
| } |
| UCNV_FROM_U_CALLBACK_ESCAPE(context, fromUArgs, codeUnits, length, codePoint, |
| reason, err); |
| } |
| |
| // Combines both gbkUrlEscapedEntityCallback and GBK character substitution. |
| static void gbkUrlEscapedEntityCallack(const void* context, |
| UConverterFromUnicodeArgs* fromUArgs, |
| const UChar* codeUnits, |
| int32_t length, |
| UChar32 codePoint, |
| UConverterCallbackReason reason, |
| UErrorCode* err) { |
| if (reason == UCNV_UNASSIGNED) { |
| if (UChar outChar = fallbackForGBK(codePoint)) { |
| const UChar* source = &outChar; |
| *err = U_ZERO_ERROR; |
| ucnv_cbFromUWriteUChars(fromUArgs, &source, source + 1, 0, err); |
| return; |
| } |
| urlEscapedEntityCallback(context, fromUArgs, codeUnits, length, codePoint, |
| reason, err); |
| return; |
| } |
| UCNV_FROM_U_CALLBACK_ESCAPE(context, fromUArgs, codeUnits, length, codePoint, |
| reason, err); |
| } |
| |
| static void gbkCallbackSubstitute(const void* context, |
| UConverterFromUnicodeArgs* fromUArgs, |
| const UChar* codeUnits, |
| int32_t length, |
| UChar32 codePoint, |
| UConverterCallbackReason reason, |
| UErrorCode* err) { |
| UChar outChar; |
| if (reason == UCNV_UNASSIGNED && (outChar = fallbackForGBK(codePoint))) { |
| const UChar* source = &outChar; |
| *err = U_ZERO_ERROR; |
| ucnv_cbFromUWriteUChars(fromUArgs, &source, source + 1, 0, err); |
| return; |
| } |
| UCNV_FROM_U_CALLBACK_SUBSTITUTE(context, fromUArgs, codeUnits, length, |
| codePoint, reason, err); |
| } |
| #endif // USING_SYSTEM_ICU |
| |
| class TextCodecInput final { |
| STACK_ALLOCATED(); |
| |
| public: |
| TextCodecInput(const TextEncoding& encoding, |
| const UChar* characters, |
| size_t length) |
| : m_begin(characters), m_end(characters + length) {} |
| |
| TextCodecInput(const TextEncoding& encoding, |
| const LChar* characters, |
| size_t length) { |
| m_buffer.reserveInitialCapacity(length); |
| for (size_t i = 0; i < length; ++i) |
| m_buffer.append(characters[i]); |
| m_begin = m_buffer.data(); |
| m_end = m_begin + m_buffer.size(); |
| } |
| |
| const UChar* begin() const { return m_begin; } |
| const UChar* end() const { return m_end; } |
| |
| private: |
| const UChar* m_begin; |
| const UChar* m_end; |
| Vector<UChar> m_buffer; |
| }; |
| |
| CString TextCodecICU::encodeInternal(const TextCodecInput& input, |
| UnencodableHandling handling) { |
| const UChar* source = input.begin(); |
| const UChar* end = input.end(); |
| |
| UErrorCode err = U_ZERO_ERROR; |
| |
| switch (handling) { |
| case QuestionMarksForUnencodables: |
| ucnv_setSubstChars(m_converterICU, "?", 1, &err); |
| #if !defined(USING_SYSTEM_ICU) |
| ucnv_setFromUCallBack(m_converterICU, UCNV_FROM_U_CALLBACK_SUBSTITUTE, 0, |
| 0, 0, &err); |
| #else |
| ucnv_setFromUCallBack( |
| m_converterICU, m_needsGBKFallbacks ? gbkCallbackSubstitute |
| : UCNV_FROM_U_CALLBACK_SUBSTITUTE, |
| 0, 0, 0, &err); |
| #endif |
| break; |
| case EntitiesForUnencodables: |
| #if !defined(USING_SYSTEM_ICU) |
| ucnv_setFromUCallBack(m_converterICU, numericEntityCallback, 0, 0, 0, |
| &err); |
| #else |
| ucnv_setFromUCallBack( |
| m_converterICU, |
| m_needsGBKFallbacks ? gbkCallbackEscape : numericEntityCallback, 0, 0, |
| 0, &err); |
| #endif |
| break; |
| case URLEncodedEntitiesForUnencodables: |
| #if !defined(USING_SYSTEM_ICU) |
| ucnv_setFromUCallBack(m_converterICU, urlEscapedEntityCallback, 0, 0, 0, |
| &err); |
| #else |
| ucnv_setFromUCallBack(m_converterICU, |
| m_needsGBKFallbacks ? gbkUrlEscapedEntityCallack |
| : urlEscapedEntityCallback, |
| 0, 0, 0, &err); |
| #endif |
| break; |
| case CSSEncodedEntitiesForUnencodables: |
| #if !defined(USING_SYSTEM_ICU) |
| ucnv_setFromUCallBack(m_converterICU, cssEscapedEntityCallback, 0, 0, 0, |
| &err); |
| #else |
| ucnv_setFromUCallBack(m_converterICU, |
| m_needsGBKFallbacks ? gbkCssEscapedEntityCallack |
| : cssEscapedEntityCallback, |
| 0, 0, 0, &err); |
| #endif |
| break; |
| } |
| |
| ASSERT(U_SUCCESS(err)); |
| if (U_FAILURE(err)) |
| return CString(); |
| |
| Vector<char> result; |
| size_t size = 0; |
| do { |
| char buffer[ConversionBufferSize]; |
| char* target = buffer; |
| char* targetLimit = target + ConversionBufferSize; |
| err = U_ZERO_ERROR; |
| ucnv_fromUnicode(m_converterICU, &target, targetLimit, &source, end, 0, |
| true, &err); |
| size_t count = target - buffer; |
| result.grow(size + count); |
| memcpy(result.data() + size, buffer, count); |
| size += count; |
| } while (err == U_BUFFER_OVERFLOW_ERROR); |
| |
| return CString(result.data(), size); |
| } |
| |
| template <typename CharType> |
| CString TextCodecICU::encodeCommon(const CharType* characters, |
| size_t length, |
| UnencodableHandling handling) { |
| if (!length) |
| return ""; |
| |
| if (!m_converterICU) |
| createICUConverter(); |
| if (!m_converterICU) |
| return CString(); |
| |
| TextCodecInput input(m_encoding, characters, length); |
| return encodeInternal(input, handling); |
| } |
| |
| CString TextCodecICU::encode(const UChar* characters, |
| size_t length, |
| UnencodableHandling handling) { |
| return encodeCommon(characters, length, handling); |
| } |
| |
| CString TextCodecICU::encode(const LChar* characters, |
| size_t length, |
| UnencodableHandling handling) { |
| return encodeCommon(characters, length, handling); |
| } |
| |
| } // namespace WTF |