Add fuzzer work in progress
[yaffs2.git] / direct / tests / nor_stress.c
1 #include "nor_stress.h"
2
3
4 #include "yaffsfs.h"
5 #include "yaffs_fsx.h"
6
7 #include <stdio.h>
8
9
10
11 #if 1
12 #define FSX_INIT(mount_pt) do{ if(interleave_fsx) yaffs_fsx_init(mount_pt); } while(0)
13 #define FSX_COMPLETE() do { if(interleave_fsx) yaffs_fsx_complete(); } while (0)
14
15 #define FSX() \
16 do { \
17   if((myrand() & 0x1) == 0){\
18      if(interleave_fsx) \
19         yaffs_fsx_do_op(); \
20   } \
21 } while(0)
22
23 #else
24 #define FSX_INIT(mount_point) do { } while(0)
25 #define FSX_COMPLETE() do { } while(0)
26 #define FSX() do { } while(0)
27 #endif
28
29
30 void (*ext_fatal)(void) = NULL;
31
32 static unsigned powerUps;
33 static unsigned cycleStarts;
34 static unsigned cycleEnds;
35
36 static int interleave_fsx;
37
38 char fullPathName[100];
39 char fullPowerUpName[100];
40 char fullStartName[100];
41 char fullEndName[100];
42 char fullMainName[100];
43 char fullTempMainName[100];
44 char fullTempCounterName[100];
45
46
47 extern int random_seed;
48
49 int myrand(void) {
50   random_seed = random_seed * 1103515245 + 12345;
51   return((unsigned)(random_seed/65536) % 32768);
52 }
53
54 void MakeName(char *fullName,const char *prefix, const char *name)
55 {
56   strcpy(fullName,prefix);
57   strcat(fullName,"/");
58   strcat(fullName,name);
59 }
60
61
62 void MakeFullNames(const char *prefix)
63 {
64   MakeName(fullPathName,prefix,"");
65   MakeName(fullPowerUpName,prefix,"powerUps");
66   MakeName(fullStartName,prefix,"starts");
67   MakeName(fullEndName,prefix,"ends");
68   MakeName(fullMainName,prefix,"main");
69   MakeName(fullTempCounterName,prefix,"tmp-counter");
70   MakeName(fullTempMainName,prefix,"tmp-main");
71 }
72
73 static void FatalError(int lineNo)
74 {
75   printf("Integrity error %d\n",lineNo);
76   if(ext_fatal)
77         ext_fatal();
78         
79   while(1){
80    sleep(1);
81   }
82 }
83 void print_stat(const char *str, struct yaffs_stat *st)
84 {
85         printf("%s inode %d\n",str,st->st_ino);
86 }
87
88 static void UpdateCounter(const char *name, unsigned *val,  int initialise)
89 {
90   int inh=-1;
91   int outh=-1;
92   unsigned x[2];
93   int nread = 0;
94   int nwritten = 0;
95   
96   x[0] = x[1] = 0;
97   
98   if(initialise){
99     x[0] = 0; 
100     x[1] = 1;
101   } else {
102     inh = yaffs_open(name,O_RDONLY, S_IREAD | S_IWRITE);
103     if(inh >= 0){
104       nread = yaffs_read(inh,x,sizeof(x));
105       yaffs_close(inh);
106     }
107
108     if(nread != sizeof(x) ||
109        x[0] + 1 != x[1]){
110       printf("Error reading counter %s handle %d, x[0] %u x[1] %u last error %d\n",
111               name, inh, x[0], x[1],yaffsfs_GetLastError());
112       FatalError(__LINE__);
113               
114     }
115     x[0]++;
116     x[1]++;
117   }
118   
119   FSX();
120   outh = yaffs_open(fullTempCounterName, O_RDWR | O_TRUNC | O_CREAT, S_IREAD | S_IWRITE);
121   if(outh >= 0){
122    struct yaffs_stat tmpstat, oldstat, tmpfstat;
123    FSX(); 
124     nwritten = yaffs_write(outh,x,sizeof(x));
125     FSX();
126     yaffs_fstat(outh,&tmpfstat);
127     yaffs_close(outh);
128     FSX();
129
130     printf("About to rename %s to %s\n",fullTempCounterName,name);
131     yaffs_stat(fullTempCounterName,&tmpstat);
132     yaffs_stat(name,&oldstat);
133     print_stat("old stat",&oldstat);
134     print_stat("new stat",&tmpstat);
135     print_stat("new fstat",&tmpfstat);
136     yaffs_rename(fullTempCounterName,name);
137     FSX();
138   }
139   
140   if(nwritten != sizeof(x)){
141       printf("Error writing counter %s handle %d, x[0] %u x[1] %u\n",
142               name, inh, x[0], x[1]);
143       FatalError(__LINE__);
144   }
145   
146   *val = x[0];
147   
148   printf("##\n"
149          "## Set counter %s to %u\n"
150          "##\n", name,x[0]);
151 }
152
153
154 static void dump_directory_tree_worker(const char *dname,int recursive)
155 {
156         yaffs_DIR *d;
157         yaffs_dirent *de;
158         struct yaffs_stat s;
159         char str[1000];
160                         
161         d = yaffs_opendir(dname);
162         
163         if(!d)
164         {
165                 printf("opendir failed\n");
166         }
167         else
168         {
169                 while((de = yaffs_readdir(d)) != NULL)
170                 {
171                         strcpy(str,dname);
172                         strcat(str,"/");
173                         strcat(str,de->d_name);
174                         
175                         yaffs_lstat(str,&s);
176                         
177                         printf("%s inode %ld %d obj %x length %d mode %X ",str, de->d_ino, s.st_ino,de->d_dont_use,(int)s.st_size,s.st_mode);\
178                         if(de->d_ino != s.st_ino){
179                                 printf(" \n\n!!!! HEY inode mismatch\n\n");
180                                 FatalError(__LINE__);
181                         }
182
183                         switch(s.st_mode & S_IFMT)
184                         {
185                                 case S_IFREG: printf("data file"); break;
186                                 case S_IFDIR: printf("directory"); break;
187                                 case S_IFLNK: printf("symlink -->");
188                                                           if(yaffs_readlink(str,str,100) < 0)
189                                                                 printf("no alias");
190                                                           else
191                                                                 printf("\"%s\"",str);    
192                                                           break;
193                                 default: printf("unknown"); break;
194                         }
195                         
196                         printf("\n");
197
198                         if((s.st_mode & S_IFMT) == S_IFDIR && recursive)
199                                 dump_directory_tree_worker(str,1);
200                                 
201                         if(s.st_ino > 10000)
202                           FatalError(__LINE__);
203                                                         
204                 }
205                 
206                 yaffs_closedir(d);
207         }
208
209 }
210
211 static void dump_directory_tree(const char *dname)
212 {
213         dump_directory_tree_worker(dname,1);
214         printf("\n");
215         printf("Free space in %s is %d\n\n",dname,(int)yaffs_freespace(dname));
216 }
217
218
219
220
221 #define XX_SIZE 500
222
223 static unsigned xx[XX_SIZE];
224
225 static int yWriteFile(const char *fname, unsigned sz32)
226 {
227         int h;
228         int r;
229         int i;
230         struct yaffs_stat st;
231         unsigned checksum = 0;
232         
233
234         FSX();
235         h = yaffs_open(fname,O_RDWR | O_CREAT | O_TRUNC, S_IREAD | S_IWRITE);
236         yaffs_fstat(h,&st);
237         printf("Writing file %s inode %d\n",fname, st.st_ino);
238         
239         FSX();
240
241         if(h < 0){
242                 printf("could not open file %s\n",fname);
243                 return h;
244         }
245
246         xx[0] = sz32;
247         checksum ^= xx[0];
248
249         if((r = yaffs_write(h,xx,sizeof(unsigned))) != sizeof(unsigned)){
250                 goto WRITE_ERROR;
251         }
252         FSX();
253         while(sz32> 0){
254                 for(i = 0; i < XX_SIZE; i++){
255                   xx[i] = sz32 + i;
256                   checksum ^= xx[i];
257                 }
258                 
259                 FSX();
260                 if((r = yaffs_write(h,xx,sizeof(xx))) != sizeof(xx)){
261                         goto WRITE_ERROR;
262                 }
263                 sz32--;
264         }
265
266         xx[0] = checksum;
267         FSX();
268         if((r = yaffs_write(h,xx,sizeof(unsigned))) != sizeof(unsigned)){
269                 goto WRITE_ERROR;
270         }
271         
272         FSX();
273         yaffs_close(h);
274         return 0;
275
276 WRITE_ERROR:
277         printf("ywrite error at position %d\n",(int)yaffs_lseek(h,0,SEEK_END));
278         yaffs_close(h);
279         return -1;
280         
281 }
282
283 static int yVerifyFile(const char *fName)
284 {
285         unsigned checksum = 0;
286         unsigned totalSize;
287         unsigned sz32;
288         unsigned recordedSize = 0;
289         int r;
290         int h;
291         int i;
292         int retval = 0;
293
294
295         printf("Verifying file %s\n",fName);
296                 
297         h = yaffs_open(fName, O_RDONLY,S_IREAD | S_IWRITE);
298
299         if(h < 0){
300                 printf("could not open file %s\n",fName);
301                 return -1;
302         }
303
304         totalSize = yaffs_lseek(h,0,SEEK_END);
305         yaffs_lseek(h,0,SEEK_SET);
306
307         r = yaffs_read(h,&sz32,sizeof(sz32));
308
309         if(r != sizeof(sz32)){
310                 printf("reading size failed ... returned %d\n",r);
311                 yaffs_close(h);
312                 return -1;
313         }
314         
315         recordedSize = sz32 * sizeof(xx) + 8;
316
317         printf("verify %s: file size is %d, recorded size is %d\n", fName, totalSize, recordedSize);
318         if(totalSize != recordedSize){
319                 printf("!!!!!!!!!!!!!!!!!!!!!!!!file size is wrong, should be %d, is %d\n", recordedSize,totalSize);
320                 yaffs_close(h);
321                 return -1;
322         }
323
324         checksum ^= sz32;
325
326
327         while(sz32 > 0){
328                 r = yaffs_read(h,xx,sizeof(xx));
329                 if(r != sizeof(xx)){
330                         printf("!!!!!!!!!!!!!!!!!!!!!!!!!!reading data failed ... returned %d\n",r);
331                         yaffs_close(h);
332                         return -1;
333                 }
334                 for(i = 0; i < XX_SIZE; i++)
335                   checksum ^= xx[i];
336                 sz32--;
337         }
338         r = yaffs_read(h,xx,sizeof(xx[0]));
339         if(r != sizeof(xx[0])){
340                 printf("!!!!!!!!!!!!!!!!!!!!!!!!!!reading data failed ... returned %d\n",r);
341                 yaffs_close(h);
342                 return -1;
343         }
344         
345         checksum ^= xx[0];
346
347         if(checksum != 0){
348                 printf("!!!!!!!!!!!!!!!!!!!!! checksum failed\n");
349                 retval = -1;
350         } else
351                 printf("verified ok\n");
352         yaffs_close(h);
353
354         return retval;
355 }
356
357
358 static void DoUpdateMainFile(void)
359 {
360         int result;
361         int sz32;
362         sz32 = (myrand() % 1000)   + 20;
363         
364         result = yWriteFile(fullTempMainName,sz32);
365         FSX();
366         if(result)
367             FatalError(__LINE__);
368         yaffs_rename(fullTempMainName,fullMainName);
369         FSX();
370 }
371
372 static void DoVerifyMainFile(void)
373 {
374         int result;
375         result = yVerifyFile(fullMainName);
376         if(result)
377             FatalError(__LINE__);
378
379 }
380
381
382 void NorStressTestInitialise(const char *prefix)
383 {
384   MakeFullNames(prefix);
385   
386   UpdateCounter(fullPowerUpName,&powerUps,1);
387   UpdateCounter(fullStartName,&cycleStarts,1);
388   UpdateCounter(fullEndName,&cycleEnds,1);
389   UpdateCounter(fullPowerUpName,&powerUps,1);
390   DoUpdateMainFile();
391   DoVerifyMainFile();
392 }
393
394
395 void NorStressTestRun(const char *prefix, int n_cycles, int do_fsx)
396 {
397   interleave_fsx = do_fsx;
398   MakeFullNames(prefix);
399   FSX_INIT(prefix);
400     
401   dump_directory_tree(fullPathName);
402   
403   UpdateCounter(fullPowerUpName,&powerUps,0);
404   dump_directory_tree(fullPathName);
405   
406   while(n_cycles < 0 || n_cycles > 0){
407     if(n_cycles > 0)
408       n_cycles--;
409     UpdateCounter(fullStartName, &cycleStarts,0);
410     dump_directory_tree(fullPathName);
411     DoVerifyMainFile();
412     DoUpdateMainFile();
413     dump_directory_tree(fullPathName);
414   
415     UpdateCounter(fullEndName,&cycleEnds,0);
416     dump_directory_tree(fullPathName);
417   }
418   FSX_COMPLETE();
419 }