mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-14 22:56:12 -04:00
113 lines
4.3 KiB
C++
113 lines
4.3 KiB
C++
// Copyright 2020 Google LLC
|
|
//
|
|
// 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
|
|
//
|
|
// https://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.
|
|
|
|
#ifndef THIRD_PARTY_NEARBY_PRESENCE_IMPLEMENTATION_SCAN_MANAGER_H_
|
|
#define THIRD_PARTY_NEARBY_PRESENCE_IMPLEMENTATION_SCAN_MANAGER_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "absl/base/thread_annotations.h"
|
|
#include "absl/container/flat_hash_map.h"
|
|
#include "absl/strings/string_view.h"
|
|
#include "internal/platform/implementation/ble_v2.h"
|
|
#include "internal/platform/mutex.h"
|
|
#include "internal/platform/mutex_lock.h"
|
|
#include "internal/platform/runnable.h"
|
|
#include "internal/platform/single_thread_executor.h"
|
|
#include "internal/proto/credential.pb.h"
|
|
#include "presence/data_types.h"
|
|
#include "presence/implementation/advertisement_filter.h"
|
|
#include "presence/implementation/credential_manager.h"
|
|
#include "presence/implementation/mediums/mediums.h"
|
|
#include "presence/scan_request.h"
|
|
|
|
#ifdef USE_RUST_DECODER
|
|
#include "presence/implementation/advertisement_decoder_rust_impl.h"
|
|
#else
|
|
#include "presence/implementation/advertisement_decoder_impl.h"
|
|
#endif
|
|
|
|
|
|
namespace nearby {
|
|
namespace presence {
|
|
|
|
// The instance of ScanManager is owned by `ServiceControllerImpl`.
|
|
// Helping service controller to manage scan requests and callbacks.
|
|
class ScanManager {
|
|
public:
|
|
using SingleThreadExecutor = ::nearby::SingleThreadExecutor;
|
|
using Mutex = ::nearby::Mutex;
|
|
using MutexLock = ::nearby::MutexLock;
|
|
using ScanningSession = ::nearby::api::ble_v2::BleMedium::ScanningSession;
|
|
using Runnable = ::nearby::Runnable;
|
|
using BleAdvertisementData = ::nearby::api::ble_v2::BleAdvertisementData;
|
|
using SharedCredential = ::nearby::internal::SharedCredential;
|
|
using IdentityType = ::nearby::internal::IdentityType;
|
|
|
|
ScanManager(Mediums& mediums, CredentialManager& credential_manager,
|
|
SingleThreadExecutor& executor) {
|
|
mediums_ = &mediums, credential_manager_ = &credential_manager;
|
|
executor_ = &executor;
|
|
}
|
|
~ScanManager() = default;
|
|
|
|
ScanSessionId StartScan(ScanRequest scan_request, ScanCallback cb);
|
|
void StopScan(ScanSessionId session_id);
|
|
// Below functions are test only.
|
|
// Reference: go/totw/135#augmenting-the-public-api-for-tests
|
|
int ScanningCallbacksLengthForTest();
|
|
|
|
private:
|
|
struct ScanSessionState {
|
|
ScanRequest request;
|
|
ScanCallback callback;
|
|
absl::flat_hash_map<IdentityType, std::vector<SharedCredential>>
|
|
credentials;
|
|
AdvertisementDecoderImpl decoder;
|
|
AdvertisementFilter advertisement_filter;
|
|
std::unique_ptr<ScanningSession> scanning_session;
|
|
};
|
|
void NotifyFoundBle(
|
|
ScanSessionId id, BleAdvertisementData data,
|
|
nearby::api::ble_v2::BlePeripheral::UniqueId peripheral_id)
|
|
ABSL_EXCLUSIVE_LOCKS_REQUIRED(*executor_);
|
|
void NotifyLostBle(ScanSessionId id,
|
|
nearby::api::ble_v2::BlePeripheral::UniqueId peripheral_id)
|
|
ABSL_EXCLUSIVE_LOCKS_REQUIRED(*executor_);
|
|
void FetchCredentials(ScanSessionId id, const ScanRequest& scan_request)
|
|
ABSL_EXCLUSIVE_LOCKS_REQUIRED(*executor_);
|
|
void UpdateCredentials(ScanSessionId id, IdentityType identity_type,
|
|
std::vector<SharedCredential> credentials)
|
|
ABSL_EXCLUSIVE_LOCKS_REQUIRED(*executor_);
|
|
void RunOnServiceControllerThread(absl::string_view name, Runnable runnable) {
|
|
executor_->Execute(std::string(name), std::move(runnable));
|
|
}
|
|
Mediums* mediums_;
|
|
CredentialManager* credential_manager_;
|
|
absl::flat_hash_map<ScanSessionId, ScanSessionState> scan_sessions_
|
|
ABSL_GUARDED_BY(*executor_);
|
|
absl::flat_hash_map<nearby::api::ble_v2::BlePeripheral::UniqueId, std::string>
|
|
device_unique_id_to_endpoint_id_map_
|
|
ABSL_GUARDED_BY(*executor_);
|
|
SingleThreadExecutor* executor_;
|
|
};
|
|
|
|
} // namespace presence
|
|
} // namespace nearby
|
|
|
|
#endif // THIRD_PARTY_NEARBY_PRESENCE_IMPLEMENTATION_SCAN_MANAGER_H_
|