1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-only
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * VFIO based driver for Mediated device
4*4882a593Smuzhiyun *
5*4882a593Smuzhiyun * Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved.
6*4882a593Smuzhiyun * Author: Neo Jia <cjia@nvidia.com>
7*4882a593Smuzhiyun * Kirti Wankhede <kwankhede@nvidia.com>
8*4882a593Smuzhiyun */
9*4882a593Smuzhiyun
10*4882a593Smuzhiyun #include <linux/init.h>
11*4882a593Smuzhiyun #include <linux/module.h>
12*4882a593Smuzhiyun #include <linux/device.h>
13*4882a593Smuzhiyun #include <linux/kernel.h>
14*4882a593Smuzhiyun #include <linux/slab.h>
15*4882a593Smuzhiyun #include <linux/vfio.h>
16*4882a593Smuzhiyun #include <linux/mdev.h>
17*4882a593Smuzhiyun
18*4882a593Smuzhiyun #include "mdev_private.h"
19*4882a593Smuzhiyun
20*4882a593Smuzhiyun #define DRIVER_VERSION "0.1"
21*4882a593Smuzhiyun #define DRIVER_AUTHOR "NVIDIA Corporation"
22*4882a593Smuzhiyun #define DRIVER_DESC "VFIO based driver for Mediated device"
23*4882a593Smuzhiyun
vfio_mdev_open(void * device_data)24*4882a593Smuzhiyun static int vfio_mdev_open(void *device_data)
25*4882a593Smuzhiyun {
26*4882a593Smuzhiyun struct mdev_device *mdev = device_data;
27*4882a593Smuzhiyun struct mdev_parent *parent = mdev->parent;
28*4882a593Smuzhiyun int ret;
29*4882a593Smuzhiyun
30*4882a593Smuzhiyun if (unlikely(!parent->ops->open))
31*4882a593Smuzhiyun return -EINVAL;
32*4882a593Smuzhiyun
33*4882a593Smuzhiyun if (!try_module_get(THIS_MODULE))
34*4882a593Smuzhiyun return -ENODEV;
35*4882a593Smuzhiyun
36*4882a593Smuzhiyun ret = parent->ops->open(mdev);
37*4882a593Smuzhiyun if (ret)
38*4882a593Smuzhiyun module_put(THIS_MODULE);
39*4882a593Smuzhiyun
40*4882a593Smuzhiyun return ret;
41*4882a593Smuzhiyun }
42*4882a593Smuzhiyun
vfio_mdev_release(void * device_data)43*4882a593Smuzhiyun static void vfio_mdev_release(void *device_data)
44*4882a593Smuzhiyun {
45*4882a593Smuzhiyun struct mdev_device *mdev = device_data;
46*4882a593Smuzhiyun struct mdev_parent *parent = mdev->parent;
47*4882a593Smuzhiyun
48*4882a593Smuzhiyun if (likely(parent->ops->release))
49*4882a593Smuzhiyun parent->ops->release(mdev);
50*4882a593Smuzhiyun
51*4882a593Smuzhiyun module_put(THIS_MODULE);
52*4882a593Smuzhiyun }
53*4882a593Smuzhiyun
vfio_mdev_unlocked_ioctl(void * device_data,unsigned int cmd,unsigned long arg)54*4882a593Smuzhiyun static long vfio_mdev_unlocked_ioctl(void *device_data,
55*4882a593Smuzhiyun unsigned int cmd, unsigned long arg)
56*4882a593Smuzhiyun {
57*4882a593Smuzhiyun struct mdev_device *mdev = device_data;
58*4882a593Smuzhiyun struct mdev_parent *parent = mdev->parent;
59*4882a593Smuzhiyun
60*4882a593Smuzhiyun if (unlikely(!parent->ops->ioctl))
61*4882a593Smuzhiyun return -EINVAL;
62*4882a593Smuzhiyun
63*4882a593Smuzhiyun return parent->ops->ioctl(mdev, cmd, arg);
64*4882a593Smuzhiyun }
65*4882a593Smuzhiyun
vfio_mdev_read(void * device_data,char __user * buf,size_t count,loff_t * ppos)66*4882a593Smuzhiyun static ssize_t vfio_mdev_read(void *device_data, char __user *buf,
67*4882a593Smuzhiyun size_t count, loff_t *ppos)
68*4882a593Smuzhiyun {
69*4882a593Smuzhiyun struct mdev_device *mdev = device_data;
70*4882a593Smuzhiyun struct mdev_parent *parent = mdev->parent;
71*4882a593Smuzhiyun
72*4882a593Smuzhiyun if (unlikely(!parent->ops->read))
73*4882a593Smuzhiyun return -EINVAL;
74*4882a593Smuzhiyun
75*4882a593Smuzhiyun return parent->ops->read(mdev, buf, count, ppos);
76*4882a593Smuzhiyun }
77*4882a593Smuzhiyun
vfio_mdev_write(void * device_data,const char __user * buf,size_t count,loff_t * ppos)78*4882a593Smuzhiyun static ssize_t vfio_mdev_write(void *device_data, const char __user *buf,
79*4882a593Smuzhiyun size_t count, loff_t *ppos)
80*4882a593Smuzhiyun {
81*4882a593Smuzhiyun struct mdev_device *mdev = device_data;
82*4882a593Smuzhiyun struct mdev_parent *parent = mdev->parent;
83*4882a593Smuzhiyun
84*4882a593Smuzhiyun if (unlikely(!parent->ops->write))
85*4882a593Smuzhiyun return -EINVAL;
86*4882a593Smuzhiyun
87*4882a593Smuzhiyun return parent->ops->write(mdev, buf, count, ppos);
88*4882a593Smuzhiyun }
89*4882a593Smuzhiyun
vfio_mdev_mmap(void * device_data,struct vm_area_struct * vma)90*4882a593Smuzhiyun static int vfio_mdev_mmap(void *device_data, struct vm_area_struct *vma)
91*4882a593Smuzhiyun {
92*4882a593Smuzhiyun struct mdev_device *mdev = device_data;
93*4882a593Smuzhiyun struct mdev_parent *parent = mdev->parent;
94*4882a593Smuzhiyun
95*4882a593Smuzhiyun if (unlikely(!parent->ops->mmap))
96*4882a593Smuzhiyun return -EINVAL;
97*4882a593Smuzhiyun
98*4882a593Smuzhiyun return parent->ops->mmap(mdev, vma);
99*4882a593Smuzhiyun }
100*4882a593Smuzhiyun
101*4882a593Smuzhiyun static const struct vfio_device_ops vfio_mdev_dev_ops = {
102*4882a593Smuzhiyun .name = "vfio-mdev",
103*4882a593Smuzhiyun .open = vfio_mdev_open,
104*4882a593Smuzhiyun .release = vfio_mdev_release,
105*4882a593Smuzhiyun .ioctl = vfio_mdev_unlocked_ioctl,
106*4882a593Smuzhiyun .read = vfio_mdev_read,
107*4882a593Smuzhiyun .write = vfio_mdev_write,
108*4882a593Smuzhiyun .mmap = vfio_mdev_mmap,
109*4882a593Smuzhiyun };
110*4882a593Smuzhiyun
vfio_mdev_probe(struct device * dev)111*4882a593Smuzhiyun static int vfio_mdev_probe(struct device *dev)
112*4882a593Smuzhiyun {
113*4882a593Smuzhiyun struct mdev_device *mdev = to_mdev_device(dev);
114*4882a593Smuzhiyun
115*4882a593Smuzhiyun return vfio_add_group_dev(dev, &vfio_mdev_dev_ops, mdev);
116*4882a593Smuzhiyun }
117*4882a593Smuzhiyun
vfio_mdev_remove(struct device * dev)118*4882a593Smuzhiyun static void vfio_mdev_remove(struct device *dev)
119*4882a593Smuzhiyun {
120*4882a593Smuzhiyun vfio_del_group_dev(dev);
121*4882a593Smuzhiyun }
122*4882a593Smuzhiyun
123*4882a593Smuzhiyun static struct mdev_driver vfio_mdev_driver = {
124*4882a593Smuzhiyun .name = "vfio_mdev",
125*4882a593Smuzhiyun .probe = vfio_mdev_probe,
126*4882a593Smuzhiyun .remove = vfio_mdev_remove,
127*4882a593Smuzhiyun };
128*4882a593Smuzhiyun
vfio_mdev_init(void)129*4882a593Smuzhiyun static int __init vfio_mdev_init(void)
130*4882a593Smuzhiyun {
131*4882a593Smuzhiyun return mdev_register_driver(&vfio_mdev_driver, THIS_MODULE);
132*4882a593Smuzhiyun }
133*4882a593Smuzhiyun
vfio_mdev_exit(void)134*4882a593Smuzhiyun static void __exit vfio_mdev_exit(void)
135*4882a593Smuzhiyun {
136*4882a593Smuzhiyun mdev_unregister_driver(&vfio_mdev_driver);
137*4882a593Smuzhiyun }
138*4882a593Smuzhiyun
139*4882a593Smuzhiyun module_init(vfio_mdev_init)
140*4882a593Smuzhiyun module_exit(vfio_mdev_exit)
141*4882a593Smuzhiyun
142*4882a593Smuzhiyun MODULE_VERSION(DRIVER_VERSION);
143*4882a593Smuzhiyun MODULE_LICENSE("GPL v2");
144*4882a593Smuzhiyun MODULE_AUTHOR(DRIVER_AUTHOR);
145*4882a593Smuzhiyun MODULE_DESCRIPTION(DRIVER_DESC);
146