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;
}