Fix compilation errors on non-windows platforms

dotnet · Jul 8, 2024 · 5458ce5 · 5458ce5
1 parent 6818023
commit 5458ce5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/tests/profiler/native/gcheapenumerationprofiler/gcheapenumerationprofiler.cpp b/src/tests/profiler/native/gcheapenumerationprofiler/gcheapenumerationprofiler.cpp
@@ -2,7 +2,7 @@
 // The .NET Foundation licenses this file to you under the MIT license.
 
 #include "gcheapenumerationprofiler.h"
-#include <wchar.h>
+#include "../profilerstring.h"
 
 #if WIN32
 #define EXPORT
@@ -34,7 +34,9 @@ HRESULT GCHeapEnumerationProfiler::Initialize(IUnknown* pICorProfilerInfoUnk)
         printf("FAIL: ICorProfilerInfo::GetEnvironmentVariable() failed hr=0x%x", hr);
         return hr;
     }
-    if (wcscmp(envVar, L"TRUE") == 0)
+    String envVarStr = envVar;
+    String expectedEnvVarStr = reinterpret_cast<const WCHAR*>(u"TRUE");
+    if (envVarStr == expectedEnvVarStr)
     {
         printf("Setting GarbageCollectionStarted event masks\n");
         eventMask |= COR_PRF_MONITOR_GC;
@@ -128,7 +130,8 @@ static BOOL STDMETHODCALLTYPE heap_walk_fn(ObjectID object, void* callbackState)
     }
 
     String classIdName = state->instance->GetClassIDNameHelper(classId);
-    if (classIdName.ToWString() == L"CustomGCHeapObject")
+    String expectedCustomObjectClass = reinterpret_cast<const WCHAR*>(u"CustomGCHeapObject");
+    if (classIdName == expectedCustomObjectClass)
     {
         state->customGcHeapObjectTypeCount->fetch_add(1, std::memory_order_relaxed);
     }