frame_msvc.ipp 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403
  1. // Copyright Antony Polukhin, 2016-2021.
  2. //
  3. // Distributed under the Boost Software License, Version 1.0. (See
  4. // accompanying file LICENSE_1_0.txt or copy at
  5. // http://www.boost.org/LICENSE_1_0.txt)
  6. #ifndef BOOST_STACKTRACE_DETAIL_FRAME_MSVC_IPP
  7. #define BOOST_STACKTRACE_DETAIL_FRAME_MSVC_IPP
  8. #include <boost/config.hpp>
  9. #ifdef BOOST_HAS_PRAGMA_ONCE
  10. # pragma once
  11. #endif
  12. #include <boost/stacktrace/frame.hpp>
  13. #include <boost/core/demangle.hpp>
  14. #include <boost/core/noncopyable.hpp>
  15. #include <boost/stacktrace/detail/to_dec_array.hpp>
  16. #include <boost/stacktrace/detail/to_hex_array.hpp>
  17. #include <windows.h>
  18. #include "dbgeng.h"
  19. #ifdef BOOST_MSVC
  20. # pragma comment(lib, "ole32.lib")
  21. # pragma comment(lib, "Dbgeng.lib")
  22. #endif
  23. #ifdef __CRT_UUID_DECL // for __MINGW32__
  24. __CRT_UUID_DECL(IDebugClient,0x27fe5639,0x8407,0x4f47,0x83,0x64,0xee,0x11,0x8f,0xb0,0x8a,0xc8)
  25. __CRT_UUID_DECL(IDebugControl,0x5182e668,0x105e,0x416e,0xad,0x92,0x24,0xef,0x80,0x04,0x24,0xba)
  26. __CRT_UUID_DECL(IDebugSymbols,0x8c31e98c,0x983a,0x48a5,0x90,0x16,0x6f,0xe5,0xd6,0x67,0xa9,0x50)
  27. #elif defined(DEFINE_GUID) && !defined(BOOST_MSVC)
  28. DEFINE_GUID(IID_IDebugClient,0x27fe5639,0x8407,0x4f47,0x83,0x64,0xee,0x11,0x8f,0xb0,0x8a,0xc8);
  29. DEFINE_GUID(IID_IDebugControl,0x5182e668,0x105e,0x416e,0xad,0x92,0x24,0xef,0x80,0x04,0x24,0xba);
  30. DEFINE_GUID(IID_IDebugSymbols,0x8c31e98c,0x983a,0x48a5,0x90,0x16,0x6f,0xe5,0xd6,0x67,0xa9,0x50);
  31. #endif
  32. // Testing. Remove later
  33. //# define __uuidof(x) ::IID_ ## x
  34. namespace boost { namespace stacktrace { namespace detail {
  35. class com_global_initer: boost::noncopyable {
  36. bool ok_;
  37. public:
  38. com_global_initer() BOOST_NOEXCEPT
  39. : ok_(false)
  40. {
  41. // COINIT_MULTITHREADED means that we must serialize access to the objects manually.
  42. // This is the fastest way to work. If user calls CoInitializeEx before us - we
  43. // can end up with other mode (which is OK for us).
  44. //
  45. // If we call CoInitializeEx befire user - user may end up with different mode, which is a problem.
  46. // So we need to call that initialization function as late as possible.
  47. const DWORD res = ::CoInitializeEx(0, COINIT_MULTITHREADED);
  48. ok_ = (res == S_OK || res == S_FALSE);
  49. }
  50. ~com_global_initer() BOOST_NOEXCEPT {
  51. if (ok_) {
  52. ::CoUninitialize();
  53. }
  54. }
  55. };
  56. template <class T>
  57. class com_holder: boost::noncopyable {
  58. T* holder_;
  59. public:
  60. com_holder(const com_global_initer&) BOOST_NOEXCEPT
  61. : holder_(0)
  62. {}
  63. T* operator->() const BOOST_NOEXCEPT {
  64. return holder_;
  65. }
  66. void** to_void_ptr_ptr() BOOST_NOEXCEPT {
  67. return reinterpret_cast<void**>(&holder_);
  68. }
  69. bool is_inited() const BOOST_NOEXCEPT {
  70. return !!holder_;
  71. }
  72. ~com_holder() BOOST_NOEXCEPT {
  73. if (holder_) {
  74. holder_->Release();
  75. }
  76. }
  77. };
  78. inline std::string mingw_demangling_workaround(const std::string& s) {
  79. #ifdef BOOST_GCC
  80. if (s.empty()) {
  81. return s;
  82. }
  83. if (s[0] != '_') {
  84. return boost::core::demangle(('_' + s).c_str());
  85. }
  86. return boost::core::demangle(s.c_str());
  87. #else
  88. return s;
  89. #endif
  90. }
  91. inline void trim_right_zeroes(std::string& s) {
  92. // MSVC-9 does not have back() and pop_back() functions in std::string
  93. while (!s.empty()) {
  94. const std::size_t last = static_cast<std::size_t>(s.size() - 1);
  95. if (s[last] != '\0') {
  96. break;
  97. }
  98. s.resize(last);
  99. }
  100. }
  101. class debugging_symbols: boost::noncopyable {
  102. static void try_init_com(com_holder< ::IDebugSymbols>& idebug, const com_global_initer& com) BOOST_NOEXCEPT {
  103. com_holder< ::IDebugClient> iclient(com);
  104. if (S_OK != ::DebugCreate(__uuidof(IDebugClient), iclient.to_void_ptr_ptr())) {
  105. return;
  106. }
  107. com_holder< ::IDebugControl> icontrol(com);
  108. const bool res0 = (S_OK == iclient->QueryInterface(
  109. __uuidof(IDebugControl),
  110. icontrol.to_void_ptr_ptr()
  111. ));
  112. if (!res0) {
  113. return;
  114. }
  115. const bool res1 = (S_OK == iclient->AttachProcess(
  116. 0,
  117. ::GetCurrentProcessId(),
  118. DEBUG_ATTACH_NONINVASIVE | DEBUG_ATTACH_NONINVASIVE_NO_SUSPEND
  119. ));
  120. if (!res1) {
  121. return;
  122. }
  123. if (S_OK != icontrol->WaitForEvent(DEBUG_WAIT_DEFAULT, INFINITE)) {
  124. return;
  125. }
  126. // No cheking: QueryInterface sets the output parameter to NULL in case of error.
  127. iclient->QueryInterface(__uuidof(IDebugSymbols), idebug.to_void_ptr_ptr());
  128. }
  129. #ifndef BOOST_STACKTRACE_USE_WINDBG_CACHED
  130. boost::stacktrace::detail::com_global_initer com_;
  131. com_holder< ::IDebugSymbols> idebug_;
  132. public:
  133. debugging_symbols() BOOST_NOEXCEPT
  134. : com_()
  135. , idebug_(com_)
  136. {
  137. try_init_com(idebug_, com_);
  138. }
  139. #else
  140. #ifdef BOOST_NO_CXX11_THREAD_LOCAL
  141. # error Your compiler does not support C++11 thread_local storage. It`s impossible to build with BOOST_STACKTRACE_USE_WINDBG_CACHED.
  142. #endif
  143. static com_holder< ::IDebugSymbols>& get_thread_local_debug_inst() BOOST_NOEXCEPT {
  144. // [class.mfct]: A static local variable or local type in a member function always refers to the same entity, whether
  145. // or not the member function is inline.
  146. static thread_local boost::stacktrace::detail::com_global_initer com;
  147. static thread_local com_holder< ::IDebugSymbols> idebug(com);
  148. if (!idebug.is_inited()) {
  149. try_init_com(idebug, com);
  150. }
  151. return idebug;
  152. }
  153. com_holder< ::IDebugSymbols>& idebug_;
  154. public:
  155. debugging_symbols() BOOST_NOEXCEPT
  156. : idebug_( get_thread_local_debug_inst() )
  157. {}
  158. #endif // #ifndef BOOST_STACKTRACE_USE_WINDBG_CACHED
  159. bool is_inited() const BOOST_NOEXCEPT {
  160. return idebug_.is_inited();
  161. }
  162. std::string get_name_impl(const void* addr, std::string* module_name = 0) const {
  163. std::string result;
  164. if (!is_inited()) {
  165. return result;
  166. }
  167. const ULONG64 offset = reinterpret_cast<ULONG64>(addr);
  168. char name[256];
  169. name[0] = '\0';
  170. ULONG size = 0;
  171. bool res = (S_OK == idebug_->GetNameByOffset(
  172. offset,
  173. name,
  174. sizeof(name),
  175. &size,
  176. 0
  177. ));
  178. if (!res && size != 0) {
  179. result.resize(size);
  180. res = (S_OK == idebug_->GetNameByOffset(
  181. offset,
  182. &result[0],
  183. static_cast<ULONG>(result.size()),
  184. &size,
  185. 0
  186. ));
  187. trim_right_zeroes(result);
  188. } else if (res) {
  189. result = name;
  190. }
  191. if (!res) {
  192. result.clear();
  193. return result;
  194. }
  195. const std::size_t delimiter = result.find_first_of('!');
  196. if (module_name) {
  197. *module_name = result.substr(0, delimiter);
  198. }
  199. if (delimiter == std::string::npos) {
  200. // If 'delimiter' is equal to 'std::string::npos' then we have only module name.
  201. result.clear();
  202. return result;
  203. }
  204. result = mingw_demangling_workaround(
  205. result.substr(delimiter + 1)
  206. );
  207. return result;
  208. }
  209. std::size_t get_line_impl(const void* addr) const BOOST_NOEXCEPT {
  210. ULONG result = 0;
  211. if (!is_inited()) {
  212. return result;
  213. }
  214. const bool is_ok = (S_OK == idebug_->GetLineByOffset(
  215. reinterpret_cast<ULONG64>(addr),
  216. &result,
  217. 0,
  218. 0,
  219. 0,
  220. 0
  221. ));
  222. return (is_ok ? result : 0);
  223. }
  224. std::pair<std::string, std::size_t> get_source_file_line_impl(const void* addr) const {
  225. std::pair<std::string, std::size_t> result;
  226. if (!is_inited()) {
  227. return result;
  228. }
  229. const ULONG64 offset = reinterpret_cast<ULONG64>(addr);
  230. char name[256];
  231. name[0] = 0;
  232. ULONG size = 0;
  233. ULONG line_num = 0;
  234. bool res = (S_OK == idebug_->GetLineByOffset(
  235. offset,
  236. &line_num,
  237. name,
  238. sizeof(name),
  239. &size,
  240. 0
  241. ));
  242. if (res) {
  243. result.first = name;
  244. result.second = line_num;
  245. return result;
  246. }
  247. if (!res && size == 0) {
  248. return result;
  249. }
  250. result.first.resize(size);
  251. res = (S_OK == idebug_->GetLineByOffset(
  252. offset,
  253. &line_num,
  254. &result.first[0],
  255. static_cast<ULONG>(result.first.size()),
  256. &size,
  257. 0
  258. ));
  259. trim_right_zeroes(result.first);
  260. result.second = line_num;
  261. if (!res) {
  262. result.first.clear();
  263. result.second = 0;
  264. }
  265. return result;
  266. }
  267. void to_string_impl(const void* addr, std::string& res) const {
  268. if (!is_inited()) {
  269. return;
  270. }
  271. std::string module_name;
  272. std::string name = this->get_name_impl(addr, &module_name);
  273. if (!name.empty()) {
  274. res += name;
  275. } else {
  276. res += to_hex_array(addr).data();
  277. }
  278. std::pair<std::string, std::size_t> source_line = this->get_source_file_line_impl(addr);
  279. if (!source_line.first.empty() && source_line.second) {
  280. res += " at ";
  281. res += source_line.first;
  282. res += ':';
  283. res += boost::stacktrace::detail::to_dec_array(source_line.second).data();
  284. } else if (!module_name.empty()) {
  285. res += " in ";
  286. res += module_name;
  287. }
  288. }
  289. };
  290. std::string to_string(const frame* frames, std::size_t size) {
  291. boost::stacktrace::detail::debugging_symbols idebug;
  292. if (!idebug.is_inited()) {
  293. return std::string();
  294. }
  295. std::string res;
  296. res.reserve(64 * size);
  297. for (std::size_t i = 0; i < size; ++i) {
  298. if (i < 10) {
  299. res += ' ';
  300. }
  301. res += boost::stacktrace::detail::to_dec_array(i).data();
  302. res += '#';
  303. res += ' ';
  304. idebug.to_string_impl(frames[i].address(), res);
  305. res += '\n';
  306. }
  307. return res;
  308. }
  309. } // namespace detail
  310. std::string frame::name() const {
  311. boost::stacktrace::detail::debugging_symbols idebug;
  312. return idebug.get_name_impl(addr_);
  313. }
  314. std::string frame::source_file() const {
  315. boost::stacktrace::detail::debugging_symbols idebug;
  316. return idebug.get_source_file_line_impl(addr_).first;
  317. }
  318. std::size_t frame::source_line() const {
  319. boost::stacktrace::detail::debugging_symbols idebug;
  320. return idebug.get_line_impl(addr_);
  321. }
  322. std::string to_string(const frame& f) {
  323. std::string res;
  324. boost::stacktrace::detail::debugging_symbols idebug;
  325. idebug.to_string_impl(f.address(), res);
  326. return res;
  327. }
  328. }} // namespace boost::stacktrace
  329. #endif // BOOST_STACKTRACE_DETAIL_FRAME_MSVC_IPP