wlantest: Fix error path in TDLS-not-found case

diff --git a/wlantest/rx_tdls.c b/wlantest/rx_tdls.c
index 1602468017babca50f8b8e883f9b8037d20ba37e..0fd4af6513f2c1eedd7bb991828312a7ef11e8fb 100644 (file)
--- a/wlantest/rx_tdls.c
+++ b/wlantest/rx_tdls.c
@@ -270,8 +270,10 @@ static void rx_data_tdls_setup_confirm(struct wlantest *wt, const u8 *bssid,
                   MAC2STR(elems.link_id + 2 * ETH_ALEN));
 
        tdls = get_tdls(wt, elems.link_id);
-       if (tdls)
-               tdls->link_up = 1;
+       if (tdls == NULL)
+               return;
+
+       tdls->link_up = 1;
        if (tdls_derive_tpk(tdls, bssid, elems.ftie, elems.ftie_len) < 1)
                return;
        if (tdls_verify_mic(tdls, 3, &elems) == 0) {
@@ -363,9 +365,10 @@ static void rx_data_tdls_teardown(struct wlantest *wt, const u8 *bssid,
                   MAC2STR(elems.link_id + 2 * ETH_ALEN));
 
        tdls = get_tdls(wt, elems.link_id);
-       if (tdls)
+       if (tdls) {
                tdls->link_up = 0;
-       tdls_verify_mic_teardown(tdls, 4, data, &elems);
+               tdls_verify_mic_teardown(tdls, 4, data, &elems);
+       }
 }