qcacld-3.0: Add set bandwidth MLO 3links support

Driver cache all links channel bandwidth config from user space,
when link switch happened, and connect complete, send bandwidth
of the link id to fw.

CRs-Fixed: 3625801
Change-Id: I7d4a13075b62c9c8928d2c33bd309e9c5b085171

components/mlme/dispatcher/inc/wlan_mlme_api.h

@@ -1210,7 +1210,7 @@ void wlan_mlme_chan_stats_scan_event_cb(struct wlan_objmgr_vdev *vdev,
  * channel bandwidth
  * @psoc: pointer to psoc object
  * @vdev: pointer to vdev object
- * @vdev_id: vdev id
+ * @link_id: mlo link id
  * @ch_width: channel width to update
  *
  * Return: none
@@ -1218,7 +1218,7 @@ void wlan_mlme_chan_stats_scan_event_cb(struct wlan_objmgr_vdev *vdev,
 QDF_STATUS
 wlan_mlme_send_ch_width_update_with_notify(struct wlan_objmgr_psoc *psoc,
 					   struct wlan_objmgr_vdev *vdev,
-					   uint8_t vdev_id,
+					   uint8_t link_id,
 					   enum phy_ch_width ch_width);
 
 /**

core/hdd/inc/wlan_hdd_main.h

@@ -749,6 +749,7 @@ struct hdd_mon_set_ch_info {
  * @ch_info: monitor mode channel information
  * @ap_supports_immediate_power_save: Does the current AP allow our STA
  *    to immediately go into power save?
+ * @user_cfg_chn_width: max channel bandwidth set by user space
  */
 struct hdd_station_ctx {
 	uint32_t reg_phymode;
@@ -757,6 +758,7 @@ struct hdd_station_ctx {
 	struct hdd_connection_info cache_conn_info;
 	struct hdd_mon_set_ch_info ch_info;
 	bool ap_supports_immediate_power_save;
+	uint8_t user_cfg_chn_width;
 };
 
 /**

core/hdd/src/wlan_hdd_cfg.c

@@ -2610,6 +2610,50 @@ static void hdd_restore_sme_config(struct wlan_hdd_link_info *link_info,
 	}
 }
 
+/**
+ * wlan_update_mlo_link_chn_width() - API to update mlo link chn width
+ * @adapter: the pointer to adapter
+ * @ch_width: channel width to update
+ * @link_id: mlo link id
+ *
+ * Get link id and channel bandwidth from user space and save in link_info.
+ * When link switch happen and host driver connect done, if the link change
+ * from standby to non-standby, ch_width will send to fw again.
+ *
+ * Return: QDF_STATUS
+ */
+
+#if defined(WLAN_FEATURE_11BE_MLO) && defined(WLAN_HDD_MULTI_VDEV_SINGLE_NDEV)
+static struct wlan_hdd_link_info *
+wlan_update_mlo_link_chn_width(struct hdd_adapter *adapter,
+			       enum phy_ch_width ch_width,
+			       uint8_t link_id)
+{
+	struct wlan_hdd_link_info *link_info;
+	struct hdd_station_ctx *sta_ctx;
+
+	link_info = hdd_get_link_info_by_ieee_link_id(adapter, link_id);
+	if (!link_info)
+		return NULL;
+
+	sta_ctx = WLAN_HDD_GET_STATION_CTX_PTR(link_info);
+
+	sta_ctx->user_cfg_chn_width = ch_width;
+	hdd_debug("save ch_width:%u to link_id:%u vdev_id:%u",
+		  ch_width, link_id, link_info->vdev_id);
+
+	return link_info;
+}
+#else
+static struct wlan_hdd_link_info *
+wlan_update_mlo_link_chn_width(struct hdd_adapter *adapter,
+			       enum phy_ch_width ch_width,
+			       uint8_t link_id)
+{
+	return NULL;
+}
+#endif
+
 int hdd_update_channel_width(struct wlan_hdd_link_info *link_info,
 			     enum eSirMacHTChannelWidth chwidth,
 			     uint32_t bonding_mode, uint8_t link_id,
@@ -2617,11 +2661,10 @@ int hdd_update_channel_width(struct wlan_hdd_link_info *link_info,
 {
 	struct hdd_context *hdd_ctx;
 	int ret;
-	enum phy_ch_width ch_width = CH_WIDTH_INVALID;
+	enum phy_ch_width ch_width;
 	struct wlan_objmgr_vdev *link_vdev;
 	struct wlan_objmgr_vdev *vdev;
 	struct wlan_hdd_link_info *link_info_t;
-	bool is_mlo_link = false;
 	uint8_t link_vdev_id;
 	enum QDF_OPMODE op_mode;
 	QDF_STATUS status;
@@ -2634,23 +2677,42 @@ int hdd_update_channel_width(struct wlan_hdd_link_info *link_info,
 		return -EINVAL;
 	}
 
-	vdev = link_info->vdev;
+	op_mode = link_info->adapter->device_mode;
+	if (op_mode != QDF_STA_MODE) {
+		hdd_debug("vdev %d: op mode %d, CW update not supported",
+			  vdev_id, op_mode);
+		return -EINVAL;
+	}
+
+	vdev = hdd_objmgr_get_vdev_by_user(link_info, WLAN_OSIF_ID);
 	if (!vdev) {
-		mlme_legacy_err("vdev %d: vdev not found", vdev_id);
-		return QDF_STATUS_E_FAILURE;
+		hdd_err("vdev %d: vdev not found", vdev_id);
+		return -EINVAL;
 	}
 
-	if (wlan_vdev_mlme_is_mlo_vdev(vdev) && link_id != 0xFF) {
-		link_vdev = mlo_get_vdev_by_link_id(vdev, link_id,
-						    WLAN_MLME_OBJMGR_ID);
-		if (!link_vdev) {
-			mlme_legacy_debug("vdev is null for the link id:%u",
-					  link_id);
-			wlan_objmgr_vdev_release_ref(vdev, WLAN_MLME_OBJMGR_ID);
-			return QDF_STATUS_E_FAILURE;
+	ch_width = hdd_convert_chwidth_to_phy_chwidth(chwidth);
+
+	/**
+	 * Link_id check is for disconnect restore process.
+	 * Disconnect will not update channel bandwidth into cache struct.
+	 */
+	if (wlan_vdev_mlme_is_mlo_vdev(vdev) &&
+	    link_id != WLAN_INVALID_LINK_ID) {
+		link_info_t = wlan_update_mlo_link_chn_width(link_info->adapter,
+							     ch_width, link_id);
+		if (!link_info_t) {
+			hdd_objmgr_put_vdev_by_user(vdev, WLAN_OSIF_ID);
+			return -EINVAL;
 		}
-		is_mlo_link = true;
-		link_vdev_id = wlan_vdev_get_id(link_vdev);
+
+		hdd_objmgr_put_vdev_by_user(vdev, WLAN_OSIF_ID);
+
+		link_vdev = hdd_objmgr_get_vdev_by_user(link_info_t,
+							WLAN_OSIF_ID);
+		if (!link_vdev)
+			return 0;
+
+		link_vdev_id = link_info_t->vdev_id;
 		status = wlan_mlme_get_bw_no_punct(hdd_ctx->psoc,
 						   link_vdev,
 						   wlan_vdev_mlme_get_des_chan(link_vdev),
@@ -2660,43 +2722,31 @@ int hdd_update_channel_width(struct wlan_hdd_link_info *link_info,
 	} else {
 		link_vdev = vdev;
 		link_vdev_id = vdev_id;
-		mlme_legacy_debug("vdev mlme is not mlo vdev");
-	}
-
-	link_info_t = hdd_get_link_info_by_vdev(hdd_ctx, link_vdev_id);
-	if (!link_info_t) {
-		mlme_legacy_debug("vdev mlme is not mlo vdev");
-		return QDF_STATUS_E_FAILURE;
+		link_info_t = link_info;
 	}
 
 	if (ucfg_mlme_is_chwidth_with_notify_supported(hdd_ctx->psoc) &&
 	    hdd_cm_is_vdev_connected(link_info_t)) {
-		op_mode = wlan_vdev_mlme_get_opmode(vdev);
-		if (op_mode != QDF_STA_MODE) {
-			mlme_legacy_debug("vdev %d: op mode %d, CW update not supported",
-				          link_vdev_id, op_mode);
-			wlan_objmgr_vdev_release_ref(vdev, WLAN_MLME_OBJMGR_ID);
-			return QDF_STATUS_E_NOSUPPORT;
-		}
-
 		ch_width = hdd_convert_chwidth_to_phy_chwidth(chwidth);
 		hdd_debug("vdev %d : process update ch width request to %d",
 			  link_vdev_id, ch_width);
-
 		status = ucfg_mlme_send_ch_width_update_with_notify(hdd_ctx->psoc,
 								    link_vdev,
 								    ch_width,
 								    link_vdev_id);
-		if (is_mlo_link)
-			wlan_objmgr_vdev_release_ref(link_vdev,
-						     WLAN_MLME_OBJMGR_ID);
-		if (QDF_IS_STATUS_ERROR(status))
+
+		if (QDF_IS_STATUS_ERROR(status)) {
+			hdd_objmgr_put_vdev_by_user(link_vdev, WLAN_OSIF_ID);
 			return -EIO;
+		}
 		status = hdd_update_bss_rate_flags(link_info_t, hdd_ctx->psoc,
 						   ch_width);
-		if (QDF_IS_STATUS_ERROR(status))
+		if (QDF_IS_STATUS_ERROR(status)) {
+			hdd_objmgr_put_vdev_by_user(link_vdev, WLAN_OSIF_ID);
 			return -EIO;
+		}
 	}
+	hdd_objmgr_put_vdev_by_user(link_vdev, WLAN_OSIF_ID);
 
 	ret = wma_cli_set_command(link_vdev_id, wmi_vdev_param_chwidth,
 				  chwidth, VDEV_CMD);

core/hdd/src/wlan_hdd_cfg80211.c

@@ -11093,8 +11093,8 @@ static int hdd_set_channel_width(struct wlan_hdd_link_info *link_info,
 				 struct nlattr *tb[])
 {
 	int rem;
-	uint8_t nl80211_chwidth = 0xFF;
-	uint8_t link_id = 0xFF;
+	uint8_t nl80211_chwidth = CH_WIDTH_INVALID;
+	uint8_t link_id = WLAN_INVALID_LINK_ID;
 	struct nlattr *tb2[QCA_WLAN_VENDOR_ATTR_CONFIG_MAX + 1];
 	struct nlattr *curr_attr;
 	struct nlattr *chn_bd = NULL;
@@ -11149,7 +11149,7 @@ static int hdd_set_channel_width(struct wlan_hdd_link_info *link_info,
 		}
 	}
 
-	if (link_id != 0xFF)
+	if (link_id != WLAN_INVALID_LINK_ID)
 		goto set_chan_width;
 
 skip_mlo:
@@ -11166,7 +11166,9 @@ skip_mlo:
 		hdd_err("Invalid channel width %u", chwidth);
 		return -EINVAL;
 	}
+
 set_chan_width:
+	hdd_debug("channel width:%u, link_id:%u", chwidth, link_id);
 	return hdd_set_mac_chan_width(link_info, chwidth, link_id, true);
 }
 
@@ -12581,6 +12583,7 @@ static int hdd_get_channel_width(struct wlan_hdd_link_info *link_info,
  *
  * Return: 0 on success; error number otherwise
  */
+#ifdef WLAN_FEATURE_11BE_MLO
 static int hdd_get_mlo_max_band_info(struct wlan_hdd_link_info *link_info,
 				     struct sk_buff *skb,
 				     const struct nlattr *attr)
@@ -12592,7 +12595,10 @@ static int hdd_get_mlo_max_band_info(struct wlan_hdd_link_info *link_info,
 	uint32_t link_id = 0;
 	struct wlan_objmgr_vdev *vdev, *link_vdev;
 	struct wlan_channel *bss_chan;
+	struct wlan_hdd_link_info *link_info_t;
+	struct hdd_station_ctx *sta_ctx;
 	uint8_t nl80211_chwidth;
+	uint8_t chn_width;
 	int8_t ret = 0;
 
 	chwidth = wma_cli_get_command(link_info->vdev_id,
@@ -12606,46 +12612,64 @@ static int hdd_get_mlo_max_band_info(struct wlan_hdd_link_info *link_info,
 	if (!vdev)
 		return -EINVAL;
 
-	if (wlan_vdev_mlme_is_mlo_vdev(vdev)) {
-		mlo_bd_info = nla_nest_start(skb, CONFIG_MLO_LINKS);
-		for (link_id = 0; link_id < WLAN_MAX_LINK_ID; link_id++) {
-			link_vdev = mlo_get_vdev_by_link_id(vdev, link_id,
-							    WLAN_OSIF_ID);
-			if (!link_vdev)
-				continue;
+	if (!wlan_vdev_mlme_is_mlo_vdev(vdev)) {
+		hdd_err("not mlo vdev");
+		hdd_objmgr_put_vdev_by_user(vdev, WLAN_OSIF_ID);
+		return -EINVAL;
+	}
 
-			mlo_bd = nla_nest_start(skb, i);
-			if (!mlo_bd) {
-				hdd_err("nla_nest_start fail");
-				ret = -EINVAL;
-				goto end;
-			}
+	mlo_bd_info = nla_nest_start(skb, CONFIG_MLO_LINKS);
+	hdd_adapter_for_each_link_info(link_info->adapter, link_info_t) {
+		sta_ctx = WLAN_HDD_GET_STATION_CTX_PTR(link_info_t);
+		if (sta_ctx->conn_info.ieee_link_id == WLAN_INVALID_LINK_ID)
+			continue;
+
+		link_id = sta_ctx->conn_info.ieee_link_id;
+
+		mlo_bd = nla_nest_start(skb, i);
+		if (!mlo_bd) {
+			hdd_err("nla_nest_start fail");
+			ret = -EINVAL;
+			goto end;
+		}
+
+		if (nla_put_u8(skb, CONFIG_MLO_LINK_ID, link_id)) {
+			hdd_err("nla_put failure");
+			ret = -EINVAL;
+			goto end;
+		}
+
+		link_vdev = mlo_get_vdev_by_link_id(vdev, link_id,
+						    WLAN_OSIF_ID);
+
+		if (link_vdev) {
 			bss_chan = wlan_vdev_mlme_get_bss_chan(link_vdev);
 			if (!bss_chan) {
 				hdd_err("fail to get bss_chan info");
 				ret = -EINVAL;
 				goto end;
 			}
-			if (nla_put_u8(skb, CONFIG_MLO_LINK_ID, link_id)) {
-				hdd_err("nla_put failure");
-				ret = -EINVAL;
-				goto end;
-			}
+			chn_width = bss_chan->ch_width;
+		} else if (link_info_t->vdev_id == WLAN_INVALID_VDEV_ID) {
+			chn_width = sta_ctx->user_cfg_chn_width;
+		} else {
+			chn_width = CH_WIDTH_INVALID;
+		}
 
-			nl80211_chwidth = hdd_phy_chwidth_to_nl80211_chwidth(bss_chan->ch_width);
-			if (nla_put_u8(skb, CONFIG_CHANNEL_WIDTH,
-				       nl80211_chwidth)) {
-				hdd_err("nla_put failure");
-				ret = -EINVAL;
-				goto end;
-			}
-			nla_nest_end(skb, mlo_bd);
-			i++;
-			hdd_objmgr_put_vdev_by_user(link_vdev, WLAN_OSIF_ID);
+		hdd_debug("get link_id:%u ch width:%u", link_id, chn_width);
+
+		nl80211_chwidth = hdd_phy_chwidth_to_nl80211_chwidth(chn_width);
+		if (nla_put_u8(skb, CONFIG_CHANNEL_WIDTH, nl80211_chwidth)) {
+			hdd_err("nla_put failure");
+			ret = -EINVAL;
+			goto end;
 		}
-		nla_nest_end(skb, mlo_bd_info);
-	}
+		nla_nest_end(skb, mlo_bd);
+		i++;
 
+		hdd_objmgr_put_vdev_by_user(link_vdev, WLAN_OSIF_ID);
+	}
+	nla_nest_end(skb, mlo_bd_info);
 end:
 	hdd_objmgr_put_vdev_by_user(vdev, WLAN_OSIF_ID);
 
@@ -12654,6 +12678,14 @@ end:
 
 	return ret;
 }
+#else
+static int hdd_get_mlo_max_band_info(struct wlan_hdd_link_info *link_info,
+				     struct sk_buff *skb,
+				     const struct nlattr *attr)
+{
+	return 0;
+}
+#endif
 
 /**
  * hdd_get_dynamic_bw() - Get dynamic bandwidth disabled / enabled

core/hdd/src/wlan_hdd_cm_connect.c

@@ -1516,6 +1516,62 @@ static void hdd_set_immediate_power_save(struct hdd_adapter *adapter,
 }
 #endif
 
+#if defined(WLAN_FEATURE_11BE_MLO) && defined(WLAN_HDD_MULTI_VDEV_SINGLE_NDEV)
+static QDF_STATUS
+hdd_cm_mlme_send_standby_link_chn_width(struct hdd_adapter *adapter,
+					struct wlan_objmgr_vdev *vdev)
+{
+	struct wlan_objmgr_psoc *psoc;
+	struct wlan_hdd_link_info *link_info;
+	struct hdd_station_ctx *sta_ctx;
+	uint8_t link_id = wlan_vdev_get_link_id(vdev);
+	uint8_t ch_width;
+	enum phy_ch_width connection_ch_width = CH_WIDTH_INVALID;
+
+	psoc = wlan_vdev_get_psoc(vdev);
+	if (!psoc) {
+		hdd_err("Failed to get PSOC Object");
+		return QDF_STATUS_E_INVAL;
+	}
+
+	link_info = hdd_get_link_info_by_ieee_link_id(adapter, link_id);
+	if (!link_info) {
+		hdd_err("Link info not found by linkid:%u", link_id);
+		return QDF_STATUS_E_INVAL;
+	}
+	sta_ctx = WLAN_HDD_GET_STATION_CTX_PTR(link_info);
+	ch_width = sta_ctx->user_cfg_chn_width;
+
+	wlan_mlme_get_sta_ch_width(vdev, &connection_ch_width);
+
+	if (ch_width == CH_WIDTH_INVALID) {
+		hdd_debug("no cached bandwidth for the link %u", link_id);
+		return QDF_STATUS_SUCCESS;
+	}
+
+	if (ch_width == connection_ch_width) {
+		hdd_debug("user config max bd same as connection ch bw:%u",
+			  ch_width);
+		return QDF_STATUS_SUCCESS;
+	}
+
+	hdd_debug("send vdev id:%u, chwidth:%u", link_info->vdev_id,
+		  ch_width);
+
+	wlan_mlme_send_ch_width_update_with_notify(psoc, vdev,
+						   link_info->vdev_id,
+						   ch_width);
+	return QDF_STATUS_SUCCESS;
+}
+#else
+static QDF_STATUS
+hdd_cm_mlme_send_standby_link_chn_width(struct hdd_adapter *adapter,
+					struct wlan_objmgr_vdev *vdev)
+{
+	return QDF_STATUS_SUCCESS;
+}
+#endif
+
 static void
 hdd_cm_connect_success_pre_user_update(struct wlan_objmgr_vdev *vdev,
 				       struct wlan_cm_connect_resp *rsp)
@@ -1537,7 +1593,7 @@ hdd_cm_connect_success_pre_user_update(struct wlan_objmgr_vdev *vdev,
 	struct hdd_adapter *assoc_link_adapter;
 	bool is_immediate_power_save;
 	struct wlan_hdd_link_info *link_info;
-	QDF_STATUS status;
+	QDF_STATUS status = QDF_STATUS_E_INVAL;
 	bool alt_pipe;
 
 	hdd_ctx = cds_get_context(QDF_MODULE_ID_HDD);
@@ -1605,6 +1661,11 @@ hdd_cm_connect_success_pre_user_update(struct wlan_objmgr_vdev *vdev,
 
 	hdd_cm_handle_assoc_event(vdev, rsp->bssid.bytes);
 
+	if (ucfg_cm_is_link_switch_connect_resp(rsp)) {
+		if (hdd_cm_mlme_send_standby_link_chn_width(adapter, vdev))
+			hdd_debug("send standby link chn width fail");
+	}
+
 	/*
 	 * check update hdd_send_update_beacon_ies_event,
 	 * hdd_send_ft_assoc_response,

core/hdd/src/wlan_hdd_main.c

@@ -14740,6 +14740,7 @@ void hdd_adapter_reset_station_ctx(struct hdd_adapter *adapter)
 		qdf_mem_zero(&sta_ctx->conn_info.bssid, QDF_MAC_ADDR_SIZE);
 
 		hdd_cm_clear_ieee_link_id(link_info);
+		sta_ctx->user_cfg_chn_width = CH_WIDTH_INVALID;
 	}
 }