ap_set_wireless: Add ABFTLRang support for 60G AP tests

The ABFTLRang parameter allows to specify a non-default ABFT length
which some AP tests need.

Signed-off-by: Alexei Avshalom Lazar <ailizaro@codeaurora.org>
diff --git a/ap.c b/ap.c
index f7cc8cb..d279736 100644
--- a/ap.c
+++ b/ap.c
@@ -1510,6 +1510,24 @@
 		dut->ap_num_ese_allocs = num_allocs;
 	}
 
+	if (is_60g_sigma_dut(dut)) {
+		unsigned int abft_len = 1; /* default is one slot */
+
+		val = get_param(cmd, "ABFTLRang");
+		if (val) {
+			sigma_dut_print(dut, DUT_MSG_DEBUG,
+					"ABFTLRang parameter %s", val);
+			if (strcasecmp(val, "Gt1") == 0)
+				abft_len = 2; /* 2 slots in this case */
+		}
+
+		if (sta_set_60g_abft_len(dut, conn, abft_len)) {
+			send_resp(dut, conn, SIGMA_ERROR,
+				  "ErrorCode,Can't set ABFT length");
+			return STATUS_SENT;
+		}
+	}
+
 	return 1;
 }
 
diff --git a/sigma_dut.h b/sigma_dut.h
index 94e3bbb..a9e44e4 100644
--- a/sigma_dut.h
+++ b/sigma_dut.h
@@ -911,6 +911,8 @@
 		       const char *val);
 int ath_set_width(struct sigma_dut *dut, struct sigma_conn *conn,
 		  const char *intf, const char *val);
+int sta_set_60g_abft_len(struct sigma_dut *dut, struct sigma_conn *conn,
+			 int abft_len);
 int wil6210_send_frame_60g(struct sigma_dut *dut, struct sigma_conn *conn,
 			   struct sigma_cmd *cmd);
 int hwaddr_aton(const char *txt, unsigned char *addr);
diff --git a/sta.c b/sta.c
index 136a291..40832b3 100644
--- a/sta.c
+++ b/sta.c
@@ -5361,8 +5361,8 @@
 }
 
 
-static int sta_set_60g_abft_len(struct sigma_dut *dut, struct sigma_conn *conn,
-				int abft_len)
+int sta_set_60g_abft_len(struct sigma_dut *dut, struct sigma_conn *conn,
+			 int abft_len)
 {
 	switch (get_driver_type()) {
 	case DRIVER_WIL6210: