qcacld-3.0: Add support to cache he operation info

Currently if station is connected in 11AX mode, AP's HE
IE is not getting updated in the cache information of the
station.

Add support to update HE operation element in the connection
info structure of the station context.

Change-Id: I214d6cec22525788e9ce21a7d832263148989715
CRs-Fixed: 2656005

core/hdd/inc/wlan_hdd_assoc.h

@@ -197,6 +197,11 @@ struct hdd_connection_info {
 	tDot11fIEhs20vendor_ie hs20vendor_ie;
 	struct ieee80211_ht_operation ht_operation;
 	struct ieee80211_vht_operation vht_operation;
+#if (LINUX_VERSION_CODE >= KERNEL_VERSION(4, 19, 0)) \
+     && defined(WLAN_FEATURE_11AX)
+	struct ieee80211_he_operation *he_operation;
+	uint32_t he_oper_len;
+#endif
 	uint32_t roam_count;
 	int8_t signal;
 	int32_t assoc_status_code;

core/hdd/inc/wlan_hdd_main.h

@@ -4326,4 +4326,34 @@ hdd_monitor_mode_qdf_create_event(struct hdd_adapter *adapter,
 }
 #endif
 
+#if (LINUX_VERSION_CODE >= KERNEL_VERSION(4, 19, 0)) && \
+     defined(WLAN_FEATURE_11AX)
+/**
+ * hdd_cleanup_conn_info() - Cleanup connectin info
+ * @adapter: Adapter upon which the command was received
+ *
+ * This function frees the memory allocated for the connection
+ * info structure
+ *
+ * Return: none
+ */
+void hdd_cleanup_conn_info(struct hdd_adapter *adapter);
+/**
+ * hdd_sta_destroy_ctx_all() - cleanup all station contexts
+ * @hdd_ctx: Global HDD context
+ *
+ * This function destroys all the station contexts
+ *
+ * Return: none
+ */
+void hdd_sta_destroy_ctx_all(struct hdd_context *hdd_ctx);
+#else
+static inline void hdd_cleanup_conn_info(struct hdd_adapter *adapter)
+{
+}
+static inline void hdd_sta_destroy_ctx_all(struct hdd_context *hdd_ctx)
+{
+}
+#endif
+
 #endif /* end #if !defined(WLAN_HDD_MAIN_H) */

core/hdd/src/wlan_hdd_assoc.c

@@ -177,6 +177,21 @@ static const int beacon_filter_table[] = {
 #endif
 };
 
+/* HE operation BIT positins */
+#if defined(WLAN_FEATURE_11AX)
+#define HE_OPERATION_DFLT_PE_DURATION_POS 0
+#define HE_OPERATION_TWT_REQUIRED_POS 3
+#define HE_OPERATION_RTS_THRESHOLD_POS 4
+#define HE_OPERATION_VHT_OPER_POS 14
+#define HE_OPERATION_CO_LOCATED_BSS_POS 15
+#define HE_OPERATION_ER_SU_DISABLE_POS 16
+#define HE_OPERATION_OPER_INFO_6G_POS 17
+#define HE_OPERATION_RESERVED_POS 18
+#define HE_OPERATION_BSS_COLOR_POS 24
+#define HE_OPERATION_PARTIAL_BSS_COLOR_POS 30
+#define HE_OPERATION_BSS_COL_DISABLED_POS 31
+#endif
+
 #if defined(WLAN_FEATURE_SAE) && \
 		defined(CFG80211_EXTERNAL_AUTH_SUPPORT)
 /**
@@ -901,6 +916,112 @@ static void hdd_copy_vht_operation(struct hdd_station_ctx *hdd_sta_ctx,
 	hdd_vht_ops->basic_mcs_set = roam_vht_ops->basicMCSSet;
 }
 
+#if (LINUX_VERSION_CODE >= KERNEL_VERSION(4, 19, 0)) && \
+     defined(WLAN_FEATURE_11AX)
+static void hdd_copy_he_operation(struct hdd_station_ctx *hdd_sta_ctx,
+				  struct csr_roam_info *roam_info)
+{
+	tDot11fIEhe_op *roam_he_operation = &roam_info->he_operation;
+	struct ieee80211_he_operation *hdd_he_operation;
+	uint32_t he_oper_params = 0;
+	uint32_t len = 0, filled = 0;
+	uint8_t he_oper_6g_params = 0;
+	uint32_t he_oper_len;
+
+	if (!roam_he_operation->present)
+		return;
+	if (roam_he_operation->vht_oper_present)
+		len += 3;
+	if (roam_he_operation->co_located_bss)
+		len += 1;
+	if (roam_he_operation->oper_info_6g_present)
+		len += 5;
+
+	he_oper_len = sizeof(struct ieee80211_he_operation) + len;
+
+	hdd_he_operation = qdf_mem_malloc(he_oper_len);
+	if (!hdd_he_operation)
+		return;
+
+	/* Fill he_oper_params */
+	he_oper_params |= roam_he_operation->default_pe <<
+					HE_OPERATION_DFLT_PE_DURATION_POS;
+	he_oper_params |= roam_he_operation->twt_required <<
+					HE_OPERATION_TWT_REQUIRED_POS;
+	he_oper_params |= roam_he_operation->txop_rts_threshold <<
+					HE_OPERATION_RTS_THRESHOLD_POS;
+	he_oper_params |= roam_he_operation->vht_oper_present <<
+					HE_OPERATION_VHT_OPER_POS;
+	he_oper_params |= roam_he_operation->co_located_bss <<
+					HE_OPERATION_CO_LOCATED_BSS_POS;
+	he_oper_params |= roam_he_operation->er_su_disable <<
+					HE_OPERATION_ER_SU_DISABLE_POS;
+	he_oper_params |= roam_he_operation->oper_info_6g_present <<
+					HE_OPERATION_OPER_INFO_6G_POS;
+	he_oper_params |= roam_he_operation->reserved2 <<
+					HE_OPERATION_RESERVED_POS;
+	he_oper_params |= roam_he_operation->bss_color <<
+					HE_OPERATION_BSS_COLOR_POS;
+	he_oper_params |= roam_he_operation->partial_bss_col <<
+					HE_OPERATION_PARTIAL_BSS_COLOR_POS;
+	he_oper_params |= roam_he_operation->bss_col_disabled <<
+					HE_OPERATION_BSS_COL_DISABLED_POS;
+
+	hdd_he_operation->he_oper_params = he_oper_params;
+
+	/* Fill he_mcs_nss set */
+	qdf_mem_copy(&hdd_he_operation->he_mcs_nss_set,
+		     roam_he_operation->basic_mcs_nss,
+		     sizeof(hdd_he_operation->he_mcs_nss_set));
+
+	/* Fill he_params_optional fields */
+
+	if (roam_he_operation->vht_oper_present) {
+		hdd_he_operation->optional[filled++] =
+			roam_he_operation->vht_oper.info.chan_width;
+		hdd_he_operation->optional[filled++] =
+			roam_he_operation->vht_oper.info.center_freq_seg0;
+		hdd_he_operation->optional[filled++] =
+			roam_he_operation->vht_oper.info.center_freq_seg1;
+	}
+	if (roam_he_operation->co_located_bss)
+		hdd_he_operation->optional[filled++] =
+				roam_he_operation->maxbssid_ind.info.data;
+
+	if (roam_he_operation->oper_info_6g_present) {
+		hdd_he_operation->optional[filled++] =
+			roam_he_operation->oper_info_6g.info.primary_ch;
+		he_oper_6g_params |=
+			roam_he_operation->oper_info_6g.info.ch_width << 0;
+		he_oper_6g_params |=
+			roam_he_operation->oper_info_6g.info.dup_bcon << 2;
+		he_oper_6g_params |=
+			roam_he_operation->oper_info_6g.info.reserved << 3;
+
+		hdd_he_operation->optional[filled++] = he_oper_6g_params;
+		hdd_he_operation->optional[filled++] =
+			roam_he_operation->oper_info_6g.info.center_freq_seg0;
+		hdd_he_operation->optional[filled++] =
+			roam_he_operation->oper_info_6g.info.center_freq_seg1;
+		hdd_he_operation->optional[filled] =
+				roam_he_operation->oper_info_6g.info.min_rate;
+	}
+
+	if (hdd_sta_ctx->cache_conn_info.he_operation) {
+		qdf_mem_free(hdd_sta_ctx->cache_conn_info.he_operation);
+		hdd_sta_ctx->cache_conn_info.he_operation = NULL;
+	}
+
+	hdd_sta_ctx->cache_conn_info.he_oper_len = he_oper_len;
+
+	hdd_sta_ctx->cache_conn_info.he_operation = hdd_he_operation;
+}
+#else
+static inline void hdd_copy_he_operation(struct hdd_station_ctx *hdd_sta_ctx,
+					 struct csr_roam_info *roam_info)
+{
+}
+#endif
 
 /**
  * hdd_save_bss_info() - save connection info in hdd sta ctx
@@ -952,9 +1073,15 @@ static void hdd_save_bss_info(struct hdd_adapter *adapter,
 	} else {
 		hdd_sta_ctx->conn_info.conn_flag.vht_op_present = false;
 	}
+
+	/* Cleanup already existing he info */
+	hdd_cleanup_conn_info(adapter);
+
 	/* Cache last connection info */
 	qdf_mem_copy(&hdd_sta_ctx->cache_conn_info, &hdd_sta_ctx->conn_info,
 		     sizeof(hdd_sta_ctx->cache_conn_info));
