From c117987811986c5c03ce12435bf69c81976d8775 Mon Sep 17 00:00:00 2001
From: Andrew Beltrano <abeltrano@gmail.com>
Date: Sat, 27 Jul 2024 06:12:30 +0000
Subject: [PATCH] Extend WifiAccessPointsEnumerate to provide access point
 capabilities.

---
 protocol/protos/NetRemoteWifi.proto     | 1 +
 src/common/service/NetRemoteService.cxx | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/protocol/protos/NetRemoteWifi.proto b/protocol/protos/NetRemoteWifi.proto
index 9e905caa..4c3f3aaf 100644
--- a/protocol/protos/NetRemoteWifi.proto
+++ b/protocol/protos/NetRemoteWifi.proto
@@ -11,6 +11,7 @@ message WifiAccessPointsEnumerateResultItem
     string AccessPointId = 1;
     string MacAddress = 2;
     Microsoft.Net.Wifi.Dot11AccessPointCapabilities Capabilities = 3;
+    Microsoft.Net.Wifi.Dot11AccessPointAttributes Attributes = 5;
     bool IsEnabled = 4;
 }
 
diff --git a/src/common/service/NetRemoteService.cxx b/src/common/service/NetRemoteService.cxx
index f8e4eb83..4f158417 100644
--- a/src/common/service/NetRemoteService.cxx
+++ b/src/common/service/NetRemoteService.cxx
@@ -209,6 +209,7 @@ IAccessPointToNetRemoteAccessPointResultItem(IAccessPoint& accessPoint)
     id.assign(std::cbegin(interfaceName), std::cend(interfaceName));
 
     std::string macAddress = Ieee80211MacAddressToString(accessPoint.GetMacAddress());
+    auto dot11AccessPointAttributes = ToDot11AccessPointAttributes(accessPoint.GetAttributes());
 
     auto accessPointController = accessPoint.CreateController();
     if (accessPointController == nullptr) {
@@ -242,6 +243,7 @@ IAccessPointToNetRemoteAccessPointResultItem(IAccessPoint& accessPoint)
     item.set_macaddress(std::move(macAddress));
     item.set_isenabled(isEnabled);
     *item.mutable_capabilities() = std::move(dot11AccessPointCapabilities);
+    *item.mutable_attributes() = std::move(dot11AccessPointAttributes);
 
     return item;
 }