From 736ed18513ec085f3cdd4c9675fae693487e330d Mon Sep 17 00:00:00 2001
From: =?utf8?q?G=C3=A9=20van=20Geldorp?= <ge@gse.nl>
Date: Sun, 1 Jan 2006 11:48:11 +0000
Subject: [PATCH] Fix memory tracking

svn path=/trunk/; revision=20505
---
 .../lib/oskittcp/include/freebsd/src/sys/sys/malloc.h     | 4 ++--
 .../lib/oskittcp/include/freebsd/src/sys/sys/mbuf.h       | 4 ++--
 reactos/drivers/lib/oskittcp/include/oskitfreebsd.h       | 2 +-
 reactos/drivers/lib/oskittcp/include/oskittcp.h           | 4 ++--
 reactos/drivers/lib/oskittcp/oskittcp.xml                 | 1 +
 reactos/drivers/lib/oskittcp/oskittcp/interface.c         | 8 ++++----
 6 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/reactos/drivers/lib/oskittcp/include/freebsd/src/sys/sys/malloc.h b/reactos/drivers/lib/oskittcp/include/freebsd/src/sys/sys/malloc.h
index 8a44caa1226..ae4a946ed5b 100644
--- a/reactos/drivers/lib/oskittcp/include/freebsd/src/sys/sys/malloc.h
+++ b/reactos/drivers/lib/oskittcp/include/freebsd/src/sys/sys/malloc.h
@@ -329,8 +329,8 @@ extern struct kmembuckets bucket[];
 extern void *malloc __P((unsigned long size, ...));
 extern void free __P((void *addr, ...));
 #else
-#define malloc fbsd_malloc
-#define free   fbsd_free
+#define malloc(size, flags, id) fbsd_malloc(size, __FILE__, __LINE__)
+#define free(area, flags)   fbsd_free(area, __FILE__, __LINE__)
 #endif
 #endif /* KERNEL */
 #endif /* !_SYS_MALLOC_H_ */
diff --git a/reactos/drivers/lib/oskittcp/include/freebsd/src/sys/sys/mbuf.h b/reactos/drivers/lib/oskittcp/include/freebsd/src/sys/sys/mbuf.h
index c42198758de..b3226e9155e 100644
--- a/reactos/drivers/lib/oskittcp/include/freebsd/src/sys/sys/mbuf.h
+++ b/reactos/drivers/lib/oskittcp/include/freebsd/src/sys/sys/mbuf.h
@@ -304,7 +304,7 @@ union mcluster {
 #ifdef __REACTOS__
 #define MCLGET(m, how) { \
           OS_DbgPrint(OSK_MID_TRACE,("(MCLGET) m = %x\n", m)); \
-          (m)->m_ext.ext_buf = malloc(MCLBYTES); \
+          (m)->m_ext.ext_buf = malloc(MCLBYTES,__FILE__,__LINE__); \
 	  if ((m)->m_ext.ext_buf != NULL) { \
               (m)->m_data = (m)->m_ext.ext_buf; \
               (m)->m_flags |= M_EXT; \
@@ -313,7 +313,7 @@ union mcluster {
         }
 
 #define MCLFREE(p) { \
-          free( (p) ); \
+          free( (p), 0 ); \
         }
 #else
 #define	MCLGET(m, how) \
diff --git a/reactos/drivers/lib/oskittcp/include/oskitfreebsd.h b/reactos/drivers/lib/oskittcp/include/oskitfreebsd.h
index 7d0f23619eb..53345afb30f 100644
--- a/reactos/drivers/lib/oskittcp/include/oskitfreebsd.h
+++ b/reactos/drivers/lib/oskittcp/include/oskitfreebsd.h
@@ -30,6 +30,6 @@ static __inline void panic ( const char* fmt, ... )
 #define panic(...) do { printf(__VA_ARGS__); \
         oskittcp_die(__FILE__,__LINE__); } while(0)
 #endif//_MSC_VER
-#define kmem_malloc(x,y,z) malloc(y)
+#define kmem_malloc(x,y,z) malloc(y,0,0)
 
 #endif//_OSKITFREEBSD_H
diff --git a/reactos/drivers/lib/oskittcp/include/oskittcp.h b/reactos/drivers/lib/oskittcp/include/oskittcp.h
index d00535ef1ab..0971f50d350 100644
--- a/reactos/drivers/lib/oskittcp/include/oskittcp.h
+++ b/reactos/drivers/lib/oskittcp/include/oskittcp.h
@@ -155,8 +155,8 @@ void OskitTCPGetAddress( void *socket,
 
 #undef errno
 
-void *fbsd_malloc( unsigned int bytes, ... );
-void fbsd_free( void *data, ... );
+void *fbsd_malloc( unsigned int bytes, char *file, unsigned line, ... );
+void fbsd_free( void *data, char *file, unsigned line, ... );
 #if 0
 #define malloc(x) fbsd_malloc(x,__FILE__,__LINE__)
 #define free(x) fbsd_free(x,__FILE__,__LINE__)
diff --git a/reactos/drivers/lib/oskittcp/oskittcp.xml b/reactos/drivers/lib/oskittcp/oskittcp.xml
index b762fb0a92d..5c47b9640be 100644
--- a/reactos/drivers/lib/oskittcp/oskittcp.xml
+++ b/reactos/drivers/lib/oskittcp/oskittcp.xml
@@ -5,6 +5,7 @@
 	<define name="_DISABLE_TIDENTS"/>
 	<define name="__USE_W32API"/>
 	<define name="__NO_CTYPE_INLINES" />
+	<define name="MEMTRACK" />
 	<include>drivers/lib/oskittcp/include/freebsd</include>
 	<include>drivers/lib/oskittcp/include/freebsd/sys/include</include>
 	<include>drivers/lib/oskittcp/include/freebsd/src/sys</include>
diff --git a/reactos/drivers/lib/oskittcp/oskittcp/interface.c b/reactos/drivers/lib/oskittcp/oskittcp/interface.c
index 5da5c2244ca..fd82994f3db 100644
--- a/reactos/drivers/lib/oskittcp/oskittcp/interface.c
+++ b/reactos/drivers/lib/oskittcp/oskittcp/interface.c
@@ -33,15 +33,15 @@ void clock_init();
 int isprint(int c);
 int _snprintf(char * buf, size_t cnt, const char *fmt, ...);
 
-void *fbsd_malloc( unsigned int bytes, ... ) {
+void *fbsd_malloc( unsigned int bytes, char *file, unsigned line, ... ) {
     if( !OtcpEvent.TCPMalloc ) panic("no malloc");
     return OtcpEvent.TCPMalloc
-	( OtcpEvent.ClientData, (OSK_UINT)bytes, "*", 0 );
+	( OtcpEvent.ClientData, (OSK_UINT)bytes, file, line );
 }
 
-void fbsd_free( void *data, ... ) {
+void fbsd_free( void *data, char *file, unsigned line, ... ) {
     if( !OtcpEvent.TCPFree ) panic("no free");
-    OtcpEvent.TCPFree( OtcpEvent.ClientData, data, "*", 0 );
+    OtcpEvent.TCPFree( OtcpEvent.ClientData, data, file, line );
 }
 
 void InitOskitTCP() {
-- 
2.17.1