From ff0be98bad9549eeb06355e848bd868c792203a5 Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Wed, 26 Feb 2020 16:19:57 +0100 Subject: externalize speedymalloc --- src/speedymalloc.c | 171 ----------------------------------------------------- 1 file changed, 171 deletions(-) delete mode 100644 src/speedymalloc.c (limited to 'src/speedymalloc.c') diff --git a/src/speedymalloc.c b/src/speedymalloc.c deleted file mode 100644 index 7df3c5a..0000000 --- a/src/speedymalloc.c +++ /dev/null @@ -1,171 +0,0 @@ -#include -#include -#include /* NULL, size_t */ -#include /* uintptr_t */ -#include /* fprintf */ -#include /* exit */ -#include /* sysconf(_SC_PAGESIZE) */ -#include /* memset */ -#include /* mmap */ - -#define MIN_ALIGNMENT 16 - -#ifndef MEMSIZE -#define MEMSIZE 1024*4*1024*1024l -#endif - -#ifndef NO_WILLNEED -#define WILLNEED_SIZE 32 * 1024 * 1024 -#endif - -// sizeof(tls_t) == 4096 -#define CACHE_BINS 511 -// max cached object: 511 * 64 - 1 = 32703 -#define CACHE_BIN_SEPERATION 64 - -#define unlikely(x) __builtin_expect((x),0) - -#ifdef __cplusplus -extern "C" { -#endif - - -typedef struct chunk { - size_t size; // Size header field for internal use - struct chunk* next; -} chunk_t; - -static inline chunk_t* ptr2chunk(void* ptr) { - return (chunk_t*)((uintptr_t)ptr - sizeof(size_t)); -} - -static inline void* chunk2ptr (chunk_t* chunk) { - return (void*)((uintptr_t)chunk + sizeof(size_t)); -} - -typedef struct TLStates { - uintptr_t ptr; - chunk_t* bins[CACHE_BINS]; -} tls_t; - -__thread tls_t* tls; -__thread uintptr_t next_willneed; - -static inline int size2bin(size_t size) { - assert(size > 0 && size < CACHE_BINS * CACHE_BIN_SEPERATION); - return (size - 1) / CACHE_BIN_SEPERATION; -} - -static inline size_t bin2size(int bin) { - assert(bin >= 0 && bin < CACHE_BINS); - return (bin + 1) * CACHE_BIN_SEPERATION; -} - -static void init_tls(void) { - void *mem = mmap(NULL, MEMSIZE, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0); - if (mem == MAP_FAILED) { - perror("mmap"); - exit(1); - } - tls = (tls_t*)mem; - - tls->ptr = ((uintptr_t)tls) + sizeof(tls_t); -#ifndef NO_WILLNEED - next_willneed = tls->ptr; -#endif -} - -static void* bump_alloc(size_t size) { - // allocate size header - tls->ptr += sizeof(size_t); - - // align ptr - size_t mask = MIN_ALIGNMENT -1; - tls->ptr = (tls->ptr + mask) & ~mask; - -#ifndef NO_WILLNEED - if(unlikely(tls->ptr >= next_willneed)) { - if (madvise((void*)next_willneed, WILLNEED_SIZE, MADV_WILLNEED) != 0) { - perror("madvice"); - } - next_willneed += WILLNEED_SIZE; - } -#endif - - void* ptr = (void*)tls->ptr; - ptr2chunk(ptr)->size = size; - tls->ptr += size; - - return ptr; -} - -void* malloc(size_t size) { - if (unlikely(tls == NULL)) { - init_tls(); - } - - // cached sizes - if (size < CACHE_BINS * CACHE_BIN_SEPERATION) { - int bin = size2bin(size); - if (tls->bins[bin] != NULL) { - chunk_t* chunk = tls->bins[bin]; - // remove first chunk from list - tls->bins[bin] = chunk->next; - return chunk2ptr(chunk); - } - return bump_alloc(bin2size(bin)); - } - - return bump_alloc(size); -} - -void free(void* ptr) { - if (unlikely(tls == NULL)) { - init_tls(); - } - - if (ptr == NULL) { - return; - } - - chunk_t* chunk = ptr2chunk(ptr); - - if (chunk->size < CACHE_BINS * CACHE_BIN_SEPERATION) { - int bin = size2bin(chunk->size); - chunk->next = tls->bins[bin]; - tls->bins[bin] = chunk; - } -} - -void* memalign(size_t alignment, size_t size) { - /* if not power of two */ - if (!((alignment != 0) && !(alignment & (alignment - 1)))) { - return NULL; - } - - if (unlikely(tls == NULL)) { - init_tls(); - } - - // allocate size header - tls->ptr += sizeof(size_t); - - // align returned pointer - size_t mask = alignment - 1; - tls->ptr = (tls->ptr + mask) & ~mask; - - void* ptr = (void*)tls->ptr; - ptr2chunk(ptr)->size = size; - tls->ptr += size; - - return ptr; -} - -void malloc_stats() { - fprintf(stderr, "speedymalloc allocator by muhq\n"); - fprintf(stderr, "Memsize: %zu, start address: %p, bump pointer %p\n", MEMSIZE, tls, tls->ptr); -} - -#ifdef __cplusplus -} -#endif -- cgit v1.2.3