Quelle  cv1800-mailbox.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2024 Sophgo Technology Inc.
 * Copyright (C) 2024 Yuntao Dai <d1581209858@live.com>
 * Copyright (C) 2025 Junhui Liu <junhui.liu@pigmoral.tech>
 */

#include <linux/bits.h>
#include <linux/device.h>
#include <linux/err.h>
#include <linux/interrupt.h>
#include <linux/io.h>
#include <linux/kfifo.h>
#include <linux/mailbox_client.h>
#include <linux/mailbox_controller.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/slab.h>

#define RECV_CPU  1

#define MAILBOX_MAX_CHAN 8
#define MAILBOX_MSG_LEN  8

#define MBOX_EN_REG(cpu) (cpu << 2)
#define MBOX_DONE_REG(cpu) ((cpu << 2) + 2)
#define MBOX_SET_CLR_REG(cpu) (0x10 + (cpu << 4))
#define MBOX_SET_INT_REG(cpu) (0x18 + (cpu << 4))
#define MBOX_SET_REG  0x60

#define MAILBOX_CONTEXT_OFFSET 0x0400
#define MAILBOX_CONTEXT_SIZE 0x0040

#define MBOX_CONTEXT_BASE_INDEX(base, index) \
 ((u64 __iomem *)(base + MAILBOX_CONTEXT_OFFSET) + index)

/**
 * struct cv1800_mbox_chan_priv - cv1800 mailbox channel private data
 * @idx: index of channel
 * @cpu: send to which processor
 */
struct cv1800_mbox_chan_priv {
 int idx;
 int cpu;
};

struct cv1800_mbox {
 struct mbox_controller mbox;
 struct cv1800_mbox_chan_priv priv[MAILBOX_MAX_CHAN];
 struct mbox_chan chans[MAILBOX_MAX_CHAN];
 u64 __iomem *content[MAILBOX_MAX_CHAN];
 void __iomem *mbox_base;
 int recvid;
};

static irqreturn_t cv1800_mbox_isr(int irq, void *dev_id)
{
 struct cv1800_mbox *mbox = (struct cv1800_mbox *)dev_id;
 size_t i;
 u64 msg;
 int ret = IRQ_NONE;

 for (i = 0; i < MAILBOX_MAX_CHAN; i++) {
  if (mbox->content[i] && mbox->chans[i].cl) {
   memcpy_fromio(&msg, mbox->content[i], MAILBOX_MSG_LEN);
   mbox->content[i] = NULL;
   mbox_chan_received_data(&mbox->chans[i], (void *)&msg);
   ret = IRQ_HANDLED;
  }
 }

 return ret;
}

static irqreturn_t cv1800_mbox_irq(int irq, void *dev_id)
{
 struct cv1800_mbox *mbox = (struct cv1800_mbox *)dev_id;
 u8 set, valid;
 size_t i;
 int ret = IRQ_NONE;

 set = readb(mbox->mbox_base + MBOX_SET_INT_REG(RECV_CPU));

 if (!set)
  return ret;

 for (i = 0; i < MAILBOX_MAX_CHAN; i++) {
  valid = set & BIT(i);
  if (valid) {
   mbox->content[i] =
    MBOX_CONTEXT_BASE_INDEX(mbox->mbox_base, i);
   writeb(valid, mbox->mbox_base +
          MBOX_SET_CLR_REG(RECV_CPU));
   writeb(~valid, mbox->mbox_base + MBOX_EN_REG(RECV_CPU));
   ret = IRQ_WAKE_THREAD;
  }
 }

 return ret;
}

