VeraCrypt
aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/Common/BootEncryption.cpp4
-rw-r--r--src/Common/BootEncryption.h2
-rw-r--r--src/Common/Dlgcode.c4
-rw-r--r--src/Common/Dlgcode.h2
-rw-r--r--src/Format/Tcformat.c4
5 files changed, 8 insertions, 8 deletions
diff --git a/src/Common/BootEncryption.cpp b/src/Common/BootEncryption.cpp
index c3aec66e..8aa40605 100644
--- a/src/Common/BootEncryption.cpp
+++ b/src/Common/BootEncryption.cpp
@@ -4568,8 +4568,8 @@ namespace VeraCrypt
}
}
- bool BootEncryption::RestartComputer (void)
+ bool BootEncryption::RestartComputer (BOOL bShutdown)
{
- return (::RestartComputer() != FALSE);
+ return (::RestartComputer(bShutdown) != FALSE);
}
}
diff --git a/src/Common/BootEncryption.h b/src/Common/BootEncryption.h
index 06da2bba..4bf31fec 100644
--- a/src/Common/BootEncryption.h
+++ b/src/Common/BootEncryption.h
@@ -276,7 +276,7 @@ namespace VeraCrypt
void RegisterSystemFavoritesService (BOOL registerService, BOOL noFileHandling);
void UpdateSystemFavoritesService ();
void RenameDeprecatedSystemLoaderBackup ();
- bool RestartComputer (void);
+ bool RestartComputer (BOOL bShutdown = FALSE);
void InitialSecurityChecksForHiddenOS ();
void RestrictPagingFilesToSystemPartition ();
void SetDriverConfigurationFlag (uint32 flag, bool state);
diff --git a/src/Common/Dlgcode.c b/src/Common/Dlgcode.c
index c2e15390..810b6e8a 100644
--- a/src/Common/Dlgcode.c
+++ b/src/Common/Dlgcode.c
@@ -10017,7 +10017,7 @@ BOOL EnableWow64FsRedirection (BOOL enable)
}
-BOOL RestartComputer (void)
+BOOL RestartComputer (BOOL bShutdown)
{
TOKEN_PRIVILEGES tokenPrivil;
HANDLE hTkn;
@@ -10038,7 +10038,7 @@ BOOL RestartComputer (void)
return false;
}
- if (!ExitWindowsEx (EWX_REBOOT,
+ if (!ExitWindowsEx (bShutdown? EWX_POWEROFF: EWX_REBOOT,
SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER | SHTDN_REASON_FLAG_PLANNED))
{
CloseHandle(hTkn);
diff --git a/src/Common/Dlgcode.h b/src/Common/Dlgcode.h
index 9ae9d5b2..a950ae22 100644
--- a/src/Common/Dlgcode.h
+++ b/src/Common/Dlgcode.h
@@ -443,7 +443,7 @@ BOOL Is64BitOs ();
BOOL IsServerOS ();
BOOL IsHiddenOSRunning (void);
BOOL EnableWow64FsRedirection (BOOL enable);
-BOOL RestartComputer (void);
+BOOL RestartComputer (BOOL bShutdown);
void Applink (char *dest, BOOL bSendOS, char *extraOutput);
wchar_t *RelativePath2Absolute (wchar_t *szFileName);
void HandleDriveNotReadyError (HWND hwnd);
diff --git a/src/Format/Tcformat.c b/src/Format/Tcformat.c
index 66ec3ac3..d83d9f49 100644
--- a/src/Format/Tcformat.c
+++ b/src/Format/Tcformat.c
@@ -8157,7 +8157,7 @@ retryCDDriveCheck:
try
{
- BootEncObj->RestartComputer ();
+ BootEncObj->RestartComputer (bSystemIsGPT);
}
catch (Exception &e)
{
@@ -10112,7 +10112,7 @@ static void AfterWMInitTasks (HWND hwndDlg)
try
{
- BootEncObj->RestartComputer ();
+ BootEncObj->RestartComputer (bSystemIsGPT);
}
catch (Exception &e)
{