xref: /openbmc/linux/drivers/virtio/virtio_ring.c (revision 5bc72234f7c65830e60806dbb73ae76bacd8a061)
1fd534e9bSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-or-later
20a8a69ddSRusty Russell /* Virtio ring implementation.
30a8a69ddSRusty Russell  *
40a8a69ddSRusty Russell  *  Copyright 2007 Rusty Russell IBM Corporation
50a8a69ddSRusty Russell  */
60a8a69ddSRusty Russell #include <linux/virtio.h>
70a8a69ddSRusty Russell #include <linux/virtio_ring.h>
8e34f8725SRusty Russell #include <linux/virtio_config.h>
90a8a69ddSRusty Russell #include <linux/device.h>
105a0e3ad6STejun Heo #include <linux/slab.h>
11b5a2c4f1SPaul Gortmaker #include <linux/module.h>
12e93300b1SRusty Russell #include <linux/hrtimer.h>
13780bc790SAndy Lutomirski #include <linux/dma-mapping.h>
1478fe3987SAndy Lutomirski #include <xen/xen.h>
150a8a69ddSRusty Russell 
160a8a69ddSRusty Russell #ifdef DEBUG
170a8a69ddSRusty Russell /* For development, we want to crash whenever the ring is screwed. */
189499f5e7SRusty Russell #define BAD_RING(_vq, fmt, args...)				\
199499f5e7SRusty Russell 	do {							\
209499f5e7SRusty Russell 		dev_err(&(_vq)->vq.vdev->dev,			\
219499f5e7SRusty Russell 			"%s:"fmt, (_vq)->vq.name, ##args);	\
229499f5e7SRusty Russell 		BUG();						\
239499f5e7SRusty Russell 	} while (0)
24c5f841f1SRusty Russell /* Caller is supposed to guarantee no reentry. */
253a35ce7dSRoel Kluin #define START_USE(_vq)						\
26c5f841f1SRusty Russell 	do {							\
27c5f841f1SRusty Russell 		if ((_vq)->in_use)				\
289499f5e7SRusty Russell 			panic("%s:in_use = %i\n",		\
299499f5e7SRusty Russell 			      (_vq)->vq.name, (_vq)->in_use);	\
30c5f841f1SRusty Russell 		(_vq)->in_use = __LINE__;			\
31c5f841f1SRusty Russell 	} while (0)
323a35ce7dSRoel Kluin #define END_USE(_vq) \
3397a545abSRusty Russell 	do { BUG_ON(!(_vq)->in_use); (_vq)->in_use = 0; } while(0)
344d6a105eSTiwei Bie #define LAST_ADD_TIME_UPDATE(_vq)				\
354d6a105eSTiwei Bie 	do {							\
364d6a105eSTiwei Bie 		ktime_t now = ktime_get();			\
374d6a105eSTiwei Bie 								\
384d6a105eSTiwei Bie 		/* No kick or get, with .1 second between?  Warn. */ \
394d6a105eSTiwei Bie 		if ((_vq)->last_add_time_valid)			\
404d6a105eSTiwei Bie 			WARN_ON(ktime_to_ms(ktime_sub(now,	\
414d6a105eSTiwei Bie 				(_vq)->last_add_time)) > 100);	\
424d6a105eSTiwei Bie 		(_vq)->last_add_time = now;			\
434d6a105eSTiwei Bie 		(_vq)->last_add_time_valid = true;		\
444d6a105eSTiwei Bie 	} while (0)
454d6a105eSTiwei Bie #define LAST_ADD_TIME_CHECK(_vq)				\
464d6a105eSTiwei Bie 	do {							\
474d6a105eSTiwei Bie 		if ((_vq)->last_add_time_valid) {		\
484d6a105eSTiwei Bie 			WARN_ON(ktime_to_ms(ktime_sub(ktime_get(), \
494d6a105eSTiwei Bie 				      (_vq)->last_add_time)) > 100); \
504d6a105eSTiwei Bie 		}						\
514d6a105eSTiwei Bie 	} while (0)
524d6a105eSTiwei Bie #define LAST_ADD_TIME_INVALID(_vq)				\
534d6a105eSTiwei Bie 	((_vq)->last_add_time_valid = false)
540a8a69ddSRusty Russell #else
559499f5e7SRusty Russell #define BAD_RING(_vq, fmt, args...)				\
569499f5e7SRusty Russell 	do {							\
579499f5e7SRusty Russell 		dev_err(&_vq->vq.vdev->dev,			\
589499f5e7SRusty Russell 			"%s:"fmt, (_vq)->vq.name, ##args);	\
599499f5e7SRusty Russell 		(_vq)->broken = true;				\
609499f5e7SRusty Russell 	} while (0)
610a8a69ddSRusty Russell #define START_USE(vq)
620a8a69ddSRusty Russell #define END_USE(vq)
634d6a105eSTiwei Bie #define LAST_ADD_TIME_UPDATE(vq)
644d6a105eSTiwei Bie #define LAST_ADD_TIME_CHECK(vq)
654d6a105eSTiwei Bie #define LAST_ADD_TIME_INVALID(vq)
660a8a69ddSRusty Russell #endif
670a8a69ddSRusty Russell 
68cbeedb72STiwei Bie struct vring_desc_state_split {
69780bc790SAndy Lutomirski 	void *data;			/* Data for callback. */
70780bc790SAndy Lutomirski 	struct vring_desc *indir_desc;	/* Indirect descriptor, if any. */
71780bc790SAndy Lutomirski };
72780bc790SAndy Lutomirski 
731ce9e605STiwei Bie struct vring_desc_state_packed {
741ce9e605STiwei Bie 	void *data;			/* Data for callback. */
751ce9e605STiwei Bie 	struct vring_packed_desc *indir_desc; /* Indirect descriptor, if any. */
761ce9e605STiwei Bie 	u16 num;			/* Descriptor list length. */
771ce9e605STiwei Bie 	u16 last;			/* The last desc state in a list. */
781ce9e605STiwei Bie };
791ce9e605STiwei Bie 
801f28750fSJason Wang struct vring_desc_extra {
811ce9e605STiwei Bie 	dma_addr_t addr;		/* Buffer DMA addr. */
821ce9e605STiwei Bie 	u32 len;			/* Buffer length. */
831ce9e605STiwei Bie 	u16 flags;			/* Descriptor flags. */
84aeef9b47SJason Wang 	u16 next;			/* The next desc state in a list. */
851ce9e605STiwei Bie };
861ce9e605STiwei Bie 
8743b4f721SMichael S. Tsirkin struct vring_virtqueue {
880a8a69ddSRusty Russell 	struct virtqueue vq;
890a8a69ddSRusty Russell 
901ce9e605STiwei Bie 	/* Is this a packed ring? */
911ce9e605STiwei Bie 	bool packed_ring;
921ce9e605STiwei Bie 
93fb3fba6bSTiwei Bie 	/* Is DMA API used? */
94fb3fba6bSTiwei Bie 	bool use_dma_api;
95fb3fba6bSTiwei Bie 
967b21e34fSRusty Russell 	/* Can we use weak barriers? */
977b21e34fSRusty Russell 	bool weak_barriers;
987b21e34fSRusty Russell 
990a8a69ddSRusty Russell 	/* Other side has made a mess, don't try any more. */
1000a8a69ddSRusty Russell 	bool broken;
1010a8a69ddSRusty Russell 
1029fa29b9dSMark McLoughlin 	/* Host supports indirect buffers */
1039fa29b9dSMark McLoughlin 	bool indirect;
1049fa29b9dSMark McLoughlin 
105a5c262c5SMichael S. Tsirkin 	/* Host publishes avail event idx */
106a5c262c5SMichael S. Tsirkin 	bool event;
107a5c262c5SMichael S. Tsirkin 
1080a8a69ddSRusty Russell 	/* Head of free buffer list. */
1090a8a69ddSRusty Russell 	unsigned int free_head;
1100a8a69ddSRusty Russell 	/* Number we've added since last sync. */
1110a8a69ddSRusty Russell 	unsigned int num_added;
1120a8a69ddSRusty Russell 
1130a8a69ddSRusty Russell 	/* Last used index we've seen. */
1141bc4953eSAnthony Liguori 	u16 last_used_idx;
1150a8a69ddSRusty Russell 
1168d622d21SMichael S. Tsirkin 	/* Hint for event idx: already triggered no need to disable. */
1178d622d21SMichael S. Tsirkin 	bool event_triggered;
1188d622d21SMichael S. Tsirkin 
1191ce9e605STiwei Bie 	union {
1201ce9e605STiwei Bie 		/* Available for split ring */
121e593bf97STiwei Bie 		struct {
1221ce9e605STiwei Bie 			/* Actual memory layout for this queue. */
123e593bf97STiwei Bie 			struct vring vring;
124e593bf97STiwei Bie 
125f277ec42SVenkatesh Srinivas 			/* Last written value to avail->flags */
126f277ec42SVenkatesh Srinivas 			u16 avail_flags_shadow;
127f277ec42SVenkatesh Srinivas 
1281ce9e605STiwei Bie 			/*
1291ce9e605STiwei Bie 			 * Last written value to avail->idx in
1301ce9e605STiwei Bie 			 * guest byte order.
1311ce9e605STiwei Bie 			 */
132f277ec42SVenkatesh Srinivas 			u16 avail_idx_shadow;
133cbeedb72STiwei Bie 
134cbeedb72STiwei Bie 			/* Per-descriptor state. */
135cbeedb72STiwei Bie 			struct vring_desc_state_split *desc_state;
136d79dca75STiwei Bie 
1371ce9e605STiwei Bie 			/* DMA address and size information */
138d79dca75STiwei Bie 			dma_addr_t queue_dma_addr;
1391ce9e605STiwei Bie 			size_t queue_size_in_bytes;
140e593bf97STiwei Bie 		} split;
141f277ec42SVenkatesh Srinivas 
1421ce9e605STiwei Bie 		/* Available for packed ring */
1431ce9e605STiwei Bie 		struct {
1441ce9e605STiwei Bie 			/* Actual memory layout for this queue. */
1459c0644eeSMichael S. Tsirkin 			struct {
1469c0644eeSMichael S. Tsirkin 				unsigned int num;
1479c0644eeSMichael S. Tsirkin 				struct vring_packed_desc *desc;
1489c0644eeSMichael S. Tsirkin 				struct vring_packed_desc_event *driver;
1499c0644eeSMichael S. Tsirkin 				struct vring_packed_desc_event *device;
1509c0644eeSMichael S. Tsirkin 			} vring;
1511ce9e605STiwei Bie 
1521ce9e605STiwei Bie 			/* Driver ring wrap counter. */
1531ce9e605STiwei Bie 			bool avail_wrap_counter;
1541ce9e605STiwei Bie 
1551ce9e605STiwei Bie 			/* Device ring wrap counter. */
1561ce9e605STiwei Bie 			bool used_wrap_counter;
1571ce9e605STiwei Bie 
1581ce9e605STiwei Bie 			/* Avail used flags. */
1591ce9e605STiwei Bie 			u16 avail_used_flags;
1601ce9e605STiwei Bie 
1611ce9e605STiwei Bie 			/* Index of the next avail descriptor. */
1621ce9e605STiwei Bie 			u16 next_avail_idx;
1631ce9e605STiwei Bie 
1641ce9e605STiwei Bie 			/*
1651ce9e605STiwei Bie 			 * Last written value to driver->flags in
1661ce9e605STiwei Bie 			 * guest byte order.
1671ce9e605STiwei Bie 			 */
1681ce9e605STiwei Bie 			u16 event_flags_shadow;
1691ce9e605STiwei Bie 
1701ce9e605STiwei Bie 			/* Per-descriptor state. */
1711ce9e605STiwei Bie 			struct vring_desc_state_packed *desc_state;
1721f28750fSJason Wang 			struct vring_desc_extra *desc_extra;
1731ce9e605STiwei Bie 
1741ce9e605STiwei Bie 			/* DMA address and size information */
1751ce9e605STiwei Bie 			dma_addr_t ring_dma_addr;
1761ce9e605STiwei Bie 			dma_addr_t driver_event_dma_addr;
1771ce9e605STiwei Bie 			dma_addr_t device_event_dma_addr;
1781ce9e605STiwei Bie 			size_t ring_size_in_bytes;
1791ce9e605STiwei Bie 			size_t event_size_in_bytes;
1801ce9e605STiwei Bie 		} packed;
1811ce9e605STiwei Bie 	};
1821ce9e605STiwei Bie 
1830a8a69ddSRusty Russell 	/* How to notify other side. FIXME: commonalize hcalls! */
18446f9c2b9SHeinz Graalfs 	bool (*notify)(struct virtqueue *vq);
1850a8a69ddSRusty Russell 
1862a2d1382SAndy Lutomirski 	/* DMA, allocation, and size information */
1872a2d1382SAndy Lutomirski 	bool we_own_ring;
1882a2d1382SAndy Lutomirski 
1890a8a69ddSRusty Russell #ifdef DEBUG
1900a8a69ddSRusty Russell 	/* They're supposed to lock for us. */
1910a8a69ddSRusty Russell 	unsigned int in_use;
192e93300b1SRusty Russell 
193e93300b1SRusty Russell 	/* Figure out if their kicks are too delayed. */
194e93300b1SRusty Russell 	bool last_add_time_valid;
195e93300b1SRusty Russell 	ktime_t last_add_time;
1960a8a69ddSRusty Russell #endif
1970a8a69ddSRusty Russell };
1980a8a69ddSRusty Russell 
199e6f633e5STiwei Bie 
200e6f633e5STiwei Bie /*
201e6f633e5STiwei Bie  * Helpers.
202e6f633e5STiwei Bie  */
203e6f633e5STiwei Bie 
2040a8a69ddSRusty Russell #define to_vvq(_vq) container_of(_vq, struct vring_virtqueue, vq)
2050a8a69ddSRusty Russell 
2062f18c2d1STiwei Bie static inline bool virtqueue_use_indirect(struct virtqueue *_vq,
2072f18c2d1STiwei Bie 					  unsigned int total_sg)
2082f18c2d1STiwei Bie {
2092f18c2d1STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
2102f18c2d1STiwei Bie 
2112f18c2d1STiwei Bie 	/*
2122f18c2d1STiwei Bie 	 * If the host supports indirect descriptor tables, and we have multiple
2132f18c2d1STiwei Bie 	 * buffers, then go indirect. FIXME: tune this threshold
2142f18c2d1STiwei Bie 	 */
2152f18c2d1STiwei Bie 	return (vq->indirect && total_sg > 1 && vq->vq.num_free);
2162f18c2d1STiwei Bie }
2172f18c2d1STiwei Bie 
218d26c96c8SAndy Lutomirski /*
2191a937693SMichael S. Tsirkin  * Modern virtio devices have feature bits to specify whether they need a
2201a937693SMichael S. Tsirkin  * quirk and bypass the IOMMU. If not there, just use the DMA API.
2211a937693SMichael S. Tsirkin  *
2221a937693SMichael S. Tsirkin  * If there, the interaction between virtio and DMA API is messy.
223d26c96c8SAndy Lutomirski  *
224d26c96c8SAndy Lutomirski  * On most systems with virtio, physical addresses match bus addresses,
225d26c96c8SAndy Lutomirski  * and it doesn't particularly matter whether we use the DMA API.
226d26c96c8SAndy Lutomirski  *
227d26c96c8SAndy Lutomirski  * On some systems, including Xen and any system with a physical device
228d26c96c8SAndy Lutomirski  * that speaks virtio behind a physical IOMMU, we must use the DMA API
229d26c96c8SAndy Lutomirski  * for virtio DMA to work at all.
230d26c96c8SAndy Lutomirski  *
231d26c96c8SAndy Lutomirski  * On other systems, including SPARC and PPC64, virtio-pci devices are
232d26c96c8SAndy Lutomirski  * enumerated as though they are behind an IOMMU, but the virtio host
233d26c96c8SAndy Lutomirski  * ignores the IOMMU, so we must either pretend that the IOMMU isn't
234d26c96c8SAndy Lutomirski  * there or somehow map everything as the identity.
235d26c96c8SAndy Lutomirski  *
236d26c96c8SAndy Lutomirski  * For the time being, we preserve historic behavior and bypass the DMA
237d26c96c8SAndy Lutomirski  * API.
2381a937693SMichael S. Tsirkin  *
2391a937693SMichael S. Tsirkin  * TODO: install a per-device DMA ops structure that does the right thing
2401a937693SMichael S. Tsirkin  * taking into account all the above quirks, and use the DMA API
2411a937693SMichael S. Tsirkin  * unconditionally on data path.
242d26c96c8SAndy Lutomirski  */
243d26c96c8SAndy Lutomirski 
244d26c96c8SAndy Lutomirski static bool vring_use_dma_api(struct virtio_device *vdev)
245d26c96c8SAndy Lutomirski {
24624b6842aSMichael S. Tsirkin 	if (!virtio_has_dma_quirk(vdev))
2471a937693SMichael S. Tsirkin 		return true;
2481a937693SMichael S. Tsirkin 
2491a937693SMichael S. Tsirkin 	/* Otherwise, we are left to guess. */
25078fe3987SAndy Lutomirski 	/*
25178fe3987SAndy Lutomirski 	 * In theory, it's possible to have a buggy QEMU-supposed
25278fe3987SAndy Lutomirski 	 * emulated Q35 IOMMU and Xen enabled at the same time.  On
25378fe3987SAndy Lutomirski 	 * such a configuration, virtio has never worked and will
25478fe3987SAndy Lutomirski 	 * not work without an even larger kludge.  Instead, enable
25578fe3987SAndy Lutomirski 	 * the DMA API if we're a Xen guest, which at least allows
25678fe3987SAndy Lutomirski 	 * all of the sensible Xen configurations to work correctly.
25778fe3987SAndy Lutomirski 	 */
25878fe3987SAndy Lutomirski 	if (xen_domain())
25978fe3987SAndy Lutomirski 		return true;
26078fe3987SAndy Lutomirski 
261d26c96c8SAndy Lutomirski 	return false;
262d26c96c8SAndy Lutomirski }
263d26c96c8SAndy Lutomirski 
264e6d6dd6cSJoerg Roedel size_t virtio_max_dma_size(struct virtio_device *vdev)
265e6d6dd6cSJoerg Roedel {
266e6d6dd6cSJoerg Roedel 	size_t max_segment_size = SIZE_MAX;
267e6d6dd6cSJoerg Roedel 
268e6d6dd6cSJoerg Roedel 	if (vring_use_dma_api(vdev))
269e6d6dd6cSJoerg Roedel 		max_segment_size = dma_max_mapping_size(&vdev->dev);
270e6d6dd6cSJoerg Roedel 
271e6d6dd6cSJoerg Roedel 	return max_segment_size;
272e6d6dd6cSJoerg Roedel }
273e6d6dd6cSJoerg Roedel EXPORT_SYMBOL_GPL(virtio_max_dma_size);
274e6d6dd6cSJoerg Roedel 
275d79dca75STiwei Bie static void *vring_alloc_queue(struct virtio_device *vdev, size_t size,
276d79dca75STiwei Bie 			      dma_addr_t *dma_handle, gfp_t flag)
277d79dca75STiwei Bie {
278d79dca75STiwei Bie 	if (vring_use_dma_api(vdev)) {
279d79dca75STiwei Bie 		return dma_alloc_coherent(vdev->dev.parent, size,
280d79dca75STiwei Bie 					  dma_handle, flag);
281d79dca75STiwei Bie 	} else {
282d79dca75STiwei Bie 		void *queue = alloc_pages_exact(PAGE_ALIGN(size), flag);
283d79dca75STiwei Bie 
284d79dca75STiwei Bie 		if (queue) {
285d79dca75STiwei Bie 			phys_addr_t phys_addr = virt_to_phys(queue);
286d79dca75STiwei Bie 			*dma_handle = (dma_addr_t)phys_addr;
287d79dca75STiwei Bie 
288d79dca75STiwei Bie 			/*
289d79dca75STiwei Bie 			 * Sanity check: make sure we dind't truncate
290d79dca75STiwei Bie 			 * the address.  The only arches I can find that
291d79dca75STiwei Bie 			 * have 64-bit phys_addr_t but 32-bit dma_addr_t
292d79dca75STiwei Bie 			 * are certain non-highmem MIPS and x86
293d79dca75STiwei Bie 			 * configurations, but these configurations
294d79dca75STiwei Bie 			 * should never allocate physical pages above 32
295d79dca75STiwei Bie 			 * bits, so this is fine.  Just in case, throw a
296d79dca75STiwei Bie 			 * warning and abort if we end up with an
297d79dca75STiwei Bie 			 * unrepresentable address.
298d79dca75STiwei Bie 			 */
299d79dca75STiwei Bie 			if (WARN_ON_ONCE(*dma_handle != phys_addr)) {
300d79dca75STiwei Bie 				free_pages_exact(queue, PAGE_ALIGN(size));
301d79dca75STiwei Bie 				return NULL;
302d79dca75STiwei Bie 			}
303d79dca75STiwei Bie 		}
304d79dca75STiwei Bie 		return queue;
305d79dca75STiwei Bie 	}
306d79dca75STiwei Bie }
307d79dca75STiwei Bie 
308d79dca75STiwei Bie static void vring_free_queue(struct virtio_device *vdev, size_t size,
309d79dca75STiwei Bie 			     void *queue, dma_addr_t dma_handle)
310d79dca75STiwei Bie {
311d79dca75STiwei Bie 	if (vring_use_dma_api(vdev))
312d79dca75STiwei Bie 		dma_free_coherent(vdev->dev.parent, size, queue, dma_handle);
313d79dca75STiwei Bie 	else
314d79dca75STiwei Bie 		free_pages_exact(queue, PAGE_ALIGN(size));
315d79dca75STiwei Bie }
316d79dca75STiwei Bie 
317780bc790SAndy Lutomirski /*
318780bc790SAndy Lutomirski  * The DMA ops on various arches are rather gnarly right now, and
319780bc790SAndy Lutomirski  * making all of the arch DMA ops work on the vring device itself
320780bc790SAndy Lutomirski  * is a mess.  For now, we use the parent device for DMA ops.
321780bc790SAndy Lutomirski  */
32275bfa81bSMichael S. Tsirkin static inline struct device *vring_dma_dev(const struct vring_virtqueue *vq)
323780bc790SAndy Lutomirski {
324780bc790SAndy Lutomirski 	return vq->vq.vdev->dev.parent;
325780bc790SAndy Lutomirski }
326780bc790SAndy Lutomirski 
327780bc790SAndy Lutomirski /* Map one sg entry. */
328780bc790SAndy Lutomirski static dma_addr_t vring_map_one_sg(const struct vring_virtqueue *vq,
329780bc790SAndy Lutomirski 				   struct scatterlist *sg,
330780bc790SAndy Lutomirski 				   enum dma_data_direction direction)
331780bc790SAndy Lutomirski {
332fb3fba6bSTiwei Bie 	if (!vq->use_dma_api)
333780bc790SAndy Lutomirski 		return (dma_addr_t)sg_phys(sg);
334780bc790SAndy Lutomirski 
335780bc790SAndy Lutomirski 	/*
336780bc790SAndy Lutomirski 	 * We can't use dma_map_sg, because we don't use scatterlists in
337780bc790SAndy Lutomirski 	 * the way it expects (we don't guarantee that the scatterlist
338780bc790SAndy Lutomirski 	 * will exist for the lifetime of the mapping).
339780bc790SAndy Lutomirski 	 */
340780bc790SAndy Lutomirski 	return dma_map_page(vring_dma_dev(vq),
341780bc790SAndy Lutomirski 			    sg_page(sg), sg->offset, sg->length,
342780bc790SAndy Lutomirski 			    direction);
343780bc790SAndy Lutomirski }
344780bc790SAndy Lutomirski 
345780bc790SAndy Lutomirski static dma_addr_t vring_map_single(const struct vring_virtqueue *vq,
346780bc790SAndy Lutomirski 				   void *cpu_addr, size_t size,
347780bc790SAndy Lutomirski 				   enum dma_data_direction direction)
348780bc790SAndy Lutomirski {
349fb3fba6bSTiwei Bie 	if (!vq->use_dma_api)
350780bc790SAndy Lutomirski 		return (dma_addr_t)virt_to_phys(cpu_addr);
351780bc790SAndy Lutomirski 
352780bc790SAndy Lutomirski 	return dma_map_single(vring_dma_dev(vq),
353780bc790SAndy Lutomirski 			      cpu_addr, size, direction);
354780bc790SAndy Lutomirski }
355780bc790SAndy Lutomirski 
356e6f633e5STiwei Bie static int vring_mapping_error(const struct vring_virtqueue *vq,
357e6f633e5STiwei Bie 			       dma_addr_t addr)
358e6f633e5STiwei Bie {
359fb3fba6bSTiwei Bie 	if (!vq->use_dma_api)
360e6f633e5STiwei Bie 		return 0;
361e6f633e5STiwei Bie 
362e6f633e5STiwei Bie 	return dma_mapping_error(vring_dma_dev(vq), addr);
363e6f633e5STiwei Bie }
364e6f633e5STiwei Bie 
365e6f633e5STiwei Bie 
366e6f633e5STiwei Bie /*
367e6f633e5STiwei Bie  * Split ring specific functions - *_split().
368e6f633e5STiwei Bie  */
369e6f633e5STiwei Bie 
370138fd251STiwei Bie static void vring_unmap_one_split(const struct vring_virtqueue *vq,
371780bc790SAndy Lutomirski 				  struct vring_desc *desc)
372780bc790SAndy Lutomirski {
373780bc790SAndy Lutomirski 	u16 flags;
374780bc790SAndy Lutomirski 
375fb3fba6bSTiwei Bie 	if (!vq->use_dma_api)
376780bc790SAndy Lutomirski 		return;
377780bc790SAndy Lutomirski 
378780bc790SAndy Lutomirski 	flags = virtio16_to_cpu(vq->vq.vdev, desc->flags);
379780bc790SAndy Lutomirski 
380780bc790SAndy Lutomirski 	if (flags & VRING_DESC_F_INDIRECT) {
381780bc790SAndy Lutomirski 		dma_unmap_single(vring_dma_dev(vq),
382780bc790SAndy Lutomirski 				 virtio64_to_cpu(vq->vq.vdev, desc->addr),
383780bc790SAndy Lutomirski 				 virtio32_to_cpu(vq->vq.vdev, desc->len),
384780bc790SAndy Lutomirski 				 (flags & VRING_DESC_F_WRITE) ?
385780bc790SAndy Lutomirski 				 DMA_FROM_DEVICE : DMA_TO_DEVICE);
386780bc790SAndy Lutomirski 	} else {
387780bc790SAndy Lutomirski 		dma_unmap_page(vring_dma_dev(vq),
388780bc790SAndy Lutomirski 			       virtio64_to_cpu(vq->vq.vdev, desc->addr),
389780bc790SAndy Lutomirski 			       virtio32_to_cpu(vq->vq.vdev, desc->len),
390780bc790SAndy Lutomirski 			       (flags & VRING_DESC_F_WRITE) ?
391780bc790SAndy Lutomirski 			       DMA_FROM_DEVICE : DMA_TO_DEVICE);
392780bc790SAndy Lutomirski 	}
393780bc790SAndy Lutomirski }
394780bc790SAndy Lutomirski 
395138fd251STiwei Bie static struct vring_desc *alloc_indirect_split(struct virtqueue *_vq,
396138fd251STiwei Bie 					       unsigned int total_sg,
397138fd251STiwei Bie 					       gfp_t gfp)
3989fa29b9dSMark McLoughlin {
3999fa29b9dSMark McLoughlin 	struct vring_desc *desc;
400b25bd251SRusty Russell 	unsigned int i;
4019fa29b9dSMark McLoughlin 
402b92b1b89SWill Deacon 	/*
403b92b1b89SWill Deacon 	 * We require lowmem mappings for the descriptors because
404b92b1b89SWill Deacon 	 * otherwise virt_to_phys will give us bogus addresses in the
405b92b1b89SWill Deacon 	 * virtqueue.
406b92b1b89SWill Deacon 	 */
40782107539SMichal Hocko 	gfp &= ~__GFP_HIGHMEM;
408b92b1b89SWill Deacon 
4096da2ec56SKees Cook 	desc = kmalloc_array(total_sg, sizeof(struct vring_desc), gfp);
4109fa29b9dSMark McLoughlin 	if (!desc)
411b25bd251SRusty Russell 		return NULL;
4129fa29b9dSMark McLoughlin 
413b25bd251SRusty Russell 	for (i = 0; i < total_sg; i++)
41400e6f3d9SMichael S. Tsirkin 		desc[i].next = cpu_to_virtio16(_vq->vdev, i + 1);
415b25bd251SRusty Russell 	return desc;
4169fa29b9dSMark McLoughlin }
4179fa29b9dSMark McLoughlin 
418fe4c3862SJason Wang static inline unsigned int virtqueue_add_desc_split(struct virtqueue *vq,
419fe4c3862SJason Wang 						    struct vring_desc *desc,
420fe4c3862SJason Wang 						    unsigned int i,
421fe4c3862SJason Wang 						    dma_addr_t addr,
422fe4c3862SJason Wang 						    unsigned int len,
423fe4c3862SJason Wang 						    u16 flags)
424fe4c3862SJason Wang {
425fe4c3862SJason Wang 	desc[i].flags = cpu_to_virtio16(vq->vdev, flags);
426fe4c3862SJason Wang 	desc[i].addr = cpu_to_virtio64(vq->vdev, addr);
427fe4c3862SJason Wang 	desc[i].len = cpu_to_virtio32(vq->vdev, len);
428fe4c3862SJason Wang 
429fe4c3862SJason Wang 	return virtio16_to_cpu(vq->vdev, desc[i].next);
430fe4c3862SJason Wang }
431fe4c3862SJason Wang 
432138fd251STiwei Bie static inline int virtqueue_add_split(struct virtqueue *_vq,
43313816c76SRusty Russell 				      struct scatterlist *sgs[],
434eeebf9b1SRusty Russell 				      unsigned int total_sg,
43513816c76SRusty Russell 				      unsigned int out_sgs,
43613816c76SRusty Russell 				      unsigned int in_sgs,
437bbd603efSMichael S. Tsirkin 				      void *data,
4385a08b04fSMichael S. Tsirkin 				      void *ctx,
439bbd603efSMichael S. Tsirkin 				      gfp_t gfp)
4400a8a69ddSRusty Russell {
4410a8a69ddSRusty Russell 	struct vring_virtqueue *vq = to_vvq(_vq);
44213816c76SRusty Russell 	struct scatterlist *sg;
443b25bd251SRusty Russell 	struct vring_desc *desc;
4443f649ab7SKees Cook 	unsigned int i, n, avail, descs_used, prev, err_idx;
4451fe9b6feSMichael S. Tsirkin 	int head;
446b25bd251SRusty Russell 	bool indirect;
4470a8a69ddSRusty Russell 
4489fa29b9dSMark McLoughlin 	START_USE(vq);
4499fa29b9dSMark McLoughlin 
4500a8a69ddSRusty Russell 	BUG_ON(data == NULL);
4515a08b04fSMichael S. Tsirkin 	BUG_ON(ctx && vq->indirect);
4529fa29b9dSMark McLoughlin 
45370670444SRusty Russell 	if (unlikely(vq->broken)) {
45470670444SRusty Russell 		END_USE(vq);
45570670444SRusty Russell 		return -EIO;
45670670444SRusty Russell 	}
45770670444SRusty Russell 
4584d6a105eSTiwei Bie 	LAST_ADD_TIME_UPDATE(vq);
459e93300b1SRusty Russell 
46013816c76SRusty Russell 	BUG_ON(total_sg == 0);
4610a8a69ddSRusty Russell 
462b25bd251SRusty Russell 	head = vq->free_head;
463b25bd251SRusty Russell 
4642f18c2d1STiwei Bie 	if (virtqueue_use_indirect(_vq, total_sg))
465138fd251STiwei Bie 		desc = alloc_indirect_split(_vq, total_sg, gfp);
46644ed8089SRichard W.M. Jones 	else {
467b25bd251SRusty Russell 		desc = NULL;
468e593bf97STiwei Bie 		WARN_ON_ONCE(total_sg > vq->split.vring.num && !vq->indirect);
46944ed8089SRichard W.M. Jones 	}
470b25bd251SRusty Russell 
471b25bd251SRusty Russell 	if (desc) {
472b25bd251SRusty Russell 		/* Use a single buffer which doesn't continue */
473780bc790SAndy Lutomirski 		indirect = true;
474b25bd251SRusty Russell 		/* Set up rest to use this indirect table. */
475b25bd251SRusty Russell 		i = 0;
476b25bd251SRusty Russell 		descs_used = 1;
477b25bd251SRusty Russell 	} else {
478780bc790SAndy Lutomirski 		indirect = false;
479e593bf97STiwei Bie 		desc = vq->split.vring.desc;
480b25bd251SRusty Russell 		i = head;
481b25bd251SRusty Russell 		descs_used = total_sg;
482b25bd251SRusty Russell 	}
483b25bd251SRusty Russell 
484b25bd251SRusty Russell 	if (vq->vq.num_free < descs_used) {
4850a8a69ddSRusty Russell 		pr_debug("Can't add buf len %i - avail = %i\n",
486b25bd251SRusty Russell 			 descs_used, vq->vq.num_free);
48744653eaeSRusty Russell 		/* FIXME: for historical reasons, we force a notify here if
48844653eaeSRusty Russell 		 * there are outgoing parts to the buffer.  Presumably the
48944653eaeSRusty Russell 		 * host should service the ring ASAP. */
49013816c76SRusty Russell 		if (out_sgs)
491426e3e0aSRusty Russell 			vq->notify(&vq->vq);
49258625edfSWei Yongjun 		if (indirect)
49358625edfSWei Yongjun 			kfree(desc);
4940a8a69ddSRusty Russell 		END_USE(vq);
4950a8a69ddSRusty Russell 		return -ENOSPC;
4960a8a69ddSRusty Russell 	}
4970a8a69ddSRusty Russell 
49813816c76SRusty Russell 	for (n = 0; n < out_sgs; n++) {
499eeebf9b1SRusty Russell 		for (sg = sgs[n]; sg; sg = sg_next(sg)) {
500780bc790SAndy Lutomirski 			dma_addr_t addr = vring_map_one_sg(vq, sg, DMA_TO_DEVICE);
501780bc790SAndy Lutomirski 			if (vring_mapping_error(vq, addr))
502780bc790SAndy Lutomirski 				goto unmap_release;
503780bc790SAndy Lutomirski 
5040a8a69ddSRusty Russell 			prev = i;
505fe4c3862SJason Wang 			i = virtqueue_add_desc_split(_vq, desc, i, addr, sg->length,
506fe4c3862SJason Wang 						     VRING_DESC_F_NEXT);
5070a8a69ddSRusty Russell 		}
50813816c76SRusty Russell 	}
50913816c76SRusty Russell 	for (; n < (out_sgs + in_sgs); n++) {
510eeebf9b1SRusty Russell 		for (sg = sgs[n]; sg; sg = sg_next(sg)) {
511780bc790SAndy Lutomirski 			dma_addr_t addr = vring_map_one_sg(vq, sg, DMA_FROM_DEVICE);
512780bc790SAndy Lutomirski 			if (vring_mapping_error(vq, addr))
513780bc790SAndy Lutomirski 				goto unmap_release;
514780bc790SAndy Lutomirski 
5150a8a69ddSRusty Russell 			prev = i;
516fe4c3862SJason Wang 			i = virtqueue_add_desc_split(_vq, desc, i, addr,
517fe4c3862SJason Wang 						     sg->length,
518fe4c3862SJason Wang 						     VRING_DESC_F_NEXT |
519fe4c3862SJason Wang 						     VRING_DESC_F_WRITE);
52013816c76SRusty Russell 		}
5210a8a69ddSRusty Russell 	}
5220a8a69ddSRusty Russell 	/* Last one doesn't continue. */
52300e6f3d9SMichael S. Tsirkin 	desc[prev].flags &= cpu_to_virtio16(_vq->vdev, ~VRING_DESC_F_NEXT);
5240a8a69ddSRusty Russell 
525780bc790SAndy Lutomirski 	if (indirect) {
526780bc790SAndy Lutomirski 		/* Now that the indirect table is filled in, map it. */
527780bc790SAndy Lutomirski 		dma_addr_t addr = vring_map_single(
528780bc790SAndy Lutomirski 			vq, desc, total_sg * sizeof(struct vring_desc),
529780bc790SAndy Lutomirski 			DMA_TO_DEVICE);
530780bc790SAndy Lutomirski 		if (vring_mapping_error(vq, addr))
531780bc790SAndy Lutomirski 			goto unmap_release;
532780bc790SAndy Lutomirski 
533fe4c3862SJason Wang 		virtqueue_add_desc_split(_vq, vq->split.vring.desc,
534fe4c3862SJason Wang 					 head, addr,
535fe4c3862SJason Wang 					 total_sg * sizeof(struct vring_desc),
536e593bf97STiwei Bie 			                 VRING_DESC_F_INDIRECT);
537780bc790SAndy Lutomirski 	}
538780bc790SAndy Lutomirski 
539780bc790SAndy Lutomirski 	/* We're using some buffers from the free list. */
540780bc790SAndy Lutomirski 	vq->vq.num_free -= descs_used;
541780bc790SAndy Lutomirski 
5420a8a69ddSRusty Russell 	/* Update free pointer */
543b25bd251SRusty Russell 	if (indirect)
544e593bf97STiwei Bie 		vq->free_head = virtio16_to_cpu(_vq->vdev,
545e593bf97STiwei Bie 					vq->split.vring.desc[head].next);
546b25bd251SRusty Russell 	else
5470a8a69ddSRusty Russell 		vq->free_head = i;
5480a8a69ddSRusty Russell 
549780bc790SAndy Lutomirski 	/* Store token and indirect buffer state. */
550cbeedb72STiwei Bie 	vq->split.desc_state[head].data = data;
551780bc790SAndy Lutomirski 	if (indirect)
552cbeedb72STiwei Bie 		vq->split.desc_state[head].indir_desc = desc;
55387646a34SJason Wang 	else
554cbeedb72STiwei Bie 		vq->split.desc_state[head].indir_desc = ctx;
5550a8a69ddSRusty Russell 
5560a8a69ddSRusty Russell 	/* Put entry in available array (but don't update avail->idx until they
5573b720b8cSRusty Russell 	 * do sync). */
558e593bf97STiwei Bie 	avail = vq->split.avail_idx_shadow & (vq->split.vring.num - 1);
559e593bf97STiwei Bie 	vq->split.vring.avail->ring[avail] = cpu_to_virtio16(_vq->vdev, head);
5600a8a69ddSRusty Russell 
561ee7cd898SRusty Russell 	/* Descriptors and available array need to be set before we expose the
562ee7cd898SRusty Russell 	 * new available array entries. */
563a9a0fef7SRusty Russell 	virtio_wmb(vq->weak_barriers);
564e593bf97STiwei Bie 	vq->split.avail_idx_shadow++;
565e593bf97STiwei Bie 	vq->split.vring.avail->idx = cpu_to_virtio16(_vq->vdev,
566e593bf97STiwei Bie 						vq->split.avail_idx_shadow);
567ee7cd898SRusty Russell 	vq->num_added++;
568ee7cd898SRusty Russell 
5695e05bf58STetsuo Handa 	pr_debug("Added buffer head %i to %p\n", head, vq);
5705e05bf58STetsuo Handa 	END_USE(vq);
5715e05bf58STetsuo Handa 
572ee7cd898SRusty Russell 	/* This is very unlikely, but theoretically possible.  Kick
573ee7cd898SRusty Russell 	 * just in case. */
574ee7cd898SRusty Russell 	if (unlikely(vq->num_added == (1 << 16) - 1))
575ee7cd898SRusty Russell 		virtqueue_kick(_vq);
576ee7cd898SRusty Russell 
57798e8c6bcSRusty Russell 	return 0;
578780bc790SAndy Lutomirski 
579780bc790SAndy Lutomirski unmap_release:
580780bc790SAndy Lutomirski 	err_idx = i;
581cf8f1696SMatthias Lange 
582cf8f1696SMatthias Lange 	if (indirect)
583cf8f1696SMatthias Lange 		i = 0;
584cf8f1696SMatthias Lange 	else
585780bc790SAndy Lutomirski 		i = head;
586780bc790SAndy Lutomirski 
587780bc790SAndy Lutomirski 	for (n = 0; n < total_sg; n++) {
588780bc790SAndy Lutomirski 		if (i == err_idx)
589780bc790SAndy Lutomirski 			break;
590138fd251STiwei Bie 		vring_unmap_one_split(vq, &desc[i]);
591cf8f1696SMatthias Lange 		i = virtio16_to_cpu(_vq->vdev, desc[i].next);
592780bc790SAndy Lutomirski 	}
593780bc790SAndy Lutomirski 
594780bc790SAndy Lutomirski 	if (indirect)
595780bc790SAndy Lutomirski 		kfree(desc);
596780bc790SAndy Lutomirski 
5973cc36f6eSMichael S. Tsirkin 	END_USE(vq);
598f7728002SHalil Pasic 	return -ENOMEM;
5990a8a69ddSRusty Russell }
60013816c76SRusty Russell 
601138fd251STiwei Bie static bool virtqueue_kick_prepare_split(struct virtqueue *_vq)
6020a8a69ddSRusty Russell {
6030a8a69ddSRusty Russell 	struct vring_virtqueue *vq = to_vvq(_vq);
604a5c262c5SMichael S. Tsirkin 	u16 new, old;
60541f0377fSRusty Russell 	bool needs_kick;
60641f0377fSRusty Russell 
6070a8a69ddSRusty Russell 	START_USE(vq);
608a72caae2SJason Wang 	/* We need to expose available array entries before checking avail
609a72caae2SJason Wang 	 * event. */
610a9a0fef7SRusty Russell 	virtio_mb(vq->weak_barriers);
6110a8a69ddSRusty Russell 
612e593bf97STiwei Bie 	old = vq->split.avail_idx_shadow - vq->num_added;
613e593bf97STiwei Bie 	new = vq->split.avail_idx_shadow;
6140a8a69ddSRusty Russell 	vq->num_added = 0;
6150a8a69ddSRusty Russell 
6164d6a105eSTiwei Bie 	LAST_ADD_TIME_CHECK(vq);
6174d6a105eSTiwei Bie 	LAST_ADD_TIME_INVALID(vq);
618e93300b1SRusty Russell 
61941f0377fSRusty Russell 	if (vq->event) {
620e593bf97STiwei Bie 		needs_kick = vring_need_event(virtio16_to_cpu(_vq->vdev,
621e593bf97STiwei Bie 					vring_avail_event(&vq->split.vring)),
62241f0377fSRusty Russell 					      new, old);
62341f0377fSRusty Russell 	} else {
624e593bf97STiwei Bie 		needs_kick = !(vq->split.vring.used->flags &
625e593bf97STiwei Bie 					cpu_to_virtio16(_vq->vdev,
626e593bf97STiwei Bie 						VRING_USED_F_NO_NOTIFY));
62741f0377fSRusty Russell 	}
6280a8a69ddSRusty Russell 	END_USE(vq);
62941f0377fSRusty Russell 	return needs_kick;
63041f0377fSRusty Russell }
631138fd251STiwei Bie 
632138fd251STiwei Bie static void detach_buf_split(struct vring_virtqueue *vq, unsigned int head,
6335a08b04fSMichael S. Tsirkin 			     void **ctx)
6340a8a69ddSRusty Russell {
635780bc790SAndy Lutomirski 	unsigned int i, j;
636c60923cbSGonglei 	__virtio16 nextflag = cpu_to_virtio16(vq->vq.vdev, VRING_DESC_F_NEXT);
6370a8a69ddSRusty Russell 
6380a8a69ddSRusty Russell 	/* Clear data ptr. */
639cbeedb72STiwei Bie 	vq->split.desc_state[head].data = NULL;
6400a8a69ddSRusty Russell 
641780bc790SAndy Lutomirski 	/* Put back on free list: unmap first-level descriptors and find end */
6420a8a69ddSRusty Russell 	i = head;
6439fa29b9dSMark McLoughlin 
644e593bf97STiwei Bie 	while (vq->split.vring.desc[i].flags & nextflag) {
645e593bf97STiwei Bie 		vring_unmap_one_split(vq, &vq->split.vring.desc[i]);
646e593bf97STiwei Bie 		i = virtio16_to_cpu(vq->vq.vdev, vq->split.vring.desc[i].next);
64706ca287dSRusty Russell 		vq->vq.num_free++;
6480a8a69ddSRusty Russell 	}
6490a8a69ddSRusty Russell 
650e593bf97STiwei Bie 	vring_unmap_one_split(vq, &vq->split.vring.desc[i]);
651e593bf97STiwei Bie 	vq->split.vring.desc[i].next = cpu_to_virtio16(vq->vq.vdev,
652e593bf97STiwei Bie 						vq->free_head);
6530a8a69ddSRusty Russell 	vq->free_head = head;
654780bc790SAndy Lutomirski 
6550a8a69ddSRusty Russell 	/* Plus final descriptor */
65606ca287dSRusty Russell 	vq->vq.num_free++;
657780bc790SAndy Lutomirski 
6585a08b04fSMichael S. Tsirkin 	if (vq->indirect) {
659cbeedb72STiwei Bie 		struct vring_desc *indir_desc =
660cbeedb72STiwei Bie 				vq->split.desc_state[head].indir_desc;
6615a08b04fSMichael S. Tsirkin 		u32 len;
6625a08b04fSMichael S. Tsirkin 
6635a08b04fSMichael S. Tsirkin 		/* Free the indirect table, if any, now that it's unmapped. */
6645a08b04fSMichael S. Tsirkin 		if (!indir_desc)
6655a08b04fSMichael S. Tsirkin 			return;
6665a08b04fSMichael S. Tsirkin 
667e593bf97STiwei Bie 		len = virtio32_to_cpu(vq->vq.vdev,
668e593bf97STiwei Bie 				vq->split.vring.desc[head].len);
669780bc790SAndy Lutomirski 
670e593bf97STiwei Bie 		BUG_ON(!(vq->split.vring.desc[head].flags &
671780bc790SAndy Lutomirski 			 cpu_to_virtio16(vq->vq.vdev, VRING_DESC_F_INDIRECT)));
672780bc790SAndy Lutomirski 		BUG_ON(len == 0 || len % sizeof(struct vring_desc));
673780bc790SAndy Lutomirski 
674780bc790SAndy Lutomirski 		for (j = 0; j < len / sizeof(struct vring_desc); j++)
675138fd251STiwei Bie 			vring_unmap_one_split(vq, &indir_desc[j]);
676780bc790SAndy Lutomirski 
6775a08b04fSMichael S. Tsirkin 		kfree(indir_desc);
678cbeedb72STiwei Bie 		vq->split.desc_state[head].indir_desc = NULL;
6795a08b04fSMichael S. Tsirkin 	} else if (ctx) {
680cbeedb72STiwei Bie 		*ctx = vq->split.desc_state[head].indir_desc;
681780bc790SAndy Lutomirski 	}
6820a8a69ddSRusty Russell }
6830a8a69ddSRusty Russell 
684138fd251STiwei Bie static inline bool more_used_split(const struct vring_virtqueue *vq)
6850a8a69ddSRusty Russell {
686e593bf97STiwei Bie 	return vq->last_used_idx != virtio16_to_cpu(vq->vq.vdev,
687e593bf97STiwei Bie 			vq->split.vring.used->idx);
6880a8a69ddSRusty Russell }
6890a8a69ddSRusty Russell 
690138fd251STiwei Bie static void *virtqueue_get_buf_ctx_split(struct virtqueue *_vq,
691138fd251STiwei Bie 					 unsigned int *len,
6925a08b04fSMichael S. Tsirkin 					 void **ctx)
6930a8a69ddSRusty Russell {
6940a8a69ddSRusty Russell 	struct vring_virtqueue *vq = to_vvq(_vq);
6950a8a69ddSRusty Russell 	void *ret;
6960a8a69ddSRusty Russell 	unsigned int i;
6973b720b8cSRusty Russell 	u16 last_used;
6980a8a69ddSRusty Russell 
6990a8a69ddSRusty Russell 	START_USE(vq);
7000a8a69ddSRusty Russell 
7015ef82752SRusty Russell 	if (unlikely(vq->broken)) {
7025ef82752SRusty Russell 		END_USE(vq);
7035ef82752SRusty Russell 		return NULL;
7045ef82752SRusty Russell 	}
7055ef82752SRusty Russell 
706138fd251STiwei Bie 	if (!more_used_split(vq)) {
7070a8a69ddSRusty Russell 		pr_debug("No more buffers in queue\n");
7080a8a69ddSRusty Russell 		END_USE(vq);
7090a8a69ddSRusty Russell 		return NULL;
7100a8a69ddSRusty Russell 	}
7110a8a69ddSRusty Russell 
7122d61ba95SMichael S. Tsirkin 	/* Only get used array entries after they have been exposed by host. */
713a9a0fef7SRusty Russell 	virtio_rmb(vq->weak_barriers);
7142d61ba95SMichael S. Tsirkin 
715e593bf97STiwei Bie 	last_used = (vq->last_used_idx & (vq->split.vring.num - 1));
716e593bf97STiwei Bie 	i = virtio32_to_cpu(_vq->vdev,
717e593bf97STiwei Bie 			vq->split.vring.used->ring[last_used].id);
718e593bf97STiwei Bie 	*len = virtio32_to_cpu(_vq->vdev,
719e593bf97STiwei Bie 			vq->split.vring.used->ring[last_used].len);
7200a8a69ddSRusty Russell 
721e593bf97STiwei Bie 	if (unlikely(i >= vq->split.vring.num)) {
7220a8a69ddSRusty Russell 		BAD_RING(vq, "id %u out of range\n", i);
7230a8a69ddSRusty Russell 		return NULL;
7240a8a69ddSRusty Russell 	}
725cbeedb72STiwei Bie 	if (unlikely(!vq->split.desc_state[i].data)) {
7260a8a69ddSRusty Russell 		BAD_RING(vq, "id %u is not a head!\n", i);
7270a8a69ddSRusty Russell 		return NULL;
7280a8a69ddSRusty Russell 	}
7290a8a69ddSRusty Russell 
730138fd251STiwei Bie 	/* detach_buf_split clears data, so grab it now. */
731cbeedb72STiwei Bie 	ret = vq->split.desc_state[i].data;
732138fd251STiwei Bie 	detach_buf_split(vq, i, ctx);
7330a8a69ddSRusty Russell 	vq->last_used_idx++;
734a5c262c5SMichael S. Tsirkin 	/* If we expect an interrupt for the next entry, tell host
735a5c262c5SMichael S. Tsirkin 	 * by writing event index and flush out the write before
736a5c262c5SMichael S. Tsirkin 	 * the read in the next get_buf call. */
737e593bf97STiwei Bie 	if (!(vq->split.avail_flags_shadow & VRING_AVAIL_F_NO_INTERRUPT))
738788e5b3aSMichael S. Tsirkin 		virtio_store_mb(vq->weak_barriers,
739e593bf97STiwei Bie 				&vring_used_event(&vq->split.vring),
740788e5b3aSMichael S. Tsirkin 				cpu_to_virtio16(_vq->vdev, vq->last_used_idx));
741a5c262c5SMichael S. Tsirkin 
7424d6a105eSTiwei Bie 	LAST_ADD_TIME_INVALID(vq);
743e93300b1SRusty Russell 
7440a8a69ddSRusty Russell 	END_USE(vq);
7450a8a69ddSRusty Russell 	return ret;
7460a8a69ddSRusty Russell }
747138fd251STiwei Bie 
748138fd251STiwei Bie static void virtqueue_disable_cb_split(struct virtqueue *_vq)
749138fd251STiwei Bie {
750138fd251STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
751138fd251STiwei Bie 
752e593bf97STiwei Bie 	if (!(vq->split.avail_flags_shadow & VRING_AVAIL_F_NO_INTERRUPT)) {
753e593bf97STiwei Bie 		vq->split.avail_flags_shadow |= VRING_AVAIL_F_NO_INTERRUPT;
7548d622d21SMichael S. Tsirkin 		if (vq->event)
7558d622d21SMichael S. Tsirkin 			/* TODO: this is a hack. Figure out a cleaner value to write. */
7568d622d21SMichael S. Tsirkin 			vring_used_event(&vq->split.vring) = 0x0;
7578d622d21SMichael S. Tsirkin 		else
758e593bf97STiwei Bie 			vq->split.vring.avail->flags =
759e593bf97STiwei Bie 				cpu_to_virtio16(_vq->vdev,
760e593bf97STiwei Bie 						vq->split.avail_flags_shadow);
761138fd251STiwei Bie 	}
762138fd251STiwei Bie }
763138fd251STiwei Bie 
764138fd251STiwei Bie static unsigned virtqueue_enable_cb_prepare_split(struct virtqueue *_vq)
765cc229884SMichael S. Tsirkin {
766cc229884SMichael S. Tsirkin 	struct vring_virtqueue *vq = to_vvq(_vq);
767cc229884SMichael S. Tsirkin 	u16 last_used_idx;
768cc229884SMichael S. Tsirkin 
769cc229884SMichael S. Tsirkin 	START_USE(vq);
770cc229884SMichael S. Tsirkin 
771cc229884SMichael S. Tsirkin 	/* We optimistically turn back on interrupts, then check if there was
772cc229884SMichael S. Tsirkin 	 * more to do. */
773cc229884SMichael S. Tsirkin 	/* Depending on the VIRTIO_RING_F_EVENT_IDX feature, we need to
774cc229884SMichael S. Tsirkin 	 * either clear the flags bit or point the event index at the next
775cc229884SMichael S. Tsirkin 	 * entry. Always do both to keep code simple. */
776e593bf97STiwei Bie 	if (vq->split.avail_flags_shadow & VRING_AVAIL_F_NO_INTERRUPT) {
777e593bf97STiwei Bie 		vq->split.avail_flags_shadow &= ~VRING_AVAIL_F_NO_INTERRUPT;
7780ea1e4a6SLadi Prosek 		if (!vq->event)
779e593bf97STiwei Bie 			vq->split.vring.avail->flags =
780e593bf97STiwei Bie 				cpu_to_virtio16(_vq->vdev,
781e593bf97STiwei Bie 						vq->split.avail_flags_shadow);
782f277ec42SVenkatesh Srinivas 	}
783e593bf97STiwei Bie 	vring_used_event(&vq->split.vring) = cpu_to_virtio16(_vq->vdev,
784e593bf97STiwei Bie 			last_used_idx = vq->last_used_idx);
785cc229884SMichael S. Tsirkin 	END_USE(vq);
786cc229884SMichael S. Tsirkin 	return last_used_idx;
787cc229884SMichael S. Tsirkin }
788138fd251STiwei Bie 
789138fd251STiwei Bie static bool virtqueue_poll_split(struct virtqueue *_vq, unsigned last_used_idx)
790138fd251STiwei Bie {
791138fd251STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
792138fd251STiwei Bie 
793138fd251STiwei Bie 	return (u16)last_used_idx != virtio16_to_cpu(_vq->vdev,
794e593bf97STiwei Bie 			vq->split.vring.used->idx);
795138fd251STiwei Bie }
796138fd251STiwei Bie 
797138fd251STiwei Bie static bool virtqueue_enable_cb_delayed_split(struct virtqueue *_vq)
7987ab358c2SMichael S. Tsirkin {
7997ab358c2SMichael S. Tsirkin 	struct vring_virtqueue *vq = to_vvq(_vq);
8007ab358c2SMichael S. Tsirkin 	u16 bufs;
8017ab358c2SMichael S. Tsirkin 
8027ab358c2SMichael S. Tsirkin 	START_USE(vq);
8037ab358c2SMichael S. Tsirkin 
8047ab358c2SMichael S. Tsirkin 	/* We optimistically turn back on interrupts, then check if there was
8057ab358c2SMichael S. Tsirkin 	 * more to do. */
8067ab358c2SMichael S. Tsirkin 	/* Depending on the VIRTIO_RING_F_USED_EVENT_IDX feature, we need to
8077ab358c2SMichael S. Tsirkin 	 * either clear the flags bit or point the event index at the next
8080ea1e4a6SLadi Prosek 	 * entry. Always update the event index to keep code simple. */
809e593bf97STiwei Bie 	if (vq->split.avail_flags_shadow & VRING_AVAIL_F_NO_INTERRUPT) {
810e593bf97STiwei Bie 		vq->split.avail_flags_shadow &= ~VRING_AVAIL_F_NO_INTERRUPT;
8110ea1e4a6SLadi Prosek 		if (!vq->event)
812e593bf97STiwei Bie 			vq->split.vring.avail->flags =
813e593bf97STiwei Bie 				cpu_to_virtio16(_vq->vdev,
814e593bf97STiwei Bie 						vq->split.avail_flags_shadow);
815f277ec42SVenkatesh Srinivas 	}
8167ab358c2SMichael S. Tsirkin 	/* TODO: tune this threshold */
817e593bf97STiwei Bie 	bufs = (u16)(vq->split.avail_idx_shadow - vq->last_used_idx) * 3 / 4;
818788e5b3aSMichael S. Tsirkin 
819788e5b3aSMichael S. Tsirkin 	virtio_store_mb(vq->weak_barriers,
820e593bf97STiwei Bie 			&vring_used_event(&vq->split.vring),
821788e5b3aSMichael S. Tsirkin 			cpu_to_virtio16(_vq->vdev, vq->last_used_idx + bufs));
822788e5b3aSMichael S. Tsirkin 
823e593bf97STiwei Bie 	if (unlikely((u16)(virtio16_to_cpu(_vq->vdev, vq->split.vring.used->idx)
824e593bf97STiwei Bie 					- vq->last_used_idx) > bufs)) {
8257ab358c2SMichael S. Tsirkin 		END_USE(vq);
8267ab358c2SMichael S. Tsirkin 		return false;
8277ab358c2SMichael S. Tsirkin 	}
8287ab358c2SMichael S. Tsirkin 
8297ab358c2SMichael S. Tsirkin 	END_USE(vq);
8307ab358c2SMichael S. Tsirkin 	return true;
8317ab358c2SMichael S. Tsirkin }
8327ab358c2SMichael S. Tsirkin 
833138fd251STiwei Bie static void *virtqueue_detach_unused_buf_split(struct virtqueue *_vq)
834c021eac4SShirley Ma {
835c021eac4SShirley Ma 	struct vring_virtqueue *vq = to_vvq(_vq);
836c021eac4SShirley Ma 	unsigned int i;
837c021eac4SShirley Ma 	void *buf;
838c021eac4SShirley Ma 
839c021eac4SShirley Ma 	START_USE(vq);
840c021eac4SShirley Ma 
841e593bf97STiwei Bie 	for (i = 0; i < vq->split.vring.num; i++) {
842cbeedb72STiwei Bie 		if (!vq->split.desc_state[i].data)
843c021eac4SShirley Ma 			continue;
844138fd251STiwei Bie 		/* detach_buf_split clears data, so grab it now. */
845cbeedb72STiwei Bie 		buf = vq->split.desc_state[i].data;
846138fd251STiwei Bie 		detach_buf_split(vq, i, NULL);
847e593bf97STiwei Bie 		vq->split.avail_idx_shadow--;
848e593bf97STiwei Bie 		vq->split.vring.avail->idx = cpu_to_virtio16(_vq->vdev,
849e593bf97STiwei Bie 				vq->split.avail_idx_shadow);
850c021eac4SShirley Ma 		END_USE(vq);
851c021eac4SShirley Ma 		return buf;
852c021eac4SShirley Ma 	}
853c021eac4SShirley Ma 	/* That should have freed everything. */
854e593bf97STiwei Bie 	BUG_ON(vq->vq.num_free != vq->split.vring.num);
855c021eac4SShirley Ma 
856c021eac4SShirley Ma 	END_USE(vq);
857c021eac4SShirley Ma 	return NULL;
858c021eac4SShirley Ma }
859138fd251STiwei Bie 
860d79dca75STiwei Bie static struct virtqueue *vring_create_virtqueue_split(
861d79dca75STiwei Bie 	unsigned int index,
862d79dca75STiwei Bie 	unsigned int num,
863d79dca75STiwei Bie 	unsigned int vring_align,
864d79dca75STiwei Bie 	struct virtio_device *vdev,
865d79dca75STiwei Bie 	bool weak_barriers,
866d79dca75STiwei Bie 	bool may_reduce_num,
867d79dca75STiwei Bie 	bool context,
868d79dca75STiwei Bie 	bool (*notify)(struct virtqueue *),
869d79dca75STiwei Bie 	void (*callback)(struct virtqueue *),
870d79dca75STiwei Bie 	const char *name)
871d79dca75STiwei Bie {
872d79dca75STiwei Bie 	struct virtqueue *vq;
873d79dca75STiwei Bie 	void *queue = NULL;
874d79dca75STiwei Bie 	dma_addr_t dma_addr;
875d79dca75STiwei Bie 	size_t queue_size_in_bytes;
876d79dca75STiwei Bie 	struct vring vring;
877d79dca75STiwei Bie 
878d79dca75STiwei Bie 	/* We assume num is a power of 2. */
879d79dca75STiwei Bie 	if (num & (num - 1)) {
880d79dca75STiwei Bie 		dev_warn(&vdev->dev, "Bad virtqueue length %u\n", num);
881d79dca75STiwei Bie 		return NULL;
882d79dca75STiwei Bie 	}
883d79dca75STiwei Bie 
884d79dca75STiwei Bie 	/* TODO: allocate each queue chunk individually */
885d79dca75STiwei Bie 	for (; num && vring_size(num, vring_align) > PAGE_SIZE; num /= 2) {
886d79dca75STiwei Bie 		queue = vring_alloc_queue(vdev, vring_size(num, vring_align),
887d79dca75STiwei Bie 					  &dma_addr,
888d79dca75STiwei Bie 					  GFP_KERNEL|__GFP_NOWARN|__GFP_ZERO);
889d79dca75STiwei Bie 		if (queue)
890d79dca75STiwei Bie 			break;
891cf94db21SCornelia Huck 		if (!may_reduce_num)
892cf94db21SCornelia Huck 			return NULL;
893d79dca75STiwei Bie 	}
894d79dca75STiwei Bie 
895d79dca75STiwei Bie 	if (!num)
896d79dca75STiwei Bie 		return NULL;
897d79dca75STiwei Bie 
898d79dca75STiwei Bie 	if (!queue) {
899d79dca75STiwei Bie 		/* Try to get a single page. You are my only hope! */
900d79dca75STiwei Bie 		queue = vring_alloc_queue(vdev, vring_size(num, vring_align),
901d79dca75STiwei Bie 					  &dma_addr, GFP_KERNEL|__GFP_ZERO);
902d79dca75STiwei Bie 	}
903d79dca75STiwei Bie 	if (!queue)
904d79dca75STiwei Bie 		return NULL;
905d79dca75STiwei Bie 
906d79dca75STiwei Bie 	queue_size_in_bytes = vring_size(num, vring_align);
907d79dca75STiwei Bie 	vring_init(&vring, num, queue, vring_align);
908d79dca75STiwei Bie 
909d79dca75STiwei Bie 	vq = __vring_new_virtqueue(index, vring, vdev, weak_barriers, context,
910d79dca75STiwei Bie 				   notify, callback, name);
911d79dca75STiwei Bie 	if (!vq) {
912d79dca75STiwei Bie 		vring_free_queue(vdev, queue_size_in_bytes, queue,
913d79dca75STiwei Bie 				 dma_addr);
914d79dca75STiwei Bie 		return NULL;
915d79dca75STiwei Bie 	}
916d79dca75STiwei Bie 
917d79dca75STiwei Bie 	to_vvq(vq)->split.queue_dma_addr = dma_addr;
918d79dca75STiwei Bie 	to_vvq(vq)->split.queue_size_in_bytes = queue_size_in_bytes;
919d79dca75STiwei Bie 	to_vvq(vq)->we_own_ring = true;
920d79dca75STiwei Bie 
921d79dca75STiwei Bie 	return vq;
922d79dca75STiwei Bie }
923d79dca75STiwei Bie 
924e6f633e5STiwei Bie 
925e6f633e5STiwei Bie /*
9261ce9e605STiwei Bie  * Packed ring specific functions - *_packed().
9271ce9e605STiwei Bie  */
9281ce9e605STiwei Bie 
9291ce9e605STiwei Bie static void vring_unmap_state_packed(const struct vring_virtqueue *vq,
9301f28750fSJason Wang 				     struct vring_desc_extra *state)
9311ce9e605STiwei Bie {
9321ce9e605STiwei Bie 	u16 flags;
9331ce9e605STiwei Bie 
9341ce9e605STiwei Bie 	if (!vq->use_dma_api)
9351ce9e605STiwei Bie 		return;
9361ce9e605STiwei Bie 
9371ce9e605STiwei Bie 	flags = state->flags;
9381ce9e605STiwei Bie 
9391ce9e605STiwei Bie 	if (flags & VRING_DESC_F_INDIRECT) {
9401ce9e605STiwei Bie 		dma_unmap_single(vring_dma_dev(vq),
9411ce9e605STiwei Bie 				 state->addr, state->len,
9421ce9e605STiwei Bie 				 (flags & VRING_DESC_F_WRITE) ?
9431ce9e605STiwei Bie 				 DMA_FROM_DEVICE : DMA_TO_DEVICE);
9441ce9e605STiwei Bie 	} else {
9451ce9e605STiwei Bie 		dma_unmap_page(vring_dma_dev(vq),
9461ce9e605STiwei Bie 			       state->addr, state->len,
9471ce9e605STiwei Bie 			       (flags & VRING_DESC_F_WRITE) ?
9481ce9e605STiwei Bie 			       DMA_FROM_DEVICE : DMA_TO_DEVICE);
9491ce9e605STiwei Bie 	}
9501ce9e605STiwei Bie }
9511ce9e605STiwei Bie 
9521ce9e605STiwei Bie static void vring_unmap_desc_packed(const struct vring_virtqueue *vq,
9531ce9e605STiwei Bie 				   struct vring_packed_desc *desc)
9541ce9e605STiwei Bie {
9551ce9e605STiwei Bie 	u16 flags;
9561ce9e605STiwei Bie 
9571ce9e605STiwei Bie 	if (!vq->use_dma_api)
9581ce9e605STiwei Bie 		return;
9591ce9e605STiwei Bie 
9601ce9e605STiwei Bie 	flags = le16_to_cpu(desc->flags);
9611ce9e605STiwei Bie 
9621ce9e605STiwei Bie 	if (flags & VRING_DESC_F_INDIRECT) {
9631ce9e605STiwei Bie 		dma_unmap_single(vring_dma_dev(vq),
9641ce9e605STiwei Bie 				 le64_to_cpu(desc->addr),
9651ce9e605STiwei Bie 				 le32_to_cpu(desc->len),
9661ce9e605STiwei Bie 				 (flags & VRING_DESC_F_WRITE) ?
9671ce9e605STiwei Bie 				 DMA_FROM_DEVICE : DMA_TO_DEVICE);
9681ce9e605STiwei Bie 	} else {
9691ce9e605STiwei Bie 		dma_unmap_page(vring_dma_dev(vq),
9701ce9e605STiwei Bie 			       le64_to_cpu(desc->addr),
9711ce9e605STiwei Bie 			       le32_to_cpu(desc->len),
9721ce9e605STiwei Bie 			       (flags & VRING_DESC_F_WRITE) ?
9731ce9e605STiwei Bie 			       DMA_FROM_DEVICE : DMA_TO_DEVICE);
9741ce9e605STiwei Bie 	}
9751ce9e605STiwei Bie }
9761ce9e605STiwei Bie 
9771ce9e605STiwei Bie static struct vring_packed_desc *alloc_indirect_packed(unsigned int total_sg,
9781ce9e605STiwei Bie 						       gfp_t gfp)
9791ce9e605STiwei Bie {
9801ce9e605STiwei Bie 	struct vring_packed_desc *desc;
9811ce9e605STiwei Bie 
9821ce9e605STiwei Bie 	/*
9831ce9e605STiwei Bie 	 * We require lowmem mappings for the descriptors because
9841ce9e605STiwei Bie 	 * otherwise virt_to_phys will give us bogus addresses in the
9851ce9e605STiwei Bie 	 * virtqueue.
9861ce9e605STiwei Bie 	 */
9871ce9e605STiwei Bie 	gfp &= ~__GFP_HIGHMEM;
9881ce9e605STiwei Bie 
9891ce9e605STiwei Bie 	desc = kmalloc_array(total_sg, sizeof(struct vring_packed_desc), gfp);
9901ce9e605STiwei Bie 
9911ce9e605STiwei Bie 	return desc;
9921ce9e605STiwei Bie }
9931ce9e605STiwei Bie 
9941ce9e605STiwei Bie static int virtqueue_add_indirect_packed(struct vring_virtqueue *vq,
9951ce9e605STiwei Bie 				       struct scatterlist *sgs[],
9961ce9e605STiwei Bie 				       unsigned int total_sg,
9971ce9e605STiwei Bie 				       unsigned int out_sgs,
9981ce9e605STiwei Bie 				       unsigned int in_sgs,
9991ce9e605STiwei Bie 				       void *data,
10001ce9e605STiwei Bie 				       gfp_t gfp)
10011ce9e605STiwei Bie {
10021ce9e605STiwei Bie 	struct vring_packed_desc *desc;
10031ce9e605STiwei Bie 	struct scatterlist *sg;
10041ce9e605STiwei Bie 	unsigned int i, n, err_idx;
10051ce9e605STiwei Bie 	u16 head, id;
10061ce9e605STiwei Bie 	dma_addr_t addr;
10071ce9e605STiwei Bie 
10081ce9e605STiwei Bie 	head = vq->packed.next_avail_idx;
10091ce9e605STiwei Bie 	desc = alloc_indirect_packed(total_sg, gfp);
10101ce9e605STiwei Bie 
10111ce9e605STiwei Bie 	if (unlikely(vq->vq.num_free < 1)) {
10121ce9e605STiwei Bie 		pr_debug("Can't add buf len 1 - avail = 0\n");
1013df0bfe75SYueHaibing 		kfree(desc);
10141ce9e605STiwei Bie 		END_USE(vq);
10151ce9e605STiwei Bie 		return -ENOSPC;
10161ce9e605STiwei Bie 	}
10171ce9e605STiwei Bie 
10181ce9e605STiwei Bie 	i = 0;
10191ce9e605STiwei Bie 	id = vq->free_head;
10201ce9e605STiwei Bie 	BUG_ON(id == vq->packed.vring.num);
10211ce9e605STiwei Bie 
10221ce9e605STiwei Bie 	for (n = 0; n < out_sgs + in_sgs; n++) {
10231ce9e605STiwei Bie 		for (sg = sgs[n]; sg; sg = sg_next(sg)) {
10241ce9e605STiwei Bie 			addr = vring_map_one_sg(vq, sg, n < out_sgs ?
10251ce9e605STiwei Bie 					DMA_TO_DEVICE : DMA_FROM_DEVICE);
10261ce9e605STiwei Bie 			if (vring_mapping_error(vq, addr))
10271ce9e605STiwei Bie 				goto unmap_release;
10281ce9e605STiwei Bie 
10291ce9e605STiwei Bie 			desc[i].flags = cpu_to_le16(n < out_sgs ?
10301ce9e605STiwei Bie 						0 : VRING_DESC_F_WRITE);
10311ce9e605STiwei Bie 			desc[i].addr = cpu_to_le64(addr);
10321ce9e605STiwei Bie 			desc[i].len = cpu_to_le32(sg->length);
10331ce9e605STiwei Bie 			i++;
10341ce9e605STiwei Bie 		}
10351ce9e605STiwei Bie 	}
10361ce9e605STiwei Bie 
10371ce9e605STiwei Bie 	/* Now that the indirect table is filled in, map it. */
10381ce9e605STiwei Bie 	addr = vring_map_single(vq, desc,
10391ce9e605STiwei Bie 			total_sg * sizeof(struct vring_packed_desc),
10401ce9e605STiwei Bie 			DMA_TO_DEVICE);
10411ce9e605STiwei Bie 	if (vring_mapping_error(vq, addr))
10421ce9e605STiwei Bie 		goto unmap_release;
10431ce9e605STiwei Bie 
10441ce9e605STiwei Bie 	vq->packed.vring.desc[head].addr = cpu_to_le64(addr);
10451ce9e605STiwei Bie 	vq->packed.vring.desc[head].len = cpu_to_le32(total_sg *
10461ce9e605STiwei Bie 				sizeof(struct vring_packed_desc));
10471ce9e605STiwei Bie 	vq->packed.vring.desc[head].id = cpu_to_le16(id);
10481ce9e605STiwei Bie 
10491ce9e605STiwei Bie 	if (vq->use_dma_api) {
10501ce9e605STiwei Bie 		vq->packed.desc_extra[id].addr = addr;
10511ce9e605STiwei Bie 		vq->packed.desc_extra[id].len = total_sg *
10521ce9e605STiwei Bie 				sizeof(struct vring_packed_desc);
10531ce9e605STiwei Bie 		vq->packed.desc_extra[id].flags = VRING_DESC_F_INDIRECT |
10541ce9e605STiwei Bie 						  vq->packed.avail_used_flags;
10551ce9e605STiwei Bie 	}
10561ce9e605STiwei Bie 
10571ce9e605STiwei Bie 	/*
10581ce9e605STiwei Bie 	 * A driver MUST NOT make the first descriptor in the list
10591ce9e605STiwei Bie 	 * available before all subsequent descriptors comprising
10601ce9e605STiwei Bie 	 * the list are made available.
10611ce9e605STiwei Bie 	 */
10621ce9e605STiwei Bie 	virtio_wmb(vq->weak_barriers);
10631ce9e605STiwei Bie 	vq->packed.vring.desc[head].flags = cpu_to_le16(VRING_DESC_F_INDIRECT |
10641ce9e605STiwei Bie 						vq->packed.avail_used_flags);
10651ce9e605STiwei Bie 
10661ce9e605STiwei Bie 	/* We're using some buffers from the free list. */
10671ce9e605STiwei Bie 	vq->vq.num_free -= 1;
10681ce9e605STiwei Bie 
10691ce9e605STiwei Bie 	/* Update free pointer */
10701ce9e605STiwei Bie 	n = head + 1;
10711ce9e605STiwei Bie 	if (n >= vq->packed.vring.num) {
10721ce9e605STiwei Bie 		n = 0;
10731ce9e605STiwei Bie 		vq->packed.avail_wrap_counter ^= 1;
10741ce9e605STiwei Bie 		vq->packed.avail_used_flags ^=
10751ce9e605STiwei Bie 				1 << VRING_PACKED_DESC_F_AVAIL |
10761ce9e605STiwei Bie 				1 << VRING_PACKED_DESC_F_USED;
10771ce9e605STiwei Bie 	}
10781ce9e605STiwei Bie 	vq->packed.next_avail_idx = n;
1079aeef9b47SJason Wang 	vq->free_head = vq->packed.desc_extra[id].next;
10801ce9e605STiwei Bie 
10811ce9e605STiwei Bie 	/* Store token and indirect buffer state. */
10821ce9e605STiwei Bie 	vq->packed.desc_state[id].num = 1;
10831ce9e605STiwei Bie 	vq->packed.desc_state[id].data = data;
10841ce9e605STiwei Bie 	vq->packed.desc_state[id].indir_desc = desc;
10851ce9e605STiwei Bie 	vq->packed.desc_state[id].last = id;
10861ce9e605STiwei Bie 
10871ce9e605STiwei Bie 	vq->num_added += 1;
10881ce9e605STiwei Bie 
10891ce9e605STiwei Bie 	pr_debug("Added buffer head %i to %p\n", head, vq);
10901ce9e605STiwei Bie 	END_USE(vq);
10911ce9e605STiwei Bie 
10921ce9e605STiwei Bie 	return 0;
10931ce9e605STiwei Bie 
10941ce9e605STiwei Bie unmap_release:
10951ce9e605STiwei Bie 	err_idx = i;
10961ce9e605STiwei Bie 
10971ce9e605STiwei Bie 	for (i = 0; i < err_idx; i++)
10981ce9e605STiwei Bie 		vring_unmap_desc_packed(vq, &desc[i]);
10991ce9e605STiwei Bie 
11001ce9e605STiwei Bie 	kfree(desc);
11011ce9e605STiwei Bie 
11021ce9e605STiwei Bie 	END_USE(vq);
1103f7728002SHalil Pasic 	return -ENOMEM;
11041ce9e605STiwei Bie }
11051ce9e605STiwei Bie 
11061ce9e605STiwei Bie static inline int virtqueue_add_packed(struct virtqueue *_vq,
11071ce9e605STiwei Bie 				       struct scatterlist *sgs[],
11081ce9e605STiwei Bie 				       unsigned int total_sg,
11091ce9e605STiwei Bie 				       unsigned int out_sgs,
11101ce9e605STiwei Bie 				       unsigned int in_sgs,
11111ce9e605STiwei Bie 				       void *data,
11121ce9e605STiwei Bie 				       void *ctx,
11131ce9e605STiwei Bie 				       gfp_t gfp)
11141ce9e605STiwei Bie {
11151ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
11161ce9e605STiwei Bie 	struct vring_packed_desc *desc;
11171ce9e605STiwei Bie 	struct scatterlist *sg;
11181ce9e605STiwei Bie 	unsigned int i, n, c, descs_used, err_idx;
11193f649ab7SKees Cook 	__le16 head_flags, flags;
11203f649ab7SKees Cook 	u16 head, id, prev, curr, avail_used_flags;
11211ce9e605STiwei Bie 
11221ce9e605STiwei Bie 	START_USE(vq);
11231ce9e605STiwei Bie 
11241ce9e605STiwei Bie 	BUG_ON(data == NULL);
11251ce9e605STiwei Bie 	BUG_ON(ctx && vq->indirect);
11261ce9e605STiwei Bie 
11271ce9e605STiwei Bie 	if (unlikely(vq->broken)) {
11281ce9e605STiwei Bie 		END_USE(vq);
11291ce9e605STiwei Bie 		return -EIO;
11301ce9e605STiwei Bie 	}
11311ce9e605STiwei Bie 
11321ce9e605STiwei Bie 	LAST_ADD_TIME_UPDATE(vq);
11331ce9e605STiwei Bie 
11341ce9e605STiwei Bie 	BUG_ON(total_sg == 0);
11351ce9e605STiwei Bie 
11361ce9e605STiwei Bie 	if (virtqueue_use_indirect(_vq, total_sg))
11371ce9e605STiwei Bie 		return virtqueue_add_indirect_packed(vq, sgs, total_sg,
11381ce9e605STiwei Bie 				out_sgs, in_sgs, data, gfp);
11391ce9e605STiwei Bie 
11401ce9e605STiwei Bie 	head = vq->packed.next_avail_idx;
11411ce9e605STiwei Bie 	avail_used_flags = vq->packed.avail_used_flags;
11421ce9e605STiwei Bie 
11431ce9e605STiwei Bie 	WARN_ON_ONCE(total_sg > vq->packed.vring.num && !vq->indirect);
11441ce9e605STiwei Bie 
11451ce9e605STiwei Bie 	desc = vq->packed.vring.desc;
11461ce9e605STiwei Bie 	i = head;
11471ce9e605STiwei Bie 	descs_used = total_sg;
11481ce9e605STiwei Bie 
11491ce9e605STiwei Bie 	if (unlikely(vq->vq.num_free < descs_used)) {
11501ce9e605STiwei Bie 		pr_debug("Can't add buf len %i - avail = %i\n",
11511ce9e605STiwei Bie 			 descs_used, vq->vq.num_free);
11521ce9e605STiwei Bie 		END_USE(vq);
11531ce9e605STiwei Bie 		return -ENOSPC;
11541ce9e605STiwei Bie 	}
11551ce9e605STiwei Bie 
11561ce9e605STiwei Bie 	id = vq->free_head;
11571ce9e605STiwei Bie 	BUG_ON(id == vq->packed.vring.num);
11581ce9e605STiwei Bie 
11591ce9e605STiwei Bie 	curr = id;
11601ce9e605STiwei Bie 	c = 0;
11611ce9e605STiwei Bie 	for (n = 0; n < out_sgs + in_sgs; n++) {
11621ce9e605STiwei Bie 		for (sg = sgs[n]; sg; sg = sg_next(sg)) {
11631ce9e605STiwei Bie 			dma_addr_t addr = vring_map_one_sg(vq, sg, n < out_sgs ?
11641ce9e605STiwei Bie 					DMA_TO_DEVICE : DMA_FROM_DEVICE);
11651ce9e605STiwei Bie 			if (vring_mapping_error(vq, addr))
11661ce9e605STiwei Bie 				goto unmap_release;
11671ce9e605STiwei Bie 
11681ce9e605STiwei Bie 			flags = cpu_to_le16(vq->packed.avail_used_flags |
11691ce9e605STiwei Bie 				    (++c == total_sg ? 0 : VRING_DESC_F_NEXT) |
11701ce9e605STiwei Bie 				    (n < out_sgs ? 0 : VRING_DESC_F_WRITE));
11711ce9e605STiwei Bie 			if (i == head)
11721ce9e605STiwei Bie 				head_flags = flags;
11731ce9e605STiwei Bie 			else
11741ce9e605STiwei Bie 				desc[i].flags = flags;
11751ce9e605STiwei Bie 
11761ce9e605STiwei Bie 			desc[i].addr = cpu_to_le64(addr);
11771ce9e605STiwei Bie 			desc[i].len = cpu_to_le32(sg->length);
11781ce9e605STiwei Bie 			desc[i].id = cpu_to_le16(id);
11791ce9e605STiwei Bie 
11801ce9e605STiwei Bie 			if (unlikely(vq->use_dma_api)) {
11811ce9e605STiwei Bie 				vq->packed.desc_extra[curr].addr = addr;
11821ce9e605STiwei Bie 				vq->packed.desc_extra[curr].len = sg->length;
11831ce9e605STiwei Bie 				vq->packed.desc_extra[curr].flags =
11841ce9e605STiwei Bie 					le16_to_cpu(flags);
11851ce9e605STiwei Bie 			}
11861ce9e605STiwei Bie 			prev = curr;
1187aeef9b47SJason Wang 			curr = vq->packed.desc_extra[curr].next;
11881ce9e605STiwei Bie 
11891ce9e605STiwei Bie 			if ((unlikely(++i >= vq->packed.vring.num))) {
11901ce9e605STiwei Bie 				i = 0;
11911ce9e605STiwei Bie 				vq->packed.avail_used_flags ^=
11921ce9e605STiwei Bie 					1 << VRING_PACKED_DESC_F_AVAIL |
11931ce9e605STiwei Bie 					1 << VRING_PACKED_DESC_F_USED;
11941ce9e605STiwei Bie 			}
11951ce9e605STiwei Bie 		}
11961ce9e605STiwei Bie 	}
11971ce9e605STiwei Bie 
11981ce9e605STiwei Bie 	if (i < head)
11991ce9e605STiwei Bie 		vq->packed.avail_wrap_counter ^= 1;
12001ce9e605STiwei Bie 
12011ce9e605STiwei Bie 	/* We're using some buffers from the free list. */
12021ce9e605STiwei Bie 	vq->vq.num_free -= descs_used;
12031ce9e605STiwei Bie 
12041ce9e605STiwei Bie 	/* Update free pointer */
12051ce9e605STiwei Bie 	vq->packed.next_avail_idx = i;
12061ce9e605STiwei Bie 	vq->free_head = curr;
12071ce9e605STiwei Bie 
12081ce9e605STiwei Bie 	/* Store token. */
12091ce9e605STiwei Bie 	vq->packed.desc_state[id].num = descs_used;
12101ce9e605STiwei Bie 	vq->packed.desc_state[id].data = data;
12111ce9e605STiwei Bie 	vq->packed.desc_state[id].indir_desc = ctx;
12121ce9e605STiwei Bie 	vq->packed.desc_state[id].last = prev;
12131ce9e605STiwei Bie 
12141ce9e605STiwei Bie 	/*
12151ce9e605STiwei Bie 	 * A driver MUST NOT make the first descriptor in the list
12161ce9e605STiwei Bie 	 * available before all subsequent descriptors comprising
12171ce9e605STiwei Bie 	 * the list are made available.
12181ce9e605STiwei Bie 	 */
12191ce9e605STiwei Bie 	virtio_wmb(vq->weak_barriers);
12201ce9e605STiwei Bie 	vq->packed.vring.desc[head].flags = head_flags;
12211ce9e605STiwei Bie 	vq->num_added += descs_used;
12221ce9e605STiwei Bie 
12231ce9e605STiwei Bie 	pr_debug("Added buffer head %i to %p\n", head, vq);
12241ce9e605STiwei Bie 	END_USE(vq);
12251ce9e605STiwei Bie 
12261ce9e605STiwei Bie 	return 0;
12271ce9e605STiwei Bie 
12281ce9e605STiwei Bie unmap_release:
12291ce9e605STiwei Bie 	err_idx = i;
12301ce9e605STiwei Bie 	i = head;
123144593865SJason Wang 	curr = vq->free_head;
12321ce9e605STiwei Bie 
12331ce9e605STiwei Bie 	vq->packed.avail_used_flags = avail_used_flags;
12341ce9e605STiwei Bie 
12351ce9e605STiwei Bie 	for (n = 0; n < total_sg; n++) {
12361ce9e605STiwei Bie 		if (i == err_idx)
12371ce9e605STiwei Bie 			break;
123844593865SJason Wang 		vring_unmap_state_packed(vq,
123944593865SJason Wang 					 &vq->packed.desc_extra[curr]);
124044593865SJason Wang 		curr = vq->packed.desc_extra[curr].next;
12411ce9e605STiwei Bie 		i++;
12421ce9e605STiwei Bie 		if (i >= vq->packed.vring.num)
12431ce9e605STiwei Bie 			i = 0;
12441ce9e605STiwei Bie 	}
12451ce9e605STiwei Bie 
12461ce9e605STiwei Bie 	END_USE(vq);
12471ce9e605STiwei Bie 	return -EIO;
12481ce9e605STiwei Bie }
12491ce9e605STiwei Bie 
12501ce9e605STiwei Bie static bool virtqueue_kick_prepare_packed(struct virtqueue *_vq)
12511ce9e605STiwei Bie {
12521ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
1253f51f9826STiwei Bie 	u16 new, old, off_wrap, flags, wrap_counter, event_idx;
12541ce9e605STiwei Bie 	bool needs_kick;
12551ce9e605STiwei Bie 	union {
12561ce9e605STiwei Bie 		struct {
12571ce9e605STiwei Bie 			__le16 off_wrap;
12581ce9e605STiwei Bie 			__le16 flags;
12591ce9e605STiwei Bie 		};
12601ce9e605STiwei Bie 		u32 u32;
12611ce9e605STiwei Bie 	} snapshot;
12621ce9e605STiwei Bie 
12631ce9e605STiwei Bie 	START_USE(vq);
12641ce9e605STiwei Bie 
12651ce9e605STiwei Bie 	/*
12661ce9e605STiwei Bie 	 * We need to expose the new flags value before checking notification
12671ce9e605STiwei Bie 	 * suppressions.
12681ce9e605STiwei Bie 	 */
12691ce9e605STiwei Bie 	virtio_mb(vq->weak_barriers);
12701ce9e605STiwei Bie 
1271f51f9826STiwei Bie 	old = vq->packed.next_avail_idx - vq->num_added;
1272f51f9826STiwei Bie 	new = vq->packed.next_avail_idx;
12731ce9e605STiwei Bie 	vq->num_added = 0;
12741ce9e605STiwei Bie 
12751ce9e605STiwei Bie 	snapshot.u32 = *(u32 *)vq->packed.vring.device;
12761ce9e605STiwei Bie 	flags = le16_to_cpu(snapshot.flags);
12771ce9e605STiwei Bie 
12781ce9e605STiwei Bie 	LAST_ADD_TIME_CHECK(vq);
12791ce9e605STiwei Bie 	LAST_ADD_TIME_INVALID(vq);
12801ce9e605STiwei Bie 
1281f51f9826STiwei Bie 	if (flags != VRING_PACKED_EVENT_FLAG_DESC) {
12821ce9e605STiwei Bie 		needs_kick = (flags != VRING_PACKED_EVENT_FLAG_DISABLE);
1283f51f9826STiwei Bie 		goto out;
1284f51f9826STiwei Bie 	}
1285f51f9826STiwei Bie 
1286f51f9826STiwei Bie 	off_wrap = le16_to_cpu(snapshot.off_wrap);
1287f51f9826STiwei Bie 
1288f51f9826STiwei Bie 	wrap_counter = off_wrap >> VRING_PACKED_EVENT_F_WRAP_CTR;
1289f51f9826STiwei Bie 	event_idx = off_wrap & ~(1 << VRING_PACKED_EVENT_F_WRAP_CTR);
1290f51f9826STiwei Bie 	if (wrap_counter != vq->packed.avail_wrap_counter)
1291f51f9826STiwei Bie 		event_idx -= vq->packed.vring.num;
1292f51f9826STiwei Bie 
1293f51f9826STiwei Bie 	needs_kick = vring_need_event(event_idx, new, old);
1294f51f9826STiwei Bie out:
12951ce9e605STiwei Bie 	END_USE(vq);
12961ce9e605STiwei Bie 	return needs_kick;
12971ce9e605STiwei Bie }
12981ce9e605STiwei Bie 
12991ce9e605STiwei Bie static void detach_buf_packed(struct vring_virtqueue *vq,
13001ce9e605STiwei Bie 			      unsigned int id, void **ctx)
13011ce9e605STiwei Bie {
13021ce9e605STiwei Bie 	struct vring_desc_state_packed *state = NULL;
13031ce9e605STiwei Bie 	struct vring_packed_desc *desc;
13041ce9e605STiwei Bie 	unsigned int i, curr;
13051ce9e605STiwei Bie 
13061ce9e605STiwei Bie 	state = &vq->packed.desc_state[id];
13071ce9e605STiwei Bie 
13081ce9e605STiwei Bie 	/* Clear data ptr. */
13091ce9e605STiwei Bie 	state->data = NULL;
13101ce9e605STiwei Bie 
1311aeef9b47SJason Wang 	vq->packed.desc_extra[state->last].next = vq->free_head;
13121ce9e605STiwei Bie 	vq->free_head = id;
13131ce9e605STiwei Bie 	vq->vq.num_free += state->num;
13141ce9e605STiwei Bie 
13151ce9e605STiwei Bie 	if (unlikely(vq->use_dma_api)) {
13161ce9e605STiwei Bie 		curr = id;
13171ce9e605STiwei Bie 		for (i = 0; i < state->num; i++) {
13181ce9e605STiwei Bie 			vring_unmap_state_packed(vq,
13191ce9e605STiwei Bie 				&vq->packed.desc_extra[curr]);
1320aeef9b47SJason Wang 			curr = vq->packed.desc_extra[curr].next;
13211ce9e605STiwei Bie 		}
13221ce9e605STiwei Bie 	}
13231ce9e605STiwei Bie 
13241ce9e605STiwei Bie 	if (vq->indirect) {
13251ce9e605STiwei Bie 		u32 len;
13261ce9e605STiwei Bie 
13271ce9e605STiwei Bie 		/* Free the indirect table, if any, now that it's unmapped. */
13281ce9e605STiwei Bie 		desc = state->indir_desc;
13291ce9e605STiwei Bie 		if (!desc)
13301ce9e605STiwei Bie 			return;
13311ce9e605STiwei Bie 
13321ce9e605STiwei Bie 		if (vq->use_dma_api) {
13331ce9e605STiwei Bie 			len = vq->packed.desc_extra[id].len;
13341ce9e605STiwei Bie 			for (i = 0; i < len / sizeof(struct vring_packed_desc);
13351ce9e605STiwei Bie 					i++)
13361ce9e605STiwei Bie 				vring_unmap_desc_packed(vq, &desc[i]);
13371ce9e605STiwei Bie 		}
13381ce9e605STiwei Bie 		kfree(desc);
13391ce9e605STiwei Bie 		state->indir_desc = NULL;
13401ce9e605STiwei Bie 	} else if (ctx) {
13411ce9e605STiwei Bie 		*ctx = state->indir_desc;
13421ce9e605STiwei Bie 	}
13431ce9e605STiwei Bie }
13441ce9e605STiwei Bie 
13451ce9e605STiwei Bie static inline bool is_used_desc_packed(const struct vring_virtqueue *vq,
13461ce9e605STiwei Bie 				       u16 idx, bool used_wrap_counter)
13471ce9e605STiwei Bie {
13481ce9e605STiwei Bie 	bool avail, used;
13491ce9e605STiwei Bie 	u16 flags;
13501ce9e605STiwei Bie 
13511ce9e605STiwei Bie 	flags = le16_to_cpu(vq->packed.vring.desc[idx].flags);
13521ce9e605STiwei Bie 	avail = !!(flags & (1 << VRING_PACKED_DESC_F_AVAIL));
13531ce9e605STiwei Bie 	used = !!(flags & (1 << VRING_PACKED_DESC_F_USED));
13541ce9e605STiwei Bie 
13551ce9e605STiwei Bie 	return avail == used && used == used_wrap_counter;
13561ce9e605STiwei Bie }
13571ce9e605STiwei Bie 
13581ce9e605STiwei Bie static inline bool more_used_packed(const struct vring_virtqueue *vq)
13591ce9e605STiwei Bie {
13601ce9e605STiwei Bie 	return is_used_desc_packed(vq, vq->last_used_idx,
13611ce9e605STiwei Bie 			vq->packed.used_wrap_counter);
13621ce9e605STiwei Bie }
13631ce9e605STiwei Bie 
13641ce9e605STiwei Bie static void *virtqueue_get_buf_ctx_packed(struct virtqueue *_vq,
13651ce9e605STiwei Bie 					  unsigned int *len,
13661ce9e605STiwei Bie 					  void **ctx)
13671ce9e605STiwei Bie {
13681ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
13691ce9e605STiwei Bie 	u16 last_used, id;
13701ce9e605STiwei Bie 	void *ret;
13711ce9e605STiwei Bie 
13721ce9e605STiwei Bie 	START_USE(vq);
13731ce9e605STiwei Bie 
13741ce9e605STiwei Bie 	if (unlikely(vq->broken)) {
13751ce9e605STiwei Bie 		END_USE(vq);
13761ce9e605STiwei Bie 		return NULL;
13771ce9e605STiwei Bie 	}
13781ce9e605STiwei Bie 
13791ce9e605STiwei Bie 	if (!more_used_packed(vq)) {
13801ce9e605STiwei Bie 		pr_debug("No more buffers in queue\n");
13811ce9e605STiwei Bie 		END_USE(vq);
13821ce9e605STiwei Bie 		return NULL;
13831ce9e605STiwei Bie 	}
13841ce9e605STiwei Bie 
13851ce9e605STiwei Bie 	/* Only get used elements after they have been exposed by host. */
13861ce9e605STiwei Bie 	virtio_rmb(vq->weak_barriers);
13871ce9e605STiwei Bie 
13881ce9e605STiwei Bie 	last_used = vq->last_used_idx;
13891ce9e605STiwei Bie 	id = le16_to_cpu(vq->packed.vring.desc[last_used].id);
13901ce9e605STiwei Bie 	*len = le32_to_cpu(vq->packed.vring.desc[last_used].len);
13911ce9e605STiwei Bie 
13921ce9e605STiwei Bie 	if (unlikely(id >= vq->packed.vring.num)) {
13931ce9e605STiwei Bie 		BAD_RING(vq, "id %u out of range\n", id);
13941ce9e605STiwei Bie 		return NULL;
13951ce9e605STiwei Bie 	}
13961ce9e605STiwei Bie 	if (unlikely(!vq->packed.desc_state[id].data)) {
13971ce9e605STiwei Bie 		BAD_RING(vq, "id %u is not a head!\n", id);
13981ce9e605STiwei Bie 		return NULL;
13991ce9e605STiwei Bie 	}
14001ce9e605STiwei Bie 
14011ce9e605STiwei Bie 	/* detach_buf_packed clears data, so grab it now. */
14021ce9e605STiwei Bie 	ret = vq->packed.desc_state[id].data;
14031ce9e605STiwei Bie 	detach_buf_packed(vq, id, ctx);
14041ce9e605STiwei Bie 
14051ce9e605STiwei Bie 	vq->last_used_idx += vq->packed.desc_state[id].num;
14061ce9e605STiwei Bie 	if (unlikely(vq->last_used_idx >= vq->packed.vring.num)) {
14071ce9e605STiwei Bie 		vq->last_used_idx -= vq->packed.vring.num;
14081ce9e605STiwei Bie 		vq->packed.used_wrap_counter ^= 1;
14091ce9e605STiwei Bie 	}
14101ce9e605STiwei Bie 
1411f51f9826STiwei Bie 	/*
1412f51f9826STiwei Bie 	 * If we expect an interrupt for the next entry, tell host
1413f51f9826STiwei Bie 	 * by writing event index and flush out the write before
1414f51f9826STiwei Bie 	 * the read in the next get_buf call.
1415f51f9826STiwei Bie 	 */
1416f51f9826STiwei Bie 	if (vq->packed.event_flags_shadow == VRING_PACKED_EVENT_FLAG_DESC)
1417f51f9826STiwei Bie 		virtio_store_mb(vq->weak_barriers,
1418f51f9826STiwei Bie 				&vq->packed.vring.driver->off_wrap,
1419f51f9826STiwei Bie 				cpu_to_le16(vq->last_used_idx |
1420f51f9826STiwei Bie 					(vq->packed.used_wrap_counter <<
1421f51f9826STiwei Bie 					 VRING_PACKED_EVENT_F_WRAP_CTR)));
1422f51f9826STiwei Bie 
14231ce9e605STiwei Bie 	LAST_ADD_TIME_INVALID(vq);
14241ce9e605STiwei Bie 
14251ce9e605STiwei Bie 	END_USE(vq);
14261ce9e605STiwei Bie 	return ret;
14271ce9e605STiwei Bie }
14281ce9e605STiwei Bie 
14291ce9e605STiwei Bie static void virtqueue_disable_cb_packed(struct virtqueue *_vq)
14301ce9e605STiwei Bie {
14311ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
14321ce9e605STiwei Bie 
14331ce9e605STiwei Bie 	if (vq->packed.event_flags_shadow != VRING_PACKED_EVENT_FLAG_DISABLE) {
14341ce9e605STiwei Bie 		vq->packed.event_flags_shadow = VRING_PACKED_EVENT_FLAG_DISABLE;
14351ce9e605STiwei Bie 		vq->packed.vring.driver->flags =
14361ce9e605STiwei Bie 			cpu_to_le16(vq->packed.event_flags_shadow);
14371ce9e605STiwei Bie 	}
14381ce9e605STiwei Bie }
14391ce9e605STiwei Bie 
14401ce9e605STiwei Bie static unsigned virtqueue_enable_cb_prepare_packed(struct virtqueue *_vq)
14411ce9e605STiwei Bie {
14421ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
14431ce9e605STiwei Bie 
14441ce9e605STiwei Bie 	START_USE(vq);
14451ce9e605STiwei Bie 
14461ce9e605STiwei Bie 	/*
14471ce9e605STiwei Bie 	 * We optimistically turn back on interrupts, then check if there was
14481ce9e605STiwei Bie 	 * more to do.
14491ce9e605STiwei Bie 	 */
14501ce9e605STiwei Bie 
1451f51f9826STiwei Bie 	if (vq->event) {
1452f51f9826STiwei Bie 		vq->packed.vring.driver->off_wrap =
1453f51f9826STiwei Bie 			cpu_to_le16(vq->last_used_idx |
1454f51f9826STiwei Bie 				(vq->packed.used_wrap_counter <<
1455f51f9826STiwei Bie 				 VRING_PACKED_EVENT_F_WRAP_CTR));
1456f51f9826STiwei Bie 		/*
1457f51f9826STiwei Bie 		 * We need to update event offset and event wrap
1458f51f9826STiwei Bie 		 * counter first before updating event flags.
1459f51f9826STiwei Bie 		 */
1460f51f9826STiwei Bie 		virtio_wmb(vq->weak_barriers);
1461f51f9826STiwei Bie 	}
1462f51f9826STiwei Bie 
14631ce9e605STiwei Bie 	if (vq->packed.event_flags_shadow == VRING_PACKED_EVENT_FLAG_DISABLE) {
1464f51f9826STiwei Bie 		vq->packed.event_flags_shadow = vq->event ?
1465f51f9826STiwei Bie 				VRING_PACKED_EVENT_FLAG_DESC :
1466f51f9826STiwei Bie 				VRING_PACKED_EVENT_FLAG_ENABLE;
14671ce9e605STiwei Bie 		vq->packed.vring.driver->flags =
14681ce9e605STiwei Bie 				cpu_to_le16(vq->packed.event_flags_shadow);
14691ce9e605STiwei Bie 	}
14701ce9e605STiwei Bie 
14711ce9e605STiwei Bie 	END_USE(vq);
14721ce9e605STiwei Bie 	return vq->last_used_idx | ((u16)vq->packed.used_wrap_counter <<
14731ce9e605STiwei Bie 			VRING_PACKED_EVENT_F_WRAP_CTR);
14741ce9e605STiwei Bie }
14751ce9e605STiwei Bie 
14761ce9e605STiwei Bie static bool virtqueue_poll_packed(struct virtqueue *_vq, u16 off_wrap)
14771ce9e605STiwei Bie {
14781ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
14791ce9e605STiwei Bie 	bool wrap_counter;
14801ce9e605STiwei Bie 	u16 used_idx;
14811ce9e605STiwei Bie 
14821ce9e605STiwei Bie 	wrap_counter = off_wrap >> VRING_PACKED_EVENT_F_WRAP_CTR;
14831ce9e605STiwei Bie 	used_idx = off_wrap & ~(1 << VRING_PACKED_EVENT_F_WRAP_CTR);
14841ce9e605STiwei Bie 
14851ce9e605STiwei Bie 	return is_used_desc_packed(vq, used_idx, wrap_counter);
14861ce9e605STiwei Bie }
14871ce9e605STiwei Bie 
14881ce9e605STiwei Bie static bool virtqueue_enable_cb_delayed_packed(struct virtqueue *_vq)
14891ce9e605STiwei Bie {
14901ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
14911ce9e605STiwei Bie 	u16 used_idx, wrap_counter;
1492f51f9826STiwei Bie 	u16 bufs;
14931ce9e605STiwei Bie 
14941ce9e605STiwei Bie 	START_USE(vq);
14951ce9e605STiwei Bie 
14961ce9e605STiwei Bie 	/*
14971ce9e605STiwei Bie 	 * We optimistically turn back on interrupts, then check if there was
14981ce9e605STiwei Bie 	 * more to do.
14991ce9e605STiwei Bie 	 */
15001ce9e605STiwei Bie 
1501f51f9826STiwei Bie 	if (vq->event) {
1502f51f9826STiwei Bie 		/* TODO: tune this threshold */
1503f51f9826STiwei Bie 		bufs = (vq->packed.vring.num - vq->vq.num_free) * 3 / 4;
15041ce9e605STiwei Bie 		wrap_counter = vq->packed.used_wrap_counter;
15051ce9e605STiwei Bie 
1506f51f9826STiwei Bie 		used_idx = vq->last_used_idx + bufs;
1507f51f9826STiwei Bie 		if (used_idx >= vq->packed.vring.num) {
1508f51f9826STiwei Bie 			used_idx -= vq->packed.vring.num;
1509f51f9826STiwei Bie 			wrap_counter ^= 1;
1510f51f9826STiwei Bie 		}
1511f51f9826STiwei Bie 
1512f51f9826STiwei Bie 		vq->packed.vring.driver->off_wrap = cpu_to_le16(used_idx |
1513f51f9826STiwei Bie 			(wrap_counter << VRING_PACKED_EVENT_F_WRAP_CTR));
1514f51f9826STiwei Bie 
1515f51f9826STiwei Bie 		/*
1516f51f9826STiwei Bie 		 * We need to update event offset and event wrap
1517f51f9826STiwei Bie 		 * counter first before updating event flags.
1518f51f9826STiwei Bie 		 */
1519f51f9826STiwei Bie 		virtio_wmb(vq->weak_barriers);
1520f51f9826STiwei Bie 	}
1521f51f9826STiwei Bie 
15221ce9e605STiwei Bie 	if (vq->packed.event_flags_shadow == VRING_PACKED_EVENT_FLAG_DISABLE) {
1523f51f9826STiwei Bie 		vq->packed.event_flags_shadow = vq->event ?
1524f51f9826STiwei Bie 				VRING_PACKED_EVENT_FLAG_DESC :
1525f51f9826STiwei Bie 				VRING_PACKED_EVENT_FLAG_ENABLE;
15261ce9e605STiwei Bie 		vq->packed.vring.driver->flags =
15271ce9e605STiwei Bie 				cpu_to_le16(vq->packed.event_flags_shadow);
15281ce9e605STiwei Bie 	}
15291ce9e605STiwei Bie 
15301ce9e605STiwei Bie 	/*
15311ce9e605STiwei Bie 	 * We need to update event suppression structure first
15321ce9e605STiwei Bie 	 * before re-checking for more used buffers.
15331ce9e605STiwei Bie 	 */
15341ce9e605STiwei Bie 	virtio_mb(vq->weak_barriers);
15351ce9e605STiwei Bie 
153640ce7919SMarvin Liu 	if (is_used_desc_packed(vq,
153740ce7919SMarvin Liu 				vq->last_used_idx,
153840ce7919SMarvin Liu 				vq->packed.used_wrap_counter)) {
15391ce9e605STiwei Bie 		END_USE(vq);
15401ce9e605STiwei Bie 		return false;
15411ce9e605STiwei Bie 	}
15421ce9e605STiwei Bie 
15431ce9e605STiwei Bie 	END_USE(vq);
15441ce9e605STiwei Bie 	return true;
15451ce9e605STiwei Bie }
15461ce9e605STiwei Bie 
15471ce9e605STiwei Bie static void *virtqueue_detach_unused_buf_packed(struct virtqueue *_vq)
15481ce9e605STiwei Bie {
15491ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
15501ce9e605STiwei Bie 	unsigned int i;
15511ce9e605STiwei Bie 	void *buf;
15521ce9e605STiwei Bie 
15531ce9e605STiwei Bie 	START_USE(vq);
15541ce9e605STiwei Bie 
15551ce9e605STiwei Bie 	for (i = 0; i < vq->packed.vring.num; i++) {
15561ce9e605STiwei Bie 		if (!vq->packed.desc_state[i].data)
15571ce9e605STiwei Bie 			continue;
15581ce9e605STiwei Bie 		/* detach_buf clears data, so grab it now. */
15591ce9e605STiwei Bie 		buf = vq->packed.desc_state[i].data;
15601ce9e605STiwei Bie 		detach_buf_packed(vq, i, NULL);
15611ce9e605STiwei Bie 		END_USE(vq);
15621ce9e605STiwei Bie 		return buf;
15631ce9e605STiwei Bie 	}
15641ce9e605STiwei Bie 	/* That should have freed everything. */
15651ce9e605STiwei Bie 	BUG_ON(vq->vq.num_free != vq->packed.vring.num);
15661ce9e605STiwei Bie 
15671ce9e605STiwei Bie 	END_USE(vq);
15681ce9e605STiwei Bie 	return NULL;
15691ce9e605STiwei Bie }
15701ce9e605STiwei Bie 
15715a222421SJason Wang static struct vring_desc_extra *vring_alloc_desc_extra(struct vring_virtqueue *vq,
15725a222421SJason Wang 						       unsigned int num)
15735a222421SJason Wang {
15745a222421SJason Wang 	struct vring_desc_extra *desc_extra;
15755a222421SJason Wang 	unsigned int i;
15765a222421SJason Wang 
15775a222421SJason Wang 	desc_extra = kmalloc_array(num, sizeof(struct vring_desc_extra),
15785a222421SJason Wang 				   GFP_KERNEL);
15795a222421SJason Wang 	if (!desc_extra)
15805a222421SJason Wang 		return NULL;
15815a222421SJason Wang 
15825a222421SJason Wang 	memset(desc_extra, 0, num * sizeof(struct vring_desc_extra));
15835a222421SJason Wang 
15845a222421SJason Wang 	for (i = 0; i < num - 1; i++)
15855a222421SJason Wang 		desc_extra[i].next = i + 1;
15865a222421SJason Wang 
15875a222421SJason Wang 	return desc_extra;
15885a222421SJason Wang }
15895a222421SJason Wang 
15901ce9e605STiwei Bie static struct virtqueue *vring_create_virtqueue_packed(
15911ce9e605STiwei Bie 	unsigned int index,
15921ce9e605STiwei Bie 	unsigned int num,
15931ce9e605STiwei Bie 	unsigned int vring_align,
15941ce9e605STiwei Bie 	struct virtio_device *vdev,
15951ce9e605STiwei Bie 	bool weak_barriers,
15961ce9e605STiwei Bie 	bool may_reduce_num,
15971ce9e605STiwei Bie 	bool context,
15981ce9e605STiwei Bie 	bool (*notify)(struct virtqueue *),
15991ce9e605STiwei Bie 	void (*callback)(struct virtqueue *),
16001ce9e605STiwei Bie 	const char *name)
16011ce9e605STiwei Bie {
16021ce9e605STiwei Bie 	struct vring_virtqueue *vq;
16031ce9e605STiwei Bie 	struct vring_packed_desc *ring;
16041ce9e605STiwei Bie 	struct vring_packed_desc_event *driver, *device;
16051ce9e605STiwei Bie 	dma_addr_t ring_dma_addr, driver_event_dma_addr, device_event_dma_addr;
16061ce9e605STiwei Bie 	size_t ring_size_in_bytes, event_size_in_bytes;
16071ce9e605STiwei Bie 
16081ce9e605STiwei Bie 	ring_size_in_bytes = num * sizeof(struct vring_packed_desc);
16091ce9e605STiwei Bie 
16101ce9e605STiwei Bie 	ring = vring_alloc_queue(vdev, ring_size_in_bytes,
16111ce9e605STiwei Bie 				 &ring_dma_addr,
16121ce9e605STiwei Bie 				 GFP_KERNEL|__GFP_NOWARN|__GFP_ZERO);
16131ce9e605STiwei Bie 	if (!ring)
16141ce9e605STiwei Bie 		goto err_ring;
16151ce9e605STiwei Bie 
16161ce9e605STiwei Bie 	event_size_in_bytes = sizeof(struct vring_packed_desc_event);
16171ce9e605STiwei Bie 
16181ce9e605STiwei Bie 	driver = vring_alloc_queue(vdev, event_size_in_bytes,
16191ce9e605STiwei Bie 				   &driver_event_dma_addr,
16201ce9e605STiwei Bie 				   GFP_KERNEL|__GFP_NOWARN|__GFP_ZERO);
16211ce9e605STiwei Bie 	if (!driver)
16221ce9e605STiwei Bie 		goto err_driver;
16231ce9e605STiwei Bie 
16241ce9e605STiwei Bie 	device = vring_alloc_queue(vdev, event_size_in_bytes,
16251ce9e605STiwei Bie 				   &device_event_dma_addr,
16261ce9e605STiwei Bie 				   GFP_KERNEL|__GFP_NOWARN|__GFP_ZERO);
16271ce9e605STiwei Bie 	if (!device)
16281ce9e605STiwei Bie 		goto err_device;
16291ce9e605STiwei Bie 
16301ce9e605STiwei Bie 	vq = kmalloc(sizeof(*vq), GFP_KERNEL);
16311ce9e605STiwei Bie 	if (!vq)
16321ce9e605STiwei Bie 		goto err_vq;
16331ce9e605STiwei Bie 
16341ce9e605STiwei Bie 	vq->vq.callback = callback;
16351ce9e605STiwei Bie 	vq->vq.vdev = vdev;
16361ce9e605STiwei Bie 	vq->vq.name = name;
16371ce9e605STiwei Bie 	vq->vq.num_free = num;
16381ce9e605STiwei Bie 	vq->vq.index = index;
16391ce9e605STiwei Bie 	vq->we_own_ring = true;
16401ce9e605STiwei Bie 	vq->notify = notify;
16411ce9e605STiwei Bie 	vq->weak_barriers = weak_barriers;
16421ce9e605STiwei Bie 	vq->broken = false;
16431ce9e605STiwei Bie 	vq->last_used_idx = 0;
16448d622d21SMichael S. Tsirkin 	vq->event_triggered = false;
16451ce9e605STiwei Bie 	vq->num_added = 0;
16461ce9e605STiwei Bie 	vq->packed_ring = true;
16471ce9e605STiwei Bie 	vq->use_dma_api = vring_use_dma_api(vdev);
16481ce9e605STiwei Bie #ifdef DEBUG
16491ce9e605STiwei Bie 	vq->in_use = false;
16501ce9e605STiwei Bie 	vq->last_add_time_valid = false;
16511ce9e605STiwei Bie #endif
16521ce9e605STiwei Bie 
16531ce9e605STiwei Bie 	vq->indirect = virtio_has_feature(vdev, VIRTIO_RING_F_INDIRECT_DESC) &&
16541ce9e605STiwei Bie 		!context;
16551ce9e605STiwei Bie 	vq->event = virtio_has_feature(vdev, VIRTIO_RING_F_EVENT_IDX);
16561ce9e605STiwei Bie 
165745383fb0STiwei Bie 	if (virtio_has_feature(vdev, VIRTIO_F_ORDER_PLATFORM))
165845383fb0STiwei Bie 		vq->weak_barriers = false;
165945383fb0STiwei Bie 
16601ce9e605STiwei Bie 	vq->packed.ring_dma_addr = ring_dma_addr;
16611ce9e605STiwei Bie 	vq->packed.driver_event_dma_addr = driver_event_dma_addr;
16621ce9e605STiwei Bie 	vq->packed.device_event_dma_addr = device_event_dma_addr;
16631ce9e605STiwei Bie 
16641ce9e605STiwei Bie 	vq->packed.ring_size_in_bytes = ring_size_in_bytes;
16651ce9e605STiwei Bie 	vq->packed.event_size_in_bytes = event_size_in_bytes;
16661ce9e605STiwei Bie 
16671ce9e605STiwei Bie 	vq->packed.vring.num = num;
16681ce9e605STiwei Bie 	vq->packed.vring.desc = ring;
16691ce9e605STiwei Bie 	vq->packed.vring.driver = driver;
16701ce9e605STiwei Bie 	vq->packed.vring.device = device;
16711ce9e605STiwei Bie 
16721ce9e605STiwei Bie 	vq->packed.next_avail_idx = 0;
16731ce9e605STiwei Bie 	vq->packed.avail_wrap_counter = 1;
16741ce9e605STiwei Bie 	vq->packed.used_wrap_counter = 1;
16751ce9e605STiwei Bie 	vq->packed.event_flags_shadow = 0;
16761ce9e605STiwei Bie 	vq->packed.avail_used_flags = 1 << VRING_PACKED_DESC_F_AVAIL;
16771ce9e605STiwei Bie 
16781ce9e605STiwei Bie 	vq->packed.desc_state = kmalloc_array(num,
16791ce9e605STiwei Bie 			sizeof(struct vring_desc_state_packed),
16801ce9e605STiwei Bie 			GFP_KERNEL);
16811ce9e605STiwei Bie 	if (!vq->packed.desc_state)
16821ce9e605STiwei Bie 		goto err_desc_state;
16831ce9e605STiwei Bie 
16841ce9e605STiwei Bie 	memset(vq->packed.desc_state, 0,
16851ce9e605STiwei Bie 		num * sizeof(struct vring_desc_state_packed));
16861ce9e605STiwei Bie 
16871ce9e605STiwei Bie 	/* Put everything in free lists. */
16881ce9e605STiwei Bie 	vq->free_head = 0;
16891ce9e605STiwei Bie 
16905a222421SJason Wang 	vq->packed.desc_extra = vring_alloc_desc_extra(vq, num);
16911ce9e605STiwei Bie 	if (!vq->packed.desc_extra)
16921ce9e605STiwei Bie 		goto err_desc_extra;
16931ce9e605STiwei Bie 
16941ce9e605STiwei Bie 	/* No callback?  Tell other side not to bother us. */
16951ce9e605STiwei Bie 	if (!callback) {
16961ce9e605STiwei Bie 		vq->packed.event_flags_shadow = VRING_PACKED_EVENT_FLAG_DISABLE;
16971ce9e605STiwei Bie 		vq->packed.vring.driver->flags =
16981ce9e605STiwei Bie 			cpu_to_le16(vq->packed.event_flags_shadow);
16991ce9e605STiwei Bie 	}
17001ce9e605STiwei Bie 
1701e152d8afSDan Carpenter 	list_add_tail(&vq->vq.list, &vdev->vqs);
17021ce9e605STiwei Bie 	return &vq->vq;
17031ce9e605STiwei Bie 
17041ce9e605STiwei Bie err_desc_extra:
17051ce9e605STiwei Bie 	kfree(vq->packed.desc_state);
17061ce9e605STiwei Bie err_desc_state:
17071ce9e605STiwei Bie 	kfree(vq);
17081ce9e605STiwei Bie err_vq:
1709ae93d8eaSDan Carpenter 	vring_free_queue(vdev, event_size_in_bytes, device, device_event_dma_addr);
17101ce9e605STiwei Bie err_device:
1711ae93d8eaSDan Carpenter 	vring_free_queue(vdev, event_size_in_bytes, driver, driver_event_dma_addr);
17121ce9e605STiwei Bie err_driver:
17131ce9e605STiwei Bie 	vring_free_queue(vdev, ring_size_in_bytes, ring, ring_dma_addr);
17141ce9e605STiwei Bie err_ring:
17151ce9e605STiwei Bie 	return NULL;
17161ce9e605STiwei Bie }
17171ce9e605STiwei Bie 
17181ce9e605STiwei Bie 
17191ce9e605STiwei Bie /*
1720e6f633e5STiwei Bie  * Generic functions and exported symbols.
1721e6f633e5STiwei Bie  */
1722e6f633e5STiwei Bie 
1723e6f633e5STiwei Bie static inline int virtqueue_add(struct virtqueue *_vq,
1724e6f633e5STiwei Bie 				struct scatterlist *sgs[],
1725e6f633e5STiwei Bie 				unsigned int total_sg,
1726e6f633e5STiwei Bie 				unsigned int out_sgs,
1727e6f633e5STiwei Bie 				unsigned int in_sgs,
1728e6f633e5STiwei Bie 				void *data,
1729e6f633e5STiwei Bie 				void *ctx,
1730e6f633e5STiwei Bie 				gfp_t gfp)
1731e6f633e5STiwei Bie {
17321ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
17331ce9e605STiwei Bie 
17341ce9e605STiwei Bie 	return vq->packed_ring ? virtqueue_add_packed(_vq, sgs, total_sg,
17351ce9e605STiwei Bie 					out_sgs, in_sgs, data, ctx, gfp) :
17361ce9e605STiwei Bie 				 virtqueue_add_split(_vq, sgs, total_sg,
1737e6f633e5STiwei Bie 					out_sgs, in_sgs, data, ctx, gfp);
1738e6f633e5STiwei Bie }
1739e6f633e5STiwei Bie 
1740e6f633e5STiwei Bie /**
1741e6f633e5STiwei Bie  * virtqueue_add_sgs - expose buffers to other end
1742a5581206SJiang Biao  * @_vq: the struct virtqueue we're talking about.
1743e6f633e5STiwei Bie  * @sgs: array of terminated scatterlists.
1744a5581206SJiang Biao  * @out_sgs: the number of scatterlists readable by other side
1745a5581206SJiang Biao  * @in_sgs: the number of scatterlists which are writable (after readable ones)
1746e6f633e5STiwei Bie  * @data: the token identifying the buffer.
1747e6f633e5STiwei Bie  * @gfp: how to do memory allocations (if necessary).
1748e6f633e5STiwei Bie  *
1749e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue operations
1750e6f633e5STiwei Bie  * at the same time (except where noted).
1751e6f633e5STiwei Bie  *
1752e6f633e5STiwei Bie  * Returns zero or a negative error (ie. ENOSPC, ENOMEM, EIO).
1753e6f633e5STiwei Bie  */
1754e6f633e5STiwei Bie int virtqueue_add_sgs(struct virtqueue *_vq,
1755e6f633e5STiwei Bie 		      struct scatterlist *sgs[],
1756e6f633e5STiwei Bie 		      unsigned int out_sgs,
1757e6f633e5STiwei Bie 		      unsigned int in_sgs,
1758e6f633e5STiwei Bie 		      void *data,
1759e6f633e5STiwei Bie 		      gfp_t gfp)
1760e6f633e5STiwei Bie {
1761e6f633e5STiwei Bie 	unsigned int i, total_sg = 0;
1762e6f633e5STiwei Bie 
1763e6f633e5STiwei Bie 	/* Count them first. */
1764e6f633e5STiwei Bie 	for (i = 0; i < out_sgs + in_sgs; i++) {
1765e6f633e5STiwei Bie 		struct scatterlist *sg;
1766e6f633e5STiwei Bie 
1767e6f633e5STiwei Bie 		for (sg = sgs[i]; sg; sg = sg_next(sg))
1768e6f633e5STiwei Bie 			total_sg++;
1769e6f633e5STiwei Bie 	}
1770e6f633e5STiwei Bie 	return virtqueue_add(_vq, sgs, total_sg, out_sgs, in_sgs,
1771e6f633e5STiwei Bie 			     data, NULL, gfp);
1772e6f633e5STiwei Bie }
1773e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_add_sgs);
1774e6f633e5STiwei Bie 
1775e6f633e5STiwei Bie /**
1776e6f633e5STiwei Bie  * virtqueue_add_outbuf - expose output buffers to other end
1777e6f633e5STiwei Bie  * @vq: the struct virtqueue we're talking about.
1778e6f633e5STiwei Bie  * @sg: scatterlist (must be well-formed and terminated!)
1779e6f633e5STiwei Bie  * @num: the number of entries in @sg readable by other side
1780e6f633e5STiwei Bie  * @data: the token identifying the buffer.
1781e6f633e5STiwei Bie  * @gfp: how to do memory allocations (if necessary).
1782e6f633e5STiwei Bie  *
1783e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue operations
1784e6f633e5STiwei Bie  * at the same time (except where noted).
1785e6f633e5STiwei Bie  *
1786e6f633e5STiwei Bie  * Returns zero or a negative error (ie. ENOSPC, ENOMEM, EIO).
1787e6f633e5STiwei Bie  */
1788e6f633e5STiwei Bie int virtqueue_add_outbuf(struct virtqueue *vq,
1789e6f633e5STiwei Bie 			 struct scatterlist *sg, unsigned int num,
1790e6f633e5STiwei Bie 			 void *data,
1791e6f633e5STiwei Bie 			 gfp_t gfp)
1792e6f633e5STiwei Bie {
1793e6f633e5STiwei Bie 	return virtqueue_add(vq, &sg, num, 1, 0, data, NULL, gfp);
1794e6f633e5STiwei Bie }
1795e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_add_outbuf);
1796e6f633e5STiwei Bie 
1797e6f633e5STiwei Bie /**
1798e6f633e5STiwei Bie  * virtqueue_add_inbuf - expose input buffers to other end
1799e6f633e5STiwei Bie  * @vq: the struct virtqueue we're talking about.
1800e6f633e5STiwei Bie  * @sg: scatterlist (must be well-formed and terminated!)
1801e6f633e5STiwei Bie  * @num: the number of entries in @sg writable by other side
1802e6f633e5STiwei Bie  * @data: the token identifying the buffer.
1803e6f633e5STiwei Bie  * @gfp: how to do memory allocations (if necessary).
1804e6f633e5STiwei Bie  *
1805e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue operations
1806e6f633e5STiwei Bie  * at the same time (except where noted).
1807e6f633e5STiwei Bie  *
1808e6f633e5STiwei Bie  * Returns zero or a negative error (ie. ENOSPC, ENOMEM, EIO).
1809e6f633e5STiwei Bie  */
1810e6f633e5STiwei Bie int virtqueue_add_inbuf(struct virtqueue *vq,
1811e6f633e5STiwei Bie 			struct scatterlist *sg, unsigned int num,
1812e6f633e5STiwei Bie 			void *data,
1813e6f633e5STiwei Bie 			gfp_t gfp)
1814e6f633e5STiwei Bie {
1815e6f633e5STiwei Bie 	return virtqueue_add(vq, &sg, num, 0, 1, data, NULL, gfp);
1816e6f633e5STiwei Bie }
1817e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_add_inbuf);
1818e6f633e5STiwei Bie 
1819e6f633e5STiwei Bie /**
1820e6f633e5STiwei Bie  * virtqueue_add_inbuf_ctx - expose input buffers to other end
1821e6f633e5STiwei Bie  * @vq: the struct virtqueue we're talking about.
1822e6f633e5STiwei Bie  * @sg: scatterlist (must be well-formed and terminated!)
1823e6f633e5STiwei Bie  * @num: the number of entries in @sg writable by other side
1824e6f633e5STiwei Bie  * @data: the token identifying the buffer.
1825e6f633e5STiwei Bie  * @ctx: extra context for the token
1826e6f633e5STiwei Bie  * @gfp: how to do memory allocations (if necessary).
1827e6f633e5STiwei Bie  *
1828e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue operations
1829e6f633e5STiwei Bie  * at the same time (except where noted).
1830e6f633e5STiwei Bie  *
1831e6f633e5STiwei Bie  * Returns zero or a negative error (ie. ENOSPC, ENOMEM, EIO).
1832e6f633e5STiwei Bie  */
1833e6f633e5STiwei Bie int virtqueue_add_inbuf_ctx(struct virtqueue *vq,
1834e6f633e5STiwei Bie 			struct scatterlist *sg, unsigned int num,
1835e6f633e5STiwei Bie 			void *data,
1836e6f633e5STiwei Bie 			void *ctx,
1837e6f633e5STiwei Bie 			gfp_t gfp)
1838e6f633e5STiwei Bie {
1839e6f633e5STiwei Bie 	return virtqueue_add(vq, &sg, num, 0, 1, data, ctx, gfp);
1840e6f633e5STiwei Bie }
1841e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_add_inbuf_ctx);
1842e6f633e5STiwei Bie 
1843e6f633e5STiwei Bie /**
1844e6f633e5STiwei Bie  * virtqueue_kick_prepare - first half of split virtqueue_kick call.
1845a5581206SJiang Biao  * @_vq: the struct virtqueue
1846e6f633e5STiwei Bie  *
1847e6f633e5STiwei Bie  * Instead of virtqueue_kick(), you can do:
1848e6f633e5STiwei Bie  *	if (virtqueue_kick_prepare(vq))
1849e6f633e5STiwei Bie  *		virtqueue_notify(vq);
1850e6f633e5STiwei Bie  *
1851e6f633e5STiwei Bie  * This is sometimes useful because the virtqueue_kick_prepare() needs
1852e6f633e5STiwei Bie  * to be serialized, but the actual virtqueue_notify() call does not.
1853e6f633e5STiwei Bie  */
1854e6f633e5STiwei Bie bool virtqueue_kick_prepare(struct virtqueue *_vq)
1855e6f633e5STiwei Bie {
18561ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
18571ce9e605STiwei Bie 
18581ce9e605STiwei Bie 	return vq->packed_ring ? virtqueue_kick_prepare_packed(_vq) :
18591ce9e605STiwei Bie 				 virtqueue_kick_prepare_split(_vq);
1860e6f633e5STiwei Bie }
1861e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_kick_prepare);
1862e6f633e5STiwei Bie 
1863e6f633e5STiwei Bie /**
1864e6f633e5STiwei Bie  * virtqueue_notify - second half of split virtqueue_kick call.
1865a5581206SJiang Biao  * @_vq: the struct virtqueue
1866e6f633e5STiwei Bie  *
1867e6f633e5STiwei Bie  * This does not need to be serialized.
1868e6f633e5STiwei Bie  *
1869e6f633e5STiwei Bie  * Returns false if host notify failed or queue is broken, otherwise true.
1870e6f633e5STiwei Bie  */
1871e6f633e5STiwei Bie bool virtqueue_notify(struct virtqueue *_vq)
1872e6f633e5STiwei Bie {
1873e6f633e5STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
1874e6f633e5STiwei Bie 
1875e6f633e5STiwei Bie 	if (unlikely(vq->broken))
1876e6f633e5STiwei Bie 		return false;
1877e6f633e5STiwei Bie 
1878e6f633e5STiwei Bie 	/* Prod other side to tell it about changes. */
1879e6f633e5STiwei Bie 	if (!vq->notify(_vq)) {
1880e6f633e5STiwei Bie 		vq->broken = true;
1881e6f633e5STiwei Bie 		return false;
1882e6f633e5STiwei Bie 	}
1883e6f633e5STiwei Bie 	return true;
1884e6f633e5STiwei Bie }
1885e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_notify);
1886e6f633e5STiwei Bie 
1887e6f633e5STiwei Bie /**
1888e6f633e5STiwei Bie  * virtqueue_kick - update after add_buf
1889e6f633e5STiwei Bie  * @vq: the struct virtqueue
1890e6f633e5STiwei Bie  *
1891e6f633e5STiwei Bie  * After one or more virtqueue_add_* calls, invoke this to kick
1892e6f633e5STiwei Bie  * the other side.
1893e6f633e5STiwei Bie  *
1894e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue
1895e6f633e5STiwei Bie  * operations at the same time (except where noted).
1896e6f633e5STiwei Bie  *
1897e6f633e5STiwei Bie  * Returns false if kick failed, otherwise true.
1898e6f633e5STiwei Bie  */
1899e6f633e5STiwei Bie bool virtqueue_kick(struct virtqueue *vq)
1900e6f633e5STiwei Bie {
1901e6f633e5STiwei Bie 	if (virtqueue_kick_prepare(vq))
1902e6f633e5STiwei Bie 		return virtqueue_notify(vq);
1903e6f633e5STiwei Bie 	return true;
1904e6f633e5STiwei Bie }
1905e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_kick);
1906e6f633e5STiwei Bie 
1907e6f633e5STiwei Bie /**
190831c11db6SYang Li  * virtqueue_get_buf_ctx - get the next used buffer
1909a5581206SJiang Biao  * @_vq: the struct virtqueue we're talking about.
1910e6f633e5STiwei Bie  * @len: the length written into the buffer
1911a5581206SJiang Biao  * @ctx: extra context for the token
1912e6f633e5STiwei Bie  *
1913e6f633e5STiwei Bie  * If the device wrote data into the buffer, @len will be set to the
1914e6f633e5STiwei Bie  * amount written.  This means you don't need to clear the buffer
1915e6f633e5STiwei Bie  * beforehand to ensure there's no data leakage in the case of short
1916e6f633e5STiwei Bie  * writes.
1917e6f633e5STiwei Bie  *
1918e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue
1919e6f633e5STiwei Bie  * operations at the same time (except where noted).
1920e6f633e5STiwei Bie  *
1921e6f633e5STiwei Bie  * Returns NULL if there are no used buffers, or the "data" token
1922e6f633e5STiwei Bie  * handed to virtqueue_add_*().
1923e6f633e5STiwei Bie  */
1924e6f633e5STiwei Bie void *virtqueue_get_buf_ctx(struct virtqueue *_vq, unsigned int *len,
1925e6f633e5STiwei Bie 			    void **ctx)
1926e6f633e5STiwei Bie {
19271ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
19281ce9e605STiwei Bie 
19291ce9e605STiwei Bie 	return vq->packed_ring ? virtqueue_get_buf_ctx_packed(_vq, len, ctx) :
19301ce9e605STiwei Bie 				 virtqueue_get_buf_ctx_split(_vq, len, ctx);
1931e6f633e5STiwei Bie }
1932e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_get_buf_ctx);
1933e6f633e5STiwei Bie 
1934e6f633e5STiwei Bie void *virtqueue_get_buf(struct virtqueue *_vq, unsigned int *len)
1935e6f633e5STiwei Bie {
1936e6f633e5STiwei Bie 	return virtqueue_get_buf_ctx(_vq, len, NULL);
1937e6f633e5STiwei Bie }
1938e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_get_buf);
1939e6f633e5STiwei Bie /**
1940e6f633e5STiwei Bie  * virtqueue_disable_cb - disable callbacks
1941a5581206SJiang Biao  * @_vq: the struct virtqueue we're talking about.
1942e6f633e5STiwei Bie  *
1943e6f633e5STiwei Bie  * Note that this is not necessarily synchronous, hence unreliable and only
1944e6f633e5STiwei Bie  * useful as an optimization.
1945e6f633e5STiwei Bie  *
1946e6f633e5STiwei Bie  * Unlike other operations, this need not be serialized.
1947e6f633e5STiwei Bie  */
1948e6f633e5STiwei Bie void virtqueue_disable_cb(struct virtqueue *_vq)
1949e6f633e5STiwei Bie {
19501ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
19511ce9e605STiwei Bie 
19528d622d21SMichael S. Tsirkin 	/* If device triggered an event already it won't trigger one again:
19538d622d21SMichael S. Tsirkin 	 * no need to disable.
19548d622d21SMichael S. Tsirkin 	 */
19558d622d21SMichael S. Tsirkin 	if (vq->event_triggered)
19568d622d21SMichael S. Tsirkin 		return;
19578d622d21SMichael S. Tsirkin 
19581ce9e605STiwei Bie 	if (vq->packed_ring)
19591ce9e605STiwei Bie 		virtqueue_disable_cb_packed(_vq);
19601ce9e605STiwei Bie 	else
1961e6f633e5STiwei Bie 		virtqueue_disable_cb_split(_vq);
1962e6f633e5STiwei Bie }
1963e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_disable_cb);
1964e6f633e5STiwei Bie 
1965e6f633e5STiwei Bie /**
1966e6f633e5STiwei Bie  * virtqueue_enable_cb_prepare - restart callbacks after disable_cb
1967a5581206SJiang Biao  * @_vq: the struct virtqueue we're talking about.
1968e6f633e5STiwei Bie  *
1969e6f633e5STiwei Bie  * This re-enables callbacks; it returns current queue state
1970e6f633e5STiwei Bie  * in an opaque unsigned value. This value should be later tested by
1971e6f633e5STiwei Bie  * virtqueue_poll, to detect a possible race between the driver checking for
1972e6f633e5STiwei Bie  * more work, and enabling callbacks.
1973e6f633e5STiwei Bie  *
1974e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue
1975e6f633e5STiwei Bie  * operations at the same time (except where noted).
1976e6f633e5STiwei Bie  */
1977e6f633e5STiwei Bie unsigned virtqueue_enable_cb_prepare(struct virtqueue *_vq)
1978e6f633e5STiwei Bie {
19791ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
19801ce9e605STiwei Bie 
19818d622d21SMichael S. Tsirkin 	if (vq->event_triggered)
19828d622d21SMichael S. Tsirkin 		vq->event_triggered = false;
19838d622d21SMichael S. Tsirkin 
19841ce9e605STiwei Bie 	return vq->packed_ring ? virtqueue_enable_cb_prepare_packed(_vq) :
19851ce9e605STiwei Bie 				 virtqueue_enable_cb_prepare_split(_vq);
1986e6f633e5STiwei Bie }
1987e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_enable_cb_prepare);
1988e6f633e5STiwei Bie 
1989e6f633e5STiwei Bie /**
1990e6f633e5STiwei Bie  * virtqueue_poll - query pending used buffers
1991a5581206SJiang Biao  * @_vq: the struct virtqueue we're talking about.
1992e6f633e5STiwei Bie  * @last_used_idx: virtqueue state (from call to virtqueue_enable_cb_prepare).
1993e6f633e5STiwei Bie  *
1994e6f633e5STiwei Bie  * Returns "true" if there are pending used buffers in the queue.
1995e6f633e5STiwei Bie  *
1996e6f633e5STiwei Bie  * This does not need to be serialized.
1997e6f633e5STiwei Bie  */
1998e6f633e5STiwei Bie bool virtqueue_poll(struct virtqueue *_vq, unsigned last_used_idx)
1999e6f633e5STiwei Bie {
2000e6f633e5STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
2001e6f633e5STiwei Bie 
2002481a0d74SMao Wenan 	if (unlikely(vq->broken))
2003481a0d74SMao Wenan 		return false;
2004481a0d74SMao Wenan 
2005e6f633e5STiwei Bie 	virtio_mb(vq->weak_barriers);
20061ce9e605STiwei Bie 	return vq->packed_ring ? virtqueue_poll_packed(_vq, last_used_idx) :
20071ce9e605STiwei Bie 				 virtqueue_poll_split(_vq, last_used_idx);
2008e6f633e5STiwei Bie }
2009e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_poll);
2010e6f633e5STiwei Bie 
2011e6f633e5STiwei Bie /**
2012e6f633e5STiwei Bie  * virtqueue_enable_cb - restart callbacks after disable_cb.
2013a5581206SJiang Biao  * @_vq: the struct virtqueue we're talking about.
2014e6f633e5STiwei Bie  *
2015e6f633e5STiwei Bie  * This re-enables callbacks; it returns "false" if there are pending
2016e6f633e5STiwei Bie  * buffers in the queue, to detect a possible race between the driver
2017e6f633e5STiwei Bie  * checking for more work, and enabling callbacks.
2018e6f633e5STiwei Bie  *
2019e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue
2020e6f633e5STiwei Bie  * operations at the same time (except where noted).
2021e6f633e5STiwei Bie  */
2022e6f633e5STiwei Bie bool virtqueue_enable_cb(struct virtqueue *_vq)
2023e6f633e5STiwei Bie {
2024e6f633e5STiwei Bie 	unsigned last_used_idx = virtqueue_enable_cb_prepare(_vq);
2025e6f633e5STiwei Bie 
2026e6f633e5STiwei Bie 	return !virtqueue_poll(_vq, last_used_idx);
2027e6f633e5STiwei Bie }
2028e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_enable_cb);
2029e6f633e5STiwei Bie 
2030e6f633e5STiwei Bie /**
2031e6f633e5STiwei Bie  * virtqueue_enable_cb_delayed - restart callbacks after disable_cb.
2032a5581206SJiang Biao  * @_vq: the struct virtqueue we're talking about.
2033e6f633e5STiwei Bie  *
2034e6f633e5STiwei Bie  * This re-enables callbacks but hints to the other side to delay
2035e6f633e5STiwei Bie  * interrupts until most of the available buffers have been processed;
2036e6f633e5STiwei Bie  * it returns "false" if there are many pending buffers in the queue,
2037e6f633e5STiwei Bie  * to detect a possible race between the driver checking for more work,
2038e6f633e5STiwei Bie  * and enabling callbacks.
2039e6f633e5STiwei Bie  *
2040e6f633e5STiwei Bie  * Caller must ensure we don't call this with other virtqueue
2041e6f633e5STiwei Bie  * operations at the same time (except where noted).
2042e6f633e5STiwei Bie  */
2043e6f633e5STiwei Bie bool virtqueue_enable_cb_delayed(struct virtqueue *_vq)
2044e6f633e5STiwei Bie {
20451ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
20461ce9e605STiwei Bie 
20478d622d21SMichael S. Tsirkin 	if (vq->event_triggered)
20488d622d21SMichael S. Tsirkin 		vq->event_triggered = false;
20498d622d21SMichael S. Tsirkin 
20501ce9e605STiwei Bie 	return vq->packed_ring ? virtqueue_enable_cb_delayed_packed(_vq) :
20511ce9e605STiwei Bie 				 virtqueue_enable_cb_delayed_split(_vq);
2052e6f633e5STiwei Bie }
2053e6f633e5STiwei Bie EXPORT_SYMBOL_GPL(virtqueue_enable_cb_delayed);
2054e6f633e5STiwei Bie 
2055138fd251STiwei Bie /**
2056138fd251STiwei Bie  * virtqueue_detach_unused_buf - detach first unused buffer
2057a5581206SJiang Biao  * @_vq: the struct virtqueue we're talking about.
2058138fd251STiwei Bie  *
2059138fd251STiwei Bie  * Returns NULL or the "data" token handed to virtqueue_add_*().
2060138fd251STiwei Bie  * This is not valid on an active queue; it is useful only for device
2061138fd251STiwei Bie  * shutdown.
2062138fd251STiwei Bie  */
2063138fd251STiwei Bie void *virtqueue_detach_unused_buf(struct virtqueue *_vq)
2064138fd251STiwei Bie {
20651ce9e605STiwei Bie 	struct vring_virtqueue *vq = to_vvq(_vq);
20661ce9e605STiwei Bie 
20671ce9e605STiwei Bie 	return vq->packed_ring ? virtqueue_detach_unused_buf_packed(_vq) :
20681ce9e605STiwei Bie 				 virtqueue_detach_unused_buf_split(_vq);
2069138fd251STiwei Bie }
20707c5e9ed0SMichael S. Tsirkin EXPORT_SYMBOL_GPL(virtqueue_detach_unused_buf);
2071c021eac4SShirley Ma 
2072138fd251STiwei Bie static inline bool more_used(const struct vring_virtqueue *vq)
2073138fd251STiwei Bie {
20741ce9e605STiwei Bie 	return vq->packed_ring ? more_used_packed(vq) : more_used_split(vq);
2075138fd251STiwei Bie }
2076138fd251STiwei Bie 
20770a8a69ddSRusty Russell irqreturn_t vring_interrupt(int irq, void *_vq)
20780a8a69ddSRusty Russell {
20790a8a69ddSRusty Russell 	struct vring_virtqueue *vq = to_vvq(_vq);
20800a8a69ddSRusty Russell 
20810a8a69ddSRusty Russell 	if (!more_used(vq)) {
20820a8a69ddSRusty Russell 		pr_debug("virtqueue interrupt with no work for %p\n", vq);
20830a8a69ddSRusty Russell 		return IRQ_NONE;
20840a8a69ddSRusty Russell 	}
20850a8a69ddSRusty Russell 
20860a8a69ddSRusty Russell 	if (unlikely(vq->broken))
20870a8a69ddSRusty Russell 		return IRQ_HANDLED;
20880a8a69ddSRusty Russell 
20898d622d21SMichael S. Tsirkin 	/* Just a hint for performance: so it's ok that this can be racy! */
20908d622d21SMichael S. Tsirkin 	if (vq->event)
20918d622d21SMichael S. Tsirkin 		vq->event_triggered = true;
20928d622d21SMichael S. Tsirkin 
20930a8a69ddSRusty Russell 	pr_debug("virtqueue callback for %p (%p)\n", vq, vq->vq.callback);
209418445c4dSRusty Russell 	if (vq->vq.callback)
209518445c4dSRusty Russell 		vq->vq.callback(&vq->vq);
20960a8a69ddSRusty Russell 
20970a8a69ddSRusty Russell 	return IRQ_HANDLED;
20980a8a69ddSRusty Russell }
2099c6fd4701SRusty Russell EXPORT_SYMBOL_GPL(vring_interrupt);
21000a8a69ddSRusty Russell 
21011ce9e605STiwei Bie /* Only available for split ring */
21022a2d1382SAndy Lutomirski struct virtqueue *__vring_new_virtqueue(unsigned int index,
21032a2d1382SAndy Lutomirski 					struct vring vring,
21040a8a69ddSRusty Russell 					struct virtio_device *vdev,
21057b21e34fSRusty Russell 					bool weak_barriers,
2106f94682ddSMichael S. Tsirkin 					bool context,
210746f9c2b9SHeinz Graalfs 					bool (*notify)(struct virtqueue *),
21089499f5e7SRusty Russell 					void (*callback)(struct virtqueue *),
21099499f5e7SRusty Russell 					const char *name)
21100a8a69ddSRusty Russell {
21110a8a69ddSRusty Russell 	unsigned int i;
21122a2d1382SAndy Lutomirski 	struct vring_virtqueue *vq;
21130a8a69ddSRusty Russell 
21141ce9e605STiwei Bie 	if (virtio_has_feature(vdev, VIRTIO_F_RING_PACKED))
21151ce9e605STiwei Bie 		return NULL;
21161ce9e605STiwei Bie 
2117cbeedb72STiwei Bie 	vq = kmalloc(sizeof(*vq), GFP_KERNEL);
21180a8a69ddSRusty Russell 	if (!vq)
21190a8a69ddSRusty Russell 		return NULL;
21200a8a69ddSRusty Russell 
21211ce9e605STiwei Bie 	vq->packed_ring = false;
21220a8a69ddSRusty Russell 	vq->vq.callback = callback;
21230a8a69ddSRusty Russell 	vq->vq.vdev = vdev;
21249499f5e7SRusty Russell 	vq->vq.name = name;
21252a2d1382SAndy Lutomirski 	vq->vq.num_free = vring.num;
212606ca287dSRusty Russell 	vq->vq.index = index;
21272a2d1382SAndy Lutomirski 	vq->we_own_ring = false;
21280a8a69ddSRusty Russell 	vq->notify = notify;
21297b21e34fSRusty Russell 	vq->weak_barriers = weak_barriers;
21300a8a69ddSRusty Russell 	vq->broken = false;
21310a8a69ddSRusty Russell 	vq->last_used_idx = 0;
21328d622d21SMichael S. Tsirkin 	vq->event_triggered = false;
21330a8a69ddSRusty Russell 	vq->num_added = 0;
2134fb3fba6bSTiwei Bie 	vq->use_dma_api = vring_use_dma_api(vdev);
21350a8a69ddSRusty Russell #ifdef DEBUG
21360a8a69ddSRusty Russell 	vq->in_use = false;
2137e93300b1SRusty Russell 	vq->last_add_time_valid = false;
21380a8a69ddSRusty Russell #endif
21390a8a69ddSRusty Russell 
21405a08b04fSMichael S. Tsirkin 	vq->indirect = virtio_has_feature(vdev, VIRTIO_RING_F_INDIRECT_DESC) &&
21415a08b04fSMichael S. Tsirkin 		!context;
2142a5c262c5SMichael S. Tsirkin 	vq->event = virtio_has_feature(vdev, VIRTIO_RING_F_EVENT_IDX);
21439fa29b9dSMark McLoughlin 
214445383fb0STiwei Bie 	if (virtio_has_feature(vdev, VIRTIO_F_ORDER_PLATFORM))
214545383fb0STiwei Bie 		vq->weak_barriers = false;
214645383fb0STiwei Bie 
2147d79dca75STiwei Bie 	vq->split.queue_dma_addr = 0;
2148d79dca75STiwei Bie 	vq->split.queue_size_in_bytes = 0;
2149d79dca75STiwei Bie 
2150e593bf97STiwei Bie 	vq->split.vring = vring;
2151e593bf97STiwei Bie 	vq->split.avail_flags_shadow = 0;
2152e593bf97STiwei Bie 	vq->split.avail_idx_shadow = 0;
2153e593bf97STiwei Bie 
21540a8a69ddSRusty Russell 	/* No callback?  Tell other side not to bother us. */
2155f277ec42SVenkatesh Srinivas 	if (!callback) {
2156e593bf97STiwei Bie 		vq->split.avail_flags_shadow |= VRING_AVAIL_F_NO_INTERRUPT;
21570ea1e4a6SLadi Prosek 		if (!vq->event)
2158e593bf97STiwei Bie 			vq->split.vring.avail->flags = cpu_to_virtio16(vdev,
2159e593bf97STiwei Bie 					vq->split.avail_flags_shadow);
2160f277ec42SVenkatesh Srinivas 	}
21610a8a69ddSRusty Russell 
2162cbeedb72STiwei Bie 	vq->split.desc_state = kmalloc_array(vring.num,
2163cbeedb72STiwei Bie 			sizeof(struct vring_desc_state_split), GFP_KERNEL);
2164*5bc72234SJason Wang 	if (!vq->split.desc_state)
2165*5bc72234SJason Wang 		goto err_state;
2166cbeedb72STiwei Bie 
21670a8a69ddSRusty Russell 	/* Put everything in free lists. */
21680a8a69ddSRusty Russell 	vq->free_head = 0;
21692a2d1382SAndy Lutomirski 	for (i = 0; i < vring.num-1; i++)
2170e593bf97STiwei Bie 		vq->split.vring.desc[i].next = cpu_to_virtio16(vdev, i + 1);
2171cbeedb72STiwei Bie 	memset(vq->split.desc_state, 0, vring.num *
2172cbeedb72STiwei Bie 			sizeof(struct vring_desc_state_split));
21730a8a69ddSRusty Russell 
2174e152d8afSDan Carpenter 	list_add_tail(&vq->vq.list, &vdev->vqs);
21750a8a69ddSRusty Russell 	return &vq->vq;
2176*5bc72234SJason Wang 
2177*5bc72234SJason Wang err_state:
2178*5bc72234SJason Wang 	kfree(vq);
2179*5bc72234SJason Wang 	return NULL;
21800a8a69ddSRusty Russell }
21812a2d1382SAndy Lutomirski EXPORT_SYMBOL_GPL(__vring_new_virtqueue);
21822a2d1382SAndy Lutomirski 
21832a2d1382SAndy Lutomirski struct virtqueue *vring_create_virtqueue(
21842a2d1382SAndy Lutomirski 	unsigned int index,
21852a2d1382SAndy Lutomirski 	unsigned int num,
21862a2d1382SAndy Lutomirski 	unsigned int vring_align,
21872a2d1382SAndy Lutomirski 	struct virtio_device *vdev,
21882a2d1382SAndy Lutomirski 	bool weak_barriers,
21892a2d1382SAndy Lutomirski 	bool may_reduce_num,
2190f94682ddSMichael S. Tsirkin 	bool context,
21912a2d1382SAndy Lutomirski 	bool (*notify)(struct virtqueue *),
21922a2d1382SAndy Lutomirski 	void (*callback)(struct virtqueue *),
21932a2d1382SAndy Lutomirski 	const char *name)
21942a2d1382SAndy Lutomirski {
21951ce9e605STiwei Bie 
21961ce9e605STiwei Bie 	if (virtio_has_feature(vdev, VIRTIO_F_RING_PACKED))
21971ce9e605STiwei Bie 		return vring_create_virtqueue_packed(index, num, vring_align,
21981ce9e605STiwei Bie 				vdev, weak_barriers, may_reduce_num,
21991ce9e605STiwei Bie 				context, notify, callback, name);
22001ce9e605STiwei Bie 
2201d79dca75STiwei Bie 	return vring_create_virtqueue_split(index, num, vring_align,
2202d79dca75STiwei Bie 			vdev, weak_barriers, may_reduce_num,
2203d79dca75STiwei Bie 			context, notify, callback, name);
22042a2d1382SAndy Lutomirski }
22052a2d1382SAndy Lutomirski EXPORT_SYMBOL_GPL(vring_create_virtqueue);
22062a2d1382SAndy Lutomirski 
22071ce9e605STiwei Bie /* Only available for split ring */
22082a2d1382SAndy Lutomirski struct virtqueue *vring_new_virtqueue(unsigned int index,
22092a2d1382SAndy Lutomirski 				      unsigned int num,
22102a2d1382SAndy Lutomirski 				      unsigned int vring_align,
22112a2d1382SAndy Lutomirski 				      struct virtio_device *vdev,
22122a2d1382SAndy Lutomirski 				      bool weak_barriers,
2213f94682ddSMichael S. Tsirkin 				      bool context,
22142a2d1382SAndy Lutomirski 				      void *pages,
22152a2d1382SAndy Lutomirski 				      bool (*notify)(struct virtqueue *vq),
22162a2d1382SAndy Lutomirski 				      void (*callback)(struct virtqueue *vq),
22172a2d1382SAndy Lutomirski 				      const char *name)
22182a2d1382SAndy Lutomirski {
22192a2d1382SAndy Lutomirski 	struct vring vring;
22201ce9e605STiwei Bie 
22211ce9e605STiwei Bie 	if (virtio_has_feature(vdev, VIRTIO_F_RING_PACKED))
22221ce9e605STiwei Bie 		return NULL;
22231ce9e605STiwei Bie 
22242a2d1382SAndy Lutomirski 	vring_init(&vring, num, pages, vring_align);
2225f94682ddSMichael S. Tsirkin 	return __vring_new_virtqueue(index, vring, vdev, weak_barriers, context,
22262a2d1382SAndy Lutomirski 				     notify, callback, name);
22272a2d1382SAndy Lutomirski }
2228c6fd4701SRusty Russell EXPORT_SYMBOL_GPL(vring_new_virtqueue);
22290a8a69ddSRusty Russell 
22302a2d1382SAndy Lutomirski void vring_del_virtqueue(struct virtqueue *_vq)
22310a8a69ddSRusty Russell {
22322a2d1382SAndy Lutomirski 	struct vring_virtqueue *vq = to_vvq(_vq);
22332a2d1382SAndy Lutomirski 
22342a2d1382SAndy Lutomirski 	if (vq->we_own_ring) {
22351ce9e605STiwei Bie 		if (vq->packed_ring) {
22361ce9e605STiwei Bie 			vring_free_queue(vq->vq.vdev,
22371ce9e605STiwei Bie 					 vq->packed.ring_size_in_bytes,
22381ce9e605STiwei Bie 					 vq->packed.vring.desc,
22391ce9e605STiwei Bie 					 vq->packed.ring_dma_addr);
22401ce9e605STiwei Bie 
22411ce9e605STiwei Bie 			vring_free_queue(vq->vq.vdev,
22421ce9e605STiwei Bie 					 vq->packed.event_size_in_bytes,
22431ce9e605STiwei Bie 					 vq->packed.vring.driver,
22441ce9e605STiwei Bie 					 vq->packed.driver_event_dma_addr);
22451ce9e605STiwei Bie 
22461ce9e605STiwei Bie 			vring_free_queue(vq->vq.vdev,
22471ce9e605STiwei Bie 					 vq->packed.event_size_in_bytes,
22481ce9e605STiwei Bie 					 vq->packed.vring.device,
22491ce9e605STiwei Bie 					 vq->packed.device_event_dma_addr);
22501ce9e605STiwei Bie 
22511ce9e605STiwei Bie 			kfree(vq->packed.desc_state);
22521ce9e605STiwei Bie 			kfree(vq->packed.desc_extra);
22531ce9e605STiwei Bie 		} else {
2254d79dca75STiwei Bie 			vring_free_queue(vq->vq.vdev,
2255d79dca75STiwei Bie 					 vq->split.queue_size_in_bytes,
2256d79dca75STiwei Bie 					 vq->split.vring.desc,
2257d79dca75STiwei Bie 					 vq->split.queue_dma_addr);
2258f13f09a1SSuman Anna 		}
2259f13f09a1SSuman Anna 	}
2260f13f09a1SSuman Anna 	if (!vq->packed_ring)
2261cbeedb72STiwei Bie 		kfree(vq->split.desc_state);
22622a2d1382SAndy Lutomirski 	list_del(&_vq->list);
22632a2d1382SAndy Lutomirski 	kfree(vq);
22640a8a69ddSRusty Russell }
2265c6fd4701SRusty Russell EXPORT_SYMBOL_GPL(vring_del_virtqueue);
22660a8a69ddSRusty Russell 
2267e34f8725SRusty Russell /* Manipulates transport-specific feature bits. */
2268e34f8725SRusty Russell void vring_transport_features(struct virtio_device *vdev)
2269e34f8725SRusty Russell {
2270e34f8725SRusty Russell 	unsigned int i;
2271e34f8725SRusty Russell 
2272e34f8725SRusty Russell 	for (i = VIRTIO_TRANSPORT_F_START; i < VIRTIO_TRANSPORT_F_END; i++) {
2273e34f8725SRusty Russell 		switch (i) {
22749fa29b9dSMark McLoughlin 		case VIRTIO_RING_F_INDIRECT_DESC:
22759fa29b9dSMark McLoughlin 			break;
2276a5c262c5SMichael S. Tsirkin 		case VIRTIO_RING_F_EVENT_IDX:
2277a5c262c5SMichael S. Tsirkin 			break;
2278747ae34aSMichael S. Tsirkin 		case VIRTIO_F_VERSION_1:
2279747ae34aSMichael S. Tsirkin 			break;
2280321bd212SMichael S. Tsirkin 		case VIRTIO_F_ACCESS_PLATFORM:
22811a937693SMichael S. Tsirkin 			break;
2282f959a128STiwei Bie 		case VIRTIO_F_RING_PACKED:
2283f959a128STiwei Bie 			break;
228445383fb0STiwei Bie 		case VIRTIO_F_ORDER_PLATFORM:
228545383fb0STiwei Bie 			break;
2286e34f8725SRusty Russell 		default:
2287e34f8725SRusty Russell 			/* We don't understand this bit. */
2288e16e12beSMichael S. Tsirkin 			__virtio_clear_bit(vdev, i);
2289e34f8725SRusty Russell 		}
2290e34f8725SRusty Russell 	}
2291e34f8725SRusty Russell }
2292e34f8725SRusty Russell EXPORT_SYMBOL_GPL(vring_transport_features);
2293e34f8725SRusty Russell 
22945dfc1762SRusty Russell /**
22955dfc1762SRusty Russell  * virtqueue_get_vring_size - return the size of the virtqueue's vring
2296a5581206SJiang Biao  * @_vq: the struct virtqueue containing the vring of interest.
22975dfc1762SRusty Russell  *
22985dfc1762SRusty Russell  * Returns the size of the vring.  This is mainly used for boasting to
22995dfc1762SRusty Russell  * userspace.  Unlike other operations, this need not be serialized.
23005dfc1762SRusty Russell  */
23018f9f4668SRick Jones unsigned int virtqueue_get_vring_size(struct virtqueue *_vq)
23028f9f4668SRick Jones {
23038f9f4668SRick Jones 
23048f9f4668SRick Jones 	struct vring_virtqueue *vq = to_vvq(_vq);
23058f9f4668SRick Jones 
23061ce9e605STiwei Bie 	return vq->packed_ring ? vq->packed.vring.num : vq->split.vring.num;
23078f9f4668SRick Jones }
23088f9f4668SRick Jones EXPORT_SYMBOL_GPL(virtqueue_get_vring_size);
23098f9f4668SRick Jones 
2310b3b32c94SHeinz Graalfs bool virtqueue_is_broken(struct virtqueue *_vq)
2311b3b32c94SHeinz Graalfs {
2312b3b32c94SHeinz Graalfs 	struct vring_virtqueue *vq = to_vvq(_vq);
2313b3b32c94SHeinz Graalfs 
2314b3b32c94SHeinz Graalfs 	return vq->broken;
2315b3b32c94SHeinz Graalfs }
2316b3b32c94SHeinz Graalfs EXPORT_SYMBOL_GPL(virtqueue_is_broken);
2317b3b32c94SHeinz Graalfs 
2318e2dcdfe9SRusty Russell /*
2319e2dcdfe9SRusty Russell  * This should prevent the device from being used, allowing drivers to
2320e2dcdfe9SRusty Russell  * recover.  You may need to grab appropriate locks to flush.
2321e2dcdfe9SRusty Russell  */
2322e2dcdfe9SRusty Russell void virtio_break_device(struct virtio_device *dev)
2323e2dcdfe9SRusty Russell {
2324e2dcdfe9SRusty Russell 	struct virtqueue *_vq;
2325e2dcdfe9SRusty Russell 
2326e2dcdfe9SRusty Russell 	list_for_each_entry(_vq, &dev->vqs, list) {
2327e2dcdfe9SRusty Russell 		struct vring_virtqueue *vq = to_vvq(_vq);
2328e2dcdfe9SRusty Russell 		vq->broken = true;
2329e2dcdfe9SRusty Russell 	}
2330e2dcdfe9SRusty Russell }
2331e2dcdfe9SRusty Russell EXPORT_SYMBOL_GPL(virtio_break_device);
2332e2dcdfe9SRusty Russell 
23332a2d1382SAndy Lutomirski dma_addr_t virtqueue_get_desc_addr(struct virtqueue *_vq)
233489062652SCornelia Huck {
233589062652SCornelia Huck 	struct vring_virtqueue *vq = to_vvq(_vq);
233689062652SCornelia Huck 
23372a2d1382SAndy Lutomirski 	BUG_ON(!vq->we_own_ring);
233889062652SCornelia Huck 
23391ce9e605STiwei Bie 	if (vq->packed_ring)
23401ce9e605STiwei Bie 		return vq->packed.ring_dma_addr;
23411ce9e605STiwei Bie 
2342d79dca75STiwei Bie 	return vq->split.queue_dma_addr;
23432a2d1382SAndy Lutomirski }
23442a2d1382SAndy Lutomirski EXPORT_SYMBOL_GPL(virtqueue_get_desc_addr);
23452a2d1382SAndy Lutomirski 
23462a2d1382SAndy Lutomirski dma_addr_t virtqueue_get_avail_addr(struct virtqueue *_vq)
234789062652SCornelia Huck {
234889062652SCornelia Huck 	struct vring_virtqueue *vq = to_vvq(_vq);
234989062652SCornelia Huck 
23502a2d1382SAndy Lutomirski 	BUG_ON(!vq->we_own_ring);
23512a2d1382SAndy Lutomirski 
23521ce9e605STiwei Bie 	if (vq->packed_ring)
23531ce9e605STiwei Bie 		return vq->packed.driver_event_dma_addr;
23541ce9e605STiwei Bie 
2355d79dca75STiwei Bie 	return vq->split.queue_dma_addr +
2356e593bf97STiwei Bie 		((char *)vq->split.vring.avail - (char *)vq->split.vring.desc);
235789062652SCornelia Huck }
23582a2d1382SAndy Lutomirski EXPORT_SYMBOL_GPL(virtqueue_get_avail_addr);
23592a2d1382SAndy Lutomirski 
23602a2d1382SAndy Lutomirski dma_addr_t virtqueue_get_used_addr(struct virtqueue *_vq)
23612a2d1382SAndy Lutomirski {
23622a2d1382SAndy Lutomirski 	struct vring_virtqueue *vq = to_vvq(_vq);
23632a2d1382SAndy Lutomirski 
23642a2d1382SAndy Lutomirski 	BUG_ON(!vq->we_own_ring);
23652a2d1382SAndy Lutomirski 
23661ce9e605STiwei Bie 	if (vq->packed_ring)
23671ce9e605STiwei Bie 		return vq->packed.device_event_dma_addr;
23681ce9e605STiwei Bie 
2369d79dca75STiwei Bie 	return vq->split.queue_dma_addr +
2370e593bf97STiwei Bie 		((char *)vq->split.vring.used - (char *)vq->split.vring.desc);
23712a2d1382SAndy Lutomirski }
23722a2d1382SAndy Lutomirski EXPORT_SYMBOL_GPL(virtqueue_get_used_addr);
23732a2d1382SAndy Lutomirski 
23741ce9e605STiwei Bie /* Only available for split ring */
23752a2d1382SAndy Lutomirski const struct vring *virtqueue_get_vring(struct virtqueue *vq)
23762a2d1382SAndy Lutomirski {
2377e593bf97STiwei Bie 	return &to_vvq(vq)->split.vring;
23782a2d1382SAndy Lutomirski }
23792a2d1382SAndy Lutomirski EXPORT_SYMBOL_GPL(virtqueue_get_vring);
238089062652SCornelia Huck 
2381c6fd4701SRusty Russell MODULE_LICENSE("GPL");
2382