Error checking: make sure that there are 3 parameters and that the

[libav.git] / tools / trasher.c

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <inttypes.h>

int main(int argc, char** argv)
{
    FILE *f;
    int count, maxburst, length;

    if (argc < 4){
        printf("USAGE: trasher <filename> <count> <maxburst>\n");
        return 1;
    }

    f= fopen(argv[1], "rb+");
    if (!f){
        perror(argv[1]);
        return 2;
    }
    count= atoi(argv[2]);
    maxburst= atoi(argv[3]);

    srand (time (0));

    fseek(f, 0, SEEK_END);
    length= ftell(f);
    fseek(f, 0, SEEK_SET);

    while(count--){
        int burst= 1 + random() * (uint64_t) (abs(maxburst)-1) / RAND_MAX;
        int pos= random() * (uint64_t) length / RAND_MAX;
        fseek(f, pos, SEEK_SET);

        if(maxburst<0) burst= -maxburst;

        if(pos + burst > length)
            continue;

        while(burst--){
            int val= random() * 256ULL / RAND_MAX;

            if(maxburst<0) val=0;

            fwrite(&val, 1, 1, f);
        }
    }

    return 0;
}