node.c 67.9 KB
Newer Older
J
Jaegeuk Kim 已提交
1
/*
J
Jaegeuk Kim 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
 * fs/f2fs/node.c
 *
 * Copyright (c) 2012 Samsung Electronics Co., Ltd.
 *             http://www.samsung.com/
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */
#include <linux/fs.h>
#include <linux/f2fs_fs.h>
#include <linux/mpage.h>
#include <linux/backing-dev.h>
#include <linux/blkdev.h>
#include <linux/pagevec.h>
#include <linux/swap.h>

#include "f2fs.h"
#include "node.h"
#include "segment.h"
22
#include "xattr.h"
J
Jaegeuk Kim 已提交
23
#include "trace.h"
24
#include <trace/events/f2fs.h>
J
Jaegeuk Kim 已提交
25

26
#define on_build_free_nids(nmi) mutex_is_locked(&(nm_i)->build_lock)
27

J
Jaegeuk Kim 已提交
28 29
static struct kmem_cache *nat_entry_slab;
static struct kmem_cache *free_nid_slab;
30
static struct kmem_cache *nat_entry_set_slab;
J
Jaegeuk Kim 已提交
31

32
bool available_free_memory(struct f2fs_sb_info *sbi, int type)
33
{
34
	struct f2fs_nm_info *nm_i = NM_I(sbi);
35
	struct sysinfo val;
36
	unsigned long avail_ram;
37
	unsigned long mem_size = 0;
38
	bool res = false;
39 40

	si_meminfo(&val);
41 42 43 44

	/* only uses low memory */
	avail_ram = val.totalram - val.totalhigh;

45 46 47
	/*
	 * give 25%, 25%, 50%, 50%, 50% memory for each components respectively
	 */
48
	if (type == FREE_NIDS) {
C
Chao Yu 已提交
49
		mem_size = (nm_i->nid_cnt[FREE_NID] *
C
Chao Yu 已提交
50
				sizeof(struct free_nid)) >> PAGE_SHIFT;
51
		res = mem_size < ((avail_ram * nm_i->ram_thresh / 100) >> 2);
52
	} else if (type == NAT_ENTRIES) {
53
		mem_size = (nm_i->nat_cnt * sizeof(struct nat_entry)) >>
54
							PAGE_SHIFT;
55
		res = mem_size < ((avail_ram * nm_i->ram_thresh / 100) >> 2);
56 57
		if (excess_cached_nats(sbi))
			res = false;
58 59 60 61 62
	} else if (type == DIRTY_DENTS) {
		if (sbi->sb->s_bdi->wb.dirty_exceeded)
			return false;
		mem_size = get_pages(sbi, F2FS_DIRTY_DENTS);
		res = mem_size < ((avail_ram * nm_i->ram_thresh / 100) >> 1);
63 64 65
	} else if (type == INO_ENTRIES) {
		int i;

C
Chao Yu 已提交
66
		for (i = 0; i < MAX_INO_ENTRY; i++)
67 68 69
			mem_size += sbi->im[i].ino_num *
						sizeof(struct ino_entry);
		mem_size >>= PAGE_SHIFT;
70
		res = mem_size < ((avail_ram * nm_i->ram_thresh / 100) >> 1);
71
	} else if (type == EXTENT_CACHE) {
72 73
		mem_size = (atomic_read(&sbi->total_ext_tree) *
				sizeof(struct extent_tree) +
74
				atomic_read(&sbi->total_ext_node) *
75
				sizeof(struct extent_node)) >> PAGE_SHIFT;
76
		res = mem_size < ((avail_ram * nm_i->ram_thresh / 100) >> 1);
J
Jaegeuk Kim 已提交
77 78 79 80
	} else if (type == INMEM_PAGES) {
		/* it allows 20% / total_ram for inmemory pages */
		mem_size = get_pages(sbi, F2FS_INMEM_PAGES);
		res = mem_size < (val.totalram / 5);
81
	} else {
82 83
		if (!sbi->sb->s_bdi->wb.dirty_exceeded)
			return true;
84 85
	}
	return res;
86 87
}

J
Jaegeuk Kim 已提交
88 89 90 91 92 93 94 95 96 97 98 99 100
static void clear_node_page_dirty(struct page *page)
{
	struct address_space *mapping = page->mapping;
	unsigned int long flags;

	if (PageDirty(page)) {
		spin_lock_irqsave(&mapping->tree_lock, flags);
		radix_tree_tag_clear(&mapping->page_tree,
				page_index(page),
				PAGECACHE_TAG_DIRTY);
		spin_unlock_irqrestore(&mapping->tree_lock, flags);

		clear_page_dirty_for_io(page);
101
		dec_page_count(F2FS_M_SB(mapping), F2FS_DIRTY_NODES);
J
Jaegeuk Kim 已提交
102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
	}
	ClearPageUptodate(page);
}

static struct page *get_current_nat_page(struct f2fs_sb_info *sbi, nid_t nid)
{
	pgoff_t index = current_nat_addr(sbi, nid);
	return get_meta_page(sbi, index);
}

static struct page *get_next_nat_page(struct f2fs_sb_info *sbi, nid_t nid)
{
	struct page *src_page;
	struct page *dst_page;
	pgoff_t src_off;
	pgoff_t dst_off;
	void *src_addr;
	void *dst_addr;
	struct f2fs_nm_info *nm_i = NM_I(sbi);

	src_off = current_nat_addr(sbi, nid);
	dst_off = next_nat_addr(sbi, src_off);

	/* get current nat block page with lock */
	src_page = get_meta_page(sbi, src_off);
	dst_page = grab_meta_page(sbi, dst_off);
128
	f2fs_bug_on(sbi, PageDirty(src_page));
J
Jaegeuk Kim 已提交
129 130 131

	src_addr = page_address(src_page);
	dst_addr = page_address(dst_page);
132
	memcpy(dst_addr, src_addr, PAGE_SIZE);
J
Jaegeuk Kim 已提交
133 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
	set_page_dirty(dst_page);
	f2fs_put_page(src_page, 1);

	set_to_next_nat(nm_i, nid);

	return dst_page;
}

static struct nat_entry *__lookup_nat_cache(struct f2fs_nm_info *nm_i, nid_t n)
{
	return radix_tree_lookup(&nm_i->nat_root, n);
}

static unsigned int __gang_lookup_nat_cache(struct f2fs_nm_info *nm_i,
		nid_t start, unsigned int nr, struct nat_entry **ep)
{
	return radix_tree_gang_lookup(&nm_i->nat_root, (void **)ep, start, nr);
}

static void __del_from_nat_cache(struct f2fs_nm_info *nm_i, struct nat_entry *e)
{
	list_del(&e->list);
	radix_tree_delete(&nm_i->nat_root, nat_get_nid(e));
	nm_i->nat_cnt--;
	kmem_cache_free(nat_entry_slab, e);
}

160 161 162 163 164 165 166 167
static void __set_nat_cache_dirty(struct f2fs_nm_info *nm_i,
						struct nat_entry *ne)
{
	nid_t set = NAT_BLOCK_OFFSET(ne->ni.nid);
	struct nat_entry_set *head;

	head = radix_tree_lookup(&nm_i->nat_set_root, set);
	if (!head) {
168
		head = f2fs_kmem_cache_alloc(nat_entry_set_slab, GFP_NOFS);
169 170 171 172 173

		INIT_LIST_HEAD(&head->entry_list);
		INIT_LIST_HEAD(&head->set_list);
		head->set = set;
		head->entry_cnt = 0;
174
		f2fs_radix_tree_insert(&nm_i->nat_set_root, set, head);
175
	}
176 177 178 179

	if (get_nat_flag(ne, IS_DIRTY))
		goto refresh_list;

180 181 182
	nm_i->dirty_nat_cnt++;
	head->entry_cnt++;
	set_nat_flag(ne, IS_DIRTY, true);
183 184 185 186 187
refresh_list:
	if (nat_get_blkaddr(ne) == NEW_ADDR)
		list_del_init(&ne->list);
	else
		list_move_tail(&ne->list, &head->entry_list);
188 189 190
}

static void __clear_nat_cache_dirty(struct f2fs_nm_info *nm_i,
191
		struct nat_entry_set *set, struct nat_entry *ne)
192
{
193 194 195 196
	list_move_tail(&ne->list, &nm_i->nat_entries);
	set_nat_flag(ne, IS_DIRTY, false);
	set->entry_cnt--;
	nm_i->dirty_nat_cnt--;
197 198 199 200 201 202 203 204 205
}

static unsigned int __gang_lookup_nat_set(struct f2fs_nm_info *nm_i,
		nid_t start, unsigned int nr, struct nat_entry_set **ep)
{
	return radix_tree_gang_lookup(&nm_i->nat_set_root, (void **)ep,
							start, nr);
}

J
Jaegeuk Kim 已提交
206
int need_dentry_mark(struct f2fs_sb_info *sbi, nid_t nid)
J
Jaegeuk Kim 已提交
207 208 209
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct nat_entry *e;
J
Jaegeuk Kim 已提交
210
	bool need = false;
J
Jaegeuk Kim 已提交
211

