// SPDX-License-Identifier: GPL-2.0-only
/*
 * This file is part of wl1271
 *
 * Copyright (C) 2010 Nokia Corporation
 *
 * Contact: Luciano Coelho <luciano.coelho@nokia.com>
 */
#include "testmode.h"

#include <linux/pm_runtime.h>
#include <linux/slab.h>
#include <net/genetlink.h>

#include "wlcore.h"
#include "debug.h"
#include "acx.h"
#include "io.h"

#define WL1271_TM_MAX_DATA_LENGTH 1024

enum wl1271_tm_commands {
 WL1271_TM_CMD_UNSPEC,
 WL1271_TM_CMD_TEST,
 WL1271_TM_CMD_INTERROGATE,
 WL1271_TM_CMD_CONFIGURE,
 WL1271_TM_CMD_NVS_PUSH,  /* Not in use. Keep to not break ABI */
 WL1271_TM_CMD_SET_PLT_MODE,
 WL1271_TM_CMD_RECOVER,  /* Not in use. Keep to not break ABI */
 WL1271_TM_CMD_GET_MAC,

 __WL1271_TM_CMD_AFTER_LAST
};
#define WL1271_TM_CMD_MAX (__WL1271_TM_CMD_AFTER_LAST - 1)

enum wl1271_tm_attrs {
 WL1271_TM_ATTR_UNSPEC,
 WL1271_TM_ATTR_CMD_ID,
 WL1271_TM_ATTR_ANSWER,
 WL1271_TM_ATTR_DATA,
 WL1271_TM_ATTR_IE_ID,
 WL1271_TM_ATTR_PLT_MODE,

 __WL1271_TM_ATTR_AFTER_LAST
};
#define WL1271_TM_ATTR_MAX (__WL1271_TM_ATTR_AFTER_LAST - 1)

static const struct nla_policy wl1271_tm_policy[WL1271_TM_ATTR_MAX + 1] = {
 [WL1271_TM_ATTR_CMD_ID] = { .type = NLA_U32 },
 [WL1271_TM_ATTR_ANSWER] = { .type = NLA_U8 },
 [WL1271_TM_ATTR_DATA] =  { .type = NLA_BINARY,
       .len = WL1271_TM_MAX_DATA_LENGTH },
 [WL1271_TM_ATTR_IE_ID] = { .type = NLA_U32 },
 [WL1271_TM_ATTR_PLT_MODE] = { .type = NLA_U32 },
};


static int wl1271_tm_cmd_test(struct wl1271 *wl, struct nlattr *tb[])
{
 int buf_len, ret, len;
 struct sk_buff *skb;
 void *buf;
 u8 answer = 0;

 wl1271_debug(DEBUG_TESTMODE, "testmode cmd test");

 if (!tb[WL1271_TM_ATTR_DATA])
  return -EINVAL;

 buf = nla_data(tb[WL1271_TM_ATTR_DATA]);
 buf_len = nla_len(tb[WL1271_TM_ATTR_DATA]);

 if (tb[WL1271_TM_ATTR_ANSWER])
  answer = nla_get_u8(tb[WL1271_TM_ATTR_ANSWER]);

 if (buf_len > sizeof(struct wl1271_command))
  return -EMSGSIZE;

 mutex_lock(&wl->mutex);

 if (unlikely(wl->state != WLCORE_STATE_ON)) {
  ret = -EINVAL;
  goto out;
 }

 ret = pm_runtime_resume_and_get(wl->dev);
 if (ret < 0)
  goto out;

 ret = wl1271_cmd_test(wl, buf, buf_len, answer);
 if (ret < 0) {
  wl1271_warning("testmode cmd test failed: %d", ret);
  goto out_sleep;
 }

 if (answer) {
  /* If we got bip calibration answer print radio status */
  struct wl1271_cmd_cal_p2g *params =
   (struct wl1271_cmd_cal_p2g *) buf;

  s16 radio_status = (s16) le16_to_cpu(params->radio_status);

  if (params->test.id == TEST_CMD_P2G_CAL &&
      radio_status < 0)
   wl1271_warning("testmode cmd: radio status=%d",
     radio_status);
  else
   wl1271_info("testmode cmd: radio status=%d",
     radio_status);

  len = nla_total_size(buf_len);
  skb = cfg80211_testmode_alloc_reply_skb(wl->hw->wiphy, len);
  if (!skb) {
   ret = -ENOMEM;
   goto out_sleep;
  }

  if (nla_put(skb, WL1271_TM_ATTR_DATA, buf_len, buf)) {
   kfree_skb(skb);
   ret = -EMSGSIZE;
   goto out_sleep;
  }

  ret = cfg80211_testmode_reply(skb);
  if (ret < 0)
   goto out_sleep;
 }

out_sleep:
 pm_runtime_mark_last_busy(wl->dev);
 pm_runtime_put_autosuspend(wl->dev);
out:
 mutex_unlock(&wl->mutex);

 return ret;
}

