v8dbg/dbgext.cpp at master · billti/v8dbg · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#include "dbgext.h"
#include <crtdbg.h>

// See https://docs.microsoft.com/en-us/visualstudio/debugger/crt-debugging-techniques
// for the memory leak and debugger reporting macros used from <crtdbg.h>
_CrtMemState memOld, memNew, memDiff;

winrt::com_ptr<IDataModelManager> spDataModelManager;
winrt::com_ptr<IDebugHost> spDebugHost;
winrt::com_ptr<IDebugControl5> spDebugControl;

extern "C" {

__declspec(dllexport) HRESULT
    __stdcall DebugExtensionInitialize(PULONG /*pVersion*/, PULONG /*pFlags*/) {
  _RPTF0(_CRT_WARN, "Entered DebugExtensionInitialize\n");
  _CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF);
  _CrtMemCheckpoint(&memOld);

  winrt::com_ptr<IDebugClient> spDebugClient;
  winrt::com_ptr<IHostDataModelAccess> spDataModelAccess;

  HRESULT hr = DebugCreate(__uuidof(IDebugClient), spDebugClient.put_void());
  if (FAILED(hr)) return E_FAIL;

  if (!spDebugClient.try_as(spDataModelAccess)) return E_FAIL;
  if (!spDebugClient.try_as(spDebugControl)) return E_FAIL;

  hr = spDataModelAccess->GetDataModel(spDataModelManager.put(),
                                       spDebugHost.put());
  if (FAILED(hr)) return E_FAIL;

  return CreateExtension() ? S_OK : E_FAIL;
  return S_OK;
}

__declspec(dllexport) void __stdcall DebugExtensionUninitialize() {
  _RPTF0(_CRT_WARN, "Entered DebugExtensionUninitialize\n");
  DestroyExtension();
  spDebugHost = nullptr;
  spDataModelManager = nullptr;

  _CrtMemCheckpoint(&memNew);
  if (_CrtMemDifference(&memDiff, &memOld, &memNew)) {
    _CrtMemDumpStatistics(&memDiff);
  }
  if (_CrtDumpMemoryLeaks()) {
    _RPTF0(_CRT_ERROR, "Memory leaks detected!\n");
  }
}

__declspec(dllexport) HRESULT __stdcall DebugExtensionCanUnload(void) {
  _RPTF0(_CRT_WARN, "Entered DebugExtensionCanUnload\n");
  uint32_t lock_count = winrt::get_module_lock().load();
  _RPTF1(_CRT_WARN, "module_lock count in DebugExtensionCanUnload is %d\n",
         lock_count);
  return lock_count == 0 ? S_OK : S_FALSE;
}

__declspec(dllexport) void __stdcall DebugExtensionUnload() {
  _RPTF0(_CRT_WARN, "Entered DebugExtensionUnload\n");
  return;
}

}  // extern "C"