212
	down_read(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
213
	e = __lookup_nat_cache(nm_i, nid);
J
Jaegeuk Kim 已提交
214 215 216 217 218
	if (e) {
		if (!get_nat_flag(e, IS_CHECKPOINTED) &&
				!get_nat_flag(e, HAS_FSYNCED_INODE))
			need = true;
	}
219
	up_read(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
220
	return need;
J
Jaegeuk Kim 已提交
221 222
}

J
Jaegeuk Kim 已提交
223
bool is_checkpointed_node(struct f2fs_sb_info *sbi, nid_t nid)
224 225 226
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct nat_entry *e;
J
Jaegeuk Kim 已提交
227
	bool is_cp = true;
228

229
	down_read(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
230 231 232
	e = __lookup_nat_cache(nm_i, nid);
	if (e && !get_nat_flag(e, IS_CHECKPOINTED))
		is_cp = false;
233
	up_read(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
234
	return is_cp;
235 236
}

237
bool need_inode_block_update(struct f2fs_sb_info *sbi, nid_t ino)
238 239 240
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct nat_entry *e;
241
	bool need_update = true;
242

243
	down_read(&nm_i->nat_tree_lock);
244 245 246 247 248
	e = __lookup_nat_cache(nm_i, ino);
	if (e && get_nat_flag(e, HAS_LAST_FSYNC) &&
			(get_nat_flag(e, IS_CHECKPOINTED) ||
			 get_nat_flag(e, HAS_FSYNCED_INODE)))
		need_update = false;
249
	up_read(&nm_i->nat_tree_lock);
250
	return need_update;
251 252
}

253 254
static struct nat_entry *grab_nat_entry(struct f2fs_nm_info *nm_i, nid_t nid,
								bool no_fail)
J
Jaegeuk Kim 已提交
255 256 257
{
	struct nat_entry *new;

258 259 260 261 262 263 264 265 266 267 268 269 270
	if (no_fail) {
		new = f2fs_kmem_cache_alloc(nat_entry_slab, GFP_NOFS);
		f2fs_radix_tree_insert(&nm_i->nat_root, nid, new);
	} else {
		new = kmem_cache_alloc(nat_entry_slab, GFP_NOFS);
		if (!new)
			return NULL;
		if (radix_tree_insert(&nm_i->nat_root, nid, new)) {
			kmem_cache_free(nat_entry_slab, new);
			return NULL;
		}
	}

J
Jaegeuk Kim 已提交
271 272
	memset(new, 0, sizeof(struct nat_entry));
	nat_set_nid(new, nid);
273
	nat_reset_flag(new);
J
Jaegeuk Kim 已提交
274 275 276 277 278
	list_add_tail(&new->list, &nm_i->nat_entries);
	nm_i->nat_cnt++;
	return new;
}

279
static void cache_nat_entry(struct f2fs_sb_info *sbi, nid_t nid,
J
Jaegeuk Kim 已提交
280 281
						struct f2fs_nat_entry *ne)
{
282
	struct f2fs_nm_info *nm_i = NM_I(sbi);
J
Jaegeuk Kim 已提交
283
	struct nat_entry *e;
284

J
Jaegeuk Kim 已提交
285 286
	e = __lookup_nat_cache(nm_i, nid);
	if (!e) {
287 288 289
		e = grab_nat_entry(nm_i, nid, false);
		if (e)
			node_info_from_raw_nat(&e->ni, ne);
290
	} else {
E
Eric Biggers 已提交
291 292 293
		f2fs_bug_on(sbi, nat_get_ino(e) != le32_to_cpu(ne->ino) ||
				nat_get_blkaddr(e) !=
					le32_to_cpu(ne->block_addr) ||
294
				nat_get_version(e) != ne->version);
J
Jaegeuk Kim 已提交
295 296 297 298
	}
}

static void set_node_addr(struct f2fs_sb_info *sbi, struct node_info *ni,
299
			block_t new_blkaddr, bool fsync_done)
J
Jaegeuk Kim 已提交
300 301 302
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct nat_entry *e;
303

304
	down_write(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
305 306
	e = __lookup_nat_cache(nm_i, ni->nid);
	if (!e) {
307
		e = grab_nat_entry(nm_i, ni->nid, true);
308
		copy_node_info(&e->ni, ni);
309
		f2fs_bug_on(sbi, ni->blk_addr == NEW_ADDR);
J
Jaegeuk Kim 已提交
310 311 312 313 314 315
	} else if (new_blkaddr == NEW_ADDR) {
		/*
		 * when nid is reallocated,
		 * previous nat entry can be remained in nat cache.
		 * So, reinitialize it with new information.
		 */
316
		copy_node_info(&e->ni, ni);
317
		f2fs_bug_on(sbi, ni->blk_addr != NULL_ADDR);
J
Jaegeuk Kim 已提交
318 319 320
	}

	/* sanity check */
321 322
	f2fs_bug_on(sbi, nat_get_blkaddr(e) != ni->blk_addr);
	f2fs_bug_on(sbi, nat_get_blkaddr(e) == NULL_ADDR &&
J
Jaegeuk Kim 已提交
323
			new_blkaddr == NULL_ADDR);
324
	f2fs_bug_on(sbi, nat_get_blkaddr(e) == NEW_ADDR &&
J
Jaegeuk Kim 已提交
325
			new_blkaddr == NEW_ADDR);
326
	f2fs_bug_on(sbi, nat_get_blkaddr(e) != NEW_ADDR &&
J
Jaegeuk Kim 已提交
327 328 329
			nat_get_blkaddr(e) != NULL_ADDR &&
			new_blkaddr == NEW_ADDR);

A
arter97 已提交
330
	/* increment version no as node is removed */
J
Jaegeuk Kim 已提交
331 332 333 334 335 336 337
	if (nat_get_blkaddr(e) != NEW_ADDR && new_blkaddr == NULL_ADDR) {
		unsigned char version = nat_get_version(e);
		nat_set_version(e, inc_node_version(version));
	}

	/* change address */
	nat_set_blkaddr(e, new_blkaddr);
338 339
	if (new_blkaddr == NEW_ADDR || new_blkaddr == NULL_ADDR)
		set_nat_flag(e, IS_CHECKPOINTED, false);
J
Jaegeuk Kim 已提交
340
	__set_nat_cache_dirty(nm_i, e);
341 342

	/* update fsync_mark if its inode nat entry is still alive */
343 344
	if (ni->nid != ni->ino)
		e = __lookup_nat_cache(nm_i, ni->ino);
345 346 347 348 349
	if (e) {
		if (fsync_done && ni->nid == ni->ino)
			set_nat_flag(e, HAS_FSYNCED_INODE, true);
		set_nat_flag(e, HAS_LAST_FSYNC, fsync_done);
	}
350
	up_write(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
351 352
}

353
int try_to_free_nats(struct f2fs_sb_info *sbi, int nr_shrink)
J
Jaegeuk Kim 已提交
354 355
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
J
Jaegeuk Kim 已提交
356
	int nr = nr_shrink;
J
Jaegeuk Kim 已提交
357

358 359
	if (!down_write_trylock(&nm_i->nat_tree_lock))
		return 0;
J
Jaegeuk Kim 已提交
360 361 362 363 364 365 366 367

	while (nr_shrink && !list_empty(&nm_i->nat_entries)) {
		struct nat_entry *ne;
		ne = list_first_entry(&nm_i->nat_entries,
					struct nat_entry, list);
		__del_from_nat_cache(nm_i, ne);
		nr_shrink--;
	}
368
	up_write(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
369
	return nr - nr_shrink;
J
Jaegeuk Kim 已提交
370 371
}

J
Jaegeuk Kim 已提交
372
/*
A
arter97 已提交
373
 * This function always returns success
J
Jaegeuk Kim 已提交
374 375 376 377 378
 */
void get_node_info(struct f2fs_sb_info *sbi, nid_t nid, struct node_info *ni)
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct curseg_info *curseg = CURSEG_I(sbi, CURSEG_HOT_DATA);
379
	struct f2fs_journal *journal = curseg->journal;
J
Jaegeuk Kim 已提交
380 381 382 383 384
	nid_t start_nid = START_NID(nid);
	struct f2fs_nat_block *nat_blk;
	struct page *page = NULL;
	struct f2fs_nat_entry ne;
	struct nat_entry *e;
385
	pgoff_t index;
J
Jaegeuk Kim 已提交
386 387 388 389 390
	int i;

	ni->nid = nid;

	/* Check nat cache */
391
	down_read(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
392 393 394 395 396
	e = __lookup_nat_cache(nm_i, nid);
	if (e) {
		ni->ino = nat_get_ino(e);
		ni->blk_addr = nat_get_blkaddr(e);
		ni->version = nat_get_version(e);
397
		up_read(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
398
		return;
399
	}
J
Jaegeuk Kim 已提交
400

401 402
	memset(&ne, 0, sizeof(struct f2fs_nat_entry));

J
Jaegeuk Kim 已提交
403
	/* Check current segment summary */
404
	down_read(&curseg->journal_rwsem);
405
	i = lookup_journal_in_cursum(journal, NAT_JOURNAL, nid, 0);
J
Jaegeuk Kim 已提交
406
	if (i >= 0) {
407
		ne = nat_in_journal(journal, i);
J
Jaegeuk Kim 已提交
408 409
		node_info_from_raw_nat(ni, &ne);
	}
410
	up_read(&curseg->journal_rwsem);
411 412
	if (i >= 0) {
		up_read(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
413
		goto cache;
414
	}
J
Jaegeuk Kim 已提交
415 416

	/* Fill node_info from nat page */
417 418 419 420
	index = current_nat_addr(sbi, nid);
	up_read(&nm_i->nat_tree_lock);

	page = get_meta_page(sbi, index);
J
Jaegeuk Kim 已提交
421 422 423 424 425 426
	nat_blk = (struct f2fs_nat_block *)page_address(page);
	ne = nat_blk->entries[nid - start_nid];
	node_info_from_raw_nat(ni, &ne);
	f2fs_put_page(page, 1);
cache:
	/* cache nat entry */
427
	down_write(&nm_i->nat_tree_lock);
428
	cache_nat_entry(sbi, nid, &ne);
429
	up_write(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
430 431
}

432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454
/*
 * readahead MAX_RA_NODE number of node pages.
 */
static void ra_node_pages(struct page *parent, int start, int n)
{
	struct f2fs_sb_info *sbi = F2FS_P_SB(parent);
	struct blk_plug plug;
	int i, end;
	nid_t nid;

	blk_start_plug(&plug);

	/* Then, try readahead for siblings of the desired node */
	end = start + n;
	end = min(end, NIDS_PER_BLOCK);
	for (i = start; i < end; i++) {
		nid = get_nid(parent, i, false);
		ra_node_page(sbi, nid);
	}

	blk_finish_plug(&plug);
}

455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485
pgoff_t get_next_page_offset(struct dnode_of_data *dn, pgoff_t pgofs)
{
	const long direct_index = ADDRS_PER_INODE(dn->inode);
	const long direct_blks = ADDRS_PER_BLOCK;
	const long indirect_blks = ADDRS_PER_BLOCK * NIDS_PER_BLOCK;
	unsigned int skipped_unit = ADDRS_PER_BLOCK;
	int cur_level = dn->cur_level;
	int max_level = dn->max_level;
	pgoff_t base = 0;

	if (!dn->max_level)
		return pgofs + 1;

	while (max_level-- > cur_level)
		skipped_unit *= NIDS_PER_BLOCK;

	switch (dn->max_level) {
	case 3:
		base += 2 * indirect_blks;
	case 2:
		base += 2 * direct_blks;
	case 1:
		base += direct_index;
		break;
	default:
		f2fs_bug_on(F2FS_I_SB(dn->inode), 1);
	}

	return ((pgofs - base) / skipped_unit + 1) * skipped_unit + base;
}

J
Jaegeuk Kim 已提交
486
/*
J
Jaegeuk Kim 已提交
487 488 489
 * The maximum depth is four.
 * Offset[0] will have raw inode offset.
 */
490
static int get_node_path(struct inode *inode, long block,
491
				int offset[4], unsigned int noffset[4])
J
Jaegeuk Kim 已提交
492
{
493
	const long direct_index = ADDRS_PER_INODE(inode);
J
Jaegeuk Kim 已提交
494 495 496 497 498 499 500 501 502 503
	const long direct_blks = ADDRS_PER_BLOCK;
	const long dptrs_per_blk = NIDS_PER_BLOCK;
	const long indirect_blks = ADDRS_PER_BLOCK * NIDS_PER_BLOCK;
	const long dindirect_blks = indirect_blks * NIDS_PER_BLOCK;
	int n = 0;
	int level = 0;

	noffset[0] = 0;

	if (block < direct_index) {
504
		offset[n] = block;
J
Jaegeuk Kim 已提交
505 506 507 508 509 510
		goto got;
	}
	block -= direct_index;
	if (block < direct_blks) {
		offset[n++] = NODE_DIR1_BLOCK;
		noffset[n] = 1;
511
		offset[n] = block;
J
Jaegeuk Kim 已提交
512 513 514 515 516 517 518
		level = 1;
		goto got;
	}
	block -= direct_blks;
	if (block < direct_blks) {
		offset[n++] = NODE_DIR2_BLOCK;
		noffset[n] = 2;
519
		offset[n] = block;
J
Jaegeuk Kim 已提交
520 521 522 523 524 525 526 527 528
		level = 1;
		goto got;
	}
	block -= direct_blks;
	if (block < indirect_blks) {
		offset[n++] = NODE_IND1_BLOCK;
		noffset[n] = 3;
		offset[n++] = block / direct_blks;
		noffset[n] = 4 + offset[n - 1];
529
		offset[n] = block % direct_blks;
J
Jaegeuk Kim 已提交
530 531 532 533 534 535 536 537 538
		level = 2;
		goto got;
	}
	block -= indirect_blks;
	if (block < indirect_blks) {
		offset[n++] = NODE_IND2_BLOCK;
		noffset[n] = 4 + dptrs_per_blk;
		offset[n++] = block / direct_blks;
		noffset[n] = 5 + dptrs_per_blk + offset[n - 1];
539
		offset[n] = block % direct_blks;
J
Jaegeuk Kim 已提交
540 541 542 543 544 545 546 547 548 549 550 551 552 553
		level = 2;
		goto got;
	}
	block -= indirect_blks;
	if (block < dindirect_blks) {
		offset[n++] = NODE_DIND_BLOCK;
		noffset[n] = 5 + (dptrs_per_blk * 2);
		offset[n++] = block / indirect_blks;
		noffset[n] = 6 + (dptrs_per_blk * 2) +
			      offset[n - 1] * (dptrs_per_blk + 1);
		offset[n++] = (block / direct_blks) % dptrs_per_blk;
		noffset[n] = 7 + (dptrs_per_blk * 2) +
			      offset[n - 2] * (dptrs_per_blk + 1) +
			      offset[n - 1];
554
		offset[n] = block % direct_blks;
J
Jaegeuk Kim 已提交
555 556 557
		level = 3;
		goto got;
	} else {
558
		return -E2BIG;
J
Jaegeuk Kim 已提交
559 560 561 562 563 564 565
	}
got:
	return level;
}

/*
 * Caller should call f2fs_put_dnode(dn).
C
Chao Yu 已提交
566 567
 * Also, it should grab and release a rwsem by calling f2fs_lock_op() and
 * f2fs_unlock_op() only if ro is not set RDONLY_NODE.
568
 * In the case of RDONLY_NODE, we don't need to care about mutex.
J
Jaegeuk Kim 已提交
569
 */
570
int get_dnode_of_data(struct dnode_of_data *dn, pgoff_t index, int mode)
J
Jaegeuk Kim 已提交
571
{
572
	struct f2fs_sb_info *sbi = F2FS_I_SB(dn->inode);
J
Jaegeuk Kim 已提交
573
	struct page *npage[4];
574
	struct page *parent = NULL;
J
Jaegeuk Kim 已提交
575 576 577
	int offset[4];
	unsigned int noffset[4];
	nid_t nids[4];
578
	int level, i = 0;
J
Jaegeuk Kim 已提交
579 580
	int err = 0;

581
	level = get_node_path(dn->inode, index, offset, noffset);
582 583
	if (level < 0)
		return level;
J
Jaegeuk Kim 已提交
584 585

	nids[0] = dn->inode->i_ino;
586
	npage[0] = dn->inode_page;
J
Jaegeuk Kim 已提交
587

588 589 590 591 592
	if (!npage[0]) {
		npage[0] = get_node_page(sbi, nids[0]);
		if (IS_ERR(npage[0]))
			return PTR_ERR(npage[0]);
	}
593 594 595

	/* if inline_data is set, should not report any block indices */
	if (f2fs_has_inline_data(dn->inode) && index) {
596
		err = -ENOENT;
597 598 599 600
		f2fs_put_page(npage[0], 1);
		goto release_out;
	}

J
Jaegeuk Kim 已提交
601
	parent = npage[0];
602 603
	if (level != 0)
		nids[1] = get_nid(parent, offset[0], true);
J
Jaegeuk Kim 已提交
604 605 606 607 608 609 610
	dn->inode_page = npage[0];
	dn->inode_page_locked = true;

	/* get indirect or direct nodes */
	for (i = 1; i <= level; i++) {
		bool done = false;

611
		if (!nids[i] && mode == ALLOC_NODE) {
J
Jaegeuk Kim 已提交
612 613 614 615 616 617 618
			/* alloc new node */
			if (!alloc_nid(sbi, &(nids[i]))) {
				err = -ENOSPC;
				goto release_pages;
			}

			dn->nid = nids[i];
Y
Yunlei He 已提交
619
			npage[i] = new_node_page(dn, noffset[i]);
J
Jaegeuk Kim 已提交
620 621 622 623 624 625 626 627 628
			if (IS_ERR(npage[i])) {
				alloc_nid_failed(sbi, nids[i]);
				err = PTR_ERR(npage[i]);
				goto release_pages;
			}

			set_nid(parent, offset[i - 1], nids[i], i == 1);
			alloc_nid_done(sbi, nids[i]);
			done = true;
629
		} else if (mode == LOOKUP_NODE_RA && i == level && level > 1) {
J
Jaegeuk Kim 已提交
630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659
			npage[i] = get_node_page_ra(parent, offset[i - 1]);
			if (IS_ERR(npage[i])) {
				err = PTR_ERR(npage[i]);
				goto release_pages;
			}
			done = true;
		}
		if (i == 1) {
			dn->inode_page_locked = false;
			unlock_page(parent);
		} else {
			f2fs_put_page(parent, 1);
		}

		if (!done) {
			npage[i] = get_node_page(sbi, nids[i]);
			if (IS_ERR(npage[i])) {
				err = PTR_ERR(npage[i]);
				f2fs_put_page(npage[0], 0);
				goto release_out;
			}
		}
		if (i < level) {
			parent = npage[i];
			nids[i + 1] = get_nid(parent, offset[i], false);
		}
	}
	dn->nid = nids[level];
	dn->ofs_in_node = offset[level];
	dn->node_page = npage[level];
660 661
	dn->data_blkaddr = datablock_addr(dn->inode,
				dn->node_page, dn->ofs_in_node);
J
Jaegeuk Kim 已提交
662 663 664 665 666 667 668 669 670
	return 0;

release_pages:
	f2fs_put_page(parent, 1);
	if (i > 1)
		f2fs_put_page(npage[0], 0);
release_out:
	dn->inode_page = NULL;
	dn->node_page = NULL;
671 672 673
	if (err == -ENOENT) {
		dn->cur_level = i;
		dn->max_level = level;
674
		dn->ofs_in_node = offset[level];
675
	}
J
Jaegeuk Kim 已提交
676 677 678 679 680
	return err;
}

static void truncate_node(struct dnode_of_data *dn)
{
681
	struct f2fs_sb_info *sbi = F2FS_I_SB(dn->inode);
J
Jaegeuk Kim 已提交
682 683 684
	struct node_info ni;

	get_node_info(sbi, dn->nid, &ni);
685
	f2fs_bug_on(sbi, ni.blk_addr == NULL_ADDR);
J
Jaegeuk Kim 已提交
686 687

	/* Deallocate node address */
688
	invalidate_blocks(sbi, ni.blk_addr);
689
	dec_valid_node_count(sbi, dn->inode, dn->nid == dn->inode->i_ino);
690
	set_node_addr(sbi, &ni, NULL_ADDR, false);
J
Jaegeuk Kim 已提交
691 692 693 694

	if (dn->nid == dn->inode->i_ino) {
		remove_orphan_inode(sbi, dn->nid);
		dec_valid_inode_count(sbi);
695
		f2fs_inode_synced(dn->inode);
J
Jaegeuk Kim 已提交
696
	}
697

J
Jaegeuk Kim 已提交
698
	clear_node_page_dirty(dn->node_page);
699
	set_sbi_flag(sbi, SBI_IS_DIRTY);
J
Jaegeuk Kim 已提交
700 701

	f2fs_put_page(dn->node_page, 1);
702 703 704 705

	invalidate_mapping_pages(NODE_MAPPING(sbi),
			dn->node_page->index, dn->node_page->index);

J
Jaegeuk Kim 已提交
706
	dn->node_page = NULL;
707
	trace_f2fs_truncate_node(dn->inode, dn->nid, ni.blk_addr);
J
Jaegeuk Kim 已提交
708 709 710 711 712 713 714 715 716 717
}

static int truncate_dnode(struct dnode_of_data *dn)
{
	struct page *page;

	if (dn->nid == 0)
		return 1;

	/* get direct node */
718
	page = get_node_page(F2FS_I_SB(dn->inode), dn->nid);
J
Jaegeuk Kim 已提交
719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745
	if (IS_ERR(page) && PTR_ERR(page) == -ENOENT)
		return 1;
	else if (IS_ERR(page))
		return PTR_ERR(page);

	/* Make dnode_of_data for parameter */
	dn->node_page = page;
	dn->ofs_in_node = 0;
	truncate_data_blocks(dn);
	truncate_node(dn);
	return 1;
}

static int truncate_nodes(struct dnode_of_data *dn, unsigned int nofs,
						int ofs, int depth)
{
	struct dnode_of_data rdn = *dn;
	struct page *page;
	struct f2fs_node *rn;
	nid_t child_nid;
	unsigned int child_nofs;
	int freed = 0;
	int i, ret;

	if (dn->nid == 0)
		return NIDS_PER_BLOCK + 1;

746 747
	trace_f2fs_truncate_nodes_enter(dn->inode, dn->nid, dn->data_blkaddr);

748
	page = get_node_page(F2FS_I_SB(dn->inode), dn->nid);
749 750
	if (IS_ERR(page)) {
		trace_f2fs_truncate_nodes_exit(dn->inode, PTR_ERR(page));
J
Jaegeuk Kim 已提交
751
		return PTR_ERR(page);
752
	}
J
Jaegeuk Kim 已提交
753

754 755
	ra_node_pages(page, ofs, NIDS_PER_BLOCK);

756
	rn = F2FS_NODE(page);
J
Jaegeuk Kim 已提交
757 758 759 760 761 762 763 764 765
	if (depth < 3) {
		for (i = ofs; i < NIDS_PER_BLOCK; i++, freed++) {
			child_nid = le32_to_cpu(rn->in.nid[i]);
			if (child_nid == 0)
				continue;
			rdn.nid = child_nid;
			ret = truncate_dnode(&rdn);
			if (ret < 0)
				goto out_err;
766 767
			if (set_nid(page, i, 0, false))
				dn->node_changed = true;
J
Jaegeuk Kim 已提交
768 769 770 771 772 773 774 775 776 777 778 779
		}
	} else {
		child_nofs = nofs + ofs * (NIDS_PER_BLOCK + 1) + 1;
		for (i = ofs; i < NIDS_PER_BLOCK; i++) {
			child_nid = le32_to_cpu(rn->in.nid[i]);
			if (child_nid == 0) {
				child_nofs += NIDS_PER_BLOCK + 1;
				continue;
			}
			rdn.nid = child_nid;
			ret = truncate_nodes(&rdn, child_nofs, 0, depth - 1);
			if (ret == (NIDS_PER_BLOCK + 1)) {
780 781
				if (set_nid(page, i, 0, false))
					dn->node_changed = true;
J
Jaegeuk Kim 已提交
782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797
				child_nofs += ret;
			} else if (ret < 0 && ret != -ENOENT) {
				goto out_err;
			}
		}
		freed = child_nofs;
	}

	if (!ofs) {
		/* remove current indirect node */
		dn->node_page = page;
		truncate_node(dn);
		freed++;
	} else {
		f2fs_put_page(page, 1);
	}
798
	trace_f2fs_truncate_nodes_exit(dn->inode, freed);
J
Jaegeuk Kim 已提交
799 800 801 802
	return freed;

out_err:
	f2fs_put_page(page, 1);
803
	trace_f2fs_truncate_nodes_exit(dn->inode, ret);
J
Jaegeuk Kim 已提交
804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821
	return ret;
}

static int truncate_partial_nodes(struct dnode_of_data *dn,
			struct f2fs_inode *ri, int *offset, int depth)
{
	struct page *pages[2];
	nid_t nid[3];
	nid_t child_nid;
	int err = 0;
	int i;
	int idx = depth - 2;

	nid[0] = le32_to_cpu(ri->i_nid[offset[0] - NODE_DIR1_BLOCK]);
	if (!nid[0])
		return 0;

	/* get indirect nodes in the path */
822
	for (i = 0; i < idx + 1; i++) {
A
arter97 已提交
823
		/* reference count'll be increased */
824
		pages[i] = get_node_page(F2FS_I_SB(dn->inode), nid[i]);
J
Jaegeuk Kim 已提交
825 826
		if (IS_ERR(pages[i])) {
			err = PTR_ERR(pages[i]);
827
			idx = i - 1;
J
Jaegeuk Kim 已提交
828 829 830 831 832
			goto fail;
		}
		nid[i + 1] = get_nid(pages[i], offset[i + 1], false);
	}

833 834
	ra_node_pages(pages[idx], offset[idx + 1], NIDS_PER_BLOCK);

J
Jaegeuk Kim 已提交
835
	/* free direct nodes linked to a partial indirect node */
836
	for (i = offset[idx + 1]; i < NIDS_PER_BLOCK; i++) {
J
Jaegeuk Kim 已提交
837 838 839 840 841 842 843
		child_nid = get_nid(pages[idx], i, false);
		if (!child_nid)
			continue;
		dn->nid = child_nid;
		err = truncate_dnode(dn);
		if (err < 0)
			goto fail;
844 845
		if (set_nid(pages[idx], i, 0, false))
			dn->node_changed = true;
J
Jaegeuk Kim 已提交
846 847
	}

848
	if (offset[idx + 1] == 0) {
J
Jaegeuk Kim 已提交
849 850 851 852 853 854 855
		dn->node_page = pages[idx];
		dn->nid = nid[idx];
		truncate_node(dn);
	} else {
		f2fs_put_page(pages[idx], 1);
	}
	offset[idx]++;
856 857
	offset[idx + 1] = 0;
	idx--;
J
Jaegeuk Kim 已提交
858
fail:
859
	for (i = idx; i >= 0; i--)
J
Jaegeuk Kim 已提交
860
		f2fs_put_page(pages[i], 1);
861 862 863

	trace_f2fs_truncate_partial_nodes(dn->inode, nid, depth, err);

J
Jaegeuk Kim 已提交
864 865 866
	return err;
}

J
Jaegeuk Kim 已提交
867
/*
J
Jaegeuk Kim 已提交
868 869 870 871
 * All the block addresses of data and nodes should be nullified.
 */
int truncate_inode_blocks(struct inode *inode, pgoff_t from)
{
872
	struct f2fs_sb_info *sbi = F2FS_I_SB(inode);
J
Jaegeuk Kim 已提交
873 874
	int err = 0, cont = 1;
	int level, offset[4], noffset[4];
J
Jaegeuk Kim 已提交
875
	unsigned int nofs = 0;
876
	struct f2fs_inode *ri;
J
Jaegeuk Kim 已提交
877 878 879
	struct dnode_of_data dn;
	struct page *page;

880 881
	trace_f2fs_truncate_inode_blocks_enter(inode, from);

882
	level = get_node_path(inode, from, offset, noffset);
883 884
	if (level < 0)
		return level;
885

J
Jaegeuk Kim 已提交
886
	page = get_node_page(sbi, inode->i_ino);
887 888
	if (IS_ERR(page)) {
		trace_f2fs_truncate_inode_blocks_exit(inode, PTR_ERR(page));
J
Jaegeuk Kim 已提交
889
		return PTR_ERR(page);
890
	}
J
Jaegeuk Kim 已提交
891 892 893 894

	set_new_dnode(&dn, inode, page, NULL, 0);
	unlock_page(page);

895
	ri = F2FS_INODE(page);
J
Jaegeuk Kim 已提交
896 897 898 899 900 901 902 903 904
	switch (level) {
	case 0:
	case 1:
		nofs = noffset[1];
		break;
	case 2:
		nofs = noffset[1];
		if (!offset[level - 1])
			goto skip_partial;
905
		err = truncate_partial_nodes(&dn, ri, offset, level);
J
Jaegeuk Kim 已提交
906 907 908 909 910 911 912 913
		if (err < 0 && err != -ENOENT)
			goto fail;
		nofs += 1 + NIDS_PER_BLOCK;
		break;
	case 3:
		nofs = 5 + 2 * NIDS_PER_BLOCK;
		if (!offset[level - 1])
			goto skip_partial;
914
		err = truncate_partial_nodes(&dn, ri, offset, level);
J
Jaegeuk Kim 已提交
915 916 917 918 919 920 921 922 923
		if (err < 0 && err != -ENOENT)
			goto fail;
		break;
	default:
		BUG();
	}

skip_partial:
	while (cont) {
924
		dn.nid = le32_to_cpu(ri->i_nid[offset[0] - NODE_DIR1_BLOCK]);
J
Jaegeuk Kim 已提交
925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946
		switch (offset[0]) {
		case NODE_DIR1_BLOCK:
		case NODE_DIR2_BLOCK:
			err = truncate_dnode(&dn);
			break;

		case NODE_IND1_BLOCK:
		case NODE_IND2_BLOCK:
			err = truncate_nodes(&dn, nofs, offset[1], 2);
			break;

		case NODE_DIND_BLOCK:
			err = truncate_nodes(&dn, nofs, offset[1], 3);
			cont = 0;
			break;

		default:
			BUG();
		}
		if (err < 0 && err != -ENOENT)
			goto fail;
		if (offset[1] == 0 &&
947
				ri->i_nid[offset[0] - NODE_DIR1_BLOCK]) {
J
Jaegeuk Kim 已提交
948
			lock_page(page);
949
			BUG_ON(page->mapping != NODE_MAPPING(sbi));
950
			f2fs_wait_on_page_writeback(page, NODE, true);
951
			ri->i_nid[offset[0] - NODE_DIR1_BLOCK] = 0;
J
Jaegeuk Kim 已提交
952 953 954 955 956 957 958 959 960
			set_page_dirty(page);
			unlock_page(page);
		}
		offset[1] = 0;
		offset[0]++;
		nofs += err;
	}
fail:
	f2fs_put_page(page, 0);
961
	trace_f2fs_truncate_inode_blocks_exit(inode, err);
J
Jaegeuk Kim 已提交
962 963 964
	return err > 0 ? 0 : err;
}

965 966
/* caller must lock inode page */
int truncate_xattr_node(struct inode *inode)
967
{
968
	struct f2fs_sb_info *sbi = F2FS_I_SB(inode);
969 970 971 972 973 974 975 976 977 978 979
	nid_t nid = F2FS_I(inode)->i_xattr_nid;
	struct dnode_of_data dn;
	struct page *npage;

	if (!nid)
		return 0;

	npage = get_node_page(sbi, nid);
	if (IS_ERR(npage))
		return PTR_ERR(npage);

980
	f2fs_i_xnid_write(inode, 0);
J
Jaegeuk Kim 已提交
981

982
	set_new_dnode(&dn, inode, NULL, npage, nid);
983 984 985 986
	truncate_node(&dn);
	return 0;
}

987
/*
C
Chao Yu 已提交
988 989
 * Caller should grab and release a rwsem by calling f2fs_lock_op() and
 * f2fs_unlock_op().
990
 */
C
Chao Yu 已提交
991
int remove_inode_page(struct inode *inode)
J
Jaegeuk Kim 已提交
992 993
{
	struct dnode_of_data dn;
C
Chao Yu 已提交
994
	int err;
J
Jaegeuk Kim 已提交
995

996
	set_new_dnode(&dn, inode, NULL, NULL, inode->i_ino);
C
Chao Yu 已提交
997 998 999
	err = get_dnode_of_data(&dn, 0, LOOKUP_NODE);
	if (err)
		return err;
J
Jaegeuk Kim 已提交
1000

1001
	err = truncate_xattr_node(inode);
C
Chao Yu 已提交
1002
	if (err) {
1003
		f2fs_put_dnode(&dn);
C
Chao Yu 已提交
1004
		return err;
J
Jaegeuk Kim 已提交
1005
	}
1006 1007 1008 1009 1010 1011

	/* remove potential inline_data blocks */
	if (S_ISREG(inode->i_mode) || S_ISDIR(inode->i_mode) ||
				S_ISLNK(inode->i_mode))
		truncate_data_blocks_range(&dn, 1);

A
arter97 已提交
1012
	/* 0 is possible, after f2fs_new_inode() has failed */
1013
	f2fs_bug_on(F2FS_I_SB(inode),
1014
			inode->i_blocks != 0 && inode->i_blocks != 8);
1015 1016

	/* will put inode & node pages */
1017
	truncate_node(&dn);
C
Chao Yu 已提交
1018
	return 0;
J
Jaegeuk Kim 已提交
1019 1020
}

1021
struct page *new_inode_page(struct inode *inode)
J
Jaegeuk Kim 已提交
1022 1023 1024 1025 1026
{
	struct dnode_of_data dn;

	/* allocate inode page for new inode */
	set_new_dnode(&dn, inode, NULL, NULL, inode->i_ino);
1027 1028

	/* caller should f2fs_put_page(page, 1); */
Y
Yunlei He 已提交
1029
	return new_node_page(&dn, 0);
J
Jaegeuk Kim 已提交
1030 1031
}

Y
Yunlei He 已提交
1032
struct page *new_node_page(struct dnode_of_data *dn, unsigned int ofs)
J
Jaegeuk Kim 已提交
1033
{
1034
	struct f2fs_sb_info *sbi = F2FS_I_SB(dn->inode);
1035
	struct node_info new_ni;
J
Jaegeuk Kim 已提交
1036 1037 1038
	struct page *page;
	int err;

1039
	if (unlikely(is_inode_flag_set(dn->inode, FI_NO_ALLOC)))
J
Jaegeuk Kim 已提交
1040 1041
		return ERR_PTR(-EPERM);

1042
	page = f2fs_grab_cache_page(NODE_MAPPING(sbi), dn->nid, false);
J
Jaegeuk Kim 已提交
1043 1044 1045
	if (!page)
		return ERR_PTR(-ENOMEM);

C
Chao Yu 已提交
1046
	if (unlikely((err = inc_valid_node_count(sbi, dn->inode, !ofs))))
1047
		goto fail;
C
Chao Yu 已提交
1048

1049 1050 1051 1052 1053
#ifdef CONFIG_F2FS_CHECK_FS
	get_node_info(sbi, dn->nid, &new_ni);
	f2fs_bug_on(sbi, new_ni.blk_addr != NULL_ADDR);
#endif
	new_ni.nid = dn->nid;
J
Jaegeuk Kim 已提交
1054
	new_ni.ino = dn->inode->i_ino;
1055 1056 1057
	new_ni.blk_addr = NULL_ADDR;
	new_ni.flag = 0;
	new_ni.version = 0;
1058
	set_node_addr(sbi, &new_ni, NEW_ADDR, false);
1059

1060
	f2fs_wait_on_page_writeback(page, NODE, true);
1061
	fill_node_footer(page, dn->nid, dn->inode->i_ino, ofs, true);
1062
	set_cold_node(dn->inode, page);
1063 1064
	if (!PageUptodate(page))
		SetPageUptodate(page);
1065 1066
	if (set_page_dirty(page))
		dn->node_changed = true;
J
Jaegeuk Kim 已提交
1067

1068
	if (f2fs_has_xattr_block(ofs))
1069
		f2fs_i_xnid_write(dn->inode, dn->nid);
1070

J
Jaegeuk Kim 已提交
1071 1072 1073 1074 1075
	if (ofs == 0)
		inc_valid_inode_count(sbi);
	return page;

fail:
1076
	clear_node_page_dirty(page);
J
Jaegeuk Kim 已提交
1077 1078 1079 1080
	f2fs_put_page(page, 1);
	return ERR_PTR(err);
}

1081 1082 1083
/*
 * Caller should do after getting the following values.
 * 0: f2fs_put_page(page, 0)
1084
 * LOCKED_PAGE or error: f2fs_put_page(page, 1)
1085
 */
M
Mike Christie 已提交
1086
static int read_node_page(struct page *page, int op_flags)
J
Jaegeuk Kim 已提交
1087
{
1088
	struct f2fs_sb_info *sbi = F2FS_P_SB(page);
J
Jaegeuk Kim 已提交
1089
	struct node_info ni;
1090
	struct f2fs_io_info fio = {
1091
		.sbi = sbi,
1092
		.type = NODE,
M
Mike Christie 已提交
1093 1094
		.op = REQ_OP_READ,
		.op_flags = op_flags,
1095
		.page = page,
1096
		.encrypted_page = NULL,
1097
	};
J
Jaegeuk Kim 已提交
1098

1099 1100 1101
	if (PageUptodate(page))
		return LOCKED_PAGE;

J
Jaegeuk Kim 已提交
1102 1103
	get_node_info(sbi, page->index, &ni);

1104
	if (unlikely(ni.blk_addr == NULL_ADDR)) {
1105
		ClearPageUptodate(page);
J
Jaegeuk Kim 已提交
1106
		return -ENOENT;
1107 1108
	}

1109
	fio.new_blkaddr = fio.old_blkaddr = ni.blk_addr;
1110
	return f2fs_submit_page_bio(&fio);
J
Jaegeuk Kim 已提交
1111 1112
}

J
Jaegeuk Kim 已提交
1113
/*
J
Jaegeuk Kim 已提交
1114 1115 1116 1117 1118
 * Readahead a node page
 */
void ra_node_page(struct f2fs_sb_info *sbi, nid_t nid)
{
	struct page *apage;
1119
	int err;
J
Jaegeuk Kim 已提交
1120

1121 1122 1123 1124
	if (!nid)
		return;
	f2fs_bug_on(sbi, check_nid_range(sbi, nid));

1125 1126 1127 1128
	rcu_read_lock();
	apage = radix_tree_lookup(&NODE_MAPPING(sbi)->page_tree, nid);
	rcu_read_unlock();
	if (apage)
1129
		return;
J
Jaegeuk Kim 已提交
1130

1131
	apage = f2fs_grab_cache_page(NODE_MAPPING(sbi), nid, false);
J
Jaegeuk Kim 已提交
1132 1133 1134
	if (!apage)
		return;

1135
	err = read_node_page(apage, REQ_RAHEAD);
1136
	f2fs_put_page(apage, err ? 1 : 0);
J
Jaegeuk Kim 已提交
1137 1138
}

J
Jaegeuk Kim 已提交
1139
static struct page *__get_node_page(struct f2fs_sb_info *sbi, pgoff_t nid,
1140
					struct page *parent, int start)
J
Jaegeuk Kim 已提交
1141 1142
{
	struct page *page;
1143
	int err;
J
Jaegeuk Kim 已提交
1144 1145 1146

	if (!nid)
		return ERR_PTR(-ENOENT);
1147
	f2fs_bug_on(sbi, check_nid_range(sbi, nid));
1148
repeat:
1149
	page = f2fs_grab_cache_page(NODE_MAPPING(sbi), nid, false);
J
Jaegeuk Kim 已提交
1150 1151 1152
	if (!page)
		return ERR_PTR(-ENOMEM);

1153
	err = read_node_page(page, 0);
1154 1155
	if (err < 0) {
		f2fs_put_page(page, 1);
J
Jaegeuk Kim 已提交
1156
		return ERR_PTR(err);
1157
	} else if (err == LOCKED_PAGE) {
1158
		err = 0;
1159
		goto page_hit;
1160
	}
J
Jaegeuk Kim 已提交
1161

1162
	if (parent)
1163
		ra_node_pages(parent, start + 1, MAX_RA_NODE);
1164

J
Jaegeuk Kim 已提交
1165
	lock_page(page);
1166

1167
	if (unlikely(page->mapping != NODE_MAPPING(sbi))) {
1168 1169 1170
		f2fs_put_page(page, 1);
		goto repeat;
	}
1171

1172 1173
	if (unlikely(!PageUptodate(page))) {
		err = -EIO;
1174
		goto out_err;
1175
	}
C
Chao Yu 已提交
1176 1177 1178 1179 1180

	if (!f2fs_inode_chksum_verify(sbi, page)) {
		err = -EBADMSG;
		goto out_err;
	}
1181
page_hit:
1182
	if(unlikely(nid != nid_of_node(page))) {
1183 1184 1185 1186 1187 1188
		f2fs_msg(sbi->sb, KERN_WARNING, "inconsistent node block, "
			"nid:%lu, node_footer[nid:%u,ino:%u,ofs:%u,cpver:%llu,blkaddr:%u]",
			nid, nid_of_node(page), ino_of_node(page),
			ofs_of_node(page), cpver_of_node(page),
			next_blkaddr_of_node(page));
		err = -EINVAL;
1189
out_err:
1190
		ClearPageUptodate(page);
1191
		f2fs_put_page(page, 1);
1192
		return ERR_PTR(err);
1193
	}
J
Jaegeuk Kim 已提交
1194 1195 1196
	return page;
}

1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209
struct page *get_node_page(struct f2fs_sb_info *sbi, pgoff_t nid)
{
	return __get_node_page(sbi, nid, NULL, 0);
}

struct page *get_node_page_ra(struct page *parent, int start)
{
	struct f2fs_sb_info *sbi = F2FS_P_SB(parent);
	nid_t nid = get_nid(parent, start, false);

	return __get_node_page(sbi, nid, parent, start);
}

1210 1211 1212 1213
static void flush_inline_data(struct f2fs_sb_info *sbi, nid_t ino)
{
	struct inode *inode;
	struct page *page;
1214
	int ret;
1215 1216 1217 1218 1219 1220

	/* should flush inline_data before evict_inode */
	inode = ilookup(sbi->sb, ino);
	if (!inode)
		return;

C
Chao Yu 已提交
1221 1222
	page = f2fs_pagecache_get_page(inode->i_mapping, 0,
					FGP_LOCK|FGP_NOWAIT, 0);
1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234
	if (!page)
		goto iput_out;

	if (!PageUptodate(page))
		goto page_out;

	if (!PageDirty(page))
		goto page_out;

	if (!clear_page_dirty_for_io(page))
		goto page_out;

1235 1236
	ret = f2fs_write_inline_data(inode, page);
	inode_dec_dirty_pages(inode);
1237
	remove_dirty_inode(inode);
1238
	if (ret)
1239 1240
		set_page_dirty(page);
page_out:
1241
	f2fs_put_page(page, 1);
1242 1243 1244 1245
iput_out:
	iput(inode);
}

1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305
static struct page *last_fsync_dnode(struct f2fs_sb_info *sbi, nid_t ino)
{
	pgoff_t index, end;
	struct pagevec pvec;
	struct page *last_page = NULL;

	pagevec_init(&pvec, 0);
	index = 0;
	end = ULONG_MAX;

	while (index <= end) {
		int i, nr_pages;
		nr_pages = pagevec_lookup_tag(&pvec, NODE_MAPPING(sbi), &index,
				PAGECACHE_TAG_DIRTY,
				min(end - index, (pgoff_t)PAGEVEC_SIZE-1) + 1);
		if (nr_pages == 0)
			break;

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

			if (unlikely(f2fs_cp_error(sbi))) {
				f2fs_put_page(last_page, 0);
				pagevec_release(&pvec);
				return ERR_PTR(-EIO);
			}

			if (!IS_DNODE(page) || !is_cold_node(page))
				continue;
			if (ino_of_node(page) != ino)
				continue;

			lock_page(page);

			if (unlikely(page->mapping != NODE_MAPPING(sbi))) {
continue_unlock:
				unlock_page(page);
				continue;
			}
			if (ino_of_node(page) != ino)
				goto continue_unlock;

			if (!PageDirty(page)) {
				/* someone wrote it for us */
				goto continue_unlock;
			}

			if (last_page)
				f2fs_put_page(last_page, 0);

			get_page(page);
			last_page = page;
			unlock_page(page);
		}
		pagevec_release(&pvec);
		cond_resched();
	}
	return last_page;
}

1306
static int __write_node_page(struct page *page, bool atomic, bool *submitted,
C
Chao Yu 已提交
1307 1308
				struct writeback_control *wbc, bool do_balance,
				enum iostat_type io_type)
1309 1310 1311 1312 1313 1314
{
	struct f2fs_sb_info *sbi = F2FS_P_SB(page);
	nid_t nid;
	struct node_info ni;
	struct f2fs_io_info fio = {
		.sbi = sbi,
C
Chao Yu 已提交
1315
		.ino = ino_of_node(page),
1316 1317 1318 1319 1320
		.type = NODE,
		.op = REQ_OP_WRITE,
		.op_flags = wbc_to_write_flags(wbc),
		.page = page,
		.encrypted_page = NULL,
1321
		.submitted = false,
C
Chao Yu 已提交
1322
		.io_type = io_type,
1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353
	};

	trace_f2fs_writepage(page, NODE);

	if (unlikely(is_sbi_flag_set(sbi, SBI_POR_DOING)))
		goto redirty_out;
	if (unlikely(f2fs_cp_error(sbi)))
		goto redirty_out;

	/* get old block addr of this node page */
	nid = nid_of_node(page);
	f2fs_bug_on(sbi, page->index != nid);

	if (wbc->for_reclaim) {
		if (!down_read_trylock(&sbi->node_write))
			goto redirty_out;
	} else {
		down_read(&sbi->node_write);
	}

	get_node_info(sbi, nid, &ni);

	/* This page is already truncated */
	if (unlikely(ni.blk_addr == NULL_ADDR)) {
		ClearPageUptodate(page);
		dec_page_count(sbi, F2FS_DIRTY_NODES);
		up_read(&sbi->node_write);
		unlock_page(page);
		return 0;
	}

1354 1355 1356
	if (atomic && !test_opt(sbi, NOBARRIER))
		fio.op_flags |= REQ_PREFLUSH | REQ_FUA;

1357 1358 1359 1360 1361 1362 1363
	set_page_writeback(page);
	fio.old_blkaddr = ni.blk_addr;
	write_node_page(nid, &fio);
	set_node_addr(sbi, &ni, fio.new_blkaddr, is_fsync_dnode(page));
	dec_page_count(sbi, F2FS_DIRTY_NODES);
	up_read(&sbi->node_write);

1364
	if (wbc->for_reclaim) {
1365 1366
		f2fs_submit_merged_write_cond(sbi, page->mapping->host, 0,
						page->index, NODE);
1367 1368
		submitted = NULL;
	}
1369 1370 1371

	unlock_page(page);

1372
	if (unlikely(f2fs_cp_error(sbi))) {
1373
		f2fs_submit_merged_write(sbi, NODE);
1374 1375 1376 1377
		submitted = NULL;
	}
	if (submitted)
		*submitted = fio.submitted;
1378

1379 1380
	if (do_balance)
		f2fs_balance_fs(sbi, false);
1381 1382 1383 1384 1385 1386 1387
	return 0;

redirty_out:
	redirty_page_for_writepage(wbc, page);
	return AOP_WRITEPAGE_ACTIVATE;
}

1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418
void move_node_page(struct page *node_page, int gc_type)
{
	if (gc_type == FG_GC) {
		struct writeback_control wbc = {
			.sync_mode = WB_SYNC_ALL,
			.nr_to_write = 1,
			.for_reclaim = 0,
		};

		set_page_dirty(node_page);
		f2fs_wait_on_page_writeback(node_page, NODE, true);

		f2fs_bug_on(F2FS_P_SB(node_page), PageWriteback(node_page));
		if (!clear_page_dirty_for_io(node_page))
			goto out_page;

		if (__write_node_page(node_page, false, NULL,
					&wbc, false, FS_GC_NODE_IO))
			unlock_page(node_page);
		goto release_page;
	} else {
		/* set page dirty and write it */
		if (!PageWriteback(node_page))
			set_page_dirty(node_page);
	}
out_page:
	unlock_page(node_page);
release_page:
	f2fs_put_page(node_page, 0);
}

1419 1420 1421
static int f2fs_write_node_page(struct page *page,
				struct writeback_control *wbc)
{
C
Chao Yu 已提交
1422
	return __write_node_page(page, false, NULL, wbc, false, FS_NODE_IO);
1423 1424
}

1425
int fsync_node_pages(struct f2fs_sb_info *sbi, struct inode *inode,
1426
			struct writeback_control *wbc, bool atomic)
J
Jaegeuk Kim 已提交
1427 1428
{
	pgoff_t index, end;
1429
	pgoff_t last_idx = ULONG_MAX;
J
Jaegeuk Kim 已提交
1430
	struct pagevec pvec;
1431
	int ret = 0;
1432 1433
	struct page *last_page = NULL;
	bool marked = false;
1434
	nid_t ino = inode->i_ino;
1435

1436 1437 1438 1439 1440 1441
	if (atomic) {
		last_page = last_fsync_dnode(sbi, ino);
		if (IS_ERR_OR_NULL(last_page))
			return PTR_ERR_OR_ZERO(last_page);
	}
retry:
1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455
	pagevec_init(&pvec, 0);
	index = 0;
	end = ULONG_MAX;

	while (index <= end) {
		int i, nr_pages;
		nr_pages = pagevec_lookup_tag(&pvec, NODE_MAPPING(sbi), &index,
				PAGECACHE_TAG_DIRTY,
				min(end - index, (pgoff_t)PAGEVEC_SIZE-1) + 1);
		if (nr_pages == 0)
			break;

		for (i = 0; i < nr_pages; i++) {
			struct page *page = pvec.pages[i];
1456
			bool submitted = false;
1457 1458

			if (unlikely(f2fs_cp_error(sbi))) {
1459
				f2fs_put_page(last_page, 0);
1460
				pagevec_release(&pvec);
1461 1462
				ret = -EIO;
				goto out;
1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479
			}

			if (!IS_DNODE(page) || !is_cold_node(page))
				continue;
			if (ino_of_node(page) != ino)
				continue;

			lock_page(page);

			if (unlikely(page->mapping != NODE_MAPPING(sbi))) {
continue_unlock:
				unlock_page(page);
				continue;
			}
			if (ino_of_node(page) != ino)
				goto continue_unlock;

1480
			if (!PageDirty(page) && page != last_page) {
1481 1482 1483 1484 1485 1486 1487
				/* someone wrote it for us */
				goto continue_unlock;
			}

			f2fs_wait_on_page_writeback(page, NODE, true);
			BUG_ON(PageWriteback(page));

1488 1489 1490
			set_fsync_mark(page, 0);
			set_dentry_mark(page, 0);

1491 1492
			if (!atomic || page == last_page) {
				set_fsync_mark(page, 1);
1493 1494 1495 1496
				if (IS_INODE(page)) {
					if (is_inode_flag_set(inode,
								FI_DIRTY_INODE))
						update_inode(inode, page);
1497
					set_dentry_mark(page,
1498
						need_dentry_mark(sbi, ino));
1499
				}
1500 1501 1502 1503 1504 1505 1506
				/*  may be written by other thread */
				if (!PageDirty(page))
					set_page_dirty(page);
			}

			if (!clear_page_dirty_for_io(page))
				goto continue_unlock;
1507

1508
			ret = __write_node_page(page, atomic &&
1509
						page == last_page,
C
Chao Yu 已提交
1510 1511
						&submitted, wbc, true,
						FS_NODE_IO);
1512
			if (ret) {
1513
				unlock_page(page);
1514 1515
				f2fs_put_page(last_page, 0);
				break;
1516
			} else if (submitted) {
1517
				last_idx = page->index;
1518
			}
1519

1520 1521 1522
			if (page == last_page) {
				f2fs_put_page(page, 0);
				marked = true;
1523
				break;
1524
			}
1525 1526 1527 1528
		}
		pagevec_release(&pvec);
		cond_resched();

1529
		if (ret || marked)
1530 1531
			break;
	}
1532 1533 1534 1535 1536
	if (!ret && atomic && !marked) {
		f2fs_msg(sbi->sb, KERN_DEBUG,
			"Retry to write fsync mark: ino=%u, idx=%lx",
					ino, last_page->index);
		lock_page(last_page);
1537
		f2fs_wait_on_page_writeback(last_page, NODE, true);
1538 1539 1540 1541
		set_page_dirty(last_page);
		unlock_page(last_page);
		goto retry;
	}
1542
out:
1543
	if (last_idx != ULONG_MAX)
1544
		f2fs_submit_merged_write_cond(sbi, NULL, ino, last_idx, NODE);
1545
	return ret ? -EIO: 0;
1546 1547
}

1548
int sync_node_pages(struct f2fs_sb_info *sbi, struct writeback_control *wbc,
C
Chao Yu 已提交
1549
				bool do_balance, enum iostat_type io_type)
1550 1551 1552 1553
{
	pgoff_t index, end;
	struct pagevec pvec;
	int step = 0;
1554
	int nwritten = 0;
1555
	int ret = 0;
J
Jaegeuk Kim 已提交
1556 1557 1558 1559 1560

	pagevec_init(&pvec, 0);

next_step:
	index = 0;
1561
	end = ULONG_MAX;
J
Jaegeuk Kim 已提交
1562 1563 1564

	while (index <= end) {
		int i, nr_pages;
1565
		nr_pages = pagevec_lookup_tag(&pvec, NODE_MAPPING(sbi), &index,
J
Jaegeuk Kim 已提交
1566 1567 1568 1569 1570 1571 1572
				PAGECACHE_TAG_DIRTY,
				min(end - index, (pgoff_t)PAGEVEC_SIZE-1) + 1);
		if (nr_pages == 0)
			break;

		for (i = 0; i < nr_pages; i++) {
			struct page *page = pvec.pages[i];
1573
			bool submitted = false;
J
Jaegeuk Kim 已提交
1574

C
Chao Yu 已提交
1575 1576
			if (unlikely(f2fs_cp_error(sbi))) {
				pagevec_release(&pvec);
1577 1578
				ret = -EIO;
				goto out;
C
Chao Yu 已提交
1579 1580
			}

J
Jaegeuk Kim 已提交
1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594
			/*
			 * flushing sequence with step:
			 * 0. indirect nodes
			 * 1. dentry dnodes
			 * 2. file dnodes
			 */
			if (step == 0 && IS_DNODE(page))
				continue;
			if (step == 1 && (!IS_DNODE(page) ||
						is_cold_node(page)))
				continue;
			if (step == 2 && (!IS_DNODE(page) ||
						!is_cold_node(page)))
				continue;
1595
lock_node:
1596
			if (!trylock_page(page))
J
Jaegeuk Kim 已提交
1597 1598
				continue;

1599
			if (unlikely(page->mapping != NODE_MAPPING(sbi))) {
J
Jaegeuk Kim 已提交
1600 1601 1602 1603 1604 1605 1606 1607 1608 1609
continue_unlock:
				unlock_page(page);
				continue;
			}

			if (!PageDirty(page)) {
				/* someone wrote it for us */
				goto continue_unlock;
			}

1610
			/* flush inline_data */
1611
			if (is_inline_node(page)) {
1612 1613 1614
				clear_inline_node(page);
				unlock_page(page);
				flush_inline_data(sbi, ino_of_node(page));
1615
				goto lock_node;
1616 1617
			}

1618 1619 1620
			f2fs_wait_on_page_writeback(page, NODE, true);

			BUG_ON(PageWriteback(page));
J
Jaegeuk Kim 已提交
1621 1622 1623
			if (!clear_page_dirty_for_io(page))
				goto continue_unlock;

1624 1625
			set_fsync_mark(page, 0);
			set_dentry_mark(page, 0);
1626

1627
			ret = __write_node_page(page, false, &submitted,
C
Chao Yu 已提交
1628
						wbc, do_balance, io_type);
1629
			if (ret)
1630
				unlock_page(page);
1631
			else if (submitted)
1632
				nwritten++;
J
Jaegeuk Kim 已提交
1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649

			if (--wbc->nr_to_write == 0)
				break;
		}
		pagevec_release(&pvec);
		cond_resched();

		if (wbc->nr_to_write == 0) {
			step = 2;
			break;
		}
	}

	if (step < 2) {
		step++;
		goto next_step;
	}
1650 1651
out:
	if (nwritten)
1652
		f2fs_submit_merged_write(sbi, NODE);
1653
	return ret;
J
Jaegeuk Kim 已提交
1654 1655
}

1656 1657
int wait_on_node_pages_writeback(struct f2fs_sb_info *sbi, nid_t ino)
{
1658
	pgoff_t index = 0, end = ULONG_MAX;
1659
	struct pagevec pvec;
1660
	int ret2, ret = 0;
1661 1662

	pagevec_init(&pvec, 0);
1663 1664 1665 1666 1667 1668 1669 1670

	while (index <= end) {
		int i, nr_pages;
		nr_pages = pagevec_lookup_tag(&pvec, NODE_MAPPING(sbi), &index,
				PAGECACHE_TAG_WRITEBACK,
				min(end - index, (pgoff_t)PAGEVEC_SIZE-1) + 1);
		if (nr_pages == 0)
			break;
1671 1672 1673 1674 1675

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

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

1679
			if (ino && ino_of_node(page) == ino) {
1680
				f2fs_wait_on_page_writeback(page, NODE, true);
1681 1682 1683
				if (TestClearPageError(page))
					ret = -EIO;
			}
1684 1685 1686 1687 1688
		}
		pagevec_release(&pvec);
		cond_resched();
	}

1689
	ret2 = filemap_check_errors(NODE_MAPPING(sbi));
1690 1691 1692 1693 1694
	if (!ret)
		ret = ret2;
	return ret;
}

J
Jaegeuk Kim 已提交
1695 1696 1697
static int f2fs_write_node_pages(struct address_space *mapping,
			    struct writeback_control *wbc)
{
1698
	struct f2fs_sb_info *sbi = F2FS_M_SB(mapping);
1699
	struct blk_plug plug;
1700
	long diff;
J
Jaegeuk Kim 已提交
1701

1702 1703 1704
	if (unlikely(is_sbi_flag_set(sbi, SBI_POR_DOING)))
		goto skip_write;

1705 1706
	/* balancing f2fs's metadata in background */
	f2fs_balance_fs_bg(sbi);
J
Jaegeuk Kim 已提交
1707

1708
	/* collect a number of dirty node pages and write together */
1709
	if (get_pages(sbi, F2FS_DIRTY_NODES) < nr_pages_to_skip(sbi, NODE))
1710
		goto skip_write;
1711

Y
Yunlei He 已提交
1712 1713
	trace_f2fs_writepages(mapping->host, wbc, NODE);

1714
	diff = nr_pages_to_write(sbi, NODE, wbc);
1715
	wbc->sync_mode = WB_SYNC_NONE;
1716
	blk_start_plug(&plug);
C
Chao Yu 已提交
1717
	sync_node_pages(sbi, wbc, true, FS_NODE_IO);
1718
	blk_finish_plug(&plug);
1719
	wbc->nr_to_write = max((long)0, wbc->nr_to_write - diff);
J
Jaegeuk Kim 已提交
1720
	return 0;
1721 1722 1723

skip_write:
	wbc->pages_skipped += get_pages(sbi, F2FS_DIRTY_NODES);
Y
Yunlei He 已提交
1724
	trace_f2fs_writepages(mapping->host, wbc, NODE);
1725
	return 0;
J
Jaegeuk Kim 已提交
1726 1727 1728 1729
}

static int f2fs_set_node_page_dirty(struct page *page)
{
1730 1731
	trace_f2fs_set_page_dirty(page, NODE);

1732 1733
	if (!PageUptodate(page))
		SetPageUptodate(page);
J
Jaegeuk Kim 已提交
1734
	if (!PageDirty(page)) {
1735
		f2fs_set_page_dirty_nobuffers(page);
1736
		inc_page_count(F2FS_P_SB(page), F2FS_DIRTY_NODES);
J
Jaegeuk Kim 已提交
1737
		SetPagePrivate(page);
J
Jaegeuk Kim 已提交
1738
		f2fs_trace_pid(page);
J
Jaegeuk Kim 已提交
1739 1740 1741 1742 1743
		return 1;
	}
	return 0;
}

J
Jaegeuk Kim 已提交
1744
/*
J
Jaegeuk Kim 已提交
1745 1746 1747 1748 1749 1750
 * Structure of the f2fs node operations
 */
const struct address_space_operations f2fs_node_aops = {
	.writepage	= f2fs_write_node_page,
	.writepages	= f2fs_write_node_pages,
	.set_page_dirty	= f2fs_set_node_page_dirty,
1751 1752
	.invalidatepage	= f2fs_invalidate_page,
	.releasepage	= f2fs_release_page,
1753 1754 1755
#ifdef CONFIG_MIGRATION
	.migratepage    = f2fs_migrate_page,
#endif
J
Jaegeuk Kim 已提交
1756 1757
};

1758 1759
static struct free_nid *__lookup_free_nid_list(struct f2fs_nm_info *nm_i,
						nid_t n)
J
Jaegeuk Kim 已提交
1760
{
1761
	return radix_tree_lookup(&nm_i->free_nid_root, n);
J
Jaegeuk Kim 已提交
1762 1763
}

C
Chao Yu 已提交
1764
static int __insert_free_nid(struct f2fs_sb_info *sbi,
F
Fan Li 已提交
1765
			struct free_nid *i, enum nid_state state)
J
Jaegeuk Kim 已提交
1766
{
C
Chao Yu 已提交
1767 1768
	struct f2fs_nm_info *nm_i = NM_I(sbi);

F
Fan Li 已提交
1769 1770 1771
	int err = radix_tree_insert(&nm_i->free_nid_root, i->nid, i);
	if (err)
		return err;
1772

C
Chao Yu 已提交
1773 1774 1775 1776
	f2fs_bug_on(sbi, state != i->state);
	nm_i->nid_cnt[state]++;
	if (state == FREE_NID)
		list_add_tail(&i->list, &nm_i->free_nid_list);
1777
	return 0;
C
Chao Yu 已提交
1778 1779
}

C
Chao Yu 已提交
1780
static void __remove_free_nid(struct f2fs_sb_info *sbi,
F
Fan Li 已提交
1781
			struct free_nid *i, enum nid_state state)
C
Chao Yu 已提交
1782 1783 1784
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);

C
Chao Yu 已提交
1785 1786 1787 1788
	f2fs_bug_on(sbi, state != i->state);
	nm_i->nid_cnt[state]--;
	if (state == FREE_NID)
		list_del(&i->list);
F
Fan Li 已提交
1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811
	radix_tree_delete(&nm_i->free_nid_root, i->nid);
}

static void __move_free_nid(struct f2fs_sb_info *sbi, struct free_nid *i,
			enum nid_state org_state, enum nid_state dst_state)
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);

	f2fs_bug_on(sbi, org_state != i->state);
	i->state = dst_state;
	nm_i->nid_cnt[org_state]--;
	nm_i->nid_cnt[dst_state]++;

	switch (dst_state) {
	case PREALLOC_NID:
		list_del(&i->list);
		break;
	case FREE_NID:
		list_add_tail(&i->list, &nm_i->free_nid_list);
		break;
	default:
		BUG_ON(1);
	}
