This is a first step so we can make more than one different address spaces. No change on the colde flow intended.
Signed-off-by: Eugenio Pérez <epere...@redhat.com> --- drivers/vdpa/vdpa_user/vduse_dev.c | 114 +++++++++++++++-------------- 1 file changed, 59 insertions(+), 55 deletions(-) diff --git a/drivers/vdpa/vdpa_user/vduse_dev.c b/drivers/vdpa/vdpa_user/vduse_dev.c index cdb3dc2b5e3f..7d2a3ed77b1e 100644 --- a/drivers/vdpa/vdpa_user/vduse_dev.c +++ b/drivers/vdpa/vdpa_user/vduse_dev.c @@ -85,6 +85,12 @@ struct vduse_umem { struct mm_struct *mm; }; +struct vduse_as { + struct vduse_iova_domain *domain; + struct vduse_umem *umem; + struct mutex mem_lock; +}; + struct vduse_vq_group_int { struct vduse_dev *dev; }; @@ -93,7 +99,7 @@ struct vduse_dev { struct vduse_vdpa *vdev; struct device *dev; struct vduse_virtqueue **vqs; - struct vduse_iova_domain *domain; + struct vduse_as as; char *name; struct mutex lock; spinlock_t msg_lock; @@ -121,9 +127,7 @@ struct vduse_dev { u32 vq_num; u32 vq_align; u32 ngroups; - struct vduse_umem *umem; struct vduse_vq_group_int *groups; - struct mutex mem_lock; unsigned int bounce_size; struct mutex domain_lock; }; @@ -438,7 +442,7 @@ static __poll_t vduse_dev_poll(struct file *file, poll_table *wait) static void vduse_dev_reset(struct vduse_dev *dev) { int i; - struct vduse_iova_domain *domain = dev->domain; + struct vduse_iova_domain *domain = dev->as.domain; /* The coherent mappings are handled in vduse_dev_free_coherent() */ if (domain && domain->bounce_map) @@ -814,13 +818,13 @@ static int vduse_vdpa_set_map(struct vdpa_device *vdpa, struct vduse_dev *dev = vdpa_to_vduse(vdpa); int ret; - ret = vduse_domain_set_map(dev->domain, iotlb); + ret = vduse_domain_set_map(dev->as.domain, iotlb); if (ret) return ret; ret = vduse_dev_update_iotlb(dev, 0ULL, ULLONG_MAX); if (ret) { - vduse_domain_clear_map(dev->domain, iotlb); + vduse_domain_clear_map(dev->as.domain, iotlb); return ret; } @@ -872,7 +876,7 @@ static void vduse_dev_sync_single_for_device(union virtio_map token, enum dma_data_direction dir) { struct vduse_dev *vdev = token.group->dev; - struct vduse_iova_domain *domain = vdev->domain; + struct vduse_iova_domain *domain = vdev->as.domain; vduse_domain_sync_single_for_device(domain, dma_addr, size, dir); } @@ -882,7 +886,7 @@ static void vduse_dev_sync_single_for_cpu(union virtio_map token, enum dma_data_direction dir) { struct vduse_dev *vdev = token.group->dev; - struct vduse_iova_domain *domain = vdev->domain; + struct vduse_iova_domain *domain = vdev->as.domain; vduse_domain_sync_single_for_cpu(domain, dma_addr, size, dir); } @@ -893,7 +897,7 @@ static dma_addr_t vduse_dev_map_page(union virtio_map token, struct page *page, unsigned long attrs) { struct vduse_dev *vdev = token.group->dev; - struct vduse_iova_domain *domain = vdev->domain; + struct vduse_iova_domain *domain = vdev->as.domain; return vduse_domain_map_page(domain, page, offset, size, dir, attrs); } @@ -903,7 +907,7 @@ static void vduse_dev_unmap_page(union virtio_map token, dma_addr_t dma_addr, unsigned long attrs) { struct vduse_dev *vdev = token.group->dev; - struct vduse_iova_domain *domain = vdev->domain; + struct vduse_iova_domain *domain = vdev->as.domain; return vduse_domain_unmap_page(domain, dma_addr, size, dir, attrs); } @@ -912,7 +916,7 @@ static void *vduse_dev_alloc_coherent(union virtio_map token, size_t size, dma_addr_t *dma_addr, gfp_t flag) { struct vduse_dev *vdev = token.group->dev; - struct vduse_iova_domain *domain = vdev->domain; + struct vduse_iova_domain *domain = vdev->as.domain; unsigned long iova; void *addr; @@ -932,7 +936,7 @@ static void vduse_dev_free_coherent(union virtio_map token, size_t size, unsigned long attrs) { struct vduse_dev *vdev = token.group->dev; - struct vduse_iova_domain *domain = vdev->domain; + struct vduse_iova_domain *domain = vdev->as.domain; vduse_domain_free_coherent(domain, size, vaddr, dma_addr, attrs); } @@ -940,7 +944,7 @@ static void vduse_dev_free_coherent(union virtio_map token, size_t size, static bool vduse_dev_need_sync(union virtio_map token, dma_addr_t dma_addr) { struct vduse_dev *vdev = token.group->dev; - struct vduse_iova_domain *domain = vdev->domain; + struct vduse_iova_domain *domain = vdev->as.domain; return dma_addr < domain->bounce_size; } @@ -955,7 +959,7 @@ static int vduse_dev_mapping_error(union virtio_map token, dma_addr_t dma_addr) static size_t vduse_dev_max_mapping_size(union virtio_map token) { struct vduse_dev *vdev = token.group->dev; - struct vduse_iova_domain *domain = vdev->domain; + struct vduse_iova_domain *domain = vdev->as.domain; return domain->bounce_size; } @@ -1102,29 +1106,29 @@ static int vduse_dev_dereg_umem(struct vduse_dev *dev, { int ret; - mutex_lock(&dev->mem_lock); + mutex_lock(&dev->as.mem_lock); ret = -ENOENT; - if (!dev->umem) + if (!dev->as.umem) goto unlock; ret = -EINVAL; - if (!dev->domain) + if (!dev->as.domain) goto unlock; - if (dev->umem->iova != iova || size != dev->domain->bounce_size) + if (dev->as.umem->iova != iova || size != dev->as.domain->bounce_size) goto unlock; - vduse_domain_remove_user_bounce_pages(dev->domain); - unpin_user_pages_dirty_lock(dev->umem->pages, - dev->umem->npages, true); - atomic64_sub(dev->umem->npages, &dev->umem->mm->pinned_vm); - mmdrop(dev->umem->mm); - vfree(dev->umem->pages); - kfree(dev->umem); - dev->umem = NULL; + vduse_domain_remove_user_bounce_pages(dev->as.domain); + unpin_user_pages_dirty_lock(dev->as.umem->pages, + dev->as.umem->npages, true); + atomic64_sub(dev->as.umem->npages, &dev->as.umem->mm->pinned_vm); + mmdrop(dev->as.umem->mm); + vfree(dev->as.umem->pages); + kfree(dev->as.umem); + dev->as.umem = NULL; ret = 0; unlock: - mutex_unlock(&dev->mem_lock); + mutex_unlock(&dev->as.mem_lock); return ret; } @@ -1137,14 +1141,14 @@ static int vduse_dev_reg_umem(struct vduse_dev *dev, unsigned long npages, lock_limit; int ret; - if (!dev->domain || !dev->domain->bounce_map || - size != dev->domain->bounce_size || + if (!dev->as.domain || !dev->as.domain->bounce_map || + size != dev->as.domain->bounce_size || iova != 0 || uaddr & ~PAGE_MASK) return -EINVAL; - mutex_lock(&dev->mem_lock); + mutex_lock(&dev->as.mem_lock); ret = -EEXIST; - if (dev->umem) + if (dev->as.umem) goto unlock; ret = -ENOMEM; @@ -1168,7 +1172,7 @@ static int vduse_dev_reg_umem(struct vduse_dev *dev, goto out; } - ret = vduse_domain_add_user_bounce_pages(dev->domain, + ret = vduse_domain_add_user_bounce_pages(dev->as.domain, page_list, pinned); if (ret) goto out; @@ -1181,7 +1185,7 @@ static int vduse_dev_reg_umem(struct vduse_dev *dev, umem->mm = current->mm; mmgrab(current->mm); - dev->umem = umem; + dev->as.umem = umem; out: if (ret && pinned > 0) unpin_user_pages(page_list, pinned); @@ -1192,7 +1196,7 @@ static int vduse_dev_reg_umem(struct vduse_dev *dev, vfree(page_list); kfree(umem); } - mutex_unlock(&dev->mem_lock); + mutex_unlock(&dev->as.mem_lock); return ret; } @@ -1238,12 +1242,12 @@ static long vduse_dev_ioctl(struct file *file, unsigned int cmd, break; mutex_lock(&dev->domain_lock); - if (!dev->domain) { + if (!dev->as.domain) { mutex_unlock(&dev->domain_lock); break; } - spin_lock(&dev->domain->iotlb_lock); - map = vhost_iotlb_itree_first(dev->domain->iotlb, + spin_lock(&dev->as.domain->iotlb_lock); + map = vhost_iotlb_itree_first(dev->as.domain->iotlb, entry.start, entry.last); if (map) { map_file = (struct vdpa_map_file *)map->opaque; @@ -1253,7 +1257,7 @@ static long vduse_dev_ioctl(struct file *file, unsigned int cmd, entry.last = map->last; entry.perm = map->perm; } - spin_unlock(&dev->domain->iotlb_lock); + spin_unlock(&dev->as.domain->iotlb_lock); mutex_unlock(&dev->domain_lock); ret = -EINVAL; if (!f) @@ -1447,22 +1451,22 @@ static long vduse_dev_ioctl(struct file *file, unsigned int cmd, break; mutex_lock(&dev->domain_lock); - if (!dev->domain) { + if (!dev->as.domain) { mutex_unlock(&dev->domain_lock); break; } - spin_lock(&dev->domain->iotlb_lock); - map = vhost_iotlb_itree_first(dev->domain->iotlb, + spin_lock(&dev->as.domain->iotlb_lock); + map = vhost_iotlb_itree_first(dev->as.domain->iotlb, info.start, info.last); if (map) { info.start = map->start; info.last = map->last; info.capability = 0; - if (dev->domain->bounce_map && map->start == 0 && - map->last == dev->domain->bounce_size - 1) + if (dev->as.domain->bounce_map && map->start == 0 && + map->last == dev->as.domain->bounce_size - 1) info.capability |= VDUSE_IOVA_CAP_UMEM; } - spin_unlock(&dev->domain->iotlb_lock); + spin_unlock(&dev->as.domain->iotlb_lock); mutex_unlock(&dev->domain_lock); if (!map) break; @@ -1487,8 +1491,8 @@ static int vduse_dev_release(struct inode *inode, struct file *file) struct vduse_dev *dev = file->private_data; mutex_lock(&dev->domain_lock); - if (dev->domain) - vduse_dev_dereg_umem(dev, 0, dev->domain->bounce_size); + if (dev->as.domain) + vduse_dev_dereg_umem(dev, 0, dev->as.domain->bounce_size); mutex_unlock(&dev->domain_lock); spin_lock(&dev->msg_lock); /* Make sure the inflight messages can processed after reconncection */ @@ -1707,7 +1711,7 @@ static struct vduse_dev *vduse_dev_create(void) return NULL; mutex_init(&dev->lock); - mutex_init(&dev->mem_lock); + mutex_init(&dev->as.mem_lock); mutex_init(&dev->domain_lock); spin_lock_init(&dev->msg_lock); INIT_LIST_HEAD(&dev->send_list); @@ -1758,8 +1762,8 @@ static int vduse_destroy_dev(char *name) idr_remove(&vduse_idr, dev->minor); kvfree(dev->config); vduse_dev_deinit_vqs(dev); - if (dev->domain) - vduse_domain_destroy(dev->domain); + if (dev->as.domain) + vduse_domain_destroy(dev->as.domain); kfree(dev->name); kfree(dev->groups); vduse_dev_destroy(dev); @@ -1875,7 +1879,7 @@ static ssize_t bounce_size_store(struct device *device, ret = -EPERM; mutex_lock(&dev->domain_lock); - if (dev->domain) + if (dev->as.domain) goto unlock; ret = kstrtouint(buf, 10, &bounce_size); @@ -2126,11 +2130,11 @@ static int vdpa_dev_add(struct vdpa_mgmt_dev *mdev, const char *name, return ret; mutex_lock(&dev->domain_lock); - if (!dev->domain) - dev->domain = vduse_domain_create(VDUSE_IOVA_SIZE - 1, + if (!dev->as.domain) + dev->as.domain = vduse_domain_create(VDUSE_IOVA_SIZE - 1, dev->bounce_size); mutex_unlock(&dev->domain_lock); - if (!dev->domain) { + if (!dev->as.domain) { put_device(&dev->vdev->vdpa.dev); return -ENOMEM; } @@ -2139,8 +2143,8 @@ static int vdpa_dev_add(struct vdpa_mgmt_dev *mdev, const char *name, if (ret) { put_device(&dev->vdev->vdpa.dev); mutex_lock(&dev->domain_lock); - vduse_domain_destroy(dev->domain); - dev->domain = NULL; + vduse_domain_destroy(dev->as.domain); + dev->as.domain = NULL; mutex_unlock(&dev->domain_lock); return ret; } -- 2.51.0