luo980/cpptrace
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Resolver tool improvements

Browse Source
This commit is contained in:
Jeremy Rifkin 2025-02-17 12:46:25 -06:00
parent 87b14c87f8
commit 87f2fd4c43
No known key found for this signature in database
GPG Key ID: 19AA8270105E8EB4
1 changed files with 43 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
tools/resolver/main.cpp
View File

@ -1,14 +1,18 @@
#include "cpptrace/formatting.hpp" #include "cpptrace/formatting.hpp"
#include "cpptrace/forward.hpp" #include "cpptrace/forward.hpp"
#include <chrono>
#include <lyra/lyra.hpp> #include <lyra/lyra.hpp>
#include <fmt/format.h> #include <fmt/format.h>
#include <fmt/std.h> #include <fmt/std.h>
#include <fmt/ostream.h> #include <fmt/ostream.h>
#include <fmt/chrono.h>
#include <cpptrace/cpptrace.hpp> #include <cpptrace/cpptrace.hpp>
#include <cpptrace/from_current.hpp> #include <cpptrace/from_current.hpp>
#include <filesystem> #include <filesystem>
#include <stdexcept> #include <stdexcept>
#include <string>
#include <thread>
#include "symbols/symbols.hpp" #include "symbols/symbols.hpp"
#include "demangle/demangle.hpp" #include "demangle/demangle.hpp"
@ -20,51 +24,71 @@ template<> struct fmt::formatter<lyra::cli> : ostream_formatter {};
auto formatter = cpptrace::formatter{}.addresses(cpptrace::formatter::address_mode::object); auto formatter = cpptrace::formatter{}.addresses(cpptrace::formatter::address_mode::object);
void resolve(const std::filesystem::path& path, cpptrace::frame_ptr address) { struct options {
cpptrace::object_frame obj_frame{0, address, path}; bool show_help = false;
std::filesystem::path path;
std::vector<std::string> address_strings;
bool from_stdin = false;
bool keepalive = false;
bool timing = false;
};
void resolve(const options& opts, cpptrace::frame_ptr address) {
cpptrace::object_frame obj_frame{0, address, opts.path};
auto start = std::chrono::high_resolution_clock::now();
std::vector<cpptrace::stacktrace_frame> trace = cpptrace::detail::resolve_frames({obj_frame}); std::vector<cpptrace::stacktrace_frame> trace = cpptrace::detail::resolve_frames({obj_frame});
auto end = std::chrono::high_resolution_clock::now();
auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(end - start);
if(trace.size() != 1) { if(trace.size() != 1) {
throw std::runtime_error("Something went wrong, trace vector size didn't match"); throw std::runtime_error("Something went wrong, trace vector size didn't match");
} }
trace[0].symbol = cpptrace::demangle(trace[0].symbol); trace[0].symbol = cpptrace::demangle(trace[0].symbol);
formatter.print(trace[0]); formatter.print(trace[0]);
std::cout<<std::endl; std::cout<<std::endl;
if(opts.timing) {
fmt::println("resolve time: {}", duration);
}
} }
int main(int argc, char** argv) CPPTRACE_TRY { int main(int argc, char** argv) CPPTRACE_TRY {
bool show_help = false; options opts;
std::filesystem::path path;
std::vector<std::string> address_strings;
bool from_stdin = false;
auto cli = lyra::cli() auto cli = lyra::cli()
| lyra::help(show_help) | lyra::help(opts.show_help)
| lyra::opt(from_stdin)["--stdin"]("read addresses from stdin") | lyra::opt(opts.from_stdin)["--stdin"]("read addresses from stdin")
| lyra::arg(path, "binary path")("binary to look in").required() | lyra::opt(opts.keepalive)["--keepalive"]("keep the program alive after resolution finishes (useful for debugging)")
| lyra::arg(address_strings, "addresses")("addresses"); | lyra::opt(opts.timing)["--timing"]("provide timing stats")
| lyra::arg(opts.path, "binary path")("binary to look in").required()
| lyra::arg(opts.address_strings, "addresses")("addresses");
if(auto result = cli.parse({ argc, argv }); !result) { if(auto result = cli.parse({ argc, argv }); !result) {
fmt::println(stderr, "Error in command line: {}", result.message()); fmt::println(stderr, "Error in command line: {}", result.message());
fmt::println("{}", cli); fmt::println("{}", cli);
return 1; return 1;
} }
if(show_help) { if(opts.show_help) {
fmt::println("{}", cli); fmt::println("{}", cli);
return 0; return 0;
} }
if(!std::filesystem::exists(path)) { if(!std::filesystem::exists(opts.path)) {
fmt::println(stderr, "Error: Path doesn't exist {}", path); fmt::println(stderr, "Error: Path doesn't exist {}", opts.path);
return 1; return 1;
} }
if(!std::filesystem::is_regular_file(path)) { if(!std::filesystem::is_regular_file(opts.path)) {
fmt::println(stderr, "Error: Path isn't a regular file {}", path); fmt::println(stderr, "Error: Path isn't a regular file {}", opts.path);
return 1; return 1;
} }
for(const auto& address : address_strings) { for(const auto& address : opts.address_strings) {
resolve(path, std::stoi(address, nullptr, 16)); resolve(opts, std::stoi(address, nullptr, 16));
} }
if(from_stdin) { if(opts.from_stdin) {
std::string word; std::string word;
while(std::cin >> word) { while(std::cin >> word) {
resolve(path, std::stoi(word, nullptr, 16)); resolve(opts, std::stoi(word, nullptr, 16));
}
}
if(opts.keepalive) {
fmt::println("Done");
while(true) {
std::this_thread::sleep_for(std::chrono::seconds(60));
} }
} }
} CPPTRACE_CATCH(const std::exception& e) { } CPPTRACE_CATCH(const std::exception& e) {