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 "TextCodec.h"
31#include "TextEncoding.h"
32#include <unicode/utypes.h>
33
34typedef struct UConverter UConverter;
35
36namespace WebCore {
37
38    class TextCodecICU : public TextCodec {
39    public:
40        static void registerEncodingNames(EncodingNameRegistrar);
41        static void registerCodecs(TextCodecRegistrar);
42
43        virtual ~TextCodecICU();
44
45    private:
46        TextCodecICU(const TextEncoding&);
47        static PassOwnPtr<TextCodec> create(const TextEncoding&, const void*);
48
49        virtual String decode(const char*, size_t length, bool flush, bool stopOnError, bool& sawError);
50        virtual CString encode(const UChar*, size_t length, UnencodableHandling);
51
52        void createICUConverter() const;
53        void releaseICUConverter() const;
54        bool needsGBKFallbacks() const { return m_needsGBKFallbacks; }
55        void setNeedsGBKFallbacks(bool needsFallbacks) { m_needsGBKFallbacks = needsFallbacks; }
56
57        int decodeToBuffer(UChar* buffer, UChar* bufferLimit, const char*& source,
58            const char* sourceLimit, int32_t* offsets, bool flush, UErrorCode& err);
59
60        TextEncoding m_encoding;
61        unsigned m_numBufferedBytes;
62        unsigned char m_bufferedBytes[16]; // bigger than any single multi-byte character
63        mutable UConverter* m_converterICU;
64        mutable bool m_needsGBKFallbacks;
65    };
66
67    struct ICUConverterWrapper {
68        ICUConverterWrapper() : converter(0) { }
69        ~ICUConverterWrapper();
70
71        UConverter* converter;
72
73        WTF_MAKE_NONCOPYABLE(ICUConverterWrapper);
74    };
75
76} // namespace WebCore
77
78#endif // TextCodecICU_h
79