J
Jaegeuk Kim 已提交
1812 1813
}

C
Chao Yu 已提交
1814 1815
/* return if the nid is recognized as free */
static bool add_free_nid(struct f2fs_sb_info *sbi, nid_t nid, bool build)
J
Jaegeuk Kim 已提交
1816
{
1817
	struct f2fs_nm_info *nm_i = NM_I(sbi);
1818
	struct free_nid *i, *e;
1819
	struct nat_entry *ne;
1820 1821
	int err = -EINVAL;
	bool ret = false;
1822 1823

	/* 0 nid should not be used */
1824
	if (unlikely(nid == 0))
C
Chao Yu 已提交
1825
		return false;
1826

1827
	i = f2fs_kmem_cache_alloc(free_nid_slab, GFP_NOFS);
J
Jaegeuk Kim 已提交
1828
	i->nid = nid;
C
Chao Yu 已提交
1829
	i->state = FREE_NID;
J
Jaegeuk Kim 已提交
1830

1831 1832
	if (radix_tree_preload(GFP_NOFS))
		goto err;
1833

C
Chao Yu 已提交
1834
	spin_lock(&nm_i->nid_list_lock);
1835 1836 1837 1838 1839 1840 1841

	if (build) {
		/*
		 *   Thread A             Thread B
		 *  - f2fs_create
		 *   - f2fs_new_inode
		 *    - alloc_nid
C
Chao Yu 已提交
1842
		 *     - __insert_nid_to_list(PREALLOC_NID)
1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854
		 *                     - f2fs_balance_fs_bg
		 *                      - build_free_nids
		 *                       - __build_free_nids
		 *                        - scan_nat_page
		 *                         - add_free_nid
		 *                          - __lookup_nat_cache
		 *  - f2fs_add_link
		 *   - init_inode_metadata
		 *    - new_inode_page
		 *     - new_node_page
		 *      - set_node_addr
		 *  - alloc_nid_done
C
Chao Yu 已提交
1855 1856
		 *   - __remove_nid_from_list(PREALLOC_NID)
		 *                         - __insert_nid_to_list(FREE_NID)
1857 1858 1859 1860 1861 1862 1863 1864
		 */
		ne = __lookup_nat_cache(nm_i, nid);
		if (ne && (!get_nat_flag(ne, IS_CHECKPOINTED) ||
				nat_get_blkaddr(ne) != NULL_ADDR))
			goto err_out;

		e = __lookup_free_nid_list(nm_i, nid);
		if (e) {
C
Chao Yu 已提交
1865
			if (e->state == FREE_NID)
1866 1867 1868 1869 1870
				ret = true;
			goto err_out;
		}
	}
	ret = true;
F
Fan Li 已提交
1871
	err = __insert_free_nid(sbi, i, FREE_NID);
1872
err_out:
1873 1874
	spin_unlock(&nm_i->nid_list_lock);
	radix_tree_preload_end();
1875 1876
err:
	if (err)
J
Jaegeuk Kim 已提交
1877
		kmem_cache_free(free_nid_slab, i);
1878
	return ret;
J
Jaegeuk Kim 已提交
1879 1880
}

