HE: Add support for txBandwidth in WCN driver

Support for txBandwidth was not present in WCN driver; add this by using
an existing ioctl. Also, for HE program reuse the sta_set_wireless_vht
functionality.

Signed-off-by: Jouni Malinen <jouni@codeaurora.org>
diff --git a/sta.c b/sta.c
index 4f1078b..2dbd823 100644
--- a/sta.c
+++ b/sta.c
@@ -5875,6 +5875,13 @@
 	 val = get_param(cmd, "txBandwidth");
 	 if (val) {
 		 switch (get_driver_type()) {
+		 case DRIVER_WCN:
+			 if (wcn_sta_set_width(dut, intf, val) < 0) {
+				 send_resp(dut, conn, SIGMA_ERROR,
+					   "ErrorCode,Failed to set txBandwidth");
+				 return 0;
+			 }
+			 break;
 		 case DRIVER_ATHEROS:
 			 if (ath_set_width(dut, conn, intf, val) < 0) {
 				 send_resp(dut, conn, SIGMA_ERROR,
@@ -5940,7 +5947,7 @@
 	if (val) {
 		if (strcasecmp(val, "11n") == 0)
 			return cmd_sta_set_11n(dut, conn, cmd);
-		if (strcasecmp(val, "VHT") == 0)
+		if (strcasecmp(val, "VHT") == 0 || strcasecmp(val, "HE") == 0)
 			return cmd_sta_set_wireless_vht(dut, conn, cmd);
 		if (strcasecmp(val, "60ghz") == 0)
 			return sta_set_wireless_60g(dut, conn, cmd);