From ded3169aa5395c8f5b6a2e490cc693218f1c4cf0 Mon Sep 17 00:00:00 2001
From: aditi_lonkar <aditi.lonkar@espressif.com>
Date: Wed, 29 Mar 2023 15:23:02 +0530
Subject: [PATCH] esp_wifi: Add APIs to get Assoc id and negotiated phymode.

---
 components/esp_wifi/include/esp_wifi.h       | 23 ++++++++++++++++++++
 components/esp_wifi/include/esp_wifi_types.h | 13 +++++++++++
 2 files changed, 36 insertions(+)

diff --git a/components/esp_wifi/include/esp_wifi.h b/components/esp_wifi/include/esp_wifi.h
index 87736e460d..1517f81d7b 100644
--- a/components/esp_wifi/include/esp_wifi.h
+++ b/components/esp_wifi/include/esp_wifi.h
@@ -1319,6 +1319,29 @@ esp_err_t esp_wifi_config_80211_tx_rate(wifi_interface_t ifx, wifi_phy_rate_t ra
   */
 esp_err_t esp_wifi_disable_pmf_config(wifi_interface_t ifx);
 
+/**
+  * @brief     Get the Association id assigned to STA by AP
+  *
+  * @param[out] aid  store the aid
+  *
+  * @attention aid = 0 if station is not connected to AP.
+  *
+  * @return
+  *    - ESP_OK: succeed
+  */
+esp_err_t esp_wifi_sta_get_aid(uint16_t *aid);
+
+/**
+  * @brief     Get the negotiated phymode after connection.
+  *
+  * @param[out] phymode  store the negotiated phymode.
+  *
+  * @attention Operation phy mode, BIT[5]: indicate whether LR enabled, BIT[0-4]: wifi_phy_mode_t
+  *
+  * @return
+  *    - ESP_OK: succeed
+  */
+esp_err_t esp_wifi_sta_get_negotiated_phymode(wifi_phy_mode_t *phymode);
 #ifdef __cplusplus
 }
 #endif
diff --git a/components/esp_wifi/include/esp_wifi_types.h b/components/esp_wifi/include/esp_wifi_types.h
index 9d945aa34b..19aa667a37 100644
--- a/components/esp_wifi/include/esp_wifi_types.h
+++ b/components/esp_wifi/include/esp_wifi_types.h
@@ -359,6 +359,19 @@ typedef enum {
 
 #define WIFI_VENDOR_IE_ELEMENT_ID 0xDD
 
+/**
+  * @brief     Operation Phymode
+  */
+typedef enum
+{
+    WIFI_PHY_MODE_LR,   /**< PHY mode for Low Rate */
+    WIFI_PHY_MODE_11B,  /**< PHY mode for 11b */
+    WIFI_PHY_MODE_11G,  /**< PHY mode for 11g */
+    WIFI_PHY_MODE_HT20, /**< PHY mode for Bandwidth HT20 */
+    WIFI_PHY_MODE_HT40, /**< PHY mode for Bandwidth HT40 */
+    WIFI_PHY_MODE_HE20, /**< PHY mode for Bandwidth HE20 */
+} wifi_phy_mode_t;
+
 /**
  * @brief Vendor Information Element header
  *