wil6210: add support for power save enable / disable

New power management wmi commands provide the ability to change
the device power save profile (enable / disable power save).

Signed-off-by: Maya Erez <qca_merez@qca.qualcomm.com>
Signed-off-by: Kalle Valo <kvalo@qca.qualcomm.com>

3 files changed, 67 insertions, 1 deletions

diff --git a/drivers/net/wireless/ath/wil6210/cfg80211.c b/drivers/net/wireless/ath/wil6210/cfg80211.c
index d117240d9a73..c6dd7a3892f9 100644
--- a/drivers/net/wireless/ath/wil6210/cfg80211.c
+++ b/drivers/net/wireless/ath/wil6210/cfg80211.c
@@ -1424,6 +1424,34 @@ static void wil_cfg80211_stop_p2p_device(struct wiphy *wiphy,
         mutex_unlock(&wil->mutex);
 }
 
+static int wil_cfg80211_set_power_mgmt(struct wiphy *wiphy,
+                                       struct net_device *dev,
+                                       bool enabled, int timeout)
+{
+        struct wil6210_priv *wil = wiphy_to_wil(wiphy);
+        enum wmi_ps_profile_type ps_profile;
+        int rc;
+
+        if (!test_bit(WMI_FW_CAPABILITY_PS_CONFIG, wil->fw_capabilities)) {
+                wil_err(wil, "set_power_mgmt not supported\n");
+                return -EOPNOTSUPP;
+        }
+
+        wil_dbg_misc(wil, "enabled=%d, timeout=%d\n",
+                     enabled, timeout);
+
+        if (enabled)
+                ps_profile = WMI_PS_PROFILE_TYPE_DEFAULT;
+        else
+                ps_profile = WMI_PS_PROFILE_TYPE_PS_DISABLED;
+
+        rc  = wmi_ps_dev_profile_cfg(wil, ps_profile);
+        if (rc)
+                wil_err(wil, "wmi_ps_dev_profile_cfg failed (%d)\n", rc);
+
+        return rc;
+}
+
 static struct cfg80211_ops wil_cfg80211_ops = {
         .add_virtual_intf = wil_cfg80211_add_iface,
         .del_virtual_intf = wil_cfg80211_del_iface,
@@ -1450,6 +1478,7 @@ static struct cfg80211_ops wil_cfg80211_ops = {
         /* P2P device */
         .start_p2p_device = wil_cfg80211_start_p2p_device,
         .stop_p2p_device = wil_cfg80211_stop_p2p_device,
+        .set_power_mgmt = wil_cfg80211_set_power_mgmt,
 };
 
 static void wil_wiphy_init(struct wiphy *wiphy)
@@ -1466,7 +1495,8 @@ static void wil_wiphy_init(struct wiphy *wiphy)
                                  BIT(NL80211_IFTYPE_MONITOR);
         wiphy->flags |= WIPHY_FLAG_HAVE_AP_SME |
                         WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL |
-                        WIPHY_FLAG_AP_PROBE_RESP_OFFLOAD;
+                        WIPHY_FLAG_AP_PROBE_RESP_OFFLOAD |
+                        WIPHY_FLAG_PS_ON_BY_DEFAULT;
         dev_dbg(wiphy_dev(wiphy), "%s : flags = 0x%08x\n",
                 __func__, wiphy->flags);
         wiphy->probe_resp_offload =
diff --git a/drivers/net/wireless/ath/wil6210/wil6210.h b/drivers/net/wireless/ath/wil6210/wil6210.h
index 12cd81bccb47..4b22ea4bc4cf 100644
--- a/drivers/net/wireless/ath/wil6210/wil6210.h
+++ b/drivers/net/wireless/ath/wil6210/wil6210.h
@@ -819,6 +819,8 @@ int wmi_delba_tx(struct wil6210_priv *wil, u8 ringid, u16 reason);
 int wmi_delba_rx(struct wil6210_priv *wil, u8 cidxtid, u16 reason);
 int wmi_addba_rx_resp(struct wil6210_priv *wil, u8 cid, u8 tid, u8 token,
                       u16 status, bool amsdu, u16 agg_wsize, u16 timeout);
+int wmi_ps_dev_profile_cfg(struct wil6210_priv *wil,
+                           enum wmi_ps_profile_type ps_profile);
 int wil_addba_rx_request(struct wil6210_priv *wil, u8 cidxtid,
                          u8 dialog_token, __le16 ba_param_set,
                          __le16 ba_timeout, __le16 ba_seq_ctrl);
diff --git a/drivers/net/wireless/ath/wil6210/wmi.c b/drivers/net/wireless/ath/wil6210/wmi.c
index 890960e9b1d3..8c60437e57e0 100644
--- a/drivers/net/wireless/ath/wil6210/wmi.c
+++ b/drivers/net/wireless/ath/wil6210/wmi.c
@@ -1563,6 +1563,40 @@ int wmi_addba_rx_resp(struct wil6210_priv *wil, u8 cid, u8 tid, u8 token,
         return rc;
 }
 
+int wmi_ps_dev_profile_cfg(struct wil6210_priv *wil,
+                           enum wmi_ps_profile_type ps_profile)
+{
+        int rc;
+        struct wmi_ps_dev_profile_cfg_cmd cmd = {
+                .ps_profile = ps_profile,
+        };
+        struct {
+                struct wmi_cmd_hdr wmi;
+                struct wmi_ps_dev_profile_cfg_event evt;
+        } __packed reply;
+        u32 status;
+
+        wil_dbg_wmi(wil, "Setting ps dev profile %d\n", ps_profile);
+
+        reply.evt.status = cpu_to_le32(WMI_PS_CFG_CMD_STATUS_ERROR);
+
+        rc = wmi_call(wil, WMI_PS_DEV_PROFILE_CFG_CMDID, &cmd, sizeof(cmd),
+                      WMI_PS_DEV_PROFILE_CFG_EVENTID, &reply, sizeof(reply),
+                      100);
+        if (rc)
+                return rc;
+
+        status = le32_to_cpu(reply.evt.status);
+
+        if (status != WMI_PS_CFG_CMD_STATUS_SUCCESS) {
+                wil_err(wil, "ps dev profile cfg failed with status %d\n",
+                        status);
+                rc = -EINVAL;
+        }
+
+        return rc;
+}
+
 void wmi_event_flush(struct wil6210_priv *wil)
 {
         struct pending_wmi_event *evt, *t;