From: Jouni Malinen <j@w1.fi>
Date: Fri, 17 Dec 2010 06:26:39 +0000 (+0200)
Subject: wlantest: Fix error path in TDLS-not-found case
X-Git-Tag: hostap-1-bp~691
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=4d00fe48e3ce935cc04c668c0b0dfd3d94719f5a;p=thirdparty%2Fhostap.git

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

diff --git a/wlantest/rx_tdls.c b/wlantest/rx_tdls.c
index 160246801..0fd4af651 100644
--- 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);
+	}
 }