6

我的stackAlloc功能如下所示:

void* stackAlloc(size_t size) {
    if (size > maxStackAllocation)
        return malloc(size);
    else 
        return _alloca(size);
}
void stackAllocFree(void *ptr, size_t size) {
    if (size > maxStackAllocation) {
        free(ptr);
    }
}

如果我改变所以stackAlloc函数总是使用malloc而不是alloca一切正常。

我将函数更改为宏,现在它按预期工作:

#define maxStackAllocation 1024
#define stackAlloc(size) \
( \
    (size > maxStackAllocation)? \
         malloc(size): \
        _alloca(size) \
)

#define stackAllocFree(ptr, size) \
( \
    (size > maxStackAllocation)? \
        free(ptr): \
    void() \
)
4

3 回答 3

9

假设您在 Windows 上运行,因为您的代码调用_alloca(),根据MSDN 文档

_alloca从程序堆栈中分配size字节。调用函数退出时自动释放分配的空间

请注意,当调用函数退出时,内存被释放——我假设这也意味着调用函数返回。

你的代码:

void* stackAlloc(size_t size) {
    if (size > maxStackAllocation)
        return malloc(size);
    else 
        return _alloca(size);
}

返回,从而释放通过_alloca().

于 2016-06-13T12:27:11.537 回答
3

从手册页

当调用 alloca() 的函数返回给它的调用者时,这个临时空间会自动释放。

因此,每当您的stackAlloc函数返回时,它都会自动释放内存。

于 2016-06-13T12:27:17.367 回答
0

这可行,但我建议不要在生产中使用它:

#include <iostream>
#include <alloca.h>

auto stackAlloc(const size_t size)
{
    return [size](){ return alloca(size); };
}

int main() {
    char *ch = (char *)stackAlloc(40000)();
    ch[39999] = '\0';

    return 0;
}

反检查:如果我减少 stackAlloc 的参数,它就不起作用(这是这里的预期行为)随意在 stackAlloc 中添加检查等(通过返回不同的 lambda 或让 lambda 进行检查)。

于 2016-06-13T20:41:09.313 回答