calloc() 函数代替 malloc()

Question

关闭。此题需要 details or clarity 。目前不接受答案。

---

想要改进这个问题吗？通过 editing this post 添加详细信息并澄清问题.

已关闭 7 年前。

Improve this question

我要改malloc()函数为 calloc() ，但我很困惑:我如何在这段代码中做到这一点？

void *mymalloc(size_t len)
{

    void *buf;
    size_t pages = (len & PAGE_MASK) + 2;
    size_t offset = PAGE_SIZE - (len & ~PAGE_MASK);

    if(offset < sizeof(size_t))
    {
        pages++;
        offset += PAGE_SIZE;
    }

    if((buf = mmap(NULL, pages << PAGE_SHIFT, PROT_READ | PROT_WRITE, 
                    MAP_PRIVATE | MAP_ANONYMOUS, 0, 0)) == -1)
    {
        perror("mymalloc/mmap");
        exit(1);
    }
    *(size_t *)buf = len;
    *(size_t *)(buf+offset+len) = len;

    if(mprotect(buf+offset+len, PAGE_SIZE, PROT_NONE) == -1)
    {
        perror("mymalloc/mprotect");
        exit(1);
    }

    return buf+offset;
}

Answer 1

calloc() 实际上只是 malloc()，其中 memset() 为零。

假设您想要一个 mycalloc() 来补充您的 mymalloc():

void *mymalloc( size_t bytes )
{
    ...
    return( ptr );
}

你会得到:

void *mycalloc( size_t bytes, size_t n )
{
    size_t total_bytes = bytes * n;
    void *ptr = mymalloc( total_bytes );
    memset( ptr, 0, total_bytes );
    return( ptr );
}

您需要添加对乘法溢出和 mymalloc() 返回 NULL 的检查。