| /* |
| * YAFFS: Yet another FFS. A NAND-flash specific file system. |
| * |
| * makeyaffsimage.c |
| * |
| * Makes a YAFFS file system image that can be used to load up a file system. |
| * |
| * Copyright (C) 2002 Aleph One Ltd. |
| * for Toby Churchill Ltd and Brightstar Engineering |
| * |
| * Created by Charles Manning <charles@aleph1.co.uk> |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License version 2 as |
| * published by the Free Software Foundation. |
| * |
| * |
| * Nick Bane modifications flagged NCB |
| * |
| * Endian handling patches by James Ng. |
| * |
| * |
| */ |
| |
| #include <stdlib.h> |
| #include <stdio.h> |
| #include <fcntl.h> |
| #include <sys/types.h> |
| #include <sys/stat.h> |
| #include <dirent.h> |
| #include <string.h> |
| #include <unistd.h> |
| #include "yaffs_ecc.h" |
| #include "yaffs_guts.h" |
| |
| |
| #define MAX_OBJECTS 10000 |
| |
| const char * mkyaffsimage_c_version = "$Id: mkyaffsimage.c,v 1.7 2003/07/16 03:00:48 charles Exp $"; |
| |
| |
| typedef struct |
| { |
| dev_t dev; |
| ino_t ino; |
| int obj; |
| } objItem; |
| |
| |
| static objItem obj_list[MAX_OBJECTS]; |
| static int n_obj = 0; |
| static int obj_id = YAFFS_NOBJECT_BUCKETS + 1; |
| |
| static int nObjects, nDirectories, nPages; |
| |
| static int outFile; |
| |
| static int error; |
| |
| static int convert_endian = 0; |
| |
| static int obj_compare(const void *a, const void * b) |
| { |
| objItem *oa, *ob; |
| |
| oa = (objItem *)a; |
| ob = (objItem *)b; |
| |
| if(oa->dev < ob->dev) return -1; |
| if(oa->dev > ob->dev) return 1; |
| if(oa->ino < ob->ino) return -1; |
| if(oa->ino > ob->ino) return 1; |
| |
| return 0; |
| } |
| |
| |
| static void add_obj_to_list(dev_t dev, ino_t ino, int obj) |
| { |
| if(n_obj < MAX_OBJECTS) |
| { |
| obj_list[n_obj].dev = dev; |
| obj_list[n_obj].ino = ino; |
| obj_list[n_obj].obj = obj; |
| n_obj++; |
| qsort(obj_list,n_obj,sizeof(objItem),obj_compare); |
| |
| } |
| else |
| { |
| // oops! not enough space in the object array |
| fprintf(stderr,"Not enough space in object array\n"); |
| exit(2); |
| } |
| } |
| |
| |
| static int find_obj_in_list(dev_t dev, ino_t ino) |
| { |
| objItem *i = NULL; |
| objItem test; |
| |
| test.dev = dev; |
| test.ino = ino; |
| |
| if(n_obj > 0) |
| { |
| i = bsearch(&test,obj_list,n_obj,sizeof(objItem),obj_compare); |
| } |
| |
| if(i) |
| { |
| return i->obj; |
| } |
| return -1; |
| } |
| |
| // NCB added 10/9/2002 |
| static __u16 yaffs_CalcNameSum(const char *name) |
| { |
| __u16 sum = 0; |
| __u16 i = 1; |
| |
| __u8 *bname = (__u8 *)name; |
| |
| while (*bname) |
| { |
| sum += (*bname) * i; |
| i++; |
| bname++; |
| } |
| return sum; |
| } |
| |
| |
| static void yaffs_CalcECC(const __u8 *data, yaffs_Spare *spare) |
| { |
| yaffs_ECCCalculate(data , spare->ecc1); |
| yaffs_ECCCalculate(&data[256] , spare->ecc2); |
| } |
| |
| static void yaffs_CalcTagsECC(yaffs_Tags *tags) |
| { |
| // Todo don't do anything yet. Need to calculate ecc |
| unsigned char *b = ((yaffs_TagsUnion *)tags)->asBytes; |
| unsigned i,j; |
| unsigned ecc = 0; |
| unsigned bit = 0; |
| |
| // Clear ECC fields |
| if (!convert_endian) |
| { |
| tags->ecc = 0; |
| } |
| else |
| { |
| // Because we're in "munged tag" mode, we have to clear it manually |
| b[6] &= 0xC0; |
| b[7] &= 0x03; |
| } |
| |
| for(i = 0; i < 8; i++) |
| { |
| // NCB modified 20-9-02 for(j = 1; j &0x7f; j<<=1) |
| for(j = 1; j &0xff; j<<=1) |
| { |
| bit++; |
| if(b[i] & j) |
| { |
| ecc ^= bit; |
| } |
| } |
| } |
| |
| // Write out ECC |
| if (!convert_endian) |
| { |
| tags->ecc = ecc; |
| } |
| else |
| { |
| // We have to munge the ECC again. |
| b[6] |= ((ecc >> 6) & 0x3F); |
| b[7] |= ((ecc & 0x3F) << 2); |
| } |
| } |
| static void yaffs_LoadTagsIntoSpare(yaffs_Spare *sparePtr, yaffs_Tags *tagsPtr) |
| { |
| yaffs_TagsUnion *tu = (yaffs_TagsUnion *)tagsPtr; |
| |
| //yaffs_CalcTagsECC(tagsPtr); |
| |
| sparePtr->tagByte0 = tu->asBytes[0]; |
| sparePtr->tagByte1 = tu->asBytes[1]; |
| sparePtr->tagByte2 = tu->asBytes[2]; |
| sparePtr->tagByte3 = tu->asBytes[3]; |
| sparePtr->tagByte4 = tu->asBytes[4]; |
| sparePtr->tagByte5 = tu->asBytes[5]; |
| sparePtr->tagByte6 = tu->asBytes[6]; |
| sparePtr->tagByte7 = tu->asBytes[7]; |
| } |
| |
| /* This little function converts a little endian tag to a big endian tag. |
| * NOTE: The tag is not usable after this other than calculating the CRC |
| * with. |
| */ |
| static void little_to_big_endian(yaffs_Tags *tagsPtr) |
| { |
| yaffs_TagsUnion * tags = (yaffs_TagsUnion* )tagsPtr; // Work in bytes. |
| yaffs_TagsUnion temp; |
| |
| memset(&temp, 0, sizeof(temp)); |
| // Ick, I hate magic numbers. |
| temp.asBytes[0] = ((tags->asBytes[2] & 0x0F) << 4) | ((tags->asBytes[1] & 0xF0) >> 4); |
| temp.asBytes[1] = ((tags->asBytes[1] & 0x0F) << 4) | ((tags->asBytes[0] & 0xF0) >> 4); |
| temp.asBytes[2] = ((tags->asBytes[0] & 0x0F) << 4) | ((tags->asBytes[2] & 0x30) >> 2) | ((tags->asBytes[3] & 0xC0) >> 6); |
| temp.asBytes[3] = ((tags->asBytes[3] & 0x3F) << 2) | ((tags->asBytes[2] & 0xC0) >> 6); |
| temp.asBytes[4] = ((tags->asBytes[6] & 0x03) << 6) | ((tags->asBytes[5] & 0xFC) >> 2); |
| temp.asBytes[5] = ((tags->asBytes[5] & 0x03) << 6) | ((tags->asBytes[4] & 0xFC) >> 2); |
| temp.asBytes[6] = ((tags->asBytes[4] & 0x03) << 6) | (tags->asBytes[7] & 0x3F); |
| temp.asBytes[7] = (tags->asBytes[6] & 0xFC) | ((tags->asBytes[7] & 0xC0) >> 6); |
| |
| // Now copy it back. |
| tags->asBytes[0] = temp.asBytes[0]; |
| tags->asBytes[1] = temp.asBytes[1]; |
| tags->asBytes[2] = temp.asBytes[2]; |
| tags->asBytes[3] = temp.asBytes[3]; |
| tags->asBytes[4] = temp.asBytes[4]; |
| tags->asBytes[5] = temp.asBytes[5]; |
| tags->asBytes[6] = temp.asBytes[6]; |
| tags->asBytes[7] = temp.asBytes[7]; |
| } |
| |
| static int write_chunk(__u8 *data, __u32 objId, __u32 chunkId, __u32 nBytes) |
| { |
| yaffs_Tags t; |
| yaffs_Spare s; |
| |
| error = write(outFile,data,512); |
| if(error < 0) return error; |
| |
| memset(&t,0xff,sizeof (yaffs_Tags)); |
| memset(&s,0xff,sizeof (yaffs_Spare)); |
| |
| t.chunkId = chunkId; |
| t.serialNumber = 0; |
| t.byteCount = nBytes; |
| t.objectId = objId; |
| |
| if (convert_endian) |
| { |
| little_to_big_endian(&t); |
| } |
| |
| yaffs_CalcTagsECC(&t); |
| yaffs_LoadTagsIntoSpare(&s,&t); |
| yaffs_CalcECC(data,&s); |
| |
| nPages++; |
| |
| return write(outFile,&s,sizeof(yaffs_Spare)); |
| |
| } |
| |
| #define SWAP32(x) ((((x) & 0x000000FF) << 24) | \ |
| (((x) & 0x0000FF00) << 8 ) | \ |
| (((x) & 0x00FF0000) >> 8 ) | \ |
| (((x) & 0xFF000000) >> 24)) |
| |
| #define SWAP16(x) ((((x) & 0x00FF) << 8) | \ |
| (((x) & 0xFF00) >> 8)) |
| |
| // This one is easier, since the types are more standard. No funky shifts here. |
| static void object_header_little_to_big_endian(yaffs_ObjectHeader* oh) |
| { |
| oh->type = SWAP32(oh->type); // GCC makes enums 32 bits. |
| oh->parentObjectId = SWAP32(oh->parentObjectId); // int |
| oh->sum__NoLongerUsed = SWAP16(oh->sum__NoLongerUsed); // __u16 - Not used, but done for completeness. |
| // name = skip. Char array. Not swapped. |
| oh->yst_mode = SWAP32(oh->yst_mode); |
| #ifdef CONFIG_YAFFS_WINCE // WinCE doesn't implement this, but we need to just in case. |
| // In fact, WinCE would be *THE* place where this would be an issue! |
| oh->notForWinCE[0] = SWAP32(oh->notForWinCE[0]); |
| oh->notForWinCE[1] = SWAP32(oh->notForWinCE[1]); |
| oh->notForWinCE[2] = SWAP32(oh->notForWinCE[2]); |
| oh->notForWinCE[3] = SWAP32(oh->notForWinCE[3]); |
| oh->notForWinCE[4] = SWAP32(oh->notForWinCE[4]); |
| #else |
| // Regular POSIX. |
| oh->yst_uid = SWAP32(oh->yst_uid); |
| oh->yst_gid = SWAP32(oh->yst_gid); |
| oh->yst_atime = SWAP32(oh->yst_atime); |
| oh->yst_mtime = SWAP32(oh->yst_mtime); |
| oh->yst_ctime = SWAP32(oh->yst_ctime); |
| #endif |
| |
| oh->fileSize = SWAP32(oh->fileSize); // Aiee. An int... signed, at that! |
| oh->equivalentObjectId = SWAP32(oh->equivalentObjectId); |
| // alias - char array. |
| oh->yst_rdev = SWAP32(oh->yst_rdev); |
| |
| #ifdef CONFIG_YAFFS_WINCE |
| oh->win_ctime[0] = SWAP32(oh->win_ctime[0]); |
| oh->win_ctime[1] = SWAP32(oh->win_ctime[1]); |
| oh->win_atime[0] = SWAP32(oh->win_atime[0]); |
| oh->win_atime[1] = SWAP32(oh->win_atime[1]); |
| oh->win_mtime[0] = SWAP32(oh->win_mtime[0]); |
| oh->win_mtime[1] = SWAP32(oh->win_mtime[1]); |
| oh->roomToGrow[0] = SWAP32(oh->roomToGrow[0]); |
| oh->roomToGrow[1] = SWAP32(oh->roomToGrow[1]); |
| oh->roomToGrow[2] = SWAP32(oh->roomToGrow[2]); |
| oh->roomToGrow[3] = SWAP32(oh->roomToGrow[3]); |
| oh->roomToGrow[4] = SWAP32(oh->roomToGrow[4]); |
| oh->roomToGrow[5] = SWAP32(oh->roomToGrow[5]); |
| #else |
| oh->roomToGrow[0] = SWAP32(oh->roomToGrow[0]); |
| oh->roomToGrow[1] = SWAP32(oh->roomToGrow[1]); |
| oh->roomToGrow[2] = SWAP32(oh->roomToGrow[2]); |
| oh->roomToGrow[3] = SWAP32(oh->roomToGrow[3]); |
| oh->roomToGrow[4] = SWAP32(oh->roomToGrow[4]); |
| oh->roomToGrow[5] = SWAP32(oh->roomToGrow[5]); |
| oh->roomToGrow[6] = SWAP32(oh->roomToGrow[6]); |
| oh->roomToGrow[7] = SWAP32(oh->roomToGrow[7]); |
| oh->roomToGrow[8] = SWAP32(oh->roomToGrow[8]); |
| oh->roomToGrow[9] = SWAP32(oh->roomToGrow[9]); |
| oh->roomToGrow[10] = SWAP32(oh->roomToGrow[10]); |
| oh->roomToGrow[11] = SWAP32(oh->roomToGrow[11]); |
| #endif |
| } |
| |
| static int write_object_header(int objId, yaffs_ObjectType t, struct stat *s, int parent, const char *name, int equivalentObj, const char * alias) |
| { |
| __u8 bytes[512]; |
| |
| |
| yaffs_ObjectHeader *oh = (yaffs_ObjectHeader *)bytes; |
| |
| memset(bytes,0xff,512); |
| |
| oh->type = t; |
| |
| oh->parentObjectId = parent; |
| |
| strncpy(oh->name,name,YAFFS_MAX_NAME_LENGTH); |
| |
| |
| if(t != YAFFS_OBJECT_TYPE_HARDLINK) |
| { |
| oh->yst_mode = s->st_mode; |
| oh->yst_uid = s->st_uid; |
| // NCB 12/9/02 oh->yst_gid = s->yst_uid; |
| oh->yst_gid = s->st_gid; |
| oh->yst_atime = s->st_atime; |
| oh->yst_mtime = s->st_mtime; |
| oh->yst_ctime = s->st_ctime; |
| oh->yst_rdev = s->st_rdev; |
| } |
| |
| if(t == YAFFS_OBJECT_TYPE_FILE) |
| { |
| oh->fileSize = s->st_size; |
| } |
| |
| if(t == YAFFS_OBJECT_TYPE_HARDLINK) |
| { |
| oh->equivalentObjectId = equivalentObj; |
| } |
| |
| if(t == YAFFS_OBJECT_TYPE_SYMLINK) |
| { |
| strncpy(oh->alias,alias,YAFFS_MAX_ALIAS_LENGTH); |
| } |
| |
| if (convert_endian) |
| { |
| object_header_little_to_big_endian(oh); |
| } |
| |
| return write_chunk(bytes,objId,0,0xffff); |
| |
| } |
| |
| |
| static int process_directory(int parent, const char *path) |
| { |
| |
| DIR *dir; |
| struct dirent *entry; |
| |
| nDirectories++; |
| |
| dir = opendir(path); |
| |
| if(dir) |
| { |
| while((entry = readdir(dir)) != NULL) |
| { |
| |
| /* Ignore . and .. */ |
| if(strcmp(entry->d_name,".") && |
| strcmp(entry->d_name,"..")) |
| { |
| char full_name[500]; |
| struct stat stats; |
| int equivalentObj; |
| int newObj; |
| |
| sprintf(full_name,"%s/%s",path,entry->d_name); |
| |
| lstat(full_name,&stats); |
| |
| if(S_ISLNK(stats.st_mode) || |
| S_ISREG(stats.st_mode) || |
| S_ISDIR(stats.st_mode) || |
| S_ISFIFO(stats.st_mode) || |
| S_ISBLK(stats.st_mode) || |
| S_ISCHR(stats.st_mode) || |
| S_ISSOCK(stats.st_mode)) |
| { |
| |
| newObj = obj_id++; |
| nObjects++; |
| |
| printf("Object %d, %s is a ",newObj,full_name); |
| |
| /* We're going to create an object for it */ |
| if((equivalentObj = find_obj_in_list(stats.st_dev, stats.st_ino)) > 0) |
| { |
| /* we need to make a hard link */ |
| printf("hard link to object %d\n",equivalentObj); |
| error = write_object_header(newObj, YAFFS_OBJECT_TYPE_HARDLINK, &stats, parent, entry->d_name, equivalentObj, NULL); |
| } |
| else |
| { |
| |
| add_obj_to_list(stats.st_dev,stats.st_ino,newObj); |
| |
| if(S_ISLNK(stats.st_mode)) |
| { |
| |
| char symname[500]; |
| |
| memset(symname,0, sizeof(symname)); |
| |
| readlink(full_name,symname,sizeof(symname) -1); |
| |
| printf("symlink to \"%s\"\n",symname); |
| error = write_object_header(newObj, YAFFS_OBJECT_TYPE_SYMLINK, &stats, parent, entry->d_name, -1, symname); |
| |
| } |
| else if(S_ISREG(stats.st_mode)) |
| { |
| printf("file, "); |
| error = write_object_header(newObj, YAFFS_OBJECT_TYPE_FILE, &stats, parent, entry->d_name, -1, NULL); |
| |
| if(error >= 0) |
| { |
| int h; |
| __u8 bytes[512]; |
| int nBytes; |
| int chunk = 0; |
| |
| h = open(full_name,O_RDONLY); |
| if(h >= 0) |
| { |
| memset(bytes,0xff,512); |
| while((nBytes = read(h,bytes,512)) > 0) |
| { |
| chunk++; |
| write_chunk(bytes,newObj,chunk,nBytes); |
| memset(bytes,0xff,512); |
| } |
| if(nBytes < 0) |
| error = nBytes; |
| |
| printf("%d data chunks written\n",chunk); |
| } |
| else |
| { |
| perror("Error opening file"); |
| } |
| close(h); |
| |
| } |
| |
| } |
| else if(S_ISSOCK(stats.st_mode)) |
| { |
| printf("socket\n"); |
| error = write_object_header(newObj, YAFFS_OBJECT_TYPE_SPECIAL, &stats, parent, entry->d_name, -1, NULL); |
| } |
| else if(S_ISFIFO(stats.st_mode)) |
| { |
| printf("fifo\n"); |
| error = write_object_header(newObj, YAFFS_OBJECT_TYPE_SPECIAL, &stats, parent, entry->d_name, -1, NULL); |
| } |
| else if(S_ISCHR(stats.st_mode)) |
| { |
| printf("character device\n"); |
| error = write_object_header(newObj, YAFFS_OBJECT_TYPE_SPECIAL, &stats, parent, entry->d_name, -1, NULL); |
| } |
| else if(S_ISBLK(stats.st_mode)) |
| { |
| printf("block device\n"); |
| error = write_object_header(newObj, YAFFS_OBJECT_TYPE_SPECIAL, &stats, parent, entry->d_name, -1, NULL); |
| } |
| else if(S_ISDIR(stats.st_mode)) |
| { |
| printf("directory\n"); |
| error = write_object_header(newObj, YAFFS_OBJECT_TYPE_DIRECTORY, &stats, parent, entry->d_name, -1, NULL); |
| // NCB modified 10/9/2001 process_directory(1,full_name); |
| process_directory(newObj,full_name); |
| } |
| } |
| } |
| else |
| { |
| printf(" we don't handle this type\n"); |
| } |
| } |
| } |
| } |
| |
| return 0; |
| |
| } |
| |
| |
| int main(int argc, char *argv[]) |
| { |
| struct stat stats; |
| |
| printf("mkyaffsimage: image building tool for YAFFS built "__DATE__"\n"); |
| |
| if(argc < 3) |
| { |
| printf("usage: mkyaffsimage dir image_file [convert]\n"); |
| printf(" dir the directory tree to be converted\n"); |
| printf(" image_file the output file to hold the image\n"); |
| printf(" 'convert' produce a big-endian image from a little-endian machine\n"); |
| exit(1); |
| } |
| |
| if ((argc == 4) && (!strncmp(argv[3], "convert", strlen("convert")))) |
| { |
| convert_endian = 1; |
| } |
| |
| if(stat(argv[1],&stats) < 0) |
| { |
| printf("Could not stat %s\n",argv[1]); |
| exit(1); |
| } |
| |
| if(!S_ISDIR(stats.st_mode)) |
| { |
| printf(" %s is not a directory\n",argv[1]); |
| exit(1); |
| } |
| |
| outFile = open(argv[2],O_CREAT | O_TRUNC | O_WRONLY, S_IREAD | S_IWRITE); |
| |
| |
| if(outFile < 0) |
| { |
| printf("Could not open output file %s\n",argv[2]); |
| exit(1); |
| } |
| |
| printf("Processing directory %s into image file %s\n",argv[1],argv[2]); |
| error = write_object_header(1, YAFFS_OBJECT_TYPE_DIRECTORY, &stats, 1,"", -1, NULL); |
| if(error) |
| error = process_directory(YAFFS_OBJECTID_ROOT,argv[1]); |
| |
| close(outFile); |
| |
| if(error < 0) |
| { |
| perror("operation incomplete"); |
| exit(1); |
| } |
| else |
| { |
| printf("Operation complete.\n" |
| "%d objects in %d directories\n" |
| "%d NAND pages\n",nObjects, nDirectories, nPages); |
| } |
| |
| close(outFile); |
| |
| exit(0); |
| } |
| |