1 files changed, 36 insertions, 0 deletions

diff --git a/src/ext/Util/ca/CheckReboot.cpp b/src/ext/Util/ca/CheckReboot.cpp
new file mode 100644
index 00000000..ce056411
--- /dev/null
+++ b/src/ext/Util/ca/CheckReboot.cpp
@@ -0,0 +1,36 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+#include "precomp.h"
+
+
+/********************************************************************
+WixCheckRebootRequired - entry point for WixCheckRebootRequired Custom Action
+
+ called as Type 1 CustomAction (binary DLL) from Windows Installer 
+ in InstallExecuteSequence after InstallFinalize
+********************************************************************/
+extern "C" UINT __stdcall WixCheckRebootRequired(
+    __in MSIHANDLE hInstall
+    )
+{
+    HRESULT hr = S_OK;
+    DWORD er = ERROR_SUCCESS;
+
+    hr = WcaInitialize(hInstall, "WixCheckRebootRequired");
+    ExitOnFailure(hr, "failed to initialize");
+
+    if (WcaDidDeferredActionRequireReboot())
+    {
+        WcaLog(LOGMSG_STANDARD, "Reboot required by deferred CustomAction.");
+
+        er = ::MsiSetMode(hInstall, MSIRUNMODE_REBOOTATEND, TRUE);
+        hr = HRESULT_FROM_WIN32(er);
+        ExitOnFailure(hr, "Failed to schedule reboot.");
+    }
+
+LExit:
+
+    if (FAILED(hr))
+        er = ERROR_INSTALL_FAILURE;
+    return WcaFinalize(er);
+}