/*
 * Copyright (c) 2011,2017-2021 The Linux Foundation. All rights reserved.
 * Copyright (c) 2022-2023 Qualcomm Innovation Center, Inc. All rights reserved.
 *
 *
 * Permission to use, copy, modify, and/or distribute this software for
 * any purpose with or without fee is hereby granted, provided that the
 * above copyright notice and this permission notice appear in all
 * copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
 * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
 * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
 * TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
 * PERFORMANCE OF THIS SOFTWARE.
 */

#include "spectral_cmn_api_i.h"
#include "spectral_ol_api_i.h"
#include <qdf_mem.h>
#include <qdf_types.h>
#include <wlan_spectral_public_structs.h>
#include <wlan_cfg80211_spectral.h>
#include <cfg_ucfg_api.h>

/**
 * spectral_get_vdev() - Get pointer to vdev to be used for Spectral
 * operations
 * @pdev: Pointer to pdev
 * @vdev_id: vdev_id
 *
 * Spectral operates on pdev. However, in order to retrieve some WLAN
 * properties, a vdev is required. To facilitate this, the function returns the
 * first vdev in our pdev. The caller should release the reference to the vdev
 * once it is done using it. Additionally, the caller should ensure it has a
 * reference to the pdev at the time of calling this function, and should
 * release the pdev reference either after this function returns or at a later
 * time when the caller is done using pdev.
 * TODO:
 *  - If the framework later provides an API to obtain the first active
 *    vdev, then it would be preferable to use this API.
 *  - Use a common get_vdev() handler for core and target_if using Rx ops. This
 *    is deferred till details emerge on framework providing API to get first
 *    active vdev.
 *
 * Return: Pointer to vdev on success, NULL on failure
 */
static struct wlan_objmgr_vdev*
spectral_get_vdev(struct wlan_objmgr_pdev *pdev, uint8_t vdev_id)
{
	struct wlan_objmgr_vdev *vdev = NULL;

	qdf_assert_always(pdev);

	if (vdev_id == WLAN_INVALID_VDEV_ID)
		vdev = wlan_objmgr_pdev_get_first_vdev(pdev, WLAN_SPECTRAL_ID);
	else
		vdev = wlan_objmgr_get_vdev_by_id_from_pdev(pdev, vdev_id,
							    WLAN_SPECTRAL_ID);
	if (!vdev) {
		spectral_warn("Unable to get first vdev of pdev");
		return NULL;
	}

	return vdev;
}

#ifdef SPECTRAL_MODULIZED_ENABLE
/**
 * spectral_register_cfg80211_handlers() - Register spectral cfg80211 handlers
 * @pdev: Pointer to pdev
 *
 * Register spectral cfg80211 handlers
 * Handlers can be different depending on whether spectral modulized or not
 *
 * Return: None
 */
static void
spectral_register_cfg80211_handlers(struct wlan_objmgr_pdev *pdev)
{
	struct spectral_cfg80211_vendor_cmd_handlers handlers = {0};

	handlers.wlan_cfg80211_spectral_scan_start =
			wlan_cfg80211_spectral_scan_config_and_start;
	handlers.wlan_cfg80211_spectral_scan_stop =
			wlan_cfg80211_spectral_scan_stop;
	handlers.wlan_cfg80211_spectral_scan_get_config =
			wlan_cfg80211_spectral_scan_get_config;
	handlers.wlan_cfg80211_spectral_scan_get_diag_stats =
			wlan_cfg80211_spectral_scan_get_diag_stats;
	handlers.wlan_cfg80211_spectral_scan_get_cap =
			wlan_cfg80211_spectral_scan_get_cap;
	handlers.wlan_cfg80211_spectral_scan_get_status =
			wlan_cfg80211_spectral_scan_get_status;

	wlan_cfg80211_register_spectral_cmd_handler(pdev, &handlers);
}
#else
static void
spectral_register_cfg80211_handlers(struct wlan_objmgr_pdev *pdev)
{
}
#endif

