filemap.c 76.7 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11
/*
 *	linux/mm/filemap.c
 *
 * Copyright (C) 1994-1999  Linus Torvalds
 */

/*
 * This file handles the generic file mmap semantics used by
 * most "normal" filesystems (but you don't /have/ to use this:
 * the NFS filesystem used to do this differently, for example)
 */
12
#include <linux/export.h>
L
Linus Torvalds 已提交
13 14
#include <linux/compiler.h>
#include <linux/fs.h>
15
#include <linux/uaccess.h>
L
Linus Torvalds 已提交
16
#include <linux/aio.h>
17
#include <linux/capability.h>
L
Linus Torvalds 已提交
18
#include <linux/kernel_stat.h>
19
#include <linux/gfp.h>
L
Linus Torvalds 已提交
20 21 22 23 24 25 26 27
#include <linux/mm.h>
#include <linux/swap.h>
#include <linux/mman.h>
#include <linux/pagemap.h>
#include <linux/file.h>
#include <linux/uio.h>
#include <linux/hash.h>
#include <linux/writeback.h>
28
#include <linux/backing-dev.h>
L
Linus Torvalds 已提交
29 30 31
#include <linux/pagevec.h>
#include <linux/blkdev.h>
#include <linux/security.h>
32
#include <linux/cpuset.h>
N
Nick Piggin 已提交
33
#include <linux/hardirq.h> /* for BUG_ON(!in_atomic()) only */
34
#include <linux/memcontrol.h>
35
#include <linux/cleancache.h>
36 37
#include "internal.h"

R
Robert Jarzmik 已提交
38 39 40
#define CREATE_TRACE_POINTS
#include <trace/events/filemap.h>

L
Linus Torvalds 已提交
41 42 43
/*
 * FIXME: remove all knowledge of the buffer layer from the core VM
 */
44
#include <linux/buffer_head.h> /* for try_to_free_buffers */
L
Linus Torvalds 已提交
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62

#include <asm/mman.h>

/*
 * Shared mappings implemented 30.11.1994. It's not fully working yet,
 * though.
 *
 * Shared mappings now work. 15.8.1995  Bruno.
 *
 * finished 'unifying' the page and buffer cache and SMP-threaded the
 * page-cache, 21.05.1999, Ingo Molnar <mingo@redhat.com>
 *
 * SMP-threaded pagemap-LRU 1999, Andrea Arcangeli <andrea@suse.de>
 */

/*
 * Lock ordering:
 *
63
 *  ->i_mmap_mutex		(truncate_pagecache)
L
Linus Torvalds 已提交
64
 *    ->private_lock		(__free_pte->__set_page_dirty_buffers)
65 66
 *      ->swap_lock		(exclusive_swap_page, others)
 *        ->mapping->tree_lock
L
Linus Torvalds 已提交
67
 *
68
 *  ->i_mutex
69
 *    ->i_mmap_mutex		(truncate->unmap_mapping_range)
L
Linus Torvalds 已提交
70 71
 *
 *  ->mmap_sem
72
 *    ->i_mmap_mutex
73
 *      ->page_table_lock or pte_lock	(various, mainly in memory.c)
L
Linus Torvalds 已提交
74 75 76 77 78
 *        ->mapping->tree_lock	(arch-dependent flush_dcache_mmap_lock)
 *
 *  ->mmap_sem
 *    ->lock_page		(access_process_vm)
 *
79 80
 *  ->i_mutex			(generic_file_buffered_write)
 *    ->mmap_sem		(fault_in_pages_readable->do_page_fault)
L
Linus Torvalds 已提交
81
 *
82
 *  bdi->wb.list_lock
83
 *    sb_lock			(fs/fs-writeback.c)
L
Linus Torvalds 已提交
84 85
 *    ->mapping->tree_lock	(__sync_single_inode)
 *
86
 *  ->i_mmap_mutex
L
Linus Torvalds 已提交
87 88 89
 *    ->anon_vma.lock		(vma_adjust)
 *
 *  ->anon_vma.lock
90
 *    ->page_table_lock or pte_lock	(anon_vma_prepare and various)
L
Linus Torvalds 已提交
91
 *
92
 *  ->page_table_lock or pte_lock
93
 *    ->swap_lock		(try_to_unmap_one)
L
Linus Torvalds 已提交
94 95 96
 *    ->private_lock		(try_to_unmap_one)
 *    ->tree_lock		(try_to_unmap_one)
 *    ->zone.lru_lock		(follow_page->mark_page_accessed)
97
 *    ->zone.lru_lock		(check_pte_range->isolate_lru_page)
L
Linus Torvalds 已提交
98 99
 *    ->private_lock		(page_remove_rmap->set_page_dirty)
 *    ->tree_lock		(page_remove_rmap->set_page_dirty)
100
 *    bdi.wb->list_lock		(page_remove_rmap->set_page_dirty)
101
 *    ->inode->i_lock		(page_remove_rmap->set_page_dirty)
102
 *    bdi.wb->list_lock		(zap_pte_range->set_page_dirty)
103
 *    ->inode->i_lock		(zap_pte_range->set_page_dirty)
L
Linus Torvalds 已提交
104 105
 *    ->private_lock		(zap_pte_range->__set_page_dirty_buffers)
 *
106 107
 * ->i_mmap_mutex
 *   ->tasklist_lock            (memory_failure, collect_procs_ao)
L
Linus Torvalds 已提交
108 109
 */

110 111 112
static void page_cache_tree_delete(struct address_space *mapping,
				   struct page *page, void *shadow)
{
113 114 115 116 117
	struct radix_tree_node *node;
	unsigned long index;
	unsigned int offset;
	unsigned int tag;
	void **slot;
118

119 120 121 122 123
	VM_BUG_ON(!PageLocked(page));

	__radix_tree_lookup(&mapping->page_tree, page->index, &node, &slot);

	if (shadow) {
124 125 126 127 128 129 130 131
		mapping->nrshadows++;
		/*
		 * Make sure the nrshadows update is committed before
		 * the nrpages update so that final truncate racing
		 * with reclaim does not see both counters 0 at the
		 * same time and miss a shadow entry.
		 */
		smp_wmb();
132
	}
133
	mapping->nrpages--;
134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170

	if (!node) {
		/* Clear direct pointer tags in root node */
		mapping->page_tree.gfp_mask &= __GFP_BITS_MASK;
		radix_tree_replace_slot(slot, shadow);
		return;
	}

	/* Clear tree tags for the removed page */
	index = page->index;
	offset = index & RADIX_TREE_MAP_MASK;
	for (tag = 0; tag < RADIX_TREE_MAX_TAGS; tag++) {
		if (test_bit(offset, node->tags[tag]))
			radix_tree_tag_clear(&mapping->page_tree, index, tag);
	}

	/* Delete page, swap shadow entry */
	radix_tree_replace_slot(slot, shadow);
	workingset_node_pages_dec(node);
	if (shadow)
		workingset_node_shadows_inc(node);
	else
		if (__radix_tree_delete_node(&mapping->page_tree, node))
			return;

	/*
	 * Track node that only contains shadow entries.
	 *
	 * Avoid acquiring the list_lru lock if already tracked.  The
	 * list_empty() test is safe as node->private_list is
	 * protected by mapping->tree_lock.
	 */
	if (!workingset_node_pages(node) &&
	    list_empty(&node->private_list)) {
		node->private_data = mapping;
		list_lru_add(&workingset_shadow_nodes, &node->private_list);
	}
171 172
}

L
Linus Torvalds 已提交
173
/*
174
 * Delete a page from the page cache and free it. Caller has to make
L
Linus Torvalds 已提交
175
 * sure the page is locked and that nobody else uses it - or that usage
N
Nick Piggin 已提交
176
 * is safe.  The caller must hold the mapping's tree_lock.
L
Linus Torvalds 已提交
177
 */
178
void __delete_from_page_cache(struct page *page, void *shadow)
L
Linus Torvalds 已提交
179 180 181
{
	struct address_space *mapping = page->mapping;

R
Robert Jarzmik 已提交
182
	trace_mm_filemap_delete_from_page_cache(page);
183 184 185 186 187 188 189 190
	/*
	 * if we're uptodate, flush out into the cleancache, otherwise
	 * invalidate any existing cleancache entries.  We can't leave
	 * stale data around in the cleancache once our page is gone
	 */
	if (PageUptodate(page) && PageMappedToDisk(page))
		cleancache_put_page(page);
	else
191
		cleancache_invalidate_page(mapping, page);
192

193 194
	page_cache_tree_delete(mapping, page, shadow);

L
Linus Torvalds 已提交
195
	page->mapping = NULL;
196
	/* Leave page->index set: truncation lookup relies upon it */
197

198
	__dec_zone_page_state(page, NR_FILE_PAGES);
199 200
	if (PageSwapBacked(page))
		__dec_zone_page_state(page, NR_SHMEM);
201
	BUG_ON(page_mapped(page));
202 203 204 205 206 207 208 209 210 211 212 213

	/*
	 * Some filesystems seem to re-dirty the page even after
	 * the VM has canceled the dirty bit (eg ext3 journaling).
	 *
	 * Fix it up by doing a final dirty accounting check after
	 * having removed the page entirely.
	 */
	if (PageDirty(page) && mapping_cap_account_dirty(mapping)) {
		dec_zone_page_state(page, NR_FILE_DIRTY);
		dec_bdi_stat(mapping->backing_dev_info, BDI_RECLAIMABLE);
	}
L
Linus Torvalds 已提交
214 215
}

216 217 218 219 220 221 222 223 224
/**
 * delete_from_page_cache - delete page from page cache
 * @page: the page which the kernel is trying to remove from page cache
 *
 * This must be called only on pages that have been verified to be in the page
 * cache and locked.  It will never put the page into the free list, the caller
 * has a reference on the page.
 */
void delete_from_page_cache(struct page *page)
L
Linus Torvalds 已提交
225 226
{
	struct address_space *mapping = page->mapping;
227
	void (*freepage)(struct page *);
L
Linus Torvalds 已提交
228

M
Matt Mackall 已提交
229
	BUG_ON(!PageLocked(page));
L
Linus Torvalds 已提交
230

231
	freepage = mapping->a_ops->freepage;
N
Nick Piggin 已提交
232
	spin_lock_irq(&mapping->tree_lock);
233
	__delete_from_page_cache(page, NULL);
N
Nick Piggin 已提交
234
	spin_unlock_irq(&mapping->tree_lock);
235
	mem_cgroup_uncharge_cache_page(page);
236 237 238

	if (freepage)
		freepage(page);
239 240 241 242
	page_cache_release(page);
}
EXPORT_SYMBOL(delete_from_page_cache);

J
Jens Axboe 已提交
243
static int sleep_on_page(void *word)
L
Linus Torvalds 已提交
244 245 246 247 248
{
	io_schedule();
	return 0;
}

J
Jens Axboe 已提交
249
static int sleep_on_page_killable(void *word)
M
Matthew Wilcox 已提交
250
{
J
Jens Axboe 已提交
251
	sleep_on_page(word);
M
Matthew Wilcox 已提交
252 253 254
	return fatal_signal_pending(current) ? -EINTR : 0;
}

255 256 257 258 259 260 261 262 263 264 265
static int filemap_check_errors(struct address_space *mapping)
{
	int ret = 0;
	/* Check for outstanding write errors */
	if (test_and_clear_bit(AS_ENOSPC, &mapping->flags))
		ret = -ENOSPC;
	if (test_and_clear_bit(AS_EIO, &mapping->flags))
		ret = -EIO;
	return ret;
}

L
Linus Torvalds 已提交
266
/**
267
 * __filemap_fdatawrite_range - start writeback on mapping dirty pages in range
268 269
 * @mapping:	address space structure to write
 * @start:	offset in bytes where the range starts
270
 * @end:	offset in bytes where the range ends (inclusive)
271
 * @sync_mode:	enable synchronous operation
L
Linus Torvalds 已提交
272
 *
273 274 275
 * Start writeback against all of a mapping's dirty pages that lie
 * within the byte offsets <start, end> inclusive.
 *
L
Linus Torvalds 已提交
276
 * If sync_mode is WB_SYNC_ALL then this is a "data integrity" operation, as
277
 * opposed to a regular memory cleansing writeback.  The difference between
L
Linus Torvalds 已提交
278 279 280
 * these two operations is that if a dirty page/buffer is encountered, it must
 * be waited upon, and not just skipped over.
 */
281 282
int __filemap_fdatawrite_range(struct address_space *mapping, loff_t start,
				loff_t end, int sync_mode)
L
Linus Torvalds 已提交
283 284 285 286
{
	int ret;
	struct writeback_control wbc = {
		.sync_mode = sync_mode,
287
		.nr_to_write = LONG_MAX,
288 289
		.range_start = start,
		.range_end = end,
L
Linus Torvalds 已提交
290 291 292 293 294 295 296 297 298 299 300 301
	};

	if (!mapping_cap_writeback_dirty(mapping))
		return 0;

	ret = do_writepages(mapping, &wbc);
	return ret;
}

static inline int __filemap_fdatawrite(struct address_space *mapping,
	int sync_mode)
{
302
	return __filemap_fdatawrite_range(mapping, 0, LLONG_MAX, sync_mode);
L
Linus Torvalds 已提交
303 304 305 306 307 308 309 310
}

int filemap_fdatawrite(struct address_space *mapping)
{
	return __filemap_fdatawrite(mapping, WB_SYNC_ALL);
}
EXPORT_SYMBOL(filemap_fdatawrite);

311
int filemap_fdatawrite_range(struct address_space *mapping, loff_t start,
312
				loff_t end)
L
Linus Torvalds 已提交
313 314 315
{
	return __filemap_fdatawrite_range(mapping, start, end, WB_SYNC_ALL);
}
316
EXPORT_SYMBOL(filemap_fdatawrite_range);
L
Linus Torvalds 已提交
317

318 319 320 321
/**
 * filemap_flush - mostly a non-blocking flush
 * @mapping:	target address_space
 *
L
Linus Torvalds 已提交
322 323 324 325 326 327 328 329 330
 * This is a mostly non-blocking flush.  Not suitable for data-integrity
 * purposes - I/O may not be started against all dirty pages.
 */
int filemap_flush(struct address_space *mapping)
{
	return __filemap_fdatawrite(mapping, WB_SYNC_NONE);
}
EXPORT_SYMBOL(filemap_flush);

