/* nsmalloc: Simple pooled malloc() thing. */
#include <stdlib.h>
-#include <stdio.h>
#include "nsmalloc.h"
#define __NSMALLOC_C
#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 nsmpool nsmpools[MAXPOOL];
-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];
-
-void nsmstats(int hookhum, void *arg);
-
-void initnsmalloc(void) {
- registerhook(HOOK_CORE_STATSREQUEST, &nsmstats);
-}
+#ifndef USE_NSMALLOC_VALGRIND
void *nsmalloc(unsigned int poolid, size_t size) {
struct nsminfo *nsmp;
return NULL;
nsmp->size=size;
- pools[poolid].size+=size;
- pools[poolid].count++;
+ nsmpools[poolid].size+=size;
+ nsmpools[poolid].count++;
- 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;
+ nsmp->next=nsmpools[poolid].first.next;
+ nsmp->prev=&nsmpools[poolid].first;
+ if (nsmpools[poolid].first.next)
+ nsmpools[poolid].first.next->prev=nsmp;
+ nsmpools[poolid].first.next=nsmp;
return (void *)nsmp->data;
}
void nsfree(unsigned int poolid, void *ptr) {
struct nsminfo *nsmp;
- if (poolid >= MAXPOOL)
+ if (!ptr || poolid >= MAXPOOL)
return;
/* evil */
if (nsmp->next)
nsmp->next->prev=nsmp->prev;
- pools[poolid].size-=nsmp->size;
- pools[poolid].count--;
+ nsmpools[poolid].size-=nsmp->size;
+ nsmpools[poolid].count--;
free(nsmp);
return;
if (!nsmpn)
return NULL;
- pools[poolid].size+=size-nsmpn->size;
+ nsmpools[poolid].size+=size-nsmpn->size;
nsmpn->size=size;
/* always set as we have a sentinel */
if (poolid >= MAXPOOL)
return;
- for (nsmp=pools[poolid].first.next;nsmp;nsmp=nnsmp) {
+ for (nsmp=nsmpools[poolid].first.next;nsmp;nsmp=nnsmp) {
nnsmp=nsmp->next;
free(nsmp);
}
- pools[poolid].first.next=NULL;
- pools[poolid].size=0;
- pools[poolid].count=0;
+ nsmpools[poolid].first.next=NULL;
+ nsmpools[poolid].size=0;
+ nsmpools[poolid].count=0;
}
void nscheckfreeall(unsigned int poolid) {
if (poolid >= MAXPOOL)
return;
- if (pools[poolid].first.next) {
- Error("core",ERR_INFO,"nsmalloc: Blocks still allocated in pool #%d (%s): %lub, %lu items",poolid,poolnames[poolid]?poolnames[poolid]:"??",pools[poolid].size,pools[poolid].count);
+ if (nsmpools[poolid].first.next) {
+ 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);
}
}
unsigned int i;
for (i=0;i<MAXPOOL;i++)
- nscheckfreeall(i);
+ nsfreeall(i);
}
-static char *formatmbuf(unsigned long count, size_t size, size_t realsize) {
- static char buf[1024];
+#else
- snprintf(buf, sizeof(buf), "%lu items, %luKb allocated for %luKb space, %luKb (%0.2f%%) overhead", count, (unsigned long)size / 1024, (unsigned long)realsize / 1024, (unsigned long)(realsize - size) / 1024, (double)(realsize - size) / (double)size * 100);
- return buf;
+void *nsmalloc(unsigned int poolid, size_t size) {
+ return malloc(size);
}
-void nsmstats(int hookhum, void *arg) {
- int i;
- char buf[1024];
- unsigned long totalcount = 0;
- size_t totalsize = 0, totalrealsize = 0;
- long level = (long)arg;
+void *nsrealloc(unsigned int poolid, void *ptr, size_t size) {
+ return realloc(ptr, size);
+}
- for (i=0;i<MAXPOOL;i++) {
- struct nsmpool *pool=&pools[i];
- size_t realsize;
+void nsfree(unsigned int poolid, void *ptr) {
+ if(ptr)
+ free(ptr);
+}
- if (!pool->count)
- continue;
+void nsfreeall(unsigned int poolid) {
+}
- realsize=pool->size + pool->count * sizeof(struct nsminfo) + sizeof(struct nsmpool);
+void nsexit(void) {
+}
- if(level > 10) {
- snprintf(buf, sizeof(buf), "NSMalloc: pool %2d (%s): %s", i, poolnames[i]?poolnames[i]:"??", formatmbuf(pool->count, pool->size, realsize));
- triggerhook(HOOK_CORE_STATSREPLY, buf);
- }
+void nscheckfreeall(unsigned int poolid) {
+}
- totalsize+=pool->size;
- totalrealsize+=realsize;
- totalcount+=pool->count;
- }
+#endif
- snprintf(buf, sizeof(buf), "NSMalloc: pool totals: %s", formatmbuf(totalcount, totalsize, totalrealsize));
- triggerhook(HOOK_CORE_STATSREPLY, buf);
-}