|
Packit |
4e8bc4 |
/* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
|
|
Packit |
4e8bc4 |
#include <stdlib.h>
|
|
Packit |
4e8bc4 |
#include <string.h>
|
|
Packit |
4e8bc4 |
#include <inttypes.h>
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
#ifndef NDEBUG
|
|
Packit |
4e8bc4 |
#include <signal.h>
|
|
Packit |
4e8bc4 |
#endif
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
#include "cache.h"
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
#ifndef NDEBUG
|
|
Packit |
4e8bc4 |
const uint64_t redzone_pattern = 0xdeadbeefcafebabe;
|
|
Packit |
4e8bc4 |
int cache_error = 0;
|
|
Packit |
4e8bc4 |
#endif
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
const int initial_pool_size = 64;
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
cache_t* cache_create(const char *name, size_t bufsize, size_t align,
|
|
Packit |
4e8bc4 |
cache_constructor_t* constructor,
|
|
Packit |
4e8bc4 |
cache_destructor_t* destructor) {
|
|
Packit |
4e8bc4 |
cache_t* ret = calloc(1, sizeof(cache_t));
|
|
Packit |
4e8bc4 |
char* nm = strdup(name);
|
|
Packit |
4e8bc4 |
void** ptr = calloc(initial_pool_size, sizeof(void*));
|
|
Packit |
4e8bc4 |
if (ret == NULL || nm == NULL || ptr == NULL ||
|
|
Packit |
4e8bc4 |
pthread_mutex_init(&ret->mutex, NULL) == -1) {
|
|
Packit |
4e8bc4 |
free(ret);
|
|
Packit |
4e8bc4 |
free(nm);
|
|
Packit |
4e8bc4 |
free(ptr);
|
|
Packit |
4e8bc4 |
return NULL;
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
ret->name = nm;
|
|
Packit |
4e8bc4 |
ret->ptr = ptr;
|
|
Packit |
4e8bc4 |
ret->freetotal = initial_pool_size;
|
|
Packit |
4e8bc4 |
ret->constructor = constructor;
|
|
Packit |
4e8bc4 |
ret->destructor = destructor;
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
#ifndef NDEBUG
|
|
Packit |
4e8bc4 |
ret->bufsize = bufsize + 2 * sizeof(redzone_pattern);
|
|
Packit |
4e8bc4 |
#else
|
|
Packit |
4e8bc4 |
ret->bufsize = bufsize;
|
|
Packit |
4e8bc4 |
#endif
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
return ret;
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
static inline void* get_object(void *ptr) {
|
|
Packit |
4e8bc4 |
#ifndef NDEBUG
|
|
Packit |
4e8bc4 |
uint64_t *pre = ptr;
|
|
Packit |
4e8bc4 |
return pre + 1;
|
|
Packit |
4e8bc4 |
#else
|
|
Packit |
4e8bc4 |
return ptr;
|
|
Packit |
4e8bc4 |
#endif
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
void cache_destroy(cache_t *cache) {
|
|
Packit |
4e8bc4 |
while (cache->freecurr > 0) {
|
|
Packit |
4e8bc4 |
void *ptr = cache->ptr[--cache->freecurr];
|
|
Packit |
4e8bc4 |
if (cache->destructor) {
|
|
Packit |
4e8bc4 |
cache->destructor(get_object(ptr), NULL);
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
free(ptr);
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
free(cache->name);
|
|
Packit |
4e8bc4 |
free(cache->ptr);
|
|
Packit |
4e8bc4 |
pthread_mutex_destroy(&cache->mutex);
|
|
Packit |
4e8bc4 |
free(cache);
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
void* cache_alloc(cache_t *cache) {
|
|
Packit |
4e8bc4 |
void *ret;
|
|
Packit |
4e8bc4 |
pthread_mutex_lock(&cache->mutex);
|
|
Packit |
4e8bc4 |
ret = do_cache_alloc(cache);
|
|
Packit |
4e8bc4 |
pthread_mutex_unlock(&cache->mutex);
|
|
Packit |
4e8bc4 |
return ret;
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
void* do_cache_alloc(cache_t *cache) {
|
|
Packit |
4e8bc4 |
void *ret;
|
|
Packit |
4e8bc4 |
void *object;
|
|
Packit |
4e8bc4 |
if (cache->freecurr > 0) {
|
|
Packit |
4e8bc4 |
ret = cache->ptr[--cache->freecurr];
|
|
Packit |
4e8bc4 |
object = get_object(ret);
|
|
Packit |
4e8bc4 |
} else {
|
|
Packit |
4e8bc4 |
object = ret = malloc(cache->bufsize);
|
|
Packit |
4e8bc4 |
if (ret != NULL) {
|
|
Packit |
4e8bc4 |
object = get_object(ret);
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
if (cache->constructor != NULL &&
|
|
Packit |
4e8bc4 |
cache->constructor(object, NULL, 0) != 0) {
|
|
Packit |
4e8bc4 |
free(ret);
|
|
Packit |
4e8bc4 |
object = NULL;
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
#ifndef NDEBUG
|
|
Packit |
4e8bc4 |
if (object != NULL) {
|
|
Packit |
4e8bc4 |
/* add a simple form of buffer-check */
|
|
Packit |
4e8bc4 |
uint64_t *pre = ret;
|
|
Packit |
4e8bc4 |
*pre = redzone_pattern;
|
|
Packit |
4e8bc4 |
ret = pre+1;
|
|
Packit |
4e8bc4 |
memcpy(((char*)ret) + cache->bufsize - (2 * sizeof(redzone_pattern)),
|
|
Packit |
4e8bc4 |
&redzone_pattern, sizeof(redzone_pattern));
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
#endif
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
return object;
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
void cache_free(cache_t *cache, void *ptr) {
|
|
Packit |
4e8bc4 |
pthread_mutex_lock(&cache->mutex);
|
|
Packit |
4e8bc4 |
do_cache_free(cache, ptr);
|
|
Packit |
4e8bc4 |
pthread_mutex_unlock(&cache->mutex);
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
void do_cache_free(cache_t *cache, void *ptr) {
|
|
Packit |
4e8bc4 |
#ifndef NDEBUG
|
|
Packit |
4e8bc4 |
/* validate redzone... */
|
|
Packit |
4e8bc4 |
if (memcmp(((char*)ptr) + cache->bufsize - (2 * sizeof(redzone_pattern)),
|
|
Packit |
4e8bc4 |
&redzone_pattern, sizeof(redzone_pattern)) != 0) {
|
|
Packit |
4e8bc4 |
raise(SIGABRT);
|
|
Packit |
4e8bc4 |
cache_error = 1;
|
|
Packit |
4e8bc4 |
return;
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
uint64_t *pre = ptr;
|
|
Packit |
4e8bc4 |
--pre;
|
|
Packit |
4e8bc4 |
if (*pre != redzone_pattern) {
|
|
Packit |
4e8bc4 |
raise(SIGABRT);
|
|
Packit |
4e8bc4 |
cache_error = -1;
|
|
Packit |
4e8bc4 |
return;
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
ptr = pre;
|
|
Packit |
4e8bc4 |
#endif
|
|
Packit |
4e8bc4 |
if (cache->freecurr < cache->freetotal) {
|
|
Packit |
4e8bc4 |
cache->ptr[cache->freecurr++] = ptr;
|
|
Packit |
4e8bc4 |
} else {
|
|
Packit |
4e8bc4 |
/* try to enlarge free connections array */
|
|
Packit |
4e8bc4 |
size_t newtotal = cache->freetotal * 2;
|
|
Packit |
4e8bc4 |
void **new_free = realloc(cache->ptr, sizeof(char *) * newtotal);
|
|
Packit |
4e8bc4 |
if (new_free) {
|
|
Packit |
4e8bc4 |
cache->freetotal = newtotal;
|
|
Packit |
4e8bc4 |
cache->ptr = new_free;
|
|
Packit |
4e8bc4 |
cache->ptr[cache->freecurr++] = ptr;
|
|
Packit |
4e8bc4 |
} else {
|
|
Packit |
4e8bc4 |
if (cache->destructor) {
|
|
Packit |
4e8bc4 |
cache->destructor(ptr, NULL);
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
free(ptr);
|
|
Packit |
4e8bc4 |
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
}
|
|
Packit |
4e8bc4 |
|