// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) 2019 MediaTek Inc.
 *
 * Author: Felix Fietkau <nbd@nbd.name>
 *    Lorenzo Bianconi <lorenzo@kernel.org>
 *    Sean Wang <sean.wang@mediatek.com>
 */
#include <linux/kernel.h>
#include <linux/module.h>

#include "mt7615.h"
#include "mac.h"
#include "mcu.h"
#include "regs.h"

static int
mt7663u_mcu_send_message(struct mt76_dev *mdev, struct sk_buff *skb,
    int cmd, int *seq)
{
 struct mt7615_dev *dev = container_of(mdev, struct mt7615_dev, mt76);
 int ret, ep, len, pad;

 mt7615_mcu_fill_msg(dev, skb, cmd, seq);
 if (cmd != MCU_CMD(FW_SCATTER))
  ep = MT_EP_OUT_INBAND_CMD;
 else
  ep = MT_EP_OUT_AC_BE;

 len = skb->len;
 put_unaligned_le32(len, skb_push(skb, sizeof(len)));
 pad = round_up(skb->len, 4) + 4 - skb->len;
 ret = mt76_skb_adjust_pad(skb, pad);
 if (ret < 0)
  goto out;

 ret = mt76u_bulk_msg(&dev->mt76, skb->data, skb->len, NULL,
        1000, ep);

out:
 dev_kfree_skb(skb);

 return ret;
}

int mt7663u_mcu_power_on(struct mt7615_dev *dev)
{
 int ret;

 ret = mt76u_vendor_request(&dev->mt76, MT_VEND_POWER_ON,
       USB_DIR_OUT | USB_TYPE_VENDOR,
       0x0, 0x1, NULL, 0);
 if (ret)
  return ret;

 if (!mt76_poll_msec(dev, MT_CONN_ON_MISC,
       MT_TOP_MISC2_FW_PWR_ON,
       FW_STATE_PWR_ON << 1, 500)) {
  dev_err(dev->mt76.dev, "Timeout for power on\n");
  ret = -EIO;
 }

 return 0;
}

int mt7663u_mcu_init(struct mt7615_dev *dev)
{
 static const struct mt76_mcu_ops mt7663u_mcu_ops = {
  .headroom = MT_USB_HDR_SIZE + sizeof(struct mt7615_mcu_txd),
  .tailroom = MT_USB_TAIL_SIZE,
  .mcu_skb_send_msg = mt7663u_mcu_send_message,
  .mcu_parse_response = mt7615_mcu_parse_response,
 };
 int ret;

 dev->mt76.mcu_ops = &mt7663u_mcu_ops;

 mt76_set(dev, MT_UDMA_TX_QSEL, MT_FW_DL_EN);
 if (test_and_clear_bit(MT76_STATE_POWER_OFF, &dev->mphy.state)) {
  ret = mt7615_mcu_restart(&dev->mt76);
  if (ret)
   return ret;

  if (!mt76_poll_msec(dev, MT_CONN_ON_MISC,
        MT_TOP_MISC2_FW_PWR_ON, 0, 500))
   return -EIO;

  ret = mt7663u_mcu_power_on(dev);
  if (ret)
   return ret;
 }

 ret = __mt7663_load_firmware(dev);
 if (ret)
  return ret;

 mt76_clear(dev, MT_UDMA_TX_QSEL, MT_FW_DL_EN);
 set_bit(MT76_STATE_MCU_RUNNING, &dev->mphy.state);

 return 0;
}