fs/jffs2/jffs2_nand_1pass.c

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