// SPDX-License-Identifier: GPL-2.0-only

#include "wcn36xx.h"
#include "firmware.h"

#define DEFINE(s)[s] = #s

static const char * const wcn36xx_firmware_caps_names[] = {
	DEFINE(MCC),
	DEFINE(P2P),
	DEFINE(DOT11AC),
	DEFINE(SLM_SESSIONIZATION),
	DEFINE(DOT11AC_OPMODE),
	DEFINE(SAP32STA),
	DEFINE(TDLS),
	DEFINE(P2P_GO_NOA_DECOUPLE_INIT_SCAN),
	DEFINE(WLANACTIVE_OFFLOAD),
	DEFINE(BEACON_OFFLOAD),
	DEFINE(SCAN_OFFLOAD),
	DEFINE(ROAM_OFFLOAD),
	DEFINE(BCN_MISS_OFFLOAD),
	DEFINE(STA_POWERSAVE),
	DEFINE(STA_ADVANCED_PWRSAVE),
	DEFINE(AP_UAPSD),
	DEFINE(AP_DFS),
	DEFINE(BLOCKACK),
	DEFINE(PHY_ERR),
	DEFINE(BCN_FILTER),
	DEFINE(RTT),
	DEFINE(RATECTRL),
	DEFINE(WOW),
	DEFINE(WLAN_ROAM_SCAN_OFFLOAD),
	DEFINE(SPECULATIVE_PS_POLL),
	DEFINE(SCAN_SCH),
	DEFINE(IBSS_HEARTBEAT_OFFLOAD),
	DEFINE(WLAN_SCAN_OFFLOAD),
	DEFINE(WLAN_PERIODIC_TX_PTRN),
	DEFINE(ADVANCE_TDLS),
	DEFINE(BATCH_SCAN),
	DEFINE(FW_IN_TX_PATH),
	DEFINE(EXTENDED_NSOFFLOAD_SLOT),
	DEFINE(CH_SWITCH_V1),
	DEFINE(HT40_OBSS_SCAN),
	DEFINE(UPDATE_CHANNEL_LIST),
	DEFINE(WLAN_MCADDR_FLT),
	DEFINE(WLAN_CH144),
	DEFINE(NAN),
	DEFINE(TDLS_SCAN_COEXISTENCE),
	DEFINE(LINK_LAYER_STATS_MEAS),
	DEFINE(MU_MIMO),
	DEFINE(EXTENDED_SCAN),
	DEFINE(DYNAMIC_WMM_PS),
	DEFINE(MAC_SPOOFED_SCAN),
	DEFINE(BMU_ERROR_GENERIC_RECOVERY),
	DEFINE(DISA),
	DEFINE(FW_STATS),
	DEFINE(WPS_PRBRSP_TMPL),
	DEFINE(BCN_IE_FLT_DELTA),
	DEFINE(TDLS_OFF_CHANNEL),
	DEFINE(RTT3),
	DEFINE(MGMT_FRAME_LOGGING),
	DEFINE(ENHANCED_TXBD_COMPLETION),
	DEFINE(LOGGING_ENHANCEMENT),
	DEFINE(EXT_SCAN_ENHANCED),
	DEFINE(MEMORY_DUMP_SUPPORTED),
	DEFINE(PER_PKT_STATS_SUPPORTED),
	DEFINE(EXT_LL_STAT),
	DEFINE(WIFI_CONFIG),
	DEFINE(ANTENNA_DIVERSITY_SELECTION),
};

#undef DEFINE

const char *wcn36xx_firmware_get_cap_name(enum wcn36xx_firmware_feat_caps x)
{
	if (x >= ARRAY_SIZE(wcn36xx_firmware_caps_names))
		return "UNKNOWN";
	return wcn36xx_firmware_caps_names[x];
}

void wcn36xx_firmware_set_feat_caps(u32 *bitmap,
				    enum wcn36xx_firmware_feat_caps cap)
{
	int arr_idx, bit_idx;

	if (cap < 0 || cap > 127) {
		wcn36xx_warn("error cap idx %d\n", cap);
		return;
	}

	arr_idx = cap / 32;
	bit_idx = cap % 32;
	bitmap[arr_idx] |= (1 << bit_idx);
}

int wcn36xx_firmware_get_feat_caps(u32 *bitmap,
				   enum wcn36xx_firmware_feat_caps cap)
{
	int arr_idx, bit_idx;

	if (cap < 0 || cap > 127) {
		wcn36xx_warn("error cap idx %d\n", cap);
		return -EINVAL;
	}

	arr_idx = cap / 32;
	bit_idx = cap % 32;

	return (bitmap[arr_idx] & (1 << bit_idx)) ? 1 : 0;
}

void wcn36xx_firmware_clear_feat_caps(u32 *bitmap,
				      enum wcn36xx_firmware_feat_caps cap)
{
	int arr_idx, bit_idx;

	if (cap < 0 || cap > 127) {
		wcn36xx_warn("error cap idx %d\n", cap);
		return;
	}

	arr_idx = cap / 32;
	bit_idx = cap % 32;
	bitmap[arr_idx] &= ~(1 << bit_idx);
}