memtest.c

Go to the documentation of this file.

/* Number of chunks to test with */
#define NUM_BLOCKS 10

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include "MEM_guardedalloc.h"

static void mem_error_cb(const char *errorStr)
{
        fprintf(stderr, "%s", errorStr);
        fflush(stderr);
}

int main (int argc, char *argv[])
{
        int verbose       = 0;
        int error_status  = 0;
        int retval        = 0;
        int *ip;

        void *p[NUM_BLOCKS];
        int i = 0;

        /* ----------------------------------------------------------------- */
        switch (argc) {
        case 2:         
                verbose = atoi(argv[1]);
                if (verbose < 0) verbose = 0;
                break;          
        case 1:
        default:
                verbose = 0;
        }
        if (verbose) {
                fprintf(stderr,"\n*** Simple memory test\n|\n");
        }

        /* ----------------------------------------------------------------- */
        /* Round one, do a normal allocation, and free the blocks again.     */
        /* ----------------------------------------------------------------- */
        /* flush mem lib output to stderr */
        MEM_set_error_callback(mem_error_cb);
        
        for (i = 0; i < NUM_BLOCKS; i++) {
                int blocksize = 10000;
                char tagstring[1000];
                if (verbose >1) printf("|--* Allocating block %d\n", i);
                sprintf(tagstring,"Memblock no. %d : ", i);
                p[i]= MEM_callocN(blocksize, strdup(tagstring));
        }

        /* report on that */
        if (verbose > 1) MEM_printmemlist();

        /* memory is there: test it */
        error_status = MEM_check_memory_integrity();

        if (verbose) {
                if (error_status) {
                        fprintf(stderr, "|--* Memory test FAILED\n|\n");
                } else {
                        fprintf(stderr, "|--* Memory tested as good (as it should be)\n|\n");
                }
        } 

        for (i = 0; i < NUM_BLOCKS; i++) {
                MEM_freeN(p[i]);
        }

        /* ----------------------------------------------------------------- */
        /* Round two, do a normal allocation, and corrupt some blocks.       */
        /* ----------------------------------------------------------------- */
        /* switch off, because it will complain about some things.           */
        MEM_set_error_callback(NULL);

        for (i = 0; i < NUM_BLOCKS; i++) {
                int blocksize = 10000;
                char tagstring[1000];
                if (verbose >1) printf("|--* Allocating block %d\n", i);
                sprintf(tagstring,"Memblock no. %d : ", i);
                p[i]= MEM_callocN(blocksize, strdup(tagstring));
        }

        /* now corrupt a few blocks...*/
        ip = (int*) p[5] - 50 ;
        for (i = 0; i< 1000; i++,ip++) *ip = i+1;
        ip = (int*) p[6];
        *(ip+10005) = 0;
        
        retval = MEM_check_memory_integrity();

        /* the test should have failed */
        error_status |= !retval;                
        if (verbose) {
                if (retval) {
                        fprintf(stderr, "|--* Memory test failed (as it should be)\n");
                } else {
                        fprintf(stderr, "|--* Memory test FAILED to find corrupted blocks \n");
                }
        } 
        
        for (i = 0; i < NUM_BLOCKS; i++) {
                MEM_freeN(p[i]);
        }


        if (verbose && error_status) {
                fprintf(stderr,"|--* Memory was corrupted\n");
        }
        /* ----------------------------------------------------------------- */ 
        if (verbose) {
                if (error_status) {
                        fprintf(stderr,"|\n|--* Errors were detected\n");
                } else {
                        fprintf(stderr,"|\n|--* Test exited succesfully\n");
                }
                
                fprintf(stderr,"|\n*** Finished test\n\n");
        }
        return error_status;
}