C语言之安全的动态内存分配

最新推荐文章于 2025-04-25 09:55:14 发布

Roger-Pang

最新推荐文章于 2025-04-25 09:55:14 发布

阅读量413

点赞数

CC 4.0 BY-SA版权

分类专栏： C语言文章标签： C语言动态内存分配 malloc

本文链接：https://blog.youkuaiyun.com/qq_31504597/article/details/79822863

C语言专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种通过宏定义实现的安全动态内存分配方法，避免了直接使用malloc可能带来的内存分配失败未检测的问题。通过自定义alloc函数确保内存分配成功，否则程序会输出错误信息并退出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

动态内存分配最常见的错误就是忘记检查所请求的内存是否分配成功. 可以利用宏实现安全的动态内存分配.

// alloc.h
#include <stdlib.h>
#define malloc 不要直接使用malloc // 直接使用malloc将无法编译
#define MALLOC(num, type) (type *)alloc((num) * sizeof(type))
extern void *alloc(size_t size);

// alloc.c
#include <stdio.h>
#include "alloc.h"
#undef malloc
void *alloc(size_t size)
{
    void *new_mem;
    new_mem = malloc(size);
    if (new_mem == NULL) {
        printf("Out of memory!\n");
        exit(1);
    }
    return new_mem;
}

// main.c
#include <stdio.h>
#include "alloc.h"
int main()
{
    int *new_mem;
    new_mem = MALLOC(25, int);
    // ...
    free(new_mem);
    return 0;
}