mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-15 07:06:11 -04:00
373 lines
11 KiB
C++
373 lines
11 KiB
C++
#include "platform/impl/g3/wifi_lan.h"
|
|
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "platform/api/wifi_lan.h"
|
|
#include "platform/base/logging.h"
|
|
#include "platform/base/medium_environment.h"
|
|
#include "platform/base/prng.h"
|
|
#include "absl/synchronization/mutex.h"
|
|
|
|
namespace location {
|
|
namespace nearby {
|
|
namespace g3 {
|
|
|
|
WifiLanSocket::~WifiLanSocket() {
|
|
absl::MutexLock lock(&mutex_);
|
|
DoClose();
|
|
}
|
|
|
|
void WifiLanSocket::Connect(WifiLanSocket& other) {
|
|
absl::MutexLock lock(&mutex_);
|
|
remote_socket_ = &other;
|
|
input_ = other.output_;
|
|
}
|
|
|
|
InputStream& WifiLanSocket::GetInputStream() {
|
|
auto* remote_socket = GetRemoteSocket();
|
|
CHECK(remote_socket != nullptr);
|
|
return remote_socket->GetLocalInputStream();
|
|
}
|
|
|
|
OutputStream& WifiLanSocket::GetOutputStream() {
|
|
return GetLocalOutputStream();
|
|
}
|
|
|
|
WifiLanSocket* WifiLanSocket::GetRemoteSocket() {
|
|
absl::MutexLock lock(&mutex_);
|
|
return remote_socket_;
|
|
}
|
|
|
|
bool WifiLanSocket::IsConnected() const {
|
|
absl::MutexLock lock(&mutex_);
|
|
return IsConnectedLocked();
|
|
}
|
|
|
|
bool WifiLanSocket::IsClosed() const {
|
|
absl::MutexLock lock(&mutex_);
|
|
return closed_;
|
|
}
|
|
|
|
Exception WifiLanSocket::Close() {
|
|
absl::MutexLock lock(&mutex_);
|
|
DoClose();
|
|
return {Exception::kSuccess};
|
|
}
|
|
|
|
WifiLanService* WifiLanSocket::GetRemoteWifiLanService() {
|
|
absl::MutexLock lock(&mutex_);
|
|
return service_;
|
|
}
|
|
|
|
void WifiLanSocket::DoClose() {
|
|
if (!closed_) {
|
|
remote_socket_ = nullptr;
|
|
output_->GetOutputStream().Close();
|
|
output_->GetInputStream().Close();
|
|
if (IsConnectedLocked()) {
|
|
input_->GetOutputStream().Close();
|
|
input_->GetInputStream().Close();
|
|
}
|
|
closed_ = true;
|
|
}
|
|
}
|
|
|
|
bool WifiLanSocket::IsConnectedLocked() const { return input_ != nullptr; }
|
|
|
|
InputStream& WifiLanSocket::GetLocalInputStream() {
|
|
absl::MutexLock lock(&mutex_);
|
|
return output_->GetInputStream();
|
|
}
|
|
|
|
OutputStream& WifiLanSocket::GetLocalOutputStream() {
|
|
absl::MutexLock lock(&mutex_);
|
|
return output_->GetOutputStream();
|
|
}
|
|
|
|
std::unique_ptr<api::WifiLanSocket> WifiLanServerSocket::Accept(
|
|
WifiLanService* service) {
|
|
absl::MutexLock lock(&mutex_);
|
|
if (closed_) return {};
|
|
while (pending_sockets_.empty()) {
|
|
cond_.Wait(&mutex_);
|
|
if (closed_) break;
|
|
}
|
|
if (closed_) return {};
|
|
auto* remote_socket =
|
|
pending_sockets_.extract(pending_sockets_.begin()).value();
|
|
CHECK(remote_socket);
|
|
auto local_socket = std::make_unique<WifiLanSocket>(service);
|
|
local_socket->Connect(*remote_socket);
|
|
remote_socket->Connect(*local_socket);
|
|
cond_.SignalAll();
|
|
return local_socket;
|
|
}
|
|
|
|
bool WifiLanServerSocket::Connect(WifiLanSocket& socket) {
|
|
absl::MutexLock lock(&mutex_);
|
|
if (closed_) return false;
|
|
if (socket.IsConnected()) {
|
|
NEARBY_LOG(ERROR,
|
|
"Failed to connect to WifiLan server socket: already connected");
|
|
return true; // already connected.
|
|
}
|
|
// add client socket to the pending list
|
|
pending_sockets_.emplace(&socket);
|
|
cond_.SignalAll();
|
|
while (!socket.IsConnected()) {
|
|
cond_.Wait(&mutex_);
|
|
if (closed_) return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void WifiLanServerSocket::SetCloseNotifier(std::function<void()> notifier) {
|
|
absl::MutexLock lock(&mutex_);
|
|
close_notifier_ = std::move(notifier);
|
|
}
|
|
|
|
WifiLanServerSocket::~WifiLanServerSocket() {
|
|
absl::MutexLock lock(&mutex_);
|
|
DoClose();
|
|
}
|
|
|
|
Exception WifiLanServerSocket::Close() {
|
|
absl::MutexLock lock(&mutex_);
|
|
return DoClose();
|
|
}
|
|
|
|
Exception WifiLanServerSocket::DoClose() {
|
|
bool should_notify = !closed_;
|
|
closed_ = true;
|
|
if (should_notify) {
|
|
cond_.SignalAll();
|
|
if (close_notifier_) {
|
|
auto notifier = std::move(close_notifier_);
|
|
mutex_.Unlock();
|
|
// Notifier may contain calls to public API, and may cause deadlock, if
|
|
// mutex_ is held during the call.
|
|
notifier();
|
|
mutex_.Lock();
|
|
}
|
|
}
|
|
return {Exception::kSuccess};
|
|
}
|
|
|
|
WifiLanMedium::WifiLanMedium() {
|
|
service_.SetMedium(this);
|
|
std::string ip_address;
|
|
ip_address.resize(4);
|
|
uint32_t raw_ip_addr = Prng().NextUint32();
|
|
uint16_t port = Prng().NextUint32();
|
|
ip_address[0] = static_cast<char>(raw_ip_addr >> 24);
|
|
ip_address[1] = static_cast<char>(raw_ip_addr >> 16);
|
|
ip_address[2] = static_cast<char>(raw_ip_addr >> 8);
|
|
ip_address[3] = static_cast<char>(raw_ip_addr >> 0);
|
|
service_.SetServiceAddress(ip_address, port);
|
|
auto& env = MediumEnvironment::Instance();
|
|
env.RegisterWifiLanMedium(*this);
|
|
}
|
|
|
|
WifiLanMedium::~WifiLanMedium() {
|
|
service_.SetMedium(nullptr);
|
|
auto& env = MediumEnvironment::Instance();
|
|
env.UnregisterWifiLanMedium(*this);
|
|
|
|
StopAdvertising(advertising_info_.service_id);
|
|
StopDiscovery(discovering_info_.service_id);
|
|
|
|
NEARBY_LOG(INFO, "WifiLanMedium dtor advertising_accept_thread_running_ = %d",
|
|
acceptance_thread_running_.load());
|
|
// If acceptance thread is still running, wait to finish.
|
|
if (acceptance_thread_running_) {
|
|
while (acceptance_thread_running_) {
|
|
CountDownLatch latch(1);
|
|
close_accept_loops_runner_.Execute([&latch]() { latch.CountDown(); });
|
|
latch.Await();
|
|
}
|
|
}
|
|
}
|
|
|
|
bool WifiLanMedium::StartAdvertising(const std::string& service_id,
|
|
const std::string& service_info_name,
|
|
const std::string& endpoint_info_name) {
|
|
NEARBY_LOG(INFO,
|
|
"G3 WifiLan StartAdvertising: service_id=%s, service_info_name=%s",
|
|
service_id.c_str(), service_info_name.c_str());
|
|
auto& env = MediumEnvironment::Instance();
|
|
service_.SetServiceName(service_info_name);
|
|
service_.SetTxtRecord("n", endpoint_info_name);
|
|
env.UpdateWifiLanMediumForAdvertising(*this, service_, service_id, true);
|
|
|
|
absl::MutexLock lock(&mutex_);
|
|
if (server_socket_ != nullptr) server_socket_.release();
|
|
server_socket_ = std::make_unique<WifiLanServerSocket>();
|
|
|
|
acceptance_thread_running_.exchange(true);
|
|
accept_loops_runner_.Execute([&env, this, service_id]() mutable {
|
|
if (!accept_loops_runner_.InShutdown()) {
|
|
while (true) {
|
|
auto client_socket = server_socket_->Accept(&service_);
|
|
if (client_socket == nullptr) break;
|
|
env.CallWifiLanAcceptedConnectionCallback(
|
|
*this, *(client_socket.release()), service_id);
|
|
}
|
|
}
|
|
acceptance_thread_running_.exchange(false);
|
|
});
|
|
advertising_info_.service_id = service_id;
|
|
return true;
|
|
}
|
|
|
|
bool WifiLanMedium::StopAdvertising(const std::string& service_id) {
|
|
NEARBY_LOG(INFO, "G3 WifiLan StopAdvertising: service_id=%s",
|
|
service_id.c_str());
|
|
{
|
|
absl::MutexLock lock(&mutex_);
|
|
if (advertising_info_.Empty()) {
|
|
NEARBY_LOG(INFO,
|
|
"G3 WifiLan StopAdvertising: Can't stop advertising because "
|
|
"we never started advertising.");
|
|
return false;
|
|
}
|
|
advertising_info_.Clear();
|
|
}
|
|
|
|
auto& env = MediumEnvironment::Instance();
|
|
env.UpdateWifiLanMediumForAdvertising(*this, service_, service_id, false);
|
|
accept_loops_runner_.Shutdown();
|
|
if (server_socket_ == nullptr) {
|
|
NEARBY_LOGS(ERROR) << "G3 WifiLan StopAdvertising: failed to find WifiLan "
|
|
"Server socket: service_id="
|
|
<< service_id;
|
|
// Fall through for server socket not found.
|
|
return true;
|
|
}
|
|
|
|
if (!server_socket_->Close().Ok()) {
|
|
NEARBY_LOG(INFO,
|
|
"G3 WifiLan StopAdvertising: Failed to close WifiLan server "
|
|
"socket for %s.",
|
|
service_id.c_str());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool WifiLanMedium::StartDiscovery(const std::string& service_id,
|
|
DiscoveredServiceCallback callback) {
|
|
NEARBY_LOG(INFO, "G3 WifiLan StartDiscovery: service_id=%s",
|
|
service_id.c_str());
|
|
auto& env = MediumEnvironment::Instance();
|
|
env.UpdateWifiLanMediumForDiscovery(*this, service_id, std::move(callback),
|
|
true);
|
|
{
|
|
absl::MutexLock lock(&mutex_);
|
|
discovering_info_.service_id = service_id;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool WifiLanMedium::StopDiscovery(const std::string& service_id) {
|
|
NEARBY_LOG(INFO, "G3 WifiLan StopDiscovery: service_id=%s",
|
|
service_id.c_str());
|
|
{
|
|
absl::MutexLock lock(&mutex_);
|
|
if (discovering_info_.Empty()) {
|
|
NEARBY_LOG(INFO,
|
|
"G3 WifiLan StopDiscovery: Can't stop discovering because we "
|
|
"never started discovering.");
|
|
return false;
|
|
}
|
|
discovering_info_.Clear();
|
|
}
|
|
|
|
auto& env = MediumEnvironment::Instance();
|
|
env.UpdateWifiLanMediumForDiscovery(*this, service_id, {}, false);
|
|
return true;
|
|
}
|
|
|
|
bool WifiLanMedium::StartAcceptingConnections(
|
|
const std::string& service_id, AcceptedConnectionCallback callback) {
|
|
NEARBY_LOG(INFO, "G3 WifiLan StartAcceptingConnections: service_id=%s",
|
|
service_id.c_str());
|
|
auto& env = MediumEnvironment::Instance();
|
|
env.UpdateWifiLanMediumForAcceptedConnection(*this, service_id, callback);
|
|
return true;
|
|
}
|
|
|
|
bool WifiLanMedium::StopAcceptingConnections(const std::string& service_id) {
|
|
NEARBY_LOG(INFO, "G3 WifiLan StopAcceptingConnections: service_id=%s",
|
|
service_id.c_str());
|
|
auto& env = MediumEnvironment::Instance();
|
|
env.UpdateWifiLanMediumForAcceptedConnection(*this, service_id, {});
|
|
return true;
|
|
}
|
|
|
|
std::unique_ptr<api::WifiLanSocket> WifiLanMedium::Connect(
|
|
api::WifiLanService& remote_service, const std::string& service_id) {
|
|
NEARBY_LOG(INFO,
|
|
"G3 WifiLan Connect: medium=%p, service=%p, service_info_name=%s, "
|
|
"service_id=%s",
|
|
this, &service_, remote_service.GetServiceName().c_str(),
|
|
service_id.c_str());
|
|
// First, find an instance of remote medium, that exposed this service.
|
|
auto* medium = static_cast<WifiLanService&>(remote_service).GetMedium();
|
|
|
|
if (!medium) return {}; // Can't find medium. Bail out.
|
|
|
|
WifiLanServerSocket* remote_server_socket = nullptr;
|
|
NEARBY_LOG(INFO,
|
|
"G3 WifiLan Connect [peer]: medium=%p, service=%p, "
|
|
"service_info_name=%s, service_id=%s",
|
|
medium, &remote_service, remote_service.GetServiceName().c_str(),
|
|
service_id.c_str());
|
|
// Then, find our server socket context in this medium.
|
|
{
|
|
absl::MutexLock medium_lock(&medium->mutex_);
|
|
remote_server_socket = medium->server_socket_.get();
|
|
if (remote_server_socket == nullptr) {
|
|
NEARBY_LOG(ERROR,
|
|
"G3 WifiLan Connect: Failed to find WifiLan Server socket: "
|
|
"service_id=%s",
|
|
service_id.c_str());
|
|
// Fall through for server socket not found.
|
|
return {};
|
|
}
|
|
}
|
|
|
|
WifiLanService service = static_cast<WifiLanService&>(remote_service);
|
|
auto socket = std::make_unique<WifiLanSocket>(&service);
|
|
// Finally, Request to connect to this socket.
|
|
if (!remote_server_socket->Connect(*socket)) {
|
|
NEARBY_LOG(ERROR,
|
|
"G3 WifiLan Connect: Failed to connect to existing WifiLan "
|
|
"Server socket: service_id=%s",
|
|
service_id.c_str());
|
|
return {};
|
|
}
|
|
|
|
NEARBY_LOG(INFO, "G3 WifiLan Connect: connected: socket=%p", socket.get());
|
|
return socket;
|
|
}
|
|
|
|
api::WifiLanService* WifiLanMedium::FindRemoteService(
|
|
const std::string& ip_address, int port) {
|
|
auto& env = MediumEnvironment::Instance();
|
|
return env.FindWifiLanService(ip_address, port);
|
|
}
|
|
|
|
std::pair<std::string, int> WifiLanMedium::GetServiceAddress(
|
|
const std::string& service_id) {
|
|
NEARBY_LOGS(INFO) << "G3 WifiLan GetServiceAddress: service_id="
|
|
<< service_id;
|
|
return service_.GetServiceAddress();
|
|
}
|
|
|
|
} // namespace g3
|
|
} // namespace nearby
|
|
} // namespace location
|