331
/**
332 333 334 335
 * filemap_fdatawait_range - wait for writeback to complete
 * @mapping:		address space structure to wait for
 * @start_byte:		offset in bytes where the range starts
 * @end_byte:		offset in bytes where the range ends (inclusive)
336
 *
337 338
 * Walk the list of under-writeback pages of the given address space
 * in the given range and wait for all of them.
L
Linus Torvalds 已提交
339
 */
340 341
int filemap_fdatawait_range(struct address_space *mapping, loff_t start_byte,
			    loff_t end_byte)
L
Linus Torvalds 已提交
342
{
343 344
	pgoff_t index = start_byte >> PAGE_CACHE_SHIFT;
	pgoff_t end = end_byte >> PAGE_CACHE_SHIFT;
L
Linus Torvalds 已提交
345 346
	struct pagevec pvec;
	int nr_pages;
347
	int ret2, ret = 0;
L
Linus Torvalds 已提交
348

349
	if (end_byte < start_byte)
350
		goto out;
L
Linus Torvalds 已提交
351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366

	pagevec_init(&pvec, 0);
	while ((index <= end) &&
			(nr_pages = pagevec_lookup_tag(&pvec, mapping, &index,
			PAGECACHE_TAG_WRITEBACK,
			min(end - index, (pgoff_t)PAGEVEC_SIZE-1) + 1)) != 0) {
		unsigned i;

		for (i = 0; i < nr_pages; i++) {
			struct page *page = pvec.pages[i];

			/* until radix tree lookup accepts end_index */
			if (page->index > end)
				continue;

			wait_on_page_writeback(page);
367
			if (TestClearPageError(page))
L
Linus Torvalds 已提交
368 369 370 371 372
				ret = -EIO;
		}
		pagevec_release(&pvec);
		cond_resched();
	}
373 374 375 376
out:
	ret2 = filemap_check_errors(mapping);
	if (!ret)
		ret = ret2;
L
Linus Torvalds 已提交
377 378 379

	return ret;
}
380 381
EXPORT_SYMBOL(filemap_fdatawait_range);

L
Linus Torvalds 已提交
382
/**
383
 * filemap_fdatawait - wait for all under-writeback pages to complete
L
Linus Torvalds 已提交
384
 * @mapping: address space structure to wait for
385 386 387
 *
 * Walk the list of under-writeback pages of the given address space
 * and wait for all of them.
L
Linus Torvalds 已提交
388 389 390 391 392 393 394 395
 */
int filemap_fdatawait(struct address_space *mapping)
{
	loff_t i_size = i_size_read(mapping->host);

	if (i_size == 0)
		return 0;

396
	return filemap_fdatawait_range(mapping, 0, i_size - 1);
L
Linus Torvalds 已提交
397 398 399 400 401
}
EXPORT_SYMBOL(filemap_fdatawait);

int filemap_write_and_wait(struct address_space *mapping)
{
402
	int err = 0;
L
Linus Torvalds 已提交
403 404

	if (mapping->nrpages) {
405 406 407 408 409 410 411 412 413 414 415 416
		err = filemap_fdatawrite(mapping);
		/*
		 * Even if the above returned error, the pages may be
		 * written partially (e.g. -ENOSPC), so we wait for it.
		 * But the -EIO is special case, it may indicate the worst
		 * thing (e.g. bug) happened, so we avoid waiting for it.
		 */
		if (err != -EIO) {
			int err2 = filemap_fdatawait(mapping);
			if (!err)
				err = err2;
		}
417 418
	} else {
		err = filemap_check_errors(mapping);
L
Linus Torvalds 已提交
419
	}
420
	return err;
L
Linus Torvalds 已提交
421
}
422
EXPORT_SYMBOL(filemap_write_and_wait);
L
Linus Torvalds 已提交
423

424 425 426 427 428 429
/**
 * filemap_write_and_wait_range - write out & wait on a file range
 * @mapping:	the address_space for the pages
 * @lstart:	offset in bytes where the range starts
 * @lend:	offset in bytes where the range ends (inclusive)
 *
430 431 432 433 434
 * Write out and wait upon file offsets lstart->lend, inclusive.
 *
 * Note that `lend' is inclusive (describes the last byte to be written) so
 * that this function can be used to write to the very end-of-file (end = -1).
 */
L
Linus Torvalds 已提交
435 436 437
int filemap_write_and_wait_range(struct address_space *mapping,
				 loff_t lstart, loff_t lend)
{
438
	int err = 0;
L
Linus Torvalds 已提交
439 440

	if (mapping->nrpages) {
441 442 443 444
		err = __filemap_fdatawrite_range(mapping, lstart, lend,
						 WB_SYNC_ALL);
		/* See comment of filemap_write_and_wait() */
		if (err != -EIO) {
445 446
			int err2 = filemap_fdatawait_range(mapping,
						lstart, lend);
447 448 449
			if (!err)
				err = err2;
		}
450 451
	} else {
		err = filemap_check_errors(mapping);
L
Linus Torvalds 已提交
452
	}
453
	return err;
L
Linus Torvalds 已提交
454
}
455
EXPORT_SYMBOL(filemap_write_and_wait_range);
L
Linus Torvalds 已提交
456

457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475
/**
 * replace_page_cache_page - replace a pagecache page with a new one
 * @old:	page to be replaced
 * @new:	page to replace with
 * @gfp_mask:	allocation mode
 *
 * This function replaces a page in the pagecache with a new one.  On
 * success it acquires the pagecache reference for the new page and
 * drops it for the old page.  Both the old and new pages must be
 * locked.  This function does not add the new page to the LRU, the
 * caller must do that.
 *
 * The remove + add is atomic.  The only way this function can fail is
 * memory allocation failure.
 */
int replace_page_cache_page(struct page *old, struct page *new, gfp_t gfp_mask)
{
	int error;

476 477 478
	VM_BUG_ON_PAGE(!PageLocked(old), old);
	VM_BUG_ON_PAGE(!PageLocked(new), new);
	VM_BUG_ON_PAGE(new->mapping, new);
479 480 481 482 483 484 485 486 487 488 489 490 491 492

	error = radix_tree_preload(gfp_mask & ~__GFP_HIGHMEM);
	if (!error) {
		struct address_space *mapping = old->mapping;
		void (*freepage)(struct page *);

		pgoff_t offset = old->index;
		freepage = mapping->a_ops->freepage;

		page_cache_get(new);
		new->mapping = mapping;
		new->index = offset;

		spin_lock_irq(&mapping->tree_lock);
493
		__delete_from_page_cache(old, NULL);
494 495 496 497 498 499 500
		error = radix_tree_insert(&mapping->page_tree, offset, new);
		BUG_ON(error);
		mapping->nrpages++;
		__inc_zone_page_state(new, NR_FILE_PAGES);
		if (PageSwapBacked(new))
			__inc_zone_page_state(new, NR_SHMEM);
		spin_unlock_irq(&mapping->tree_lock);
501 502
		/* mem_cgroup codes must not be called under tree_lock */
		mem_cgroup_replace_page_cache(old, new);
503 504 505 506 507 508 509 510 511 512
		radix_tree_preload_end();
		if (freepage)
			freepage(old);
		page_cache_release(old);
	}

	return error;
}
EXPORT_SYMBOL_GPL(replace_page_cache_page);

513
static int page_cache_tree_insert(struct address_space *mapping,
514
				  struct page *page, void **shadowp)
515
{
516
	struct radix_tree_node *node;
517 518 519
	void **slot;
	int error;

520 521 522 523 524
	error = __radix_tree_create(&mapping->page_tree, page->index,
				    &node, &slot);
	if (error)
		return error;
	if (*slot) {
525 526 527 528 529
		void *p;

		p = radix_tree_deref_slot_protected(slot, &mapping->tree_lock);
		if (!radix_tree_exceptional_entry(p))
			return -EEXIST;
530 531
		if (shadowp)
			*shadowp = p;
532 533 534
		mapping->nrshadows--;
		if (node)
			workingset_node_shadows_dec(node);
535
	}
536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
	radix_tree_replace_slot(slot, page);
	mapping->nrpages++;
	if (node) {
		workingset_node_pages_inc(node);
		/*
		 * Don't track node that contains actual pages.
		 *
		 * Avoid acquiring the list_lru lock if already
		 * untracked.  The list_empty() test is safe as
		 * node->private_list is protected by
		 * mapping->tree_lock.
		 */
		if (!list_empty(&node->private_list))
			list_lru_del(&workingset_shadow_nodes,
				     &node->private_list);
	}
	return 0;
553 554
}

555 556 557 558
static int __add_to_page_cache_locked(struct page *page,
				      struct address_space *mapping,
				      pgoff_t offset, gfp_t gfp_mask,
				      void **shadowp)
L
Linus Torvalds 已提交
559
{
N
Nick Piggin 已提交
560 561
	int error;

562 563
	VM_BUG_ON_PAGE(!PageLocked(page), page);
	VM_BUG_ON_PAGE(PageSwapBacked(page), page);
N
Nick Piggin 已提交
564 565

	error = mem_cgroup_cache_charge(page, current->mm,
K
KAMEZAWA Hiroyuki 已提交
566
					gfp_mask & GFP_RECLAIM_MASK);
B
Balbir Singh 已提交
567
	if (error)
568
		return error;
L
Linus Torvalds 已提交
569

570
	error = radix_tree_maybe_preload(gfp_mask & ~__GFP_HIGHMEM);
571
	if (error) {
572
		mem_cgroup_uncharge_cache_page(page);
573 574 575 576 577 578 579 580
		return error;
	}

	page_cache_get(page);
	page->mapping = mapping;
	page->index = offset;

	spin_lock_irq(&mapping->tree_lock);
581
	error = page_cache_tree_insert(mapping, page, shadowp);
582 583 584 585 586 587 588 589 590 591 592 593 594
	radix_tree_preload_end();
	if (unlikely(error))
		goto err_insert;
	__inc_zone_page_state(page, NR_FILE_PAGES);
	spin_unlock_irq(&mapping->tree_lock);
	trace_mm_filemap_add_to_page_cache(page);
	return 0;
err_insert:
	page->mapping = NULL;
	/* Leave page->index set: truncation relies upon it */
	spin_unlock_irq(&mapping->tree_lock);
	mem_cgroup_uncharge_cache_page(page);
	page_cache_release(page);
L
Linus Torvalds 已提交
595 596
	return error;
}
597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613

/**
 * add_to_page_cache_locked - add a locked page to the pagecache
 * @page:	page to add
 * @mapping:	the page's address_space
 * @offset:	page index
 * @gfp_mask:	page allocation mode
 *
 * This function is used to add a page to the pagecache. It must be locked.
 * This function does not add the page to the LRU.  The caller must do that.
 */
int add_to_page_cache_locked(struct page *page, struct address_space *mapping,
		pgoff_t offset, gfp_t gfp_mask)
{
	return __add_to_page_cache_locked(page, mapping, offset,
					  gfp_mask, NULL);
}
N
Nick Piggin 已提交
614
EXPORT_SYMBOL(add_to_page_cache_locked);
L
Linus Torvalds 已提交
615 616

int add_to_page_cache_lru(struct page *page, struct address_space *mapping,
A
Al Viro 已提交
617
				pgoff_t offset, gfp_t gfp_mask)
L
Linus Torvalds 已提交
618
{
619
	void *shadow = NULL;
620 621
	int ret;

622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639
	__set_page_locked(page);
	ret = __add_to_page_cache_locked(page, mapping, offset,
					 gfp_mask, &shadow);
	if (unlikely(ret))
		__clear_page_locked(page);
	else {
		/*
		 * The page might have been evicted from cache only
		 * recently, in which case it should be activated like
		 * any other repeatedly accessed page.
		 */
		if (shadow && workingset_refault(shadow)) {
			SetPageActive(page);
			workingset_activation(page);
		} else
			ClearPageActive(page);
		lru_cache_add(page);
	}
L
Linus Torvalds 已提交
640 641
	return ret;
}
642
EXPORT_SYMBOL_GPL(add_to_page_cache_lru);
L
Linus Torvalds 已提交
643

644
#ifdef CONFIG_NUMA
645
struct page *__page_cache_alloc(gfp_t gfp)
646
{
647 648 649
	int n;
	struct page *page;

650
	if (cpuset_do_page_mem_spread()) {
651 652
		unsigned int cpuset_mems_cookie;
		do {
653
			cpuset_mems_cookie = read_mems_allowed_begin();
654 655
			n = cpuset_mem_spread_node();
			page = alloc_pages_exact_node(n, gfp, 0);
656
		} while (!page && read_mems_allowed_retry(cpuset_mems_cookie));
657

658
		return page;
659
	}
660
	return alloc_pages(gfp, 0);
661
}
662
EXPORT_SYMBOL(__page_cache_alloc);
663 664
#endif

L
Linus Torvalds 已提交
665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686
/*
 * In order to wait for pages to become available there must be
 * waitqueues associated with pages. By using a hash table of
 * waitqueues where the bucket discipline is to maintain all
 * waiters on the same queue and wake all when any of the pages
 * become available, and for the woken contexts to check to be
 * sure the appropriate page became available, this saves space
 * at a cost of "thundering herd" phenomena during rare hash
 * collisions.
 */
static wait_queue_head_t *page_waitqueue(struct page *page)
{
	const struct zone *zone = page_zone(page);

	return &zone->wait_table[hash_ptr(page, zone->wait_table_bits)];
}

static inline void wake_up_page(struct page *page, int bit)
{
	__wake_up_bit(page_waitqueue(page), &page->flags, bit);
}

H
Harvey Harrison 已提交
687
void wait_on_page_bit(struct page *page, int bit_nr)
L
Linus Torvalds 已提交
688 689 690 691
{
	DEFINE_WAIT_BIT(wait, &page->flags, bit_nr);

	if (test_bit(bit_nr, &page->flags))
J
Jens Axboe 已提交
692
		__wait_on_bit(page_waitqueue(page), &wait, sleep_on_page,
L
Linus Torvalds 已提交
693 694 695 696
							TASK_UNINTERRUPTIBLE);
}
EXPORT_SYMBOL(wait_on_page_bit);

697 698 699 700 701 702 703 704 705 706 707
int wait_on_page_bit_killable(struct page *page, int bit_nr)
{
	DEFINE_WAIT_BIT(wait, &page->flags, bit_nr);

	if (!test_bit(bit_nr, &page->flags))
		return 0;

	return __wait_on_bit(page_waitqueue(page), &wait,
			     sleep_on_page_killable, TASK_KILLABLE);
}

