[reactos] 02/02: [WINLOGON] Implement logoff and MessageBeep sounds support (#4755) CORE-13951

[Thamatip Chitpong]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=9d48f26d2992af33c0fbeb...

commit 9d48f26d2992af33c0fbeb9ad51bf776cf334ccc
Author:     Thamatip Chitpong [e-mail]
AuthorDate: Wed Oct 5 08:33:28 2022 +0700
Commit:     Hermès Bélusca-Maïto [e-mail]
CommitDate: Fri May 5 20:18:34 2023 +0200

[WINLOGON] Implement logoff and MessageBeep sounds support (#4755)
    CORE-13951
---
 base/system/winlogon/sas.c | 186 ++++++++++++++++++---------------------------
 1 file changed, 75 insertions(+), 111 deletions(-)

diff --git a/base/system/winlogon/sas.c b/base/system/winlogon/sas.c
index bc37d5db49d..b94663b1c55 100644
--- a/base/system/winlogon/sas.c
+++ b/base/system/winlogon/sas.c
@@ -4,7 +4,7 @@
  * FILE:            base/system/winlogon/sas.c
  * PURPOSE:         Secure Attention Sequence
  * PROGRAMMERS:     Thomas Weidenmueller ([e-mail])
- *                  Herv� Poussineau ([e-mail])
+ *                  Hervé Poussineau ([e-mail])
  *                  Arnav Bhatt ([e-mail])
  * UPDATE HISTORY:
  *                  Created 28/03/2004
@@ -290,88 +290,11 @@ WINAPI
 PlayLogonSoundThread(
     IN LPVOID lpParameter)
 {
-    BYTE TokenUserBuffer[256];
-    PTOKEN_USER pTokenUser = (TOKEN_USER*)TokenUserBuffer;
-    ULONG Length;
-    HKEY hKey;
-    WCHAR wszBuffer[MAX_PATH] = {0};
-    WCHAR wszDest[MAX_PATH];
-    DWORD dwSize = sizeof(wszBuffer), dwType;
     SERVICE_STATUS_PROCESS Info;
-    UNICODE_STRING SidString;
-    NTSTATUS Status;
+    DWORD dwSize;
     ULONG Index = 0;
     SC_HANDLE hSCManager, hService;

-    //
-    // FIXME: Isn't it possible to *JUST* impersonate the current user
-    // *AND* open its HKCU??
-    //
-
-    /* Get SID of current user */
-    Status = NtQueryInformationToken((HANDLE)lpParameter,
-                                     TokenUser,
-                                     TokenUserBuffer,
-                                     sizeof(TokenUserBuffer),
-                                     &Length);
-    if (!NT_SUCCESS(Status))
-    {
-        ERR("NtQueryInformationToken failed: %x!\n", Status);
-        return 0;
-    }
-
-    /* Convert SID to string */
-    RtlInitEmptyUnicodeString(&SidString, wszBuffer, sizeof(wszBuffer));
-    Status = RtlConvertSidToUnicodeString(&SidString, pTokenUser->User.Sid, FALSE);
-    if (!NT_SUCCESS(Status))
-    {
-        ERR("RtlConvertSidToUnicodeString failed: %x!\n", Status);
-        return 0;
-    }
-
-    /* Build path to logon sound registry key.
-       Note: We can't use HKCU here, because Winlogon is owned by SYSTEM user */
-    if (FAILED(StringCbCopyW(wszBuffer + SidString.Length/sizeof(WCHAR),
-                             sizeof(wszBuffer) - SidString.Length,
-                             L"\AppEvents\Schemes\Apps\.Default\WindowsLogon\.Current")))
-    {
-        /* SID is too long. Should not happen. */
-        ERR("StringCbCopyW failed!\n");
-        return 0;
-    }
-
-    /* Open registry key and query sound path */
-    if (RegOpenKeyExW(HKEY_USERS, wszBuffer, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
-    {
-        ERR("RegOpenKeyExW(%ls) failed!\n", wszBuffer);
-        return 0;
-    }
-
-    if (RegQueryValueExW(hKey, NULL, NULL, &dwType,
-                      (LPBYTE)wszBuffer, &dwSize) != ERROR_SUCCESS ||
-        (dwType != REG_SZ && dwType != REG_EXPAND_SZ))
-    {
-        ERR("RegQueryValueExW failed!\n");
-        RegCloseKey(hKey);
-        return 0;
-    }
-
-    RegCloseKey(hKey);
-
-    if (!wszBuffer[0])
-    {
-        /* No sound has been set */
-        ERR("No sound has been set\n");
-        return 0;
-    }
-
-    /* Expand environment variables */
-    if (!ExpandEnvironmentStringsW(wszBuffer, wszDest, MAX_PATH))
-    {
-        ERR("ExpandEnvironmentStringsW failed!\n");
-        return 0;
-    }
-
     /* Open the service manager */
     hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CONNECT);
     if (!hSCManager)
@@ -417,8 +340,16 @@ PlayLogonSoundThread(
     }

/* Sound subsystem is running. Play logon sound. */
-    TRACE("Playing logon sound: %ls\n", wszDest);
-    PlaySoundRoutine(wszDest, TRUE, SND_FILENAME);
+    TRACE("Playing logon sound\n");
+    if (!ImpersonateLoggedOnUser((HANDLE)lpParameter))
+    {
+        ERR("ImpersonateLoggedOnUser failed (%x)\n", GetLastError());
+    }
+    else
+    {
+        PlaySoundRoutine(L"WindowsLogon", TRUE, SND_ALIAS | SND_NODEFAULT);
+        RevertToSelf();
+    }
     return 0;
 }

@@ -434,6 +365,37 @@ PlayLogonSound(
         CloseHandle(hThread);
 }

+static
+VOID
+PlayLogoffSound(
+    _In_ PWLSESSION Session)
+{
+    if (!ImpersonateLoggedOnUser(Session->UserToken))
+        return;
+
+    PlaySoundRoutine(L"WindowsLogoff", FALSE, SND_ALIAS | SND_NODEFAULT);
+
+    RevertToSelf();
+}
+
+static
+BOOL
+PlayEventSound(
+    _In_ PWLSESSION Session,
+    _In_ LPCWSTR EventName)
+{
+    BOOL bRet;
+
+    if (!ImpersonateLoggedOnUser(Session->UserToken))
+        return FALSE;
+
+    bRet = PlaySoundRoutine(EventName, FALSE, SND_ALIAS | SND_ASYNC | SND_NODEFAULT);
+
+    RevertToSelf();
+
+    return bRet;
+}
+
 static
 VOID
 RestoreAllConnections(PWLSESSION Session)
@@ -880,7 +842,7 @@ HandleLogoff(

SwitchDesktop(Session->WinlogonDesktop);

-    // TODO: Play logoff sound!
+    PlayLogoffSound(Session);

SetWindowStationUser(Session->InteractiveWindowStation,
                          &LuidNone, NULL, 0);
@@ -1274,38 +1236,40 @@ UnregisterHotKeys(
     return TRUE;
 }

+static
 BOOL
-WINAPI
-HandleMessageBeep(UINT uType)
+HandleMessageBeep(
+    _In_ PWLSESSION Session,
+    _In_ UINT uType)
 {
     LPWSTR EventName;

-    switch(uType)
-    {
-    case 0xFFFFFFFF:
-        EventName = NULL;
-        break;
-    case MB_OK:
-        EventName = L"SystemDefault";
-        break;
-    case MB_ICONASTERISK:
-        EventName = L"SystemAsterisk";
-        break;
-    case MB_ICONEXCLAMATION:
-        EventName = L"SystemExclamation";
-        break;
-    case MB_ICONHAND:
-        EventName = L"SystemHand";
-        break;
-    case MB_ICONQUESTION:
-        EventName = L"SystemQuestion";
-        break;
-    default:
-        WARN("Unhandled type %d\n", uType);
-        EventName = L"SystemDefault";
-    }
-
-    return PlaySoundRoutine(EventName, FALSE, SND_ALIAS | SND_NOWAIT | SND_NOSTOP | SND_ASYNC);
+    switch (uType)
+    {
+        case 0xFFFFFFFF:
+            EventName = NULL;
+            break;
+        case MB_OK:
+            EventName = L"SystemDefault";
+            break;
+        case MB_ICONASTERISK:
+            EventName = L"SystemAsterisk";
+            break;
+        case MB_ICONEXCLAMATION:
+            EventName = L"SystemExclamation";
+            break;
+        case MB_ICONHAND:
+            EventName = L"SystemHand";
+            break;
+        case MB_ICONQUESTION:
+            EventName = L"SystemQuestion";
+            break;
+        default:
+            WARN("Unhandled type %d\n", uType);
+            EventName = L"SystemDefault";
+    }
+
+    return PlayEventSound(Session, EventName);
 }

static
@@ -1376,7 +1340,7 @@ SASWindowProc(
             {
                 case LN_MESSAGE_BEEP:
                 {
-                    return HandleMessageBeep(lParam);
+                    return HandleMessageBeep(Session, lParam);
                 }
                 case LN_SHELL_EXITED:
                 {