drivers: wifi: siwx91x: Add support for TWT

- Implemented `siwx91x_set_twt_setup` and `siwx91x_set_twt_teardown` for
  handling TWT setup and teardown requests.
- Added validation for TWT parameters, including negotiation type,
  device state, and operation type.
- Integrated TWT functionality into the `wifi_mgmt_ops` structure for
  seamless management via the Zephyr Wi-Fi API.

Signed-off-by: Muzaffar Ahmed <[email protected]>

Author: muahmed-silabs

drivers/wifi/siwx91x/siwx91x_wifi.c

@@ -28,6 +28,12 @@ LOG_MODULE_REGISTER(siwx91x_wifi);
 
 NET_BUF_POOL_FIXED_DEFINE(siwx91x_tx_pool, 1, _NET_ETH_MAX_FRAME_SIZE, 0, NULL);
 
+enum {
+	REQUEST_TWT = 0,
+	SUGGEST_TWT = 1,
+	DEMAND_TWT = 2,
+};
+
 static int siwx91x_sl_to_z_mode(sl_wifi_interface_t interface)
 {
 	switch (interface) {
@@ -1067,6 +1073,137 @@ static int siwx91x_dev_init(const struct device *dev)
 	return 0;
 }
 
+static int siwx91x_convert_z_sl_twt_req_type(enum wifi_twt_setup_cmd z_req_cmd)
+{
+	switch (z_req_cmd) {
+	case WIFI_TWT_SETUP_CMD_REQUEST:
+		return REQUEST_TWT;
+	case WIFI_TWT_SETUP_CMD_SUGGEST:
+		return SUGGEST_TWT;
+	case WIFI_TWT_SETUP_CMD_DEMAND:
+		return DEMAND_TWT;
+	default:
+		return -EINVAL;
+	}
+}
+
+static int siwx91x_set_twt_setup(struct wifi_twt_params *params)
+{
+	sl_status_t status;
+	int twt_req_type = siwx91x_convert_z_sl_twt_req_type(params->setup_cmd);
+
+	sl_wifi_twt_request_t twt_req = {
+		.wake_duration_unit = 0,
+		.wake_int_mantissa = params->setup.twt_mantissa,
+		.un_announced_twt = !params->setup.announce,
+		.wake_duration = params->setup.twt_wake_interval,
+		.triggered_twt = params->setup.trigger,
+		.wake_int_exp = params->setup.twt_exponent,
+		.implicit_twt = 1,
+		.twt_flow_id = params->flow_id,
+		.twt_enable = 1,
+		.req_type = twt_req_type,
+	};
+
+	if (twt_req_type < 0) {
+		params->fail_reason = WIFI_TWT_FAIL_CMD_EXEC_FAIL;
+		return -EINVAL;
+	}
+
+	if (!params->setup.twt_info_disable) {
+		params->fail_reason = WIFI_TWT_FAIL_OPERATION_NOT_SUPPORTED;
+		return -ENOTSUP;
+	}
+
+	if (params->setup.responder) {
+		params->fail_reason = WIFI_TWT_FAIL_OPERATION_NOT_SUPPORTED;
+		return -ENOTSUP;
+	}
+
+	/* implicit -> won't do renegotiation
+	 * explicit -> must do renegotiation for each session
+	 */
+	if (!params->setup.implicit) {
+		/* explicit twt is not supported */
+		params->fail_reason = WIFI_TWT_FAIL_OPERATION_NOT_SUPPORTED;
+		return -ENOTSUP;
+	}
+
+	if (params->setup.twt_wake_interval > 255 * 256) {
+		twt_req.wake_duration_unit = 1;
+		twt_req.wake_duration = params->setup.twt_wake_interval / 256;
+	} else {
+		twt_req.wake_duration_unit = 0;
+		twt_req.wake_duration = params->setup.twt_wake_interval / 1024;
+	}
+
+	status = sl_wifi_enable_target_wake_time(&twt_req);
+	if (status != SL_STATUS_OK) {
+		params->fail_reason = WIFI_TWT_FAIL_CMD_EXEC_FAIL;
+		params->resp_status = WIFI_TWT_RESP_NOT_RECEIVED;
+		return -EINVAL;
+	}
+
+	return 0;
+}
+
+static int siwx91x_set_twt_teardown(struct wifi_twt_params *params)
+{
+	sl_status_t status;
+	sl_wifi_twt_request_t twt_req = { };
+
+	twt_req.twt_enable = 0;
+
+	if (params->teardown.teardown_all) {
+		twt_req.twt_flow_id = 0xFF;
+	} else {
+		twt_req.twt_flow_id = params->flow_id;
+	}
+
+	status = sl_wifi_disable_target_wake_time(&twt_req);
+	if (status != SL_STATUS_OK) {
+		params->fail_reason = WIFI_TWT_FAIL_CMD_EXEC_FAIL;
+		params->teardown_status = WIFI_TWT_TEARDOWN_FAILED;
+		return -EINVAL;
+	}
+
+	params->teardown_status = WIFI_TWT_TEARDOWN_SUCCESS;
+
+	return 0;
+}
+
+static int siwx91x_set_twt(const struct device *dev, struct wifi_twt_params *params)
+{
+	sl_wifi_interface_t interface = sl_wifi_get_default_interface();
+	struct siwx91x_dev *sidev = dev->data;
+
+	__ASSERT(params, "params cannot be a NULL");
+
+	if (FIELD_GET(SIWX91X_INTERFACE_MASK, interface) != SL_WIFI_CLIENT_INTERFACE) {
+		params->fail_reason = WIFI_TWT_FAIL_OPERATION_NOT_SUPPORTED;
+		return -ENOTSUP;
+	}
+
+	if (sidev->state != WIFI_STATE_DISCONNECTED && sidev->state != WIFI_STATE_INACTIVE &&
+	    sidev->state != WIFI_STATE_COMPLETED) {
+		LOG_ERR("Command given in invalid state");
+		return -EBUSY;
+	}
+
+	if (params->negotiation_type != WIFI_TWT_INDIVIDUAL) {
+		params->fail_reason = WIFI_TWT_FAIL_OPERATION_NOT_SUPPORTED;
+		return -ENOTSUP;
+	}
+
+	if (params->operation == WIFI_TWT_SETUP) {
+		return siwx91x_set_twt_setup(params);
+	} else if (params->operation == WIFI_TWT_TEARDOWN) {
+		return siwx91x_set_twt_teardown(params);
+	}
+	params->fail_reason = WIFI_TWT_FAIL_OPERATION_NOT_SUPPORTED;
+	return -ENOTSUP;
+}
+
 static const struct wifi_mgmt_ops siwx91x_mgmt = {
 	.scan			= siwx91x_scan,
 	.connect		= siwx91x_connect,
@@ -1076,6 +1213,7 @@ static const struct wifi_mgmt_ops siwx91x_mgmt = {
 	.ap_sta_disconnect	= siwx91x_ap_sta_disconnect,
 	.iface_status		= siwx91x_status,
 	.mode			= siwx91x_mode,
+	.set_twt		= siwx91x_set_twt,
 #if defined(CONFIG_NET_STATISTICS_WIFI)
 	.get_stats		= siwx91x_stats,
 #endif