C 内存管理模块的编写

C语言手动管理内存很困难,有时候很难发现内存泄漏,这两天看了一下<自制编程语言>里面有写了一个简单的内存管理模块,发现挺精巧,可以有效检测内存泄漏

原理很简单,就是把C的malloc函数分配的内存用一个链表记录起来,最后查看这个链表是否还有节点就可以判断是否有内存泄漏了

首先我们先看一下这个链表的数据结构

//标记数组的大小
#define MARK_SIZE 4
typedef struct Mem_Header Mem_Header;
struct Mem_Header
{
int size;//后面内存分配的大小
Mem_Header *next;//下一个节点的指针
Mem_Header *prev;//上一个节点的指针
char *filename;//文件名指针
int line;//在第几行
unsigned char mark[MARK_SIZE];//标记数组,这个数据用来表示这块内存是否被破坏了
}

这个mark数组我们将用 0xcd填充,如果这个数组被动过了就说明这块内存被破坏了

我们每次malloc一块内存的时候,这个结构体就会被添加到这块内存的头部

就像这样:

void *p = malloc(size);
Mem_Header *header = (Mem_Header *)p;
//然后设置header信息
/*省略一些赋值过程*/
return (char*)header+sizeof(Mem_Header);

这样外部拿到的指针就可以随便用了,回收的时候还可以重新推出头部指针的信息,然后从链表中删除这个节点

为了防止后面溢出,可以在后面内存也加上 mark数组 这样我们每次分配的内存大小是 sizeof(Mem_Header) + size + MARK_SIZE其中size是需要分配的大小,MARK_SIZE是在后面加的标记数组

我们用malloc初始化的时候经常碰到0,我们把这块内存填充0xcc(没意义的值别的其实也行),更容易发现错误

memset(p, 0xcc, willAlloc);

我们先来看第一个函数:

//size 需要分配的大小
//filename 文件名
//line 行好
void *Mem_alloc(Mem_Controller *controller, size_t size, char *filename, int line)
{
int willAlloc = size + MEM_HEADER_SIZE + MARK_SIZE;//实际分配大小
void *p = malloc(willAlloc);
if (!p)
controller->handle(controller, filename, line, "malloc null");
//填充数组
memset(p, 0xcc, willAlloc);
//设置头部一些信息
Mem_setHeader(p, size, filename, line);
//设置tail的信息
Mem_setTail(p, size);
//把这个节点添加到链表中
Mem_addChain(controller, (Mem_Header *)p);
return (char *)p + MEM_HEADER_SIZE;
}

这个函数就是分配一个内存,controller是控制器,实际上就是负责打印和保存头指针的

filename和line就是分配空间时候的文件和行号,可以用宏定义解决可以百度一下这两个宏__LINE__,__FILE__

内存释放:从链表中删除这个节点

void Mem_free(Mem_Controller *controller, void *p, char *filename, int line)
{
if (!p)
return;
//头信息
Mem_Header *header = (char *)p - MEM_HEADER_SIZE;
//check header and tail
//检查是不是mark数组损坏了
Mem_check(controller, header, filename, line);
//移除节点
Mem_rmChain(controller, header);
free(header);
}

打印所有链表中的节点

void Mem_dumps(Mem_Controller *controller)
{
Mem_Header *pos = controller->memHeader;
FILE *fp = stderr;
int counter = 0;
for (pos = controller->memHeader; pos; pos = pos->next)
{
Mem_check(controller, pos, pos->filename, pos->line);
fprintf(fp, "[%04d]%p********************\n", counter,
(char *)pos + MEM_HEADER_SIZE);
fprintf(fp, "%s line %d size..%d\n",
pos->filename, pos->line, pos->size);
counter++;
}
}

为了方便我们使用,我们可以设置几个宏定义

typedef struct Mem_Controller Mem_Controller;
extern Mem_Controller *pcontrol; void *Mem_alloc(Mem_Controller *controller, size_t size, char *filename, int line);
void Mem_free(Mem_Controller *controller, void *p, char *filename, int line);
void Mem_dumps(Mem_Controller *controller); #define current_mem_controller pcontrol
#define Mem_alloc_func(size) Mem_alloc(current_mem_controller, size, __FILE__, __LINE__)
#define Mem_free_func(p) Mem_free(current_mem_controller, p, __FILE__, __LINE__)
#define Mem_dump_func() Mem_dumps(current_mem_controller)

那个extern全局变量是别的文件定义的指针,__FILE__会自动替换文件名,这样我们就可以看到是哪里的内存没有被释放

使用的时候也简单了很多直接用宏定义就可以了

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "MEM.h" void fill_buffer(unsigned char *buf, int size)
{
int i; for (i = 0; i < size; i++)
{
buf[i] = i;
}
}
void mem_print(unsigned char *p, int size)
{
for (int i = 0; i < size; i++)
{
fprintf(stderr, "%02x", p[i]);
}
fprintf(stderr, "\n");
} int main()
{
void *p = Mem_alloc_func(10);
mem_print(p, 10);
void *p2 = Mem_alloc_func(10);
mem_print(p2, 10);
void *p3 = Mem_alloc_func(10);
mem_print(p3, 10);
Mem_dump_func();
fprintf(stderr, "try to clear\n");
Mem_free_func(p2);
Mem_free_func(p);
Mem_free_func(p3);
Mem_dump_func();
fprintf(stderr, "then to malloc\n");
p = Mem_alloc_func(10);
fill_buffer(p, 10);
p = Mem_realloc_func(p, 20);
mem_print(p, 24);
Mem_dump_func();
return 0;
}

