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 
34 typedef struct UConverter UConverter;
35 
36 namespace 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;
needsGBKFallbacks()54         bool needsGBKFallbacks() const { return m_needsGBKFallbacks; }
setNeedsGBKFallbacks(bool needsFallbacks)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 {
ICUConverterWrapperICUConverterWrapper68         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