1 #include <common.h>
2 
3 #include <malloc.h>
4 #include <linux/stat.h>
5 #include <linux/time.h>
6 
7 #include <jffs2/jffs2.h>
8 #include <jffs2/jffs2_1pass.h>
9 #include <nand.h>
10 
11 #include "jffs2_nand_private.h"
12 
13 #define	NODE_CHUNK	1024	/* size of memory allocation chunk in b_nodes */
14 
15 /* Debugging switches */
16 #undef	DEBUG_DIRENTS		/* print directory entry list after scan */
17 #undef	DEBUG_FRAGMENTS		/* print fragment list after scan */
18 #undef	DEBUG			/* enable debugging messages */
19 
20 #ifdef  DEBUG
21 # define DEBUGF(fmt,args...)	printf(fmt ,##args)
22 #else
23 # define DEBUGF(fmt,args...)
24 #endif
25 
26 static nand_info_t *nand;
27 
28 /* Compression names */
29 static char *compr_names[] = {
30 	"NONE",
31 	"ZERO",
32 	"RTIME",
33 	"RUBINMIPS",
34 	"COPY",
35 	"DYNRUBIN",
36 	"ZLIB",
37 #if defined(CONFIG_JFFS2_LZO)
38 	"LZO",
39 #endif
40 };
41 
42 /* Spinning wheel */
43 static char spinner[] = { '|', '/', '-', '\\' };
44 
45 /* Memory management */
46 struct mem_block {
47 	unsigned index;
48 	struct mem_block *next;
49 	char nodes[0];
50 };
51 
52 static void
free_nodes(struct b_list * list)53 free_nodes(struct b_list *list)
54 {
55 	while (list->listMemBase != NULL) {
56 		struct mem_block *next = list->listMemBase->next;
57 		free(list->listMemBase);
58 		list->listMemBase = next;
59 	}
60 }
61 
62 static struct b_node *
add_node(struct b_list * list,int size)63 add_node(struct b_list *list, int size)
64 {
65 	u32 index = 0;
66 	struct mem_block *memBase;
67 	struct b_node *b;
68 
69 	memBase = list->listMemBase;
70 	if (memBase != NULL)
71 		index = memBase->index;
72 
73 	if (memBase == NULL || index >= NODE_CHUNK) {
74 		/* we need more space before we continue */
75 		memBase = mmalloc(sizeof(struct mem_block) + NODE_CHUNK * size);
76 		if (memBase == NULL) {
77 			putstr("add_node: malloc failed\n");
78 			return NULL;
79 		}
80 		memBase->next = list->listMemBase;
81 		index = 0;
82 	}
83 	/* now we have room to add it. */
84 	b = (struct b_node *)&memBase->nodes[size * index];
85 	index ++;
86 
87 	memBase->index = index;
88 	list->listMemBase = memBase;
89 	list->listCount++;
90 	return b;
91 }
92 
93 static struct b_node *
insert_node(struct b_list * list,struct b_node * new)94 insert_node(struct b_list *list, struct b_node *new)
95 {
96 #ifdef CONFIG_SYS_JFFS2_SORT_FRAGMENTS
97 	struct b_node *b, *prev;
98 
99 	if (list->listTail != NULL && list->listCompare(new, list->listTail))
100 		prev = list->listTail;
101 	else if (list->listLast != NULL && list->listCompare(new, list->listLast))
102 		prev = list->listLast;
103 	else
104 		prev = NULL;
105 
106 	for (b = (prev ? prev->next : list->listHead);
107 	     b != NULL && list->listCompare(new, b);
108 	     prev = b, b = b->next) {
109 		list->listLoops++;
110 	}
111 	if (b != NULL)
112 		list->listLast = prev;
113 
114 	if (b != NULL) {
115 		new->next = b;
116 		if (prev != NULL)
117 			prev->next = new;
118 		else
119 			list->listHead = new;
120 	} else
121 #endif
122 	{
123 		new->next = (struct b_node *) NULL;
124 		if (list->listTail != NULL) {
125 			list->listTail->next = new;
126 			list->listTail = new;
127 		} else {
128 			list->listTail = list->listHead = new;
129 		}
130 	}
131 
132 	return new;
133 }
134 
135 static struct b_node *
insert_inode(struct b_list * list,struct jffs2_raw_inode * node,u32 offset)136 insert_inode(struct b_list *list, struct jffs2_raw_inode *node, u32 offset)
137 {
138 	struct b_inode *new;
139 
140 	if (!(new = (struct b_inode *)add_node(list, sizeof(struct b_inode)))) {
141 		putstr("add_node failed!\r\n");
142 		return NULL;
143 	}
144 	new->offset = offset;
145 	new->version = node->version;
146 	new->ino = node->ino;
147 	new->isize = node->isize;
148 	new->csize = node->csize;
149 
150 	return insert_node(list, (struct b_node *)new);
151 }
152 
153 static struct b_node *
insert_dirent(struct b_list * list,struct jffs2_raw_dirent * node,u32 offset)154 insert_dirent(struct b_list *list, struct jffs2_raw_dirent *node, u32 offset)
155 {
156 	struct b_dirent *new;
157 
158 	if (!(new = (struct b_dirent *)add_node(list, sizeof(struct b_dirent)))) {
159 		putstr("add_node failed!\r\n");
160 		return NULL;
161 	}
162 	new->offset = offset;
163 	new->version = node->version;
164 	new->pino = node->pino;
165 	new->ino = node->ino;
166 	new->nhash = full_name_hash(node->name, node->nsize);
167 	new->nsize = node->nsize;
168 	new->type = node->type;
169 
170 	return insert_node(list, (struct b_node *)new);
171 }
172 
173 #ifdef CONFIG_SYS_JFFS2_SORT_FRAGMENTS
174 /* Sort data entries with the latest version last, so that if there
175  * is overlapping data the latest version will be used.
176  */
compare_inodes(struct b_node * new,struct b_node * old)177 static int compare_inodes(struct b_node *new, struct b_node *old)
178 {
179 	struct jffs2_raw_inode ojNew;
180 	struct jffs2_raw_inode ojOld;
181 	struct jffs2_raw_inode *jNew =
182 		(struct jffs2_raw_inode *)get_fl_mem(new->offset, sizeof(ojNew), &ojNew);
183 	struct jffs2_raw_inode *jOld =
184 		(struct jffs2_raw_inode *)get_fl_mem(old->offset, sizeof(ojOld), &ojOld);
185 
186 	return jNew->version > jOld->version;
187 }
188 
189 /* Sort directory entries so all entries in the same directory
190  * with the same name are grouped together, with the latest version
191  * last. This makes it easy to eliminate all but the latest version
192  * by marking the previous version dead by setting the inode to 0.
193  */
compare_dirents(struct b_node * new,struct b_node * old)194 static int compare_dirents(struct b_node *new, struct b_node *old)
195 {
196 	struct jffs2_raw_dirent ojNew;
197 	struct jffs2_raw_dirent ojOld;
198 	struct jffs2_raw_dirent *jNew =
199 		(struct jffs2_raw_dirent *)get_fl_mem(new->offset, sizeof(ojNew), &ojNew);
200 	struct jffs2_raw_dirent *jOld =
201 		(struct jffs2_raw_dirent *)get_fl_mem(old->offset, sizeof(ojOld), &ojOld);
202 	int cmp;
203 
204 	/* ascending sort by pino */
205 	if (jNew->pino != jOld->pino)
206 		return jNew->pino > jOld->pino;
207 
208 	/* pino is the same, so use ascending sort by nsize, so
209 	 * we don't do strncmp unless we really must.
210 	 */
211 	if (jNew->nsize != jOld->nsize)
212 		return jNew->nsize > jOld->nsize;
213 
214 	/* length is also the same, so use ascending sort by name
215 	 */
216 	cmp = strncmp(jNew->name, jOld->name, jNew->nsize);
217 	if (cmp != 0)
218 		return cmp > 0;
219 
220 	/* we have duplicate names in this directory, so use ascending
221 	 * sort by version
222 	 */
223 	if (jNew->version > jOld->version) {
224 		/* since jNew is newer, we know jOld is not valid, so
225 		 * mark it with inode 0 and it will not be used
226 		 */
227 		jOld->ino = 0;
228 		return 1;
229 	}
230 
231 	return 0;
232 }
233 #endif
234 
235 static u32
jffs_init_1pass_list(struct part_info * part)236 jffs_init_1pass_list(struct part_info *part)
237 {
238 	struct b_lists *pL;
239 
240 	if (part->jffs2_priv != NULL) {
241 		pL = (struct b_lists *)part->jffs2_priv;
242 		free_nodes(&pL->frag);
243 		free_nodes(&pL->dir);
244 		free(pL);
245 	}
246 	if (NULL != (part->jffs2_priv = malloc(sizeof(struct b_lists)))) {
247 		pL = (struct b_lists *)part->jffs2_priv;
248 
249 		memset(pL, 0, sizeof(*pL));
250 #ifdef CONFIG_SYS_JFFS2_SORT_FRAGMENTS
251 		pL->dir.listCompare = compare_dirents;
252 		pL->frag.listCompare = compare_inodes;
253 #endif
254 	}
255 	return 0;
256 }
257 
258 /* find the inode from the slashless name given a parent */
259 static long
jffs2_1pass_read_inode(struct b_lists * pL,u32 ino,char * dest,struct stat * stat)260 jffs2_1pass_read_inode(struct b_lists *pL, u32 ino, char *dest,
261 		       struct stat *stat)
262 {
263 	struct b_inode *jNode;
264 	u32 totalSize = 0;
265 	u32 latestVersion = 0;
266 	long ret;
267 
268 #ifdef CONFIG_SYS_JFFS2_SORT_FRAGMENTS
269 	/* Find file size before loading any data, so fragments that
270 	 * start past the end of file can be ignored. A fragment
271 	 * that is partially in the file is loaded, so extra data may
272 	 * be loaded up to the next 4K boundary above the file size.
273 	 * This shouldn't cause trouble when loading kernel images, so
274 	 * we will live with it.
275 	 */
276 	for (jNode = (struct b_inode *)pL->frag.listHead; jNode; jNode = jNode->next) {
277 		if ((ino == jNode->ino)) {
278 			/* get actual file length from the newest node */
279 			if (jNode->version >= latestVersion) {
280 				totalSize = jNode->isize;
281 				latestVersion = jNode->version;
282 			}
283 		}
284 	}
285 #endif
286 
287 	for (jNode = (struct b_inode *)pL->frag.listHead; jNode; jNode = jNode->next) {
288 		if ((ino != jNode->ino))
289 			continue;
290 #ifndef CONFIG_SYS_JFFS2_SORT_FRAGMENTS
291 		/* get actual file length from the newest node */
292 		if (jNode->version >= latestVersion) {
293 			totalSize = jNode->isize;
294 			latestVersion = jNode->version;
295 		}
296 #endif
297 		if (dest || stat) {
298 			char *src, *dst;
299 			char data[4096 + sizeof(struct jffs2_raw_inode)];
300 			struct jffs2_raw_inode *inode;
301 			size_t len;
302 
303 			inode = (struct jffs2_raw_inode *)&data;
304 			len = sizeof(struct jffs2_raw_inode);
305 			if (dest)
306 				len += jNode->csize;
307 			nand_read(nand, jNode->offset, &len, inode);
308 			/* ignore data behind latest known EOF */
309 			if (inode->offset > totalSize)
310 				continue;
311 
312 			if (stat) {
313 				stat->st_mtime = inode->mtime;
314 				stat->st_mode = inode->mode;
315 				stat->st_ino = inode->ino;
316 				stat->st_size = totalSize;
317 			}
318 
319 			if (!dest)
320 				continue;
321 
322 			src = ((char *) inode) + sizeof(struct jffs2_raw_inode);
323 			dst = (char *) (dest + inode->offset);
324 
325 			switch (inode->compr) {
326 			case JFFS2_COMPR_NONE:
327 				ret = 0;
328 				memcpy(dst, src, inode->dsize);
329 				break;
330 			case JFFS2_COMPR_ZERO:
331 				ret = 0;
332 				memset(dst, 0, inode->dsize);
333 				break;
334 			case JFFS2_COMPR_RTIME:
335 				ret = 0;
336 				rtime_decompress(src, dst, inode->csize, inode->dsize);
337 				break;
338 			case JFFS2_COMPR_DYNRUBIN:
339 				/* this is slow but it works */
340 				ret = 0;
341 				dynrubin_decompress(src, dst, inode->csize, inode->dsize);
342 				break;
343 			case JFFS2_COMPR_ZLIB:
344 				ret = zlib_decompress(src, dst, inode->csize, inode->dsize);
345 				break;
346 #if defined(CONFIG_JFFS2_LZO)
347 			case JFFS2_COMPR_LZO:
348 				ret = lzo_decompress(src, dst, inode->csize, inode->dsize);
349 				break;
350 #endif
351 			default:
352 				/* unknown */
353 				putLabeledWord("UNKOWN COMPRESSION METHOD = ", inode->compr);
354 				return -1;
355 			}
356 		}
357 	}
358 
359 	return totalSize;
360 }
361 
362 /* find the inode from the slashless name given a parent */
363 static u32
jffs2_1pass_find_inode(struct b_lists * pL,const char * name,u32 pino)364 jffs2_1pass_find_inode(struct b_lists * pL, const char *name, u32 pino)
365 {
366 	struct b_dirent *jDir;
367 	int len = strlen(name);	/* name is assumed slash free */
368 	unsigned int nhash = full_name_hash(name, len);
369 	u32 version = 0;
370 	u32 inode = 0;
371 
372 	/* we need to search all and return the inode with the highest version */
373 	for (jDir = (struct b_dirent *)pL->dir.listHead; jDir; jDir = jDir->next) {
374 		if ((pino == jDir->pino) && (jDir->ino) &&	/* 0 for unlink */
375 		    (len == jDir->nsize) && (nhash == jDir->nhash)) {
376 			/* TODO: compare name */
377 			if (jDir->version < version)
378 				continue;
379 
380 			if (jDir->version == version && inode != 0) {
381 				/* I'm pretty sure this isn't legal */
382 				putstr(" ** ERROR ** ");
383 /*				putnstr(jDir->name, jDir->nsize); */
384 /*				putLabeledWord(" has dup version =", version); */
385 			}
386 			inode = jDir->ino;
387 			version = jDir->version;
388 		}
389 	}
390 	return inode;
391 }
392 
mkmodestr(unsigned long mode,char * str)393 char *mkmodestr(unsigned long mode, char *str)
394 {
395 	static const char *l = "xwr";
396 	int mask = 1, i;
397 	char c;
398 
399 	switch (mode & S_IFMT) {
400 		case S_IFDIR:    str[0] = 'd'; break;
401 		case S_IFBLK:    str[0] = 'b'; break;
402 		case S_IFCHR:    str[0] = 'c'; break;
403 		case S_IFIFO:    str[0] = 'f'; break;
404 		case S_IFLNK:    str[0] = 'l'; break;
405 		case S_IFSOCK:   str[0] = 's'; break;
406 		case S_IFREG:    str[0] = '-'; break;
407 		default:         str[0] = '?';
408 	}
409 
410 	for(i = 0; i < 9; i++) {
411 		c = l[i%3];
412 		str[9-i] = (mode & mask)?c:'-';
413 		mask = mask<<1;
414 	}
415 
416 	if(mode & S_ISUID) str[3] = (mode & S_IXUSR)?'s':'S';
417 	if(mode & S_ISGID) str[6] = (mode & S_IXGRP)?'s':'S';
418 	if(mode & S_ISVTX) str[9] = (mode & S_IXOTH)?'t':'T';
419 	str[10] = '\0';
420 	return str;
421 }
422 
dump_stat(struct stat * st,const char * name)423 static inline void dump_stat(struct stat *st, const char *name)
424 {
425 	char str[20];
426 	char s[64], *p;
427 
428 	if (st->st_mtime == (time_t)(-1)) /* some ctimes really hate -1 */
429 		st->st_mtime = 1;
430 
431 	ctime_r(&st->st_mtime, s/*,64*/); /* newlib ctime doesn't have buflen */
432 
433 	if ((p = strchr(s,'\n')) != NULL) *p = '\0';
434 	if ((p = strchr(s,'\r')) != NULL) *p = '\0';
435 
436 /*
437 	printf("%6lo %s %8ld %s %s\n", st->st_mode, mkmodestr(st->st_mode, str),
438 		st->st_size, s, name);
439 */
440 
441 	printf(" %s %8ld %s %s", mkmodestr(st->st_mode,str), st->st_size, s, name);
442 }
443 
444 static inline int
dump_inode(struct b_lists * pL,struct b_dirent * d,struct b_inode * i)445 dump_inode(struct b_lists *pL, struct b_dirent *d, struct b_inode *i)
446 {
447 	char fname[JFFS2_MAX_NAME_LEN + 1];
448 	struct stat st;
449 	size_t len;
450 
451 	if(!d || !i) return -1;
452 	len = d->nsize;
453 	nand_read(nand, d->offset + sizeof(struct jffs2_raw_dirent),
454 		  &len, &fname);
455 	fname[d->nsize] = '\0';
456 
457 	memset(&st, 0, sizeof(st));
458 
459 	jffs2_1pass_read_inode(pL, i->ino, NULL, &st);
460 
461 	dump_stat(&st, fname);
462 /* FIXME
463 	if (d->type == DT_LNK) {
464 		unsigned char *src = (unsigned char *) (&i[1]);
465 	        putstr(" -> ");
466 		putnstr(src, (int)i->dsize);
467 	}
468 */
469 	putstr("\r\n");
470 
471 	return 0;
472 }
473 
474 /* list inodes with the given pino */
475 static u32
jffs2_1pass_list_inodes(struct b_lists * pL,u32 pino)476 jffs2_1pass_list_inodes(struct b_lists * pL, u32 pino)
477 {
478 	struct b_dirent *jDir;
479 	u32 i_version = 0;
480 
481 	for (jDir = (struct b_dirent *)pL->dir.listHead; jDir; jDir = jDir->next) {
482 		if ((pino == jDir->pino) && (jDir->ino)) { /* ino=0 -> unlink */
483 			struct b_inode *jNode = (struct b_inode *)pL->frag.listHead;
484 			struct b_inode *i = NULL;
485 
486 			while (jNode) {
487 				if (jNode->ino == jDir->ino && jNode->version >= i_version) {
488 					i_version = jNode->version;
489 					i = jNode;
490 				}
491 				jNode = jNode->next;
492 			}
493 			dump_inode(pL, jDir, i);
494 		}
495 	}
496 	return pino;
497 }
498 
499 static u32
jffs2_1pass_search_inode(struct b_lists * pL,const char * fname,u32 pino)500 jffs2_1pass_search_inode(struct b_lists * pL, const char *fname, u32 pino)
501 {
502 	int i;
503 	char tmp[256];
504 	char working_tmp[256];
505 	char *c;
506 
507 	/* discard any leading slash */
508 	i = 0;
509 	while (fname[i] == '/')
510 		i++;
511 	strcpy(tmp, &fname[i]);
512 
513 	while ((c = (char *) strchr(tmp, '/')))	/* we are still dired searching */
514 	{
515 		strncpy(working_tmp, tmp, c - tmp);
516 		working_tmp[c - tmp] = '\0';
517 #if 0
518 		putstr("search_inode: tmp = ");
519 		putstr(tmp);
520 		putstr("\r\n");
521 		putstr("search_inode: wtmp = ");
522 		putstr(working_tmp);
523 		putstr("\r\n");
524 		putstr("search_inode: c = ");
525 		putstr(c);
526 		putstr("\r\n");
527 #endif
528 		for (i = 0; i < strlen(c) - 1; i++)
529 			tmp[i] = c[i + 1];
530 		tmp[i] = '\0';
531 #if 0
532 		putstr("search_inode: post tmp = ");
533 		putstr(tmp);
534 		putstr("\r\n");
535 #endif
536 
537 		if (!(pino = jffs2_1pass_find_inode(pL, working_tmp, pino))) {
538 			putstr("find_inode failed for name=");
539 			putstr(working_tmp);
540 			putstr("\r\n");
541 			return 0;
542 		}
543 	}
544 	/* this is for the bare filename, directories have already been mapped */
545 	if (!(pino = jffs2_1pass_find_inode(pL, tmp, pino))) {
546 		putstr("find_inode failed for name=");
547 		putstr(tmp);
548 		putstr("\r\n");
549 		return 0;
550 	}
551 	return pino;
552 
553 }
554 
555 static u32
jffs2_1pass_resolve_inode(struct b_lists * pL,u32 ino)556 jffs2_1pass_resolve_inode(struct b_lists * pL, u32 ino)
557 {
558 	struct b_dirent *jDir;
559 	struct b_inode *jNode;
560 	u8 jDirFoundType = 0;
561 	u32 jDirFoundIno = 0;
562 	u32 jDirFoundPino = 0;
563 	char tmp[JFFS2_MAX_NAME_LEN + 1];
564 	u32 version = 0;
565 	u32 pino;
566 
567 	/* we need to search all and return the inode with the highest version */
568 	for (jDir = (struct b_dirent *)pL->dir.listHead; jDir; jDir = jDir->next) {
569 		if (ino == jDir->ino) {
570 			if (jDir->version < version)
571 				continue;
572 
573 			if (jDir->version == version && jDirFoundType) {
574 				/* I'm pretty sure this isn't legal */
575 				putstr(" ** ERROR ** ");
576 /*				putnstr(jDir->name, jDir->nsize); */
577 /*				putLabeledWord(" has dup version (resolve) = ", */
578 /*					version); */
579 			}
580 
581 			jDirFoundType = jDir->type;
582 			jDirFoundIno = jDir->ino;
583 			jDirFoundPino = jDir->pino;
584 			version = jDir->version;
585 		}
586 	}
587 	/* now we found the right entry again. (shoulda returned inode*) */
588 	if (jDirFoundType != DT_LNK)
589 		return jDirFoundIno;
590 
591 	/* it's a soft link so we follow it again. */
592 	for (jNode = (struct b_inode *)pL->frag.listHead; jNode; jNode = jNode->next) {
593 		if (jNode->ino == jDirFoundIno) {
594 			size_t len = jNode->csize;
595 			nand_read(nand, jNode->offset + sizeof(struct jffs2_raw_inode), &len, &tmp);
596 			tmp[jNode->csize] = '\0';
597 			break;
598 		}
599 	}
600 	/* ok so the name of the new file to find is in tmp */
601 	/* if it starts with a slash it is root based else shared dirs */
602 	if (tmp[0] == '/')
603 		pino = 1;
604 	else
605 		pino = jDirFoundPino;
606 
607 	return jffs2_1pass_search_inode(pL, tmp, pino);
608 }
609 
610 static u32
jffs2_1pass_search_list_inodes(struct b_lists * pL,const char * fname,u32 pino)611 jffs2_1pass_search_list_inodes(struct b_lists * pL, const char *fname, u32 pino)
612 {
613 	int i;
614 	char tmp[256];
615 	char working_tmp[256];
616 	char *c;
617 
618 	/* discard any leading slash */
619 	i = 0;
620 	while (fname[i] == '/')
621 		i++;
622 	strcpy(tmp, &fname[i]);
623 	working_tmp[0] = '\0';
624 	while ((c = (char *) strchr(tmp, '/')))	/* we are still dired searching */
625 	{
626 		strncpy(working_tmp, tmp, c - tmp);
627 		working_tmp[c - tmp] = '\0';
628 		for (i = 0; i < strlen(c) - 1; i++)
629 			tmp[i] = c[i + 1];
630 		tmp[i] = '\0';
631 		/* only a failure if we arent looking at top level */
632 		if (!(pino = jffs2_1pass_find_inode(pL, working_tmp, pino)) &&
633 		    (working_tmp[0])) {
634 			putstr("find_inode failed for name=");
635 			putstr(working_tmp);
636 			putstr("\r\n");
637 			return 0;
638 		}
639 	}
640 
641 	if (tmp[0] && !(pino = jffs2_1pass_find_inode(pL, tmp, pino))) {
642 		putstr("find_inode failed for name=");
643 		putstr(tmp);
644 		putstr("\r\n");
645 		return 0;
646 	}
647 	/* this is for the bare filename, directories have already been mapped */
648 	if (!(pino = jffs2_1pass_list_inodes(pL, pino))) {
649 		putstr("find_inode failed for name=");
650 		putstr(tmp);
651 		putstr("\r\n");
652 		return 0;
653 	}
654 	return pino;
655 
656 }
657 
658 unsigned char
jffs2_1pass_rescan_needed(struct part_info * part)659 jffs2_1pass_rescan_needed(struct part_info *part)
660 {
661 	struct b_node *b;
662 	struct jffs2_unknown_node onode;
663 	struct jffs2_unknown_node *node;
664 	struct b_lists *pL = (struct b_lists *)part->jffs2_priv;
665 
666 	if (part->jffs2_priv == 0){
667 		DEBUGF ("rescan: First time in use\n");
668 		return 1;
669 	}
670 	/* if we have no list, we need to rescan */
671 	if (pL->frag.listCount == 0) {
672 		DEBUGF ("rescan: fraglist zero\n");
673 		return 1;
674 	}
675 
676 	/* or if we are scanning a new partition */
677 	if (pL->partOffset != part->offset) {
678 		DEBUGF ("rescan: different partition\n");
679 		return 1;
680 	}
681 
682 	/* FIXME */
683 #if 0
684 	/* but suppose someone reflashed a partition at the same offset... */
685 	b = pL->dir.listHead;
686 	while (b) {
687 		node = (struct jffs2_unknown_node *) get_fl_mem(b->offset,
688 			sizeof(onode), &onode);
689 		if (node->nodetype != JFFS2_NODETYPE_DIRENT) {
690 			DEBUGF ("rescan: fs changed beneath me? (%lx)\n",
691 					(unsigned long) b->offset);
692 			return 1;
693 		}
694 		b = b->next;
695 	}
696 #endif
697 	return 0;
698 }
699 
700 #ifdef DEBUG_FRAGMENTS
701 static void
dump_fragments(struct b_lists * pL)702 dump_fragments(struct b_lists *pL)
703 {
704 	struct b_node *b;
705 	struct jffs2_raw_inode ojNode;
706 	struct jffs2_raw_inode *jNode;
707 
708 	putstr("\r\n\r\n******The fragment Entries******\r\n");
709 	b = pL->frag.listHead;
710 	while (b) {
711 		jNode = (struct jffs2_raw_inode *) get_fl_mem(b->offset,
712 			sizeof(ojNode), &ojNode);
713 		putLabeledWord("\r\n\tbuild_list: FLASH_OFFSET = ", b->offset);
714 		putLabeledWord("\tbuild_list: totlen = ", jNode->totlen);
715 		putLabeledWord("\tbuild_list: inode = ", jNode->ino);
716 		putLabeledWord("\tbuild_list: version = ", jNode->version);
717 		putLabeledWord("\tbuild_list: isize = ", jNode->isize);
718 		putLabeledWord("\tbuild_list: atime = ", jNode->atime);
719 		putLabeledWord("\tbuild_list: offset = ", jNode->offset);
720 		putLabeledWord("\tbuild_list: csize = ", jNode->csize);
721 		putLabeledWord("\tbuild_list: dsize = ", jNode->dsize);
722 		putLabeledWord("\tbuild_list: compr = ", jNode->compr);
723 		putLabeledWord("\tbuild_list: usercompr = ", jNode->usercompr);
724 		putLabeledWord("\tbuild_list: flags = ", jNode->flags);
725 		putLabeledWord("\tbuild_list: offset = ", b->offset);	/* FIXME: ? [RS] */
726 		b = b->next;
727 	}
728 }
729 #endif
730 
731 #ifdef DEBUG_DIRENTS
732 static void
dump_dirents(struct b_lists * pL)733 dump_dirents(struct b_lists *pL)
734 {
735 	struct b_node *b;
736 	struct jffs2_raw_dirent *jDir;
737 
738 	putstr("\r\n\r\n******The directory Entries******\r\n");
739 	b = pL->dir.listHead;
740 	while (b) {
741 		jDir = (struct jffs2_raw_dirent *) get_node_mem(b->offset);
742 		putstr("\r\n");
743 		putnstr(jDir->name, jDir->nsize);
744 		putLabeledWord("\r\n\tbuild_list: magic = ", jDir->magic);
745 		putLabeledWord("\tbuild_list: nodetype = ", jDir->nodetype);
746 		putLabeledWord("\tbuild_list: hdr_crc = ", jDir->hdr_crc);
747 		putLabeledWord("\tbuild_list: pino = ", jDir->pino);
748 		putLabeledWord("\tbuild_list: version = ", jDir->version);
749 		putLabeledWord("\tbuild_list: ino = ", jDir->ino);
750 		putLabeledWord("\tbuild_list: mctime = ", jDir->mctime);
751 		putLabeledWord("\tbuild_list: nsize = ", jDir->nsize);
752 		putLabeledWord("\tbuild_list: type = ", jDir->type);
753 		putLabeledWord("\tbuild_list: node_crc = ", jDir->node_crc);
754 		putLabeledWord("\tbuild_list: name_crc = ", jDir->name_crc);
755 		putLabeledWord("\tbuild_list: offset = ", b->offset);	/* FIXME: ? [RS] */
756 		b = b->next;
757 		put_fl_mem(jDir);
758 	}
759 }
760 #endif
761 
762 static int
jffs2_fill_scan_buf(nand_info_t * nand,unsigned char * buf,unsigned ofs,unsigned len)763 jffs2_fill_scan_buf(nand_info_t *nand, unsigned char *buf,
764 		    unsigned ofs, unsigned len)
765 {
766 	int ret;
767 	unsigned olen;
768 
769 	olen = len;
770 	ret = nand_read(nand, ofs, &olen, buf);
771 	if (ret) {
772 		printf("nand_read(0x%x bytes from 0x%x) returned %d\n", len, ofs, ret);
773 		return ret;
774 	}
775 	if (olen < len) {
776 		printf("Read at 0x%x gave only 0x%x bytes\n", ofs, olen);
777 		return -1;
778 	}
779 	return 0;
780 }
781 
782 #define	EMPTY_SCAN_SIZE	1024
783 static u32
jffs2_1pass_build_lists(struct part_info * part)784 jffs2_1pass_build_lists(struct part_info * part)
785 {
786 	struct b_lists *pL;
787 	struct jffs2_unknown_node *node;
788 	unsigned nr_blocks, sectorsize, ofs, offset;
789 	char *buf;
790 	int i;
791 	u32 counter = 0;
792 	u32 counter4 = 0;
793 	u32 counterF = 0;
794 	u32 counterN = 0;
795 
796 	struct mtdids *id = part->dev->id;
797 	nand = nand_info + id->num;
798 
799 	/* if we are building a list we need to refresh the cache. */
800 	jffs_init_1pass_list(part);
801 	pL = (struct b_lists *)part->jffs2_priv;
802 	pL->partOffset = part->offset;
803 	puts ("Scanning JFFS2 FS:   ");
804 
805 	sectorsize = nand->erasesize;
806 	nr_blocks = part->size / sectorsize;
807 	buf = malloc(sectorsize);
808 	if (!buf)
809 		return 0;
810 
811 	for (i = 0; i < nr_blocks; i++) {
812 		printf("\b\b%c ", spinner[counter++ % sizeof(spinner)]);
813 
814 		offset = part->offset + i * sectorsize;
815 
816 		if (nand_block_isbad(nand, offset))
817 			continue;
818 
819 		if (jffs2_fill_scan_buf(nand, buf, offset, EMPTY_SCAN_SIZE))
820 			return 0;
821 
822 		ofs = 0;
823 		/* Scan only 4KiB of 0xFF before declaring it's empty */
824 		while (ofs < EMPTY_SCAN_SIZE && *(uint32_t *)(&buf[ofs]) == 0xFFFFFFFF)
825 			ofs += 4;
826 		if (ofs == EMPTY_SCAN_SIZE)
827 			continue;
828 
829 		if (jffs2_fill_scan_buf(nand, buf + EMPTY_SCAN_SIZE, offset + EMPTY_SCAN_SIZE, sectorsize - EMPTY_SCAN_SIZE))
830 			return 0;
831 		offset += ofs;
832 
833 		while (ofs < sectorsize - sizeof(struct jffs2_unknown_node)) {
834 			node = (struct jffs2_unknown_node *)&buf[ofs];
835 			if (node->magic != JFFS2_MAGIC_BITMASK || !hdr_crc(node)) {
836 				offset += 4;
837 				ofs += 4;
838 				counter4++;
839 				continue;
840 			}
841 			/* if its a fragment add it */
842 			if (node->nodetype == JFFS2_NODETYPE_INODE &&
843 				    inode_crc((struct jffs2_raw_inode *) node)) {
844 				if (insert_inode(&pL->frag, (struct jffs2_raw_inode *) node,
845 						 offset) == NULL) {
846 					return 0;
847 				}
848 			} else if (node->nodetype == JFFS2_NODETYPE_DIRENT &&
849 				   dirent_crc((struct jffs2_raw_dirent *) node)  &&
850 				   dirent_name_crc((struct jffs2_raw_dirent *) node)) {
851 				if (! (counterN%100))
852 					puts ("\b\b.  ");
853 				if (insert_dirent(&pL->dir, (struct jffs2_raw_dirent *) node,
854 						  offset) == NULL) {
855 					return 0;
856 				}
857 				counterN++;
858 			} else if (node->nodetype == JFFS2_NODETYPE_CLEANMARKER) {
859 				if (node->totlen != sizeof(struct jffs2_unknown_node))
860 					printf("OOPS Cleanmarker has bad size "
861 						"%d != %zu\n",
862 						node->totlen,
863 						sizeof(struct jffs2_unknown_node));
864 			} else if (node->nodetype == JFFS2_NODETYPE_PADDING) {
865 				if (node->totlen < sizeof(struct jffs2_unknown_node))
866 					printf("OOPS Padding has bad size "
867 						"%d < %zu\n",
868 						node->totlen,
869 						sizeof(struct jffs2_unknown_node));
870 			} else {
871 				printf("Unknown node type: %x len %d offset 0x%x\n",
872 					node->nodetype,
873 					node->totlen, offset);
874 			}
875 			offset += ((node->totlen + 3) & ~3);
876 			ofs += ((node->totlen + 3) & ~3);
877 			counterF++;
878 		}
879 	}
880 
881 	putstr("\b\b done.\r\n");		/* close off the dots */
882 
883 #if 0
884 	putLabeledWord("dir entries = ", pL->dir.listCount);
885 	putLabeledWord("frag entries = ", pL->frag.listCount);
886 	putLabeledWord("+4 increments = ", counter4);
887 	putLabeledWord("+file_offset increments = ", counterF);
888 #endif
889 
890 #ifdef DEBUG_DIRENTS
891 	dump_dirents(pL);
892 #endif
893 
894 #ifdef DEBUG_FRAGMENTS
895 	dump_fragments(pL);
896 #endif
897 
898 	/* give visual feedback that we are done scanning the flash */
899 	led_blink(0x0, 0x0, 0x1, 0x1);	/* off, forever, on 100ms, off 100ms */
900 	free(buf);
901 
902 	return 1;
903 }
904 
905 
906 static u32
jffs2_1pass_fill_info(struct b_lists * pL,struct b_jffs2_info * piL)907 jffs2_1pass_fill_info(struct b_lists * pL, struct b_jffs2_info * piL)
908 {
909 	struct b_node *b;
910 	struct jffs2_raw_inode ojNode;
911 	struct jffs2_raw_inode *jNode;
912 	int i;
913 
914 	for (i = 0; i < JFFS2_NUM_COMPR; i++) {
915 		piL->compr_info[i].num_frags = 0;
916 		piL->compr_info[i].compr_sum = 0;
917 		piL->compr_info[i].decompr_sum = 0;
918 	}
919 /*	FIXME
920 	b = pL->frag.listHead;
921 	while (b) {
922 		jNode = (struct jffs2_raw_inode *) get_fl_mem(b->offset,
923 			sizeof(ojNode), &ojNode);
924 		if (jNode->compr < JFFS2_NUM_COMPR) {
925 			piL->compr_info[jNode->compr].num_frags++;
926 			piL->compr_info[jNode->compr].compr_sum += jNode->csize;
927 			piL->compr_info[jNode->compr].decompr_sum += jNode->dsize;
928 		}
929 		b = b->next;
930 	}
931 */
932 	return 0;
933 }
934 
935 
936 static struct b_lists *
jffs2_get_list(struct part_info * part,const char * who)937 jffs2_get_list(struct part_info * part, const char *who)
938 {
939 	if (jffs2_1pass_rescan_needed(part)) {
940 		if (!jffs2_1pass_build_lists(part)) {
941 			printf("%s: Failed to scan JFFSv2 file structure\n", who);
942 			return NULL;
943 		}
944 	}
945 	return (struct b_lists *)part->jffs2_priv;
946 }
947 
948 
949 /* Print directory / file contents */
950 u32
jffs2_1pass_ls(struct part_info * part,const char * fname)951 jffs2_1pass_ls(struct part_info * part, const char *fname)
952 {
953 	struct b_lists *pl;
954 	long ret = 0;
955 	u32 inode;
956 
957 	if (! (pl = jffs2_get_list(part, "ls")))
958 		return 0;
959 
960 	if (! (inode = jffs2_1pass_search_list_inodes(pl, fname, 1))) {
961 		putstr("ls: Failed to scan jffs2 file structure\r\n");
962 		return 0;
963 	}
964 
965 #if 0
966 	putLabeledWord("found file at inode = ", inode);
967 	putLabeledWord("read_inode returns = ", ret);
968 #endif
969 
970 	return ret;
971 }
972 
973 
974 /* Load a file from flash into memory. fname can be a full path */
975 u32
jffs2_1pass_load(char * dest,struct part_info * part,const char * fname)976 jffs2_1pass_load(char *dest, struct part_info * part, const char *fname)
977 {
978 
979 	struct b_lists *pl;
980 	long ret = 0;
981 	u32 inode;
982 
983 	if (! (pl = jffs2_get_list(part, "load")))
984 		return 0;
985 
986 	if (! (inode = jffs2_1pass_search_inode(pl, fname, 1))) {
987 		putstr("load: Failed to find inode\r\n");
988 		return 0;
989 	}
990 
991 	/* Resolve symlinks */
992 	if (! (inode = jffs2_1pass_resolve_inode(pl, inode))) {
993 		putstr("load: Failed to resolve inode structure\r\n");
994 		return 0;
995 	}
996 
997 	if ((ret = jffs2_1pass_read_inode(pl, inode, dest, NULL)) < 0) {
998 		putstr("load: Failed to read inode\r\n");
999 		return 0;
1000 	}
1001 
1002 	DEBUGF ("load: loaded '%s' to 0x%lx (%ld bytes)\n", fname,
1003 				(unsigned long) dest, ret);
1004 	return ret;
1005 }
1006 
1007 /* Return information about the fs on this partition */
1008 u32
jffs2_1pass_info(struct part_info * part)1009 jffs2_1pass_info(struct part_info * part)
1010 {
1011 	struct b_jffs2_info info;
1012 	struct b_lists *pl;
1013 	int i;
1014 
1015 	if (! (pl = jffs2_get_list(part, "info")))
1016 		return 0;
1017 
1018 	jffs2_1pass_fill_info(pl, &info);
1019 	for (i = 0; i < JFFS2_NUM_COMPR; i++) {
1020 		printf ("Compression: %s\n"
1021 			"\tfrag count: %d\n"
1022 			"\tcompressed sum: %d\n"
1023 			"\tuncompressed sum: %d\n",
1024 			compr_names[i],
1025 			info.compr_info[i].num_frags,
1026 			info.compr_info[i].compr_sum,
1027 			info.compr_info[i].decompr_sum);
1028 	}
1029 	return 1;
1030 }
1031