mallocHook.h

Go to the documentation of this file.

// Copyright 2016 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
// Modified by Jeremy Retailleau.
 
#ifndef PXR_ARCH_MALLOC_HOOK_H
#define PXR_ARCH_MALLOC_HOOK_H

 
#include "pxr/arch/pxr.h"
#include "pxr/arch/api.h"
 
#include <stdlib.h>
#include <string>
 
ARCH_NAMESPACE_OPEN_SCOPE

ARCH_API bool ArchIsPtmallocActive();

ARCH_API bool ArchIsStlAllocatorOff();

class ArchMallocHook {
public:
    ARCH_API
    bool Initialize(void* (*mallocWrapper)(size_t, const void*),
            void* (*reallocWrapper)(void*, size_t, const void*),
            void* (*memalignWrapper)(size_t, size_t, const void*),
            void  (*freeWrapper)(void*, const void*),
            std::string* errMsg);

    ARCH_API
    bool IsInitialized();

    ARCH_API
    void* Malloc(size_t nBytes) {
        return (*_underlyingMallocFunc)(nBytes);
    }

    ARCH_API
    void* Realloc(void* ptr, size_t nBytes) {
        return (*_underlyingReallocFunc)(ptr, nBytes);
    }

    ARCH_API
    void* Memalign(size_t alignment, size_t nBytes) {
        return (*_underlyingMemalignFunc)(alignment, nBytes);
    }

    ARCH_API
    void Free(void* ptr) {
        (*_underlyingFreeFunc)(ptr);
    }
 
private:
    // Note: this is a POD (plain 'ol data structure) so we depend on zero
    // initialization here to null these out.  Do not add a constructor or
    // destructor to this class.
 
    void* (*_underlyingMallocFunc)(size_t);
    void* (*_underlyingReallocFunc)(void*, size_t);
    void* (*_underlyingMemalignFunc)(size_t, size_t);
    void  (*_underlyingFreeFunc)(void*);
};
 
ARCH_NAMESPACE_CLOSE_SCOPE
 
#endif // PXR_ARCH_MALLOC_HOOK_H