708 709
/**
 * add_page_wait_queue - Add an arbitrary waiter to a page's wait queue
R
Randy Dunlap 已提交
710 711
 * @page: Page defining the wait queue of interest
 * @waiter: Waiter to add to the queue
712 713 714 715 716 717 718 719 720 721 722 723 724 725
 *
 * Add an arbitrary @waiter to the wait queue for the nominated @page.
 */
void add_page_wait_queue(struct page *page, wait_queue_t *waiter)
{
	wait_queue_head_t *q = page_waitqueue(page);
	unsigned long flags;

	spin_lock_irqsave(&q->lock, flags);
	__add_wait_queue(q, waiter);
	spin_unlock_irqrestore(&q->lock, flags);
}
EXPORT_SYMBOL_GPL(add_page_wait_queue);

L
Linus Torvalds 已提交
726
/**
727
 * unlock_page - unlock a locked page
L
Linus Torvalds 已提交
728 729 730 731 732 733 734
 * @page: the page
 *
 * Unlocks the page and wakes up sleepers in ___wait_on_page_locked().
 * Also wakes sleepers in wait_on_page_writeback() because the wakeup
 * mechananism between PageLocked pages and PageWriteback pages is shared.
 * But that's OK - sleepers in wait_on_page_writeback() just go back to sleep.
 *
N
Nick Piggin 已提交
735 736
 * The mb is necessary to enforce ordering between the clear_bit and the read
 * of the waitqueue (to avoid SMP races with a parallel wait_on_page_locked()).
L
Linus Torvalds 已提交
737
 */
H
Harvey Harrison 已提交
738
void unlock_page(struct page *page)
L
Linus Torvalds 已提交
739
{
740
	VM_BUG_ON_PAGE(!PageLocked(page), page);
N
Nick Piggin 已提交
741 742
	clear_bit_unlock(PG_locked, &page->flags);
	smp_mb__after_clear_bit();
L
Linus Torvalds 已提交
743 744 745 746
	wake_up_page(page, PG_locked);
}
EXPORT_SYMBOL(unlock_page);

747 748 749
/**
 * end_page_writeback - end writeback against a page
 * @page: the page
L
Linus Torvalds 已提交
750 751 752
 */
void end_page_writeback(struct page *page)
{
753 754 755 756 757 758
	if (TestClearPageReclaim(page))
		rotate_reclaimable_page(page);

	if (!test_clear_page_writeback(page))
		BUG();

L
Linus Torvalds 已提交
759 760 761 762 763
	smp_mb__after_clear_bit();
	wake_up_page(page, PG_writeback);
}
EXPORT_SYMBOL(end_page_writeback);

764 765 766
/**
 * __lock_page - get a lock on the page, assuming we need to sleep to get it
 * @page: the page to lock
L
Linus Torvalds 已提交
767
 */
H
Harvey Harrison 已提交
768
void __lock_page(struct page *page)
L
Linus Torvalds 已提交
769 770 771
{
	DEFINE_WAIT_BIT(wait, &page->flags, PG_locked);

J
Jens Axboe 已提交
772
	__wait_on_bit_lock(page_waitqueue(page), &wait, sleep_on_page,
L
Linus Torvalds 已提交
773 774 775 776
							TASK_UNINTERRUPTIBLE);
}
EXPORT_SYMBOL(__lock_page);

H
Harvey Harrison 已提交
777
int __lock_page_killable(struct page *page)
M
Matthew Wilcox 已提交
778 779 780 781
{
	DEFINE_WAIT_BIT(wait, &page->flags, PG_locked);

	return __wait_on_bit_lock(page_waitqueue(page), &wait,
J
Jens Axboe 已提交
782
					sleep_on_page_killable, TASK_KILLABLE);
M
Matthew Wilcox 已提交
783
}
784
EXPORT_SYMBOL_GPL(__lock_page_killable);
M
Matthew Wilcox 已提交
785

786 787 788
int __lock_page_or_retry(struct page *page, struct mm_struct *mm,
			 unsigned int flags)
{
789 790 791 792 793 794 795 796 797 798 799 800
	if (flags & FAULT_FLAG_ALLOW_RETRY) {
		/*
		 * CAUTION! In this case, mmap_sem is not released
		 * even though return 0.
		 */
		if (flags & FAULT_FLAG_RETRY_NOWAIT)
			return 0;

		up_read(&mm->mmap_sem);
		if (flags & FAULT_FLAG_KILLABLE)
			wait_on_page_locked_killable(page);
		else
801
			wait_on_page_locked(page);
802
		return 0;
803 804 805 806 807 808 809 810 811 812 813 814
	} else {
		if (flags & FAULT_FLAG_KILLABLE) {
			int ret;

			ret = __lock_page_killable(page);
			if (ret) {
				up_read(&mm->mmap_sem);
				return 0;
			}
		} else
			__lock_page(page);
		return 1;
815 816 817
	}
}

818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844
/**
 * page_cache_next_hole - find the next hole (not-present entry)
 * @mapping: mapping
 * @index: index
 * @max_scan: maximum range to search
 *
 * Search the set [index, min(index+max_scan-1, MAX_INDEX)] for the
 * lowest indexed hole.
 *
 * Returns: the index of the hole if found, otherwise returns an index
 * outside of the set specified (in which case 'return - index >=
 * max_scan' will be true). In rare cases of index wrap-around, 0 will
 * be returned.
 *
 * page_cache_next_hole may be called under rcu_read_lock. However,
 * like radix_tree_gang_lookup, this will not atomically search a
 * snapshot of the tree at a single point in time. For example, if a
 * hole is created at index 5, then subsequently a hole is created at
 * index 10, page_cache_next_hole covering both indexes may return 10
 * if called under rcu_read_lock.
 */
pgoff_t page_cache_next_hole(struct address_space *mapping,
			     pgoff_t index, unsigned long max_scan)
{
	unsigned long i;

	for (i = 0; i < max_scan; i++) {
845 846 847 848
		struct page *page;

		page = radix_tree_lookup(&mapping->page_tree, index);
		if (!page || radix_tree_exceptional_entry(page))
849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885
			break;
		index++;
		if (index == 0)
			break;
	}

	return index;
}
EXPORT_SYMBOL(page_cache_next_hole);

/**
 * page_cache_prev_hole - find the prev hole (not-present entry)
 * @mapping: mapping
 * @index: index
 * @max_scan: maximum range to search
 *
 * Search backwards in the range [max(index-max_scan+1, 0), index] for
 * the first hole.
 *
 * Returns: the index of the hole if found, otherwise returns an index
 * outside of the set specified (in which case 'index - return >=
 * max_scan' will be true). In rare cases of wrap-around, ULONG_MAX
 * will be returned.
 *
 * page_cache_prev_hole may be called under rcu_read_lock. However,
 * like radix_tree_gang_lookup, this will not atomically search a
 * snapshot of the tree at a single point in time. For example, if a
 * hole is created at index 10, then subsequently a hole is created at
 * index 5, page_cache_prev_hole covering both indexes may return 5 if
 * called under rcu_read_lock.
 */
pgoff_t page_cache_prev_hole(struct address_space *mapping,
			     pgoff_t index, unsigned long max_scan)
{
	unsigned long i;

	for (i = 0; i < max_scan; i++) {
886 887 888 889
		struct page *page;

		page = radix_tree_lookup(&mapping->page_tree, index);
		if (!page || radix_tree_exceptional_entry(page))
890 891 892 893 894 895 896 897 898 899
			break;
		index--;
		if (index == ULONG_MAX)
			break;
	}

	return index;
}
EXPORT_SYMBOL(page_cache_prev_hole);

900
/**
901
 * find_get_entry - find and get a page cache entry
902
 * @mapping: the address_space to search
903 904 905 906
 * @offset: the page cache index
 *
 * Looks up the page cache slot at @mapping & @offset.  If there is a
 * page cache page, it is returned with an increased refcount.
907
 *
908 909 910 911
 * If the slot holds a shadow entry of a previously evicted page, it
 * is returned.
 *
 * Otherwise, %NULL is returned.
L
Linus Torvalds 已提交
912
 */
913
struct page *find_get_entry(struct address_space *mapping, pgoff_t offset)
L
Linus Torvalds 已提交
914
{
N
Nick Piggin 已提交
915
	void **pagep;
L
Linus Torvalds 已提交
916 917
	struct page *page;

N
Nick Piggin 已提交
918 919 920 921 922 923
	rcu_read_lock();
repeat:
	page = NULL;
	pagep = radix_tree_lookup_slot(&mapping->page_tree, offset);
	if (pagep) {
		page = radix_tree_deref_slot(pagep);
N
Nick Piggin 已提交
924 925
		if (unlikely(!page))
			goto out;
926
		if (radix_tree_exception(page)) {
927 928 929 930 931 932 933 934
			if (radix_tree_deref_retry(page))
				goto repeat;
			/*
			 * Otherwise, shmem/tmpfs must be storing a swap entry
			 * here as an exceptional entry: so return it without
			 * attempting to raise page count.
			 */
			goto out;
935
		}
N
Nick Piggin 已提交
936 937 938 939 940 941 942 943 944 945 946 947 948
		if (!page_cache_get_speculative(page))
			goto repeat;

		/*
		 * Has the page moved?
		 * This is part of the lockless pagecache protocol. See
		 * include/linux/pagemap.h for details.
		 */
		if (unlikely(page != *pagep)) {
			page_cache_release(page);
			goto repeat;
		}
	}
N
Nick Piggin 已提交
949
out:
N
Nick Piggin 已提交
950 951
	rcu_read_unlock();

L
Linus Torvalds 已提交
952 953
	return page;
}
954
EXPORT_SYMBOL(find_get_entry);
L
Linus Torvalds 已提交
955 956

/**
957
 * find_get_page - find and get a page reference
958 959
 * @mapping: the address_space to search
 * @offset: the page index
L
Linus Torvalds 已提交
960
 *
961 962
 * Looks up the page cache slot at @mapping & @offset.  If there is a
 * page cache page, it is returned with an increased refcount.
L
Linus Torvalds 已提交
963
 *
964
 * Otherwise, %NULL is returned.
L
Linus Torvalds 已提交
965
 */
966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992
struct page *find_get_page(struct address_space *mapping, pgoff_t offset)
{
	struct page *page = find_get_entry(mapping, offset);

	if (radix_tree_exceptional_entry(page))
		page = NULL;
	return page;
}
EXPORT_SYMBOL(find_get_page);

/**
 * find_lock_entry - locate, pin and lock a page cache entry
 * @mapping: the address_space to search
 * @offset: the page cache index
 *
 * Looks up the page cache slot at @mapping & @offset.  If there is a
 * page cache page, it is returned locked and with an increased
 * refcount.
 *
 * If the slot holds a shadow entry of a previously evicted page, it
 * is returned.
 *
 * Otherwise, %NULL is returned.
 *
 * find_lock_entry() may sleep.
 */
struct page *find_lock_entry(struct address_space *mapping, pgoff_t offset)
L
Linus Torvalds 已提交
993 994 995 996
{
	struct page *page;

repeat:
997
	page = find_get_entry(mapping, offset);
998
	if (page && !radix_tree_exception(page)) {
N
Nick Piggin 已提交
999 1000 1001 1002 1003 1004
		lock_page(page);
		/* Has the page been truncated? */
		if (unlikely(page->mapping != mapping)) {
			unlock_page(page);
			page_cache_release(page);
			goto repeat;
L
Linus Torvalds 已提交
1005
		}
1006
		VM_BUG_ON_PAGE(page->index != offset, page);
L
Linus Torvalds 已提交
1007 1008 1009
	}
	return page;
}
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032
EXPORT_SYMBOL(find_lock_entry);

/**
 * find_lock_page - locate, pin and lock a pagecache page
 * @mapping: the address_space to search
 * @offset: the page index
 *
 * Looks up the page cache slot at @mapping & @offset.  If there is a
 * page cache page, it is returned locked and with an increased
 * refcount.
 *
 * Otherwise, %NULL is returned.
 *
 * find_lock_page() may sleep.
 */
struct page *find_lock_page(struct address_space *mapping, pgoff_t offset)
{
	struct page *page = find_lock_entry(mapping, offset);

	if (radix_tree_exceptional_entry(page))
		page = NULL;
	return page;
}
L
Linus Torvalds 已提交
1033 1034 1035 1036
EXPORT_SYMBOL(find_lock_page);

/**
 * find_or_create_page - locate or add a pagecache page
1037 1038 1039
 * @mapping: the page's address_space
 * @index: the page's index into the mapping
 * @gfp_mask: page allocation mode
L
Linus Torvalds 已提交
1040
 *
1041 1042 1043 1044 1045 1046 1047
 * Looks up the page cache slot at @mapping & @offset.  If there is a
 * page cache page, it is returned locked and with an increased
 * refcount.
 *
 * If the page is not present, a new page is allocated using @gfp_mask
 * and added to the page cache and the VM's LRU list.  The page is
 * returned locked and with an increased refcount.
L
Linus Torvalds 已提交
1048
 *
1049
 * On memory exhaustion, %NULL is returned.
L
Linus Torvalds 已提交
1050
 *
1051 1052
 * find_or_create_page() may sleep, even if @gfp_flags specifies an
 * atomic allocation!
L
Linus Torvalds 已提交
1053 1054
 */
struct page *find_or_create_page(struct address_space *mapping,
1055
		pgoff_t index, gfp_t gfp_mask)
L
Linus Torvalds 已提交
1056
{
N
Nick Piggin 已提交
1057
	struct page *page;
L
Linus Torvalds 已提交
1058 1059 1060 1061
	int err;
repeat:
	page = find_lock_page(mapping, index);
	if (!page) {
N
Nick Piggin 已提交
1062 1063 1064
		page = __page_cache_alloc(gfp_mask);
		if (!page)
			return NULL;
N
Nick Piggin 已提交
1065 1066 1067 1068 1069 1070 1071 1072
		/*
		 * We want a regular kernel memory (not highmem or DMA etc)
		 * allocation for the radix tree nodes, but we need to honour
		 * the context-specific requirements the caller has asked for.
		 * GFP_RECLAIM_MASK collects those requirements.
		 */
		err = add_to_page_cache_lru(page, mapping, index,
			(gfp_mask & GFP_RECLAIM_MASK));
N
Nick Piggin 已提交
1073 1074 1075 1076 1077
		if (unlikely(err)) {
			page_cache_release(page);
			page = NULL;
			if (err == -EEXIST)
				goto repeat;
L
Linus Torvalds 已提交
1078 1079 1080 1081 1082 1083
		}
	}
	return page;
}
EXPORT_SYMBOL(find_or_create_page);

