blob: feeb7a32cdcf97d085c7d2e9efad01f545313b75 [file] [log] [blame]
/*
* Copyright 2021 HIMSA II K/S - www.himsa.com.
* Represented by EHIMA - www.ehima.com
*
* 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 <base/logging.h>
#include <algorithm>
#include <limits>
#include <map>
#include <mutex>
#include <unordered_set>
#include "bta_groups.h"
#include "btif/include/btif_profile_storage.h"
#include "os/logging/log_adapter.h"
#include "stack/include/bt_types.h"
#include "types/bluetooth/uuid.h"
#include "types/raw_address.h"
using bluetooth::Uuid;
namespace bluetooth {
namespace groups {
class DeviceGroupsImpl;
DeviceGroupsImpl* instance;
std::mutex instance_mutex;
static constexpr int kMaxGroupId = 0xEF;
class DeviceGroup {
public:
DeviceGroup(int group_id, Uuid uuid)
: group_id_(group_id), group_uuid_(uuid) {}
void Add(const RawAddress& addr) { devices_.insert(addr); }
void Remove(const RawAddress& addr) { devices_.erase(addr); }
bool Contains(const RawAddress& addr) const {
return (devices_.count(addr) != 0);
}
void ForEachDevice(std::function<void(const RawAddress&)> cb) const {
for (auto const& addr : devices_) {
cb(addr);
}
}
int Size(void) const { return devices_.size(); }
int GetGroupId(void) const { return group_id_; }
const Uuid& GetUuid(void) const { return group_uuid_; }
private:
friend std::ostream& operator<<(std::ostream& out,
const bluetooth::groups::DeviceGroup& value);
int group_id_;
Uuid group_uuid_;
std::unordered_set<RawAddress> devices_;
};
class DeviceGroupsImpl : public DeviceGroups {
static constexpr uint8_t GROUP_STORAGE_CURRENT_LAYOUT_MAGIC = 0x10;
static constexpr size_t GROUP_STORAGE_HEADER_SZ =
sizeof(GROUP_STORAGE_CURRENT_LAYOUT_MAGIC) +
sizeof(uint8_t); /* num_of_groups */
static constexpr size_t GROUP_STORAGE_ENTRY_SZ =
sizeof(uint8_t) /* group_id */ + Uuid::kNumBytes128;
public:
DeviceGroupsImpl(DeviceGroupsCallbacks* callbacks) {
AddCallbacks(callbacks);
btif_storage_load_bonded_groups();
}
int GetGroupId(const RawAddress& addr, Uuid uuid) const override {
for (const auto& [id, g] : groups_) {
if ((g.Contains(addr)) && (uuid == g.GetUuid())) return id;
}
return kGroupUnknown;
}
void add_to_group(const RawAddress& addr, DeviceGroup* group) {
group->Add(addr);
bool first_device_in_group = (group->Size() == 1);
for (auto c : callbacks_) {
if (first_device_in_group) {
c->OnGroupAdded(addr, group->GetUuid(), group->GetGroupId());
} else {
c->OnGroupMemberAdded(addr, group->GetGroupId());
}
}
}
int AddDevice(const RawAddress& addr, Uuid uuid, int group_id) override {
DeviceGroup* group = nullptr;
if (group_id == kGroupUnknown) {
auto gid = GetGroupId(addr, uuid);
if (gid != kGroupUnknown) return gid;
group = create_group(uuid);
} else {
group = get_or_create_group_with_id(group_id, uuid);
if (!group) {
return kGroupUnknown;
}
}
LOG_ASSERT(group);
if (group->Contains(addr)) {
LOG(ERROR) << __func__ << " device " << ADDRESS_TO_LOGGABLE_STR(addr)
<< " already in the group: " << group_id;
return group->GetGroupId();
}
add_to_group(addr, group);
btif_storage_add_groups(addr);
return group->GetGroupId();
}
void RemoveDevice(const RawAddress& addr, int group_id) override {
int num_of_groups_dev_belongs = 0;
/* Remove from all the groups. Usually happens on unbond */
for (auto it = groups_.begin(); it != groups_.end();) {
auto& [id, g] = *it;
if (!g.Contains(addr)) {
++it;
continue;
}
num_of_groups_dev_belongs++;
if ((group_id != bluetooth::groups::kGroupUnknown) && (group_id != id)) {
++it;
continue;
}
num_of_groups_dev_belongs--;
g.Remove(addr);
for (auto c : callbacks_) {
c->OnGroupMemberRemoved(addr, id);
}
if (g.Size() == 0) {
for (auto c : callbacks_) {
c->OnGroupRemoved(g.GetUuid(), g.GetGroupId());
}
it = groups_.erase(it);
} else {
++it;
}
}
btif_storage_remove_groups(addr);
if (num_of_groups_dev_belongs > 0) {
btif_storage_add_groups(addr);
}
}
bool SerializeGroups(const RawAddress& addr,
std::vector<uint8_t>& out) const {
auto num_groups = std::count_if(
groups_.begin(), groups_.end(), [&addr](auto& id_group_pair) {
return id_group_pair.second.Contains(addr);
});
if ((num_groups == 0) || (num_groups > std::numeric_limits<uint8_t>::max()))
return false;
out.resize(GROUP_STORAGE_HEADER_SZ + (num_groups * GROUP_STORAGE_ENTRY_SZ));
auto* ptr = out.data();
/* header */
UINT8_TO_STREAM(ptr, GROUP_STORAGE_CURRENT_LAYOUT_MAGIC);
UINT8_TO_STREAM(ptr, num_groups);
/* group entries */
for (const auto& [id, g] : groups_) {
if (g.Contains(addr)) {
UINT8_TO_STREAM(ptr, id);
Uuid::UUID128Bit uuid128 = g.GetUuid().To128BitLE();
memcpy(ptr, uuid128.data(), Uuid::kNumBytes128);
ptr += Uuid::kNumBytes128;
}
}
return true;
}
void DeserializeGroups(const RawAddress& addr,
const std::vector<uint8_t>& in) {
if (in.size() < GROUP_STORAGE_HEADER_SZ + GROUP_STORAGE_ENTRY_SZ) return;
auto* ptr = in.data();
uint8_t magic;
STREAM_TO_UINT8(magic, ptr);
if (magic == GROUP_STORAGE_CURRENT_LAYOUT_MAGIC) {
uint8_t num_groups;
STREAM_TO_UINT8(num_groups, ptr);
if (in.size() <
GROUP_STORAGE_HEADER_SZ + (num_groups * GROUP_STORAGE_ENTRY_SZ)) {
LOG(ERROR) << "Invalid persistent storage data";
return;
}
/* group entries */
while (num_groups--) {
uint8_t id;
STREAM_TO_UINT8(id, ptr);
Uuid::UUID128Bit uuid128;
STREAM_TO_ARRAY(uuid128.data(), ptr, (int)Uuid::kNumBytes128);
auto* group =
get_or_create_group_with_id(id, Uuid::From128BitLE(uuid128));
if (group) add_to_group(addr, group);
for (auto c : callbacks_) {
c->OnGroupAddFromStorage(addr, Uuid::From128BitLE(uuid128), id);
}
}
}
}
void AddCallbacks(DeviceGroupsCallbacks* callbacks) {
callbacks_.push_back(std::move(callbacks));
/* Notify new user about known groups */
for (const auto& [id, g] : groups_) {
auto group_uuid = g.GetUuid();
auto group_id = g.GetGroupId();
g.ForEachDevice([&](auto& dev) {
callbacks->OnGroupAdded(dev, group_uuid, group_id);
});
}
}
bool Clear(DeviceGroupsCallbacks* callbacks) {
auto it = find_if(callbacks_.begin(), callbacks_.end(),
[callbacks](auto c) { return c == callbacks; });
if (it != callbacks_.end()) callbacks_.erase(it);
if (callbacks_.size() != 0) {
return false;
}
/* When all clients were unregistered */
groups_.clear();
return true;
}
void Dump(int fd) {
std::stringstream stream;
stream << " Num. registered clients: " << callbacks_.size() << std::endl;
stream << " Groups:\n";
for (const auto& kv_pair : groups_) {
stream << kv_pair.second << std::endl;
}
dprintf(fd, "%s", stream.str().c_str());
}
private:
DeviceGroup* find_device_group(int group_id) {
return groups_.count(group_id) ? &groups_.at(group_id) : nullptr;
}
DeviceGroup* get_or_create_group_with_id(int group_id, Uuid uuid) {
auto group = find_device_group(group_id);
if (group) {
if (group->GetUuid() != uuid) {
LOG(ERROR) << __func__ << " group " << group_id
<< " exists but for different uuid: " << group->GetUuid()
<< ", user request uuid: " << uuid;
return nullptr;
}
LOG(INFO) << __func__ << " group already exists: " << group_id;
return group;
}
DeviceGroup new_group(group_id, uuid);
groups_.insert({group_id, std::move(new_group)});
return &groups_.at(group_id);
}
DeviceGroup* create_group(Uuid& uuid) {
/* Generate new group id and return empty group */
/* Find first free id */
int group_id = -1;
for (int i = 1; i < kMaxGroupId; i++) {
if (groups_.count(i) == 0) {
group_id = i;
break;
}
}
if (group_id < 0) {
LOG(ERROR) << __func__ << " too many groups";
return nullptr;
}
DeviceGroup group(group_id, uuid);
groups_.insert({group_id, std::move(group)});
return &groups_.at(group_id);
}
std::map<int, DeviceGroup> groups_;
std::list<DeviceGroupsCallbacks*> callbacks_;
};
void DeviceGroups::Initialize(DeviceGroupsCallbacks* callbacks) {
std::scoped_lock<std::mutex> lock(instance_mutex);
if (instance == nullptr) {
instance = new DeviceGroupsImpl(callbacks);
return;
}
instance->AddCallbacks(callbacks);
}
void DeviceGroups::AddFromStorage(const RawAddress& addr,
const std::vector<uint8_t>& in) {
if (!instance) {
LOG(ERROR) << __func__ << ": Not initialized yet";
return;
}
instance->DeserializeGroups(addr, in);
}
bool DeviceGroups::GetForStorage(const RawAddress& addr,
std::vector<uint8_t>& out) {
if (!instance) {
LOG(ERROR) << __func__ << ": Not initialized yet";
return false;
}
return instance->SerializeGroups(addr, out);
}
void DeviceGroups::CleanUp(DeviceGroupsCallbacks* callbacks) {
std::scoped_lock<std::mutex> lock(instance_mutex);
if (!instance) return;
if (instance->Clear(callbacks)) {
delete (instance);
instance = nullptr;
}
}
std::ostream& operator<<(std::ostream& out,
bluetooth::groups::DeviceGroup const& group) {
out << " == Group id: " << group.group_id_ << " == \n"
<< " Uuid: " << group.group_uuid_ << std::endl;
out << " Devices:\n";
for (auto const& addr : group.devices_) {
out << " " << ADDRESS_TO_LOGGABLE_STR(addr) << std::endl;
}
return out;
}
void DeviceGroups::DebugDump(int fd) {
std::scoped_lock<std::mutex> lock(instance_mutex);
dprintf(fd, "Device Groups Manager:\n");
if (instance)
instance->Dump(fd);
else
dprintf(fd, " Not initialized \n");
}
DeviceGroups* DeviceGroups::Get() { return instance; }
} // namespace groups
} // namespace bluetooth