--- /dev/null
+/*
+ * COPYRIGHT: See COPYING in the top level directory
+ * PROJECT: ReactOS Winlogon
+ * FILE: base/system/winlogon/notify.c
+ * PURPOSE: Logon notifications
+ * PROGRAMMERS: Eric Kohl
+ */
+
+/* INCLUDES *****************************************************************/
+
+#include "winlogon.h"
+
+/* GLOBALS ******************************************************************/
+
+
+// void Event_Handler_Function_Name(PWLX_NOTIFICATION_INFO pInfo);
+typedef VOID (WINAPI *PWLX_NOTIFY_HANDLER)(PWLX_NOTIFICATION_INFO pInfo);
+
+static PSTR FuncNames[LastHandler] =
+{
+ "Logon",
+ "Logoff",
+ "Lock",
+ "Unlock",
+ "Startup",
+ "Shutdown",
+ "StartScreenSaver",
+ "StopScreenSaver",
+ "Disconnect",
+ "Reconnect",
+ "StartShell",
+ "PostShell"
+};
+
+typedef struct _NOTIFICATION_ITEM
+{
+ LIST_ENTRY ListEntry;
+
+ HINSTANCE hInstance;
+ BOOL bEnabled;
+ BOOL bAsynchronous;
+ BOOL bSafe;
+ BOOL bImpersonate;
+ BOOL bSmartCardLogon;
+ DWORD dwMaxWait;
+ PWLX_NOTIFY_HANDLER Handler[LastHandler];
+} NOTIFICATION_ITEM, *PNOTIFICATION_ITEM;
+
+
+static LIST_ENTRY NotificationDllListHead;
+
+
+/* FUNCTIONS *****************************************************************/
+
+PWLX_NOTIFY_HANDLER
+GetNotificationHandler(
+ HKEY hDllKey,
+ HINSTANCE hInstance,
+ PSTR pNotification)
+{
+ CHAR szFuncBuffer[128];
+ DWORD dwSize;
+ DWORD dwType;
+ LONG lError;
+
+ dwSize = 128;
+ lError = RegQueryValueExA(hDllKey,
+ pNotification,
+ NULL,
+ &dwType,
+ (PBYTE)szFuncBuffer,
+ &dwSize);
+ if (lError == ERROR_SUCCESS)
+ {
+ return (PWLX_NOTIFY_HANDLER)GetProcAddress(hInstance, szFuncBuffer);
+ }
+
+ return NULL;
+}
+
+
+static
+VOID
+LoadNotificationDll(
+ HKEY hNotifyKey,
+ PWSTR pKeyName)
+{
+ HKEY hDllKey = NULL;
+ PNOTIFICATION_ITEM NotificationDll = NULL;
+ LONG lError;
+ WCHAR szBuffer[80];
+ DWORD dwSize;
+ DWORD dwType;
+ HINSTANCE hInstance;
+ NOTIFICATION_TYPE i;
+
+ TRACE("LoadNotificationDll(%p %S)\n", hNotifyKey, pKeyName);
+
+ lError = RegOpenKeyExW(hNotifyKey,
+ pKeyName,
+ 0,
+ KEY_READ,
+ &hDllKey);
+ if (lError != ERROR_SUCCESS)
+ return;
+
+ dwSize = 80 * sizeof(WCHAR);
+ lError = RegQueryValueExW(hDllKey,
+ L"DllName",
+ NULL,
+ &dwType,
+ (PBYTE)szBuffer,
+ &dwSize);
+ if (lError == ERROR_SUCCESS)
+ {
+ hInstance = LoadLibraryW(szBuffer);
+ if (hInstance == NULL)
+ return;
+
+ NotificationDll = RtlAllocateHeap(RtlGetProcessHeap(),
+ HEAP_ZERO_MEMORY,
+ sizeof(NOTIFICATION_ITEM));
+ if (NotificationDll == NULL)
+ {
+ FreeLibrary(hInstance);
+ return;
+ }
+
+ NotificationDll->bEnabled = TRUE;
+ NotificationDll->dwMaxWait = 30; /* FIXME: ??? */
+ NotificationDll->hInstance = hInstance;
+
+ dwSize = sizeof(BOOL);
+ RegQueryValueExW(hDllKey,
+ L"Asynchronous",
+ NULL,
+ &dwType,
+ (PBYTE)&NotificationDll->bAsynchronous,
+ &dwSize);
+
+ dwSize = sizeof(BOOL);
+ RegQueryValueExW(hDllKey,
+ L"Enabled",
+ NULL,
+ &dwType,
+ (PBYTE)&NotificationDll->bEnabled,
+ &dwSize);
+
+ dwSize = sizeof(BOOL);
+ RegQueryValueExW(hDllKey,
+ L"Impersonate",
+ NULL,
+ &dwType,
+ (PBYTE)&NotificationDll->bImpersonate,
+ &dwSize);
+
+ dwSize = sizeof(BOOL);
+ RegQueryValueExW(hDllKey,
+ L"Safe",
+ NULL,
+ &dwType,
+ (PBYTE)&NotificationDll->bSafe,
+ &dwSize);
+
+ dwSize = sizeof(BOOL);
+ RegQueryValueExW(hDllKey,
+ L"SmartCardLogonNotify",
+ NULL,
+ &dwType,
+ (PBYTE)&NotificationDll->bSmartCardLogon,
+ &dwSize);
+
+ dwSize = sizeof(DWORD);
+ RegQueryValueExW(hDllKey,
+ L"MaxWait",
+ NULL,
+ &dwType,
+ (PBYTE)&NotificationDll->dwMaxWait,
+ &dwSize);
+
+ for (i = LogonHandler; i < LastHandler; i++)
+ {
+ NotificationDll->Handler[i] = GetNotificationHandler(hDllKey, hInstance, FuncNames[i]);
+ TRACE("%s: %p\n", FuncNames[i], NotificationDll->Handler[i]);
+ }
+
+ InsertHeadList(&NotificationDllListHead,
+ &NotificationDll->ListEntry);
+ }
+
+ RegCloseKey(hDllKey);
+}
+
+
+BOOL
+InitNotifications(VOID)
+{
+ HKEY hNotifyKey = NULL;
+ LONG lError;
+ DWORD dwIndex;
+ WCHAR szKeyName[80];
+ DWORD dwKeyName;
+
+ TRACE("InitNotifications()\n");
+
+ InitializeListHead(&NotificationDllListHead);
+
+ lError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
+ L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon\\Notify",
+ 0,
+ KEY_READ | KEY_ENUMERATE_SUB_KEYS,
+ &hNotifyKey);
+ if (lError != ERROR_SUCCESS)
+ {
+ TRACE("RegOpenKeyExW()\n");
+ return TRUE;
+ }
+
+ dwIndex = 0;
+ for(;;)
+ {
+ dwKeyName = 80 * sizeof(WCHAR);
+ lError = RegEnumKeyExW(hNotifyKey,
+ dwIndex,
+ szKeyName,
+ &dwKeyName,
+ NULL,
+ NULL,
+ NULL,
+ NULL);
+ if (lError != ERROR_SUCCESS)
+ break;
+
+ TRACE("Notification DLL: %S\n", szKeyName);
+ LoadNotificationDll(hNotifyKey, szKeyName);
+
+ dwIndex++;
+ }
+
+ RegCloseKey(hNotifyKey);
+
+ TRACE("InitNotifications() done\n");
+
+ return TRUE;
+}
+
+
+VOID
+CallNotificationDlls(
+ PWLSESSION pSession,
+ NOTIFICATION_TYPE Type)
+{
+ PLIST_ENTRY ListEntry;
+ PNOTIFICATION_ITEM NotificationDll;
+ WLX_NOTIFICATION_INFO Info;
+
+ TRACE("CallNotificationDlls()\n");
+
+ Info.Size = sizeof(WLX_NOTIFICATION_INFO);
+
+ switch (Type)
+ {
+ case LogoffHandler:
+ case ShutdownHandler:
+ Info.Flags = 3;
+ break;
+
+ default:
+ Info.Flags = 0;
+ break;
+ }
+
+ Info.UserName = NULL; //UserName;
+ Info.Domain = NULL; //Domain;
+ Info.WindowStation = pSession->InteractiveWindowStationName;
+ Info.hToken = pSession->UserToken;
+
+ switch (Type)
+ {
+ case LogonHandler:
+ case StartShellHandler:
+ Info.hDesktop = pSession->ApplicationDesktop;
+ break;
+
+ case StartScreenSaverHandler:
+ Info.hDesktop = pSession->ApplicationDesktop;
+ break;
+
+ default:
+ Info.hDesktop = pSession->WinlogonDesktop;
+ break;
+ }
+
+ Info.pStatusCallback = NULL;
+
+ ListEntry = NotificationDllListHead.Flink;
+ while (ListEntry != &NotificationDllListHead)
+ {
+TRACE("ListEntry %p\n", ListEntry);
+
+ NotificationDll = CONTAINING_RECORD(ListEntry,
+ NOTIFICATION_ITEM,
+ ListEntry);
+TRACE("NotificationDll: %p\n", NotificationDll);
+ if (NotificationDll != NULL && NotificationDll->bEnabled)
+ {
+TRACE("NotificationDll->Handler: %p\n", NotificationDll->Handler[Type]);
+ if (NotificationDll->Handler[Type] != NULL)
+ NotificationDll->Handler[Type](&Info);
+ }
+
+ ListEntry = ListEntry->Flink;
+ }
+}
+
+
+VOID
+CleanupNotifications(VOID)
+{
+ PLIST_ENTRY ListEntry;
+ PNOTIFICATION_ITEM NotificationDll;
+
+ ListEntry = NotificationDllListHead.Flink;
+ while (ListEntry != &NotificationDllListHead)
+ {
+ NotificationDll = CONTAINING_RECORD(ListEntry,
+ NOTIFICATION_ITEM,
+ ListEntry);
+ if (NotificationDll != NULL)
+ {
+ FreeLibrary(NotificationDll->hInstance);
+ }
+
+ ListEntry = ListEntry->Flink;
+
+ RemoveEntryList(&NotificationDll->ListEntry);
+
+ RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll);
+ }
+}
+
+/* EOF */
projects / reactos.git / blobdiff