X-Git-Url: https://jxself.org/git/?p=carl9170fw.git;a=blobdiff_plain;f=carlfw%2Fsrc%2Fwlan.c;h=947652486205e9092689327a08b5000e3bc56c63;hp=8bc3d4fa4dcc21cdb493d05357c7f941aa72dc1d;hb=0c5091c6bc2a036c1eb3272db0768ecb84559671;hpb=84b0f7825de5dbf4d1af761fe205ec77d2680e7a

diff --git a/carlfw/src/wlan.c b/carlfw/src/wlan.c
index 8bc3d4f..9476524 100644
--- a/carlfw/src/wlan.c
+++ b/carlfw/src/wlan.c
@@ -625,16 +625,11 @@ void wlan_cab_flush_queue(const unsigned int vif)
 	}
 }
 
-static uint8_t *beacon_find_ie(uint8_t ie)
+static uint8_t *beacon_find_ie(uint8_t ie, void *addr,
+			       const unsigned int len)
 {
-	struct ieee80211_mgmt *mgmt = getp(AR9170_MAC_REG_BCN_ADDR);
+	struct ieee80211_mgmt *mgmt = addr;
 	uint8_t *pos, *end;
-	unsigned int len;
-
-	len = get(AR9170_MAC_REG_BCN_LENGTH);
-
-	if (len < FCS_LEN + sizeof(mgmt))
-		return NULL;
 
 	pos = mgmt->u.beacon.variable;
 	end = (uint8_t *) ((unsigned long)mgmt + (len - FCS_LEN));
@@ -651,12 +646,13 @@ static uint8_t *beacon_find_ie(uint8_t ie)
 	return NULL;
 }
 
-void wlan_cab_modify_dtim_beacon(const unsigned int vif)
+void wlan_cab_modify_dtim_beacon(const unsigned int vif,
+	const unsigned int addr, const unsigned int len)
 {
 	uint8_t *_ie;
 	struct ieee80211_tim_ie *ie;
 
-	_ie = beacon_find_ie(WLAN_EID_TIM);
+	_ie = beacon_find_ie(WLAN_EID_TIM, (void *)addr, len);
 	if (likely(_ie)) {
 		ie = (struct ieee80211_tim_ie *) &_ie[2];