d0f0c9e24acd575dbca39511f3e3718b8f601500
[WebKit-https.git] / Source / WebCore / platform / cf / KeyedEncoderCF.cpp
1 /*
2  * Copyright (C) 2013 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "KeyedEncoderCF.h"
28
29 #include "SharedBuffer.h"
30 #include <CoreFoundation/CoreFoundation.h>
31 #include <wtf/text/WTFString.h>
32
33 namespace WebCore {
34
35 std::unique_ptr<KeyedEncoder> KeyedEncoder::encoder()
36 {
37     return std::make_unique<KeyedEncoderCF>();
38 }
39
40 static RetainPtr<CFMutableDictionaryRef> createDictionary()
41 {
42     return adoptCF(CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks));
43 }
44
45 KeyedEncoderCF::KeyedEncoderCF()
46     : m_rootDictionary(createDictionary())
47 {
48     m_dictionaryStack.append(m_rootDictionary.get());
49 }
50     
51 KeyedEncoderCF::~KeyedEncoderCF()
52 {
53     ASSERT(m_dictionaryStack.size() == 1);
54     ASSERT(m_dictionaryStack.last() == m_rootDictionary);
55     ASSERT(m_arrayStack.isEmpty());
56 }
57
58 void KeyedEncoderCF::encodeBytes(const String& key, const uint8_t* bytes, size_t size)
59 {
60     auto data = adoptCF(CFDataCreateWithBytesNoCopy(kCFAllocatorDefault, bytes, size, kCFAllocatorNull));
61     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), data.get());
62 }
63
64 void KeyedEncoderCF::encodeBool(const String& key, bool value)
65 {
66     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), value ? kCFBooleanTrue : kCFBooleanFalse);
67 }
68
69 void KeyedEncoderCF::encodeUInt32(const String& key, uint32_t value)
70 {
71     auto number = adoptCF(CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, &value));
72     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), number.get());
73 }
74
75 void KeyedEncoderCF::encodeInt32(const String& key, int32_t value)
76 {
77     auto number = adoptCF(CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, &value));
78     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), number.get());
79 }
80
81 void KeyedEncoderCF::encodeInt64(const String& key, int64_t value)
82 {
83     auto number = adoptCF(CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt64Type, &value));
84     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), number.get());
85 }
86
87 void KeyedEncoderCF::encodeFloat(const String& key, float value)
88 {
89     auto number = adoptCF(CFNumberCreate(kCFAllocatorDefault, kCFNumberFloatType, &value));
90     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), number.get());
91 }
92
93 void KeyedEncoderCF::encodeDouble(const String& key, double value)
94 {
95     auto number = adoptCF(CFNumberCreate(kCFAllocatorDefault, kCFNumberDoubleType, &value));
96     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), number.get());
97 }
98
99 void KeyedEncoderCF::encodeString(const String& key, const String& value)
100 {
101     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), value.createCFString().get());
102 }
103
104 void KeyedEncoderCF::beginObject(const String& key)
105 {
106     auto dictionary = createDictionary();
107     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), dictionary.get());
108
109     m_dictionaryStack.append(dictionary.get());
110 }
111
112 void KeyedEncoderCF::endObject()
113 {
114     m_dictionaryStack.removeLast();
115 }
116
117 void KeyedEncoderCF::beginArray(const String& key)
118 {
119     auto array = adoptCF(CFArrayCreateMutable(kCFAllocatorDefault, 0, &kCFTypeArrayCallBacks));
120     CFDictionarySetValue(m_dictionaryStack.last(), key.createCFString().get(), array.get());
121
122     m_arrayStack.append(array.get());
123 }
124
125 void KeyedEncoderCF::beginArrayElement()
126 {
127     auto dictionary = createDictionary();
128     CFArrayAppendValue(m_arrayStack.last(), dictionary.get());
129
130     m_dictionaryStack.append(dictionary.get());
131 }
132
133 void KeyedEncoderCF::endArrayElement()
134 {
135     m_dictionaryStack.removeLast();
136 }
137
138 void KeyedEncoderCF::endArray()
139 {
140     m_arrayStack.removeLast();
141 }
142
143 RefPtr<SharedBuffer> KeyedEncoderCF::finishEncoding()
144 {
145     auto data = adoptCF(CFPropertyListCreateData(kCFAllocatorDefault, m_rootDictionary.get(), kCFPropertyListBinaryFormat_v1_0, 0, nullptr));
146     if (!data)
147         return nullptr;
148     return SharedBuffer::create(data.get());
149 }
150
151 } // namespace WebCore