Ticket #2584: rw_test.cpp

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <kernel/OS.h>

char testbuffer[] = "thisisatestthisisatestthisisatestthisisatestthisisatest"
        "thisisatestthisisatestthisisatestthisisatestthisisatestthisisatest"
        "thisisatestthisisatestthisisatestthisisatestthisisatestthisisatest"
        "thisisatestthisisatestthisisatestthisisatestthisisatestthisisatest"
        "thisisatestthisisatestthisisatestthisisatestthisisatestthisisatest"
        "thisisatestthisisatestthisisatestthisisatestthisisatestthisisatest";       

static status_t
rw_thread(void *arg)
{
    int threadnum = (int)arg;
    size_t result = 0;
    char filename[64];
    char readbuffer[sizeof(testbuffer)];
    
    snprintf(filename, sizeof(filename), "/boot/home/rw_test_%d", threadnum);
    for (int i = 0; i < 100; i++) {
        FILE *testfile = fopen(filename, "w");
        if (testfile != NULL) {
            for (int j = 0; j < 100; j++)  {
                result = fwrite(testbuffer, 1, sizeof(testbuffer), testfile);
                if (result < sizeof(testbuffer)) {
                    fprintf(stderr, "Thread %ld: write error - wrote fewer bytes than expected (actual: %ld, expected: %ld\n", threadnum, result, sizeof(testbuffer));
                    exit(8);
                }
            }
            fclose(testfile);
            
            testfile = fopen(filename, "r");
            if (testfile != NULL) {
                for (int k = 0; k < 100; k++) {
                    memset(readbuffer, 0, sizeof(readbuffer));
                    result = fread(readbuffer, 1, sizeof(readbuffer), testfile);
                    if (feof(testfile))
                        break;
                    else if (result < sizeof(readbuffer)) {
                        fprintf(stderr, "Thread %ld: read: size less than expected (actual: %ld, expected: %ld)\n", threadnum, result, sizeof(readbuffer));
                        exit(8);
                    } else 
                        if (memcmp(readbuffer, testbuffer, sizeof(testbuffer)))
                            fprintf(stderr, "Thread %ld: Data returned from read mismatched\n", threadnum);
                }
                fclose(testfile);
            }           
        }
    }
    
    return B_OK;        
}

int main(int argc, char **argv)
{
    int thread_count = 1;
    status_t result;
    
    if (argc == 2)
        thread_count = atoi(argv[1]);
    
    thread_id threads[thread_count];
    
    for (int i = 0; i < thread_count; i++)
        threads[i] = spawn_thread(rw_thread, "Reader/Writer test", B_NORMAL_PRIORITY, (void *)i);
    
    for (int j = 0; j < thread_count; j++)
        wait_for_thread(threads[j], &result);
    
    printf("Test complete!\n");
    
    return 0;
}