@@ -8,3 +8,11 @@ config MDEV
See Documentation/vfio-mediated-device.txt for more details.
If you don't know what do here, say N.
+
+config VFIO_MDEV
+ tristate "VFIO Bus driver for Mediated devices"
+ depends on VFIO && MDEV
+ default n
+ help
+ VFIO Bus driver for mediated devices.
+
@@ -2,3 +2,4 @@
mdev-y := mdev_core.o mdev_sysfs.o mdev_driver.o
obj-$(CONFIG_MDEV) += mdev.o
+obj-$(CONFIG_VFIO_MDEV) += vfio_mdev.o
@@ -1,10 +1,15 @@
/*
- * VFIO based Mediated PCI device driver
+ * VFIO Bus driver for Mediated device
*
* Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved.
* Author: Neo Jia <cjia@nvidia.com>
* Kirti Wankhede <kwankhede@nvidia.com>
*
+ * Copyright (c) 2016 Intel Corporation.
+ * Author:
+ * Xiao Guangrong <guangrong.xiao@linux.intel.com>
+ * Jike Song <jike.song@intel.com>
+ *
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 as
* published by the Free Software Foundation.
@@ -22,24 +27,21 @@
#include "mdev_private.h"
-#define DRIVER_VERSION "0.1"
+#define DRIVER_VERSION "0.2"
#define DRIVER_AUTHOR "NVIDIA Corporation"
-#define DRIVER_DESC "VFIO based Mediated PCI device driver"
+#define DRIVER_DESC "VFIO Bus driver for Mediated device"
struct vfio_mdev {
struct iommu_group *group;
struct mdev_device *mdev;
- struct vfio_device_info dev_info;
};
static int vfio_mdev_open(void *device_data)
{
- int ret = 0;
-
if (!try_module_get(THIS_MODULE))
return -ENODEV;
- return ret;
+ return 0;
}
static void vfio_mdev_close(void *device_data)
@@ -47,220 +49,17 @@ static void vfio_mdev_close(void *device_data)
module_put(THIS_MODULE);
}
-static int sparse_mmap_cap(struct vfio_info_cap *caps, void *cap_type)
-{
- struct vfio_info_cap_header *header;
- struct vfio_region_info_cap_sparse_mmap *sparse_cap, *sparse = cap_type;
- size_t size;
-
- size = sizeof(*sparse) + sparse->nr_areas * sizeof(*sparse->areas);
- header = vfio_info_cap_add(caps, size,
- VFIO_REGION_INFO_CAP_SPARSE_MMAP, 1);
- if (IS_ERR(header))
- return PTR_ERR(header);
-
- sparse_cap = container_of(header,
- struct vfio_region_info_cap_sparse_mmap, header);
- sparse_cap->nr_areas = sparse->nr_areas;
- memcpy(sparse_cap->areas, sparse->areas,
- sparse->nr_areas * sizeof(*sparse->areas));
- return 0;
-}
-
-static int region_type_cap(struct vfio_info_cap *caps, void *cap_type)
-{
- struct vfio_info_cap_header *header;
- struct vfio_region_info_cap_type *type_cap, *cap = cap_type;
-
- header = vfio_info_cap_add(caps, sizeof(*cap),
- VFIO_REGION_INFO_CAP_TYPE, 1);
- if (IS_ERR(header))
- return PTR_ERR(header);
-
- type_cap = container_of(header, struct vfio_region_info_cap_type,
- header);
- type_cap->type = cap->type;
- type_cap->subtype = cap->type;
- return 0;
-}
-
static long vfio_mdev_unlocked_ioctl(void *device_data,
unsigned int cmd, unsigned long arg)
{
- int ret = 0;
struct vfio_mdev *vmdev = device_data;
- struct parent_device *parent = vmdev->mdev->parent;
- unsigned long minsz;
-
- switch (cmd) {
- case VFIO_DEVICE_GET_INFO:
- {
- struct vfio_device_info info;
-
- minsz = offsetofend(struct vfio_device_info, num_irqs);
-
- if (copy_from_user(&info, (void __user *)arg, minsz))
- return -EFAULT;
-
- if (info.argsz < minsz)
- return -EINVAL;
-
- if (parent->ops->get_device_info)
- ret = parent->ops->get_device_info(vmdev->mdev, &info);
- else
- return -EINVAL;
-
- if (ret)
- return ret;
-
- if (parent->ops->reset)
- info.flags |= VFIO_DEVICE_FLAGS_RESET;
-
- memcpy(&vmdev->dev_info, &info, sizeof(info));
-
- return copy_to_user((void __user *)arg, &info, minsz);
- }
- case VFIO_DEVICE_GET_REGION_INFO:
- {
- struct vfio_region_info info;
- struct vfio_info_cap caps = { .buf = NULL, .size = 0 };
- u16 cap_type_id = 0;
- void *cap_type = NULL;
-
- minsz = offsetofend(struct vfio_region_info, offset);
-
- if (copy_from_user(&info, (void __user *)arg, minsz))
- return -EFAULT;
-
- if (info.argsz < minsz)
- return -EINVAL;
-
- if (parent->ops->get_region_info)
- ret = parent->ops->get_region_info(vmdev->mdev, &info,
- &cap_type_id, &cap_type);
- else
- return -EINVAL;
-
- if (ret)
- return ret;
-
- if ((info.flags & VFIO_REGION_INFO_FLAG_CAPS) && cap_type) {
- switch (cap_type_id) {
- case VFIO_REGION_INFO_CAP_SPARSE_MMAP:
- ret = sparse_mmap_cap(&caps, cap_type);
- if (ret)
- return ret;
- break;
-
- case VFIO_REGION_INFO_CAP_TYPE:
- ret = region_type_cap(&caps, cap_type);
- if (ret)
- return ret;
- break;
- default:
- return -EINVAL;
- }
- }
-
- if (caps.size) {
- if (info.argsz < sizeof(info) + caps.size) {
- info.argsz = sizeof(info) + caps.size;
- info.cap_offset = 0;
- } else {
- vfio_info_cap_shift(&caps, sizeof(info));
- if (copy_to_user((void __user *)arg +
- sizeof(info), caps.buf,
- caps.size)) {
- kfree(caps.buf);
- return -EFAULT;
- }
- info.cap_offset = sizeof(info);
- }
- kfree(caps.buf);
- }
-
- return copy_to_user((void __user *)arg, &info, minsz);
- }
- case VFIO_DEVICE_GET_IRQ_INFO:
- {
- struct vfio_irq_info info;
-
- minsz = offsetofend(struct vfio_irq_info, count);
-
- if (copy_from_user(&info, (void __user *)arg, minsz))
- return -EFAULT;
-
- if ((info.argsz < minsz) ||
- (info.index >= vmdev->dev_info.num_irqs))
- return -EINVAL;
-
- if (parent->ops->get_irq_info)
- ret = parent->ops->get_irq_info(vmdev->mdev, &info);
- else
- return -EINVAL;
-
- if (ret)
- return ret;
-
- if (info.count == -1)
- return -EINVAL;
-
- return copy_to_user((void __user *)arg, &info, minsz);
- }
- case VFIO_DEVICE_SET_IRQS:
- {
- struct vfio_irq_set hdr;
- u8 *data = NULL, *ptr = NULL;
-
- minsz = offsetofend(struct vfio_irq_set, count);
-
- if (copy_from_user(&hdr, (void __user *)arg, minsz))
- return -EFAULT;
-
- if ((hdr.argsz < minsz) ||
- (hdr.index >= vmdev->dev_info.num_irqs) ||
- (hdr.flags & ~(VFIO_IRQ_SET_DATA_TYPE_MASK |
- VFIO_IRQ_SET_ACTION_TYPE_MASK)))
- return -EINVAL;
-
- if (!(hdr.flags & VFIO_IRQ_SET_DATA_NONE)) {
- size_t size;
-
- if (hdr.flags & VFIO_IRQ_SET_DATA_BOOL)
- size = sizeof(uint8_t);
- else if (hdr.flags & VFIO_IRQ_SET_DATA_EVENTFD)
- size = sizeof(int32_t);
- else
- return -EINVAL;
-
- if (hdr.argsz - minsz < hdr.count * size)
- return -EINVAL;
-
- ptr = data = memdup_user((void __user *)(arg + minsz),
- hdr.count * size);
- if (IS_ERR(data))
- return PTR_ERR(data);
- }
-
- if (parent->ops->set_irqs)
- ret = parent->ops->set_irqs(vmdev->mdev, hdr.flags,
- hdr.index, hdr.start,
- hdr.count, data);
- else
- ret = -EINVAL;
-
- kfree(ptr);
- return ret;
- }
- case VFIO_DEVICE_RESET:
- {
- if (parent->ops->reset)
- return parent->ops->reset(vmdev->mdev);
-
- return -EINVAL;
- }
- }
- return -ENOTTY;
+ struct mdev_device *mdev = vmdev->mdev;
+ struct mdev_host *host = dev_to_host(mdev->dev.parent);
+
+ if (host->ops->ioctl)
+ return host->ops->ioctl(mdev, cmd, arg);
+
+ return -ENODEV;
}
static ssize_t vfio_mdev_read(void *device_data, char __user *buf,
@@ -268,63 +67,12 @@ static ssize_t vfio_mdev_read(void *device_data, char __user *buf,
{
struct vfio_mdev *vmdev = device_data;
struct mdev_device *mdev = vmdev->mdev;
- struct parent_device *parent = mdev->parent;
- unsigned int done = 0;
- int ret;
-
- if (!parent->ops->read)
- return -EINVAL;
-
- while (count) {
- size_t filled;
-
- if (count >= 4 && !(*ppos % 4)) {
- u32 val;
-
- ret = parent->ops->read(mdev, (char *)&val, sizeof(val),
- *ppos);
- if (ret <= 0)
- goto read_err;
-
- if (copy_to_user(buf, &val, sizeof(val)))
- goto read_err;
-
- filled = 4;
- } else if (count >= 2 && !(*ppos % 2)) {
- u16 val;
-
- ret = parent->ops->read(mdev, (char *)&val, sizeof(val),
- *ppos);
- if (ret <= 0)
- goto read_err;
-
- if (copy_to_user(buf, &val, sizeof(val)))
- goto read_err;
-
- filled = 2;
- } else {
- u8 val;
-
- ret = parent->ops->read(mdev, &val, sizeof(val), *ppos);
- if (ret <= 0)
- goto read_err;
+ struct mdev_host *host = dev_to_host(mdev->dev.parent);
- if (copy_to_user(buf, &val, sizeof(val)))
- goto read_err;
+ if (host->ops->read)
+ return host->ops->read(mdev, buf, count, ppos);
- filled = 1;
- }
-
- count -= filled;
- done += filled;
- *ppos += filled;
- buf += filled;
- }
-
- return done;
-
-read_err:
- return -EFAULT;
+ return -ENODEV;
}
static ssize_t vfio_mdev_write(void *device_data, const char __user *buf,
@@ -332,75 +80,24 @@ static ssize_t vfio_mdev_write(void *device_data, const char __user *buf,
{
struct vfio_mdev *vmdev = device_data;
struct mdev_device *mdev = vmdev->mdev;
- struct parent_device *parent = mdev->parent;
- unsigned int done = 0;
- int ret;
-
- if (!parent->ops->write)
- return -EINVAL;
-
- while (count) {
- size_t filled;
-
- if (count >= 4 && !(*ppos % 4)) {
- u32 val;
-
- if (copy_from_user(&val, buf, sizeof(val)))
- goto write_err;
-
- ret = parent->ops->write(mdev, (char *)&val,
- sizeof(val), *ppos);
- if (ret <= 0)
- goto write_err;
-
- filled = 4;
- } else if (count >= 2 && !(*ppos % 2)) {
- u16 val;
-
- if (copy_from_user(&val, buf, sizeof(val)))
- goto write_err;
-
- ret = parent->ops->write(mdev, (char *)&val,
- sizeof(val), *ppos);
- if (ret <= 0)
- goto write_err;
-
- filled = 2;
- } else {
- u8 val;
+ struct mdev_host *host = dev_to_host(mdev->dev.parent);
- if (copy_from_user(&val, buf, sizeof(val)))
- goto write_err;
+ if (host->ops->write)
+ return host->ops->write(mdev, buf, count, ppos);
- ret = parent->ops->write(mdev, &val, sizeof(val),
- *ppos);
- if (ret <= 0)
- goto write_err;
-
- filled = 1;
- }
-
- count -= filled;
- done += filled;
- *ppos += filled;
- buf += filled;
- }
-
- return done;
-write_err:
- return -EFAULT;
+ return -ENODEV;
}
static int vfio_mdev_mmap(void *device_data, struct vm_area_struct *vma)
{
struct vfio_mdev *vmdev = device_data;
struct mdev_device *mdev = vmdev->mdev;
- struct parent_device *parent = mdev->parent;
+ struct mdev_host *host = dev_to_host(mdev->dev.parent);
- if (parent->ops->mmap)
- return parent->ops->mmap(mdev, vma);
+ if (host->ops->mmap)
+ return host->ops->mmap(mdev, vma);
- return -EINVAL;
+ return -ENODEV;
}
static const struct vfio_device_ops vfio_mdev_dev_ops = {
@@ -413,28 +110,27 @@ static const struct vfio_device_ops vfio_mdev_dev_ops = {
.mmap = vfio_mdev_mmap,
};
-int vfio_mdev_probe(struct device *dev)
+static int vfio_mdev_probe(struct device *dev)
{
struct vfio_mdev *vmdev;
- struct mdev_device *mdev = to_mdev_device(dev);
+ struct mdev_device *mdev = dev_to_mdev(dev);
int ret;
vmdev = kzalloc(sizeof(*vmdev), GFP_KERNEL);
if (IS_ERR(vmdev))
return PTR_ERR(vmdev);
- vmdev->mdev = mdev_get_device(mdev);
+ vmdev->mdev = mdev;
vmdev->group = mdev->group;
ret = vfio_add_group_dev(dev, &vfio_mdev_dev_ops, vmdev);
if (ret)
kfree(vmdev);
- mdev_put_device(mdev);
return ret;
}
-void vfio_mdev_remove(struct device *dev)
+static void vfio_mdev_remove(struct device *dev)
{
struct vfio_mdev *vmdev;
@@ -442,10 +138,34 @@ void vfio_mdev_remove(struct device *dev)
kfree(vmdev);
}
-struct mdev_driver vfio_mdev_driver = {
- .name = "vfio_mdev",
- .probe = vfio_mdev_probe,
- .remove = vfio_mdev_remove,
+static int vfio_mdev_online(struct device *dev)
+{
+ struct mdev_device *mdev = dev_to_mdev(dev);
+ struct mdev_host *host = dev_to_host(mdev->dev.parent);
+
+ if (host->ops->start)
+ return host->ops->start(mdev);
+
+ return -ENOTSUPP;
+}
+
+static int vfio_mdev_offline(struct device *dev)
+{
+ struct mdev_device *mdev = dev_to_mdev(dev);
+ struct mdev_host *host = dev_to_host(mdev->dev.parent);
+
+ if (host->ops->stop)
+ return host->ops->stop(mdev);
+
+ return -ENOTSUPP;
+}
+
+static struct mdev_driver vfio_mdev_driver = {
+ .name = "vfio_mdev",
+ .probe = vfio_mdev_probe,
+ .remove = vfio_mdev_remove,
+ .online = vfio_mdev_online,
+ .offline = vfio_mdev_offline,
};
static int __init vfio_mdev_init(void)