1/*
2 * Copyright (C) 2004-2017 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 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 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#pragma once
28
29#include "TextCodec.h"
30#include <unicode/ucnv.h>
31
32namespace WebCore {
33
34using ICUConverterPtr = std::unique_ptr<UConverter, void (*)(UConverter*)>;
35
36class TextCodecICU : public TextCodec {
37public:
38 TextCodecICU(const char* encoding, const char* canonicalConverterName);
39 virtual ~TextCodecICU();
40
41 static void registerEncodingNames(EncodingNameRegistrar);
42 static void registerCodecs(TextCodecRegistrar);
43
44private:
45 String decode(const char*, size_t length, bool flush, bool stopOnError, bool& sawError) final;
46 Vector<uint8_t> encode(StringView, UnencodableHandling) final;
47
48 void createICUConverter() const;
49 void releaseICUConverter() const;
50 bool needsGBKFallbacks() const { return m_needsGBKFallbacks; }
51 void setNeedsGBKFallbacks(bool needsFallbacks) { m_needsGBKFallbacks = needsFallbacks; }
52
53 int decodeToBuffer(UChar* buffer, UChar* bufferLimit, const char*& source, const char* sourceLimit, int32_t* offsets, bool flush, UErrorCode&);
54
55 const char* const m_encodingName;
56 const char* const m_canonicalConverterName;
57 mutable ICUConverterPtr m_converter { nullptr, ucnv_close };
58 mutable bool m_needsGBKFallbacks { false };
59};
60
61struct ICUConverterWrapper {
62 WTF_MAKE_STRUCT_FAST_ALLOCATED;
63
64 ICUConverterPtr converter { nullptr, ucnv_close };
65};
66
67} // namespace WebCore
68
69