C ++ 11兼容的线性分配器实现

Question

我已经实现了C ++ 11兼容的线性或竞技场分配器。 代码如下。

linear_allocator.hpp：

#pragma once

#include <cstddef>
#include <cassert>
#include <new>
#include "aligned_mallocations.hpp"

template <typename T>
class LinearAllocator
{
public:
    using value_type = T;
    using pointer = T*;
    using const_pointer = const T*;
    using reference = T&;
    using const_reference = const T&;
    //using propagate_on_container_copy_assignment = std::true_type;
    //using propagate_on_container_move_assignment = std::true_type;
    //using propagate_on_container_swap = std::true_type;

    LinearAllocator(std::size_t count = 64)
        : m_memUsed(0),
        m_memStartAddress(nullptr)
    {
        allocate(count);
    }
    ~LinearAllocator()
    {
        clear();
    }

    template <class U>
    LinearAllocator(const LinearAllocator<U>&) noexcept 
    {}

    /// \brief allocates memory equal to # count objects of type T
    pointer allocate(std::size_t count)
    {
        if (count > std::size_t(-1) / sizeof(T))
        {
            throw std::bad_alloc{};
        }
        if (m_memStartAddress != nullptr)
        {
            alignedFree(m_memStartAddress);
        }
        m_memUsed = count * sizeof(T);
        m_memStartAddress = static_cast<pointer>(alignedMalloc(m_memUsed, alignof(T)));
        return m_memStartAddress;
    }
    /// \brief deallocates previously allocated memory
    /// \brief Linear/arena allocators do not support free() operations. Use clear() instead.
    void deallocate([[maybe_unused]] pointer p, [[maybe_unused]] std::size_t count) noexcept
    {
        //assert(false);
        clear();
    }

    /// \brief simply resets memory
    void clear()
    {
        if (m_memStartAddress != nullptr)
        {
            alignedFree(m_memStartAddress);
            m_memStartAddress = nullptr;
        }
        this->m_memUsed = 0;
    }

    /// \brief GETTERS
    pointer getStartAddress() const
    {
        return this->m_memStartAddress;
    }
    std::size_t getUsedMemory() const
    {
        return this->m_memUsed;
    }
private:
    std::size_t m_memUsed;
    pointer m_memStartAddress;
};

template <class T, class U>
bool operator==(const LinearAllocator<T> &, const LinearAllocator<U> &)
{
    return true;
}

template <class T, class U>
bool operator!=(const LinearAllocator<T> &, const LinearAllocator<U> &)
{
    return false;
}

不用担心alignedMalloc和alignedFree 。 他们是正确的。

这是我的测试程序（linear_allocator.cpp）：

#include "linear_allocator.hpp"
#include <vector>
#include <deque>
#include <iostream>
#include <string>
#include <typeinfo>

int main()
{
    [[maybe_unused]]
    LinearAllocator<int> a{1024};
    std::cout << a.getStartAddress() << '\n';
    std::cout << a.getUsedMemory() << '\n';
    std::vector<std::string, LinearAllocator<std::string>> v;
    v.reserve(100);
    std::cout << "Vector capacity = " << v.capacity() << '\n';
    //std::cout << v.get_allocator().getStartAddress() << '\n';
    //std::cout << v.get_allocator().getUsedMemory() << '\n';
    v.push_back("Hello");
    v.push_back("w/e");
    v.push_back("whatever");
    v.push_back("there is ist sofi j");
    v.push_back("wisdom");
    v.push_back("fear");
    v.push_back("there's more than meets the eye");
    for (const auto &s : v)
    {
        std::cout << s << '\n';
    }
    std::cout << typeid(v.get_allocator()).name() << '\n';

    std::deque<int, LinearAllocator<int>> dq;
    dq.push_back(23);
    dq.push_back(90);
    dq.push_back(38794);
    dq.push_back(7);
    dq.push_back(0);
    dq.push_back(2);
    dq.push_back(13);
    dq.push_back(24323);
    dq.push_back(0);
    dq.push_back(1234);
    for (const auto &i : dq)
    {
        std::cout << i << '\n';
    }
    std::cout << typeid(dq.get_allocator()).name() << '\n';
}

使用g++ -std=c++17 -O2 -march=native -Wall linear_allocator.cpp -o linear_allocator.gpp.exe并运行linear_allocator.gpp.exe会得到输出：

0x4328b8
4096
Vector capacity = 100
Hello
w/e
whatever
there is ist sofi j
wisdom
fear
there's more than meets the eye
15LinearAllocatorINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEE

如您所见，deque的输出根本不存在。 如果我取消注释这两行：

//std::cout << v.get_allocator().getStartAddress() << '\n';
//std::cout << v.get_allocator().getUsedMemory() << '\n';

向量的输出也不会显示。

使用MSVS cl进行编译可得到以下输出：

000000B47A1CAF88
4096

更糟的是。

好像有UB，肯定有一些我想念的东西，但是我无法确定它的位置。 我的分配器设计基于C ++ 11 +准则。 我想知道我在做什么错。

Answer 1

尽管分配器负责提供和释放用于存储容器数据的内存，但它仍然仅应容器的请求才这样做。 即，所提供的存储的实际管理（尤其是其寿命）仍然在容器的一边。 想象一下，当向量执行其元素的重定位时会发生什么：

1. 请求一个新的内存块，其大小比当前（旧）内存大一个给定因子。

2. 存储在“旧”块中的元素将被复制/移动到新块中。

3. 只有这样，才能释放“旧”内存块。

在您的实现中，一次只能有一个内存块处于活动状态-旧的内存块在分配新的内存块之前就已释放（特别是，当容器仅请求新的内存块时，这种情况会发生，元素可能是移至）。 当向量尝试从先前的存储位置重定位元素时，您已经调用了UB，因为它们所居住的内存已经失效。

此外，通过不为您的分配器类型提供复制构造函数，编译器提供的实现将执行浅表复制（即，它复制指针，而不是该地址下存储的数据），然后将其释放到析构函数中。 也就是说，调用：

v.get_allocator()

将复制分配器的浅表副本，创建分配器类型的prvalue，并在临时对象的生命周期结束时（即，在包含cout调用的完整语句结束时）释放存储的指针，从而导致加倍调用同一指针上的alignedFree 。