xref: /OK3568_Linux_fs/kernel/fs/jffs2/nodelist.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun  * JFFS2 -- Journalling Flash File System, Version 2.
3*4882a593Smuzhiyun  *
4*4882a593Smuzhiyun  * Copyright © 2001-2007 Red Hat, Inc.
5*4882a593Smuzhiyun  *
6*4882a593Smuzhiyun  * Created by David Woodhouse <dwmw2@infradead.org>
7*4882a593Smuzhiyun  *
8*4882a593Smuzhiyun  * For licensing information, see the file 'LICENCE' in this directory.
9*4882a593Smuzhiyun  *
10*4882a593Smuzhiyun  */
11*4882a593Smuzhiyun 
12*4882a593Smuzhiyun #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
13*4882a593Smuzhiyun 
14*4882a593Smuzhiyun #include <linux/kernel.h>
15*4882a593Smuzhiyun #include <linux/sched.h>
16*4882a593Smuzhiyun #include <linux/fs.h>
17*4882a593Smuzhiyun #include <linux/mtd/mtd.h>
18*4882a593Smuzhiyun #include <linux/rbtree.h>
19*4882a593Smuzhiyun #include <linux/crc32.h>
20*4882a593Smuzhiyun #include <linux/pagemap.h>
21*4882a593Smuzhiyun #include "nodelist.h"
22*4882a593Smuzhiyun 
23*4882a593Smuzhiyun static void jffs2_obsolete_node_frag(struct jffs2_sb_info *c,
24*4882a593Smuzhiyun 				     struct jffs2_node_frag *this);
25*4882a593Smuzhiyun 
jffs2_add_fd_to_list(struct jffs2_sb_info * c,struct jffs2_full_dirent * new,struct jffs2_full_dirent ** list)26*4882a593Smuzhiyun void jffs2_add_fd_to_list(struct jffs2_sb_info *c, struct jffs2_full_dirent *new, struct jffs2_full_dirent **list)
27*4882a593Smuzhiyun {
28*4882a593Smuzhiyun 	struct jffs2_full_dirent **prev = list;
29*4882a593Smuzhiyun 
30*4882a593Smuzhiyun 	dbg_dentlist("add dirent \"%s\", ino #%u\n", new->name, new->ino);
31*4882a593Smuzhiyun 
32*4882a593Smuzhiyun 	while ((*prev) && (*prev)->nhash <= new->nhash) {
33*4882a593Smuzhiyun 		if ((*prev)->nhash == new->nhash && !strcmp((*prev)->name, new->name)) {
34*4882a593Smuzhiyun 			/* Duplicate. Free one */
35*4882a593Smuzhiyun 			if (new->version < (*prev)->version) {
36*4882a593Smuzhiyun 				dbg_dentlist("Eep! Marking new dirent node obsolete, old is \"%s\", ino #%u\n",
37*4882a593Smuzhiyun 					(*prev)->name, (*prev)->ino);
38*4882a593Smuzhiyun 				jffs2_mark_node_obsolete(c, new->raw);
39*4882a593Smuzhiyun 				jffs2_free_full_dirent(new);
40*4882a593Smuzhiyun 			} else {
41*4882a593Smuzhiyun 				dbg_dentlist("marking old dirent \"%s\", ino #%u obsolete\n",
42*4882a593Smuzhiyun 					(*prev)->name, (*prev)->ino);
43*4882a593Smuzhiyun 				new->next = (*prev)->next;
44*4882a593Smuzhiyun 				/* It may have been a 'placeholder' deletion dirent,
45*4882a593Smuzhiyun 				   if jffs2_can_mark_obsolete() (see jffs2_do_unlink()) */
46*4882a593Smuzhiyun 				if ((*prev)->raw)
47*4882a593Smuzhiyun 					jffs2_mark_node_obsolete(c, ((*prev)->raw));
48*4882a593Smuzhiyun 				jffs2_free_full_dirent(*prev);
49*4882a593Smuzhiyun 				*prev = new;
50*4882a593Smuzhiyun 			}
51*4882a593Smuzhiyun 			return;
52*4882a593Smuzhiyun 		}
53*4882a593Smuzhiyun 		prev = &((*prev)->next);
54*4882a593Smuzhiyun 	}
55*4882a593Smuzhiyun 	new->next = *prev;
56*4882a593Smuzhiyun 	*prev = new;
57*4882a593Smuzhiyun }
58*4882a593Smuzhiyun 
jffs2_truncate_fragtree(struct jffs2_sb_info * c,struct rb_root * list,uint32_t size)59*4882a593Smuzhiyun uint32_t jffs2_truncate_fragtree(struct jffs2_sb_info *c, struct rb_root *list, uint32_t size)
60*4882a593Smuzhiyun {
61*4882a593Smuzhiyun 	struct jffs2_node_frag *frag = jffs2_lookup_node_frag(list, size);
62*4882a593Smuzhiyun 
63*4882a593Smuzhiyun 	dbg_fragtree("truncating fragtree to 0x%08x bytes\n", size);
64*4882a593Smuzhiyun 
65*4882a593Smuzhiyun 	/* We know frag->ofs <= size. That's what lookup does for us */
66*4882a593Smuzhiyun 	if (frag && frag->ofs != size) {
67*4882a593Smuzhiyun 		if (frag->ofs+frag->size > size) {
68*4882a593Smuzhiyun 			frag->size = size - frag->ofs;
69*4882a593Smuzhiyun 		}
70*4882a593Smuzhiyun 		frag = frag_next(frag);
71*4882a593Smuzhiyun 	}
72*4882a593Smuzhiyun 	while (frag && frag->ofs >= size) {
73*4882a593Smuzhiyun 		struct jffs2_node_frag *next = frag_next(frag);
74*4882a593Smuzhiyun 
75*4882a593Smuzhiyun 		frag_erase(frag, list);
76*4882a593Smuzhiyun 		jffs2_obsolete_node_frag(c, frag);
77*4882a593Smuzhiyun 		frag = next;
78*4882a593Smuzhiyun 	}
79*4882a593Smuzhiyun 
80*4882a593Smuzhiyun 	if (size == 0)
81*4882a593Smuzhiyun 		return 0;
82*4882a593Smuzhiyun 
83*4882a593Smuzhiyun 	frag = frag_last(list);
84*4882a593Smuzhiyun 
85*4882a593Smuzhiyun 	/* Sanity check for truncation to longer than we started with... */
86*4882a593Smuzhiyun 	if (!frag)
87*4882a593Smuzhiyun 		return 0;
88*4882a593Smuzhiyun 	if (frag->ofs + frag->size < size)
89*4882a593Smuzhiyun 		return frag->ofs + frag->size;
90*4882a593Smuzhiyun 
91*4882a593Smuzhiyun 	/* If the last fragment starts at the RAM page boundary, it is
92*4882a593Smuzhiyun 	 * REF_PRISTINE irrespective of its size. */
93*4882a593Smuzhiyun 	if (frag->node && (frag->ofs & (PAGE_SIZE - 1)) == 0) {
94*4882a593Smuzhiyun 		dbg_fragtree2("marking the last fragment 0x%08x-0x%08x REF_PRISTINE.\n",
95*4882a593Smuzhiyun 			frag->ofs, frag->ofs + frag->size);
96*4882a593Smuzhiyun 		frag->node->raw->flash_offset = ref_offset(frag->node->raw) | REF_PRISTINE;
97*4882a593Smuzhiyun 	}
98*4882a593Smuzhiyun 	return size;
99*4882a593Smuzhiyun }
100*4882a593Smuzhiyun 
jffs2_obsolete_node_frag(struct jffs2_sb_info * c,struct jffs2_node_frag * this)101*4882a593Smuzhiyun static void jffs2_obsolete_node_frag(struct jffs2_sb_info *c,
102*4882a593Smuzhiyun 				     struct jffs2_node_frag *this)
103*4882a593Smuzhiyun {
104*4882a593Smuzhiyun 	if (this->node) {
105*4882a593Smuzhiyun 		this->node->frags--;
106*4882a593Smuzhiyun 		if (!this->node->frags) {
107*4882a593Smuzhiyun 			/* The node has no valid frags left. It's totally obsoleted */
108*4882a593Smuzhiyun 			dbg_fragtree2("marking old node @0x%08x (0x%04x-0x%04x) obsolete\n",
109*4882a593Smuzhiyun 				ref_offset(this->node->raw), this->node->ofs, this->node->ofs+this->node->size);
110*4882a593Smuzhiyun 			jffs2_mark_node_obsolete(c, this->node->raw);
111*4882a593Smuzhiyun 			jffs2_free_full_dnode(this->node);
112*4882a593Smuzhiyun 		} else {
113*4882a593Smuzhiyun 			dbg_fragtree2("marking old node @0x%08x (0x%04x-0x%04x) REF_NORMAL. frags is %d\n",
114*4882a593Smuzhiyun 				ref_offset(this->node->raw), this->node->ofs, this->node->ofs+this->node->size, this->node->frags);
115*4882a593Smuzhiyun 			mark_ref_normal(this->node->raw);
116*4882a593Smuzhiyun 		}
117*4882a593Smuzhiyun 
118*4882a593Smuzhiyun 	}
119*4882a593Smuzhiyun 	jffs2_free_node_frag(this);
120*4882a593Smuzhiyun }
121*4882a593Smuzhiyun 
jffs2_fragtree_insert(struct jffs2_node_frag * newfrag,struct jffs2_node_frag * base)122*4882a593Smuzhiyun static void jffs2_fragtree_insert(struct jffs2_node_frag *newfrag, struct jffs2_node_frag *base)
123*4882a593Smuzhiyun {
124*4882a593Smuzhiyun 	struct rb_node *parent = &base->rb;
125*4882a593Smuzhiyun 	struct rb_node **link = &parent;
126*4882a593Smuzhiyun 
127*4882a593Smuzhiyun 	dbg_fragtree2("insert frag (0x%04x-0x%04x)\n", newfrag->ofs, newfrag->ofs + newfrag->size);
128*4882a593Smuzhiyun 
129*4882a593Smuzhiyun 	while (*link) {
130*4882a593Smuzhiyun 		parent = *link;
131*4882a593Smuzhiyun 		base = rb_entry(parent, struct jffs2_node_frag, rb);
132*4882a593Smuzhiyun 
133*4882a593Smuzhiyun 		if (newfrag->ofs > base->ofs)
134*4882a593Smuzhiyun 			link = &base->rb.rb_right;
135*4882a593Smuzhiyun 		else if (newfrag->ofs < base->ofs)
136*4882a593Smuzhiyun 			link = &base->rb.rb_left;
137*4882a593Smuzhiyun 		else {
138*4882a593Smuzhiyun 			JFFS2_ERROR("duplicate frag at %08x (%p,%p)\n", newfrag->ofs, newfrag, base);
139*4882a593Smuzhiyun 			BUG();
140*4882a593Smuzhiyun 		}
141*4882a593Smuzhiyun 	}
142*4882a593Smuzhiyun 
143*4882a593Smuzhiyun 	rb_link_node(&newfrag->rb, &base->rb, link);
144*4882a593Smuzhiyun }
145*4882a593Smuzhiyun 
146*4882a593Smuzhiyun /*
147*4882a593Smuzhiyun  * Allocate and initializes a new fragment.
148*4882a593Smuzhiyun  */
new_fragment(struct jffs2_full_dnode * fn,uint32_t ofs,uint32_t size)149*4882a593Smuzhiyun static struct jffs2_node_frag * new_fragment(struct jffs2_full_dnode *fn, uint32_t ofs, uint32_t size)
150*4882a593Smuzhiyun {
151*4882a593Smuzhiyun 	struct jffs2_node_frag *newfrag;
152*4882a593Smuzhiyun 
153*4882a593Smuzhiyun 	newfrag = jffs2_alloc_node_frag();
154*4882a593Smuzhiyun 	if (likely(newfrag)) {
155*4882a593Smuzhiyun 		newfrag->ofs = ofs;
156*4882a593Smuzhiyun 		newfrag->size = size;
157*4882a593Smuzhiyun 		newfrag->node = fn;
158*4882a593Smuzhiyun 	} else {
159*4882a593Smuzhiyun 		JFFS2_ERROR("cannot allocate a jffs2_node_frag object\n");
160*4882a593Smuzhiyun 	}
161*4882a593Smuzhiyun 
162*4882a593Smuzhiyun 	return newfrag;
163*4882a593Smuzhiyun }
164*4882a593Smuzhiyun 
165*4882a593Smuzhiyun /*
166*4882a593Smuzhiyun  * Called when there is no overlapping fragment exist. Inserts a hole before the new
167*4882a593Smuzhiyun  * fragment and inserts the new fragment to the fragtree.
168*4882a593Smuzhiyun  */
no_overlapping_node(struct jffs2_sb_info * c,struct rb_root * root,struct jffs2_node_frag * newfrag,struct jffs2_node_frag * this,uint32_t lastend)169*4882a593Smuzhiyun static int no_overlapping_node(struct jffs2_sb_info *c, struct rb_root *root,
170*4882a593Smuzhiyun 		 	       struct jffs2_node_frag *newfrag,
171*4882a593Smuzhiyun 			       struct jffs2_node_frag *this, uint32_t lastend)
172*4882a593Smuzhiyun {
173*4882a593Smuzhiyun 	if (lastend < newfrag->node->ofs) {
174*4882a593Smuzhiyun 		/* put a hole in before the new fragment */
175*4882a593Smuzhiyun 		struct jffs2_node_frag *holefrag;
176*4882a593Smuzhiyun 
177*4882a593Smuzhiyun 		holefrag= new_fragment(NULL, lastend, newfrag->node->ofs - lastend);
178*4882a593Smuzhiyun 		if (unlikely(!holefrag)) {
179*4882a593Smuzhiyun 			jffs2_free_node_frag(newfrag);
180*4882a593Smuzhiyun 			return -ENOMEM;
181*4882a593Smuzhiyun 		}
182*4882a593Smuzhiyun 
183*4882a593Smuzhiyun 		if (this) {
184*4882a593Smuzhiyun 			/* By definition, the 'this' node has no right-hand child,
185*4882a593Smuzhiyun 			   because there are no frags with offset greater than it.
186*4882a593Smuzhiyun 			   So that's where we want to put the hole */
187*4882a593Smuzhiyun 			dbg_fragtree2("add hole frag %#04x-%#04x on the right of the new frag.\n",
188*4882a593Smuzhiyun 				holefrag->ofs, holefrag->ofs + holefrag->size);
189*4882a593Smuzhiyun 			rb_link_node(&holefrag->rb, &this->rb, &this->rb.rb_right);
190*4882a593Smuzhiyun 		} else {
191*4882a593Smuzhiyun 			dbg_fragtree2("Add hole frag %#04x-%#04x to the root of the tree.\n",
192*4882a593Smuzhiyun 				holefrag->ofs, holefrag->ofs + holefrag->size);
193*4882a593Smuzhiyun 			rb_link_node(&holefrag->rb, NULL, &root->rb_node);
194*4882a593Smuzhiyun 		}
195*4882a593Smuzhiyun 		rb_insert_color(&holefrag->rb, root);
196*4882a593Smuzhiyun 		this = holefrag;
197*4882a593Smuzhiyun 	}
198*4882a593Smuzhiyun 
199*4882a593Smuzhiyun 	if (this) {
200*4882a593Smuzhiyun 		/* By definition, the 'this' node has no right-hand child,
201*4882a593Smuzhiyun 		   because there are no frags with offset greater than it.
202*4882a593Smuzhiyun 		   So that's where we want to put new fragment */
203*4882a593Smuzhiyun 		dbg_fragtree2("add the new node at the right\n");
204*4882a593Smuzhiyun 		rb_link_node(&newfrag->rb, &this->rb, &this->rb.rb_right);
205*4882a593Smuzhiyun 	} else {
206*4882a593Smuzhiyun 		dbg_fragtree2("insert the new node at the root of the tree\n");
207*4882a593Smuzhiyun 		rb_link_node(&newfrag->rb, NULL, &root->rb_node);
208*4882a593Smuzhiyun 	}
209*4882a593Smuzhiyun 	rb_insert_color(&newfrag->rb, root);
210*4882a593Smuzhiyun 
211*4882a593Smuzhiyun 	return 0;
212*4882a593Smuzhiyun }
213*4882a593Smuzhiyun 
214*4882a593Smuzhiyun /* Doesn't set inode->i_size */
jffs2_add_frag_to_fragtree(struct jffs2_sb_info * c,struct rb_root * root,struct jffs2_node_frag * newfrag)215*4882a593Smuzhiyun static int jffs2_add_frag_to_fragtree(struct jffs2_sb_info *c, struct rb_root *root, struct jffs2_node_frag *newfrag)
216*4882a593Smuzhiyun {
217*4882a593Smuzhiyun 	struct jffs2_node_frag *this;
218*4882a593Smuzhiyun 	uint32_t lastend;
219*4882a593Smuzhiyun 
220*4882a593Smuzhiyun 	/* Skip all the nodes which are completed before this one starts */
221*4882a593Smuzhiyun 	this = jffs2_lookup_node_frag(root, newfrag->node->ofs);
222*4882a593Smuzhiyun 
223*4882a593Smuzhiyun 	if (this) {
224*4882a593Smuzhiyun 		dbg_fragtree2("lookup gave frag 0x%04x-0x%04x; phys 0x%08x (*%p)\n",
225*4882a593Smuzhiyun 			  this->ofs, this->ofs+this->size, this->node?(ref_offset(this->node->raw)):0xffffffff, this);
226*4882a593Smuzhiyun 		lastend = this->ofs + this->size;
227*4882a593Smuzhiyun 	} else {
228*4882a593Smuzhiyun 		dbg_fragtree2("lookup gave no frag\n");
229*4882a593Smuzhiyun 		lastend = 0;
230*4882a593Smuzhiyun 	}
231*4882a593Smuzhiyun 
232*4882a593Smuzhiyun 	/* See if we ran off the end of the fragtree */
233*4882a593Smuzhiyun 	if (lastend <= newfrag->ofs) {
234*4882a593Smuzhiyun 		/* We did */
235*4882a593Smuzhiyun 
236*4882a593Smuzhiyun 		/* Check if 'this' node was on the same page as the new node.
237*4882a593Smuzhiyun 		   If so, both 'this' and the new node get marked REF_NORMAL so
238*4882a593Smuzhiyun 		   the GC can take a look.
239*4882a593Smuzhiyun 		*/
240*4882a593Smuzhiyun 		if (lastend && (lastend-1) >> PAGE_SHIFT == newfrag->ofs >> PAGE_SHIFT) {
241*4882a593Smuzhiyun 			if (this->node)
242*4882a593Smuzhiyun 				mark_ref_normal(this->node->raw);
243*4882a593Smuzhiyun 			mark_ref_normal(newfrag->node->raw);
244*4882a593Smuzhiyun 		}
245*4882a593Smuzhiyun 
246*4882a593Smuzhiyun 		return no_overlapping_node(c, root, newfrag, this, lastend);
247*4882a593Smuzhiyun 	}
248*4882a593Smuzhiyun 
249*4882a593Smuzhiyun 	if (this->node)
250*4882a593Smuzhiyun 		dbg_fragtree2("dealing with frag %u-%u, phys %#08x(%d).\n",
251*4882a593Smuzhiyun 		this->ofs, this->ofs + this->size,
252*4882a593Smuzhiyun 		ref_offset(this->node->raw), ref_flags(this->node->raw));
253*4882a593Smuzhiyun 	else
254*4882a593Smuzhiyun 		dbg_fragtree2("dealing with hole frag %u-%u.\n",
255*4882a593Smuzhiyun 		this->ofs, this->ofs + this->size);
256*4882a593Smuzhiyun 
257*4882a593Smuzhiyun 	/* OK. 'this' is pointing at the first frag that newfrag->ofs at least partially obsoletes,
258*4882a593Smuzhiyun 	 * - i.e. newfrag->ofs < this->ofs+this->size && newfrag->ofs >= this->ofs
259*4882a593Smuzhiyun 	 */
260*4882a593Smuzhiyun 	if (newfrag->ofs > this->ofs) {
261*4882a593Smuzhiyun 		/* This node isn't completely obsoleted. The start of it remains valid */
262*4882a593Smuzhiyun 
263*4882a593Smuzhiyun 		/* Mark the new node and the partially covered node REF_NORMAL -- let
264*4882a593Smuzhiyun 		   the GC take a look at them */
265*4882a593Smuzhiyun 		mark_ref_normal(newfrag->node->raw);
266*4882a593Smuzhiyun 		if (this->node)
267*4882a593Smuzhiyun 			mark_ref_normal(this->node->raw);
268*4882a593Smuzhiyun 
269*4882a593Smuzhiyun 		if (this->ofs + this->size > newfrag->ofs + newfrag->size) {
270*4882a593Smuzhiyun 			/* The new node splits 'this' frag into two */
271*4882a593Smuzhiyun 			struct jffs2_node_frag *newfrag2;
272*4882a593Smuzhiyun 
273*4882a593Smuzhiyun 			if (this->node)
274*4882a593Smuzhiyun 				dbg_fragtree2("split old frag 0x%04x-0x%04x, phys 0x%08x\n",
275*4882a593Smuzhiyun 					this->ofs, this->ofs+this->size, ref_offset(this->node->raw));
276*4882a593Smuzhiyun 			else
277*4882a593Smuzhiyun 				dbg_fragtree2("split old hole frag 0x%04x-0x%04x\n",
278*4882a593Smuzhiyun 					this->ofs, this->ofs+this->size);
279*4882a593Smuzhiyun 
280*4882a593Smuzhiyun 			/* New second frag pointing to this's node */
281*4882a593Smuzhiyun 			newfrag2 = new_fragment(this->node, newfrag->ofs + newfrag->size,
282*4882a593Smuzhiyun 						this->ofs + this->size - newfrag->ofs - newfrag->size);
283*4882a593Smuzhiyun 			if (unlikely(!newfrag2))
284*4882a593Smuzhiyun 				return -ENOMEM;
285*4882a593Smuzhiyun 			if (this->node)
286*4882a593Smuzhiyun 				this->node->frags++;
287*4882a593Smuzhiyun 
288*4882a593Smuzhiyun 			/* Adjust size of original 'this' */
289*4882a593Smuzhiyun 			this->size = newfrag->ofs - this->ofs;
290*4882a593Smuzhiyun 
291*4882a593Smuzhiyun 			/* Now, we know there's no node with offset
292*4882a593Smuzhiyun 			   greater than this->ofs but smaller than
293*4882a593Smuzhiyun 			   newfrag2->ofs or newfrag->ofs, for obvious
294*4882a593Smuzhiyun 			   reasons. So we can do a tree insert from
295*4882a593Smuzhiyun 			   'this' to insert newfrag, and a tree insert
296*4882a593Smuzhiyun 			   from newfrag to insert newfrag2. */
297*4882a593Smuzhiyun 			jffs2_fragtree_insert(newfrag, this);
298*4882a593Smuzhiyun 			rb_insert_color(&newfrag->rb, root);
299*4882a593Smuzhiyun 
300*4882a593Smuzhiyun 			jffs2_fragtree_insert(newfrag2, newfrag);
301*4882a593Smuzhiyun 			rb_insert_color(&newfrag2->rb, root);
302*4882a593Smuzhiyun 
303*4882a593Smuzhiyun 			return 0;
304*4882a593Smuzhiyun 		}
305*4882a593Smuzhiyun 		/* New node just reduces 'this' frag in size, doesn't split it */
306*4882a593Smuzhiyun 		this->size = newfrag->ofs - this->ofs;
307*4882a593Smuzhiyun 
308*4882a593Smuzhiyun 		/* Again, we know it lives down here in the tree */
309*4882a593Smuzhiyun 		jffs2_fragtree_insert(newfrag, this);
310*4882a593Smuzhiyun 		rb_insert_color(&newfrag->rb, root);
311*4882a593Smuzhiyun 	} else {
312*4882a593Smuzhiyun 		/* New frag starts at the same point as 'this' used to. Replace
313*4882a593Smuzhiyun 		   it in the tree without doing a delete and insertion */
314*4882a593Smuzhiyun 		dbg_fragtree2("inserting newfrag (*%p),%d-%d in before 'this' (*%p),%d-%d\n",
315*4882a593Smuzhiyun 			  newfrag, newfrag->ofs, newfrag->ofs+newfrag->size, this, this->ofs, this->ofs+this->size);
316*4882a593Smuzhiyun 
317*4882a593Smuzhiyun 		rb_replace_node(&this->rb, &newfrag->rb, root);
318*4882a593Smuzhiyun 
319*4882a593Smuzhiyun 		if (newfrag->ofs + newfrag->size >= this->ofs+this->size) {
320*4882a593Smuzhiyun 			dbg_fragtree2("obsoleting node frag %p (%x-%x)\n", this, this->ofs, this->ofs+this->size);
321*4882a593Smuzhiyun 			jffs2_obsolete_node_frag(c, this);
322*4882a593Smuzhiyun 		} else {
323*4882a593Smuzhiyun 			this->ofs += newfrag->size;
324*4882a593Smuzhiyun 			this->size -= newfrag->size;
325*4882a593Smuzhiyun 
326*4882a593Smuzhiyun 			jffs2_fragtree_insert(this, newfrag);
327*4882a593Smuzhiyun 			rb_insert_color(&this->rb, root);
328*4882a593Smuzhiyun 			return 0;
329*4882a593Smuzhiyun 		}
330*4882a593Smuzhiyun 	}
331*4882a593Smuzhiyun 	/* OK, now we have newfrag added in the correct place in the tree, but
332*4882a593Smuzhiyun 	   frag_next(newfrag) may be a fragment which is overlapped by it
333*4882a593Smuzhiyun 	*/
334*4882a593Smuzhiyun 	while ((this = frag_next(newfrag)) && newfrag->ofs + newfrag->size >= this->ofs + this->size) {
335*4882a593Smuzhiyun 		/* 'this' frag is obsoleted completely. */
336*4882a593Smuzhiyun 		dbg_fragtree2("obsoleting node frag %p (%x-%x) and removing from tree\n",
337*4882a593Smuzhiyun 			this, this->ofs, this->ofs+this->size);
338*4882a593Smuzhiyun 		rb_erase(&this->rb, root);
339*4882a593Smuzhiyun 		jffs2_obsolete_node_frag(c, this);
340*4882a593Smuzhiyun 	}
341*4882a593Smuzhiyun 	/* Now we're pointing at the first frag which isn't totally obsoleted by
342*4882a593Smuzhiyun 	   the new frag */
343*4882a593Smuzhiyun 
344*4882a593Smuzhiyun 	if (!this || newfrag->ofs + newfrag->size == this->ofs)
345*4882a593Smuzhiyun 		return 0;
346*4882a593Smuzhiyun 
347*4882a593Smuzhiyun 	/* Still some overlap but we don't need to move it in the tree */
348*4882a593Smuzhiyun 	this->size = (this->ofs + this->size) - (newfrag->ofs + newfrag->size);
349*4882a593Smuzhiyun 	this->ofs = newfrag->ofs + newfrag->size;
350*4882a593Smuzhiyun 
351*4882a593Smuzhiyun 	/* And mark them REF_NORMAL so the GC takes a look at them */
352*4882a593Smuzhiyun 	if (this->node)
353*4882a593Smuzhiyun 		mark_ref_normal(this->node->raw);
354*4882a593Smuzhiyun 	mark_ref_normal(newfrag->node->raw);
355*4882a593Smuzhiyun 
356*4882a593Smuzhiyun 	return 0;
357*4882a593Smuzhiyun }
358*4882a593Smuzhiyun 
359*4882a593Smuzhiyun /*
360*4882a593Smuzhiyun  * Given an inode, probably with existing tree of fragments, add the new node
361*4882a593Smuzhiyun  * to the fragment tree.
362*4882a593Smuzhiyun  */
jffs2_add_full_dnode_to_inode(struct jffs2_sb_info * c,struct jffs2_inode_info * f,struct jffs2_full_dnode * fn)363*4882a593Smuzhiyun int jffs2_add_full_dnode_to_inode(struct jffs2_sb_info *c, struct jffs2_inode_info *f, struct jffs2_full_dnode *fn)
364*4882a593Smuzhiyun {
365*4882a593Smuzhiyun 	int ret;
366*4882a593Smuzhiyun 	struct jffs2_node_frag *newfrag;
367*4882a593Smuzhiyun 
368*4882a593Smuzhiyun 	if (unlikely(!fn->size))
369*4882a593Smuzhiyun 		return 0;
370*4882a593Smuzhiyun 
371*4882a593Smuzhiyun 	newfrag = new_fragment(fn, fn->ofs, fn->size);
372*4882a593Smuzhiyun 	if (unlikely(!newfrag))
373*4882a593Smuzhiyun 		return -ENOMEM;
374*4882a593Smuzhiyun 	newfrag->node->frags = 1;
375*4882a593Smuzhiyun 
376*4882a593Smuzhiyun 	dbg_fragtree("adding node %#04x-%#04x @0x%08x on flash, newfrag *%p\n",
377*4882a593Smuzhiyun 		  fn->ofs, fn->ofs+fn->size, ref_offset(fn->raw), newfrag);
378*4882a593Smuzhiyun 
379*4882a593Smuzhiyun 	ret = jffs2_add_frag_to_fragtree(c, &f->fragtree, newfrag);
380*4882a593Smuzhiyun 	if (unlikely(ret))
381*4882a593Smuzhiyun 		return ret;
382*4882a593Smuzhiyun 
383*4882a593Smuzhiyun 	/* If we now share a page with other nodes, mark either previous
384*4882a593Smuzhiyun 	   or next node REF_NORMAL, as appropriate.  */
385*4882a593Smuzhiyun 	if (newfrag->ofs & (PAGE_SIZE-1)) {
386*4882a593Smuzhiyun 		struct jffs2_node_frag *prev = frag_prev(newfrag);
387*4882a593Smuzhiyun 
388*4882a593Smuzhiyun 		mark_ref_normal(fn->raw);
389*4882a593Smuzhiyun 		/* If we don't start at zero there's _always_ a previous */
390*4882a593Smuzhiyun 		if (prev->node)
391*4882a593Smuzhiyun 			mark_ref_normal(prev->node->raw);
392*4882a593Smuzhiyun 	}
393*4882a593Smuzhiyun 
394*4882a593Smuzhiyun 	if ((newfrag->ofs+newfrag->size) & (PAGE_SIZE-1)) {
395*4882a593Smuzhiyun 		struct jffs2_node_frag *next = frag_next(newfrag);
396*4882a593Smuzhiyun 
397*4882a593Smuzhiyun 		if (next) {
398*4882a593Smuzhiyun 			mark_ref_normal(fn->raw);
399*4882a593Smuzhiyun 			if (next->node)
400*4882a593Smuzhiyun 				mark_ref_normal(next->node->raw);
401*4882a593Smuzhiyun 		}
402*4882a593Smuzhiyun 	}
403*4882a593Smuzhiyun 	jffs2_dbg_fragtree_paranoia_check_nolock(f);
404*4882a593Smuzhiyun 
405*4882a593Smuzhiyun 	return 0;
406*4882a593Smuzhiyun }
407*4882a593Smuzhiyun 
jffs2_set_inocache_state(struct jffs2_sb_info * c,struct jffs2_inode_cache * ic,int state)408*4882a593Smuzhiyun void jffs2_set_inocache_state(struct jffs2_sb_info *c, struct jffs2_inode_cache *ic, int state)
409*4882a593Smuzhiyun {
410*4882a593Smuzhiyun 	spin_lock(&c->inocache_lock);
411*4882a593Smuzhiyun 	ic->state = state;
412*4882a593Smuzhiyun 	wake_up(&c->inocache_wq);
413*4882a593Smuzhiyun 	spin_unlock(&c->inocache_lock);
414*4882a593Smuzhiyun }
415*4882a593Smuzhiyun 
416*4882a593Smuzhiyun /* During mount, this needs no locking. During normal operation, its
417*4882a593Smuzhiyun    callers want to do other stuff while still holding the inocache_lock.
418*4882a593Smuzhiyun    Rather than introducing special case get_ino_cache functions or
419*4882a593Smuzhiyun    callbacks, we just let the caller do the locking itself. */
420*4882a593Smuzhiyun 
jffs2_get_ino_cache(struct jffs2_sb_info * c,uint32_t ino)421*4882a593Smuzhiyun struct jffs2_inode_cache *jffs2_get_ino_cache(struct jffs2_sb_info *c, uint32_t ino)
422*4882a593Smuzhiyun {
423*4882a593Smuzhiyun 	struct jffs2_inode_cache *ret;
424*4882a593Smuzhiyun 
425*4882a593Smuzhiyun 	ret = c->inocache_list[ino % c->inocache_hashsize];
426*4882a593Smuzhiyun 	while (ret && ret->ino < ino) {
427*4882a593Smuzhiyun 		ret = ret->next;
428*4882a593Smuzhiyun 	}
429*4882a593Smuzhiyun 
430*4882a593Smuzhiyun 	if (ret && ret->ino != ino)
431*4882a593Smuzhiyun 		ret = NULL;
432*4882a593Smuzhiyun 
433*4882a593Smuzhiyun 	return ret;
434*4882a593Smuzhiyun }
435*4882a593Smuzhiyun 
jffs2_add_ino_cache(struct jffs2_sb_info * c,struct jffs2_inode_cache * new)436*4882a593Smuzhiyun void jffs2_add_ino_cache (struct jffs2_sb_info *c, struct jffs2_inode_cache *new)
437*4882a593Smuzhiyun {
438*4882a593Smuzhiyun 	struct jffs2_inode_cache **prev;
439*4882a593Smuzhiyun 
440*4882a593Smuzhiyun 	spin_lock(&c->inocache_lock);
441*4882a593Smuzhiyun 	if (!new->ino)
442*4882a593Smuzhiyun 		new->ino = ++c->highest_ino;
443*4882a593Smuzhiyun 
444*4882a593Smuzhiyun 	dbg_inocache("add %p (ino #%u)\n", new, new->ino);
445*4882a593Smuzhiyun 
446*4882a593Smuzhiyun 	prev = &c->inocache_list[new->ino % c->inocache_hashsize];
447*4882a593Smuzhiyun 
448*4882a593Smuzhiyun 	while ((*prev) && (*prev)->ino < new->ino) {
449*4882a593Smuzhiyun 		prev = &(*prev)->next;
450*4882a593Smuzhiyun 	}
451*4882a593Smuzhiyun 	new->next = *prev;
452*4882a593Smuzhiyun 	*prev = new;
453*4882a593Smuzhiyun 
454*4882a593Smuzhiyun 	spin_unlock(&c->inocache_lock);
455*4882a593Smuzhiyun }
456*4882a593Smuzhiyun 
jffs2_del_ino_cache(struct jffs2_sb_info * c,struct jffs2_inode_cache * old)457*4882a593Smuzhiyun void jffs2_del_ino_cache(struct jffs2_sb_info *c, struct jffs2_inode_cache *old)
458*4882a593Smuzhiyun {
459*4882a593Smuzhiyun 	struct jffs2_inode_cache **prev;
460*4882a593Smuzhiyun 
461*4882a593Smuzhiyun #ifdef CONFIG_JFFS2_FS_XATTR
462*4882a593Smuzhiyun 	BUG_ON(old->xref);
463*4882a593Smuzhiyun #endif
464*4882a593Smuzhiyun 	dbg_inocache("del %p (ino #%u)\n", old, old->ino);
465*4882a593Smuzhiyun 	spin_lock(&c->inocache_lock);
466*4882a593Smuzhiyun 
467*4882a593Smuzhiyun 	prev = &c->inocache_list[old->ino % c->inocache_hashsize];
468*4882a593Smuzhiyun 
469*4882a593Smuzhiyun 	while ((*prev) && (*prev)->ino < old->ino) {
470*4882a593Smuzhiyun 		prev = &(*prev)->next;
471*4882a593Smuzhiyun 	}
472*4882a593Smuzhiyun 	if ((*prev) == old) {
473*4882a593Smuzhiyun 		*prev = old->next;
474*4882a593Smuzhiyun 	}
475*4882a593Smuzhiyun 
476*4882a593Smuzhiyun 	/* Free it now unless it's in READING or CLEARING state, which
477*4882a593Smuzhiyun 	   are the transitions upon read_inode() and clear_inode(). The
478*4882a593Smuzhiyun 	   rest of the time we know nobody else is looking at it, and
479*4882a593Smuzhiyun 	   if it's held by read_inode() or clear_inode() they'll free it
480*4882a593Smuzhiyun 	   for themselves. */
481*4882a593Smuzhiyun 	if (old->state != INO_STATE_READING && old->state != INO_STATE_CLEARING)
482*4882a593Smuzhiyun 		jffs2_free_inode_cache(old);
483*4882a593Smuzhiyun 
484*4882a593Smuzhiyun 	spin_unlock(&c->inocache_lock);
485*4882a593Smuzhiyun }
486*4882a593Smuzhiyun 
jffs2_free_ino_caches(struct jffs2_sb_info * c)487*4882a593Smuzhiyun void jffs2_free_ino_caches(struct jffs2_sb_info *c)
488*4882a593Smuzhiyun {
489*4882a593Smuzhiyun 	int i;
490*4882a593Smuzhiyun 	struct jffs2_inode_cache *this, *next;
491*4882a593Smuzhiyun 
492*4882a593Smuzhiyun 	for (i=0; i < c->inocache_hashsize; i++) {
493*4882a593Smuzhiyun 		this = c->inocache_list[i];
494*4882a593Smuzhiyun 		while (this) {
495*4882a593Smuzhiyun 			next = this->next;
496*4882a593Smuzhiyun 			jffs2_xattr_free_inode(c, this);
497*4882a593Smuzhiyun 			jffs2_free_inode_cache(this);
498*4882a593Smuzhiyun 			this = next;
499*4882a593Smuzhiyun 		}
500*4882a593Smuzhiyun 		c->inocache_list[i] = NULL;
501*4882a593Smuzhiyun 	}
502*4882a593Smuzhiyun }
503*4882a593Smuzhiyun 
jffs2_free_raw_node_refs(struct jffs2_sb_info * c)504*4882a593Smuzhiyun void jffs2_free_raw_node_refs(struct jffs2_sb_info *c)
505*4882a593Smuzhiyun {
506*4882a593Smuzhiyun 	int i;
507*4882a593Smuzhiyun 	struct jffs2_raw_node_ref *this, *next;
508*4882a593Smuzhiyun 
509*4882a593Smuzhiyun 	for (i=0; i<c->nr_blocks; i++) {
510*4882a593Smuzhiyun 		this = c->blocks[i].first_node;
511*4882a593Smuzhiyun 		while (this) {
512*4882a593Smuzhiyun 			if (this[REFS_PER_BLOCK].flash_offset == REF_LINK_NODE)
513*4882a593Smuzhiyun 				next = this[REFS_PER_BLOCK].next_in_ino;
514*4882a593Smuzhiyun 			else
515*4882a593Smuzhiyun 				next = NULL;
516*4882a593Smuzhiyun 
517*4882a593Smuzhiyun 			jffs2_free_refblock(this);
518*4882a593Smuzhiyun 			this = next;
519*4882a593Smuzhiyun 		}
520*4882a593Smuzhiyun 		c->blocks[i].first_node = c->blocks[i].last_node = NULL;
521*4882a593Smuzhiyun 	}
522*4882a593Smuzhiyun }
523*4882a593Smuzhiyun 
jffs2_lookup_node_frag(struct rb_root * fragtree,uint32_t offset)524*4882a593Smuzhiyun struct jffs2_node_frag *jffs2_lookup_node_frag(struct rb_root *fragtree, uint32_t offset)
525*4882a593Smuzhiyun {
526*4882a593Smuzhiyun 	/* The common case in lookup is that there will be a node
527*4882a593Smuzhiyun 	   which precisely matches. So we go looking for that first */
528*4882a593Smuzhiyun 	struct rb_node *next;
529*4882a593Smuzhiyun 	struct jffs2_node_frag *prev = NULL;
530*4882a593Smuzhiyun 	struct jffs2_node_frag *frag = NULL;
531*4882a593Smuzhiyun 
532*4882a593Smuzhiyun 	dbg_fragtree2("root %p, offset %d\n", fragtree, offset);
533*4882a593Smuzhiyun 
534*4882a593Smuzhiyun 	next = fragtree->rb_node;
535*4882a593Smuzhiyun 
536*4882a593Smuzhiyun 	while(next) {
537*4882a593Smuzhiyun 		frag = rb_entry(next, struct jffs2_node_frag, rb);
538*4882a593Smuzhiyun 
539*4882a593Smuzhiyun 		if (frag->ofs + frag->size <= offset) {
540*4882a593Smuzhiyun 			/* Remember the closest smaller match on the way down */
541*4882a593Smuzhiyun 			if (!prev || frag->ofs > prev->ofs)
542*4882a593Smuzhiyun 				prev = frag;
543*4882a593Smuzhiyun 			next = frag->rb.rb_right;
544*4882a593Smuzhiyun 		} else if (frag->ofs > offset) {
545*4882a593Smuzhiyun 			next = frag->rb.rb_left;
546*4882a593Smuzhiyun 		} else {
547*4882a593Smuzhiyun 			return frag;
548*4882a593Smuzhiyun 		}
549*4882a593Smuzhiyun 	}
550*4882a593Smuzhiyun 
551*4882a593Smuzhiyun 	/* Exact match not found. Go back up looking at each parent,
552*4882a593Smuzhiyun 	   and return the closest smaller one */
553*4882a593Smuzhiyun 
554*4882a593Smuzhiyun 	if (prev)
555*4882a593Smuzhiyun 		dbg_fragtree2("no match. Returning frag %#04x-%#04x, closest previous\n",
556*4882a593Smuzhiyun 			  prev->ofs, prev->ofs+prev->size);
557*4882a593Smuzhiyun 	else
558*4882a593Smuzhiyun 		dbg_fragtree2("returning NULL, empty fragtree\n");
559*4882a593Smuzhiyun 
560*4882a593Smuzhiyun 	return prev;
561*4882a593Smuzhiyun }
562*4882a593Smuzhiyun 
563*4882a593Smuzhiyun /* Pass 'c' argument to indicate that nodes should be marked obsolete as
564*4882a593Smuzhiyun    they're killed. */
jffs2_kill_fragtree(struct rb_root * root,struct jffs2_sb_info * c)565*4882a593Smuzhiyun void jffs2_kill_fragtree(struct rb_root *root, struct jffs2_sb_info *c)
566*4882a593Smuzhiyun {
567*4882a593Smuzhiyun 	struct jffs2_node_frag *frag, *next;
568*4882a593Smuzhiyun 
569*4882a593Smuzhiyun 	dbg_fragtree("killing\n");
570*4882a593Smuzhiyun 	rbtree_postorder_for_each_entry_safe(frag, next, root, rb) {
571*4882a593Smuzhiyun 		if (frag->node && !(--frag->node->frags)) {
572*4882a593Smuzhiyun 			/* Not a hole, and it's the final remaining frag
573*4882a593Smuzhiyun 			   of this node. Free the node */
574*4882a593Smuzhiyun 			if (c)
575*4882a593Smuzhiyun 				jffs2_mark_node_obsolete(c, frag->node->raw);
576*4882a593Smuzhiyun 
577*4882a593Smuzhiyun 			jffs2_free_full_dnode(frag->node);
578*4882a593Smuzhiyun 		}
579*4882a593Smuzhiyun 
580*4882a593Smuzhiyun 		jffs2_free_node_frag(frag);
581*4882a593Smuzhiyun 		cond_resched();
582*4882a593Smuzhiyun 	}
583*4882a593Smuzhiyun }
584*4882a593Smuzhiyun 
jffs2_link_node_ref(struct jffs2_sb_info * c,struct jffs2_eraseblock * jeb,uint32_t ofs,uint32_t len,struct jffs2_inode_cache * ic)585*4882a593Smuzhiyun struct jffs2_raw_node_ref *jffs2_link_node_ref(struct jffs2_sb_info *c,
586*4882a593Smuzhiyun 					       struct jffs2_eraseblock *jeb,
587*4882a593Smuzhiyun 					       uint32_t ofs, uint32_t len,
588*4882a593Smuzhiyun 					       struct jffs2_inode_cache *ic)
589*4882a593Smuzhiyun {
590*4882a593Smuzhiyun 	struct jffs2_raw_node_ref *ref;
591*4882a593Smuzhiyun 
592*4882a593Smuzhiyun 	BUG_ON(!jeb->allocated_refs);
593*4882a593Smuzhiyun 	jeb->allocated_refs--;
594*4882a593Smuzhiyun 
595*4882a593Smuzhiyun 	ref = jeb->last_node;
596*4882a593Smuzhiyun 
597*4882a593Smuzhiyun 	dbg_noderef("Last node at %p is (%08x,%p)\n", ref, ref->flash_offset,
598*4882a593Smuzhiyun 		    ref->next_in_ino);
599*4882a593Smuzhiyun 
600*4882a593Smuzhiyun 	while (ref->flash_offset != REF_EMPTY_NODE) {
601*4882a593Smuzhiyun 		if (ref->flash_offset == REF_LINK_NODE)
602*4882a593Smuzhiyun 			ref = ref->next_in_ino;
603*4882a593Smuzhiyun 		else
604*4882a593Smuzhiyun 			ref++;
605*4882a593Smuzhiyun 	}
606*4882a593Smuzhiyun 
607*4882a593Smuzhiyun 	dbg_noderef("New ref is %p (%08x becomes %08x,%p) len 0x%x\n", ref,
608*4882a593Smuzhiyun 		    ref->flash_offset, ofs, ref->next_in_ino, len);
609*4882a593Smuzhiyun 
610*4882a593Smuzhiyun 	ref->flash_offset = ofs;
611*4882a593Smuzhiyun 
612*4882a593Smuzhiyun 	if (!jeb->first_node) {
613*4882a593Smuzhiyun 		jeb->first_node = ref;
614*4882a593Smuzhiyun 		BUG_ON(ref_offset(ref) != jeb->offset);
615*4882a593Smuzhiyun 	} else if (unlikely(ref_offset(ref) != jeb->offset + c->sector_size - jeb->free_size)) {
616*4882a593Smuzhiyun 		uint32_t last_len = ref_totlen(c, jeb, jeb->last_node);
617*4882a593Smuzhiyun 
618*4882a593Smuzhiyun 		JFFS2_ERROR("Adding new ref %p at (0x%08x-0x%08x) not immediately after previous (0x%08x-0x%08x)\n",
619*4882a593Smuzhiyun 			    ref, ref_offset(ref), ref_offset(ref)+len,
620*4882a593Smuzhiyun 			    ref_offset(jeb->last_node),
621*4882a593Smuzhiyun 			    ref_offset(jeb->last_node)+last_len);
622*4882a593Smuzhiyun 		BUG();
623*4882a593Smuzhiyun 	}
624*4882a593Smuzhiyun 	jeb->last_node = ref;
625*4882a593Smuzhiyun 
626*4882a593Smuzhiyun 	if (ic) {
627*4882a593Smuzhiyun 		ref->next_in_ino = ic->nodes;
628*4882a593Smuzhiyun 		ic->nodes = ref;
629*4882a593Smuzhiyun 	} else {
630*4882a593Smuzhiyun 		ref->next_in_ino = NULL;
631*4882a593Smuzhiyun 	}
632*4882a593Smuzhiyun 
633*4882a593Smuzhiyun 	switch(ref_flags(ref)) {
634*4882a593Smuzhiyun 	case REF_UNCHECKED:
635*4882a593Smuzhiyun 		c->unchecked_size += len;
636*4882a593Smuzhiyun 		jeb->unchecked_size += len;
637*4882a593Smuzhiyun 		break;
638*4882a593Smuzhiyun 
639*4882a593Smuzhiyun 	case REF_NORMAL:
640*4882a593Smuzhiyun 	case REF_PRISTINE:
641*4882a593Smuzhiyun 		c->used_size += len;
642*4882a593Smuzhiyun 		jeb->used_size += len;
643*4882a593Smuzhiyun 		break;
644*4882a593Smuzhiyun 
645*4882a593Smuzhiyun 	case REF_OBSOLETE:
646*4882a593Smuzhiyun 		c->dirty_size += len;
647*4882a593Smuzhiyun 		jeb->dirty_size += len;
648*4882a593Smuzhiyun 		break;
649*4882a593Smuzhiyun 	}
650*4882a593Smuzhiyun 	c->free_size -= len;
651*4882a593Smuzhiyun 	jeb->free_size -= len;
652*4882a593Smuzhiyun 
653*4882a593Smuzhiyun #ifdef TEST_TOTLEN
654*4882a593Smuzhiyun 	/* Set (and test) __totlen field... for now */
655*4882a593Smuzhiyun 	ref->__totlen = len;
656*4882a593Smuzhiyun 	ref_totlen(c, jeb, ref);
657*4882a593Smuzhiyun #endif
658*4882a593Smuzhiyun 	return ref;
659*4882a593Smuzhiyun }
660*4882a593Smuzhiyun 
661*4882a593Smuzhiyun /* No locking, no reservation of 'ref'. Do not use on a live file system */
jffs2_scan_dirty_space(struct jffs2_sb_info * c,struct jffs2_eraseblock * jeb,uint32_t size)662*4882a593Smuzhiyun int jffs2_scan_dirty_space(struct jffs2_sb_info *c, struct jffs2_eraseblock *jeb,
663*4882a593Smuzhiyun 			   uint32_t size)
664*4882a593Smuzhiyun {
665*4882a593Smuzhiyun 	if (!size)
666*4882a593Smuzhiyun 		return 0;
667*4882a593Smuzhiyun 	if (unlikely(size > jeb->free_size)) {
668*4882a593Smuzhiyun 		pr_crit("Dirty space 0x%x larger then free_size 0x%x (wasted 0x%x)\n",
669*4882a593Smuzhiyun 			size, jeb->free_size, jeb->wasted_size);
670*4882a593Smuzhiyun 		BUG();
671*4882a593Smuzhiyun 	}
672*4882a593Smuzhiyun 	/* REF_EMPTY_NODE is !obsolete, so that works OK */
673*4882a593Smuzhiyun 	if (jeb->last_node && ref_obsolete(jeb->last_node)) {
674*4882a593Smuzhiyun #ifdef TEST_TOTLEN
675*4882a593Smuzhiyun 		jeb->last_node->__totlen += size;
676*4882a593Smuzhiyun #endif
677*4882a593Smuzhiyun 		c->dirty_size += size;
678*4882a593Smuzhiyun 		c->free_size -= size;
679*4882a593Smuzhiyun 		jeb->dirty_size += size;
680*4882a593Smuzhiyun 		jeb->free_size -= size;
681*4882a593Smuzhiyun 	} else {
682*4882a593Smuzhiyun 		uint32_t ofs = jeb->offset + c->sector_size - jeb->free_size;
683*4882a593Smuzhiyun 		ofs |= REF_OBSOLETE;
684*4882a593Smuzhiyun 
685*4882a593Smuzhiyun 		jffs2_link_node_ref(c, jeb, ofs, size, NULL);
686*4882a593Smuzhiyun 	}
687*4882a593Smuzhiyun 
688*4882a593Smuzhiyun 	return 0;
689*4882a593Smuzhiyun }
690*4882a593Smuzhiyun 
691*4882a593Smuzhiyun /* Calculate totlen from surrounding nodes or eraseblock */
__ref_totlen(struct jffs2_sb_info * c,struct jffs2_eraseblock * jeb,struct jffs2_raw_node_ref * ref)692*4882a593Smuzhiyun static inline uint32_t __ref_totlen(struct jffs2_sb_info *c,
693*4882a593Smuzhiyun 				    struct jffs2_eraseblock *jeb,
694*4882a593Smuzhiyun 				    struct jffs2_raw_node_ref *ref)
695*4882a593Smuzhiyun {
696*4882a593Smuzhiyun 	uint32_t ref_end;
697*4882a593Smuzhiyun 	struct jffs2_raw_node_ref *next_ref = ref_next(ref);
698*4882a593Smuzhiyun 
699*4882a593Smuzhiyun 	if (next_ref)
700*4882a593Smuzhiyun 		ref_end = ref_offset(next_ref);
701*4882a593Smuzhiyun 	else {
702*4882a593Smuzhiyun 		if (!jeb)
703*4882a593Smuzhiyun 			jeb = &c->blocks[ref->flash_offset / c->sector_size];
704*4882a593Smuzhiyun 
705*4882a593Smuzhiyun 		/* Last node in block. Use free_space */
706*4882a593Smuzhiyun 		if (unlikely(ref != jeb->last_node)) {
707*4882a593Smuzhiyun 			pr_crit("ref %p @0x%08x is not jeb->last_node (%p @0x%08x)\n",
708*4882a593Smuzhiyun 				ref, ref_offset(ref), jeb->last_node,
709*4882a593Smuzhiyun 				jeb->last_node ?
710*4882a593Smuzhiyun 				ref_offset(jeb->last_node) : 0);
711*4882a593Smuzhiyun 			BUG();
712*4882a593Smuzhiyun 		}
713*4882a593Smuzhiyun 		ref_end = jeb->offset + c->sector_size - jeb->free_size;
714*4882a593Smuzhiyun 	}
715*4882a593Smuzhiyun 	return ref_end - ref_offset(ref);
716*4882a593Smuzhiyun }
717*4882a593Smuzhiyun 
__jffs2_ref_totlen(struct jffs2_sb_info * c,struct jffs2_eraseblock * jeb,struct jffs2_raw_node_ref * ref)718*4882a593Smuzhiyun uint32_t __jffs2_ref_totlen(struct jffs2_sb_info *c, struct jffs2_eraseblock *jeb,
719*4882a593Smuzhiyun 			    struct jffs2_raw_node_ref *ref)
720*4882a593Smuzhiyun {
721*4882a593Smuzhiyun 	uint32_t ret;
722*4882a593Smuzhiyun 
723*4882a593Smuzhiyun 	ret = __ref_totlen(c, jeb, ref);
724*4882a593Smuzhiyun 
725*4882a593Smuzhiyun #ifdef TEST_TOTLEN
726*4882a593Smuzhiyun 	if (unlikely(ret != ref->__totlen)) {
727*4882a593Smuzhiyun 		if (!jeb)
728*4882a593Smuzhiyun 			jeb = &c->blocks[ref->flash_offset / c->sector_size];
729*4882a593Smuzhiyun 
730*4882a593Smuzhiyun 		pr_crit("Totlen for ref at %p (0x%08x-0x%08x) miscalculated as 0x%x instead of %x\n",
731*4882a593Smuzhiyun 			ref, ref_offset(ref), ref_offset(ref) + ref->__totlen,
732*4882a593Smuzhiyun 			ret, ref->__totlen);
733*4882a593Smuzhiyun 		if (ref_next(ref)) {
734*4882a593Smuzhiyun 			pr_crit("next %p (0x%08x-0x%08x)\n",
735*4882a593Smuzhiyun 				ref_next(ref), ref_offset(ref_next(ref)),
736*4882a593Smuzhiyun 				ref_offset(ref_next(ref)) + ref->__totlen);
737*4882a593Smuzhiyun 		} else
738*4882a593Smuzhiyun 			pr_crit("No next ref. jeb->last_node is %p\n",
739*4882a593Smuzhiyun 				jeb->last_node);
740*4882a593Smuzhiyun 
741*4882a593Smuzhiyun 		pr_crit("jeb->wasted_size %x, dirty_size %x, used_size %x, free_size %x\n",
742*4882a593Smuzhiyun 			jeb->wasted_size, jeb->dirty_size, jeb->used_size,
743*4882a593Smuzhiyun 			jeb->free_size);
744*4882a593Smuzhiyun 
745*4882a593Smuzhiyun #if defined(JFFS2_DBG_DUMPS) || defined(JFFS2_DBG_PARANOIA_CHECKS)
746*4882a593Smuzhiyun 		__jffs2_dbg_dump_node_refs_nolock(c, jeb);
747*4882a593Smuzhiyun #endif
748*4882a593Smuzhiyun 
749*4882a593Smuzhiyun 		WARN_ON(1);
750*4882a593Smuzhiyun 
751*4882a593Smuzhiyun 		ret = ref->__totlen;
752*4882a593Smuzhiyun 	}
753*4882a593Smuzhiyun #endif /* TEST_TOTLEN */
754*4882a593Smuzhiyun 	return ret;
755*4882a593Smuzhiyun }
756