1 files changed, 30 insertions, 0 deletions

diff --git a/src/stub/stub.cpp b/src/stub/stub.cpp
index 2f09eede..d3ace1f3 100644
--- a/src/stub/stub.cpp
+++ b/src/stub/stub.cpp
@@ -3,6 +3,16 @@
 #include "precomp.h"
 
 
+static void CALLBACK BurnTraceError(
+    __in_z LPCSTR szFile,
+    __in int iLine,
+    __in REPORT_LEVEL rl,
+    __in UINT source,
+    __in HRESULT hrError,
+    __in_z __format_string LPCSTR szFormat,
+    __in va_list args
+    );
+
 int WINAPI wWinMain(
     __in HINSTANCE hInstance,
     __in_opt HINSTANCE /* hPrevInstance */,
@@ -30,6 +40,8 @@ int WINAPI wWinMain(
         L"feclient.dll", // unsafely loaded by DecryptFile().
     };
 
+    DutilInitialize(&BurnTraceError);
+
     // Best effort attempt to get our file handle as soon as possible.
     hr = PathForCurrentProcess(&sczPath, NULL);
     if (SUCCEEDED(hr))
@@ -60,5 +72,23 @@ LExit:
     ReleaseFileHandle(hEngineFile);
     ReleaseStr(sczPath);
 
+    DutilUninitialize();
+
     return FAILED(hr) ? (int)hr : (int)dwExitCode;
 }
+
+static void CALLBACK BurnTraceError(
+    __in_z LPCSTR /*szFile*/,
+    __in int /*iLine*/,
+    __in REPORT_LEVEL /*rl*/,
+    __in UINT source,
+    __in HRESULT hrError,
+    __in_z __format_string LPCSTR szFormat,
+    __in va_list args
+    )
+{
+    if (DUTIL_SOURCE_DEFAULT == source)
+    {
+        LogErrorStringArgs(hrError, szFormat, args);
+    }
+}