wlantest: Fix management frame injection

The TDLS special case was supposed to apply to only Data frames.

diff --git a/wlantest/inject.c b/wlantest/inject.c
index 51990e4..3defdf7 100644 (file)
--- a/wlantest/inject.c
+++ b/wlantest/inject.c
@@ -216,7 +216,8 @@ static int wlantest_inject_prot(struct wlantest *wt, struct wlantest_bss *bss,
        hdrlen = 24;
        fc = le_to_host16(hdr->frame_control);
 
-       if ((fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) == 0) {
+       if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_DATA &&
+           (fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) == 0) {
                struct wlantest_sta *sta2;
                bss = bss_get(wt, hdr->addr3);
                if (bss == NULL)