143 lines
3.8 KiB
C++
143 lines
3.8 KiB
C++
|
|
#pragma once
|
|
|
|
#if !defined(MIJIN_DEBUG_STACKTRACE_HPP_INCLUDED)
|
|
#define MIJIN_DEBUG_STACKTRACE_HPP_INCLUDED 1
|
|
|
|
#include <cmath>
|
|
#include <iomanip>
|
|
#include <vector>
|
|
#if __has_include(<fmt/format.h>)
|
|
# include <fmt/format.h>
|
|
#endif
|
|
#include "./symbol_info.hpp"
|
|
#include "../internal/common.hpp"
|
|
#include "../types/result.hpp"
|
|
#include "../util/iterators.hpp"
|
|
|
|
namespace mijin
|
|
{
|
|
|
|
//
|
|
// public defines
|
|
//
|
|
|
|
//
|
|
// public constants
|
|
//
|
|
|
|
//
|
|
// public types
|
|
//
|
|
|
|
struct Stackframe
|
|
{
|
|
void* address;
|
|
std::string filename;
|
|
std::string function;
|
|
int lineNumber;
|
|
};
|
|
|
|
class Stacktrace
|
|
{
|
|
private:
|
|
std::vector<Stackframe> frames_;
|
|
public:
|
|
Stacktrace() = default;
|
|
Stacktrace(const Stacktrace&) = default;
|
|
Stacktrace(Stacktrace&&) = default;
|
|
explicit Stacktrace(std::vector<Stackframe> frames) MIJIN_NOEXCEPT : frames_(std::move(frames)) {}
|
|
|
|
Stacktrace& operator=(const Stacktrace&) = default;
|
|
Stacktrace& operator=(Stacktrace&&) = default;
|
|
|
|
[[nodiscard]] const std::vector<Stackframe>& getFrames() const MIJIN_NOEXCEPT { return frames_; }
|
|
};
|
|
|
|
//
|
|
// public functions
|
|
//
|
|
|
|
[[nodiscard]] Result<Stacktrace> captureStacktrace(unsigned skipFrames = 0) MIJIN_NOEXCEPT;
|
|
[[nodiscard]] const Optional<Stacktrace>& getExceptionStacktrace() MIJIN_NOEXCEPT;
|
|
|
|
template<typename TStream>
|
|
TStream& operator<<(TStream& stream, const Stackframe& stackframe)
|
|
{
|
|
stream << "[" << stackframe.address << "] " << stackframe.filename << ":" << stackframe.lineNumber << " in " << demangleCPPIdentifier(stackframe.function.c_str());
|
|
|
|
return stream;
|
|
}
|
|
|
|
template<typename TStream>
|
|
TStream& operator<<(TStream& stream, const Stacktrace& stacktrace)
|
|
{
|
|
const std::streamsize oldWidth = stream.width();
|
|
const std::ios::fmtflags oldFlags = stream.flags();
|
|
const int numDigits = static_cast<int>(std::ceil(std::log10(stacktrace.getFrames().size())));
|
|
stream << std::left;
|
|
for (const auto& [idx, frame] : mijin::enumerate(stacktrace.getFrames()))
|
|
{
|
|
stream << " #" << std::setw(numDigits) << idx << std::setw(0) << " at " << frame << "\n";
|
|
}
|
|
stream << std::setw(oldWidth);
|
|
stream.flags(oldFlags);
|
|
return stream;
|
|
}
|
|
|
|
} // namespace mijin
|
|
|
|
#if __has_include(<fmt/format.h>)
|
|
template<>
|
|
struct fmt::formatter<mijin::Stackframe>
|
|
{
|
|
constexpr auto parse(format_parse_context& ctx) -> decltype(ctx.begin())
|
|
{
|
|
auto it = ctx.begin();
|
|
auto end = ctx.end();
|
|
|
|
if (it != end && *it != '}') throw format_error("invalid format");
|
|
|
|
return it;
|
|
}
|
|
|
|
template<typename TContext>
|
|
auto format(const mijin::Stackframe& stackframe, TContext& ctx) const -> decltype(ctx.out())
|
|
{
|
|
auto it = ctx.out();
|
|
it = fmt::format_to(it, "[{}] {}:{} in {}", stackframe.address, stackframe.filename,
|
|
stackframe.lineNumber, mijin::demangleCPPIdentifier(stackframe.function.c_str()));
|
|
return it;
|
|
}
|
|
};
|
|
|
|
template<>
|
|
struct fmt::formatter<mijin::Stacktrace>
|
|
{
|
|
constexpr auto parse(format_parse_context& ctx) -> decltype(ctx.begin())
|
|
{
|
|
auto it = ctx.begin();
|
|
auto end = ctx.end();
|
|
|
|
if (it != end && *it != '}') throw format_error("invalid format");
|
|
|
|
return it;
|
|
}
|
|
|
|
template<typename TContext>
|
|
auto format(const mijin::Stacktrace& stacktrace, TContext& ctx) const -> decltype(ctx.out())
|
|
{
|
|
const int numDigits = static_cast<int>(std::ceil(std::log10(stacktrace.getFrames().size())));
|
|
auto it = ctx.out();
|
|
it = fmt::format_to(it, "[{} frames]", stacktrace.getFrames().size());
|
|
for (const auto& [idx, frame] : mijin::enumerate(stacktrace.getFrames()))
|
|
{
|
|
it = fmt::format_to(it, "\n #{:<{}} at {}", idx, numDigits, frame);
|
|
}
|
|
return it;
|
|
}
|
|
};
|
|
#endif // __has_include(<fmt/format.h>)
|
|
|
|
#endif // !defined(MIJIN_DEBUG_STACKTRACE_HPP_INCLUDED)
|