xref: /qemu/hw/virtio/vhost-vdpa.c (revision a8516e5c9719cf45fd81d6790bc9ffdcf753376b)
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;
50a5583457SEugenio Pérez     bool is_ram = memory_region_is_ram(section->mr);
51a5583457SEugenio Pérez     bool is_iommu = memory_region_is_iommu(section->mr);
52a5583457SEugenio Pérez     bool is_protected = memory_region_is_protected(section->mr);
53013108b6SEugenio Pérez 
54d60c75d2SJason Wang     /* vhost-vDPA doesn't allow MMIO to be mapped  */
55a5583457SEugenio Pérez     bool is_ram_device = memory_region_is_ram_device(section->mr);
56a5583457SEugenio Pérez 
57a5583457SEugenio Pérez     if ((!is_ram && !is_iommu) || is_protected || is_ram_device) {
58*a8516e5cSEugenio Pérez         trace_vhost_vdpa_skipped_memory_section(is_ram, is_iommu, is_protected,
59*a8516e5cSEugenio Pérez                                                 is_ram_device, iova_min,
60*a8516e5cSEugenio Pérez                                                 iova_max, page_mask);
61013108b6SEugenio Pérez         return true;
62013108b6SEugenio Pérez     }
63013108b6SEugenio Pérez 
64013108b6SEugenio Pérez     if (section->offset_within_address_space < iova_min) {
65013108b6SEugenio Pérez         error_report("RAM section out of device range (min=0x%" PRIx64
66013108b6SEugenio Pérez                      ", addr=0x%" HWADDR_PRIx ")",
67013108b6SEugenio Pérez                      iova_min, section->offset_within_address_space);
68013108b6SEugenio Pérez         return true;
69013108b6SEugenio Pérez     }
70bc7b0cacSCindy Lu     /*
71bc7b0cacSCindy Lu      * While using vIOMMU, sometimes the section will be larger than iova_max,
72bc7b0cacSCindy Lu      * but the memory that actually maps is smaller, so move the check to
73bc7b0cacSCindy Lu      * function vhost_vdpa_iommu_map_notify(). That function will use the actual
74bc7b0cacSCindy Lu      * size that maps to the kernel
75bc7b0cacSCindy Lu      */
76013108b6SEugenio Pérez 
77a5583457SEugenio Pérez     if (!is_iommu) {
788b1a8884SPhilippe Mathieu-Daudé         llend = vhost_vdpa_section_end(section, page_mask);
79013108b6SEugenio Pérez         if (int128_gt(llend, int128_make64(iova_max))) {
80013108b6SEugenio Pérez             error_report("RAM section out of device range (max=0x%" PRIx64
81013108b6SEugenio Pérez                          ", end addr=0x%" PRIx64 ")",
82013108b6SEugenio Pérez                          iova_max, int128_get64(llend));
83013108b6SEugenio Pérez             return true;
84013108b6SEugenio Pérez         }
85bc7b0cacSCindy Lu     }
86013108b6SEugenio Pérez 
87013108b6SEugenio Pérez     return false;
88108a6481SCindy Lu }
89108a6481SCindy Lu 
90cd831ed5SEugenio Pérez /*
91cd831ed5SEugenio Pérez  * The caller must set asid = 0 if the device does not support asid.
92cd831ed5SEugenio Pérez  * This is not an ABI break since it is set to 0 by the initializer anyway.
93cd831ed5SEugenio Pérez  */
946f03d9efSEugenio Pérez int vhost_vdpa_dma_map(VhostVDPAShared *s, uint32_t asid, hwaddr iova,
95cd831ed5SEugenio Pérez                        hwaddr size, void *vaddr, bool readonly)
96108a6481SCindy Lu {
97386494f2SCindy Lu     struct vhost_msg_v2 msg = {};
986f03d9efSEugenio Pérez     int fd = s->device_fd;
99108a6481SCindy Lu     int ret = 0;
100108a6481SCindy Lu 
10174e76c7dSEugenio Pérez     msg.type = VHOST_IOTLB_MSG_V2;
102cd831ed5SEugenio Pérez     msg.asid = asid;
103108a6481SCindy Lu     msg.iotlb.iova = iova;
104108a6481SCindy Lu     msg.iotlb.size = size;
105108a6481SCindy Lu     msg.iotlb.uaddr = (uint64_t)(uintptr_t)vaddr;
106108a6481SCindy Lu     msg.iotlb.perm = readonly ? VHOST_ACCESS_RO : VHOST_ACCESS_RW;
107108a6481SCindy Lu     msg.iotlb.type = VHOST_IOTLB_UPDATE;
108108a6481SCindy Lu 
1096f03d9efSEugenio Pérez     trace_vhost_vdpa_dma_map(s, fd, msg.type, msg.asid, msg.iotlb.iova,
110cd831ed5SEugenio Pérez                              msg.iotlb.size, msg.iotlb.uaddr, msg.iotlb.perm,
111cd831ed5SEugenio Pérez                              msg.iotlb.type);
112778e67deSLaurent Vivier 
113108a6481SCindy Lu     if (write(fd, &msg, sizeof(msg)) != sizeof(msg)) {
114108a6481SCindy Lu         error_report("failed to write, fd=%d, errno=%d (%s)",
115108a6481SCindy Lu             fd, errno, strerror(errno));
116108a6481SCindy Lu         return -EIO ;
117108a6481SCindy Lu     }
118108a6481SCindy Lu 
119108a6481SCindy Lu     return ret;
120108a6481SCindy Lu }
121108a6481SCindy Lu 
122cd831ed5SEugenio Pérez /*
123cd831ed5SEugenio Pérez  * The caller must set asid = 0 if the device does not support asid.
124cd831ed5SEugenio Pérez  * This is not an ABI break since it is set to 0 by the initializer anyway.
125cd831ed5SEugenio Pérez  */
1266f03d9efSEugenio Pérez int vhost_vdpa_dma_unmap(VhostVDPAShared *s, uint32_t asid, hwaddr iova,
127cd831ed5SEugenio Pérez                          hwaddr size)
128108a6481SCindy Lu {
129386494f2SCindy Lu     struct vhost_msg_v2 msg = {};
1306f03d9efSEugenio Pérez     int fd = s->device_fd;
131108a6481SCindy Lu     int ret = 0;
132108a6481SCindy Lu 
13374e76c7dSEugenio Pérez     msg.type = VHOST_IOTLB_MSG_V2;
134cd831ed5SEugenio Pérez     msg.asid = asid;
135108a6481SCindy Lu     msg.iotlb.iova = iova;
136108a6481SCindy Lu     msg.iotlb.size = size;
137108a6481SCindy Lu     msg.iotlb.type = VHOST_IOTLB_INVALIDATE;
138108a6481SCindy Lu 
1396f03d9efSEugenio Pérez     trace_vhost_vdpa_dma_unmap(s, fd, msg.type, msg.asid, msg.iotlb.iova,
1406f03d9efSEugenio Pérez                                msg.iotlb.size, msg.iotlb.type);
141778e67deSLaurent Vivier 
142108a6481SCindy Lu     if (write(fd, &msg, sizeof(msg)) != sizeof(msg)) {
143108a6481SCindy Lu         error_report("failed to write, fd=%d, errno=%d (%s)",
144108a6481SCindy Lu             fd, errno, strerror(errno));
145108a6481SCindy Lu         return -EIO ;
146108a6481SCindy Lu     }
147108a6481SCindy Lu 
148108a6481SCindy Lu     return ret;
149108a6481SCindy Lu }
150108a6481SCindy Lu 
1516f03d9efSEugenio Pérez static void vhost_vdpa_listener_begin_batch(VhostVDPAShared *s)
152a5bd0580SJason Wang {
1536f03d9efSEugenio Pérez     int fd = s->device_fd;
154e6db5df7SEugenio Pérez     struct vhost_msg_v2 msg = {
15574e76c7dSEugenio Pérez         .type = VHOST_IOTLB_MSG_V2,
156e6db5df7SEugenio Pérez         .iotlb.type = VHOST_IOTLB_BATCH_BEGIN,
157e6db5df7SEugenio Pérez     };
158a5bd0580SJason Wang 
1596f03d9efSEugenio Pérez     trace_vhost_vdpa_listener_begin_batch(s, fd, msg.type, msg.iotlb.type);
160a5bd0580SJason Wang     if (write(fd, &msg, sizeof(msg)) != sizeof(msg)) {
161a5bd0580SJason Wang         error_report("failed to write, fd=%d, errno=%d (%s)",
162a5bd0580SJason Wang                      fd, errno, strerror(errno));
163a5bd0580SJason Wang     }
164a5bd0580SJason Wang }
165a5bd0580SJason Wang 
1666f03d9efSEugenio Pérez static void vhost_vdpa_iotlb_batch_begin_once(VhostVDPAShared *s)
167e6db5df7SEugenio Pérez {
1686f03d9efSEugenio Pérez     if (s->backend_cap & (0x1ULL << VHOST_BACKEND_F_IOTLB_BATCH) &&
1696f03d9efSEugenio Pérez         !s->iotlb_batch_begin_sent) {
1706f03d9efSEugenio Pérez         vhost_vdpa_listener_begin_batch(s);
171e6db5df7SEugenio Pérez     }
172e6db5df7SEugenio Pérez 
1736f03d9efSEugenio Pérez     s->iotlb_batch_begin_sent = true;
174e6db5df7SEugenio Pérez }
175e6db5df7SEugenio Pérez 
176a5bd0580SJason Wang static void vhost_vdpa_listener_commit(MemoryListener *listener)
177a5bd0580SJason Wang {
178f6fe3e33SEugenio Pérez     VhostVDPAShared *s = container_of(listener, VhostVDPAShared, listener);
1798acb3218SPhilippe Mathieu-Daudé     struct vhost_msg_v2 msg = {};
180f6fe3e33SEugenio Pérez     int fd = s->device_fd;
181a5bd0580SJason Wang 
182f6fe3e33SEugenio Pérez     if (!(s->backend_cap & (0x1ULL << VHOST_BACKEND_F_IOTLB_BATCH))) {
183a5bd0580SJason Wang         return;
184a5bd0580SJason Wang     }
185a5bd0580SJason Wang 
186f6fe3e33SEugenio Pérez     if (!s->iotlb_batch_begin_sent) {
187e6db5df7SEugenio Pérez         return;
188e6db5df7SEugenio Pérez     }
189e6db5df7SEugenio Pérez 
19074e76c7dSEugenio Pérez     msg.type = VHOST_IOTLB_MSG_V2;
191a5bd0580SJason Wang     msg.iotlb.type = VHOST_IOTLB_BATCH_END;
192a5bd0580SJason Wang 
193f6fe3e33SEugenio Pérez     trace_vhost_vdpa_listener_commit(s, fd, msg.type, msg.iotlb.type);
194a5bd0580SJason Wang     if (write(fd, &msg, sizeof(msg)) != sizeof(msg)) {
195a5bd0580SJason Wang         error_report("failed to write, fd=%d, errno=%d (%s)",
196a5bd0580SJason Wang                      fd, errno, strerror(errno));
197a5bd0580SJason Wang     }
198e6db5df7SEugenio Pérez 
199f6fe3e33SEugenio Pérez     s->iotlb_batch_begin_sent = false;
200a5bd0580SJason Wang }
201a5bd0580SJason Wang 
202bc7b0cacSCindy Lu static void vhost_vdpa_iommu_map_notify(IOMMUNotifier *n, IOMMUTLBEntry *iotlb)
203bc7b0cacSCindy Lu {
204bc7b0cacSCindy Lu     struct vdpa_iommu *iommu = container_of(n, struct vdpa_iommu, n);
205bc7b0cacSCindy Lu 
206bc7b0cacSCindy Lu     hwaddr iova = iotlb->iova + iommu->iommu_offset;
207b06a38f2SEugenio Pérez     VhostVDPAShared *s = iommu->dev_shared;
208bc7b0cacSCindy Lu     void *vaddr;
209bc7b0cacSCindy Lu     int ret;
210bc7b0cacSCindy Lu     Int128 llend;
211bc7b0cacSCindy Lu 
212bc7b0cacSCindy Lu     if (iotlb->target_as != &address_space_memory) {
213bc7b0cacSCindy Lu         error_report("Wrong target AS \"%s\", only system memory is allowed",
214bc7b0cacSCindy Lu                      iotlb->target_as->name ? iotlb->target_as->name : "none");
215bc7b0cacSCindy Lu         return;
216bc7b0cacSCindy Lu     }
217bc7b0cacSCindy Lu     RCU_READ_LOCK_GUARD();
218bc7b0cacSCindy Lu     /* check if RAM section out of device range */
219bc7b0cacSCindy Lu     llend = int128_add(int128_makes64(iotlb->addr_mask), int128_makes64(iova));
220b06a38f2SEugenio Pérez     if (int128_gt(llend, int128_make64(s->iova_range.last))) {
221bc7b0cacSCindy Lu         error_report("RAM section out of device range (max=0x%" PRIx64
222bc7b0cacSCindy Lu                      ", end addr=0x%" PRIx64 ")",
223b06a38f2SEugenio Pérez                      s->iova_range.last, int128_get64(llend));
224bc7b0cacSCindy Lu         return;
225bc7b0cacSCindy Lu     }
226bc7b0cacSCindy Lu 
227bc7b0cacSCindy Lu     if ((iotlb->perm & IOMMU_RW) != IOMMU_NONE) {
228bc7b0cacSCindy Lu         bool read_only;
229bc7b0cacSCindy Lu 
230bc7b0cacSCindy Lu         if (!memory_get_xlat_addr(iotlb, &vaddr, NULL, &read_only, NULL)) {
231bc7b0cacSCindy Lu             return;
232bc7b0cacSCindy Lu         }
233b06a38f2SEugenio Pérez         ret = vhost_vdpa_dma_map(s, VHOST_VDPA_GUEST_PA_ASID, iova,
234bc7b0cacSCindy Lu                                  iotlb->addr_mask + 1, vaddr, read_only);
235bc7b0cacSCindy Lu         if (ret) {
236bc7b0cacSCindy Lu             error_report("vhost_vdpa_dma_map(%p, 0x%" HWADDR_PRIx ", "
237bc7b0cacSCindy Lu                          "0x%" HWADDR_PRIx ", %p) = %d (%m)",
238b06a38f2SEugenio Pérez                          s, iova, iotlb->addr_mask + 1, vaddr, ret);
239bc7b0cacSCindy Lu         }
240bc7b0cacSCindy Lu     } else {
241b06a38f2SEugenio Pérez         ret = vhost_vdpa_dma_unmap(s, VHOST_VDPA_GUEST_PA_ASID, iova,
242bc7b0cacSCindy Lu                                    iotlb->addr_mask + 1);
243bc7b0cacSCindy Lu         if (ret) {
244bc7b0cacSCindy Lu             error_report("vhost_vdpa_dma_unmap(%p, 0x%" HWADDR_PRIx ", "
245bc7b0cacSCindy Lu                          "0x%" HWADDR_PRIx ") = %d (%m)",
246b06a38f2SEugenio Pérez                          s, iova, iotlb->addr_mask + 1, ret);
247bc7b0cacSCindy Lu         }
248bc7b0cacSCindy Lu     }
249bc7b0cacSCindy Lu }
250bc7b0cacSCindy Lu 
251bc7b0cacSCindy Lu static void vhost_vdpa_iommu_region_add(MemoryListener *listener,
252bc7b0cacSCindy Lu                                         MemoryRegionSection *section)
253bc7b0cacSCindy Lu {
254f6fe3e33SEugenio Pérez     VhostVDPAShared *s = container_of(listener, VhostVDPAShared, listener);
255bc7b0cacSCindy Lu 
256bc7b0cacSCindy Lu     struct vdpa_iommu *iommu;
257bc7b0cacSCindy Lu     Int128 end;
258bc7b0cacSCindy Lu     int iommu_idx;
259bc7b0cacSCindy Lu     IOMMUMemoryRegion *iommu_mr;
260bc7b0cacSCindy Lu     int ret;
261bc7b0cacSCindy Lu 
262bc7b0cacSCindy Lu     iommu_mr = IOMMU_MEMORY_REGION(section->mr);
263bc7b0cacSCindy Lu 
264bc7b0cacSCindy Lu     iommu = g_malloc0(sizeof(*iommu));
265bc7b0cacSCindy Lu     end = int128_add(int128_make64(section->offset_within_region),
266bc7b0cacSCindy Lu                      section->size);
267bc7b0cacSCindy Lu     end = int128_sub(end, int128_one());
268bc7b0cacSCindy Lu     iommu_idx = memory_region_iommu_attrs_to_index(iommu_mr,
269bc7b0cacSCindy Lu                                                    MEMTXATTRS_UNSPECIFIED);
270bc7b0cacSCindy Lu     iommu->iommu_mr = iommu_mr;
271bc7b0cacSCindy Lu     iommu_notifier_init(&iommu->n, vhost_vdpa_iommu_map_notify,
272bc7b0cacSCindy Lu                         IOMMU_NOTIFIER_IOTLB_EVENTS,
273bc7b0cacSCindy Lu                         section->offset_within_region,
274bc7b0cacSCindy Lu                         int128_get64(end),
275bc7b0cacSCindy Lu                         iommu_idx);
276bc7b0cacSCindy Lu     iommu->iommu_offset = section->offset_within_address_space -
277bc7b0cacSCindy Lu                           section->offset_within_region;
278f6fe3e33SEugenio Pérez     iommu->dev_shared = s;
279bc7b0cacSCindy Lu 
280bc7b0cacSCindy Lu     ret = memory_region_register_iommu_notifier(section->mr, &iommu->n, NULL);
281bc7b0cacSCindy Lu     if (ret) {
282bc7b0cacSCindy Lu         g_free(iommu);
283bc7b0cacSCindy Lu         return;
284bc7b0cacSCindy Lu     }
285bc7b0cacSCindy Lu 
286f6fe3e33SEugenio Pérez     QLIST_INSERT_HEAD(&s->iommu_list, iommu, iommu_next);
287bc7b0cacSCindy Lu     memory_region_iommu_replay(iommu->iommu_mr, &iommu->n);
288bc7b0cacSCindy Lu 
289bc7b0cacSCindy Lu     return;
290bc7b0cacSCindy Lu }
291bc7b0cacSCindy Lu 
292bc7b0cacSCindy Lu static void vhost_vdpa_iommu_region_del(MemoryListener *listener,
293bc7b0cacSCindy Lu                                         MemoryRegionSection *section)
294bc7b0cacSCindy Lu {
295f6fe3e33SEugenio Pérez     VhostVDPAShared *s = container_of(listener, VhostVDPAShared, listener);
296bc7b0cacSCindy Lu 
297bc7b0cacSCindy Lu     struct vdpa_iommu *iommu;
298bc7b0cacSCindy Lu 
299f6fe3e33SEugenio Pérez     QLIST_FOREACH(iommu, &s->iommu_list, iommu_next)
300bc7b0cacSCindy Lu     {
301bc7b0cacSCindy Lu         if (MEMORY_REGION(iommu->iommu_mr) == section->mr &&
302bc7b0cacSCindy Lu             iommu->n.start == section->offset_within_region) {
303bc7b0cacSCindy Lu             memory_region_unregister_iommu_notifier(section->mr, &iommu->n);
304bc7b0cacSCindy Lu             QLIST_REMOVE(iommu, iommu_next);
305bc7b0cacSCindy Lu             g_free(iommu);
306bc7b0cacSCindy Lu             break;
307bc7b0cacSCindy Lu         }
308bc7b0cacSCindy Lu     }
309bc7b0cacSCindy Lu }
310bc7b0cacSCindy Lu 
311108a6481SCindy Lu static void vhost_vdpa_listener_region_add(MemoryListener *listener,
312108a6481SCindy Lu                                            MemoryRegionSection *section)
313108a6481SCindy Lu {
3147dab70beSEugenio Pérez     DMAMap mem_region = {};
315f6fe3e33SEugenio Pérez     VhostVDPAShared *s = container_of(listener, VhostVDPAShared, listener);
316108a6481SCindy Lu     hwaddr iova;
317108a6481SCindy Lu     Int128 llend, llsize;
318108a6481SCindy Lu     void *vaddr;
319108a6481SCindy Lu     int ret;
32033f21860SPhilippe Mathieu-Daudé     int page_size = qemu_target_page_size();
32133f21860SPhilippe Mathieu-Daudé     int page_mask = -page_size;
322108a6481SCindy Lu 
323f6fe3e33SEugenio Pérez     if (vhost_vdpa_listener_skipped_section(section, s->iova_range.first,
324f6fe3e33SEugenio Pérez                                             s->iova_range.last, page_mask)) {
325108a6481SCindy Lu         return;
326108a6481SCindy Lu     }
327bc7b0cacSCindy Lu     if (memory_region_is_iommu(section->mr)) {
328bc7b0cacSCindy Lu         vhost_vdpa_iommu_region_add(listener, section);
329bc7b0cacSCindy Lu         return;
330bc7b0cacSCindy Lu     }
331108a6481SCindy Lu 
33233f21860SPhilippe Mathieu-Daudé     if (unlikely((section->offset_within_address_space & ~page_mask) !=
33333f21860SPhilippe Mathieu-Daudé                  (section->offset_within_region & ~page_mask))) {
334f6fe3e33SEugenio Pérez         trace_vhost_vdpa_listener_region_add_unaligned(s, section->mr->name,
33533f21860SPhilippe Mathieu-Daudé                        section->offset_within_address_space & ~page_mask,
33633f21860SPhilippe Mathieu-Daudé                        section->offset_within_region & ~page_mask);
337108a6481SCindy Lu         return;
338108a6481SCindy Lu     }
339108a6481SCindy Lu 
34033f21860SPhilippe Mathieu-Daudé     iova = ROUND_UP(section->offset_within_address_space, page_size);
34133f21860SPhilippe Mathieu-Daudé     llend = vhost_vdpa_section_end(section, page_mask);
342108a6481SCindy Lu     if (int128_ge(int128_make64(iova), llend)) {
343108a6481SCindy Lu         return;
344108a6481SCindy Lu     }
345108a6481SCindy Lu 
346108a6481SCindy Lu     memory_region_ref(section->mr);
347108a6481SCindy Lu 
348108a6481SCindy Lu     /* Here we assume that memory_region_is_ram(section->mr)==true */
349108a6481SCindy Lu 
350108a6481SCindy Lu     vaddr = memory_region_get_ram_ptr(section->mr) +
351108a6481SCindy Lu             section->offset_within_region +
352108a6481SCindy Lu             (iova - section->offset_within_address_space);
353108a6481SCindy Lu 
354f6fe3e33SEugenio Pérez     trace_vhost_vdpa_listener_region_add(s, iova, int128_get64(llend),
355778e67deSLaurent Vivier                                          vaddr, section->readonly);
356778e67deSLaurent Vivier 
357108a6481SCindy Lu     llsize = int128_sub(llend, int128_make64(iova));
358f6fe3e33SEugenio Pérez     if (s->shadow_data) {
3597dab70beSEugenio Pérez         int r;
36034e3c94eSEugenio Pérez 
3617dab70beSEugenio Pérez         mem_region.translated_addr = (hwaddr)(uintptr_t)vaddr,
3627dab70beSEugenio Pérez         mem_region.size = int128_get64(llsize) - 1,
3637dab70beSEugenio Pérez         mem_region.perm = IOMMU_ACCESS_FLAG(true, section->readonly),
3647dab70beSEugenio Pérez 
365f6fe3e33SEugenio Pérez         r = vhost_iova_tree_map_alloc(s->iova_tree, &mem_region);
36634e3c94eSEugenio Pérez         if (unlikely(r != IOVA_OK)) {
36734e3c94eSEugenio Pérez             error_report("Can't allocate a mapping (%d)", r);
36834e3c94eSEugenio Pérez             goto fail;
36934e3c94eSEugenio Pérez         }
37034e3c94eSEugenio Pérez 
37134e3c94eSEugenio Pérez         iova = mem_region.iova;
37234e3c94eSEugenio Pérez     }
373108a6481SCindy Lu 
374f6fe3e33SEugenio Pérez     vhost_vdpa_iotlb_batch_begin_once(s);
375f6fe3e33SEugenio Pérez     ret = vhost_vdpa_dma_map(s, VHOST_VDPA_GUEST_PA_ASID, iova,
376cd831ed5SEugenio Pérez                              int128_get64(llsize), vaddr, section->readonly);
377108a6481SCindy Lu     if (ret) {
378108a6481SCindy Lu         error_report("vhost vdpa map fail!");
3797dab70beSEugenio Pérez         goto fail_map;
380108a6481SCindy Lu     }
381108a6481SCindy Lu 
382108a6481SCindy Lu     return;
383108a6481SCindy Lu 
3847dab70beSEugenio Pérez fail_map:
385f6fe3e33SEugenio Pérez     if (s->shadow_data) {
386f6fe3e33SEugenio Pérez         vhost_iova_tree_remove(s->iova_tree, mem_region);
3877dab70beSEugenio Pérez     }
3887dab70beSEugenio Pérez 
389108a6481SCindy Lu fail:
390108a6481SCindy Lu     /*
391108a6481SCindy Lu      * On the initfn path, store the first error in the container so we
392108a6481SCindy Lu      * can gracefully fail.  Runtime, there's not much we can do other
393108a6481SCindy Lu      * than throw a hardware error.
394108a6481SCindy Lu      */
395108a6481SCindy Lu     error_report("vhost-vdpa: DMA mapping failed, unable to continue");
396108a6481SCindy Lu     return;
397108a6481SCindy Lu 
398108a6481SCindy Lu }
399108a6481SCindy Lu 
400108a6481SCindy Lu static void vhost_vdpa_listener_region_del(MemoryListener *listener,
401108a6481SCindy Lu                                            MemoryRegionSection *section)
402108a6481SCindy Lu {
403f6fe3e33SEugenio Pérez     VhostVDPAShared *s = container_of(listener, VhostVDPAShared, listener);
404108a6481SCindy Lu     hwaddr iova;
405108a6481SCindy Lu     Int128 llend, llsize;
406108a6481SCindy Lu     int ret;
40733f21860SPhilippe Mathieu-Daudé     int page_size = qemu_target_page_size();
40833f21860SPhilippe Mathieu-Daudé     int page_mask = -page_size;
409108a6481SCindy Lu 
410f6fe3e33SEugenio Pérez     if (vhost_vdpa_listener_skipped_section(section, s->iova_range.first,
411f6fe3e33SEugenio Pérez                                             s->iova_range.last, page_mask)) {
412108a6481SCindy Lu         return;
413108a6481SCindy Lu     }
414bc7b0cacSCindy Lu     if (memory_region_is_iommu(section->mr)) {
415bc7b0cacSCindy Lu         vhost_vdpa_iommu_region_del(listener, section);
416bc7b0cacSCindy Lu     }
417108a6481SCindy Lu 
41833f21860SPhilippe Mathieu-Daudé     if (unlikely((section->offset_within_address_space & ~page_mask) !=
41933f21860SPhilippe Mathieu-Daudé                  (section->offset_within_region & ~page_mask))) {
420f6fe3e33SEugenio Pérez         trace_vhost_vdpa_listener_region_del_unaligned(s, section->mr->name,
42133f21860SPhilippe Mathieu-Daudé                        section->offset_within_address_space & ~page_mask,
42233f21860SPhilippe Mathieu-Daudé                        section->offset_within_region & ~page_mask);
423108a6481SCindy Lu         return;
424108a6481SCindy Lu     }
425108a6481SCindy Lu 
42633f21860SPhilippe Mathieu-Daudé     iova = ROUND_UP(section->offset_within_address_space, page_size);
42733f21860SPhilippe Mathieu-Daudé     llend = vhost_vdpa_section_end(section, page_mask);
428108a6481SCindy Lu 
429f6fe3e33SEugenio Pérez     trace_vhost_vdpa_listener_region_del(s, iova,
4303d1e4d34SCindy Lu         int128_get64(int128_sub(llend, int128_one())));
431778e67deSLaurent Vivier 
432108a6481SCindy Lu     if (int128_ge(int128_make64(iova), llend)) {
433108a6481SCindy Lu         return;
434108a6481SCindy Lu     }
435108a6481SCindy Lu 
436108a6481SCindy Lu     llsize = int128_sub(llend, int128_make64(iova));
437108a6481SCindy Lu 
438f6fe3e33SEugenio Pérez     if (s->shadow_data) {
43934e3c94eSEugenio Pérez         const DMAMap *result;
44034e3c94eSEugenio Pérez         const void *vaddr = memory_region_get_ram_ptr(section->mr) +
44134e3c94eSEugenio Pérez             section->offset_within_region +
44234e3c94eSEugenio Pérez             (iova - section->offset_within_address_space);
44334e3c94eSEugenio Pérez         DMAMap mem_region = {
44434e3c94eSEugenio Pérez             .translated_addr = (hwaddr)(uintptr_t)vaddr,
44534e3c94eSEugenio Pérez             .size = int128_get64(llsize) - 1,
44634e3c94eSEugenio Pérez         };
44734e3c94eSEugenio Pérez 
448f6fe3e33SEugenio Pérez         result = vhost_iova_tree_find_iova(s->iova_tree, &mem_region);
44910dab9f2SEugenio Pérez         if (!result) {
45010dab9f2SEugenio Pérez             /* The memory listener map wasn't mapped */
45110dab9f2SEugenio Pérez             return;
45210dab9f2SEugenio Pérez         }
45334e3c94eSEugenio Pérez         iova = result->iova;
454f6fe3e33SEugenio Pérez         vhost_iova_tree_remove(s->iova_tree, *result);
45534e3c94eSEugenio Pérez     }
456f6fe3e33SEugenio Pérez     vhost_vdpa_iotlb_batch_begin_once(s);
4572fbef6aaSCindy Lu     /*
4582fbef6aaSCindy Lu      * The unmap ioctl doesn't accept a full 64-bit. need to check it
4592fbef6aaSCindy Lu      */
4602fbef6aaSCindy Lu     if (int128_eq(llsize, int128_2_64())) {
4612fbef6aaSCindy Lu         llsize = int128_rshift(llsize, 1);
462f6fe3e33SEugenio Pérez         ret = vhost_vdpa_dma_unmap(s, VHOST_VDPA_GUEST_PA_ASID, iova,
463cd831ed5SEugenio Pérez                                    int128_get64(llsize));
4642fbef6aaSCindy Lu 
465108a6481SCindy Lu         if (ret) {
4662fbef6aaSCindy Lu             error_report("vhost_vdpa_dma_unmap(%p, 0x%" HWADDR_PRIx ", "
4672fbef6aaSCindy Lu                          "0x%" HWADDR_PRIx ") = %d (%m)",
468f6fe3e33SEugenio Pérez                          s, iova, int128_get64(llsize), ret);
4692fbef6aaSCindy Lu         }
4702fbef6aaSCindy Lu         iova += int128_get64(llsize);
4712fbef6aaSCindy Lu     }
472f6fe3e33SEugenio Pérez     ret = vhost_vdpa_dma_unmap(s, VHOST_VDPA_GUEST_PA_ASID, iova,
4732fbef6aaSCindy Lu                                int128_get64(llsize));
4742fbef6aaSCindy Lu 
4752fbef6aaSCindy Lu     if (ret) {
4762fbef6aaSCindy Lu         error_report("vhost_vdpa_dma_unmap(%p, 0x%" HWADDR_PRIx ", "
4772fbef6aaSCindy Lu                      "0x%" HWADDR_PRIx ") = %d (%m)",
478f6fe3e33SEugenio Pérez                      s, iova, int128_get64(llsize), ret);
479108a6481SCindy Lu     }
480108a6481SCindy Lu 
481108a6481SCindy Lu     memory_region_unref(section->mr);
482108a6481SCindy Lu }
483108a6481SCindy Lu /*
484ef4ff56cSStefano Garzarella  * IOTLB API is used by vhost-vdpa which requires incremental updating
485108a6481SCindy Lu  * of the mapping. So we can not use generic vhost memory listener which
486108a6481SCindy Lu  * depends on the addnop().
487108a6481SCindy Lu  */
488108a6481SCindy Lu static const MemoryListener vhost_vdpa_memory_listener = {
489142518bdSPeter Xu     .name = "vhost-vdpa",
490a5bd0580SJason Wang     .commit = vhost_vdpa_listener_commit,
491108a6481SCindy Lu     .region_add = vhost_vdpa_listener_region_add,
492108a6481SCindy Lu     .region_del = vhost_vdpa_listener_region_del,
493108a6481SCindy Lu };
494108a6481SCindy Lu 
495108a6481SCindy Lu static int vhost_vdpa_call(struct vhost_dev *dev, unsigned long int request,
496108a6481SCindy Lu                              void *arg)
497108a6481SCindy Lu {
498108a6481SCindy Lu     struct vhost_vdpa *v = dev->opaque;
499f12b2498SEugenio Pérez     int fd = v->shared->device_fd;
500f2a6e6c4SKevin Wolf     int ret;
501108a6481SCindy Lu 
502108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
503108a6481SCindy Lu 
504f2a6e6c4SKevin Wolf     ret = ioctl(fd, request, arg);
505f2a6e6c4SKevin Wolf     return ret < 0 ? -errno : ret;
506108a6481SCindy Lu }
507108a6481SCindy Lu 
5083631151bSRoman Kagan static int vhost_vdpa_add_status(struct vhost_dev *dev, uint8_t status)
509108a6481SCindy Lu {
510108a6481SCindy Lu     uint8_t s;
5113631151bSRoman Kagan     int ret;
512108a6481SCindy Lu 
513778e67deSLaurent Vivier     trace_vhost_vdpa_add_status(dev, status);
5143631151bSRoman Kagan     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_STATUS, &s);
5153631151bSRoman Kagan     if (ret < 0) {
5163631151bSRoman Kagan         return ret;
517108a6481SCindy Lu     }
518bc865bfeSEugenio Pérez     if ((s & status) == status) {
519bc865bfeSEugenio Pérez         /* Don't set bits already set */
520bc865bfeSEugenio Pérez         return 0;
521bc865bfeSEugenio Pérez     }
522108a6481SCindy Lu 
523108a6481SCindy Lu     s |= status;
524108a6481SCindy Lu 
5253631151bSRoman Kagan     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_STATUS, &s);
5263631151bSRoman Kagan     if (ret < 0) {
5273631151bSRoman Kagan         return ret;
5283631151bSRoman Kagan     }
5293631151bSRoman Kagan 
5303631151bSRoman Kagan     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_STATUS, &s);
5313631151bSRoman Kagan     if (ret < 0) {
5323631151bSRoman Kagan         return ret;
5333631151bSRoman Kagan     }
5343631151bSRoman Kagan 
5353631151bSRoman Kagan     if (!(s & status)) {
5363631151bSRoman Kagan         return -EIO;
5373631151bSRoman Kagan     }
5383631151bSRoman Kagan 
5393631151bSRoman Kagan     return 0;
540108a6481SCindy Lu }
541108a6481SCindy Lu 
542c672f348SLongpeng int vhost_vdpa_get_iova_range(int fd, struct vhost_vdpa_iova_range *iova_range)
543c672f348SLongpeng {
544c672f348SLongpeng     int ret = ioctl(fd, VHOST_VDPA_GET_IOVA_RANGE, iova_range);
545c672f348SLongpeng 
546c672f348SLongpeng     return ret < 0 ? -errno : 0;
547c672f348SLongpeng }
548c672f348SLongpeng 
549d71b0609SSi-Wei Liu /*
550d71b0609SSi-Wei Liu  * The use of this function is for requests that only need to be
551d71b0609SSi-Wei Liu  * applied once. Typically such request occurs at the beginning
552d71b0609SSi-Wei Liu  * of operation, and before setting up queues. It should not be
553d71b0609SSi-Wei Liu  * used for request that performs operation until all queues are
554d71b0609SSi-Wei Liu  * set, which would need to check dev->vq_index_end instead.
555d71b0609SSi-Wei Liu  */
556d71b0609SSi-Wei Liu static bool vhost_vdpa_first_dev(struct vhost_dev *dev)
5574d191cfdSJason Wang {
5584d191cfdSJason Wang     struct vhost_vdpa *v = dev->opaque;
5594d191cfdSJason Wang 
560d71b0609SSi-Wei Liu     return v->index == 0;
5614d191cfdSJason Wang }
5624d191cfdSJason Wang 
563c812b065SSi-Wei Liu static bool vhost_vdpa_last_dev(struct vhost_dev *dev)
564c812b065SSi-Wei Liu {
565c812b065SSi-Wei Liu     return dev->vq_index + dev->nvqs == dev->vq_index_end;
566c812b065SSi-Wei Liu }
567c812b065SSi-Wei Liu 
56812a195faSEugenio Pérez static int vhost_vdpa_get_dev_features(struct vhost_dev *dev,
56912a195faSEugenio Pérez                                        uint64_t *features)
57012a195faSEugenio Pérez {
57112a195faSEugenio Pérez     int ret;
57212a195faSEugenio Pérez 
57312a195faSEugenio Pérez     ret = vhost_vdpa_call(dev, VHOST_GET_FEATURES, features);
57412a195faSEugenio Pérez     trace_vhost_vdpa_get_features(dev, *features);
57512a195faSEugenio Pérez     return ret;
57612a195faSEugenio Pérez }
57712a195faSEugenio Pérez 
578258a0394SEugenio Pérez static void vhost_vdpa_init_svq(struct vhost_dev *hdev, struct vhost_vdpa *v)
579dff4426fSEugenio Pérez {
580dff4426fSEugenio Pérez     g_autoptr(GPtrArray) shadow_vqs = NULL;
5814725a418SEugenio Pérez 
582dff4426fSEugenio Pérez     shadow_vqs = g_ptr_array_new_full(hdev->nvqs, vhost_svq_free);
583dff4426fSEugenio Pérez     for (unsigned n = 0; n < hdev->nvqs; ++n) {
5843cfb4d06SEugenio Pérez         VhostShadowVirtqueue *svq;
585dff4426fSEugenio Pérez 
5865fde952bSEugenio Pérez         svq = vhost_svq_new(v->shadow_vq_ops, v->shadow_vq_ops_opaque);
5873cfb4d06SEugenio Pérez         g_ptr_array_add(shadow_vqs, svq);
588dff4426fSEugenio Pérez     }
589dff4426fSEugenio Pérez 
590dff4426fSEugenio Pérez     v->shadow_vqs = g_steal_pointer(&shadow_vqs);
591dff4426fSEugenio Pérez }
592dff4426fSEugenio Pérez 
59328770ff9SKevin Wolf static int vhost_vdpa_init(struct vhost_dev *dev, void *opaque, Error **errp)
594108a6481SCindy Lu {
595e36b9992SEugenio Pérez     struct vhost_vdpa *v = opaque;
596108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
597e36b9992SEugenio Pérez     trace_vhost_vdpa_init(dev, v->shared, opaque);
598e1c1915bSDavid Hildenbrand     int ret;
599e1c1915bSDavid Hildenbrand 
600a5bd0580SJason Wang     v->dev = dev;
601108a6481SCindy Lu     dev->opaque =  opaque ;
602f6fe3e33SEugenio Pérez     v->shared->listener = vhost_vdpa_memory_listener;
603258a0394SEugenio Pérez     vhost_vdpa_init_svq(dev, v);
604108a6481SCindy Lu 
6059c363cf6SEugenio Pérez     error_propagate(&dev->migration_blocker, v->migration_blocker);
606d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
6074d191cfdSJason Wang         return 0;
6084d191cfdSJason Wang     }
6094d191cfdSJason Wang 
610a230c471SEugenio Pérez     /*
61157ac8318SEugenio Pérez      * If dev->shadow_vqs_enabled at initialization that means the device has
61257ac8318SEugenio Pérez      * been started with x-svq=on, so don't block migration
61357ac8318SEugenio Pérez      */
61457ac8318SEugenio Pérez     if (dev->migration_blocker == NULL && !v->shadow_vqs_enabled) {
61557ac8318SEugenio Pérez         /* We don't have dev->features yet */
61657ac8318SEugenio Pérez         uint64_t features;
61757ac8318SEugenio Pérez         ret = vhost_vdpa_get_dev_features(dev, &features);
61857ac8318SEugenio Pérez         if (unlikely(ret)) {
61957ac8318SEugenio Pérez             error_setg_errno(errp, -ret, "Could not get device features");
62057ac8318SEugenio Pérez             return ret;
62157ac8318SEugenio Pérez         }
62257ac8318SEugenio Pérez         vhost_svq_valid_features(features, &dev->migration_blocker);
62357ac8318SEugenio Pérez     }
62457ac8318SEugenio Pérez 
62557ac8318SEugenio Pérez     /*
626a230c471SEugenio Pérez      * Similar to VFIO, we end up pinning all guest memory and have to
627a230c471SEugenio Pérez      * disable discarding of RAM.
628a230c471SEugenio Pérez      */
629a230c471SEugenio Pérez     ret = ram_block_discard_disable(true);
630a230c471SEugenio Pérez     if (ret) {
631a230c471SEugenio Pérez         error_report("Cannot set discarding of RAM broken");
632a230c471SEugenio Pérez         return ret;
633a230c471SEugenio Pérez     }
634a230c471SEugenio Pérez 
635108a6481SCindy Lu     vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_ACKNOWLEDGE |
636108a6481SCindy Lu                                VIRTIO_CONFIG_S_DRIVER);
637108a6481SCindy Lu 
638108a6481SCindy Lu     return 0;
639108a6481SCindy Lu }
640108a6481SCindy Lu 
641d0416d48SJason Wang static void vhost_vdpa_host_notifier_uninit(struct vhost_dev *dev,
642d0416d48SJason Wang                                             int queue_index)
643d0416d48SJason Wang {
6448e3b0cbbSMarc-André Lureau     size_t page_size = qemu_real_host_page_size();
645d0416d48SJason Wang     struct vhost_vdpa *v = dev->opaque;
646d0416d48SJason Wang     VirtIODevice *vdev = dev->vdev;
647d0416d48SJason Wang     VhostVDPAHostNotifier *n;
648d0416d48SJason Wang 
649d0416d48SJason Wang     n = &v->notifier[queue_index];
650d0416d48SJason Wang 
651d0416d48SJason Wang     if (n->addr) {
652d0416d48SJason Wang         virtio_queue_set_host_notifier_mr(vdev, queue_index, &n->mr, false);
653d0416d48SJason Wang         object_unparent(OBJECT(&n->mr));
654d0416d48SJason Wang         munmap(n->addr, page_size);
655d0416d48SJason Wang         n->addr = NULL;
656d0416d48SJason Wang     }
657d0416d48SJason Wang }
658d0416d48SJason Wang 
659d0416d48SJason Wang static int vhost_vdpa_host_notifier_init(struct vhost_dev *dev, int queue_index)
660d0416d48SJason Wang {
6618e3b0cbbSMarc-André Lureau     size_t page_size = qemu_real_host_page_size();
662d0416d48SJason Wang     struct vhost_vdpa *v = dev->opaque;
663d0416d48SJason Wang     VirtIODevice *vdev = dev->vdev;
664d0416d48SJason Wang     VhostVDPAHostNotifier *n;
665f12b2498SEugenio Pérez     int fd = v->shared->device_fd;
666d0416d48SJason Wang     void *addr;
667d0416d48SJason Wang     char *name;
668d0416d48SJason Wang 
669d0416d48SJason Wang     vhost_vdpa_host_notifier_uninit(dev, queue_index);
670d0416d48SJason Wang 
671d0416d48SJason Wang     n = &v->notifier[queue_index];
672d0416d48SJason Wang 
673d0416d48SJason Wang     addr = mmap(NULL, page_size, PROT_WRITE, MAP_SHARED, fd,
674d0416d48SJason Wang                 queue_index * page_size);
675d0416d48SJason Wang     if (addr == MAP_FAILED) {
676d0416d48SJason Wang         goto err;
677d0416d48SJason Wang     }
678d0416d48SJason Wang 
679d0416d48SJason Wang     name = g_strdup_printf("vhost-vdpa/host-notifier@%p mmaps[%d]",
680d0416d48SJason Wang                            v, queue_index);
681d0416d48SJason Wang     memory_region_init_ram_device_ptr(&n->mr, OBJECT(vdev), name,
682d0416d48SJason Wang                                       page_size, addr);
683d0416d48SJason Wang     g_free(name);
684d0416d48SJason Wang 
685d0416d48SJason Wang     if (virtio_queue_set_host_notifier_mr(vdev, queue_index, &n->mr, true)) {
68698f7607eSLaurent Vivier         object_unparent(OBJECT(&n->mr));
687d0416d48SJason Wang         munmap(addr, page_size);
688d0416d48SJason Wang         goto err;
689d0416d48SJason Wang     }
690d0416d48SJason Wang     n->addr = addr;
691d0416d48SJason Wang 
692d0416d48SJason Wang     return 0;
693d0416d48SJason Wang 
694d0416d48SJason Wang err:
695d0416d48SJason Wang     return -1;
696d0416d48SJason Wang }
697d0416d48SJason Wang 
698b1f030a0SLaurent Vivier static void vhost_vdpa_host_notifiers_uninit(struct vhost_dev *dev, int n)
699b1f030a0SLaurent Vivier {
700b1f030a0SLaurent Vivier     int i;
701b1f030a0SLaurent Vivier 
702e66f2311SLongpeng     /*
703e66f2311SLongpeng      * Pack all the changes to the memory regions in a single
704e66f2311SLongpeng      * transaction to avoid a few updating of the address space
705e66f2311SLongpeng      * topology.
706e66f2311SLongpeng      */
707e66f2311SLongpeng     memory_region_transaction_begin();
708e66f2311SLongpeng 
709b1f030a0SLaurent Vivier     for (i = dev->vq_index; i < dev->vq_index + n; i++) {
710b1f030a0SLaurent Vivier         vhost_vdpa_host_notifier_uninit(dev, i);
711b1f030a0SLaurent Vivier     }
712e66f2311SLongpeng 
713e66f2311SLongpeng     memory_region_transaction_commit();
714b1f030a0SLaurent Vivier }
715b1f030a0SLaurent Vivier 
716d0416d48SJason Wang static void vhost_vdpa_host_notifiers_init(struct vhost_dev *dev)
717d0416d48SJason Wang {
718dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
719d0416d48SJason Wang     int i;
720d0416d48SJason Wang 
721dff4426fSEugenio Pérez     if (v->shadow_vqs_enabled) {
722dff4426fSEugenio Pérez         /* FIXME SVQ is not compatible with host notifiers mr */
723dff4426fSEugenio Pérez         return;
724dff4426fSEugenio Pérez     }
725dff4426fSEugenio Pérez 
726e66f2311SLongpeng     /*
727e66f2311SLongpeng      * Pack all the changes to the memory regions in a single
728e66f2311SLongpeng      * transaction to avoid a few updating of the address space
729e66f2311SLongpeng      * topology.
730e66f2311SLongpeng      */
731e66f2311SLongpeng     memory_region_transaction_begin();
732e66f2311SLongpeng 
733d0416d48SJason Wang     for (i = dev->vq_index; i < dev->vq_index + dev->nvqs; i++) {
734d0416d48SJason Wang         if (vhost_vdpa_host_notifier_init(dev, i)) {
735b1f030a0SLaurent Vivier             vhost_vdpa_host_notifiers_uninit(dev, i - dev->vq_index);
736e66f2311SLongpeng             break;
737e66f2311SLongpeng         }
738e66f2311SLongpeng     }
739e66f2311SLongpeng 
740e66f2311SLongpeng     memory_region_transaction_commit();
741d0416d48SJason Wang }
742d0416d48SJason Wang 
743dff4426fSEugenio Pérez static void vhost_vdpa_svq_cleanup(struct vhost_dev *dev)
744dff4426fSEugenio Pérez {
745dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
746dff4426fSEugenio Pérez     size_t idx;
747dff4426fSEugenio Pérez 
748dff4426fSEugenio Pérez     for (idx = 0; idx < v->shadow_vqs->len; ++idx) {
749dff4426fSEugenio Pérez         vhost_svq_stop(g_ptr_array_index(v->shadow_vqs, idx));
750dff4426fSEugenio Pérez     }
751dff4426fSEugenio Pérez     g_ptr_array_free(v->shadow_vqs, true);
752dff4426fSEugenio Pérez }
753dff4426fSEugenio Pérez 
754108a6481SCindy Lu static int vhost_vdpa_cleanup(struct vhost_dev *dev)
755108a6481SCindy Lu {
756108a6481SCindy Lu     struct vhost_vdpa *v;
757108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
758108a6481SCindy Lu     v = dev->opaque;
759778e67deSLaurent Vivier     trace_vhost_vdpa_cleanup(dev, v);
760a230c471SEugenio Pérez     if (vhost_vdpa_first_dev(dev)) {
761a230c471SEugenio Pérez         ram_block_discard_disable(false);
762f6fe3e33SEugenio Pérez         memory_listener_unregister(&v->shared->listener);
763a230c471SEugenio Pérez     }
764a230c471SEugenio Pérez 
765d0416d48SJason Wang     vhost_vdpa_host_notifiers_uninit(dev, dev->nvqs);
766dff4426fSEugenio Pérez     vhost_vdpa_svq_cleanup(dev);
767108a6481SCindy Lu 
768108a6481SCindy Lu     dev->opaque = NULL;
769e1c1915bSDavid Hildenbrand 
770108a6481SCindy Lu     return 0;
771108a6481SCindy Lu }
772108a6481SCindy Lu 
773108a6481SCindy Lu static int vhost_vdpa_memslots_limit(struct vhost_dev *dev)
774108a6481SCindy Lu {
775778e67deSLaurent Vivier     trace_vhost_vdpa_memslots_limit(dev, INT_MAX);
776108a6481SCindy Lu     return INT_MAX;
777108a6481SCindy Lu }
778108a6481SCindy Lu 
779108a6481SCindy Lu static int vhost_vdpa_set_mem_table(struct vhost_dev *dev,
780108a6481SCindy Lu                                     struct vhost_memory *mem)
781108a6481SCindy Lu {
782d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
7834d191cfdSJason Wang         return 0;
7844d191cfdSJason Wang     }
7854d191cfdSJason Wang 
786778e67deSLaurent Vivier     trace_vhost_vdpa_set_mem_table(dev, mem->nregions, mem->padding);
787778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_SET_MEM_TABLE) &&
788778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_REGIONS)) {
789778e67deSLaurent Vivier         int i;
790778e67deSLaurent Vivier         for (i = 0; i < mem->nregions; i++) {
791778e67deSLaurent Vivier             trace_vhost_vdpa_dump_regions(dev, i,
792778e67deSLaurent Vivier                                           mem->regions[i].guest_phys_addr,
793778e67deSLaurent Vivier                                           mem->regions[i].memory_size,
794778e67deSLaurent Vivier                                           mem->regions[i].userspace_addr,
795778e67deSLaurent Vivier                                           mem->regions[i].flags_padding);
796778e67deSLaurent Vivier         }
797778e67deSLaurent Vivier     }
798108a6481SCindy Lu     if (mem->padding) {
7993631151bSRoman Kagan         return -EINVAL;
800108a6481SCindy Lu     }
801108a6481SCindy Lu 
802108a6481SCindy Lu     return 0;
803108a6481SCindy Lu }
804108a6481SCindy Lu 
805108a6481SCindy Lu static int vhost_vdpa_set_features(struct vhost_dev *dev,
806108a6481SCindy Lu                                    uint64_t features)
807108a6481SCindy Lu {
80812a195faSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
809108a6481SCindy Lu     int ret;
8104d191cfdSJason Wang 
811d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
8124d191cfdSJason Wang         return 0;
8134d191cfdSJason Wang     }
8144d191cfdSJason Wang 
81512a195faSEugenio Pérez     if (v->shadow_vqs_enabled) {
81612a195faSEugenio Pérez         if ((v->acked_features ^ features) == BIT_ULL(VHOST_F_LOG_ALL)) {
81712a195faSEugenio Pérez             /*
81812a195faSEugenio Pérez              * QEMU is just trying to enable or disable logging. SVQ handles
81912a195faSEugenio Pérez              * this sepparately, so no need to forward this.
82012a195faSEugenio Pérez              */
82112a195faSEugenio Pérez             v->acked_features = features;
82212a195faSEugenio Pérez             return 0;
82312a195faSEugenio Pérez         }
82412a195faSEugenio Pérez 
82512a195faSEugenio Pérez         v->acked_features = features;
82612a195faSEugenio Pérez 
82712a195faSEugenio Pérez         /* We must not ack _F_LOG if SVQ is enabled */
82812a195faSEugenio Pérez         features &= ~BIT_ULL(VHOST_F_LOG_ALL);
82912a195faSEugenio Pérez     }
83012a195faSEugenio Pérez 
831778e67deSLaurent Vivier     trace_vhost_vdpa_set_features(dev, features);
832108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_SET_FEATURES, &features);
833108a6481SCindy Lu     if (ret) {
834108a6481SCindy Lu         return ret;
835108a6481SCindy Lu     }
836108a6481SCindy Lu 
8373631151bSRoman Kagan     return vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_FEATURES_OK);
838108a6481SCindy Lu }
839108a6481SCindy Lu 
840a5bd0580SJason Wang static int vhost_vdpa_set_backend_cap(struct vhost_dev *dev)
841a5bd0580SJason Wang {
8424da38d1aSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
8434da38d1aSEugenio Pérez 
844a5bd0580SJason Wang     uint64_t features;
845a5bd0580SJason Wang     uint64_t f = 0x1ULL << VHOST_BACKEND_F_IOTLB_MSG_V2 |
846c1a10086SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_IOTLB_BATCH |
847d83b4945SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_IOTLB_ASID |
848d83b4945SEugenio Pérez         0x1ULL << VHOST_BACKEND_F_SUSPEND;
849a5bd0580SJason Wang     int r;
850a5bd0580SJason Wang 
851a5bd0580SJason Wang     if (vhost_vdpa_call(dev, VHOST_GET_BACKEND_FEATURES, &features)) {
8522a83e97eSJason Wang         return -EFAULT;
853a5bd0580SJason Wang     }
854a5bd0580SJason Wang 
855a5bd0580SJason Wang     features &= f;
8564d191cfdSJason Wang 
857d71b0609SSi-Wei Liu     if (vhost_vdpa_first_dev(dev)) {
858a5bd0580SJason Wang         r = vhost_vdpa_call(dev, VHOST_SET_BACKEND_FEATURES, &features);
859a5bd0580SJason Wang         if (r) {
8602a83e97eSJason Wang             return -EFAULT;
861a5bd0580SJason Wang         }
8624d191cfdSJason Wang     }
863a5bd0580SJason Wang 
864a5bd0580SJason Wang     dev->backend_cap = features;
8654da38d1aSEugenio Pérez     v->shared->backend_cap = features;
866a5bd0580SJason Wang 
867a5bd0580SJason Wang     return 0;
868a5bd0580SJason Wang }
869a5bd0580SJason Wang 
870c232b8f4SZenghui Yu static int vhost_vdpa_get_device_id(struct vhost_dev *dev,
871108a6481SCindy Lu                                     uint32_t *device_id)
872108a6481SCindy Lu {
873778e67deSLaurent Vivier     int ret;
874778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_DEVICE_ID, device_id);
875778e67deSLaurent Vivier     trace_vhost_vdpa_get_device_id(dev, *device_id);
876778e67deSLaurent Vivier     return ret;
877108a6481SCindy Lu }
878108a6481SCindy Lu 
879108a6481SCindy Lu static int vhost_vdpa_reset_device(struct vhost_dev *dev)
880108a6481SCindy Lu {
8810bb302a9SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
882778e67deSLaurent Vivier     int ret;
883108a6481SCindy Lu     uint8_t status = 0;
884108a6481SCindy Lu 
885778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_STATUS, &status);
8862b5de4d7SEugenio Pérez     trace_vhost_vdpa_reset_device(dev);
8870bb302a9SEugenio Pérez     v->suspended = false;
888778e67deSLaurent Vivier     return ret;
889108a6481SCindy Lu }
890108a6481SCindy Lu 
891108a6481SCindy Lu static int vhost_vdpa_get_vq_index(struct vhost_dev *dev, int idx)
892108a6481SCindy Lu {
893108a6481SCindy Lu     assert(idx >= dev->vq_index && idx < dev->vq_index + dev->nvqs);
894108a6481SCindy Lu 
895353244d8SJason Wang     trace_vhost_vdpa_get_vq_index(dev, idx, idx);
896353244d8SJason Wang     return idx;
897108a6481SCindy Lu }
898108a6481SCindy Lu 
899d7ce0841SEugenio Pérez int vhost_vdpa_set_vring_ready(struct vhost_vdpa *v, unsigned idx)
900108a6481SCindy Lu {
901d7ce0841SEugenio Pérez     struct vhost_dev *dev = v->dev;
902108a6481SCindy Lu     struct vhost_vring_state state = {
903d7ce0841SEugenio Pérez         .index = idx,
904108a6481SCindy Lu         .num = 1,
905108a6481SCindy Lu     };
906d7ce0841SEugenio Pérez     int r = vhost_vdpa_call(dev, VHOST_VDPA_SET_VRING_ENABLE, &state);
907d7ce0841SEugenio Pérez 
908d7ce0841SEugenio Pérez     trace_vhost_vdpa_set_vring_ready(dev, idx, r);
909d7ce0841SEugenio Pérez     return r;
910108a6481SCindy Lu }
911108a6481SCindy Lu 
912259f3accSCindy Lu static int vhost_vdpa_set_config_call(struct vhost_dev *dev,
913259f3accSCindy Lu                                        int fd)
914259f3accSCindy Lu {
915259f3accSCindy Lu     trace_vhost_vdpa_set_config_call(dev, fd);
916259f3accSCindy Lu     return vhost_vdpa_call(dev, VHOST_VDPA_SET_CONFIG_CALL, &fd);
917259f3accSCindy Lu }
918259f3accSCindy Lu 
919778e67deSLaurent Vivier static void vhost_vdpa_dump_config(struct vhost_dev *dev, const uint8_t *config,
920778e67deSLaurent Vivier                                    uint32_t config_len)
921778e67deSLaurent Vivier {
922778e67deSLaurent Vivier     int b, len;
923778e67deSLaurent Vivier     char line[QEMU_HEXDUMP_LINE_LEN];
924778e67deSLaurent Vivier 
925778e67deSLaurent Vivier     for (b = 0; b < config_len; b += 16) {
926778e67deSLaurent Vivier         len = config_len - b;
927778e67deSLaurent Vivier         qemu_hexdump_line(line, b, config, len, false);
928778e67deSLaurent Vivier         trace_vhost_vdpa_dump_config(dev, line);
929778e67deSLaurent Vivier     }
930778e67deSLaurent Vivier }
931778e67deSLaurent Vivier 
932108a6481SCindy Lu static int vhost_vdpa_set_config(struct vhost_dev *dev, const uint8_t *data,
933108a6481SCindy Lu                                    uint32_t offset, uint32_t size,
934108a6481SCindy Lu                                    uint32_t flags)
935108a6481SCindy Lu {
936108a6481SCindy Lu     struct vhost_vdpa_config *config;
937108a6481SCindy Lu     int ret;
938108a6481SCindy Lu     unsigned long config_size = offsetof(struct vhost_vdpa_config, buf);
939986d4f78SLi Qiang 
940778e67deSLaurent Vivier     trace_vhost_vdpa_set_config(dev, offset, size, flags);
941108a6481SCindy Lu     config = g_malloc(size + config_size);
942108a6481SCindy Lu     config->off = offset;
943108a6481SCindy Lu     config->len = size;
944108a6481SCindy Lu     memcpy(config->buf, data, size);
945778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_SET_CONFIG) &&
946778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_CONFIG)) {
947778e67deSLaurent Vivier         vhost_vdpa_dump_config(dev, data, size);
948778e67deSLaurent Vivier     }
949108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_VDPA_SET_CONFIG, config);
950108a6481SCindy Lu     g_free(config);
951108a6481SCindy Lu     return ret;
952108a6481SCindy Lu }
953108a6481SCindy Lu 
954108a6481SCindy Lu static int vhost_vdpa_get_config(struct vhost_dev *dev, uint8_t *config,
95550de5138SKevin Wolf                                    uint32_t config_len, Error **errp)
956108a6481SCindy Lu {
957108a6481SCindy Lu     struct vhost_vdpa_config *v_config;
958108a6481SCindy Lu     unsigned long config_size = offsetof(struct vhost_vdpa_config, buf);
959108a6481SCindy Lu     int ret;
960108a6481SCindy Lu 
961778e67deSLaurent Vivier     trace_vhost_vdpa_get_config(dev, config, config_len);
962108a6481SCindy Lu     v_config = g_malloc(config_len + config_size);
963108a6481SCindy Lu     v_config->len = config_len;
964108a6481SCindy Lu     v_config->off = 0;
965108a6481SCindy Lu     ret = vhost_vdpa_call(dev, VHOST_VDPA_GET_CONFIG, v_config);
966108a6481SCindy Lu     memcpy(config, v_config->buf, config_len);
967108a6481SCindy Lu     g_free(v_config);
968778e67deSLaurent Vivier     if (trace_event_get_state_backends(TRACE_VHOST_VDPA_GET_CONFIG) &&
969778e67deSLaurent Vivier         trace_event_get_state_backends(TRACE_VHOST_VDPA_DUMP_CONFIG)) {
970778e67deSLaurent Vivier         vhost_vdpa_dump_config(dev, config, config_len);
971778e67deSLaurent Vivier     }
972108a6481SCindy Lu     return ret;
973108a6481SCindy Lu  }
974108a6481SCindy Lu 
975d96be4c8SEugenio Pérez static int vhost_vdpa_set_dev_vring_base(struct vhost_dev *dev,
976d96be4c8SEugenio Pérez                                          struct vhost_vring_state *ring)
977d96be4c8SEugenio Pérez {
97819a060bcSSi-Wei Liu     struct vhost_vdpa *v = dev->opaque;
97919a060bcSSi-Wei Liu 
98019a060bcSSi-Wei Liu     trace_vhost_vdpa_set_dev_vring_base(dev, ring->index, ring->num,
98119a060bcSSi-Wei Liu                                         v->shadow_vqs_enabled);
982d96be4c8SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_BASE, ring);
983d96be4c8SEugenio Pérez }
984d96be4c8SEugenio Pérez 
985dff4426fSEugenio Pérez static int vhost_vdpa_set_vring_dev_kick(struct vhost_dev *dev,
986dff4426fSEugenio Pérez                                          struct vhost_vring_file *file)
987dff4426fSEugenio Pérez {
988dff4426fSEugenio Pérez     trace_vhost_vdpa_set_vring_kick(dev, file->index, file->fd);
989dff4426fSEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_KICK, file);
990dff4426fSEugenio Pérez }
991dff4426fSEugenio Pérez 
992a8ac8858SEugenio Pérez static int vhost_vdpa_set_vring_dev_call(struct vhost_dev *dev,
993a8ac8858SEugenio Pérez                                          struct vhost_vring_file *file)
994a8ac8858SEugenio Pérez {
995a8ac8858SEugenio Pérez     trace_vhost_vdpa_set_vring_call(dev, file->index, file->fd);
996a8ac8858SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_CALL, file);
997a8ac8858SEugenio Pérez }
998a8ac8858SEugenio Pérez 
999d96be4c8SEugenio Pérez static int vhost_vdpa_set_vring_dev_addr(struct vhost_dev *dev,
1000d96be4c8SEugenio Pérez                                          struct vhost_vring_addr *addr)
1001d96be4c8SEugenio Pérez {
1002d96be4c8SEugenio Pérez     trace_vhost_vdpa_set_vring_addr(dev, addr->index, addr->flags,
1003d96be4c8SEugenio Pérez                                 addr->desc_user_addr, addr->used_user_addr,
1004d96be4c8SEugenio Pérez                                 addr->avail_user_addr,
1005d96be4c8SEugenio Pérez                                 addr->log_guest_addr);
1006d96be4c8SEugenio Pérez 
1007d96be4c8SEugenio Pérez     return vhost_vdpa_call(dev, VHOST_SET_VRING_ADDR, addr);
1008d96be4c8SEugenio Pérez 
1009d96be4c8SEugenio Pérez }
1010d96be4c8SEugenio Pérez 
1011dff4426fSEugenio Pérez /**
1012dff4426fSEugenio Pérez  * Set the shadow virtqueue descriptors to the device
1013dff4426fSEugenio Pérez  *
1014dff4426fSEugenio Pérez  * @dev: The vhost device model
1015dff4426fSEugenio Pérez  * @svq: The shadow virtqueue
1016dff4426fSEugenio Pérez  * @idx: The index of the virtqueue in the vhost device
1017dff4426fSEugenio Pérez  * @errp: Error
1018a8ac8858SEugenio Pérez  *
1019a8ac8858SEugenio Pérez  * Note that this function does not rewind kick file descriptor if cannot set
1020a8ac8858SEugenio Pérez  * call one.
1021dff4426fSEugenio Pérez  */
1022100890f7SEugenio Pérez static int vhost_vdpa_svq_set_fds(struct vhost_dev *dev,
1023dff4426fSEugenio Pérez                                   VhostShadowVirtqueue *svq, unsigned idx,
1024dff4426fSEugenio Pérez                                   Error **errp)
1025dff4426fSEugenio Pérez {
1026dff4426fSEugenio Pérez     struct vhost_vring_file file = {
1027dff4426fSEugenio Pérez         .index = dev->vq_index + idx,
1028dff4426fSEugenio Pérez     };
1029dff4426fSEugenio Pérez     const EventNotifier *event_notifier = &svq->hdev_kick;
1030dff4426fSEugenio Pérez     int r;
1031dff4426fSEugenio Pérez 
10323cfb4d06SEugenio Pérez     r = event_notifier_init(&svq->hdev_kick, 0);
10333cfb4d06SEugenio Pérez     if (r != 0) {
10343cfb4d06SEugenio Pérez         error_setg_errno(errp, -r, "Couldn't create kick event notifier");
10353cfb4d06SEugenio Pérez         goto err_init_hdev_kick;
10363cfb4d06SEugenio Pérez     }
10373cfb4d06SEugenio Pérez 
10383cfb4d06SEugenio Pérez     r = event_notifier_init(&svq->hdev_call, 0);
10393cfb4d06SEugenio Pérez     if (r != 0) {
10403cfb4d06SEugenio Pérez         error_setg_errno(errp, -r, "Couldn't create call event notifier");
10413cfb4d06SEugenio Pérez         goto err_init_hdev_call;
10423cfb4d06SEugenio Pérez     }
10433cfb4d06SEugenio Pérez 
1044dff4426fSEugenio Pérez     file.fd = event_notifier_get_fd(event_notifier);
1045dff4426fSEugenio Pérez     r = vhost_vdpa_set_vring_dev_kick(dev, &file);
1046dff4426fSEugenio Pérez     if (unlikely(r != 0)) {
1047dff4426fSEugenio Pérez         error_setg_errno(errp, -r, "Can't set device kick fd");
10483cfb4d06SEugenio Pérez         goto err_init_set_dev_fd;
1049a8ac8858SEugenio Pérez     }
1050a8ac8858SEugenio Pérez 
1051a8ac8858SEugenio Pérez     event_notifier = &svq->hdev_call;
1052a8ac8858SEugenio Pérez     file.fd = event_notifier_get_fd(event_notifier);
1053a8ac8858SEugenio Pérez     r = vhost_vdpa_set_vring_dev_call(dev, &file);
1054a8ac8858SEugenio Pérez     if (unlikely(r != 0)) {
1055a8ac8858SEugenio Pérez         error_setg_errno(errp, -r, "Can't set device call fd");
10563cfb4d06SEugenio Pérez         goto err_init_set_dev_fd;
1057dff4426fSEugenio Pérez     }
1058dff4426fSEugenio Pérez 
10593cfb4d06SEugenio Pérez     return 0;
10603cfb4d06SEugenio Pérez 
10613cfb4d06SEugenio Pérez err_init_set_dev_fd:
10623cfb4d06SEugenio Pérez     event_notifier_set_handler(&svq->hdev_call, NULL);
10633cfb4d06SEugenio Pérez 
10643cfb4d06SEugenio Pérez err_init_hdev_call:
10653cfb4d06SEugenio Pérez     event_notifier_cleanup(&svq->hdev_kick);
10663cfb4d06SEugenio Pérez 
10673cfb4d06SEugenio Pérez err_init_hdev_kick:
1068100890f7SEugenio Pérez     return r;
1069100890f7SEugenio Pérez }
1070100890f7SEugenio Pérez 
1071100890f7SEugenio Pérez /**
1072100890f7SEugenio Pérez  * Unmap a SVQ area in the device
1073100890f7SEugenio Pérez  */
10748b6d6119SEugenio Pérez static void vhost_vdpa_svq_unmap_ring(struct vhost_vdpa *v, hwaddr addr)
1075100890f7SEugenio Pérez {
10768b6d6119SEugenio Pérez     const DMAMap needle = {
10778b6d6119SEugenio Pérez         .translated_addr = addr,
10788b6d6119SEugenio Pérez     };
10795edb02e8SEugenio Pérez     const DMAMap *result = vhost_iova_tree_find_iova(v->shared->iova_tree,
10805edb02e8SEugenio Pérez                                                      &needle);
108134e3c94eSEugenio Pérez     hwaddr size;
1082100890f7SEugenio Pérez     int r;
1083100890f7SEugenio Pérez 
108434e3c94eSEugenio Pérez     if (unlikely(!result)) {
108534e3c94eSEugenio Pérez         error_report("Unable to find SVQ address to unmap");
10865b590f51SEugenio Pérez         return;
108734e3c94eSEugenio Pérez     }
108834e3c94eSEugenio Pérez 
10898e3b0cbbSMarc-André Lureau     size = ROUND_UP(result->size, qemu_real_host_page_size());
10906f03d9efSEugenio Pérez     r = vhost_vdpa_dma_unmap(v->shared, v->address_space_id, result->iova,
10916f03d9efSEugenio Pérez                              size);
1092b37c12beSEugenio Pérez     if (unlikely(r < 0)) {
1093b37c12beSEugenio Pérez         error_report("Unable to unmap SVQ vring: %s (%d)", g_strerror(-r), -r);
10945b590f51SEugenio Pérez         return;
1095b37c12beSEugenio Pérez     }
1096b37c12beSEugenio Pérez 
10975edb02e8SEugenio Pérez     vhost_iova_tree_remove(v->shared->iova_tree, *result);
1098100890f7SEugenio Pérez }
1099100890f7SEugenio Pérez 
11005b590f51SEugenio Pérez static void vhost_vdpa_svq_unmap_rings(struct vhost_dev *dev,
1101100890f7SEugenio Pérez                                        const VhostShadowVirtqueue *svq)
1102100890f7SEugenio Pérez {
1103100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1104100890f7SEugenio Pérez     struct vhost_vring_addr svq_addr;
1105100890f7SEugenio Pérez 
1106100890f7SEugenio Pérez     vhost_svq_get_vring_addr(svq, &svq_addr);
1107100890f7SEugenio Pérez 
11088b6d6119SEugenio Pérez     vhost_vdpa_svq_unmap_ring(v, svq_addr.desc_user_addr);
1109100890f7SEugenio Pérez 
11108b6d6119SEugenio Pérez     vhost_vdpa_svq_unmap_ring(v, svq_addr.used_user_addr);
111134e3c94eSEugenio Pérez }
111234e3c94eSEugenio Pérez 
111334e3c94eSEugenio Pérez /**
111434e3c94eSEugenio Pérez  * Map the SVQ area in the device
111534e3c94eSEugenio Pérez  *
111634e3c94eSEugenio Pérez  * @v: Vhost-vdpa device
111734e3c94eSEugenio Pérez  * @needle: The area to search iova
111834e3c94eSEugenio Pérez  * @errorp: Error pointer
111934e3c94eSEugenio Pérez  */
112034e3c94eSEugenio Pérez static bool vhost_vdpa_svq_map_ring(struct vhost_vdpa *v, DMAMap *needle,
112134e3c94eSEugenio Pérez                                     Error **errp)
112234e3c94eSEugenio Pérez {
112334e3c94eSEugenio Pérez     int r;
112434e3c94eSEugenio Pérez 
11255edb02e8SEugenio Pérez     r = vhost_iova_tree_map_alloc(v->shared->iova_tree, needle);
112634e3c94eSEugenio Pérez     if (unlikely(r != IOVA_OK)) {
112734e3c94eSEugenio Pérez         error_setg(errp, "Cannot allocate iova (%d)", r);
112834e3c94eSEugenio Pérez         return false;
112934e3c94eSEugenio Pérez     }
113034e3c94eSEugenio Pérez 
11316f03d9efSEugenio Pérez     r = vhost_vdpa_dma_map(v->shared, v->address_space_id, needle->iova,
1132cd831ed5SEugenio Pérez                            needle->size + 1,
113334e3c94eSEugenio Pérez                            (void *)(uintptr_t)needle->translated_addr,
113434e3c94eSEugenio Pérez                            needle->perm == IOMMU_RO);
113534e3c94eSEugenio Pérez     if (unlikely(r != 0)) {
113634e3c94eSEugenio Pérez         error_setg_errno(errp, -r, "Cannot map region to device");
11375edb02e8SEugenio Pérez         vhost_iova_tree_remove(v->shared->iova_tree, *needle);
113834e3c94eSEugenio Pérez     }
113934e3c94eSEugenio Pérez 
114034e3c94eSEugenio Pérez     return r == 0;
1141100890f7SEugenio Pérez }
1142100890f7SEugenio Pérez 
1143100890f7SEugenio Pérez /**
1144100890f7SEugenio Pérez  * Map the shadow virtqueue rings in the device
1145100890f7SEugenio Pérez  *
1146100890f7SEugenio Pérez  * @dev: The vhost device
1147100890f7SEugenio Pérez  * @svq: The shadow virtqueue
1148100890f7SEugenio Pérez  * @addr: Assigned IOVA addresses
1149100890f7SEugenio Pérez  * @errp: Error pointer
1150100890f7SEugenio Pérez  */
1151100890f7SEugenio Pérez static bool vhost_vdpa_svq_map_rings(struct vhost_dev *dev,
1152100890f7SEugenio Pérez                                      const VhostShadowVirtqueue *svq,
1153100890f7SEugenio Pérez                                      struct vhost_vring_addr *addr,
1154100890f7SEugenio Pérez                                      Error **errp)
1155100890f7SEugenio Pérez {
115605e385d2SMarkus Armbruster     ERRP_GUARD();
115734e3c94eSEugenio Pérez     DMAMap device_region, driver_region;
115834e3c94eSEugenio Pérez     struct vhost_vring_addr svq_addr;
1159100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1160100890f7SEugenio Pérez     size_t device_size = vhost_svq_device_area_size(svq);
1161100890f7SEugenio Pérez     size_t driver_size = vhost_svq_driver_area_size(svq);
116234e3c94eSEugenio Pérez     size_t avail_offset;
116334e3c94eSEugenio Pérez     bool ok;
1164100890f7SEugenio Pérez 
116534e3c94eSEugenio Pérez     vhost_svq_get_vring_addr(svq, &svq_addr);
1166100890f7SEugenio Pérez 
116734e3c94eSEugenio Pérez     driver_region = (DMAMap) {
116834e3c94eSEugenio Pérez         .translated_addr = svq_addr.desc_user_addr,
116934e3c94eSEugenio Pérez         .size = driver_size - 1,
117034e3c94eSEugenio Pérez         .perm = IOMMU_RO,
117134e3c94eSEugenio Pérez     };
117234e3c94eSEugenio Pérez     ok = vhost_vdpa_svq_map_ring(v, &driver_region, errp);
117334e3c94eSEugenio Pérez     if (unlikely(!ok)) {
117434e3c94eSEugenio Pérez         error_prepend(errp, "Cannot create vq driver region: ");
1175100890f7SEugenio Pérez         return false;
1176100890f7SEugenio Pérez     }
117734e3c94eSEugenio Pérez     addr->desc_user_addr = driver_region.iova;
117834e3c94eSEugenio Pérez     avail_offset = svq_addr.avail_user_addr - svq_addr.desc_user_addr;
117934e3c94eSEugenio Pérez     addr->avail_user_addr = driver_region.iova + avail_offset;
1180100890f7SEugenio Pérez 
118134e3c94eSEugenio Pérez     device_region = (DMAMap) {
118234e3c94eSEugenio Pérez         .translated_addr = svq_addr.used_user_addr,
118334e3c94eSEugenio Pérez         .size = device_size - 1,
118434e3c94eSEugenio Pérez         .perm = IOMMU_RW,
118534e3c94eSEugenio Pérez     };
118634e3c94eSEugenio Pérez     ok = vhost_vdpa_svq_map_ring(v, &device_region, errp);
118734e3c94eSEugenio Pérez     if (unlikely(!ok)) {
118834e3c94eSEugenio Pérez         error_prepend(errp, "Cannot create vq device region: ");
11898b6d6119SEugenio Pérez         vhost_vdpa_svq_unmap_ring(v, driver_region.translated_addr);
1190100890f7SEugenio Pérez     }
119134e3c94eSEugenio Pérez     addr->used_user_addr = device_region.iova;
1192100890f7SEugenio Pérez 
119334e3c94eSEugenio Pérez     return ok;
1194100890f7SEugenio Pérez }
1195100890f7SEugenio Pérez 
1196100890f7SEugenio Pérez static bool vhost_vdpa_svq_setup(struct vhost_dev *dev,
1197100890f7SEugenio Pérez                                  VhostShadowVirtqueue *svq, unsigned idx,
1198100890f7SEugenio Pérez                                  Error **errp)
1199100890f7SEugenio Pérez {
1200100890f7SEugenio Pérez     uint16_t vq_index = dev->vq_index + idx;
1201100890f7SEugenio Pérez     struct vhost_vring_state s = {
1202100890f7SEugenio Pérez         .index = vq_index,
1203100890f7SEugenio Pérez     };
1204100890f7SEugenio Pérez     int r;
1205100890f7SEugenio Pérez 
1206100890f7SEugenio Pérez     r = vhost_vdpa_set_dev_vring_base(dev, &s);
1207100890f7SEugenio Pérez     if (unlikely(r)) {
1208100890f7SEugenio Pérez         error_setg_errno(errp, -r, "Cannot set vring base");
1209100890f7SEugenio Pérez         return false;
1210100890f7SEugenio Pérez     }
1211100890f7SEugenio Pérez 
1212100890f7SEugenio Pérez     r = vhost_vdpa_svq_set_fds(dev, svq, idx, errp);
1213dff4426fSEugenio Pérez     return r == 0;
1214dff4426fSEugenio Pérez }
1215dff4426fSEugenio Pérez 
1216dff4426fSEugenio Pérez static bool vhost_vdpa_svqs_start(struct vhost_dev *dev)
1217dff4426fSEugenio Pérez {
1218dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1219dff4426fSEugenio Pérez     Error *err = NULL;
1220dff4426fSEugenio Pérez     unsigned i;
1221dff4426fSEugenio Pérez 
1222712c1a31SEugenio Pérez     if (!v->shadow_vqs_enabled) {
1223dff4426fSEugenio Pérez         return true;
1224dff4426fSEugenio Pérez     }
1225dff4426fSEugenio Pérez 
1226dff4426fSEugenio Pérez     for (i = 0; i < v->shadow_vqs->len; ++i) {
1227100890f7SEugenio Pérez         VirtQueue *vq = virtio_get_queue(dev->vdev, dev->vq_index + i);
1228dff4426fSEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, i);
1229100890f7SEugenio Pérez         struct vhost_vring_addr addr = {
12301c82fdfeSEugenio Pérez             .index = dev->vq_index + i,
1231100890f7SEugenio Pérez         };
1232100890f7SEugenio Pérez         int r;
1233dff4426fSEugenio Pérez         bool ok = vhost_vdpa_svq_setup(dev, svq, i, &err);
1234dff4426fSEugenio Pérez         if (unlikely(!ok)) {
1235100890f7SEugenio Pérez             goto err;
1236100890f7SEugenio Pérez         }
1237100890f7SEugenio Pérez 
12385edb02e8SEugenio Pérez         vhost_svq_start(svq, dev->vdev, vq, v->shared->iova_tree);
1239100890f7SEugenio Pérez         ok = vhost_vdpa_svq_map_rings(dev, svq, &addr, &err);
1240100890f7SEugenio Pérez         if (unlikely(!ok)) {
1241100890f7SEugenio Pérez             goto err_map;
1242100890f7SEugenio Pérez         }
1243100890f7SEugenio Pérez 
1244100890f7SEugenio Pérez         /* Override vring GPA set by vhost subsystem */
1245100890f7SEugenio Pérez         r = vhost_vdpa_set_vring_dev_addr(dev, &addr);
1246100890f7SEugenio Pérez         if (unlikely(r != 0)) {
1247100890f7SEugenio Pérez             error_setg_errno(&err, -r, "Cannot set device address");
1248100890f7SEugenio Pérez             goto err_set_addr;
1249100890f7SEugenio Pérez         }
1250100890f7SEugenio Pérez     }
1251100890f7SEugenio Pérez 
1252100890f7SEugenio Pérez     return true;
1253100890f7SEugenio Pérez 
1254100890f7SEugenio Pérez err_set_addr:
1255100890f7SEugenio Pérez     vhost_vdpa_svq_unmap_rings(dev, g_ptr_array_index(v->shadow_vqs, i));
1256100890f7SEugenio Pérez 
1257100890f7SEugenio Pérez err_map:
1258100890f7SEugenio Pérez     vhost_svq_stop(g_ptr_array_index(v->shadow_vqs, i));
1259100890f7SEugenio Pérez 
1260100890f7SEugenio Pérez err:
1261dff4426fSEugenio Pérez     error_reportf_err(err, "Cannot setup SVQ %u: ", i);
1262100890f7SEugenio Pérez     for (unsigned j = 0; j < i; ++j) {
1263100890f7SEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, j);
1264100890f7SEugenio Pérez         vhost_vdpa_svq_unmap_rings(dev, svq);
1265100890f7SEugenio Pérez         vhost_svq_stop(svq);
1266100890f7SEugenio Pérez     }
1267100890f7SEugenio Pérez 
1268100890f7SEugenio Pérez     return false;
1269100890f7SEugenio Pérez }
1270100890f7SEugenio Pérez 
12715b590f51SEugenio Pérez static void vhost_vdpa_svqs_stop(struct vhost_dev *dev)
1272100890f7SEugenio Pérez {
1273100890f7SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1274100890f7SEugenio Pérez 
1275712c1a31SEugenio Pérez     if (!v->shadow_vqs_enabled) {
12765b590f51SEugenio Pérez         return;
1277100890f7SEugenio Pérez     }
1278100890f7SEugenio Pérez 
1279100890f7SEugenio Pérez     for (unsigned i = 0; i < v->shadow_vqs->len; ++i) {
1280100890f7SEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, i);
12812e1a9de9SEugenio Pérez 
12822e1a9de9SEugenio Pérez         vhost_svq_stop(svq);
12835b590f51SEugenio Pérez         vhost_vdpa_svq_unmap_rings(dev, svq);
12843cfb4d06SEugenio Pérez 
12853cfb4d06SEugenio Pérez         event_notifier_cleanup(&svq->hdev_kick);
12863cfb4d06SEugenio Pérez         event_notifier_cleanup(&svq->hdev_call);
1287dff4426fSEugenio Pérez     }
1288dff4426fSEugenio Pérez }
1289dff4426fSEugenio Pérez 
12900bb302a9SEugenio Pérez static void vhost_vdpa_suspend(struct vhost_dev *dev)
12910bb302a9SEugenio Pérez {
12920bb302a9SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
12930bb302a9SEugenio Pérez     int r;
12940bb302a9SEugenio Pérez 
12950bb302a9SEugenio Pérez     if (!vhost_vdpa_first_dev(dev)) {
12960bb302a9SEugenio Pérez         return;
12970bb302a9SEugenio Pérez     }
12980bb302a9SEugenio Pérez 
12990bb302a9SEugenio Pérez     if (dev->backend_cap & BIT_ULL(VHOST_BACKEND_F_SUSPEND)) {
13000bb302a9SEugenio Pérez         trace_vhost_vdpa_suspend(dev);
1301f12b2498SEugenio Pérez         r = ioctl(v->shared->device_fd, VHOST_VDPA_SUSPEND);
13020bb302a9SEugenio Pérez         if (unlikely(r)) {
13030bb302a9SEugenio Pérez             error_report("Cannot suspend: %s(%d)", g_strerror(errno), errno);
13040bb302a9SEugenio Pérez         } else {
13050bb302a9SEugenio Pérez             v->suspended = true;
13060bb302a9SEugenio Pérez             return;
13070bb302a9SEugenio Pérez         }
13080bb302a9SEugenio Pérez     }
13090bb302a9SEugenio Pérez 
13100bb302a9SEugenio Pérez     vhost_vdpa_reset_device(dev);
13110bb302a9SEugenio Pérez }
13120bb302a9SEugenio Pérez 
1313108a6481SCindy Lu static int vhost_vdpa_dev_start(struct vhost_dev *dev, bool started)
1314108a6481SCindy Lu {
1315108a6481SCindy Lu     struct vhost_vdpa *v = dev->opaque;
1316dff4426fSEugenio Pérez     bool ok;
1317778e67deSLaurent Vivier     trace_vhost_vdpa_dev_start(dev, started);
13184d191cfdSJason Wang 
13194d191cfdSJason Wang     if (started) {
13204d191cfdSJason Wang         vhost_vdpa_host_notifiers_init(dev);
1321dff4426fSEugenio Pérez         ok = vhost_vdpa_svqs_start(dev);
1322dff4426fSEugenio Pérez         if (unlikely(!ok)) {
1323dff4426fSEugenio Pérez             return -1;
1324dff4426fSEugenio Pérez         }
13254d191cfdSJason Wang     } else {
13260bb302a9SEugenio Pérez         vhost_vdpa_suspend(dev);
13275b590f51SEugenio Pérez         vhost_vdpa_svqs_stop(dev);
13284d191cfdSJason Wang         vhost_vdpa_host_notifiers_uninit(dev, dev->nvqs);
13294d191cfdSJason Wang     }
13304d191cfdSJason Wang 
1331c812b065SSi-Wei Liu     if (!vhost_vdpa_last_dev(dev)) {
13324d191cfdSJason Wang         return 0;
13334d191cfdSJason Wang     }
13344d191cfdSJason Wang 
1335108a6481SCindy Lu     if (started) {
1336bc7b0cacSCindy Lu         if (vhost_dev_has_iommu(dev) && (v->shadow_vqs_enabled)) {
1337bc7b0cacSCindy Lu             error_report("SVQ can not work while IOMMU enable, please disable"
1338bc7b0cacSCindy Lu                          "IOMMU and try again");
1339bc7b0cacSCindy Lu             return -1;
1340bc7b0cacSCindy Lu         }
1341f6fe3e33SEugenio Pérez         memory_listener_register(&v->shared->listener, dev->vdev->dma_as);
1342bc7b0cacSCindy Lu 
13433631151bSRoman Kagan         return vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_DRIVER_OK);
1344c3716f26SEugenio Pérez     }
1345c3716f26SEugenio Pérez 
1346c3716f26SEugenio Pérez     return 0;
1347c3716f26SEugenio Pérez }
1348c3716f26SEugenio Pérez 
1349c3716f26SEugenio Pérez static void vhost_vdpa_reset_status(struct vhost_dev *dev)
1350c3716f26SEugenio Pérez {
1351c3716f26SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1352c3716f26SEugenio Pérez 
1353c812b065SSi-Wei Liu     if (!vhost_vdpa_last_dev(dev)) {
1354c3716f26SEugenio Pérez         return;
1355c3716f26SEugenio Pérez     }
1356c3716f26SEugenio Pérez 
1357108a6481SCindy Lu     vhost_vdpa_reset_device(dev);
1358108a6481SCindy Lu     vhost_vdpa_add_status(dev, VIRTIO_CONFIG_S_ACKNOWLEDGE |
1359108a6481SCindy Lu                                VIRTIO_CONFIG_S_DRIVER);
1360f6fe3e33SEugenio Pérez     memory_listener_unregister(&v->shared->listener);
1361108a6481SCindy Lu }
1362108a6481SCindy Lu 
1363108a6481SCindy Lu static int vhost_vdpa_set_log_base(struct vhost_dev *dev, uint64_t base,
1364108a6481SCindy Lu                                      struct vhost_log *log)
1365108a6481SCindy Lu {
1366773ebc95SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1367d71b0609SSi-Wei Liu     if (v->shadow_vqs_enabled || !vhost_vdpa_first_dev(dev)) {
13684d191cfdSJason Wang         return 0;
13694d191cfdSJason Wang     }
13704d191cfdSJason Wang 
1371778e67deSLaurent Vivier     trace_vhost_vdpa_set_log_base(dev, base, log->size, log->refcnt, log->fd,
1372778e67deSLaurent Vivier                                   log->log);
1373108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_LOG_BASE, &base);
1374108a6481SCindy Lu }
1375108a6481SCindy Lu 
1376108a6481SCindy Lu static int vhost_vdpa_set_vring_addr(struct vhost_dev *dev,
1377108a6481SCindy Lu                                        struct vhost_vring_addr *addr)
1378108a6481SCindy Lu {
1379d96be4c8SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1380d96be4c8SEugenio Pérez 
1381d96be4c8SEugenio Pérez     if (v->shadow_vqs_enabled) {
1382d96be4c8SEugenio Pérez         /*
1383d96be4c8SEugenio Pérez          * Device vring addr was set at device start. SVQ base is handled by
1384d96be4c8SEugenio Pérez          * VirtQueue code.
1385d96be4c8SEugenio Pérez          */
1386d96be4c8SEugenio Pérez         return 0;
1387d96be4c8SEugenio Pérez     }
1388d96be4c8SEugenio Pérez 
1389d96be4c8SEugenio Pérez     return vhost_vdpa_set_vring_dev_addr(dev, addr);
1390108a6481SCindy Lu }
1391108a6481SCindy Lu 
1392108a6481SCindy Lu static int vhost_vdpa_set_vring_num(struct vhost_dev *dev,
1393108a6481SCindy Lu                                       struct vhost_vring_state *ring)
1394108a6481SCindy Lu {
1395778e67deSLaurent Vivier     trace_vhost_vdpa_set_vring_num(dev, ring->index, ring->num);
1396108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_VRING_NUM, ring);
1397108a6481SCindy Lu }
1398108a6481SCindy Lu 
1399108a6481SCindy Lu static int vhost_vdpa_set_vring_base(struct vhost_dev *dev,
1400108a6481SCindy Lu                                        struct vhost_vring_state *ring)
1401108a6481SCindy Lu {
1402d96be4c8SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1403d96be4c8SEugenio Pérez 
1404d96be4c8SEugenio Pérez     if (v->shadow_vqs_enabled) {
1405d96be4c8SEugenio Pérez         /*
1406d96be4c8SEugenio Pérez          * Device vring base was set at device start. SVQ base is handled by
1407d96be4c8SEugenio Pérez          * VirtQueue code.
1408d96be4c8SEugenio Pérez          */
1409d96be4c8SEugenio Pérez         return 0;
1410d96be4c8SEugenio Pérez     }
1411d96be4c8SEugenio Pérez 
1412d96be4c8SEugenio Pérez     return vhost_vdpa_set_dev_vring_base(dev, ring);
1413108a6481SCindy Lu }
1414108a6481SCindy Lu 
1415108a6481SCindy Lu static int vhost_vdpa_get_vring_base(struct vhost_dev *dev,
1416108a6481SCindy Lu                                        struct vhost_vring_state *ring)
1417108a6481SCindy Lu {
14186d0b2226SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1419778e67deSLaurent Vivier     int ret;
1420778e67deSLaurent Vivier 
14216d0b2226SEugenio Pérez     if (v->shadow_vqs_enabled) {
14222fdac348SEugenio Pérez         ring->num = virtio_queue_get_last_avail_idx(dev->vdev, ring->index);
14236ec0a746SSi-Wei Liu         trace_vhost_vdpa_get_vring_base(dev, ring->index, ring->num, true);
14246d0b2226SEugenio Pérez         return 0;
14256d0b2226SEugenio Pérez     }
14266d0b2226SEugenio Pérez 
1427b6662cb7SEugenio Pérez     if (!v->suspended) {
1428b6662cb7SEugenio Pérez         /*
1429b6662cb7SEugenio Pérez          * Cannot trust in value returned by device, let vhost recover used
1430b6662cb7SEugenio Pérez          * idx from guest.
1431b6662cb7SEugenio Pérez          */
1432b6662cb7SEugenio Pérez         return -1;
1433b6662cb7SEugenio Pérez     }
1434b6662cb7SEugenio Pérez 
1435778e67deSLaurent Vivier     ret = vhost_vdpa_call(dev, VHOST_GET_VRING_BASE, ring);
14366ec0a746SSi-Wei Liu     trace_vhost_vdpa_get_vring_base(dev, ring->index, ring->num, false);
1437778e67deSLaurent Vivier     return ret;
1438108a6481SCindy Lu }
1439108a6481SCindy Lu 
1440108a6481SCindy Lu static int vhost_vdpa_set_vring_kick(struct vhost_dev *dev,
1441108a6481SCindy Lu                                        struct vhost_vring_file *file)
1442108a6481SCindy Lu {
1443dff4426fSEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1444dff4426fSEugenio Pérez     int vdpa_idx = file->index - dev->vq_index;
1445dff4426fSEugenio Pérez 
1446dff4426fSEugenio Pérez     if (v->shadow_vqs_enabled) {
1447dff4426fSEugenio Pérez         VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, vdpa_idx);
1448dff4426fSEugenio Pérez         vhost_svq_set_svq_kick_fd(svq, file->fd);
1449dff4426fSEugenio Pérez         return 0;
1450dff4426fSEugenio Pérez     } else {
1451dff4426fSEugenio Pérez         return vhost_vdpa_set_vring_dev_kick(dev, file);
1452dff4426fSEugenio Pérez     }
1453108a6481SCindy Lu }
1454108a6481SCindy Lu 
1455108a6481SCindy Lu static int vhost_vdpa_set_vring_call(struct vhost_dev *dev,
1456108a6481SCindy Lu                                        struct vhost_vring_file *file)
1457108a6481SCindy Lu {
1458a8ac8858SEugenio Pérez     struct vhost_vdpa *v = dev->opaque;
1459a8ac8858SEugenio Pérez     int vdpa_idx = file->index - dev->vq_index;
1460a8ac8858SEugenio Pérez     VhostShadowVirtqueue *svq = g_ptr_array_index(v->shadow_vqs, vdpa_idx);
1461a8ac8858SEugenio Pérez 
1462b2765243SEugenio Pérez     /* Remember last call fd because we can switch to SVQ anytime. */
1463a8ac8858SEugenio Pérez     vhost_svq_set_svq_call_fd(svq, file->fd);
14649ed62809SSi-Wei Liu     /*
14659ed62809SSi-Wei Liu      * When SVQ is transitioning to off, shadow_vqs_enabled has
14669ed62809SSi-Wei Liu      * not been set back to false yet, but the underlying call fd
14679ed62809SSi-Wei Liu      * will have to switch back to the guest notifier to signal the
14689ed62809SSi-Wei Liu      * passthrough virtqueues. In other situations, SVQ's own call
14699ed62809SSi-Wei Liu      * fd shall be used to signal the device model.
14709ed62809SSi-Wei Liu      */
14719ed62809SSi-Wei Liu     if (v->shadow_vqs_enabled &&
14729ed62809SSi-Wei Liu         v->shared->svq_switching != SVQ_TSTATE_DISABLING) {
1473a8ac8858SEugenio Pérez         return 0;
1474a8ac8858SEugenio Pérez     }
1475b2765243SEugenio Pérez 
1476b2765243SEugenio Pérez     return vhost_vdpa_set_vring_dev_call(dev, file);
1477108a6481SCindy Lu }
1478108a6481SCindy Lu 
1479108a6481SCindy Lu static int vhost_vdpa_get_features(struct vhost_dev *dev,
1480108a6481SCindy Lu                                      uint64_t *features)
1481108a6481SCindy Lu {
148212a195faSEugenio Pérez     int ret = vhost_vdpa_get_dev_features(dev, features);
1483778e67deSLaurent Vivier 
1484ab7337e3SEugenio Pérez     if (ret == 0) {
148512a195faSEugenio Pérez         /* Add SVQ logging capabilities */
148612a195faSEugenio Pérez         *features |= BIT_ULL(VHOST_F_LOG_ALL);
148712a195faSEugenio Pérez     }
148812a195faSEugenio Pérez 
1489778e67deSLaurent Vivier     return ret;
1490108a6481SCindy Lu }
1491108a6481SCindy Lu 
1492108a6481SCindy Lu static int vhost_vdpa_set_owner(struct vhost_dev *dev)
1493108a6481SCindy Lu {
1494d71b0609SSi-Wei Liu     if (!vhost_vdpa_first_dev(dev)) {
14954d191cfdSJason Wang         return 0;
14964d191cfdSJason Wang     }
14974d191cfdSJason Wang 
1498778e67deSLaurent Vivier     trace_vhost_vdpa_set_owner(dev);
1499108a6481SCindy Lu     return vhost_vdpa_call(dev, VHOST_SET_OWNER, NULL);
1500108a6481SCindy Lu }
1501108a6481SCindy Lu 
1502108a6481SCindy Lu static int vhost_vdpa_vq_get_addr(struct vhost_dev *dev,
1503108a6481SCindy Lu                     struct vhost_vring_addr *addr, struct vhost_virtqueue *vq)
1504108a6481SCindy Lu {
1505108a6481SCindy Lu     assert(dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_VDPA);
1506108a6481SCindy Lu     addr->desc_user_addr = (uint64_t)(unsigned long)vq->desc_phys;
1507108a6481SCindy Lu     addr->avail_user_addr = (uint64_t)(unsigned long)vq->avail_phys;
1508108a6481SCindy Lu     addr->used_user_addr = (uint64_t)(unsigned long)vq->used_phys;
1509778e67deSLaurent Vivier     trace_vhost_vdpa_vq_get_addr(dev, vq, addr->desc_user_addr,
1510778e67deSLaurent Vivier                                  addr->avail_user_addr, addr->used_user_addr);
1511108a6481SCindy Lu     return 0;
1512108a6481SCindy Lu }
1513108a6481SCindy Lu 
1514108a6481SCindy Lu static bool  vhost_vdpa_force_iommu(struct vhost_dev *dev)
1515108a6481SCindy Lu {
1516108a6481SCindy Lu     return true;
1517108a6481SCindy Lu }
1518108a6481SCindy Lu 
1519108a6481SCindy Lu const VhostOps vdpa_ops = {
1520108a6481SCindy Lu         .backend_type = VHOST_BACKEND_TYPE_VDPA,
1521108a6481SCindy Lu         .vhost_backend_init = vhost_vdpa_init,
1522108a6481SCindy Lu         .vhost_backend_cleanup = vhost_vdpa_cleanup,
1523108a6481SCindy Lu         .vhost_set_log_base = vhost_vdpa_set_log_base,
1524108a6481SCindy Lu         .vhost_set_vring_addr = vhost_vdpa_set_vring_addr,
1525108a6481SCindy Lu         .vhost_set_vring_num = vhost_vdpa_set_vring_num,
1526108a6481SCindy Lu         .vhost_set_vring_base = vhost_vdpa_set_vring_base,
1527108a6481SCindy Lu         .vhost_get_vring_base = vhost_vdpa_get_vring_base,
1528108a6481SCindy Lu         .vhost_set_vring_kick = vhost_vdpa_set_vring_kick,
1529108a6481SCindy Lu         .vhost_set_vring_call = vhost_vdpa_set_vring_call,
1530108a6481SCindy Lu         .vhost_get_features = vhost_vdpa_get_features,
1531a5bd0580SJason Wang         .vhost_set_backend_cap = vhost_vdpa_set_backend_cap,
1532108a6481SCindy Lu         .vhost_set_owner = vhost_vdpa_set_owner,
1533108a6481SCindy Lu         .vhost_set_vring_endian = NULL,
1534108a6481SCindy Lu         .vhost_backend_memslots_limit = vhost_vdpa_memslots_limit,
1535108a6481SCindy Lu         .vhost_set_mem_table = vhost_vdpa_set_mem_table,
1536108a6481SCindy Lu         .vhost_set_features = vhost_vdpa_set_features,
1537108a6481SCindy Lu         .vhost_reset_device = vhost_vdpa_reset_device,
1538108a6481SCindy Lu         .vhost_get_vq_index = vhost_vdpa_get_vq_index,
1539108a6481SCindy Lu         .vhost_get_config  = vhost_vdpa_get_config,
1540108a6481SCindy Lu         .vhost_set_config = vhost_vdpa_set_config,
1541108a6481SCindy Lu         .vhost_requires_shm_log = NULL,
1542108a6481SCindy Lu         .vhost_migration_done = NULL,
1543108a6481SCindy Lu         .vhost_net_set_mtu = NULL,
1544108a6481SCindy Lu         .vhost_set_iotlb_callback = NULL,
1545108a6481SCindy Lu         .vhost_send_device_iotlb_msg = NULL,
1546108a6481SCindy Lu         .vhost_dev_start = vhost_vdpa_dev_start,
1547108a6481SCindy Lu         .vhost_get_device_id = vhost_vdpa_get_device_id,
1548108a6481SCindy Lu         .vhost_vq_get_addr = vhost_vdpa_vq_get_addr,
1549108a6481SCindy Lu         .vhost_force_iommu = vhost_vdpa_force_iommu,
1550259f3accSCindy Lu         .vhost_set_config_call = vhost_vdpa_set_config_call,
1551c3716f26SEugenio Pérez         .vhost_reset_status = vhost_vdpa_reset_status,
1552108a6481SCindy Lu };
1553