Merge branch 'master' of ssh://www.aleph1.co.uk/home/aleph1/git/yaffs2
[yaffs2.git] / direct / test-framework / timothy_tests / stress_tester / yaffs_tester.c
1 /*
2  * YAFFS: Yet another FFS. A NAND-flash specific file system.
3  *
4  * Copyright (C) 2002-2011 Aleph One Ltd.
5  *   for Toby Churchill Ltd and Brightstar Engineering
6  *
7  * Created by Timothy Manning <timothy@yaffs.net>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License version 2 as
11  * published by the Free Software Foundation.
12  */
13
14 /*
15  * yaffs_tester.c designed to stress test yaffs2 direct.
16  */
17
18
19 #include "yaffs_tester.h"
20
21
22
23 int random_seed;
24 int simulate_power_failure = 0;
25
26
27 buffer message_buffer;  /*create  message_buffer */
28
29 char yaffs_test_dir[] ="/yaffs2/test_dir";      /*the path to the directory where all of the testing will take place*/
30 char yaffs_mount_dir[]="/yaffs2/";              /*the path to the mount point which yaffs will mount*/
31
32
33 int main(int argc, char *argv[]){       
34         
35         
36         init(yaffs_test_dir,yaffs_mount_dir,argc,argv);
37         test(yaffs_test_dir);
38         yaffs_unmount(yaffs_mount_dir);
39         return 0;
40 }
41
42
43
44 void init(char *yaffs_test_dir,char *yaffs_mount_dir,int argc, char *argv[]){
45         char output=0;
46         int x=0;
47         int seed=-1;
48         FILE *log_handle;
49         /*these variables are already set to zero, but it is better not to take chances*/
50         message_buffer.head=0;                           
51         message_buffer.tail=0;
52
53
54         log_handle=fopen(LOG_FILE,"w");
55         if (log_handle!=NULL){
56                 fputs("log file for yaffs tester\n",log_handle);
57                 fclose(log_handle);
58         }
59         add_to_buffer(&message_buffer,"welcome to the yaffs tester",MESSAGE_LEVEL_BASIC_TASKS,PRINT);/* print boot up message*/ 
60         yaffs_start_up();
61         yaffs_mount(yaffs_mount_dir);
62         for (x=0;x<argc;x++){
63 //              add_to_buffer(&message_buffer,"argv ",MESSAGE_LEVEL_BASIC_TASKS,PRINT);
64 //              add_to_buffer(&message_buffer,argv[x],MESSAGE_LEVEL_BASIC_TASKS,PRINT);
65                 if (strcmp("-seed",argv[x])==0){                        /*warning only compares the length of the strings, quick fix*/
66                         seed= atoi(argv[x+1]);
67                         /*add_to_buffer(&message_buffer,"setting seed to ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
68                         append_int_to_buffer(&message_buffer,seed,MESSAGE_LEVEL_BASIC_TASKS,NPRINT);                    
69                         append_to_buffer(&message_buffer,"\n",MESSAGE_LEVEL_BASIC_TASKS,PRINT);*/
70                 }
71         }
72         if (seed==-1){
73                 seed=time(NULL);
74                 srand(seed); 
75         }
76         else {
77         srand(seed);
78         }
79         add_to_buffer(&message_buffer,"setting seed to ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
80         append_int_to_buffer(&message_buffer,seed,MESSAGE_LEVEL_BASIC_TASKS,PRINT);/* print boot up message*/   
81
82         if (yaffs_access(yaffs_test_dir,0))     /* if the test folder does not exist then create it */
83         {
84                 add_to_buffer(&message_buffer,"creating dir: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
85                 append_to_buffer(&message_buffer,yaffs_test_dir,MESSAGE_LEVEL_BASIC_TASKS,PRINT);       
86                 output=yaffs_mkdir(yaffs_test_dir,S_IREAD | S_IWRITE);
87                 yaffs_check_for_errors(output, &message_buffer,"could not create dir","created dir\n\n");
88         }
89         
90 }
91
92 void quit_program(){
93         yaffs_unmount(yaffs_mount_dir);
94         exit(1);
95 }
96 void join_paths(char *path1,char *path2,char *new_path ){
97
98 /*      printf("strlen path1:%d\n",strlen(path1));
99         printf("strlen path2:%d\n",strlen(path2));
100         printf("path1; %s\n",path1);
101 */
102         add_to_buffer(&message_buffer, "joining paths:",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
103         append_to_buffer(&message_buffer,path1,MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
104         append_to_buffer(&message_buffer, " and ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
105         append_to_buffer(&message_buffer, path2,MESSAGE_LEVEL_BASIC_TASKS,PRINT);
106         if ( (path1[(sizeof(path1)/sizeof(char))-2]=='/') && path2[0]!='/') {
107                 /*paths are compatiable. concatanate them. note -2 is because of \0*/  
108                 strcat(new_path,path1);
109                 strcat(new_path,path2);         
110                 //char new_path[(sizeof(path1)/sizeof(char))+(sizeof(path2)/sizeof(char))];
111                 //strcpy(new_path,strcat(path1,path2)); 
112                 //return new_path;
113         }       
114         else if ((path1[(sizeof(path1)/sizeof(char))-2]!='/') && path2[0]=='/') {
115                 /*paths are compatiable. concatanate them*/  
116                 strcat(new_path,path1);
117                 strcat(new_path,path2);         
118                 //char new_path[(sizeof(path1)/sizeof(char))+(sizeof(path2)/sizeof(char))];
119                 //strcpy(new_path,strcat(path1,path2)); 
120                 //return new_path;
121         }
122         else if ((path1[(sizeof(path1)/sizeof(char))-2]!='/') && path2[0]!='/') {
123                         /*need to add a "/". */  
124                 strcat(new_path,path1);
125                 strcat(new_path,"/");
126                 strcat(new_path,path2);
127                 //strcpy(new_path,strcat(path1,strcat("/\0",path2)));
128
129 #if 0
130                 copy_array(path1,new_path,0,0);
131                 copy_array('\0',new_path,0,(sizeof(path1)/sizeof(char)));
132                 copy_array(path2,new_path,0,(sizeof(path1)/sizeof(char))+1);
133  old method now trying to use copy_array
134                 //char new_path[(sizeof(path1)/sizeof(char))+(sizeof(path2)/sizeof(char))+1];
135                 for (x=0;x<=(sizeof(path1)/sizeof(char))-1;x++){ 
136                         new_path[x]=path1[x];
137                 }
138                 new_path[x+1]='/';
139                 for (x=(sizeof(path1)/sizeof(char)) ,i=0 ;i<=(sizeof(path2)/sizeof(char));x++,i++){ 
140                         new_path[x]=path2[i]; 
141                 }
142 #endif
143
144                 //return new_path;
145         }
146         else if ((path1[(sizeof(path1)/sizeof(char))-2]=='/') && path2[0]=='/') {
147                 /*need to remove a "/". */
148                 /*yaffs does not mind the extra slash. */
149                 //char new_path[(sizeof(path1)/sizeof(char))+(sizeof(path2)/sizeof(char))-1];
150                 
151                 strcat(new_path,path1);
152                 strcat(new_path,path2);
153                 //strcpy(new_path,strcat(path1,strncat("",path2,(sizeof(path1)/sizeof(char))-1))); 
154                 //return new_path;
155         } 
156         else{
157                 //error 
158                 //return -1;
159         }
160 }
161
162 void open_random_file(char *yaffs_test_dir, handle_regster *P_open_handles_array){
163         int output=0;
164         int x=0;
165         char name[MAX_FILE_NAME_SIZE+3 ]="apple\0";
166         char path[MAX_FILE_NAME_SIZE+strlen(yaffs_test_dir)];
167         path[0]='\0';// this should clear the path
168         add_to_buffer(&message_buffer,"\n\number of opened handles: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
169         append_int_to_buffer(&message_buffer,P_open_handles_array->number_of_open_handles,MESSAGE_LEVEL_BASIC_TASKS,PRINT);
170         if (P_open_handles_array->number_of_open_handles<MAX_NUMBER_OF_OPENED_HANDLES)
171         {
172                 generate_random_string(name,MAX_FILE_NAME_SIZE);
173                 printf("before %d %d %d\n",strlen(yaffs_test_dir),strlen(name),strlen(path));
174                 join_paths(yaffs_test_dir,name,path);
175                 printf("after %d %d %d\n",strlen(yaffs_test_dir),strlen(name),strlen(path));
176                 add_to_buffer(&message_buffer,"trying to open file: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
177                 append_to_buffer(&message_buffer,path,MESSAGE_LEVEL_BASIC_TASKS,PRINT);
178                 if (yaffs_access(path,0)==0){
179                         stat_file(path);
180                 }
181                 else {
182                         add_to_buffer(&message_buffer,"file does not exists, creating file",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
183                 }
184
185                 output=yaffs_open(path,O_CREAT | O_TRUNC| O_RDWR, S_IREAD | S_IWRITE);
186                 x=0;
187                 for (x=0;P_open_handles_array->handle[x]!=-3 && x<MAX_NUMBER_OF_OPENED_HANDLES;x++){}   /*find an empty handle*/         
188
189                 add_to_buffer(&message_buffer,"handle array id ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
190                 append_int_to_buffer(&message_buffer,x,MESSAGE_LEVEL_BASIC_TASKS,PRINT);
191                         
192                 //for (x=0;handle_pointers[x]!=NULL;x++){}
193                 P_open_handles_array->handle[x]=output;
194                 P_open_handles_array->path[x][0]='\0';
195                 strcat(P_open_handles_array->path[x],path);
196                 add_to_buffer(&message_buffer,"yaffs handle: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
197                 append_int_to_buffer(&message_buffer,output,MESSAGE_LEVEL_BASIC_TASKS,PRINT);
198                 add_to_buffer(&message_buffer,"stored handle: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
199         
200                 //yaffs_open will return a null pointer if it cannot open a file. check for errors will not work.                       
201                 yaffs_check_for_errors(output, &message_buffer,"failed to open file","opened file");
202                 
203                 P_open_handles_array->number_of_open_handles++; 
204         }
205         else  close_random_file(P_open_handles_array);
206  
207 }
208
209 void write_to_random_file(handle_regster *P_open_handles_array){
210         int number_of_lines_of_text=0;
211         int length=100;
212         char text[length+1];
213         text[0]='\0';
214         int seek=0;
215         int x=0;
216         int output=0;
217         if (P_open_handles_array->number_of_open_handles>0){
218                 
219                 while (P_open_handles_array->handle[x]==-3){            /*find a random open handle*/
220                         x=rand() % (MAX_NUMBER_OF_OPENED_HANDLES-1);
221                 }
222                 add_to_buffer(&message_buffer,"\n\ntrying to write to ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
223                 append_to_buffer(&message_buffer,P_open_handles_array->path[x],MESSAGE_LEVEL_BASIC_TASKS,PRINT);
224                 
225                 stat_file(P_open_handles_array->path[x]);
226                 number_of_lines_of_text=rand() %1000;
227                 add_to_buffer(&message_buffer,"writing  ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
228                 append_int_to_buffer(&message_buffer,number_of_lines_of_text,MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
229                 append_to_buffer(&message_buffer," lines of text",MESSAGE_LEVEL_BASIC_TASKS,PRINT);
230
231                 
232                 for (;number_of_lines_of_text>0;number_of_lines_of_text--)
233                 {
234                         generate_random_string(text,length);
235                         seek=rand()%1000;
236                         add_to_buffer(&message_buffer,"trying to seek to  ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
237                         append_int_to_buffer(&message_buffer,seek,MESSAGE_LEVEL_BASIC_TASKS,PRINT);
238                         output=yaffs_lseek(P_open_handles_array->handle[x],seek,SEEK_SET);
239                         yaffs_check_for_errors(output, &message_buffer,"failed to seek","seeked file");
240                         add_to_buffer(&message_buffer,"trying to write to file",MESSAGE_LEVEL_BASIC_TASKS,PRINT);
241                         output=yaffs_write(P_open_handles_array->handle[x], text, strlen(text));
242                         yaffs_check_for_errors(output, &message_buffer,"failed to write text","wrote text");
243                                         
244                 }       
245          }
246 }
247
248 void truncate_random_file(handle_regster *P_open_handles_array){
249         int x=0;
250         int truncate_size=0;
251         int output=0;
252         
253         if (P_open_handles_array->number_of_open_handles>0){    
254                 add_to_buffer(&message_buffer,"\n\ntruncate function ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
255                 while (P_open_handles_array->handle[x]==-3){            /*find a random open handle*/
256                         x=rand() % (MAX_NUMBER_OF_OPENED_HANDLES-1);
257                 }
258                 add_to_buffer(&message_buffer,"trying to truncate ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
259                 append_to_buffer(&message_buffer,P_open_handles_array->path[x],MESSAGE_LEVEL_BASIC_TASKS,PRINT);
260                 
261                 stat_file(P_open_handles_array->path[x]);
262                 truncate_size=rand() %10000;
263                 output=yaffs_ftruncate(P_open_handles_array->handle[x], truncate_size);
264                 yaffs_check_for_errors(output, &message_buffer,"failed to truncate file","truncated file");
265         }
266 }
267
268 void close_random_file(handle_regster *P_open_handles_array){
269         /*run out of space on the handle pointer array*/        
270         /*make space*/
271         int x=0;
272         int output=0;
273         int start=0;
274         printf("trying to clear handle");
275         if (P_open_handles_array->number_of_open_handles>0){
276                 start=rand() % (MAX_NUMBER_OF_OPENED_HANDLES-1);
277
278                 for (x=start;(x+1>start ||x<start);){
279                         x++;                    
280                         if (x>MAX_NUMBER_OF_OPENED_HANDLES-1) x=0;
281                         if (P_open_handles_array->handle[x] !=-3 ){
282                                 //the handle is open, so try to close it.
283                                 break;
284                         }
285
286                 }
287                 if (P_open_handles_array->handle[x]!=-3)
288                 {
289                         add_to_buffer(&message_buffer,"\n\ntrying to close file: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
290                         append_to_buffer(&message_buffer,P_open_handles_array->path[x],MESSAGE_LEVEL_BASIC_TASKS,PRINT);
291                         add_to_buffer(&message_buffer,"file handle: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
292                         append_int_to_buffer(&message_buffer,P_open_handles_array->handle[x],MESSAGE_LEVEL_BASIC_TASKS,PRINT);
293         
294                         stat_file(P_open_handles_array->path[x]);
295                         output=yaffs_close(P_open_handles_array->handle[x]);
296
297                         if (output==-1) yaffs_check_for_errors(output, &message_buffer,"failed to close file","closed file");
298                         else {
299                                 yaffs_check_for_errors(output, &message_buffer,"failed to close file","closed file");
300                                 P_open_handles_array->handle[x]=-3;
301                                 P_open_handles_array->path[x][0]='\0';
302                                 P_open_handles_array->number_of_open_handles--;
303                         }
304                 }
305                 else {
306                         add_to_buffer(&message_buffer,"\n\ntried to close file but could not find a open file ",MESSAGE_LEVEL_BASIC_TASKS,PRINT);       
307                 }       
308         }
309 }
310
311 void stat_file(char *path){
312         int output=0;
313         struct yaffs_stat stat;
314         if (yaffs_access(path,0)==0){
315                 add_to_buffer(&message_buffer,"file exists, trying to stat: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
316                 add_to_buffer(&message_buffer,path,MESSAGE_LEVEL_BASIC_TASKS,PRINT);
317                 output=yaffs_lstat(path,&stat);
318                 yaffs_check_for_errors(output, &message_buffer,"failed to stat file","statted file");
319                 //stat.st_ino,(int)stat.st_size,stat.st_mode
320                 add_to_buffer(&message_buffer,"yaffs inode: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
321                 append_int_to_buffer(&message_buffer,stat.st_ino,MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
322                 append_to_buffer(&message_buffer," file size: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
323                 append_int_to_buffer(&message_buffer,(int)stat.st_size,MESSAGE_LEVEL_BASIC_TASKS,NPRINT);                       
324                 append_to_buffer(&message_buffer," file mode: ",MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
325                 append_int_to_buffer(&message_buffer,stat.st_mode,MESSAGE_LEVEL_BASIC_TASKS,PRINT);
326         }
327         else{
328                 add_to_buffer(&message_buffer,path,MESSAGE_LEVEL_BASIC_TASKS,NPRINT);
329                 append_to_buffer(&message_buffer," does not exist,could not stat",MESSAGE_LEVEL_BASIC_TASKS,PRINT);
330         }
331 }
332
333 void test(char*yaffs_test_dir){
334         struct yaffs_stat stat;
335         int output=0;
336         //char name[MAX_FILE_NAME_SIZE+3 ]="apple\0";
337         //char path[MAX_FILE_NAME_SIZE];
338         handle_regster open_handles_array;
339         //int handle_pointers[MAX_NUMBER_OF_OPENED_HANDLES];
340         //int number_of_opened_handles=0;
341         int x=0;
342         
343                 
344         open_handles_array.number_of_open_handles=0;
345         for (x=0;x<MAX_NUMBER_OF_OPENED_HANDLES;x++){
346                 open_handles_array.handle[x]=-3;
347                 open_handles_array.path[x][0]='\0';
348
349         }
350
351         while(1)
352         {
353
354                 x=rand() % 3;
355                 printf("running test: %d",x);
356                 switch(x){
357                         case 0 :open_random_file(yaffs_test_dir,&open_handles_array);break;
358                         case 3 :write_to_random_file(&open_handles_array);break;
359 //                      case 1 :close_random_file(&open_handles_array);break;
360                         case 2 :truncate_random_file(&open_handles_array);break;
361                 }
362         }
363 }
364 void  generate_random_string(char *ptr,int length_of_str){
365         unsigned int x;
366         unsigned int length=((rand() %length_of_str)+1);        /*creates a int with the number of charecters been between 1 and 51*/           
367         char letter='\0';
368
369         //printf("generating string\n");
370         //printf("string length is %d\n",length);
371         for (x=0; x <= (length-2) &&length>2 ; x++)
372         {
373                 //printf("x=%d\n",x);   
374                 /* keep generating a charecter until the charecter is legal*/
375                 while((letter=='\0' )||(letter=='/')||(letter=='\\')){
376                         letter=(rand() % 126-32)+32;    /*generate a number between 32 and 126 and uses it as a charecter (letter) */
377                 }       
378                 ptr[x]=letter;
379                 //printf("charecter generated is %c\n",ptr[x]);
380         }
381         ptr[x+1]='\0';  /*adds NULL charecter to turn it into a string*/
382         
383 }
384