Add size info for precompiled methods

microsoft · Aug 4, 2019 · 932e547 · 932e547
1 parent d51ab3f
commit 932e547
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 0 deletions.
diff --git a/CoreCLRProfiler/native/BPerfProfilerCallback.cpp b/CoreCLRProfiler/native/BPerfProfilerCallback.cpp
@@ -27,6 +27,7 @@ BPerfProfilerCallback::BPerfProfilerCallback()
     /* Precompiled Methods Counters */
     this->totalCachedMethodsSearched = 0;
     this->totalCachedMethodsRestored = 0;
+    this->totalCachedMethodsMachineCodeBytesRestored = 0;
 
     /* Runtime Suspension Counters */
     this->totalNumberOfRuntimeSuspensions = 0;
@@ -273,6 +274,19 @@ HRESULT STDMETHODCALLTYPE BPerfProfilerCallback::JITCachedFunctionSearchFinished
     if (result == COR_PRF_CACHED_FUNCTION_FOUND)
     {
         ++this->totalCachedMethodsRestored;
+
+        ULONG32 cCodeInfos;
+        IfFailRet(this->corProfilerInfo->GetCodeInfo2(functionId, 0, &cCodeInfos, nullptr));
+        std::vector<COR_PRF_CODE_INFO> codeInfos(cCodeInfos);
+        IfFailRet(this->corProfilerInfo->GetCodeInfo2(functionId, cCodeInfos, &cCodeInfos, codeInfos.data()));
+
+        size_t codeSize = 0;
+        for (auto& s : codeInfos)
+        {
+            codeSize += s.size;
+        }
+
+        this->totalCachedMethodsMachineCodeBytesRestored += codeSize;
     }
 
     return S_OK;
@@ -796,6 +810,11 @@ size_t BPerfProfilerCallback::GetTotalCachedMethodsRestored() const
     return this->totalCachedMethodsRestored;
 }
 
+size_t BPerfProfilerCallback::GetTotalCachedMethodsMachineCodeBytesRestored() const
+{
+    return this->totalCachedMethodsMachineCodeBytesRestored;
+}
+
 /* Runtime Suspension Counters */
 size_t BPerfProfilerCallback::GetTotalNumberOfRuntimeSuspsensions() const
 {

diff --git a/CoreCLRProfiler/native/BPerfProfilerCallback.h b/CoreCLRProfiler/native/BPerfProfilerCallback.h
@@ -139,6 +139,7 @@ class BPerfProfilerCallback final : public ICorProfilerCallback9
     /* Precompiled Method Counters */
     size_t GetTotalCachedMethodsSearched() const;
     size_t GetTotalCachedMethodsRestored() const;
+    size_t GetTotalCachedMethodsMachineCodeBytesRestored() const;
 
     /* Runtime Suspension Counters */
     size_t GetTotalNumberOfRuntimeSuspsensions() const;
@@ -237,6 +238,7 @@ class BPerfProfilerCallback final : public ICorProfilerCallback9
     /* Precompiled Methods Counters */
     std::atomic<size_t> totalCachedMethodsSearched;
     std::atomic<size_t> totalCachedMethodsRestored;
+    std::atomic<size_t> totalCachedMethodsMachineCodeBytesRestored;
 
     /* Runtime Suspension Counters */
     std::atomic<size_t> totalNumberOfRuntimeSuspensions;

diff --git a/CoreCLRProfiler/native/Microsoft.BPerf.CoreCLRProfiler.cpp b/CoreCLRProfiler/native/Microsoft.BPerf.CoreCLRProfiler.cpp
@@ -74,6 +74,11 @@ extern "C" size_t GetTotalCachedMethodsRestored()
     return ProfilerInstance->GetTotalCachedMethodsRestored();
 }
 
+extern "C" size_t GetTotalCachedMethodsMachineCodeBytesRestored()
+{
+    return ProfilerInstance->GetTotalCachedMethodsMachineCodeBytesRestored();
+}
+
 /* Runtime Suspension Counters */
 extern "C" size_t GetTotalNumberOfRuntimeSuspsensions()
 {

diff --git a/CoreCLRProfiler/native/exports.def b/CoreCLRProfiler/native/exports.def
@@ -16,6 +16,7 @@ EXPORTS
     GetCurrentNumberOfDynamicMethods
     GetTotalCachedMethodsSearched
     GetTotalCachedMethodsRestored
+    GetTotalCachedMethodsMachineCodeBytesRestored
     GetTotalNumberOfRuntimeSuspsensions
     GetTotalNumberOfRuntimeSuspensionsForGC
     GetTotalNumberOfRuntimeSuspensionsForGCPrep