xref: /openbmc/linux/fs/fuse/virtio_fs.c (revision a7f0d7aab0b4f3f0780b1f77356e2fe7202ac0cb)
1a62a8ef9SStefan Hajnoczi // SPDX-License-Identifier: GPL-2.0
2a62a8ef9SStefan Hajnoczi /*
3a62a8ef9SStefan Hajnoczi  * virtio-fs: Virtio Filesystem
4a62a8ef9SStefan Hajnoczi  * Copyright (C) 2018 Red Hat, Inc.
5a62a8ef9SStefan Hajnoczi  */
6a62a8ef9SStefan Hajnoczi 
7a62a8ef9SStefan Hajnoczi #include <linux/fs.h>
822f3787eSStefan Hajnoczi #include <linux/dax.h>
922f3787eSStefan Hajnoczi #include <linux/pci.h>
1022f3787eSStefan Hajnoczi #include <linux/pfn_t.h>
11a62a8ef9SStefan Hajnoczi #include <linux/module.h>
12a62a8ef9SStefan Hajnoczi #include <linux/virtio.h>
13a62a8ef9SStefan Hajnoczi #include <linux/virtio_fs.h>
14a62a8ef9SStefan Hajnoczi #include <linux/delay.h>
15a62a8ef9SStefan Hajnoczi #include <linux/fs_context.h>
161dd53957SVivek Goyal #include <linux/fs_parser.h>
17a62a8ef9SStefan Hajnoczi #include <linux/highmem.h>
1822f3787eSStefan Hajnoczi #include <linux/uio.h>
19a62a8ef9SStefan Hajnoczi #include "fuse_i.h"
20a62a8ef9SStefan Hajnoczi 
21*a7f0d7aaSConnor Kuehl /* Used to help calculate the FUSE connection's max_pages limit for a request's
22*a7f0d7aaSConnor Kuehl  * size. Parts of the struct fuse_req are sliced into scattergather lists in
23*a7f0d7aaSConnor Kuehl  * addition to the pages used, so this can help account for that overhead.
24*a7f0d7aaSConnor Kuehl  */
25*a7f0d7aaSConnor Kuehl #define FUSE_HEADER_OVERHEAD    4
26*a7f0d7aaSConnor Kuehl 
27a62a8ef9SStefan Hajnoczi /* List of virtio-fs device instances and a lock for the list. Also provides
28a62a8ef9SStefan Hajnoczi  * mutual exclusion in device removal and mounting path
29a62a8ef9SStefan Hajnoczi  */
30a62a8ef9SStefan Hajnoczi static DEFINE_MUTEX(virtio_fs_mutex);
31a62a8ef9SStefan Hajnoczi static LIST_HEAD(virtio_fs_instances);
32a62a8ef9SStefan Hajnoczi 
33a62a8ef9SStefan Hajnoczi enum {
34a62a8ef9SStefan Hajnoczi 	VQ_HIPRIO,
35a62a8ef9SStefan Hajnoczi 	VQ_REQUEST
36a62a8ef9SStefan Hajnoczi };
37a62a8ef9SStefan Hajnoczi 
38b43b7e81SVivek Goyal #define VQ_NAME_LEN	24
39b43b7e81SVivek Goyal 
40a62a8ef9SStefan Hajnoczi /* Per-virtqueue state */
41a62a8ef9SStefan Hajnoczi struct virtio_fs_vq {
42a62a8ef9SStefan Hajnoczi 	spinlock_t lock;
43a62a8ef9SStefan Hajnoczi 	struct virtqueue *vq;     /* protected by ->lock */
44a62a8ef9SStefan Hajnoczi 	struct work_struct done_work;
45a62a8ef9SStefan Hajnoczi 	struct list_head queued_reqs;
4651fecdd2SVivek Goyal 	struct list_head end_reqs;	/* End these requests */
47a62a8ef9SStefan Hajnoczi 	struct delayed_work dispatch_work;
48a62a8ef9SStefan Hajnoczi 	struct fuse_dev *fud;
49a62a8ef9SStefan Hajnoczi 	bool connected;
50a62a8ef9SStefan Hajnoczi 	long in_flight;
51724c15a4SVivek Goyal 	struct completion in_flight_zero; /* No inflight requests */
52b43b7e81SVivek Goyal 	char name[VQ_NAME_LEN];
53a62a8ef9SStefan Hajnoczi } ____cacheline_aligned_in_smp;
54a62a8ef9SStefan Hajnoczi 
55a62a8ef9SStefan Hajnoczi /* A virtio-fs device instance */
56a62a8ef9SStefan Hajnoczi struct virtio_fs {
57a62a8ef9SStefan Hajnoczi 	struct kref refcount;
58a62a8ef9SStefan Hajnoczi 	struct list_head list;    /* on virtio_fs_instances */
59a62a8ef9SStefan Hajnoczi 	char *tag;
60a62a8ef9SStefan Hajnoczi 	struct virtio_fs_vq *vqs;
61a62a8ef9SStefan Hajnoczi 	unsigned int nvqs;               /* number of virtqueues */
62a62a8ef9SStefan Hajnoczi 	unsigned int num_request_queues; /* number of request queues */
6322f3787eSStefan Hajnoczi 	struct dax_device *dax_dev;
6422f3787eSStefan Hajnoczi 
6522f3787eSStefan Hajnoczi 	/* DAX memory window where file contents are mapped */
6622f3787eSStefan Hajnoczi 	void *window_kaddr;
6722f3787eSStefan Hajnoczi 	phys_addr_t window_phys_addr;
6822f3787eSStefan Hajnoczi 	size_t window_len;
69a62a8ef9SStefan Hajnoczi };
70a62a8ef9SStefan Hajnoczi 
711efcf39eSVivek Goyal struct virtio_fs_forget_req {
72a62a8ef9SStefan Hajnoczi 	struct fuse_in_header ih;
73a62a8ef9SStefan Hajnoczi 	struct fuse_forget_in arg;
741efcf39eSVivek Goyal };
751efcf39eSVivek Goyal 
761efcf39eSVivek Goyal struct virtio_fs_forget {
77a62a8ef9SStefan Hajnoczi 	/* This request can be temporarily queued on virt queue */
78a62a8ef9SStefan Hajnoczi 	struct list_head list;
791efcf39eSVivek Goyal 	struct virtio_fs_forget_req req;
80a62a8ef9SStefan Hajnoczi };
81a62a8ef9SStefan Hajnoczi 
82bb737bbeSVivek Goyal struct virtio_fs_req_work {
83bb737bbeSVivek Goyal 	struct fuse_req *req;
84bb737bbeSVivek Goyal 	struct virtio_fs_vq *fsvq;
85bb737bbeSVivek Goyal 	struct work_struct done_work;
86bb737bbeSVivek Goyal };
87bb737bbeSVivek Goyal 
88a9bfd9ddSVivek Goyal static int virtio_fs_enqueue_req(struct virtio_fs_vq *fsvq,
89a9bfd9ddSVivek Goyal 				 struct fuse_req *req, bool in_flight);
90a9bfd9ddSVivek Goyal 
911dd53957SVivek Goyal enum {
921dd53957SVivek Goyal 	OPT_DAX,
931dd53957SVivek Goyal };
941dd53957SVivek Goyal 
951dd53957SVivek Goyal static const struct fs_parameter_spec virtio_fs_parameters[] = {
961dd53957SVivek Goyal 	fsparam_flag("dax", OPT_DAX),
971dd53957SVivek Goyal 	{}
981dd53957SVivek Goyal };
991dd53957SVivek Goyal 
1001dd53957SVivek Goyal static int virtio_fs_parse_param(struct fs_context *fc,
1011dd53957SVivek Goyal 				 struct fs_parameter *param)
1021dd53957SVivek Goyal {
1031dd53957SVivek Goyal 	struct fs_parse_result result;
1041dd53957SVivek Goyal 	struct fuse_fs_context *ctx = fc->fs_private;
1051dd53957SVivek Goyal 	int opt;
1061dd53957SVivek Goyal 
1071dd53957SVivek Goyal 	opt = fs_parse(fc, virtio_fs_parameters, param, &result);
1081dd53957SVivek Goyal 	if (opt < 0)
1091dd53957SVivek Goyal 		return opt;
1101dd53957SVivek Goyal 
1111dd53957SVivek Goyal 	switch (opt) {
1121dd53957SVivek Goyal 	case OPT_DAX:
1131dd53957SVivek Goyal 		ctx->dax = 1;
1141dd53957SVivek Goyal 		break;
1151dd53957SVivek Goyal 	default:
1161dd53957SVivek Goyal 		return -EINVAL;
1171dd53957SVivek Goyal 	}
1181dd53957SVivek Goyal 
1191dd53957SVivek Goyal 	return 0;
1201dd53957SVivek Goyal }
1211dd53957SVivek Goyal 
1221dd53957SVivek Goyal static void virtio_fs_free_fc(struct fs_context *fc)
1231dd53957SVivek Goyal {
1241dd53957SVivek Goyal 	struct fuse_fs_context *ctx = fc->fs_private;
1251dd53957SVivek Goyal 
1261dd53957SVivek Goyal 	kfree(ctx);
1271dd53957SVivek Goyal }
1281dd53957SVivek Goyal 
129a62a8ef9SStefan Hajnoczi static inline struct virtio_fs_vq *vq_to_fsvq(struct virtqueue *vq)
130a62a8ef9SStefan Hajnoczi {
131a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs = vq->vdev->priv;
132a62a8ef9SStefan Hajnoczi 
133a62a8ef9SStefan Hajnoczi 	return &fs->vqs[vq->index];
134a62a8ef9SStefan Hajnoczi }
135a62a8ef9SStefan Hajnoczi 
136a62a8ef9SStefan Hajnoczi static inline struct fuse_pqueue *vq_to_fpq(struct virtqueue *vq)
137a62a8ef9SStefan Hajnoczi {
138a62a8ef9SStefan Hajnoczi 	return &vq_to_fsvq(vq)->fud->pq;
139a62a8ef9SStefan Hajnoczi }
140a62a8ef9SStefan Hajnoczi 
141c17ea009SVivek Goyal /* Should be called with fsvq->lock held. */
142c17ea009SVivek Goyal static inline void inc_in_flight_req(struct virtio_fs_vq *fsvq)
143c17ea009SVivek Goyal {
144c17ea009SVivek Goyal 	fsvq->in_flight++;
145c17ea009SVivek Goyal }
146c17ea009SVivek Goyal 
147c17ea009SVivek Goyal /* Should be called with fsvq->lock held. */
148c17ea009SVivek Goyal static inline void dec_in_flight_req(struct virtio_fs_vq *fsvq)
149c17ea009SVivek Goyal {
150c17ea009SVivek Goyal 	WARN_ON(fsvq->in_flight <= 0);
151c17ea009SVivek Goyal 	fsvq->in_flight--;
152724c15a4SVivek Goyal 	if (!fsvq->in_flight)
153724c15a4SVivek Goyal 		complete(&fsvq->in_flight_zero);
154c17ea009SVivek Goyal }
155c17ea009SVivek Goyal 
156a62a8ef9SStefan Hajnoczi static void release_virtio_fs_obj(struct kref *ref)
157a62a8ef9SStefan Hajnoczi {
158a62a8ef9SStefan Hajnoczi 	struct virtio_fs *vfs = container_of(ref, struct virtio_fs, refcount);
159a62a8ef9SStefan Hajnoczi 
160a62a8ef9SStefan Hajnoczi 	kfree(vfs->vqs);
161a62a8ef9SStefan Hajnoczi 	kfree(vfs);
162a62a8ef9SStefan Hajnoczi }
163a62a8ef9SStefan Hajnoczi 
164a62a8ef9SStefan Hajnoczi /* Make sure virtiofs_mutex is held */
165a62a8ef9SStefan Hajnoczi static void virtio_fs_put(struct virtio_fs *fs)
166a62a8ef9SStefan Hajnoczi {
167a62a8ef9SStefan Hajnoczi 	kref_put(&fs->refcount, release_virtio_fs_obj);
168a62a8ef9SStefan Hajnoczi }
169a62a8ef9SStefan Hajnoczi 
170a62a8ef9SStefan Hajnoczi static void virtio_fs_fiq_release(struct fuse_iqueue *fiq)
171a62a8ef9SStefan Hajnoczi {
172a62a8ef9SStefan Hajnoczi 	struct virtio_fs *vfs = fiq->priv;
173a62a8ef9SStefan Hajnoczi 
174a62a8ef9SStefan Hajnoczi 	mutex_lock(&virtio_fs_mutex);
175a62a8ef9SStefan Hajnoczi 	virtio_fs_put(vfs);
176a62a8ef9SStefan Hajnoczi 	mutex_unlock(&virtio_fs_mutex);
177a62a8ef9SStefan Hajnoczi }
178a62a8ef9SStefan Hajnoczi 
179a62a8ef9SStefan Hajnoczi static void virtio_fs_drain_queue(struct virtio_fs_vq *fsvq)
180a62a8ef9SStefan Hajnoczi {
181a62a8ef9SStefan Hajnoczi 	WARN_ON(fsvq->in_flight < 0);
182a62a8ef9SStefan Hajnoczi 
183a62a8ef9SStefan Hajnoczi 	/* Wait for in flight requests to finish.*/
184a62a8ef9SStefan Hajnoczi 	spin_lock(&fsvq->lock);
185724c15a4SVivek Goyal 	if (fsvq->in_flight) {
186724c15a4SVivek Goyal 		/* We are holding virtio_fs_mutex. There should not be any
187724c15a4SVivek Goyal 		 * waiters waiting for completion.
188724c15a4SVivek Goyal 		 */
189724c15a4SVivek Goyal 		reinit_completion(&fsvq->in_flight_zero);
190a62a8ef9SStefan Hajnoczi 		spin_unlock(&fsvq->lock);
191724c15a4SVivek Goyal 		wait_for_completion(&fsvq->in_flight_zero);
192724c15a4SVivek Goyal 	} else {
193a62a8ef9SStefan Hajnoczi 		spin_unlock(&fsvq->lock);
194a62a8ef9SStefan Hajnoczi 	}
195a62a8ef9SStefan Hajnoczi 
196a62a8ef9SStefan Hajnoczi 	flush_work(&fsvq->done_work);
197a62a8ef9SStefan Hajnoczi 	flush_delayed_work(&fsvq->dispatch_work);
198a62a8ef9SStefan Hajnoczi }
199a62a8ef9SStefan Hajnoczi 
200724c15a4SVivek Goyal static void virtio_fs_drain_all_queues_locked(struct virtio_fs *fs)
201a62a8ef9SStefan Hajnoczi {
202a62a8ef9SStefan Hajnoczi 	struct virtio_fs_vq *fsvq;
203a62a8ef9SStefan Hajnoczi 	int i;
204a62a8ef9SStefan Hajnoczi 
205a62a8ef9SStefan Hajnoczi 	for (i = 0; i < fs->nvqs; i++) {
206a62a8ef9SStefan Hajnoczi 		fsvq = &fs->vqs[i];
207a62a8ef9SStefan Hajnoczi 		virtio_fs_drain_queue(fsvq);
208a62a8ef9SStefan Hajnoczi 	}
209a62a8ef9SStefan Hajnoczi }
210a62a8ef9SStefan Hajnoczi 
211724c15a4SVivek Goyal static void virtio_fs_drain_all_queues(struct virtio_fs *fs)
212724c15a4SVivek Goyal {
213724c15a4SVivek Goyal 	/* Provides mutual exclusion between ->remove and ->kill_sb
214724c15a4SVivek Goyal 	 * paths. We don't want both of these draining queue at the
215724c15a4SVivek Goyal 	 * same time. Current completion logic reinits completion
216724c15a4SVivek Goyal 	 * and that means there should not be any other thread
217724c15a4SVivek Goyal 	 * doing reinit or waiting for completion already.
218724c15a4SVivek Goyal 	 */
219724c15a4SVivek Goyal 	mutex_lock(&virtio_fs_mutex);
220724c15a4SVivek Goyal 	virtio_fs_drain_all_queues_locked(fs);
221724c15a4SVivek Goyal 	mutex_unlock(&virtio_fs_mutex);
222724c15a4SVivek Goyal }
223724c15a4SVivek Goyal 
224a62a8ef9SStefan Hajnoczi static void virtio_fs_start_all_queues(struct virtio_fs *fs)
225a62a8ef9SStefan Hajnoczi {
226a62a8ef9SStefan Hajnoczi 	struct virtio_fs_vq *fsvq;
227a62a8ef9SStefan Hajnoczi 	int i;
228a62a8ef9SStefan Hajnoczi 
229a62a8ef9SStefan Hajnoczi 	for (i = 0; i < fs->nvqs; i++) {
230a62a8ef9SStefan Hajnoczi 		fsvq = &fs->vqs[i];
231a62a8ef9SStefan Hajnoczi 		spin_lock(&fsvq->lock);
232a62a8ef9SStefan Hajnoczi 		fsvq->connected = true;
233a62a8ef9SStefan Hajnoczi 		spin_unlock(&fsvq->lock);
234a62a8ef9SStefan Hajnoczi 	}
235a62a8ef9SStefan Hajnoczi }
236a62a8ef9SStefan Hajnoczi 
237a62a8ef9SStefan Hajnoczi /* Add a new instance to the list or return -EEXIST if tag name exists*/
238a62a8ef9SStefan Hajnoczi static int virtio_fs_add_instance(struct virtio_fs *fs)
239a62a8ef9SStefan Hajnoczi {
240a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs2;
241a62a8ef9SStefan Hajnoczi 	bool duplicate = false;
242a62a8ef9SStefan Hajnoczi 
243a62a8ef9SStefan Hajnoczi 	mutex_lock(&virtio_fs_mutex);
244a62a8ef9SStefan Hajnoczi 
245a62a8ef9SStefan Hajnoczi 	list_for_each_entry(fs2, &virtio_fs_instances, list) {
246a62a8ef9SStefan Hajnoczi 		if (strcmp(fs->tag, fs2->tag) == 0)
247a62a8ef9SStefan Hajnoczi 			duplicate = true;
248a62a8ef9SStefan Hajnoczi 	}
249a62a8ef9SStefan Hajnoczi 
250a62a8ef9SStefan Hajnoczi 	if (!duplicate)
251a62a8ef9SStefan Hajnoczi 		list_add_tail(&fs->list, &virtio_fs_instances);
252a62a8ef9SStefan Hajnoczi 
253a62a8ef9SStefan Hajnoczi 	mutex_unlock(&virtio_fs_mutex);
254a62a8ef9SStefan Hajnoczi 
255a62a8ef9SStefan Hajnoczi 	if (duplicate)
256a62a8ef9SStefan Hajnoczi 		return -EEXIST;
257a62a8ef9SStefan Hajnoczi 	return 0;
258a62a8ef9SStefan Hajnoczi }
259a62a8ef9SStefan Hajnoczi 
260a62a8ef9SStefan Hajnoczi /* Return the virtio_fs with a given tag, or NULL */
261a62a8ef9SStefan Hajnoczi static struct virtio_fs *virtio_fs_find_instance(const char *tag)
262a62a8ef9SStefan Hajnoczi {
263a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs;
264a62a8ef9SStefan Hajnoczi 
265a62a8ef9SStefan Hajnoczi 	mutex_lock(&virtio_fs_mutex);
266a62a8ef9SStefan Hajnoczi 
267a62a8ef9SStefan Hajnoczi 	list_for_each_entry(fs, &virtio_fs_instances, list) {
268a62a8ef9SStefan Hajnoczi 		if (strcmp(fs->tag, tag) == 0) {
269a62a8ef9SStefan Hajnoczi 			kref_get(&fs->refcount);
270a62a8ef9SStefan Hajnoczi 			goto found;
271a62a8ef9SStefan Hajnoczi 		}
272a62a8ef9SStefan Hajnoczi 	}
273a62a8ef9SStefan Hajnoczi 
274a62a8ef9SStefan Hajnoczi 	fs = NULL; /* not found */
275a62a8ef9SStefan Hajnoczi 
276a62a8ef9SStefan Hajnoczi found:
277a62a8ef9SStefan Hajnoczi 	mutex_unlock(&virtio_fs_mutex);
278a62a8ef9SStefan Hajnoczi 
279a62a8ef9SStefan Hajnoczi 	return fs;
280a62a8ef9SStefan Hajnoczi }
281a62a8ef9SStefan Hajnoczi 
282a62a8ef9SStefan Hajnoczi static void virtio_fs_free_devs(struct virtio_fs *fs)
283a62a8ef9SStefan Hajnoczi {
284a62a8ef9SStefan Hajnoczi 	unsigned int i;
285a62a8ef9SStefan Hajnoczi 
286a62a8ef9SStefan Hajnoczi 	for (i = 0; i < fs->nvqs; i++) {
287a62a8ef9SStefan Hajnoczi 		struct virtio_fs_vq *fsvq = &fs->vqs[i];
288a62a8ef9SStefan Hajnoczi 
289a62a8ef9SStefan Hajnoczi 		if (!fsvq->fud)
290a62a8ef9SStefan Hajnoczi 			continue;
291a62a8ef9SStefan Hajnoczi 
292a62a8ef9SStefan Hajnoczi 		fuse_dev_free(fsvq->fud);
293a62a8ef9SStefan Hajnoczi 		fsvq->fud = NULL;
294a62a8ef9SStefan Hajnoczi 	}
295a62a8ef9SStefan Hajnoczi }
296a62a8ef9SStefan Hajnoczi 
297a62a8ef9SStefan Hajnoczi /* Read filesystem name from virtio config into fs->tag (must kfree()). */
298a62a8ef9SStefan Hajnoczi static int virtio_fs_read_tag(struct virtio_device *vdev, struct virtio_fs *fs)
299a62a8ef9SStefan Hajnoczi {
300a62a8ef9SStefan Hajnoczi 	char tag_buf[sizeof_field(struct virtio_fs_config, tag)];
301a62a8ef9SStefan Hajnoczi 	char *end;
302a62a8ef9SStefan Hajnoczi 	size_t len;
303a62a8ef9SStefan Hajnoczi 
304a62a8ef9SStefan Hajnoczi 	virtio_cread_bytes(vdev, offsetof(struct virtio_fs_config, tag),
305a62a8ef9SStefan Hajnoczi 			   &tag_buf, sizeof(tag_buf));
306a62a8ef9SStefan Hajnoczi 	end = memchr(tag_buf, '\0', sizeof(tag_buf));
307a62a8ef9SStefan Hajnoczi 	if (end == tag_buf)
308a62a8ef9SStefan Hajnoczi 		return -EINVAL; /* empty tag */
309a62a8ef9SStefan Hajnoczi 	if (!end)
310a62a8ef9SStefan Hajnoczi 		end = &tag_buf[sizeof(tag_buf)];
311a62a8ef9SStefan Hajnoczi 
312a62a8ef9SStefan Hajnoczi 	len = end - tag_buf;
313a62a8ef9SStefan Hajnoczi 	fs->tag = devm_kmalloc(&vdev->dev, len + 1, GFP_KERNEL);
314a62a8ef9SStefan Hajnoczi 	if (!fs->tag)
315a62a8ef9SStefan Hajnoczi 		return -ENOMEM;
316a62a8ef9SStefan Hajnoczi 	memcpy(fs->tag, tag_buf, len);
317a62a8ef9SStefan Hajnoczi 	fs->tag[len] = '\0';
318a62a8ef9SStefan Hajnoczi 	return 0;
319a62a8ef9SStefan Hajnoczi }
320a62a8ef9SStefan Hajnoczi 
321a62a8ef9SStefan Hajnoczi /* Work function for hiprio completion */
322a62a8ef9SStefan Hajnoczi static void virtio_fs_hiprio_done_work(struct work_struct *work)
323a62a8ef9SStefan Hajnoczi {
324a62a8ef9SStefan Hajnoczi 	struct virtio_fs_vq *fsvq = container_of(work, struct virtio_fs_vq,
325a62a8ef9SStefan Hajnoczi 						 done_work);
326a62a8ef9SStefan Hajnoczi 	struct virtqueue *vq = fsvq->vq;
327a62a8ef9SStefan Hajnoczi 
328a62a8ef9SStefan Hajnoczi 	/* Free completed FUSE_FORGET requests */
329a62a8ef9SStefan Hajnoczi 	spin_lock(&fsvq->lock);
330a62a8ef9SStefan Hajnoczi 	do {
331a62a8ef9SStefan Hajnoczi 		unsigned int len;
332a62a8ef9SStefan Hajnoczi 		void *req;
333a62a8ef9SStefan Hajnoczi 
334a62a8ef9SStefan Hajnoczi 		virtqueue_disable_cb(vq);
335a62a8ef9SStefan Hajnoczi 
336a62a8ef9SStefan Hajnoczi 		while ((req = virtqueue_get_buf(vq, &len)) != NULL) {
337a62a8ef9SStefan Hajnoczi 			kfree(req);
338c17ea009SVivek Goyal 			dec_in_flight_req(fsvq);
339a62a8ef9SStefan Hajnoczi 		}
340a62a8ef9SStefan Hajnoczi 	} while (!virtqueue_enable_cb(vq) && likely(!virtqueue_is_broken(vq)));
341a62a8ef9SStefan Hajnoczi 	spin_unlock(&fsvq->lock);
342a62a8ef9SStefan Hajnoczi }
343a62a8ef9SStefan Hajnoczi 
34451fecdd2SVivek Goyal static void virtio_fs_request_dispatch_work(struct work_struct *work)
345a62a8ef9SStefan Hajnoczi {
34651fecdd2SVivek Goyal 	struct fuse_req *req;
34751fecdd2SVivek Goyal 	struct virtio_fs_vq *fsvq = container_of(work, struct virtio_fs_vq,
34851fecdd2SVivek Goyal 						 dispatch_work.work);
349a9bfd9ddSVivek Goyal 	int ret;
35051fecdd2SVivek Goyal 
35151fecdd2SVivek Goyal 	pr_debug("virtio-fs: worker %s called.\n", __func__);
35251fecdd2SVivek Goyal 	while (1) {
35351fecdd2SVivek Goyal 		spin_lock(&fsvq->lock);
35451fecdd2SVivek Goyal 		req = list_first_entry_or_null(&fsvq->end_reqs, struct fuse_req,
35551fecdd2SVivek Goyal 					       list);
35651fecdd2SVivek Goyal 		if (!req) {
35751fecdd2SVivek Goyal 			spin_unlock(&fsvq->lock);
358a9bfd9ddSVivek Goyal 			break;
35951fecdd2SVivek Goyal 		}
36051fecdd2SVivek Goyal 
36151fecdd2SVivek Goyal 		list_del_init(&req->list);
36251fecdd2SVivek Goyal 		spin_unlock(&fsvq->lock);
3638f622e94SMax Reitz 		fuse_request_end(req);
36451fecdd2SVivek Goyal 	}
365a9bfd9ddSVivek Goyal 
366a9bfd9ddSVivek Goyal 	/* Dispatch pending requests */
367a9bfd9ddSVivek Goyal 	while (1) {
368a9bfd9ddSVivek Goyal 		spin_lock(&fsvq->lock);
369a9bfd9ddSVivek Goyal 		req = list_first_entry_or_null(&fsvq->queued_reqs,
370a9bfd9ddSVivek Goyal 					       struct fuse_req, list);
371a9bfd9ddSVivek Goyal 		if (!req) {
372a9bfd9ddSVivek Goyal 			spin_unlock(&fsvq->lock);
373a9bfd9ddSVivek Goyal 			return;
374a9bfd9ddSVivek Goyal 		}
375a9bfd9ddSVivek Goyal 		list_del_init(&req->list);
376a9bfd9ddSVivek Goyal 		spin_unlock(&fsvq->lock);
377a9bfd9ddSVivek Goyal 
378a9bfd9ddSVivek Goyal 		ret = virtio_fs_enqueue_req(fsvq, req, true);
379a9bfd9ddSVivek Goyal 		if (ret < 0) {
380a9bfd9ddSVivek Goyal 			if (ret == -ENOMEM || ret == -ENOSPC) {
381a9bfd9ddSVivek Goyal 				spin_lock(&fsvq->lock);
382a9bfd9ddSVivek Goyal 				list_add_tail(&req->list, &fsvq->queued_reqs);
383a9bfd9ddSVivek Goyal 				schedule_delayed_work(&fsvq->dispatch_work,
384a9bfd9ddSVivek Goyal 						      msecs_to_jiffies(1));
385a9bfd9ddSVivek Goyal 				spin_unlock(&fsvq->lock);
386a9bfd9ddSVivek Goyal 				return;
387a9bfd9ddSVivek Goyal 			}
388a9bfd9ddSVivek Goyal 			req->out.h.error = ret;
389a9bfd9ddSVivek Goyal 			spin_lock(&fsvq->lock);
390a9bfd9ddSVivek Goyal 			dec_in_flight_req(fsvq);
391a9bfd9ddSVivek Goyal 			spin_unlock(&fsvq->lock);
392a9bfd9ddSVivek Goyal 			pr_err("virtio-fs: virtio_fs_enqueue_req() failed %d\n",
393a9bfd9ddSVivek Goyal 			       ret);
3948f622e94SMax Reitz 			fuse_request_end(req);
395a9bfd9ddSVivek Goyal 		}
396a9bfd9ddSVivek Goyal 	}
397a62a8ef9SStefan Hajnoczi }
398a62a8ef9SStefan Hajnoczi 
39958ada94fSVivek Goyal /*
40058ada94fSVivek Goyal  * Returns 1 if queue is full and sender should wait a bit before sending
40158ada94fSVivek Goyal  * next request, 0 otherwise.
40258ada94fSVivek Goyal  */
40358ada94fSVivek Goyal static int send_forget_request(struct virtio_fs_vq *fsvq,
40458ada94fSVivek Goyal 			       struct virtio_fs_forget *forget,
40558ada94fSVivek Goyal 			       bool in_flight)
40658ada94fSVivek Goyal {
40758ada94fSVivek Goyal 	struct scatterlist sg;
40858ada94fSVivek Goyal 	struct virtqueue *vq;
40958ada94fSVivek Goyal 	int ret = 0;
41058ada94fSVivek Goyal 	bool notify;
4111efcf39eSVivek Goyal 	struct virtio_fs_forget_req *req = &forget->req;
41258ada94fSVivek Goyal 
41358ada94fSVivek Goyal 	spin_lock(&fsvq->lock);
41458ada94fSVivek Goyal 	if (!fsvq->connected) {
41558ada94fSVivek Goyal 		if (in_flight)
41658ada94fSVivek Goyal 			dec_in_flight_req(fsvq);
41758ada94fSVivek Goyal 		kfree(forget);
41858ada94fSVivek Goyal 		goto out;
41958ada94fSVivek Goyal 	}
42058ada94fSVivek Goyal 
4211efcf39eSVivek Goyal 	sg_init_one(&sg, req, sizeof(*req));
42258ada94fSVivek Goyal 	vq = fsvq->vq;
42358ada94fSVivek Goyal 	dev_dbg(&vq->vdev->dev, "%s\n", __func__);
42458ada94fSVivek Goyal 
42558ada94fSVivek Goyal 	ret = virtqueue_add_outbuf(vq, &sg, 1, forget, GFP_ATOMIC);
42658ada94fSVivek Goyal 	if (ret < 0) {
42758ada94fSVivek Goyal 		if (ret == -ENOMEM || ret == -ENOSPC) {
42858ada94fSVivek Goyal 			pr_debug("virtio-fs: Could not queue FORGET: err=%d. Will try later\n",
42958ada94fSVivek Goyal 				 ret);
43058ada94fSVivek Goyal 			list_add_tail(&forget->list, &fsvq->queued_reqs);
43158ada94fSVivek Goyal 			schedule_delayed_work(&fsvq->dispatch_work,
43258ada94fSVivek Goyal 					      msecs_to_jiffies(1));
43358ada94fSVivek Goyal 			if (!in_flight)
43458ada94fSVivek Goyal 				inc_in_flight_req(fsvq);
43558ada94fSVivek Goyal 			/* Queue is full */
43658ada94fSVivek Goyal 			ret = 1;
43758ada94fSVivek Goyal 		} else {
43858ada94fSVivek Goyal 			pr_debug("virtio-fs: Could not queue FORGET: err=%d. Dropping it.\n",
43958ada94fSVivek Goyal 				 ret);
44058ada94fSVivek Goyal 			kfree(forget);
44158ada94fSVivek Goyal 			if (in_flight)
44258ada94fSVivek Goyal 				dec_in_flight_req(fsvq);
44358ada94fSVivek Goyal 		}
44458ada94fSVivek Goyal 		goto out;
44558ada94fSVivek Goyal 	}
44658ada94fSVivek Goyal 
44758ada94fSVivek Goyal 	if (!in_flight)
44858ada94fSVivek Goyal 		inc_in_flight_req(fsvq);
44958ada94fSVivek Goyal 	notify = virtqueue_kick_prepare(vq);
45058ada94fSVivek Goyal 	spin_unlock(&fsvq->lock);
45158ada94fSVivek Goyal 
45258ada94fSVivek Goyal 	if (notify)
45358ada94fSVivek Goyal 		virtqueue_notify(vq);
45458ada94fSVivek Goyal 	return ret;
45558ada94fSVivek Goyal out:
45658ada94fSVivek Goyal 	spin_unlock(&fsvq->lock);
45758ada94fSVivek Goyal 	return ret;
45858ada94fSVivek Goyal }
45958ada94fSVivek Goyal 
460a62a8ef9SStefan Hajnoczi static void virtio_fs_hiprio_dispatch_work(struct work_struct *work)
461a62a8ef9SStefan Hajnoczi {
462a62a8ef9SStefan Hajnoczi 	struct virtio_fs_forget *forget;
463a62a8ef9SStefan Hajnoczi 	struct virtio_fs_vq *fsvq = container_of(work, struct virtio_fs_vq,
464a62a8ef9SStefan Hajnoczi 						 dispatch_work.work);
465a62a8ef9SStefan Hajnoczi 	pr_debug("virtio-fs: worker %s called.\n", __func__);
466a62a8ef9SStefan Hajnoczi 	while (1) {
467a62a8ef9SStefan Hajnoczi 		spin_lock(&fsvq->lock);
468a62a8ef9SStefan Hajnoczi 		forget = list_first_entry_or_null(&fsvq->queued_reqs,
469a62a8ef9SStefan Hajnoczi 					struct virtio_fs_forget, list);
470a62a8ef9SStefan Hajnoczi 		if (!forget) {
471a62a8ef9SStefan Hajnoczi 			spin_unlock(&fsvq->lock);
472a62a8ef9SStefan Hajnoczi 			return;
473a62a8ef9SStefan Hajnoczi 		}
474a62a8ef9SStefan Hajnoczi 
475a62a8ef9SStefan Hajnoczi 		list_del(&forget->list);
476a62a8ef9SStefan Hajnoczi 		spin_unlock(&fsvq->lock);
47758ada94fSVivek Goyal 		if (send_forget_request(fsvq, forget, true))
478a62a8ef9SStefan Hajnoczi 			return;
479a62a8ef9SStefan Hajnoczi 	}
480a62a8ef9SStefan Hajnoczi }
481a62a8ef9SStefan Hajnoczi 
482a62a8ef9SStefan Hajnoczi /* Allocate and copy args into req->argbuf */
483a62a8ef9SStefan Hajnoczi static int copy_args_to_argbuf(struct fuse_req *req)
484a62a8ef9SStefan Hajnoczi {
485a62a8ef9SStefan Hajnoczi 	struct fuse_args *args = req->args;
486a62a8ef9SStefan Hajnoczi 	unsigned int offset = 0;
487a62a8ef9SStefan Hajnoczi 	unsigned int num_in;
488a62a8ef9SStefan Hajnoczi 	unsigned int num_out;
489a62a8ef9SStefan Hajnoczi 	unsigned int len;
490a62a8ef9SStefan Hajnoczi 	unsigned int i;
491a62a8ef9SStefan Hajnoczi 
492a62a8ef9SStefan Hajnoczi 	num_in = args->in_numargs - args->in_pages;
493a62a8ef9SStefan Hajnoczi 	num_out = args->out_numargs - args->out_pages;
494a62a8ef9SStefan Hajnoczi 	len = fuse_len_args(num_in, (struct fuse_arg *) args->in_args) +
495a62a8ef9SStefan Hajnoczi 	      fuse_len_args(num_out, args->out_args);
496a62a8ef9SStefan Hajnoczi 
497a62a8ef9SStefan Hajnoczi 	req->argbuf = kmalloc(len, GFP_ATOMIC);
498a62a8ef9SStefan Hajnoczi 	if (!req->argbuf)
499a62a8ef9SStefan Hajnoczi 		return -ENOMEM;
500a62a8ef9SStefan Hajnoczi 
501a62a8ef9SStefan Hajnoczi 	for (i = 0; i < num_in; i++) {
502a62a8ef9SStefan Hajnoczi 		memcpy(req->argbuf + offset,
503a62a8ef9SStefan Hajnoczi 		       args->in_args[i].value,
504a62a8ef9SStefan Hajnoczi 		       args->in_args[i].size);
505a62a8ef9SStefan Hajnoczi 		offset += args->in_args[i].size;
506a62a8ef9SStefan Hajnoczi 	}
507a62a8ef9SStefan Hajnoczi 
508a62a8ef9SStefan Hajnoczi 	return 0;
509a62a8ef9SStefan Hajnoczi }
510a62a8ef9SStefan Hajnoczi 
511a62a8ef9SStefan Hajnoczi /* Copy args out of and free req->argbuf */
512a62a8ef9SStefan Hajnoczi static void copy_args_from_argbuf(struct fuse_args *args, struct fuse_req *req)
513a62a8ef9SStefan Hajnoczi {
514a62a8ef9SStefan Hajnoczi 	unsigned int remaining;
515a62a8ef9SStefan Hajnoczi 	unsigned int offset;
516a62a8ef9SStefan Hajnoczi 	unsigned int num_in;
517a62a8ef9SStefan Hajnoczi 	unsigned int num_out;
518a62a8ef9SStefan Hajnoczi 	unsigned int i;
519a62a8ef9SStefan Hajnoczi 
520a62a8ef9SStefan Hajnoczi 	remaining = req->out.h.len - sizeof(req->out.h);
521a62a8ef9SStefan Hajnoczi 	num_in = args->in_numargs - args->in_pages;
522a62a8ef9SStefan Hajnoczi 	num_out = args->out_numargs - args->out_pages;
523a62a8ef9SStefan Hajnoczi 	offset = fuse_len_args(num_in, (struct fuse_arg *)args->in_args);
524a62a8ef9SStefan Hajnoczi 
525a62a8ef9SStefan Hajnoczi 	for (i = 0; i < num_out; i++) {
526a62a8ef9SStefan Hajnoczi 		unsigned int argsize = args->out_args[i].size;
527a62a8ef9SStefan Hajnoczi 
528a62a8ef9SStefan Hajnoczi 		if (args->out_argvar &&
529a62a8ef9SStefan Hajnoczi 		    i == args->out_numargs - 1 &&
530a62a8ef9SStefan Hajnoczi 		    argsize > remaining) {
531a62a8ef9SStefan Hajnoczi 			argsize = remaining;
532a62a8ef9SStefan Hajnoczi 		}
533a62a8ef9SStefan Hajnoczi 
534a62a8ef9SStefan Hajnoczi 		memcpy(args->out_args[i].value, req->argbuf + offset, argsize);
535a62a8ef9SStefan Hajnoczi 		offset += argsize;
536a62a8ef9SStefan Hajnoczi 
537a62a8ef9SStefan Hajnoczi 		if (i != args->out_numargs - 1)
538a62a8ef9SStefan Hajnoczi 			remaining -= argsize;
539a62a8ef9SStefan Hajnoczi 	}
540a62a8ef9SStefan Hajnoczi 
541a62a8ef9SStefan Hajnoczi 	/* Store the actual size of the variable-length arg */
542a62a8ef9SStefan Hajnoczi 	if (args->out_argvar)
543a62a8ef9SStefan Hajnoczi 		args->out_args[args->out_numargs - 1].size = remaining;
544a62a8ef9SStefan Hajnoczi 
545a62a8ef9SStefan Hajnoczi 	kfree(req->argbuf);
546a62a8ef9SStefan Hajnoczi 	req->argbuf = NULL;
547a62a8ef9SStefan Hajnoczi }
548a62a8ef9SStefan Hajnoczi 
549a62a8ef9SStefan Hajnoczi /* Work function for request completion */
550bb737bbeSVivek Goyal static void virtio_fs_request_complete(struct fuse_req *req,
551bb737bbeSVivek Goyal 				       struct virtio_fs_vq *fsvq)
552a62a8ef9SStefan Hajnoczi {
553a62a8ef9SStefan Hajnoczi 	struct fuse_pqueue *fpq = &fsvq->fud->pq;
554a62a8ef9SStefan Hajnoczi 	struct fuse_args *args;
555bb737bbeSVivek Goyal 	struct fuse_args_pages *ap;
556a62a8ef9SStefan Hajnoczi 	unsigned int len, i, thislen;
557a62a8ef9SStefan Hajnoczi 	struct page *page;
558a62a8ef9SStefan Hajnoczi 
559a62a8ef9SStefan Hajnoczi 	/*
560a62a8ef9SStefan Hajnoczi 	 * TODO verify that server properly follows FUSE protocol
561a62a8ef9SStefan Hajnoczi 	 * (oh.uniq, oh.len)
562a62a8ef9SStefan Hajnoczi 	 */
563a62a8ef9SStefan Hajnoczi 	args = req->args;
564a62a8ef9SStefan Hajnoczi 	copy_args_from_argbuf(args, req);
565a62a8ef9SStefan Hajnoczi 
566a62a8ef9SStefan Hajnoczi 	if (args->out_pages && args->page_zeroing) {
567a62a8ef9SStefan Hajnoczi 		len = args->out_args[args->out_numargs - 1].size;
568a62a8ef9SStefan Hajnoczi 		ap = container_of(args, typeof(*ap), args);
569a62a8ef9SStefan Hajnoczi 		for (i = 0; i < ap->num_pages; i++) {
570a62a8ef9SStefan Hajnoczi 			thislen = ap->descs[i].length;
571a62a8ef9SStefan Hajnoczi 			if (len < thislen) {
572a62a8ef9SStefan Hajnoczi 				WARN_ON(ap->descs[i].offset);
573a62a8ef9SStefan Hajnoczi 				page = ap->pages[i];
574a62a8ef9SStefan Hajnoczi 				zero_user_segment(page, len, thislen);
575a62a8ef9SStefan Hajnoczi 				len = 0;
576a62a8ef9SStefan Hajnoczi 			} else {
577a62a8ef9SStefan Hajnoczi 				len -= thislen;
578a62a8ef9SStefan Hajnoczi 			}
579a62a8ef9SStefan Hajnoczi 		}
580a62a8ef9SStefan Hajnoczi 	}
581a62a8ef9SStefan Hajnoczi 
582a62a8ef9SStefan Hajnoczi 	spin_lock(&fpq->lock);
583a62a8ef9SStefan Hajnoczi 	clear_bit(FR_SENT, &req->flags);
584a62a8ef9SStefan Hajnoczi 	spin_unlock(&fpq->lock);
585a62a8ef9SStefan Hajnoczi 
5868f622e94SMax Reitz 	fuse_request_end(req);
587a62a8ef9SStefan Hajnoczi 	spin_lock(&fsvq->lock);
588c17ea009SVivek Goyal 	dec_in_flight_req(fsvq);
589a62a8ef9SStefan Hajnoczi 	spin_unlock(&fsvq->lock);
590a62a8ef9SStefan Hajnoczi }
591bb737bbeSVivek Goyal 
592bb737bbeSVivek Goyal static void virtio_fs_complete_req_work(struct work_struct *work)
593bb737bbeSVivek Goyal {
594bb737bbeSVivek Goyal 	struct virtio_fs_req_work *w =
595bb737bbeSVivek Goyal 		container_of(work, typeof(*w), done_work);
596bb737bbeSVivek Goyal 
597bb737bbeSVivek Goyal 	virtio_fs_request_complete(w->req, w->fsvq);
598bb737bbeSVivek Goyal 	kfree(w);
599bb737bbeSVivek Goyal }
600bb737bbeSVivek Goyal 
601bb737bbeSVivek Goyal static void virtio_fs_requests_done_work(struct work_struct *work)
602bb737bbeSVivek Goyal {
603bb737bbeSVivek Goyal 	struct virtio_fs_vq *fsvq = container_of(work, struct virtio_fs_vq,
604bb737bbeSVivek Goyal 						 done_work);
605bb737bbeSVivek Goyal 	struct fuse_pqueue *fpq = &fsvq->fud->pq;
606bb737bbeSVivek Goyal 	struct virtqueue *vq = fsvq->vq;
607bb737bbeSVivek Goyal 	struct fuse_req *req;
608bb737bbeSVivek Goyal 	struct fuse_req *next;
609bb737bbeSVivek Goyal 	unsigned int len;
610bb737bbeSVivek Goyal 	LIST_HEAD(reqs);
611bb737bbeSVivek Goyal 
612bb737bbeSVivek Goyal 	/* Collect completed requests off the virtqueue */
613bb737bbeSVivek Goyal 	spin_lock(&fsvq->lock);
614bb737bbeSVivek Goyal 	do {
615bb737bbeSVivek Goyal 		virtqueue_disable_cb(vq);
616bb737bbeSVivek Goyal 
617bb737bbeSVivek Goyal 		while ((req = virtqueue_get_buf(vq, &len)) != NULL) {
618bb737bbeSVivek Goyal 			spin_lock(&fpq->lock);
619bb737bbeSVivek Goyal 			list_move_tail(&req->list, &reqs);
620bb737bbeSVivek Goyal 			spin_unlock(&fpq->lock);
621bb737bbeSVivek Goyal 		}
622bb737bbeSVivek Goyal 	} while (!virtqueue_enable_cb(vq) && likely(!virtqueue_is_broken(vq)));
623bb737bbeSVivek Goyal 	spin_unlock(&fsvq->lock);
624bb737bbeSVivek Goyal 
625bb737bbeSVivek Goyal 	/* End requests */
626bb737bbeSVivek Goyal 	list_for_each_entry_safe(req, next, &reqs, list) {
627bb737bbeSVivek Goyal 		list_del_init(&req->list);
628bb737bbeSVivek Goyal 
629bb737bbeSVivek Goyal 		/* blocking async request completes in a worker context */
630bb737bbeSVivek Goyal 		if (req->args->may_block) {
631bb737bbeSVivek Goyal 			struct virtio_fs_req_work *w;
632bb737bbeSVivek Goyal 
633bb737bbeSVivek Goyal 			w = kzalloc(sizeof(*w), GFP_NOFS | __GFP_NOFAIL);
634bb737bbeSVivek Goyal 			INIT_WORK(&w->done_work, virtio_fs_complete_req_work);
635bb737bbeSVivek Goyal 			w->fsvq = fsvq;
636bb737bbeSVivek Goyal 			w->req = req;
637bb737bbeSVivek Goyal 			schedule_work(&w->done_work);
638bb737bbeSVivek Goyal 		} else {
639bb737bbeSVivek Goyal 			virtio_fs_request_complete(req, fsvq);
640bb737bbeSVivek Goyal 		}
641bb737bbeSVivek Goyal 	}
642a62a8ef9SStefan Hajnoczi }
643a62a8ef9SStefan Hajnoczi 
644a62a8ef9SStefan Hajnoczi /* Virtqueue interrupt handler */
645a62a8ef9SStefan Hajnoczi static void virtio_fs_vq_done(struct virtqueue *vq)
646a62a8ef9SStefan Hajnoczi {
647a62a8ef9SStefan Hajnoczi 	struct virtio_fs_vq *fsvq = vq_to_fsvq(vq);
648a62a8ef9SStefan Hajnoczi 
649a62a8ef9SStefan Hajnoczi 	dev_dbg(&vq->vdev->dev, "%s %s\n", __func__, fsvq->name);
650a62a8ef9SStefan Hajnoczi 
651a62a8ef9SStefan Hajnoczi 	schedule_work(&fsvq->done_work);
652a62a8ef9SStefan Hajnoczi }
653a62a8ef9SStefan Hajnoczi 
654b43b7e81SVivek Goyal static void virtio_fs_init_vq(struct virtio_fs_vq *fsvq, char *name,
655b43b7e81SVivek Goyal 			      int vq_type)
656b43b7e81SVivek Goyal {
657b43b7e81SVivek Goyal 	strncpy(fsvq->name, name, VQ_NAME_LEN);
658b43b7e81SVivek Goyal 	spin_lock_init(&fsvq->lock);
659b43b7e81SVivek Goyal 	INIT_LIST_HEAD(&fsvq->queued_reqs);
660b43b7e81SVivek Goyal 	INIT_LIST_HEAD(&fsvq->end_reqs);
661b43b7e81SVivek Goyal 	init_completion(&fsvq->in_flight_zero);
662b43b7e81SVivek Goyal 
663b43b7e81SVivek Goyal 	if (vq_type == VQ_REQUEST) {
664b43b7e81SVivek Goyal 		INIT_WORK(&fsvq->done_work, virtio_fs_requests_done_work);
665b43b7e81SVivek Goyal 		INIT_DELAYED_WORK(&fsvq->dispatch_work,
666b43b7e81SVivek Goyal 				  virtio_fs_request_dispatch_work);
667b43b7e81SVivek Goyal 	} else {
668b43b7e81SVivek Goyal 		INIT_WORK(&fsvq->done_work, virtio_fs_hiprio_done_work);
669b43b7e81SVivek Goyal 		INIT_DELAYED_WORK(&fsvq->dispatch_work,
670b43b7e81SVivek Goyal 				  virtio_fs_hiprio_dispatch_work);
671b43b7e81SVivek Goyal 	}
672b43b7e81SVivek Goyal }
673b43b7e81SVivek Goyal 
674a62a8ef9SStefan Hajnoczi /* Initialize virtqueues */
675a62a8ef9SStefan Hajnoczi static int virtio_fs_setup_vqs(struct virtio_device *vdev,
676a62a8ef9SStefan Hajnoczi 			       struct virtio_fs *fs)
677a62a8ef9SStefan Hajnoczi {
678a62a8ef9SStefan Hajnoczi 	struct virtqueue **vqs;
679a62a8ef9SStefan Hajnoczi 	vq_callback_t **callbacks;
680a62a8ef9SStefan Hajnoczi 	const char **names;
681a62a8ef9SStefan Hajnoczi 	unsigned int i;
682a62a8ef9SStefan Hajnoczi 	int ret = 0;
683a62a8ef9SStefan Hajnoczi 
6842c0349ecSMichael S. Tsirkin 	virtio_cread_le(vdev, struct virtio_fs_config, num_request_queues,
685a62a8ef9SStefan Hajnoczi 			&fs->num_request_queues);
686a62a8ef9SStefan Hajnoczi 	if (fs->num_request_queues == 0)
687a62a8ef9SStefan Hajnoczi 		return -EINVAL;
688a62a8ef9SStefan Hajnoczi 
689b43b7e81SVivek Goyal 	fs->nvqs = VQ_REQUEST + fs->num_request_queues;
690a62a8ef9SStefan Hajnoczi 	fs->vqs = kcalloc(fs->nvqs, sizeof(fs->vqs[VQ_HIPRIO]), GFP_KERNEL);
691a62a8ef9SStefan Hajnoczi 	if (!fs->vqs)
692a62a8ef9SStefan Hajnoczi 		return -ENOMEM;
693a62a8ef9SStefan Hajnoczi 
694a62a8ef9SStefan Hajnoczi 	vqs = kmalloc_array(fs->nvqs, sizeof(vqs[VQ_HIPRIO]), GFP_KERNEL);
695a62a8ef9SStefan Hajnoczi 	callbacks = kmalloc_array(fs->nvqs, sizeof(callbacks[VQ_HIPRIO]),
696a62a8ef9SStefan Hajnoczi 					GFP_KERNEL);
697a62a8ef9SStefan Hajnoczi 	names = kmalloc_array(fs->nvqs, sizeof(names[VQ_HIPRIO]), GFP_KERNEL);
698a62a8ef9SStefan Hajnoczi 	if (!vqs || !callbacks || !names) {
699a62a8ef9SStefan Hajnoczi 		ret = -ENOMEM;
700a62a8ef9SStefan Hajnoczi 		goto out;
701a62a8ef9SStefan Hajnoczi 	}
702a62a8ef9SStefan Hajnoczi 
703b43b7e81SVivek Goyal 	/* Initialize the hiprio/forget request virtqueue */
704a62a8ef9SStefan Hajnoczi 	callbacks[VQ_HIPRIO] = virtio_fs_vq_done;
705b43b7e81SVivek Goyal 	virtio_fs_init_vq(&fs->vqs[VQ_HIPRIO], "hiprio", VQ_HIPRIO);
706a62a8ef9SStefan Hajnoczi 	names[VQ_HIPRIO] = fs->vqs[VQ_HIPRIO].name;
707a62a8ef9SStefan Hajnoczi 
708a62a8ef9SStefan Hajnoczi 	/* Initialize the requests virtqueues */
709a62a8ef9SStefan Hajnoczi 	for (i = VQ_REQUEST; i < fs->nvqs; i++) {
710b43b7e81SVivek Goyal 		char vq_name[VQ_NAME_LEN];
711b43b7e81SVivek Goyal 
712b43b7e81SVivek Goyal 		snprintf(vq_name, VQ_NAME_LEN, "requests.%u", i - VQ_REQUEST);
713b43b7e81SVivek Goyal 		virtio_fs_init_vq(&fs->vqs[i], vq_name, VQ_REQUEST);
714a62a8ef9SStefan Hajnoczi 		callbacks[i] = virtio_fs_vq_done;
715a62a8ef9SStefan Hajnoczi 		names[i] = fs->vqs[i].name;
716a62a8ef9SStefan Hajnoczi 	}
717a62a8ef9SStefan Hajnoczi 
718a62a8ef9SStefan Hajnoczi 	ret = virtio_find_vqs(vdev, fs->nvqs, vqs, callbacks, names, NULL);
719a62a8ef9SStefan Hajnoczi 	if (ret < 0)
720a62a8ef9SStefan Hajnoczi 		goto out;
721a62a8ef9SStefan Hajnoczi 
722a62a8ef9SStefan Hajnoczi 	for (i = 0; i < fs->nvqs; i++)
723a62a8ef9SStefan Hajnoczi 		fs->vqs[i].vq = vqs[i];
724a62a8ef9SStefan Hajnoczi 
725a62a8ef9SStefan Hajnoczi 	virtio_fs_start_all_queues(fs);
726a62a8ef9SStefan Hajnoczi out:
727a62a8ef9SStefan Hajnoczi 	kfree(names);
728a62a8ef9SStefan Hajnoczi 	kfree(callbacks);
729a62a8ef9SStefan Hajnoczi 	kfree(vqs);
730a62a8ef9SStefan Hajnoczi 	if (ret)
731a62a8ef9SStefan Hajnoczi 		kfree(fs->vqs);
732a62a8ef9SStefan Hajnoczi 	return ret;
733a62a8ef9SStefan Hajnoczi }
734a62a8ef9SStefan Hajnoczi 
735a62a8ef9SStefan Hajnoczi /* Free virtqueues (device must already be reset) */
736a62a8ef9SStefan Hajnoczi static void virtio_fs_cleanup_vqs(struct virtio_device *vdev,
737a62a8ef9SStefan Hajnoczi 				  struct virtio_fs *fs)
738a62a8ef9SStefan Hajnoczi {
739a62a8ef9SStefan Hajnoczi 	vdev->config->del_vqs(vdev);
740a62a8ef9SStefan Hajnoczi }
741a62a8ef9SStefan Hajnoczi 
74222f3787eSStefan Hajnoczi /* Map a window offset to a page frame number.  The window offset will have
74322f3787eSStefan Hajnoczi  * been produced by .iomap_begin(), which maps a file offset to a window
74422f3787eSStefan Hajnoczi  * offset.
74522f3787eSStefan Hajnoczi  */
74622f3787eSStefan Hajnoczi static long virtio_fs_direct_access(struct dax_device *dax_dev, pgoff_t pgoff,
74722f3787eSStefan Hajnoczi 				    long nr_pages, void **kaddr, pfn_t *pfn)
74822f3787eSStefan Hajnoczi {
74922f3787eSStefan Hajnoczi 	struct virtio_fs *fs = dax_get_private(dax_dev);
75022f3787eSStefan Hajnoczi 	phys_addr_t offset = PFN_PHYS(pgoff);
75122f3787eSStefan Hajnoczi 	size_t max_nr_pages = fs->window_len/PAGE_SIZE - pgoff;
75222f3787eSStefan Hajnoczi 
75322f3787eSStefan Hajnoczi 	if (kaddr)
75422f3787eSStefan Hajnoczi 		*kaddr = fs->window_kaddr + offset;
75522f3787eSStefan Hajnoczi 	if (pfn)
75622f3787eSStefan Hajnoczi 		*pfn = phys_to_pfn_t(fs->window_phys_addr + offset,
75722f3787eSStefan Hajnoczi 					PFN_DEV | PFN_MAP);
75822f3787eSStefan Hajnoczi 	return nr_pages > max_nr_pages ? max_nr_pages : nr_pages;
75922f3787eSStefan Hajnoczi }
76022f3787eSStefan Hajnoczi 
76122f3787eSStefan Hajnoczi static size_t virtio_fs_copy_from_iter(struct dax_device *dax_dev,
76222f3787eSStefan Hajnoczi 				       pgoff_t pgoff, void *addr,
76322f3787eSStefan Hajnoczi 				       size_t bytes, struct iov_iter *i)
76422f3787eSStefan Hajnoczi {
76522f3787eSStefan Hajnoczi 	return copy_from_iter(addr, bytes, i);
76622f3787eSStefan Hajnoczi }
76722f3787eSStefan Hajnoczi 
76822f3787eSStefan Hajnoczi static size_t virtio_fs_copy_to_iter(struct dax_device *dax_dev,
76922f3787eSStefan Hajnoczi 				       pgoff_t pgoff, void *addr,
77022f3787eSStefan Hajnoczi 				       size_t bytes, struct iov_iter *i)
77122f3787eSStefan Hajnoczi {
77222f3787eSStefan Hajnoczi 	return copy_to_iter(addr, bytes, i);
77322f3787eSStefan Hajnoczi }
77422f3787eSStefan Hajnoczi 
77522f3787eSStefan Hajnoczi static int virtio_fs_zero_page_range(struct dax_device *dax_dev,
77622f3787eSStefan Hajnoczi 				     pgoff_t pgoff, size_t nr_pages)
77722f3787eSStefan Hajnoczi {
77822f3787eSStefan Hajnoczi 	long rc;
77922f3787eSStefan Hajnoczi 	void *kaddr;
78022f3787eSStefan Hajnoczi 
78122f3787eSStefan Hajnoczi 	rc = dax_direct_access(dax_dev, pgoff, nr_pages, &kaddr, NULL);
78222f3787eSStefan Hajnoczi 	if (rc < 0)
78322f3787eSStefan Hajnoczi 		return rc;
78422f3787eSStefan Hajnoczi 	memset(kaddr, 0, nr_pages << PAGE_SHIFT);
78522f3787eSStefan Hajnoczi 	dax_flush(dax_dev, kaddr, nr_pages << PAGE_SHIFT);
78622f3787eSStefan Hajnoczi 	return 0;
78722f3787eSStefan Hajnoczi }
78822f3787eSStefan Hajnoczi 
78922f3787eSStefan Hajnoczi static const struct dax_operations virtio_fs_dax_ops = {
79022f3787eSStefan Hajnoczi 	.direct_access = virtio_fs_direct_access,
79122f3787eSStefan Hajnoczi 	.copy_from_iter = virtio_fs_copy_from_iter,
79222f3787eSStefan Hajnoczi 	.copy_to_iter = virtio_fs_copy_to_iter,
79322f3787eSStefan Hajnoczi 	.zero_page_range = virtio_fs_zero_page_range,
79422f3787eSStefan Hajnoczi };
79522f3787eSStefan Hajnoczi 
79622f3787eSStefan Hajnoczi static void virtio_fs_cleanup_dax(void *data)
79722f3787eSStefan Hajnoczi {
79822f3787eSStefan Hajnoczi 	struct dax_device *dax_dev = data;
79922f3787eSStefan Hajnoczi 
80022f3787eSStefan Hajnoczi 	kill_dax(dax_dev);
80122f3787eSStefan Hajnoczi 	put_dax(dax_dev);
80222f3787eSStefan Hajnoczi }
80322f3787eSStefan Hajnoczi 
80422f3787eSStefan Hajnoczi static int virtio_fs_setup_dax(struct virtio_device *vdev, struct virtio_fs *fs)
80522f3787eSStefan Hajnoczi {
80622f3787eSStefan Hajnoczi 	struct virtio_shm_region cache_reg;
80722f3787eSStefan Hajnoczi 	struct dev_pagemap *pgmap;
80822f3787eSStefan Hajnoczi 	bool have_cache;
80922f3787eSStefan Hajnoczi 
81022f3787eSStefan Hajnoczi 	if (!IS_ENABLED(CONFIG_FUSE_DAX))
81122f3787eSStefan Hajnoczi 		return 0;
81222f3787eSStefan Hajnoczi 
81322f3787eSStefan Hajnoczi 	/* Get cache region */
81422f3787eSStefan Hajnoczi 	have_cache = virtio_get_shm_region(vdev, &cache_reg,
81522f3787eSStefan Hajnoczi 					   (u8)VIRTIO_FS_SHMCAP_ID_CACHE);
81622f3787eSStefan Hajnoczi 	if (!have_cache) {
81722f3787eSStefan Hajnoczi 		dev_notice(&vdev->dev, "%s: No cache capability\n", __func__);
81822f3787eSStefan Hajnoczi 		return 0;
81922f3787eSStefan Hajnoczi 	}
82022f3787eSStefan Hajnoczi 
82122f3787eSStefan Hajnoczi 	if (!devm_request_mem_region(&vdev->dev, cache_reg.addr, cache_reg.len,
82222f3787eSStefan Hajnoczi 				     dev_name(&vdev->dev))) {
82322f3787eSStefan Hajnoczi 		dev_warn(&vdev->dev, "could not reserve region addr=0x%llx len=0x%llx\n",
82422f3787eSStefan Hajnoczi 			 cache_reg.addr, cache_reg.len);
82522f3787eSStefan Hajnoczi 		return -EBUSY;
82622f3787eSStefan Hajnoczi 	}
82722f3787eSStefan Hajnoczi 
82822f3787eSStefan Hajnoczi 	dev_notice(&vdev->dev, "Cache len: 0x%llx @ 0x%llx\n", cache_reg.len,
82922f3787eSStefan Hajnoczi 		   cache_reg.addr);
83022f3787eSStefan Hajnoczi 
83122f3787eSStefan Hajnoczi 	pgmap = devm_kzalloc(&vdev->dev, sizeof(*pgmap), GFP_KERNEL);
83222f3787eSStefan Hajnoczi 	if (!pgmap)
83322f3787eSStefan Hajnoczi 		return -ENOMEM;
83422f3787eSStefan Hajnoczi 
83522f3787eSStefan Hajnoczi 	pgmap->type = MEMORY_DEVICE_FS_DAX;
83622f3787eSStefan Hajnoczi 
83722f3787eSStefan Hajnoczi 	/* Ideally we would directly use the PCI BAR resource but
83822f3787eSStefan Hajnoczi 	 * devm_memremap_pages() wants its own copy in pgmap.  So
83922f3787eSStefan Hajnoczi 	 * initialize a struct resource from scratch (only the start
84022f3787eSStefan Hajnoczi 	 * and end fields will be used).
84122f3787eSStefan Hajnoczi 	 */
84269456535SLinus Torvalds 	pgmap->range = (struct range) {
84322f3787eSStefan Hajnoczi 		.start = (phys_addr_t) cache_reg.addr,
84422f3787eSStefan Hajnoczi 		.end = (phys_addr_t) cache_reg.addr + cache_reg.len - 1,
84522f3787eSStefan Hajnoczi 	};
84669456535SLinus Torvalds 	pgmap->nr_range = 1;
84722f3787eSStefan Hajnoczi 
84822f3787eSStefan Hajnoczi 	fs->window_kaddr = devm_memremap_pages(&vdev->dev, pgmap);
84922f3787eSStefan Hajnoczi 	if (IS_ERR(fs->window_kaddr))
85022f3787eSStefan Hajnoczi 		return PTR_ERR(fs->window_kaddr);
85122f3787eSStefan Hajnoczi 
85222f3787eSStefan Hajnoczi 	fs->window_phys_addr = (phys_addr_t) cache_reg.addr;
85322f3787eSStefan Hajnoczi 	fs->window_len = (phys_addr_t) cache_reg.len;
85422f3787eSStefan Hajnoczi 
85522f3787eSStefan Hajnoczi 	dev_dbg(&vdev->dev, "%s: window kaddr 0x%px phys_addr 0x%llx len 0x%llx\n",
85622f3787eSStefan Hajnoczi 		__func__, fs->window_kaddr, cache_reg.addr, cache_reg.len);
85722f3787eSStefan Hajnoczi 
85822f3787eSStefan Hajnoczi 	fs->dax_dev = alloc_dax(fs, NULL, &virtio_fs_dax_ops, 0);
85922f3787eSStefan Hajnoczi 	if (IS_ERR(fs->dax_dev))
86022f3787eSStefan Hajnoczi 		return PTR_ERR(fs->dax_dev);
86122f3787eSStefan Hajnoczi 
86222f3787eSStefan Hajnoczi 	return devm_add_action_or_reset(&vdev->dev, virtio_fs_cleanup_dax,
86322f3787eSStefan Hajnoczi 					fs->dax_dev);
86422f3787eSStefan Hajnoczi }
86522f3787eSStefan Hajnoczi 
866a62a8ef9SStefan Hajnoczi static int virtio_fs_probe(struct virtio_device *vdev)
867a62a8ef9SStefan Hajnoczi {
868a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs;
869a62a8ef9SStefan Hajnoczi 	int ret;
870a62a8ef9SStefan Hajnoczi 
871a62a8ef9SStefan Hajnoczi 	fs = kzalloc(sizeof(*fs), GFP_KERNEL);
872a62a8ef9SStefan Hajnoczi 	if (!fs)
873a62a8ef9SStefan Hajnoczi 		return -ENOMEM;
874a62a8ef9SStefan Hajnoczi 	kref_init(&fs->refcount);
875a62a8ef9SStefan Hajnoczi 	vdev->priv = fs;
876a62a8ef9SStefan Hajnoczi 
877a62a8ef9SStefan Hajnoczi 	ret = virtio_fs_read_tag(vdev, fs);
878a62a8ef9SStefan Hajnoczi 	if (ret < 0)
879a62a8ef9SStefan Hajnoczi 		goto out;
880a62a8ef9SStefan Hajnoczi 
881a62a8ef9SStefan Hajnoczi 	ret = virtio_fs_setup_vqs(vdev, fs);
882a62a8ef9SStefan Hajnoczi 	if (ret < 0)
883a62a8ef9SStefan Hajnoczi 		goto out;
884a62a8ef9SStefan Hajnoczi 
885a62a8ef9SStefan Hajnoczi 	/* TODO vq affinity */
886a62a8ef9SStefan Hajnoczi 
88722f3787eSStefan Hajnoczi 	ret = virtio_fs_setup_dax(vdev, fs);
88822f3787eSStefan Hajnoczi 	if (ret < 0)
88922f3787eSStefan Hajnoczi 		goto out_vqs;
89022f3787eSStefan Hajnoczi 
891a62a8ef9SStefan Hajnoczi 	/* Bring the device online in case the filesystem is mounted and
892a62a8ef9SStefan Hajnoczi 	 * requests need to be sent before we return.
893a62a8ef9SStefan Hajnoczi 	 */
894a62a8ef9SStefan Hajnoczi 	virtio_device_ready(vdev);
895a62a8ef9SStefan Hajnoczi 
896a62a8ef9SStefan Hajnoczi 	ret = virtio_fs_add_instance(fs);
897a62a8ef9SStefan Hajnoczi 	if (ret < 0)
898a62a8ef9SStefan Hajnoczi 		goto out_vqs;
899a62a8ef9SStefan Hajnoczi 
900a62a8ef9SStefan Hajnoczi 	return 0;
901a62a8ef9SStefan Hajnoczi 
902a62a8ef9SStefan Hajnoczi out_vqs:
903a62a8ef9SStefan Hajnoczi 	vdev->config->reset(vdev);
904a62a8ef9SStefan Hajnoczi 	virtio_fs_cleanup_vqs(vdev, fs);
905c79c5e01SLuis Henriques 	kfree(fs->vqs);
906a62a8ef9SStefan Hajnoczi 
907a62a8ef9SStefan Hajnoczi out:
908a62a8ef9SStefan Hajnoczi 	vdev->priv = NULL;
909a62a8ef9SStefan Hajnoczi 	kfree(fs);
910a62a8ef9SStefan Hajnoczi 	return ret;
911a62a8ef9SStefan Hajnoczi }
912a62a8ef9SStefan Hajnoczi 
913a62a8ef9SStefan Hajnoczi static void virtio_fs_stop_all_queues(struct virtio_fs *fs)
914a62a8ef9SStefan Hajnoczi {
915a62a8ef9SStefan Hajnoczi 	struct virtio_fs_vq *fsvq;
916a62a8ef9SStefan Hajnoczi 	int i;
917a62a8ef9SStefan Hajnoczi 
918a62a8ef9SStefan Hajnoczi 	for (i = 0; i < fs->nvqs; i++) {
919a62a8ef9SStefan Hajnoczi 		fsvq = &fs->vqs[i];
920a62a8ef9SStefan Hajnoczi 		spin_lock(&fsvq->lock);
921a62a8ef9SStefan Hajnoczi 		fsvq->connected = false;
922a62a8ef9SStefan Hajnoczi 		spin_unlock(&fsvq->lock);
923a62a8ef9SStefan Hajnoczi 	}
924a62a8ef9SStefan Hajnoczi }
925a62a8ef9SStefan Hajnoczi 
926a62a8ef9SStefan Hajnoczi static void virtio_fs_remove(struct virtio_device *vdev)
927a62a8ef9SStefan Hajnoczi {
928a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs = vdev->priv;
929a62a8ef9SStefan Hajnoczi 
930a62a8ef9SStefan Hajnoczi 	mutex_lock(&virtio_fs_mutex);
931a62a8ef9SStefan Hajnoczi 	/* This device is going away. No one should get new reference */
932a62a8ef9SStefan Hajnoczi 	list_del_init(&fs->list);
933a62a8ef9SStefan Hajnoczi 	virtio_fs_stop_all_queues(fs);
934724c15a4SVivek Goyal 	virtio_fs_drain_all_queues_locked(fs);
935a62a8ef9SStefan Hajnoczi 	vdev->config->reset(vdev);
936a62a8ef9SStefan Hajnoczi 	virtio_fs_cleanup_vqs(vdev, fs);
937a62a8ef9SStefan Hajnoczi 
938a62a8ef9SStefan Hajnoczi 	vdev->priv = NULL;
939a62a8ef9SStefan Hajnoczi 	/* Put device reference on virtio_fs object */
940a62a8ef9SStefan Hajnoczi 	virtio_fs_put(fs);
941a62a8ef9SStefan Hajnoczi 	mutex_unlock(&virtio_fs_mutex);
942a62a8ef9SStefan Hajnoczi }
943a62a8ef9SStefan Hajnoczi 
944a62a8ef9SStefan Hajnoczi #ifdef CONFIG_PM_SLEEP
945a62a8ef9SStefan Hajnoczi static int virtio_fs_freeze(struct virtio_device *vdev)
946a62a8ef9SStefan Hajnoczi {
947a62a8ef9SStefan Hajnoczi 	/* TODO need to save state here */
948a62a8ef9SStefan Hajnoczi 	pr_warn("virtio-fs: suspend/resume not yet supported\n");
949a62a8ef9SStefan Hajnoczi 	return -EOPNOTSUPP;
950a62a8ef9SStefan Hajnoczi }
951a62a8ef9SStefan Hajnoczi 
952a62a8ef9SStefan Hajnoczi static int virtio_fs_restore(struct virtio_device *vdev)
953a62a8ef9SStefan Hajnoczi {
954a62a8ef9SStefan Hajnoczi 	 /* TODO need to restore state here */
955a62a8ef9SStefan Hajnoczi 	return 0;
956a62a8ef9SStefan Hajnoczi }
957a62a8ef9SStefan Hajnoczi #endif /* CONFIG_PM_SLEEP */
958a62a8ef9SStefan Hajnoczi 
95900929447SYueHaibing static const struct virtio_device_id id_table[] = {
960a62a8ef9SStefan Hajnoczi 	{ VIRTIO_ID_FS, VIRTIO_DEV_ANY_ID },
961a62a8ef9SStefan Hajnoczi 	{},
962a62a8ef9SStefan Hajnoczi };
963a62a8ef9SStefan Hajnoczi 
96400929447SYueHaibing static const unsigned int feature_table[] = {};
965a62a8ef9SStefan Hajnoczi 
966a62a8ef9SStefan Hajnoczi static struct virtio_driver virtio_fs_driver = {
967a62a8ef9SStefan Hajnoczi 	.driver.name		= KBUILD_MODNAME,
968a62a8ef9SStefan Hajnoczi 	.driver.owner		= THIS_MODULE,
969a62a8ef9SStefan Hajnoczi 	.id_table		= id_table,
970a62a8ef9SStefan Hajnoczi 	.feature_table		= feature_table,
971a62a8ef9SStefan Hajnoczi 	.feature_table_size	= ARRAY_SIZE(feature_table),
972a62a8ef9SStefan Hajnoczi 	.probe			= virtio_fs_probe,
973a62a8ef9SStefan Hajnoczi 	.remove			= virtio_fs_remove,
974a62a8ef9SStefan Hajnoczi #ifdef CONFIG_PM_SLEEP
975a62a8ef9SStefan Hajnoczi 	.freeze			= virtio_fs_freeze,
976a62a8ef9SStefan Hajnoczi 	.restore		= virtio_fs_restore,
977a62a8ef9SStefan Hajnoczi #endif
978a62a8ef9SStefan Hajnoczi };
979a62a8ef9SStefan Hajnoczi 
980a62a8ef9SStefan Hajnoczi static void virtio_fs_wake_forget_and_unlock(struct fuse_iqueue *fiq)
981a62a8ef9SStefan Hajnoczi __releases(fiq->lock)
982a62a8ef9SStefan Hajnoczi {
983a62a8ef9SStefan Hajnoczi 	struct fuse_forget_link *link;
984a62a8ef9SStefan Hajnoczi 	struct virtio_fs_forget *forget;
9851efcf39eSVivek Goyal 	struct virtio_fs_forget_req *req;
986a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs;
987a62a8ef9SStefan Hajnoczi 	struct virtio_fs_vq *fsvq;
988a62a8ef9SStefan Hajnoczi 	u64 unique;
989a62a8ef9SStefan Hajnoczi 
990a62a8ef9SStefan Hajnoczi 	link = fuse_dequeue_forget(fiq, 1, NULL);
991a62a8ef9SStefan Hajnoczi 	unique = fuse_get_unique(fiq);
992a62a8ef9SStefan Hajnoczi 
993a62a8ef9SStefan Hajnoczi 	fs = fiq->priv;
994a62a8ef9SStefan Hajnoczi 	fsvq = &fs->vqs[VQ_HIPRIO];
995a62a8ef9SStefan Hajnoczi 	spin_unlock(&fiq->lock);
996a62a8ef9SStefan Hajnoczi 
997a62a8ef9SStefan Hajnoczi 	/* Allocate a buffer for the request */
998a62a8ef9SStefan Hajnoczi 	forget = kmalloc(sizeof(*forget), GFP_NOFS | __GFP_NOFAIL);
9991efcf39eSVivek Goyal 	req = &forget->req;
1000a62a8ef9SStefan Hajnoczi 
10011efcf39eSVivek Goyal 	req->ih = (struct fuse_in_header){
1002a62a8ef9SStefan Hajnoczi 		.opcode = FUSE_FORGET,
1003a62a8ef9SStefan Hajnoczi 		.nodeid = link->forget_one.nodeid,
1004a62a8ef9SStefan Hajnoczi 		.unique = unique,
10051efcf39eSVivek Goyal 		.len = sizeof(*req),
1006a62a8ef9SStefan Hajnoczi 	};
10071efcf39eSVivek Goyal 	req->arg = (struct fuse_forget_in){
1008a62a8ef9SStefan Hajnoczi 		.nlookup = link->forget_one.nlookup,
1009a62a8ef9SStefan Hajnoczi 	};
1010a62a8ef9SStefan Hajnoczi 
101158ada94fSVivek Goyal 	send_forget_request(fsvq, forget, false);
1012a62a8ef9SStefan Hajnoczi 	kfree(link);
1013a62a8ef9SStefan Hajnoczi }
1014a62a8ef9SStefan Hajnoczi 
1015a62a8ef9SStefan Hajnoczi static void virtio_fs_wake_interrupt_and_unlock(struct fuse_iqueue *fiq)
1016a62a8ef9SStefan Hajnoczi __releases(fiq->lock)
1017a62a8ef9SStefan Hajnoczi {
1018a62a8ef9SStefan Hajnoczi 	/*
1019a62a8ef9SStefan Hajnoczi 	 * TODO interrupts.
1020a62a8ef9SStefan Hajnoczi 	 *
1021a62a8ef9SStefan Hajnoczi 	 * Normal fs operations on a local filesystems aren't interruptible.
1022a62a8ef9SStefan Hajnoczi 	 * Exceptions are blocking lock operations; for example fcntl(F_SETLKW)
1023a62a8ef9SStefan Hajnoczi 	 * with shared lock between host and guest.
1024a62a8ef9SStefan Hajnoczi 	 */
1025a62a8ef9SStefan Hajnoczi 	spin_unlock(&fiq->lock);
1026a62a8ef9SStefan Hajnoczi }
1027a62a8ef9SStefan Hajnoczi 
102842d3e2d0SVivek Goyal /* Count number of scatter-gather elements required */
102942d3e2d0SVivek Goyal static unsigned int sg_count_fuse_pages(struct fuse_page_desc *page_descs,
103042d3e2d0SVivek Goyal 				       unsigned int num_pages,
103142d3e2d0SVivek Goyal 				       unsigned int total_len)
103242d3e2d0SVivek Goyal {
103342d3e2d0SVivek Goyal 	unsigned int i;
103442d3e2d0SVivek Goyal 	unsigned int this_len;
103542d3e2d0SVivek Goyal 
103642d3e2d0SVivek Goyal 	for (i = 0; i < num_pages && total_len; i++) {
103742d3e2d0SVivek Goyal 		this_len =  min(page_descs[i].length, total_len);
103842d3e2d0SVivek Goyal 		total_len -= this_len;
103942d3e2d0SVivek Goyal 	}
104042d3e2d0SVivek Goyal 
104142d3e2d0SVivek Goyal 	return i;
104242d3e2d0SVivek Goyal }
104342d3e2d0SVivek Goyal 
1044a62a8ef9SStefan Hajnoczi /* Return the number of scatter-gather list elements required */
1045a62a8ef9SStefan Hajnoczi static unsigned int sg_count_fuse_req(struct fuse_req *req)
1046a62a8ef9SStefan Hajnoczi {
1047a62a8ef9SStefan Hajnoczi 	struct fuse_args *args = req->args;
1048a62a8ef9SStefan Hajnoczi 	struct fuse_args_pages *ap = container_of(args, typeof(*ap), args);
104942d3e2d0SVivek Goyal 	unsigned int size, total_sgs = 1 /* fuse_in_header */;
1050a62a8ef9SStefan Hajnoczi 
1051a62a8ef9SStefan Hajnoczi 	if (args->in_numargs - args->in_pages)
1052a62a8ef9SStefan Hajnoczi 		total_sgs += 1;
1053a62a8ef9SStefan Hajnoczi 
105442d3e2d0SVivek Goyal 	if (args->in_pages) {
105542d3e2d0SVivek Goyal 		size = args->in_args[args->in_numargs - 1].size;
105642d3e2d0SVivek Goyal 		total_sgs += sg_count_fuse_pages(ap->descs, ap->num_pages,
105742d3e2d0SVivek Goyal 						 size);
105842d3e2d0SVivek Goyal 	}
1059a62a8ef9SStefan Hajnoczi 
1060a62a8ef9SStefan Hajnoczi 	if (!test_bit(FR_ISREPLY, &req->flags))
1061a62a8ef9SStefan Hajnoczi 		return total_sgs;
1062a62a8ef9SStefan Hajnoczi 
1063a62a8ef9SStefan Hajnoczi 	total_sgs += 1 /* fuse_out_header */;
1064a62a8ef9SStefan Hajnoczi 
1065a62a8ef9SStefan Hajnoczi 	if (args->out_numargs - args->out_pages)
1066a62a8ef9SStefan Hajnoczi 		total_sgs += 1;
1067a62a8ef9SStefan Hajnoczi 
106842d3e2d0SVivek Goyal 	if (args->out_pages) {
106942d3e2d0SVivek Goyal 		size = args->out_args[args->out_numargs - 1].size;
107042d3e2d0SVivek Goyal 		total_sgs += sg_count_fuse_pages(ap->descs, ap->num_pages,
107142d3e2d0SVivek Goyal 						 size);
107242d3e2d0SVivek Goyal 	}
1073a62a8ef9SStefan Hajnoczi 
1074a62a8ef9SStefan Hajnoczi 	return total_sgs;
1075a62a8ef9SStefan Hajnoczi }
1076a62a8ef9SStefan Hajnoczi 
1077a62a8ef9SStefan Hajnoczi /* Add pages to scatter-gather list and return number of elements used */
1078a62a8ef9SStefan Hajnoczi static unsigned int sg_init_fuse_pages(struct scatterlist *sg,
1079a62a8ef9SStefan Hajnoczi 				       struct page **pages,
1080a62a8ef9SStefan Hajnoczi 				       struct fuse_page_desc *page_descs,
1081a62a8ef9SStefan Hajnoczi 				       unsigned int num_pages,
1082a62a8ef9SStefan Hajnoczi 				       unsigned int total_len)
1083a62a8ef9SStefan Hajnoczi {
1084a62a8ef9SStefan Hajnoczi 	unsigned int i;
1085a62a8ef9SStefan Hajnoczi 	unsigned int this_len;
1086a62a8ef9SStefan Hajnoczi 
1087a62a8ef9SStefan Hajnoczi 	for (i = 0; i < num_pages && total_len; i++) {
1088a62a8ef9SStefan Hajnoczi 		sg_init_table(&sg[i], 1);
1089a62a8ef9SStefan Hajnoczi 		this_len =  min(page_descs[i].length, total_len);
1090a62a8ef9SStefan Hajnoczi 		sg_set_page(&sg[i], pages[i], this_len, page_descs[i].offset);
1091a62a8ef9SStefan Hajnoczi 		total_len -= this_len;
1092a62a8ef9SStefan Hajnoczi 	}
1093a62a8ef9SStefan Hajnoczi 
1094a62a8ef9SStefan Hajnoczi 	return i;
1095a62a8ef9SStefan Hajnoczi }
1096a62a8ef9SStefan Hajnoczi 
1097a62a8ef9SStefan Hajnoczi /* Add args to scatter-gather list and return number of elements used */
1098a62a8ef9SStefan Hajnoczi static unsigned int sg_init_fuse_args(struct scatterlist *sg,
1099a62a8ef9SStefan Hajnoczi 				      struct fuse_req *req,
1100a62a8ef9SStefan Hajnoczi 				      struct fuse_arg *args,
1101a62a8ef9SStefan Hajnoczi 				      unsigned int numargs,
1102a62a8ef9SStefan Hajnoczi 				      bool argpages,
1103a62a8ef9SStefan Hajnoczi 				      void *argbuf,
1104a62a8ef9SStefan Hajnoczi 				      unsigned int *len_used)
1105a62a8ef9SStefan Hajnoczi {
1106a62a8ef9SStefan Hajnoczi 	struct fuse_args_pages *ap = container_of(req->args, typeof(*ap), args);
1107a62a8ef9SStefan Hajnoczi 	unsigned int total_sgs = 0;
1108a62a8ef9SStefan Hajnoczi 	unsigned int len;
1109a62a8ef9SStefan Hajnoczi 
1110a62a8ef9SStefan Hajnoczi 	len = fuse_len_args(numargs - argpages, args);
1111a62a8ef9SStefan Hajnoczi 	if (len)
1112a62a8ef9SStefan Hajnoczi 		sg_init_one(&sg[total_sgs++], argbuf, len);
1113a62a8ef9SStefan Hajnoczi 
1114a62a8ef9SStefan Hajnoczi 	if (argpages)
1115a62a8ef9SStefan Hajnoczi 		total_sgs += sg_init_fuse_pages(&sg[total_sgs],
1116a62a8ef9SStefan Hajnoczi 						ap->pages, ap->descs,
1117a62a8ef9SStefan Hajnoczi 						ap->num_pages,
1118a62a8ef9SStefan Hajnoczi 						args[numargs - 1].size);
1119a62a8ef9SStefan Hajnoczi 
1120a62a8ef9SStefan Hajnoczi 	if (len_used)
1121a62a8ef9SStefan Hajnoczi 		*len_used = len;
1122a62a8ef9SStefan Hajnoczi 
1123a62a8ef9SStefan Hajnoczi 	return total_sgs;
1124a62a8ef9SStefan Hajnoczi }
1125a62a8ef9SStefan Hajnoczi 
1126a62a8ef9SStefan Hajnoczi /* Add a request to a virtqueue and kick the device */
1127a62a8ef9SStefan Hajnoczi static int virtio_fs_enqueue_req(struct virtio_fs_vq *fsvq,
1128a9bfd9ddSVivek Goyal 				 struct fuse_req *req, bool in_flight)
1129a62a8ef9SStefan Hajnoczi {
1130a62a8ef9SStefan Hajnoczi 	/* requests need at least 4 elements */
1131a62a8ef9SStefan Hajnoczi 	struct scatterlist *stack_sgs[6];
1132a62a8ef9SStefan Hajnoczi 	struct scatterlist stack_sg[ARRAY_SIZE(stack_sgs)];
1133a62a8ef9SStefan Hajnoczi 	struct scatterlist **sgs = stack_sgs;
1134a62a8ef9SStefan Hajnoczi 	struct scatterlist *sg = stack_sg;
1135a62a8ef9SStefan Hajnoczi 	struct virtqueue *vq;
1136a62a8ef9SStefan Hajnoczi 	struct fuse_args *args = req->args;
1137a62a8ef9SStefan Hajnoczi 	unsigned int argbuf_used = 0;
1138a62a8ef9SStefan Hajnoczi 	unsigned int out_sgs = 0;
1139a62a8ef9SStefan Hajnoczi 	unsigned int in_sgs = 0;
1140a62a8ef9SStefan Hajnoczi 	unsigned int total_sgs;
1141a62a8ef9SStefan Hajnoczi 	unsigned int i;
1142a62a8ef9SStefan Hajnoczi 	int ret;
1143a62a8ef9SStefan Hajnoczi 	bool notify;
11445dbe190fSVivek Goyal 	struct fuse_pqueue *fpq;
1145a62a8ef9SStefan Hajnoczi 
1146a62a8ef9SStefan Hajnoczi 	/* Does the sglist fit on the stack? */
1147a62a8ef9SStefan Hajnoczi 	total_sgs = sg_count_fuse_req(req);
1148a62a8ef9SStefan Hajnoczi 	if (total_sgs > ARRAY_SIZE(stack_sgs)) {
1149a62a8ef9SStefan Hajnoczi 		sgs = kmalloc_array(total_sgs, sizeof(sgs[0]), GFP_ATOMIC);
1150a62a8ef9SStefan Hajnoczi 		sg = kmalloc_array(total_sgs, sizeof(sg[0]), GFP_ATOMIC);
1151a62a8ef9SStefan Hajnoczi 		if (!sgs || !sg) {
1152a62a8ef9SStefan Hajnoczi 			ret = -ENOMEM;
1153a62a8ef9SStefan Hajnoczi 			goto out;
1154a62a8ef9SStefan Hajnoczi 		}
1155a62a8ef9SStefan Hajnoczi 	}
1156a62a8ef9SStefan Hajnoczi 
1157a62a8ef9SStefan Hajnoczi 	/* Use a bounce buffer since stack args cannot be mapped */
1158a62a8ef9SStefan Hajnoczi 	ret = copy_args_to_argbuf(req);
1159a62a8ef9SStefan Hajnoczi 	if (ret < 0)
1160a62a8ef9SStefan Hajnoczi 		goto out;
1161a62a8ef9SStefan Hajnoczi 
1162a62a8ef9SStefan Hajnoczi 	/* Request elements */
1163a62a8ef9SStefan Hajnoczi 	sg_init_one(&sg[out_sgs++], &req->in.h, sizeof(req->in.h));
1164a62a8ef9SStefan Hajnoczi 	out_sgs += sg_init_fuse_args(&sg[out_sgs], req,
1165a62a8ef9SStefan Hajnoczi 				     (struct fuse_arg *)args->in_args,
1166a62a8ef9SStefan Hajnoczi 				     args->in_numargs, args->in_pages,
1167a62a8ef9SStefan Hajnoczi 				     req->argbuf, &argbuf_used);
1168a62a8ef9SStefan Hajnoczi 
1169a62a8ef9SStefan Hajnoczi 	/* Reply elements */
1170a62a8ef9SStefan Hajnoczi 	if (test_bit(FR_ISREPLY, &req->flags)) {
1171a62a8ef9SStefan Hajnoczi 		sg_init_one(&sg[out_sgs + in_sgs++],
1172a62a8ef9SStefan Hajnoczi 			    &req->out.h, sizeof(req->out.h));
1173a62a8ef9SStefan Hajnoczi 		in_sgs += sg_init_fuse_args(&sg[out_sgs + in_sgs], req,
1174a62a8ef9SStefan Hajnoczi 					    args->out_args, args->out_numargs,
1175a62a8ef9SStefan Hajnoczi 					    args->out_pages,
1176a62a8ef9SStefan Hajnoczi 					    req->argbuf + argbuf_used, NULL);
1177a62a8ef9SStefan Hajnoczi 	}
1178a62a8ef9SStefan Hajnoczi 
1179a62a8ef9SStefan Hajnoczi 	WARN_ON(out_sgs + in_sgs != total_sgs);
1180a62a8ef9SStefan Hajnoczi 
1181a62a8ef9SStefan Hajnoczi 	for (i = 0; i < total_sgs; i++)
1182a62a8ef9SStefan Hajnoczi 		sgs[i] = &sg[i];
1183a62a8ef9SStefan Hajnoczi 
1184a62a8ef9SStefan Hajnoczi 	spin_lock(&fsvq->lock);
1185a62a8ef9SStefan Hajnoczi 
1186a62a8ef9SStefan Hajnoczi 	if (!fsvq->connected) {
1187a62a8ef9SStefan Hajnoczi 		spin_unlock(&fsvq->lock);
1188a62a8ef9SStefan Hajnoczi 		ret = -ENOTCONN;
1189a62a8ef9SStefan Hajnoczi 		goto out;
1190a62a8ef9SStefan Hajnoczi 	}
1191a62a8ef9SStefan Hajnoczi 
1192a62a8ef9SStefan Hajnoczi 	vq = fsvq->vq;
1193a62a8ef9SStefan Hajnoczi 	ret = virtqueue_add_sgs(vq, sgs, out_sgs, in_sgs, req, GFP_ATOMIC);
1194a62a8ef9SStefan Hajnoczi 	if (ret < 0) {
1195a62a8ef9SStefan Hajnoczi 		spin_unlock(&fsvq->lock);
1196a62a8ef9SStefan Hajnoczi 		goto out;
1197a62a8ef9SStefan Hajnoczi 	}
1198a62a8ef9SStefan Hajnoczi 
11995dbe190fSVivek Goyal 	/* Request successfully sent. */
12005dbe190fSVivek Goyal 	fpq = &fsvq->fud->pq;
12015dbe190fSVivek Goyal 	spin_lock(&fpq->lock);
12025dbe190fSVivek Goyal 	list_add_tail(&req->list, fpq->processing);
12035dbe190fSVivek Goyal 	spin_unlock(&fpq->lock);
12045dbe190fSVivek Goyal 	set_bit(FR_SENT, &req->flags);
12055dbe190fSVivek Goyal 	/* matches barrier in request_wait_answer() */
12065dbe190fSVivek Goyal 	smp_mb__after_atomic();
12075dbe190fSVivek Goyal 
1208a9bfd9ddSVivek Goyal 	if (!in_flight)
1209c17ea009SVivek Goyal 		inc_in_flight_req(fsvq);
1210a62a8ef9SStefan Hajnoczi 	notify = virtqueue_kick_prepare(vq);
1211a62a8ef9SStefan Hajnoczi 
1212a62a8ef9SStefan Hajnoczi 	spin_unlock(&fsvq->lock);
1213a62a8ef9SStefan Hajnoczi 
1214a62a8ef9SStefan Hajnoczi 	if (notify)
1215a62a8ef9SStefan Hajnoczi 		virtqueue_notify(vq);
1216a62a8ef9SStefan Hajnoczi 
1217a62a8ef9SStefan Hajnoczi out:
1218a62a8ef9SStefan Hajnoczi 	if (ret < 0 && req->argbuf) {
1219a62a8ef9SStefan Hajnoczi 		kfree(req->argbuf);
1220a62a8ef9SStefan Hajnoczi 		req->argbuf = NULL;
1221a62a8ef9SStefan Hajnoczi 	}
1222a62a8ef9SStefan Hajnoczi 	if (sgs != stack_sgs) {
1223a62a8ef9SStefan Hajnoczi 		kfree(sgs);
1224a62a8ef9SStefan Hajnoczi 		kfree(sg);
1225a62a8ef9SStefan Hajnoczi 	}
1226a62a8ef9SStefan Hajnoczi 
1227a62a8ef9SStefan Hajnoczi 	return ret;
1228a62a8ef9SStefan Hajnoczi }
1229a62a8ef9SStefan Hajnoczi 
1230a62a8ef9SStefan Hajnoczi static void virtio_fs_wake_pending_and_unlock(struct fuse_iqueue *fiq)
1231a62a8ef9SStefan Hajnoczi __releases(fiq->lock)
1232a62a8ef9SStefan Hajnoczi {
1233a62a8ef9SStefan Hajnoczi 	unsigned int queue_id = VQ_REQUEST; /* TODO multiqueue */
1234a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs;
1235a62a8ef9SStefan Hajnoczi 	struct fuse_req *req;
123651fecdd2SVivek Goyal 	struct virtio_fs_vq *fsvq;
1237a62a8ef9SStefan Hajnoczi 	int ret;
1238a62a8ef9SStefan Hajnoczi 
1239a62a8ef9SStefan Hajnoczi 	WARN_ON(list_empty(&fiq->pending));
1240a62a8ef9SStefan Hajnoczi 	req = list_last_entry(&fiq->pending, struct fuse_req, list);
1241a62a8ef9SStefan Hajnoczi 	clear_bit(FR_PENDING, &req->flags);
1242a62a8ef9SStefan Hajnoczi 	list_del_init(&req->list);
1243a62a8ef9SStefan Hajnoczi 	WARN_ON(!list_empty(&fiq->pending));
1244a62a8ef9SStefan Hajnoczi 	spin_unlock(&fiq->lock);
1245a62a8ef9SStefan Hajnoczi 
1246a62a8ef9SStefan Hajnoczi 	fs = fiq->priv;
1247a62a8ef9SStefan Hajnoczi 
1248a62a8ef9SStefan Hajnoczi 	pr_debug("%s: opcode %u unique %#llx nodeid %#llx in.len %u out.len %u\n",
1249a62a8ef9SStefan Hajnoczi 		  __func__, req->in.h.opcode, req->in.h.unique,
1250a62a8ef9SStefan Hajnoczi 		 req->in.h.nodeid, req->in.h.len,
1251a62a8ef9SStefan Hajnoczi 		 fuse_len_args(req->args->out_numargs, req->args->out_args));
1252a62a8ef9SStefan Hajnoczi 
125351fecdd2SVivek Goyal 	fsvq = &fs->vqs[queue_id];
1254a9bfd9ddSVivek Goyal 	ret = virtio_fs_enqueue_req(fsvq, req, false);
1255a62a8ef9SStefan Hajnoczi 	if (ret < 0) {
1256a62a8ef9SStefan Hajnoczi 		if (ret == -ENOMEM || ret == -ENOSPC) {
1257a9bfd9ddSVivek Goyal 			/*
1258a9bfd9ddSVivek Goyal 			 * Virtqueue full. Retry submission from worker
1259a9bfd9ddSVivek Goyal 			 * context as we might be holding fc->bg_lock.
1260a9bfd9ddSVivek Goyal 			 */
1261a9bfd9ddSVivek Goyal 			spin_lock(&fsvq->lock);
1262a9bfd9ddSVivek Goyal 			list_add_tail(&req->list, &fsvq->queued_reqs);
1263a9bfd9ddSVivek Goyal 			inc_in_flight_req(fsvq);
1264a9bfd9ddSVivek Goyal 			schedule_delayed_work(&fsvq->dispatch_work,
1265a9bfd9ddSVivek Goyal 						msecs_to_jiffies(1));
1266a9bfd9ddSVivek Goyal 			spin_unlock(&fsvq->lock);
1267a9bfd9ddSVivek Goyal 			return;
1268a62a8ef9SStefan Hajnoczi 		}
1269a62a8ef9SStefan Hajnoczi 		req->out.h.error = ret;
1270a62a8ef9SStefan Hajnoczi 		pr_err("virtio-fs: virtio_fs_enqueue_req() failed %d\n", ret);
127151fecdd2SVivek Goyal 
127251fecdd2SVivek Goyal 		/* Can't end request in submission context. Use a worker */
127351fecdd2SVivek Goyal 		spin_lock(&fsvq->lock);
127451fecdd2SVivek Goyal 		list_add_tail(&req->list, &fsvq->end_reqs);
127551fecdd2SVivek Goyal 		schedule_delayed_work(&fsvq->dispatch_work, 0);
127651fecdd2SVivek Goyal 		spin_unlock(&fsvq->lock);
1277a62a8ef9SStefan Hajnoczi 		return;
1278a62a8ef9SStefan Hajnoczi 	}
1279a62a8ef9SStefan Hajnoczi }
1280a62a8ef9SStefan Hajnoczi 
128100929447SYueHaibing static const struct fuse_iqueue_ops virtio_fs_fiq_ops = {
1282a62a8ef9SStefan Hajnoczi 	.wake_forget_and_unlock		= virtio_fs_wake_forget_and_unlock,
1283a62a8ef9SStefan Hajnoczi 	.wake_interrupt_and_unlock	= virtio_fs_wake_interrupt_and_unlock,
1284a62a8ef9SStefan Hajnoczi 	.wake_pending_and_unlock	= virtio_fs_wake_pending_and_unlock,
1285a62a8ef9SStefan Hajnoczi 	.release			= virtio_fs_fiq_release,
1286a62a8ef9SStefan Hajnoczi };
1287a62a8ef9SStefan Hajnoczi 
12881dd53957SVivek Goyal static inline void virtio_fs_ctx_set_defaults(struct fuse_fs_context *ctx)
1289a62a8ef9SStefan Hajnoczi {
12901dd53957SVivek Goyal 	ctx->rootmode = S_IFDIR;
12911dd53957SVivek Goyal 	ctx->default_permissions = 1;
12921dd53957SVivek Goyal 	ctx->allow_other = 1;
12931dd53957SVivek Goyal 	ctx->max_read = UINT_MAX;
12941dd53957SVivek Goyal 	ctx->blksize = 512;
12951dd53957SVivek Goyal 	ctx->destroy = true;
12961dd53957SVivek Goyal 	ctx->no_control = true;
12971dd53957SVivek Goyal 	ctx->no_force_umount = true;
12981dd53957SVivek Goyal }
12991dd53957SVivek Goyal 
13001dd53957SVivek Goyal static int virtio_fs_fill_super(struct super_block *sb, struct fs_context *fsc)
1301a62a8ef9SStefan Hajnoczi {
1302fcee216bSMax Reitz 	struct fuse_mount *fm = get_fuse_mount_super(sb);
1303fcee216bSMax Reitz 	struct fuse_conn *fc = fm->fc;
1304a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs = fc->iq.priv;
13051dd53957SVivek Goyal 	struct fuse_fs_context *ctx = fsc->fs_private;
1306a62a8ef9SStefan Hajnoczi 	unsigned int i;
1307a62a8ef9SStefan Hajnoczi 	int err;
1308a62a8ef9SStefan Hajnoczi 
13091dd53957SVivek Goyal 	virtio_fs_ctx_set_defaults(ctx);
1310a62a8ef9SStefan Hajnoczi 	mutex_lock(&virtio_fs_mutex);
1311a62a8ef9SStefan Hajnoczi 
1312a62a8ef9SStefan Hajnoczi 	/* After holding mutex, make sure virtiofs device is still there.
1313a62a8ef9SStefan Hajnoczi 	 * Though we are holding a reference to it, drive ->remove might
1314a62a8ef9SStefan Hajnoczi 	 * still have cleaned up virtual queues. In that case bail out.
1315a62a8ef9SStefan Hajnoczi 	 */
1316a62a8ef9SStefan Hajnoczi 	err = -EINVAL;
1317a62a8ef9SStefan Hajnoczi 	if (list_empty(&fs->list)) {
1318a62a8ef9SStefan Hajnoczi 		pr_info("virtio-fs: tag <%s> not found\n", fs->tag);
1319a62a8ef9SStefan Hajnoczi 		goto err;
1320a62a8ef9SStefan Hajnoczi 	}
1321a62a8ef9SStefan Hajnoczi 
1322a62a8ef9SStefan Hajnoczi 	err = -ENOMEM;
1323a62a8ef9SStefan Hajnoczi 	/* Allocate fuse_dev for hiprio and notification queues */
13247fd3abfaSVivek Goyal 	for (i = 0; i < fs->nvqs; i++) {
1325a62a8ef9SStefan Hajnoczi 		struct virtio_fs_vq *fsvq = &fs->vqs[i];
1326a62a8ef9SStefan Hajnoczi 
1327a62a8ef9SStefan Hajnoczi 		fsvq->fud = fuse_dev_alloc();
1328a62a8ef9SStefan Hajnoczi 		if (!fsvq->fud)
1329a62a8ef9SStefan Hajnoczi 			goto err_free_fuse_devs;
1330a62a8ef9SStefan Hajnoczi 	}
1331a62a8ef9SStefan Hajnoczi 
13327fd3abfaSVivek Goyal 	/* virtiofs allocates and installs its own fuse devices */
13331dd53957SVivek Goyal 	ctx->fudptr = NULL;
13343f9b9efdSVivek Goyal 	if (ctx->dax) {
13353f9b9efdSVivek Goyal 		if (!fs->dax_dev) {
13363f9b9efdSVivek Goyal 			err = -EINVAL;
13373f9b9efdSVivek Goyal 			pr_err("virtio-fs: dax can't be enabled as filesystem"
13383f9b9efdSVivek Goyal 			       " device does not support it.\n");
13393f9b9efdSVivek Goyal 			goto err_free_fuse_devs;
13403f9b9efdSVivek Goyal 		}
13411dd53957SVivek Goyal 		ctx->dax_dev = fs->dax_dev;
13423f9b9efdSVivek Goyal 	}
13431dd53957SVivek Goyal 	err = fuse_fill_super_common(sb, ctx);
1344a62a8ef9SStefan Hajnoczi 	if (err < 0)
1345a62a8ef9SStefan Hajnoczi 		goto err_free_fuse_devs;
1346a62a8ef9SStefan Hajnoczi 
1347a62a8ef9SStefan Hajnoczi 	for (i = 0; i < fs->nvqs; i++) {
1348a62a8ef9SStefan Hajnoczi 		struct virtio_fs_vq *fsvq = &fs->vqs[i];
1349a62a8ef9SStefan Hajnoczi 
1350a62a8ef9SStefan Hajnoczi 		fuse_dev_install(fsvq->fud, fc);
1351a62a8ef9SStefan Hajnoczi 	}
1352a62a8ef9SStefan Hajnoczi 
1353a62a8ef9SStefan Hajnoczi 	/* Previous unmount will stop all queues. Start these again */
1354a62a8ef9SStefan Hajnoczi 	virtio_fs_start_all_queues(fs);
1355fcee216bSMax Reitz 	fuse_send_init(fm);
1356a62a8ef9SStefan Hajnoczi 	mutex_unlock(&virtio_fs_mutex);
1357a62a8ef9SStefan Hajnoczi 	return 0;
1358a62a8ef9SStefan Hajnoczi 
1359a62a8ef9SStefan Hajnoczi err_free_fuse_devs:
1360a62a8ef9SStefan Hajnoczi 	virtio_fs_free_devs(fs);
1361a62a8ef9SStefan Hajnoczi err:
1362a62a8ef9SStefan Hajnoczi 	mutex_unlock(&virtio_fs_mutex);
1363a62a8ef9SStefan Hajnoczi 	return err;
1364a62a8ef9SStefan Hajnoczi }
1365a62a8ef9SStefan Hajnoczi 
1366fcee216bSMax Reitz static void virtio_fs_conn_destroy(struct fuse_mount *fm)
1367a62a8ef9SStefan Hajnoczi {
1368fcee216bSMax Reitz 	struct fuse_conn *fc = fm->fc;
1369fcee216bSMax Reitz 	struct virtio_fs *vfs = fc->iq.priv;
1370fcee216bSMax Reitz 	struct virtio_fs_vq *fsvq = &vfs->vqs[VQ_HIPRIO];
1371a62a8ef9SStefan Hajnoczi 
1372fcee216bSMax Reitz 	/* Stop dax worker. Soon evict_inodes() will be called which
1373fcee216bSMax Reitz 	 * will free all memory ranges belonging to all inodes.
13749a752d18SVivek Goyal 	 */
13759a752d18SVivek Goyal 	if (IS_ENABLED(CONFIG_FUSE_DAX))
13769a752d18SVivek Goyal 		fuse_dax_cancel_work(fc);
1377a62a8ef9SStefan Hajnoczi 
1378a62a8ef9SStefan Hajnoczi 	/* Stop forget queue. Soon destroy will be sent */
1379a62a8ef9SStefan Hajnoczi 	spin_lock(&fsvq->lock);
1380a62a8ef9SStefan Hajnoczi 	fsvq->connected = false;
1381a62a8ef9SStefan Hajnoczi 	spin_unlock(&fsvq->lock);
1382a62a8ef9SStefan Hajnoczi 	virtio_fs_drain_all_queues(vfs);
1383a62a8ef9SStefan Hajnoczi 
1384fcee216bSMax Reitz 	fuse_conn_destroy(fm);
1385a62a8ef9SStefan Hajnoczi 
1386fcee216bSMax Reitz 	/* fuse_conn_destroy() must have sent destroy. Stop all queues
1387a62a8ef9SStefan Hajnoczi 	 * and drain one more time and free fuse devices. Freeing fuse
1388a62a8ef9SStefan Hajnoczi 	 * devices will drop their reference on fuse_conn and that in
1389a62a8ef9SStefan Hajnoczi 	 * turn will drop its reference on virtio_fs object.
1390a62a8ef9SStefan Hajnoczi 	 */
1391a62a8ef9SStefan Hajnoczi 	virtio_fs_stop_all_queues(vfs);
1392a62a8ef9SStefan Hajnoczi 	virtio_fs_drain_all_queues(vfs);
1393a62a8ef9SStefan Hajnoczi 	virtio_fs_free_devs(vfs);
1394a62a8ef9SStefan Hajnoczi }
1395a62a8ef9SStefan Hajnoczi 
1396fcee216bSMax Reitz static void virtio_kill_sb(struct super_block *sb)
1397fcee216bSMax Reitz {
1398fcee216bSMax Reitz 	struct fuse_mount *fm = get_fuse_mount_super(sb);
1399fcee216bSMax Reitz 	bool last;
1400fcee216bSMax Reitz 
1401fcee216bSMax Reitz 	/* If mount failed, we can still be called without any fc */
1402fcee216bSMax Reitz 	if (fm) {
1403fcee216bSMax Reitz 		last = fuse_mount_remove(fm);
1404fcee216bSMax Reitz 		if (last)
1405fcee216bSMax Reitz 			virtio_fs_conn_destroy(fm);
1406fcee216bSMax Reitz 	}
1407fcee216bSMax Reitz 	kill_anon_super(sb);
1408fcee216bSMax Reitz }
1409fcee216bSMax Reitz 
1410a62a8ef9SStefan Hajnoczi static int virtio_fs_test_super(struct super_block *sb,
1411a62a8ef9SStefan Hajnoczi 				struct fs_context *fsc)
1412a62a8ef9SStefan Hajnoczi {
1413fcee216bSMax Reitz 	struct fuse_mount *fsc_fm = fsc->s_fs_info;
1414fcee216bSMax Reitz 	struct fuse_mount *sb_fm = get_fuse_mount_super(sb);
1415a62a8ef9SStefan Hajnoczi 
1416fcee216bSMax Reitz 	return fsc_fm->fc->iq.priv == sb_fm->fc->iq.priv;
1417a62a8ef9SStefan Hajnoczi }
1418a62a8ef9SStefan Hajnoczi 
1419a62a8ef9SStefan Hajnoczi static int virtio_fs_get_tree(struct fs_context *fsc)
1420a62a8ef9SStefan Hajnoczi {
1421a62a8ef9SStefan Hajnoczi 	struct virtio_fs *fs;
1422a62a8ef9SStefan Hajnoczi 	struct super_block *sb;
1423*a7f0d7aaSConnor Kuehl 	struct fuse_conn *fc = NULL;
1424fcee216bSMax Reitz 	struct fuse_mount *fm;
1425*a7f0d7aaSConnor Kuehl 	unsigned int virtqueue_size;
1426*a7f0d7aaSConnor Kuehl 	int err = -EIO;
1427a62a8ef9SStefan Hajnoczi 
1428a62a8ef9SStefan Hajnoczi 	/* This gets a reference on virtio_fs object. This ptr gets installed
1429a62a8ef9SStefan Hajnoczi 	 * in fc->iq->priv. Once fuse_conn is going away, it calls ->put()
1430a62a8ef9SStefan Hajnoczi 	 * to drop the reference to this object.
1431a62a8ef9SStefan Hajnoczi 	 */
1432a62a8ef9SStefan Hajnoczi 	fs = virtio_fs_find_instance(fsc->source);
1433a62a8ef9SStefan Hajnoczi 	if (!fs) {
1434a62a8ef9SStefan Hajnoczi 		pr_info("virtio-fs: tag <%s> not found\n", fsc->source);
1435a62a8ef9SStefan Hajnoczi 		return -EINVAL;
1436a62a8ef9SStefan Hajnoczi 	}
1437a62a8ef9SStefan Hajnoczi 
1438*a7f0d7aaSConnor Kuehl 	virtqueue_size = virtqueue_get_vring_size(fs->vqs[VQ_REQUEST].vq);
1439*a7f0d7aaSConnor Kuehl 	if (WARN_ON(virtqueue_size <= FUSE_HEADER_OVERHEAD))
1440*a7f0d7aaSConnor Kuehl 		goto out_err;
1441*a7f0d7aaSConnor Kuehl 
1442833c5a42SMiklos Szeredi 	err = -ENOMEM;
1443a62a8ef9SStefan Hajnoczi 	fc = kzalloc(sizeof(struct fuse_conn), GFP_KERNEL);
1444833c5a42SMiklos Szeredi 	if (!fc)
1445833c5a42SMiklos Szeredi 		goto out_err;
1446a62a8ef9SStefan Hajnoczi 
1447fcee216bSMax Reitz 	fm = kzalloc(sizeof(struct fuse_mount), GFP_KERNEL);
1448833c5a42SMiklos Szeredi 	if (!fm)
1449833c5a42SMiklos Szeredi 		goto out_err;
1450fcee216bSMax Reitz 
1451fcee216bSMax Reitz 	fuse_conn_init(fc, fm, get_user_ns(current_user_ns()),
1452fcee216bSMax Reitz 		       &virtio_fs_fiq_ops, fs);
1453a62a8ef9SStefan Hajnoczi 	fc->release = fuse_free_conn;
1454a62a8ef9SStefan Hajnoczi 	fc->delete_stale = true;
1455bf109c64SMax Reitz 	fc->auto_submounts = true;
1456a62a8ef9SStefan Hajnoczi 
1457*a7f0d7aaSConnor Kuehl 	/* Tell FUSE to split requests that exceed the virtqueue's size */
1458*a7f0d7aaSConnor Kuehl 	fc->max_pages_limit = min_t(unsigned int, fc->max_pages_limit,
1459*a7f0d7aaSConnor Kuehl 				    virtqueue_size - FUSE_HEADER_OVERHEAD);
1460*a7f0d7aaSConnor Kuehl 
1461fcee216bSMax Reitz 	fsc->s_fs_info = fm;
1462b19d3d00SMiklos Szeredi 	sb = sget_fc(fsc, virtio_fs_test_super, set_anon_super_fc);
1463514b5e3fSMiklos Szeredi 	if (fsc->s_fs_info) {
1464514b5e3fSMiklos Szeredi 		fuse_conn_put(fc);
1465514b5e3fSMiklos Szeredi 		kfree(fm);
1466514b5e3fSMiklos Szeredi 	}
1467a62a8ef9SStefan Hajnoczi 	if (IS_ERR(sb))
1468a62a8ef9SStefan Hajnoczi 		return PTR_ERR(sb);
1469a62a8ef9SStefan Hajnoczi 
1470a62a8ef9SStefan Hajnoczi 	if (!sb->s_root) {
14711dd53957SVivek Goyal 		err = virtio_fs_fill_super(sb, fsc);
1472a62a8ef9SStefan Hajnoczi 		if (err) {
1473514b5e3fSMiklos Szeredi 			fuse_conn_put(fc);
1474514b5e3fSMiklos Szeredi 			kfree(fm);
147566ab33bfSMiklos Szeredi 			sb->s_fs_info = NULL;
1476a62a8ef9SStefan Hajnoczi 			deactivate_locked_super(sb);
1477a62a8ef9SStefan Hajnoczi 			return err;
1478a62a8ef9SStefan Hajnoczi 		}
1479a62a8ef9SStefan Hajnoczi 
1480a62a8ef9SStefan Hajnoczi 		sb->s_flags |= SB_ACTIVE;
1481a62a8ef9SStefan Hajnoczi 	}
1482a62a8ef9SStefan Hajnoczi 
1483a62a8ef9SStefan Hajnoczi 	WARN_ON(fsc->root);
1484a62a8ef9SStefan Hajnoczi 	fsc->root = dget(sb->s_root);
1485a62a8ef9SStefan Hajnoczi 	return 0;
1486833c5a42SMiklos Szeredi 
1487833c5a42SMiklos Szeredi out_err:
1488833c5a42SMiklos Szeredi 	kfree(fc);
1489833c5a42SMiklos Szeredi 	mutex_lock(&virtio_fs_mutex);
1490833c5a42SMiklos Szeredi 	virtio_fs_put(fs);
1491833c5a42SMiklos Szeredi 	mutex_unlock(&virtio_fs_mutex);
1492833c5a42SMiklos Szeredi 	return err;
1493a62a8ef9SStefan Hajnoczi }
1494a62a8ef9SStefan Hajnoczi 
1495a62a8ef9SStefan Hajnoczi static const struct fs_context_operations virtio_fs_context_ops = {
14961dd53957SVivek Goyal 	.free		= virtio_fs_free_fc,
14971dd53957SVivek Goyal 	.parse_param	= virtio_fs_parse_param,
1498a62a8ef9SStefan Hajnoczi 	.get_tree	= virtio_fs_get_tree,
1499a62a8ef9SStefan Hajnoczi };
1500a62a8ef9SStefan Hajnoczi 
1501a62a8ef9SStefan Hajnoczi static int virtio_fs_init_fs_context(struct fs_context *fsc)
1502a62a8ef9SStefan Hajnoczi {
15031dd53957SVivek Goyal 	struct fuse_fs_context *ctx;
15041dd53957SVivek Goyal 
15051dd53957SVivek Goyal 	ctx = kzalloc(sizeof(struct fuse_fs_context), GFP_KERNEL);
15061dd53957SVivek Goyal 	if (!ctx)
15071dd53957SVivek Goyal 		return -ENOMEM;
15081dd53957SVivek Goyal 	fsc->fs_private = ctx;
1509a62a8ef9SStefan Hajnoczi 	fsc->ops = &virtio_fs_context_ops;
1510a62a8ef9SStefan Hajnoczi 	return 0;
1511a62a8ef9SStefan Hajnoczi }
1512a62a8ef9SStefan Hajnoczi 
1513a62a8ef9SStefan Hajnoczi static struct file_system_type virtio_fs_type = {
1514a62a8ef9SStefan Hajnoczi 	.owner		= THIS_MODULE,
1515a62a8ef9SStefan Hajnoczi 	.name		= "virtiofs",
1516a62a8ef9SStefan Hajnoczi 	.init_fs_context = virtio_fs_init_fs_context,
1517a62a8ef9SStefan Hajnoczi 	.kill_sb	= virtio_kill_sb,
1518a62a8ef9SStefan Hajnoczi };
1519a62a8ef9SStefan Hajnoczi 
1520a62a8ef9SStefan Hajnoczi static int __init virtio_fs_init(void)
1521a62a8ef9SStefan Hajnoczi {
1522a62a8ef9SStefan Hajnoczi 	int ret;
1523a62a8ef9SStefan Hajnoczi 
1524a62a8ef9SStefan Hajnoczi 	ret = register_virtio_driver(&virtio_fs_driver);
1525a62a8ef9SStefan Hajnoczi 	if (ret < 0)
1526a62a8ef9SStefan Hajnoczi 		return ret;
1527a62a8ef9SStefan Hajnoczi 
1528a62a8ef9SStefan Hajnoczi 	ret = register_filesystem(&virtio_fs_type);
1529a62a8ef9SStefan Hajnoczi 	if (ret < 0) {
1530a62a8ef9SStefan Hajnoczi 		unregister_virtio_driver(&virtio_fs_driver);
1531a62a8ef9SStefan Hajnoczi 		return ret;
1532a62a8ef9SStefan Hajnoczi 	}
1533a62a8ef9SStefan Hajnoczi 
1534a62a8ef9SStefan Hajnoczi 	return 0;
1535a62a8ef9SStefan Hajnoczi }
1536a62a8ef9SStefan Hajnoczi module_init(virtio_fs_init);
1537a62a8ef9SStefan Hajnoczi 
1538a62a8ef9SStefan Hajnoczi static void __exit virtio_fs_exit(void)
1539a62a8ef9SStefan Hajnoczi {
1540a62a8ef9SStefan Hajnoczi 	unregister_filesystem(&virtio_fs_type);
1541a62a8ef9SStefan Hajnoczi 	unregister_virtio_driver(&virtio_fs_driver);
1542a62a8ef9SStefan Hajnoczi }
1543a62a8ef9SStefan Hajnoczi module_exit(virtio_fs_exit);
1544a62a8ef9SStefan Hajnoczi 
1545a62a8ef9SStefan Hajnoczi MODULE_AUTHOR("Stefan Hajnoczi <stefanha@redhat.com>");
1546a62a8ef9SStefan Hajnoczi MODULE_DESCRIPTION("Virtio Filesystem");
1547a62a8ef9SStefan Hajnoczi MODULE_LICENSE("GPL");
1548a62a8ef9SStefan Hajnoczi MODULE_ALIAS_FS(KBUILD_MODNAME);
1549a62a8ef9SStefan Hajnoczi MODULE_DEVICE_TABLE(virtio, id_table);
1550