llstacktrace.cpp
上传用户:king477883
上传日期:2021-03-01
资源大小:9553k
文件大小:4k
源码类别:

游戏引擎

开发平台:

C++ Builder

  1. /** 
  2.  * @file llstacktrace.cpp
  3.  * @brief stack tracing functionality
  4.  *
  5.  * $LicenseInfo:firstyear=2001&license=viewergpl$
  6.  * 
  7.  * Copyright (c) 2001-2010, Linden Research, Inc.
  8.  * 
  9.  * Second Life Viewer Source Code
  10.  * The source code in this file ("Source Code") is provided by Linden Lab
  11.  * to you under the terms of the GNU General Public License, version 2.0
  12.  * ("GPL"), unless you have obtained a separate licensing agreement
  13.  * ("Other License"), formally executed by you and Linden Lab.  Terms of
  14.  * the GPL can be found in doc/GPL-license.txt in this distribution, or
  15.  * online at http://secondlifegrid.net/programs/open_source/licensing/gplv2
  16.  * 
  17.  * There are special exceptions to the terms and conditions of the GPL as
  18.  * it is applied to this Source Code. View the full text of the exception
  19.  * in the file doc/FLOSS-exception.txt in this software distribution, or
  20.  * online at
  21.  * http://secondlifegrid.net/programs/open_source/licensing/flossexception
  22.  * 
  23.  * By copying, modifying or distributing this software, you acknowledge
  24.  * that you have read and understood your obligations described above,
  25.  * and agree to abide by those obligations.
  26.  * 
  27.  * ALL LINDEN LAB SOURCE CODE IS PROVIDED "AS IS." LINDEN LAB MAKES NO
  28.  * WARRANTIES, EXPRESS, IMPLIED OR OTHERWISE, REGARDING ITS ACCURACY,
  29.  * COMPLETENESS OR PERFORMANCE.
  30.  * $/LicenseInfo$
  31.  */
  32. #include "linden_common.h"
  33. #include "llstacktrace.h"
  34. #ifdef LL_WINDOWS
  35. #include <iostream>
  36. #include <sstream>
  37. #include "windows.h"
  38. #include "Dbghelp.h"
  39. typedef USHORT NTAPI RtlCaptureStackBackTrace_Function(
  40.     IN ULONG frames_to_skip,
  41.     IN ULONG frames_to_capture,
  42.     OUT PVOID *backtrace,
  43.     OUT PULONG backtrace_hash);
  44. static RtlCaptureStackBackTrace_Function* const RtlCaptureStackBackTrace_fn =
  45.    (RtlCaptureStackBackTrace_Function*)
  46.    GetProcAddress(GetModuleHandleA("ntdll.dll"), "RtlCaptureStackBackTrace");
  47. bool ll_get_stack_trace(std::vector<std::string>& lines)
  48. {
  49. const S32 MAX_STACK_DEPTH = 32;
  50. const S32 STRING_NAME_LENGTH = 200;
  51. const S32 FRAME_SKIP = 2;
  52. static BOOL symbolsLoaded = false;
  53. static BOOL firstCall = true;
  54. HANDLE hProc = GetCurrentProcess();
  55. // load the symbols if they're not loaded
  56. if(!symbolsLoaded && firstCall)
  57. {
  58. symbolsLoaded = SymInitialize(hProc, NULL, true);
  59. firstCall = false;
  60. }
  61. // if loaded, get the call stack
  62. if(symbolsLoaded)
  63. {
  64. // create the frames to hold the addresses
  65. void* frames[MAX_STACK_DEPTH];
  66. memset(frames, 0, sizeof(void*)*MAX_STACK_DEPTH);
  67. S32 depth = 0;
  68. // get the addresses
  69. depth = RtlCaptureStackBackTrace_fn(FRAME_SKIP, MAX_STACK_DEPTH, frames, NULL);
  70. IMAGEHLP_LINE64 line;
  71. memset(&line, 0, sizeof(IMAGEHLP_LINE64));
  72. line.SizeOfStruct = sizeof(IMAGEHLP_LINE64);
  73. // create something to hold address info
  74. PIMAGEHLP_SYMBOL64 pSym;
  75. pSym = (PIMAGEHLP_SYMBOL64)malloc(sizeof(IMAGEHLP_SYMBOL64) + STRING_NAME_LENGTH);
  76. memset(pSym, 0, sizeof(IMAGEHLP_SYMBOL64) + STRING_NAME_LENGTH);
  77. pSym->MaxNameLength = STRING_NAME_LENGTH;
  78. pSym->SizeOfStruct = sizeof(IMAGEHLP_SYMBOL64);
  79. // get address info for each address frame
  80. // and store
  81. for(S32 i=0; i < depth; i++)
  82. {
  83. std::stringstream stack_line;
  84. BOOL ret;
  85. DWORD64 addr = (DWORD64)frames[i];
  86. ret = SymGetSymFromAddr64(hProc, addr, 0, pSym);
  87. if(ret)
  88. {
  89. stack_line << pSym->Name << " ";
  90. }
  91. DWORD dummy;
  92. ret = SymGetLineFromAddr64(hProc, addr, &dummy, &line);
  93. if(ret)
  94. {
  95. std::string file_name = line.FileName;
  96. std::string::size_type index = file_name.rfind("\");
  97. stack_line << file_name.substr(index + 1, file_name.size()) << ":" << line.LineNumber; 
  98. }
  99. lines.push_back(stack_line.str());
  100. }
  101. free(pSym);
  102. // TODO: figure out a way to cleanup symbol loading
  103. // Not hugely necessary, however.
  104. //SymCleanup(hProc);
  105. return true;
  106. }
  107. else
  108. {
  109. lines.push_back("Stack Trace Failed.  PDB symbol info not loaded");
  110. }
  111. return false;
  112. }
  113. #else
  114. bool ll_get_stack_trace(std::vector<std::string>& lines)
  115. {
  116. return false;
  117. }
  118. #endif