MemoryAllocations.hpp

Go to the documentation of this file.

#pragma once
 
#include "MemoryTracker.hpp"
#include <new>
 
void* operator new(std::size_t sz) {
    void* p = std::malloc(sz);
 
    if (!p) 
        throw std::bad_alloc{};
 
    memviz::on_alloc(p, sz);
 
    return p;
}
 
void operator delete(void* p) noexcept {
    memviz::on_free(p);
    std::free(p);
}
 
void operator delete(void* p, std::size_t) noexcept {
    memviz::on_free(p);
    std::free(p);
}
 
void* operator new(std::size_t sz, std::align_val_t al) {
    std::size_t align = static_cast<std::size_t>(al);
    std::size_t n = (sz + align - 1) / align * align;
 
    void* p = std::aligned_alloc(align, n);
    if (!p) 
        throw std::bad_alloc{};
    
    memviz::on_alloc(p, sz);
    return p;
}
 
void operator delete(void* p, std::align_val_t) noexcept {
    memviz::on_free(p);
    std::free(p);
}
 
void operator delete(void* p, std::size_t, std::align_val_t) noexcept {
    memviz::on_free(p);
    std::free(p);
}