--- libaitio/src/aitio.c	2012/05/23 11:59:51	1.10.6.2
+++ libaitio/src/aitio.c	2012/11/15 23:23:54	1.13
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@elwix.org>
 *
 * $Author: misho $
-* $Id: aitio.c,v 1.10.6.2 2012/05/23 11:59:51 misho Exp $
+* $Id: aitio.c,v 1.13 2012/11/15 23:23:54 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -47,11 +47,21 @@ SUCH DAMAGE.
 
 
 int io_Debug;
+mpool_t *io_mpool;
 
 
+/* Memory management */
+
+void *(*io_malloc)(size_t) = malloc;
+void *(*io_calloc)(size_t, size_t) = calloc;
+void *(*io_realloc)(void*, size_t) = realloc;
+char *(*io_strdup)(const char*) = strdup;
+void (*io_free)(void*) = free;
+
+
 #pragma GCC visibility push(hidden)
 
-int io_mpool;
+int use_mm;
 
 int io_Errno;
 char io_Error[STRSIZ];
@@ -86,15 +96,92 @@ io_SetErr(int eno, char *estr, ...)
 	va_end(lst);
 }
 
-// mpool_inuse() Check for mpool usage
+// io_mm_inuse() Check for memory management model
 inline int
-mpool_inuse()
+io_mm_inuse()
 {
-	return io_mpool;
+	return use_mm & IO_MPOOL;
 }
 
 
+// init libaitio routine
+__attribute__((constructor)) void
+_io_init()
+{
+	ioLibInit(IO_MPOOL, 0);
+}
+
+// fini libaitio routine
+__attribute__((destructor)) void
+_io_fini()
+{
+	ioLibFini();
+}
+
 /*
+ * ioLibInit() - Init libaitio library memory management
+ *
+ * @mm = memory management (IO_SYSM or IO_MPOOL)
+ * @maxmem = memory limit
+ * return: -1 error or !=-1 used memory management model
+ */
+inline int
+ioLibInit(int mm, u_long maxmem)
+{
+	switch (mm) {
+		case IO_MPOOL:	/* mpool */
+			io_mpool = mpool_init(maxmem);
+			if (io_mpool) {
+				io_malloc = mpool_xmalloc;
+				io_calloc = mpool_xcalloc;
+				io_realloc = mpool_xrealloc;
+				io_strdup = mpool_xstrdup;
+				io_free = mpool_xfree;
+				use_mm = mm;
+				break;
+			} else {
+				#undef USE_MPOOL
+			}
+		case IO_SYSM: 	/* system */
+			io_malloc = malloc;
+			io_calloc = calloc;
+			io_realloc = realloc;
+			io_strdup = strdup;
+			io_free = free;
+			use_mm = mm;
+			break;
+		default:	/* not supported */
+			io_SetErr(EINVAL, "Not supported memory management");
+			return -1;
+	}
+
+	return use_mm;
+}
+
+/*
+ * ioLibFini() - Finish libaitio library memory management
+ *
+ * return: none
+ */
+inline void
+ioLibFini()
+{
+	switch (use_mm) {
+		case IO_MPOOL:
+			io_malloc = malloc;
+			io_calloc = calloc;
+			io_realloc = realloc;
+			io_strdup = strdup;
+			io_free = free;
+			use_mm = IO_SYSM;
+
+			mpool_destroy(&io_mpool);
+			break;
+	}
+}
+
+
+/*
  * ioPromptRead() - Read data from input h[0] with prompt to output h[1]
  *
  * @h = file handles h[0] = input, h[1] = output, if NULL use stdin, stdout
@@ -321,7 +408,7 @@ ioRegexGet(const char *csRegex, const char *csData, ch
  * @csRegex = Regulare expression pattern
  * @csData = Source data
  * @csNew = Data for replace
- * return: NULL not match or error; !=NULL allocated new string, must be xfree after use!
+ * return: NULL not match or error; !=NULL allocated new string, must be io_free after use!
 */
 char *
 ioRegexReplace(const char *csRegex, const char *csData, const char *csNew)
@@ -337,7 +424,7 @@ ioRegexReplace(const char *csRegex, const char *csData
 
 	// ___ before match
 	len = sp + 1;
-	str = xmalloc(len);
+	str = io_malloc(len);
 	if (!str) {
 		LOGERR;
 		return NULL;
@@ -346,7 +433,7 @@ ioRegexReplace(const char *csRegex, const char *csData
 	// * replace match *
 	if (csNew) {
 		len += strlen(csNew);
-		str = xrealloc(str, len);
+		str = io_realloc(str, len);
 		if (!str) {
 			LOGERR;
 			return NULL;
@@ -355,7 +442,7 @@ ioRegexReplace(const char *csRegex, const char *csData
 	}
 	// after match ___
 	len += strlen(csData) - ep;
-	str = xrealloc(str, len);
+	str = io_realloc(str, len);
 	if (!str) {
 		LOGERR;
 		return NULL;
@@ -369,7 +456,7 @@ ioRegexReplace(const char *csRegex, const char *csData
  * ioStrAst() - Function for evaluate string like asterisk variable "{text[:[-]#[:#]]}"
  *
  * @csString = Input string
- * return: NULL error, !=NULL Allocated new string evaluated from input string, must be xfree()
+ * return: NULL error, !=NULL Allocated new string evaluated from input string, must be io_free()
 */
 char *
 ioStrAst(const char *csString)
@@ -387,7 +474,7 @@ ioStrAst(const char *csString)
 		io_Errno = EINVAL;
 		return NULL;
 	} else {
-		str = strdup(strchr(csString, '{') + 1);
+		str = io_strdup(strchr(csString, '{') + 1);
 		*strrchr(str, '}') = 0;
 	}
 
@@ -409,8 +496,8 @@ ioStrAst(const char *csString)
 		/* ok, clear show */
 		ext = str;
 
-	out = strdup(ext);
-	xfree(str);
+	out = io_strdup(ext);
+	io_free(str);
 
 	return out;
 }
@@ -432,7 +519,7 @@ ioMkDir(const char *csDir, int mode)
 	if (!csDir)
 		return cx;
 
-	str = strdup(csDir);
+	str = io_strdup(csDir);
 	if (!str) {
 		LOGERR;
 		return cx;
@@ -460,7 +547,7 @@ ioMkDir(const char *csDir, int mode)
 	}
 end:
 	chdir(szOld);
-	xfree(str);
+	io_free(str);
 	return cx;
 }