QDF_STATUS
spectral_control_cmn(struct wlan_objmgr_pdev *pdev,
		     struct spectral_cp_request *sscan_req)
{
	QDF_STATUS status = QDF_STATUS_E_FAILURE;
	int temp_debug;
	struct spectral_config sp_out;
	struct spectral_config *sp_in;
	struct spectral_config *spectralparams;
	struct spectral_context *sc;
	struct wlan_objmgr_vdev *vdev = NULL;
	uint8_t vdev_rxchainmask = 0;
	enum spectral_scan_mode smode = sscan_req->ss_mode;
	enum spectral_cp_error_code *err;
	QDF_STATUS ret;
	struct spectral_cp_param param;

	if (!pdev) {
		spectral_err("PDEV is NULL!");
		goto bad;
	}
	sc = spectral_get_spectral_ctx_from_pdev(pdev);
	if (!sc) {
		spectral_err("Spectral context is NULL!");
		goto bad;
	}

	switch (sscan_req->req_id) {
	case SPECTRAL_SET_CONFIG:
		err =  &sscan_req->config_req.sscan_err_code;
		sp_in = &sscan_req->config_req.sscan_config;
		if (sp_in->ss_count != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_SCAN_COUNT;
			param.value = sp_in->ss_count;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_fft_period != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_FFT_PERIOD;
			param.value = sp_in->ss_fft_period;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_period != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_SCAN_PERIOD;
			param.value = sp_in->ss_period;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_recapture != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_FFT_RECAPTURE;
			param.value = sp_in->ss_recapture;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_short_report != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_SHORT_REPORT;
			param.value = (uint32_t)sp_in->ss_short_report ? 1 : 0;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_spectral_pri != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_SPECT_PRI;
			param.value = (uint32_t)sp_in->ss_spectral_pri;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_fft_size != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_FFT_SIZE;
			param.value = sp_in->ss_fft_size;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_gc_ena != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_GC_ENA;
			param.value = sp_in->ss_gc_ena;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_restart_ena != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_RESTART_ENA;
			param.value = sp_in->ss_restart_ena;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_noise_floor_ref != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_NOISE_FLOOR_REF;
			param.value = sp_in->ss_noise_floor_ref;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_init_delay != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_INIT_DELAY;
			param.value = sp_in->ss_init_delay;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_nb_tone_thr != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_NB_TONE_THR;
			param.value = sp_in->ss_nb_tone_thr;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_str_bin_thr != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_STR_BIN_THR;
			param.value = sp_in->ss_str_bin_thr;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_wb_rpt_mode != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_WB_RPT_MODE;
			param.value = sp_in->ss_wb_rpt_mode;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_rssi_rpt_mode != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_RSSI_RPT_MODE;
			param.value = sp_in->ss_rssi_rpt_mode;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_rssi_thr != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_RSSI_THR;
			param.value = sp_in->ss_rssi_thr;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_pwr_format != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_PWR_FORMAT;
			param.value = sp_in->ss_pwr_format;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_rpt_mode != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_RPT_MODE;
			param.value = sp_in->ss_rpt_mode;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_bin_scale != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_BIN_SCALE;
			param.value = sp_in->ss_bin_scale;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_dbm_adj != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_DBM_ADJ;
			param.value = sp_in->ss_dbm_adj;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_chn_mask != SPECTRAL_PHYERR_PARAM_NOVAL) {
			/*
			 * Check if any of the inactive Rx antenna
			 * chains is set active in spectral chainmask
			 */
			vdev = spectral_get_vdev(pdev, sscan_req->vdev_id);
			if (!vdev)
				goto bad;

			vdev_rxchainmask =
			    wlan_vdev_mlme_get_rxchainmask(vdev);
			wlan_objmgr_vdev_release_ref(vdev,
						     WLAN_SPECTRAL_ID);

			if (!(sp_in->ss_chn_mask & vdev_rxchainmask)) {
				spectral_err("Invalid Spectral Chainmask - Inactive Rx antenna chain cannot be an active spectral chain");
				goto bad;
			} else {
				param.id = SPECTRAL_PARAM_CHN_MASK;
				param.value = sp_in->ss_chn_mask;
				ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
				if (QDF_IS_STATUS_ERROR(ret))
					goto bad;
			}
		}

		if (sp_in->ss_frequency.cfreq1 != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_FREQUENCY;
			param.freq.cfreq1 = sp_in->ss_frequency.cfreq1;
			param.freq.cfreq2 = sp_in->ss_frequency.cfreq2;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		if (sp_in->ss_bandwidth != SPECTRAL_PHYERR_PARAM_NOVAL) {
			param.id = SPECTRAL_PARAM_CHAN_WIDTH;
			param.value = sp_in->ss_bandwidth;
			ret = sc->sptrlc_set_spectral_config
						(pdev, &param, smode, err);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}

		break;

	case SPECTRAL_GET_CONFIG:
		sc->sptrlc_get_spectral_config(pdev, &sp_out, smode);
		spectralparams = &sscan_req->config_req.sscan_config;
		spectralparams->ss_fft_period = sp_out.ss_fft_period;
		spectralparams->ss_period = sp_out.ss_period;
		spectralparams->ss_recapture = sp_out.ss_recapture;
		spectralparams->ss_count = sp_out.ss_count;
		spectralparams->ss_short_report =
				sp_out.ss_short_report;
		spectralparams->ss_spectral_pri =
				sp_out.ss_spectral_pri;
		spectralparams->ss_fft_size = sp_out.ss_fft_size;
		spectralparams->ss_gc_ena = sp_out.ss_gc_ena;
		spectralparams->ss_restart_ena = sp_out.ss_restart_ena;
		spectralparams->ss_noise_floor_ref =
				sp_out.ss_noise_floor_ref;
		spectralparams->ss_init_delay = sp_out.ss_init_delay;
		spectralparams->ss_nb_tone_thr = sp_out.ss_nb_tone_thr;
		spectralparams->ss_str_bin_thr = sp_out.ss_str_bin_thr;
		spectralparams->ss_wb_rpt_mode = sp_out.ss_wb_rpt_mode;
		spectralparams->ss_rssi_rpt_mode =
				sp_out.ss_rssi_rpt_mode;
		spectralparams->ss_rssi_thr = sp_out.ss_rssi_thr;
		spectralparams->ss_pwr_format = sp_out.ss_pwr_format;
		spectralparams->ss_rpt_mode = sp_out.ss_rpt_mode;
		spectralparams->ss_bin_scale = sp_out.ss_bin_scale;
		spectralparams->ss_dbm_adj = sp_out.ss_dbm_adj;
		spectralparams->ss_chn_mask = sp_out.ss_chn_mask;
		spectralparams->ss_frequency = sp_out.ss_frequency;
		spectralparams->ss_bandwidth = sp_out.ss_bandwidth;
		break;

	case SPECTRAL_IS_ACTIVE:
		sscan_req->status_req.is_active =
					sc->sptrlc_is_spectral_active(pdev,
								      smode);
		break;

	case SPECTRAL_IS_ENABLED:
		sscan_req->status_req.is_enabled =
					sc->sptrlc_is_spectral_enabled(pdev,
								       smode);
		break;

	case SPECTRAL_SET_DEBUG_LEVEL:
		temp_debug = sscan_req->debug_req.spectral_dbg_level;
		sc->sptrlc_set_debug_level(pdev, temp_debug);
		break;

	case SPECTRAL_GET_DEBUG_LEVEL:
		sscan_req->debug_req.spectral_dbg_level =
					sc->sptrlc_get_debug_level(pdev);
		break;

	case SPECTRAL_ACTIVATE_SCAN:
		err = &sscan_req->action_req.sscan_err_code;
		ret = sc->sptrlc_start_spectral_scan(pdev, sscan_req->vdev_id,
						     smode, err);
		if (QDF_IS_STATUS_ERROR(ret))
			goto bad;
		break;

	case SPECTRAL_STOP_SCAN:
		err = &sscan_req->action_req.sscan_err_code;
		ret = sc->sptrlc_stop_spectral_scan(pdev, smode, err);
		if (QDF_IS_STATUS_ERROR(ret))
			goto bad;
		break;

	case SPECTRAL_GET_CAPABILITY_INFO:
		{
			struct spectral_caps *caps;

			caps  = &sscan_req->caps_req.sscan_caps;
			ret = sc->sptrlc_get_spectral_capinfo(pdev, caps);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}
		break;

	case SPECTRAL_GET_DIAG_STATS:
		{
			struct spectral_diag_stats *diag;

			diag  = &sscan_req->diag_req.sscan_diag;
			ret = sc->sptrlc_get_spectral_diagstats(pdev, diag);
			if (QDF_IS_STATUS_ERROR(ret))
				goto bad;
		}
		break;

	case SPECTRAL_GET_CHAN_WIDTH:
		{
			uint32_t chan_width;

			vdev = spectral_get_vdev(pdev, sscan_req->vdev_id);
			if (!vdev)
				goto bad;

			chan_width = spectral_vdev_get_ch_width(vdev);
			wlan_objmgr_vdev_release_ref(vdev, WLAN_SPECTRAL_ID);

			sscan_req->chan_width_req.chan_width =
							(uint32_t)chan_width;
		}
		break;

	case SPECTRAL_SET_DMA_DEBUG:
		if (sc->sptrlc_set_dma_debug)
			sc->sptrlc_set_dma_debug(
			     pdev,
			     sscan_req->dma_debug_req.dma_debug_type,
			     sscan_req->dma_debug_req.dma_debug_enable);
		break;

	default:
		goto bad;
		break;
	}

	status = QDF_STATUS_SUCCESS;
bad:
	return status;
}

/**
 * spectral_ctx_deinit() - De-initialize function pointers from spectral context
 * @sc: Reference to spectral_context object
 *
 * Return: None
 */
static void
spectral_ctx_deinit(struct spectral_context *sc)
{
	if (sc) {
		sc->sptrlc_ucfg_phyerr_config = NULL;
		sc->sptrlc_pdev_spectral_init = NULL;
		sc->sptrlc_pdev_spectral_deinit = NULL;
		sc->sptrlc_psoc_spectral_init = NULL;
		sc->sptrlc_psoc_spectral_deinit = NULL;
		sc->sptrlc_set_spectral_config = NULL;
		sc->sptrlc_get_spectral_config = NULL;
		sc->sptrlc_start_spectral_scan = NULL;
		sc->sptrlc_stop_spectral_scan = NULL;
		sc->sptrlc_is_spectral_active = NULL;
		sc->sptrlc_is_spectral_enabled = NULL;
		sc->sptrlc_set_debug_level = NULL;
		sc->sptrlc_get_debug_level = NULL;
		sc->sptrlc_get_spectral_capinfo = NULL;
		sc->sptrlc_get_spectral_diagstats = NULL;
	}
}

QDF_STATUS
wlan_spectral_psoc_obj_create_handler(struct wlan_objmgr_psoc *psoc, void *arg)
{
	struct spectral_context *sc = NULL;
	QDF_STATUS status;

	if (!psoc) {
		spectral_err("PSOC is NULL");
		return QDF_STATUS_E_FAILURE;
	}

	status = wlan_spectral_init_psoc_feature_cap(psoc);
	if (QDF_IS_STATUS_ERROR(status)) {
		spectral_err("Failed to initialize spectral pdev feature caps");
		return QDF_STATUS_E_FAILURE;
	}

	if (wlan_spectral_is_feature_disabled_psoc(psoc)) {
		spectral_info("Spectral feature is disabled");
		return QDF_STATUS_COMP_DISABLED;
	}

	sc = (struct spectral_context *)
	    qdf_mem_malloc(sizeof(struct spectral_context));
	if (!sc)
		return QDF_STATUS_E_NOMEM;

	qdf_mem_zero(sc, sizeof(struct spectral_context));
	sc->psoc_obj = psoc;
	if (wlan_objmgr_psoc_get_dev_type(psoc) == WLAN_DEV_OL)
		spectral_ctx_init_ol(sc);
	wlan_objmgr_psoc_component_obj_attach(psoc, WLAN_UMAC_COMP_SPECTRAL,
					      (void *)sc, QDF_STATUS_SUCCESS);

	return QDF_STATUS_SUCCESS;
}

QDF_STATUS
wlan_spectral_psoc_obj_destroy_handler(struct wlan_objmgr_psoc *psoc,
				       void *arg)
{
	struct spectral_context *sc = NULL;

	if (!psoc) {
		spectral_err("PSOC is NULL");
		return QDF_STATUS_E_FAILURE;
	}

	if (wlan_spectral_is_feature_disabled_psoc(psoc)) {
		spectral_info("Spectral feature is disabled");
		return QDF_STATUS_COMP_DISABLED;
	}

	sc = wlan_objmgr_psoc_get_comp_private_obj(psoc,
						   WLAN_UMAC_COMP_SPECTRAL);
	if (sc) {
		wlan_objmgr_psoc_component_obj_detach(psoc,
						      WLAN_UMAC_COMP_SPECTRAL,
						      (void *)sc);
		/* Deinitilise function pointers from spectral context */
		spectral_ctx_deinit(sc);
		qdf_mem_free(sc);
	}

	return QDF_STATUS_SUCCESS;
}

QDF_STATUS
wlan_spectral_pdev_obj_create_handler(struct wlan_objmgr_pdev *pdev, void *arg)
{
	struct pdev_spectral *ps = NULL;
	struct spectral_context *sc = NULL;
	void *target_handle = NULL;
	QDF_STATUS status;

	if (!pdev) {
		spectral_err("PDEV is NULL");
		return QDF_STATUS_E_FAILURE;
	}

	status = wlan_spectral_init_pdev_feature_caps(pdev);
	if (QDF_IS_STATUS_ERROR(status)) {
		spectral_err("Failed to initialize spectral pdev feature caps");
		return QDF_STATUS_E_FAILURE;
	}

	if (wlan_spectral_is_feature_disabled_pdev(pdev)) {
		spectral_info("Spectral feature is disabled");
		return QDF_STATUS_COMP_DISABLED;
	}

	ps = (struct pdev_spectral *)
	    qdf_mem_malloc(sizeof(struct pdev_spectral));
	if (!ps)
		return QDF_STATUS_E_NOMEM;

	sc = spectral_get_spectral_ctx_from_pdev(pdev);
	if (!sc) {
		spectral_err("Spectral context is NULL!");
		goto cleanup;
	}

	qdf_mem_zero(ps, sizeof(struct pdev_spectral));
	ps->psptrl_pdev = pdev;

	spectral_register_cfg80211_handlers(pdev);
	if (sc->sptrlc_pdev_spectral_init) {
		target_handle = sc->sptrlc_pdev_spectral_init(pdev);
		if (!target_handle) {
			spectral_err("Spectral lmac object is NULL!");
			goto cleanup;
		}
		ps->psptrl_target_handle = target_handle;
	}
	wlan_objmgr_pdev_component_obj_attach(pdev, WLAN_UMAC_COMP_SPECTRAL,
					      (void *)ps, QDF_STATUS_SUCCESS);

	return QDF_STATUS_SUCCESS;
 cleanup:
	qdf_mem_free(ps);
	return QDF_STATUS_E_FAILURE;
}

QDF_STATUS
wlan_spectral_pdev_obj_destroy_handler(struct wlan_objmgr_pdev *pdev,
				       void *arg)
{
	struct pdev_spectral *ps = NULL;
	struct spectral_context *sc = NULL;

	if (!pdev) {
		spectral_err("PDEV is NULL");
		return QDF_STATUS_E_FAILURE;
	}

	if (wlan_spectral_is_feature_disabled_pdev(pdev)) {
		spectral_info("Spectral feature is disabled");
		return QDF_STATUS_COMP_DISABLED;
	}

	sc = spectral_get_spectral_ctx_from_pdev(pdev);
	if (!sc) {
		spectral_err("Spectral context is NULL!");
		return QDF_STATUS_E_FAILURE;
	}
	ps = wlan_objmgr_pdev_get_comp_private_obj(pdev,
						   WLAN_UMAC_COMP_SPECTRAL);
	if (ps) {
		if (sc->sptrlc_pdev_spectral_deinit)
			sc->sptrlc_pdev_spectral_deinit(pdev);
		ps->psptrl_target_handle = NULL;
		wlan_objmgr_pdev_component_obj_detach(pdev,
						      WLAN_UMAC_COMP_SPECTRAL,
						      (void *)ps);
		qdf_mem_free(ps);
	}

	return QDF_STATUS_SUCCESS;
}