xref: /qemu/hw/virtio/vhost-vdpa.c (revision d7ce0841767d01c226fc0e22436ce22a0ec74226)
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     }
514108a6481SCindy Lu 
515108a6481SCindy Lu     s |= status;
516108a6481SCindy Lu 
5173631151bSRoman Kagan     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_STATUS, &s);
5183631151bSRoman Kagan     if (ret < 0) {
5193631151bSRoman Kagan         return ret;
5203631151bSRoman Kagan     }
5213631151bSRoman Kagan 
5223631151bSRoman Kagan     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_STATUS, &s);
5233631151bSRoman Kagan     if (ret < 0) {
5243631151bSRoman Kagan         return ret;
5253631151bSRoman Kagan     }
5263631151bSRoman Kagan 
5273631151bSRoman Kagan     if (!(s & status)) {
5283631151bSRoman Kagan         return -EIO;
5293631151bSRoman Kagan     }
5303631151bSRoman Kagan 
5313631151bSRoman Kagan     return 0;
532108a6481SCindy Lu }
533108a6481SCindy Lu 
534c672f348SLongpeng int vhost_vdpa_get_iova_range(int fd, struct vhost_vdpa_iova_range *iova_range)
535c672f348SLongpeng {
536c672f348SLongpeng     int ret = ioctl(fd, VHOST_VDPA_GET_IOVA_RANGE, iova_range);
537c672f348SLongpeng 
538c672f348SLongpeng     return ret < 0 ? -errno : 0;
539c672f348SLongpeng }
540c672f348SLongpeng 
541d71b0609SSi-Wei Liu /*
542d71b0609SSi-Wei Liu  * The use of this function is for requests that only need to be
543d71b0609SSi-Wei Liu  * applied once. Typically such request occurs at the beginning
544d71b0609SSi-Wei Liu  * of operation, and before setting up queues. It should not be
545d71b0609SSi-Wei Liu  * used for request that performs operation until all queues are
546d71b0609SSi-Wei Liu  * set, which would need to check dev->vq_index_end instead.
547d71b0609SSi-Wei Liu  */
548d71b0609SSi-Wei Liu static bool vhost_vdpa_first_dev(struct vhost_dev *dev)
5494d191cfdSJason Wang {
5504d191cfdSJason Wang     struct vhost_vdpa *v = dev->opaque;
5514d191cfdSJason Wang 
552d71b0609SSi-Wei Liu     return v->index == 0;
5534d191cfdSJason Wang }
5544d191cfdSJason Wang 
55512a195faSEugenio Pérez static int vhost_vdpa_get_dev_features(struct vhost_dev *dev,
55612a195faSEugenio Pérez                                        uint64_t *features)
55712a195faSEugenio Pérez {
55812a195faSEugenio Pérez     int ret;
55912a195faSEugenio Pérez 
56012a195faSEugenio Pérez     ret = vhost_vdpa_call(dev, VHOST_GET_FEATURES, features);
56112a195faSEugenio Pérez     trace_vhost_vdpa_get_features(dev, *features);
56212a195faSEugenio Pérez     return ret;
56312a195faSEugenio Pérez }
56412a195faSEugenio Pérez 
565258a0394SEugenio Pérez static void vhost_vdpa_init_svq(struct vhost_dev *hdev, struct vhost_vdpa *v)
566dff4426fSEugenio Pérez {
567dff4426fSEugenio Pérez     g_autoptr(GPtrArray) shadow_vqs = NULL;
5684725a418SEugenio Pérez 
569dff4426fSEugenio Pérez     shadow_vqs = g_ptr_array_new_full(hdev->nvqs, vhost_svq_free);
570dff4426fSEugenio Pérez     for (unsigned n = 0; n < hdev->nvqs; ++n) {
5713cfb4d06SEugenio Pérez         VhostShadowVirtqueue *svq;
572dff4426fSEugenio Pérez 
5735fde952bSEugenio Pérez         svq = vhost_svq_new(v->shadow_vq_ops, v->shadow_vq_ops_opaque);
5743cfb4d06SEugenio Pérez         g_ptr_array_add(shadow_vqs, svq);
575dff4426fSEugenio Pérez     }
576dff4426fSEugenio Pérez 
577dff4426fSEugenio Pérez     v->shadow_vqs = g_steal_pointer(&shadow_vqs);
578dff4426fSEugenio Pérez }
579dff4426fSEugenio Pérez 
58028770ff9SKevin Wolf static int vhost_vdpa_init(struct vhost_dev *dev, void *opaque, Error **errp)
581108a6481SCindy Lu {
582108a6481SCindy Lu     struct vhost_vdpa *v;
583108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
584778e67deSLaurent Vivier     trace_vhost_vdpa_init(dev, opaque);
585e1c1915bSDavid Hildenbrand     int ret;
586e1c1915bSDavid Hildenbrand 
587108a6481SCindy Lu     v = opaque;
588a5bd0580SJason Wang     v->dev = dev;
589108a6481SCindy Lu     dev->opaque =  opaque ;
590108a6481SCindy Lu     v->listener = vhost_vdpa_memory_listener;
591108a6481SCindy Lu     v->msg_type = VHOST_IOTLB_MSG_V2;
592258a0394SEugenio Pérez     vhost_vdpa_init_svq(dev, v);
593108a6481SCindy Lu 
5949c363cf6SEugenio Pérez     error_propagate(&dev->migration_blocker, v->migration_blocker);
595d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
5964d191cfdSJason Wang         return 0;
5974d191cfdSJason Wang     }
5984d191cfdSJason Wang 
599a230c471SEugenio Pérez     /*
60057ac8318SEugenio Pérez      * If dev->shadow_vqs_enabled at initialization that means the device has
60157ac8318SEugenio Pérez      * been started with x-svq=on, so don't block migration
60257ac8318SEugenio Pérez      */
60357ac8318SEugenio Pérez     if (dev->migration_blocker == NULL && !v->shadow_vqs_enabled) {
60457ac8318SEugenio Pérez         /* We don't have dev->features yet */
60557ac8318SEugenio Pérez         uint64_t features;
60657ac8318SEugenio Pérez         ret = vhost_vdpa_get_dev_features(dev, &features);
60757ac8318SEugenio Pérez         if (unlikely(ret)) {
60857ac8318SEugenio Pérez             error_setg_errno(errp, -ret, "Could not get device features");
60957ac8318SEugenio Pérez             return ret;
61057ac8318SEugenio Pérez         }
61157ac8318SEugenio Pérez         vhost_svq_valid_features(features, &dev->migration_blocker);
61257ac8318SEugenio Pérez     }
61357ac8318SEugenio Pérez 
61457ac8318SEugenio Pérez     /*
615a230c471SEugenio Pérez      * Similar to VFIO, we end up pinning all guest memory and have to
616a230c471SEugenio Pérez      * disable discarding of RAM.
617a230c471SEugenio Pérez      */
618a230c471SEugenio Pérez     ret = ram_block_discard_disable(true);
619a230c471SEugenio Pérez     if (ret) {
620a230c471SEugenio Pérez         error_report("Cannot set discarding of RAM broken");
621a230c471SEugenio Pérez         return ret;
622a230c471SEugenio Pérez     }
623a230c471SEugenio Pérez 
624108a6481SCindy Lu     vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_ACKNOWLEDGE |
625108a6481SCindy Lu                                VIRTIO_CONFIG_S_DRIVER);
626108a6481SCindy Lu 
627108a6481SCindy Lu     return 0;
628108a6481SCindy Lu }
629108a6481SCindy Lu 
630d0416d48SJason Wang static void vhost_vdpa_host_notifier_uninit(struct vhost_dev *dev,
631d0416d48SJason Wang                                             int queue_index)
632d0416d48SJason Wang {
6338e3b0cbbSMarc-André Lureau     size_t page_size = qemu_real_host_page_size();
634d0416d48SJason Wang     struct vhost_vdpa *v = dev->opaque;
635d0416d48SJason Wang     VirtIODevice *vdev = dev->vdev;
636d0416d48SJason Wang     VhostVDPAHostNotifier *n;
637d0416d48SJason Wang 
638d0416d48SJason Wang     n = &v->notifier[queue_index];
639d0416d48SJason Wang 
640d0416d48SJason Wang     if (n->addr) {
641d0416d48SJason Wang         virtio_queue_set_host_notifier_mr(vdev, queue_index, &n->mr, false);
642d0416d48SJason Wang         object_unparent(OBJECT(&n->mr));
643d0416d48SJason Wang         munmap(n->addr, page_size);
644d0416d48SJason Wang         n->addr = NULL;
645d0416d48SJason Wang     }
646d0416d48SJason Wang }
647d0416d48SJason Wang 
648d0416d48SJason Wang static int vhost_vdpa_host_notifier_init(struct vhost_dev *dev, int queue_index)
649d0416d48SJason Wang {
6508e3b0cbbSMarc-André Lureau     size_t page_size = qemu_real_host_page_size();
651d0416d48SJason Wang     struct vhost_vdpa *v = dev->opaque;
652d0416d48SJason Wang     VirtIODevice *vdev = dev->vdev;
653d0416d48SJason Wang     VhostVDPAHostNotifier *n;
654d0416d48SJason Wang     int fd = v->device_fd;
655d0416d48SJason Wang     void *addr;
656d0416d48SJason Wang     char *name;
657d0416d48SJason Wang 
658d0416d48SJason Wang     vhost_vdpa_host_notifier_uninit(dev, queue_index);
659d0416d48SJason Wang 
660d0416d48SJason Wang     n = &v->notifier[queue_index];
661d0416d48SJason Wang 
662d0416d48SJason Wang     addr = mmap(NULL, page_size, PROT_WRITE, MAP_SHARED, fd,
663d0416d48SJason Wang                 queue_index * page_size);
664d0416d48SJason Wang     if (addr == MAP_FAILED) {
665d0416d48SJason Wang         goto err;
666d0416d48SJason Wang     }
667d0416d48SJason Wang 
668d0416d48SJason Wang     name = g_strdup_printf("vhost-vdpa/host-notifier@%p mmaps[%d]",
669d0416d48SJason Wang                            v, queue_index);
670d0416d48SJason Wang     memory_region_init_ram_device_ptr(&n->mr, OBJECT(vdev), name,
671d0416d48SJason Wang                                       page_size, addr);
672d0416d48SJason Wang     g_free(name);
673d0416d48SJason Wang 
674d0416d48SJason Wang     if (virtio_queue_set_host_notifier_mr(vdev, queue_index, &n->mr, true)) {
67598f7607eSLaurent Vivier         object_unparent(OBJECT(&n->mr));
676d0416d48SJason Wang         munmap(addr, page_size);
677d0416d48SJason Wang         goto err;
678d0416d48SJason Wang     }
679d0416d48SJason Wang     n->addr = addr;
680d0416d48SJason Wang 
681d0416d48SJason Wang     return 0;
682d0416d48SJason Wang 
683d0416d48SJason Wang err:
684d0416d48SJason Wang     return -1;
685d0416d48SJason Wang }
686d0416d48SJason Wang 
687b1f030a0SLaurent Vivier static void vhost_vdpa_host_notifiers_uninit(struct vhost_dev *dev, int n)
688b1f030a0SLaurent Vivier {
689b1f030a0SLaurent Vivier     int i;
690b1f030a0SLaurent Vivier 
691e66f2311SLongpeng     /*
692e66f2311SLongpeng      * Pack all the changes to the memory regions in a single
693e66f2311SLongpeng      * transaction to avoid a few updating of the address space
694e66f2311SLongpeng      * topology.
695e66f2311SLongpeng      */
696e66f2311SLongpeng     memory_region_transaction_begin();
697e66f2311SLongpeng 
698b1f030a0SLaurent Vivier     for (i = dev->vq_index; i < dev->vq_index + n; i++) {
699b1f030a0SLaurent Vivier         vhost_vdpa_host_notifier_uninit(dev, i);
700b1f030a0SLaurent Vivier     }
701e66f2311SLongpeng 
702e66f2311SLongpeng     memory_region_transaction_commit();
703b1f030a0SLaurent Vivier }
704b1f030a0SLaurent Vivier 
705d0416d48SJason Wang static void vhost_vdpa_host_notifiers_init(struct vhost_dev *dev)
706d0416d48SJason Wang {
707dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
708d0416d48SJason Wang     int i;
709d0416d48SJason Wang 
710dff4426fSEugenio Pérez     if (v->shadow_vqs_enabled) {
711dff4426fSEugenio Pérez         /* FIXME SVQ is not compatible with host notifiers mr */
712dff4426fSEugenio Pérez         return;
713dff4426fSEugenio Pérez     }
714dff4426fSEugenio Pérez 
715e66f2311SLongpeng     /*
716e66f2311SLongpeng      * Pack all the changes to the memory regions in a single
717e66f2311SLongpeng      * transaction to avoid a few updating of the address space
718e66f2311SLongpeng      * topology.
719e66f2311SLongpeng      */
720e66f2311SLongpeng     memory_region_transaction_begin();
721e66f2311SLongpeng 
722d0416d48SJason Wang     for (i = dev->vq_index; i < dev->vq_index + dev->nvqs; i++) {
723d0416d48SJason Wang         if (vhost_vdpa_host_notifier_init(dev, i)) {
724b1f030a0SLaurent Vivier             vhost_vdpa_host_notifiers_uninit(dev, i - dev->vq_index);
725e66f2311SLongpeng             break;
726e66f2311SLongpeng         }
727e66f2311SLongpeng     }
728e66f2311SLongpeng 
729e66f2311SLongpeng     memory_region_transaction_commit();
730d0416d48SJason Wang }
731d0416d48SJason Wang 
732dff4426fSEugenio Pérez static void vhost_vdpa_svq_cleanup(struct vhost_dev *dev)
733dff4426fSEugenio Pérez {
734dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
735dff4426fSEugenio Pérez     size_t idx;
736dff4426fSEugenio Pérez 
737dff4426fSEugenio Pérez     for (idx = 0; idx < v->shadow_vqs->len; ++idx) {
738dff4426fSEugenio Pérez         vhost_svq_stop(g_ptr_array_index(v->shadow_vqs, idx));
739dff4426fSEugenio Pérez     }
740dff4426fSEugenio Pérez     g_ptr_array_free(v->shadow_vqs, true);
741dff4426fSEugenio Pérez }
742dff4426fSEugenio Pérez 
743108a6481SCindy Lu static int vhost_vdpa_cleanup(struct vhost_dev *dev)
744108a6481SCindy Lu {
745108a6481SCindy Lu     struct vhost_vdpa *v;
746108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
747108a6481SCindy Lu     v = dev->opaque;
748778e67deSLaurent Vivier     trace_vhost_vdpa_cleanup(dev, v);
749a230c471SEugenio Pérez     if (vhost_vdpa_first_dev(dev)) {
750a230c471SEugenio Pérez         ram_block_discard_disable(false);
751a230c471SEugenio Pérez     }
752a230c471SEugenio Pérez 
753d0416d48SJason Wang     vhost_vdpa_host_notifiers_uninit(dev, dev->nvqs);
754108a6481SCindy Lu     memory_listener_unregister(&v->listener);
755dff4426fSEugenio Pérez     vhost_vdpa_svq_cleanup(dev);
756108a6481SCindy Lu 
757108a6481SCindy Lu     dev->opaque = NULL;
758e1c1915bSDavid Hildenbrand 
759108a6481SCindy Lu     return 0;
760108a6481SCindy Lu }
761108a6481SCindy Lu 
762108a6481SCindy Lu static int vhost_vdpa_memslots_limit(struct vhost_dev *dev)
763108a6481SCindy Lu {
764778e67deSLaurent Vivier     trace_vhost_vdpa_memslots_limit(dev, INT_MAX);
765108a6481SCindy Lu     return INT_MAX;
766108a6481SCindy Lu }
767108a6481SCindy Lu 
768108a6481SCindy Lu static int vhost_vdpa_set_mem_table(struct vhost_dev *dev,
769108a6481SCindy Lu                                     struct vhost_memory *mem)
770108a6481SCindy Lu {
771d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
7724d191cfdSJason Wang         return 0;
7734d191cfdSJason Wang     }
7744d191cfdSJason Wang 
775778e67deSLaurent Vivier     trace_vhost_vdpa_set_mem_table(dev, mem->nregions, mem->padding);
776778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_SET_MEM_TABLE) &&
777778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_REGIONS)) {
778778e67deSLaurent Vivier         int i;
779778e67deSLaurent Vivier         for (i = 0; i < mem->nregions; i++) {
780778e67deSLaurent Vivier             trace_vhost_vdpa_dump_regions(dev, i,
781778e67deSLaurent Vivier                                           mem->regions[i].guest_phys_addr,
782778e67deSLaurent Vivier                                           mem->regions[i].memory_size,
783778e67deSLaurent Vivier                                           mem->regions[i].userspace_addr,
784778e67deSLaurent Vivier                                           mem->regions[i].flags_padding);
785778e67deSLaurent Vivier         }
786778e67deSLaurent Vivier     }
787108a6481SCindy Lu     if (mem->padding) {
7883631151bSRoman Kagan         return -EINVAL;
789108a6481SCindy Lu     }
790108a6481SCindy Lu 
791108a6481SCindy Lu     return 0;
792108a6481SCindy Lu }
793108a6481SCindy Lu 
794108a6481SCindy Lu static int vhost_vdpa_set_features(struct vhost_dev *dev,
795108a6481SCindy Lu                                    uint64_t features)
796108a6481SCindy Lu {
79712a195faSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
798108a6481SCindy Lu     int ret;
7994d191cfdSJason Wang 
800d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
8014d191cfdSJason Wang         return 0;
8024d191cfdSJason Wang     }
8034d191cfdSJason Wang 
80412a195faSEugenio Pérez     if (v->shadow_vqs_enabled) {
80512a195faSEugenio Pérez         if ((v->acked_features ^ features) == BIT_ULL(VHOST_F_LOG_ALL)) {
80612a195faSEugenio Pérez             /*
80712a195faSEugenio Pérez              * QEMU is just trying to enable or disable logging. SVQ handles
80812a195faSEugenio Pérez              * this sepparately, so no need to forward this.
80912a195faSEugenio Pérez              */
81012a195faSEugenio Pérez             v->acked_features = features;
81112a195faSEugenio Pérez             return 0;
81212a195faSEugenio Pérez         }
81312a195faSEugenio Pérez 
81412a195faSEugenio Pérez         v->acked_features = features;
81512a195faSEugenio Pérez 
81612a195faSEugenio Pérez         /* We must not ack _F_LOG if SVQ is enabled */
81712a195faSEugenio Pérez         features &= ~BIT_ULL(VHOST_F_LOG_ALL);
81812a195faSEugenio Pérez     }
81912a195faSEugenio Pérez 
820778e67deSLaurent Vivier     trace_vhost_vdpa_set_features(dev, features);
821108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_SET_FEATURES, &features);
822108a6481SCindy Lu     if (ret) {
823108a6481SCindy Lu         return ret;
824108a6481SCindy Lu     }
825108a6481SCindy Lu 
8263631151bSRoman Kagan     return vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_FEATURES_OK);
827108a6481SCindy Lu }
828108a6481SCindy Lu 
829a5bd0580SJason Wang static int vhost_vdpa_set_backend_cap(struct vhost_dev *dev)
830a5bd0580SJason Wang {
831a5bd0580SJason Wang     uint64_t features;
832a5bd0580SJason Wang     uint64_t f = 0x1ULL << VHOST_BACKEND_F_IOTLB_MSG_V2 |
833c1a10086SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_IOTLB_BATCH |
834d83b4945SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_IOTLB_ASID |
835d83b4945SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_SUSPEND;
836a5bd0580SJason Wang     int r;
837a5bd0580SJason Wang 
838a5bd0580SJason Wang     if (vhost_vdpa_call(dev, VHOST_GET_BACKEND_FEATURES, &features)) {
8392a83e97eSJason Wang         return -EFAULT;
840a5bd0580SJason Wang     }
841a5bd0580SJason Wang 
842a5bd0580SJason Wang     features &= f;
8434d191cfdSJason Wang 
844d71b0609SSi-Wei Liu     if (vhost_vdpa_first_dev(dev)) {
845a5bd0580SJason Wang         r = vhost_vdpa_call(dev, VHOST_SET_BACKEND_FEATURES, &features);
846a5bd0580SJason Wang         if (r) {
8472a83e97eSJason Wang             return -EFAULT;
848a5bd0580SJason Wang         }
8494d191cfdSJason Wang     }
850a5bd0580SJason Wang 
851a5bd0580SJason Wang     dev->backend_cap = features;
852a5bd0580SJason Wang 
853a5bd0580SJason Wang     return 0;
854a5bd0580SJason Wang }
855a5bd0580SJason Wang 
856c232b8f4SZenghui Yu static int vhost_vdpa_get_device_id(struct vhost_dev *dev,
857108a6481SCindy Lu                                     uint32_t *device_id)
858108a6481SCindy Lu {
859778e67deSLaurent Vivier     int ret;
860778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_DEVICE_ID, device_id);
861778e67deSLaurent Vivier     trace_vhost_vdpa_get_device_id(dev, *device_id);
862778e67deSLaurent Vivier     return ret;
863108a6481SCindy Lu }
864108a6481SCindy Lu 
865108a6481SCindy Lu static int vhost_vdpa_reset_device(struct vhost_dev *dev)
866108a6481SCindy Lu {
8670bb302a9SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
868778e67deSLaurent Vivier     int ret;
869108a6481SCindy Lu     uint8_t status = 0;
870108a6481SCindy Lu 
871778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_STATUS, &status);
8722b5de4d7SEugenio Pérez     trace_vhost_vdpa_reset_device(dev);
8730bb302a9SEugenio Pérez     v->suspended = false;
874778e67deSLaurent Vivier     return ret;
875108a6481SCindy Lu }
876108a6481SCindy Lu 
877108a6481SCindy Lu static int vhost_vdpa_get_vq_index(struct vhost_dev *dev, int idx)
878108a6481SCindy Lu {
879108a6481SCindy Lu     assert(idx >= dev->vq_index && idx < dev->vq_index + dev->nvqs);
880108a6481SCindy Lu 
881353244d8SJason Wang     trace_vhost_vdpa_get_vq_index(dev, idx, idx);
882353244d8SJason Wang     return idx;
883108a6481SCindy Lu }
884108a6481SCindy Lu 
885*d7ce0841SEugenio Pérez int vhost_vdpa_set_vring_ready(struct vhost_vdpa *v, unsigned idx)
886108a6481SCindy Lu {
887*d7ce0841SEugenio Pérez     struct vhost_dev *dev = v->dev;
888108a6481SCindy Lu     struct vhost_vring_state state = {
889*d7ce0841SEugenio Pérez         .index = idx,
890108a6481SCindy Lu         .num = 1,
891108a6481SCindy Lu     };
892*d7ce0841SEugenio Pérez     int r = vhost_vdpa_call(dev, VHOST_VDPA_SET_VRING_ENABLE, &state);
893*d7ce0841SEugenio Pérez 
894*d7ce0841SEugenio Pérez     trace_vhost_vdpa_set_vring_ready(dev, idx, r);
895*d7ce0841SEugenio Pérez     return r;
896108a6481SCindy Lu }
897108a6481SCindy Lu 
898259f3accSCindy Lu static int vhost_vdpa_set_config_call(struct vhost_dev *dev,
899259f3accSCindy Lu                                        int fd)
900259f3accSCindy Lu {
901259f3accSCindy Lu     trace_vhost_vdpa_set_config_call(dev, fd);
902259f3accSCindy Lu     return vhost_vdpa_call(dev, VHOST_VDPA_SET_CONFIG_CALL, &fd);
903259f3accSCindy Lu }
904259f3accSCindy Lu 
905778e67deSLaurent Vivier static void vhost_vdpa_dump_config(struct vhost_dev *dev, const uint8_t *config,
906778e67deSLaurent Vivier                                    uint32_t config_len)
907778e67deSLaurent Vivier {
908778e67deSLaurent Vivier     int b, len;
909778e67deSLaurent Vivier     char line[QEMU_HEXDUMP_LINE_LEN];
910778e67deSLaurent Vivier 
911778e67deSLaurent Vivier     for (b = 0; b < config_len; b += 16) {
912778e67deSLaurent Vivier         len = config_len - b;
913778e67deSLaurent Vivier         qemu_hexdump_line(line, b, config, len, false);
914778e67deSLaurent Vivier         trace_vhost_vdpa_dump_config(dev, line);
915778e67deSLaurent Vivier     }
916778e67deSLaurent Vivier }
917778e67deSLaurent Vivier 
918108a6481SCindy Lu static int vhost_vdpa_set_config(struct vhost_dev *dev, const uint8_t *data,
919108a6481SCindy Lu                                    uint32_t offset, uint32_t size,
920108a6481SCindy Lu                                    uint32_t flags)
921108a6481SCindy Lu {
922108a6481SCindy Lu     struct vhost_vdpa_config *config;
923108a6481SCindy Lu     int ret;
924108a6481SCindy Lu     unsigned long config_size = offsetof(struct vhost_vdpa_config, buf);
925986d4f78SLi Qiang 
926778e67deSLaurent Vivier     trace_vhost_vdpa_set_config(dev, offset, size, flags);
927108a6481SCindy Lu     config = g_malloc(size + config_size);
928108a6481SCindy Lu     config->off = offset;
929108a6481SCindy Lu     config->len = size;
930108a6481SCindy Lu     memcpy(config->buf, data, size);
931778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_SET_CONFIG) &&
932778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_CONFIG)) {
933778e67deSLaurent Vivier         vhost_vdpa_dump_config(dev, data, size);
934778e67deSLaurent Vivier     }
935108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_CONFIG, config);
936108a6481SCindy Lu     g_free(config);
937108a6481SCindy Lu     return ret;
938108a6481SCindy Lu }
939108a6481SCindy Lu 
940108a6481SCindy Lu static int vhost_vdpa_get_config(struct vhost_dev *dev, uint8_t *config,
94150de5138SKevin Wolf                                    uint32_t config_len, Error **errp)
942108a6481SCindy Lu {
943108a6481SCindy Lu     struct vhost_vdpa_config *v_config;
944108a6481SCindy Lu     unsigned long config_size = offsetof(struct vhost_vdpa_config, buf);
945108a6481SCindy Lu     int ret;
946108a6481SCindy Lu 
947778e67deSLaurent Vivier     trace_vhost_vdpa_get_config(dev, config, config_len);
948108a6481SCindy Lu     v_config = g_malloc(config_len + config_size);
949108a6481SCindy Lu     v_config->len = config_len;
950108a6481SCindy Lu     v_config->off = 0;
951108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_CONFIG, v_config);
952108a6481SCindy Lu     memcpy(config, v_config->buf, config_len);
953108a6481SCindy Lu     g_free(v_config);
954778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_GET_CONFIG) &&
955778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_CONFIG)) {
956778e67deSLaurent Vivier         vhost_vdpa_dump_config(dev, config, config_len);
957778e67deSLaurent Vivier     }
958108a6481SCindy Lu     return ret;
959108a6481SCindy Lu  }
960108a6481SCindy Lu 
961d96be4c8SEugenio Pérez static int vhost_vdpa_set_dev_vring_base(struct vhost_dev *dev,
962d96be4c8SEugenio Pérez                                          struct vhost_vring_state *ring)
963d96be4c8SEugenio Pérez {
964d96be4c8SEugenio Pérez     trace_vhost_vdpa_set_vring_base(dev, ring->index, ring->num);
965d96be4c8SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_BASE, ring);
966d96be4c8SEugenio Pérez }
967d96be4c8SEugenio Pérez 
968dff4426fSEugenio Pérez static int vhost_vdpa_set_vring_dev_kick(struct vhost_dev *dev,
969dff4426fSEugenio Pérez                                          struct vhost_vring_file *file)
970dff4426fSEugenio Pérez {
971dff4426fSEugenio Pérez     trace_vhost_vdpa_set_vring_kick(dev, file->index, file->fd);
972dff4426fSEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_KICK, file);
973dff4426fSEugenio Pérez }
974dff4426fSEugenio Pérez 
975a8ac8858SEugenio Pérez static int vhost_vdpa_set_vring_dev_call(struct vhost_dev *dev,
976a8ac8858SEugenio Pérez                                          struct vhost_vring_file *file)
977a8ac8858SEugenio Pérez {
978a8ac8858SEugenio Pérez     trace_vhost_vdpa_set_vring_call(dev, file->index, file->fd);
979a8ac8858SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_CALL, file);
980a8ac8858SEugenio Pérez }
981a8ac8858SEugenio Pérez 
982d96be4c8SEugenio Pérez static int vhost_vdpa_set_vring_dev_addr(struct vhost_dev *dev,
983d96be4c8SEugenio Pérez                                          struct vhost_vring_addr *addr)
984d96be4c8SEugenio Pérez {
985d96be4c8SEugenio Pérez     trace_vhost_vdpa_set_vring_addr(dev, addr->index, addr->flags,
986d96be4c8SEugenio Pérez                                 addr->desc_user_addr, addr->used_user_addr,
987d96be4c8SEugenio Pérez                                 addr->avail_user_addr,
988d96be4c8SEugenio Pérez                                 addr->log_guest_addr);
989d96be4c8SEugenio Pérez 
990d96be4c8SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_ADDR, addr);
991d96be4c8SEugenio Pérez 
992d96be4c8SEugenio Pérez }
993d96be4c8SEugenio Pérez 
994dff4426fSEugenio Pérez /**
995dff4426fSEugenio Pérez  * Set the shadow virtqueue descriptors to the device
996dff4426fSEugenio Pérez  *
997dff4426fSEugenio Pérez  * @dev: The vhost device model
998dff4426fSEugenio Pérez  * @svq: The shadow virtqueue
999dff4426fSEugenio Pérez  * @idx: The index of the virtqueue in the vhost device
1000dff4426fSEugenio Pérez  * @errp: Error
1001a8ac8858SEugenio Pérez  *
1002a8ac8858SEugenio Pérez  * Note that this function does not rewind kick file descriptor if cannot set
1003a8ac8858SEugenio Pérez  * call one.
1004dff4426fSEugenio Pérez  */
1005100890f7SEugenio Pérez static int vhost_vdpa_svq_set_fds(struct vhost_dev *dev,
1006dff4426fSEugenio Pérez                                   VhostShadowVirtqueue *svq, unsigned idx,
1007dff4426fSEugenio Pérez                                   Error **errp)
1008dff4426fSEugenio Pérez {
1009dff4426fSEugenio Pérez     struct vhost_vring_file file = {
1010dff4426fSEugenio Pérez         .index = dev->vq_index + idx,
1011dff4426fSEugenio Pérez     };
1012dff4426fSEugenio Pérez     const EventNotifier *event_notifier = &svq->hdev_kick;
1013dff4426fSEugenio Pérez     int r;
1014dff4426fSEugenio Pérez 
10153cfb4d06SEugenio Pérez     r = event_notifier_init(&svq->hdev_kick, 0);
10163cfb4d06SEugenio Pérez     if (r != 0) {
10173cfb4d06SEugenio Pérez         error_setg_errno(errp, -r, "Couldn't create kick event notifier");
10183cfb4d06SEugenio Pérez         goto err_init_hdev_kick;
10193cfb4d06SEugenio Pérez     }
10203cfb4d06SEugenio Pérez 
10213cfb4d06SEugenio Pérez     r = event_notifier_init(&svq->hdev_call, 0);
10223cfb4d06SEugenio Pérez     if (r != 0) {
10233cfb4d06SEugenio Pérez         error_setg_errno(errp, -r, "Couldn't create call event notifier");
10243cfb4d06SEugenio Pérez         goto err_init_hdev_call;
10253cfb4d06SEugenio Pérez     }
10263cfb4d06SEugenio Pérez 
1027dff4426fSEugenio Pérez     file.fd = event_notifier_get_fd(event_notifier);
1028dff4426fSEugenio Pérez     r = vhost_vdpa_set_vring_dev_kick(dev, &file);
1029dff4426fSEugenio Pérez     if (unlikely(r != 0)) {
1030dff4426fSEugenio Pérez         error_setg_errno(errp, -r, "Can't set device kick fd");
10313cfb4d06SEugenio Pérez         goto err_init_set_dev_fd;
1032a8ac8858SEugenio Pérez     }
1033a8ac8858SEugenio Pérez 
1034a8ac8858SEugenio Pérez     event_notifier = &svq->hdev_call;
1035a8ac8858SEugenio Pérez     file.fd = event_notifier_get_fd(event_notifier);
1036a8ac8858SEugenio Pérez     r = vhost_vdpa_set_vring_dev_call(dev, &file);
1037a8ac8858SEugenio Pérez     if (unlikely(r != 0)) {
1038a8ac8858SEugenio Pérez         error_setg_errno(errp, -r, "Can't set device call fd");
10393cfb4d06SEugenio Pérez         goto err_init_set_dev_fd;
1040dff4426fSEugenio Pérez     }
1041dff4426fSEugenio Pérez 
10423cfb4d06SEugenio Pérez     return 0;
10433cfb4d06SEugenio Pérez 
10443cfb4d06SEugenio Pérez err_init_set_dev_fd:
10453cfb4d06SEugenio Pérez     event_notifier_set_handler(&svq->hdev_call, NULL);
10463cfb4d06SEugenio Pérez 
10473cfb4d06SEugenio Pérez err_init_hdev_call:
10483cfb4d06SEugenio Pérez     event_notifier_cleanup(&svq->hdev_kick);
10493cfb4d06SEugenio Pérez 
10503cfb4d06SEugenio Pérez err_init_hdev_kick:
1051100890f7SEugenio Pérez     return r;
1052100890f7SEugenio Pérez }
1053100890f7SEugenio Pérez 
1054100890f7SEugenio Pérez /**
1055100890f7SEugenio Pérez  * Unmap a SVQ area in the device
1056100890f7SEugenio Pérez  */
10578b6d6119SEugenio Pérez static void vhost_vdpa_svq_unmap_ring(struct vhost_vdpa *v, hwaddr addr)
1058100890f7SEugenio Pérez {
10598b6d6119SEugenio Pérez     const DMAMap needle = {
10608b6d6119SEugenio Pérez         .translated_addr = addr,
10618b6d6119SEugenio Pérez     };
10628b6d6119SEugenio Pérez     const DMAMap *result = vhost_iova_tree_find_iova(v->iova_tree, &needle);
106334e3c94eSEugenio Pérez     hwaddr size;
1064100890f7SEugenio Pérez     int r;
1065100890f7SEugenio Pérez 
106634e3c94eSEugenio Pérez     if (unlikely(!result)) {
106734e3c94eSEugenio Pérez         error_report("Unable to find SVQ address to unmap");
10685b590f51SEugenio Pérez         return;
106934e3c94eSEugenio Pérez     }
107034e3c94eSEugenio Pérez 
10718e3b0cbbSMarc-André Lureau     size = ROUND_UP(result->size, qemu_real_host_page_size());
1072cd831ed5SEugenio Pérez     r = vhost_vdpa_dma_unmap(v, v->address_space_id, result->iova, size);
1073b37c12beSEugenio Pérez     if (unlikely(r < 0)) {
1074b37c12beSEugenio Pérez         error_report("Unable to unmap SVQ vring: %s (%d)", g_strerror(-r), -r);
10755b590f51SEugenio Pérez         return;
1076b37c12beSEugenio Pérez     }
1077b37c12beSEugenio Pérez 
1078b37c12beSEugenio Pérez     vhost_iova_tree_remove(v->iova_tree, *result);
1079100890f7SEugenio Pérez }
1080100890f7SEugenio Pérez 
10815b590f51SEugenio Pérez static void vhost_vdpa_svq_unmap_rings(struct vhost_dev *dev,
1082100890f7SEugenio Pérez                                        const VhostShadowVirtqueue *svq)
1083100890f7SEugenio Pérez {
1084100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1085100890f7SEugenio Pérez     struct vhost_vring_addr svq_addr;
1086100890f7SEugenio Pérez 
1087100890f7SEugenio Pérez     vhost_svq_get_vring_addr(svq, &svq_addr);
1088100890f7SEugenio Pérez 
10898b6d6119SEugenio Pérez     vhost_vdpa_svq_unmap_ring(v, svq_addr.desc_user_addr);
1090100890f7SEugenio Pérez 
10918b6d6119SEugenio Pérez     vhost_vdpa_svq_unmap_ring(v, svq_addr.used_user_addr);
109234e3c94eSEugenio Pérez }
109334e3c94eSEugenio Pérez 
109434e3c94eSEugenio Pérez /**
109534e3c94eSEugenio Pérez  * Map the SVQ area in the device
109634e3c94eSEugenio Pérez  *
109734e3c94eSEugenio Pérez  * @v: Vhost-vdpa device
109834e3c94eSEugenio Pérez  * @needle: The area to search iova
109934e3c94eSEugenio Pérez  * @errorp: Error pointer
110034e3c94eSEugenio Pérez  */
110134e3c94eSEugenio Pérez static bool vhost_vdpa_svq_map_ring(struct vhost_vdpa *v, DMAMap *needle,
110234e3c94eSEugenio Pérez                                     Error **errp)
110334e3c94eSEugenio Pérez {
110434e3c94eSEugenio Pérez     int r;
110534e3c94eSEugenio Pérez 
110634e3c94eSEugenio Pérez     r = vhost_iova_tree_map_alloc(v->iova_tree, needle);
110734e3c94eSEugenio Pérez     if (unlikely(r != IOVA_OK)) {
110834e3c94eSEugenio Pérez         error_setg(errp, "Cannot allocate iova (%d)", r);
110934e3c94eSEugenio Pérez         return false;
111034e3c94eSEugenio Pérez     }
111134e3c94eSEugenio Pérez 
1112cd831ed5SEugenio Pérez     r = vhost_vdpa_dma_map(v, v->address_space_id, needle->iova,
1113cd831ed5SEugenio Pérez                            needle->size + 1,
111434e3c94eSEugenio Pérez                            (void *)(uintptr_t)needle->translated_addr,
111534e3c94eSEugenio Pérez                            needle->perm == IOMMU_RO);
111634e3c94eSEugenio Pérez     if (unlikely(r != 0)) {
111734e3c94eSEugenio Pérez         error_setg_errno(errp, -r, "Cannot map region to device");
111869292a8eSEugenio Pérez         vhost_iova_tree_remove(v->iova_tree, *needle);
111934e3c94eSEugenio Pérez     }
112034e3c94eSEugenio Pérez 
112134e3c94eSEugenio Pérez     return r == 0;
1122100890f7SEugenio Pérez }
1123100890f7SEugenio Pérez 
1124100890f7SEugenio Pérez /**
1125100890f7SEugenio Pérez  * Map the shadow virtqueue rings in the device
1126100890f7SEugenio Pérez  *
1127100890f7SEugenio Pérez  * @dev: The vhost device
1128100890f7SEugenio Pérez  * @svq: The shadow virtqueue
1129100890f7SEugenio Pérez  * @addr: Assigned IOVA addresses
1130100890f7SEugenio Pérez  * @errp: Error pointer
1131100890f7SEugenio Pérez  */
1132100890f7SEugenio Pérez static bool vhost_vdpa_svq_map_rings(struct vhost_dev *dev,
1133100890f7SEugenio Pérez                                      const VhostShadowVirtqueue *svq,
1134100890f7SEugenio Pérez                                      struct vhost_vring_addr *addr,
1135100890f7SEugenio Pérez                                      Error **errp)
1136100890f7SEugenio Pérez {
113705e385d2SMarkus Armbruster     ERRP_GUARD();
113834e3c94eSEugenio Pérez     DMAMap device_region, driver_region;
113934e3c94eSEugenio Pérez     struct vhost_vring_addr svq_addr;
1140100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1141100890f7SEugenio Pérez     size_t device_size = vhost_svq_device_area_size(svq);
1142100890f7SEugenio Pérez     size_t driver_size = vhost_svq_driver_area_size(svq);
114334e3c94eSEugenio Pérez     size_t avail_offset;
114434e3c94eSEugenio Pérez     bool ok;
1145100890f7SEugenio Pérez 
114634e3c94eSEugenio Pérez     vhost_svq_get_vring_addr(svq, &svq_addr);
1147100890f7SEugenio Pérez 
114834e3c94eSEugenio Pérez     driver_region = (DMAMap) {
114934e3c94eSEugenio Pérez         .translated_addr = svq_addr.desc_user_addr,
115034e3c94eSEugenio Pérez         .size = driver_size - 1,
115134e3c94eSEugenio Pérez         .perm = IOMMU_RO,
115234e3c94eSEugenio Pérez     };
115334e3c94eSEugenio Pérez     ok = vhost_vdpa_svq_map_ring(v, &driver_region, errp);
115434e3c94eSEugenio Pérez     if (unlikely(!ok)) {
115534e3c94eSEugenio Pérez         error_prepend(errp, "Cannot create vq driver region: ");
1156100890f7SEugenio Pérez         return false;
1157100890f7SEugenio Pérez     }
115834e3c94eSEugenio Pérez     addr->desc_user_addr = driver_region.iova;
115934e3c94eSEugenio Pérez     avail_offset = svq_addr.avail_user_addr - svq_addr.desc_user_addr;
116034e3c94eSEugenio Pérez     addr->avail_user_addr = driver_region.iova + avail_offset;
1161100890f7SEugenio Pérez 
116234e3c94eSEugenio Pérez     device_region = (DMAMap) {
116334e3c94eSEugenio Pérez         .translated_addr = svq_addr.used_user_addr,
116434e3c94eSEugenio Pérez         .size = device_size - 1,
116534e3c94eSEugenio Pérez         .perm = IOMMU_RW,
116634e3c94eSEugenio Pérez     };
116734e3c94eSEugenio Pérez     ok = vhost_vdpa_svq_map_ring(v, &device_region, errp);
116834e3c94eSEugenio Pérez     if (unlikely(!ok)) {
116934e3c94eSEugenio Pérez         error_prepend(errp, "Cannot create vq device region: ");
11708b6d6119SEugenio Pérez         vhost_vdpa_svq_unmap_ring(v, driver_region.translated_addr);
1171100890f7SEugenio Pérez     }
117234e3c94eSEugenio Pérez     addr->used_user_addr = device_region.iova;
1173100890f7SEugenio Pérez 
117434e3c94eSEugenio Pérez     return ok;
1175100890f7SEugenio Pérez }
1176100890f7SEugenio Pérez 
1177100890f7SEugenio Pérez static bool vhost_vdpa_svq_setup(struct vhost_dev *dev,
1178100890f7SEugenio Pérez                                  VhostShadowVirtqueue *svq, unsigned idx,
1179100890f7SEugenio Pérez                                  Error **errp)
1180100890f7SEugenio Pérez {
1181100890f7SEugenio Pérez     uint16_t vq_index = dev->vq_index + idx;
1182100890f7SEugenio Pérez     struct vhost_vring_state s = {
1183100890f7SEugenio Pérez         .index = vq_index,
1184100890f7SEugenio Pérez     };
1185100890f7SEugenio Pérez     int r;
1186100890f7SEugenio Pérez 
1187100890f7SEugenio Pérez     r = vhost_vdpa_set_dev_vring_base(dev, &s);
1188100890f7SEugenio Pérez     if (unlikely(r)) {
1189100890f7SEugenio Pérez         error_setg_errno(errp, -r, "Cannot set vring base");
1190100890f7SEugenio Pérez         return false;
1191100890f7SEugenio Pérez     }
1192100890f7SEugenio Pérez 
1193100890f7SEugenio Pérez     r = vhost_vdpa_svq_set_fds(dev, svq, idx, errp);
1194dff4426fSEugenio Pérez     return r == 0;
1195dff4426fSEugenio Pérez }
1196dff4426fSEugenio Pérez 
1197dff4426fSEugenio Pérez static bool vhost_vdpa_svqs_start(struct vhost_dev *dev)
1198dff4426fSEugenio Pérez {
1199dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1200dff4426fSEugenio Pérez     Error *err = NULL;
1201dff4426fSEugenio Pérez     unsigned i;
1202dff4426fSEugenio Pérez 
1203712c1a31SEugenio Pérez     if (!v->shadow_vqs_enabled) {
1204dff4426fSEugenio Pérez         return true;
1205dff4426fSEugenio Pérez     }
1206dff4426fSEugenio Pérez 
1207dff4426fSEugenio Pérez     for (i = 0; i < v->shadow_vqs->len; ++i) {
1208100890f7SEugenio Pérez         VirtQueue *vq = virtio_get_queue(dev->vdev, dev->vq_index + i);
1209dff4426fSEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, i);
1210100890f7SEugenio Pérez         struct vhost_vring_addr addr = {
12111c82fdfeSEugenio Pérez             .index = dev->vq_index + i,
1212100890f7SEugenio Pérez         };
1213100890f7SEugenio Pérez         int r;
1214dff4426fSEugenio Pérez         bool ok = vhost_vdpa_svq_setup(dev, svq, i, &err);
1215dff4426fSEugenio Pérez         if (unlikely(!ok)) {
1216100890f7SEugenio Pérez             goto err;
1217100890f7SEugenio Pérez         }
1218100890f7SEugenio Pérez 
12195fde952bSEugenio Pérez         vhost_svq_start(svq, dev->vdev, vq, v->iova_tree);
1220100890f7SEugenio Pérez         ok = vhost_vdpa_svq_map_rings(dev, svq, &addr, &err);
1221100890f7SEugenio Pérez         if (unlikely(!ok)) {
1222100890f7SEugenio Pérez             goto err_map;
1223100890f7SEugenio Pérez         }
1224100890f7SEugenio Pérez 
1225100890f7SEugenio Pérez         /* Override vring GPA set by vhost subsystem */
1226100890f7SEugenio Pérez         r = vhost_vdpa_set_vring_dev_addr(dev, &addr);
1227100890f7SEugenio Pérez         if (unlikely(r != 0)) {
1228100890f7SEugenio Pérez             error_setg_errno(&err, -r, "Cannot set device address");
1229100890f7SEugenio Pérez             goto err_set_addr;
1230100890f7SEugenio Pérez         }
1231100890f7SEugenio Pérez     }
1232100890f7SEugenio Pérez 
1233100890f7SEugenio Pérez     return true;
1234100890f7SEugenio Pérez 
1235100890f7SEugenio Pérez err_set_addr:
1236100890f7SEugenio Pérez     vhost_vdpa_svq_unmap_rings(dev, g_ptr_array_index(v->shadow_vqs, i));
1237100890f7SEugenio Pérez 
1238100890f7SEugenio Pérez err_map:
1239100890f7SEugenio Pérez     vhost_svq_stop(g_ptr_array_index(v->shadow_vqs, i));
1240100890f7SEugenio Pérez 
1241100890f7SEugenio Pérez err:
1242dff4426fSEugenio Pérez     error_reportf_err(err, "Cannot setup SVQ %u: ", i);
1243100890f7SEugenio Pérez     for (unsigned j = 0; j < i; ++j) {
1244100890f7SEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, j);
1245100890f7SEugenio Pérez         vhost_vdpa_svq_unmap_rings(dev, svq);
1246100890f7SEugenio Pérez         vhost_svq_stop(svq);
1247100890f7SEugenio Pérez     }
1248100890f7SEugenio Pérez 
1249100890f7SEugenio Pérez     return false;
1250100890f7SEugenio Pérez }
1251100890f7SEugenio Pérez 
12525b590f51SEugenio Pérez static void vhost_vdpa_svqs_stop(struct vhost_dev *dev)
1253100890f7SEugenio Pérez {
1254100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1255100890f7SEugenio Pérez 
1256712c1a31SEugenio Pérez     if (!v->shadow_vqs_enabled) {
12575b590f51SEugenio Pérez         return;
1258100890f7SEugenio Pérez     }
1259100890f7SEugenio Pérez 
1260100890f7SEugenio Pérez     for (unsigned i = 0; i < v->shadow_vqs->len; ++i) {
1261100890f7SEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, i);
12622e1a9de9SEugenio Pérez 
12632e1a9de9SEugenio Pérez         vhost_svq_stop(svq);
12645b590f51SEugenio Pérez         vhost_vdpa_svq_unmap_rings(dev, svq);
12653cfb4d06SEugenio Pérez 
12663cfb4d06SEugenio Pérez         event_notifier_cleanup(&svq->hdev_kick);
12673cfb4d06SEugenio Pérez         event_notifier_cleanup(&svq->hdev_call);
1268dff4426fSEugenio Pérez     }
1269dff4426fSEugenio Pérez }
1270dff4426fSEugenio Pérez 
12710bb302a9SEugenio Pérez static void vhost_vdpa_suspend(struct vhost_dev *dev)
12720bb302a9SEugenio Pérez {
12730bb302a9SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
12740bb302a9SEugenio Pérez     int r;
12750bb302a9SEugenio Pérez 
12760bb302a9SEugenio Pérez     if (!vhost_vdpa_first_dev(dev)) {
12770bb302a9SEugenio Pérez         return;
12780bb302a9SEugenio Pérez     }
12790bb302a9SEugenio Pérez 
12800bb302a9SEugenio Pérez     if (dev->backend_cap & BIT_ULL(VHOST_BACKEND_F_SUSPEND)) {
12810bb302a9SEugenio Pérez         trace_vhost_vdpa_suspend(dev);
12820bb302a9SEugenio Pérez         r = ioctl(v->device_fd, VHOST_VDPA_SUSPEND);
12830bb302a9SEugenio Pérez         if (unlikely(r)) {
12840bb302a9SEugenio Pérez             error_report("Cannot suspend: %s(%d)", g_strerror(errno), errno);
12850bb302a9SEugenio Pérez         } else {
12860bb302a9SEugenio Pérez             v->suspended = true;
12870bb302a9SEugenio Pérez             return;
12880bb302a9SEugenio Pérez         }
12890bb302a9SEugenio Pérez     }
12900bb302a9SEugenio Pérez 
12910bb302a9SEugenio Pérez     vhost_vdpa_reset_device(dev);
12920bb302a9SEugenio Pérez }
12930bb302a9SEugenio Pérez 
1294108a6481SCindy Lu static int vhost_vdpa_dev_start(struct vhost_dev *dev, bool started)
1295108a6481SCindy Lu {
1296108a6481SCindy Lu     struct vhost_vdpa *v = dev->opaque;
1297dff4426fSEugenio Pérez     bool ok;
1298778e67deSLaurent Vivier     trace_vhost_vdpa_dev_start(dev, started);
12994d191cfdSJason Wang 
13004d191cfdSJason Wang     if (started) {
13014d191cfdSJason Wang         vhost_vdpa_host_notifiers_init(dev);
1302dff4426fSEugenio Pérez         ok = vhost_vdpa_svqs_start(dev);
1303dff4426fSEugenio Pérez         if (unlikely(!ok)) {
1304dff4426fSEugenio Pérez             return -1;
1305dff4426fSEugenio Pérez         }
1306*d7ce0841SEugenio Pérez         for (int i = 0; i < dev->nvqs; ++i) {
1307*d7ce0841SEugenio Pérez             vhost_vdpa_set_vring_ready(v, dev->vq_index + i);
1308*d7ce0841SEugenio Pérez         }
13094d191cfdSJason Wang     } else {
13100bb302a9SEugenio Pérez         vhost_vdpa_suspend(dev);
13115b590f51SEugenio Pérez         vhost_vdpa_svqs_stop(dev);
13124d191cfdSJason Wang         vhost_vdpa_host_notifiers_uninit(dev, dev->nvqs);
13134d191cfdSJason Wang     }
13144d191cfdSJason Wang 
1315245cf2c2SEugenio Pérez     if (dev->vq_index + dev->nvqs != dev->vq_index_end) {
13164d191cfdSJason Wang         return 0;
13174d191cfdSJason Wang     }
13184d191cfdSJason Wang 
1319108a6481SCindy Lu     if (started) {
1320bc7b0cacSCindy Lu         if (vhost_dev_has_iommu(dev) && (v->shadow_vqs_enabled)) {
1321bc7b0cacSCindy Lu             error_report("SVQ can not work while IOMMU enable, please disable"
1322bc7b0cacSCindy Lu                          "IOMMU and try again");
1323bc7b0cacSCindy Lu             return -1;
1324bc7b0cacSCindy Lu         }
1325bc7b0cacSCindy Lu         memory_listener_register(&v->listener, dev->vdev->dma_as);
1326bc7b0cacSCindy Lu 
13273631151bSRoman Kagan         return vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_DRIVER_OK);
1328c3716f26SEugenio Pérez     }
1329c3716f26SEugenio Pérez 
1330c3716f26SEugenio Pérez     return 0;
1331c3716f26SEugenio Pérez }
1332c3716f26SEugenio Pérez 
1333c3716f26SEugenio Pérez static void vhost_vdpa_reset_status(struct vhost_dev *dev)
1334c3716f26SEugenio Pérez {
1335c3716f26SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1336c3716f26SEugenio Pérez 
1337c3716f26SEugenio Pérez     if (dev->vq_index + dev->nvqs != dev->vq_index_end) {
1338c3716f26SEugenio Pérez         return;
1339c3716f26SEugenio Pérez     }
1340c3716f26SEugenio Pérez 
1341108a6481SCindy Lu     vhost_vdpa_reset_device(dev);
1342108a6481SCindy Lu     vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_ACKNOWLEDGE |
1343108a6481SCindy Lu                                VIRTIO_CONFIG_S_DRIVER);
1344108a6481SCindy Lu     memory_listener_unregister(&v->listener);
1345108a6481SCindy Lu }
1346108a6481SCindy Lu 
1347108a6481SCindy Lu static int vhost_vdpa_set_log_base(struct vhost_dev *dev, uint64_t base,
1348108a6481SCindy Lu                                      struct vhost_log *log)
1349108a6481SCindy Lu {
1350773ebc95SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1351d71b0609SSi-Wei Liu     if (v->shadow_vqs_enabled || !vhost_vdpa_first_dev(dev)) {
13524d191cfdSJason Wang         return 0;
13534d191cfdSJason Wang     }
13544d191cfdSJason Wang 
1355778e67deSLaurent Vivier     trace_vhost_vdpa_set_log_base(dev, base, log->size, log->refcnt, log->fd,
1356778e67deSLaurent Vivier                                   log->log);
1357108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_LOG_BASE, &base);
1358108a6481SCindy Lu }
1359108a6481SCindy Lu 
1360108a6481SCindy Lu static int vhost_vdpa_set_vring_addr(struct vhost_dev *dev,
1361108a6481SCindy Lu                                        struct vhost_vring_addr *addr)
1362108a6481SCindy Lu {
1363d96be4c8SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1364d96be4c8SEugenio Pérez 
1365d96be4c8SEugenio Pérez     if (v->shadow_vqs_enabled) {
1366d96be4c8SEugenio Pérez         /*
1367d96be4c8SEugenio Pérez          * Device vring addr was set at device start. SVQ base is handled by
1368d96be4c8SEugenio Pérez          * VirtQueue code.
1369d96be4c8SEugenio Pérez          */
1370d96be4c8SEugenio Pérez         return 0;
1371d96be4c8SEugenio Pérez     }
1372d96be4c8SEugenio Pérez 
1373d96be4c8SEugenio Pérez     return vhost_vdpa_set_vring_dev_addr(dev, addr);
1374108a6481SCindy Lu }
1375108a6481SCindy Lu 
1376108a6481SCindy Lu static int vhost_vdpa_set_vring_num(struct vhost_dev *dev,
1377108a6481SCindy Lu                                       struct vhost_vring_state *ring)
1378108a6481SCindy Lu {
1379778e67deSLaurent Vivier     trace_vhost_vdpa_set_vring_num(dev, ring->index, ring->num);
1380108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_VRING_NUM, ring);
1381108a6481SCindy Lu }
1382108a6481SCindy Lu 
1383108a6481SCindy Lu static int vhost_vdpa_set_vring_base(struct vhost_dev *dev,
1384108a6481SCindy Lu                                        struct vhost_vring_state *ring)
1385108a6481SCindy Lu {
1386d96be4c8SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1387d96be4c8SEugenio Pérez 
1388d96be4c8SEugenio Pérez     if (v->shadow_vqs_enabled) {
1389d96be4c8SEugenio Pérez         /*
1390d96be4c8SEugenio Pérez          * Device vring base was set at device start. SVQ base is handled by
1391d96be4c8SEugenio Pérez          * VirtQueue code.
1392d96be4c8SEugenio Pérez          */
1393d96be4c8SEugenio Pérez         return 0;
1394d96be4c8SEugenio Pérez     }
1395d96be4c8SEugenio Pérez 
1396d96be4c8SEugenio Pérez     return vhost_vdpa_set_dev_vring_base(dev, ring);
1397108a6481SCindy Lu }
1398108a6481SCindy Lu 
1399108a6481SCindy Lu static int vhost_vdpa_get_vring_base(struct vhost_dev *dev,
1400108a6481SCindy Lu                                        struct vhost_vring_state *ring)
1401108a6481SCindy Lu {
14026d0b2226SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1403778e67deSLaurent Vivier     int ret;
1404778e67deSLaurent Vivier 
14056d0b2226SEugenio Pérez     if (v->shadow_vqs_enabled) {
14062fdac348SEugenio Pérez         ring->num = virtio_queue_get_last_avail_idx(dev->vdev, ring->index);
14076d0b2226SEugenio Pérez         return 0;
14086d0b2226SEugenio Pérez     }
14096d0b2226SEugenio Pérez 
1410b6662cb7SEugenio Pérez     if (!v->suspended) {
1411b6662cb7SEugenio Pérez         /*
1412b6662cb7SEugenio Pérez          * Cannot trust in value returned by device, let vhost recover used
1413b6662cb7SEugenio Pérez          * idx from guest.
1414b6662cb7SEugenio Pérez          */
1415b6662cb7SEugenio Pérez         return -1;
1416b6662cb7SEugenio Pérez     }
1417b6662cb7SEugenio Pérez 
1418778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_GET_VRING_BASE, ring);
1419778e67deSLaurent Vivier     trace_vhost_vdpa_get_vring_base(dev, ring->index, ring->num);
1420778e67deSLaurent Vivier     return ret;
1421108a6481SCindy Lu }
1422108a6481SCindy Lu 
1423108a6481SCindy Lu static int vhost_vdpa_set_vring_kick(struct vhost_dev *dev,
1424108a6481SCindy Lu                                        struct vhost_vring_file *file)
1425108a6481SCindy Lu {
1426dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1427dff4426fSEugenio Pérez     int vdpa_idx = file->index - dev->vq_index;
1428dff4426fSEugenio Pérez 
1429dff4426fSEugenio Pérez     if (v->shadow_vqs_enabled) {
1430dff4426fSEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, vdpa_idx);
1431dff4426fSEugenio Pérez         vhost_svq_set_svq_kick_fd(svq, file->fd);
1432dff4426fSEugenio Pérez         return 0;
1433dff4426fSEugenio Pérez     } else {
1434dff4426fSEugenio Pérez         return vhost_vdpa_set_vring_dev_kick(dev, file);
1435dff4426fSEugenio Pérez     }
1436108a6481SCindy Lu }
1437108a6481SCindy Lu 
1438108a6481SCindy Lu static int vhost_vdpa_set_vring_call(struct vhost_dev *dev,
1439108a6481SCindy Lu                                        struct vhost_vring_file *file)
1440108a6481SCindy Lu {
1441a8ac8858SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1442a8ac8858SEugenio Pérez     int vdpa_idx = file->index - dev->vq_index;
1443a8ac8858SEugenio Pérez     VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, vdpa_idx);
1444a8ac8858SEugenio Pérez 
1445b2765243SEugenio Pérez     /* Remember last call fd because we can switch to SVQ anytime. */
1446a8ac8858SEugenio Pérez     vhost_svq_set_svq_call_fd(svq, file->fd);
1447b2765243SEugenio Pérez     if (v->shadow_vqs_enabled) {
1448a8ac8858SEugenio Pérez         return 0;
1449a8ac8858SEugenio Pérez     }
1450b2765243SEugenio Pérez 
1451b2765243SEugenio Pérez     return vhost_vdpa_set_vring_dev_call(dev, file);
1452108a6481SCindy Lu }
1453108a6481SCindy Lu 
1454108a6481SCindy Lu static int vhost_vdpa_get_features(struct vhost_dev *dev,
1455108a6481SCindy Lu                                      uint64_t *features)
1456108a6481SCindy Lu {
145712a195faSEugenio Pérez     int ret = vhost_vdpa_get_dev_features(dev, features);
1458778e67deSLaurent Vivier 
1459ab7337e3SEugenio Pérez     if (ret == 0) {
146012a195faSEugenio Pérez         /* Add SVQ logging capabilities */
146112a195faSEugenio Pérez         *features |= BIT_ULL(VHOST_F_LOG_ALL);
146212a195faSEugenio Pérez     }
146312a195faSEugenio Pérez 
1464778e67deSLaurent Vivier     return ret;
1465108a6481SCindy Lu }
1466108a6481SCindy Lu 
1467108a6481SCindy Lu static int vhost_vdpa_set_owner(struct vhost_dev *dev)
1468108a6481SCindy Lu {
1469d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
14704d191cfdSJason Wang         return 0;
14714d191cfdSJason Wang     }
14724d191cfdSJason Wang 
1473778e67deSLaurent Vivier     trace_vhost_vdpa_set_owner(dev);
1474108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_OWNER, NULL);
1475108a6481SCindy Lu }
1476108a6481SCindy Lu 
1477108a6481SCindy Lu static int vhost_vdpa_vq_get_addr(struct vhost_dev *dev,
1478108a6481SCindy Lu                     struct vhost_vring_addr *addr, struct vhost_virtqueue *vq)
1479108a6481SCindy Lu {
1480108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
1481108a6481SCindy Lu     addr->desc_user_addr = (uint64_t)(unsigned long)vq->desc_phys;
1482108a6481SCindy Lu     addr->avail_user_addr = (uint64_t)(unsigned long)vq->avail_phys;
1483108a6481SCindy Lu     addr->used_user_addr = (uint64_t)(unsigned long)vq->used_phys;
1484778e67deSLaurent Vivier     trace_vhost_vdpa_vq_get_addr(dev, vq, addr->desc_user_addr,
1485778e67deSLaurent Vivier                                  addr->avail_user_addr, addr->used_user_addr);
1486108a6481SCindy Lu     return 0;
1487108a6481SCindy Lu }
1488108a6481SCindy Lu 
1489108a6481SCindy Lu static bool  vhost_vdpa_force_iommu(struct vhost_dev *dev)
1490108a6481SCindy Lu {
1491108a6481SCindy Lu     return true;
1492108a6481SCindy Lu }
1493108a6481SCindy Lu 
1494108a6481SCindy Lu const VhostOps vdpa_ops = {
1495108a6481SCindy Lu         .backend_type = VHOST_BACKEND_TYPE_VDPA,
1496108a6481SCindy Lu         .vhost_backend_init = vhost_vdpa_init,
1497108a6481SCindy Lu         .vhost_backend_cleanup = vhost_vdpa_cleanup,
1498108a6481SCindy Lu         .vhost_set_log_base = vhost_vdpa_set_log_base,
1499108a6481SCindy Lu         .vhost_set_vring_addr = vhost_vdpa_set_vring_addr,
1500108a6481SCindy Lu         .vhost_set_vring_num = vhost_vdpa_set_vring_num,
1501108a6481SCindy Lu         .vhost_set_vring_base = vhost_vdpa_set_vring_base,
1502108a6481SCindy Lu         .vhost_get_vring_base = vhost_vdpa_get_vring_base,
1503108a6481SCindy Lu         .vhost_set_vring_kick = vhost_vdpa_set_vring_kick,
1504108a6481SCindy Lu         .vhost_set_vring_call = vhost_vdpa_set_vring_call,
1505108a6481SCindy Lu         .vhost_get_features = vhost_vdpa_get_features,
1506a5bd0580SJason Wang         .vhost_set_backend_cap = vhost_vdpa_set_backend_cap,
1507108a6481SCindy Lu         .vhost_set_owner = vhost_vdpa_set_owner,
1508108a6481SCindy Lu         .vhost_set_vring_endian = NULL,
1509108a6481SCindy Lu         .vhost_backend_memslots_limit = vhost_vdpa_memslots_limit,
1510108a6481SCindy Lu         .vhost_set_mem_table = vhost_vdpa_set_mem_table,
1511108a6481SCindy Lu         .vhost_set_features = vhost_vdpa_set_features,
1512108a6481SCindy Lu         .vhost_reset_device = vhost_vdpa_reset_device,
1513108a6481SCindy Lu         .vhost_get_vq_index = vhost_vdpa_get_vq_index,
1514108a6481SCindy Lu         .vhost_get_config  = vhost_vdpa_get_config,
1515108a6481SCindy Lu         .vhost_set_config = vhost_vdpa_set_config,
1516108a6481SCindy Lu         .vhost_requires_shm_log = NULL,
1517108a6481SCindy Lu         .vhost_migration_done = NULL,
1518108a6481SCindy Lu         .vhost_backend_can_merge = 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