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
34typedef struct UConverter UConverter;
35
36namespace WTF {
37
38class TextCodecInput;
39
40class TextCodecICU final : public TextCodec {
41public:
42    static void registerEncodingNames(EncodingNameRegistrar);
43    static void registerCodecs(TextCodecRegistrar);
44
45    virtual ~TextCodecICU();
46
47private:
48    TextCodecICU(const TextEncoding&);
49    static PassOwnPtr<TextCodec> create(const TextEncoding&, const void*);
50
51    virtual String decode(const char*, size_t length, FlushBehavior, bool stopOnError, bool& sawError) override;
52    virtual CString encode(const UChar*, size_t length, UnencodableHandling) override;
53    virtual 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    bool needsGBKFallbacks() const { return m_needsGBKFallbacks; }
62    void setNeedsGBKFallbacks(bool needsFallbacks) { m_needsGBKFallbacks = needsFallbacks; }
63
64    int decodeToBuffer(UChar* buffer, UChar* bufferLimit, const char*& source,
65        const char* sourceLimit, int32_t* offsets, bool flush, UErrorCode&);
66
67    TextEncoding m_encoding;
68    mutable UConverter* m_converterICU;
69    mutable bool m_needsGBKFallbacks;
70};
71
72struct ICUConverterWrapper {
73    WTF_MAKE_NONCOPYABLE(ICUConverterWrapper); WTF_MAKE_FAST_ALLOCATED;
74public:
75    ICUConverterWrapper() : converter(0) { }
76    ~ICUConverterWrapper();
77
78    UConverter* converter;
79};
80
81} // namespace WTF
82
83#endif // TextCodecICU_h
84