1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153
/**
 * find_get_entries - gang pagecache lookup
 * @mapping:	The address_space to search
 * @start:	The starting page cache index
 * @nr_entries:	The maximum number of entries
 * @entries:	Where the resulting entries are placed
 * @indices:	The cache indices corresponding to the entries in @entries
 *
 * find_get_entries() will search for and return a group of up to
 * @nr_entries entries in the mapping.  The entries are placed at
 * @entries.  find_get_entries() takes a reference against any actual
 * pages it returns.
 *
 * The search returns a group of mapping-contiguous page cache entries
 * with ascending indexes.  There may be holes in the indices due to
 * not-present pages.
 *
 * Any shadow entries of evicted pages are included in the returned
 * array.
 *
 * find_get_entries() returns the number of pages and shadow entries
 * which were found.
 */
unsigned find_get_entries(struct address_space *mapping,
			  pgoff_t start, unsigned int nr_entries,
			  struct page **entries, pgoff_t *indices)
{
	void **slot;
	unsigned int ret = 0;
	struct radix_tree_iter iter;

	if (!nr_entries)
		return 0;

	rcu_read_lock();
restart:
	radix_tree_for_each_slot(slot, &mapping->page_tree, &iter, start) {
		struct page *page;
repeat:
		page = radix_tree_deref_slot(slot);
		if (unlikely(!page))
			continue;
		if (radix_tree_exception(page)) {
			if (radix_tree_deref_retry(page))
				goto restart;
			/*
			 * Otherwise, we must be storing a swap entry
			 * here as an exceptional entry: so return it
			 * without attempting to raise page count.
			 */
			goto export;
		}
		if (!page_cache_get_speculative(page))
			goto repeat;

		/* Has the page moved? */
		if (unlikely(page != *slot)) {
			page_cache_release(page);
			goto repeat;
		}
export:
		indices[ret] = iter.index;
		entries[ret] = page;
		if (++ret == nr_entries)
			break;
	}
	rcu_read_unlock();
	return ret;
}

L
Linus Torvalds 已提交
1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172
/**
 * find_get_pages - gang pagecache lookup
 * @mapping:	The address_space to search
 * @start:	The starting page index
 * @nr_pages:	The maximum number of pages
 * @pages:	Where the resulting pages are placed
 *
 * find_get_pages() will search for and return a group of up to
 * @nr_pages pages in the mapping.  The pages are placed at @pages.
 * find_get_pages() takes a reference against the returned pages.
 *
 * The search returns a group of mapping-contiguous pages with ascending
 * indexes.  There may be holes in the indices due to not-present pages.
 *
 * find_get_pages() returns the number of pages which were found.
 */
unsigned find_get_pages(struct address_space *mapping, pgoff_t start,
			    unsigned int nr_pages, struct page **pages)
{
1173 1174 1175 1176 1177 1178
	struct radix_tree_iter iter;
	void **slot;
	unsigned ret = 0;

	if (unlikely(!nr_pages))
		return 0;
N
Nick Piggin 已提交
1179 1180 1181

	rcu_read_lock();
restart:
1182
	radix_tree_for_each_slot(slot, &mapping->page_tree, &iter, start) {
N
Nick Piggin 已提交
1183 1184
		struct page *page;
repeat:
1185
		page = radix_tree_deref_slot(slot);
N
Nick Piggin 已提交
1186 1187
		if (unlikely(!page))
			continue;
1188

1189
		if (radix_tree_exception(page)) {
1190 1191 1192 1193 1194 1195
			if (radix_tree_deref_retry(page)) {
				/*
				 * Transient condition which can only trigger
				 * when entry at index 0 moves out of or back
				 * to root: none yet gotten, safe to restart.
				 */
1196
				WARN_ON(iter.index);
1197 1198
				goto restart;
			}
1199
			/*
1200 1201 1202
			 * Otherwise, shmem/tmpfs must be storing a swap entry
			 * here as an exceptional entry: so skip over it -
			 * we only reach this from invalidate_mapping_pages().
1203
			 */
1204
			continue;
N
Nick Piggin 已提交
1205
		}
N
Nick Piggin 已提交
1206 1207 1208 1209 1210

		if (!page_cache_get_speculative(page))
			goto repeat;

		/* Has the page moved? */
1211
		if (unlikely(page != *slot)) {
N
Nick Piggin 已提交
1212 1213 1214
			page_cache_release(page);
			goto repeat;
		}
L
Linus Torvalds 已提交
1215

N
Nick Piggin 已提交
1216
		pages[ret] = page;
1217 1218
		if (++ret == nr_pages)
			break;
N
Nick Piggin 已提交
1219
	}
1220

N
Nick Piggin 已提交
1221
	rcu_read_unlock();
L
Linus Torvalds 已提交
1222 1223 1224
	return ret;
}

1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239
/**
 * find_get_pages_contig - gang contiguous pagecache lookup
 * @mapping:	The address_space to search
 * @index:	The starting page index
 * @nr_pages:	The maximum number of pages
 * @pages:	Where the resulting pages are placed
 *
 * find_get_pages_contig() works exactly like find_get_pages(), except
 * that the returned number of pages are guaranteed to be contiguous.
 *
 * find_get_pages_contig() returns the number of pages which were found.
 */
unsigned find_get_pages_contig(struct address_space *mapping, pgoff_t index,
			       unsigned int nr_pages, struct page **pages)
{
1240 1241 1242 1243 1244 1245
	struct radix_tree_iter iter;
	void **slot;
	unsigned int ret = 0;

	if (unlikely(!nr_pages))
		return 0;
N
Nick Piggin 已提交
1246 1247 1248

	rcu_read_lock();
restart:
1249
	radix_tree_for_each_contig(slot, &mapping->page_tree, &iter, index) {
N
Nick Piggin 已提交
1250 1251
		struct page *page;
repeat:
1252 1253
		page = radix_tree_deref_slot(slot);
		/* The hole, there no reason to continue */
N
Nick Piggin 已提交
1254
		if (unlikely(!page))
1255
			break;
1256

1257
		if (radix_tree_exception(page)) {
1258 1259 1260 1261 1262 1263 1264 1265
			if (radix_tree_deref_retry(page)) {
				/*
				 * Transient condition which can only trigger
				 * when entry at index 0 moves out of or back
				 * to root: none yet gotten, safe to restart.
				 */
				goto restart;
			}
1266
			/*
1267 1268 1269
			 * Otherwise, shmem/tmpfs must be storing a swap entry
			 * here as an exceptional entry: so stop looking for
			 * contiguous pages.
1270
			 */
1271
			break;
1272
		}
1273

N
Nick Piggin 已提交
1274 1275 1276 1277
		if (!page_cache_get_speculative(page))
			goto repeat;

		/* Has the page moved? */
1278
		if (unlikely(page != *slot)) {
N
Nick Piggin 已提交
1279 1280 1281 1282
			page_cache_release(page);
			goto repeat;
		}

N
Nick Piggin 已提交
1283 1284 1285 1286 1287
		/*
		 * must check mapping and index after taking the ref.
		 * otherwise we can get both false positives and false
		 * negatives, which is just confusing to the caller.
		 */
1288
		if (page->mapping == NULL || page->index != iter.index) {
N
Nick Piggin 已提交
1289 1290 1291 1292
			page_cache_release(page);
			break;
		}

N
Nick Piggin 已提交
1293
		pages[ret] = page;
1294 1295
		if (++ret == nr_pages)
			break;
1296
	}
N
Nick Piggin 已提交
1297 1298
	rcu_read_unlock();
	return ret;
1299
}
1300
EXPORT_SYMBOL(find_get_pages_contig);
1301

1302 1303 1304 1305 1306 1307 1308 1309
/**
 * find_get_pages_tag - find and return pages that match @tag
 * @mapping:	the address_space to search
 * @index:	the starting page index
 * @tag:	the tag index
 * @nr_pages:	the maximum number of pages
 * @pages:	where the resulting pages are placed
 *
L
Linus Torvalds 已提交
1310
 * Like find_get_pages, except we only return pages which are tagged with
1311
 * @tag.   We update @index to index the next page for the traversal.
L
Linus Torvalds 已提交
1312 1313 1314 1315
 */
unsigned find_get_pages_tag(struct address_space *mapping, pgoff_t *index,
			int tag, unsigned int nr_pages, struct page **pages)
{
1316 1317 1318 1319 1320 1321
	struct radix_tree_iter iter;
	void **slot;
	unsigned ret = 0;

	if (unlikely(!nr_pages))
		return 0;
N
Nick Piggin 已提交
1322 1323 1324

	rcu_read_lock();
restart:
1325 1326
	radix_tree_for_each_tagged(slot, &mapping->page_tree,
				   &iter, *index, tag) {
N
Nick Piggin 已提交
1327 1328
		struct page *page;
repeat:
1329
		page = radix_tree_deref_slot(slot);
N
Nick Piggin 已提交
1330 1331
		if (unlikely(!page))
			continue;
1332

1333
		if (radix_tree_exception(page)) {
1334 1335 1336 1337 1338 1339 1340 1341
			if (radix_tree_deref_retry(page)) {
				/*
				 * Transient condition which can only trigger
				 * when entry at index 0 moves out of or back
				 * to root: none yet gotten, safe to restart.
				 */
				goto restart;
			}
1342
			/*
1343 1344
			 * This function is never used on a shmem/tmpfs
			 * mapping, so a swap entry won't be found here.
1345
			 */
1346
			BUG();
1347
		}
N
Nick Piggin 已提交
1348 1349 1350 1351 1352

		if (!page_cache_get_speculative(page))
			goto repeat;

		/* Has the page moved? */
1353
		if (unlikely(page != *slot)) {
N
Nick Piggin 已提交
1354 1355 1356 1357 1358
			page_cache_release(page);
			goto repeat;
		}

		pages[ret] = page;
1359 1360
		if (++ret == nr_pages)
			break;
N
Nick Piggin 已提交
1361
	}
1362

N
Nick Piggin 已提交
1363
	rcu_read_unlock();
L
Linus Torvalds 已提交
1364 1365 1366

	if (ret)
		*index = pages[ret - 1]->index + 1;
N
Nick Piggin 已提交
1367

L
Linus Torvalds 已提交
1368 1369
	return ret;
}
1370
EXPORT_SYMBOL(find_get_pages_tag);
L
Linus Torvalds 已提交
1371

1372 1373 1374 1375 1376
/**
 * grab_cache_page_nowait - returns locked page at given index in given cache
 * @mapping: target address_space
 * @index: the page index
 *
1377
 * Same as grab_cache_page(), but do not wait if the page is unavailable.
L
Linus Torvalds 已提交
1378 1379 1380 1381 1382 1383 1384 1385
 * This is intended for speculative data generators, where the data can
 * be regenerated if the page couldn't be grabbed.  This routine should
 * be safe to call while holding the lock for another page.
 *
 * Clear __GFP_FS when allocating the page to avoid recursion into the fs
 * and deadlock against the caller's locked page.
 */
struct page *
1386
grab_cache_page_nowait(struct address_space *mapping, pgoff_t index)
L
Linus Torvalds 已提交
1387 1388 1389 1390
{
	struct page *page = find_get_page(mapping, index);

	if (page) {
N
Nick Piggin 已提交
1391
		if (trylock_page(page))
L
Linus Torvalds 已提交
1392 1393 1394 1395
			return page;
		page_cache_release(page);
		return NULL;
	}
1396
	page = __page_cache_alloc(mapping_gfp_mask(mapping) & ~__GFP_FS);
N
Nick Piggin 已提交
1397
	if (page && add_to_page_cache_lru(page, mapping, index, GFP_NOFS)) {
L
Linus Torvalds 已提交
1398 1399 1400 1401 1402 1403 1404
		page_cache_release(page);
		page = NULL;
	}
	return page;
}
EXPORT_SYMBOL(grab_cache_page_nowait);

1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425
/*
 * CD/DVDs are error prone. When a medium error occurs, the driver may fail
 * a _large_ part of the i/o request. Imagine the worst scenario:
 *
 *      ---R__________________________________________B__________
 *         ^ reading here                             ^ bad block(assume 4k)
 *
 * read(R) => miss => readahead(R...B) => media error => frustrating retries
 * => failing the whole request => read(R) => read(R+1) =>
 * readahead(R+1...B+1) => bang => read(R+2) => read(R+3) =>
 * readahead(R+3...B+2) => bang => read(R+3) => read(R+4) =>
 * readahead(R+4...B+3) => bang => read(R+4) => read(R+5) => ......
 *
 * It is going insane. Fix it by quickly scaling down the readahead size.
 */
static void shrink_readahead_size_eio(struct file *filp,
					struct file_ra_state *ra)
{
	ra->ra_pages /= 4;
}

1426
/**
C
Christoph Hellwig 已提交
1427
 * do_generic_file_read - generic file read routine
1428 1429 1430 1431
 * @filp:	the file to read
 * @ppos:	current file position
 * @desc:	read_descriptor
 *
L
Linus Torvalds 已提交
1432
 * This is a generic file read routine, and uses the
1433
 * mapping->a_ops->readpage() function for the actual low-level stuff.
L
Linus Torvalds 已提交
1434 1435 1436 1437
 *
 * This is really ugly. But the goto's actually try to clarify some
 * of the logic when it comes to error handling etc.
 */
C
Christoph Hellwig 已提交
1438
static void do_generic_file_read(struct file *filp, loff_t *ppos,
1439
		read_descriptor_t *desc)
