xref: /freebsd/tools/test/malloc/main.c (revision adeb92a24c57f97d5cd3c3c45be239cbb23aed68)

/* $FreeBSD$ */
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>

u_long NBUCKETS		= 2000;
u_long NOPS		= 200000;
u_long NSIZE		= (16*1024);

char **foo;

int
main(int argc, char **argv)
{
    int i,j,k;

    if (argc > 1) NOPS     = strtoul(argv[1],0,0);
    if (argc > 2) NBUCKETS = strtoul(argv[2],0,0);
    if (argc > 3) NSIZE	   = strtoul(argv[3],0,0);
    printf("BRK(0)=%x ",sbrk(0));
    foo = malloc (sizeof *foo * NBUCKETS);
    memset(foo,0,sizeof *foo * NBUCKETS);
    for (i = 1; i <= 4096; i *= 2) {
        for (j = 0 ; j < 40960/i && j < NBUCKETS; j++) {
	    foo[j] = malloc(i);
        }
        for (j = 0 ; j < 40960/i && j < NBUCKETS; j++) {
	    free(foo[j]);
	    foo[j] = 0;
        }
    }

    for (i = 0 ; i < NOPS ; i++) {
	j = random() % NBUCKETS;
	k = random() % NSIZE;
	foo[j] = realloc(foo[j], k & 1 ? 0 : k);
	if (k & 1 || k == 0) {
		/*
		 * Workaround because realloc return bogus pointer rather than
		 * NULL if passed zero length.
		 */
		foo[j] = 0;
	}
	if (foo[j])
	    foo[j][0] = 1;
    }
    printf("BRK(1)=%x ",sbrk(0));
    for (j = 0 ; j < NBUCKETS ; j++) {
	if (foo[j]) {
	    free(foo[j]);
	    foo[j] = 0;
	}
    }
    printf("BRK(2)=%x NOPS=%lu NBUCKETS=%lu NSIZE=%lu\n",
	sbrk(0),NOPS,NBUCKETS,NSIZE);
    return 0;
}