yaffs More tests
[yaffs2.git] / direct / tests / fuzzer.c
1 /*
2  * YAFFS: Yet another FFS. A NAND-flash specific file system.
3  *
4  * Copyright (C) 2002-2010 Aleph One Ltd.
5  *   for Toby Churchill Ltd and Brightstar Engineering
6  *
7  * Created by Charles Manning <charles@aleph1.co.uk>
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  * Fuzzer to fuzz a file
16  */
17
18 #include <string.h>
19 #include <unistd.h>
20 #include <stdlib.h>
21 #include <stdio.h>
22 #include <pthread.h>
23 #include <sys/stat.h>
24 #include <sys/types.h>
25 #include <fcntl.h>
26                      
27 int main(int argc, char *argv[])
28 {
29         int prob = 10000;
30         int h;
31         int flen = 0;
32         int changesPerBuffer = 0;
33         int b;
34         char c;
35         unsigned char buffer[1000000];
36         int bufsize;
37         int x;
38         int i;
39         int nbuffers;
40
41         while((c = getopt(argc,argv,"p:")) != -1){
42                 switch(c){
43                         case 'p':
44                                 prob = atoi(optarg);
45                                 break;
46                 }
47         }
48         if(prob < 100){
49                 printf("-p value less than 100 is invalid\n");
50                 return 1;
51         }
52         
53         if(optind >= argc){
54                 printf(" Needs a file name to fuzz\n");
55                 return 1;
56         }
57         
58         h = open(argv[optind], O_RDWR);
59         flen = lseek(h,0,SEEK_END);
60         lseek(h,0,SEEK_SET);
61         if(flen < 1){
62                 printf(" File is too short\n");
63                 return 1;
64         }
65         
66         nbuffers = (flen + sizeof(buffer) - 1) / sizeof(buffer);
67
68         changesPerBuffer = 1+ (sizeof(buffer) * 8) / prob;
69
70         printf("Fuzzing file %s. Size %d, probablity 1/%d, changing %d bits in each of %d buffers\n",
71                 argv[optind],flen,prob,changesPerBuffer,nbuffers);
72
73         srand(time(0));
74
75         for(b = 0; b < nbuffers; b++){
76                 /* printf("buffer %d\n",b); */
77                 lseek(h,b * sizeof(buffer),SEEK_SET);
78                 bufsize = read(h,buffer,sizeof(buffer));
79                 for(i = 0; i < changesPerBuffer; i++){
80                         x = rand() % (sizeof(buffer) * 8);
81                         buffer[x >> 3] ^= (1 << (x & 7));
82                 }
83                 lseek(h,b * sizeof(buffer),SEEK_SET);
84                 write(h,buffer,bufsize);
85         }
86         close(h);
87         return 0;
88 }