HE: Add support for setting ADDBA buffer size (WCN)

This adds support for using the ADDBAResp_BufSize and ADDBAReq_BufSize
parameters.

Signed-off-by: Jouni Malinen <jouni@codeaurora.org>
diff --git a/sta.c b/sta.c
index a3a04ac..9eb4def 100644
--- a/sta.c
+++ b/sta.c
@@ -7066,6 +7066,39 @@
 			return 0;
 		}
 	}
+
+	val = get_param(cmd, "ADDBAResp_BufSize");
+	if (val) {
+		int buf_size;
+
+		if (strcasecmp(val, "gt64") == 0)
+			buf_size = 256;
+		else
+			buf_size = 64;
+		if (get_driver_type() == DRIVER_WCN &&
+		    sta_set_addba_buf_size(dut, intf, buf_size)) {
+			send_resp(dut, conn, SIGMA_ERROR,
+				  "ErrorCode,set addbaresp_buff_size failed");
+			return 0;
+		}
+	}
+
+	val = get_param(cmd, "ADDBAReq_BufSize");
+	if (val) {
+		int buf_size;
+
+		if (strcasecmp(val, "gt64") == 0)
+			buf_size = 256;
+		else
+			buf_size = 64;
+		if (get_driver_type() == DRIVER_WCN &&
+		    sta_set_addba_buf_size(dut, intf, buf_size)) {
+			send_resp(dut, conn, SIGMA_ERROR,
+				  "ErrorCode,set addbareq_buff_size failed");
+			return 0;
+		}
+	}
+
 	return cmd_sta_set_wireless_common(intf, dut, conn, cmd);
 }