162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (c) 2022 MediaTek Corporation. All rights reserved.
462306a36Sopenharmony_ci * Author: Allen-KH Cheng <allen-kh.cheng@mediatek.com>
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#include <linux/firmware/mediatek/mtk-adsp-ipc.h>
862306a36Sopenharmony_ci#include <linux/kernel.h>
962306a36Sopenharmony_ci#include <linux/mailbox_client.h>
1062306a36Sopenharmony_ci#include <linux/module.h>
1162306a36Sopenharmony_ci#include <linux/platform_device.h>
1262306a36Sopenharmony_ci#include <linux/slab.h>
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_cistatic const char * const adsp_mbox_ch_names[MTK_ADSP_MBOX_NUM] = { "rx", "tx" };
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci/*
1762306a36Sopenharmony_ci * mtk_adsp_ipc_send - send ipc cmd to MTK ADSP
1862306a36Sopenharmony_ci *
1962306a36Sopenharmony_ci * @ipc: ADSP IPC handle
2062306a36Sopenharmony_ci * @idx: index of the mailbox channel
2162306a36Sopenharmony_ci * @msg: IPC cmd (reply or request)
2262306a36Sopenharmony_ci *
2362306a36Sopenharmony_ci * Returns zero for success from mbox_send_message
2462306a36Sopenharmony_ci * negative value for error
2562306a36Sopenharmony_ci */
2662306a36Sopenharmony_ciint mtk_adsp_ipc_send(struct mtk_adsp_ipc *ipc, unsigned int idx, uint32_t msg)
2762306a36Sopenharmony_ci{
2862306a36Sopenharmony_ci	struct mtk_adsp_chan *adsp_chan;
2962306a36Sopenharmony_ci	int ret;
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci	if (idx >= MTK_ADSP_MBOX_NUM)
3262306a36Sopenharmony_ci		return -EINVAL;
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci	adsp_chan = &ipc->chans[idx];
3562306a36Sopenharmony_ci	ret = mbox_send_message(adsp_chan->ch, &msg);
3662306a36Sopenharmony_ci	if (ret < 0)
3762306a36Sopenharmony_ci		return ret;
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	return 0;
4062306a36Sopenharmony_ci}
4162306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(mtk_adsp_ipc_send);
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci/*
4462306a36Sopenharmony_ci * mtk_adsp_ipc_recv - recv callback used by MTK ADSP mailbox
4562306a36Sopenharmony_ci *
4662306a36Sopenharmony_ci * @c: mbox client
4762306a36Sopenharmony_ci * @msg: message received
4862306a36Sopenharmony_ci *
4962306a36Sopenharmony_ci * Users of ADSP IPC will need to privde handle_reply and handle_request
5062306a36Sopenharmony_ci * callbacks.
5162306a36Sopenharmony_ci */
5262306a36Sopenharmony_cistatic void mtk_adsp_ipc_recv(struct mbox_client *c, void *msg)
5362306a36Sopenharmony_ci{
5462306a36Sopenharmony_ci	struct mtk_adsp_chan *chan = container_of(c, struct mtk_adsp_chan, cl);
5562306a36Sopenharmony_ci	struct device *dev = c->dev;
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	switch (chan->idx) {
5862306a36Sopenharmony_ci	case MTK_ADSP_MBOX_REPLY:
5962306a36Sopenharmony_ci		chan->ipc->ops->handle_reply(chan->ipc);
6062306a36Sopenharmony_ci		break;
6162306a36Sopenharmony_ci	case MTK_ADSP_MBOX_REQUEST:
6262306a36Sopenharmony_ci		chan->ipc->ops->handle_request(chan->ipc);
6362306a36Sopenharmony_ci		break;
6462306a36Sopenharmony_ci	default:
6562306a36Sopenharmony_ci		dev_err(dev, "wrong mbox chan %d\n", chan->idx);
6662306a36Sopenharmony_ci		break;
6762306a36Sopenharmony_ci	}
6862306a36Sopenharmony_ci}
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_cistatic int mtk_adsp_ipc_probe(struct platform_device *pdev)
7162306a36Sopenharmony_ci{
7262306a36Sopenharmony_ci	struct device *dev = &pdev->dev;
7362306a36Sopenharmony_ci	struct mtk_adsp_ipc *adsp_ipc;
7462306a36Sopenharmony_ci	struct mtk_adsp_chan *adsp_chan;
7562306a36Sopenharmony_ci	struct mbox_client *cl;
7662306a36Sopenharmony_ci	int ret;
7762306a36Sopenharmony_ci	int i, j;
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	device_set_of_node_from_dev(&pdev->dev, pdev->dev.parent);
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	adsp_ipc = devm_kzalloc(dev, sizeof(*adsp_ipc), GFP_KERNEL);
8262306a36Sopenharmony_ci	if (!adsp_ipc)
8362306a36Sopenharmony_ci		return -ENOMEM;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	for (i = 0; i < MTK_ADSP_MBOX_NUM; i++) {
8662306a36Sopenharmony_ci		adsp_chan = &adsp_ipc->chans[i];
8762306a36Sopenharmony_ci		cl = &adsp_chan->cl;
8862306a36Sopenharmony_ci		cl->dev = dev->parent;
8962306a36Sopenharmony_ci		cl->tx_block = false;
9062306a36Sopenharmony_ci		cl->knows_txdone = false;
9162306a36Sopenharmony_ci		cl->tx_prepare = NULL;
9262306a36Sopenharmony_ci		cl->rx_callback = mtk_adsp_ipc_recv;
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ci		adsp_chan->ipc = adsp_ipc;
9562306a36Sopenharmony_ci		adsp_chan->idx = i;
9662306a36Sopenharmony_ci		adsp_chan->ch = mbox_request_channel_byname(cl, adsp_mbox_ch_names[i]);
9762306a36Sopenharmony_ci		if (IS_ERR(adsp_chan->ch)) {
9862306a36Sopenharmony_ci			ret = PTR_ERR(adsp_chan->ch);
9962306a36Sopenharmony_ci			if (ret != -EPROBE_DEFER)
10062306a36Sopenharmony_ci				dev_err(dev, "Failed to request mbox chan %s ret %d\n",
10162306a36Sopenharmony_ci					adsp_mbox_ch_names[i], ret);
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_ci			for (j = 0; j < i; j++) {
10462306a36Sopenharmony_ci				adsp_chan = &adsp_ipc->chans[j];
10562306a36Sopenharmony_ci				mbox_free_channel(adsp_chan->ch);
10662306a36Sopenharmony_ci			}
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_ci			return ret;
10962306a36Sopenharmony_ci		}
11062306a36Sopenharmony_ci	}
11162306a36Sopenharmony_ci
11262306a36Sopenharmony_ci	adsp_ipc->dev = dev;
11362306a36Sopenharmony_ci	dev_set_drvdata(dev, adsp_ipc);
11462306a36Sopenharmony_ci	dev_dbg(dev, "MTK ADSP IPC initialized\n");
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_ci	return 0;
11762306a36Sopenharmony_ci}
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_cistatic int mtk_adsp_ipc_remove(struct platform_device *pdev)
12062306a36Sopenharmony_ci{
12162306a36Sopenharmony_ci	struct mtk_adsp_ipc *adsp_ipc = dev_get_drvdata(&pdev->dev);
12262306a36Sopenharmony_ci	struct mtk_adsp_chan *adsp_chan;
12362306a36Sopenharmony_ci	int i;
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	for (i = 0; i < MTK_ADSP_MBOX_NUM; i++) {
12662306a36Sopenharmony_ci		adsp_chan = &adsp_ipc->chans[i];
12762306a36Sopenharmony_ci		mbox_free_channel(adsp_chan->ch);
12862306a36Sopenharmony_ci	}
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ci	return 0;
13162306a36Sopenharmony_ci}
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_cistatic struct platform_driver mtk_adsp_ipc_driver = {
13462306a36Sopenharmony_ci	.driver = {
13562306a36Sopenharmony_ci		.name = "mtk-adsp-ipc",
13662306a36Sopenharmony_ci	},
13762306a36Sopenharmony_ci	.probe = mtk_adsp_ipc_probe,
13862306a36Sopenharmony_ci	.remove = mtk_adsp_ipc_remove,
13962306a36Sopenharmony_ci};
14062306a36Sopenharmony_cibuiltin_platform_driver(mtk_adsp_ipc_driver);
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_ciMODULE_AUTHOR("Allen-KH Cheng <allen-kh.cheng@mediatek.com>");
14362306a36Sopenharmony_ciMODULE_DESCRIPTION("MTK ADSP IPC Driver");
14462306a36Sopenharmony_ciMODULE_LICENSE("GPL");
145