1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /* kiocb-using read/write
3  *
4  * Copyright (C) 2021 Red Hat, Inc. All Rights Reserved.
5  * Written by David Howells (dhowells@redhat.com)
6  */
7 
8 #include <linux/mount.h>
9 #include <linux/slab.h>
10 #include <linux/file.h>
11 #include <linux/uio.h>
12 #include <linux/bio.h>
13 #include <linux/falloc.h>
14 #include <linux/sched/mm.h>
15 #include <trace/events/fscache.h>
16 #include "internal.h"
17 
18 struct cachefiles_kiocb {
19 	struct kiocb		iocb;
20 	refcount_t		ki_refcnt;
21 	loff_t			start;
22 	union {
23 		size_t		skipped;
24 		size_t		len;
25 	};
26 	struct cachefiles_object *object;
27 	netfs_io_terminated_t	term_func;
28 	void			*term_func_priv;
29 	bool			was_async;
30 	unsigned int		inval_counter;	/* Copy of cookie->inval_counter */
31 	u64			b_writing;
32 };
33 
cachefiles_put_kiocb(struct cachefiles_kiocb * ki)34 static inline void cachefiles_put_kiocb(struct cachefiles_kiocb *ki)
35 {
36 	if (refcount_dec_and_test(&ki->ki_refcnt)) {
37 		cachefiles_put_object(ki->object, cachefiles_obj_put_ioreq);
38 		fput(ki->iocb.ki_filp);
39 		kfree(ki);
40 	}
41 }
42 
43 /*
44  * Handle completion of a read from the cache.
45  */
cachefiles_read_complete(struct kiocb * iocb,long ret)46 static void cachefiles_read_complete(struct kiocb *iocb, long ret)
47 {
48 	struct cachefiles_kiocb *ki = container_of(iocb, struct cachefiles_kiocb, iocb);
49 	struct inode *inode = file_inode(ki->iocb.ki_filp);
50 
51 	_enter("%ld", ret);
52 
53 	if (ret < 0)
54 		trace_cachefiles_io_error(ki->object, inode, ret,
55 					  cachefiles_trace_read_error);
56 
57 	if (ki->term_func) {
58 		if (ret >= 0) {
59 			if (ki->object->cookie->inval_counter == ki->inval_counter)
60 				ki->skipped += ret;
61 			else
62 				ret = -ESTALE;
63 		}
64 
65 		ki->term_func(ki->term_func_priv, ret, ki->was_async);
66 	}
67 
68 	cachefiles_put_kiocb(ki);
69 }
70 
71 /*
72  * Initiate a read from the cache.
73  */
cachefiles_read(struct netfs_cache_resources * cres,loff_t start_pos,struct iov_iter * iter,enum netfs_read_from_hole read_hole,netfs_io_terminated_t term_func,void * term_func_priv)74 static int cachefiles_read(struct netfs_cache_resources *cres,
75 			   loff_t start_pos,
76 			   struct iov_iter *iter,
77 			   enum netfs_read_from_hole read_hole,
78 			   netfs_io_terminated_t term_func,
79 			   void *term_func_priv)
80 {
81 	struct cachefiles_object *object;
82 	struct cachefiles_kiocb *ki;
83 	struct file *file;
84 	unsigned int old_nofs;
85 	ssize_t ret = -ENOBUFS;
86 	size_t len = iov_iter_count(iter), skipped = 0;
87 
88 	if (!fscache_wait_for_operation(cres, FSCACHE_WANT_READ))
89 		goto presubmission_error;
90 
91 	fscache_count_read();
92 	object = cachefiles_cres_object(cres);
93 	file = cachefiles_cres_file(cres);
94 
95 	_enter("%pD,%li,%llx,%zx/%llx",
96 	       file, file_inode(file)->i_ino, start_pos, len,
97 	       i_size_read(file_inode(file)));
98 
99 	/* If the caller asked us to seek for data before doing the read, then
100 	 * we should do that now.  If we find a gap, we fill it with zeros.
101 	 */
102 	if (read_hole != NETFS_READ_HOLE_IGNORE) {
103 		loff_t off = start_pos, off2;
104 
105 		off2 = cachefiles_inject_read_error();
106 		if (off2 == 0)
107 			off2 = vfs_llseek(file, off, SEEK_DATA);
108 		if (off2 < 0 && off2 >= (loff_t)-MAX_ERRNO && off2 != -ENXIO) {
109 			skipped = 0;
110 			ret = off2;
111 			goto presubmission_error;
112 		}
113 
114 		if (off2 == -ENXIO || off2 >= start_pos + len) {
115 			/* The region is beyond the EOF or there's no more data
116 			 * in the region, so clear the rest of the buffer and
117 			 * return success.
118 			 */
119 			ret = -ENODATA;
120 			if (read_hole == NETFS_READ_HOLE_FAIL)
121 				goto presubmission_error;
122 
123 			iov_iter_zero(len, iter);
124 			skipped = len;
125 			ret = 0;
126 			goto presubmission_error;
127 		}
128 
129 		skipped = off2 - off;
130 		iov_iter_zero(skipped, iter);
131 	}
132 
133 	ret = -ENOMEM;
134 	ki = kzalloc(sizeof(struct cachefiles_kiocb), GFP_KERNEL);
135 	if (!ki)
136 		goto presubmission_error;
137 
138 	refcount_set(&ki->ki_refcnt, 2);
139 	ki->iocb.ki_filp	= file;
140 	ki->iocb.ki_pos		= start_pos + skipped;
141 	ki->iocb.ki_flags	= IOCB_DIRECT;
142 	ki->iocb.ki_ioprio	= get_current_ioprio();
143 	ki->skipped		= skipped;
144 	ki->object		= object;
145 	ki->inval_counter	= cres->inval_counter;
146 	ki->term_func		= term_func;
147 	ki->term_func_priv	= term_func_priv;
148 	ki->was_async		= true;
149 
150 	if (ki->term_func)
151 		ki->iocb.ki_complete = cachefiles_read_complete;
152 
153 	get_file(ki->iocb.ki_filp);
154 	cachefiles_grab_object(object, cachefiles_obj_get_ioreq);
155 
156 	trace_cachefiles_read(object, file_inode(file), ki->iocb.ki_pos, len - skipped);
157 	old_nofs = memalloc_nofs_save();
158 	ret = cachefiles_inject_read_error();
159 	if (ret == 0)
160 		ret = vfs_iocb_iter_read(file, &ki->iocb, iter);
161 	memalloc_nofs_restore(old_nofs);
162 	switch (ret) {
163 	case -EIOCBQUEUED:
164 		goto in_progress;
165 
166 	case -ERESTARTSYS:
167 	case -ERESTARTNOINTR:
168 	case -ERESTARTNOHAND:
169 	case -ERESTART_RESTARTBLOCK:
170 		/* There's no easy way to restart the syscall since other AIO's
171 		 * may be already running. Just fail this IO with EINTR.
172 		 */
173 		ret = -EINTR;
174 		fallthrough;
175 	default:
176 		ki->was_async = false;
177 		cachefiles_read_complete(&ki->iocb, ret);
178 		if (ret > 0)
179 			ret = 0;
180 		break;
181 	}
182 
183 in_progress:
184 	cachefiles_put_kiocb(ki);
185 	_leave(" = %zd", ret);
186 	return ret;
187 
188 presubmission_error:
189 	if (term_func)
190 		term_func(term_func_priv, ret < 0 ? ret : skipped, false);
191 	return ret;
192 }
193 
194 /*
195  * Query the occupancy of the cache in a region, returning where the next chunk
196  * of data starts and how long it is.
197  */
cachefiles_query_occupancy(struct netfs_cache_resources * cres,loff_t start,size_t len,size_t granularity,loff_t * _data_start,size_t * _data_len)198 static int cachefiles_query_occupancy(struct netfs_cache_resources *cres,
199 				      loff_t start, size_t len, size_t granularity,
200 				      loff_t *_data_start, size_t *_data_len)
201 {
202 	struct cachefiles_object *object;
203 	struct file *file;
204 	loff_t off, off2;
205 
206 	*_data_start = -1;
207 	*_data_len = 0;
208 
209 	if (!fscache_wait_for_operation(cres, FSCACHE_WANT_READ))
210 		return -ENOBUFS;
211 
212 	object = cachefiles_cres_object(cres);
213 	file = cachefiles_cres_file(cres);
214 	granularity = max_t(size_t, object->volume->cache->bsize, granularity);
215 
216 	_enter("%pD,%li,%llx,%zx/%llx",
217 	       file, file_inode(file)->i_ino, start, len,
218 	       i_size_read(file_inode(file)));
219 
220 	off = cachefiles_inject_read_error();
221 	if (off == 0)
222 		off = vfs_llseek(file, start, SEEK_DATA);
223 	if (off == -ENXIO)
224 		return -ENODATA; /* Beyond EOF */
225 	if (off < 0 && off >= (loff_t)-MAX_ERRNO)
226 		return -ENOBUFS; /* Error. */
227 	if (round_up(off, granularity) >= start + len)
228 		return -ENODATA; /* No data in range */
229 
230 	off2 = cachefiles_inject_read_error();
231 	if (off2 == 0)
232 		off2 = vfs_llseek(file, off, SEEK_HOLE);
233 	if (off2 == -ENXIO)
234 		return -ENODATA; /* Beyond EOF */
235 	if (off2 < 0 && off2 >= (loff_t)-MAX_ERRNO)
236 		return -ENOBUFS; /* Error. */
237 
238 	/* Round away partial blocks */
239 	off = round_up(off, granularity);
240 	off2 = round_down(off2, granularity);
241 	if (off2 <= off)
242 		return -ENODATA;
243 
244 	*_data_start = off;
245 	if (off2 > start + len)
246 		*_data_len = len;
247 	else
248 		*_data_len = off2 - off;
249 	return 0;
250 }
251 
252 /*
253  * Handle completion of a write to the cache.
254  */
cachefiles_write_complete(struct kiocb * iocb,long ret)255 static void cachefiles_write_complete(struct kiocb *iocb, long ret)
256 {
257 	struct cachefiles_kiocb *ki = container_of(iocb, struct cachefiles_kiocb, iocb);
258 	struct cachefiles_object *object = ki->object;
259 	struct inode *inode = file_inode(ki->iocb.ki_filp);
260 
261 	_enter("%ld", ret);
262 
263 	if (ki->was_async)
264 		kiocb_end_write(iocb);
265 
266 	if (ret < 0)
267 		trace_cachefiles_io_error(object, inode, ret,
268 					  cachefiles_trace_write_error);
269 
270 	atomic_long_sub(ki->b_writing, &object->volume->cache->b_writing);
271 	set_bit(FSCACHE_COOKIE_HAVE_DATA, &object->cookie->flags);
272 	if (ki->term_func)
273 		ki->term_func(ki->term_func_priv, ret, ki->was_async);
274 	cachefiles_put_kiocb(ki);
275 }
276 
277 /*
278  * Initiate a write to the cache.
279  */
__cachefiles_write(struct cachefiles_object * object,struct file * file,loff_t start_pos,struct iov_iter * iter,netfs_io_terminated_t term_func,void * term_func_priv)280 int __cachefiles_write(struct cachefiles_object *object,
281 		       struct file *file,
282 		       loff_t start_pos,
283 		       struct iov_iter *iter,
284 		       netfs_io_terminated_t term_func,
285 		       void *term_func_priv)
286 {
287 	struct cachefiles_cache *cache;
288 	struct cachefiles_kiocb *ki;
289 	unsigned int old_nofs;
290 	ssize_t ret;
291 	size_t len = iov_iter_count(iter);
292 
293 	fscache_count_write();
294 	cache = object->volume->cache;
295 
296 	_enter("%pD,%li,%llx,%zx/%llx",
297 	       file, file_inode(file)->i_ino, start_pos, len,
298 	       i_size_read(file_inode(file)));
299 
300 	ki = kzalloc(sizeof(struct cachefiles_kiocb), GFP_KERNEL);
301 	if (!ki) {
302 		if (term_func)
303 			term_func(term_func_priv, -ENOMEM, false);
304 		return -ENOMEM;
305 	}
306 
307 	refcount_set(&ki->ki_refcnt, 2);
308 	ki->iocb.ki_filp	= file;
309 	ki->iocb.ki_pos		= start_pos;
310 	ki->iocb.ki_flags	= IOCB_DIRECT | IOCB_WRITE;
311 	ki->iocb.ki_ioprio	= get_current_ioprio();
312 	ki->object		= object;
313 	ki->start		= start_pos;
314 	ki->len			= len;
315 	ki->term_func		= term_func;
316 	ki->term_func_priv	= term_func_priv;
317 	ki->was_async		= true;
318 	ki->b_writing		= (len + (1 << cache->bshift) - 1) >> cache->bshift;
319 
320 	if (ki->term_func)
321 		ki->iocb.ki_complete = cachefiles_write_complete;
322 	atomic_long_add(ki->b_writing, &cache->b_writing);
323 
324 	get_file(ki->iocb.ki_filp);
325 	cachefiles_grab_object(object, cachefiles_obj_get_ioreq);
326 
327 	trace_cachefiles_write(object, file_inode(file), ki->iocb.ki_pos, len);
328 	old_nofs = memalloc_nofs_save();
329 	ret = cachefiles_inject_write_error();
330 	if (ret == 0)
331 		ret = vfs_iocb_iter_write(file, &ki->iocb, iter);
332 	memalloc_nofs_restore(old_nofs);
333 	switch (ret) {
334 	case -EIOCBQUEUED:
335 		goto in_progress;
336 
337 	case -ERESTARTSYS:
338 	case -ERESTARTNOINTR:
339 	case -ERESTARTNOHAND:
340 	case -ERESTART_RESTARTBLOCK:
341 		/* There's no easy way to restart the syscall since other AIO's
342 		 * may be already running. Just fail this IO with EINTR.
343 		 */
344 		ret = -EINTR;
345 		fallthrough;
346 	default:
347 		ki->was_async = false;
348 		cachefiles_write_complete(&ki->iocb, ret);
349 		break;
350 	}
351 
352 in_progress:
353 	cachefiles_put_kiocb(ki);
354 	_leave(" = %zd", ret);
355 	return ret;
356 }
357 
cachefiles_write(struct netfs_cache_resources * cres,loff_t start_pos,struct iov_iter * iter,netfs_io_terminated_t term_func,void * term_func_priv)358 static int cachefiles_write(struct netfs_cache_resources *cres,
359 			    loff_t start_pos,
360 			    struct iov_iter *iter,
361 			    netfs_io_terminated_t term_func,
362 			    void *term_func_priv)
363 {
364 	if (!fscache_wait_for_operation(cres, FSCACHE_WANT_WRITE)) {
365 		if (term_func)
366 			term_func(term_func_priv, -ENOBUFS, false);
367 		return -ENOBUFS;
368 	}
369 
370 	return __cachefiles_write(cachefiles_cres_object(cres),
371 				  cachefiles_cres_file(cres),
372 				  start_pos, iter,
373 				  term_func, term_func_priv);
374 }
375 
376 static inline enum netfs_io_source
cachefiles_do_prepare_read(struct netfs_cache_resources * cres,loff_t start,size_t * _len,loff_t i_size,unsigned long * _flags,ino_t netfs_ino)377 cachefiles_do_prepare_read(struct netfs_cache_resources *cres,
378 			   loff_t start, size_t *_len, loff_t i_size,
379 			   unsigned long *_flags, ino_t netfs_ino)
380 {
381 	enum cachefiles_prepare_read_trace why;
382 	struct cachefiles_object *object = NULL;
383 	struct cachefiles_cache *cache;
384 	struct fscache_cookie *cookie = fscache_cres_cookie(cres);
385 	const struct cred *saved_cred;
386 	struct file *file = cachefiles_cres_file(cres);
387 	enum netfs_io_source ret = NETFS_DOWNLOAD_FROM_SERVER;
388 	size_t len = *_len;
389 	loff_t off, to;
390 	ino_t ino = file ? file_inode(file)->i_ino : 0;
391 	int rc;
392 
393 	_enter("%zx @%llx/%llx", len, start, i_size);
394 
395 	if (start >= i_size) {
396 		ret = NETFS_FILL_WITH_ZEROES;
397 		why = cachefiles_trace_read_after_eof;
398 		goto out_no_object;
399 	}
400 
401 	if (test_bit(FSCACHE_COOKIE_NO_DATA_TO_READ, &cookie->flags)) {
402 		__set_bit(NETFS_SREQ_COPY_TO_CACHE, _flags);
403 		why = cachefiles_trace_read_no_data;
404 		if (!test_bit(NETFS_SREQ_ONDEMAND, _flags))
405 			goto out_no_object;
406 	}
407 
408 	/* The object and the file may be being created in the background. */
409 	if (!file) {
410 		why = cachefiles_trace_read_no_file;
411 		if (!fscache_wait_for_operation(cres, FSCACHE_WANT_READ))
412 			goto out_no_object;
413 		file = cachefiles_cres_file(cres);
414 		if (!file)
415 			goto out_no_object;
416 		ino = file_inode(file)->i_ino;
417 	}
418 
419 	object = cachefiles_cres_object(cres);
420 	cache = object->volume->cache;
421 	cachefiles_begin_secure(cache, &saved_cred);
422 retry:
423 	off = cachefiles_inject_read_error();
424 	if (off == 0)
425 		off = vfs_llseek(file, start, SEEK_DATA);
426 	if (off < 0 && off >= (loff_t)-MAX_ERRNO) {
427 		if (off == (loff_t)-ENXIO) {
428 			why = cachefiles_trace_read_seek_nxio;
429 			goto download_and_store;
430 		}
431 		trace_cachefiles_io_error(object, file_inode(file), off,
432 					  cachefiles_trace_seek_error);
433 		why = cachefiles_trace_read_seek_error;
434 		goto out;
435 	}
436 
437 	if (off >= start + len) {
438 		why = cachefiles_trace_read_found_hole;
439 		goto download_and_store;
440 	}
441 
442 	if (off > start) {
443 		off = round_up(off, cache->bsize);
444 		len = off - start;
445 		*_len = len;
446 		why = cachefiles_trace_read_found_part;
447 		goto download_and_store;
448 	}
449 
450 	to = cachefiles_inject_read_error();
451 	if (to == 0)
452 		to = vfs_llseek(file, start, SEEK_HOLE);
453 	if (to < 0 && to >= (loff_t)-MAX_ERRNO) {
454 		trace_cachefiles_io_error(object, file_inode(file), to,
455 					  cachefiles_trace_seek_error);
456 		why = cachefiles_trace_read_seek_error;
457 		goto out;
458 	}
459 
460 	if (to < start + len) {
461 		if (start + len >= i_size)
462 			to = round_up(to, cache->bsize);
463 		else
464 			to = round_down(to, cache->bsize);
465 		len = to - start;
466 		*_len = len;
467 	}
468 
469 	why = cachefiles_trace_read_have_data;
470 	ret = NETFS_READ_FROM_CACHE;
471 	goto out;
472 
473 download_and_store:
474 	__set_bit(NETFS_SREQ_COPY_TO_CACHE, _flags);
475 	if (test_bit(NETFS_SREQ_ONDEMAND, _flags)) {
476 		rc = cachefiles_ondemand_read(object, start, len);
477 		if (!rc) {
478 			__clear_bit(NETFS_SREQ_ONDEMAND, _flags);
479 			goto retry;
480 		}
481 		ret = NETFS_INVALID_READ;
482 	}
483 out:
484 	cachefiles_end_secure(cache, saved_cred);
485 out_no_object:
486 	trace_cachefiles_prep_read(object, start, len, *_flags, ret, why, ino, netfs_ino);
487 	return ret;
488 }
489 
490 /*
491  * Prepare a read operation, shortening it to a cached/uncached
492  * boundary as appropriate.
493  */
cachefiles_prepare_read(struct netfs_io_subrequest * subreq,unsigned long long i_size)494 static enum netfs_io_source cachefiles_prepare_read(struct netfs_io_subrequest *subreq,
495 						    unsigned long long i_size)
496 {
497 	return cachefiles_do_prepare_read(&subreq->rreq->cache_resources,
498 					  subreq->start, &subreq->len, i_size,
499 					  &subreq->flags, subreq->rreq->inode->i_ino);
500 }
501 
502 /*
503  * Prepare an on-demand read operation, shortening it to a cached/uncached
504  * boundary as appropriate.
505  */
506 static enum netfs_io_source
cachefiles_prepare_ondemand_read(struct netfs_cache_resources * cres,loff_t start,size_t * _len,loff_t i_size,unsigned long * _flags,ino_t ino)507 cachefiles_prepare_ondemand_read(struct netfs_cache_resources *cres,
508 				 loff_t start, size_t *_len, loff_t i_size,
509 				 unsigned long *_flags, ino_t ino)
510 {
511 	return cachefiles_do_prepare_read(cres, start, _len, i_size, _flags, ino);
512 }
513 
514 /*
515  * Prepare for a write to occur.
516  */
__cachefiles_prepare_write(struct cachefiles_object * object,struct file * file,loff_t * _start,size_t * _len,size_t upper_len,bool no_space_allocated_yet)517 int __cachefiles_prepare_write(struct cachefiles_object *object,
518 			       struct file *file,
519 			       loff_t *_start, size_t *_len, size_t upper_len,
520 			       bool no_space_allocated_yet)
521 {
522 	struct cachefiles_cache *cache = object->volume->cache;
523 	loff_t start = *_start, pos;
524 	size_t len = *_len;
525 	int ret;
526 
527 	/* Round to DIO size */
528 	start = round_down(*_start, PAGE_SIZE);
529 	if (start != *_start || *_len > upper_len) {
530 		/* Probably asked to cache a streaming write written into the
531 		 * pagecache when the cookie was temporarily out of service to
532 		 * culling.
533 		 */
534 		fscache_count_dio_misfit();
535 		return -ENOBUFS;
536 	}
537 
538 	*_len = round_up(len, PAGE_SIZE);
539 
540 	/* We need to work out whether there's sufficient disk space to perform
541 	 * the write - but we can skip that check if we have space already
542 	 * allocated.
543 	 */
544 	if (no_space_allocated_yet)
545 		goto check_space;
546 
547 	pos = cachefiles_inject_read_error();
548 	if (pos == 0)
549 		pos = vfs_llseek(file, start, SEEK_DATA);
550 	if (pos < 0 && pos >= (loff_t)-MAX_ERRNO) {
551 		if (pos == -ENXIO)
552 			goto check_space; /* Unallocated tail */
553 		trace_cachefiles_io_error(object, file_inode(file), pos,
554 					  cachefiles_trace_seek_error);
555 		return pos;
556 	}
557 	if ((u64)pos >= (u64)start + *_len)
558 		goto check_space; /* Unallocated region */
559 
560 	/* We have a block that's at least partially filled - if we're low on
561 	 * space, we need to see if it's fully allocated.  If it's not, we may
562 	 * want to cull it.
563 	 */
564 	if (cachefiles_has_space(cache, 0, *_len / PAGE_SIZE,
565 				 cachefiles_has_space_check) == 0)
566 		return 0; /* Enough space to simply overwrite the whole block */
567 
568 	pos = cachefiles_inject_read_error();
569 	if (pos == 0)
570 		pos = vfs_llseek(file, start, SEEK_HOLE);
571 	if (pos < 0 && pos >= (loff_t)-MAX_ERRNO) {
572 		trace_cachefiles_io_error(object, file_inode(file), pos,
573 					  cachefiles_trace_seek_error);
574 		return pos;
575 	}
576 	if ((u64)pos >= (u64)start + *_len)
577 		return 0; /* Fully allocated */
578 
579 	/* Partially allocated, but insufficient space: cull. */
580 	fscache_count_no_write_space();
581 	ret = cachefiles_inject_remove_error();
582 	if (ret == 0)
583 		ret = vfs_fallocate(file, FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
584 				    start, *_len);
585 	if (ret < 0) {
586 		trace_cachefiles_io_error(object, file_inode(file), ret,
587 					  cachefiles_trace_fallocate_error);
588 		cachefiles_io_error_obj(object,
589 					"CacheFiles: fallocate failed (%d)\n", ret);
590 		ret = -EIO;
591 	}
592 
593 	return ret;
594 
595 check_space:
596 	return cachefiles_has_space(cache, 0, *_len / PAGE_SIZE,
597 				    cachefiles_has_space_for_write);
598 }
599 
cachefiles_prepare_write(struct netfs_cache_resources * cres,loff_t * _start,size_t * _len,size_t upper_len,loff_t i_size,bool no_space_allocated_yet)600 static int cachefiles_prepare_write(struct netfs_cache_resources *cres,
601 				    loff_t *_start, size_t *_len, size_t upper_len,
602 				    loff_t i_size, bool no_space_allocated_yet)
603 {
604 	struct cachefiles_object *object = cachefiles_cres_object(cres);
605 	struct cachefiles_cache *cache = object->volume->cache;
606 	const struct cred *saved_cred;
607 	int ret;
608 
609 	if (!cachefiles_cres_file(cres)) {
610 		if (!fscache_wait_for_operation(cres, FSCACHE_WANT_WRITE))
611 			return -ENOBUFS;
612 		if (!cachefiles_cres_file(cres))
613 			return -ENOBUFS;
614 	}
615 
616 	cachefiles_begin_secure(cache, &saved_cred);
617 	ret = __cachefiles_prepare_write(object, cachefiles_cres_file(cres),
618 					 _start, _len, upper_len,
619 					 no_space_allocated_yet);
620 	cachefiles_end_secure(cache, saved_cred);
621 	return ret;
622 }
623 
cachefiles_prepare_write_subreq(struct netfs_io_subrequest * subreq)624 static void cachefiles_prepare_write_subreq(struct netfs_io_subrequest *subreq)
625 {
626 	struct netfs_io_request *wreq = subreq->rreq;
627 	struct netfs_cache_resources *cres = &wreq->cache_resources;
628 	struct netfs_io_stream *stream = &wreq->io_streams[subreq->stream_nr];
629 
630 	_enter("W=%x[%x] %llx", wreq->debug_id, subreq->debug_index, subreq->start);
631 
632 	stream->sreq_max_len = MAX_RW_COUNT;
633 	stream->sreq_max_segs = BIO_MAX_VECS;
634 
635 	if (!cachefiles_cres_file(cres)) {
636 		if (!fscache_wait_for_operation(cres, FSCACHE_WANT_WRITE))
637 			return netfs_prepare_write_failed(subreq);
638 		if (!cachefiles_cres_file(cres))
639 			return netfs_prepare_write_failed(subreq);
640 	}
641 }
642 
cachefiles_issue_write(struct netfs_io_subrequest * subreq)643 static void cachefiles_issue_write(struct netfs_io_subrequest *subreq)
644 {
645 	struct netfs_io_request *wreq = subreq->rreq;
646 	struct netfs_cache_resources *cres = &wreq->cache_resources;
647 	struct cachefiles_object *object = cachefiles_cres_object(cres);
648 	struct cachefiles_cache *cache = object->volume->cache;
649 	struct netfs_io_stream *stream = &wreq->io_streams[subreq->stream_nr];
650 	const struct cred *saved_cred;
651 	size_t off, pre, post, len = subreq->len;
652 	loff_t start = subreq->start;
653 	int ret;
654 
655 	_enter("W=%x[%x] %llx-%llx",
656 	       wreq->debug_id, subreq->debug_index, start, start + len - 1);
657 
658 	/* We need to start on the cache granularity boundary */
659 	off = start & (CACHEFILES_DIO_BLOCK_SIZE - 1);
660 	if (off) {
661 		pre = CACHEFILES_DIO_BLOCK_SIZE - off;
662 		if (pre >= len) {
663 			fscache_count_dio_misfit();
664 			netfs_write_subrequest_terminated(subreq, len, false);
665 			return;
666 		}
667 		subreq->transferred += pre;
668 		start += pre;
669 		len -= pre;
670 		iov_iter_advance(&subreq->io_iter, pre);
671 	}
672 
673 	/* We also need to end on the cache granularity boundary */
674 	if (start + len == wreq->i_size) {
675 		size_t part = len % CACHEFILES_DIO_BLOCK_SIZE;
676 		size_t need = CACHEFILES_DIO_BLOCK_SIZE - part;
677 
678 		if (part && stream->submit_extendable_to >= need) {
679 			len += need;
680 			subreq->len += need;
681 			subreq->io_iter.count += need;
682 		}
683 	}
684 
685 	post = len & (CACHEFILES_DIO_BLOCK_SIZE - 1);
686 	if (post) {
687 		len -= post;
688 		if (len == 0) {
689 			fscache_count_dio_misfit();
690 			netfs_write_subrequest_terminated(subreq, post, false);
691 			return;
692 		}
693 		iov_iter_truncate(&subreq->io_iter, len);
694 	}
695 
696 	cachefiles_begin_secure(cache, &saved_cred);
697 	ret = __cachefiles_prepare_write(object, cachefiles_cres_file(cres),
698 					 &start, &len, len, true);
699 	cachefiles_end_secure(cache, saved_cred);
700 	if (ret < 0) {
701 		netfs_write_subrequest_terminated(subreq, ret, false);
702 		return;
703 	}
704 
705 	cachefiles_write(&subreq->rreq->cache_resources,
706 			 subreq->start, &subreq->io_iter,
707 			 netfs_write_subrequest_terminated, subreq);
708 }
709 
710 /*
711  * Clean up an operation.
712  */
cachefiles_end_operation(struct netfs_cache_resources * cres)713 static void cachefiles_end_operation(struct netfs_cache_resources *cres)
714 {
715 	struct file *file = cachefiles_cres_file(cres);
716 
717 	if (file)
718 		fput(file);
719 	fscache_end_cookie_access(fscache_cres_cookie(cres), fscache_access_io_end);
720 }
721 
722 static const struct netfs_cache_ops cachefiles_netfs_cache_ops = {
723 	.end_operation		= cachefiles_end_operation,
724 	.read			= cachefiles_read,
725 	.write			= cachefiles_write,
726 	.issue_write		= cachefiles_issue_write,
727 	.prepare_read		= cachefiles_prepare_read,
728 	.prepare_write		= cachefiles_prepare_write,
729 	.prepare_write_subreq	= cachefiles_prepare_write_subreq,
730 	.prepare_ondemand_read	= cachefiles_prepare_ondemand_read,
731 	.query_occupancy	= cachefiles_query_occupancy,
732 };
733 
734 /*
735  * Open the cache file when beginning a cache operation.
736  */
cachefiles_begin_operation(struct netfs_cache_resources * cres,enum fscache_want_state want_state)737 bool cachefiles_begin_operation(struct netfs_cache_resources *cres,
738 				enum fscache_want_state want_state)
739 {
740 	struct cachefiles_object *object = cachefiles_cres_object(cres);
741 
742 	if (!cachefiles_cres_file(cres)) {
743 		cres->ops = &cachefiles_netfs_cache_ops;
744 		if (object->file) {
745 			spin_lock(&object->lock);
746 			if (!cres->cache_priv2 && object->file)
747 				cres->cache_priv2 = get_file(object->file);
748 			spin_unlock(&object->lock);
749 		}
750 	}
751 
752 	if (!cachefiles_cres_file(cres) && want_state != FSCACHE_WANT_PARAMS) {
753 		pr_err("failed to get cres->file\n");
754 		return false;
755 	}
756 
757 	return true;
758 }
759