+
+	hdd_copy_he_operation(hdd_sta_ctx, roam_info);
 }
 
 /**

core/hdd/src/wlan_hdd_main.c

@@ -5351,6 +5351,34 @@ void hdd_deinit_adapter(struct hdd_context *hdd_ctx,
 	hdd_exit();
 }
 
+#if (LINUX_VERSION_CODE >= KERNEL_VERSION(4, 19, 0)) && \
+     defined(WLAN_FEATURE_11AX)
+void hdd_cleanup_conn_info(struct hdd_adapter *adapter)
+{
+	struct hdd_station_ctx *hdd_sta_ctx =
+					WLAN_HDD_GET_STATION_CTX_PTR(adapter);
+
+	if (!hdd_sta_ctx)
+		return;
+
+	if (hdd_sta_ctx->cache_conn_info.he_operation) {
+		qdf_mem_free(hdd_sta_ctx->cache_conn_info.he_operation);
+		hdd_sta_ctx->cache_conn_info.he_operation = NULL;
+	}
+}
+
+void hdd_sta_destroy_ctx_all(struct hdd_context *hdd_ctx)
+{
+	struct hdd_adapter *adapter;
+
+	hdd_for_each_adapter_dev_held(hdd_ctx, adapter) {
+		if (adapter->device_mode == QDF_STA_MODE)
+			hdd_cleanup_conn_info(adapter);
+		dev_put(adapter->dev);
+	}
+}
+#endif
+
 static void hdd_cleanup_adapter(struct hdd_context *hdd_ctx,
 				struct hdd_adapter *adapter,
 				bool rtnl_held)
@@ -6454,7 +6482,7 @@ QDF_STATUS hdd_stop_adapter(struct hdd_context *hdd_ctx,
 			ucfg_ipa_flush_pending_vdev_events(hdd_ctx->pdev,
 							   adapter->vdev_id);
 		}
-
+		hdd_cleanup_conn_info(adapter);
 		hdd_vdev_destroy(adapter);
 		break;
 
@@ -12882,6 +12910,7 @@ int hdd_wlan_stop_modules(struct hdd_context *hdd_ctx, bool ftm_mode)
 		wlan_hdd_free_sar_config(hdd_ctx);
 
 	hdd_sap_destroy_ctx_all(hdd_ctx, is_recovery_stop);
+	hdd_sta_destroy_ctx_all(hdd_ctx);
 	pld_request_bus_bandwidth(hdd_ctx->parent_dev, PLD_BUS_WIDTH_NONE);
 	hdd_bus_bandwidth_deinit(hdd_ctx);
 	hdd_check_for_leaks(hdd_ctx, is_recovery_stop);

core/hdd/src/wlan_hdd_station_info.c

@@ -68,6 +68,8 @@
 	QCA_WLAN_VENDOR_ATTR_GET_STATION_INFO_HT_OPERATION
 #define VHT_OPERATION \
 	QCA_WLAN_VENDOR_ATTR_GET_STATION_INFO_VHT_OPERATION
+#define HE_OPERATION \
+	QCA_WLAN_VENDOR_ATTR_GET_STATION_INFO_HE_OPERATION
 #define INFO_ASSOC_FAIL_REASON \
 	QCA_WLAN_VENDOR_ATTR_GET_STATION_INFO_ASSOC_FAIL_REASON
 #define REMOTE_MAX_PHY_RATE \
@@ -581,6 +583,43 @@ fail:
 	return -EINVAL;
 }
 
+#if (LINUX_VERSION_CODE >= KERNEL_VERSION(4, 19, 0)) && \
+     defined(WLAN_FEATURE_11AX)
+static int32_t hdd_add_he_oper_info(
+				struct sk_buff *skb,
+				struct hdd_station_ctx *hdd_sta_ctx)
+{
+	int32_t ret = 0;
+
+	if (nla_put(skb, HE_OPERATION,
+		    hdd_sta_ctx->cache_conn_info.he_oper_len,
+		     hdd_sta_ctx->cache_conn_info.he_operation))
+		ret = -EINVAL;
+
+	qdf_mem_free(hdd_sta_ctx->cache_conn_info.he_operation);
+	hdd_sta_ctx->cache_conn_info.he_operation = NULL;
+	hdd_sta_ctx->cache_conn_info.he_oper_len = 0;
+	return ret;
+}
+
+static int32_t hdd_get_he_op_len(struct hdd_station_ctx *hdd_sta_ctx)
+{
+	return hdd_sta_ctx->cache_conn_info.he_oper_len;
+}
+#else
+static inline uint32_t hdd_add_he_oper_info(
+					struct sk_buff *skb,
+					struct hdd_station_ctx *hdd_sta_ctx)
+{
+	return 0;
+}
+
+static uint32_t hdd_get_he_op_len(struct hdd_station_ctx *hdd_sta_ctx)
+{
+	return 0;
+}
+#endif
+
 /**
  * hdd_get_station_info() - send BSS information to supplicant
  * @hdd_ctx: pointer to hdd context
@@ -593,7 +632,7 @@ static int hdd_get_station_info(struct hdd_context *hdd_ctx,
 {
 	struct sk_buff *skb = NULL;
 	uint8_t *tmp_hs20 = NULL, *ies = NULL;
-	uint32_t nl_buf_len, ie_len = 0;
+	uint32_t nl_buf_len, ie_len = 0, hdd_he_op_len = 0;
 	struct hdd_station_ctx *hdd_sta_ctx;
 	QDF_STATUS status;
 
@@ -634,6 +673,9 @@ static int hdd_get_station_info(struct hdd_context *hdd_ctx,
 	if (QDF_IS_STATUS_SUCCESS(status))
 		nl_buf_len += ie_len;
 
+	hdd_he_op_len = hdd_get_he_op_len(hdd_sta_ctx);
+	nl_buf_len += hdd_he_op_len;
+
 	skb = cfg80211_vendor_cmd_alloc_reply_skb(hdd_ctx->wiphy, nl_buf_len);
 	if (!skb) {
 		hdd_err("cfg80211_vendor_cmd_alloc_reply_skb failed");
@@ -676,6 +718,10 @@ static int hdd_get_station_info(struct hdd_context *hdd_ctx,
 			hdd_err("put fail");
 			goto fail;
 		}
+	if (hdd_add_he_oper_info(skb, hdd_sta_ctx)) {
+		hdd_err("put fail");
+		goto fail;
+	}
 	if (hdd_sta_ctx->cache_conn_info.conn_flag.hs20_present)
 		if (nla_put(skb, AP_INFO_HS20_INDICATION,
 			    (sizeof(hdd_sta_ctx->cache_conn_info.hs20vendor_ie)

core/mac/inc/sir_api.h

@@ -1045,6 +1045,9 @@ struct join_rsp {
 	tDot11fIEVHTCaps vht_caps;
 	tDot11fIEHTInfo ht_operation;
 	tDot11fIEVHTOperation vht_operation;
+#ifdef WLAN_FEATURE_11AX
+	tDot11fIEhe_op he_operation;
+#endif
 	tDot11fIEhs20vendor_ie hs20vendor_ie;
 	bool is_fils_connection;
 	uint16_t fils_seq_num;

core/mac/src/include/dph_global.h

@@ -77,6 +77,9 @@ struct parsed_ies {
 	tDot11fIEHTInfo ht_operation;
 	tDot11fIEVHTOperation vht_operation;
 	tDot11fIEhs20vendor_ie hs20vendor_ie;
+#ifdef WLAN_FEATURE_11AX
+	tDot11fIEhe_op he_operation;
+#endif
 };
 
 /* STA state node */

