mirror of https://github.com/hak5/openwrt-owl.git
57 lines
1.4 KiB
Diff
57 lines
1.4 KiB
Diff
|
--- a/src/common/wpa_common.c
|
||
|
|
+++ b/src/common/wpa_common.c
|
||
|
|
@@ -811,6 +811,31 @@ const char * wpa_key_mgmt_txt(int key_mg
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
+static void wpa_fixup_wpa_ie_rsn(u8 *assoc_ie, const u8 *wpa_msg_ie,
|
||
|
|
+ size_t rsn_ie_len)
|
||
|
|
+{
|
||
|
|
+ int pos, count;
|
||
|
|
+
|
||
|
|
+ pos = sizeof(struct rsn_ie_hdr) + RSN_SELECTOR_LEN;
|
||
|
|
+ if (rsn_ie_len < pos + 2)
|
||
|
|
+ return;
|
||
|
|
+
|
||
|
|
+ count = WPA_GET_LE16(wpa_msg_ie + pos);
|
||
|
|
+ pos += 2 + count * RSN_SELECTOR_LEN;
|
||
|
|
+ if (rsn_ie_len < pos + 2)
|
||
|
|
+ return;
|
||
|
|
+
|
||
|
|
+ count = WPA_GET_LE16(wpa_msg_ie + pos);
|
||
|
|
+ pos += 2 + count * RSN_SELECTOR_LEN;
|
||
|
|
+ if (rsn_ie_len < pos + 2)
|
||
|
|
+ return;
|
||
|
|
+
|
||
|
|
+ if (!assoc_ie[pos] && !assoc_ie[pos + 1] &&
|
||
|
|
+ (wpa_msg_ie[pos] || wpa_msg_ie[pos + 1]))
|
||
|
|
+ memcpy(&assoc_ie[pos], &wpa_msg_ie[pos], 2);
|
||
|
|
+}
|
||
|
|
+
|
||
|
|
+
|
||
|
|
int wpa_compare_rsn_ie(int ft_initial_assoc,
|
||
|
|
const u8 *ie1, size_t ie1len,
|
||
|
|
const u8 *ie2, size_t ie2len)
|
||
|
|
@@ -818,8 +843,19 @@ int wpa_compare_rsn_ie(int ft_initial_as
|
||
|
|
if (ie1 == NULL || ie2 == NULL)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
- if (ie1len == ie2len && os_memcmp(ie1, ie2, ie1len) == 0)
|
||
|
|
- return 0; /* identical IEs */
|
||
|
|
+ if (ie1len == ie2len) {
|
||
|
|
+ u8 *ie_tmp;
|
||
|
|
+
|
||
|
|
+ if (os_memcmp(ie1, ie2, ie1len) == 0)
|
||
|
|
+ return 0; /* identical IEs */
|
||
|
|
+
|
||
|
|
+ ie_tmp = alloca(ie1len);
|
||
|
|
+ memcpy(ie_tmp, ie1, ie1len);
|
||
|
|
+ wpa_fixup_wpa_ie_rsn(ie_tmp, ie2, ie1len);
|
||
|
|
+
|
||
|
|
+ if (os_memcmp(ie_tmp, ie2, ie1len) == 0)
|
||
|
|
+ return 0; /* only mismatch in RSN capabilties */
|
||
|
|
+ }
|
||
|
|
|
||
|
|
#ifdef CONFIG_IEEE80211R
|
||
|
|
if (ft_initial_assoc) {
|