blob: 35d8452e58159f852dd03e3bec8511df7c397fec [file] [log] [blame]
/*
* Copyright 2019 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 "hci/facade/acl_manager_facade.h"
#include <condition_variable>
#include <memory>
#include <mutex>
#include "common/bind.h"
#include "grpc/grpc_event_queue.h"
#include "hci/acl_manager.h"
#include "hci/facade/acl_manager_facade.grpc.pb.h"
#include "hci/facade/acl_manager_facade.pb.h"
#include "hci/hci_packets.h"
#include "packet/raw_builder.h"
using ::grpc::ServerAsyncResponseWriter;
using ::grpc::ServerAsyncWriter;
using ::grpc::ServerContext;
using ::bluetooth::packet::RawBuilder;
namespace bluetooth {
namespace hci {
namespace facade {
class AclManagerFacadeService : public AclManagerFacade::Service,
public ::bluetooth::hci::ConnectionCallbacks,
public ::bluetooth::hci::ConnectionManagementCallbacks,
public ::bluetooth::hci::AclManagerCallbacks {
public:
AclManagerFacadeService(AclManager* acl_manager, ::bluetooth::os::Handler* facade_handler)
: acl_manager_(acl_manager), facade_handler_(facade_handler) {
acl_manager_->RegisterCallbacks(this, facade_handler_);
acl_manager_->RegisterAclManagerCallbacks(this, facade_handler_);
}
~AclManagerFacadeService() override {
std::unique_lock<std::mutex> lock(acl_connections_mutex_);
for (auto connection : acl_connections_) {
connection.second->GetAclQueueEnd()->UnregisterDequeue();
}
}
::grpc::Status CreateConnection(::grpc::ServerContext* context, const ConnectionMsg* request,
::grpc::ServerWriter<ConnectionEvent>* writer) override {
Address peer;
ASSERT(Address::FromString(request->address(), peer));
acl_manager_->CreateConnection(peer);
if (per_connection_events_.size() > current_connection_request_) {
return ::grpc::Status(::grpc::StatusCode::RESOURCE_EXHAUSTED, "Only one outstanding request is supported");
}
per_connection_events_.emplace_back(std::make_unique<::bluetooth::grpc::GrpcEventQueue<ConnectionEvent>>(
std::string("connection attempt ") + std::to_string(current_connection_request_)));
return per_connection_events_[current_connection_request_]->RunLoop(context, writer);
}
::grpc::Status Disconnect(::grpc::ServerContext* context, const HandleMsg* request,
::google::protobuf::Empty* response) override {
std::unique_lock<std::mutex> lock(acl_connections_mutex_);
auto connection = acl_connections_.find(request->handle());
if (connection == acl_connections_.end()) {
LOG_ERROR("Invalid handle");
return ::grpc::Status(::grpc::StatusCode::INVALID_ARGUMENT, "Invalid handle");
} else {
connection->second->Disconnect(DisconnectReason::REMOTE_USER_TERMINATED_CONNECTION);
return ::grpc::Status::OK;
}
}
::grpc::Status AuthenticationRequested(::grpc::ServerContext* context, const HandleMsg* request,
::google::protobuf::Empty* response) override {
std::unique_lock<std::mutex> lock(acl_connections_mutex_);
auto connection = acl_connections_.find(request->handle());
if (connection == acl_connections_.end()) {
LOG_ERROR("Invalid handle");
return ::grpc::Status(::grpc::StatusCode::INVALID_ARGUMENT, "Invalid handle");
} else {
connection->second->AuthenticationRequested();
return ::grpc::Status::OK;
}
};
::grpc::Status FetchIncomingConnection(::grpc::ServerContext* context, const google::protobuf::Empty* request,
::grpc::ServerWriter<ConnectionEvent>* writer) override {
if (per_connection_events_.size() > current_connection_request_) {
return ::grpc::Status(::grpc::StatusCode::RESOURCE_EXHAUSTED, "Only one outstanding connection is supported");
}
per_connection_events_.emplace_back(std::make_unique<::bluetooth::grpc::GrpcEventQueue<ConnectionEvent>>(
std::string("incoming connection ") + std::to_string(current_connection_request_)));
return per_connection_events_[current_connection_request_]->RunLoop(context, writer);
}
::grpc::Status SendAclData(::grpc::ServerContext* context, const AclData* request,
::google::protobuf::Empty* response) override {
std::promise<void> promise;
auto future = promise.get_future();
{
std::unique_lock<std::mutex> lock(acl_connections_mutex_);
auto connection = acl_connections_.find(request->handle());
if (connection == acl_connections_.end()) {
LOG_ERROR("Invalid handle");
return ::grpc::Status(::grpc::StatusCode::INVALID_ARGUMENT, "Invalid handle");
} else {
connection->second->GetAclQueueEnd()->RegisterEnqueue(
facade_handler_, common::Bind(&AclManagerFacadeService::enqueue_packet, common::Unretained(this),
common::Unretained(request), common::Passed(std::move(promise))));
}
}
future.wait();
return ::grpc::Status::OK;
}
std::unique_ptr<BasePacketBuilder> enqueue_packet(const AclData* request, std::promise<void> promise) {
acl_connections_[request->handle()]->GetAclQueueEnd()->UnregisterEnqueue();
std::unique_ptr<RawBuilder> packet =
std::make_unique<RawBuilder>(std::vector<uint8_t>(request->payload().begin(), request->payload().end()));
promise.set_value();
return packet;
}
::grpc::Status FetchAclData(::grpc::ServerContext* context, const ::google::protobuf::Empty* request,
::grpc::ServerWriter<AclData>* writer) override {
return pending_acl_data_.RunLoop(context, writer);
}
static inline uint16_t to_handle(uint32_t current_request) {
return (current_request + 0x10) % 0xe00;
}
static inline std::string builder_to_string(std::unique_ptr<BasePacketBuilder> builder) {
std::vector<uint8_t> bytes;
BitInserter bit_inserter(bytes);
builder->Serialize(bit_inserter);
return std::string(bytes.begin(), bytes.end());
}
void on_incoming_acl(std::shared_ptr<AclConnection> connection, uint16_t handle) {
auto packet = connection->GetAclQueueEnd()->TryDequeue();
AclData acl_data;
acl_data.set_handle(handle);
acl_data.set_payload(std::string(packet->begin(), packet->end()));
pending_acl_data_.OnIncomingEvent(acl_data);
}
void on_disconnect(std::shared_ptr<AclConnection> connection, uint32_t entry, ErrorCode code) {
connection->GetAclQueueEnd()->UnregisterDequeue();
connection->Finish();
std::unique_ptr<BasePacketBuilder> builder =
DisconnectBuilder::Create(to_handle(entry), static_cast<DisconnectReason>(code));
ConnectionEvent disconnection;
disconnection.set_event(builder_to_string(std::move(builder)));
per_connection_events_[entry]->OnIncomingEvent(disconnection);
}
void OnConnectSuccess(std::unique_ptr<::bluetooth::hci::AclConnection> connection) override {
std::unique_lock<std::mutex> lock(acl_connections_mutex_);
auto addr = connection->GetAddress();
std::shared_ptr<::bluetooth::hci::AclConnection> shared_connection = std::move(connection);
acl_connections_.emplace(to_handle(current_connection_request_), shared_connection);
auto remote_address = shared_connection->GetAddress().ToString();
shared_connection->GetAclQueueEnd()->RegisterDequeue(
facade_handler_, common::Bind(&AclManagerFacadeService::on_incoming_acl, common::Unretained(this),
shared_connection, to_handle(current_connection_request_)));
shared_connection->RegisterDisconnectCallback(
common::BindOnce(&AclManagerFacadeService::on_disconnect, common::Unretained(this), shared_connection,
current_connection_request_),
facade_handler_);
shared_connection->RegisterCallbacks(this, facade_handler_);
std::unique_ptr<BasePacketBuilder> builder = ConnectionCompleteBuilder::Create(
ErrorCode::SUCCESS, to_handle(current_connection_request_), addr, LinkType::ACL, Enable::DISABLED);
ConnectionEvent success;
success.set_event(builder_to_string(std::move(builder)));
per_connection_events_[current_connection_request_]->OnIncomingEvent(success);
current_connection_request_++;
}
void OnMasterLinkKeyComplete(uint16_t connection_handle, KeyFlag key_flag) override {
LOG_DEBUG("OnMasterLinkKeyComplete connection_handle:%d", connection_handle);
}
void OnRoleChange(Address bd_addr, Role new_role) override {
LOG_DEBUG("OnRoleChange bd_addr:%s, new_role:%d", bd_addr.ToString().c_str(), (uint8_t)new_role);
}
void OnReadDefaultLinkPolicySettingsComplete(uint16_t default_link_policy_settings) override {
LOG_DEBUG("OnReadDefaultLinkPolicySettingsComplete default_link_policy_settings:%d", default_link_policy_settings);
}
void OnConnectFail(Address address, ErrorCode reason) override {
std::unique_ptr<BasePacketBuilder> builder =
ConnectionCompleteBuilder::Create(reason, 0, address, LinkType::ACL, Enable::DISABLED);
ConnectionEvent fail;
fail.set_event(builder_to_string(std::move(builder)));
per_connection_events_[current_connection_request_]->OnIncomingEvent(fail);
current_connection_request_++;
}
void OnConnectionPacketTypeChanged(uint16_t packet_type) override {
LOG_DEBUG("OnConnectionPacketTypeChanged packet_type:%d", packet_type);
}
void OnAuthenticationComplete() override {
LOG_DEBUG("OnAuthenticationComplete");
}
void OnEncryptionChange(EncryptionEnabled enabled) override {
LOG_DEBUG("OnConnectionPacketTypeChanged enabled:%d", (uint8_t)enabled);
}
void OnChangeConnectionLinkKeyComplete() override {
LOG_DEBUG("OnChangeConnectionLinkKeyComplete");
};
void OnReadClockOffsetComplete(uint16_t clock_offset) override {
LOG_DEBUG("OnReadClockOffsetComplete clock_offset:%d", clock_offset);
};
void OnModeChange(Mode current_mode, uint16_t interval) override {
LOG_DEBUG("OnModeChange Mode:%d, interval:%d", (uint8_t)current_mode, interval);
};
void OnQosSetupComplete(ServiceType service_type, uint32_t token_rate, uint32_t peak_bandwidth, uint32_t latency,
uint32_t delay_variation) override {
LOG_DEBUG("OnQosSetupComplete service_type:%d, token_rate:%d, peak_bandwidth:%d, latency:%d, delay_variation:%d",
(uint8_t)service_type, token_rate, peak_bandwidth, latency, delay_variation);
}
void OnFlowSpecificationComplete(FlowDirection flow_direction, ServiceType service_type, uint32_t token_rate,
uint32_t token_bucket_size, uint32_t peak_bandwidth,
uint32_t access_latency) override {
LOG_DEBUG(
"OnFlowSpecificationComplete flow_direction:%d. service_type:%d, token_rate:%d, token_bucket_size:%d, "
"peak_bandwidth:%d, access_latency:%d",
(uint8_t)flow_direction, (uint8_t)service_type, token_rate, token_bucket_size, peak_bandwidth, access_latency);
}
void OnFlushOccurred() override {
LOG_DEBUG("OnFlushOccurred");
}
void OnRoleDiscoveryComplete(Role current_role) override {
LOG_DEBUG("OnRoleDiscoveryComplete current_role:%d", (uint8_t)current_role);
}
void OnReadLinkPolicySettingsComplete(uint16_t link_policy_settings) override {
LOG_DEBUG("OnReadLinkPolicySettingsComplete link_policy_settings:%d", link_policy_settings);
}
void OnReadAutomaticFlushTimeoutComplete(uint16_t flush_timeout) override {
LOG_DEBUG("OnReadAutomaticFlushTimeoutComplete flush_timeout:%d", flush_timeout);
}
void OnReadTransmitPowerLevelComplete(uint8_t transmit_power_level) override {
LOG_DEBUG("OnReadTransmitPowerLevelComplete transmit_power_level:%d", transmit_power_level);
}
void OnReadLinkSupervisionTimeoutComplete(uint16_t link_supervision_timeout) override {
LOG_DEBUG("OnReadLinkSupervisionTimeoutComplete link_supervision_timeout:%d", link_supervision_timeout);
}
void OnReadFailedContactCounterComplete(uint16_t failed_contact_counter) override {
LOG_DEBUG("OnReadFailedContactCounterComplete failed_contact_counter:%d", failed_contact_counter);
}
void OnReadLinkQualityComplete(uint8_t link_quality) override {
LOG_DEBUG("OnReadLinkQualityComplete link_quality:%d", link_quality);
}
void OnReadAfhChannelMapComplete(AfhMode afh_mode, std::array<uint8_t, 10> afh_channel_map) override {
LOG_DEBUG("OnReadAfhChannelMapComplete afh_mode:%d", (uint8_t)afh_mode);
}
void OnReadRssiComplete(uint8_t rssi) override {
LOG_DEBUG("OnReadRssiComplete rssi:%d", rssi);
}
void OnReadClockComplete(uint32_t clock, uint16_t accuracy) override {
LOG_DEBUG("OnReadClockComplete clock:%d, accuracy:%d", clock, accuracy);
}
private:
AclManager* acl_manager_;
::bluetooth::os::Handler* facade_handler_;
mutable std::mutex acl_connections_mutex_;
std::map<uint16_t, std::shared_ptr<AclConnection>> acl_connections_;
::bluetooth::grpc::GrpcEventQueue<AclData> pending_acl_data_{"FetchAclData"};
std::vector<std::unique_ptr<::bluetooth::grpc::GrpcEventQueue<ConnectionEvent>>> per_connection_events_;
uint32_t current_connection_request_{0};
};
void AclManagerFacadeModule::ListDependencies(ModuleList* list) {
::bluetooth::grpc::GrpcFacadeModule::ListDependencies(list);
list->add<AclManager>();
}
void AclManagerFacadeModule::Start() {
::bluetooth::grpc::GrpcFacadeModule::Start();
service_ = new AclManagerFacadeService(GetDependency<AclManager>(), GetHandler());
}
void AclManagerFacadeModule::Stop() {
delete service_;
::bluetooth::grpc::GrpcFacadeModule::Stop();
}
::grpc::Service* AclManagerFacadeModule::GetService() const {
return service_;
}
const ModuleFactory AclManagerFacadeModule::Factory =
::bluetooth::ModuleFactory([]() { return new AclManagerFacadeModule(); });
} // namespace facade
} // namespace hci
} // namespace bluetooth