/* nsmalloc: Simple pooled malloc() thing. */
#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
#include "nsmalloc.h"
#define __NSMALLOC_C
#undef __NSMALLOC_H
#include "nsmalloc.h"
+#include "../lib/valgrind.h"
+#include "../lib/memcheck.h"
+#include "../core/hooks.h"
#include "../core/error.h"
-void *nsmalloc(unsigned int poolid, size_t size);
-void nsfree(unsigned int poolid, void *ptr);
-void nsfreeall(unsigned int poolid);
-
-struct nsminfo {
- struct nsminfo *next;
- struct nsminfo *prev;
-
- size_t size;
- char data[];
-};
-
-struct nsmpool {
- struct nsminfo first;
-
- unsigned long count;
- size_t size;
-};
-
-struct nsmpool pools[MAXPOOL];
+struct nsmpool nsmpools[MAXPOOL];
void *nsmalloc(unsigned int poolid, size_t size) {
struct nsminfo *nsmp;
if (!nsmp)
return NULL;
-
+
+ VALGRIND_CREATE_MEMPOOL(nsmp, 0, 0);
+
nsmp->size=size;
- pools[poolid].size+=size;
- pools[poolid].count++;
+ nsmpools[poolid].size+=size;
+ nsmpools[poolid].count++;
+
+ if (nsmpools[poolid].blocks) {
+ nsmpools[poolid].blocks->prev = nsmp;
+ }
+ nsmp->next=nsmpools[poolid].blocks;
+ nsmp->prev=NULL;
+ nsmpools[poolid].blocks=nsmp;
- nsmp->next=pools[poolid].first.next;
- nsmp->prev=&pools[poolid].first;
- if (pools[poolid].first.next)
- pools[poolid].first.next->prev=nsmp;
- pools[poolid].first.next=nsmp;
+ VALGRIND_MEMPOOL_ALLOC(nsmp, nsmp->data, nsmp->size);
+
+ nsmp->redzone = REDZONE_MAGIC;
+ VALGRIND_MAKE_MEM_NOACCESS(&nsmp->redzone, sizeof(nsmp->redzone));
return (void *)nsmp->data;
}
-/* we dump core on ptr == NULL */
+void *nscalloc(unsigned int poolid, size_t nmemb, size_t size) {
+ size_t total = nmemb * size;
+ void *m;
+
+ m = nsmalloc(poolid, total);
+ if(!m)
+ return NULL;
+
+ memset(m, 0, total);
+
+ return m;
+}
+
void nsfree(unsigned int poolid, void *ptr) {
struct nsminfo *nsmp;
- if (poolid >= MAXPOOL)
+ if (!ptr || poolid >= MAXPOOL)
return;
/* evil */
nsmp=(struct nsminfo*)ptr - 1;
- /* always set as we have a sentinel */
- nsmp->prev->next=nsmp->next;
- if (nsmp->next)
- nsmp->next->prev=nsmp->prev;
+ VALGRIND_MAKE_MEM_DEFINED(&nsmp->redzone, sizeof(nsmp->redzone));
+ assert(nsmp->redzone == REDZONE_MAGIC);
- pools[poolid].size-=nsmp->size;
- pools[poolid].count--;
+ if (nsmp->prev) {
+ nsmp->prev->next = nsmp->next;
+ } else
+ nsmpools[poolid].blocks = NULL;
+ if (nsmp->next) {
+ nsmp->next->prev = nsmp->prev;
+ }
+
+ nsmpools[poolid].size-=nsmp->size;
+ nsmpools[poolid].count--;
+
+ VALGRIND_MEMPOOL_FREE(nsmp, nsmp->data);
free(nsmp);
+ VALGRIND_DESTROY_MEMPOOL(nsmp);
+
return;
}
/* evil */
nsmp=(struct nsminfo *)ptr - 1;
+ VALGRIND_MAKE_MEM_DEFINED(nsmp, sizeof(struct nsminfo));
+
if (size == nsmp->size)
return (void *)nsmp->data;
if (!nsmpn)
return NULL;
- pools[poolid].size+=size-nsmpn->size;
+ VALGRIND_MOVE_MEMPOOL(nsmp, nsmpn);
+
+ nsmpools[poolid].size+=size-nsmpn->size;
nsmpn->size=size;
- /* always set as we have a sentinel */
- nsmpn->prev->next=nsmpn;
+ if (nsmpn->prev) {
+ nsmpn->prev->next=nsmpn;
+ } else
+ nsmpools[poolid].blocks = nsmpn;
- if (nsmpn->next)
+ if (nsmpn->next) {
nsmpn->next->prev=nsmpn;
+ }
+
+ VALGRIND_MEMPOOL_CHANGE(nsmpn, nsmp->data, nsmpn->data, nsmpn->size);
return (void *)nsmpn->data;
}
if (poolid >= MAXPOOL)
return;
- for (nsmp=pools[poolid].first.next;nsmp;nsmp=nnsmp) {
+ for (nsmp=nsmpools[poolid].blocks;nsmp;nsmp=nnsmp) {
nnsmp=nsmp->next;
+ VALGRIND_MEMPOOL_FREE(nsmp, nsmp->data);
free(nsmp);
+ VALGRIND_DESTROY_MEMPOOL(nsmp);
}
- pools[poolid].first.next=NULL;
- pools[poolid].size=0;
- pools[poolid].count=0;
+ nsmpools[poolid].blocks=NULL;
+ nsmpools[poolid].size=0;
+ nsmpools[poolid].count=0;
}
-void nsexit() {
- unsigned int i;
-
- for (i=0;i<MAXPOOL;i++) {
- if (pools[i].first.next) {
- Error("core",ERR_INFO,"nsmalloc: Blocks still allocated in pool #%d (%lub, %lu items)\n",i,pools[i].size, pools[i].count);
- nsfreeall(i);
- }
+void nscheckfreeall(unsigned int poolid) {
+ if (poolid >= MAXPOOL)
+ return;
+
+ if (nsmpools[poolid].blocks) {
+ Error("core",ERR_INFO,"nsmalloc: Blocks still allocated in pool #%d (%s): %zub, %lu items",poolid,nsmpoolnames[poolid]?nsmpoolnames[poolid]:"??",nsmpools[poolid].size,nsmpools[poolid].count);
+ nsfreeall(poolid);
}
}
-int nspoolstats(unsigned int poolid, size_t *size, unsigned long *count, char **poolname, size_t *realsize) {
- if (poolid >= MAXPOOL)
- return 0;
-
- *size = pools[poolid].size;
- *realsize = pools[poolid].size + pools[poolid].count * sizeof(struct nsminfo) + sizeof(struct nsmpool);
- *count = pools[poolid].count;
- *poolname = poolnames[poolid];
+void nsinit(void) {
+ memset(nsmpools, 0, sizeof(nsmpools));
+}
- return 1;
+void nsexit(void) {
+ unsigned int i;
+
+ for (i=0;i<MAXPOOL;i++)
+ nsfreeall(i);
}
+