我有一些代码包含使用 calloc 和 malloc 进行内存分配的自制哈希表。我想使用带有自定义删除器的 shared_ptr 修改这些部分,该删除器会自动释放分配的内存。该代码是 mmseg 中文分段器算法的一部分,它运行良好,但由于它留下内存泄漏而变得如此混乱。我正在考虑使用 unordered_map 等重写该代码,但现在我想做这些更改。
我阅读了类似问题的答案,例如shared_ptr with malloc and free或Accessing calloc'd data through a shared_ptr,但我在下面的代码中使用它时遇到问题。
我有这些行,我无法用智能指针包装调用。所以也许有人可以帮我解决这个问题:
struct Word {
unsigned char nbytes; /* number of bytes */
char length; /* number of characters */
unsigned short freq;
char text[word_embed_len];
};
struct Entry {
Word *word;
Entry *next;
};
static Entry **new_bins = static_cast<Entry **>(std::calloc(init_size,
sizeof(Entry *)));
Entry *entry, ...;
...
new_bins[hash_val] = entry;
....
free(new_bins);
上面的 calloc 调用我会用 calloc 的结果提供共享指针,例如
std::shared_ptr<Entry *> binsPtr(new_bins, freePtr());
我不确定这是否正确。
mmseg 使用 malloc() 的池分配例程,如下所示:
inline void *pool_alloc(int len) {
void *mem = _pool_base;
if (len <= _pool_size) {
_pool_size -= len;
_pool_base += len;
return mem;
}
_pool_base = static_cast<char *>(std::malloc(REALLOC_SIZE));
mem = _pool_base;
_pool_base += len;
_pool_size = REALLOC_SIZE - len;
return mem;
}
然后分配器是这样调用的:
Entry *entry = bins[h];
...
entry = static_cast<Entry *>(pool_alloc(sizeof(Entry)));
entry->word = word;
entry->next = NULL;
bins[h] = entry;
是否可以修改 pool_alloc 例程,例如我可以用共享指针包装 malloc() 并定义自定义删除器(甚至可以跳过完整的 pool_alloc fct 并仅使用 shared_ptr),例如
std::shared_ptr<Entry> entry((Entry *)malloc(sizeof(Entry)), freePtr());
struct freePtr {
void operator()(void* x) {
free(x);
}
};
如果有人可以帮助我解决这个问题,那就太好了。提前致谢!
更新:
我为我的问题编写了一个简单的内存池类,因此所有指针都会自动销毁。shared_ptr 中的包装 calloc() 似乎工作正常并且按预期工作。Valgrind 不再报告内存泄漏和错误。