37 static int make_idx_node(struct ubifs_info *c, struct ubifs_idx_node *idx,  in make_idx_node()  argument
48 		struct ubifs_branch *br = ubifs_idx_branch(c, idx, i);  in make_idx_node()
51 		key_write_idx(c, &zbr->key, &br->key);  in make_idx_node()
56 			ubifs_err(c, "bad ref in znode");  in make_idx_node()
57 			ubifs_dump_znode(c, znode);  in make_idx_node()
59 				ubifs_dump_znode(c, zbr->znode);  in make_idx_node()
64 	ubifs_prepare_node(c, idx, len, 0);  in make_idx_node()
70 	err = insert_old_idx_znode(c, znode);  in make_idx_node()
82 		c->zroot.lnum = lnum;  in make_idx_node()
83 		c->zroot.offs = offs;  in make_idx_node()
84 		c->zroot.len = len;  in make_idx_node()
86 	c->calc_idx_sz += ALIGN(len, 8);  in make_idx_node()
88 	atomic_long_dec(&c->dirty_zn_cnt);  in make_idx_node()
90 	ubifs_assert(c, ubifs_zn_dirty(znode));  in make_idx_node()
91 	ubifs_assert(c, ubifs_zn_cow(znode));  in make_idx_node()
113 static int fill_gap(struct ubifs_info *c, int lnum, int gap_start, int gap_end,  in fill_gap()  argument
118 	ubifs_assert(c, (gap_start & 7) == 0);  in fill_gap()
119 	ubifs_assert(c, (gap_end & 7) == 0);  in fill_gap()
120 	ubifs_assert(c, gap_end >= gap_start);  in fill_gap()
127 	while (c->enext) {  in fill_gap()
128 		len = ubifs_idx_node_sz(c, c->enext->child_cnt);  in fill_gap()
130 			struct ubifs_znode *znode = c->enext;  in fill_gap()
134 			ubifs_assert(c, alen <= gap_remains);  in fill_gap()
135 			err = make_idx_node(c, c->ileb_buf + gap_pos, znode,  in fill_gap()
141 			c->enext = znode->cnext;  in fill_gap()
142 			if (c->enext == c->cnext)  in fill_gap()
143 				c->enext = NULL;  in fill_gap()
148 	if (gap_end == c->leb_size) {  in fill_gap()
149 		c->ileb_len = ALIGN(gap_pos, c->min_io_size);  in fill_gap()
151 		pad_len = c->ileb_len - gap_pos;  in fill_gap()
157 	ubifs_pad(c, c->ileb_buf + gap_pos, pad_len);  in fill_gap()
170 static int find_old_idx(struct ubifs_info *c, int lnum, int offs)  in find_old_idx()  argument
175 	p = c->old_idx.rb_node;  in find_old_idx()
205 static int is_idx_node_in_use(struct ubifs_info *c, union ubifs_key *key,  in is_idx_node_in_use()  argument
210 	ret = is_idx_node_in_tnc(c, key, level, lnum, offs);  in is_idx_node_in_use()
214 		if (find_old_idx(c, lnum, offs))  in is_idx_node_in_use()
231 static int layout_leb_in_gaps(struct ubifs_info *c, int *p)  in layout_leb_in_gaps()  argument
239 	lnum = ubifs_find_dirty_idx_leb(c);  in layout_leb_in_gaps()
253 	sleb = ubifs_scan(c, lnum, 0, c->ileb_buf, 0);  in layout_leb_in_gaps()
254 	c->ileb_len = 0;  in layout_leb_in_gaps()
262 		ubifs_assert(c, snod->type == UBIFS_IDX_NODE);  in layout_leb_in_gaps()
264 		key_read(c, ubifs_idx_key(c, idx), &snod->key);  in layout_leb_in_gaps()
267 		in_use = is_idx_node_in_use(c, &snod->key, level, lnum,  in layout_leb_in_gaps()
284 			written = fill_gap(c, lnum, gap_start, gap_end, &dirt);  in layout_leb_in_gaps()
294 	c->ileb_len = c->leb_size;  in layout_leb_in_gaps()
295 	gap_end = c->leb_size;  in layout_leb_in_gaps()
297 	written = fill_gap(c, lnum, gap_start, gap_end, &dirt);  in layout_leb_in_gaps()
305 		err = ubifs_read_one_lp(c, lnum, &lp);  in layout_leb_in_gaps()
308 		if (lp.free == c->leb_size) {  in layout_leb_in_gaps()
313 			err = ubifs_change_one_lp(c, lnum,  in layout_leb_in_gaps()
314 						  c->leb_size - c->ileb_len,  in layout_leb_in_gaps()
321 	err = ubifs_change_one_lp(c, lnum, c->leb_size - c->ileb_len, dirt,  in layout_leb_in_gaps()
325 	err = ubifs_leb_change(c, lnum, c->ileb_buf, c->ileb_len);  in layout_leb_in_gaps()
341 static int get_leb_cnt(struct ubifs_info *c, int cnt)  in get_leb_cnt()  argument
346 	cnt -= (c->leb_size - c->ihead_offs) / c->max_idx_node_sz;  in get_leb_cnt()
349 	d = c->leb_size / c->max_idx_node_sz;  in get_leb_cnt()
363 static int layout_in_gaps(struct ubifs_info *c, int cnt)  in layout_in_gaps()  argument
369 	c->gap_lebs = kmalloc_array(c->lst.idx_lebs + 1, sizeof(int),  in layout_in_gaps()
371 	if (!c->gap_lebs)  in layout_in_gaps()
374 	p = c->gap_lebs;  in layout_in_gaps()
376 		ubifs_assert(c, p < c->gap_lebs + c->lst.idx_lebs);  in layout_in_gaps()
377 		written = layout_leb_in_gaps(c, p);  in layout_in_gaps()
381 				kfree(c->gap_lebs);  in layout_in_gaps()
382 				c->gap_lebs = NULL;  in layout_in_gaps()
385 			if (!dbg_is_chk_index(c)) {  in layout_in_gaps()
390 				ubifs_warn(c, "out of space");  in layout_in_gaps()
391 				ubifs_dump_budg(c, &c->bi);  in layout_in_gaps()
392 				ubifs_dump_lprops(c);  in layout_in_gaps()
399 		leb_needed_cnt = get_leb_cnt(c, cnt);  in layout_in_gaps()
401 		       leb_needed_cnt, c->ileb_cnt);  in layout_in_gaps()
402 	} while (leb_needed_cnt > c->ileb_cnt);  in layout_in_gaps()
416 static int layout_in_empty_space(struct ubifs_info *c)  in layout_in_empty_space()  argument
422 	cnext = c->enext;  in layout_in_empty_space()
426 	lnum = c->ihead_lnum;  in layout_in_empty_space()
427 	buf_offs = c->ihead_offs;  in layout_in_empty_space()
429 	buf_len = ubifs_idx_node_sz(c, c->fanout);  in layout_in_empty_space()
430 	buf_len = ALIGN(buf_len, c->min_io_size);  in layout_in_empty_space()
435 	next_len = ubifs_idx_node_sz(c, cnext->child_cnt);  in layout_in_empty_space()
436 	if (buf_offs + next_len > c->leb_size)  in layout_in_empty_space()
442 		len = ubifs_idx_node_sz(c, znode->child_cnt);  in layout_in_empty_space()
446 			if (c->ileb_nxt >= c->ileb_cnt) {  in layout_in_empty_space()
447 				ubifs_err(c, "out of space");  in layout_in_empty_space()
450 			lnum = c->ilebs[c->ileb_nxt++];  in layout_in_empty_space()
474 			c->zroot.lnum = lnum;  in layout_in_empty_space()
475 			c->zroot.offs = offs;  in layout_in_empty_space()
476 			c->zroot.len = len;  in layout_in_empty_space()
478 		c->calc_idx_sz += ALIGN(len, 8);  in layout_in_empty_space()
484 		atomic_long_dec(&c->dirty_zn_cnt);  in layout_in_empty_space()
491 		if (cnext == c->cnext)  in layout_in_empty_space()
494 			next_len = ubifs_idx_node_sz(c, cnext->child_cnt);  in layout_in_empty_space()
502 		    buf_offs + used + next_len <= c->leb_size &&  in layout_in_empty_space()
507 		    buf_offs + used + next_len <= c->leb_size)  in layout_in_empty_space()
510 			blen = ALIGN(wlen, c->min_io_size);  in layout_in_empty_space()
515 			if (buf_offs + next_len > c->leb_size) {  in layout_in_empty_space()
516 				err = ubifs_update_one_lp(c, lnum,  in layout_in_empty_space()
517 					c->leb_size - buf_offs, blen - used,  in layout_in_empty_space()
529 		err = ubifs_update_one_lp(c, lnum, c->leb_size - buf_offs,  in layout_in_empty_space()
536 	c->dbg->new_ihead_lnum = lnum;  in layout_in_empty_space()
537 	c->dbg->new_ihead_offs = buf_offs;  in layout_in_empty_space()
555 static int layout_commit(struct ubifs_info *c, int no_space, int cnt)  in layout_commit()  argument
560 		err = layout_in_gaps(c, cnt);  in layout_commit()
564 	err = layout_in_empty_space(c);  in layout_commit()
629 static int get_znodes_to_commit(struct ubifs_info *c)  in get_znodes_to_commit()  argument
634 	c->cnext = find_first_dirty(c->zroot.znode);  in get_znodes_to_commit()
635 	znode = c->enext = c->cnext;  in get_znodes_to_commit()
642 		ubifs_assert(c, !ubifs_zn_cow(znode));  in get_znodes_to_commit()
647 			znode->cnext = c->cnext;  in get_znodes_to_commit()
655 	ubifs_assert(c, cnt == atomic_long_read(&c->dirty_zn_cnt));  in get_znodes_to_commit()
668 static int alloc_idx_lebs(struct ubifs_info *c, int cnt)  in alloc_idx_lebs()  argument
672 	c->ileb_cnt = 0;  in alloc_idx_lebs()
673 	c->ileb_nxt = 0;  in alloc_idx_lebs()
674 	leb_cnt = get_leb_cnt(c, cnt);  in alloc_idx_lebs()
678 	c->ilebs = kmalloc_array(leb_cnt, sizeof(int), GFP_NOFS);  in alloc_idx_lebs()
679 	if (!c->ilebs)  in alloc_idx_lebs()
682 		lnum = ubifs_find_free_leb_for_idx(c);  in alloc_idx_lebs()
685 		c->ilebs[c->ileb_cnt++] = lnum;  in alloc_idx_lebs()
688 	if (dbg_is_chk_index(c) && !(prandom_u32() & 7))  in alloc_idx_lebs()
702 static int free_unused_idx_lebs(struct ubifs_info *c)  in free_unused_idx_lebs()  argument
706 	for (i = c->ileb_nxt; i < c->ileb_cnt; i++) {  in free_unused_idx_lebs()
707 		lnum = c->ilebs[i];  in free_unused_idx_lebs()
709 		er = ubifs_change_one_lp(c, lnum, LPROPS_NC, LPROPS_NC, 0,  in free_unused_idx_lebs()
723 static int free_idx_lebs(struct ubifs_info *c)  in free_idx_lebs()  argument
727 	err = free_unused_idx_lebs(c);  in free_idx_lebs()
728 	kfree(c->ilebs);  in free_idx_lebs()
729 	c->ilebs = NULL;  in free_idx_lebs()
743 int ubifs_tnc_start_commit(struct ubifs_info *c, struct ubifs_zbranch *zroot)  in ubifs_tnc_start_commit()  argument
747 	mutex_lock(&c->tnc_mutex);  in ubifs_tnc_start_commit()
748 	err = dbg_check_tnc(c, 1);  in ubifs_tnc_start_commit()
751 	cnt = get_znodes_to_commit(c);  in ubifs_tnc_start_commit()
755 		err = alloc_idx_lebs(c, cnt);  in ubifs_tnc_start_commit()
760 		err = layout_commit(c, no_space, cnt);  in ubifs_tnc_start_commit()
763 		ubifs_assert(c, atomic_long_read(&c->dirty_zn_cnt) == 0);  in ubifs_tnc_start_commit()
764 		err = free_unused_idx_lebs(c);  in ubifs_tnc_start_commit()
768 	destroy_old_idx(c);  in ubifs_tnc_start_commit()
769 	memcpy(zroot, &c->zroot, sizeof(struct ubifs_zbranch));  in ubifs_tnc_start_commit()
771 	err = ubifs_save_dirty_idx_lnums(c);  in ubifs_tnc_start_commit()
775 	spin_lock(&c->space_lock);  in ubifs_tnc_start_commit()
784 	ubifs_assert(c, c->bi.min_idx_lebs == ubifs_calc_min_idx_lebs(c));  in ubifs_tnc_start_commit()
785 	c->bi.old_idx_sz = c->calc_idx_sz;  in ubifs_tnc_start_commit()
786 	c->bi.uncommitted_idx = 0;  in ubifs_tnc_start_commit()
787 	c->bi.min_idx_lebs = ubifs_calc_min_idx_lebs(c);  in ubifs_tnc_start_commit()
788 	spin_unlock(&c->space_lock);  in ubifs_tnc_start_commit()
789 	mutex_unlock(&c->tnc_mutex);  in ubifs_tnc_start_commit()
791 	dbg_cmt("number of index LEBs %d", c->lst.idx_lebs);  in ubifs_tnc_start_commit()
792 	dbg_cmt("size of index %llu", c->calc_idx_sz);  in ubifs_tnc_start_commit()
796 	free_idx_lebs(c);  in ubifs_tnc_start_commit()
798 	mutex_unlock(&c->tnc_mutex);  in ubifs_tnc_start_commit()
809 static int write_index(struct ubifs_info *c)  in write_index()  argument
816 	cnext = c->enext;  in write_index()
824 	lnum = c->ihead_lnum;  in write_index()
825 	buf_offs = c->ihead_offs;  in write_index()
828 	buf_len = ALIGN(c->max_idx_node_sz, c->min_io_size);  in write_index()
833 	next_len = ubifs_idx_node_sz(c, cnext->child_cnt);  in write_index()
834 	if (buf_offs + next_len > c->leb_size) {  in write_index()
835 		err = ubifs_update_one_lp(c, lnum, LPROPS_NC, 0, 0,  in write_index()
846 		idx = c->cbuf + used;  in write_index()
853 			struct ubifs_branch *br = ubifs_idx_branch(c, idx, i);  in write_index()
856 			key_write_idx(c, &zbr->key, &br->key);  in write_index()
861 				ubifs_err(c, "bad ref in znode");  in write_index()
862 				ubifs_dump_znode(c, znode);  in write_index()
864 					ubifs_dump_znode(c, zbr->znode);  in write_index()
869 		len = ubifs_idx_node_sz(c, znode->child_cnt);  in write_index()
870 		ubifs_prepare_node(c, idx, len, 0);  in write_index()
874 			lnum = c->ilebs[lnum_pos++];  in write_index()
883 			ubifs_err(c, "inconsistent znode posn");  in write_index()
890 		ubifs_assert(c, ubifs_zn_dirty(znode));  in write_index()
891 		ubifs_assert(c, ubifs_zn_cow(znode));  in write_index()
939 		if (cnext == c->cnext)  in write_index()
942 			next_len = ubifs_idx_node_sz(c, cnext->child_cnt);  in write_index()
945 		if (next_len && nxt_offs <= c->leb_size) {  in write_index()
952 			blen = ALIGN(wlen, c->min_io_size);  in write_index()
953 			ubifs_pad(c, c->cbuf + wlen, blen - wlen);  in write_index()
957 		err = ubifs_leb_write(c, lnum, c->cbuf, buf_offs, blen);  in write_index()
962 			if (nxt_offs > c->leb_size) {  in write_index()
963 				err = ubifs_update_one_lp(c, lnum, LPROPS_NC, 0,  in write_index()
973 			memmove(c->cbuf, c->cbuf + blen, used);  in write_index()
979 	if (lnum != c->dbg->new_ihead_lnum ||  in write_index()
980 	    buf_offs != c->dbg->new_ihead_offs) {  in write_index()
981 		ubifs_err(c, "inconsistent ihead");  in write_index()
985 	c->ihead_lnum = lnum;  in write_index()
986 	c->ihead_offs = buf_offs;  in write_index()
997 static void free_obsolete_znodes(struct ubifs_info *c)  in free_obsolete_znodes()  argument
1001 	cnext = c->cnext;  in free_obsolete_znodes()
1009 			atomic_long_inc(&c->clean_zn_cnt);  in free_obsolete_znodes()
1012 	} while (cnext != c->cnext);  in free_obsolete_znodes()
1022 static int return_gap_lebs(struct ubifs_info *c)  in return_gap_lebs()  argument
1026 	if (!c->gap_lebs)  in return_gap_lebs()
1030 	for (p = c->gap_lebs; *p != -1; p++) {  in return_gap_lebs()
1031 		err = ubifs_change_one_lp(c, *p, LPROPS_NC, LPROPS_NC, 0,  in return_gap_lebs()
1037 	kfree(c->gap_lebs);  in return_gap_lebs()
1038 	c->gap_lebs = NULL;  in return_gap_lebs()
1048 int ubifs_tnc_end_commit(struct ubifs_info *c)  in ubifs_tnc_end_commit()  argument
1052 	if (!c->cnext)  in ubifs_tnc_end_commit()
1055 	err = return_gap_lebs(c);  in ubifs_tnc_end_commit()
1059 	err = write_index(c);  in ubifs_tnc_end_commit()
1063 	mutex_lock(&c->tnc_mutex);  in ubifs_tnc_end_commit()
1065 	dbg_cmt("TNC height is %d", c->zroot.znode->level + 1);  in ubifs_tnc_end_commit()
1067 	free_obsolete_znodes(c);  in ubifs_tnc_end_commit()
1069 	c->cnext = NULL;  in ubifs_tnc_end_commit()
1070 	kfree(c->ilebs);  in ubifs_tnc_end_commit()
1071 	c->ilebs = NULL;  in ubifs_tnc_end_commit()
1073 	mutex_unlock(&c->tnc_mutex);  in ubifs_tnc_end_commit()