自定义分配器有时会与stl向量一起崩溃

Custom allocator sometimes crashes with stl vector

本文关键字:stl 向量 一起 崩溃 分配器 自定义      更新时间:2023-10-16

我得到了一个自定义分配器,它实现了自己的分配策略、自己的malloc/free等。现在,我被要求将这个自定义分配器与STL容器一起使用(无论是向量还是其他)。我创建了一个类,比如my_stdAllocator,它是一个符合ISO C++标准的接口。通过这个类,我调用了分配器的方法。例如:

template <class T>
class my_stdAllocator {
// ...other required stuff...
// allocate 'num' elements of type T
pointer allocate(size_type num, const_pointer = 0) {
return static_cast<T*>(MYAllocator::instance()->malloc(num));
}
// deallocate memory at 'p'
void deallocate(pointer p, size_type num=0) { 
MYAllocator::instance()->free(p);
}
// initialize allocated memory at 'p' with value 'value'
void construct(pointer p, const T& value) {   
::new ((void*)p) T(value);  
}

// destroy elements of initialized memory at p
void destroy(pointer p) {
p->~T();
}
// ...other required stuff...
} // end my_stdAllocator

自定义分配器的工作原理通常很有魅力:它经过了广泛的测试,它确实提高了性能,限制了碎片等。当我将它用作stl容器(比如向量)的分配器时,它有这种奇怪的行为,因为它有时工作正常,而在其他时候它会因segfault错误而崩溃。

举个例子,它用char:的向量正确地分配和释放

typedef char TP;
int main(int argc, char* argv[]) { 
std::vector<TP, my_stdAllocator<TP> > vec;
std::string s ("Whatever string, no matter how long...");
std::string::iterator it;
for (it=s.begin(); it<s.end(); ++it)
vec.push_back(*it);
...

在矢量内"手动"推送数字时可以

typedef double TP;
int main(int argc, char* argv[]) { 
std::vector<TP, my_stdAllocator<TP> > vec;
// "manual" push_back
vec.push_back(3.2);
vec.push_back(6.4);
vec.push_back(9.6);
vec.push_back(12.8);
vec.push_back(15.1);
vec.push_back(18.3);
vec.push_back(21.5);
...

当通过循环插入元素时,它因分段故障而停止:

typedef int TP;
int main(int argc, char* argv[]) { 
std::vector<TP, ff_stdAllocatorInst<TP> > vec;
for(unsigned int i=0; i<size; ++i)
vec.push_back( (TP) i );
...

当为至少保留一定数量的元素时,它就像一个魅力:

typedef int TP;
int main(int argc, char* argv[]) { 
std::vector<TP, ff_stdAllocatorInst<TP> > vec;
vec.reserve(size);
for(unsigned int i=0; i<size+150; ++i)
vec.push_back( (TP) i );
...

注意,当使用放置新时,上面提到的segfault不会发生,就像下面这样:

void *p = MYAllocator::instance()->malloc(size);
std::vector<TP> *vec = new (p) std::vector<TP>
for(unsigned int i=0; i<size; ++i)
vec->push_back( (TP) i );
...    

正如我已经说过的,自定义分配器已经过测试并且运行良好。我的类是C++标准和自定义分配器之间的一个简单接口。我试着用gdb调试它,但没有帮助:底层分配器还可以,我的代码中一定有一些错误,但我不明白哪里出了问题!

在对自定义分配器的malloc函数的调用中,需要乘以正在分配的对象的大小:

pointer allocate(size_type num, const_pointer = 0) {
return static_cast<T*>(MYAllocator::instance()->malloc(num*sizeof(T)));
}