L
Linus Torvalds 已提交
1440
{
C
Christoph Hellwig 已提交
1441
	struct address_space *mapping = filp->f_mapping;
L
Linus Torvalds 已提交
1442
	struct inode *inode = mapping->host;
C
Christoph Hellwig 已提交
1443
	struct file_ra_state *ra = &filp->f_ra;
1444 1445 1446 1447
	pgoff_t index;
	pgoff_t last_index;
	pgoff_t prev_index;
	unsigned long offset;      /* offset into pagecache page */
1448
	unsigned int prev_offset;
L
Linus Torvalds 已提交
1449 1450 1451
	int error;

	index = *ppos >> PAGE_CACHE_SHIFT;
1452 1453
	prev_index = ra->prev_pos >> PAGE_CACHE_SHIFT;
	prev_offset = ra->prev_pos & (PAGE_CACHE_SIZE-1);
L
Linus Torvalds 已提交
1454 1455 1456 1457 1458
	last_index = (*ppos + desc->count + PAGE_CACHE_SIZE-1) >> PAGE_CACHE_SHIFT;
	offset = *ppos & ~PAGE_CACHE_MASK;

	for (;;) {
		struct page *page;
1459
		pgoff_t end_index;
N
NeilBrown 已提交
1460
		loff_t isize;
L
Linus Torvalds 已提交
1461 1462 1463 1464 1465
		unsigned long nr, ret;

		cond_resched();
find_page:
		page = find_get_page(mapping, index);
1466
		if (!page) {
1467
			page_cache_sync_readahead(mapping,
1468
					ra, filp,
1469 1470 1471 1472 1473 1474
					index, last_index - index);
			page = find_get_page(mapping, index);
			if (unlikely(page == NULL))
				goto no_cached_page;
		}
		if (PageReadahead(page)) {
1475
			page_cache_async_readahead(mapping,
1476
					ra, filp, page,
1477
					index, last_index - index);
L
Linus Torvalds 已提交
1478
		}
1479 1480 1481 1482
		if (!PageUptodate(page)) {
			if (inode->i_blkbits == PAGE_CACHE_SHIFT ||
					!mapping->a_ops->is_partially_uptodate)
				goto page_not_up_to_date;
N
Nick Piggin 已提交
1483
			if (!trylock_page(page))
1484
				goto page_not_up_to_date;
1485 1486 1487
			/* Did it get truncated before we got the lock? */
			if (!page->mapping)
				goto page_not_up_to_date_locked;
1488 1489 1490 1491 1492
			if (!mapping->a_ops->is_partially_uptodate(page,
								desc, offset))
				goto page_not_up_to_date_locked;
			unlock_page(page);
		}
L
Linus Torvalds 已提交
1493
page_ok:
N
NeilBrown 已提交
1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519
		/*
		 * i_size must be checked after we know the page is Uptodate.
		 *
		 * Checking i_size after the check allows us to calculate
		 * the correct value for "nr", which means the zero-filled
		 * part of the page is not copied back to userspace (unless
		 * another truncate extends the file - this is desired though).
		 */

		isize = i_size_read(inode);
		end_index = (isize - 1) >> PAGE_CACHE_SHIFT;
		if (unlikely(!isize || index > end_index)) {
			page_cache_release(page);
			goto out;
		}

		/* nr is the maximum number of bytes to copy from this page */
		nr = PAGE_CACHE_SIZE;
		if (index == end_index) {
			nr = ((isize - 1) & ~PAGE_CACHE_MASK) + 1;
			if (nr <= offset) {
				page_cache_release(page);
				goto out;
			}
		}
		nr = nr - offset;
L
Linus Torvalds 已提交
1520 1521 1522 1523 1524 1525 1526 1527 1528

		/* If users can be writing to this page using arbitrary
		 * virtual addresses, take care about potential aliasing
		 * before reading the page on the kernel side.
		 */
		if (mapping_writably_mapped(mapping))
			flush_dcache_page(page);

		/*
1529 1530
		 * When a sequential read accesses a page several times,
		 * only mark it as accessed the first time.
L
Linus Torvalds 已提交
1531
		 */
1532
		if (prev_index != index || offset != prev_offset)
L
Linus Torvalds 已提交
1533 1534 1535 1536 1537 1538 1539
			mark_page_accessed(page);
		prev_index = index;

		/*
		 * Ok, we have the page, and it's up-to-date, so
		 * now we can copy it to user space...
		 *
1540 1541
		 * The file_read_actor routine returns how many bytes were
		 * actually used..
L
Linus Torvalds 已提交
1542 1543 1544 1545 1546
		 * NOTE! This may not be the same as how much of a user buffer
		 * we filled up (we may be padding etc), so we can only update
		 * "pos" here (the actor routine has to update the user buffer
		 * pointers and the remaining count).
		 */
1547
		ret = file_read_actor(desc, page, offset, nr);
L
Linus Torvalds 已提交
1548 1549 1550
		offset += ret;
		index += offset >> PAGE_CACHE_SHIFT;
		offset &= ~PAGE_CACHE_MASK;
J
Jan Kara 已提交
1551
		prev_offset = offset;
L
Linus Torvalds 已提交
1552 1553 1554 1555 1556 1557 1558 1559

		page_cache_release(page);
		if (ret == nr && desc->count)
			continue;
		goto out;

page_not_up_to_date:
		/* Get exclusive access to the page ... */
1560 1561 1562
		error = lock_page_killable(page);
		if (unlikely(error))
			goto readpage_error;
L
Linus Torvalds 已提交
1563

1564
page_not_up_to_date_locked:
N
Nick Piggin 已提交
1565
		/* Did it get truncated before we got the lock? */
L
Linus Torvalds 已提交
1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578
		if (!page->mapping) {
			unlock_page(page);
			page_cache_release(page);
			continue;
		}

		/* Did somebody else fill it already? */
		if (PageUptodate(page)) {
			unlock_page(page);
			goto page_ok;
		}

readpage:
1579 1580 1581 1582 1583 1584
		/*
		 * A previous I/O error may have been due to temporary
		 * failures, eg. multipath errors.
		 * PG_error will be set again if readpage fails.
		 */
		ClearPageError(page);
L
Linus Torvalds 已提交
1585 1586 1587
		/* Start the actual read. The read will unlock the page. */
		error = mapping->a_ops->readpage(filp, page);

1588 1589 1590 1591 1592
		if (unlikely(error)) {
			if (error == AOP_TRUNCATED_PAGE) {
				page_cache_release(page);
				goto find_page;
			}
L
Linus Torvalds 已提交
1593
			goto readpage_error;
1594
		}
L
Linus Torvalds 已提交
1595 1596

		if (!PageUptodate(page)) {
1597 1598 1599
			error = lock_page_killable(page);
			if (unlikely(error))
				goto readpage_error;
L
Linus Torvalds 已提交
1600 1601 1602
			if (!PageUptodate(page)) {
				if (page->mapping == NULL) {
					/*
1603
					 * invalidate_mapping_pages got it
L
Linus Torvalds 已提交
1604 1605 1606 1607 1608 1609
					 */
					unlock_page(page);
					page_cache_release(page);
					goto find_page;
				}
				unlock_page(page);
1610
				shrink_readahead_size_eio(filp, ra);
1611 1612
				error = -EIO;
				goto readpage_error;
L
Linus Torvalds 已提交
1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629
			}
			unlock_page(page);
		}

		goto page_ok;

readpage_error:
		/* UHHUH! A synchronous read error occurred. Report it */
		desc->error = error;
		page_cache_release(page);
		goto out;

no_cached_page:
		/*
		 * Ok, it wasn't cached, so we need to create a new
		 * page..
		 */
N
Nick Piggin 已提交
1630 1631 1632 1633
		page = page_cache_alloc_cold(mapping);
		if (!page) {
			desc->error = -ENOMEM;
			goto out;
L
Linus Torvalds 已提交
1634
		}
N
Nick Piggin 已提交
1635
		error = add_to_page_cache_lru(page, mapping,
L
Linus Torvalds 已提交
1636 1637
						index, GFP_KERNEL);
		if (error) {
N
Nick Piggin 已提交
1638
			page_cache_release(page);
L
Linus Torvalds 已提交
1639 1640 1641 1642 1643 1644 1645 1646 1647
			if (error == -EEXIST)
				goto find_page;
			desc->error = error;
			goto out;
		}
		goto readpage;
	}

out:
1648 1649 1650
	ra->prev_pos = prev_index;
	ra->prev_pos <<= PAGE_CACHE_SHIFT;
	ra->prev_pos |= prev_offset;
L
Linus Torvalds 已提交
1651

1652
	*ppos = ((loff_t)index << PAGE_CACHE_SHIFT) + offset;
1653
	file_accessed(filp);
L
Linus Torvalds 已提交
1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669
}

int file_read_actor(read_descriptor_t *desc, struct page *page,
			unsigned long offset, unsigned long size)
{
	char *kaddr;
	unsigned long left, count = desc->count;

	if (size > count)
		size = count;

	/*
	 * Faults on the destination of a read are common, so do it before
	 * taking the kmap.
	 */
	if (!fault_in_pages_writeable(desc->arg.buf, size)) {
1670
		kaddr = kmap_atomic(page);
L
Linus Torvalds 已提交
1671 1672
		left = __copy_to_user_inatomic(desc->arg.buf,
						kaddr + offset, size);
1673
		kunmap_atomic(kaddr);
L
Linus Torvalds 已提交
1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693
		if (left == 0)
			goto success;
	}

	/* Do it the slow way */
	kaddr = kmap(page);
	left = __copy_to_user(desc->arg.buf, kaddr + offset, size);
	kunmap(page);

	if (left) {
		size -= left;
		desc->error = -EFAULT;
	}
success:
	desc->count = count - size;
	desc->written += size;
	desc->arg.buf += size;
	return size;
}

1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732
/*
 * Performs necessary checks before doing a write
 * @iov:	io vector request
 * @nr_segs:	number of segments in the iovec
 * @count:	number of bytes to write
 * @access_flags: type of access: %VERIFY_READ or %VERIFY_WRITE
 *
 * Adjust number of segments and amount of bytes to write (nr_segs should be
 * properly initialized first). Returns appropriate error code that caller
 * should return or zero in case that write should be allowed.
 */
int generic_segment_checks(const struct iovec *iov,
			unsigned long *nr_segs, size_t *count, int access_flags)
{
	unsigned long   seg;
	size_t cnt = 0;
	for (seg = 0; seg < *nr_segs; seg++) {
		const struct iovec *iv = &iov[seg];

		/*
		 * If any segment has a negative length, or the cumulative
		 * length ever wraps negative then return -EINVAL.
		 */
		cnt += iv->iov_len;
		if (unlikely((ssize_t)(cnt|iv->iov_len) < 0))
			return -EINVAL;
		if (access_ok(access_flags, iv->iov_base, iv->iov_len))
			continue;
		if (seg == 0)
			return -EFAULT;
		*nr_segs = seg;
		cnt -= iv->iov_len;	/* This segment is no good */
		break;
	}
	*count = cnt;
	return 0;
}
EXPORT_SYMBOL(generic_segment_checks);

1733
/**
H
Henrik Kretzschmar 已提交
1734
 * generic_file_aio_read - generic filesystem read routine
1735 1736 1737
 * @iocb:	kernel I/O control block
 * @iov:	io vector request
 * @nr_segs:	number of segments in the iovec
H
Henrik Kretzschmar 已提交
1738
 * @pos:	current file position
1739
 *
L
Linus Torvalds 已提交
1740 1741 1742 1743
 * This is the "read()" routine for all filesystems
 * that can use the page cache directly.
 */
ssize_t
1744 1745
generic_file_aio_read(struct kiocb *iocb, const struct iovec *iov,
		unsigned long nr_segs, loff_t pos)
L
Linus Torvalds 已提交
1746 1747 1748
{
	struct file *filp = iocb->ki_filp;
	ssize_t retval;
1749
	unsigned long seg = 0;
L
Linus Torvalds 已提交
1750
	size_t count;
1751
	loff_t *ppos = &iocb->ki_pos;
L
Linus Torvalds 已提交
1752 1753

	count = 0;
1754 1755 1756
	retval = generic_segment_checks(iov, &nr_segs, &count, VERIFY_WRITE);
	if (retval)
		return retval;
L
Linus Torvalds 已提交
1757 1758 1759

	/* coalesce the iovecs and go direct-to-BIO for O_DIRECT */
	if (filp->f_flags & O_DIRECT) {
1760
		loff_t size;
L
Linus Torvalds 已提交
1761 1762 1763 1764 1765 1766 1767 1768
		struct address_space *mapping;
		struct inode *inode;

		mapping = filp->f_mapping;
		inode = mapping->host;
		if (!count)
			goto out; /* skip atime */
		size = i_size_read(inode);
1769
		retval = filemap_write_and_wait_range(mapping, pos,
1770
					pos + iov_length(iov, nr_segs) - 1);
1771 1772 1773 1774 1775 1776 1777 1778
		if (!retval) {
			retval = mapping->a_ops->direct_IO(READ, iocb,
							   iov, pos, nr_segs);
		}
		if (retval > 0) {
			*ppos = pos + retval;
			count -= retval;
		}
1779

1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790
		/*
		 * Btrfs can have a short DIO read if we encounter
		 * compressed extents, so if there was an error, or if
		 * we've already read everything we wanted to, or if
		 * there was a short read because we hit EOF, go ahead
		 * and return.  Otherwise fallthrough to buffered io for
		 * the rest of the read.
		 */
		if (retval < 0 || !count || *ppos >= size) {
			file_accessed(filp);
			goto out;
1791
		}
L
Linus Torvalds 已提交
1792 1793
	}

1794
	count = retval;
H
Hugh Dickins 已提交
1795 1796
	for (seg = 0; seg < nr_segs; seg++) {
		read_descriptor_t desc;
1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810
		loff_t offset = 0;

		/*
		 * If we did a short DIO read we need to skip the section of the
		 * iov that we've already read data into.
		 */
		if (count) {
			if (count > iov[seg].iov_len) {
				count -= iov[seg].iov_len;
				continue;
			}
			offset = count;
			count = 0;
		}
L
Linus Torvalds 已提交
1811

H
Hugh Dickins 已提交
1812
		desc.written = 0;
1813 1814
		desc.arg.buf = iov[seg].iov_base + offset;
		desc.count = iov[seg].iov_len - offset;
H
Hugh Dickins 已提交
1815 1816 1817
		if (desc.count == 0)
			continue;
		desc.error = 0;
1818
		do_generic_file_read(filp, ppos, &desc);
H
Hugh Dickins 已提交
1819 1820 1821 1822
		retval += desc.written;
		if (desc.error) {
			retval = retval ?: desc.error;
			break;
L
Linus Torvalds 已提交
1823
		}
H
Hugh Dickins 已提交
1824 1825
		if (desc.count > 0)
			break;
L
Linus Torvalds 已提交
1826 1827 1828 1829 1830 1831 1832
	}
out:
	return retval;
}
EXPORT_SYMBOL(generic_file_aio_read);

