123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326 |
- // Copyright Antony Polukhin, 2016-2019.
- //
- // Distributed under the Boost Software License, Version 1.0. (See
- // accompanying file LICENSE_1_0.txt or copy at
- // http://www.boost.org/LICENSE_1_0.txt)
- #include <boost/array.hpp>
- BOOST_NOINLINE void foo(int i);
- BOOST_NOINLINE void bar(int i);
-
- BOOST_NOINLINE void bar(int i) {
- boost::array<int, 5> a = {{-1, -231, -123, -23, -32}};
- if (i >= 0) {
- foo(a[i]);
- } else {
- std::terminate();
- }
- }
- BOOST_NOINLINE void foo(int i) {
- bar(--i);
- }
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- //[getting_started_terminate_handlers
- #include <signal.h> // ::signal, ::raise
- #include <boost/stacktrace.hpp>
- void my_signal_handler(int signum) {
- ::signal(signum, SIG_DFL);
- boost::stacktrace::safe_dump_to("./backtrace.dump");
- ::raise(SIGABRT);
- }
- //]
- void setup_handlers() {
- //[getting_started_setup_handlers
- ::signal(SIGSEGV, &my_signal_handler);
- ::signal(SIGABRT, &my_signal_handler);
- //]
- }
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- BOOST_CONSTEXPR_OR_CONST std::size_t shared_memory_size = 4096 * 8;
- //[getting_started_terminate_handlers_shmem
- #include <boost/stacktrace.hpp>
- #include <boost/interprocess/shared_memory_object.hpp>
- #include <boost/interprocess/mapped_region.hpp>
- boost::interprocess::shared_memory_object g_shm; // inited at program start
- boost::interprocess::mapped_region g_region; // inited at program start
- void my_signal_handler2(int signum) {
- ::signal(signum, SIG_DFL);
- void** f = static_cast<void**>(g_region.get_address());
- *f = reinterpret_cast<void*>(1); // Setting flag that shared memory now constains stacktrace.
- boost::stacktrace::safe_dump_to(f + 1, g_region.get_size() - sizeof(void*));
- ::raise(SIGABRT);
- }
- //]
- #include <iostream> // std::cerr
- #include <fstream> // std::ifstream
- #include <boost/filesystem/path.hpp>
- #include <boost/filesystem/operations.hpp>
- inline void copy_and_run(const char* exec_name, char param, bool not_null) {
- std::cout << "Running with param " << param << std::endl;
- boost::filesystem::path command = exec_name;
- command = command.parent_path() / (command.stem().string() + param + command.extension().string());
- boost::filesystem::copy_file(exec_name, command, boost::filesystem::copy_option::overwrite_if_exists);
- boost::filesystem::path command_args = command;
- command_args += ' ';
- command_args += param;
- const int ret = std::system(command_args.string().c_str());
- std::cout << "End Running with param " << param << "; ret code is " << ret << std::endl;
- boost::system::error_code ignore;
- boost::filesystem::remove(command, ignore);
- if (not_null && !ret) {
- std::exit(97);
- } else if (!not_null && ret) {
- std::exit(ret);
- }
- }
- int run_1(const char* /*argv*/[]) {
- setup_handlers();
- foo(5);
- return 11;
- }
- int run_2(const char* argv[]) {
- if (!boost::filesystem::exists("./backtrace.dump")) {
- if (std::string(argv[0]).find("noop") == std::string::npos) {
- return 21;
- }
- boost::stacktrace::stacktrace st = boost::stacktrace::stacktrace::from_dump(std::cin);
- if (st) {
- return 22;
- }
- return 0;
- }
- //[getting_started_on_program_restart
- if (boost::filesystem::exists("./backtrace.dump")) {
- // there is a backtrace
- std::ifstream ifs("./backtrace.dump");
- boost::stacktrace::stacktrace st = boost::stacktrace::stacktrace::from_dump(ifs);
- std::cout << "Previous run crashed:\n" << st << std::endl; /*<-*/
- if (!st) {
- return 23;
- } /*->*/
- // cleaning up
- ifs.close();
- boost::filesystem::remove("./backtrace.dump");
- }
- //]
- return 0;
- }
- int run_3(const char* /*argv*/[]) {
- using namespace boost::interprocess;
- {
- shared_memory_object shm_obj(open_or_create, "shared_memory", read_write);
- shm_obj.swap(g_shm);
- }
- g_shm.truncate(shared_memory_size);
- {
- mapped_region m(g_shm, read_write, 0, shared_memory_size);
- m.swap(g_region);
- }
- void** f = static_cast<void**>(g_region.get_address());
- *f = 0;
- ::signal(SIGSEGV, &my_signal_handler2);
- ::signal(SIGABRT, &my_signal_handler2);
- foo(5);
- return 31;
- }
- int run_4(const char* argv[]) {
- using namespace boost::interprocess;
- {
- shared_memory_object shm_obj(open_only, "shared_memory", read_write);
- shm_obj.swap(g_shm);
- }
- {
- mapped_region m(g_shm, read_write, 0, shared_memory_size);
- m.swap(g_region);
- }
- //[getting_started_on_program_restart_shmem
- void** f = static_cast<void**>(g_region.get_address());
- if (*f) { // Checking if memory constains stacktrace.
- boost::stacktrace::stacktrace st
- = boost::stacktrace::stacktrace::from_dump(f + 1, g_region.get_size() - sizeof(bool));
- std::cout << "Previous run crashed and left trace in shared memory:\n" << st << std::endl;
- *f = 0; /*<-*/
- shared_memory_object::remove("shared_memory");
- if (std::string(argv[0]).find("noop") == std::string::npos) {
- if (!st) {
- return 43;
- }
- } else {
- if (st) {
- return 44;
- }
- }
- } else {
- return 42; /*->*/
- }
- //]
- return 0;
- }
- #include <sstream>
- int test_inplace() {
- const bool is_noop = !boost::stacktrace::stacktrace();
- {
- // This is very dependent on compiler and link flags. No sane way to make it work, because:
- // * BOOST_NOINLINE could be ignored by MSVC compiler if link-time optimization is enabled.
- // * BOOST_FORCEINLINE could be ignored by GCC depending on the std::vector default constructor length.
- const std::size_t frames_ss1 = boost::stacktrace::safe_dump_to("./backtrace2.dump");
- boost::stacktrace::stacktrace ss2;
- std::ifstream ifs("./backtrace2.dump");
- boost::stacktrace::stacktrace ss1 = boost::stacktrace::stacktrace::from_dump(ifs);
- ifs.close();
- boost::filesystem::remove("./backtrace2.dump");
- if (ss1.size() + 1 != frames_ss1 || ss2.size() != ss1.size()) {
- std::cerr << "51: Stacktraces differ. Dumped size == " << frames_ss1 << ".\n" << ss1 << "\n vs \n" << ss2 << '\n';
- } else if (ss1.size() > 1 && ss1[1].name() != ss2[1].name()) {
- std::cerr << "52: Stacktraces differ:\n" << ss1 << "\n vs \n" << ss2 << '\n';
- }
- }
- {
- // This is very dependent on compiler and link flags. No sane way to make it work, because:
- // * BOOST_NOINLINE could be ignored by MSVC compiler if link-time optimization is enabled.
- // * BOOST_FORCEINLINE could be ignored by GCC depending on the std::vector default constructor length.
- void* data[1024];
- const std::size_t frames_ss1 = boost::stacktrace::safe_dump_to(data, sizeof(data));
- boost::stacktrace::stacktrace ss2;
- boost::stacktrace::stacktrace ss1 = boost::stacktrace::stacktrace::from_dump(data, sizeof(data));
- if (ss1.size() + 1 != frames_ss1 || ss1.size() != ss2.size()) {
- std::cerr << "53: Stacktraces differ. Dumped size == " << frames_ss1 << ".\n" << ss1 << "\n vs \n" << ss2 << '\n';
- } else if (ss1.size() > 1 && ss1[1].name() != ss2[1].name()) {
- std::cerr << "54: Stacktraces differ:\n" << ss1 << "\n vs \n" << ss2 << '\n';
- }
- }
- {
- void* data[1024];
- boost::stacktrace::safe_dump_to(1024, data, sizeof(data));
- if (boost::stacktrace::stacktrace::from_dump(data, sizeof(data))) {
- std::cerr << "Stacktrace not empty!\n";
- return 55;
- }
- }
- {
- void* data[1024];
- boost::stacktrace::safe_dump_to(1, data, sizeof(data));
- if (!is_noop && !boost::stacktrace::stacktrace::from_dump(data, sizeof(data))) {
- std::cerr << "Stacktrace empty!\n";
- return 56;
- }
- const std::size_t size_1_skipped = boost::stacktrace::stacktrace::from_dump(data, sizeof(data)).size();
- boost::stacktrace::safe_dump_to(0, data, sizeof(data));
- const std::size_t size_0_skipped = boost::stacktrace::stacktrace::from_dump(data, sizeof(data)).size();
- if (!is_noop && (size_1_skipped + 1 != size_0_skipped)) {
- std::cerr << "failed to skip 1 frame!\n";
- return 57;
- }
- }
- {
- boost::stacktrace::safe_dump_to(0, 1, "./backtrace3.dump");
- std::ifstream ifs("./backtrace3.dump");
- boost::stacktrace::stacktrace ss1 = boost::stacktrace::stacktrace::from_dump(ifs);
- ifs.close();
- boost::stacktrace::safe_dump_to(1, 1, "./backtrace3.dump");
- ifs.open("./backtrace3.dump");
- boost::stacktrace::stacktrace ss2 = boost::stacktrace::stacktrace::from_dump(ifs);
- ifs.close();
- boost::filesystem::remove("./backtrace3.dump");
- #ifdef BOOST_WINDOWS
- // `ss2` could be empty on some combinations of Windows+MSVC.
- if (!ss2) {
- return 0;
- }
- #endif
- if (ss1.size() != ss2.size()) {
- std::cerr << "Stacktraces differ:\n" << ss1 << "\n vs \n" << ss2 << '\n';
- return 58;
- }
- if (!is_noop && ss1.size() != 1) {
- std::cerr << "Stacktraces does not have size 1:\n" << ss1 << '\n';
- return 59;
- }
- if (ss1 && ss1[0].address() == ss2[0].address()) {
- std::cerr << "Stacktraces must differ:\n" << ss1 << "\n vs \n" << ss2 << '\n';
- return 60;
- }
- }
- return 0;
- }
- int main(int argc, const char* argv[]) {
- if (argc < 2) {
- #ifndef BOOST_WINDOWS
- // We are copying files to make sure that stacktrace printing works independently from executable name
- copy_and_run(argv[0], '1', true);
- copy_and_run(argv[0], '2', false);
- // There are some issues with async-safety of shared memory writes on Windows.
- copy_and_run(argv[0], '3', true);
- copy_and_run(argv[0], '4', false);
- #endif
- return test_inplace();
- }
- switch (argv[1][0]) {
- case '1': return run_1(argv);
- case '2': return run_2(argv);
- case '3': return run_3(argv);
- case '4': return run_4(argv);
- }
- return 404;
- }
|