static int wl1271_tm_cmd_interrogate(struct wl1271 *wl, struct nlattr *tb[])
{
 int ret;
 struct wl1271_command *cmd;
 struct sk_buff *skb;
 u8 ie_id;

 wl1271_debug(DEBUG_TESTMODE, "testmode cmd interrogate");

 if (!tb[WL1271_TM_ATTR_IE_ID])
  return -EINVAL;

 ie_id = nla_get_u8(tb[WL1271_TM_ATTR_IE_ID]);

 mutex_lock(&wl->mutex);

 if (unlikely(wl->state != WLCORE_STATE_ON)) {
  ret = -EINVAL;
  goto out;
 }

 ret = pm_runtime_resume_and_get(wl->dev);
 if (ret < 0)
  goto out;

 cmd = kzalloc(sizeof(*cmd), GFP_KERNEL);
 if (!cmd) {
  ret = -ENOMEM;
  goto out_sleep;
 }

 ret = wl1271_cmd_interrogate(wl, ie_id, cmd,
         sizeof(struct acx_header), sizeof(*cmd));
 if (ret < 0) {
  wl1271_warning("testmode cmd interrogate failed: %d", ret);
  goto out_free;
 }

 skb = cfg80211_testmode_alloc_reply_skb(wl->hw->wiphy, sizeof(*cmd));
 if (!skb) {
  ret = -ENOMEM;
  goto out_free;
 }

 if (nla_put(skb, WL1271_TM_ATTR_DATA, sizeof(*cmd), cmd)) {
  kfree_skb(skb);
  ret = -EMSGSIZE;
  goto out_free;
 }

 ret = cfg80211_testmode_reply(skb);
 if (ret < 0)
  goto out_free;

out_free:
 kfree(cmd);
out_sleep:
 pm_runtime_mark_last_busy(wl->dev);
 pm_runtime_put_autosuspend(wl->dev);
out:
 mutex_unlock(&wl->mutex);

 return ret;
}

static int wl1271_tm_cmd_configure(struct wl1271 *wl, struct nlattr *tb[])
{
 int buf_len, ret;
 void *buf;
 u8 ie_id;

 wl1271_debug(DEBUG_TESTMODE, "testmode cmd configure");

 if (!tb[WL1271_TM_ATTR_DATA])
  return -EINVAL;
 if (!tb[WL1271_TM_ATTR_IE_ID])
  return -EINVAL;

 ie_id = nla_get_u8(tb[WL1271_TM_ATTR_IE_ID]);
 buf = nla_data(tb[WL1271_TM_ATTR_DATA]);
 buf_len = nla_len(tb[WL1271_TM_ATTR_DATA]);

 if (buf_len > sizeof(struct wl1271_command))
  return -EMSGSIZE;

 mutex_lock(&wl->mutex);
 ret = wl1271_cmd_configure(wl, ie_id, buf, buf_len);
 mutex_unlock(&wl->mutex);

 if (ret < 0) {
  wl1271_warning("testmode cmd configure failed: %d", ret);
  return ret;
 }

 return 0;
}

static int wl1271_tm_detect_fem(struct wl1271 *wl, struct nlattr *tb[])
{
 /* return FEM type */
 int ret, len;
 struct sk_buff *skb;

 ret = wl1271_plt_start(wl, PLT_FEM_DETECT);
 if (ret < 0)
  goto out;

 mutex_lock(&wl->mutex);

 len = nla_total_size(sizeof(wl->fem_manuf));
 skb = cfg80211_testmode_alloc_reply_skb(wl->hw->wiphy, len);
 if (!skb) {
  ret = -ENOMEM;
  goto out_mutex;
 }

 if (nla_put(skb, WL1271_TM_ATTR_DATA, sizeof(wl->fem_manuf),
           &wl->fem_manuf)) {
  kfree_skb(skb);
  ret = -EMSGSIZE;
  goto out_mutex;
 }

 ret = cfg80211_testmode_reply(skb);

out_mutex:
 mutex_unlock(&wl->mutex);

 /* We always stop plt after DETECT mode */
 wl1271_plt_stop(wl);
out:
 return ret;
}

