#include <linux/virtio_ring.h>
 #include <linux/virtio_pci.h>
 #include <linux/virtio_pci_modern.h>
+#include <uapi/linux/vdpa.h>
 
 #define VP_VDPA_QUEUE_MAX 256
 #define VP_VDPA_DRIVER_NAME "vp_vdpa"
        struct virtio_pci_modern_device *mdev;
        struct vp_vring *vring;
        struct vdpa_callback config_cb;
+       u64 device_features;
        char msix_name[VP_VDPA_NAME_SIZE];
        int config_irq;
        int queues;
 
 static u64 vp_vdpa_get_device_features(struct vdpa_device *vdpa)
 {
-       struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
+       struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
 
-       return vp_modern_get_features(mdev);
+       return vp_vdpa->device_features;
 }
 
 static int vp_vdpa_set_driver_features(struct vdpa_device *vdpa, u64 features)
        struct pci_dev *pdev = mdev->pci_dev;
        struct device *dev = &pdev->dev;
        struct vp_vdpa *vp_vdpa = NULL;
+       u64 device_features;
        int ret, i;
 
        vp_vdpa = vdpa_alloc_device(struct vp_vdpa, vdpa,
        vp_vdpa->queues = vp_modern_get_num_queues(mdev);
        vp_vdpa->mdev = mdev;
 
+       device_features = vp_modern_get_features(mdev);
+       if (add_config->mask & BIT_ULL(VDPA_ATTR_DEV_FEATURES)) {
+               if (add_config->device_features & ~device_features) {
+                       ret = -EINVAL;
+                       dev_err(&pdev->dev, "Try to provision features "
+                               "that are not supported by the device: "
+                               "device_features 0x%llx provisioned 0x%llx\n",
+                               device_features, add_config->device_features);
+                       goto err;
+               }
+               device_features = add_config->device_features;
+       }
+       vp_vdpa->device_features = device_features;
+
        ret = devm_add_action_or_reset(dev, vp_vdpa_free_irq_vectors, pdev);
        if (ret) {
                dev_err(&pdev->dev,
        mgtdev->id_table = mdev_id;
        mgtdev->max_supported_vqs = vp_modern_get_num_queues(mdev);
        mgtdev->supported_features = vp_modern_get_features(mdev);
+       mgtdev->config_attr_mask = (1 << VDPA_ATTR_DEV_FEATURES);
        pci_set_master(pdev);
        pci_set_drvdata(pdev, vp_vdpa_mgtdev);