From 0a9ddd92cd85f47bba0520c4d87414066b43955c Mon Sep 17 00:00:00 2001
From: Jouni Malinen <jouni.malinen@atheros.com>
Date: Fri, 7 Jan 2011 17:09:09 +0200
Subject: [PATCH] wlantest: Add STA counter prot_data_tx

---
 wlantest/rx_data.c       | 2 ++
 wlantest/wlantest_cli.c  | 1 +
 wlantest/wlantest_ctrl.h | 1 +
 3 files changed, 4 insertions(+)

diff --git a/wlantest/rx_data.c b/wlantest/rx_data.c
index fc264acb8..a8312b631 100644
--- a/wlantest/rx_data.c
+++ b/wlantest/rx_data.c
@@ -210,6 +210,8 @@ static void rx_data_bss_prot(struct wlantest *wt,
 		if (bss == NULL)
 			return;
 		sta = sta_get(bss, hdr->addr2);
+		if (sta)
+			sta->counters[WLANTEST_STA_COUNTER_PROT_DATA_TX]++;
 	} else if (fc & WLAN_FC_FROMDS) {
 		bss = bss_get(wt, hdr->addr2);
 		if (bss == NULL)
diff --git a/wlantest/wlantest_cli.c b/wlantest/wlantest_cli.c
index da2552470..4b99c0efb 100644
--- a/wlantest/wlantest_cli.c
+++ b/wlantest/wlantest_cli.c
@@ -551,6 +551,7 @@ static const struct sta_counters sta_counters[] = {
 	{ "deauth_rx_awake", WLANTEST_STA_COUNTER_DEAUTH_RX_AWAKE },
 	{ "disassoc_rx_asleep", WLANTEST_STA_COUNTER_DISASSOC_RX_ASLEEP },
 	{ "disassoc_rx_awake", WLANTEST_STA_COUNTER_DISASSOC_RX_AWAKE },
+	{ "prot_data_tx", WLANTEST_STA_COUNTER_PROT_DATA_TX },
 	{ NULL, 0 }
 };
 
diff --git a/wlantest/wlantest_ctrl.h b/wlantest/wlantest_ctrl.h
index fa6b7660a..be26eff14 100644
--- a/wlantest/wlantest_ctrl.h
+++ b/wlantest/wlantest_ctrl.h
@@ -105,6 +105,7 @@ enum wlantest_sta_counter {
 	WLANTEST_STA_COUNTER_DEAUTH_RX_AWAKE,
 	WLANTEST_STA_COUNTER_DISASSOC_RX_ASLEEP,
 	WLANTEST_STA_COUNTER_DISASSOC_RX_AWAKE,
+	WLANTEST_STA_COUNTER_PROT_DATA_TX,
 	NUM_WLANTEST_STA_COUNTER
 };