From b6bd52d31723bd00d52c7bdd758556abb34fd8a8 Mon Sep 17 00:00:00 2001
From: Christian B <christian@aircrack-ng.org>
Date: Thu, 14 Feb 2019 20:28:49 +0100
Subject: [PATCH] Add check for psta->aid out of bounds.

---
 core/rtw_sta_mgt.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/core/rtw_sta_mgt.c b/core/rtw_sta_mgt.c
index a2a0781..69d63b0 100644
--- a/core/rtw_sta_mgt.c
+++ b/core/rtw_sta_mgt.c
@@ -775,6 +775,10 @@ u32	rtw_free_stainfo(_adapter *padapter , struct sta_info *psta)
 			pstapriv->sta_aid[psta->cmn.aid - 1] = NULL;
 			psta->cmn.aid = 0;
 		}
+		if (psta->aid > 31) {
+			pr_err("***** psta->aid (%d) out of bounds\n", psta->aid);
+			return _FAIL;
+		}
 	}
 
 #endif /* CONFIG_NATIVEAP_MLME	 */