From 62a060588d003f55180512be45dbac15f99c519c Mon Sep 17 00:00:00 2001
From: Blake Harnden <32446120+bharnden@users.noreply.github.com>
Date: Tue, 19 Apr 2022 16:19:19 -0700
Subject: [PATCH] grpc: update to make sure retrieving session data returns
 wireless configs for nodes

---
 daemon/core/api/grpc/grpcutils.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/daemon/core/api/grpc/grpcutils.py b/daemon/core/api/grpc/grpcutils.py
index 05cdd3cd..c6e375f9 100644
--- a/daemon/core/api/grpc/grpcutils.py
+++ b/daemon/core/api/grpc/grpcutils.py
@@ -307,6 +307,10 @@ def get_node_proto(
     )
     if wlan_config:
         wlan_config = get_config_options(wlan_config, BasicRangeModel)
+    # check for wireless config
+    wireless_config = None
+    if isinstance(node, WirelessNode):
+        wireless_config = node.get_config()
     # check for mobility config
     mobility_config = session.mobility.get_configs(
         node.id, config_type=Ns2ScriptedMobility.name
@@ -353,6 +357,7 @@ def get_node_proto(
         channel=channel,
         canvas=node.canvas,
         wlan_config=wlan_config,
+        wireless_config=wireless_config,
         mobility_config=mobility_config,
         service_configs=service_configs,
         config_service_configs=config_service_configs,