shared_payload.hpp

#pragma once
 
#include "nalchi/export.hpp"
 
#include <atomic>
#include <cstdint>
 
struct SteamNetworkingMessage_t;
 
namespace nalchi
{
 
struct shared_payload
{
    using ref_count_t = std::atomic<std::int32_t>;
    using alloc_size_t = std::uint32_t;
 
    void* ptr; 
 
    NALCHI_API static shared_payload allocate(alloc_size_t size);
 
    NALCHI_API static void force_deallocate(shared_payload payload);
 
    NALCHI_API auto size() const -> alloc_size_t;
 
    NALCHI_API auto word_ceiled_size() const -> alloc_size_t;
 
    NALCHI_API auto internal_alloc_size() const -> alloc_size_t;
 
    NALCHI_API bool used_bit_stream() const;
 
private:
    friend class bit_stream_writer;
 
    void set_used_bit_stream(bool);
 
private:
    friend class socket_extensions;
 
    NALCHI_API void add_to_message(SteamNetworkingMessage_t* msg, int logical_bytes_length);
 
    void increase_ref_count();
    void decrease_ref_count_and_deallocate_if_zero();
 
    static void decrease_ref_count_and_deallocate_if_zero_callback(SteamNetworkingMessage_t* msg);
 
private:
    auto ref_count() -> ref_count_t&;
    auto ref_count() const -> const ref_count_t&;
 
    auto payload_size_and_bit_stream_used_flag() -> alloc_size_t&;
    auto payload_size_and_bit_stream_used_flag() const -> const alloc_size_t&;
};
 
} // namespace nalchi