stack_usage.cc

#include <iostream>
#include <pthread.h>
#include <unistd.h>
#include <cstdint>

#define BYTES_TO_MB(x) ((double)(x) / (1024.0 * 1024.0))

inline void printStackUsage()
{
    volatile int localVar;
    void* currentStackPtr = (void*)&localVar;

    // Get stack info using pthread APIs
    pthread_attr_t attr;
    pthread_getattr_np(pthread_self(), &attr);

    void* stackAddr;
    size_t stackSize;
    pthread_attr_getstack(&attr, &stackAddr, &stackSize);
    pthread_attr_destroy(&attr);

    uintptr_t stackBase  = reinterpret_cast<uintptr_t>(stackAddr);
    uintptr_t stackEnd   = stackBase + stackSize;
    uintptr_t currentPtr = reinterpret_cast<uintptr_t>(currentStackPtr);

    size_t used = 0;
    if (currentPtr < stackEnd && currentPtr >= stackBase)
        used = stackEnd - currentPtr;

    size_t total = stackSize;

    std::cout << "Stack used:  " << (100.0 * used / total) << "%\n";
    std::cout << "Stack total: " << BYTES_TO_MB(total) << " MB\n\n";
}