qcacld-3.0: Set peer QOS Flag for eht/he capable peer

Set peer QOS Flag if peer is either eht capable or he
capable.

Change-Id: I93fddd7aed3a687c79442d83fb4b359d00022f6c
CRs-Fixed: 3105589

core/wma/src/wma_eht.c

@@ -1,5 +1,6 @@
 /*
  * Copyright (c) 2021, The Linux Foundation. All rights reserved.
+ * Copyright (c) 2022 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
@@ -386,11 +387,12 @@ void wma_populate_peer_eht_cap(struct peer_assoc_params *peer,
 	uint32_t *phy_cap = peer->peer_eht_cap_phyinfo;
 	uint32_t *mac_cap = peer->peer_eht_cap_macinfo;
 
-	if (params->eht_capable)
-		peer->eht_flag = 1;
-	else
+	if (!params->eht_capable)
 		return;
 
+	peer->eht_flag = 1;
+	peer->qos_flag = 1;
+
 	/* EHT MAC Capabilities */
 	WMI_EHTCAP_MAC_NSEPPRIACCESS_SET(mac_cap, eht_cap->nsep_pri_access);
 	WMI_EHTCAP_MAC_EHTOMCTRL_SET(mac_cap, eht_cap->eht_om_ctl);

core/wma/src/wma_he.c

@@ -1,6 +1,6 @@
 /*
  * Copyright (c) 2017-2020 The Linux Foundation. All rights reserved.
- * Copyright (c) 2021 Qualcomm Innovation Center, Inc. All rights reserved.
+ * Copyright (c) 2021-2022 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
@@ -1118,11 +1118,12 @@ void wma_populate_peer_he_cap(struct peer_assoc_params *peer,
 	uint32_t mac_cap[PSOC_HOST_MAX_MAC_SIZE] = {0}, he_ops = 0;
 	uint8_t temp, i, chan_width;
 
-	if (params->he_capable)
-		peer->he_flag = 1;
-	else
+	if (!params->he_capable)
 		return;
 
+	peer->he_flag = 1;
+	peer->qos_flag = 1;
+
 	/* HE MAC capabilities */
 	WMI_HECAP_MAC_HECTRL_SET(mac_cap[0], he_cap->htc_he);
 	WMI_HECAP_MAC_TWTREQ_SET(mac_cap[0], he_cap->twt_request);