1/* SPDX-License-Identifier: GPL-2.0 */
2#ifndef _LINUX_VIRTIO_H
3#define _LINUX_VIRTIO_H
4/* Everything a virtio driver needs to work with any particular virtio
5 * implementation. */
6#include <linux/types.h>
7#include <linux/scatterlist.h>
8#include <linux/spinlock.h>
9#include <linux/device.h>
10#include <linux/mod_devicetable.h>
11#include <linux/gfp.h>
12#include <linux/dma-mapping.h>
13
14/**
15 * struct virtqueue - a queue to register buffers for sending or receiving.
16 * @list: the chain of virtqueues for this device
17 * @callback: the function to call when buffers are consumed (can be NULL).
18 * @name: the name of this virtqueue (mainly for debugging)
19 * @vdev: the virtio device this queue was created for.
20 * @priv: a pointer for the virtqueue implementation to use.
21 * @index: the zero-based ordinal number for this queue.
22 * @num_free: number of elements we expect to be able to fit.
23 * @num_max: the maximum number of elements supported by the device.
24 * @reset: vq is in reset state or not.
25 *
26 * A note on @num_free: with indirect buffers, each buffer needs one
27 * element in the queue, otherwise a buffer will need one element per
28 * sg element.
29 */
30struct virtqueue {
31	struct list_head list;
32	void (*callback)(struct virtqueue *vq);
33	const char *name;
34	struct virtio_device *vdev;
35	unsigned int index;
36	unsigned int num_free;
37	unsigned int num_max;
38	bool reset;
39	void *priv;
40};
41
42int virtqueue_add_outbuf(struct virtqueue *vq,
43			 struct scatterlist sg[], unsigned int num,
44			 void *data,
45			 gfp_t gfp);
46
47int virtqueue_add_inbuf(struct virtqueue *vq,
48			struct scatterlist sg[], unsigned int num,
49			void *data,
50			gfp_t gfp);
51
52int virtqueue_add_inbuf_ctx(struct virtqueue *vq,
53			    struct scatterlist sg[], unsigned int num,
54			    void *data,
55			    void *ctx,
56			    gfp_t gfp);
57
58int virtqueue_add_sgs(struct virtqueue *vq,
59		      struct scatterlist *sgs[],
60		      unsigned int out_sgs,
61		      unsigned int in_sgs,
62		      void *data,
63		      gfp_t gfp);
64
65struct device *virtqueue_dma_dev(struct virtqueue *vq);
66
67bool virtqueue_kick(struct virtqueue *vq);
68
69bool virtqueue_kick_prepare(struct virtqueue *vq);
70
71bool virtqueue_notify(struct virtqueue *vq);
72
73void *virtqueue_get_buf(struct virtqueue *vq, unsigned int *len);
74
75void *virtqueue_get_buf_ctx(struct virtqueue *vq, unsigned int *len,
76			    void **ctx);
77
78void virtqueue_disable_cb(struct virtqueue *vq);
79
80bool virtqueue_enable_cb(struct virtqueue *vq);
81
82unsigned virtqueue_enable_cb_prepare(struct virtqueue *vq);
83
84int virtqueue_set_dma_premapped(struct virtqueue *_vq);
85
86bool virtqueue_poll(struct virtqueue *vq, unsigned);
87
88bool virtqueue_enable_cb_delayed(struct virtqueue *vq);
89
90void *virtqueue_detach_unused_buf(struct virtqueue *vq);
91
92unsigned int virtqueue_get_vring_size(const struct virtqueue *vq);
93
94bool virtqueue_is_broken(const struct virtqueue *vq);
95
96const struct vring *virtqueue_get_vring(const struct virtqueue *vq);
97dma_addr_t virtqueue_get_desc_addr(const struct virtqueue *vq);
98dma_addr_t virtqueue_get_avail_addr(const struct virtqueue *vq);
99dma_addr_t virtqueue_get_used_addr(const struct virtqueue *vq);
100
101int virtqueue_resize(struct virtqueue *vq, u32 num,
102		     void (*recycle)(struct virtqueue *vq, void *buf));
103int virtqueue_reset(struct virtqueue *vq,
104		    void (*recycle)(struct virtqueue *vq, void *buf));
105
106struct virtio_admin_cmd {
107	__le16 opcode;
108	__le16 group_type;
109	__le64 group_member_id;
110	struct scatterlist *data_sg;
111	struct scatterlist *result_sg;
112};
113
114/**
115 * struct virtio_device - representation of a device using virtio
116 * @index: unique position on the virtio bus
117 * @failed: saved value for VIRTIO_CONFIG_S_FAILED bit (for restore)
118 * @config_enabled: configuration change reporting enabled
119 * @config_change_pending: configuration change reported while disabled
120 * @config_lock: protects configuration change reporting
121 * @vqs_list_lock: protects @vqs.
122 * @dev: underlying device.
123 * @id: the device type identification (used to match it with a driver).
124 * @config: the configuration ops for this device.
125 * @vringh_config: configuration ops for host vrings.
126 * @vqs: the list of virtqueues for this device.
127 * @features: the features supported by both driver and device.
128 * @priv: private pointer for the driver's use.
129 */
130struct virtio_device {
131	int index;
132	bool failed;
133	bool config_enabled;
134	bool config_change_pending;
135	spinlock_t config_lock;
136	spinlock_t vqs_list_lock;
137	struct device dev;
138	struct virtio_device_id id;
139	const struct virtio_config_ops *config;
140	const struct vringh_config_ops *vringh_config;
141	struct list_head vqs;
142	u64 features;
143	void *priv;
144};
145
146#define dev_to_virtio(_dev)	container_of_const(_dev, struct virtio_device, dev)
147
148void virtio_add_status(struct virtio_device *dev, unsigned int status);
149int register_virtio_device(struct virtio_device *dev);
150void unregister_virtio_device(struct virtio_device *dev);
151bool is_virtio_device(struct device *dev);
152
153void virtio_break_device(struct virtio_device *dev);
154void __virtio_unbreak_device(struct virtio_device *dev);
155
156void __virtqueue_break(struct virtqueue *_vq);
157void __virtqueue_unbreak(struct virtqueue *_vq);
158
159void virtio_config_changed(struct virtio_device *dev);
160#ifdef CONFIG_PM_SLEEP
161int virtio_device_freeze(struct virtio_device *dev);
162int virtio_device_restore(struct virtio_device *dev);
163#endif
164void virtio_reset_device(struct virtio_device *dev);
165
166size_t virtio_max_dma_size(const struct virtio_device *vdev);
167
168#define virtio_device_for_each_vq(vdev, vq) \
169	list_for_each_entry(vq, &vdev->vqs, list)
170
171/**
172 * struct virtio_driver - operations for a virtio I/O driver
173 * @driver: underlying device driver (populate name).
174 * @id_table: the ids serviced by this driver.
175 * @feature_table: an array of feature numbers supported by this driver.
176 * @feature_table_size: number of entries in the feature table array.
177 * @feature_table_legacy: same as feature_table but when working in legacy mode.
178 * @feature_table_size_legacy: number of entries in feature table legacy array.
179 * @validate: the function to call to validate features and config space.
180 *            Returns 0 or -errno.
181 * @probe: the function to call when a device is found.  Returns 0 or -errno.
182 * @scan: optional function to call after successful probe; intended
183 *    for virtio-scsi to invoke a scan.
184 * @remove: the function to call when a device is removed.
185 * @config_changed: optional function to call when the device configuration
186 *    changes; may be called in interrupt context.
187 * @freeze: optional function to call during suspend/hibernation.
188 * @restore: optional function to call on resume.
189 */
190struct virtio_driver {
191	struct device_driver driver;
192	const struct virtio_device_id *id_table;
193	const unsigned int *feature_table;
194	unsigned int feature_table_size;
195	const unsigned int *feature_table_legacy;
196	unsigned int feature_table_size_legacy;
197	int (*validate)(struct virtio_device *dev);
198	int (*probe)(struct virtio_device *dev);
199	void (*scan)(struct virtio_device *dev);
200	void (*remove)(struct virtio_device *dev);
201	void (*config_changed)(struct virtio_device *dev);
202	int (*freeze)(struct virtio_device *dev);
203	int (*restore)(struct virtio_device *dev);
204};
205
206static inline struct virtio_driver *drv_to_virtio(struct device_driver *drv)
207{
208	return container_of(drv, struct virtio_driver, driver);
209}
210
211/* use a macro to avoid include chaining to get THIS_MODULE */
212#define register_virtio_driver(drv) \
213	__register_virtio_driver(drv, THIS_MODULE)
214int __register_virtio_driver(struct virtio_driver *drv, struct module *owner);
215void unregister_virtio_driver(struct virtio_driver *drv);
216
217/* module_virtio_driver() - Helper macro for drivers that don't do
218 * anything special in module init/exit.  This eliminates a lot of
219 * boilerplate.  Each module may only use this macro once, and
220 * calling it replaces module_init() and module_exit()
221 */
222#define module_virtio_driver(__virtio_driver) \
223	module_driver(__virtio_driver, register_virtio_driver, \
224			unregister_virtio_driver)
225
226dma_addr_t virtqueue_dma_map_single_attrs(struct virtqueue *_vq, void *ptr, size_t size,
227					  enum dma_data_direction dir, unsigned long attrs);
228void virtqueue_dma_unmap_single_attrs(struct virtqueue *_vq, dma_addr_t addr,
229				      size_t size, enum dma_data_direction dir,
230				      unsigned long attrs);
231int virtqueue_dma_mapping_error(struct virtqueue *_vq, dma_addr_t addr);
232
233bool virtqueue_dma_need_sync(struct virtqueue *_vq, dma_addr_t addr);
234void virtqueue_dma_sync_single_range_for_cpu(struct virtqueue *_vq, dma_addr_t addr,
235					     unsigned long offset, size_t size,
236					     enum dma_data_direction dir);
237void virtqueue_dma_sync_single_range_for_device(struct virtqueue *_vq, dma_addr_t addr,
238						unsigned long offset, size_t size,
239						enum dma_data_direction dir);
240#endif /* _LINUX_VIRTIO_H */
241