drivers: wifi: infineon: add .iface_status method

The .iface_status method of the wifi_mgmt_ops API needs to be added
so that the "wifi status" command on the network shell will work.

Signed-off-by: Dave Rensberger <[email protected]>

Author: drensber

drivers/wifi/infineon/airoc_wifi.c

@@ -13,6 +13,7 @@
 #include <zephyr/net/conn_mgr/connectivity_wifi_mgmt.h>
 #include <airoc_wifi.h>
 #include <airoc_whd_hal_common.h>
+#include <whd_wlioctl.h>
 
 LOG_MODULE_REGISTER(infineon_airoc_wifi, CONFIG_WIFI_LOG_LEVEL);
 
@@ -25,7 +26,7 @@ LOG_MODULE_REGISTER(infineon_airoc_wifi, CONFIG_WIFI_LOG_LEVEL);
 #endif
 
 #ifndef AIROC_WIFI_PACKET_POOL_SIZE
-#define AIROC_WIFI_PACKET_POOL_SIZE     (1600)
+#define AIROC_WIFI_PACKET_POOL_SIZE (1600)
 #endif
 
 #define AIROC_WIFI_PACKET_POOL_COUNT                                                               \
@@ -49,8 +50,8 @@ int airoc_wifi_init_primary(const struct device *dev, whd_interface_t *interface
 			    whd_netif_funcs_t *netif_funcs, whd_buffer_funcs_t *buffer_if);
 
 /* Allocate network pool */
-NET_BUF_POOL_FIXED_DEFINE(airoc_pool, AIROC_WIFI_PACKET_POOL_COUNT,
-				AIROC_WIFI_PACKET_POOL_SIZE, 0, NULL);
+NET_BUF_POOL_FIXED_DEFINE(airoc_pool, AIROC_WIFI_PACKET_POOL_COUNT, AIROC_WIFI_PACKET_POOL_SIZE, 0,
+			  NULL);
 
 /* AIROC globals */
 static uint16_t ap_event_handler_index = 0xFF;
@@ -236,7 +237,7 @@ static whd_result_t airoc_wifi_host_buffer_get(whd_buffer_t *buffer, whd_buffer_
 	*buffer = buf;
 
 	/* Set buffer size */
-	(void) airoc_wifi_buffer_set_size(*buffer, size);
+	(void)airoc_wifi_buffer_set_size(*buffer, size);
 
 	return WHD_SUCCESS;
 }
@@ -302,7 +303,7 @@ static int airoc_mgmt_send(const struct device *dev, struct net_pkt *pkt)
 	}
 
 	/* Allocate Network Buffer from pool with Packet Length + Data Header */
