| // 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 CORE_INTERNAL_WIFI_LAN_BWU_HANDLER_H_ |
| #define CORE_INTERNAL_WIFI_LAN_BWU_HANDLER_H_ |
| |
| #include "connections/implementation/base_bwu_handler.h" |
| #include "connections/implementation/client_proxy.h" |
| #include "connections/implementation/endpoint_channel_manager.h" |
| #include "connections/implementation/mediums/mediums.h" |
| |
| namespace location { |
| namespace nearby { |
| namespace connections { |
| |
| // Defines the set of methods that need to be implemented to handle the |
| // per-Medium-specific operations needed to upgrade an EndpointChannel. |
| class WifiLanBwuHandler : public BaseBwuHandler { |
| public: |
| WifiLanBwuHandler(Mediums& mediums, EndpointChannelManager& channel_manager, |
| BwuNotifications notifications); |
| ~WifiLanBwuHandler() override = default; |
| |
| private: |
| ByteArray InitializeUpgradedMediumForEndpoint( |
| ClientProxy* client, const std::string& service_id, |
| const std::string& endpoint_id) override; |
| |
| void Revert() override; |
| |
| std::unique_ptr<EndpointChannel> CreateUpgradedEndpointChannel( |
| ClientProxy* client, const std::string& service_id, |
| const std::string& endpoint_id, |
| const UpgradePathInfo& upgrade_path_info) override; |
| |
| Medium GetUpgradeMedium() const override { return Medium::WIFI_LAN; } |
| |
| void OnEndpointDisconnect(ClientProxy* client, |
| const std::string& endpoint_id) override {} |
| |
| void OnIncomingWifiLanConnection(ClientProxy* client, |
| const std::string& service_id, |
| WifiLanSocket socket); |
| |
| class WifiLanIncomingSocket : public BwuHandler::IncomingSocket { |
| public: |
| explicit WifiLanIncomingSocket(const std::string& name, |
| WifiLanSocket socket) |
| : name_(name), socket_(socket) {} |
| ~WifiLanIncomingSocket() override = default; |
| |
| std::string ToString() override { return name_; } |
| void Close() override { socket_.Close(); } |
| |
| private: |
| std::string name_; |
| WifiLanSocket socket_; |
| }; |
| |
| Mediums& mediums_; |
| WifiLan& wifi_lan_medium_{mediums_.GetWifiLan()}; |
| absl::flat_hash_set<std::string> active_service_ids_; |
| }; |
| |
| } // namespace connections |
| } // namespace nearby |
| } // namespace location |
| |
| #endif // CORE_INTERNAL_WIFI_LAN_BWU_HANDLER_H_ |