cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

vpu_drv.c (5954B)

      1// SPDX-License-Identifier: GPL-2.0
      2/*
      3 * Copyright 2020-2021 NXP
      4 */
      5
      6#include <linux/init.h>
      7#include <linux/interconnect.h>
      8#include <linux/ioctl.h>
      9#include <linux/list.h>
     10#include <linux/kernel.h>
     11#include <linux/module.h>
     12#include <linux/dma-map-ops.h>
     13#include <linux/of_device.h>
     14#include <linux/of_address.h>
     15#include <linux/platform_device.h>
     16#include <linux/slab.h>
     17#include <linux/types.h>
     18#include <linux/pm_runtime.h>
     19#include <linux/videodev2.h>
     20#include <linux/of_reserved_mem.h>
     21#include <media/v4l2-device.h>
     22#include <media/videobuf2-v4l2.h>
     23#include <media/v4l2-mem2mem.h>
     24#include <media/v4l2-ioctl.h>
     25#include <linux/debugfs.h>
     26#include "vpu.h"
     27#include "vpu_imx8q.h"
     28
     29bool debug;
     30module_param(debug, bool, 0644);
     31
     32void vpu_writel(struct vpu_dev *vpu, u32 reg, u32 val)
     33{
     34	writel(val, vpu->base + reg);
     35}
     36
     37u32 vpu_readl(struct vpu_dev *vpu, u32 reg)
     38{
     39	return readl(vpu->base + reg);
     40}
     41
     42static void vpu_dev_get(struct vpu_dev *vpu)
     43{
     44	if (atomic_inc_return(&vpu->ref_vpu) == 1 && vpu->res->setup)
     45		vpu->res->setup(vpu);
     46}
     47
     48static void vpu_dev_put(struct vpu_dev *vpu)
     49{
     50	atomic_dec(&vpu->ref_vpu);
     51}
     52
     53static void vpu_enc_get(struct vpu_dev *vpu)
     54{
     55	if (atomic_inc_return(&vpu->ref_enc) == 1 && vpu->res->setup_encoder)
     56		vpu->res->setup_encoder(vpu);
     57}
     58
     59static void vpu_enc_put(struct vpu_dev *vpu)
     60{
     61	atomic_dec(&vpu->ref_enc);
     62}
     63
     64static void vpu_dec_get(struct vpu_dev *vpu)
     65{
     66	if (atomic_inc_return(&vpu->ref_dec) == 1 && vpu->res->setup_decoder)
     67		vpu->res->setup_decoder(vpu);
     68}
     69
     70static void vpu_dec_put(struct vpu_dev *vpu)
     71{
     72	atomic_dec(&vpu->ref_dec);
     73}
     74
     75static int vpu_init_media_device(struct vpu_dev *vpu)
     76{
     77	vpu->mdev.dev = vpu->dev;
     78	strscpy(vpu->mdev.model, "amphion-vpu", sizeof(vpu->mdev.model));
     79	strscpy(vpu->mdev.bus_info, "platform: amphion-vpu", sizeof(vpu->mdev.bus_info));
     80	media_device_init(&vpu->mdev);
     81	vpu->v4l2_dev.mdev = &vpu->mdev;
     82
     83	return 0;
     84}
     85
     86static int vpu_probe(struct platform_device *pdev)
     87{
     88	struct device *dev = &pdev->dev;
     89	struct vpu_dev *vpu;
     90	int ret;
     91
     92	dev_dbg(dev, "probe\n");
     93	vpu = devm_kzalloc(dev, sizeof(*vpu), GFP_KERNEL);
     94	if (!vpu)
     95		return -ENOMEM;
     96
     97	vpu->pdev = pdev;
     98	vpu->dev = dev;
     99	mutex_init(&vpu->lock);
    100	INIT_LIST_HEAD(&vpu->cores);
    101	platform_set_drvdata(pdev, vpu);
    102	atomic_set(&vpu->ref_vpu, 0);
    103	atomic_set(&vpu->ref_enc, 0);
    104	atomic_set(&vpu->ref_dec, 0);
    105	vpu->get_vpu = vpu_dev_get;
    106	vpu->put_vpu = vpu_dev_put;
    107	vpu->get_enc = vpu_enc_get;
    108	vpu->put_enc = vpu_enc_put;
    109	vpu->get_dec = vpu_dec_get;
    110	vpu->put_dec = vpu_dec_put;
    111
    112	vpu->base = devm_platform_ioremap_resource(pdev, 0);
    113	if (IS_ERR(vpu->base))
    114		return PTR_ERR(vpu->base);
    115
    116	vpu->res = of_device_get_match_data(dev);
    117	if (!vpu->res)
    118		return -ENODEV;
    119
    120	pm_runtime_enable(dev);
    121
    122	ret = v4l2_device_register(dev, &vpu->v4l2_dev);
    123	if (ret)
    124		goto err_vpu_deinit;
    125
    126	vpu_init_media_device(vpu);
    127	vpu->encoder.type = VPU_CORE_TYPE_ENC;
    128	vpu->encoder.function = MEDIA_ENT_F_PROC_VIDEO_ENCODER;
    129	vpu->decoder.type = VPU_CORE_TYPE_DEC;
    130	vpu->decoder.function = MEDIA_ENT_F_PROC_VIDEO_DECODER;
    131	ret = vpu_add_func(vpu, &vpu->decoder);
    132	if (ret)
    133		goto err_add_decoder;
    134	ret = vpu_add_func(vpu, &vpu->encoder);
    135	if (ret)
    136		goto err_add_encoder;
    137	ret = media_device_register(&vpu->mdev);
    138	if (ret)
    139		goto err_vpu_media;
    140	vpu->debugfs = debugfs_create_dir("amphion_vpu", NULL);
    141
    142	of_platform_populate(dev->of_node, NULL, NULL, dev);
    143
    144	return 0;
    145
    146err_vpu_media:
    147	vpu_remove_func(&vpu->encoder);
    148err_add_encoder:
    149	vpu_remove_func(&vpu->decoder);
    150err_add_decoder:
    151	media_device_cleanup(&vpu->mdev);
    152	v4l2_device_unregister(&vpu->v4l2_dev);
    153err_vpu_deinit:
    154	pm_runtime_set_suspended(dev);
    155	pm_runtime_disable(dev);
    156
    157	return ret;
    158}
    159
    160static int vpu_remove(struct platform_device *pdev)
    161{
    162	struct vpu_dev *vpu = platform_get_drvdata(pdev);
    163	struct device *dev = &pdev->dev;
    164
    165	debugfs_remove_recursive(vpu->debugfs);
    166	vpu->debugfs = NULL;
    167
    168	pm_runtime_disable(dev);
    169
    170	media_device_unregister(&vpu->mdev);
    171	vpu_remove_func(&vpu->decoder);
    172	vpu_remove_func(&vpu->encoder);
    173	media_device_cleanup(&vpu->mdev);
    174	v4l2_device_unregister(&vpu->v4l2_dev);
    175	mutex_destroy(&vpu->lock);
    176
    177	return 0;
    178}
    179
    180static int __maybe_unused vpu_runtime_resume(struct device *dev)
    181{
    182	return 0;
    183}
    184
    185static int __maybe_unused vpu_runtime_suspend(struct device *dev)
    186{
    187	return 0;
    188}
    189
    190static int __maybe_unused vpu_resume(struct device *dev)
    191{
    192	return 0;
    193}
    194
    195static int __maybe_unused vpu_suspend(struct device *dev)
    196{
    197	return 0;
    198}
    199
    200static const struct dev_pm_ops vpu_pm_ops = {
    201	SET_RUNTIME_PM_OPS(vpu_runtime_suspend, vpu_runtime_resume, NULL)
    202	SET_SYSTEM_SLEEP_PM_OPS(vpu_suspend, vpu_resume)
    203};
    204
    205static struct vpu_resources imx8qxp_res = {
    206	.plat_type = IMX8QXP,
    207	.mreg_base = 0x40000000,
    208	.setup = vpu_imx8q_setup,
    209	.setup_encoder = vpu_imx8q_setup_enc,
    210	.setup_decoder = vpu_imx8q_setup_dec,
    211	.reset = vpu_imx8q_reset
    212};
    213
    214static struct vpu_resources imx8qm_res = {
    215	.plat_type = IMX8QM,
    216	.mreg_base = 0x40000000,
    217	.setup = vpu_imx8q_setup,
    218	.setup_encoder = vpu_imx8q_setup_enc,
    219	.setup_decoder = vpu_imx8q_setup_dec,
    220	.reset = vpu_imx8q_reset
    221};
    222
    223static const struct of_device_id vpu_dt_match[] = {
    224	{ .compatible = "nxp,imx8qxp-vpu", .data = &imx8qxp_res },
    225	{ .compatible = "nxp,imx8qm-vpu", .data = &imx8qm_res },
    226	{}
    227};
    228MODULE_DEVICE_TABLE(of, vpu_dt_match);
    229
    230static struct platform_driver amphion_vpu_driver = {
    231	.probe = vpu_probe,
    232	.remove = vpu_remove,
    233	.driver = {
    234		.name = "amphion-vpu",
    235		.of_match_table = vpu_dt_match,
    236		.pm = &vpu_pm_ops,
    237	},
    238};
    239
    240static int __init vpu_driver_init(void)
    241{
    242	int ret;
    243
    244	ret = platform_driver_register(&amphion_vpu_driver);
    245	if (ret)
    246		return ret;
    247
    248	return vpu_core_driver_init();
    249}
    250
    251static void __exit vpu_driver_exit(void)
    252{
    253	vpu_core_driver_exit();
    254	platform_driver_unregister(&amphion_vpu_driver);
    255}
    256module_init(vpu_driver_init);
    257module_exit(vpu_driver_exit);
    258
    259MODULE_AUTHOR("Freescale Semiconductor, Inc.");
    260MODULE_DESCRIPTION("Linux VPU driver for Freescale i.MX8Q");
    261MODULE_LICENSE("GPL v2");