You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
340 lines
14 KiB
340 lines
14 KiB
/*
|
|
* Copyright 2020 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include "storage/config_cache.h"
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <cstdio>
|
|
|
|
#include "hci/enum_helper.h"
|
|
#include "storage/device.h"
|
|
|
|
namespace testing {
|
|
|
|
namespace {
|
|
std::string GetTestAddress(int i) {
|
|
std::string res = "00:00:00:00:00:00";
|
|
res.reserve(res.size() + 1);
|
|
std::snprintf(res.data(), res.capacity(), "AA:BB:CC:DD:EE:%02d", i);
|
|
return res;
|
|
}
|
|
} // namespace
|
|
|
|
using bluetooth::storage::ConfigCache;
|
|
using bluetooth::storage::Device;
|
|
using SectionAndPropertyValue = bluetooth::storage::ConfigCache::SectionAndPropertyValue;
|
|
|
|
TEST(ConfigCacheTest, simple_set_get_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
auto value = config.GetProperty("A", "B");
|
|
ASSERT_TRUE(value);
|
|
ASSERT_EQ(*value, "C");
|
|
}
|
|
|
|
TEST(ConfigCacheTest, empty_values_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
ASSERT_DEATH({ config.SetProperty("", "B", "C"); }, "Empty section name not allowed");
|
|
ASSERT_DEATH({ config.SetProperty("A", "", "C"); }, "Empty property name not allowed");
|
|
// empty value is allowed
|
|
config.SetProperty("A", "B", "");
|
|
auto value = config.GetProperty("A", "B");
|
|
ASSERT_TRUE(value);
|
|
ASSERT_EQ(*value, "");
|
|
}
|
|
|
|
TEST(ConfigCacheTest, insert_boundary_device_with_linkkey_test) {
|
|
ConfigCache config(2, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("CC:DD:EE:FF:00:10", "Name", "Hello");
|
|
config.SetProperty("CC:DD:EE:FF:00:09", "Name", "Hello 2");
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
ASSERT_TRUE(config.GetProperty("CC:DD:EE:FF:00:10", "Name"));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, comparison_test) {
|
|
ConfigCache config_1(2, Device::kLinkKeyProperties);
|
|
config_1.SetProperty("A", "B", "C");
|
|
config_1.SetProperty("CC:DD:EE:FF:00:10", "Name", "Hello");
|
|
config_1.SetProperty("CC:DD:EE:FF:00:09", "Name", "Hello 2");
|
|
config_1.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
ConfigCache config_2(2, Device::kLinkKeyProperties);
|
|
config_2.SetProperty("A", "B", "C");
|
|
config_2.SetProperty("CC:DD:EE:FF:00:10", "Name", "Hello");
|
|
config_2.SetProperty("CC:DD:EE:FF:00:09", "Name", "Hello 2");
|
|
config_2.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
ASSERT_EQ(config_1, config_2);
|
|
// Config with different temp device order should not be equal
|
|
ASSERT_TRUE(config_2.GetProperty("CC:DD:EE:FF:00:10", "Name"));
|
|
ASSERT_NE(config_1, config_2);
|
|
ASSERT_TRUE(config_1.GetProperty("CC:DD:EE:FF:00:10", "Name"));
|
|
ASSERT_EQ(config_1, config_2);
|
|
// Config with different persistent device order should not be equal
|
|
config_1.SetProperty("CC:DD:EE:FF:00:12", "LinkKey", "AABBAABBCCDDEE");
|
|
config_2.RemoveSection("CC:DD:EE:FF:00:11");
|
|
config_2.SetProperty("CC:DD:EE:FF:00:12", "LinkKey", "AABBAABBCCDDEE");
|
|
config_2.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
ASSERT_NE(config_1, config_2);
|
|
// Config with different capacity should not be equal
|
|
ConfigCache config_3(3, Device::kLinkKeyProperties);
|
|
config_3.SetProperty("A", "B", "C");
|
|
config_3.SetProperty("CC:DD:EE:FF:00:10", "Name", "Hello");
|
|
config_3.SetProperty("CC:DD:EE:FF:00:09", "Name", "Hello 2");
|
|
config_3.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
config_3.SetProperty("CC:DD:EE:FF:00:12", "LinkKey", "AABBAABBCCDDEE");
|
|
ASSERT_NE(config_1, config_3);
|
|
// Empty config should not be equal to non-empty ones
|
|
ConfigCache config_4(2, Device::kLinkKeyProperties);
|
|
ASSERT_NE(config_1, config_4);
|
|
// Empty configs should be equal
|
|
ConfigCache config_5(2, Device::kLinkKeyProperties);
|
|
ASSERT_EQ(config_4, config_5);
|
|
// Empty configs with different capacity should not be equal
|
|
ConfigCache config_6(3, Device::kLinkKeyProperties);
|
|
ASSERT_NE(config_4, config_6);
|
|
}
|
|
|
|
TEST(ConfigCacheTest, empty_string_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "");
|
|
auto value = config.GetProperty("A", "B");
|
|
ASSERT_TRUE(value);
|
|
ASSERT_EQ(*value, "");
|
|
}
|
|
|
|
TEST(ConfigCacheTest, mac_address_set_get_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
auto value = config.GetProperty("A", "B");
|
|
ASSERT_TRUE(value);
|
|
ASSERT_EQ(*value, "C");
|
|
value = config.GetProperty("AA:BB:CC:DD:EE:FF", "B");
|
|
ASSERT_TRUE(value);
|
|
ASSERT_EQ(*value, "C");
|
|
ASSERT_FALSE(config.GetProperty("A", "BC"));
|
|
ASSERT_FALSE(config.GetProperty("ABC", "B"));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, has_section_and_property_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
ASSERT_TRUE(config.HasSection("A"));
|
|
ASSERT_TRUE(config.HasSection("AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_TRUE(config.HasProperty("A", "B"));
|
|
ASSERT_TRUE(config.HasProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
auto value = config.GetProperty("AA:BB:CC:DD:EE:FF", "C");
|
|
ASSERT_TRUE(value);
|
|
ASSERT_EQ(*value, "D");
|
|
value = config.GetProperty("AA:BB:CC:DD:EE:FF", "B");
|
|
ASSERT_TRUE(value);
|
|
ASSERT_EQ(*value, "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "E");
|
|
value = config.GetProperty("AA:BB:CC:DD:EE:FF", "B");
|
|
ASSERT_TRUE(value);
|
|
ASSERT_THAT(value, Optional(StrEq("E")));
|
|
ASSERT_FALSE(config.HasSection("Ab"));
|
|
ASSERT_FALSE(config.HasSection("AA:11:CC:DD:EE:FF"));
|
|
ASSERT_FALSE(config.HasProperty("A", "bB"));
|
|
ASSERT_FALSE(config.HasProperty("AA:BB:11:DD:EE:FF", "B"));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, remove_section_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
ASSERT_TRUE(config.HasSection("A"));
|
|
ASSERT_TRUE(config.HasSection("AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_TRUE(config.HasProperty("A", "B"));
|
|
ASSERT_TRUE(config.HasProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
ASSERT_TRUE(config.RemoveSection("AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_TRUE(config.RemoveSection("A"));
|
|
ASSERT_FALSE(config.HasProperty("A", "B"));
|
|
ASSERT_FALSE(config.HasProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, remove_property_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
ASSERT_TRUE(config.HasSection("A"));
|
|
ASSERT_TRUE(config.HasSection("AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_TRUE(config.HasProperty("A", "B"));
|
|
ASSERT_TRUE(config.HasProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
ASSERT_TRUE(config.HasProperty("AA:BB:CC:DD:EE:FF", "C"));
|
|
ASSERT_TRUE(config.RemoveProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
ASSERT_FALSE(config.HasProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
ASSERT_FALSE(config.GetProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, remove_all_properties_from_section_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
ASSERT_TRUE(config.HasSection("A"));
|
|
ASSERT_TRUE(config.HasSection("AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_TRUE(config.HasProperty("A", "B"));
|
|
ASSERT_TRUE(config.HasProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
ASSERT_TRUE(config.HasProperty("AA:BB:CC:DD:EE:FF", "C"));
|
|
ASSERT_TRUE(config.RemoveSection("AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_FALSE(config.HasSection("AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_FALSE(config.HasProperty("AA:BB:CC:DD:EE:FF", "B"));
|
|
ASSERT_FALSE(config.GetProperty("AA:BB:CC:DD:EE:FF", "C"));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, get_persistent_devices_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
ASSERT_TRUE(config.HasProperty("CC:DD:EE:FF:00:11", "LinkKey"));
|
|
ASSERT_THAT(config.GetPersistentSections(), ElementsAre("CC:DD:EE:FF:00:11"));
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "LinkKey", "DEERDEERDEER");
|
|
ASSERT_THAT(config.GetPersistentSections(), ElementsAre("CC:DD:EE:FF:00:11", "AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_TRUE(config.RemoveProperty("CC:DD:EE:FF:00:11", "LinkKey"));
|
|
ASSERT_THAT(config.GetPersistentSections(), ElementsAre("AA:BB:CC:DD:EE:FF"));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, appoaching_temporary_config_limit_test) {
|
|
ConfigCache config(2, Device::kLinkKeyProperties);
|
|
for (int i = 0; i < 10; ++i) {
|
|
config.SetProperty(GetTestAddress(i), "Name", "Hello" + std::to_string(i));
|
|
if (i % 2 == 0) {
|
|
config.SetProperty(GetTestAddress(i), "LinkKey", "Key" + std::to_string(i));
|
|
}
|
|
}
|
|
for (int i = 0; i < 10; ++i) {
|
|
if (i % 2 == 0) {
|
|
ASSERT_TRUE(config.HasSection(GetTestAddress(i)));
|
|
ASSERT_TRUE(config.HasProperty(GetTestAddress(i), "LinkKey"));
|
|
ASSERT_THAT(config.GetProperty(GetTestAddress(i), "Name"), Optional(StrEq("Hello" + std::to_string(i))));
|
|
} else if (i >= 7) {
|
|
ASSERT_TRUE(config.HasSection(GetTestAddress(i)));
|
|
ASSERT_THAT(config.GetProperty(GetTestAddress(i), "Name"), Optional(StrEq("Hello" + std::to_string(i))));
|
|
} else {
|
|
ASSERT_FALSE(config.HasSection(GetTestAddress(i)));
|
|
}
|
|
}
|
|
ASSERT_THAT(
|
|
config.GetPersistentSections(),
|
|
ElementsAre(GetTestAddress(0), GetTestAddress(2), GetTestAddress(4), GetTestAddress(6), GetTestAddress(8)));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, remove_section_with_property_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "B", "AABBAABBCCDDEE");
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
config.RemoveSectionWithProperty("B");
|
|
ASSERT_FALSE(config.HasSection("A"));
|
|
ASSERT_FALSE(config.HasSection("AA:BB:CC:DD:EE:FF"));
|
|
ASSERT_FALSE(config.HasSection("CC:DD:EE:FF:00:11"));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, persistent_config_changed_callback_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
int num_change = 0;
|
|
config.SetPersistentConfigChangedCallback([&num_change] { num_change++; });
|
|
config.SetProperty("A", "B", "C");
|
|
ASSERT_EQ(num_change, 1);
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
ASSERT_EQ(num_change, 1);
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
ASSERT_EQ(num_change, 1);
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "B", "AABBAABBCCDDEE");
|
|
ASSERT_EQ(num_change, 1);
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
ASSERT_EQ(num_change, 2);
|
|
config.RemoveProperty("CC:DD:EE:FF:00:11", "LinkKey");
|
|
ASSERT_EQ(num_change, 3);
|
|
config.RemoveSectionWithProperty("B");
|
|
ASSERT_EQ(num_change, 4);
|
|
}
|
|
|
|
TEST(ConfigCacheTest, fix_device_type_inconsistency_test) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "C", "D");
|
|
ASSERT_TRUE(config.FixDeviceTypeInconsistencies());
|
|
ASSERT_THAT(
|
|
config.GetProperty("AA:BB:CC:DD:EE:FF", "DevType"),
|
|
Optional(StrEq(std::to_string(bluetooth::hci::DeviceType::BR_EDR))));
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "B", "AABBAABBCCDDEE");
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "DevType", std::to_string(bluetooth::hci::DeviceType::BR_EDR));
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "LinkKey", "AABBAABBCCDDEE");
|
|
ASSERT_FALSE(config.FixDeviceTypeInconsistencies());
|
|
ASSERT_THAT(
|
|
config.GetProperty("CC:DD:EE:FF:00:11", "DevType"),
|
|
Optional(StrEq(std::to_string(bluetooth::hci::DeviceType::BR_EDR))));
|
|
config.SetProperty("CC:DD:EE:FF:00:11", "LE_KEY_PENC", "AABBAABBCCDDEE");
|
|
ASSERT_TRUE(config.FixDeviceTypeInconsistencies());
|
|
ASSERT_THAT(
|
|
config.GetProperty("CC:DD:EE:FF:00:11", "DevType"),
|
|
Optional(StrEq(std::to_string(bluetooth::hci::DeviceType::DUAL))));
|
|
config.RemoveProperty("CC:DD:EE:FF:00:11", "LinkKey");
|
|
ASSERT_TRUE(config.FixDeviceTypeInconsistencies());
|
|
ASSERT_THAT(
|
|
config.GetProperty("CC:DD:EE:FF:00:11", "DevType"),
|
|
Optional(StrEq(std::to_string(bluetooth::hci::DeviceType::LE))));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, test_get_section_with_property) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:EF", "C", "D");
|
|
ASSERT_THAT(
|
|
config.GetSectionNamesWithProperty("B"),
|
|
ElementsAre(
|
|
SectionAndPropertyValue{.section = "A", .property = "C"},
|
|
SectionAndPropertyValue{.section = "AA:BB:CC:DD:EE:FF", .property = "C"}));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, test_get_sections_matching_at_least_one_property) {
|
|
ConfigCache config(100, Device::kLinkKeyProperties);
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:EF", "C", "D");
|
|
ASSERT_TRUE(config.HasAtLeastOneMatchingPropertiesInSection("AA:BB:CC:DD:EE:FF", {"B", "C", "D"}));
|
|
ASSERT_TRUE(config.HasAtLeastOneMatchingPropertiesInSection("A", {"B", "C", "D"}));
|
|
ASSERT_FALSE(config.HasAtLeastOneMatchingPropertiesInSection("AA:BB:CC:DD:EE:FF", {"BC", "D"}));
|
|
}
|
|
|
|
TEST(ConfigCacheTest, test_empty_persistent_properties) {
|
|
ConfigCache config(100, {});
|
|
config.SetProperty("A", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:FF", "B", "C");
|
|
config.SetProperty("AA:BB:CC:DD:EE:EF", "C", "D");
|
|
config.SetProperty("AA:BB:CC:DD:EE:EF", "LinkKey", "D");
|
|
ASSERT_TRUE(config.HasAtLeastOneMatchingPropertiesInSection("AA:BB:CC:DD:EE:FF", {"B", "C", "D"}));
|
|
ASSERT_TRUE(config.HasAtLeastOneMatchingPropertiesInSection("A", {"B", "C", "D"}));
|
|
ASSERT_FALSE(config.HasAtLeastOneMatchingPropertiesInSection("AA:BB:CC:DD:EE:FF", {"BC", "D"}));
|
|
ASSERT_THAT(config.GetPersistentSections(), ElementsAre());
|
|
}
|
|
|
|
} // namespace testing
|