C
Chao Yu 已提交
1881
static void remove_free_nid(struct f2fs_sb_info *sbi, nid_t nid)
J
Jaegeuk Kim 已提交
1882
{
C
Chao Yu 已提交
1883
	struct f2fs_nm_info *nm_i = NM_I(sbi);
J
Jaegeuk Kim 已提交
1884
	struct free_nid *i;
1885 1886
	bool need_free = false;

C
Chao Yu 已提交
1887
	spin_lock(&nm_i->nid_list_lock);
1888
	i = __lookup_free_nid_list(nm_i, nid);
C
Chao Yu 已提交
1889
	if (i && i->state == FREE_NID) {
F
Fan Li 已提交
1890
		__remove_free_nid(sbi, i, FREE_NID);
1891
		need_free = true;
J
Jaegeuk Kim 已提交
1892
	}
C
Chao Yu 已提交
1893
	spin_unlock(&nm_i->nid_list_lock);
1894 1895 1896

	if (need_free)
		kmem_cache_free(free_nid_slab, i);
J
Jaegeuk Kim 已提交
1897 1898
}

J
Jaegeuk Kim 已提交
1899
static void update_free_nid_bitmap(struct f2fs_sb_info *sbi, nid_t nid,
1900
							bool set, bool build)
C
Chao Yu 已提交
1901 1902 1903 1904 1905 1906 1907 1908 1909
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	unsigned int nat_ofs = NAT_BLOCK_OFFSET(nid);
	unsigned int nid_ofs = nid - START_NID(nid);

	if (!test_bit_le(nat_ofs, nm_i->nat_block_bitmap))
		return;

	if (set)