static int cv1800_mbox_send_data(struct mbox_chan *chan, void *data)
{
 struct cv1800_mbox_chan_priv *priv =
  (struct cv1800_mbox_chan_priv *)chan->con_priv;
 struct cv1800_mbox *mbox = dev_get_drvdata(chan->mbox->dev);
 int idx = priv->idx;
 int cpu = priv->cpu;
 u8 en, valid;

 memcpy_toio(MBOX_CONTEXT_BASE_INDEX(mbox->mbox_base, idx),
      data, MAILBOX_MSG_LEN);

 valid = BIT(idx);
 writeb(valid, mbox->mbox_base + MBOX_SET_CLR_REG(cpu));
 en = readb(mbox->mbox_base + MBOX_EN_REG(cpu));
 writeb(en | valid, mbox->mbox_base + MBOX_EN_REG(cpu));
 writeb(valid, mbox->mbox_base + MBOX_SET_REG);

 return 0;
}

static bool cv1800_last_tx_done(struct mbox_chan *chan)
{
 struct cv1800_mbox_chan_priv *priv =
  (struct cv1800_mbox_chan_priv *)chan->con_priv;
 struct cv1800_mbox *mbox = dev_get_drvdata(chan->mbox->dev);
 u8 en;

 en = readb(mbox->mbox_base + MBOX_EN_REG(priv->cpu));

 return !(en & BIT(priv->idx));
}

static const struct mbox_chan_ops cv1800_mbox_chan_ops = {
 .send_data = cv1800_mbox_send_data,
 .last_tx_done = cv1800_last_tx_done,
};

static struct mbox_chan *cv1800_mbox_xlate(struct mbox_controller *mbox,
        const struct of_phandle_args *spec)
{
 struct cv1800_mbox_chan_priv *priv;

 int idx = spec->args[0];
 int cpu = spec->args[1];

 if (idx >= mbox->num_chans)
  return ERR_PTR(-EINVAL);

 priv = mbox->chans[idx].con_priv;
 priv->cpu = cpu;

 return &mbox->chans[idx];
}

static const struct of_device_id cv1800_mbox_of_match[] = {
 { .compatible = "sophgo,cv1800b-mailbox", },
 {},
};
MODULE_DEVICE_TABLE(of, cv1800_mbox_of_match);

static int cv1800_mbox_probe(struct platform_device *pdev)
{
 struct device *dev = &pdev->dev;
 struct cv1800_mbox *mb;
 int irq, idx, err;

 mb = devm_kzalloc(dev, sizeof(*mb), GFP_KERNEL);
 if (!mb)
  return -ENOMEM;

 mb->mbox_base = devm_platform_ioremap_resource(pdev, 0);
 if (IS_ERR(mb->mbox_base))
  return dev_err_probe(dev, PTR_ERR(mb->mbox_base),
         "Failed to map resource\n");

 mb->mbox.dev = dev;
 mb->mbox.chans = mb->chans;
 mb->mbox.txdone_poll = true;
 mb->mbox.ops = &cv1800_mbox_chan_ops;
 mb->mbox.num_chans = MAILBOX_MAX_CHAN;
 mb->mbox.of_xlate = cv1800_mbox_xlate;

 irq = platform_get_irq(pdev, 0);
 if (irq < 0)
  return irq;

 err = devm_request_threaded_irq(dev, irq, cv1800_mbox_irq,
     cv1800_mbox_isr, IRQF_ONESHOT,
     dev_name(&pdev->dev), mb);
 if (err < 0)
  return dev_err_probe(dev, err, "Failed to register irq\n");

 for (idx = 0; idx < MAILBOX_MAX_CHAN; idx++) {
  mb->priv[idx].idx = idx;
  mb->mbox.chans[idx].con_priv = &mb->priv[idx];
 }

 platform_set_drvdata(pdev, mb);

 err = devm_mbox_controller_register(dev, &mb->mbox);
 if (err)
  return dev_err_probe(dev, err, "Failed to register mailbox\n");

 return 0;
}

static struct platform_driver cv1800_mbox_driver = {
 .driver = {
  .name = "cv1800-mbox",
  .of_match_table = cv1800_mbox_of_match,
 },
 .probe = cv1800_mbox_probe,
};

module_platform_driver(cv1800_mbox_driver);

MODULE_DESCRIPTION("cv1800 mailbox driver");
MODULE_LICENSE("GPL");