xref: /qemu/hw/virtio/vhost-vdpa.c (revision bc865bfe2d54ac56daa20a6395abff68fab6a5d6)
1108a6481SCindy Lu /*
2108a6481SCindy Lu  * vhost-vdpa
3108a6481SCindy Lu  *
4108a6481SCindy Lu  *  Copyright(c) 2017-2018 Intel Corporation.
5108a6481SCindy Lu  *  Copyright(c) 2020 Red Hat, Inc.
6108a6481SCindy Lu  *
7108a6481SCindy Lu  * This work is licensed under the terms of the GNU GPL, version 2 or later.
8108a6481SCindy Lu  * See the COPYING file in the top-level directory.
9108a6481SCindy Lu  *
10108a6481SCindy Lu  */
11108a6481SCindy Lu 
12108a6481SCindy Lu #include "qemu/osdep.h"
13108a6481SCindy Lu #include <linux/vhost.h>
14108a6481SCindy Lu #include <linux/vfio.h>
15108a6481SCindy Lu #include <sys/eventfd.h>
16108a6481SCindy Lu #include <sys/ioctl.h>
1733f21860SPhilippe Mathieu-Daudé #include "exec/target_page.h"
18108a6481SCindy Lu #include "hw/virtio/vhost.h"
19108a6481SCindy Lu #include "hw/virtio/vhost-backend.h"
20108a6481SCindy Lu #include "hw/virtio/virtio-net.h"
21dff4426fSEugenio Pérez #include "hw/virtio/vhost-shadow-virtqueue.h"
22108a6481SCindy Lu #include "hw/virtio/vhost-vdpa.h"
23df77d45aSXie Yongji #include "exec/address-spaces.h"
24c156d5bfSEugenio Pérez #include "migration/blocker.h"
25415b7327SMarc-André Lureau #include "qemu/cutils.h"
26108a6481SCindy Lu #include "qemu/main-loop.h"
27778e67deSLaurent Vivier #include "trace.h"
28dff4426fSEugenio Pérez #include "qapi/error.h"
29108a6481SCindy Lu 
30032e4d68SEugenio Pérez /*
31032e4d68SEugenio Pérez  * Return one past the end of the end of section. Be careful with uint64_t
32032e4d68SEugenio Pérez  * conversions!
33032e4d68SEugenio Pérez  */
348b1a8884SPhilippe Mathieu-Daudé static Int128 vhost_vdpa_section_end(const MemoryRegionSection *section,
358b1a8884SPhilippe Mathieu-Daudé                                      int page_mask)
36032e4d68SEugenio Pérez {
37032e4d68SEugenio Pérez     Int128 llend = int128_make64(section->offset_within_address_space);
38032e4d68SEugenio Pérez     llend = int128_add(llend, section->size);
398b1a8884SPhilippe Mathieu-Daudé     llend = int128_and(llend, int128_exts64(page_mask));
40032e4d68SEugenio Pérez 
41032e4d68SEugenio Pérez     return llend;
42032e4d68SEugenio Pérez }
43032e4d68SEugenio Pérez 
44013108b6SEugenio Pérez static bool vhost_vdpa_listener_skipped_section(MemoryRegionSection *section,
45013108b6SEugenio Pérez                                                 uint64_t iova_min,
46961d60e9SPhilippe Mathieu-Daudé                                                 uint64_t iova_max,
47961d60e9SPhilippe Mathieu-Daudé                                                 int page_mask)
48108a6481SCindy Lu {
49013108b6SEugenio Pérez     Int128 llend;
50013108b6SEugenio Pérez 
51013108b6SEugenio Pérez     if ((!memory_region_is_ram(section->mr) &&
52108a6481SCindy Lu          !memory_region_is_iommu(section->mr)) ||
53c64038c9SEugenio Pérez         memory_region_is_protected(section->mr) ||
54d60c75d2SJason Wang         /* vhost-vDPA doesn't allow MMIO to be mapped  */
55013108b6SEugenio Pérez         memory_region_is_ram_device(section->mr)) {
56013108b6SEugenio Pérez         return true;
57013108b6SEugenio Pérez     }
58013108b6SEugenio Pérez 
59013108b6SEugenio Pérez     if (section->offset_within_address_space < iova_min) {
60013108b6SEugenio Pérez         error_report("RAM section out of device range (min=0x%" PRIx64
61013108b6SEugenio Pérez                      ", addr=0x%" HWADDR_PRIx ")",
62013108b6SEugenio Pérez                      iova_min, section->offset_within_address_space);
63013108b6SEugenio Pérez         return true;
64013108b6SEugenio Pérez     }
65bc7b0cacSCindy Lu     /*
66bc7b0cacSCindy Lu      * While using vIOMMU, sometimes the section will be larger than iova_max,
67bc7b0cacSCindy Lu      * but the memory that actually maps is smaller, so move the check to
68bc7b0cacSCindy Lu      * function vhost_vdpa_iommu_map_notify(). That function will use the actual
69bc7b0cacSCindy Lu      * size that maps to the kernel
70bc7b0cacSCindy Lu      */
71013108b6SEugenio Pérez 
72bc7b0cacSCindy Lu     if (!memory_region_is_iommu(section->mr)) {
738b1a8884SPhilippe Mathieu-Daudé         llend = vhost_vdpa_section_end(section, page_mask);
74013108b6SEugenio Pérez         if (int128_gt(llend, int128_make64(iova_max))) {
75013108b6SEugenio Pérez             error_report("RAM section out of device range (max=0x%" PRIx64
76013108b6SEugenio Pérez                          ", end addr=0x%" PRIx64 ")",
77013108b6SEugenio Pérez                          iova_max, int128_get64(llend));
78013108b6SEugenio Pérez             return true;
79013108b6SEugenio Pérez         }
80bc7b0cacSCindy Lu     }
81013108b6SEugenio Pérez 
82013108b6SEugenio Pérez     return false;
83108a6481SCindy Lu }
84108a6481SCindy Lu 
85cd831ed5SEugenio Pérez /*
86cd831ed5SEugenio Pérez  * The caller must set asid = 0 if the device does not support asid.
87cd831ed5SEugenio Pérez  * This is not an ABI break since it is set to 0 by the initializer anyway.
88cd831ed5SEugenio Pérez  */
89cd831ed5SEugenio Pérez int vhost_vdpa_dma_map(struct vhost_vdpa *v, uint32_t asid, hwaddr iova,
90cd831ed5SEugenio Pérez                        hwaddr size, void *vaddr, bool readonly)
91108a6481SCindy Lu {
92386494f2SCindy Lu     struct vhost_msg_v2 msg = {};
93108a6481SCindy Lu     int fd = v->device_fd;
94108a6481SCindy Lu     int ret = 0;
95108a6481SCindy Lu 
96108a6481SCindy Lu     msg.type = v->msg_type;
97cd831ed5SEugenio Pérez     msg.asid = asid;
98108a6481SCindy Lu     msg.iotlb.iova = iova;
99108a6481SCindy Lu     msg.iotlb.size = size;
100108a6481SCindy Lu     msg.iotlb.uaddr = (uint64_t)(uintptr_t)vaddr;
101108a6481SCindy Lu     msg.iotlb.perm = readonly ? VHOST_ACCESS_RO : VHOST_ACCESS_RW;
102108a6481SCindy Lu     msg.iotlb.type = VHOST_IOTLB_UPDATE;
103108a6481SCindy Lu 
104cd831ed5SEugenio Pérez     trace_vhost_vdpa_dma_map(v, fd, msg.type, msg.asid, msg.iotlb.iova,
105cd831ed5SEugenio Pérez                              msg.iotlb.size, msg.iotlb.uaddr, msg.iotlb.perm,
106cd831ed5SEugenio Pérez                              msg.iotlb.type);
107778e67deSLaurent Vivier 
108108a6481SCindy Lu     if (write(fd, &msg, sizeof(msg)) != sizeof(msg)) {
109108a6481SCindy Lu         error_report("failed to write, fd=%d, errno=%d (%s)",
110108a6481SCindy Lu             fd, errno, strerror(errno));
111108a6481SCindy Lu         return -EIO ;
112108a6481SCindy Lu     }
113108a6481SCindy Lu 
114108a6481SCindy Lu     return ret;
115108a6481SCindy Lu }
116108a6481SCindy Lu 
117cd831ed5SEugenio Pérez /*
118cd831ed5SEugenio Pérez  * The caller must set asid = 0 if the device does not support asid.
119cd831ed5SEugenio Pérez  * This is not an ABI break since it is set to 0 by the initializer anyway.
120cd831ed5SEugenio Pérez  */
121cd831ed5SEugenio Pérez int vhost_vdpa_dma_unmap(struct vhost_vdpa *v, uint32_t asid, hwaddr iova,
122cd831ed5SEugenio Pérez                          hwaddr size)
123108a6481SCindy Lu {
124386494f2SCindy Lu     struct vhost_msg_v2 msg = {};
125108a6481SCindy Lu     int fd = v->device_fd;
126108a6481SCindy Lu     int ret = 0;
127108a6481SCindy Lu 
128108a6481SCindy Lu     msg.type = v->msg_type;
129cd831ed5SEugenio Pérez     msg.asid = asid;
130108a6481SCindy Lu     msg.iotlb.iova = iova;
131108a6481SCindy Lu     msg.iotlb.size = size;
132108a6481SCindy Lu     msg.iotlb.type = VHOST_IOTLB_INVALIDATE;
133108a6481SCindy Lu 
134cd831ed5SEugenio Pérez     trace_vhost_vdpa_dma_unmap(v, fd, msg.type, msg.asid, msg.iotlb.iova,
135778e67deSLaurent Vivier                                msg.iotlb.size, msg.iotlb.type);
136778e67deSLaurent Vivier 
137108a6481SCindy Lu     if (write(fd, &msg, sizeof(msg)) != sizeof(msg)) {
138108a6481SCindy Lu         error_report("failed to write, fd=%d, errno=%d (%s)",
139108a6481SCindy Lu             fd, errno, strerror(errno));
140108a6481SCindy Lu         return -EIO ;
141108a6481SCindy Lu     }
142108a6481SCindy Lu 
143108a6481SCindy Lu     return ret;
144108a6481SCindy Lu }
145108a6481SCindy Lu 
146e6db5df7SEugenio Pérez static void vhost_vdpa_listener_begin_batch(struct vhost_vdpa *v)
147a5bd0580SJason Wang {
148a5bd0580SJason Wang     int fd = v->device_fd;
149e6db5df7SEugenio Pérez     struct vhost_msg_v2 msg = {
150e6db5df7SEugenio Pérez         .type = v->msg_type,
151e6db5df7SEugenio Pérez         .iotlb.type = VHOST_IOTLB_BATCH_BEGIN,
152e6db5df7SEugenio Pérez     };
153a5bd0580SJason Wang 
1545580b9f0SEugenio Pérez     trace_vhost_vdpa_listener_begin_batch(v, fd, msg.type, msg.iotlb.type);
155a5bd0580SJason Wang     if (write(fd, &msg, sizeof(msg)) != sizeof(msg)) {
156a5bd0580SJason Wang         error_report("failed to write, fd=%d, errno=%d (%s)",
157a5bd0580SJason Wang                      fd, errno, strerror(errno));
158a5bd0580SJason Wang     }
159a5bd0580SJason Wang }
160a5bd0580SJason Wang 
161e6db5df7SEugenio Pérez static void vhost_vdpa_iotlb_batch_begin_once(struct vhost_vdpa *v)
162e6db5df7SEugenio Pérez {
163e6db5df7SEugenio Pérez     if (v->dev->backend_cap & (0x1ULL << VHOST_BACKEND_F_IOTLB_BATCH) &&
164e6db5df7SEugenio Pérez         !v->iotlb_batch_begin_sent) {
165e6db5df7SEugenio Pérez         vhost_vdpa_listener_begin_batch(v);
166e6db5df7SEugenio Pérez     }
167e6db5df7SEugenio Pérez 
168e6db5df7SEugenio Pérez     v->iotlb_batch_begin_sent = true;
169e6db5df7SEugenio Pérez }
170e6db5df7SEugenio Pérez 
171a5bd0580SJason Wang static void vhost_vdpa_listener_commit(MemoryListener *listener)
172a5bd0580SJason Wang {
173a5bd0580SJason Wang     struct vhost_vdpa *v = container_of(listener, struct vhost_vdpa, listener);
174a5bd0580SJason Wang     struct vhost_dev *dev = v->dev;
1758acb3218SPhilippe Mathieu-Daudé     struct vhost_msg_v2 msg = {};
176a5bd0580SJason Wang     int fd = v->device_fd;
177a5bd0580SJason Wang 
178a5bd0580SJason Wang     if (!(dev->backend_cap & (0x1ULL << VHOST_BACKEND_F_IOTLB_BATCH))) {
179a5bd0580SJason Wang         return;
180a5bd0580SJason Wang     }
181a5bd0580SJason Wang 
182e6db5df7SEugenio Pérez     if (!v->iotlb_batch_begin_sent) {
183e6db5df7SEugenio Pérez         return;
184e6db5df7SEugenio Pérez     }
185e6db5df7SEugenio Pérez 
186a5bd0580SJason Wang     msg.type = v->msg_type;
187a5bd0580SJason Wang     msg.iotlb.type = VHOST_IOTLB_BATCH_END;
188a5bd0580SJason Wang 
1895580b9f0SEugenio Pérez     trace_vhost_vdpa_listener_commit(v, fd, msg.type, msg.iotlb.type);
190a5bd0580SJason Wang     if (write(fd, &msg, sizeof(msg)) != sizeof(msg)) {
191a5bd0580SJason Wang         error_report("failed to write, fd=%d, errno=%d (%s)",
192a5bd0580SJason Wang                      fd, errno, strerror(errno));
193a5bd0580SJason Wang     }
194e6db5df7SEugenio Pérez 
195e6db5df7SEugenio Pérez     v->iotlb_batch_begin_sent = false;
196a5bd0580SJason Wang }
197a5bd0580SJason Wang 
198bc7b0cacSCindy Lu static void vhost_vdpa_iommu_map_notify(IOMMUNotifier *n, IOMMUTLBEntry *iotlb)
199bc7b0cacSCindy Lu {
200bc7b0cacSCindy Lu     struct vdpa_iommu *iommu = container_of(n, struct vdpa_iommu, n);
201bc7b0cacSCindy Lu 
202bc7b0cacSCindy Lu     hwaddr iova = iotlb->iova + iommu->iommu_offset;
203bc7b0cacSCindy Lu     struct vhost_vdpa *v = iommu->dev;
204bc7b0cacSCindy Lu     void *vaddr;
205bc7b0cacSCindy Lu     int ret;
206bc7b0cacSCindy Lu     Int128 llend;
207bc7b0cacSCindy Lu 
208bc7b0cacSCindy Lu     if (iotlb->target_as != &address_space_memory) {
209bc7b0cacSCindy Lu         error_report("Wrong target AS \"%s\", only system memory is allowed",
210bc7b0cacSCindy Lu                      iotlb->target_as->name ? iotlb->target_as->name : "none");
211bc7b0cacSCindy Lu         return;
212bc7b0cacSCindy Lu     }
213bc7b0cacSCindy Lu     RCU_READ_LOCK_GUARD();
214bc7b0cacSCindy Lu     /* check if RAM section out of device range */
215bc7b0cacSCindy Lu     llend = int128_add(int128_makes64(iotlb->addr_mask), int128_makes64(iova));
216bc7b0cacSCindy Lu     if (int128_gt(llend, int128_make64(v->iova_range.last))) {
217bc7b0cacSCindy Lu         error_report("RAM section out of device range (max=0x%" PRIx64
218bc7b0cacSCindy Lu                      ", end addr=0x%" PRIx64 ")",
219bc7b0cacSCindy Lu                      v->iova_range.last, int128_get64(llend));
220bc7b0cacSCindy Lu         return;
221bc7b0cacSCindy Lu     }
222bc7b0cacSCindy Lu 
223bc7b0cacSCindy Lu     if ((iotlb->perm & IOMMU_RW) != IOMMU_NONE) {
224bc7b0cacSCindy Lu         bool read_only;
225bc7b0cacSCindy Lu 
226bc7b0cacSCindy Lu         if (!memory_get_xlat_addr(iotlb, &vaddr, NULL, &read_only, NULL)) {
227bc7b0cacSCindy Lu             return;
228bc7b0cacSCindy Lu         }
229bc7b0cacSCindy Lu         ret = vhost_vdpa_dma_map(v, VHOST_VDPA_GUEST_PA_ASID, iova,
230bc7b0cacSCindy Lu                                  iotlb->addr_mask + 1, vaddr, read_only);
231bc7b0cacSCindy Lu         if (ret) {
232bc7b0cacSCindy Lu             error_report("vhost_vdpa_dma_map(%p, 0x%" HWADDR_PRIx ", "
233bc7b0cacSCindy Lu                          "0x%" HWADDR_PRIx ", %p) = %d (%m)",
234bc7b0cacSCindy Lu                          v, iova, iotlb->addr_mask + 1, vaddr, ret);
235bc7b0cacSCindy Lu         }
236bc7b0cacSCindy Lu     } else {
237bc7b0cacSCindy Lu         ret = vhost_vdpa_dma_unmap(v, VHOST_VDPA_GUEST_PA_ASID, iova,
238bc7b0cacSCindy Lu                                    iotlb->addr_mask + 1);
239bc7b0cacSCindy Lu         if (ret) {
240bc7b0cacSCindy Lu             error_report("vhost_vdpa_dma_unmap(%p, 0x%" HWADDR_PRIx ", "
241bc7b0cacSCindy Lu                          "0x%" HWADDR_PRIx ") = %d (%m)",
242bc7b0cacSCindy Lu                          v, iova, iotlb->addr_mask + 1, ret);
243bc7b0cacSCindy Lu         }
244bc7b0cacSCindy Lu     }
245bc7b0cacSCindy Lu }
246bc7b0cacSCindy Lu 
247bc7b0cacSCindy Lu static void vhost_vdpa_iommu_region_add(MemoryListener *listener,
248bc7b0cacSCindy Lu                                         MemoryRegionSection *section)
249bc7b0cacSCindy Lu {
250bc7b0cacSCindy Lu     struct vhost_vdpa *v = container_of(listener, struct vhost_vdpa, listener);
251bc7b0cacSCindy Lu 
252bc7b0cacSCindy Lu     struct vdpa_iommu *iommu;
253bc7b0cacSCindy Lu     Int128 end;
254bc7b0cacSCindy Lu     int iommu_idx;
255bc7b0cacSCindy Lu     IOMMUMemoryRegion *iommu_mr;
256bc7b0cacSCindy Lu     int ret;
257bc7b0cacSCindy Lu 
258bc7b0cacSCindy Lu     iommu_mr = IOMMU_MEMORY_REGION(section->mr);
259bc7b0cacSCindy Lu 
260bc7b0cacSCindy Lu     iommu = g_malloc0(sizeof(*iommu));
261bc7b0cacSCindy Lu     end = int128_add(int128_make64(section->offset_within_region),
262bc7b0cacSCindy Lu                      section->size);
263bc7b0cacSCindy Lu     end = int128_sub(end, int128_one());
264bc7b0cacSCindy Lu     iommu_idx = memory_region_iommu_attrs_to_index(iommu_mr,
265bc7b0cacSCindy Lu                                                    MEMTXATTRS_UNSPECIFIED);
266bc7b0cacSCindy Lu     iommu->iommu_mr = iommu_mr;
267bc7b0cacSCindy Lu     iommu_notifier_init(&iommu->n, vhost_vdpa_iommu_map_notify,
268bc7b0cacSCindy Lu                         IOMMU_NOTIFIER_IOTLB_EVENTS,
269bc7b0cacSCindy Lu                         section->offset_within_region,
270bc7b0cacSCindy Lu                         int128_get64(end),
271bc7b0cacSCindy Lu                         iommu_idx);
272bc7b0cacSCindy Lu     iommu->iommu_offset = section->offset_within_address_space -
273bc7b0cacSCindy Lu                           section->offset_within_region;
274bc7b0cacSCindy Lu     iommu->dev = v;
275bc7b0cacSCindy Lu 
276bc7b0cacSCindy Lu     ret = memory_region_register_iommu_notifier(section->mr, &iommu->n, NULL);
277bc7b0cacSCindy Lu     if (ret) {
278bc7b0cacSCindy Lu         g_free(iommu);
279bc7b0cacSCindy Lu         return;
280bc7b0cacSCindy Lu     }
281bc7b0cacSCindy Lu 
282bc7b0cacSCindy Lu     QLIST_INSERT_HEAD(&v->iommu_list, iommu, iommu_next);
283bc7b0cacSCindy Lu     memory_region_iommu_replay(iommu->iommu_mr, &iommu->n);
284bc7b0cacSCindy Lu 
285bc7b0cacSCindy Lu     return;
286bc7b0cacSCindy Lu }
287bc7b0cacSCindy Lu 
288bc7b0cacSCindy Lu static void vhost_vdpa_iommu_region_del(MemoryListener *listener,
289bc7b0cacSCindy Lu                                         MemoryRegionSection *section)
290bc7b0cacSCindy Lu {
291bc7b0cacSCindy Lu     struct vhost_vdpa *v = container_of(listener, struct vhost_vdpa, listener);
292bc7b0cacSCindy Lu 
293bc7b0cacSCindy Lu     struct vdpa_iommu *iommu;
294bc7b0cacSCindy Lu 
295bc7b0cacSCindy Lu     QLIST_FOREACH(iommu, &v->iommu_list, iommu_next)
296bc7b0cacSCindy Lu     {
297bc7b0cacSCindy Lu         if (MEMORY_REGION(iommu->iommu_mr) == section->mr &&
298bc7b0cacSCindy Lu             iommu->n.start == section->offset_within_region) {
299bc7b0cacSCindy Lu             memory_region_unregister_iommu_notifier(section->mr, &iommu->n);
300bc7b0cacSCindy Lu             QLIST_REMOVE(iommu, iommu_next);
301bc7b0cacSCindy Lu             g_free(iommu);
302bc7b0cacSCindy Lu             break;
303bc7b0cacSCindy Lu         }
304bc7b0cacSCindy Lu     }
305bc7b0cacSCindy Lu }
306bc7b0cacSCindy Lu 
307108a6481SCindy Lu static void vhost_vdpa_listener_region_add(MemoryListener *listener,
308108a6481SCindy Lu                                            MemoryRegionSection *section)
309108a6481SCindy Lu {
3107dab70beSEugenio Pérez     DMAMap mem_region = {};
311108a6481SCindy Lu     struct vhost_vdpa *v = container_of(listener, struct vhost_vdpa, listener);
312108a6481SCindy Lu     hwaddr iova;
313108a6481SCindy Lu     Int128 llend, llsize;
314108a6481SCindy Lu     void *vaddr;
315108a6481SCindy Lu     int ret;
31633f21860SPhilippe Mathieu-Daudé     int page_size = qemu_target_page_size();
31733f21860SPhilippe Mathieu-Daudé     int page_mask = -page_size;
318108a6481SCindy Lu 
319013108b6SEugenio Pérez     if (vhost_vdpa_listener_skipped_section(section, v->iova_range.first,
32033f21860SPhilippe Mathieu-Daudé                                             v->iova_range.last, page_mask)) {
321108a6481SCindy Lu         return;
322108a6481SCindy Lu     }
323bc7b0cacSCindy Lu     if (memory_region_is_iommu(section->mr)) {
324bc7b0cacSCindy Lu         vhost_vdpa_iommu_region_add(listener, section);
325bc7b0cacSCindy Lu         return;
326bc7b0cacSCindy Lu     }
327108a6481SCindy Lu 
32833f21860SPhilippe Mathieu-Daudé     if (unlikely((section->offset_within_address_space & ~page_mask) !=
32933f21860SPhilippe Mathieu-Daudé                  (section->offset_within_region & ~page_mask))) {
33077812aa7SLaurent Vivier         trace_vhost_vdpa_listener_region_add_unaligned(v, section->mr->name,
33133f21860SPhilippe Mathieu-Daudé                        section->offset_within_address_space & ~page_mask,
33233f21860SPhilippe Mathieu-Daudé                        section->offset_within_region & ~page_mask);
333108a6481SCindy Lu         return;
334108a6481SCindy Lu     }
335108a6481SCindy Lu 
33633f21860SPhilippe Mathieu-Daudé     iova = ROUND_UP(section->offset_within_address_space, page_size);
33733f21860SPhilippe Mathieu-Daudé     llend = vhost_vdpa_section_end(section, page_mask);
338108a6481SCindy Lu     if (int128_ge(int128_make64(iova), llend)) {
339108a6481SCindy Lu         return;
340108a6481SCindy Lu     }
341108a6481SCindy Lu 
342108a6481SCindy Lu     memory_region_ref(section->mr);
343108a6481SCindy Lu 
344108a6481SCindy Lu     /* Here we assume that memory_region_is_ram(section->mr)==true */
345108a6481SCindy Lu 
346108a6481SCindy Lu     vaddr = memory_region_get_ram_ptr(section->mr) +
347108a6481SCindy Lu             section->offset_within_region +
348108a6481SCindy Lu             (iova - section->offset_within_address_space);
349108a6481SCindy Lu 
350778e67deSLaurent Vivier     trace_vhost_vdpa_listener_region_add(v, iova, int128_get64(llend),
351778e67deSLaurent Vivier                                          vaddr, section->readonly);
352778e67deSLaurent Vivier 
353108a6481SCindy Lu     llsize = int128_sub(llend, int128_make64(iova));
3546188d78aSEugenio Pérez     if (v->shadow_data) {
3557dab70beSEugenio Pérez         int r;
35634e3c94eSEugenio Pérez 
3577dab70beSEugenio Pérez         mem_region.translated_addr = (hwaddr)(uintptr_t)vaddr,
3587dab70beSEugenio Pérez         mem_region.size = int128_get64(llsize) - 1,
3597dab70beSEugenio Pérez         mem_region.perm = IOMMU_ACCESS_FLAG(true, section->readonly),
3607dab70beSEugenio Pérez 
3617dab70beSEugenio Pérez         r = vhost_iova_tree_map_alloc(v->iova_tree, &mem_region);
36234e3c94eSEugenio Pérez         if (unlikely(r != IOVA_OK)) {
36334e3c94eSEugenio Pérez             error_report("Can't allocate a mapping (%d)", r);
36434e3c94eSEugenio Pérez             goto fail;
36534e3c94eSEugenio Pérez         }
36634e3c94eSEugenio Pérez 
36734e3c94eSEugenio Pérez         iova = mem_region.iova;
36834e3c94eSEugenio Pérez     }
369108a6481SCindy Lu 
370e6db5df7SEugenio Pérez     vhost_vdpa_iotlb_batch_begin_once(v);
371cd831ed5SEugenio Pérez     ret = vhost_vdpa_dma_map(v, VHOST_VDPA_GUEST_PA_ASID, iova,
372cd831ed5SEugenio Pérez                              int128_get64(llsize), vaddr, section->readonly);
373108a6481SCindy Lu     if (ret) {
374108a6481SCindy Lu         error_report("vhost vdpa map fail!");
3757dab70beSEugenio Pérez         goto fail_map;
376108a6481SCindy Lu     }
377108a6481SCindy Lu 
378108a6481SCindy Lu     return;
379108a6481SCindy Lu 
3807dab70beSEugenio Pérez fail_map:
3816188d78aSEugenio Pérez     if (v->shadow_data) {
38269292a8eSEugenio Pérez         vhost_iova_tree_remove(v->iova_tree, mem_region);
3837dab70beSEugenio Pérez     }
3847dab70beSEugenio Pérez 
385108a6481SCindy Lu fail:
386108a6481SCindy Lu     /*
387108a6481SCindy Lu      * On the initfn path, store the first error in the container so we
388108a6481SCindy Lu      * can gracefully fail.  Runtime, there's not much we can do other
389108a6481SCindy Lu      * than throw a hardware error.
390108a6481SCindy Lu      */
391108a6481SCindy Lu     error_report("vhost-vdpa: DMA mapping failed, unable to continue");
392108a6481SCindy Lu     return;
393108a6481SCindy Lu 
394108a6481SCindy Lu }
395108a6481SCindy Lu 
396108a6481SCindy Lu static void vhost_vdpa_listener_region_del(MemoryListener *listener,
397108a6481SCindy Lu                                            MemoryRegionSection *section)
398108a6481SCindy Lu {
399108a6481SCindy Lu     struct vhost_vdpa *v = container_of(listener, struct vhost_vdpa, listener);
400108a6481SCindy Lu     hwaddr iova;
401108a6481SCindy Lu     Int128 llend, llsize;
402108a6481SCindy Lu     int ret;
40333f21860SPhilippe Mathieu-Daudé     int page_size = qemu_target_page_size();
40433f21860SPhilippe Mathieu-Daudé     int page_mask = -page_size;
405108a6481SCindy Lu 
406013108b6SEugenio Pérez     if (vhost_vdpa_listener_skipped_section(section, v->iova_range.first,
40733f21860SPhilippe Mathieu-Daudé                                             v->iova_range.last, page_mask)) {
408108a6481SCindy Lu         return;
409108a6481SCindy Lu     }
410bc7b0cacSCindy Lu     if (memory_region_is_iommu(section->mr)) {
411bc7b0cacSCindy Lu         vhost_vdpa_iommu_region_del(listener, section);
412bc7b0cacSCindy Lu     }
413108a6481SCindy Lu 
41433f21860SPhilippe Mathieu-Daudé     if (unlikely((section->offset_within_address_space & ~page_mask) !=
41533f21860SPhilippe Mathieu-Daudé                  (section->offset_within_region & ~page_mask))) {
41677812aa7SLaurent Vivier         trace_vhost_vdpa_listener_region_del_unaligned(v, section->mr->name,
41733f21860SPhilippe Mathieu-Daudé                        section->offset_within_address_space & ~page_mask,
41833f21860SPhilippe Mathieu-Daudé                        section->offset_within_region & ~page_mask);
419108a6481SCindy Lu         return;
420108a6481SCindy Lu     }
421108a6481SCindy Lu 
42233f21860SPhilippe Mathieu-Daudé     iova = ROUND_UP(section->offset_within_address_space, page_size);
42333f21860SPhilippe Mathieu-Daudé     llend = vhost_vdpa_section_end(section, page_mask);
424108a6481SCindy Lu 
4253d1e4d34SCindy Lu     trace_vhost_vdpa_listener_region_del(v, iova,
4263d1e4d34SCindy Lu         int128_get64(int128_sub(llend, int128_one())));
427778e67deSLaurent Vivier 
428108a6481SCindy Lu     if (int128_ge(int128_make64(iova), llend)) {
429108a6481SCindy Lu         return;
430108a6481SCindy Lu     }
431108a6481SCindy Lu 
432108a6481SCindy Lu     llsize = int128_sub(llend, int128_make64(iova));
433108a6481SCindy Lu 
4346188d78aSEugenio Pérez     if (v->shadow_data) {
43534e3c94eSEugenio Pérez         const DMAMap *result;
43634e3c94eSEugenio Pérez         const void *vaddr = memory_region_get_ram_ptr(section->mr) +
43734e3c94eSEugenio Pérez             section->offset_within_region +
43834e3c94eSEugenio Pérez             (iova - section->offset_within_address_space);
43934e3c94eSEugenio Pérez         DMAMap mem_region = {
44034e3c94eSEugenio Pérez             .translated_addr = (hwaddr)(uintptr_t)vaddr,
44134e3c94eSEugenio Pérez             .size = int128_get64(llsize) - 1,
44234e3c94eSEugenio Pérez         };
44334e3c94eSEugenio Pérez 
44434e3c94eSEugenio Pérez         result = vhost_iova_tree_find_iova(v->iova_tree, &mem_region);
44510dab9f2SEugenio Pérez         if (!result) {
44610dab9f2SEugenio Pérez             /* The memory listener map wasn't mapped */
44710dab9f2SEugenio Pérez             return;
44810dab9f2SEugenio Pérez         }
44934e3c94eSEugenio Pérez         iova = result->iova;
45069292a8eSEugenio Pérez         vhost_iova_tree_remove(v->iova_tree, *result);
45134e3c94eSEugenio Pérez     }
452e6db5df7SEugenio Pérez     vhost_vdpa_iotlb_batch_begin_once(v);
4532fbef6aaSCindy Lu     /*
4542fbef6aaSCindy Lu      * The unmap ioctl doesn't accept a full 64-bit. need to check it
4552fbef6aaSCindy Lu      */
4562fbef6aaSCindy Lu     if (int128_eq(llsize, int128_2_64())) {
4572fbef6aaSCindy Lu         llsize = int128_rshift(llsize, 1);
458cd831ed5SEugenio Pérez         ret = vhost_vdpa_dma_unmap(v, VHOST_VDPA_GUEST_PA_ASID, iova,
459cd831ed5SEugenio Pérez                                    int128_get64(llsize));
4602fbef6aaSCindy Lu 
461108a6481SCindy Lu         if (ret) {
4622fbef6aaSCindy Lu             error_report("vhost_vdpa_dma_unmap(%p, 0x%" HWADDR_PRIx ", "
4632fbef6aaSCindy Lu                          "0x%" HWADDR_PRIx ") = %d (%m)",
4642fbef6aaSCindy Lu                          v, iova, int128_get64(llsize), ret);
4652fbef6aaSCindy Lu         }
4662fbef6aaSCindy Lu         iova += int128_get64(llsize);
4672fbef6aaSCindy Lu     }
4682fbef6aaSCindy Lu     ret = vhost_vdpa_dma_unmap(v, VHOST_VDPA_GUEST_PA_ASID, iova,
4692fbef6aaSCindy Lu                                int128_get64(llsize));
4702fbef6aaSCindy Lu 
4712fbef6aaSCindy Lu     if (ret) {
4722fbef6aaSCindy Lu         error_report("vhost_vdpa_dma_unmap(%p, 0x%" HWADDR_PRIx ", "
4732fbef6aaSCindy Lu                      "0x%" HWADDR_PRIx ") = %d (%m)",
4742fbef6aaSCindy Lu                      v, iova, int128_get64(llsize), ret);
475108a6481SCindy Lu     }
476108a6481SCindy Lu 
477108a6481SCindy Lu     memory_region_unref(section->mr);
478108a6481SCindy Lu }
479108a6481SCindy Lu /*
480ef4ff56cSStefano Garzarella  * IOTLB API is used by vhost-vdpa which requires incremental updating
481108a6481SCindy Lu  * of the mapping. So we can not use generic vhost memory listener which
482108a6481SCindy Lu  * depends on the addnop().
483108a6481SCindy Lu  */
484108a6481SCindy Lu static const MemoryListener vhost_vdpa_memory_listener = {
485142518bdSPeter Xu     .name = "vhost-vdpa",
486a5bd0580SJason Wang     .commit = vhost_vdpa_listener_commit,
487108a6481SCindy Lu     .region_add = vhost_vdpa_listener_region_add,
488108a6481SCindy Lu     .region_del = vhost_vdpa_listener_region_del,
489108a6481SCindy Lu };
490108a6481SCindy Lu 
491108a6481SCindy Lu static int vhost_vdpa_call(struct vhost_dev *dev, unsigned long int request,
492108a6481SCindy Lu                              void *arg)
493108a6481SCindy Lu {
494108a6481SCindy Lu     struct vhost_vdpa *v = dev->opaque;
495108a6481SCindy Lu     int fd = v->device_fd;
496f2a6e6c4SKevin Wolf     int ret;
497108a6481SCindy Lu 
498108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
499108a6481SCindy Lu 
500f2a6e6c4SKevin Wolf     ret = ioctl(fd, request, arg);
501f2a6e6c4SKevin Wolf     return ret < 0 ? -errno : ret;
502108a6481SCindy Lu }
503108a6481SCindy Lu 
5043631151bSRoman Kagan static int vhost_vdpa_add_status(struct vhost_dev *dev, uint8_t status)
505108a6481SCindy Lu {
506108a6481SCindy Lu     uint8_t s;
5073631151bSRoman Kagan     int ret;
508108a6481SCindy Lu 
509778e67deSLaurent Vivier     trace_vhost_vdpa_add_status(dev, status);
5103631151bSRoman Kagan     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_STATUS, &s);
5113631151bSRoman Kagan     if (ret < 0) {
5123631151bSRoman Kagan         return ret;
513108a6481SCindy Lu     }
514*bc865bfeSEugenio Pérez     if ((s & status) == status) {
515*bc865bfeSEugenio Pérez         /* Don't set bits already set */
516*bc865bfeSEugenio Pérez         return 0;
517*bc865bfeSEugenio Pérez     }
518108a6481SCindy Lu 
519108a6481SCindy Lu     s |= status;
520108a6481SCindy Lu 
5213631151bSRoman Kagan     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_STATUS, &s);
5223631151bSRoman Kagan     if (ret < 0) {
5233631151bSRoman Kagan         return ret;
5243631151bSRoman Kagan     }
5253631151bSRoman Kagan 
5263631151bSRoman Kagan     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_STATUS, &s);
5273631151bSRoman Kagan     if (ret < 0) {
5283631151bSRoman Kagan         return ret;
5293631151bSRoman Kagan     }
5303631151bSRoman Kagan 
5313631151bSRoman Kagan     if (!(s & status)) {
5323631151bSRoman Kagan         return -EIO;
5333631151bSRoman Kagan     }
5343631151bSRoman Kagan 
5353631151bSRoman Kagan     return 0;
536108a6481SCindy Lu }
537108a6481SCindy Lu 
538c672f348SLongpeng int vhost_vdpa_get_iova_range(int fd, struct vhost_vdpa_iova_range *iova_range)
539c672f348SLongpeng {
540c672f348SLongpeng     int ret = ioctl(fd, VHOST_VDPA_GET_IOVA_RANGE, iova_range);
541c672f348SLongpeng 
542c672f348SLongpeng     return ret < 0 ? -errno : 0;
543c672f348SLongpeng }
544c672f348SLongpeng 
545d71b0609SSi-Wei Liu /*
546d71b0609SSi-Wei Liu  * The use of this function is for requests that only need to be
547d71b0609SSi-Wei Liu  * applied once. Typically such request occurs at the beginning
548d71b0609SSi-Wei Liu  * of operation, and before setting up queues. It should not be
549d71b0609SSi-Wei Liu  * used for request that performs operation until all queues are
550d71b0609SSi-Wei Liu  * set, which would need to check dev->vq_index_end instead.
551d71b0609SSi-Wei Liu  */
552d71b0609SSi-Wei Liu static bool vhost_vdpa_first_dev(struct vhost_dev *dev)
5534d191cfdSJason Wang {
5544d191cfdSJason Wang     struct vhost_vdpa *v = dev->opaque;
5554d191cfdSJason Wang 
556d71b0609SSi-Wei Liu     return v->index == 0;
5574d191cfdSJason Wang }
5584d191cfdSJason Wang 
55912a195faSEugenio Pérez static int vhost_vdpa_get_dev_features(struct vhost_dev *dev,
56012a195faSEugenio Pérez                                        uint64_t *features)
56112a195faSEugenio Pérez {
56212a195faSEugenio Pérez     int ret;
56312a195faSEugenio Pérez 
56412a195faSEugenio Pérez     ret = vhost_vdpa_call(dev, VHOST_GET_FEATURES, features);
56512a195faSEugenio Pérez     trace_vhost_vdpa_get_features(dev, *features);
56612a195faSEugenio Pérez     return ret;
56712a195faSEugenio Pérez }
56812a195faSEugenio Pérez 
569258a0394SEugenio Pérez static void vhost_vdpa_init_svq(struct vhost_dev *hdev, struct vhost_vdpa *v)
570dff4426fSEugenio Pérez {
571dff4426fSEugenio Pérez     g_autoptr(GPtrArray) shadow_vqs = NULL;
5724725a418SEugenio Pérez 
573dff4426fSEugenio Pérez     shadow_vqs = g_ptr_array_new_full(hdev->nvqs, vhost_svq_free);
574dff4426fSEugenio Pérez     for (unsigned n = 0; n < hdev->nvqs; ++n) {
5753cfb4d06SEugenio Pérez         VhostShadowVirtqueue *svq;
576dff4426fSEugenio Pérez 
5775fde952bSEugenio Pérez         svq = vhost_svq_new(v->shadow_vq_ops, v->shadow_vq_ops_opaque);
5783cfb4d06SEugenio Pérez         g_ptr_array_add(shadow_vqs, svq);
579dff4426fSEugenio Pérez     }
580dff4426fSEugenio Pérez 
581dff4426fSEugenio Pérez     v->shadow_vqs = g_steal_pointer(&shadow_vqs);
582dff4426fSEugenio Pérez }
583dff4426fSEugenio Pérez 
58428770ff9SKevin Wolf static int vhost_vdpa_init(struct vhost_dev *dev, void *opaque, Error **errp)
585108a6481SCindy Lu {
586108a6481SCindy Lu     struct vhost_vdpa *v;
587108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
588778e67deSLaurent Vivier     trace_vhost_vdpa_init(dev, opaque);
589e1c1915bSDavid Hildenbrand     int ret;
590e1c1915bSDavid Hildenbrand 
591108a6481SCindy Lu     v = opaque;
592a5bd0580SJason Wang     v->dev = dev;
593108a6481SCindy Lu     dev->opaque =  opaque ;
594108a6481SCindy Lu     v->listener = vhost_vdpa_memory_listener;
595108a6481SCindy Lu     v->msg_type = VHOST_IOTLB_MSG_V2;
596258a0394SEugenio Pérez     vhost_vdpa_init_svq(dev, v);
597108a6481SCindy Lu 
5989c363cf6SEugenio Pérez     error_propagate(&dev->migration_blocker, v->migration_blocker);
599d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
6004d191cfdSJason Wang         return 0;
6014d191cfdSJason Wang     }
6024d191cfdSJason Wang 
603a230c471SEugenio Pérez     /*
60457ac8318SEugenio Pérez      * If dev->shadow_vqs_enabled at initialization that means the device has
60557ac8318SEugenio Pérez      * been started with x-svq=on, so don't block migration
60657ac8318SEugenio Pérez      */
60757ac8318SEugenio Pérez     if (dev->migration_blocker == NULL && !v->shadow_vqs_enabled) {
60857ac8318SEugenio Pérez         /* We don't have dev->features yet */
60957ac8318SEugenio Pérez         uint64_t features;
61057ac8318SEugenio Pérez         ret = vhost_vdpa_get_dev_features(dev, &features);
61157ac8318SEugenio Pérez         if (unlikely(ret)) {
61257ac8318SEugenio Pérez             error_setg_errno(errp, -ret, "Could not get device features");
61357ac8318SEugenio Pérez             return ret;
61457ac8318SEugenio Pérez         }
61557ac8318SEugenio Pérez         vhost_svq_valid_features(features, &dev->migration_blocker);
61657ac8318SEugenio Pérez     }
61757ac8318SEugenio Pérez 
61857ac8318SEugenio Pérez     /*
619a230c471SEugenio Pérez      * Similar to VFIO, we end up pinning all guest memory and have to
620a230c471SEugenio Pérez      * disable discarding of RAM.
621a230c471SEugenio Pérez      */
622a230c471SEugenio Pérez     ret = ram_block_discard_disable(true);
623a230c471SEugenio Pérez     if (ret) {
624a230c471SEugenio Pérez         error_report("Cannot set discarding of RAM broken");
625a230c471SEugenio Pérez         return ret;
626a230c471SEugenio Pérez     }
627a230c471SEugenio Pérez 
628108a6481SCindy Lu     vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_ACKNOWLEDGE |
629108a6481SCindy Lu                                VIRTIO_CONFIG_S_DRIVER);
630108a6481SCindy Lu 
631108a6481SCindy Lu     return 0;
632108a6481SCindy Lu }
633108a6481SCindy Lu 
634d0416d48SJason Wang static void vhost_vdpa_host_notifier_uninit(struct vhost_dev *dev,
635d0416d48SJason Wang                                             int queue_index)
636d0416d48SJason Wang {
6378e3b0cbbSMarc-André Lureau     size_t page_size = qemu_real_host_page_size();
638d0416d48SJason Wang     struct vhost_vdpa *v = dev->opaque;
639d0416d48SJason Wang     VirtIODevice *vdev = dev->vdev;
640d0416d48SJason Wang     VhostVDPAHostNotifier *n;
641d0416d48SJason Wang 
642d0416d48SJason Wang     n = &v->notifier[queue_index];
643d0416d48SJason Wang 
644d0416d48SJason Wang     if (n->addr) {
645d0416d48SJason Wang         virtio_queue_set_host_notifier_mr(vdev, queue_index, &n->mr, false);
646d0416d48SJason Wang         object_unparent(OBJECT(&n->mr));
647d0416d48SJason Wang         munmap(n->addr, page_size);
648d0416d48SJason Wang         n->addr = NULL;
649d0416d48SJason Wang     }
650d0416d48SJason Wang }
651d0416d48SJason Wang 
652d0416d48SJason Wang static int vhost_vdpa_host_notifier_init(struct vhost_dev *dev, int queue_index)
653d0416d48SJason Wang {
6548e3b0cbbSMarc-André Lureau     size_t page_size = qemu_real_host_page_size();
655d0416d48SJason Wang     struct vhost_vdpa *v = dev->opaque;
656d0416d48SJason Wang     VirtIODevice *vdev = dev->vdev;
657d0416d48SJason Wang     VhostVDPAHostNotifier *n;
658d0416d48SJason Wang     int fd = v->device_fd;
659d0416d48SJason Wang     void *addr;
660d0416d48SJason Wang     char *name;
661d0416d48SJason Wang 
662d0416d48SJason Wang     vhost_vdpa_host_notifier_uninit(dev, queue_index);
663d0416d48SJason Wang 
664d0416d48SJason Wang     n = &v->notifier[queue_index];
665d0416d48SJason Wang 
666d0416d48SJason Wang     addr = mmap(NULL, page_size, PROT_WRITE, MAP_SHARED, fd,
667d0416d48SJason Wang                 queue_index * page_size);
668d0416d48SJason Wang     if (addr == MAP_FAILED) {
669d0416d48SJason Wang         goto err;
670d0416d48SJason Wang     }
671d0416d48SJason Wang 
672d0416d48SJason Wang     name = g_strdup_printf("vhost-vdpa/host-notifier@%p mmaps[%d]",
673d0416d48SJason Wang                            v, queue_index);
674d0416d48SJason Wang     memory_region_init_ram_device_ptr(&n->mr, OBJECT(vdev), name,
675d0416d48SJason Wang                                       page_size, addr);
676d0416d48SJason Wang     g_free(name);
677d0416d48SJason Wang 
678d0416d48SJason Wang     if (virtio_queue_set_host_notifier_mr(vdev, queue_index, &n->mr, true)) {
67998f7607eSLaurent Vivier         object_unparent(OBJECT(&n->mr));
680d0416d48SJason Wang         munmap(addr, page_size);
681d0416d48SJason Wang         goto err;
682d0416d48SJason Wang     }
683d0416d48SJason Wang     n->addr = addr;
684d0416d48SJason Wang 
685d0416d48SJason Wang     return 0;
686d0416d48SJason Wang 
687d0416d48SJason Wang err:
688d0416d48SJason Wang     return -1;
689d0416d48SJason Wang }
690d0416d48SJason Wang 
691b1f030a0SLaurent Vivier static void vhost_vdpa_host_notifiers_uninit(struct vhost_dev *dev, int n)
692b1f030a0SLaurent Vivier {
693b1f030a0SLaurent Vivier     int i;
694b1f030a0SLaurent Vivier 
695e66f2311SLongpeng     /*
696e66f2311SLongpeng      * Pack all the changes to the memory regions in a single
697e66f2311SLongpeng      * transaction to avoid a few updating of the address space
698e66f2311SLongpeng      * topology.
699e66f2311SLongpeng      */
700e66f2311SLongpeng     memory_region_transaction_begin();
701e66f2311SLongpeng 
702b1f030a0SLaurent Vivier     for (i = dev->vq_index; i < dev->vq_index + n; i++) {
703b1f030a0SLaurent Vivier         vhost_vdpa_host_notifier_uninit(dev, i);
704b1f030a0SLaurent Vivier     }
705e66f2311SLongpeng 
706e66f2311SLongpeng     memory_region_transaction_commit();
707b1f030a0SLaurent Vivier }
708b1f030a0SLaurent Vivier 
709d0416d48SJason Wang static void vhost_vdpa_host_notifiers_init(struct vhost_dev *dev)
710d0416d48SJason Wang {
711dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
712d0416d48SJason Wang     int i;
713d0416d48SJason Wang 
714dff4426fSEugenio Pérez     if (v->shadow_vqs_enabled) {
715dff4426fSEugenio Pérez         /* FIXME SVQ is not compatible with host notifiers mr */
716dff4426fSEugenio Pérez         return;
717dff4426fSEugenio Pérez     }
718dff4426fSEugenio Pérez 
719e66f2311SLongpeng     /*
720e66f2311SLongpeng      * Pack all the changes to the memory regions in a single
721e66f2311SLongpeng      * transaction to avoid a few updating of the address space
722e66f2311SLongpeng      * topology.
723e66f2311SLongpeng      */
724e66f2311SLongpeng     memory_region_transaction_begin();
725e66f2311SLongpeng 
726d0416d48SJason Wang     for (i = dev->vq_index; i < dev->vq_index + dev->nvqs; i++) {
727d0416d48SJason Wang         if (vhost_vdpa_host_notifier_init(dev, i)) {
728b1f030a0SLaurent Vivier             vhost_vdpa_host_notifiers_uninit(dev, i - dev->vq_index);
729e66f2311SLongpeng             break;
730e66f2311SLongpeng         }
731e66f2311SLongpeng     }
732e66f2311SLongpeng 
733e66f2311SLongpeng     memory_region_transaction_commit();
734d0416d48SJason Wang }
735d0416d48SJason Wang 
736dff4426fSEugenio Pérez static void vhost_vdpa_svq_cleanup(struct vhost_dev *dev)
737dff4426fSEugenio Pérez {
738dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
739dff4426fSEugenio Pérez     size_t idx;
740dff4426fSEugenio Pérez 
741dff4426fSEugenio Pérez     for (idx = 0; idx < v->shadow_vqs->len; ++idx) {
742dff4426fSEugenio Pérez         vhost_svq_stop(g_ptr_array_index(v->shadow_vqs, idx));
743dff4426fSEugenio Pérez     }
744dff4426fSEugenio Pérez     g_ptr_array_free(v->shadow_vqs, true);
745dff4426fSEugenio Pérez }
746dff4426fSEugenio Pérez 
747108a6481SCindy Lu static int vhost_vdpa_cleanup(struct vhost_dev *dev)
748108a6481SCindy Lu {
749108a6481SCindy Lu     struct vhost_vdpa *v;
750108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
751108a6481SCindy Lu     v = dev->opaque;
752778e67deSLaurent Vivier     trace_vhost_vdpa_cleanup(dev, v);
753a230c471SEugenio Pérez     if (vhost_vdpa_first_dev(dev)) {
754a230c471SEugenio Pérez         ram_block_discard_disable(false);
755a230c471SEugenio Pérez     }
756a230c471SEugenio Pérez 
757d0416d48SJason Wang     vhost_vdpa_host_notifiers_uninit(dev, dev->nvqs);
758108a6481SCindy Lu     memory_listener_unregister(&v->listener);
759dff4426fSEugenio Pérez     vhost_vdpa_svq_cleanup(dev);
760108a6481SCindy Lu 
761108a6481SCindy Lu     dev->opaque = NULL;
762e1c1915bSDavid Hildenbrand 
763108a6481SCindy Lu     return 0;
764108a6481SCindy Lu }
765108a6481SCindy Lu 
766108a6481SCindy Lu static int vhost_vdpa_memslots_limit(struct vhost_dev *dev)
767108a6481SCindy Lu {
768778e67deSLaurent Vivier     trace_vhost_vdpa_memslots_limit(dev, INT_MAX);
769108a6481SCindy Lu     return INT_MAX;
770108a6481SCindy Lu }
771108a6481SCindy Lu 
772108a6481SCindy Lu static int vhost_vdpa_set_mem_table(struct vhost_dev *dev,
773108a6481SCindy Lu                                     struct vhost_memory *mem)
774108a6481SCindy Lu {
775d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
7764d191cfdSJason Wang         return 0;
7774d191cfdSJason Wang     }
7784d191cfdSJason Wang 
779778e67deSLaurent Vivier     trace_vhost_vdpa_set_mem_table(dev, mem->nregions, mem->padding);
780778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_SET_MEM_TABLE) &&
781778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_REGIONS)) {
782778e67deSLaurent Vivier         int i;
783778e67deSLaurent Vivier         for (i = 0; i < mem->nregions; i++) {
784778e67deSLaurent Vivier             trace_vhost_vdpa_dump_regions(dev, i,
785778e67deSLaurent Vivier                                           mem->regions[i].guest_phys_addr,
786778e67deSLaurent Vivier                                           mem->regions[i].memory_size,
787778e67deSLaurent Vivier                                           mem->regions[i].userspace_addr,
788778e67deSLaurent Vivier                                           mem->regions[i].flags_padding);
789778e67deSLaurent Vivier         }
790778e67deSLaurent Vivier     }
791108a6481SCindy Lu     if (mem->padding) {
7923631151bSRoman Kagan         return -EINVAL;
793108a6481SCindy Lu     }
794108a6481SCindy Lu 
795108a6481SCindy Lu     return 0;
796108a6481SCindy Lu }
797108a6481SCindy Lu 
798108a6481SCindy Lu static int vhost_vdpa_set_features(struct vhost_dev *dev,
799108a6481SCindy Lu                                    uint64_t features)
800108a6481SCindy Lu {
80112a195faSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
802108a6481SCindy Lu     int ret;
8034d191cfdSJason Wang 
804d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
8054d191cfdSJason Wang         return 0;
8064d191cfdSJason Wang     }
8074d191cfdSJason Wang 
80812a195faSEugenio Pérez     if (v->shadow_vqs_enabled) {
80912a195faSEugenio Pérez         if ((v->acked_features ^ features) == BIT_ULL(VHOST_F_LOG_ALL)) {
81012a195faSEugenio Pérez             /*
81112a195faSEugenio Pérez              * QEMU is just trying to enable or disable logging. SVQ handles
81212a195faSEugenio Pérez              * this sepparately, so no need to forward this.
81312a195faSEugenio Pérez              */
81412a195faSEugenio Pérez             v->acked_features = features;
81512a195faSEugenio Pérez             return 0;
81612a195faSEugenio Pérez         }
81712a195faSEugenio Pérez 
81812a195faSEugenio Pérez         v->acked_features = features;
81912a195faSEugenio Pérez 
82012a195faSEugenio Pérez         /* We must not ack _F_LOG if SVQ is enabled */
82112a195faSEugenio Pérez         features &= ~BIT_ULL(VHOST_F_LOG_ALL);
82212a195faSEugenio Pérez     }
82312a195faSEugenio Pérez 
824778e67deSLaurent Vivier     trace_vhost_vdpa_set_features(dev, features);
825108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_SET_FEATURES, &features);
826108a6481SCindy Lu     if (ret) {
827108a6481SCindy Lu         return ret;
828108a6481SCindy Lu     }
829108a6481SCindy Lu 
8303631151bSRoman Kagan     return vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_FEATURES_OK);
831108a6481SCindy Lu }
832108a6481SCindy Lu 
833a5bd0580SJason Wang static int vhost_vdpa_set_backend_cap(struct vhost_dev *dev)
834a5bd0580SJason Wang {
835a5bd0580SJason Wang     uint64_t features;
836a5bd0580SJason Wang     uint64_t f = 0x1ULL << VHOST_BACKEND_F_IOTLB_MSG_V2 |
837c1a10086SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_IOTLB_BATCH |
838d83b4945SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_IOTLB_ASID |
839d83b4945SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_SUSPEND;
840a5bd0580SJason Wang     int r;
841a5bd0580SJason Wang 
842a5bd0580SJason Wang     if (vhost_vdpa_call(dev, VHOST_GET_BACKEND_FEATURES, &features)) {
8432a83e97eSJason Wang         return -EFAULT;
844a5bd0580SJason Wang     }
845a5bd0580SJason Wang 
846a5bd0580SJason Wang     features &= f;
8474d191cfdSJason Wang 
848d71b0609SSi-Wei Liu     if (vhost_vdpa_first_dev(dev)) {
849a5bd0580SJason Wang         r = vhost_vdpa_call(dev, VHOST_SET_BACKEND_FEATURES, &features);
850a5bd0580SJason Wang         if (r) {
8512a83e97eSJason Wang             return -EFAULT;
852a5bd0580SJason Wang         }
8534d191cfdSJason Wang     }
854a5bd0580SJason Wang 
855a5bd0580SJason Wang     dev->backend_cap = features;
856a5bd0580SJason Wang 
857a5bd0580SJason Wang     return 0;
858a5bd0580SJason Wang }
859a5bd0580SJason Wang 
860c232b8f4SZenghui Yu static int vhost_vdpa_get_device_id(struct vhost_dev *dev,
861108a6481SCindy Lu                                     uint32_t *device_id)
862108a6481SCindy Lu {
863778e67deSLaurent Vivier     int ret;
864778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_DEVICE_ID, device_id);
865778e67deSLaurent Vivier     trace_vhost_vdpa_get_device_id(dev, *device_id);
866778e67deSLaurent Vivier     return ret;
867108a6481SCindy Lu }
868108a6481SCindy Lu 
869108a6481SCindy Lu static int vhost_vdpa_reset_device(struct vhost_dev *dev)
870108a6481SCindy Lu {
8710bb302a9SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
872778e67deSLaurent Vivier     int ret;
873108a6481SCindy Lu     uint8_t status = 0;
874108a6481SCindy Lu 
875778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_STATUS, &status);
8762b5de4d7SEugenio Pérez     trace_vhost_vdpa_reset_device(dev);
8770bb302a9SEugenio Pérez     v->suspended = false;
878778e67deSLaurent Vivier     return ret;
879108a6481SCindy Lu }
880108a6481SCindy Lu 
881108a6481SCindy Lu static int vhost_vdpa_get_vq_index(struct vhost_dev *dev, int idx)
882108a6481SCindy Lu {
883108a6481SCindy Lu     assert(idx >= dev->vq_index && idx < dev->vq_index + dev->nvqs);
884108a6481SCindy Lu 
885353244d8SJason Wang     trace_vhost_vdpa_get_vq_index(dev, idx, idx);
886353244d8SJason Wang     return idx;
887108a6481SCindy Lu }
888108a6481SCindy Lu 
889d7ce0841SEugenio Pérez int vhost_vdpa_set_vring_ready(struct vhost_vdpa *v, unsigned idx)
890108a6481SCindy Lu {
891d7ce0841SEugenio Pérez     struct vhost_dev *dev = v->dev;
892108a6481SCindy Lu     struct vhost_vring_state state = {
893d7ce0841SEugenio Pérez         .index = idx,
894108a6481SCindy Lu         .num = 1,
895108a6481SCindy Lu     };
896d7ce0841SEugenio Pérez     int r = vhost_vdpa_call(dev, VHOST_VDPA_SET_VRING_ENABLE, &state);
897d7ce0841SEugenio Pérez 
898d7ce0841SEugenio Pérez     trace_vhost_vdpa_set_vring_ready(dev, idx, r);
899d7ce0841SEugenio Pérez     return r;
900108a6481SCindy Lu }
901108a6481SCindy Lu 
902259f3accSCindy Lu static int vhost_vdpa_set_config_call(struct vhost_dev *dev,
903259f3accSCindy Lu                                        int fd)
904259f3accSCindy Lu {
905259f3accSCindy Lu     trace_vhost_vdpa_set_config_call(dev, fd);
906259f3accSCindy Lu     return vhost_vdpa_call(dev, VHOST_VDPA_SET_CONFIG_CALL, &fd);
907259f3accSCindy Lu }
908259f3accSCindy Lu 
909778e67deSLaurent Vivier static void vhost_vdpa_dump_config(struct vhost_dev *dev, const uint8_t *config,
910778e67deSLaurent Vivier                                    uint32_t config_len)
911778e67deSLaurent Vivier {
912778e67deSLaurent Vivier     int b, len;
913778e67deSLaurent Vivier     char line[QEMU_HEXDUMP_LINE_LEN];
914778e67deSLaurent Vivier 
915778e67deSLaurent Vivier     for (b = 0; b < config_len; b += 16) {
916778e67deSLaurent Vivier         len = config_len - b;
917778e67deSLaurent Vivier         qemu_hexdump_line(line, b, config, len, false);
918778e67deSLaurent Vivier         trace_vhost_vdpa_dump_config(dev, line);
919778e67deSLaurent Vivier     }
920778e67deSLaurent Vivier }
921778e67deSLaurent Vivier 
922108a6481SCindy Lu static int vhost_vdpa_set_config(struct vhost_dev *dev, const uint8_t *data,
923108a6481SCindy Lu                                    uint32_t offset, uint32_t size,
924108a6481SCindy Lu                                    uint32_t flags)
925108a6481SCindy Lu {
926108a6481SCindy Lu     struct vhost_vdpa_config *config;
927108a6481SCindy Lu     int ret;
928108a6481SCindy Lu     unsigned long config_size = offsetof(struct vhost_vdpa_config, buf);
929986d4f78SLi Qiang 
930778e67deSLaurent Vivier     trace_vhost_vdpa_set_config(dev, offset, size, flags);
931108a6481SCindy Lu     config = g_malloc(size + config_size);
932108a6481SCindy Lu     config->off = offset;
933108a6481SCindy Lu     config->len = size;
934108a6481SCindy Lu     memcpy(config->buf, data, size);
935778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_SET_CONFIG) &&
936778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_CONFIG)) {
937778e67deSLaurent Vivier         vhost_vdpa_dump_config(dev, data, size);
938778e67deSLaurent Vivier     }
939108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_CONFIG, config);
940108a6481SCindy Lu     g_free(config);
941108a6481SCindy Lu     return ret;
942108a6481SCindy Lu }
943108a6481SCindy Lu 
944108a6481SCindy Lu static int vhost_vdpa_get_config(struct vhost_dev *dev, uint8_t *config,
94550de5138SKevin Wolf                                    uint32_t config_len, Error **errp)
946108a6481SCindy Lu {
947108a6481SCindy Lu     struct vhost_vdpa_config *v_config;
948108a6481SCindy Lu     unsigned long config_size = offsetof(struct vhost_vdpa_config, buf);
949108a6481SCindy Lu     int ret;
950108a6481SCindy Lu 
951778e67deSLaurent Vivier     trace_vhost_vdpa_get_config(dev, config, config_len);
952108a6481SCindy Lu     v_config = g_malloc(config_len + config_size);
953108a6481SCindy Lu     v_config->len = config_len;
954108a6481SCindy Lu     v_config->off = 0;
955108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_CONFIG, v_config);
956108a6481SCindy Lu     memcpy(config, v_config->buf, config_len);
957108a6481SCindy Lu     g_free(v_config);
958778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_GET_CONFIG) &&
959778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_CONFIG)) {
960778e67deSLaurent Vivier         vhost_vdpa_dump_config(dev, config, config_len);
961778e67deSLaurent Vivier     }
962108a6481SCindy Lu     return ret;
963108a6481SCindy Lu  }
964108a6481SCindy Lu 
965d96be4c8SEugenio Pérez static int vhost_vdpa_set_dev_vring_base(struct vhost_dev *dev,
966d96be4c8SEugenio Pérez                                          struct vhost_vring_state *ring)
967d96be4c8SEugenio Pérez {
968d96be4c8SEugenio Pérez     trace_vhost_vdpa_set_vring_base(dev, ring->index, ring->num);
969d96be4c8SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_BASE, ring);
970d96be4c8SEugenio Pérez }
971d96be4c8SEugenio Pérez 
972dff4426fSEugenio Pérez static int vhost_vdpa_set_vring_dev_kick(struct vhost_dev *dev,
973dff4426fSEugenio Pérez                                          struct vhost_vring_file *file)
974dff4426fSEugenio Pérez {
975dff4426fSEugenio Pérez     trace_vhost_vdpa_set_vring_kick(dev, file->index, file->fd);
976dff4426fSEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_KICK, file);
977dff4426fSEugenio Pérez }
978dff4426fSEugenio Pérez 
979a8ac8858SEugenio Pérez static int vhost_vdpa_set_vring_dev_call(struct vhost_dev *dev,
980a8ac8858SEugenio Pérez                                          struct vhost_vring_file *file)
981a8ac8858SEugenio Pérez {
982a8ac8858SEugenio Pérez     trace_vhost_vdpa_set_vring_call(dev, file->index, file->fd);
983a8ac8858SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_CALL, file);
984a8ac8858SEugenio Pérez }
985a8ac8858SEugenio Pérez 
986d96be4c8SEugenio Pérez static int vhost_vdpa_set_vring_dev_addr(struct vhost_dev *dev,
987d96be4c8SEugenio Pérez                                          struct vhost_vring_addr *addr)
988d96be4c8SEugenio Pérez {
989d96be4c8SEugenio Pérez     trace_vhost_vdpa_set_vring_addr(dev, addr->index, addr->flags,
990d96be4c8SEugenio Pérez                                 addr->desc_user_addr, addr->used_user_addr,
991d96be4c8SEugenio Pérez                                 addr->avail_user_addr,
992d96be4c8SEugenio Pérez                                 addr->log_guest_addr);
993d96be4c8SEugenio Pérez 
994d96be4c8SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_ADDR, addr);
995d96be4c8SEugenio Pérez 
996d96be4c8SEugenio Pérez }
997d96be4c8SEugenio Pérez 
998dff4426fSEugenio Pérez /**
999dff4426fSEugenio Pérez  * Set the shadow virtqueue descriptors to the device
1000dff4426fSEugenio Pérez  *
1001dff4426fSEugenio Pérez  * @dev: The vhost device model
1002dff4426fSEugenio Pérez  * @svq: The shadow virtqueue
1003dff4426fSEugenio Pérez  * @idx: The index of the virtqueue in the vhost device
1004dff4426fSEugenio Pérez  * @errp: Error
1005a8ac8858SEugenio Pérez  *
1006a8ac8858SEugenio Pérez  * Note that this function does not rewind kick file descriptor if cannot set
1007a8ac8858SEugenio Pérez  * call one.
1008dff4426fSEugenio Pérez  */
1009100890f7SEugenio Pérez static int vhost_vdpa_svq_set_fds(struct vhost_dev *dev,
1010dff4426fSEugenio Pérez                                   VhostShadowVirtqueue *svq, unsigned idx,
1011dff4426fSEugenio Pérez                                   Error **errp)
1012dff4426fSEugenio Pérez {
1013dff4426fSEugenio Pérez     struct vhost_vring_file file = {
1014dff4426fSEugenio Pérez         .index = dev->vq_index + idx,
1015dff4426fSEugenio Pérez     };
1016dff4426fSEugenio Pérez     const EventNotifier *event_notifier = &svq->hdev_kick;
1017dff4426fSEugenio Pérez     int r;
1018dff4426fSEugenio Pérez 
10193cfb4d06SEugenio Pérez     r = event_notifier_init(&svq->hdev_kick, 0);
10203cfb4d06SEugenio Pérez     if (r != 0) {
10213cfb4d06SEugenio Pérez         error_setg_errno(errp, -r, "Couldn't create kick event notifier");
10223cfb4d06SEugenio Pérez         goto err_init_hdev_kick;
10233cfb4d06SEugenio Pérez     }
10243cfb4d06SEugenio Pérez 
10253cfb4d06SEugenio Pérez     r = event_notifier_init(&svq->hdev_call, 0);
10263cfb4d06SEugenio Pérez     if (r != 0) {
10273cfb4d06SEugenio Pérez         error_setg_errno(errp, -r, "Couldn't create call event notifier");
10283cfb4d06SEugenio Pérez         goto err_init_hdev_call;
10293cfb4d06SEugenio Pérez     }
10303cfb4d06SEugenio Pérez 
1031dff4426fSEugenio Pérez     file.fd = event_notifier_get_fd(event_notifier);
1032dff4426fSEugenio Pérez     r = vhost_vdpa_set_vring_dev_kick(dev, &file);
1033dff4426fSEugenio Pérez     if (unlikely(r != 0)) {
1034dff4426fSEugenio Pérez         error_setg_errno(errp, -r, "Can't set device kick fd");
10353cfb4d06SEugenio Pérez         goto err_init_set_dev_fd;
1036a8ac8858SEugenio Pérez     }
1037a8ac8858SEugenio Pérez 
1038a8ac8858SEugenio Pérez     event_notifier = &svq->hdev_call;
1039a8ac8858SEugenio Pérez     file.fd = event_notifier_get_fd(event_notifier);
1040a8ac8858SEugenio Pérez     r = vhost_vdpa_set_vring_dev_call(dev, &file);
1041a8ac8858SEugenio Pérez     if (unlikely(r != 0)) {
1042a8ac8858SEugenio Pérez         error_setg_errno(errp, -r, "Can't set device call fd");
10433cfb4d06SEugenio Pérez         goto err_init_set_dev_fd;
1044dff4426fSEugenio Pérez     }
1045dff4426fSEugenio Pérez 
10463cfb4d06SEugenio Pérez     return 0;
10473cfb4d06SEugenio Pérez 
10483cfb4d06SEugenio Pérez err_init_set_dev_fd:
10493cfb4d06SEugenio Pérez     event_notifier_set_handler(&svq->hdev_call, NULL);
10503cfb4d06SEugenio Pérez 
10513cfb4d06SEugenio Pérez err_init_hdev_call:
10523cfb4d06SEugenio Pérez     event_notifier_cleanup(&svq->hdev_kick);
10533cfb4d06SEugenio Pérez 
10543cfb4d06SEugenio Pérez err_init_hdev_kick:
1055100890f7SEugenio Pérez     return r;
1056100890f7SEugenio Pérez }
1057100890f7SEugenio Pérez 
1058100890f7SEugenio Pérez /**
1059100890f7SEugenio Pérez  * Unmap a SVQ area in the device
1060100890f7SEugenio Pérez  */
10618b6d6119SEugenio Pérez static void vhost_vdpa_svq_unmap_ring(struct vhost_vdpa *v, hwaddr addr)
1062100890f7SEugenio Pérez {
10638b6d6119SEugenio Pérez     const DMAMap needle = {
10648b6d6119SEugenio Pérez         .translated_addr = addr,
10658b6d6119SEugenio Pérez     };
10668b6d6119SEugenio Pérez     const DMAMap *result = vhost_iova_tree_find_iova(v->iova_tree, &needle);
106734e3c94eSEugenio Pérez     hwaddr size;
1068100890f7SEugenio Pérez     int r;
1069100890f7SEugenio Pérez 
107034e3c94eSEugenio Pérez     if (unlikely(!result)) {
107134e3c94eSEugenio Pérez         error_report("Unable to find SVQ address to unmap");
10725b590f51SEugenio Pérez         return;
107334e3c94eSEugenio Pérez     }
107434e3c94eSEugenio Pérez 
10758e3b0cbbSMarc-André Lureau     size = ROUND_UP(result->size, qemu_real_host_page_size());
1076cd831ed5SEugenio Pérez     r = vhost_vdpa_dma_unmap(v, v->address_space_id, result->iova, size);
1077b37c12beSEugenio Pérez     if (unlikely(r < 0)) {
1078b37c12beSEugenio Pérez         error_report("Unable to unmap SVQ vring: %s (%d)", g_strerror(-r), -r);
10795b590f51SEugenio Pérez         return;
1080b37c12beSEugenio Pérez     }
1081b37c12beSEugenio Pérez 
1082b37c12beSEugenio Pérez     vhost_iova_tree_remove(v->iova_tree, *result);
1083100890f7SEugenio Pérez }
1084100890f7SEugenio Pérez 
10855b590f51SEugenio Pérez static void vhost_vdpa_svq_unmap_rings(struct vhost_dev *dev,
1086100890f7SEugenio Pérez                                        const VhostShadowVirtqueue *svq)
1087100890f7SEugenio Pérez {
1088100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1089100890f7SEugenio Pérez     struct vhost_vring_addr svq_addr;
1090100890f7SEugenio Pérez 
1091100890f7SEugenio Pérez     vhost_svq_get_vring_addr(svq, &svq_addr);
1092100890f7SEugenio Pérez 
10938b6d6119SEugenio Pérez     vhost_vdpa_svq_unmap_ring(v, svq_addr.desc_user_addr);
1094100890f7SEugenio Pérez 
10958b6d6119SEugenio Pérez     vhost_vdpa_svq_unmap_ring(v, svq_addr.used_user_addr);
109634e3c94eSEugenio Pérez }
109734e3c94eSEugenio Pérez 
109834e3c94eSEugenio Pérez /**
109934e3c94eSEugenio Pérez  * Map the SVQ area in the device
110034e3c94eSEugenio Pérez  *
110134e3c94eSEugenio Pérez  * @v: Vhost-vdpa device
110234e3c94eSEugenio Pérez  * @needle: The area to search iova
110334e3c94eSEugenio Pérez  * @errorp: Error pointer
110434e3c94eSEugenio Pérez  */
110534e3c94eSEugenio Pérez static bool vhost_vdpa_svq_map_ring(struct vhost_vdpa *v, DMAMap *needle,
110634e3c94eSEugenio Pérez                                     Error **errp)
110734e3c94eSEugenio Pérez {
110834e3c94eSEugenio Pérez     int r;
110934e3c94eSEugenio Pérez 
111034e3c94eSEugenio Pérez     r = vhost_iova_tree_map_alloc(v->iova_tree, needle);
111134e3c94eSEugenio Pérez     if (unlikely(r != IOVA_OK)) {
111234e3c94eSEugenio Pérez         error_setg(errp, "Cannot allocate iova (%d)", r);
111334e3c94eSEugenio Pérez         return false;
111434e3c94eSEugenio Pérez     }
111534e3c94eSEugenio Pérez 
1116cd831ed5SEugenio Pérez     r = vhost_vdpa_dma_map(v, v->address_space_id, needle->iova,
1117cd831ed5SEugenio Pérez                            needle->size + 1,
111834e3c94eSEugenio Pérez                            (void *)(uintptr_t)needle->translated_addr,
111934e3c94eSEugenio Pérez                            needle->perm == IOMMU_RO);
112034e3c94eSEugenio Pérez     if (unlikely(r != 0)) {
112134e3c94eSEugenio Pérez         error_setg_errno(errp, -r, "Cannot map region to device");
112269292a8eSEugenio Pérez         vhost_iova_tree_remove(v->iova_tree, *needle);
112334e3c94eSEugenio Pérez     }
112434e3c94eSEugenio Pérez 
112534e3c94eSEugenio Pérez     return r == 0;
1126100890f7SEugenio Pérez }
1127100890f7SEugenio Pérez 
1128100890f7SEugenio Pérez /**
1129100890f7SEugenio Pérez  * Map the shadow virtqueue rings in the device
1130100890f7SEugenio Pérez  *
1131100890f7SEugenio Pérez  * @dev: The vhost device
1132100890f7SEugenio Pérez  * @svq: The shadow virtqueue
1133100890f7SEugenio Pérez  * @addr: Assigned IOVA addresses
1134100890f7SEugenio Pérez  * @errp: Error pointer
1135100890f7SEugenio Pérez  */
1136100890f7SEugenio Pérez static bool vhost_vdpa_svq_map_rings(struct vhost_dev *dev,
1137100890f7SEugenio Pérez                                      const VhostShadowVirtqueue *svq,
1138100890f7SEugenio Pérez                                      struct vhost_vring_addr *addr,
1139100890f7SEugenio Pérez                                      Error **errp)
1140100890f7SEugenio Pérez {
114105e385d2SMarkus Armbruster     ERRP_GUARD();
114234e3c94eSEugenio Pérez     DMAMap device_region, driver_region;
114334e3c94eSEugenio Pérez     struct vhost_vring_addr svq_addr;
1144100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1145100890f7SEugenio Pérez     size_t device_size = vhost_svq_device_area_size(svq);
1146100890f7SEugenio Pérez     size_t driver_size = vhost_svq_driver_area_size(svq);
114734e3c94eSEugenio Pérez     size_t avail_offset;
114834e3c94eSEugenio Pérez     bool ok;
1149100890f7SEugenio Pérez 
115034e3c94eSEugenio Pérez     vhost_svq_get_vring_addr(svq, &svq_addr);
1151100890f7SEugenio Pérez 
115234e3c94eSEugenio Pérez     driver_region = (DMAMap) {
115334e3c94eSEugenio Pérez         .translated_addr = svq_addr.desc_user_addr,
115434e3c94eSEugenio Pérez         .size = driver_size - 1,
115534e3c94eSEugenio Pérez         .perm = IOMMU_RO,
115634e3c94eSEugenio Pérez     };
115734e3c94eSEugenio Pérez     ok = vhost_vdpa_svq_map_ring(v, &driver_region, errp);
115834e3c94eSEugenio Pérez     if (unlikely(!ok)) {
115934e3c94eSEugenio Pérez         error_prepend(errp, "Cannot create vq driver region: ");
1160100890f7SEugenio Pérez         return false;
1161100890f7SEugenio Pérez     }
116234e3c94eSEugenio Pérez     addr->desc_user_addr = driver_region.iova;
116334e3c94eSEugenio Pérez     avail_offset = svq_addr.avail_user_addr - svq_addr.desc_user_addr;
116434e3c94eSEugenio Pérez     addr->avail_user_addr = driver_region.iova + avail_offset;
1165100890f7SEugenio Pérez 
116634e3c94eSEugenio Pérez     device_region = (DMAMap) {
116734e3c94eSEugenio Pérez         .translated_addr = svq_addr.used_user_addr,
116834e3c94eSEugenio Pérez         .size = device_size - 1,
116934e3c94eSEugenio Pérez         .perm = IOMMU_RW,
117034e3c94eSEugenio Pérez     };
117134e3c94eSEugenio Pérez     ok = vhost_vdpa_svq_map_ring(v, &device_region, errp);
117234e3c94eSEugenio Pérez     if (unlikely(!ok)) {
117334e3c94eSEugenio Pérez         error_prepend(errp, "Cannot create vq device region: ");
11748b6d6119SEugenio Pérez         vhost_vdpa_svq_unmap_ring(v, driver_region.translated_addr);
1175100890f7SEugenio Pérez     }
117634e3c94eSEugenio Pérez     addr->used_user_addr = device_region.iova;
1177100890f7SEugenio Pérez 
117834e3c94eSEugenio Pérez     return ok;
1179100890f7SEugenio Pérez }
1180100890f7SEugenio Pérez 
1181100890f7SEugenio Pérez static bool vhost_vdpa_svq_setup(struct vhost_dev *dev,
1182100890f7SEugenio Pérez                                  VhostShadowVirtqueue *svq, unsigned idx,
1183100890f7SEugenio Pérez                                  Error **errp)
1184100890f7SEugenio Pérez {
1185100890f7SEugenio Pérez     uint16_t vq_index = dev->vq_index + idx;
1186100890f7SEugenio Pérez     struct vhost_vring_state s = {
1187100890f7SEugenio Pérez         .index = vq_index,
1188100890f7SEugenio Pérez     };
1189100890f7SEugenio Pérez     int r;
1190100890f7SEugenio Pérez 
1191100890f7SEugenio Pérez     r = vhost_vdpa_set_dev_vring_base(dev, &s);
1192100890f7SEugenio Pérez     if (unlikely(r)) {
1193100890f7SEugenio Pérez         error_setg_errno(errp, -r, "Cannot set vring base");
1194100890f7SEugenio Pérez         return false;
1195100890f7SEugenio Pérez     }
1196100890f7SEugenio Pérez 
1197100890f7SEugenio Pérez     r = vhost_vdpa_svq_set_fds(dev, svq, idx, errp);
1198dff4426fSEugenio Pérez     return r == 0;
1199dff4426fSEugenio Pérez }
1200dff4426fSEugenio Pérez 
1201dff4426fSEugenio Pérez static bool vhost_vdpa_svqs_start(struct vhost_dev *dev)
1202dff4426fSEugenio Pérez {
1203dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1204dff4426fSEugenio Pérez     Error *err = NULL;
1205dff4426fSEugenio Pérez     unsigned i;
1206dff4426fSEugenio Pérez 
1207712c1a31SEugenio Pérez     if (!v->shadow_vqs_enabled) {
1208dff4426fSEugenio Pérez         return true;
1209dff4426fSEugenio Pérez     }
1210dff4426fSEugenio Pérez 
1211dff4426fSEugenio Pérez     for (i = 0; i < v->shadow_vqs->len; ++i) {
1212100890f7SEugenio Pérez         VirtQueue *vq = virtio_get_queue(dev->vdev, dev->vq_index + i);
1213dff4426fSEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, i);
1214100890f7SEugenio Pérez         struct vhost_vring_addr addr = {
12151c82fdfeSEugenio Pérez             .index = dev->vq_index + i,
1216100890f7SEugenio Pérez         };
1217100890f7SEugenio Pérez         int r;
1218dff4426fSEugenio Pérez         bool ok = vhost_vdpa_svq_setup(dev, svq, i, &err);
1219dff4426fSEugenio Pérez         if (unlikely(!ok)) {
1220100890f7SEugenio Pérez             goto err;
1221100890f7SEugenio Pérez         }
1222100890f7SEugenio Pérez 
12235fde952bSEugenio Pérez         vhost_svq_start(svq, dev->vdev, vq, v->iova_tree);
1224100890f7SEugenio Pérez         ok = vhost_vdpa_svq_map_rings(dev, svq, &addr, &err);
1225100890f7SEugenio Pérez         if (unlikely(!ok)) {
1226100890f7SEugenio Pérez             goto err_map;
1227100890f7SEugenio Pérez         }
1228100890f7SEugenio Pérez 
1229100890f7SEugenio Pérez         /* Override vring GPA set by vhost subsystem */
1230100890f7SEugenio Pérez         r = vhost_vdpa_set_vring_dev_addr(dev, &addr);
1231100890f7SEugenio Pérez         if (unlikely(r != 0)) {
1232100890f7SEugenio Pérez             error_setg_errno(&err, -r, "Cannot set device address");
1233100890f7SEugenio Pérez             goto err_set_addr;
1234100890f7SEugenio Pérez         }
1235100890f7SEugenio Pérez     }
1236100890f7SEugenio Pérez 
1237100890f7SEugenio Pérez     return true;
1238100890f7SEugenio Pérez 
1239100890f7SEugenio Pérez err_set_addr:
1240100890f7SEugenio Pérez     vhost_vdpa_svq_unmap_rings(dev, g_ptr_array_index(v->shadow_vqs, i));
1241100890f7SEugenio Pérez 
1242100890f7SEugenio Pérez err_map:
1243100890f7SEugenio Pérez     vhost_svq_stop(g_ptr_array_index(v->shadow_vqs, i));
1244100890f7SEugenio Pérez 
1245100890f7SEugenio Pérez err:
1246dff4426fSEugenio Pérez     error_reportf_err(err, "Cannot setup SVQ %u: ", i);
1247100890f7SEugenio Pérez     for (unsigned j = 0; j < i; ++j) {
1248100890f7SEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, j);
1249100890f7SEugenio Pérez         vhost_vdpa_svq_unmap_rings(dev, svq);
1250100890f7SEugenio Pérez         vhost_svq_stop(svq);
1251100890f7SEugenio Pérez     }
1252100890f7SEugenio Pérez 
1253100890f7SEugenio Pérez     return false;
1254100890f7SEugenio Pérez }
1255100890f7SEugenio Pérez 
12565b590f51SEugenio Pérez static void vhost_vdpa_svqs_stop(struct vhost_dev *dev)
1257100890f7SEugenio Pérez {
1258100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1259100890f7SEugenio Pérez 
1260712c1a31SEugenio Pérez     if (!v->shadow_vqs_enabled) {
12615b590f51SEugenio Pérez         return;
1262100890f7SEugenio Pérez     }
1263100890f7SEugenio Pérez 
1264100890f7SEugenio Pérez     for (unsigned i = 0; i < v->shadow_vqs->len; ++i) {
1265100890f7SEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, i);
12662e1a9de9SEugenio Pérez 
12672e1a9de9SEugenio Pérez         vhost_svq_stop(svq);
12685b590f51SEugenio Pérez         vhost_vdpa_svq_unmap_rings(dev, svq);
12693cfb4d06SEugenio Pérez 
12703cfb4d06SEugenio Pérez         event_notifier_cleanup(&svq->hdev_kick);
12713cfb4d06SEugenio Pérez         event_notifier_cleanup(&svq->hdev_call);
1272dff4426fSEugenio Pérez     }
1273dff4426fSEugenio Pérez }
1274dff4426fSEugenio Pérez 
12750bb302a9SEugenio Pérez static void vhost_vdpa_suspend(struct vhost_dev *dev)
12760bb302a9SEugenio Pérez {
12770bb302a9SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
12780bb302a9SEugenio Pérez     int r;
12790bb302a9SEugenio Pérez 
12800bb302a9SEugenio Pérez     if (!vhost_vdpa_first_dev(dev)) {
12810bb302a9SEugenio Pérez         return;
12820bb302a9SEugenio Pérez     }
12830bb302a9SEugenio Pérez 
12840bb302a9SEugenio Pérez     if (dev->backend_cap & BIT_ULL(VHOST_BACKEND_F_SUSPEND)) {
12850bb302a9SEugenio Pérez         trace_vhost_vdpa_suspend(dev);
12860bb302a9SEugenio Pérez         r = ioctl(v->device_fd, VHOST_VDPA_SUSPEND);
12870bb302a9SEugenio Pérez         if (unlikely(r)) {
12880bb302a9SEugenio Pérez             error_report("Cannot suspend: %s(%d)", g_strerror(errno), errno);
12890bb302a9SEugenio Pérez         } else {
12900bb302a9SEugenio Pérez             v->suspended = true;
12910bb302a9SEugenio Pérez             return;
12920bb302a9SEugenio Pérez         }
12930bb302a9SEugenio Pérez     }
12940bb302a9SEugenio Pérez 
12950bb302a9SEugenio Pérez     vhost_vdpa_reset_device(dev);
12960bb302a9SEugenio Pérez }
12970bb302a9SEugenio Pérez 
1298108a6481SCindy Lu static int vhost_vdpa_dev_start(struct vhost_dev *dev, bool started)
1299108a6481SCindy Lu {
1300108a6481SCindy Lu     struct vhost_vdpa *v = dev->opaque;
1301dff4426fSEugenio Pérez     bool ok;
1302778e67deSLaurent Vivier     trace_vhost_vdpa_dev_start(dev, started);
13034d191cfdSJason Wang 
13044d191cfdSJason Wang     if (started) {
13054d191cfdSJason Wang         vhost_vdpa_host_notifiers_init(dev);
1306dff4426fSEugenio Pérez         ok = vhost_vdpa_svqs_start(dev);
1307dff4426fSEugenio Pérez         if (unlikely(!ok)) {
1308dff4426fSEugenio Pérez             return -1;
1309dff4426fSEugenio Pérez         }
13104d191cfdSJason Wang     } else {
13110bb302a9SEugenio Pérez         vhost_vdpa_suspend(dev);
13125b590f51SEugenio Pérez         vhost_vdpa_svqs_stop(dev);
13134d191cfdSJason Wang         vhost_vdpa_host_notifiers_uninit(dev, dev->nvqs);
13144d191cfdSJason Wang     }
13154d191cfdSJason Wang 
1316245cf2c2SEugenio Pérez     if (dev->vq_index + dev->nvqs != dev->vq_index_end) {
13174d191cfdSJason Wang         return 0;
13184d191cfdSJason Wang     }
13194d191cfdSJason Wang 
1320108a6481SCindy Lu     if (started) {
1321bc7b0cacSCindy Lu         if (vhost_dev_has_iommu(dev) && (v->shadow_vqs_enabled)) {
1322bc7b0cacSCindy Lu             error_report("SVQ can not work while IOMMU enable, please disable"
1323bc7b0cacSCindy Lu                          "IOMMU and try again");
1324bc7b0cacSCindy Lu             return -1;
1325bc7b0cacSCindy Lu         }
1326bc7b0cacSCindy Lu         memory_listener_register(&v->listener, dev->vdev->dma_as);
1327bc7b0cacSCindy Lu 
13283631151bSRoman Kagan         return vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_DRIVER_OK);
1329c3716f26SEugenio Pérez     }
1330c3716f26SEugenio Pérez 
1331c3716f26SEugenio Pérez     return 0;
1332c3716f26SEugenio Pérez }
1333c3716f26SEugenio Pérez 
1334c3716f26SEugenio Pérez static void vhost_vdpa_reset_status(struct vhost_dev *dev)
1335c3716f26SEugenio Pérez {
1336c3716f26SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1337c3716f26SEugenio Pérez 
1338c3716f26SEugenio Pérez     if (dev->vq_index + dev->nvqs != dev->vq_index_end) {
1339c3716f26SEugenio Pérez         return;
1340c3716f26SEugenio Pérez     }
1341c3716f26SEugenio Pérez 
1342108a6481SCindy Lu     vhost_vdpa_reset_device(dev);
1343108a6481SCindy Lu     vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_ACKNOWLEDGE |
1344108a6481SCindy Lu                                VIRTIO_CONFIG_S_DRIVER);
1345108a6481SCindy Lu     memory_listener_unregister(&v->listener);
1346108a6481SCindy Lu }
1347108a6481SCindy Lu 
1348108a6481SCindy Lu static int vhost_vdpa_set_log_base(struct vhost_dev *dev, uint64_t base,
1349108a6481SCindy Lu                                      struct vhost_log *log)
1350108a6481SCindy Lu {
1351773ebc95SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1352d71b0609SSi-Wei Liu     if (v->shadow_vqs_enabled || !vhost_vdpa_first_dev(dev)) {
13534d191cfdSJason Wang         return 0;
13544d191cfdSJason Wang     }
13554d191cfdSJason Wang 
1356778e67deSLaurent Vivier     trace_vhost_vdpa_set_log_base(dev, base, log->size, log->refcnt, log->fd,
1357778e67deSLaurent Vivier                                   log->log);
1358108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_LOG_BASE, &base);
1359108a6481SCindy Lu }
1360108a6481SCindy Lu 
1361108a6481SCindy Lu static int vhost_vdpa_set_vring_addr(struct vhost_dev *dev,
1362108a6481SCindy Lu                                        struct vhost_vring_addr *addr)
1363108a6481SCindy Lu {
1364d96be4c8SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1365d96be4c8SEugenio Pérez 
1366d96be4c8SEugenio Pérez     if (v->shadow_vqs_enabled) {
1367d96be4c8SEugenio Pérez         /*
1368d96be4c8SEugenio Pérez          * Device vring addr was set at device start. SVQ base is handled by
1369d96be4c8SEugenio Pérez          * VirtQueue code.
1370d96be4c8SEugenio Pérez          */
1371d96be4c8SEugenio Pérez         return 0;
1372d96be4c8SEugenio Pérez     }
1373d96be4c8SEugenio Pérez 
1374d96be4c8SEugenio Pérez     return vhost_vdpa_set_vring_dev_addr(dev, addr);
1375108a6481SCindy Lu }
1376108a6481SCindy Lu 
1377108a6481SCindy Lu static int vhost_vdpa_set_vring_num(struct vhost_dev *dev,
1378108a6481SCindy Lu                                       struct vhost_vring_state *ring)
1379108a6481SCindy Lu {
1380778e67deSLaurent Vivier     trace_vhost_vdpa_set_vring_num(dev, ring->index, ring->num);
1381108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_VRING_NUM, ring);
1382108a6481SCindy Lu }
1383108a6481SCindy Lu 
1384108a6481SCindy Lu static int vhost_vdpa_set_vring_base(struct vhost_dev *dev,
1385108a6481SCindy Lu                                        struct vhost_vring_state *ring)
1386108a6481SCindy Lu {
1387d96be4c8SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1388d96be4c8SEugenio Pérez 
1389d96be4c8SEugenio Pérez     if (v->shadow_vqs_enabled) {
1390d96be4c8SEugenio Pérez         /*
1391d96be4c8SEugenio Pérez          * Device vring base was set at device start. SVQ base is handled by
1392d96be4c8SEugenio Pérez          * VirtQueue code.
1393d96be4c8SEugenio Pérez          */
1394d96be4c8SEugenio Pérez         return 0;
1395d96be4c8SEugenio Pérez     }
1396d96be4c8SEugenio Pérez 
1397d96be4c8SEugenio Pérez     return vhost_vdpa_set_dev_vring_base(dev, ring);
1398108a6481SCindy Lu }
1399108a6481SCindy Lu 
1400108a6481SCindy Lu static int vhost_vdpa_get_vring_base(struct vhost_dev *dev,
1401108a6481SCindy Lu                                        struct vhost_vring_state *ring)
1402108a6481SCindy Lu {
14036d0b2226SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1404778e67deSLaurent Vivier     int ret;
1405778e67deSLaurent Vivier 
14066d0b2226SEugenio Pérez     if (v->shadow_vqs_enabled) {
14072fdac348SEugenio Pérez         ring->num = virtio_queue_get_last_avail_idx(dev->vdev, ring->index);
14086d0b2226SEugenio Pérez         return 0;
14096d0b2226SEugenio Pérez     }
14106d0b2226SEugenio Pérez 
1411b6662cb7SEugenio Pérez     if (!v->suspended) {
1412b6662cb7SEugenio Pérez         /*
1413b6662cb7SEugenio Pérez          * Cannot trust in value returned by device, let vhost recover used
1414b6662cb7SEugenio Pérez          * idx from guest.
1415b6662cb7SEugenio Pérez          */
1416b6662cb7SEugenio Pérez         return -1;
1417b6662cb7SEugenio Pérez     }
1418b6662cb7SEugenio Pérez 
1419778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_GET_VRING_BASE, ring);
1420778e67deSLaurent Vivier     trace_vhost_vdpa_get_vring_base(dev, ring->index, ring->num);
1421778e67deSLaurent Vivier     return ret;
1422108a6481SCindy Lu }
1423108a6481SCindy Lu 
1424108a6481SCindy Lu static int vhost_vdpa_set_vring_kick(struct vhost_dev *dev,
1425108a6481SCindy Lu                                        struct vhost_vring_file *file)
1426108a6481SCindy Lu {
1427dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1428dff4426fSEugenio Pérez     int vdpa_idx = file->index - dev->vq_index;
1429dff4426fSEugenio Pérez 
1430dff4426fSEugenio Pérez     if (v->shadow_vqs_enabled) {
1431dff4426fSEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, vdpa_idx);
1432dff4426fSEugenio Pérez         vhost_svq_set_svq_kick_fd(svq, file->fd);
1433dff4426fSEugenio Pérez         return 0;
1434dff4426fSEugenio Pérez     } else {
1435dff4426fSEugenio Pérez         return vhost_vdpa_set_vring_dev_kick(dev, file);
1436dff4426fSEugenio Pérez     }
1437108a6481SCindy Lu }
1438108a6481SCindy Lu 
1439108a6481SCindy Lu static int vhost_vdpa_set_vring_call(struct vhost_dev *dev,
1440108a6481SCindy Lu                                        struct vhost_vring_file *file)
1441108a6481SCindy Lu {
1442a8ac8858SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1443a8ac8858SEugenio Pérez     int vdpa_idx = file->index - dev->vq_index;
1444a8ac8858SEugenio Pérez     VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, vdpa_idx);
1445a8ac8858SEugenio Pérez 
1446b2765243SEugenio Pérez     /* Remember last call fd because we can switch to SVQ anytime. */
1447a8ac8858SEugenio Pérez     vhost_svq_set_svq_call_fd(svq, file->fd);
1448b2765243SEugenio Pérez     if (v->shadow_vqs_enabled) {
1449a8ac8858SEugenio Pérez         return 0;
1450a8ac8858SEugenio Pérez     }
1451b2765243SEugenio Pérez 
1452b2765243SEugenio Pérez     return vhost_vdpa_set_vring_dev_call(dev, file);
1453108a6481SCindy Lu }
1454108a6481SCindy Lu 
1455108a6481SCindy Lu static int vhost_vdpa_get_features(struct vhost_dev *dev,
1456108a6481SCindy Lu                                      uint64_t *features)
1457108a6481SCindy Lu {
145812a195faSEugenio Pérez     int ret = vhost_vdpa_get_dev_features(dev, features);
1459778e67deSLaurent Vivier 
1460ab7337e3SEugenio Pérez     if (ret == 0) {
146112a195faSEugenio Pérez         /* Add SVQ logging capabilities */
146212a195faSEugenio Pérez         *features |= BIT_ULL(VHOST_F_LOG_ALL);
146312a195faSEugenio Pérez     }
146412a195faSEugenio Pérez 
1465778e67deSLaurent Vivier     return ret;
1466108a6481SCindy Lu }
1467108a6481SCindy Lu 
1468108a6481SCindy Lu static int vhost_vdpa_set_owner(struct vhost_dev *dev)
1469108a6481SCindy Lu {
1470d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
14714d191cfdSJason Wang         return 0;
14724d191cfdSJason Wang     }
14734d191cfdSJason Wang 
1474778e67deSLaurent Vivier     trace_vhost_vdpa_set_owner(dev);
1475108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_OWNER, NULL);
1476108a6481SCindy Lu }
1477108a6481SCindy Lu 
1478108a6481SCindy Lu static int vhost_vdpa_vq_get_addr(struct vhost_dev *dev,
1479108a6481SCindy Lu                     struct vhost_vring_addr *addr, struct vhost_virtqueue *vq)
1480108a6481SCindy Lu {
1481108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
1482108a6481SCindy Lu     addr->desc_user_addr = (uint64_t)(unsigned long)vq->desc_phys;
1483108a6481SCindy Lu     addr->avail_user_addr = (uint64_t)(unsigned long)vq->avail_phys;
1484108a6481SCindy Lu     addr->used_user_addr = (uint64_t)(unsigned long)vq->used_phys;
1485778e67deSLaurent Vivier     trace_vhost_vdpa_vq_get_addr(dev, vq, addr->desc_user_addr,
1486778e67deSLaurent Vivier                                  addr->avail_user_addr, addr->used_user_addr);
1487108a6481SCindy Lu     return 0;
1488108a6481SCindy Lu }
1489108a6481SCindy Lu 
1490108a6481SCindy Lu static bool  vhost_vdpa_force_iommu(struct vhost_dev *dev)
1491108a6481SCindy Lu {
1492108a6481SCindy Lu     return true;
1493108a6481SCindy Lu }
1494108a6481SCindy Lu 
1495108a6481SCindy Lu const VhostOps vdpa_ops = {
1496108a6481SCindy Lu         .backend_type = VHOST_BACKEND_TYPE_VDPA,
1497108a6481SCindy Lu         .vhost_backend_init = vhost_vdpa_init,
1498108a6481SCindy Lu         .vhost_backend_cleanup = vhost_vdpa_cleanup,
1499108a6481SCindy Lu         .vhost_set_log_base = vhost_vdpa_set_log_base,
1500108a6481SCindy Lu         .vhost_set_vring_addr = vhost_vdpa_set_vring_addr,
1501108a6481SCindy Lu         .vhost_set_vring_num = vhost_vdpa_set_vring_num,
1502108a6481SCindy Lu         .vhost_set_vring_base = vhost_vdpa_set_vring_base,
1503108a6481SCindy Lu         .vhost_get_vring_base = vhost_vdpa_get_vring_base,
1504108a6481SCindy Lu         .vhost_set_vring_kick = vhost_vdpa_set_vring_kick,
1505108a6481SCindy Lu         .vhost_set_vring_call = vhost_vdpa_set_vring_call,
1506108a6481SCindy Lu         .vhost_get_features = vhost_vdpa_get_features,
1507a5bd0580SJason Wang         .vhost_set_backend_cap = vhost_vdpa_set_backend_cap,
1508108a6481SCindy Lu         .vhost_set_owner = vhost_vdpa_set_owner,
1509108a6481SCindy Lu         .vhost_set_vring_endian = NULL,
1510108a6481SCindy Lu         .vhost_backend_memslots_limit = vhost_vdpa_memslots_limit,
1511108a6481SCindy Lu         .vhost_set_mem_table = vhost_vdpa_set_mem_table,
1512108a6481SCindy Lu         .vhost_set_features = vhost_vdpa_set_features,
1513108a6481SCindy Lu         .vhost_reset_device = vhost_vdpa_reset_device,
1514108a6481SCindy Lu         .vhost_get_vq_index = vhost_vdpa_get_vq_index,
1515108a6481SCindy Lu         .vhost_get_config  = vhost_vdpa_get_config,
1516108a6481SCindy Lu         .vhost_set_config = vhost_vdpa_set_config,
1517108a6481SCindy Lu         .vhost_requires_shm_log = NULL,
1518108a6481SCindy Lu         .vhost_migration_done = NULL,
1519108a6481SCindy Lu         .vhost_net_set_mtu = NULL,
1520108a6481SCindy Lu         .vhost_set_iotlb_callback = NULL,
1521108a6481SCindy Lu         .vhost_send_device_iotlb_msg = NULL,
1522108a6481SCindy Lu         .vhost_dev_start = vhost_vdpa_dev_start,
1523108a6481SCindy Lu         .vhost_get_device_id = vhost_vdpa_get_device_id,
1524108a6481SCindy Lu         .vhost_vq_get_addr = vhost_vdpa_vq_get_addr,
1525108a6481SCindy Lu         .vhost_force_iommu = vhost_vdpa_force_iommu,
1526259f3accSCindy Lu         .vhost_set_config_call = vhost_vdpa_set_config_call,
1527c3716f26SEugenio Pérez         .vhost_reset_status = vhost_vdpa_reset_status,
1528108a6481SCindy Lu };
1529