J
Jaegeuk Kim 已提交
1910
		__set_bit_le(nid_ofs, nm_i->free_nid_bitmap[nat_ofs]);
C
Chao Yu 已提交
1911
	else
J
Jaegeuk Kim 已提交
1912
		__clear_bit_le(nid_ofs, nm_i->free_nid_bitmap[nat_ofs]);
1913 1914 1915 1916 1917

	if (set)
		nm_i->free_nid_count[nat_ofs]++;
	else if (!build)
		nm_i->free_nid_count[nat_ofs]--;
C
Chao Yu 已提交
1918 1919
}

1920
static void scan_nat_page(struct f2fs_sb_info *sbi,
J
Jaegeuk Kim 已提交
1921 1922
			struct page *nat_page, nid_t start_nid)
{
1923
	struct f2fs_nm_info *nm_i = NM_I(sbi);
J
Jaegeuk Kim 已提交
1924 1925
	struct f2fs_nat_block *nat_blk = page_address(nat_page);
	block_t blk_addr;
C
Chao Yu 已提交
1926
	unsigned int nat_ofs = NAT_BLOCK_OFFSET(start_nid);
J
Jaegeuk Kim 已提交
1927 1928
	int i;

1929 1930 1931
	if (test_bit_le(nat_ofs, nm_i->nat_block_bitmap))
		return;

J
Jaegeuk Kim 已提交
1932
	__set_bit_le(nat_ofs, nm_i->nat_block_bitmap);
C
Chao Yu 已提交
1933

J
Jaegeuk Kim 已提交
1934 1935 1936
	i = start_nid % NAT_ENTRY_PER_BLOCK;