#ifdef CONFIG_MMU
1833 1834 1835 1836 1837
/**
 * page_cache_read - adds requested page to the page cache if not already there
 * @file:	file to read
 * @offset:	page index
 *
L
Linus Torvalds 已提交
1838 1839 1840
 * This adds the requested page to the page cache if it isn't already there,
 * and schedules an I/O to read in its contents from disk.
 */
H
Harvey Harrison 已提交
1841
static int page_cache_read(struct file *file, pgoff_t offset)
L
Linus Torvalds 已提交
1842 1843 1844
{
	struct address_space *mapping = file->f_mapping;
	struct page *page; 
1845
	int ret;
L
Linus Torvalds 已提交
1846

1847 1848 1849 1850 1851 1852 1853 1854 1855 1856
	do {
		page = page_cache_alloc_cold(mapping);
		if (!page)
			return -ENOMEM;

		ret = add_to_page_cache_lru(page, mapping, offset, GFP_KERNEL);
		if (ret == 0)
			ret = mapping->a_ops->readpage(file, page);
		else if (ret == -EEXIST)
			ret = 0; /* losing race to add is OK */
L
Linus Torvalds 已提交
1857 1858 1859

		page_cache_release(page);

1860 1861 1862
	} while (ret == AOP_TRUNCATED_PAGE);
		
	return ret;
L
Linus Torvalds 已提交
1863 1864 1865 1866
}

#define MMAP_LOTSAMISS  (100)

1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879
/*
 * Synchronous readahead happens when we don't even find
 * a page in the page cache at all.
 */
static void do_sync_mmap_readahead(struct vm_area_struct *vma,
				   struct file_ra_state *ra,
				   struct file *file,
				   pgoff_t offset)
{
	unsigned long ra_pages;
	struct address_space *mapping = file->f_mapping;

	/* If we don't want any read-ahead, don't bother */
1880
	if (vma->vm_flags & VM_RAND_READ)
1881
		return;
1882 1883
	if (!ra->ra_pages)
		return;
1884

1885
	if (vma->vm_flags & VM_SEQ_READ) {
1886 1887
		page_cache_sync_readahead(mapping, ra, file, offset,
					  ra->ra_pages);
1888 1889 1890
		return;
	}

1891 1892
	/* Avoid banging the cache line if not needed */
	if (ra->mmap_miss < MMAP_LOTSAMISS * 10)
1893 1894 1895 1896 1897 1898 1899 1900 1901
		ra->mmap_miss++;

	/*
	 * Do we miss much more than hit in this file? If so,
	 * stop bothering with read-ahead. It will only hurt.
	 */
	if (ra->mmap_miss > MMAP_LOTSAMISS)
		return;

1902 1903 1904
	/*
	 * mmap read-around
	 */
1905
	ra_pages = max_sane_readahead(ra->ra_pages);
1906 1907
	ra->start = max_t(long, 0, offset - ra_pages / 2);
	ra->size = ra_pages;
1908
	ra->async_size = ra_pages / 4;
1909
	ra_submit(ra, mapping, file);
1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924
}

/*
 * Asynchronous readahead happens when we find the page and PG_readahead,
 * so we want to possibly extend the readahead further..
 */
static void do_async_mmap_readahead(struct vm_area_struct *vma,
				    struct file_ra_state *ra,
				    struct file *file,
				    struct page *page,
				    pgoff_t offset)
{
	struct address_space *mapping = file->f_mapping;

	/* If we don't want any read-ahead, don't bother */
1925
	if (vma->vm_flags & VM_RAND_READ)
1926 1927 1928 1929
		return;
	if (ra->mmap_miss > 0)
		ra->mmap_miss--;
	if (PageReadahead(page))
1930 1931
		page_cache_async_readahead(mapping, ra, file,
					   page, offset, ra->ra_pages);
1932 1933
}

1934
/**
1935
 * filemap_fault - read in file data for page fault handling
N
Nick Piggin 已提交
1936 1937
 * @vma:	vma in which the fault was taken
 * @vmf:	struct vm_fault containing details of the fault
1938
 *
1939
 * filemap_fault() is invoked via the vma operations vector for a
L
Linus Torvalds 已提交
1940 1941 1942 1943 1944 1945
 * mapped memory region to read in file data during a page fault.
 *
 * The goto's are kind of ugly, but this streamlines the normal case of having
 * it in the page cache, and handles the special cases reasonably without
 * having a lot of duplicated code.
 */
