#include "nor_stress.h"#include "yaffsfs.h"#include "yaffs_fsx.h"#include <stdio.h>#if 1#define FSX_INIT(mount_pt) do{ if(interleave_fsx) yaffs_fsx_init(mount_pt); } while(0)#define FSX_COMPLETE() do { if(interleave_fsx) yaffs_fsx_complete(); } while (0)#define FSX() \do { \  if((myrand() & 0x1) == 0){\     if(interleave_fsx) \        yaffs_fsx_do_op(); \  } \} while(0)#else#define FSX_INIT(mount_point) do { } while(0)#define FSX_COMPLETE() do { } while(0)#define FSX() do { } while(0)#endifstatic unsigned powerUps;static unsigned cycleStarts;static unsigned cycleEnds;static int interleave_fsx;char fullPathName[100];char fullPowerUpName[100];char fullStartName[100];char fullEndName[100];char fullMainName[100];char fullTempMainName[100];char fullTempCounterName[100];extern int random_seed;int myrand(void) {  random_seed = random_seed * 1103515245 + 12345;  return((unsigned)(random_seed/65536) % 32768);}void MakeName(char *fullName,const char *prefix, const char *name){  strcpy(fullName,prefix);  strcat(fullName,"/");  strcat(fullName,name);}void MakeFullNames(const char *prefix){  MakeName(fullPathName,prefix,"");  MakeName(fullPowerUpName,prefix,"powerUps");  MakeName(fullStartName,prefix,"starts");  MakeName(fullEndName,prefix,"ends");  MakeName(fullMainName,prefix,"main");  MakeName(fullTempCounterName,prefix,"tmp-counter");  MakeName(fullTempMainName,prefix,"tmp-main");}static void FatalError(void){  printf("Integrity error\n");  while(1){   sleep(1);  }}static void UpdateCounter(const char *name, unsigned *val,  int initialise){  int inh=-1;  int outh=-1;  unsigned x[2];  int nread = 0;  int nwritten = 0;    x[0] = x[1] = 0;    if(initialise){    x[0] = 0;     x[1] = 1;  } else {    inh = yaffs_open(name,O_RDONLY, S_IREAD | S_IWRITE);    if(inh >= 0){      nread = yaffs_read(inh,x,sizeof(x));      yaffs_close(inh);    }    if(nread != sizeof(x) ||       x[0] + 1 != x[1]){      printf("Error reading counter %s handle %d, x[0] %u x[1] %u last error %d\n",              name, inh, x[0], x[1],yaffsfs_GetLastError());      FatalError();                  }    x[0]++;    x[1]++;  }    FSX();  outh = yaffs_open(fullTempCounterName, O_RDWR | O_TRUNC | O_CREAT, S_IREAD | S_IWRITE);  if(outh >= 0){   FSX();     nwritten = yaffs_write(outh,x,sizeof(x));    FSX();    yaffs_close(outh);    FSX();    yaffs_rename(fullTempCounterName,name);    FSX();  }    if(nwritten != sizeof(x)){      printf("Error writing counter %s handle %d, x[0] %u x[1] %u\n",              name, inh, x[0], x[1]);      FatalError();  }    *val = x[0];    printf("##\n"         "## Set counter %s to %u\n"         "##\n", name,x[0]);}static void dump_directory_tree_worker(const char *dname,int recursive){	yaffs_DIR *d;	yaffs_dirent *de;	struct yaffs_stat s;	char str[1000];				d = yaffs_opendir(dname);		if(!d)	{		printf("opendir failed\n");	}	else	{		while((de = yaffs_readdir(d)) != NULL)		{			strcpy(str,dname);			strcat(str,"/");			strcat(str,de->d_name);						yaffs_lstat(str,&s);						printf("%s inode %d obj %x length %d mode %X ",str,s.st_ino,de->d_dont_use,(int)s.st_size,s.st_mode);			switch(s.st_mode & S_IFMT)			{				case S_IFREG: printf("data file"); break;				case S_IFDIR: printf("directory"); break;				case S_IFLNK: printf("symlink -->");							  if(yaffs_readlink(str,str,100) < 0)								printf("no alias");							  else								printf("\"%s\"",str);    							  break;				default: printf("unknown"); break;			}						printf("\n");			if((s.st_mode & S_IFMT) == S_IFDIR && recursive)				dump_directory_tree_worker(str,1);				                        if(s.st_ino > 10000)                          FatalError();									}				yaffs_closedir(d);	}}static void dump_directory_tree(const char *dname){	dump_directory_tree_worker(dname,1);	printf("\n");	printf("Free space in %s is %d\n\n",dname,(int)yaffs_freespace(dname));}#define XX_SIZE 500static unsigned xx[XX_SIZE];static int yWriteFile(const char *fname, unsigned sz32){	int h;	int r;	int i;	unsigned checksum = 0;		printf("Writing file %s\n",fname);	FSX();	h = yaffs_open(fname,O_RDWR | O_CREAT | O_TRUNC, S_IREAD | S_IWRITE);	FSX();	if(h < 0){		printf("could not open file %s\n",fname);		return h;	}        xx[0] = sz32;        checksum ^= xx[0];	if((r = yaffs_write(h,xx,sizeof(unsigned))) != sizeof(unsigned)){		goto WRITE_ERROR;	}        FSX();	while(sz32> 0){        	for(i = 0; i < XX_SIZE; i++){        	  xx[i] = sz32 + i;        	  checksum ^= xx[i];                }                                FSX();		if((r = yaffs_write(h,xx,sizeof(xx))) != sizeof(xx)){			goto WRITE_ERROR;		}		sz32--;	}	xx[0] = checksum;	FSX();	if((r = yaffs_write(h,xx,sizeof(unsigned))) != sizeof(unsigned)){		goto WRITE_ERROR;	}		FSX();	yaffs_close(h);	return 0;WRITE_ERROR:	printf("ywrite error at position %d\n",(int)yaffs_lseek(h,0,SEEK_END));	yaffs_close(h);	return -1;	}static int yVerifyFile(const char *fName){	unsigned checksum = 0;	unsigned totalSize;	unsigned sz32;	unsigned recordedSize = 0;	int r;	int h;	int i;	int retval = 0;        printf("Verifying file %s\n",fName);        		h = yaffs_open(fName, O_RDONLY,S_IREAD | S_IWRITE);	if(h < 0){		printf("could not open file %s\n",fName);		return -1;	}	totalSize = yaffs_lseek(h,0,SEEK_END);	yaffs_lseek(h,0,SEEK_SET);	r = yaffs_read(h,&sz32,sizeof(sz32));	if(r != sizeof(sz32)){		printf("reading size failed ... returned %d\n",r);		yaffs_close(h);		return -1;	}		recordedSize = sz32 * sizeof(xx) + 8;	printf("verify %s: file size is %d, recorded size is %d\n", fName, totalSize, recordedSize);	if(totalSize != recordedSize){		printf("!!!!!!!!!!!!!!!!!!!!!!!!file size is wrong, should be %d, is %d\n", recordedSize,totalSize);		yaffs_close(h);		return -1;	}	checksum ^= sz32;	while(sz32 > 0){		r = yaffs_read(h,xx,sizeof(xx));		if(r != sizeof(xx)){			printf("!!!!!!!!!!!!!!!!!!!!!!!!!!reading data failed ... returned %d\n",r);			yaffs_close(h);			return -1;		}		for(i = 0; i < XX_SIZE; i++)		  checksum ^= xx[i];		sz32--;	}	r = yaffs_read(h,xx,sizeof(xx[0]));	if(r != sizeof(xx[0])){		printf("!!!!!!!!!!!!!!!!!!!!!!!!!!reading data failed ... returned %d\n",r);		yaffs_close(h);		return -1;	}		checksum ^= xx[0];	if(checksum != 0){		printf("!!!!!!!!!!!!!!!!!!!!! checksum failed\n");		retval = -1;        } else		printf("verified ok\n");	yaffs_close(h);	return retval;}static void DoUpdateMainFile(void){        int result;        int sz32;        sz32 = (myrand() % 1000)   + 20;        	result = yWriteFile(fullTempMainName,sz32);	FSX();	if(result)	    FatalError();	yaffs_rename(fullTempMainName,fullMainName);	FSX();}static void DoVerifyMainFile(void){        int result;	result = yVerifyFile(fullMainName);	if(result)	    FatalError();}void NorStressTestInitialise(const char *prefix){  MakeFullNames(prefix);    UpdateCounter(fullPowerUpName,&powerUps,1);  UpdateCounter(fullStartName,&cycleStarts,1);  UpdateCounter(fullEndName,&cycleEnds,1);  UpdateCounter(fullPowerUpName,&powerUps,1);  DoUpdateMainFile();  DoVerifyMainFile();}void NorStressTestRun(const char *prefix, int n_cycles, int do_fsx){  interleave_fsx = do_fsx;  MakeFullNames(prefix);  FSX_INIT(prefix);      dump_directory_tree(fullPathName);    UpdateCounter(fullPowerUpName,&powerUps,0);  dump_directory_tree(fullPathName);    while(n_cycles < 0 || n_cycles > 0){    if(n_cycles > 0)      n_cycles--;    UpdateCounter(fullStartName, &cycleStarts,0);    dump_directory_tree(fullPathName);    DoVerifyMainFile();    DoUpdateMainFile();    dump_directory_tree(fullPathName);      UpdateCounter(fullEndName,&cycleEnds,0);    dump_directory_tree(fullPathName);  }  FSX_COMPLETE();}