	for (; i < NAT_ENTRY_PER_BLOCK; i++, start_nid++) {
C
Chao Yu 已提交
1937
		bool freed = false;
H
Haicheng Li 已提交
1938

1939
		if (unlikely(start_nid >= nm_i->max_nid))
J
Jaegeuk Kim 已提交
1940
			break;
H
Haicheng Li 已提交
1941 1942

		blk_addr = le32_to_cpu(nat_blk->entries[i].block_addr);
1943
		f2fs_bug_on(sbi, blk_addr == NEW_ADDR);
1944
		if (blk_addr == NULL_ADDR)
C
Chao Yu 已提交
1945
			freed = add_free_nid(sbi, start_nid, true);
1946 1947 1948
		spin_lock(&NM_I(sbi)->nid_list_lock);
		update_free_nid_bitmap(sbi, start_nid, freed, true);
		spin_unlock(&NM_I(sbi)->nid_list_lock);
C
Chao Yu 已提交
1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963
	}
}

static void scan_free_nid_bits(struct f2fs_sb_info *sbi)
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct curseg_info *curseg = CURSEG_I(sbi, CURSEG_HOT_DATA);
	struct f2fs_journal *journal = curseg->journal;
	unsigned int i, idx;

	down_read(&nm_i->nat_tree_lock);

	for (i = 0; i < nm_i->nat_blocks; i++) {
		if (!test_bit_le(i, nm_i->nat_block_bitmap))
			continue;
1964 1965
		if (!nm_i->free_nid_count[i])
			continue;
C
Chao Yu 已提交
1966 1967 1968 1969 1970 1971 1972 1973 1974
		for (idx = 0; idx < NAT_ENTRY_PER_BLOCK; idx++) {
			nid_t nid;

			if (!test_bit_le(idx, nm_i->free_nid_bitmap[i]))
				continue;

			nid = i * NAT_ENTRY_PER_BLOCK + idx;
			add_free_nid(sbi, nid, true);

C
Chao Yu 已提交
1975
			if (nm_i->nid_cnt[FREE_NID] >= MAX_FREE_NIDS)
C
Chao Yu 已提交
1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990
				goto out;
		}
	}
out:
	down_read(&curseg->journal_rwsem);
	for (i = 0; i < nats_in_cursum(journal); i++) {
		block_t addr;
		nid_t nid;

		addr = le32_to_cpu(nat_in_journal(journal, i).block_addr);
		nid = le32_to_cpu(nid_in_journal(journal, i));
		if (addr == NULL_ADDR)
			add_free_nid(sbi, nid, true);
		else
			remove_free_nid(sbi, nid);
J
Jaegeuk Kim 已提交
1991
	}
C
Chao Yu 已提交
1992 1993
	up_read(&curseg->journal_rwsem);
	up_read(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
1994 1995
}

1996
static void __build_free_nids(struct f2fs_sb_info *sbi, bool sync, bool mount)
J
Jaegeuk Kim 已提交
1997 1998 1999
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct curseg_info *curseg = CURSEG_I(sbi, CURSEG_HOT_DATA);
2000
	struct f2fs_journal *journal = curseg->journal;
2001
	int i = 0;
2002
	nid_t nid = nm_i->next_scan_nid;
J
Jaegeuk Kim 已提交
2003

2004 2005 2006
	if (unlikely(nid >= nm_i->max_nid))
		nid = 0;

2007
	/* Enough entries */
C
Chao Yu 已提交
2008
	if (nm_i->nid_cnt[FREE_NID] >= NAT_ENTRY_PER_BLOCK)
2009
		return;
J
Jaegeuk Kim 已提交
2010

2011
	if (!sync && !available_free_memory(sbi, FREE_NIDS))
2012
		return;
J
Jaegeuk Kim 已提交
2013

C
Chao Yu 已提交
2014 2015 2016 2017
	if (!mount) {
		/* try to find free nids in free_nid_bitmap */
		scan_free_nid_bits(sbi);

C
Chao Yu 已提交
2018
		if (nm_i->nid_cnt[FREE_NID])
C
Chao Yu 已提交
2019
			return;
2020 2021
	}

2022
	/* readahead nat pages to be scanned */
2023 2024
	ra_meta_pages(sbi, NAT_BLOCK_OFFSET(nid), FREE_NID_PAGES,
							META_NAT, true);
J
Jaegeuk Kim 已提交
2025

2026
	down_read(&nm_i->nat_tree_lock);
2027

J
Jaegeuk Kim 已提交
2028 2029 2030
	while (1) {
		struct page *page = get_current_nat_page(sbi, nid);

2031
		scan_nat_page(sbi, page, nid);
J
Jaegeuk Kim 已提交
2032 2033 2034
		f2fs_put_page(page, 1);

		nid += (NAT_ENTRY_PER_BLOCK - (nid % NAT_ENTRY_PER_BLOCK));
2035
		if (unlikely(nid >= nm_i->max_nid))
J
Jaegeuk Kim 已提交
2036
			nid = 0;
2037

2038
		if (++i >= FREE_NID_PAGES)
J
Jaegeuk Kim 已提交
2039 2040 2041
			break;
	}

2042 2043
	/* go to the next free nat pages to find free nids abundantly */
	nm_i->next_scan_nid = nid;
J
Jaegeuk Kim 已提交
2044 2045

	/* find free nids from current sum_pages */
2046
	down_read(&curseg->journal_rwsem);
2047 2048 2049 2050 2051
	for (i = 0; i < nats_in_cursum(journal); i++) {
		block_t addr;

		addr = le32_to_cpu(nat_in_journal(journal, i).block_addr);
		nid = le32_to_cpu(nid_in_journal(journal, i));
J
Jaegeuk Kim 已提交
2052
		if (addr == NULL_ADDR)
2053
			add_free_nid(sbi, nid, true);
J
Jaegeuk Kim 已提交
2054
		else
C
Chao Yu 已提交
2055
			remove_free_nid(sbi, nid);
J
Jaegeuk Kim 已提交
2056
	}
2057
	up_read(&curseg->journal_rwsem);
2058
	up_read(&nm_i->nat_tree_lock);
C
Chao Yu 已提交
2059 2060

	ra_meta_pages(sbi, NAT_BLOCK_OFFSET(nm_i->next_scan_nid),
C
Chao Yu 已提交
2061
					nm_i->ra_nid_pages, META_NAT, false);
J
Jaegeuk Kim 已提交
2062 2063
}

2064
void build_free_nids(struct f2fs_sb_info *sbi, bool sync, bool mount)
2065 2066
{
	mutex_lock(&NM_I(sbi)->build_lock);
2067
	__build_free_nids(sbi, sync, mount);
2068 2069 2070
	mutex_unlock(&NM_I(sbi)->build_lock);
}

J
Jaegeuk Kim 已提交
2071 2072 2073 2074 2075 2076 2077 2078 2079 2080
/*
 * If this function returns success, caller can obtain a new nid
 * from second parameter of this function.
 * The returned nid could be used ino as well as nid when inode is created.
 */
bool alloc_nid(struct f2fs_sb_info *sbi, nid_t *nid)
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct free_nid *i = NULL;
retry:
J
Jaegeuk Kim 已提交
2081
#ifdef CONFIG_F2FS_FAULT_INJECTION
2082 2083
	if (time_to_inject(sbi, FAULT_ALLOC_NID)) {
		f2fs_show_injection_info(FAULT_ALLOC_NID);
J
Jaegeuk Kim 已提交
2084
		return false;
2085
	}
J
Jaegeuk Kim 已提交
2086
#endif
C
Chao Yu 已提交
2087
	spin_lock(&nm_i->nid_list_lock);
J
Jaegeuk Kim 已提交
2088

2089 2090 2091 2092
	if (unlikely(nm_i->available_nids == 0)) {
		spin_unlock(&nm_i->nid_list_lock);
		return false;
	}
J
Jaegeuk Kim 已提交
2093

2094
	/* We should not use stale free nids created by build_free_nids */
C
Chao Yu 已提交
2095 2096 2097
	if (nm_i->nid_cnt[FREE_NID] && !on_build_free_nids(nm_i)) {
		f2fs_bug_on(sbi, list_empty(&nm_i->free_nid_list));
		i = list_first_entry(&nm_i->free_nid_list,
C
Chao Yu 已提交
2098
					struct free_nid, list);
2099
		*nid = i->nid;
C
Chao Yu 已提交
2100

F
Fan Li 已提交
2101
		__move_free_nid(sbi, i, FREE_NID, PREALLOC_NID);
2102
		nm_i->available_nids--;
C
Chao Yu 已提交
2103

2104
		update_free_nid_bitmap(sbi, *nid, false, false);
C
Chao Yu 已提交
2105

C
Chao Yu 已提交
2106
		spin_unlock(&nm_i->nid_list_lock);
2107 2108
		return true;
	}
C
Chao Yu 已提交
2109
	spin_unlock(&nm_i->nid_list_lock);
2110 2111

	/* Let's scan nat pages and its caches to get free nids */
2112
	build_free_nids(sbi, true, false);
2113
	goto retry;
J
Jaegeuk Kim 已提交
2114 2115
}

J
Jaegeuk Kim 已提交
2116
/*
J
Jaegeuk Kim 已提交
2117 2118 2119 2120 2121 2122 2123
 * alloc_nid() should be called prior to this function.
 */
void alloc_nid_done(struct f2fs_sb_info *sbi, nid_t nid)
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct free_nid *i;

C
Chao Yu 已提交
2124
	spin_lock(&nm_i->nid_list_lock);
2125
	i = __lookup_free_nid_list(nm_i, nid);
C
Chao Yu 已提交
2126
	f2fs_bug_on(sbi, !i);
F
Fan Li 已提交
2127
	__remove_free_nid(sbi, i, PREALLOC_NID);
C
Chao Yu 已提交
2128
	spin_unlock(&nm_i->nid_list_lock);
2129 2130

	kmem_cache_free(free_nid_slab, i);
J
Jaegeuk Kim 已提交
2131 2132
}

J
Jaegeuk Kim 已提交
2133
/*
J
Jaegeuk Kim 已提交
2134 2135 2136 2137
 * alloc_nid() should be called prior to this function.
 */
void alloc_nid_failed(struct f2fs_sb_info *sbi, nid_t nid)
{
2138 2139
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct free_nid *i;
2140
	bool need_free = false;
2141

J
Jaegeuk Kim 已提交
2142 2143 2144
	if (!nid)
		return;

C
Chao Yu 已提交
2145
	spin_lock(&nm_i->nid_list_lock);
2146
	i = __lookup_free_nid_list(nm_i, nid);
C
Chao Yu 已提交
2147 2148
	f2fs_bug_on(sbi, !i);

2149
	if (!available_free_memory(sbi, FREE_NIDS)) {
F
Fan Li 已提交
2150
		__remove_free_nid(sbi, i, PREALLOC_NID);
2151
		need_free = true;
2152
	} else {
F
Fan Li 已提交
2153
		__move_free_nid(sbi, i, PREALLOC_NID, FREE_NID);
2154
	}
2155 2156 2157

	nm_i->available_nids++;

2158
	update_free_nid_bitmap(sbi, nid, true, false);
C
Chao Yu 已提交
2159

C
Chao Yu 已提交
2160
	spin_unlock(&nm_i->nid_list_lock);
2161 2162 2163

	if (need_free)
		kmem_cache_free(free_nid_slab, i);
J
Jaegeuk Kim 已提交
2164 2165
}

C
Chao Yu 已提交
2166 2167 2168 2169 2170 2171
int try_to_free_nids(struct f2fs_sb_info *sbi, int nr_shrink)
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct free_nid *i, *next;
	int nr = nr_shrink;

C
Chao Yu 已提交
2172
	if (nm_i->nid_cnt[FREE_NID] <= MAX_FREE_NIDS)
2173 2174
		return 0;

C
Chao Yu 已提交
2175 2176 2177
	if (!mutex_trylock(&nm_i->build_lock))
		return 0;

C
Chao Yu 已提交
2178
	spin_lock(&nm_i->nid_list_lock);
C
Chao Yu 已提交
2179
	list_for_each_entry_safe(i, next, &nm_i->free_nid_list, list) {
C
Chao Yu 已提交
2180
		if (nr_shrink <= 0 ||
C
Chao Yu 已提交
2181
				nm_i->nid_cnt[FREE_NID] <= MAX_FREE_NIDS)
C
Chao Yu 已提交
2182
			break;
C
Chao Yu 已提交
2183

F
Fan Li 已提交
2184
		__remove_free_nid(sbi, i, FREE_NID);
C
Chao Yu 已提交
2185 2186 2187
		kmem_cache_free(free_nid_slab, i);
		nr_shrink--;
	}
C
Chao Yu 已提交
2188
	spin_unlock(&nm_i->nid_list_lock);
C
Chao Yu 已提交
2189 2190 2191 2192 2193
	mutex_unlock(&nm_i->build_lock);

	return nr - nr_shrink;
}