N
Nick Piggin 已提交
1946
int filemap_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
L
Linus Torvalds 已提交
1947 1948
{
	int error;
1949
	struct file *file = vma->vm_file;
L
Linus Torvalds 已提交
1950 1951 1952
	struct address_space *mapping = file->f_mapping;
	struct file_ra_state *ra = &file->f_ra;
	struct inode *inode = mapping->host;
1953
	pgoff_t offset = vmf->pgoff;
L
Linus Torvalds 已提交
1954
	struct page *page;
J
Jan Kara 已提交
1955
	pgoff_t size;
N
Nick Piggin 已提交
1956
	int ret = 0;
L
Linus Torvalds 已提交
1957 1958

	size = (i_size_read(inode) + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
1959
	if (offset >= size)
1960
		return VM_FAULT_SIGBUS;
L
Linus Torvalds 已提交
1961 1962

	/*
1963
	 * Do we have something in the page cache already?
L
Linus Torvalds 已提交
1964
	 */
1965
	page = find_get_page(mapping, offset);
1966
	if (likely(page) && !(vmf->flags & FAULT_FLAG_TRIED)) {
L
Linus Torvalds 已提交
1967
		/*
1968 1969
		 * We found the page, so try async readahead before
		 * waiting for the lock.
L
Linus Torvalds 已提交
1970
		 */
1971
		do_async_mmap_readahead(vma, ra, file, page, offset);
1972
	} else if (!page) {
1973 1974 1975
		/* No page in the page cache at all */
		do_sync_mmap_readahead(vma, ra, file, offset);
		count_vm_event(PGMAJFAULT);
1976
		mem_cgroup_count_vm_event(vma->vm_mm, PGMAJFAULT);
1977 1978
		ret = VM_FAULT_MAJOR;
retry_find:
1979
		page = find_get_page(mapping, offset);
L
Linus Torvalds 已提交
1980 1981 1982 1983
		if (!page)
			goto no_cached_page;
	}

1984 1985
	if (!lock_page_or_retry(page, vma->vm_mm, vmf->flags)) {
		page_cache_release(page);
1986
		return ret | VM_FAULT_RETRY;
1987
	}
1988 1989 1990 1991 1992 1993 1994

	/* Did it get truncated? */
	if (unlikely(page->mapping != mapping)) {
		unlock_page(page);
		put_page(page);
		goto retry_find;
	}
1995
	VM_BUG_ON_PAGE(page->index != offset, page);
1996

L
Linus Torvalds 已提交
1997
	/*
1998 1999
	 * We have a locked page in the page cache, now we need to check
	 * that it's up-to-date. If not, it is going to be due to an error.
L
Linus Torvalds 已提交
2000
	 */
2001
	if (unlikely(!PageUptodate(page)))
L
Linus Torvalds 已提交
2002 2003
		goto page_not_uptodate;

2004 2005 2006 2007
	/*
	 * Found the page and have a reference on it.
	 * We must recheck i_size under page lock.
	 */
2008
	size = (i_size_read(inode) + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
2009
	if (unlikely(offset >= size)) {
2010
		unlock_page(page);
2011
		page_cache_release(page);
2012
		return VM_FAULT_SIGBUS;
2013 2014
	}

N
Nick Piggin 已提交
2015
	vmf->page = page;
N
Nick Piggin 已提交
2016
	return ret | VM_FAULT_LOCKED;
L
Linus Torvalds 已提交
2017 2018 2019 2020 2021 2022

no_cached_page:
	/*
	 * We're only likely to ever get here if MADV_RANDOM is in
	 * effect.
	 */
2023
	error = page_cache_read(file, offset);
L
Linus Torvalds 已提交
2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038

	/*
	 * The page we want has now been added to the page cache.
	 * In the unlikely event that someone removed it in the
	 * meantime, we'll just come back here and read it again.
	 */
	if (error >= 0)
		goto retry_find;

	/*
	 * An error return from page_cache_read can result if the
	 * system is low on memory, or a problem occurs while trying
	 * to schedule I/O.
	 */
	if (error == -ENOMEM)
N
Nick Piggin 已提交
2039 2040
		return VM_FAULT_OOM;
	return VM_FAULT_SIGBUS;
L
Linus Torvalds 已提交
2041 2042 2043 2044 2045 2046 2047 2048 2049

page_not_uptodate:
	/*
	 * Umm, take care of errors if the page isn't up-to-date.
	 * Try to re-read it _once_. We do this synchronously,
	 * because there really aren't any performance issues here
	 * and we need to check for errors.
	 */
	ClearPageError(page);
2050
	error = mapping->a_ops->readpage(file, page);
2051 2052 2053 2054 2055
	if (!error) {
		wait_on_page_locked(page);
		if (!PageUptodate(page))
			error = -EIO;
	}
2056 2057 2058
	page_cache_release(page);

	if (!error || error == AOP_TRUNCATED_PAGE)
2059
		goto retry_find;
L
Linus Torvalds 已提交
2060

2061
	/* Things didn't work out. Return zero to tell the mm layer so. */
2062
	shrink_readahead_size_eio(file, ra);
N
Nick Piggin 已提交
2063
	return VM_FAULT_SIGBUS;
2064 2065 2066
}
EXPORT_SYMBOL(filemap_fault);

2067 2068 2069
int filemap_page_mkwrite(struct vm_area_struct *vma, struct vm_fault *vmf)
{
	struct page *page = vmf->page;
A
Al Viro 已提交
2070
	struct inode *inode = file_inode(vma->vm_file);
2071 2072
	int ret = VM_FAULT_LOCKED;

2073
	sb_start_pagefault(inode->i_sb);
2074 2075 2076 2077 2078 2079 2080
	file_update_time(vma->vm_file);
	lock_page(page);
	if (page->mapping != inode->i_mapping) {
		unlock_page(page);
		ret = VM_FAULT_NOPAGE;
		goto out;
	}
2081 2082 2083 2084 2085 2086
	/*
	 * We mark the page dirty already here so that when freeze is in
	 * progress, we are guaranteed that writeback during freezing will
	 * see the dirty page and writeprotect it again.
	 */
	set_page_dirty(page);
2087
	wait_for_stable_page(page);
2088
out:
2089
	sb_end_pagefault(inode->i_sb);
2090 2091 2092 2093
	return ret;
}
EXPORT_SYMBOL(filemap_page_mkwrite);

2094
const struct vm_operations_struct generic_file_vm_ops = {
2095
	.fault		= filemap_fault,
2096
	.page_mkwrite	= filemap_page_mkwrite,
2097
	.remap_pages	= generic_file_remap_pages,
L
Linus Torvalds 已提交
2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135
};

/* This is used for a general mmap of a disk file */

int generic_file_mmap(struct file * file, struct vm_area_struct * vma)
{
	struct address_space *mapping = file->f_mapping;

	if (!mapping->a_ops->readpage)
		return -ENOEXEC;
	file_accessed(file);
	vma->vm_ops = &generic_file_vm_ops;
	return 0;
}

/*
 * This is for filesystems which do not implement ->writepage.
 */
int generic_file_readonly_mmap(struct file *file, struct vm_area_struct *vma)
{
	if ((vma->vm_flags & VM_SHARED) && (vma->vm_flags & VM_MAYWRITE))
		return -EINVAL;
	return generic_file_mmap(file, vma);
}
#else
int generic_file_mmap(struct file * file, struct vm_area_struct * vma)
{
	return -ENOSYS;
}
int generic_file_readonly_mmap(struct file * file, struct vm_area_struct * vma)
{
	return -ENOSYS;
}
#endif /* CONFIG_MMU */

EXPORT_SYMBOL(generic_file_mmap);
EXPORT_SYMBOL(generic_file_readonly_mmap);

2136
static struct page *__read_cache_page(struct address_space *mapping,
2137
				pgoff_t index,
2138
				int (*filler)(void *, struct page *),
2139 2140
				void *data,
				gfp_t gfp)
L
Linus Torvalds 已提交
2141
{
N
Nick Piggin 已提交
2142
	struct page *page;
L
Linus Torvalds 已提交
2143 2144 2145 2146
	int err;
repeat:
	page = find_get_page(mapping, index);
	if (!page) {
2147
		page = __page_cache_alloc(gfp | __GFP_COLD);
N
Nick Piggin 已提交
2148 2149
		if (!page)
			return ERR_PTR(-ENOMEM);
2150
		err = add_to_page_cache_lru(page, mapping, index, gfp);
N
Nick Piggin 已提交
2151 2152 2153 2154
		if (unlikely(err)) {
			page_cache_release(page);
			if (err == -EEXIST)
				goto repeat;
L
Linus Torvalds 已提交
2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166
			/* Presumably ENOMEM for radix tree node */
			return ERR_PTR(err);
		}
		err = filler(data, page);
		if (err < 0) {
			page_cache_release(page);
			page = ERR_PTR(err);
		}
	}
	return page;
}

2167
static struct page *do_read_cache_page(struct address_space *mapping,
2168
				pgoff_t index,
2169
				int (*filler)(void *, struct page *),
2170 2171 2172
				void *data,
				gfp_t gfp)

L
Linus Torvalds 已提交
2173 2174 2175 2176 2177
{
	struct page *page;
	int err;

retry:
2178
	page = __read_cache_page(mapping, index, filler, data, gfp);
L
Linus Torvalds 已提交
2179
	if (IS_ERR(page))
2180
		return page;
L
Linus Torvalds 已提交
2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196
	if (PageUptodate(page))
		goto out;

	lock_page(page);
	if (!page->mapping) {
		unlock_page(page);
		page_cache_release(page);
		goto retry;
	}
	if (PageUptodate(page)) {
		unlock_page(page);
		goto out;
	}
	err = filler(data, page);
	if (err < 0) {
		page_cache_release(page);
2197
		return ERR_PTR(err);
L
Linus Torvalds 已提交
2198
	}
2199
out:
2200 2201 2202
	mark_page_accessed(page);
	return page;
}
2203 2204 2205 2206 2207 2208

/**
 * read_cache_page_async - read into page cache, fill it if needed
 * @mapping:	the page's address_space
 * @index:	the page index
 * @filler:	function to perform the read
2209
 * @data:	first arg to filler(data, page) function, often left as NULL
2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220
 *
 * Same as read_cache_page, but don't wait for page to become unlocked
 * after submitting it to the filler.
 *
 * Read into the page cache. If a page already exists, and PageUptodate() is
 * not set, try to fill the page but don't wait for it to become unlocked.
 *
 * If the page does not get brought uptodate, return -EIO.
 */
struct page *read_cache_page_async(struct address_space *mapping,
				pgoff_t index,
2221
				int (*filler)(void *, struct page *),
2222 2223 2224 2225
				void *data)
{
	return do_read_cache_page(mapping, index, filler, data, mapping_gfp_mask(mapping));
}
2226 2227
EXPORT_SYMBOL(read_cache_page_async);

2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246
static struct page *wait_on_page_read(struct page *page)
{
	if (!IS_ERR(page)) {
		wait_on_page_locked(page);
		if (!PageUptodate(page)) {
			page_cache_release(page);
			page = ERR_PTR(-EIO);
		}
	}
	return page;
}

/**
 * read_cache_page_gfp - read into page cache, using specified page allocation flags.
 * @mapping:	the page's address_space
 * @index:	the page index
 * @gfp:	the page allocator flags to use if allocating
 *
 * This is the same as "read_mapping_page(mapping, index, NULL)", but with
2247
 * any new page allocations done using the specified allocation flags.
2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260
 *
 * If the page does not get brought uptodate, return -EIO.
 */
struct page *read_cache_page_gfp(struct address_space *mapping,
				pgoff_t index,
				gfp_t gfp)
{
	filler_t *filler = (filler_t *)mapping->a_ops->readpage;

	return wait_on_page_read(do_read_cache_page(mapping, index, filler, NULL, gfp));
}
EXPORT_SYMBOL(read_cache_page_gfp);

2261 2262 2263 2264 2265
/**
 * read_cache_page - read into page cache, fill it if needed
 * @mapping:	the page's address_space
 * @index:	the page index
 * @filler:	function to perform the read
2266
 * @data:	first arg to filler(data, page) function, often left as NULL
2267 2268 2269 2270 2271 2272 2273
 *
 * Read into the page cache. If a page already exists, and PageUptodate() is
 * not set, try to fill the page then wait for it to become unlocked.
 *
 * If the page does not get brought uptodate, return -EIO.
 */
struct page *read_cache_page(struct address_space *mapping,
2274
				pgoff_t index,
2275
				int (*filler)(void *, struct page *),
2276 2277
				void *data)
{
2278
	return wait_on_page_read(read_cache_page_async(mapping, index, filler, data));
L
Linus Torvalds 已提交
2279 2280 2281
}
EXPORT_SYMBOL(read_cache_page);

N
Nick Piggin 已提交
2282
static size_t __iovec_copy_from_user_inatomic(char *vaddr,
L
Linus Torvalds 已提交
2283 2284
			const struct iovec *iov, size_t base, size_t bytes)
{
2285
	size_t copied = 0, left = 0;
L
Linus Torvalds 已提交
2286 2287 2288 2289 2290 2291

	while (bytes) {
		char __user *buf = iov->iov_base + base;
		int copy = min(bytes, iov->iov_len - base);

		base = 0;
2292
		left = __copy_from_user_inatomic(vaddr, buf, copy);
L
Linus Torvalds 已提交
2293 2294 2295 2296 2297
		copied += copy;
		bytes -= copy;
		vaddr += copy;
		iov++;

2298
		if (unlikely(left))
L
Linus Torvalds 已提交
2299 2300 2301 2302 2303
			break;
	}
	return copied - left;
}

N
Nick Piggin 已提交
2304 2305
/*
 * Copy as much as we can into the page and return the number of bytes which
2306
 * were successfully copied.  If a fault is encountered then return the number of
N
Nick Piggin 已提交
2307 2308 2309 2310 2311 2312 2313 2314 2315
 * bytes which were copied.
 */
size_t iov_iter_copy_from_user_atomic(struct page *page,
		struct iov_iter *i, unsigned long offset, size_t bytes)
{
	char *kaddr;
	size_t copied;

	BUG_ON(!in_atomic());
2316
	kaddr = kmap_atomic(page);
N
Nick Piggin 已提交
2317 2318 2319
	if (likely(i->nr_segs == 1)) {
		int left;
		char __user *buf = i->iov->iov_base + i->iov_offset;
2320
		left = __copy_from_user_inatomic(kaddr + offset, buf, bytes);
N
Nick Piggin 已提交
2321 2322 2323 2324 2325
		copied = bytes - left;
	} else {
		copied = __iovec_copy_from_user_inatomic(kaddr + offset,
						i->iov, i->iov_offset, bytes);
	}
2326
	kunmap_atomic(kaddr);
N
Nick Piggin 已提交
2327 2328 2329

	return copied;
}
N
Nick Piggin 已提交
2330
EXPORT_SYMBOL(iov_iter_copy_from_user_atomic);
N
Nick Piggin 已提交
2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347

/*
 * This has the same sideeffects and return value as
 * iov_iter_copy_from_user_atomic().
 * The difference is that it attempts to resolve faults.
 * Page must not be locked.
 */
size_t iov_iter_copy_from_user(struct page *page,
		struct iov_iter *i, unsigned long offset, size_t bytes)
{
	char *kaddr;
	size_t copied;

	kaddr = kmap(page);
	if (likely(i->nr_segs == 1)) {
		int left;
		char __user *buf = i->iov->iov_base + i->iov_offset;
2348
		left = __copy_from_user(kaddr + offset, buf, bytes);
N
Nick Piggin 已提交
2349 2350 2351 2352 2353 2354 2355 2356
		copied = bytes - left;
	} else {
		copied = __iovec_copy_from_user_inatomic(kaddr + offset,
						i->iov, i->iov_offset, bytes);
	}
	kunmap(page);
	return copied;
}
N
Nick Piggin 已提交
2357
EXPORT_SYMBOL(iov_iter_copy_from_user);
N
Nick Piggin 已提交
2358

N
Nick Piggin 已提交
2359
void iov_iter_advance(struct iov_iter *i, size_t bytes)
N
Nick Piggin 已提交
2360
{
N
Nick Piggin 已提交
2361 2362
	BUG_ON(i->count < bytes);

N
Nick Piggin 已提交
2363 2364
	if (likely(i->nr_segs == 1)) {
		i->iov_offset += bytes;
N
Nick Piggin 已提交
2365
		i->count -= bytes;
N
Nick Piggin 已提交
2366 2367 2368
	} else {
		const struct iovec *iov = i->iov;
		size_t base = i->iov_offset;
2369
		unsigned long nr_segs = i->nr_segs;
N
Nick Piggin 已提交
2370

2371 2372
		/*
		 * The !iov->iov_len check ensures we skip over unlikely
N
Nick Piggin 已提交
2373
		 * zero-length segments (without overruning the iovec).
2374
		 */
2375
		while (bytes || unlikely(i->count && !iov->iov_len)) {
N
Nick Piggin 已提交
2376
			int copy;
N
Nick Piggin 已提交
2377

N
Nick Piggin 已提交
2378 2379 2380
			copy = min(bytes, iov->iov_len - base);
			BUG_ON(!i->count || i->count < copy);
			i->count -= copy;
N
Nick Piggin 已提交
2381 2382 2383 2384
			bytes -= copy;
			base += copy;
			if (iov->iov_len == base) {
				iov++;
2385
				nr_segs--;
N
Nick Piggin 已提交
2386 2387 2388 2389 2390
				base = 0;
			}
		}
		i->iov = iov;
		i->iov_offset = base;
2391
		i->nr_segs = nr_segs;
N
Nick Piggin 已提交
2392 2393
	}
}
N
Nick Piggin 已提交
2394
EXPORT_SYMBOL(iov_iter_advance);
N
Nick Piggin 已提交
2395

2396 2397 2398 2399 2400 2401 2402 2403 2404 2405
/*
 * Fault in the first iovec of the given iov_iter, to a maximum length
 * of bytes. Returns 0 on success, or non-zero if the memory could not be
 * accessed (ie. because it is an invalid address).
 *
 * writev-intensive code may want this to prefault several iovecs -- that
 * would be possible (callers must not rely on the fact that _only_ the
 * first iovec will be faulted with the current implementation).
 */
int iov_iter_fault_in_readable(struct iov_iter *i, size_t bytes)
N
Nick Piggin 已提交
2406 2407
{
	char __user *buf = i->iov->iov_base + i->iov_offset;
2408 2409
	bytes = min(bytes, i->iov->iov_len - i->iov_offset);
	return fault_in_pages_readable(buf, bytes);
N
Nick Piggin 已提交
2410
}
N
Nick Piggin 已提交
2411
EXPORT_SYMBOL(iov_iter_fault_in_readable);
N
Nick Piggin 已提交
2412 2413 2414 2415

/*
 * Return the count of just the current iov_iter segment.
 */
2416
size_t iov_iter_single_seg_count(const struct iov_iter *i)
N
Nick Piggin 已提交
2417 2418 2419 2420 2421 2422 2423
{
	const struct iovec *iov = i->iov;
	if (i->nr_segs == 1)
		return i->count;
	else
		return min(i->count, iov->iov_len - i->iov_offset);
}
N
Nick Piggin 已提交
2424
EXPORT_SYMBOL(iov_iter_single_seg_count);
N
Nick Piggin 已提交
2425

L
Linus Torvalds 已提交
2426 2427 2428
/*
 * Performs necessary checks before doing a write
 *
2429
 * Can adjust writing position or amount of bytes to write.
L
Linus Torvalds 已提交
2430 2431 2432 2433 2434 2435
 * Returns appropriate error code that caller should return or
 * zero in case that write should be allowed.
 */
inline int generic_write_checks(struct file *file, loff_t *pos, size_t *count, int isblk)
{
	struct inode *inode = file->f_mapping->host;
J
Jiri Slaby 已提交
2436
	unsigned long limit = rlimit(RLIMIT_FSIZE);
L
Linus Torvalds 已提交
2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487

        if (unlikely(*pos < 0))
                return -EINVAL;

	if (!isblk) {
		/* FIXME: this is for backwards compatibility with 2.4 */
		if (file->f_flags & O_APPEND)
                        *pos = i_size_read(inode);

		if (limit != RLIM_INFINITY) {
			if (*pos >= limit) {
				send_sig(SIGXFSZ, current, 0);
				return -EFBIG;
			}
			if (*count > limit - (typeof(limit))*pos) {
				*count = limit - (typeof(limit))*pos;
			}
		}
	}

	/*
	 * LFS rule
	 */
	if (unlikely(*pos + *count > MAX_NON_LFS &&
				!(file->f_flags & O_LARGEFILE))) {
		if (*pos >= MAX_NON_LFS) {
			return -EFBIG;
		}
		if (*count > MAX_NON_LFS - (unsigned long)*pos) {
			*count = MAX_NON_LFS - (unsigned long)*pos;
		}
	}

	/*
	 * Are we about to exceed the fs block limit ?
	 *
	 * If we have written data it becomes a short write.  If we have
	 * exceeded without writing data we send a signal and return EFBIG.
	 * Linus frestrict idea will clean these up nicely..
	 */
	if (likely(!isblk)) {
		if (unlikely(*pos >= inode->i_sb->s_maxbytes)) {
			if (*count || *pos > inode->i_sb->s_maxbytes) {
				return -EFBIG;
			}
			/* zero-length writes at ->s_maxbytes are OK */
		}

		if (unlikely(*pos + *count > inode->i_sb->s_maxbytes))
			*count = inode->i_sb->s_maxbytes - *pos;
	} else {
2488
#ifdef CONFIG_BLOCK
L
Linus Torvalds 已提交
2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499
		loff_t isize;
		if (bdev_read_only(I_BDEV(inode)))
			return -EPERM;
		isize = i_size_read(inode);
		if (*pos >= isize) {
			if (*count || *pos > isize)
				return -ENOSPC;
		}

		if (*pos + *count > isize)
			*count = isize - *pos;
2500 2501 2502
#else
		return -EPERM;
#endif
L
Linus Torvalds 已提交
2503 2504 2505 2506 2507
	}
	return 0;
}
EXPORT_SYMBOL(generic_write_checks);

2508 2509 2510 2511 2512 2513
int pagecache_write_begin(struct file *file, struct address_space *mapping,
				loff_t pos, unsigned len, unsigned flags,
				struct page **pagep, void **fsdata)
{
	const struct address_space_operations *aops = mapping->a_ops;

2514
	return aops->write_begin(file, mapping, pos, len, flags,
2515 2516 2517 2518 2519 2520 2521 2522 2523 2524
							pagep, fsdata);
}
EXPORT_SYMBOL(pagecache_write_begin);

int pagecache_write_end(struct file *file, struct address_space *mapping,
				loff_t pos, unsigned len, unsigned copied,
				struct page *page, void *fsdata)
{
	const struct address_space_operations *aops = mapping->a_ops;

2525 2526
	mark_page_accessed(page);
	return aops->write_end(file, mapping, pos, len, copied, page, fsdata);
2527 2528 2529
}
EXPORT_SYMBOL(pagecache_write_end);

L
Linus Torvalds 已提交
2530 2531 2532 2533 2534 2535 2536 2537 2538
ssize_t
generic_file_direct_write(struct kiocb *iocb, const struct iovec *iov,
		unsigned long *nr_segs, loff_t pos, loff_t *ppos,
		size_t count, size_t ocount)
{
	struct file	*file = iocb->ki_filp;
	struct address_space *mapping = file->f_mapping;
	struct inode	*inode = mapping->host;
	ssize_t		written;
2539 2540
	size_t		write_len;
	pgoff_t		end;
L
Linus Torvalds 已提交
2541 2542 2543 2544

	if (count != ocount)
		*nr_segs = iov_shorten((struct iovec *)iov, *nr_segs, count);

2545 2546 2547
	write_len = iov_length(iov, *nr_segs);
	end = (pos + write_len - 1) >> PAGE_CACHE_SHIFT;

2548
	written = filemap_write_and_wait_range(mapping, pos, pos + write_len - 1);
2549 2550 2551 2552 2553 2554 2555
	if (written)
		goto out;

	/*
	 * After a write we want buffered reads to be sure to go to disk to get
	 * the new data.  We invalidate clean cached page from the region we're
	 * about to write.  We do this *before* the write so that we can return
2556
	 * without clobbering -EIOCBQUEUED from ->direct_IO().
2557 2558 2559 2560
	 */
	if (mapping->nrpages) {
		written = invalidate_inode_pages2_range(mapping,
					pos >> PAGE_CACHE_SHIFT, end);
2561 2562 2563 2564 2565 2566 2567
		/*
		 * If a page can not be invalidated, return 0 to fall back
		 * to buffered write.
		 */
		if (written) {
			if (written == -EBUSY)
				return 0;
2568
			goto out;
2569
		}
2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586
	}

	written = mapping->a_ops->direct_IO(WRITE, iocb, iov, pos, *nr_segs);

	/*
	 * Finally, try again to invalidate clean pages which might have been
	 * cached by non-direct readahead, or faulted in by get_user_pages()
	 * if the source of the write was an mmap'ed region of the file
	 * we're writing.  Either one is a pretty crazy thing to do,
	 * so we don't support it 100%.  If this invalidation
	 * fails, tough, the write still worked...
	 */
	if (mapping->nrpages) {
		invalidate_inode_pages2_range(mapping,
					      pos >> PAGE_CACHE_SHIFT, end);
	}

L
Linus Torvalds 已提交
2587
	if (written > 0) {
2588 2589 2590
		pos += written;
		if (pos > i_size_read(inode) && !S_ISBLK(inode->i_mode)) {
			i_size_write(inode, pos);
L
Linus Torvalds 已提交
2591 2592
			mark_inode_dirty(inode);
		}
2593
		*ppos = pos;
L
Linus Torvalds 已提交
2594
	}
2595
out:
L
Linus Torvalds 已提交
2596 2597 2598 2599
	return written;
}
EXPORT_SYMBOL(generic_file_direct_write);

N
Nick Piggin 已提交
2600 2601 2602 2603
/*
 * Find or create a page at the given pagecache position. Return the locked
 * page. This function is specifically for buffered writes.
 */
2604 2605
struct page *grab_cache_page_write_begin(struct address_space *mapping,
					pgoff_t index, unsigned flags)
N
Nick Piggin 已提交
2606 2607
{
	int status;
2608
	gfp_t gfp_mask;
N
Nick Piggin 已提交
2609
	struct page *page;
2610
	gfp_t gfp_notmask = 0;
2611

2612 2613 2614
	gfp_mask = mapping_gfp_mask(mapping);
	if (mapping_cap_account_dirty(mapping))
		gfp_mask |= __GFP_WRITE;
2615 2616
	if (flags & AOP_FLAG_NOFS)
		gfp_notmask = __GFP_FS;
N
Nick Piggin 已提交
2617 2618
repeat:
	page = find_lock_page(mapping, index);
2619
	if (page)
2620
		goto found;
N
Nick Piggin 已提交
2621

2622
	page = __page_cache_alloc(gfp_mask & ~gfp_notmask);
N
Nick Piggin 已提交
2623 2624
	if (!page)
		return NULL;
2625 2626
	status = add_to_page_cache_lru(page, mapping, index,
						GFP_KERNEL & ~gfp_notmask);
N
Nick Piggin 已提交
2627 2628 2629 2630 2631 2632
	if (unlikely(status)) {
		page_cache_release(page);
		if (status == -EEXIST)
			goto repeat;
		return NULL;
	}
2633
found:
2634
	wait_for_stable_page(page);
N
Nick Piggin 已提交
2635 2636
	return page;
}
2637
EXPORT_SYMBOL(grab_cache_page_write_begin);
N
Nick Piggin 已提交
2638

2639 2640 2641 2642 2643 2644 2645
static ssize_t generic_perform_write(struct file *file,
				struct iov_iter *i, loff_t pos)
{
	struct address_space *mapping = file->f_mapping;
	const struct address_space_operations *a_ops = mapping->a_ops;
	long status = 0;
	ssize_t written = 0;
N
Nick Piggin 已提交
2646 2647 2648 2649 2650 2651 2652
	unsigned int flags = 0;

	/*
	 * Copies from kernel address space cannot fail (NFSD is a big user).
	 */
	if (segment_eq(get_fs(), KERNEL_DS))
		flags |= AOP_FLAG_UNINTERRUPTIBLE;
2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680

	do {
		struct page *page;
		unsigned long offset;	/* Offset into pagecache page */
		unsigned long bytes;	/* Bytes to write to page */
		size_t copied;		/* Bytes copied from user */
		void *fsdata;

		offset = (pos & (PAGE_CACHE_SIZE - 1));
		bytes = min_t(unsigned long, PAGE_CACHE_SIZE - offset,
						iov_iter_count(i));

again:
		/*
		 * Bring in the user page that we will copy from _first_.
		 * Otherwise there's a nasty deadlock on copying from the
		 * same page as we're writing to, without it being marked
		 * up-to-date.
		 *
		 * Not only is this an optimisation, but it is also required
		 * to check that the address is actually valid, when atomic
		 * usercopies are used, below.
		 */
		if (unlikely(iov_iter_fault_in_readable(i, bytes))) {
			status = -EFAULT;
			break;
		}

N
Nick Piggin 已提交
2681
		status = a_ops->write_begin(file, mapping, pos, bytes, flags,
2682 2683 2684 2685
						&page, &fsdata);
		if (unlikely(status))
			break;

2686 2687 2688
		if (mapping_writably_mapped(mapping))
			flush_dcache_page(page);

2689 2690 2691 2692 2693
		pagefault_disable();
		copied = iov_iter_copy_from_user_atomic(page, i, offset, bytes);
		pagefault_enable();
		flush_dcache_page(page);

2694
		mark_page_accessed(page);
2695 2696 2697 2698 2699 2700 2701 2702
		status = a_ops->write_end(file, mapping, pos, bytes, copied,
						page, fsdata);
		if (unlikely(status < 0))
			break;
		copied = status;

		cond_resched();

2703
		iov_iter_advance(i, copied);
2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720
		if (unlikely(copied == 0)) {
			/*
			 * If we were unable to copy any data at all, we must
			 * fall back to a single segment length write.
			 *
			 * If we didn't fallback here, we could livelock
			 * because not all segments in the iov can be copied at
			 * once without a pagefault.
			 */
			bytes = min_t(unsigned long, PAGE_CACHE_SIZE - offset,
						iov_iter_single_seg_count(i));
			goto again;
		}
		pos += copied;
		written += copied;

		balance_dirty_pages_ratelimited(mapping);
2721 2722 2723 2724
		if (fatal_signal_pending(current)) {
			status = -EINTR;
			break;
		}
2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739
	} while (iov_iter_count(i));

	return written ? written : status;
}

ssize_t
generic_file_buffered_write(struct kiocb *iocb, const struct iovec *iov,
		unsigned long nr_segs, loff_t pos, loff_t *ppos,
		size_t count, ssize_t written)
{
	struct file *file = iocb->ki_filp;
	ssize_t status;
	struct iov_iter i;

	iov_iter_init(&i, iov, nr_segs, count, written);
2740
	status = generic_perform_write(file, &i, pos);
L
Linus Torvalds 已提交
2741 2742

	if (likely(status >= 0)) {
2743 2744
		written += status;
		*ppos = pos + status;
L
Linus Torvalds 已提交
2745 2746 2747 2748 2749 2750
  	}
	
	return written ? written : status;
}
EXPORT_SYMBOL(generic_file_buffered_write);

2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771
/**
 * __generic_file_aio_write - write data to a file
 * @iocb:	IO state structure (file, offset, etc.)
 * @iov:	vector with data to write
 * @nr_segs:	number of segments in the vector
 * @ppos:	position where to write
 *
 * This function does all the work needed for actually writing data to a
 * file. It does all basic checks, removes SUID from the file, updates
 * modification times and calls proper subroutines depending on whether we
 * do direct IO or a standard buffered write.
 *
 * It expects i_mutex to be grabbed unless we work on a block device or similar
 * object which does not need locking at all.
 *
 * This function does *not* take care of syncing data in case of O_SYNC write.
 * A caller has to handle it. This is mainly due to the fact that we want to
 * avoid syncing under i_mutex.
 */
ssize_t __generic_file_aio_write(struct kiocb *iocb, const struct iovec *iov,
				 unsigned long nr_segs, loff_t *ppos)
L
Linus Torvalds 已提交
2772 2773
{
	struct file *file = iocb->ki_filp;
2774
	struct address_space * mapping = file->f_mapping;
L
Linus Torvalds 已提交
2775 2776 2777 2778 2779 2780 2781 2782
	size_t ocount;		/* original count */
	size_t count;		/* after file limit checks */
	struct inode 	*inode = mapping->host;
	loff_t		pos;
	ssize_t		written;
	ssize_t		err;

	ocount = 0;
2783 2784 2785
	err = generic_segment_checks(iov, &nr_segs, &ocount, VERIFY_READ);
	if (err)
		return err;
L
Linus Torvalds 已提交
2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800

	count = ocount;
	pos = *ppos;

	/* We can write back this queue in page reclaim */
	current->backing_dev_info = mapping->backing_dev_info;
	written = 0;

	err = generic_write_checks(file, &pos, &count, S_ISBLK(inode->i_mode));
	if (err)
		goto out;

	if (count == 0)
		goto out;

2801
	err = file_remove_suid(file);
L
Linus Torvalds 已提交
2802 2803 2804
	if (err)
		goto out;

2805 2806 2807
	err = file_update_time(file);
	if (err)
		goto out;
L
Linus Torvalds 已提交
2808 2809 2810

	/* coalesce the iovecs and go direct-to-BIO for O_DIRECT */
	if (unlikely(file->f_flags & O_DIRECT)) {
2811 2812 2813 2814 2815
		loff_t endbyte;
		ssize_t written_buffered;

		written = generic_file_direct_write(iocb, iov, &nr_segs, pos,
							ppos, count, ocount);
L
Linus Torvalds 已提交
2816 2817 2818 2819 2820 2821 2822 2823
		if (written < 0 || written == count)
			goto out;
		/*
		 * direct-io write to a hole: fall through to buffered I/O
		 * for completing the rest of the request.
		 */
		pos += written;
		count -= written;
2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837
		written_buffered = generic_file_buffered_write(iocb, iov,
						nr_segs, pos, ppos, count,
						written);
		/*
		 * If generic_file_buffered_write() retuned a synchronous error
		 * then we want to return the number of bytes which were
		 * direct-written, or the error code if that was zero.  Note
		 * that this differs from normal direct-io semantics, which
		 * will return -EFOO even if some bytes were written.
		 */
		if (written_buffered < 0) {
			err = written_buffered;
			goto out;
		}
L
Linus Torvalds 已提交
2838

2839 2840 2841 2842 2843 2844
		/*
		 * We need to ensure that the page cache pages are written to
		 * disk and invalidated to preserve the expected O_DIRECT
		 * semantics.
		 */
		endbyte = pos + written_buffered - written - 1;
2845
		err = filemap_write_and_wait_range(file->f_mapping, pos, endbyte);
2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860
		if (err == 0) {
			written = written_buffered;
			invalidate_mapping_pages(mapping,
						 pos >> PAGE_CACHE_SHIFT,
						 endbyte >> PAGE_CACHE_SHIFT);
		} else {
			/*
			 * We don't know how much we wrote, so just return
			 * the number of bytes which were direct-written
			 */
		}
	} else {
		written = generic_file_buffered_write(iocb, iov, nr_segs,
				pos, ppos, count, written);
	}
L
Linus Torvalds 已提交
2861 2862 2863 2864
out:
	current->backing_dev_info = NULL;
	return written ? written : err;
}
2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877
EXPORT_SYMBOL(__generic_file_aio_write);

/**
 * generic_file_aio_write - write data to a file
 * @iocb:	IO state structure
 * @iov:	vector with data to write
 * @nr_segs:	number of segments in the vector
 * @pos:	position in file where to write
 *
 * This is a wrapper around __generic_file_aio_write() to be used by most
 * filesystems. It takes care of syncing the file in case of O_SYNC file
 * and acquires i_mutex as needed.
 */
2878 2879
ssize_t generic_file_aio_write(struct kiocb *iocb, const struct iovec *iov,
		unsigned long nr_segs, loff_t pos)
L
Linus Torvalds 已提交
2880 2881
{
	struct file *file = iocb->ki_filp;
2882
	struct inode *inode = file->f_mapping->host;
L
Linus Torvalds 已提交
2883 2884 2885 2886
	ssize_t ret;

	BUG_ON(iocb->ki_pos != pos);

2887
	mutex_lock(&inode->i_mutex);
2888
	ret = __generic_file_aio_write(iocb, iov, nr_segs, &iocb->ki_pos);
2889
	mutex_unlock(&inode->i_mutex);
L
Linus Torvalds 已提交
2890

2891
	if (ret > 0) {
L
Linus Torvalds 已提交
2892 2893
		ssize_t err;

2894 2895
		err = generic_write_sync(file, iocb->ki_pos - ret, ret);
		if (err < 0)
L
Linus Torvalds 已提交
2896 2897 2898 2899 2900 2901
			ret = err;
	}
	return ret;
}
EXPORT_SYMBOL(generic_file_aio_write);

2902 2903 2904 2905 2906 2907 2908 2909 2910 2911
/**
 * try_to_release_page() - release old fs-specific metadata on a page
 *
 * @page: the page which the kernel is trying to free
 * @gfp_mask: memory allocation flags (and I/O mode)
 *
 * The address_space is to try to release any data against the page
 * (presumably at page->private).  If the release was successful, return `1'.
 * Otherwise return zero.
 *
2912 2913 2914
 * This may also be called if PG_fscache is set on a page, indicating that the
 * page is known to the local caching routines.
 *
2915
 * The @gfp_mask argument specifies whether I/O may be performed to release
2916
 * this page (__GFP_IO), and whether the call may block (__GFP_WAIT & __GFP_FS).
2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932
 *
 */
int try_to_release_page(struct page *page, gfp_t gfp_mask)
{
	struct address_space * const mapping = page->mapping;

	BUG_ON(!PageLocked(page));
	if (PageWriteback(page))
		return 0;

	if (mapping && mapping->a_ops->releasepage)
		return mapping->a_ops->releasepage(page, gfp_mask);
	return try_to_free_buffers(page);
}

EXPORT_SYMBOL(try_to_release_page);