bench.c

/*
 * Copyright 2023 Stanislav Paskalev <spaskalev@protonmail.com>
 */

#include <assert.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#define BUDDY_ALLOC_ALIGN 64
#define BUDDY_ALLOC_IMPLEMENTATION
#include "buddy_alloc.h"
#undef BUDDY_ALLOC_IMPLEMENTATION

double test_malloc(struct buddy *buddy, size_t alloc_size);
double test_malloc_firstfit(size_t alloc_size);
void *freeing_callback(void *ctx, void *addr, size_t slot_size, size_t allocated);

int main() {
    setvbuf(stdout, NULL, _IONBF, 0);

    size_t arena_size = 1 << 30;
    unsigned char *buddy_buf = (unsigned char *) malloc(buddy_sizeof_alignment(arena_size, 64));
    unsigned char *data_buf = (unsigned char *) malloc(arena_size);
    struct buddy *buddy = buddy_init_alignment(buddy_buf, data_buf, arena_size, 64);

    double total = 0;
    for (size_t i = 0; i <= 6; i++) {
        total += test_malloc(buddy, 64 << i);
        total += test_malloc(buddy, 64 << i);
        total += test_malloc(buddy, 64 << i);
    }
    printf("Total malloc runtime was %f seconds.\n\n", total);

    free(data_buf);
    free(buddy_buf);
}

double test_malloc(struct buddy *buddy, size_t alloc_size) {
    printf("Starting test with alloc size [%zu].\n", alloc_size);
    time_t start_time = time(NULL);

    while (buddy_malloc(buddy, alloc_size)) {
        // fill it up
    }
    time_t alloc_time = time(NULL);

    assert(buddy_is_full(buddy));

    buddy_walk(buddy, freeing_callback, buddy);
    assert(buddy_is_empty(buddy));

    time_t end_time = time(NULL);
    double delta = difftime(end_time, start_time);
    printf("Test took %.f seconds in total. Allocation: %.f freeing: %.f\n", delta,
        difftime(alloc_time, start_time), difftime(end_time, alloc_time));

    return delta;
}

void *freeing_callback(void *ctx, void *addr, size_t slot_size, size_t allocated) {
    if (! allocated) {
        return NULL;
    }
    struct buddy *buddy = (struct buddy*) ctx;
    buddy_free(buddy, addr);
    return NULL;
}