我们可以看到泄漏的大小和行号

原版的代码的联合体是用来内存对齐的加快访问速度

一个C语言内存管理模块的实现的更多相关文章

  1. Go语言内存管理(一)内存分配

    Go语言内存管理(一)内存分配 golang作为一种"高级语言",也提供了自己的内存管理机制.这样一方面可以简化编码的流程,降低因内存使用导致出现问题的频率(C语言使用者尤其是初学 ...

  2. C语言 内存管理(转)

     转自 https://blog.csdn.net/u011616739/article/details/61621815 C语言 内存管理 1.内存分区 C源代码进过预处理.编译.汇编和链接4步生成 ...

  3. 深入理解Linux C语言内存管理

    问题不能拖,我这就来学习一下吧,争取一次搞定. 在任何程序设计环境及语言中,内存管理都十分重要. 内存管理的基本概念 分析C语言内存的分布先从Linux下可执行的C程序入手.现在有一个简单的C源程序h ...

  4. rt-thread中动态内存分配之小内存管理模块方法的一点理解

    @2019-01-18 [小记] rt-thread中动态内存分配之小内存管理模块方法的一点理解 > 内存初始化后的布局示意 lfree指向内存空闲区首地址 /** * @ingroup Sys ...

  5. C语言内存管理(转)

    伟大的Bill Gates 曾经失言: 640K ought to be enough for everybody — Bill Gates 1981 程序员们经常编写内存管理程序,往往提心吊胆.如果 ...

  6. iOS学习08之C语言内存管理

    本次主要学习和理解C语言中的内存管理 1.存储区划分 按照地址从高到低的顺序:栈区,堆区,静态区,常量区,代码区 1> 栈区:局部变量的存储区域 局部变量基本都在函数.循环.分支中定义 栈区的内 ...

  7. 深入理解C语言内存管理

    之前在学Java的时候对于Java虚拟机中的内存分布有一定的了解,但是最近在看一些C,发现居然自己对于C语言的内存分配了解的太少. 问题不能拖,我这就来学习一下吧,争取一次搞定. 在任何程序设计环境及 ...

  8. Objective-C语言内存管理

    • Objective-C为每个对象提供一个内部计数器,这个计数器跟踪对象的引用次数.所有类都继承自 NSObject 的对象retain和release方法. 如果使用了new.alloc或copy ...

  9. 黑马程序员——OC语言 内存管理

    Java培训.Android培训.iOS培训..Net培训.期待与您交流! (以下内容是对黑马苹果入学视频的个人知识点总结) (一)计数器 每个对象内部都保存了一个与之相关联的整数,称为引用计数器,当 ...

随机推荐

  1. JQuery 引用方式

    # head里面 <script src="jquery.min.js"></script> <script src="common.js& ...

  2. Java中的Number和Math类简单介绍

    Java Number类 一般地,当需要使用数字的时候,我们通常使用内置数据类型,如:byte.int.long.double 等. 实例: int a = 5000; float b = 13.65 ...

  3. OpenLDAP权限配置

    安装好了openldap之后,就是对它进行配置了,其中一项就是设置访问控制,限制普通用户只能修改/访问他们能修改/访问的项.这就是ACL需要做的事情. 设置方法 1.可以将 include行放在/et ...

  4. CentOS随笔 - 5.CentOS7安装Sql Server 2017

    前言 转帖请注明出处: http://www.cnblogs.com/Troy-Lv5/ 开发环境嘛, 作为.Net系Sql Server那是必备的. 听过Sql server可以安装在Linux上了 ...

  5. 百度开源的分布式 id 生成器

    UidGenerator是Java实现的, 基于Snowflake算法的唯一ID生成器.UidGenerator以组件形式工作在应用项目中, 支持自定义workerId位数和初始化策略, 从而适用于d ...

  6. redis几种数据类型以及使用场景

    1. string类型 string为最简单类型,一个key对应一个value set mykey "wangzai" ##设置key,第二次赋值会直接覆盖之前的 setnx my ...

  7. Java编程练习题

      曾经,有人说过,没有刷题的人生是不完整的.看了几天Java,我试着做了几道练习题,好让我的人生完整一点.(偷笑--)这里挑了一些题来跟大家分享,本文不定期更新. 题目集  1. 最后一个单词的长度 ...

  8. 获取URL网页信息

    static string GetHtml(string url) {string strHTML = ""; WebClient myWebClient = new WebCli ...

  9. JDK/bin目录下的不同exe文件的用途

    新安装完JDk 大家是否发现安装目录的bin文件夹有很多exe文件 下面就为大家讲解不同exe文件的用途 javac:Java编译器,将Java源代码换成字节代 java:Java解释器,直接从类文件 ...

  10. python处理数据(二)

    处理PDF文件 PyPDF2简介 作为 PDF 工具包构建的纯 python 库. 它可以:提取文档信息(标题,作者,... ...)一页一页地分割文件一页一页地合并文件裁剪页面将多个页面合并成一个页 ...