From 515e9989a03dc5262fa5cdc1ec186ae2409f33c7 Mon Sep 17 00:00:00 2001
From: phadiman <phadiman@codeaurora.org>
Date: Wed, 20 Feb 2019 12:56:42 +0530
Subject: [PATCH] qcacmn: Add target type checks

spectral register and unregister is
supported for QCA8074, QCA8074V2 and
QCA6018

Missing target check in spectral un-
register function lead to memory leak
issue

Add missing target check in spectral
unregister to resolve the leak

CRs-Fixed: 2401813
Change-Id: I04283eaacc2d51934fb08ada46bca25d43c1dfa4
---
 spectral/dispatcher/src/wlan_spectral_tgt_api.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/spectral/dispatcher/src/wlan_spectral_tgt_api.c b/spectral/dispatcher/src/wlan_spectral_tgt_api.c
index e0e639979029..8e08605e441e 100644
--- a/spectral/dispatcher/src/wlan_spectral_tgt_api.c
+++ b/spectral/dispatcher/src/wlan_spectral_tgt_api.c
@@ -306,7 +306,9 @@ tgt_spectral_unregister_to_dbr(struct wlan_objmgr_pdev *pdev)
 	psoc = wlan_pdev_get_psoc(pdev);
 	dbr_tx_ops = &psoc->soc_cb.tx_ops.dbr_tx_ops;
 
-	if (tgt_spectral_get_target_type(psoc) == TARGET_TYPE_QCA8074)
+	if ((tgt_spectral_get_target_type(psoc) == TARGET_TYPE_QCA8074) ||
+	    (tgt_spectral_get_target_type(psoc) == TARGET_TYPE_QCA8074V2) ||
+	    (tgt_spectral_get_target_type(psoc) == TARGET_TYPE_QCA6018))
 		if (dbr_tx_ops->direct_buf_rx_module_unregister)
 			return dbr_tx_ops->direct_buf_rx_module_unregister
 				(pdev, 0);