From: Chris Porter <redacted>
Date: Fri, 29 Feb 2008 21:04:49 +0000 (+0000)
Subject: Add malloc checking to lua/nterfacer/trojanscan.
X-Git-Url: https://jfr.im/git/irc/quakenet/newserv.git/commitdiff_plain/f8d5cfcf45e5ae05ebe3ca8ac759fd2465faded7

Add malloc checking to lua/nterfacer/trojanscan.
Cosmetic alteration to nsmalloc output.
---

diff --git a/core/nsmalloc.c b/core/nsmalloc.c
index e4b03334..3c23f03b 100644
--- a/core/nsmalloc.c
+++ b/core/nsmalloc.c
@@ -129,15 +129,21 @@ void nsfreeall(unsigned int poolid) {
   pools[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);
+    nsfreeall(poolid);
+  }
+}
+
 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);
-    }
-  }
+  for (i=0;i<MAXPOOL;i++)
+    nscheckfreeall(i);
 }
 
 int nspoolstats(unsigned int poolid, size_t *size, unsigned long *count, char **poolname, size_t *realsize) {
diff --git a/core/nsmalloc.h b/core/nsmalloc.h
index 8b25aa1a..ea5d862c 100644
--- a/core/nsmalloc.h
+++ b/core/nsmalloc.h
@@ -20,6 +20,7 @@ void nsfreeall(unsigned int poolid);
 void nsexit(void);
 void *nsrealloc(unsigned int poolid, void *ptr, size_t size);
 int nspoolstats(unsigned int poolid, size_t *size, unsigned long *count, char **poolname, size_t *realsize);
+void nscheckfreeall(unsigned int poolid);
 
 #define MAXPOOL		100
 
diff --git a/lua/lua.c b/lua/lua.c
index c23da67a..a3d54f3b 100644
--- a/lua/lua.c
+++ b/lua/lua.c
@@ -151,6 +151,7 @@ void _fini() {
   freesstring(dummy.name);
 
   lua_freedebugsocket();
+  nscheckfreeall(POOL_LUA);
 }
 
 void lua_loadscripts(void) {
diff --git a/nterfacer/nterfacer.c b/nterfacer/nterfacer.c
index cf95e8f8..6e68d0c1 100644
--- a/nterfacer/nterfacer.c
+++ b/nterfacer/nterfacer.c
@@ -128,6 +128,7 @@ void _fini(void) {
   }
 
   nrl = nterface_close_log(nrl);
+  nscheckfreeall(POOL_NTERFACER);
 }
 
 int load_permits(void) {
diff --git a/trojanscan/trojanscan.c b/trojanscan/trojanscan.c
index efdb804d..960594b5 100644
--- a/trojanscan/trojanscan.c
+++ b/trojanscan/trojanscan.c
@@ -129,6 +129,8 @@ void _fini(void) {
   for (i=0;i<trojanscan_tailpoolsize;i++)
     freesstring(trojanscan_tailpool[i]);
   trojanscan_database_close();
+
+  nscheckfreeall(POOL_TROJANSCAN);
 }
 
 static void trojanscan_connect_nick(void *arg) {