From 0570165f4e44833840dd26af90d9c1ae896b220e Mon Sep 17 00:00:00 2001
From: Vulupala Shashank Reddy <vulupa@codeaurora.org>
Date: Tue, 2 Jun 2020 18:49:11 +0530
Subject: [PATCH] qcacld-3.0: Update pktcapture support

Update pktcapture support for packet capture mode.

Change-Id: Id7281d57b7eb547027a999e9662e6ca3c612673d
CRs-Fixed: 2700679
---
 core/wma/src/wma_main.c | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/core/wma/src/wma_main.c b/core/wma/src/wma_main.c
index 7aae99460125..d10529c54919 100644
--- a/core/wma/src/wma_main.c
+++ b/core/wma/src/wma_main.c
@@ -109,9 +109,7 @@
 #include <target_if_direct_buf_rx_api.h>
 #endif
 
-#ifdef WLAN_FEATURE_PKT_CAPTURE
 #include "wlan_pkt_capture_ucfg_api.h"
-#endif
 
 #define WMA_LOG_COMPLETION_TIMER 3000 /* 3 seconds */
 #define WMI_TLV_HEADROOM 128
@@ -6732,6 +6730,13 @@ int wma_rx_service_ready_ext_event(void *handle, uint8_t *event,
 		wma_update_num_peers_tids(wma_handle, wlan_res_cfg);
 	}
 
+	if (ucfg_pkt_capture_get_mode(wma_handle->psoc) &&
+	    wmi_service_enabled(wmi_handle,
+				wmi_service_packet_capture_support))
+		wlan_res_cfg->pktcapture_support = true;
+	else
+		wlan_res_cfg->pktcapture_support = false;
+
 	WMA_LOGD("%s: num_vdevs: %u", __func__, wlan_res_cfg->num_vdevs);
 
 	wma_init_dbr_params(wma_handle);