|  | // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 
|  | // Use of this source code is governed by a BSD-style license that can be | 
|  | // found in the LICENSE file. | 
|  |  | 
|  | #include "base/i18n/icu_string_conversions.h" | 
|  |  | 
|  | #include <stddef.h> | 
|  | #include <stdint.h> | 
|  |  | 
|  | #include <memory> | 
|  | #include <vector> | 
|  |  | 
|  | #include "base/logging.h" | 
|  | #include "base/strings/string_util.h" | 
|  | #include "base/strings/utf_string_conversions.h" | 
|  | #include "third_party/icu/source/common/unicode/normalizer2.h" | 
|  | #include "third_party/icu/source/common/unicode/ucnv.h" | 
|  | #include "third_party/icu/source/common/unicode/ucnv_cb.h" | 
|  | #include "third_party/icu/source/common/unicode/ucnv_err.h" | 
|  | #include "third_party/icu/source/common/unicode/ustring.h" | 
|  |  | 
|  | namespace base { | 
|  |  | 
|  | namespace { | 
|  | // ToUnicodeCallbackSubstitute() is based on UCNV_TO_U_CALLBACK_SUBSTITUTE | 
|  | // in source/common/ucnv_err.c. | 
|  |  | 
|  | // Copyright (c) 1995-2006 International Business Machines Corporation | 
|  | // and others | 
|  | // | 
|  | // All rights reserved. | 
|  | // | 
|  |  | 
|  | // Permission is hereby granted, free of charge, to any person obtaining a | 
|  | // copy of this software and associated documentation files (the "Software"), | 
|  | // to deal in the Software without restriction, including without limitation | 
|  | // the rights to use, copy, modify, merge, publish, distribute, and/or | 
|  | // sell copies of the Software, and to permit persons to whom the Software | 
|  | // is furnished to do so, provided that the above copyright notice(s) and | 
|  | // this permission notice appear in all copies of the Software and that | 
|  | // both the above copyright notice(s) and this permission notice appear in | 
|  | // supporting documentation. | 
|  | // | 
|  | // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | 
|  | // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | 
|  | // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT | 
|  | // OF THIRD PARTY RIGHTS. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR HOLDERS | 
|  | // INCLUDED IN THIS NOTICE BE LIABLE FOR ANY CLAIM, OR ANY SPECIAL INDIRECT | 
|  | // OR CONSEQUENTIAL DAMAGES, OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS | 
|  | // OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE | 
|  | // OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE | 
|  | // OR PERFORMANCE OF THIS SOFTWARE. | 
|  | // | 
|  | // Except as contained in this notice, the name of a copyright holder | 
|  | // shall not be used in advertising or otherwise to promote the sale, use | 
|  | // or other dealings in this Software without prior written authorization | 
|  | // of the copyright holder. | 
|  |  | 
|  | //  ___________________________________________________________________________ | 
|  | // | 
|  | // All trademarks and registered trademarks mentioned herein are the property | 
|  | // of their respective owners. | 
|  |  | 
|  | void ToUnicodeCallbackSubstitute(const void* context, | 
|  | UConverterToUnicodeArgs *to_args, | 
|  | const char* code_units, | 
|  | int32_t length, | 
|  | UConverterCallbackReason reason, | 
|  | UErrorCode * err) { | 
|  | static const UChar kReplacementChar = 0xFFFD; | 
|  | if (reason <= UCNV_IRREGULAR) { | 
|  | if (context == nullptr || | 
|  | (*(reinterpret_cast<const char*>(context)) == 'i' && | 
|  | reason == UCNV_UNASSIGNED)) { | 
|  | *err = U_ZERO_ERROR; | 
|  | ucnv_cbToUWriteUChars(to_args, &kReplacementChar, 1, 0, err); | 
|  | } | 
|  | // else the caller must have set the error code accordingly. | 
|  | } | 
|  | // else ignore the reset, close and clone calls. | 
|  | } | 
|  |  | 
|  | bool ConvertFromUTF16(UConverter* converter, const UChar* uchar_src, | 
|  | int uchar_len, OnStringConversionError::Type on_error, | 
|  | std::string* encoded) { | 
|  | int encoded_max_length = UCNV_GET_MAX_BYTES_FOR_STRING(uchar_len, | 
|  | ucnv_getMaxCharSize(converter)); | 
|  | encoded->resize(encoded_max_length); | 
|  |  | 
|  | UErrorCode status = U_ZERO_ERROR; | 
|  |  | 
|  | // Setup our error handler. | 
|  | switch (on_error) { | 
|  | case OnStringConversionError::FAIL: | 
|  | ucnv_setFromUCallBack(converter, UCNV_FROM_U_CALLBACK_STOP, nullptr, | 
|  | nullptr, nullptr, &status); | 
|  | break; | 
|  | case OnStringConversionError::SKIP: | 
|  | case OnStringConversionError::SUBSTITUTE: | 
|  | ucnv_setFromUCallBack(converter, UCNV_FROM_U_CALLBACK_SKIP, nullptr, | 
|  | nullptr, nullptr, &status); | 
|  | break; | 
|  | default: | 
|  | NOTREACHED(); | 
|  | } | 
|  |  | 
|  | // ucnv_fromUChars returns size not including terminating null | 
|  | int actual_size = ucnv_fromUChars(converter, &(*encoded)[0], | 
|  | encoded_max_length, uchar_src, uchar_len, &status); | 
|  | encoded->resize(actual_size); | 
|  | ucnv_close(converter); | 
|  | if (U_SUCCESS(status)) | 
|  | return true; | 
|  | encoded->clear();  // Make sure the output is empty on error. | 
|  | return false; | 
|  | } | 
|  |  | 
|  | // Set up our error handler for ToUTF-16 converters | 
|  | void SetUpErrorHandlerForToUChars(OnStringConversionError::Type on_error, | 
|  | UConverter* converter, UErrorCode* status) { | 
|  | switch (on_error) { | 
|  | case OnStringConversionError::FAIL: | 
|  | ucnv_setToUCallBack(converter, UCNV_TO_U_CALLBACK_STOP, nullptr, nullptr, | 
|  | nullptr, status); | 
|  | break; | 
|  | case OnStringConversionError::SKIP: | 
|  | ucnv_setToUCallBack(converter, UCNV_TO_U_CALLBACK_SKIP, nullptr, nullptr, | 
|  | nullptr, status); | 
|  | break; | 
|  | case OnStringConversionError::SUBSTITUTE: | 
|  | ucnv_setToUCallBack(converter, ToUnicodeCallbackSubstitute, nullptr, | 
|  | nullptr, nullptr, status); | 
|  | break; | 
|  | default: | 
|  | NOTREACHED(); | 
|  | } | 
|  | } | 
|  |  | 
|  | }  // namespace | 
|  |  | 
|  | // Codepage <-> Wide/UTF-16  --------------------------------------------------- | 
|  |  | 
|  | bool UTF16ToCodepage(const string16& utf16, | 
|  | const char* codepage_name, | 
|  | OnStringConversionError::Type on_error, | 
|  | std::string* encoded) { | 
|  | encoded->clear(); | 
|  |  | 
|  | UErrorCode status = U_ZERO_ERROR; | 
|  | UConverter* converter = ucnv_open(codepage_name, &status); | 
|  | if (!U_SUCCESS(status)) | 
|  | return false; | 
|  |  | 
|  | return ConvertFromUTF16(converter, utf16.c_str(), | 
|  | static_cast<int>(utf16.length()), on_error, encoded); | 
|  | } | 
|  |  | 
|  | bool CodepageToUTF16(const std::string& encoded, | 
|  | const char* codepage_name, | 
|  | OnStringConversionError::Type on_error, | 
|  | string16* utf16) { | 
|  | utf16->clear(); | 
|  |  | 
|  | UErrorCode status = U_ZERO_ERROR; | 
|  | UConverter* converter = ucnv_open(codepage_name, &status); | 
|  | if (!U_SUCCESS(status)) | 
|  | return false; | 
|  |  | 
|  | // Even in the worst case, the maximum length in 2-byte units of UTF-16 | 
|  | // output would be at most the same as the number of bytes in input. There | 
|  | // is no single-byte encoding in which a character is mapped to a | 
|  | // non-BMP character requiring two 2-byte units. | 
|  | // | 
|  | // Moreover, non-BMP characters in legacy multibyte encodings | 
|  | // (e.g. EUC-JP, GB18030) take at least 2 bytes. The only exceptions are | 
|  | // BOCU and SCSU, but we don't care about them. | 
|  | size_t uchar_max_length = encoded.length() + 1; | 
|  |  | 
|  | SetUpErrorHandlerForToUChars(on_error, converter, &status); | 
|  | std::unique_ptr<char16[]> buffer(new char16[uchar_max_length]); | 
|  | int actual_size = ucnv_toUChars(converter, buffer.get(), | 
|  | static_cast<int>(uchar_max_length), encoded.data(), | 
|  | static_cast<int>(encoded.length()), &status); | 
|  | ucnv_close(converter); | 
|  | if (!U_SUCCESS(status)) { | 
|  | utf16->clear();  // Make sure the output is empty on error. | 
|  | return false; | 
|  | } | 
|  |  | 
|  | utf16->assign(buffer.get(), actual_size); | 
|  | return true; | 
|  | } | 
|  |  | 
|  | bool ConvertToUtf8AndNormalize(const std::string& text, | 
|  | const std::string& charset, | 
|  | std::string* result) { | 
|  | result->clear(); | 
|  | string16 utf16; | 
|  | if (!CodepageToUTF16( | 
|  | text, charset.c_str(), OnStringConversionError::FAIL, &utf16)) | 
|  | return false; | 
|  |  | 
|  | UErrorCode status = U_ZERO_ERROR; | 
|  | const icu::Normalizer2* normalizer = icu::Normalizer2::getNFCInstance(status); | 
|  | DCHECK(U_SUCCESS(status)); | 
|  | if (U_FAILURE(status)) | 
|  | return false; | 
|  | int32_t utf16_length = static_cast<int32_t>(utf16.length()); | 
|  | icu::UnicodeString normalized(utf16.data(), utf16_length); | 
|  | int32_t normalized_prefix_length = | 
|  | normalizer->spanQuickCheckYes(normalized, status); | 
|  | if (normalized_prefix_length < utf16_length) { | 
|  | icu::UnicodeString un_normalized(normalized, normalized_prefix_length); | 
|  | normalized.truncate(normalized_prefix_length); | 
|  | normalizer->normalizeSecondAndAppend(normalized, un_normalized, status); | 
|  | } | 
|  | if (U_FAILURE(status)) | 
|  | return false; | 
|  | normalized.toUTF8String(*result); | 
|  | return true; | 
|  | } | 
|  |  | 
|  | }  // namespace base |