source: proiecte/swift/trunk/src/swift_allocator.c @ 176

/*
 * swift_allocator.c
 *
 * (c) 2009 Ionut Rosoiu <ionut.rosoiu@gmail.com>
 *
 */

#include <stdlib.h>
#include <stdio.h>
#include "swift_assert.h"
#include "swift_allocator.h"
#include "swift_log.h"

inline
void* swift_malloc_system(size_t size) {
        void *ret = malloc(size);

        swift_assert(ret);

        return ret;
}

inline
void swift_free_system(void *addr) {
        free(addr);
}

inline
void* swift_malloc(swift_thread_t *thread, size_t size) {
        //TODO: check
        void *ret;

#ifdef SWIFT_USE_CUSTOM_ALLOCATOR
        swift_allocator_list_t **b, *elem;
        size_t sz;

        /* allocate space for the allocator's list */
        size += sizeof(swift_allocator_list_t);
        size = SWIFT_MEM_ALIGN(size);

        for (b = thread->buckets, sz = SWIFT_MALLOC_ALIGN;
                 sz < size;
                 sz <<= 1, b++) {
        }

        elem = *b;
        if (elem) {
                /* reuse from the bucket */
                *b = elem->next;
                SWIFT_LOG(INFO, "get %d from bucket %p  --> elem=%p\n", sz, (void*)b, (void*) elem);

        } else {
                /* get from the heap */
                elem = thread->heap_top;
                thread->heap_top = ((char*)thread->heap_top + SWIFT_MEM_ALIGN(sz));

                SWIFT_LOG(INFO, "get %d from heap for bucket %p --> elem=%p\n", sz, (void*)b,  (void*)elem);
        }

        /* store from what bucket this came from */
        elem->next = b;

        swift_assert((void*)elem < thread->heap_limit);

        ret = elem;
        ret = (void*) ((char*)ret + SWIFT_MEM_ALIGN(sizeof(swift_allocator_list_t)));

        SWIFT_LOG(INFO, "malloc return %p\n", ret);
#else
        ret = malloc(size);
#endif

        return ret;
}

inline
void swift_free(swift_thread_t *thread, void *addr) {
#ifdef SWIFT_USE_CUSTOM_ALLOCATOR
        swift_allocator_list_t *elem;
        swift_allocator_list_t **b;

        elem = (swift_allocator_list_t*) ((char*)addr - SWIFT_MEM_ALIGN(sizeof(swift_allocator_list_t)));

        /* get the bucket this element came from */
        b = elem->next;

        SWIFT_LOG(INFO, "return addr=%p to bucket %p (*b=%p) --> elem=%p\n", addr, (void*)b, (void*)*b, (void*)elem);

        /* return to the bucket */
        elem->next = *b;
        *b = elem;
#else
        //free(addr);
#endif
}

inline
void swift_retire_frame(swift_thread_t *thread, swift_frame_t *frame) {
        /* drop into the pool */
        swift_free(thread, frame);
}