1 /*
2 * Copyright 2020 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at:
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "osi/include/config.h"
18
19 #include <base/files/file_util.h>
20 #include <gtest/gtest.h>
21
22 #include <filesystem>
23
24 static const std::filesystem::path kConfigFile =
25 std::filesystem::temp_directory_path() / "config_test.conf";
26 static const char* CONFIG_FILE = kConfigFile.c_str();
27 static const char CONFIG_FILE_CONTENT[] =
28 " \n\
29 first_key=value \n\
30 \n\
31 # Device ID (DID) configuration \n\
32 [DID] \n\
33 \n\
34 # Record Number: 1, 2 or 3 - maximum of 3 records \n\
35 recordNumber = 1 \n\
36 \n\
37 # Primary Record - true or false (default) \n\
38 # There can be only one primary record \n\
39 primaryRecord = true \n\
40 \n\
41 # Vendor ID '0xFFFF' indicates no Device ID Service Record is present in the device \n\
42 # 0x000F = Broadcom Corporation (default) \n\
43 #vendorId = 0x000F \n\
44 \n\
45 # Vendor ID Source \n\
46 # 0x0001 = Bluetooth SIG assigned Device ID Vendor ID value (default) \n\
47 # 0x0002 = USB Implementer's Forum assigned Device ID Vendor ID value \n\
48 #vendorIdSource = 0x0001 \n\
49 \n\
50 # Product ID & Product Version \n\
51 # Per spec DID v1.3 0xJJMN for version is interpreted as JJ.M.N \n\
52 # JJ: major version number, M: minor version number, N: sub-minor version number \n\
53 # For example: 1200, v14.3.6 \n\
54 productId = 0x1200 \n\
55 version = 0x1111 \n\
56 \n\
57 # Optional attributes \n\
58 #clientExecutableURL = \n\
59 #serviceDescription = \n\
60 #documentationURL = \n\
61 \n\
62 # Additional optional DID records. Bluedroid supports up to 3 records. \n\
63 [DID] \n\
64 [DID] \n\
65 version = 0x1436 \n\
66 \n\
67 HiSyncId = 18446744073709551615 \n\
68 HiSyncId2 = 15001900 \n\
69 ";
70
71 class ConfigTest : public ::testing::Test {
72 protected:
SetUp()73 void SetUp() override {
74 FILE* fp = fopen(CONFIG_FILE, "wt");
75 ASSERT_NE(fp, nullptr);
76 ASSERT_EQ(fwrite(CONFIG_FILE_CONTENT, 1, sizeof(CONFIG_FILE_CONTENT), fp),
77 sizeof(CONFIG_FILE_CONTENT));
78 ASSERT_EQ(fclose(fp), 0);
79 }
80
TearDown()81 void TearDown() override { EXPECT_TRUE(std::filesystem::remove(kConfigFile)); }
82 };
83
TEST_F(ConfigTest,config_find)84 TEST_F(ConfigTest, config_find) {
85 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
86 ASSERT_NE(config, nullptr);
87 EXPECT_TRUE(config->Has("DID"));
88 auto section_iter = config->Find("DID");
89 ASSERT_NE(section_iter, config->sections.end());
90 EXPECT_FALSE(config->Has("random"));
91 EXPECT_EQ(config->Find("random"), config->sections.end());
92 }
93
TEST_F(ConfigTest,section_find)94 TEST_F(ConfigTest, section_find) {
95 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
96 ASSERT_NE(config, nullptr);
97 EXPECT_TRUE(config->Has("DID"));
98 auto section_iter = config->Find("DID");
99 ASSERT_NE(section_iter, config->sections.end());
100 EXPECT_EQ(section_iter->name, "DID");
101 EXPECT_TRUE(section_iter->Has("version"));
102 auto entry_iter = section_iter->Find("version");
103 ASSERT_NE(entry_iter, section_iter->entries.end());
104 EXPECT_EQ(entry_iter->key, "version");
105 EXPECT_EQ(entry_iter->value, "0x1436");
106 EXPECT_EQ(section_iter->Find("random"), section_iter->entries.end());
107 EXPECT_FALSE(section_iter->Has("random"));
108 }
109
TEST_F(ConfigTest,section_set)110 TEST_F(ConfigTest, section_set) {
111 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
112 ASSERT_NE(config, nullptr);
113 EXPECT_TRUE(config->Has("DID"));
114 auto section_iter = config->Find("DID");
115 ASSERT_NE(section_iter, config->sections.end());
116 EXPECT_EQ(section_iter->name, "DID");
117 EXPECT_FALSE(section_iter->Has("random"));
118 section_iter->Set("random", "foo");
119 EXPECT_TRUE(section_iter->Has("random"));
120 auto entry_iter = section_iter->Find("random");
121 ASSERT_NE(entry_iter, section_iter->entries.end());
122 EXPECT_EQ(entry_iter->key, "random");
123 EXPECT_EQ(entry_iter->value, "foo");
124 section_iter->Set("random", "bar");
125 EXPECT_EQ(entry_iter->value, "bar");
126 entry_iter = section_iter->Find("random");
127 ASSERT_NE(entry_iter, section_iter->entries.end());
128 EXPECT_EQ(entry_iter->value, "bar");
129 }
130
TEST_F(ConfigTest,config_new_empty)131 TEST_F(ConfigTest, config_new_empty) {
132 std::unique_ptr<config_t> config = config_new_empty();
133 EXPECT_TRUE(config.get() != NULL);
134 }
135
TEST_F(ConfigTest,config_new_no_file)136 TEST_F(ConfigTest, config_new_no_file) {
137 std::unique_ptr<config_t> config = config_new("/meow");
138 EXPECT_TRUE(config.get() == NULL);
139 }
140
TEST_F(ConfigTest,config_new)141 TEST_F(ConfigTest, config_new) {
142 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
143 EXPECT_TRUE(config.get() != NULL);
144 }
145
TEST_F(ConfigTest,config_new_clone)146 TEST_F(ConfigTest, config_new_clone) {
147 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
148 std::unique_ptr<config_t> clone = config_new_clone(*config);
149
150 config_set_string(clone.get(), CONFIG_DEFAULT_SECTION, "first_key", "not_value");
151
152 std::string one = std::string("one");
153 EXPECT_STRNE(config_get_string(*config, CONFIG_DEFAULT_SECTION, "first_key", &one)->c_str(),
154 config_get_string(*clone, CONFIG_DEFAULT_SECTION, "first_key", &one)->c_str());
155 }
156
TEST_F(ConfigTest,config_has_section)157 TEST_F(ConfigTest, config_has_section) {
158 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
159 EXPECT_TRUE(config_has_section(*config, "DID"));
160 }
161
TEST_F(ConfigTest,config_has_key_in_default_section)162 TEST_F(ConfigTest, config_has_key_in_default_section) {
163 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
164 EXPECT_TRUE(config_has_key(*config, CONFIG_DEFAULT_SECTION, "first_key"));
165 EXPECT_STREQ(config_get_string(*config, CONFIG_DEFAULT_SECTION, "first_key", nullptr)->c_str(),
166 "value");
167 }
168
TEST_F(ConfigTest,config_has_keys)169 TEST_F(ConfigTest, config_has_keys) {
170 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
171 EXPECT_TRUE(config_has_key(*config, "DID", "recordNumber"));
172 EXPECT_TRUE(config_has_key(*config, "DID", "primaryRecord"));
173 EXPECT_TRUE(config_has_key(*config, "DID", "productId"));
174 EXPECT_TRUE(config_has_key(*config, "DID", "version"));
175 }
176
TEST_F(ConfigTest,config_no_bad_keys)177 TEST_F(ConfigTest, config_no_bad_keys) {
178 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
179 EXPECT_FALSE(config_has_key(*config, "DID_BAD", "primaryRecord"));
180 EXPECT_FALSE(config_has_key(*config, "DID", "primaryRecord_BAD"));
181 EXPECT_FALSE(config_has_key(*config, CONFIG_DEFAULT_SECTION, "primaryRecord"));
182 }
183
TEST_F(ConfigTest,config_get_int_version)184 TEST_F(ConfigTest, config_get_int_version) {
185 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
186 EXPECT_EQ(config_get_int(*config, "DID", "version", 0), 0x1436);
187 }
188
TEST_F(ConfigTest,config_get_int_default)189 TEST_F(ConfigTest, config_get_int_default) {
190 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
191 EXPECT_EQ(config_get_int(*config, "DID", "primaryRecord", 123), 123);
192 }
193
TEST_F(ConfigTest,config_get_uint64)194 TEST_F(ConfigTest, config_get_uint64) {
195 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
196 EXPECT_EQ(config_get_uint64(*config, "DID", "HiSyncId", 0), 0xFFFFFFFFFFFFFFFF);
197 EXPECT_EQ(config_get_uint64(*config, "DID", "HiSyncId2", 0), uint64_t(15001900));
198 }
199
TEST_F(ConfigTest,config_get_uint64_default)200 TEST_F(ConfigTest, config_get_uint64_default) {
201 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
202 EXPECT_EQ(config_get_uint64(*config, "DID", "primaryRecord", 123), uint64_t(123));
203 }
204
TEST_F(ConfigTest,config_remove_section)205 TEST_F(ConfigTest, config_remove_section) {
206 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
207 EXPECT_TRUE(config_remove_section(config.get(), "DID"));
208 EXPECT_FALSE(config_has_section(*config, "DID"));
209 EXPECT_FALSE(config_has_key(*config, "DID", "productId"));
210 }
211
TEST_F(ConfigTest,config_remove_section_missing)212 TEST_F(ConfigTest, config_remove_section_missing) {
213 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
214 EXPECT_FALSE(config_remove_section(config.get(), "not a section"));
215 }
216
TEST_F(ConfigTest,config_remove_key)217 TEST_F(ConfigTest, config_remove_key) {
218 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
219 EXPECT_EQ(config_get_int(*config, "DID", "productId", 999), 0x1200);
220 EXPECT_TRUE(config_remove_key(config.get(), "DID", "productId"));
221 EXPECT_FALSE(config_has_key(*config, "DID", "productId"));
222 }
223
TEST_F(ConfigTest,config_remove_key_missing)224 TEST_F(ConfigTest, config_remove_key_missing) {
225 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
226 EXPECT_EQ(config_get_int(*config, "DID", "productId", 999), 0x1200);
227 EXPECT_TRUE(config_remove_key(config.get(), "DID", "productId"));
228 EXPECT_EQ(config_get_int(*config, "DID", "productId", 999), 999);
229 }
230
TEST_F(ConfigTest,config_save_basic)231 TEST_F(ConfigTest, config_save_basic) {
232 std::unique_ptr<config_t> config = config_new(CONFIG_FILE);
233 EXPECT_TRUE(config_save(*config, CONFIG_FILE));
234 }
235
TEST_F(ConfigTest,checksum_read)236 TEST_F(ConfigTest, checksum_read) {
237 auto tmp_dir = std::filesystem::temp_directory_path();
238 auto filename = tmp_dir / "test.checksum";
239 std::string checksum = "0x1234";
240 base::FilePath file_path(filename.string());
241
242 EXPECT_EQ(base::WriteFile(file_path, checksum.data(), checksum.size()), (int)checksum.size());
243
244 EXPECT_EQ(checksum_read(filename.c_str()), checksum.c_str());
245
246 EXPECT_TRUE(std::filesystem::remove(filename));
247 }
248
TEST_F(ConfigTest,checksum_save)249 TEST_F(ConfigTest, checksum_save) {
250 auto tmp_dir = std::filesystem::temp_directory_path();
251 auto filename = tmp_dir / "test.checksum";
252 std::string checksum = "0x1234";
253 base::FilePath file_path(filename.string());
254
255 EXPECT_TRUE(checksum_save(checksum, filename));
256
257 EXPECT_TRUE(base::PathExists(file_path));
258
259 EXPECT_TRUE(std::filesystem::remove(filename));
260 }
261