diff options
| author | Florian Fischer <florian.fl.fischer@fau.de> | 2020-02-26 16:19:57 +0100 |
|---|---|---|
| committer | Florian Fischer <florian.fl.fischer@fau.de> | 2020-02-26 16:19:57 +0100 |
| commit | ff0be98bad9549eeb06355e848bd868c792203a5 (patch) | |
| tree | a0444b0320b1b0d80b9b4215d16805113fb63871 /src/speedymalloc.c | |
| parent | da69bb193b5249c3f489b18a746e3800b36e37d0 (diff) | |
| download | allocbench-ff0be98bad9549eeb06355e848bd868c792203a5.tar.gz allocbench-ff0be98bad9549eeb06355e848bd868c792203a5.zip | |
externalize speedymalloc
Diffstat (limited to 'src/speedymalloc.c')
| -rw-r--r-- | src/speedymalloc.c | 171 |
1 files changed, 0 insertions, 171 deletions
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 <assert.h> -#include <errno.h> -#include <stddef.h> /* NULL, size_t */ -#include <stdint.h> /* uintptr_t */ -#include <stdio.h> /* fprintf */ -#include <stdlib.h> /* exit */ -#include <unistd.h> /* sysconf(_SC_PAGESIZE) */ -#include <string.h> /* memset */ -#include <sys/mman.h> /* 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 |