2194
void recover_inline_xattr(struct inode *inode, struct page *page)
2195 2196 2197 2198 2199 2200
{
	void *src_addr, *dst_addr;
	size_t inline_size;
	struct page *ipage;
	struct f2fs_inode *ri;

2201
	ipage = get_node_page(F2FS_I_SB(inode), inode->i_ino);
2202
	f2fs_bug_on(F2FS_I_SB(inode), IS_ERR(ipage));
2203

2204 2205
	ri = F2FS_INODE(page);
	if (!(ri->i_inline & F2FS_INLINE_XATTR)) {
2206
		clear_inode_flag(inode, FI_INLINE_XATTR);
2207 2208 2209
		goto update_inode;
	}

C
Chao Yu 已提交
2210 2211
	dst_addr = inline_xattr_addr(inode, ipage);
	src_addr = inline_xattr_addr(inode, page);
2212 2213
	inline_size = inline_xattr_size(inode);

2214
	f2fs_wait_on_page_writeback(ipage, NODE, true);
2215
	memcpy(dst_addr, src_addr, inline_size);
2216
update_inode:
2217 2218 2219 2220
	update_inode(inode, ipage);
	f2fs_put_page(ipage, 1);
}

2221
int recover_xattr_data(struct inode *inode, struct page *page, block_t blkaddr)
2222
{
2223
	struct f2fs_sb_info *sbi = F2FS_I_SB(inode);
2224
	nid_t prev_xnid = F2FS_I(inode)->i_xattr_nid;
2225 2226
	nid_t new_xnid;
	struct dnode_of_data dn;
2227
	struct node_info ni;
2228
	struct page *xpage;
2229 2230 2231 2232

	if (!prev_xnid)
		goto recover_xnid;

2233
	/* 1: invalidate the previous xattr nid */
2234
	get_node_info(sbi, prev_xnid, &ni);
2235
	f2fs_bug_on(sbi, ni.blk_addr == NULL_ADDR);
2236
	invalidate_blocks(sbi, ni.blk_addr);
2237
	dec_valid_node_count(sbi, inode, false);
2238
	set_node_addr(sbi, &ni, NULL_ADDR, false);
2239 2240

recover_xnid:
2241
	/* 2: update xattr nid in inode */
2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252
	if (!alloc_nid(sbi, &new_xnid))
		return -ENOSPC;

	set_new_dnode(&dn, inode, NULL, NULL, new_xnid);
	xpage = new_node_page(&dn, XATTR_NODE_OFFSET);
	if (IS_ERR(xpage)) {
		alloc_nid_failed(sbi, new_xnid);
		return PTR_ERR(xpage);
	}

	alloc_nid_done(sbi, new_xnid);
2253 2254 2255
	update_inode_page(inode);

	/* 3: update and set xattr node page dirty */
2256
	memcpy(F2FS_NODE(xpage), F2FS_NODE(page), VALID_XATTR_BLOCK_SIZE);
2257

2258 2259
	set_page_dirty(xpage);
	f2fs_put_page(xpage, 1);
2260

2261
	return 0;
2262 2263
}

J
Jaegeuk Kim 已提交
2264 2265
int recover_inode_page(struct f2fs_sb_info *sbi, struct page *page)
{
2266
	struct f2fs_inode *src, *dst;
J
Jaegeuk Kim 已提交
2267 2268 2269 2270
	nid_t ino = ino_of_node(page);
	struct node_info old_ni, new_ni;
	struct page *ipage;

2271 2272 2273 2274
	get_node_info(sbi, ino, &old_ni);

	if (unlikely(old_ni.blk_addr != NULL_ADDR))
		return -EINVAL;
2275
retry:
2276
	ipage = f2fs_grab_cache_page(NODE_MAPPING(sbi), ino, false);
2277 2278 2279 2280
	if (!ipage) {
		congestion_wait(BLK_RW_ASYNC, HZ/50);
		goto retry;
	}
J
Jaegeuk Kim 已提交
2281

A
arter97 已提交
2282
	/* Should not use this inode from free nid list */
C
Chao Yu 已提交
2283
	remove_free_nid(sbi, ino);
J
Jaegeuk Kim 已提交
2284

2285 2286
	if (!PageUptodate(ipage))
		SetPageUptodate(ipage);
J
Jaegeuk Kim 已提交
2287 2288
	fill_node_footer(ipage, ino, ino, 0, true);

2289 2290
	src = F2FS_INODE(page);
	dst = F2FS_INODE(ipage);
J
Jaegeuk Kim 已提交
2291

2292 2293 2294 2295 2296
	memcpy(dst, src, (unsigned long)&src->i_ext - (unsigned long)src);
	dst->i_size = 0;
	dst->i_blocks = cpu_to_le64(1);
	dst->i_links = cpu_to_le32(1);
	dst->i_xattr_nid = 0;
2297
	dst->i_inline = src->i_inline & (F2FS_INLINE_XATTR | F2FS_EXTRA_ATTR);
C
Chao Yu 已提交
2298
	if (dst->i_inline & F2FS_EXTRA_ATTR) {
2299
		dst->i_extra_isize = src->i_extra_isize;
C
Chao Yu 已提交
2300 2301 2302 2303 2304 2305

		if (f2fs_sb_has_flexible_inline_xattr(sbi->sb) &&
			F2FS_FITS_IN_INODE(src, le16_to_cpu(src->i_extra_isize),
							i_inline_xattr_size))
			dst->i_inline_xattr_size = src->i_inline_xattr_size;

C
Chao Yu 已提交
2306 2307 2308 2309 2310
		if (f2fs_sb_has_project_quota(sbi->sb) &&
			F2FS_FITS_IN_INODE(src, le16_to_cpu(src->i_extra_isize),
								i_projid))
			dst->i_projid = src->i_projid;
	}
J
Jaegeuk Kim 已提交
2311 2312 2313 2314

	new_ni = old_ni;
	new_ni.ino = ino;

C
Chao Yu 已提交
2315
	if (unlikely(inc_valid_node_count(sbi, NULL, true)))
2316
		WARN_ON(1);
2317
	set_node_addr(sbi, &new_ni, NEW_ADDR, false);
J
Jaegeuk Kim 已提交
2318
	inc_valid_inode_count(sbi);
2319
	set_page_dirty(ipage);
J
Jaegeuk Kim 已提交
2320 2321 2322 2323 2324 2325 2326 2327 2328 2329
	f2fs_put_page(ipage, 1);
	return 0;
}

int restore_node_summary(struct f2fs_sb_info *sbi,
			unsigned int segno, struct f2fs_summary_block *sum)
{
	struct f2fs_node *rn;
	struct f2fs_summary *sum_entry;
	block_t addr;
2330
	int i, idx, last_offset, nrpages;
J
Jaegeuk Kim 已提交
2331 2332 2333 2334 2335 2336

	/* scan the node segment */
	last_offset = sbi->blocks_per_seg;
	addr = START_BLOCK(sbi, segno);
	sum_entry = &sum->entries[0];

2337
	for (i = 0; i < last_offset; i += nrpages, addr += nrpages) {
2338
		nrpages = min(last_offset - i, BIO_MAX_PAGES);
2339

A
arter97 已提交
2340
		/* readahead node pages */
2341
		ra_meta_pages(sbi, addr, nrpages, META_POR, true);
J
Jaegeuk Kim 已提交
2342

2343
		for (idx = addr; idx < addr + nrpages; idx++) {
2344
			struct page *page = get_tmp_page(sbi, idx);
2345

2346 2347 2348 2349 2350 2351
			rn = F2FS_NODE(page);
			sum_entry->nid = rn->footer.nid;
			sum_entry->version = 0;
			sum_entry->ofs_in_node = 0;
			sum_entry++;
			f2fs_put_page(page, 1);
2352
		}
2353

2354
		invalidate_mapping_pages(META_MAPPING(sbi), addr,
2355
							addr + nrpages);
J
Jaegeuk Kim 已提交
2356
	}
2357
	return 0;
J
Jaegeuk Kim 已提交
2358 2359
}

2360
static void remove_nats_in_journal(struct f2fs_sb_info *sbi)
J
Jaegeuk Kim 已提交
2361 2362 2363
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct curseg_info *curseg = CURSEG_I(sbi, CURSEG_HOT_DATA);
2364
	struct f2fs_journal *journal = curseg->journal;
J
Jaegeuk Kim 已提交
2365 2366
	int i;

2367
	down_write(&curseg->journal_rwsem);
2368
	for (i = 0; i < nats_in_cursum(journal); i++) {
J
Jaegeuk Kim 已提交
2369 2370
		struct nat_entry *ne;
		struct f2fs_nat_entry raw_ne;
2371
		nid_t nid = le32_to_cpu(nid_in_journal(journal, i));
J
Jaegeuk Kim 已提交
2372

2373
		raw_ne = nat_in_journal(journal, i);
2374

J
Jaegeuk Kim 已提交
2375 2376
		ne = __lookup_nat_cache(nm_i, nid);
		if (!ne) {
2377
			ne = grab_nat_entry(nm_i, nid, true);
2378
			node_info_from_raw_nat(&ne->ni, &raw_ne);
J
Jaegeuk Kim 已提交
2379
		}
2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392

		/*
		 * if a free nat in journal has not been used after last
		 * checkpoint, we should remove it from available nids,
		 * since later we will add it again.
		 */
		if (!get_nat_flag(ne, IS_DIRTY) &&
				le32_to_cpu(raw_ne.block_addr) == NULL_ADDR) {
			spin_lock(&nm_i->nid_list_lock);
			nm_i->available_nids--;
			spin_unlock(&nm_i->nid_list_lock);
		}

J
Jaegeuk Kim 已提交
2393 2394
		__set_nat_cache_dirty(nm_i, ne);
	}
2395
	update_nats_in_cursum(journal, -i);
2396
	up_write(&curseg->journal_rwsem);
J
Jaegeuk Kim 已提交
2397 2398
}

2399 2400
static void __adjust_nat_entry_set(struct nat_entry_set *nes,
						struct list_head *head, int max)
J
Jaegeuk Kim 已提交
2401
{
2402
	struct nat_entry_set *cur;
J
Jaegeuk Kim 已提交
2403

2404 2405
	if (nes->entry_cnt >= max)
		goto add_out;
J
Jaegeuk Kim 已提交
2406

2407 2408 2409 2410 2411
	list_for_each_entry(cur, head, set_list) {
		if (cur->entry_cnt >= nes->entry_cnt) {
			list_add(&nes->set_list, cur->set_list.prev);
			return;
		}
2412
	}
2413 2414 2415
add_out:
	list_add_tail(&nes->set_list, head);
}
J
Jaegeuk Kim 已提交
2416

J
Jaegeuk Kim 已提交
2417
static void __update_nat_bits(struct f2fs_sb_info *sbi, nid_t start_nid,
2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435
						struct page *page)
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	unsigned int nat_index = start_nid / NAT_ENTRY_PER_BLOCK;
	struct f2fs_nat_block *nat_blk = page_address(page);
	int valid = 0;
	int i;

	if (!enabled_nat_bits(sbi, NULL))
		return;

	for (i = 0; i < NAT_ENTRY_PER_BLOCK; i++) {
		if (start_nid == 0 && i == 0)
			valid++;
		if (nat_blk->entries[i].block_addr)
			valid++;
	}
	if (valid == 0) {
J
Jaegeuk Kim 已提交
2436 2437
		__set_bit_le(nat_index, nm_i->empty_nat_bits);
		__clear_bit_le(nat_index, nm_i->full_nat_bits);
2438 2439 2440
		return;
	}

J
Jaegeuk Kim 已提交
2441
	__clear_bit_le(nat_index, nm_i->empty_nat_bits);
2442
	if (valid == NAT_ENTRY_PER_BLOCK)
J
Jaegeuk Kim 已提交
2443
		__set_bit_le(nat_index, nm_i->full_nat_bits);
2444
	else
J
Jaegeuk Kim 已提交
2445
		__clear_bit_le(nat_index, nm_i->full_nat_bits);
2446 2447
}

2448
static void __flush_nat_entry_set(struct f2fs_sb_info *sbi,
2449
		struct nat_entry_set *set, struct cp_control *cpc)
2450 2451
{
	struct curseg_info *curseg = CURSEG_I(sbi, CURSEG_HOT_DATA);
2452
	struct f2fs_journal *journal = curseg->journal;
2453 2454 2455 2456 2457
	nid_t start_nid = set->set * NAT_ENTRY_PER_BLOCK;
	bool to_journal = true;
	struct f2fs_nat_block *nat_blk;
	struct nat_entry *ne, *cur;
	struct page *page = NULL;
J
Jaegeuk Kim 已提交
2458

2459 2460 2461 2462 2463
	/*
	 * there are two steps to flush nat entries:
	 * #1, flush nat entries to journal in current hot data summary block.
	 * #2, flush nat entries to nat page.
	 */
2464 2465
	if (enabled_nat_bits(sbi, cpc) ||
		!__has_cursum_space(journal, set->entry_cnt, NAT_JOURNAL))
2466 2467 2468
		to_journal = false;

	if (to_journal) {
2469
		down_write(&curseg->journal_rwsem);
2470 2471 2472 2473 2474
	} else {
		page = get_next_nat_page(sbi, start_nid);
		nat_blk = page_address(page);
		f2fs_bug_on(sbi, !nat_blk);
	}
2475

2476 2477 2478 2479 2480 2481
	/* flush dirty nats in nat entry set */
	list_for_each_entry_safe(ne, cur, &set->entry_list, list) {
		struct f2fs_nat_entry *raw_ne;
		nid_t nid = nat_get_nid(ne);
		int offset;

2482
		f2fs_bug_on(sbi, nat_get_blkaddr(ne) == NEW_ADDR);
2483 2484

		if (to_journal) {
2485
			offset = lookup_journal_in_cursum(journal,
2486 2487
							NAT_JOURNAL, nid, 1);
			f2fs_bug_on(sbi, offset < 0);
2488 2489
			raw_ne = &nat_in_journal(journal, offset);
			nid_in_journal(journal, offset) = cpu_to_le32(nid);
2490
		} else {
2491
			raw_ne = &nat_blk->entries[nid - start_nid];
J
Jaegeuk Kim 已提交
2492
		}
2493 2494
		raw_nat_from_node_info(raw_ne, &ne->ni);
		nat_reset_flag(ne);
2495
		__clear_nat_cache_dirty(NM_I(sbi), set, ne);
2496
		if (nat_get_blkaddr(ne) == NULL_ADDR) {
2497
			add_free_nid(sbi, nid, false);
2498 2499
			spin_lock(&NM_I(sbi)->nid_list_lock);
			NM_I(sbi)->available_nids++;
2500
			update_free_nid_bitmap(sbi, nid, true, false);
C
Chao Yu 已提交
2501 2502 2503
			spin_unlock(&NM_I(sbi)->nid_list_lock);
		} else {
			spin_lock(&NM_I(sbi)->nid_list_lock);
2504
			update_free_nid_bitmap(sbi, nid, false, false);
2505 2506
			spin_unlock(&NM_I(sbi)->nid_list_lock);
		}
2507
	}
J
Jaegeuk Kim 已提交
2508

2509
	if (to_journal) {
2510
		up_write(&curseg->journal_rwsem);
2511 2512
	} else {
		__update_nat_bits(sbi, start_nid, page);
2513
		f2fs_put_page(page, 1);
2514
	}
2515

2516 2517 2518 2519 2520
	/* Allow dirty nats by node block allocation in write_begin */
	if (!set->entry_cnt) {
		radix_tree_delete(&NM_I(sbi)->nat_set_root, set->set);
		kmem_cache_free(nat_entry_set_slab, set);
	}
2521
}
2522

2523 2524 2525
/*
 * This function is called during the checkpointing process.
 */
2526
void flush_nat_entries(struct f2fs_sb_info *sbi, struct cp_control *cpc)
2527 2528 2529
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct curseg_info *curseg = CURSEG_I(sbi, CURSEG_HOT_DATA);
2530
	struct f2fs_journal *journal = curseg->journal;
