android13/kernel-5.10/drivers/rpmsg/rockchip_rpmsg_test.c

123 lines
3.1 KiB
C

// SPDX-License-Identifier: GPL-2.0
/*
* Rockchip Remote Processors Messaging Test.
*
* Copyright (c) 2022 Rockchip Electronics Co. Ltd.
* Author: Hongming Zou <hongming.zou@rock-chips.com>
*/
#include <linux/delay.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/rpmsg.h>
#include <linux/rpmsg/rockchip_rpmsg.h>
#include <linux/time.h>
#include <linux/virtio.h>
#define LINUX_TEST_MSG_1 "Announce master ept id!"
#define LINUX_TEST_MSG_2 "Rockchip rpmsg linux test pingpong!"
#define MSG_LIMIT 10000
/* different processor cores may need to adjust the value of this definition */
#define LINUX_RPMSG_COMPENSATION (1) //ms
struct rpmsg_info_t {
int rx_count;
};
static int rockchip_rpmsg_test_cb(struct rpmsg_device *rp, void *payload,
int payload_len, void *priv, u32 src)
{
int ret;
uint32_t remote_ept_id;
struct rpmsg_info_t *info = dev_get_drvdata(&rp->dev);
remote_ept_id = src;
dev_info(&rp->dev, "rx msg %s rx_count %d(remote_ept_id: 0x%x)\n",
(char *)payload, ++info->rx_count, remote_ept_id);
/* test should not live forever */
if (info->rx_count >= MSG_LIMIT) {
dev_info(&rp->dev, "Rockchip rpmsg test exit!\n");
return 0;
}
mdelay(LINUX_RPMSG_COMPENSATION);
/* send a new message now */
ret = rpmsg_sendto(rp->ept, LINUX_TEST_MSG_2, strlen(LINUX_TEST_MSG_2), remote_ept_id);
if (ret)
dev_err(&rp->dev, "rpmsg_send failed: %d\n", ret);
return ret;
}
static int rockchip_rpmsg_test_probe(struct rpmsg_device *rp)
{
int ret;
uint32_t master_ept_id, remote_ept_id;
struct rpmsg_info_t *info;
master_ept_id = rp->src;
remote_ept_id = rp->dst;
dev_info(&rp->dev, "new channel: 0x%x -> 0x%x!\n", master_ept_id, remote_ept_id);
info = devm_kzalloc(&rp->dev, sizeof(*info), GFP_KERNEL);
if (!info)
return -ENOMEM;
dev_set_drvdata(&rp->dev, info);
/*
* send a message to our remote processor, and tell remote
* processor about this channel
*/
ret = rpmsg_send(rp->ept, LINUX_TEST_MSG_1, strlen(LINUX_TEST_MSG_1));
if (ret) {
dev_err(&rp->dev, "rpmsg_send failed: %d\n", ret);
return ret;
}
mdelay(LINUX_RPMSG_COMPENSATION);
ret = rpmsg_sendto(rp->ept, LINUX_TEST_MSG_2, strlen(LINUX_TEST_MSG_2), remote_ept_id);
if (ret) {
dev_err(&rp->dev, "rpmsg_send failed: %d\n", ret);
return ret;
}
return 0;
}
static void rockchip_rpmsg_test_remove(struct rpmsg_device *rp)
{
dev_info(&rp->dev, "rockchip rpmsg test is removed\n");
}
static struct rpmsg_device_id rockchip_rpmsg_test_id_table[] = {
{ .name = "rpmsg-ap3-ch0" },
{ /* sentinel */ },
};
static struct rpmsg_driver rockchip_rpmsg_test = {
.drv.name = KBUILD_MODNAME,
.drv.owner = THIS_MODULE,
.id_table = rockchip_rpmsg_test_id_table,
.probe = rockchip_rpmsg_test_probe,
.callback = rockchip_rpmsg_test_cb,
.remove = rockchip_rpmsg_test_remove,
};
static int __init init(void)
{
return register_rpmsg_driver(&rockchip_rpmsg_test);
}
static void __exit fini(void)
{
unregister_rpmsg_driver(&rockchip_rpmsg_test);
}
module_init(init);
module_exit(fini);
MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Rockchip Remote Processors Messaging Test");
MODULE_AUTHOR("Hongming Zou <hongming.zou@rock-chips.com>");