Question

这可能是重复的，但我无法在任何地方找到解决方案。

我有这样的源代码：

struct Blob{
    //...    

    static void *operator new(size_t size_reported, size_t size) {
        return ::operator new(size);
    }
};

我这样用：

std::shared_ptr<Blob> blob;
// ...
size_t size = calcSize(); // it returns say 231
Blob *p = new(size) Blob();
blob.reset(p);

我可以以某种方式更改代码，以便我可以使用std::make_shared或std::allocate_shared，因此我single allocation代替two allocations吗？

更新

我能够消除new并将代码简化为以下内容：

struct Blob{
    //...    
};

std::shared_ptr<Blob> blob;
// ...
size_t size = calcSize(); // it returns say 231

// allocate memory
void *addr = ::operator new(size);

// placement new
Blob *p = ::new(addr) Blob();

blob.reset(p);

它完全相同，但我认为现在我更清楚我想要做什么。

Answer 1

以下是提出的建议。

由于无法将大小传递给分配器，您可以通过global variable或class member来完成。

在这两种情况下，解决方案都不优雅，而且相当危险 - 当需要维护代码时，灾难等待现在或稍后发生。

如果allocate_shared将shared_ptr控制块放在缓冲区类之后，可能会出现另一个意外问题。

在这种情况下会出现明显的缓冲区溢出，因为sizeof(buffer)将报告大小大约1个字节左右。

再一次 - 代码正在运行，但将来肯定会遇到问题。

#include <stdio.h>
#include <string.h>

#include <memory>

// ============================

class buffer{
public:
    buffer(const char *s){
        strcpy(x, s);
    }

    char x[1];
};

// ============================

template <typename T>
struct buffer_allocator {
    using value_type = T;

    buffer_allocator() = default;

    template <class U>
    buffer_allocator(const buffer_allocator<U>&) {}

    T* allocate(size_t n) {
        void *p = operator new(n * sizeof(T));

        printf("Allocate   %p, (%zu)\n", p, get_size());

        return (T*) p;
    }

    void deallocate(T* p, size_t n) {
        delete p;

        printf("Deallocate %p, (%zu)\n", p, get_size());
    }

    size_t get_size(){
        return size;
    }

    void set_size(size_t size){
        this->size = size;
    }

private:
    size_t size = 0;
};

template <typename T, typename U>
inline bool operator == (const buffer_allocator<T>&, const buffer_allocator<U>&) {
  return true;
}

template <typename T, typename U>
inline bool operator != (const buffer_allocator<T>& a, const buffer_allocator<U>& b) {
  return !(a == b);
}

// ============================

int main(int argc, char *argv[]){
    buffer_allocator<buffer> ba;

    const char *s = "hello world!";

    ba.set_size( strlen(s) + 1 );

    auto b = std::allocate_shared<buffer>(ba, s);

    printf("Pointer    %p\n", b.get());

    printf("%s\n", b->x);
    printf("%zu\n", b.use_count());
    auto b1 = b;
    printf("%zu\n", b1.use_count());

    return 0;
}

使用自定义新运算符make_shared

更新

1 个答案: