third_party/WebKit/Source/wtf/text/TextCodecICU.h

   1 /*
   2  * Copyright (C) 2004, 2006, 2007, 2011 Apple Inc. All rights reserved.
   3  * Copyright (C) 2006 Alexey Proskuryakov <ap@nypop.com>
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
  15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
  18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #ifndef TextCodecICU_h
  28 #define TextCodecICU_h
  29
  30 #include <unicode/utypes.h>
  31 #include "wtf/text/TextCodec.h"
  32 #include "wtf/text/TextEncoding.h"
  33
  34 typedef struct UConverter UConverter;
  35
  36 namespace WTF {
  37
  38 class TextCodecInput;
  39
  40 class TextCodecICU final : public TextCodec {
  41 public:
  42     static void registerEncodingNames(EncodingNameRegistrar);
  43     static void registerCodecs(TextCodecRegistrar);
  44
  45     ~TextCodecICU() override;
  46
  47 private:
  48     TextCodecICU(const TextEncoding&);
  49     static PassOwnPtr<TextCodec> create(const TextEncoding&, const void*);
  50
  51     String decode(const char*, size_t length, FlushBehavior, bool stopOnError, bool& sawError) override;
  52     CString encode(const UChar*, size_t length, UnencodableHandling) override;
  53     CString encode(const LChar*, size_t length, UnencodableHandling) override;
  54
  55     template<typename CharType>
  56     CString encodeCommon(const CharType*, size_t length, UnencodableHandling);
  57     CString encodeInternal(const TextCodecInput&, UnencodableHandling);
  58
  59     void createICUConverter() const;
  60     void releaseICUConverter() const;
  61 #if defined(USING_SYSTEM_ICU)
  62     bool needsGBKFallbacks() const { return m_needsGBKFallbacks; }
  63     void setNeedsGBKFallbacks(bool needsFallbacks) { m_needsGBKFallbacks = needsFallbacks; }
  64 #endif
  65
  66     int decodeToBuffer(UChar* buffer, UChar* bufferLimit, const char*& source,
  67         const char* sourceLimit, int32_t* offsets, bool flush, UErrorCode&);
  68
  69     TextEncoding m_encoding;
  70     mutable UConverter* m_converterICU;
  71 #if defined(USING_SYSTEM_ICU)
  72     mutable bool m_needsGBKFallbacks;
  73 #endif
  74 };
  75
  76 struct ICUConverterWrapper {
  77     WTF_MAKE_NONCOPYABLE(ICUConverterWrapper); WTF_MAKE_FAST_ALLOCATED(ICUConverterWrapper);
  78 public:
  79     ICUConverterWrapper() : converter(0) { }
  80     ~ICUConverterWrapper();
  81
  82     UConverter* converter;
  83 };
  84
  85 } // namespace WTF
  86
  87 #endif // TextCodecICU_h