-	ret = airoc_wifi_host_buffer_get((whd_buffer_t *) &buf, WHD_NETWORK_TX,
+	ret = airoc_wifi_host_buffer_get((whd_buffer_t *)&buf, WHD_NETWORK_TX,
 					 pkt_len + sizeof(data_header_t), 0);
 	if ((ret != WHD_SUCCESS) || (buf == NULL)) {
 		return -EIO;
@@ -381,8 +382,7 @@ static enum ethernet_hw_caps airoc_get_capabilities(const struct device *dev)
 	return ETHERNET_HW_FILTERING;
 }
 
-static int airoc_set_config(const struct device *dev,
-			    enum ethernet_config_type type,
+static int airoc_set_config(const struct device *dev, enum ethernet_config_type type,
 			    const struct ethernet_config *config)
 {
 	ARG_UNUSED(dev);
@@ -602,10 +602,8 @@ static void *airoc_wifi_ap_link_events_handler(whd_interface_t ifp,
 					       const whd_event_header_t *event_header,
 					       const uint8_t *event_data, void *handler_user_data)
 {
-	struct airoc_wifi_event_t airoc_event = {
-		.is_ap_event = 1,
-		.event_type = event_header->event_type
-	};
+	struct airoc_wifi_event_t airoc_event = {.is_ap_event = 1,
+						 .event_type = event_header->event_type};
 
 	k_msgq_put(&airoc_wifi_msgq, &airoc_event, K_FOREVER);
 
@@ -658,8 +656,7 @@ static int airoc_mgmt_ap_enable(const struct device *dev, struct wifi_connect_re
 		channel = params->channel;
 	} else {
 		channel = 1;
-		LOG_WRN("Discard of setting unsupported channel: %u (will set 1)",
-			params->channel);
+		LOG_WRN("Discard of setting unsupported channel: %u (will set 1)", params->channel);
 	}
 
 	switch (params->security) {
@@ -760,6 +757,80 @@ static int airoc_mgmt_ap_disable(const struct device *dev)
 	return 0;
 }
 
+static int airoc_iface_status(const struct device *dev, struct wifi_iface_status *status)
+{
+	struct airoc_wifi_data *data = dev->data;
+	whd_result_t result;
+	wl_bss_info_t bss_info;
+	whd_security_t security_info = 0;
+	uint32_t wpa_data_rate_value = 0;
+	uint32_t join_status;
+
+	if (airoc_if == NULL) {
+		return -ENOTSUP;
+	}
+
+	status->iface_mode =
+		(data->is_ap_up ? WIFI_MODE_AP
+				: (data->is_sta_connected ? WIFI_MODE_INFRA : WIFI_MODE_UNKNOWN));
+
+	join_status = whd_wifi_is_ready_to_transceive(airoc_if);
+
+	if (join_status == WHD_SUCCESS) {
+		status->state = WIFI_STATE_COMPLETED;
+	} else if (join_status == WHD_JOIN_IN_PROGRESS) {
+		status->state = WIFI_STATE_ASSOCIATING;
+	} else if (join_status == WHD_NOT_KEYED) {
+		status->state = WIFI_STATE_AUTHENTICATING;
+	} else {
+		status->state = WIFI_STATE_DISCONNECTED;
+	}
+
+	result = whd_wifi_get_ap_info(airoc_if, &bss_info, &security_info);
+
+	if (result == WHD_SUCCESS) {
+		memcpy(&(status->bssid[0]), &(bss_info.BSSID), sizeof(whd_mac_t));
+
+		whd_wifi_get_channel(airoc_if, (int *)&status->channel);
+
+		status->band = (status->channel <= CH_MAX_2G_CHANNEL) ? WIFI_FREQ_BAND_2_4_GHZ
+								      : WIFI_FREQ_BAND_5_GHZ;
+
+		status->rssi = (int)bss_info.RSSI;
+
+		status->ssid_len = bss_info.SSID_len;
+		strncpy(status->ssid, bss_info.SSID, bss_info.SSID_len);
+
+		status->security = convert_whd_security_to_zephyr(security_info);
+
+		status->beacon_interval = (unsigned short)bss_info.beacon_period;
+		status->dtim_period = (unsigned char)bss_info.dtim_period;
+
+		status->twt_capable = false;
+	}
+
+	whd_wifi_get_ioctl_value(airoc_if, WLC_GET_RATE, &wpa_data_rate_value);
+	status->current_phy_tx_rate = wpa_data_rate_value;
+
+	/* Unbelievably, this appears to be the only way to determine the phy mode with
+	 *  the whd SDK that we're currently using. Note that the logic below is only valid on
+	 *  devices that are limited to the 2.4Ghz band. Other versions of the SDK and chip
+	 * evidently allow one to obtain a phy_mode value directly from bss_info
+	 */
+	if (wpa_data_rate_value > 54) {
+		status->link_mode = WIFI_4;
+	} else if (wpa_data_rate_value == 6 || wpa_data_rate_value == 9 ||
+		   wpa_data_rate_value == 12 || wpa_data_rate_value == 18 ||
+		   wpa_data_rate_value == 24 || wpa_data_rate_value == 36 ||
+		   wpa_data_rate_value == 48 || wpa_data_rate_value == 54) {
+		status->link_mode = WIFI_3;
+	} else {
+		status->link_mode = WIFI_1;
+	}
+
+	return 0;
+}
+
 static int airoc_init(const struct device *dev)
 {
 	int ret;
@@ -785,8 +856,8 @@ static int airoc_init(const struct device *dev)
 	}
 	airoc_if = airoc_sta_if;
 
-	whd_ret = whd_management_set_event_handler(airoc_sta_if, sta_link_events,
-				link_events_handler, NULL, &sta_event_handler_index);
+	whd_ret = whd_management_set_event_handler(
+		airoc_sta_if, sta_link_events, link_events_handler, NULL, &sta_event_handler_index);
 	if (whd_ret != CY_RSLT_SUCCESS) {
 		LOG_ERR("whd_management_set_event_handler failed ret = %d \r\n", whd_ret);
 		return -EAGAIN;
@@ -813,6 +884,7 @@ static const struct wifi_mgmt_ops airoc_wifi_mgmt = {
 	.disconnect = airoc_mgmt_disconnect,
 	.ap_enable = airoc_mgmt_ap_enable,
 	.ap_disable = airoc_mgmt_ap_disable,
+	.iface_status = airoc_iface_status,
 #if defined(CONFIG_NET_STATISTICS_WIFI)
 	.get_stats = airoc_mgmt_wifi_stats,
 #endif