#include "shared_data.h"

#include "memory_tracker.h"

#include <util/system/sys_alloc.h>
#include <util/system/sanitizers.h>

namespace NActors {

    static constexpr char MemoryLabelSharedData[] = "Tablet/TSharedData/Buffers";

    char* TSharedData::Allocate(size_t size) {
        char* data = nullptr;
        if (size > 0) {
            if (size >= MaxDataSize) {
                throw std::length_error("Allocate size overflow");
            }
            auto allocSize = OverheadSize + size;
            char* raw = reinterpret_cast<char*>(y_allocate(allocSize));

            auto* privateHeader = reinterpret_cast<TPrivateHeader*>(raw);
            privateHeader->AllocSize = allocSize;
            NActors::NMemory::TLabel<MemoryLabelSharedData>::Add(allocSize);

            auto* header = reinterpret_cast<THeader*>(raw + PrivateHeaderSize);
            header->RefCount = 1;
            header->Owner = nullptr;

            data = raw + OverheadSize;
            NSan::Poison(data, size);
        }
        return data;
    }

    void TSharedData::Deallocate(char* data) noexcept {
        if (data) {
            char* raw = data - OverheadSize;

            auto* privateHeader = reinterpret_cast<TPrivateHeader*>(raw);
            NActors::NMemory::TLabel<MemoryLabelSharedData>::Sub(privateHeader->AllocSize);

            auto* header = reinterpret_cast<THeader*>(raw + PrivateHeaderSize);
            Y_VERIFY_DEBUG(header->Owner == nullptr);

            y_deallocate(raw);
        }
    }

}