core/mac/src/pe/lim/lim_process_assoc_rsp_frame.c

@@ -555,6 +555,20 @@ static uint8_t lim_get_nss_supported_by_ap(tDot11fIEVHTCaps *vht_caps,
 	return NSS_1x1_MODE;
 }
 
+#ifdef WLAN_FEATURE_11AX
+static void lim_process_he_info(tpSirProbeRespBeacon beacon,
+				tpDphHashNode sta_ds)
+{
+	if (beacon->he_op.present)
+		sta_ds->parsed_ies.he_operation = beacon->he_op;
+}
+#else
+static inline void lim_process_he_info(tpSirProbeRespBeacon beacon,
+				       tpDphHashNode sta_ds)
+{
+}
+#endif
+
 /**
  * lim_process_assoc_rsp_frame() - Processes assoc response
  * @mac_ctx: Pointer to Global MAC structure
@@ -1054,6 +1068,8 @@ lim_process_assoc_rsp_frame(struct mac_context *mac_ctx,
 	if (beacon->VHTOperation.present)
 		sta_ds->parsed_ies.vht_operation = beacon->VHTOperation;
 
+	lim_process_he_info(beacon, sta_ds);
+
 	if (mac_ctx->lim.gLimProtectionControl !=
 	    MLME_FORCE_POLICY_PROTECTION_DISABLE)
 		lim_decide_sta_protection_on_assoc(mac_ctx, beacon,

core/mac/src/pe/lim/lim_send_sme_rsp_messages.c

@@ -347,6 +347,20 @@ static void lim_handle_join_rsp_status(struct mac_context *mac_ctx,
 	}
 }
 
+#ifdef WLAN_FEATURE_11AX
+static void lim_add_he_info(struct parsed_ies *parsed_ies,
+			    struct join_rsp *sme_join_rsp)
+{
+	if (parsed_ies->he_operation.present)
+		sme_join_rsp->he_operation = parsed_ies->he_operation;
+}
+#else
+static inline void lim_add_he_info(struct parsed_ies *parsed_ies,
+				   struct join_rsp *sme_join_rsp)
+{
+}
+#endif
+
 /**
  * lim_add_bss_info() - copy data from session entry to join rsp
  * @sta_ds: Station dph entry
@@ -369,6 +383,7 @@ static void lim_add_bss_info(tpDphHashNode sta_ds,
 		sme_join_rsp->ht_operation = parsed_ies->ht_operation;
 	if (parsed_ies->vht_operation.present)
 		sme_join_rsp->vht_operation = parsed_ies->vht_operation;
+	lim_add_he_info(parsed_ies, sme_join_rsp);
 }
 
 #ifdef WLAN_FEATURE_FILS_SK

core/sme/inc/csr_api.h

@@ -1085,6 +1085,9 @@ struct csr_roam_info {
 	tDot11fIEhs20vendor_ie hs20vendor_ie;
 	tDot11fIEVHTOperation vht_operation;
 	tDot11fIEHTInfo ht_operation;
+#ifdef WLAN_FEATURE_11AX
+	tDot11fIEhe_op he_operation;
+#endif
 	bool reassoc;
 	bool ampdu;
 	bool sgi_enable;

core/sme/src/csr/csr_api_roam.c

@@ -7641,6 +7641,19 @@ static inline void csr_process_fils_join_rsp(struct mac_context *mac_ctx,
 {}
 #endif
 
+#ifdef WLAN_FEATURE_11AX
+static void csr_roam_process_he_info(struct join_rsp *sme_join_rsp,
+				     struct csr_roam_info *roam_info)
+{
+	roam_info->he_operation = sme_join_rsp->he_operation;
+}
+#else
+static inline void csr_roam_process_he_info(struct join_rsp *sme_join_rsp,
+					    struct csr_roam_info *roam_info)
+{
+}
+#endif
+
 /**
  * csr_roam_process_join_res() - Process the Join results
  * @mac_ctx:          Global MAC Context
@@ -7949,6 +7962,7 @@ static void csr_roam_process_join_res(struct mac_context *mac_ctx,
 			roam_info->hs20vendor_ie = join_rsp->hs20vendor_ie;
 			roam_info->ht_operation = join_rsp->ht_operation;
 			roam_info->vht_operation = join_rsp->vht_operation;
+			csr_roam_process_he_info(join_rsp, roam_info);
 		} else {
 			if (cmd->u.roamCmd.fReassoc) {
 				roam_info->fReassocReq =