GNU Linux-libre 4.9.282-gnu1
[releases.git] / fs / fuse / file.c
1 /*
2   FUSE: Filesystem in Userspace
3   Copyright (C) 2001-2008  Miklos Szeredi <miklos@szeredi.hu>
4
5   This program can be distributed under the terms of the GNU GPL.
6   See the file COPYING.
7 */
8
9 #include "fuse_i.h"
10
11 #include <linux/pagemap.h>
12 #include <linux/slab.h>
13 #include <linux/kernel.h>
14 #include <linux/sched.h>
15 #include <linux/module.h>
16 #include <linux/compat.h>
17 #include <linux/swap.h>
18 #include <linux/falloc.h>
19 #include <linux/uio.h>
20 #include <linux/fs.h>
21
22 static const struct file_operations fuse_direct_io_file_operations;
23
24 static int fuse_send_open(struct fuse_conn *fc, u64 nodeid, struct file *file,
25                           int opcode, struct fuse_open_out *outargp)
26 {
27         struct fuse_open_in inarg;
28         FUSE_ARGS(args);
29
30         memset(&inarg, 0, sizeof(inarg));
31         inarg.flags = file->f_flags & ~(O_CREAT | O_EXCL | O_NOCTTY);
32         if (!fc->atomic_o_trunc)
33                 inarg.flags &= ~O_TRUNC;
34         args.in.h.opcode = opcode;
35         args.in.h.nodeid = nodeid;
36         args.in.numargs = 1;
37         args.in.args[0].size = sizeof(inarg);
38         args.in.args[0].value = &inarg;
39         args.out.numargs = 1;
40         args.out.args[0].size = sizeof(*outargp);
41         args.out.args[0].value = outargp;
42
43         return fuse_simple_request(fc, &args);
44 }
45
46 struct fuse_file *fuse_file_alloc(struct fuse_conn *fc)
47 {
48         struct fuse_file *ff;
49
50         ff = kzalloc(sizeof(struct fuse_file), GFP_KERNEL);
51         if (unlikely(!ff))
52                 return NULL;
53
54         ff->fc = fc;
55         ff->reserved_req = fuse_request_alloc(0);
56         if (unlikely(!ff->reserved_req)) {
57                 kfree(ff);
58                 return NULL;
59         }
60
61         INIT_LIST_HEAD(&ff->write_entry);
62         atomic_set(&ff->count, 0);
63         RB_CLEAR_NODE(&ff->polled_node);
64         init_waitqueue_head(&ff->poll_wait);
65
66         spin_lock(&fc->lock);
67         ff->kh = ++fc->khctr;
68         spin_unlock(&fc->lock);
69
70         return ff;
71 }
72
73 void fuse_file_free(struct fuse_file *ff)
74 {
75         fuse_request_free(ff->reserved_req);
76         kfree(ff);
77 }
78
79 struct fuse_file *fuse_file_get(struct fuse_file *ff)
80 {
81         atomic_inc(&ff->count);
82         return ff;
83 }
84
85 static void fuse_release_end(struct fuse_conn *fc, struct fuse_req *req)
86 {
87         iput(req->misc.release.inode);
88 }
89
90 static void fuse_file_put(struct fuse_file *ff, bool sync)
91 {
92         if (atomic_dec_and_test(&ff->count)) {
93                 struct fuse_req *req = ff->reserved_req;
94
95                 if (ff->fc->no_open) {
96                         /*
97                          * Drop the release request when client does not
98                          * implement 'open'
99                          */
100                         __clear_bit(FR_BACKGROUND, &req->flags);
101                         iput(req->misc.release.inode);
102                         fuse_put_request(ff->fc, req);
103                 } else if (sync) {
104                         __set_bit(FR_FORCE, &req->flags);
105                         __clear_bit(FR_BACKGROUND, &req->flags);
106                         fuse_request_send(ff->fc, req);
107                         iput(req->misc.release.inode);
108                         fuse_put_request(ff->fc, req);
109                 } else {
110                         req->end = fuse_release_end;
111                         __set_bit(FR_BACKGROUND, &req->flags);
112                         fuse_request_send_background(ff->fc, req);
113                 }
114                 kfree(ff);
115         }
116 }
117
118 int fuse_do_open(struct fuse_conn *fc, u64 nodeid, struct file *file,
119                  bool isdir)
120 {
121         struct fuse_file *ff;
122         int opcode = isdir ? FUSE_OPENDIR : FUSE_OPEN;
123
124         ff = fuse_file_alloc(fc);
125         if (!ff)
126                 return -ENOMEM;
127
128         ff->fh = 0;
129         ff->open_flags = FOPEN_KEEP_CACHE; /* Default for no-open */
130         if (!fc->no_open || isdir) {
131                 struct fuse_open_out outarg;
132                 int err;
133
134                 err = fuse_send_open(fc, nodeid, file, opcode, &outarg);
135                 if (!err) {
136                         ff->fh = outarg.fh;
137                         ff->open_flags = outarg.open_flags;
138
139                 } else if (err != -ENOSYS || isdir) {
140                         fuse_file_free(ff);
141                         return err;
142                 } else {
143                         fc->no_open = 1;
144                 }
145         }
146
147         if (isdir)
148                 ff->open_flags &= ~FOPEN_DIRECT_IO;
149
150         ff->nodeid = nodeid;
151         file->private_data = fuse_file_get(ff);
152
153         return 0;
154 }
155 EXPORT_SYMBOL_GPL(fuse_do_open);
156
157 static void fuse_link_write_file(struct file *file)
158 {
159         struct inode *inode = file_inode(file);
160         struct fuse_conn *fc = get_fuse_conn(inode);
161         struct fuse_inode *fi = get_fuse_inode(inode);
162         struct fuse_file *ff = file->private_data;
163         /*
164          * file may be written through mmap, so chain it onto the
165          * inodes's write_file list
166          */
167         spin_lock(&fc->lock);
168         if (list_empty(&ff->write_entry))
169                 list_add(&ff->write_entry, &fi->write_files);
170         spin_unlock(&fc->lock);
171 }
172
173 void fuse_finish_open(struct inode *inode, struct file *file)
174 {
175         struct fuse_file *ff = file->private_data;
176         struct fuse_conn *fc = get_fuse_conn(inode);
177
178         if (ff->open_flags & FOPEN_DIRECT_IO)
179                 file->f_op = &fuse_direct_io_file_operations;
180         if (!(ff->open_flags & FOPEN_KEEP_CACHE))
181                 invalidate_inode_pages2(inode->i_mapping);
182         if (ff->open_flags & FOPEN_STREAM)
183                 stream_open(inode, file);
184         else if (ff->open_flags & FOPEN_NONSEEKABLE)
185                 nonseekable_open(inode, file);
186         if (fc->atomic_o_trunc && (file->f_flags & O_TRUNC)) {
187                 struct fuse_inode *fi = get_fuse_inode(inode);
188
189                 spin_lock(&fc->lock);
190                 fi->attr_version = ++fc->attr_version;
191                 i_size_write(inode, 0);
192                 spin_unlock(&fc->lock);
193                 fuse_invalidate_attr(inode);
194                 if (fc->writeback_cache)
195                         file_update_time(file);
196         }
197         if ((file->f_mode & FMODE_WRITE) && fc->writeback_cache)
198                 fuse_link_write_file(file);
199 }
200
201 int fuse_open_common(struct inode *inode, struct file *file, bool isdir)
202 {
203         struct fuse_conn *fc = get_fuse_conn(inode);
204         int err;
205         bool is_wb_truncate = (file->f_flags & O_TRUNC) &&
206                           fc->atomic_o_trunc &&
207                           fc->writeback_cache;
208
209         err = generic_file_open(inode, file);
210         if (err)
211                 return err;
212
213         if (is_wb_truncate) {
214                 inode_lock(inode);
215                 fuse_set_nowrite(inode);
216         }
217
218         err = fuse_do_open(fc, get_node_id(inode), file, isdir);
219
220         if (!err)
221                 fuse_finish_open(inode, file);
222
223         if (is_wb_truncate) {
224                 fuse_release_nowrite(inode);
225                 inode_unlock(inode);
226         }
227
228         return err;
229 }
230
231 static void fuse_prepare_release(struct fuse_file *ff, int flags, int opcode)
232 {
233         struct fuse_conn *fc = ff->fc;
234         struct fuse_req *req = ff->reserved_req;
235         struct fuse_release_in *inarg = &req->misc.release.in;
236
237         spin_lock(&fc->lock);
238         list_del(&ff->write_entry);
239         if (!RB_EMPTY_NODE(&ff->polled_node))
240                 rb_erase(&ff->polled_node, &fc->polled_files);
241         spin_unlock(&fc->lock);
242
243         wake_up_interruptible_all(&ff->poll_wait);
244
245         inarg->fh = ff->fh;
246         inarg->flags = flags;
247         req->in.h.opcode = opcode;
248         req->in.h.nodeid = ff->nodeid;
249         req->in.numargs = 1;
250         req->in.args[0].size = sizeof(struct fuse_release_in);
251         req->in.args[0].value = inarg;
252 }
253
254 void fuse_release_common(struct file *file, int opcode)
255 {
256         struct fuse_file *ff;
257         struct fuse_req *req;
258
259         ff = file->private_data;
260         if (unlikely(!ff))
261                 return;
262
263         req = ff->reserved_req;
264         fuse_prepare_release(ff, file->f_flags, opcode);
265
266         if (ff->flock) {
267                 struct fuse_release_in *inarg = &req->misc.release.in;
268                 inarg->release_flags |= FUSE_RELEASE_FLOCK_UNLOCK;
269                 inarg->lock_owner = fuse_lock_owner_id(ff->fc,
270                                                        (fl_owner_t) file);
271         }
272         /* Hold inode until release is finished */
273         req->misc.release.inode = igrab(file_inode(file));
274
275         /*
276          * Normally this will send the RELEASE request, however if
277          * some asynchronous READ or WRITE requests are outstanding,
278          * the sending will be delayed.
279          *
280          * Make the release synchronous if this is a fuseblk mount,
281          * synchronous RELEASE is allowed (and desirable) in this case
282          * because the server can be trusted not to screw up.
283          */
284         fuse_file_put(ff, ff->fc->destroy_req != NULL);
285 }
286
287 static int fuse_open(struct inode *inode, struct file *file)
288 {
289         return fuse_open_common(inode, file, false);
290 }
291
292 static int fuse_release(struct inode *inode, struct file *file)
293 {
294         struct fuse_conn *fc = get_fuse_conn(inode);
295
296         /* see fuse_vma_close() for !writeback_cache case */
297         if (fc->writeback_cache)
298                 write_inode_now(inode, 1);
299
300         fuse_release_common(file, FUSE_RELEASE);
301
302         /* return value is ignored by VFS */
303         return 0;
304 }
305
306 void fuse_sync_release(struct fuse_file *ff, int flags)
307 {
308         WARN_ON(atomic_read(&ff->count) > 1);
309         fuse_prepare_release(ff, flags, FUSE_RELEASE);
310         __set_bit(FR_FORCE, &ff->reserved_req->flags);
311         __clear_bit(FR_BACKGROUND, &ff->reserved_req->flags);
312         fuse_request_send(ff->fc, ff->reserved_req);
313         fuse_put_request(ff->fc, ff->reserved_req);
314         kfree(ff);
315 }
316 EXPORT_SYMBOL_GPL(fuse_sync_release);
317
318 /*
319  * Scramble the ID space with XTEA, so that the value of the files_struct
320  * pointer is not exposed to userspace.
321  */
322 u64 fuse_lock_owner_id(struct fuse_conn *fc, fl_owner_t id)
323 {
324         u32 *k = fc->scramble_key;
325         u64 v = (unsigned long) id;
326         u32 v0 = v;
327         u32 v1 = v >> 32;
328         u32 sum = 0;
329         int i;
330
331         for (i = 0; i < 32; i++) {
332                 v0 += ((v1 << 4 ^ v1 >> 5) + v1) ^ (sum + k[sum & 3]);
333                 sum += 0x9E3779B9;
334                 v1 += ((v0 << 4 ^ v0 >> 5) + v0) ^ (sum + k[sum>>11 & 3]);
335         }
336
337         return (u64) v0 + ((u64) v1 << 32);
338 }
339
340 /*
341  * Check if any page in a range is under writeback
342  *
343  * This is currently done by walking the list of writepage requests
344  * for the inode, which can be pretty inefficient.
345  */
346 static bool fuse_range_is_writeback(struct inode *inode, pgoff_t idx_from,
347                                    pgoff_t idx_to)
348 {
349         struct fuse_conn *fc = get_fuse_conn(inode);
350         struct fuse_inode *fi = get_fuse_inode(inode);
351         struct fuse_req *req;
352         bool found = false;
353
354         spin_lock(&fc->lock);
355         list_for_each_entry(req, &fi->writepages, writepages_entry) {
356                 pgoff_t curr_index;
357
358                 BUG_ON(req->inode != inode);
359                 curr_index = req->misc.write.in.offset >> PAGE_SHIFT;
360                 if (idx_from < curr_index + req->num_pages &&
361                     curr_index <= idx_to) {
362                         found = true;
363                         break;
364                 }
365         }
366         spin_unlock(&fc->lock);
367
368         return found;
369 }
370
371 static inline bool fuse_page_is_writeback(struct inode *inode, pgoff_t index)
372 {
373         return fuse_range_is_writeback(inode, index, index);
374 }
375
376 /*
377  * Wait for page writeback to be completed.
378  *
379  * Since fuse doesn't rely on the VM writeback tracking, this has to
380  * use some other means.
381  */
382 static int fuse_wait_on_page_writeback(struct inode *inode, pgoff_t index)
383 {
384         struct fuse_inode *fi = get_fuse_inode(inode);
385
386         wait_event(fi->page_waitq, !fuse_page_is_writeback(inode, index));
387         return 0;
388 }
389
390 /*
391  * Wait for all pending writepages on the inode to finish.
392  *
393  * This is currently done by blocking further writes with FUSE_NOWRITE
394  * and waiting for all sent writes to complete.
395  *
396  * This must be called under i_mutex, otherwise the FUSE_NOWRITE usage
397  * could conflict with truncation.
398  */
399 static void fuse_sync_writes(struct inode *inode)
400 {
401         fuse_set_nowrite(inode);
402         fuse_release_nowrite(inode);
403 }
404
405 static int fuse_flush(struct file *file, fl_owner_t id)
406 {
407         struct inode *inode = file_inode(file);
408         struct fuse_conn *fc = get_fuse_conn(inode);
409         struct fuse_file *ff = file->private_data;
410         struct fuse_req *req;
411         struct fuse_flush_in inarg;
412         int err;
413
414         if (is_bad_inode(inode))
415                 return -EIO;
416
417         if (fc->no_flush)
418                 return 0;
419
420         err = write_inode_now(inode, 1);
421         if (err)
422                 return err;
423
424         inode_lock(inode);
425         fuse_sync_writes(inode);
426         inode_unlock(inode);
427
428         err = filemap_check_errors(file->f_mapping);
429         if (err)
430                 return err;
431
432         req = fuse_get_req_nofail_nopages(fc, file);
433         memset(&inarg, 0, sizeof(inarg));
434         inarg.fh = ff->fh;
435         inarg.lock_owner = fuse_lock_owner_id(fc, id);
436         req->in.h.opcode = FUSE_FLUSH;
437         req->in.h.nodeid = get_node_id(inode);
438         req->in.numargs = 1;
439         req->in.args[0].size = sizeof(inarg);
440         req->in.args[0].value = &inarg;
441         __set_bit(FR_FORCE, &req->flags);
442         fuse_request_send(fc, req);
443         err = req->out.h.error;
444         fuse_put_request(fc, req);
445         if (err == -ENOSYS) {
446                 fc->no_flush = 1;
447                 err = 0;
448         }
449         return err;
450 }
451
452 int fuse_fsync_common(struct file *file, loff_t start, loff_t end,
453                       int datasync, int isdir)
454 {
455         struct inode *inode = file->f_mapping->host;
456         struct fuse_conn *fc = get_fuse_conn(inode);
457         struct fuse_file *ff = file->private_data;
458         FUSE_ARGS(args);
459         struct fuse_fsync_in inarg;
460         int err;
461
462         if (is_bad_inode(inode))
463                 return -EIO;
464
465         inode_lock(inode);
466
467         /*
468          * Start writeback against all dirty pages of the inode, then
469          * wait for all outstanding writes, before sending the FSYNC
470          * request.
471          */
472         err = filemap_write_and_wait_range(inode->i_mapping, start, end);
473         if (err)
474                 goto out;
475
476         fuse_sync_writes(inode);
477
478         /*
479          * Due to implementation of fuse writeback
480          * filemap_write_and_wait_range() does not catch errors.
481          * We have to do this directly after fuse_sync_writes()
482          */
483         err = filemap_check_errors(file->f_mapping);
484         if (err)
485                 goto out;
486
487         err = sync_inode_metadata(inode, 1);
488         if (err)
489                 goto out;
490
491         if ((!isdir && fc->no_fsync) || (isdir && fc->no_fsyncdir))
492                 goto out;
493
494         memset(&inarg, 0, sizeof(inarg));
495         inarg.fh = ff->fh;
496         inarg.fsync_flags = datasync ? 1 : 0;
497         args.in.h.opcode = isdir ? FUSE_FSYNCDIR : FUSE_FSYNC;
498         args.in.h.nodeid = get_node_id(inode);
499         args.in.numargs = 1;
500         args.in.args[0].size = sizeof(inarg);
501         args.in.args[0].value = &inarg;
502         err = fuse_simple_request(fc, &args);
503         if (err == -ENOSYS) {
504                 if (isdir)
505                         fc->no_fsyncdir = 1;
506                 else
507                         fc->no_fsync = 1;
508                 err = 0;
509         }
510 out:
511         inode_unlock(inode);
512         return err;
513 }
514
515 static int fuse_fsync(struct file *file, loff_t start, loff_t end,
516                       int datasync)
517 {
518         return fuse_fsync_common(file, start, end, datasync, 0);
519 }
520
521 void fuse_read_fill(struct fuse_req *req, struct file *file, loff_t pos,
522                     size_t count, int opcode)
523 {
524         struct fuse_read_in *inarg = &req->misc.read.in;
525         struct fuse_file *ff = file->private_data;
526
527         inarg->fh = ff->fh;
528         inarg->offset = pos;
529         inarg->size = count;
530         inarg->flags = file->f_flags;
531         req->in.h.opcode = opcode;
532         req->in.h.nodeid = ff->nodeid;
533         req->in.numargs = 1;
534         req->in.args[0].size = sizeof(struct fuse_read_in);
535         req->in.args[0].value = inarg;
536         req->out.argvar = 1;
537         req->out.numargs = 1;
538         req->out.args[0].size = count;
539 }
540
541 static void fuse_release_user_pages(struct fuse_req *req, bool should_dirty)
542 {
543         unsigned i;
544
545         for (i = 0; i < req->num_pages; i++) {
546                 struct page *page = req->pages[i];
547                 if (should_dirty)
548                         set_page_dirty_lock(page);
549                 put_page(page);
550         }
551 }
552
553 static void fuse_io_release(struct kref *kref)
554 {
555         kfree(container_of(kref, struct fuse_io_priv, refcnt));
556 }
557
558 static ssize_t fuse_get_res_by_io(struct fuse_io_priv *io)
559 {
560         if (io->err)
561                 return io->err;
562
563         if (io->bytes >= 0 && io->write)
564                 return -EIO;
565
566         return io->bytes < 0 ? io->size : io->bytes;
567 }
568
569 /**
570  * In case of short read, the caller sets 'pos' to the position of
571  * actual end of fuse request in IO request. Otherwise, if bytes_requested
572  * == bytes_transferred or rw == WRITE, the caller sets 'pos' to -1.
573  *
574  * An example:
575  * User requested DIO read of 64K. It was splitted into two 32K fuse requests,
576  * both submitted asynchronously. The first of them was ACKed by userspace as
577  * fully completed (req->out.args[0].size == 32K) resulting in pos == -1. The
578  * second request was ACKed as short, e.g. only 1K was read, resulting in
579  * pos == 33K.
580  *
581  * Thus, when all fuse requests are completed, the minimal non-negative 'pos'
582  * will be equal to the length of the longest contiguous fragment of
583  * transferred data starting from the beginning of IO request.
584  */
585 static void fuse_aio_complete(struct fuse_io_priv *io, int err, ssize_t pos)
586 {
587         int left;
588
589         spin_lock(&io->lock);
590         if (err)
591                 io->err = io->err ? : err;
592         else if (pos >= 0 && (io->bytes < 0 || pos < io->bytes))
593                 io->bytes = pos;
594
595         left = --io->reqs;
596         if (!left && io->blocking)
597                 complete(io->done);
598         spin_unlock(&io->lock);
599
600         if (!left && !io->blocking) {
601                 ssize_t res = fuse_get_res_by_io(io);
602
603                 if (res >= 0) {
604                         struct inode *inode = file_inode(io->iocb->ki_filp);
605                         struct fuse_conn *fc = get_fuse_conn(inode);
606                         struct fuse_inode *fi = get_fuse_inode(inode);
607
608                         spin_lock(&fc->lock);
609                         fi->attr_version = ++fc->attr_version;
610                         spin_unlock(&fc->lock);
611                 }
612
613                 io->iocb->ki_complete(io->iocb, res, 0);
614         }
615
616         kref_put(&io->refcnt, fuse_io_release);
617 }
618
619 static void fuse_aio_complete_req(struct fuse_conn *fc, struct fuse_req *req)
620 {
621         struct fuse_io_priv *io = req->io;
622         ssize_t pos = -1;
623
624         fuse_release_user_pages(req, !io->write);
625
626         if (io->write) {
627                 if (req->misc.write.in.size != req->misc.write.out.size)
628                         pos = req->misc.write.in.offset - io->offset +
629                                 req->misc.write.out.size;
630         } else {
631                 if (req->misc.read.in.size != req->out.args[0].size)
632                         pos = req->misc.read.in.offset - io->offset +
633                                 req->out.args[0].size;
634         }
635
636         fuse_aio_complete(io, req->out.h.error, pos);
637 }
638
639 static size_t fuse_async_req_send(struct fuse_conn *fc, struct fuse_req *req,
640                 size_t num_bytes, struct fuse_io_priv *io)
641 {
642         spin_lock(&io->lock);
643         kref_get(&io->refcnt);
644         io->size += num_bytes;
645         io->reqs++;
646         spin_unlock(&io->lock);
647
648         req->io = io;
649         req->end = fuse_aio_complete_req;
650
651         __fuse_get_request(req);
652         fuse_request_send_background(fc, req);
653
654         return num_bytes;
655 }
656
657 static size_t fuse_send_read(struct fuse_req *req, struct fuse_io_priv *io,
658                              loff_t pos, size_t count, fl_owner_t owner)
659 {
660         struct file *file = io->file;
661         struct fuse_file *ff = file->private_data;
662         struct fuse_conn *fc = ff->fc;
663
664         fuse_read_fill(req, file, pos, count, FUSE_READ);
665         if (owner != NULL) {
666                 struct fuse_read_in *inarg = &req->misc.read.in;
667
668                 inarg->read_flags |= FUSE_READ_LOCKOWNER;
669                 inarg->lock_owner = fuse_lock_owner_id(fc, owner);
670         }
671
672         if (io->async)
673                 return fuse_async_req_send(fc, req, count, io);
674
675         fuse_request_send(fc, req);
676         return req->out.args[0].size;
677 }
678
679 static void fuse_read_update_size(struct inode *inode, loff_t size,
680                                   u64 attr_ver)
681 {
682         struct fuse_conn *fc = get_fuse_conn(inode);
683         struct fuse_inode *fi = get_fuse_inode(inode);
684
685         spin_lock(&fc->lock);
686         if (attr_ver == fi->attr_version && size < inode->i_size &&
687             !test_bit(FUSE_I_SIZE_UNSTABLE, &fi->state)) {
688                 fi->attr_version = ++fc->attr_version;
689                 i_size_write(inode, size);
690         }
691         spin_unlock(&fc->lock);
692 }
693
694 static void fuse_short_read(struct fuse_req *req, struct inode *inode,
695                             u64 attr_ver)
696 {
697         size_t num_read = req->out.args[0].size;
698         struct fuse_conn *fc = get_fuse_conn(inode);
699
700         if (fc->writeback_cache) {
701                 /*
702                  * A hole in a file. Some data after the hole are in page cache,
703                  * but have not reached the client fs yet. So, the hole is not
704                  * present there.
705                  */
706                 int i;
707                 int start_idx = num_read >> PAGE_SHIFT;
708                 size_t off = num_read & (PAGE_SIZE - 1);
709
710                 for (i = start_idx; i < req->num_pages; i++) {
711                         zero_user_segment(req->pages[i], off, PAGE_SIZE);
712                         off = 0;
713                 }
714         } else {
715                 loff_t pos = page_offset(req->pages[0]) + num_read;
716                 fuse_read_update_size(inode, pos, attr_ver);
717         }
718 }
719
720 static int fuse_do_readpage(struct file *file, struct page *page)
721 {
722         struct fuse_io_priv io = FUSE_IO_PRIV_SYNC(file);
723         struct inode *inode = page->mapping->host;
724         struct fuse_conn *fc = get_fuse_conn(inode);
725         struct fuse_req *req;
726         size_t num_read;
727         loff_t pos = page_offset(page);
728         size_t count = PAGE_SIZE;
729         u64 attr_ver;
730         int err;
731
732         /*
733          * Page writeback can extend beyond the lifetime of the
734          * page-cache page, so make sure we read a properly synced
735          * page.
736          */
737         fuse_wait_on_page_writeback(inode, page->index);
738
739         req = fuse_get_req(fc, 1);
740         if (IS_ERR(req))
741                 return PTR_ERR(req);
742
743         attr_ver = fuse_get_attr_version(fc);
744
745         req->out.page_zeroing = 1;
746         req->out.argpages = 1;
747         req->num_pages = 1;
748         req->pages[0] = page;
749         req->page_descs[0].length = count;
750         num_read = fuse_send_read(req, &io, pos, count, NULL);
751         err = req->out.h.error;
752
753         if (!err) {
754                 /*
755                  * Short read means EOF.  If file size is larger, truncate it
756                  */
757                 if (num_read < count)
758                         fuse_short_read(req, inode, attr_ver);
759
760                 SetPageUptodate(page);
761         }
762
763         fuse_put_request(fc, req);
764
765         return err;
766 }
767
768 static int fuse_readpage(struct file *file, struct page *page)
769 {
770         struct inode *inode = page->mapping->host;
771         int err;
772
773         err = -EIO;
774         if (is_bad_inode(inode))
775                 goto out;
776
777         err = fuse_do_readpage(file, page);
778         fuse_invalidate_atime(inode);
779  out:
780         unlock_page(page);
781         return err;
782 }
783
784 static void fuse_readpages_end(struct fuse_conn *fc, struct fuse_req *req)
785 {
786         int i;
787         size_t count = req->misc.read.in.size;
788         size_t num_read = req->out.args[0].size;
789         struct address_space *mapping = NULL;
790
791         for (i = 0; mapping == NULL && i < req->num_pages; i++)
792                 mapping = req->pages[i]->mapping;
793
794         if (mapping) {
795                 struct inode *inode = mapping->host;
796
797                 /*
798                  * Short read means EOF. If file size is larger, truncate it
799                  */
800                 if (!req->out.h.error && num_read < count)
801                         fuse_short_read(req, inode, req->misc.read.attr_ver);
802
803                 fuse_invalidate_atime(inode);
804         }
805
806         for (i = 0; i < req->num_pages; i++) {
807                 struct page *page = req->pages[i];
808                 if (!req->out.h.error)
809                         SetPageUptodate(page);
810                 else
811                         SetPageError(page);
812                 unlock_page(page);
813                 put_page(page);
814         }
815         if (req->ff)
816                 fuse_file_put(req->ff, false);
817 }
818
819 static void fuse_send_readpages(struct fuse_req *req, struct file *file)
820 {
821         struct fuse_file *ff = file->private_data;
822         struct fuse_conn *fc = ff->fc;
823         loff_t pos = page_offset(req->pages[0]);
824         size_t count = req->num_pages << PAGE_SHIFT;
825
826         req->out.argpages = 1;
827         req->out.page_zeroing = 1;
828         req->out.page_replace = 1;
829         fuse_read_fill(req, file, pos, count, FUSE_READ);
830         req->misc.read.attr_ver = fuse_get_attr_version(fc);
831         if (fc->async_read) {
832                 req->ff = fuse_file_get(ff);
833                 req->end = fuse_readpages_end;
834                 fuse_request_send_background(fc, req);
835         } else {
836                 fuse_request_send(fc, req);
837                 fuse_readpages_end(fc, req);
838                 fuse_put_request(fc, req);
839         }
840 }
841
842 struct fuse_fill_data {
843         struct fuse_req *req;
844         struct file *file;
845         struct inode *inode;
846         unsigned nr_pages;
847 };
848
849 static int fuse_readpages_fill(void *_data, struct page *page)
850 {
851         struct fuse_fill_data *data = _data;
852         struct fuse_req *req = data->req;
853         struct inode *inode = data->inode;
854         struct fuse_conn *fc = get_fuse_conn(inode);
855
856         fuse_wait_on_page_writeback(inode, page->index);
857
858         if (req->num_pages &&
859             (req->num_pages == FUSE_MAX_PAGES_PER_REQ ||
860              (req->num_pages + 1) * PAGE_SIZE > fc->max_read ||
861              req->pages[req->num_pages - 1]->index + 1 != page->index)) {
862                 int nr_alloc = min_t(unsigned, data->nr_pages,
863                                      FUSE_MAX_PAGES_PER_REQ);
864                 fuse_send_readpages(req, data->file);
865                 if (fc->async_read)
866                         req = fuse_get_req_for_background(fc, nr_alloc);
867                 else
868                         req = fuse_get_req(fc, nr_alloc);
869
870                 data->req = req;
871                 if (IS_ERR(req)) {
872                         unlock_page(page);
873                         return PTR_ERR(req);
874                 }
875         }
876
877         if (WARN_ON(req->num_pages >= req->max_pages)) {
878                 unlock_page(page);
879                 fuse_put_request(fc, req);
880                 return -EIO;
881         }
882
883         get_page(page);
884         req->pages[req->num_pages] = page;
885         req->page_descs[req->num_pages].length = PAGE_SIZE;
886         req->num_pages++;
887         data->nr_pages--;
888         return 0;
889 }
890
891 static int fuse_readpages(struct file *file, struct address_space *mapping,
892                           struct list_head *pages, unsigned nr_pages)
893 {
894         struct inode *inode = mapping->host;
895         struct fuse_conn *fc = get_fuse_conn(inode);
896         struct fuse_fill_data data;
897         int err;
898         int nr_alloc = min_t(unsigned, nr_pages, FUSE_MAX_PAGES_PER_REQ);
899
900         err = -EIO;
901         if (is_bad_inode(inode))
902                 goto out;
903
904         data.file = file;
905         data.inode = inode;
906         if (fc->async_read)
907                 data.req = fuse_get_req_for_background(fc, nr_alloc);
908         else
909                 data.req = fuse_get_req(fc, nr_alloc);
910         data.nr_pages = nr_pages;
911         err = PTR_ERR(data.req);
912         if (IS_ERR(data.req))
913                 goto out;
914
915         err = read_cache_pages(mapping, pages, fuse_readpages_fill, &data);
916         if (!err) {
917                 if (data.req->num_pages)
918                         fuse_send_readpages(data.req, file);
919                 else
920                         fuse_put_request(fc, data.req);
921         }
922 out:
923         return err;
924 }
925
926 static ssize_t fuse_file_read_iter(struct kiocb *iocb, struct iov_iter *to)
927 {
928         struct inode *inode = iocb->ki_filp->f_mapping->host;
929         struct fuse_conn *fc = get_fuse_conn(inode);
930
931         /*
932          * In auto invalidate mode, always update attributes on read.
933          * Otherwise, only update if we attempt to read past EOF (to ensure
934          * i_size is up to date).
935          */
936         if (fc->auto_inval_data ||
937             (iocb->ki_pos + iov_iter_count(to) > i_size_read(inode))) {
938                 int err;
939                 err = fuse_update_attributes(inode, NULL, iocb->ki_filp, NULL);
940                 if (err)
941                         return err;
942         }
943
944         return generic_file_read_iter(iocb, to);
945 }
946
947 static void fuse_write_fill(struct fuse_req *req, struct fuse_file *ff,
948                             loff_t pos, size_t count)
949 {
950         struct fuse_write_in *inarg = &req->misc.write.in;
951         struct fuse_write_out *outarg = &req->misc.write.out;
952
953         inarg->fh = ff->fh;
954         inarg->offset = pos;
955         inarg->size = count;
956         req->in.h.opcode = FUSE_WRITE;
957         req->in.h.nodeid = ff->nodeid;
958         req->in.numargs = 2;
959         if (ff->fc->minor < 9)
960                 req->in.args[0].size = FUSE_COMPAT_WRITE_IN_SIZE;
961         else
962                 req->in.args[0].size = sizeof(struct fuse_write_in);
963         req->in.args[0].value = inarg;
964         req->in.args[1].size = count;
965         req->out.numargs = 1;
966         req->out.args[0].size = sizeof(struct fuse_write_out);
967         req->out.args[0].value = outarg;
968 }
969
970 static size_t fuse_send_write(struct fuse_req *req, struct fuse_io_priv *io,
971                               loff_t pos, size_t count, fl_owner_t owner)
972 {
973         struct file *file = io->file;
974         struct fuse_file *ff = file->private_data;
975         struct fuse_conn *fc = ff->fc;
976         struct fuse_write_in *inarg = &req->misc.write.in;
977
978         fuse_write_fill(req, ff, pos, count);
979         inarg->flags = file->f_flags;
980         if (owner != NULL) {
981                 inarg->write_flags |= FUSE_WRITE_LOCKOWNER;
982                 inarg->lock_owner = fuse_lock_owner_id(fc, owner);
983         }
984
985         if (io->async)
986                 return fuse_async_req_send(fc, req, count, io);
987
988         fuse_request_send(fc, req);
989         return req->misc.write.out.size;
990 }
991
992 bool fuse_write_update_size(struct inode *inode, loff_t pos)
993 {
994         struct fuse_conn *fc = get_fuse_conn(inode);
995         struct fuse_inode *fi = get_fuse_inode(inode);
996         bool ret = false;
997
998         spin_lock(&fc->lock);
999         fi->attr_version = ++fc->attr_version;
1000         if (pos > inode->i_size) {
1001                 i_size_write(inode, pos);
1002                 ret = true;
1003         }
1004         spin_unlock(&fc->lock);
1005
1006         return ret;
1007 }
1008
1009 static size_t fuse_send_write_pages(struct fuse_req *req, struct file *file,
1010                                     struct inode *inode, loff_t pos,
1011                                     size_t count)
1012 {
1013         size_t res;
1014         unsigned offset;
1015         unsigned i;
1016         struct fuse_io_priv io = FUSE_IO_PRIV_SYNC(file);
1017
1018         for (i = 0; i < req->num_pages; i++)
1019                 fuse_wait_on_page_writeback(inode, req->pages[i]->index);
1020
1021         res = fuse_send_write(req, &io, pos, count, NULL);
1022
1023         offset = req->page_descs[0].offset;
1024         count = res;
1025         for (i = 0; i < req->num_pages; i++) {
1026                 struct page *page = req->pages[i];
1027
1028                 if (!req->out.h.error && !offset && count >= PAGE_SIZE)
1029                         SetPageUptodate(page);
1030
1031                 if (count > PAGE_SIZE - offset)
1032                         count -= PAGE_SIZE - offset;
1033                 else
1034                         count = 0;
1035                 offset = 0;
1036
1037                 unlock_page(page);
1038                 put_page(page);
1039         }
1040
1041         return res;
1042 }
1043
1044 static ssize_t fuse_fill_write_pages(struct fuse_req *req,
1045                                struct address_space *mapping,
1046                                struct iov_iter *ii, loff_t pos)
1047 {
1048         struct fuse_conn *fc = get_fuse_conn(mapping->host);
1049         unsigned offset = pos & (PAGE_SIZE - 1);
1050         size_t count = 0;
1051         int err;
1052
1053         req->in.argpages = 1;
1054         req->page_descs[0].offset = offset;
1055
1056         do {
1057                 size_t tmp;
1058                 struct page *page;
1059                 pgoff_t index = pos >> PAGE_SHIFT;
1060                 size_t bytes = min_t(size_t, PAGE_SIZE - offset,
1061                                      iov_iter_count(ii));
1062
1063                 bytes = min_t(size_t, bytes, fc->max_write - count);
1064
1065  again:
1066                 err = -EFAULT;
1067                 if (iov_iter_fault_in_readable(ii, bytes))
1068                         break;
1069
1070                 err = -ENOMEM;
1071                 page = grab_cache_page_write_begin(mapping, index, 0);
1072                 if (!page)
1073                         break;
1074
1075                 if (mapping_writably_mapped(mapping))
1076                         flush_dcache_page(page);
1077
1078                 tmp = iov_iter_copy_from_user_atomic(page, ii, offset, bytes);
1079                 flush_dcache_page(page);
1080
1081                 iov_iter_advance(ii, tmp);
1082                 if (!tmp) {
1083                         unlock_page(page);
1084                         put_page(page);
1085                         bytes = min(bytes, iov_iter_single_seg_count(ii));
1086                         goto again;
1087                 }
1088
1089                 err = 0;
1090                 req->pages[req->num_pages] = page;
1091                 req->page_descs[req->num_pages].length = tmp;
1092                 req->num_pages++;
1093
1094                 count += tmp;
1095                 pos += tmp;
1096                 offset += tmp;
1097                 if (offset == PAGE_SIZE)
1098                         offset = 0;
1099
1100                 if (!fc->big_writes)
1101                         break;
1102         } while (iov_iter_count(ii) && count < fc->max_write &&
1103                  req->num_pages < req->max_pages && offset == 0);
1104
1105         return count > 0 ? count : err;
1106 }
1107
1108 static inline unsigned fuse_wr_pages(loff_t pos, size_t len)
1109 {
1110         return min_t(unsigned,
1111                      ((pos + len - 1) >> PAGE_SHIFT) -
1112                      (pos >> PAGE_SHIFT) + 1,
1113                      FUSE_MAX_PAGES_PER_REQ);
1114 }
1115
1116 static ssize_t fuse_perform_write(struct file *file,
1117                                   struct address_space *mapping,
1118                                   struct iov_iter *ii, loff_t pos)
1119 {
1120         struct inode *inode = mapping->host;
1121         struct fuse_conn *fc = get_fuse_conn(inode);
1122         struct fuse_inode *fi = get_fuse_inode(inode);
1123         int err = 0;
1124         ssize_t res = 0;
1125
1126         if (is_bad_inode(inode))
1127                 return -EIO;
1128
1129         if (inode->i_size < pos + iov_iter_count(ii))
1130                 set_bit(FUSE_I_SIZE_UNSTABLE, &fi->state);
1131
1132         do {
1133                 struct fuse_req *req;
1134                 ssize_t count;
1135                 unsigned nr_pages = fuse_wr_pages(pos, iov_iter_count(ii));
1136
1137                 req = fuse_get_req(fc, nr_pages);
1138                 if (IS_ERR(req)) {
1139                         err = PTR_ERR(req);
1140                         break;
1141                 }
1142
1143                 count = fuse_fill_write_pages(req, mapping, ii, pos);
1144                 if (count <= 0) {
1145                         err = count;
1146                 } else {
1147                         size_t num_written;
1148
1149                         num_written = fuse_send_write_pages(req, file, inode,
1150                                                             pos, count);
1151                         err = req->out.h.error;
1152                         if (!err) {
1153                                 res += num_written;
1154                                 pos += num_written;
1155
1156                                 /* break out of the loop on short write */
1157                                 if (num_written != count)
1158                                         err = -EIO;
1159                         }
1160                 }
1161                 fuse_put_request(fc, req);
1162         } while (!err && iov_iter_count(ii));
1163
1164         if (res > 0)
1165                 fuse_write_update_size(inode, pos);
1166
1167         clear_bit(FUSE_I_SIZE_UNSTABLE, &fi->state);
1168         fuse_invalidate_attr(inode);
1169
1170         return res > 0 ? res : err;
1171 }
1172
1173 static ssize_t fuse_file_write_iter(struct kiocb *iocb, struct iov_iter *from)
1174 {
1175         struct file *file = iocb->ki_filp;
1176         struct address_space *mapping = file->f_mapping;
1177         ssize_t written = 0;
1178         ssize_t written_buffered = 0;
1179         struct inode *inode = mapping->host;
1180         ssize_t err;
1181         loff_t endbyte = 0;
1182
1183         if (get_fuse_conn(inode)->writeback_cache) {
1184                 /* Update size (EOF optimization) and mode (SUID clearing) */
1185                 err = fuse_update_attributes(mapping->host, NULL, file, NULL);
1186                 if (err)
1187                         return err;
1188
1189                 return generic_file_write_iter(iocb, from);
1190         }
1191
1192         inode_lock(inode);
1193
1194         /* We can write back this queue in page reclaim */
1195         current->backing_dev_info = inode_to_bdi(inode);
1196
1197         err = generic_write_checks(iocb, from);
1198         if (err <= 0)
1199                 goto out;
1200
1201         err = file_remove_privs(file);
1202         if (err)
1203                 goto out;
1204
1205         err = file_update_time(file);
1206         if (err)
1207                 goto out;
1208
1209         if (iocb->ki_flags & IOCB_DIRECT) {
1210                 loff_t pos = iocb->ki_pos;
1211                 written = generic_file_direct_write(iocb, from);
1212                 if (written < 0 || !iov_iter_count(from))
1213                         goto out;
1214
1215                 pos += written;
1216
1217                 written_buffered = fuse_perform_write(file, mapping, from, pos);
1218                 if (written_buffered < 0) {
1219                         err = written_buffered;
1220                         goto out;
1221                 }
1222                 endbyte = pos + written_buffered - 1;
1223
1224                 err = filemap_write_and_wait_range(file->f_mapping, pos,
1225                                                    endbyte);
1226                 if (err)
1227                         goto out;
1228
1229                 invalidate_mapping_pages(file->f_mapping,
1230                                          pos >> PAGE_SHIFT,
1231                                          endbyte >> PAGE_SHIFT);
1232
1233                 written += written_buffered;
1234                 iocb->ki_pos = pos + written_buffered;
1235         } else {
1236                 written = fuse_perform_write(file, mapping, from, iocb->ki_pos);
1237                 if (written >= 0)
1238                         iocb->ki_pos += written;
1239         }
1240 out:
1241         current->backing_dev_info = NULL;
1242         inode_unlock(inode);
1243
1244         return written ? written : err;
1245 }
1246
1247 static inline void fuse_page_descs_length_init(struct fuse_req *req,
1248                 unsigned index, unsigned nr_pages)
1249 {
1250         int i;
1251
1252         for (i = index; i < index + nr_pages; i++)
1253                 req->page_descs[i].length = PAGE_SIZE -
1254                         req->page_descs[i].offset;
1255 }
1256
1257 static inline unsigned long fuse_get_user_addr(const struct iov_iter *ii)
1258 {
1259         return (unsigned long)ii->iov->iov_base + ii->iov_offset;
1260 }
1261
1262 static inline size_t fuse_get_frag_size(const struct iov_iter *ii,
1263                                         size_t max_size)
1264 {
1265         return min(iov_iter_single_seg_count(ii), max_size);
1266 }
1267
1268 static int fuse_get_user_pages(struct fuse_req *req, struct iov_iter *ii,
1269                                size_t *nbytesp, int write)
1270 {
1271         size_t nbytes = 0;  /* # bytes already packed in req */
1272         ssize_t ret = 0;
1273
1274         /* Special case for kernel I/O: can copy directly into the buffer */
1275         if (ii->type & ITER_KVEC) {
1276                 unsigned long user_addr = fuse_get_user_addr(ii);
1277                 size_t frag_size = fuse_get_frag_size(ii, *nbytesp);
1278
1279                 if (write)
1280                         req->in.args[1].value = (void *) user_addr;
1281                 else
1282                         req->out.args[0].value = (void *) user_addr;
1283
1284                 iov_iter_advance(ii, frag_size);
1285                 *nbytesp = frag_size;
1286                 return 0;
1287         }
1288
1289         while (nbytes < *nbytesp && req->num_pages < req->max_pages) {
1290                 unsigned npages;
1291                 size_t start;
1292                 ret = iov_iter_get_pages(ii, &req->pages[req->num_pages],
1293                                         *nbytesp - nbytes,
1294                                         req->max_pages - req->num_pages,
1295                                         &start);
1296                 if (ret < 0)
1297                         break;
1298
1299                 iov_iter_advance(ii, ret);
1300                 nbytes += ret;
1301
1302                 ret += start;
1303                 npages = (ret + PAGE_SIZE - 1) / PAGE_SIZE;
1304
1305                 req->page_descs[req->num_pages].offset = start;
1306                 fuse_page_descs_length_init(req, req->num_pages, npages);
1307
1308                 req->num_pages += npages;
1309                 req->page_descs[req->num_pages - 1].length -=
1310                         (PAGE_SIZE - ret) & (PAGE_SIZE - 1);
1311         }
1312
1313         if (write)
1314                 req->in.argpages = 1;
1315         else
1316                 req->out.argpages = 1;
1317
1318         *nbytesp = nbytes;
1319
1320         return ret < 0 ? ret : 0;
1321 }
1322
1323 static inline int fuse_iter_npages(const struct iov_iter *ii_p)
1324 {
1325         return iov_iter_npages(ii_p, FUSE_MAX_PAGES_PER_REQ);
1326 }
1327
1328 ssize_t fuse_direct_io(struct fuse_io_priv *io, struct iov_iter *iter,
1329                        loff_t *ppos, int flags)
1330 {
1331         int write = flags & FUSE_DIO_WRITE;
1332         bool should_dirty = !write && iter_is_iovec(iter);
1333         int cuse = flags & FUSE_DIO_CUSE;
1334         struct file *file = io->file;
1335         struct inode *inode = file->f_mapping->host;
1336         struct fuse_file *ff = file->private_data;
1337         struct fuse_conn *fc = ff->fc;
1338         size_t nmax = write ? fc->max_write : fc->max_read;
1339         loff_t pos = *ppos;
1340         size_t count = iov_iter_count(iter);
1341         pgoff_t idx_from = pos >> PAGE_SHIFT;
1342         pgoff_t idx_to = (pos + count - 1) >> PAGE_SHIFT;
1343         ssize_t res = 0;
1344         struct fuse_req *req;
1345         int err = 0;
1346
1347         if (io->async)
1348                 req = fuse_get_req_for_background(fc, fuse_iter_npages(iter));
1349         else
1350                 req = fuse_get_req(fc, fuse_iter_npages(iter));
1351         if (IS_ERR(req))
1352                 return PTR_ERR(req);
1353
1354         if (!cuse && fuse_range_is_writeback(inode, idx_from, idx_to)) {
1355                 if (!write)
1356                         inode_lock(inode);
1357                 fuse_sync_writes(inode);
1358                 if (!write)
1359                         inode_unlock(inode);
1360         }
1361
1362         while (count) {
1363                 size_t nres;
1364                 fl_owner_t owner = current->files;
1365                 size_t nbytes = min(count, nmax);
1366                 err = fuse_get_user_pages(req, iter, &nbytes, write);
1367                 if (err && !nbytes)
1368                         break;
1369
1370                 if (write)
1371                         nres = fuse_send_write(req, io, pos, nbytes, owner);
1372                 else
1373                         nres = fuse_send_read(req, io, pos, nbytes, owner);
1374
1375                 if (!io->async)
1376                         fuse_release_user_pages(req, should_dirty);
1377                 if (req->out.h.error) {
1378                         err = req->out.h.error;
1379                         break;
1380                 } else if (nres > nbytes) {
1381                         res = 0;
1382                         err = -EIO;
1383                         break;
1384                 }
1385                 count -= nres;
1386                 res += nres;
1387                 pos += nres;
1388                 if (nres != nbytes)
1389                         break;
1390                 if (count) {
1391                         fuse_put_request(fc, req);
1392                         if (io->async)
1393                                 req = fuse_get_req_for_background(fc,
1394                                         fuse_iter_npages(iter));
1395                         else
1396                                 req = fuse_get_req(fc, fuse_iter_npages(iter));
1397                         if (IS_ERR(req))
1398                                 break;
1399                 }
1400         }
1401         if (!IS_ERR(req))
1402                 fuse_put_request(fc, req);
1403         if (res > 0)
1404                 *ppos = pos;
1405
1406         return res > 0 ? res : err;
1407 }
1408 EXPORT_SYMBOL_GPL(fuse_direct_io);
1409
1410 static ssize_t __fuse_direct_read(struct fuse_io_priv *io,
1411                                   struct iov_iter *iter,
1412                                   loff_t *ppos)
1413 {
1414         ssize_t res;
1415         struct file *file = io->file;
1416         struct inode *inode = file_inode(file);
1417
1418         if (is_bad_inode(inode))
1419                 return -EIO;
1420
1421         res = fuse_direct_io(io, iter, ppos, 0);
1422
1423         fuse_invalidate_attr(inode);
1424
1425         return res;
1426 }
1427
1428 static ssize_t fuse_direct_read_iter(struct kiocb *iocb, struct iov_iter *to)
1429 {
1430         struct fuse_io_priv io = FUSE_IO_PRIV_SYNC(iocb->ki_filp);
1431         return __fuse_direct_read(&io, to, &iocb->ki_pos);
1432 }
1433
1434 static ssize_t fuse_direct_write_iter(struct kiocb *iocb, struct iov_iter *from)
1435 {
1436         struct file *file = iocb->ki_filp;
1437         struct inode *inode = file_inode(file);
1438         struct fuse_io_priv io = FUSE_IO_PRIV_SYNC(file);
1439         ssize_t res;
1440
1441         if (is_bad_inode(inode))
1442                 return -EIO;
1443
1444         /* Don't allow parallel writes to the same file */
1445         inode_lock(inode);
1446         res = generic_write_checks(iocb, from);
1447         if (res > 0)
1448                 res = fuse_direct_io(&io, from, &iocb->ki_pos, FUSE_DIO_WRITE);
1449         fuse_invalidate_attr(inode);
1450         if (res > 0)
1451                 fuse_write_update_size(inode, iocb->ki_pos);
1452         inode_unlock(inode);
1453
1454         return res;
1455 }
1456
1457 static void fuse_writepage_free(struct fuse_conn *fc, struct fuse_req *req)
1458 {
1459         int i;
1460
1461         for (i = 0; i < req->num_pages; i++)
1462                 __free_page(req->pages[i]);
1463
1464         if (req->ff)
1465                 fuse_file_put(req->ff, false);
1466 }
1467
1468 static void fuse_writepage_finish(struct fuse_conn *fc, struct fuse_req *req)
1469 {
1470         struct inode *inode = req->inode;
1471         struct fuse_inode *fi = get_fuse_inode(inode);
1472         struct backing_dev_info *bdi = inode_to_bdi(inode);
1473         int i;
1474
1475         list_del(&req->writepages_entry);
1476         for (i = 0; i < req->num_pages; i++) {
1477                 dec_wb_stat(&bdi->wb, WB_WRITEBACK);
1478                 dec_node_page_state(req->pages[i], NR_WRITEBACK_TEMP);
1479                 wb_writeout_inc(&bdi->wb);
1480         }
1481         wake_up(&fi->page_waitq);
1482 }
1483
1484 /* Called under fc->lock, may release and reacquire it */
1485 static void fuse_send_writepage(struct fuse_conn *fc, struct fuse_req *req,
1486                                 loff_t size)
1487 __releases(fc->lock)
1488 __acquires(fc->lock)
1489 {
1490         struct fuse_inode *fi = get_fuse_inode(req->inode);
1491         struct fuse_write_in *inarg = &req->misc.write.in;
1492         __u64 data_size = req->num_pages * PAGE_SIZE;
1493
1494         if (!fc->connected)
1495                 goto out_free;
1496
1497         if (inarg->offset + data_size <= size) {
1498                 inarg->size = data_size;
1499         } else if (inarg->offset < size) {
1500                 inarg->size = size - inarg->offset;
1501         } else {
1502                 /* Got truncated off completely */
1503                 goto out_free;
1504         }
1505
1506         req->in.args[1].size = inarg->size;
1507         fi->writectr++;
1508         fuse_request_send_background_locked(fc, req);
1509         return;
1510
1511  out_free:
1512         fuse_writepage_finish(fc, req);
1513         spin_unlock(&fc->lock);
1514         fuse_writepage_free(fc, req);
1515         fuse_put_request(fc, req);
1516         spin_lock(&fc->lock);
1517 }
1518
1519 /*
1520  * If fi->writectr is positive (no truncate or fsync going on) send
1521  * all queued writepage requests.
1522  *
1523  * Called with fc->lock
1524  */
1525 void fuse_flush_writepages(struct inode *inode)
1526 __releases(fc->lock)
1527 __acquires(fc->lock)
1528 {
1529         struct fuse_conn *fc = get_fuse_conn(inode);
1530         struct fuse_inode *fi = get_fuse_inode(inode);
1531         loff_t crop = i_size_read(inode);
1532         struct fuse_req *req;
1533
1534         while (fi->writectr >= 0 && !list_empty(&fi->queued_writes)) {
1535                 req = list_entry(fi->queued_writes.next, struct fuse_req, list);
1536                 list_del_init(&req->list);
1537                 fuse_send_writepage(fc, req, crop);
1538         }
1539 }
1540
1541 static void fuse_writepage_end(struct fuse_conn *fc, struct fuse_req *req)
1542 {
1543         struct inode *inode = req->inode;
1544         struct fuse_inode *fi = get_fuse_inode(inode);
1545
1546         mapping_set_error(inode->i_mapping, req->out.h.error);
1547         spin_lock(&fc->lock);
1548         while (req->misc.write.next) {
1549                 struct fuse_conn *fc = get_fuse_conn(inode);
1550                 struct fuse_write_in *inarg = &req->misc.write.in;
1551                 struct fuse_req *next = req->misc.write.next;
1552                 req->misc.write.next = next->misc.write.next;
1553                 next->misc.write.next = NULL;
1554                 next->ff = fuse_file_get(req->ff);
1555                 list_add(&next->writepages_entry, &fi->writepages);
1556
1557                 /*
1558                  * Skip fuse_flush_writepages() to make it easy to crop requests
1559                  * based on primary request size.
1560                  *
1561                  * 1st case (trivial): there are no concurrent activities using
1562                  * fuse_set/release_nowrite.  Then we're on safe side because
1563                  * fuse_flush_writepages() would call fuse_send_writepage()
1564                  * anyway.
1565                  *
1566                  * 2nd case: someone called fuse_set_nowrite and it is waiting
1567                  * now for completion of all in-flight requests.  This happens
1568                  * rarely and no more than once per page, so this should be
1569                  * okay.
1570                  *
1571                  * 3rd case: someone (e.g. fuse_do_setattr()) is in the middle
1572                  * of fuse_set_nowrite..fuse_release_nowrite section.  The fact
1573                  * that fuse_set_nowrite returned implies that all in-flight
1574                  * requests were completed along with all of their secondary
1575                  * requests.  Further primary requests are blocked by negative
1576                  * writectr.  Hence there cannot be any in-flight requests and
1577                  * no invocations of fuse_writepage_end() while we're in
1578                  * fuse_set_nowrite..fuse_release_nowrite section.
1579                  */
1580                 fuse_send_writepage(fc, next, inarg->offset + inarg->size);
1581         }
1582         fi->writectr--;
1583         fuse_writepage_finish(fc, req);
1584         spin_unlock(&fc->lock);
1585         fuse_writepage_free(fc, req);
1586 }
1587
1588 static struct fuse_file *__fuse_write_file_get(struct fuse_conn *fc,
1589                                                struct fuse_inode *fi)
1590 {
1591         struct fuse_file *ff = NULL;
1592
1593         spin_lock(&fc->lock);
1594         if (!list_empty(&fi->write_files)) {
1595                 ff = list_entry(fi->write_files.next, struct fuse_file,
1596                                 write_entry);
1597                 fuse_file_get(ff);
1598         }
1599         spin_unlock(&fc->lock);
1600
1601         return ff;
1602 }
1603
1604 static struct fuse_file *fuse_write_file_get(struct fuse_conn *fc,
1605                                              struct fuse_inode *fi)
1606 {
1607         struct fuse_file *ff = __fuse_write_file_get(fc, fi);
1608         WARN_ON(!ff);
1609         return ff;
1610 }
1611
1612 int fuse_write_inode(struct inode *inode, struct writeback_control *wbc)
1613 {
1614         struct fuse_conn *fc = get_fuse_conn(inode);
1615         struct fuse_inode *fi = get_fuse_inode(inode);
1616         struct fuse_file *ff;
1617         int err;
1618
1619         ff = __fuse_write_file_get(fc, fi);
1620         err = fuse_flush_times(inode, ff);
1621         if (ff)
1622                 fuse_file_put(ff, 0);
1623
1624         return err;
1625 }
1626
1627 static int fuse_writepage_locked(struct page *page)
1628 {
1629         struct address_space *mapping = page->mapping;
1630         struct inode *inode = mapping->host;
1631         struct fuse_conn *fc = get_fuse_conn(inode);
1632         struct fuse_inode *fi = get_fuse_inode(inode);
1633         struct fuse_req *req;
1634         struct page *tmp_page;
1635         int error = -ENOMEM;
1636
1637         set_page_writeback(page);
1638
1639         req = fuse_request_alloc_nofs(1);
1640         if (!req)
1641                 goto err;
1642
1643         /* writeback always goes to bg_queue */
1644         __set_bit(FR_BACKGROUND, &req->flags);
1645         tmp_page = alloc_page(GFP_NOFS | __GFP_HIGHMEM);
1646         if (!tmp_page)
1647                 goto err_free;
1648
1649         error = -EIO;
1650         req->ff = fuse_write_file_get(fc, fi);
1651         if (!req->ff)
1652                 goto err_nofile;
1653
1654         fuse_write_fill(req, req->ff, page_offset(page), 0);
1655
1656         copy_highpage(tmp_page, page);
1657         req->misc.write.in.write_flags |= FUSE_WRITE_CACHE;
1658         req->misc.write.next = NULL;
1659         req->in.argpages = 1;
1660         req->num_pages = 1;
1661         req->pages[0] = tmp_page;
1662         req->page_descs[0].offset = 0;
1663         req->page_descs[0].length = PAGE_SIZE;
1664         req->end = fuse_writepage_end;
1665         req->inode = inode;
1666
1667         inc_wb_stat(&inode_to_bdi(inode)->wb, WB_WRITEBACK);
1668         inc_node_page_state(tmp_page, NR_WRITEBACK_TEMP);
1669
1670         spin_lock(&fc->lock);
1671         list_add(&req->writepages_entry, &fi->writepages);
1672         list_add_tail(&req->list, &fi->queued_writes);
1673         fuse_flush_writepages(inode);
1674         spin_unlock(&fc->lock);
1675
1676         end_page_writeback(page);
1677
1678         return 0;
1679
1680 err_nofile:
1681         __free_page(tmp_page);
1682 err_free:
1683         fuse_request_free(req);
1684 err:
1685         end_page_writeback(page);
1686         return error;
1687 }
1688
1689 static int fuse_writepage(struct page *page, struct writeback_control *wbc)
1690 {
1691         int err;
1692
1693         if (fuse_page_is_writeback(page->mapping->host, page->index)) {
1694                 /*
1695                  * ->writepages() should be called for sync() and friends.  We
1696                  * should only get here on direct reclaim and then we are
1697                  * allowed to skip a page which is already in flight
1698                  */
1699                 WARN_ON(wbc->sync_mode == WB_SYNC_ALL);
1700
1701                 redirty_page_for_writepage(wbc, page);
1702                 unlock_page(page);
1703                 return 0;
1704         }
1705
1706         err = fuse_writepage_locked(page);
1707         unlock_page(page);
1708
1709         return err;
1710 }
1711
1712 struct fuse_fill_wb_data {
1713         struct fuse_req *req;
1714         struct fuse_file *ff;
1715         struct inode *inode;
1716         struct page **orig_pages;
1717 };
1718
1719 static void fuse_writepages_send(struct fuse_fill_wb_data *data)
1720 {
1721         struct fuse_req *req = data->req;
1722         struct inode *inode = data->inode;
1723         struct fuse_conn *fc = get_fuse_conn(inode);
1724         struct fuse_inode *fi = get_fuse_inode(inode);
1725         int num_pages = req->num_pages;
1726         int i;
1727
1728         req->ff = fuse_file_get(data->ff);
1729         spin_lock(&fc->lock);
1730         list_add_tail(&req->list, &fi->queued_writes);
1731         fuse_flush_writepages(inode);
1732         spin_unlock(&fc->lock);
1733
1734         for (i = 0; i < num_pages; i++)
1735                 end_page_writeback(data->orig_pages[i]);
1736 }
1737
1738 static bool fuse_writepage_in_flight(struct fuse_req *new_req,
1739                                      struct page *page)
1740 {
1741         struct fuse_conn *fc = get_fuse_conn(new_req->inode);
1742         struct fuse_inode *fi = get_fuse_inode(new_req->inode);
1743         struct fuse_req *tmp;
1744         struct fuse_req *old_req;
1745         bool found = false;
1746         pgoff_t curr_index;
1747
1748         BUG_ON(new_req->num_pages != 0);
1749
1750         spin_lock(&fc->lock);
1751         list_del(&new_req->writepages_entry);
1752         list_for_each_entry(old_req, &fi->writepages, writepages_entry) {
1753                 BUG_ON(old_req->inode != new_req->inode);
1754                 curr_index = old_req->misc.write.in.offset >> PAGE_SHIFT;
1755                 if (curr_index <= page->index &&
1756                     page->index < curr_index + old_req->num_pages) {
1757                         found = true;
1758                         break;
1759                 }
1760         }
1761         if (!found) {
1762                 list_add(&new_req->writepages_entry, &fi->writepages);
1763                 goto out_unlock;
1764         }
1765
1766         new_req->num_pages = 1;
1767         for (tmp = old_req; tmp != NULL; tmp = tmp->misc.write.next) {
1768                 BUG_ON(tmp->inode != new_req->inode);
1769                 curr_index = tmp->misc.write.in.offset >> PAGE_SHIFT;
1770                 if (tmp->num_pages == 1 &&
1771                     curr_index == page->index) {
1772                         old_req = tmp;
1773                 }
1774         }
1775
1776         if (old_req->num_pages == 1 && test_bit(FR_PENDING, &old_req->flags)) {
1777                 struct backing_dev_info *bdi = inode_to_bdi(page->mapping->host);
1778
1779                 copy_highpage(old_req->pages[0], page);
1780                 spin_unlock(&fc->lock);
1781
1782                 dec_wb_stat(&bdi->wb, WB_WRITEBACK);
1783                 dec_node_page_state(new_req->pages[0], NR_WRITEBACK_TEMP);
1784                 wb_writeout_inc(&bdi->wb);
1785                 fuse_writepage_free(fc, new_req);
1786                 fuse_request_free(new_req);
1787                 goto out;
1788         } else {
1789                 new_req->misc.write.next = old_req->misc.write.next;
1790                 old_req->misc.write.next = new_req;
1791         }
1792 out_unlock:
1793         spin_unlock(&fc->lock);
1794 out:
1795         return found;
1796 }
1797
1798 static int fuse_writepages_fill(struct page *page,
1799                 struct writeback_control *wbc, void *_data)
1800 {
1801         struct fuse_fill_wb_data *data = _data;
1802         struct fuse_req *req = data->req;
1803         struct inode *inode = data->inode;
1804         struct fuse_conn *fc = get_fuse_conn(inode);
1805         struct page *tmp_page;
1806         bool is_writeback;
1807         int err;
1808
1809         if (!data->ff) {
1810                 err = -EIO;
1811                 data->ff = fuse_write_file_get(fc, get_fuse_inode(inode));
1812                 if (!data->ff)
1813                         goto out_unlock;
1814         }
1815
1816         /*
1817          * Being under writeback is unlikely but possible.  For example direct
1818          * read to an mmaped fuse file will set the page dirty twice; once when
1819          * the pages are faulted with get_user_pages(), and then after the read
1820          * completed.
1821          */
1822         is_writeback = fuse_page_is_writeback(inode, page->index);
1823
1824         if (req && req->num_pages &&
1825             (is_writeback || req->num_pages == FUSE_MAX_PAGES_PER_REQ ||
1826              (req->num_pages + 1) * PAGE_SIZE > fc->max_write ||
1827              data->orig_pages[req->num_pages - 1]->index + 1 != page->index)) {
1828                 fuse_writepages_send(data);
1829                 data->req = NULL;
1830         }
1831         err = -ENOMEM;
1832         tmp_page = alloc_page(GFP_NOFS | __GFP_HIGHMEM);
1833         if (!tmp_page)
1834                 goto out_unlock;
1835
1836         /*
1837          * The page must not be redirtied until the writeout is completed
1838          * (i.e. userspace has sent a reply to the write request).  Otherwise
1839          * there could be more than one temporary page instance for each real
1840          * page.
1841          *
1842          * This is ensured by holding the page lock in page_mkwrite() while
1843          * checking fuse_page_is_writeback().  We already hold the page lock
1844          * since clear_page_dirty_for_io() and keep it held until we add the
1845          * request to the fi->writepages list and increment req->num_pages.
1846          * After this fuse_page_is_writeback() will indicate that the page is
1847          * under writeback, so we can release the page lock.
1848          */
1849         if (data->req == NULL) {
1850                 struct fuse_inode *fi = get_fuse_inode(inode);
1851
1852                 err = -ENOMEM;
1853                 req = fuse_request_alloc_nofs(FUSE_MAX_PAGES_PER_REQ);
1854                 if (!req) {
1855                         __free_page(tmp_page);
1856                         goto out_unlock;
1857                 }
1858
1859                 fuse_write_fill(req, data->ff, page_offset(page), 0);
1860                 req->misc.write.in.write_flags |= FUSE_WRITE_CACHE;
1861                 req->misc.write.next = NULL;
1862                 req->in.argpages = 1;
1863                 __set_bit(FR_BACKGROUND, &req->flags);
1864                 req->num_pages = 0;
1865                 req->end = fuse_writepage_end;
1866                 req->inode = inode;
1867
1868                 spin_lock(&fc->lock);
1869                 list_add(&req->writepages_entry, &fi->writepages);
1870                 spin_unlock(&fc->lock);
1871
1872                 data->req = req;
1873         }
1874         set_page_writeback(page);
1875
1876         copy_highpage(tmp_page, page);
1877         req->pages[req->num_pages] = tmp_page;
1878         req->page_descs[req->num_pages].offset = 0;
1879         req->page_descs[req->num_pages].length = PAGE_SIZE;
1880
1881         inc_wb_stat(&inode_to_bdi(inode)->wb, WB_WRITEBACK);
1882         inc_node_page_state(tmp_page, NR_WRITEBACK_TEMP);
1883
1884         err = 0;
1885         if (is_writeback && fuse_writepage_in_flight(req, page)) {
1886                 end_page_writeback(page);
1887                 data->req = NULL;
1888                 goto out_unlock;
1889         }
1890         data->orig_pages[req->num_pages] = page;
1891
1892         /*
1893          * Protected by fc->lock against concurrent access by
1894          * fuse_page_is_writeback().
1895          */
1896         spin_lock(&fc->lock);
1897         req->num_pages++;
1898         spin_unlock(&fc->lock);
1899
1900 out_unlock:
1901         unlock_page(page);
1902
1903         return err;
1904 }
1905
1906 static int fuse_writepages(struct address_space *mapping,
1907                            struct writeback_control *wbc)
1908 {
1909         struct inode *inode = mapping->host;
1910         struct fuse_fill_wb_data data;
1911         int err;
1912
1913         err = -EIO;
1914         if (is_bad_inode(inode))
1915                 goto out;
1916
1917         data.inode = inode;
1918         data.req = NULL;
1919         data.ff = NULL;
1920
1921         err = -ENOMEM;
1922         data.orig_pages = kcalloc(FUSE_MAX_PAGES_PER_REQ,
1923                                   sizeof(struct page *),
1924                                   GFP_NOFS);
1925         if (!data.orig_pages)
1926                 goto out;
1927
1928         err = write_cache_pages(mapping, wbc, fuse_writepages_fill, &data);
1929         if (data.req) {
1930                 /* Ignore errors if we can write at least one page */
1931                 BUG_ON(!data.req->num_pages);
1932                 fuse_writepages_send(&data);
1933                 err = 0;
1934         }
1935         if (data.ff)
1936                 fuse_file_put(data.ff, false);
1937
1938         kfree(data.orig_pages);
1939 out:
1940         return err;
1941 }
1942
1943 /*
1944  * It's worthy to make sure that space is reserved on disk for the write,
1945  * but how to implement it without killing performance need more thinking.
1946  */
1947 static int fuse_write_begin(struct file *file, struct address_space *mapping,
1948                 loff_t pos, unsigned len, unsigned flags,
1949                 struct page **pagep, void **fsdata)
1950 {
1951         pgoff_t index = pos >> PAGE_SHIFT;
1952         struct fuse_conn *fc = get_fuse_conn(file_inode(file));
1953         struct page *page;
1954         loff_t fsize;
1955         int err = -ENOMEM;
1956
1957         WARN_ON(!fc->writeback_cache);
1958
1959         page = grab_cache_page_write_begin(mapping, index, flags);
1960         if (!page)
1961                 goto error;
1962
1963         fuse_wait_on_page_writeback(mapping->host, page->index);
1964
1965         if (PageUptodate(page) || len == PAGE_SIZE)
1966                 goto success;
1967         /*
1968          * Check if the start this page comes after the end of file, in which
1969          * case the readpage can be optimized away.
1970          */
1971         fsize = i_size_read(mapping->host);
1972         if (fsize <= (pos & PAGE_MASK)) {
1973                 size_t off = pos & ~PAGE_MASK;
1974                 if (off)
1975                         zero_user_segment(page, 0, off);
1976                 goto success;
1977         }
1978         err = fuse_do_readpage(file, page);
1979         if (err)
1980                 goto cleanup;
1981 success:
1982         *pagep = page;
1983         return 0;
1984
1985 cleanup:
1986         unlock_page(page);
1987         put_page(page);
1988 error:
1989         return err;
1990 }
1991
1992 static int fuse_write_end(struct file *file, struct address_space *mapping,
1993                 loff_t pos, unsigned len, unsigned copied,
1994                 struct page *page, void *fsdata)
1995 {
1996         struct inode *inode = page->mapping->host;
1997
1998         /* Haven't copied anything?  Skip zeroing, size extending, dirtying. */
1999         if (!copied)
2000                 goto unlock;
2001
2002         if (!PageUptodate(page)) {
2003                 /* Zero any unwritten bytes at the end of the page */
2004                 size_t endoff = (pos + copied) & ~PAGE_MASK;
2005                 if (endoff)
2006                         zero_user_segment(page, endoff, PAGE_SIZE);
2007                 SetPageUptodate(page);
2008         }
2009
2010         fuse_write_update_size(inode, pos + copied);
2011         set_page_dirty(page);
2012
2013 unlock:
2014         unlock_page(page);
2015         put_page(page);
2016
2017         return copied;
2018 }
2019
2020 static int fuse_launder_page(struct page *page)
2021 {
2022         int err = 0;
2023         if (clear_page_dirty_for_io(page)) {
2024                 struct inode *inode = page->mapping->host;
2025                 err = fuse_writepage_locked(page);
2026                 if (!err)
2027                         fuse_wait_on_page_writeback(inode, page->index);
2028         }
2029         return err;
2030 }
2031
2032 /*
2033  * Write back dirty pages now, because there may not be any suitable
2034  * open files later
2035  */
2036 static void fuse_vma_close(struct vm_area_struct *vma)
2037 {
2038         filemap_write_and_wait(vma->vm_file->f_mapping);
2039 }
2040
2041 /*
2042  * Wait for writeback against this page to complete before allowing it
2043  * to be marked dirty again, and hence written back again, possibly
2044  * before the previous writepage completed.
2045  *
2046  * Block here, instead of in ->writepage(), so that the userspace fs
2047  * can only block processes actually operating on the filesystem.
2048  *
2049  * Otherwise unprivileged userspace fs would be able to block
2050  * unrelated:
2051  *
2052  * - page migration
2053  * - sync(2)
2054  * - try_to_free_pages() with order > PAGE_ALLOC_COSTLY_ORDER
2055  */
2056 static int fuse_page_mkwrite(struct vm_area_struct *vma, struct vm_fault *vmf)
2057 {
2058         struct page *page = vmf->page;
2059         struct inode *inode = file_inode(vma->vm_file);
2060
2061         file_update_time(vma->vm_file);
2062         lock_page(page);
2063         if (page->mapping != inode->i_mapping) {
2064                 unlock_page(page);
2065                 return VM_FAULT_NOPAGE;
2066         }
2067
2068         fuse_wait_on_page_writeback(inode, page->index);
2069         return VM_FAULT_LOCKED;
2070 }
2071
2072 static const struct vm_operations_struct fuse_file_vm_ops = {
2073         .close          = fuse_vma_close,
2074         .fault          = filemap_fault,
2075         .map_pages      = filemap_map_pages,
2076         .page_mkwrite   = fuse_page_mkwrite,
2077 };
2078
2079 static int fuse_file_mmap(struct file *file, struct vm_area_struct *vma)
2080 {
2081         if ((vma->vm_flags & VM_SHARED) && (vma->vm_flags & VM_MAYWRITE))
2082                 fuse_link_write_file(file);
2083
2084         file_accessed(file);
2085         vma->vm_ops = &fuse_file_vm_ops;
2086         return 0;
2087 }
2088
2089 static int fuse_direct_mmap(struct file *file, struct vm_area_struct *vma)
2090 {
2091         /* Can't provide the coherency needed for MAP_SHARED */
2092         if (vma->vm_flags & VM_MAYSHARE)
2093                 return -ENODEV;
2094
2095         invalidate_inode_pages2(file->f_mapping);
2096
2097         return generic_file_mmap(file, vma);
2098 }
2099
2100 static int convert_fuse_file_lock(const struct fuse_file_lock *ffl,
2101                                   struct file_lock *fl)
2102 {
2103         switch (ffl->type) {
2104         case F_UNLCK:
2105                 break;
2106
2107         case F_RDLCK:
2108         case F_WRLCK:
2109                 if (ffl->start > OFFSET_MAX || ffl->end > OFFSET_MAX ||
2110                     ffl->end < ffl->start)
2111                         return -EIO;
2112
2113                 fl->fl_start = ffl->start;
2114                 fl->fl_end = ffl->end;
2115                 fl->fl_pid = ffl->pid;
2116                 break;
2117
2118         default:
2119                 return -EIO;
2120         }
2121         fl->fl_type = ffl->type;
2122         return 0;
2123 }
2124
2125 static void fuse_lk_fill(struct fuse_args *args, struct file *file,
2126                          const struct file_lock *fl, int opcode, pid_t pid,
2127                          int flock, struct fuse_lk_in *inarg)
2128 {
2129         struct inode *inode = file_inode(file);
2130         struct fuse_conn *fc = get_fuse_conn(inode);
2131         struct fuse_file *ff = file->private_data;
2132
2133         memset(inarg, 0, sizeof(*inarg));
2134         inarg->fh = ff->fh;
2135         inarg->owner = fuse_lock_owner_id(fc, fl->fl_owner);
2136         inarg->lk.start = fl->fl_start;
2137         inarg->lk.end = fl->fl_end;
2138         inarg->lk.type = fl->fl_type;
2139         inarg->lk.pid = pid;
2140         if (flock)
2141                 inarg->lk_flags |= FUSE_LK_FLOCK;
2142         args->in.h.opcode = opcode;
2143         args->in.h.nodeid = get_node_id(inode);
2144         args->in.numargs = 1;
2145         args->in.args[0].size = sizeof(*inarg);
2146         args->in.args[0].value = inarg;
2147 }
2148
2149 static int fuse_getlk(struct file *file, struct file_lock *fl)
2150 {
2151         struct inode *inode = file_inode(file);
2152         struct fuse_conn *fc = get_fuse_conn(inode);
2153         FUSE_ARGS(args);
2154         struct fuse_lk_in inarg;
2155         struct fuse_lk_out outarg;
2156         int err;
2157
2158         fuse_lk_fill(&args, file, fl, FUSE_GETLK, 0, 0, &inarg);
2159         args.out.numargs = 1;
2160         args.out.args[0].size = sizeof(outarg);
2161         args.out.args[0].value = &outarg;
2162         err = fuse_simple_request(fc, &args);
2163         if (!err)
2164                 err = convert_fuse_file_lock(&outarg.lk, fl);
2165
2166         return err;
2167 }
2168
2169 static int fuse_setlk(struct file *file, struct file_lock *fl, int flock)
2170 {
2171         struct inode *inode = file_inode(file);
2172         struct fuse_conn *fc = get_fuse_conn(inode);
2173         FUSE_ARGS(args);
2174         struct fuse_lk_in inarg;
2175         int opcode = (fl->fl_flags & FL_SLEEP) ? FUSE_SETLKW : FUSE_SETLK;
2176         pid_t pid = fl->fl_type != F_UNLCK ? current->tgid : 0;
2177         int err;
2178
2179         if (fl->fl_lmops && fl->fl_lmops->lm_grant) {
2180                 /* NLM needs asynchronous locks, which we don't support yet */
2181                 return -ENOLCK;
2182         }
2183
2184         /* Unlock on close is handled by the flush method */
2185         if (fl->fl_flags & FL_CLOSE)
2186                 return 0;
2187
2188         fuse_lk_fill(&args, file, fl, opcode, pid, flock, &inarg);
2189         err = fuse_simple_request(fc, &args);
2190
2191         /* locking is restartable */
2192         if (err == -EINTR)
2193                 err = -ERESTARTSYS;
2194
2195         return err;
2196 }
2197
2198 static int fuse_file_lock(struct file *file, int cmd, struct file_lock *fl)
2199 {
2200         struct inode *inode = file_inode(file);
2201         struct fuse_conn *fc = get_fuse_conn(inode);
2202         int err;
2203
2204         if (cmd == F_CANCELLK) {
2205                 err = 0;
2206         } else if (cmd == F_GETLK) {
2207                 if (fc->no_lock) {
2208                         posix_test_lock(file, fl);
2209                         err = 0;
2210                 } else
2211                         err = fuse_getlk(file, fl);
2212         } else {
2213                 if (fc->no_lock)
2214                         err = posix_lock_file(file, fl, NULL);
2215                 else
2216                         err = fuse_setlk(file, fl, 0);
2217         }
2218         return err;
2219 }
2220
2221 static int fuse_file_flock(struct file *file, int cmd, struct file_lock *fl)
2222 {
2223         struct inode *inode = file_inode(file);
2224         struct fuse_conn *fc = get_fuse_conn(inode);
2225         int err;
2226
2227         if (fc->no_flock) {
2228                 err = locks_lock_file_wait(file, fl);
2229         } else {
2230                 struct fuse_file *ff = file->private_data;
2231
2232                 /* emulate flock with POSIX locks */
2233                 ff->flock = true;
2234                 err = fuse_setlk(file, fl, 1);
2235         }
2236
2237         return err;
2238 }
2239
2240 static sector_t fuse_bmap(struct address_space *mapping, sector_t block)
2241 {
2242         struct inode *inode = mapping->host;
2243         struct fuse_conn *fc = get_fuse_conn(inode);
2244         FUSE_ARGS(args);
2245         struct fuse_bmap_in inarg;
2246         struct fuse_bmap_out outarg;
2247         int err;
2248
2249         if (!inode->i_sb->s_bdev || fc->no_bmap)
2250                 return 0;
2251
2252         memset(&inarg, 0, sizeof(inarg));
2253         inarg.block = block;
2254         inarg.blocksize = inode->i_sb->s_blocksize;
2255         args.in.h.opcode = FUSE_BMAP;
2256         args.in.h.nodeid = get_node_id(inode);
2257         args.in.numargs = 1;
2258         args.in.args[0].size = sizeof(inarg);
2259         args.in.args[0].value = &inarg;
2260         args.out.numargs = 1;
2261         args.out.args[0].size = sizeof(outarg);
2262         args.out.args[0].value = &outarg;
2263         err = fuse_simple_request(fc, &args);
2264         if (err == -ENOSYS)
2265                 fc->no_bmap = 1;
2266
2267         return err ? 0 : outarg.block;
2268 }
2269
2270 static loff_t fuse_lseek(struct file *file, loff_t offset, int whence)
2271 {
2272         struct inode *inode = file->f_mapping->host;
2273         struct fuse_conn *fc = get_fuse_conn(inode);
2274         struct fuse_file *ff = file->private_data;
2275         FUSE_ARGS(args);
2276         struct fuse_lseek_in inarg = {
2277                 .fh = ff->fh,
2278                 .offset = offset,
2279                 .whence = whence
2280         };
2281         struct fuse_lseek_out outarg;
2282         int err;
2283
2284         if (fc->no_lseek)
2285                 goto fallback;
2286
2287         args.in.h.opcode = FUSE_LSEEK;
2288         args.in.h.nodeid = ff->nodeid;
2289         args.in.numargs = 1;
2290         args.in.args[0].size = sizeof(inarg);
2291         args.in.args[0].value = &inarg;
2292         args.out.numargs = 1;
2293         args.out.args[0].size = sizeof(outarg);
2294         args.out.args[0].value = &outarg;
2295         err = fuse_simple_request(fc, &args);
2296         if (err) {
2297                 if (err == -ENOSYS) {
2298                         fc->no_lseek = 1;
2299                         goto fallback;
2300                 }
2301                 return err;
2302         }
2303
2304         return vfs_setpos(file, outarg.offset, inode->i_sb->s_maxbytes);
2305
2306 fallback:
2307         err = fuse_update_attributes(inode, NULL, file, NULL);
2308         if (!err)
2309                 return generic_file_llseek(file, offset, whence);
2310         else
2311                 return err;
2312 }
2313
2314 static loff_t fuse_file_llseek(struct file *file, loff_t offset, int whence)
2315 {
2316         loff_t retval;
2317         struct inode *inode = file_inode(file);
2318
2319         switch (whence) {
2320         case SEEK_SET:
2321         case SEEK_CUR:
2322                  /* No i_mutex protection necessary for SEEK_CUR and SEEK_SET */
2323                 retval = generic_file_llseek(file, offset, whence);
2324                 break;
2325         case SEEK_END:
2326                 inode_lock(inode);
2327                 retval = fuse_update_attributes(inode, NULL, file, NULL);
2328                 if (!retval)
2329                         retval = generic_file_llseek(file, offset, whence);
2330                 inode_unlock(inode);
2331                 break;
2332         case SEEK_HOLE:
2333         case SEEK_DATA:
2334                 inode_lock(inode);
2335                 retval = fuse_lseek(file, offset, whence);
2336                 inode_unlock(inode);
2337                 break;
2338         default:
2339                 retval = -EINVAL;
2340         }
2341
2342         return retval;
2343 }
2344
2345 /*
2346  * CUSE servers compiled on 32bit broke on 64bit kernels because the
2347  * ABI was defined to be 'struct iovec' which is different on 32bit
2348  * and 64bit.  Fortunately we can determine which structure the server
2349  * used from the size of the reply.
2350  */
2351 static int fuse_copy_ioctl_iovec_old(struct iovec *dst, void *src,
2352                                      size_t transferred, unsigned count,
2353                                      bool is_compat)
2354 {
2355 #ifdef CONFIG_COMPAT
2356         if (count * sizeof(struct compat_iovec) == transferred) {
2357                 struct compat_iovec *ciov = src;
2358                 unsigned i;
2359
2360                 /*
2361                  * With this interface a 32bit server cannot support
2362                  * non-compat (i.e. ones coming from 64bit apps) ioctl
2363                  * requests
2364                  */
2365                 if (!is_compat)
2366                         return -EINVAL;
2367
2368                 for (i = 0; i < count; i++) {
2369                         dst[i].iov_base = compat_ptr(ciov[i].iov_base);
2370                         dst[i].iov_len = ciov[i].iov_len;
2371                 }
2372                 return 0;
2373         }
2374 #endif
2375
2376         if (count * sizeof(struct iovec) != transferred)
2377                 return -EIO;
2378
2379         memcpy(dst, src, transferred);
2380         return 0;
2381 }
2382
2383 /* Make sure iov_length() won't overflow */
2384 static int fuse_verify_ioctl_iov(struct iovec *iov, size_t count)
2385 {
2386         size_t n;
2387         u32 max = FUSE_MAX_PAGES_PER_REQ << PAGE_SHIFT;
2388
2389         for (n = 0; n < count; n++, iov++) {
2390                 if (iov->iov_len > (size_t) max)
2391                         return -ENOMEM;
2392                 max -= iov->iov_len;
2393         }
2394         return 0;
2395 }
2396
2397 static int fuse_copy_ioctl_iovec(struct fuse_conn *fc, struct iovec *dst,
2398                                  void *src, size_t transferred, unsigned count,
2399                                  bool is_compat)
2400 {
2401         unsigned i;
2402         struct fuse_ioctl_iovec *fiov = src;
2403
2404         if (fc->minor < 16) {
2405                 return fuse_copy_ioctl_iovec_old(dst, src, transferred,
2406                                                  count, is_compat);
2407         }
2408
2409         if (count * sizeof(struct fuse_ioctl_iovec) != transferred)
2410                 return -EIO;
2411
2412         for (i = 0; i < count; i++) {
2413                 /* Did the server supply an inappropriate value? */
2414                 if (fiov[i].base != (unsigned long) fiov[i].base ||
2415                     fiov[i].len != (unsigned long) fiov[i].len)
2416                         return -EIO;
2417
2418                 dst[i].iov_base = (void __user *) (unsigned long) fiov[i].base;
2419                 dst[i].iov_len = (size_t) fiov[i].len;
2420
2421 #ifdef CONFIG_COMPAT
2422                 if (is_compat &&
2423                     (ptr_to_compat(dst[i].iov_base) != fiov[i].base ||
2424                      (compat_size_t) dst[i].iov_len != fiov[i].len))
2425                         return -EIO;
2426 #endif
2427         }
2428
2429         return 0;
2430 }
2431
2432
2433 /*
2434  * For ioctls, there is no generic way to determine how much memory
2435  * needs to be read and/or written.  Furthermore, ioctls are allowed
2436  * to dereference the passed pointer, so the parameter requires deep
2437  * copying but FUSE has no idea whatsoever about what to copy in or
2438  * out.
2439  *
2440  * This is solved by allowing FUSE server to retry ioctl with
2441  * necessary in/out iovecs.  Let's assume the ioctl implementation
2442  * needs to read in the following structure.
2443  *
2444  * struct a {
2445  *      char    *buf;
2446  *      size_t  buflen;
2447  * }
2448  *
2449  * On the first callout to FUSE server, inarg->in_size and
2450  * inarg->out_size will be NULL; then, the server completes the ioctl
2451  * with FUSE_IOCTL_RETRY set in out->flags, out->in_iovs set to 1 and
2452  * the actual iov array to
2453  *
2454  * { { .iov_base = inarg.arg,   .iov_len = sizeof(struct a) } }
2455  *
2456  * which tells FUSE to copy in the requested area and retry the ioctl.
2457  * On the second round, the server has access to the structure and
2458  * from that it can tell what to look for next, so on the invocation,
2459  * it sets FUSE_IOCTL_RETRY, out->in_iovs to 2 and iov array to
2460  *
2461  * { { .iov_base = inarg.arg,   .iov_len = sizeof(struct a)     },
2462  *   { .iov_base = a.buf,       .iov_len = a.buflen             } }
2463  *
2464  * FUSE will copy both struct a and the pointed buffer from the
2465  * process doing the ioctl and retry ioctl with both struct a and the
2466  * buffer.
2467  *
2468  * This time, FUSE server has everything it needs and completes ioctl
2469  * without FUSE_IOCTL_RETRY which finishes the ioctl call.
2470  *
2471  * Copying data out works the same way.
2472  *
2473  * Note that if FUSE_IOCTL_UNRESTRICTED is clear, the kernel
2474  * automatically initializes in and out iovs by decoding @cmd with
2475  * _IOC_* macros and the server is not allowed to request RETRY.  This
2476  * limits ioctl data transfers to well-formed ioctls and is the forced
2477  * behavior for all FUSE servers.
2478  */
2479 long fuse_do_ioctl(struct file *file, unsigned int cmd, unsigned long arg,
2480                    unsigned int flags)
2481 {
2482         struct fuse_file *ff = file->private_data;
2483         struct fuse_conn *fc = ff->fc;
2484         struct fuse_ioctl_in inarg = {
2485                 .fh = ff->fh,
2486                 .cmd = cmd,
2487                 .arg = arg,
2488                 .flags = flags
2489         };
2490         struct fuse_ioctl_out outarg;
2491         struct fuse_req *req = NULL;
2492         struct page **pages = NULL;
2493         struct iovec *iov_page = NULL;
2494         struct iovec *in_iov = NULL, *out_iov = NULL;
2495         unsigned int in_iovs = 0, out_iovs = 0, num_pages = 0, max_pages;
2496         size_t in_size, out_size, transferred, c;
2497         int err, i;
2498         struct iov_iter ii;
2499
2500 #if BITS_PER_LONG == 32
2501         inarg.flags |= FUSE_IOCTL_32BIT;
2502 #else
2503         if (flags & FUSE_IOCTL_COMPAT)
2504                 inarg.flags |= FUSE_IOCTL_32BIT;
2505 #endif
2506
2507         /* assume all the iovs returned by client always fits in a page */
2508         BUILD_BUG_ON(sizeof(struct fuse_ioctl_iovec) * FUSE_IOCTL_MAX_IOV > PAGE_SIZE);
2509
2510         err = -ENOMEM;
2511         pages = kcalloc(FUSE_MAX_PAGES_PER_REQ, sizeof(pages[0]), GFP_KERNEL);
2512         iov_page = (struct iovec *) __get_free_page(GFP_KERNEL);
2513         if (!pages || !iov_page)
2514                 goto out;
2515
2516         /*
2517          * If restricted, initialize IO parameters as encoded in @cmd.
2518          * RETRY from server is not allowed.
2519          */
2520         if (!(flags & FUSE_IOCTL_UNRESTRICTED)) {
2521                 struct iovec *iov = iov_page;
2522
2523                 iov->iov_base = (void __user *)arg;
2524
2525                 switch (cmd) {
2526                 case FS_IOC_GETFLAGS:
2527                 case FS_IOC_SETFLAGS:
2528                         iov->iov_len = sizeof(int);
2529                         break;
2530                 default:
2531                         iov->iov_len = _IOC_SIZE(cmd);
2532                         break;
2533                 }
2534
2535                 if (_IOC_DIR(cmd) & _IOC_WRITE) {
2536                         in_iov = iov;
2537                         in_iovs = 1;
2538                 }
2539
2540                 if (_IOC_DIR(cmd) & _IOC_READ) {
2541                         out_iov = iov;
2542                         out_iovs = 1;
2543                 }
2544         }
2545
2546  retry:
2547         inarg.in_size = in_size = iov_length(in_iov, in_iovs);
2548         inarg.out_size = out_size = iov_length(out_iov, out_iovs);
2549
2550         /*
2551          * Out data can be used either for actual out data or iovs,
2552          * make sure there always is at least one page.
2553          */
2554         out_size = max_t(size_t, out_size, PAGE_SIZE);
2555         max_pages = DIV_ROUND_UP(max(in_size, out_size), PAGE_SIZE);
2556
2557         /* make sure there are enough buffer pages and init request with them */
2558         err = -ENOMEM;
2559         if (max_pages > FUSE_MAX_PAGES_PER_REQ)
2560                 goto out;
2561         while (num_pages < max_pages) {
2562                 pages[num_pages] = alloc_page(GFP_KERNEL | __GFP_HIGHMEM);
2563                 if (!pages[num_pages])
2564                         goto out;
2565                 num_pages++;
2566         }
2567
2568         req = fuse_get_req(fc, num_pages);
2569         if (IS_ERR(req)) {
2570                 err = PTR_ERR(req);
2571                 req = NULL;
2572                 goto out;
2573         }
2574         memcpy(req->pages, pages, sizeof(req->pages[0]) * num_pages);
2575         req->num_pages = num_pages;
2576         fuse_page_descs_length_init(req, 0, req->num_pages);
2577
2578         /* okay, let's send it to the client */
2579         req->in.h.opcode = FUSE_IOCTL;
2580         req->in.h.nodeid = ff->nodeid;
2581         req->in.numargs = 1;
2582         req->in.args[0].size = sizeof(inarg);
2583         req->in.args[0].value = &inarg;
2584         if (in_size) {
2585                 req->in.numargs++;
2586                 req->in.args[1].size = in_size;
2587                 req->in.argpages = 1;
2588
2589                 err = -EFAULT;
2590                 iov_iter_init(&ii, WRITE, in_iov, in_iovs, in_size);
2591                 for (i = 0; iov_iter_count(&ii) && !WARN_ON(i >= num_pages); i++) {
2592                         c = copy_page_from_iter(pages[i], 0, PAGE_SIZE, &ii);
2593                         if (c != PAGE_SIZE && iov_iter_count(&ii))
2594                                 goto out;
2595                 }
2596         }
2597
2598         req->out.numargs = 2;
2599         req->out.args[0].size = sizeof(outarg);
2600         req->out.args[0].value = &outarg;
2601         req->out.args[1].size = out_size;
2602         req->out.argpages = 1;
2603         req->out.argvar = 1;
2604
2605         fuse_request_send(fc, req);
2606         err = req->out.h.error;
2607         transferred = req->out.args[1].size;
2608         fuse_put_request(fc, req);
2609         req = NULL;
2610         if (err)
2611                 goto out;
2612
2613         /* did it ask for retry? */
2614         if (outarg.flags & FUSE_IOCTL_RETRY) {
2615                 void *vaddr;
2616
2617                 /* no retry if in restricted mode */
2618                 err = -EIO;
2619                 if (!(flags & FUSE_IOCTL_UNRESTRICTED))
2620                         goto out;
2621
2622                 in_iovs = outarg.in_iovs;
2623                 out_iovs = outarg.out_iovs;
2624
2625                 /*
2626                  * Make sure things are in boundary, separate checks
2627                  * are to protect against overflow.
2628                  */
2629                 err = -ENOMEM;
2630                 if (in_iovs > FUSE_IOCTL_MAX_IOV ||
2631                     out_iovs > FUSE_IOCTL_MAX_IOV ||
2632                     in_iovs + out_iovs > FUSE_IOCTL_MAX_IOV)
2633                         goto out;
2634
2635                 vaddr = kmap_atomic(pages[0]);
2636                 err = fuse_copy_ioctl_iovec(fc, iov_page, vaddr,
2637                                             transferred, in_iovs + out_iovs,
2638                                             (flags & FUSE_IOCTL_COMPAT) != 0);
2639                 kunmap_atomic(vaddr);
2640                 if (err)
2641                         goto out;
2642
2643                 in_iov = iov_page;
2644                 out_iov = in_iov + in_iovs;
2645
2646                 err = fuse_verify_ioctl_iov(in_iov, in_iovs);
2647                 if (err)
2648                         goto out;
2649
2650                 err = fuse_verify_ioctl_iov(out_iov, out_iovs);
2651                 if (err)
2652                         goto out;
2653
2654                 goto retry;
2655         }
2656
2657         err = -EIO;
2658         if (transferred > inarg.out_size)
2659                 goto out;
2660
2661         err = -EFAULT;
2662         iov_iter_init(&ii, READ, out_iov, out_iovs, transferred);
2663         for (i = 0; iov_iter_count(&ii) && !WARN_ON(i >= num_pages); i++) {
2664                 c = copy_page_to_iter(pages[i], 0, PAGE_SIZE, &ii);
2665                 if (c != PAGE_SIZE && iov_iter_count(&ii))
2666                         goto out;
2667         }
2668         err = 0;
2669  out:
2670         if (req)
2671                 fuse_put_request(fc, req);
2672         free_page((unsigned long) iov_page);
2673         while (num_pages)
2674                 __free_page(pages[--num_pages]);
2675         kfree(pages);
2676
2677         return err ? err : outarg.result;
2678 }
2679 EXPORT_SYMBOL_GPL(fuse_do_ioctl);
2680
2681 long fuse_ioctl_common(struct file *file, unsigned int cmd,
2682                        unsigned long arg, unsigned int flags)
2683 {
2684         struct inode *inode = file_inode(file);
2685         struct fuse_conn *fc = get_fuse_conn(inode);
2686
2687         if (!fuse_allow_current_process(fc))
2688                 return -EACCES;
2689
2690         if (is_bad_inode(inode))
2691                 return -EIO;
2692
2693         return fuse_do_ioctl(file, cmd, arg, flags);
2694 }
2695
2696 static long fuse_file_ioctl(struct file *file, unsigned int cmd,
2697                             unsigned long arg)
2698 {
2699         return fuse_ioctl_common(file, cmd, arg, 0);
2700 }
2701
2702 static long fuse_file_compat_ioctl(struct file *file, unsigned int cmd,
2703                                    unsigned long arg)
2704 {
2705         return fuse_ioctl_common(file, cmd, arg, FUSE_IOCTL_COMPAT);
2706 }
2707
2708 /*
2709  * All files which have been polled are linked to RB tree
2710  * fuse_conn->polled_files which is indexed by kh.  Walk the tree and
2711  * find the matching one.
2712  */
2713 static struct rb_node **fuse_find_polled_node(struct fuse_conn *fc, u64 kh,
2714                                               struct rb_node **parent_out)
2715 {
2716         struct rb_node **link = &fc->polled_files.rb_node;
2717         struct rb_node *last = NULL;
2718
2719         while (*link) {
2720                 struct fuse_file *ff;
2721
2722                 last = *link;
2723                 ff = rb_entry(last, struct fuse_file, polled_node);
2724
2725                 if (kh < ff->kh)
2726                         link = &last->rb_left;
2727                 else if (kh > ff->kh)
2728                         link = &last->rb_right;
2729                 else
2730                         return link;
2731         }
2732
2733         if (parent_out)
2734                 *parent_out = last;
2735         return link;
2736 }
2737
2738 /*
2739  * The file is about to be polled.  Make sure it's on the polled_files
2740  * RB tree.  Note that files once added to the polled_files tree are
2741  * not removed before the file is released.  This is because a file
2742  * polled once is likely to be polled again.
2743  */
2744 static void fuse_register_polled_file(struct fuse_conn *fc,
2745                                       struct fuse_file *ff)
2746 {
2747         spin_lock(&fc->lock);
2748         if (RB_EMPTY_NODE(&ff->polled_node)) {
2749                 struct rb_node **link, *uninitialized_var(parent);
2750
2751                 link = fuse_find_polled_node(fc, ff->kh, &parent);
2752                 BUG_ON(*link);
2753                 rb_link_node(&ff->polled_node, parent, link);
2754                 rb_insert_color(&ff->polled_node, &fc->polled_files);
2755         }
2756         spin_unlock(&fc->lock);
2757 }
2758
2759 unsigned fuse_file_poll(struct file *file, poll_table *wait)
2760 {
2761         struct fuse_file *ff = file->private_data;
2762         struct fuse_conn *fc = ff->fc;
2763         struct fuse_poll_in inarg = { .fh = ff->fh, .kh = ff->kh };
2764         struct fuse_poll_out outarg;
2765         FUSE_ARGS(args);
2766         int err;
2767
2768         if (fc->no_poll)
2769                 return DEFAULT_POLLMASK;
2770
2771         poll_wait(file, &ff->poll_wait, wait);
2772         inarg.events = (__u32)poll_requested_events(wait);
2773
2774         /*
2775          * Ask for notification iff there's someone waiting for it.
2776          * The client may ignore the flag and always notify.
2777          */
2778         if (waitqueue_active(&ff->poll_wait)) {
2779                 inarg.flags |= FUSE_POLL_SCHEDULE_NOTIFY;
2780                 fuse_register_polled_file(fc, ff);
2781         }
2782
2783         args.in.h.opcode = FUSE_POLL;
2784         args.in.h.nodeid = ff->nodeid;
2785         args.in.numargs = 1;
2786         args.in.args[0].size = sizeof(inarg);
2787         args.in.args[0].value = &inarg;
2788         args.out.numargs = 1;
2789         args.out.args[0].size = sizeof(outarg);
2790         args.out.args[0].value = &outarg;
2791         err = fuse_simple_request(fc, &args);
2792
2793         if (!err)
2794                 return outarg.revents;
2795         if (err == -ENOSYS) {
2796                 fc->no_poll = 1;
2797                 return DEFAULT_POLLMASK;
2798         }
2799         return POLLERR;
2800 }
2801 EXPORT_SYMBOL_GPL(fuse_file_poll);
2802
2803 /*
2804  * This is called from fuse_handle_notify() on FUSE_NOTIFY_POLL and
2805  * wakes up the poll waiters.
2806  */
2807 int fuse_notify_poll_wakeup(struct fuse_conn *fc,
2808                             struct fuse_notify_poll_wakeup_out *outarg)
2809 {
2810         u64 kh = outarg->kh;
2811         struct rb_node **link;
2812
2813         spin_lock(&fc->lock);
2814
2815         link = fuse_find_polled_node(fc, kh, NULL);
2816         if (*link) {
2817                 struct fuse_file *ff;
2818
2819                 ff = rb_entry(*link, struct fuse_file, polled_node);
2820                 wake_up_interruptible_sync(&ff->poll_wait);
2821         }
2822
2823         spin_unlock(&fc->lock);
2824         return 0;
2825 }
2826
2827 static void fuse_do_truncate(struct file *file)
2828 {
2829         struct inode *inode = file->f_mapping->host;
2830         struct iattr attr;
2831
2832         attr.ia_valid = ATTR_SIZE;
2833         attr.ia_size = i_size_read(inode);
2834
2835         attr.ia_file = file;
2836         attr.ia_valid |= ATTR_FILE;
2837
2838         fuse_do_setattr(file_dentry(file), &attr, file);
2839 }
2840
2841 static inline loff_t fuse_round_up(loff_t off)
2842 {
2843         return round_up(off, FUSE_MAX_PAGES_PER_REQ << PAGE_SHIFT);
2844 }
2845
2846 static ssize_t
2847 fuse_direct_IO(struct kiocb *iocb, struct iov_iter *iter)
2848 {
2849         DECLARE_COMPLETION_ONSTACK(wait);
2850         ssize_t ret = 0;
2851         struct file *file = iocb->ki_filp;
2852         struct fuse_file *ff = file->private_data;
2853         bool async_dio = ff->fc->async_dio;
2854         loff_t pos = 0;
2855         struct inode *inode;
2856         loff_t i_size;
2857         size_t count = iov_iter_count(iter);
2858         loff_t offset = iocb->ki_pos;
2859         struct fuse_io_priv *io;
2860
2861         pos = offset;
2862         inode = file->f_mapping->host;
2863         i_size = i_size_read(inode);
2864
2865         if ((iov_iter_rw(iter) == READ) && (offset > i_size))
2866                 return 0;
2867
2868         /* optimization for short read */
2869         if (async_dio && iov_iter_rw(iter) != WRITE && offset + count > i_size) {
2870                 if (offset >= i_size)
2871                         return 0;
2872                 iov_iter_truncate(iter, fuse_round_up(i_size - offset));
2873                 count = iov_iter_count(iter);
2874         }
2875
2876         io = kmalloc(sizeof(struct fuse_io_priv), GFP_KERNEL);
2877         if (!io)
2878                 return -ENOMEM;
2879         spin_lock_init(&io->lock);
2880         kref_init(&io->refcnt);
2881         io->reqs = 1;
2882         io->bytes = -1;
2883         io->size = 0;
2884         io->offset = offset;
2885         io->write = (iov_iter_rw(iter) == WRITE);
2886         io->err = 0;
2887         io->file = file;
2888         /*
2889          * By default, we want to optimize all I/Os with async request
2890          * submission to the client filesystem if supported.
2891          */
2892         io->async = async_dio;
2893         io->iocb = iocb;
2894         io->blocking = is_sync_kiocb(iocb);
2895
2896         /*
2897          * We cannot asynchronously extend the size of a file.
2898          * In such case the aio will behave exactly like sync io.
2899          */
2900         if ((offset + count > i_size) && iov_iter_rw(iter) == WRITE)
2901                 io->blocking = true;
2902
2903         if (io->async && io->blocking) {
2904                 /*
2905                  * Additional reference to keep io around after
2906                  * calling fuse_aio_complete()
2907                  */
2908                 kref_get(&io->refcnt);
2909                 io->done = &wait;
2910         }
2911
2912         if (iov_iter_rw(iter) == WRITE) {
2913                 ret = fuse_direct_io(io, iter, &pos, FUSE_DIO_WRITE);
2914                 fuse_invalidate_attr(inode);
2915         } else {
2916                 ret = __fuse_direct_read(io, iter, &pos);
2917         }
2918
2919         if (io->async) {
2920                 bool blocking = io->blocking;
2921
2922                 fuse_aio_complete(io, ret < 0 ? ret : 0, -1);
2923
2924                 /* we have a non-extending, async request, so return */
2925                 if (!blocking)
2926                         return -EIOCBQUEUED;
2927
2928                 wait_for_completion(&wait);
2929                 ret = fuse_get_res_by_io(io);
2930         }
2931
2932         kref_put(&io->refcnt, fuse_io_release);
2933
2934         if (iov_iter_rw(iter) == WRITE) {
2935                 if (ret > 0)
2936                         fuse_write_update_size(inode, pos);
2937                 else if (ret < 0 && offset + count > i_size)
2938                         fuse_do_truncate(file);
2939         }
2940
2941         return ret;
2942 }
2943
2944 static long fuse_file_fallocate(struct file *file, int mode, loff_t offset,
2945                                 loff_t length)
2946 {
2947         struct fuse_file *ff = file->private_data;
2948         struct inode *inode = file_inode(file);
2949         struct fuse_inode *fi = get_fuse_inode(inode);
2950         struct fuse_conn *fc = ff->fc;
2951         FUSE_ARGS(args);
2952         struct fuse_fallocate_in inarg = {
2953                 .fh = ff->fh,
2954                 .offset = offset,
2955                 .length = length,
2956                 .mode = mode
2957         };
2958         int err;
2959         bool lock_inode = !(mode & FALLOC_FL_KEEP_SIZE) ||
2960                            (mode & FALLOC_FL_PUNCH_HOLE);
2961
2962         if (mode & ~(FALLOC_FL_KEEP_SIZE | FALLOC_FL_PUNCH_HOLE))
2963                 return -EOPNOTSUPP;
2964
2965         if (fc->no_fallocate)
2966                 return -EOPNOTSUPP;
2967
2968         if (lock_inode) {
2969                 inode_lock(inode);
2970                 if (mode & FALLOC_FL_PUNCH_HOLE) {
2971                         loff_t endbyte = offset + length - 1;
2972                         err = filemap_write_and_wait_range(inode->i_mapping,
2973                                                            offset, endbyte);
2974                         if (err)
2975                                 goto out;
2976
2977                         fuse_sync_writes(inode);
2978                 }
2979         }
2980
2981         if (!(mode & FALLOC_FL_KEEP_SIZE) &&
2982             offset + length > i_size_read(inode)) {
2983                 err = inode_newsize_ok(inode, offset + length);
2984                 if (err)
2985                         goto out;
2986         }
2987
2988         if (!(mode & FALLOC_FL_KEEP_SIZE))
2989                 set_bit(FUSE_I_SIZE_UNSTABLE, &fi->state);
2990
2991         args.in.h.opcode = FUSE_FALLOCATE;
2992         args.in.h.nodeid = ff->nodeid;
2993         args.in.numargs = 1;
2994         args.in.args[0].size = sizeof(inarg);
2995         args.in.args[0].value = &inarg;
2996         err = fuse_simple_request(fc, &args);
2997         if (err == -ENOSYS) {
2998                 fc->no_fallocate = 1;
2999                 err = -EOPNOTSUPP;
3000         }
3001         if (err)
3002                 goto out;
3003
3004         /* we could have extended the file */
3005         if (!(mode & FALLOC_FL_KEEP_SIZE)) {
3006                 bool changed = fuse_write_update_size(inode, offset + length);
3007
3008                 if (changed && fc->writeback_cache)
3009                         file_update_time(file);
3010         }
3011
3012         if (mode & FALLOC_FL_PUNCH_HOLE)
3013                 truncate_pagecache_range(inode, offset, offset + length - 1);
3014
3015         fuse_invalidate_attr(inode);
3016
3017 out:
3018         if (!(mode & FALLOC_FL_KEEP_SIZE))
3019                 clear_bit(FUSE_I_SIZE_UNSTABLE, &fi->state);
3020
3021         if (lock_inode)
3022                 inode_unlock(inode);
3023
3024         return err;
3025 }
3026
3027 static const struct file_operations fuse_file_operations = {
3028         .llseek         = fuse_file_llseek,
3029         .read_iter      = fuse_file_read_iter,
3030         .write_iter     = fuse_file_write_iter,
3031         .mmap           = fuse_file_mmap,
3032         .open           = fuse_open,
3033         .flush          = fuse_flush,
3034         .release        = fuse_release,
3035         .fsync          = fuse_fsync,
3036         .lock           = fuse_file_lock,
3037         .flock          = fuse_file_flock,
3038         .splice_read    = generic_file_splice_read,
3039         .unlocked_ioctl = fuse_file_ioctl,
3040         .compat_ioctl   = fuse_file_compat_ioctl,
3041         .poll           = fuse_file_poll,
3042         .fallocate      = fuse_file_fallocate,
3043 };
3044
3045 static const struct file_operations fuse_direct_io_file_operations = {
3046         .llseek         = fuse_file_llseek,
3047         .read_iter      = fuse_direct_read_iter,
3048         .write_iter     = fuse_direct_write_iter,
3049         .mmap           = fuse_direct_mmap,
3050         .open           = fuse_open,
3051         .flush          = fuse_flush,
3052         .release        = fuse_release,
3053         .fsync          = fuse_fsync,
3054         .lock           = fuse_file_lock,
3055         .flock          = fuse_file_flock,
3056         .unlocked_ioctl = fuse_file_ioctl,
3057         .compat_ioctl   = fuse_file_compat_ioctl,
3058         .poll           = fuse_file_poll,
3059         .fallocate      = fuse_file_fallocate,
3060         /* no splice_read */
3061 };
3062
3063 static const struct address_space_operations fuse_file_aops  = {
3064         .readpage       = fuse_readpage,
3065         .writepage      = fuse_writepage,
3066         .writepages     = fuse_writepages,
3067         .launder_page   = fuse_launder_page,
3068         .readpages      = fuse_readpages,
3069         .set_page_dirty = __set_page_dirty_nobuffers,
3070         .bmap           = fuse_bmap,
3071         .direct_IO      = fuse_direct_IO,
3072         .write_begin    = fuse_write_begin,
3073         .write_end      = fuse_write_end,
3074 };
3075
3076 void fuse_init_file_inode(struct inode *inode)
3077 {
3078         inode->i_fop = &fuse_file_operations;
3079         inode->i_data.a_ops = &fuse_file_aops;
3080 }