static int wl1271_tm_cmd_set_plt_mode(struct wl1271 *wl, struct nlattr *tb[])
{
 u32 val;
 int ret;

 wl1271_debug(DEBUG_TESTMODE, "testmode cmd set plt mode");

 if (!tb[WL1271_TM_ATTR_PLT_MODE])
  return -EINVAL;

 val = nla_get_u32(tb[WL1271_TM_ATTR_PLT_MODE]);

 switch (val) {
 case PLT_OFF:
  ret = wl1271_plt_stop(wl);
  break;
 case PLT_ON:
 case PLT_CHIP_AWAKE:
  ret = wl1271_plt_start(wl, val);
  break;
 case PLT_FEM_DETECT:
  ret = wl1271_tm_detect_fem(wl, tb);
  break;
 default:
  ret = -EINVAL;
  break;
 }

 return ret;
}

static int wl12xx_tm_cmd_get_mac(struct wl1271 *wl, struct nlattr *tb[])
{
 struct sk_buff *skb;
 u8 mac_addr[ETH_ALEN];
 int ret = 0;

 mutex_lock(&wl->mutex);

 if (!wl->plt) {
  ret = -EINVAL;
  goto out;
 }

 if (wl->fuse_oui_addr == 0 && wl->fuse_nic_addr == 0) {
  ret = -EOPNOTSUPP;
  goto out;
 }

 mac_addr[0] = (u8)(wl->fuse_oui_addr >> 16);
 mac_addr[1] = (u8)(wl->fuse_oui_addr >> 8);
 mac_addr[2] = (u8) wl->fuse_oui_addr;
 mac_addr[3] = (u8)(wl->fuse_nic_addr >> 16);
 mac_addr[4] = (u8)(wl->fuse_nic_addr >> 8);
 mac_addr[5] = (u8) wl->fuse_nic_addr;

 skb = cfg80211_testmode_alloc_reply_skb(wl->hw->wiphy, ETH_ALEN);
 if (!skb) {
  ret = -ENOMEM;
  goto out;
 }

 if (nla_put(skb, WL1271_TM_ATTR_DATA, ETH_ALEN, mac_addr)) {
  kfree_skb(skb);
  ret = -EMSGSIZE;
  goto out;
 }

 ret = cfg80211_testmode_reply(skb);
 if (ret < 0)
  goto out;

out:
 mutex_unlock(&wl->mutex);
 return ret;
}

int wl1271_tm_cmd(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
    void *data, int len)
{
 struct wl1271 *wl = hw->priv;
 struct nlattr *tb[WL1271_TM_ATTR_MAX + 1];
 u32 nla_cmd;
 int err;

 err = nla_parse_deprecated(tb, WL1271_TM_ATTR_MAX, data, len,
       wl1271_tm_policy, NULL);
 if (err)
  return err;

 if (!tb[WL1271_TM_ATTR_CMD_ID])
  return -EINVAL;

 nla_cmd = nla_get_u32(tb[WL1271_TM_ATTR_CMD_ID]);

 /* Only SET_PLT_MODE is allowed in case of mode PLT_CHIP_AWAKE */
 if (wl->plt_mode == PLT_CHIP_AWAKE &&
     nla_cmd != WL1271_TM_CMD_SET_PLT_MODE)
  return -EOPNOTSUPP;

 switch (nla_cmd) {
 case WL1271_TM_CMD_TEST:
  return wl1271_tm_cmd_test(wl, tb);
 case WL1271_TM_CMD_INTERROGATE:
  return wl1271_tm_cmd_interrogate(wl, tb);
 case WL1271_TM_CMD_CONFIGURE:
  return wl1271_tm_cmd_configure(wl, tb);
 case WL1271_TM_CMD_SET_PLT_MODE:
  return wl1271_tm_cmd_set_plt_mode(wl, tb);
 case WL1271_TM_CMD_GET_MAC:
  return wl12xx_tm_cmd_get_mac(wl, tb);
 default:
  return -EOPNOTSUPP;
 }
}