memtrack.h

来自「基于TDI驱动编写的个人防火墙程序。包括驱动模块、应用层规则配置及加载模块。」· C头文件 代码 · 共 39 行

// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil -*- (for GNU Emacs)
//
// $Id: memtrack.h,v 1.1.1.1 2002/09/24 11:12:16 dev Exp $

#ifndef _memtrack_h_
#define _memtrack_h_

#define MEM_TAG		'1VRD'

#if DBG

void	memtrack_init(void);
void	memtrack_free(void);

void	*mt_malloc(ULONG size, const char *file, ULONG line);

// allocate memory from nonpaged pool
#define malloc_np(size)	mt_malloc((size), __FILE__, __LINE__)	

void free(void *ptr);

// debug macro
#define _TEST_ME_	__asm int 3

#else /* DBG */

#define memtrack_init()
#define memtrack_free()

#define malloc_np(size)	ExAllocatePoolWithTag(NonPagedPool, (size), MEM_TAG)
#define free(ptr)		ExFreePool(ptr)


#define _TEST_ME_

#endif /* DBG */

#endif