struct vfio_info_cap caps = {};
int ret;
+ if (unlikely(!device->ops->get_region_info_caps))
+ return -EINVAL;
+
if (copy_from_user(&info, arg, minsz))
return -EFAULT;
if (info.argsz < minsz)
return -EINVAL;
- if (device->ops->get_region_info_caps) {
- ret = device->ops->get_region_info_caps(device, &info, &caps);
- if (ret)
- goto out_free;
-
- if (caps.size) {
- info.flags |= VFIO_REGION_INFO_FLAG_CAPS;
- 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(arg + 1, caps.buf,
- caps.size)) {
- ret = -EFAULT;
- goto out_free;
- }
- info.cap_offset = sizeof(info);
+ ret = device->ops->get_region_info_caps(device, &info, &caps);
+ if (ret)
+ goto out_free;
+
+ if (caps.size) {
+ info.flags |= VFIO_REGION_INFO_FLAG_CAPS;
+ 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(arg + 1, caps.buf, caps.size)) {
+ ret = -EFAULT;
+ goto out_free;
}
+ info.cap_offset = sizeof(info);
}
+ }
- if (copy_to_user(arg, &info, minsz)) {
- ret = -EFAULT;
- goto out_free;
- }
- } else if (device->ops->get_region_info) {
- ret = device->ops->get_region_info(device, arg);
- if (ret)
- return ret;
- } else {
- return -EINVAL;
+ if (copy_to_user(arg, &info, minsz)){
+ ret = -EFAULT;
+ goto out_free;
}
out_free:
size_t count, loff_t *size);
long (*ioctl)(struct vfio_device *vdev, unsigned int cmd,
unsigned long arg);
- int (*get_region_info)(struct vfio_device *vdev,
- struct vfio_region_info __user *arg);
int (*get_region_info_caps)(struct vfio_device *vdev,
struct vfio_region_info *info,
struct vfio_info_cap *caps);