MLD STA: Store PMKSA with AP MLD address for MLO connection event

Store PMKSA with AP MLD address while processing connect event for OWE
and FILS when the connection is MLO capable.

Signed-off-by: Veerendranath Jakkam <quic_vjakkam@quicinc.com>

diff --git a/wpa_supplicant/events.c b/wpa_supplicant/events.c
index b0648780525be0754684418419e4f06f92805168..146191d03429b2a2d4aa415505cc12958b9acfec 100644 (file)
--- a/wpa_supplicant/events.c
+++ b/wpa_supplicant/events.c
@@ -3167,7 +3167,9 @@ static int wpa_supplicant_event_associnfo(struct wpa_supplicant *wpa_s,
 #ifdef CONFIG_OWE
        if (wpa_s->key_mgmt == WPA_KEY_MGMT_OWE &&
            (!bssid_known ||
-            owe_process_assoc_resp(wpa_s->wpa, bssid,
+            owe_process_assoc_resp(wpa_s->wpa,
+                                   wpa_s->valid_links ?
+                                   wpa_s->ap_mld_addr : bssid,
                                    data->assoc_info.resp_ies,
                                    data->assoc_info.resp_ies_len) < 0)) {
                wpa_supplicant_deauthenticate(wpa_s, WLAN_REASON_UNSPECIFIED);
@@ -4960,7 +4962,10 @@ static void wpa_supplicant_event_assoc_auth(struct wpa_supplicant *wpa_s,
                                               data->assoc_info.fils_pmk,
                                               data->assoc_info.fils_pmk_len,
                                               data->assoc_info.fils_pmkid,
-                                              wpa_s->bssid, fils_cache_id);
+                                              wpa_s->valid_links ?
+                                              wpa_s->ap_mld_addr :
+                                              wpa_s->bssid,
+                                              fils_cache_id);
                } else if (data->assoc_info.fils_pmkid) {
                        /* Update the current PMKSA used for this connection */
                        pmksa_cache_set_current(wpa_s->wpa,