检查 malloc/calloc 的分配

Question

每次使用 malloc/calloc 后检查地址是否为 asagin 似乎是一个很好的编程实践。

是否可以创建一个函数来检查分配是否成功？ 当我们转换时，我们每次都转换点的类型，所以函数不会知道它是哪种指针类型。

例如：

newUser -> name = (char*)malloc(NAME_LENGTH*sizeof(char));
    if (newUser -> name == NULL){
        printf("Allocation of newUser failed\n");
        exit(1);
    } 

User *newUser = (User*)malloc(sizeof(User));
    if(newUser == NULL){
        printf("Allocation of newUser failed\n");
        exit(1);
    }

是否可以创建一个函数来获取 newUser 和 newUser -> name 并在分配失败时退出？

Answer 1

首先， 不要malloc的返回值，因为它可以隐藏其他错误。

将malloc包装在一个可以为您进行空检查的函数中没有问题。 由于malloc需要一个size_t并返回一个void *你的包装函数应该做同样的事情。 例如：

void *safe_malloc(size_t s)
{
    void *p = malloc(s);
    if (!p) {
        perror("malloc failed");
        exit(1);
    }
    return p;
}

然后，您可以在任何使用malloc地方使用此函数，而无需显式执行 NULL 检查。