关于rt_malloc的调试方法

发布于 2010-01-29 17:17:10 浏览：9733 订阅该版

11 个回答

ffxz 2010-01-30

这家伙很懒，什么也没写！

ws_d1 2010-01-31

这家伙很懒，什么也没写！

bernard 2010-02-01

这家伙很懒，什么也没写！

bernard 2010-02-01

这家伙很懒，什么也没写！

McuPlayer 2010-02-04

这家伙很懒，什么也没写！

coldfish 2010-02-05

这家伙很懒，什么也没写！

bernard 2010-02-06

这家伙很懒，什么也没写！

ws_d1 2010-07-07

这家伙很懒，什么也没写！

```c
#include <rtthread.h>

typedef struct {
	unsigned short addr; //首地址（仅仅使用后两个字节）
	unsigned short ownr; //注册号，标识注册者
}mm_slot;

//静态1K空间
mm_slot mm[256]; // 1K

//地址为0，表示未被占用
#define IS_SLOT_AVAL(slot) (slot.addr == 0)
//取末16位
#define USHORT_VALUE(v) ((unsigned short)(0x0000FFFF & (v)))

// 统计租借者登记的次数 
static int mm_owner_occur(unsigned short owner)
{
	int i,cnt = 0;
	for(i = 0; i < NELEMENTS(mm); i++){
		if(owner == mm[i].ownr){
			cnt++;
		}
	}
	return cnt;
}

//全部清零
void mm_reset(void)
{
	os_memset(mm, 0, sizeof(mm));
}

//租用注册
void mm_reg(void* addr, unsigned short owner)
{
	int i;
	for(i = 0; i < NELEMENTS(mm); i++){
		if(IS_SLOT_AVAL(mm[i])){
			mm[i].addr = USHORT_VALUE(((int)addr));
			mm[i].ownr = owner;
			break;
		}
	}

//空间满了，有泄漏！！！
	if(i >= NELEMENTS(mm))
		os_printf("------------mm alert !!! ------------n");
}

//归还注销
void mm_unreg(void* addr)
{
	int i;
	unsigned short a = USHORT_VALUE(((int)addr));
	
	for(i = 0; i < NELEMENTS(mm); i++){
		if(a == mm[i].addr){
			mm[i].addr = 0;
			mm[i].ownr = 0;
			break;
		}
	}
}

#define rt_list_entry(node, type, member) 
    ((type *)((char *)(node) - (unsigned long)(&((type *)0)->member)))
extern struct rt_object_information rt_object_container[];

//一个finsh命令，输出当前内存出租情况
void list_mmon(void)
{
	int i;

struct rt_thread *thread;
	struct rt_list_node *list, *node;
	os_printf("========================================n");
	list = &rt_object_container[RT_Object_Class_Thread].object_list;
	for (node = list->next; node != list; node = node->next)
	{
		thread = rt_list_entry(node, struct rt_thread, list);
		os_printf("%-8s: %05u timesn", thread->name, mm_owner_occur(USHORT_VALUE(((int)thread))));
	}
	os_printf("----------------------------------------n");

for (i = MMON_OWNER_MIN; i < MMON_OWNER_MAX; i++)
	{
		os_printf("0x%-8x: %05u timesn", i, mm_owner_occur(USHORT_VALUE(i)));
	}
	os_printf("========================================n");
}

//这里就是对malloc的包装了，租借者主动提供 身份ID，作为参数
 void * os_malloc_mon(unsigned long size,  unsigned short owner)
{
	void *ptr = rt_malloc(size);

if(ptr){
		mm_reg(ptr, owner);
	}
	return ptr;
}

//这里也是对malloc的包装，默认使用thread ID 作为租借身份
 void * os_malloc_thread_mon(unsigned long size)
{
	void *ptr = rt_malloc(size);

if(ptr){
		unsigned short owner = USHORT_VALUE(((int)rt_thread_self ()));
		mm_reg(ptr, owner);
	}
	return ptr;
}

//对free的包装
 void os_free_mon(void *ptr)
{
	if(ptr){
		mm_unreg(ptr);
		rt_free(ptr);
	}
}

#ifdef MEM_MON_ENABLE
#include "finsh.h"
FINSH_FUNCTION_EXPORT(list_mmon, list_mmon)
#endif
```

对该代码感兴趣的同学，可以先参见本系列一楼，了解设计思想。

代码是其实现。使用了1K的内存，能够提供256个登记位置；能够按进程来注册，也可以指定ID。提供了一个finsh命令，能够列出哪个进程登记了多少次，或者哪个ID登记了多少次；
如果出现泄漏，看看那个登记最多的，就是光记账不还钱的主了。

按我的情况，长期运行大约可以起伏稳定在60多个登记项，占用资源不多。

工具简单，能够找出问题，我已经靠它剔除两个深藏的BUG了。

此外，该工具有点粗糙，有些效率问题，或者有竞争问题，各位当工具使用应该问题不大，用完就扔呗。 [s:154]

使用的时候请根据自己的情况 定义开关宏。希望能够对大家有用；