/* * 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_guts.h"#define MAX_OBJECTS 10000const char * mkyaffsimage_c_version = "$Id: mkyaffsimage.c,v 1.1.8.2 2003/12/01 09:09:51 bushi Exp $";// External functions for ECC on data//void nand_calculate_ecc (const u_char *dat, u_char *ecc_code);void yaffs_ECCCalculate(const unsigned char *data,unsigned char *ecc);#define nand_calculate_ecc(__dat, __code) yaffs_ECCCalculate(__dat, __code)int nand_correct_data (u_char *dat, u_char *read_ecc, u_char *calc_ecc);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/2002static __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){	nand_calculate_ecc (data , spare->ecc1);	nand_calculate_ecc (&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->st_mode = SWAP32(oh->st_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->st_uid = SWAP32(oh->st_uid);    oh->st_gid = SWAP32(oh->st_gid);    oh->st_atime_ = SWAP32(oh->st_atime_);    oh->st_mtime_ = SWAP32(oh->st_mtime_);    oh->st_ctime_ = SWAP32(oh->st_ctime_);#endif    oh->fileSize = SWAP32(oh->fileSize); // Aiee. An int... signed, at that!    oh->equivalentObjectId = SWAP32(oh->equivalentObjectId);    // alias  - char array.    oh->st_rdev = SWAP32(oh->st_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->st_mode = s->st_mode;		oh->st_uid = s->st_uid;// NCB 12/9/02		oh->st_gid = s->st_uid;		oh->st_gid = s->st_gid;		oh->st_atime_ = s->st_atime;		oh->st_mtime_ = s->st_mtime;		oh->st_ctime_ = s->st_ctime;		oh->st_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);}