cfg80211: allow finding vendor with OUI without specifying the OUI type

This allows finding vendor IE from a specific vendor.

Signed-off-by: Emmanuel Grumbach <emmanuel.grumbach@intel.com>
Signed-off-by: Luca Coelho <luciano.coelho@intel.com>
Signed-off-by: Johannes Berg <johannes.berg@intel.com>

1 files changed, 6 insertions, 2 deletions

diff --git a/net/wireless/scan.c b/net/wireless/scan.c
index abdf651a70d9..ef2955c89a00 100644
--- a/net/wireless/scan.c
+++ b/net/wireless/scan.c
@@ -364,13 +364,16 @@ const u8 *cfg80211_find_ie(u8 eid, const u8 *ies, int len)
 }
 EXPORT_SYMBOL(cfg80211_find_ie);
 
-const u8 *cfg80211_find_vendor_ie(unsigned int oui, u8 oui_type,
+const u8 *cfg80211_find_vendor_ie(unsigned int oui, int oui_type,
                                   const u8 *ies, int len)
 {
         struct ieee80211_vendor_ie *ie;
         const u8 *pos = ies, *end = ies + len;
         int ie_oui;
 
+        if (WARN_ON(oui_type > 0xff))
+                return NULL;
+
         while (pos < end) {
                 pos = cfg80211_find_ie(WLAN_EID_VENDOR_SPECIFIC, pos,
                                        end - pos);
@@ -386,7 +389,8 @@ const u8 *cfg80211_find_vendor_ie(unsigned int oui, u8 oui_type,
                         goto cont;
 
                 ie_oui = ie->oui[0] << 16 | ie->oui[1] << 8 | ie->oui[2];
-                if (ie_oui == oui && ie->oui_type == oui_type)
+                if (ie_oui == oui &&
+                    (oui_type < 0 || ie->oui_type == oui_type))
                         return pos;
 cont:
                 pos += 2 + ie->len;