2531
	struct nat_entry_set *setvec[SETVEC_SIZE];
2532 2533 2534 2535 2536
	struct nat_entry_set *set, *tmp;
	unsigned int found;
	nid_t set_idx = 0;
	LIST_HEAD(sets);

2537 2538
	if (!nm_i->dirty_nat_cnt)
		return;
2539

2540
	down_write(&nm_i->nat_tree_lock);
2541

2542 2543 2544 2545 2546
	/*
	 * if there are no enough space in journal to store dirty nat
	 * entries, remove all entries from journal and merge them
	 * into nat entry set.
	 */
2547
	if (enabled_nat_bits(sbi, cpc) ||
2548
		!__has_cursum_space(journal, nm_i->dirty_nat_cnt, NAT_JOURNAL))
2549 2550 2551
		remove_nats_in_journal(sbi);

	while ((found = __gang_lookup_nat_set(nm_i,
2552
					set_idx, SETVEC_SIZE, setvec))) {
2553 2554 2555 2556
		unsigned idx;
		set_idx = setvec[found - 1]->set + 1;
		for (idx = 0; idx < found; idx++)
			__adjust_nat_entry_set(setvec[idx], &sets,
2557
						MAX_NAT_JENTRIES(journal));
J
Jaegeuk Kim 已提交
2558
	}
2559

2560 2561
	/* flush dirty nats in nat entry set */
	list_for_each_entry_safe(set, tmp, &sets, set_list)
2562
		__flush_nat_entry_set(sbi, set, cpc);
2563

2564
	up_write(&nm_i->nat_tree_lock);
2565
	/* Allow dirty nats by node block allocation in write_begin */
J
Jaegeuk Kim 已提交
2566 2567
}

2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596
static int __get_nat_bitmaps(struct f2fs_sb_info *sbi)
{
	struct f2fs_checkpoint *ckpt = F2FS_CKPT(sbi);
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	unsigned int nat_bits_bytes = nm_i->nat_blocks / BITS_PER_BYTE;
	unsigned int i;
	__u64 cp_ver = cur_cp_version(ckpt);
	block_t nat_bits_addr;

	if (!enabled_nat_bits(sbi, NULL))
		return 0;

	nm_i->nat_bits_blocks = F2FS_BYTES_TO_BLK((nat_bits_bytes << 1) + 8 +
						F2FS_BLKSIZE - 1);
	nm_i->nat_bits = kzalloc(nm_i->nat_bits_blocks << F2FS_BLKSIZE_BITS,
						GFP_KERNEL);
	if (!nm_i->nat_bits)
		return -ENOMEM;

	nat_bits_addr = __start_cp_addr(sbi) + sbi->blocks_per_seg -
						nm_i->nat_bits_blocks;
	for (i = 0; i < nm_i->nat_bits_blocks; i++) {
		struct page *page = get_meta_page(sbi, nat_bits_addr++);

		memcpy(nm_i->nat_bits + (i << F2FS_BLKSIZE_BITS),
					page_address(page), F2FS_BLKSIZE);
		f2fs_put_page(page, 1);
	}

2597
	cp_ver |= (cur_cp_crc(ckpt) << 32);
2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609
	if (cpu_to_le64(cp_ver) != *(__le64 *)nm_i->nat_bits) {
		disable_nat_bits(sbi, true);
		return 0;
	}

	nm_i->full_nat_bits = nm_i->nat_bits + 8;
	nm_i->empty_nat_bits = nm_i->full_nat_bits + nat_bits_bytes;

	f2fs_msg(sbi->sb, KERN_NOTICE, "Found nat_bits in checkpoint");
	return 0;
}

2610
static inline void load_free_nid_bitmap(struct f2fs_sb_info *sbi)
2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	unsigned int i = 0;
	nid_t nid, last_nid;

	if (!enabled_nat_bits(sbi, NULL))
		return;

	for (i = 0; i < nm_i->nat_blocks; i++) {
		i = find_next_bit_le(nm_i->empty_nat_bits, nm_i->nat_blocks, i);
		if (i >= nm_i->nat_blocks)
			break;

		__set_bit_le(i, nm_i->nat_block_bitmap);

		nid = i * NAT_ENTRY_PER_BLOCK;
		last_nid = (i + 1) * NAT_ENTRY_PER_BLOCK;

2629
		spin_lock(&NM_I(sbi)->nid_list_lock);
2630
		for (; nid < last_nid; nid++)
2631 2632
			update_free_nid_bitmap(sbi, nid, true, true);
		spin_unlock(&NM_I(sbi)->nid_list_lock);
2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643
	}

	for (i = 0; i < nm_i->nat_blocks; i++) {
		i = find_next_bit_le(nm_i->full_nat_bits, nm_i->nat_blocks, i);
		if (i >= nm_i->nat_blocks)
			break;

		__set_bit_le(i, nm_i->nat_block_bitmap);
	}
}

J
Jaegeuk Kim 已提交
2644 2645 2646 2647 2648
static int init_node_manager(struct f2fs_sb_info *sbi)
{
	struct f2fs_super_block *sb_raw = F2FS_RAW_SUPER(sbi);
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	unsigned char *version_bitmap;
2649 2650
	unsigned int nat_segs;
	int err;
J
Jaegeuk Kim 已提交
2651 2652 2653 2654 2655

	nm_i->nat_blkaddr = le32_to_cpu(sb_raw->nat_blkaddr);

	/* segment_count_nat includes pair segment so divide to 2. */
	nat_segs = le32_to_cpu(sb_raw->segment_count_nat) >> 1;
2656 2657
	nm_i->nat_blocks = nat_segs << le32_to_cpu(sb_raw->log_blocks_per_seg);
	nm_i->max_nid = NAT_ENTRY_PER_BLOCK * nm_i->nat_blocks;
2658

2659
	/* not used nids: 0, node, meta, (and root counted as valid node) */
2660 2661
	nm_i->available_nids = nm_i->max_nid - sbi->total_valid_node_count -
							F2FS_RESERVED_NODE_NUM;
C
Chao Yu 已提交
2662 2663
	nm_i->nid_cnt[FREE_NID] = 0;
	nm_i->nid_cnt[PREALLOC_NID] = 0;
J
Jaegeuk Kim 已提交
2664
	nm_i->nat_cnt = 0;
2665
	nm_i->ram_thresh = DEF_RAM_THRESHOLD;
C
Chao Yu 已提交
2666
	nm_i->ra_nid_pages = DEF_RA_NID_PAGES;
C
Chao Yu 已提交
2667
	nm_i->dirty_nats_ratio = DEF_DIRTY_NAT_RATIO_THRESHOLD;
J
Jaegeuk Kim 已提交
2668

2669
	INIT_RADIX_TREE(&nm_i->free_nid_root, GFP_ATOMIC);
C
Chao Yu 已提交
2670
	INIT_LIST_HEAD(&nm_i->free_nid_list);
2671 2672
	INIT_RADIX_TREE(&nm_i->nat_root, GFP_NOIO);
	INIT_RADIX_TREE(&nm_i->nat_set_root, GFP_NOIO);
J
Jaegeuk Kim 已提交
2673 2674 2675
	INIT_LIST_HEAD(&nm_i->nat_entries);

	mutex_init(&nm_i->build_lock);
C
Chao Yu 已提交
2676
	spin_lock_init(&nm_i->nid_list_lock);
2677
	init_rwsem(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
2678 2679

	nm_i->next_scan_nid = le32_to_cpu(sbi->ckpt->next_free_nid);
A
Alexandru Gheorghiu 已提交
2680
	nm_i->bitmap_size = __bitmap_size(sbi, NAT_BITMAP);
J
Jaegeuk Kim 已提交
2681 2682 2683 2684
	version_bitmap = __bitmap_ptr(sbi, NAT_BITMAP);
	if (!version_bitmap)
		return -EFAULT;

A
Alexandru Gheorghiu 已提交
2685 2686 2687 2688
	nm_i->nat_bitmap = kmemdup(version_bitmap, nm_i->bitmap_size,
					GFP_KERNEL);
	if (!nm_i->nat_bitmap)
		return -ENOMEM;
2689

2690 2691 2692 2693
	err = __get_nat_bitmaps(sbi);
	if (err)
		return err;

2694 2695 2696 2697 2698 2699 2700
#ifdef CONFIG_F2FS_CHECK_FS
	nm_i->nat_bitmap_mir = kmemdup(version_bitmap, nm_i->bitmap_size,
					GFP_KERNEL);
	if (!nm_i->nat_bitmap_mir)
		return -ENOMEM;
#endif

J
Jaegeuk Kim 已提交
2701 2702 2703
	return 0;
}

J
Jaegeuk Kim 已提交
2704
static int init_free_nid_cache(struct f2fs_sb_info *sbi)
C
Chao Yu 已提交
2705 2706 2707
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);

M
Michal Hocko 已提交
2708
	nm_i->free_nid_bitmap = kvzalloc(nm_i->nat_blocks *
C
Chao Yu 已提交
2709 2710 2711 2712
					NAT_ENTRY_BITMAP_SIZE, GFP_KERNEL);
	if (!nm_i->free_nid_bitmap)
		return -ENOMEM;

M
Michal Hocko 已提交
2713
	nm_i->nat_block_bitmap = kvzalloc(nm_i->nat_blocks / 8,
C
Chao Yu 已提交
2714 2715 2716
								GFP_KERNEL);
	if (!nm_i->nat_block_bitmap)
		return -ENOMEM;
2717

M
Michal Hocko 已提交
2718
	nm_i->free_nid_count = kvzalloc(nm_i->nat_blocks *
2719 2720 2721
					sizeof(unsigned short), GFP_KERNEL);
	if (!nm_i->free_nid_count)
		return -ENOMEM;
C
Chao Yu 已提交
2722 2723 2724
	return 0;
}

J
Jaegeuk Kim 已提交
2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736
int build_node_manager(struct f2fs_sb_info *sbi)
{
	int err;

	sbi->nm_info = kzalloc(sizeof(struct f2fs_nm_info), GFP_KERNEL);
	if (!sbi->nm_info)
		return -ENOMEM;

	err = init_node_manager(sbi);
	if (err)
		return err;

C
Chao Yu 已提交
2737 2738 2739 2740
	err = init_free_nid_cache(sbi);
	if (err)
		return err;

2741 2742 2743
	/* load free nid status from nat_bits table */
	load_free_nid_bitmap(sbi);

2744
	build_free_nids(sbi, true, true);
J
Jaegeuk Kim 已提交
2745 2746 2747 2748 2749 2750 2751 2752
	return 0;
}

void destroy_node_manager(struct f2fs_sb_info *sbi)
{
	struct f2fs_nm_info *nm_i = NM_I(sbi);
	struct free_nid *i, *next_i;
	struct nat_entry *natvec[NATVEC_SIZE];
2753
	struct nat_entry_set *setvec[SETVEC_SIZE];
J
Jaegeuk Kim 已提交
2754 2755 2756 2757 2758 2759 2760
	nid_t nid = 0;
	unsigned int found;

	if (!nm_i)
		return;

	/* destroy free nid list */
C
Chao Yu 已提交
2761
	spin_lock(&nm_i->nid_list_lock);
C
Chao Yu 已提交
2762
	list_for_each_entry_safe(i, next_i, &nm_i->free_nid_list, list) {
F
Fan Li 已提交
2763
		__remove_free_nid(sbi, i, FREE_NID);
C
Chao Yu 已提交
2764
		spin_unlock(&nm_i->nid_list_lock);
2765
		kmem_cache_free(free_nid_slab, i);
C
Chao Yu 已提交
2766
		spin_lock(&nm_i->nid_list_lock);
J
Jaegeuk Kim 已提交
2767
	}
C
Chao Yu 已提交
2768 2769 2770
	f2fs_bug_on(sbi, nm_i->nid_cnt[FREE_NID]);
	f2fs_bug_on(sbi, nm_i->nid_cnt[PREALLOC_NID]);
	f2fs_bug_on(sbi, !list_empty(&nm_i->free_nid_list));
C
Chao Yu 已提交
2771
	spin_unlock(&nm_i->nid_list_lock);
J
Jaegeuk Kim 已提交
2772 2773

	/* destroy nat cache */
2774
	down_write(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
2775 2776 2777
	while ((found = __gang_lookup_nat_cache(nm_i,
					nid, NATVEC_SIZE, natvec))) {
		unsigned idx;
2778

2779 2780 2781
		nid = nat_get_nid(natvec[found - 1]) + 1;
		for (idx = 0; idx < found; idx++)
			__del_from_nat_cache(nm_i, natvec[idx]);
J
Jaegeuk Kim 已提交
2782
	}
2783
	f2fs_bug_on(sbi, nm_i->nat_cnt);
2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798

	/* destroy nat set cache */
	nid = 0;
	while ((found = __gang_lookup_nat_set(nm_i,
					nid, SETVEC_SIZE, setvec))) {
		unsigned idx;

		nid = setvec[found - 1]->set + 1;
		for (idx = 0; idx < found; idx++) {
			/* entry_cnt is not zero, when cp_error was occurred */
			f2fs_bug_on(sbi, !list_empty(&setvec[idx]->entry_list));
			radix_tree_delete(&nm_i->nat_set_root, setvec[idx]->set);
			kmem_cache_free(nat_entry_set_slab, setvec[idx]);
		}
	}
2799
	up_write(&nm_i->nat_tree_lock);
J
Jaegeuk Kim 已提交
2800

C
Chao Yu 已提交
2801 2802
	kvfree(nm_i->nat_block_bitmap);
	kvfree(nm_i->free_nid_bitmap);
2803
	kvfree(nm_i->free_nid_count);
C
Chao Yu 已提交
2804

J
Jaegeuk Kim 已提交
2805
	kfree(nm_i->nat_bitmap);
2806
	kfree(nm_i->nat_bits);
2807 2808 2809
#ifdef CONFIG_F2FS_CHECK_FS
	kfree(nm_i->nat_bitmap_mir);
#endif
J
Jaegeuk Kim 已提交
2810 2811 2812 2813
	sbi->nm_info = NULL;
	kfree(nm_i);
}

2814
int __init create_node_manager_caches(void)
J
Jaegeuk Kim 已提交
2815 2816
{
	nat_entry_slab = f2fs_kmem_cache_create("nat_entry",
2817
			sizeof(struct nat_entry));
J
Jaegeuk Kim 已提交
2818
	if (!nat_entry_slab)
2819
		goto fail;
J
Jaegeuk Kim 已提交
2820 2821

	free_nid_slab = f2fs_kmem_cache_create("free_nid",
2822
			sizeof(struct free_nid));
2823
	if (!free_nid_slab)
2824
		goto destroy_nat_entry;
2825 2826 2827 2828

	nat_entry_set_slab = f2fs_kmem_cache_create("nat_entry_set",
			sizeof(struct nat_entry_set));
	if (!nat_entry_set_slab)
2829
		goto destroy_free_nid;
J
Jaegeuk Kim 已提交
2830
	return 0;
2831

2832
destroy_free_nid:
2833
	kmem_cache_destroy(free_nid_slab);
2834
destroy_nat_entry:
2835 2836 2837
	kmem_cache_destroy(nat_entry_slab);
fail:
	return -ENOMEM;
J
Jaegeuk Kim 已提交
2838 2839 2840 2841
}

void destroy_node_manager_caches(void)
{
2842
	kmem_cache_destroy(nat_entry_set_slab);
J
Jaegeuk Kim 已提交
2843 2844 2845
	kmem_cache_destroy(free_nid_slab);
	kmem_cache_destroy(nat_entry_slab);
}