mirror of
https://github.com/reactos/reactos.git
synced 2025-08-03 13:45:50 +00:00
[WINLOGON] Load the notification dlls on demand
Do not keep the notifications loaded until shutdown. Load and unload them when a notification routine must be called.
This commit is contained in:
Eric Kohl
parent
47303268e1
commit
5d4f69bf35
1 changed files with 200 additions and 130 deletions
|
|
@ -35,15 +35,14 @@ static PSTR FuncNames[LastHandler] =
|
||||||
typedef struct _NOTIFICATION_ITEM
|
typedef struct _NOTIFICATION_ITEM
|
||||||
{
|
{
|
||||||
LIST_ENTRY ListEntry;
|
LIST_ENTRY ListEntry;
|
||||||
|
PWSTR pszKeyName;
|
||||||
HINSTANCE hInstance;
|
PWSTR pszDllName;
|
||||||
BOOL bEnabled;
|
BOOL bEnabled;
|
||||||
BOOL bAsynchronous;
|
BOOL bAsynchronous;
|
||||||
BOOL bSafe;
|
BOOL bSafe;
|
||||||
BOOL bImpersonate;
|
BOOL bImpersonate;
|
||||||
BOOL bSmartCardLogon;
|
BOOL bSmartCardLogon;
|
||||||
DWORD dwMaxWait;
|
DWORD dwMaxWait;
|
||||||
PWLX_NOTIFY_HANDLER Handler[LastHandler];
|
|
||||||
} NOTIFICATION_ITEM, *PNOTIFICATION_ITEM;
|
} NOTIFICATION_ITEM, *PNOTIFICATION_ITEM;
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -52,140 +51,145 @@ static LIST_ENTRY NotificationDllListHead;
|
||||||
|
|
||||||
/* FUNCTIONS *****************************************************************/
|
/* 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
|
static
|
||||||
VOID
|
VOID
|
||||||
LoadNotificationDll(
|
AddNotificationDll(
|
||||||
HKEY hNotifyKey,
|
HKEY hNotifyKey,
|
||||||
PWSTR pKeyName)
|
PWSTR pszKeyName)
|
||||||
{
|
{
|
||||||
HKEY hDllKey = NULL;
|
HKEY hDllKey = NULL;
|
||||||
PNOTIFICATION_ITEM NotificationDll = NULL;
|
PNOTIFICATION_ITEM NotificationDll = NULL;
|
||||||
LONG lError;
|
DWORD dwSize, dwType;
|
||||||
WCHAR szBuffer[80];
|
DWORD dwError;
|
||||||
DWORD dwSize;
|
|
||||||
DWORD dwType;
|
|
||||||
HINSTANCE hInstance;
|
|
||||||
NOTIFICATION_TYPE i;
|
|
||||||
|
|
||||||
TRACE("LoadNotificationDll(%p %S)\n", hNotifyKey, pKeyName);
|
TRACE("AddNotificationDll(%p %S)\n", hNotifyKey, pszKeyName);
|
||||||
|
|
||||||
lError = RegOpenKeyExW(hNotifyKey,
|
dwError = RegOpenKeyExW(hNotifyKey,
|
||||||
pKeyName,
|
pszKeyName,
|
||||||
0,
|
0,
|
||||||
KEY_READ,
|
KEY_READ,
|
||||||
&hDllKey);
|
&hDllKey);
|
||||||
if (lError != ERROR_SUCCESS)
|
if (dwError != ERROR_SUCCESS)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
dwSize = 80 * sizeof(WCHAR);
|
NotificationDll = RtlAllocateHeap(RtlGetProcessHeap(),
|
||||||
lError = RegQueryValueExW(hDllKey,
|
HEAP_ZERO_MEMORY,
|
||||||
L"DllName",
|
sizeof(NOTIFICATION_ITEM));
|
||||||
NULL,
|
if (NotificationDll == NULL)
|
||||||
&dwType,
|
|
||||||
(PBYTE)szBuffer,
|
|
||||||
&dwSize);
|
|
||||||
if (lError == ERROR_SUCCESS)
|
|
||||||
{
|
{
|
||||||
hInstance = LoadLibraryW(szBuffer);
|
dwError = ERROR_OUTOFMEMORY;
|
||||||
if (hInstance == NULL)
|
goto done;
|
||||||
return;
|
}
|
||||||
|
|
||||||
NotificationDll = RtlAllocateHeap(RtlGetProcessHeap(),
|
NotificationDll->pszKeyName = RtlAllocateHeap(RtlGetProcessHeap(),
|
||||||
HEAP_ZERO_MEMORY,
|
HEAP_ZERO_MEMORY,
|
||||||
sizeof(NOTIFICATION_ITEM));
|
(wcslen(pszKeyName) + 1) * sizeof(WCHAR));
|
||||||
if (NotificationDll == NULL)
|
if (NotificationDll->pszKeyName == NULL)
|
||||||
|
{
|
||||||
|
dwError = ERROR_OUTOFMEMORY;
|
||||||
|
goto done;
|
||||||
|
}
|
||||||
|
|
||||||
|
wcscpy(NotificationDll->pszKeyName, pszKeyName);
|
||||||
|
|
||||||
|
dwSize = 0;
|
||||||
|
RegQueryValueExW(hDllKey,
|
||||||
|
L"DllName",
|
||||||
|
NULL,
|
||||||
|
&dwType,
|
||||||
|
NULL,
|
||||||
|
&dwSize);
|
||||||
|
if (dwSize == 0)
|
||||||
|
{
|
||||||
|
dwError = ERROR_FILE_NOT_FOUND;
|
||||||
|
goto done;
|
||||||
|
}
|
||||||
|
|
||||||
|
NotificationDll->pszDllName = RtlAllocateHeap(RtlGetProcessHeap(),
|
||||||
|
HEAP_ZERO_MEMORY,
|
||||||
|
dwSize);
|
||||||
|
if (NotificationDll->pszDllName == NULL)
|
||||||
|
{
|
||||||
|
dwError = ERROR_OUTOFMEMORY;
|
||||||
|
goto done;
|
||||||
|
}
|
||||||
|
|
||||||
|
dwError = RegQueryValueExW(hDllKey,
|
||||||
|
L"DllName",
|
||||||
|
NULL,
|
||||||
|
&dwType,
|
||||||
|
(PBYTE)NotificationDll->pszDllName,
|
||||||
|
&dwSize);
|
||||||
|
if (dwError != ERROR_SUCCESS)
|
||||||
|
goto done;
|
||||||
|
|
||||||
|
NotificationDll->bEnabled = TRUE;
|
||||||
|
NotificationDll->dwMaxWait = 30; /* FIXME: ??? */
|
||||||
|
|
||||||
|
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);
|
||||||
|
|
||||||
|
InsertHeadList(&NotificationDllListHead,
|
||||||
|
&NotificationDll->ListEntry);
|
||||||
|
|
||||||
|
done:
|
||||||
|
if (dwError != ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
if (NotificationDll != NULL)
|
||||||
{
|
{
|
||||||
FreeLibrary(hInstance);
|
if (NotificationDll->pszKeyName != NULL)
|
||||||
return;
|
RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszKeyName);
|
||||||
|
|
||||||
|
if (NotificationDll->pszDllName != NULL)
|
||||||
|
RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszDllName);
|
||||||
|
|
||||||
|
RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll);
|
||||||
}
|
}
|
||||||
|
|
||||||
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);
|
RegCloseKey(hDllKey);
|
||||||
|
|
@ -232,7 +236,7 @@ InitNotifications(VOID)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
TRACE("Notification DLL: %S\n", szKeyName);
|
TRACE("Notification DLL: %S\n", szKeyName);
|
||||||
LoadNotificationDll(hNotifyKey, szKeyName);
|
AddNotificationDll(hNotifyKey, szKeyName);
|
||||||
|
|
||||||
dwIndex++;
|
dwIndex++;
|
||||||
}
|
}
|
||||||
|
|
@ -245,6 +249,57 @@ InitNotifications(VOID)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static
|
||||||
|
VOID
|
||||||
|
CallNotificationDll(
|
||||||
|
HKEY hNotifyKey,
|
||||||
|
PNOTIFICATION_ITEM NotificationDll,
|
||||||
|
NOTIFICATION_TYPE Type,
|
||||||
|
PWLX_NOTIFICATION_INFO pInfo)
|
||||||
|
{
|
||||||
|
HKEY hDllKey = NULL;
|
||||||
|
HMODULE hModule = NULL;
|
||||||
|
CHAR szFuncBuffer[128];
|
||||||
|
DWORD dwSize;
|
||||||
|
DWORD dwType;
|
||||||
|
DWORD dwError;
|
||||||
|
PWLX_NOTIFY_HANDLER pNotifyHandler;
|
||||||
|
|
||||||
|
dwError = RegOpenKeyExW(hNotifyKey,
|
||||||
|
NotificationDll->pszKeyName,
|
||||||
|
0,
|
||||||
|
KEY_READ,
|
||||||
|
&hDllKey);
|
||||||
|
if (dwError != ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
TRACE("RegOpenKeyExW()\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
dwSize = sizeof(szFuncBuffer);
|
||||||
|
dwError = RegQueryValueExA(hDllKey,
|
||||||
|
FuncNames[Type],
|
||||||
|
NULL,
|
||||||
|
&dwType,
|
||||||
|
(PBYTE)szFuncBuffer,
|
||||||
|
&dwSize);
|
||||||
|
if (dwError == ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
hModule = LoadLibraryW(NotificationDll->pszDllName);
|
||||||
|
if (hModule != NULL)
|
||||||
|
{
|
||||||
|
pNotifyHandler = (PWLX_NOTIFY_HANDLER)GetProcAddress(hModule, szFuncBuffer);
|
||||||
|
if (pNotifyHandler != NULL)
|
||||||
|
pNotifyHandler(pInfo);
|
||||||
|
|
||||||
|
FreeLibrary(hModule);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
RegCloseKey(hDllKey);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
VOID
|
VOID
|
||||||
CallNotificationDlls(
|
CallNotificationDlls(
|
||||||
PWLSESSION pSession,
|
PWLSESSION pSession,
|
||||||
|
|
@ -253,9 +308,22 @@ CallNotificationDlls(
|
||||||
PLIST_ENTRY ListEntry;
|
PLIST_ENTRY ListEntry;
|
||||||
PNOTIFICATION_ITEM NotificationDll;
|
PNOTIFICATION_ITEM NotificationDll;
|
||||||
WLX_NOTIFICATION_INFO Info;
|
WLX_NOTIFICATION_INFO Info;
|
||||||
|
HKEY hNotifyKey = NULL;
|
||||||
|
DWORD dwError;
|
||||||
|
|
||||||
TRACE("CallNotificationDlls()\n");
|
TRACE("CallNotificationDlls()\n");
|
||||||
|
|
||||||
|
dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
|
||||||
|
L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon\\Notify",
|
||||||
|
0,
|
||||||
|
KEY_READ | KEY_ENUMERATE_SUB_KEYS,
|
||||||
|
&hNotifyKey);
|
||||||
|
if (dwError != ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
TRACE("RegOpenKeyExW()\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
Info.Size = sizeof(WLX_NOTIFICATION_INFO);
|
Info.Size = sizeof(WLX_NOTIFICATION_INFO);
|
||||||
|
|
||||||
switch (Type)
|
switch (Type)
|
||||||
|
|
@ -303,14 +371,12 @@ TRACE("ListEntry %p\n", ListEntry);
|
||||||
ListEntry);
|
ListEntry);
|
||||||
TRACE("NotificationDll: %p\n", NotificationDll);
|
TRACE("NotificationDll: %p\n", NotificationDll);
|
||||||
if (NotificationDll != NULL && NotificationDll->bEnabled)
|
if (NotificationDll != NULL && NotificationDll->bEnabled)
|
||||||
{
|
CallNotificationDll(hNotifyKey, NotificationDll, Type, &Info);
|
||||||
TRACE("NotificationDll->Handler: %p\n", NotificationDll->Handler[Type]);
|
|
||||||
if (NotificationDll->Handler[Type] != NULL)
|
|
||||||
NotificationDll->Handler[Type](&Info);
|
|
||||||
}
|
|
||||||
|
|
||||||
ListEntry = ListEntry->Flink;
|
ListEntry = ListEntry->Flink;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
RegCloseKey(hNotifyKey);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -328,7 +394,11 @@ CleanupNotifications(VOID)
|
||||||
ListEntry);
|
ListEntry);
|
||||||
if (NotificationDll != NULL)
|
if (NotificationDll != NULL)
|
||||||
{
|
{
|
||||||
FreeLibrary(NotificationDll->hInstance);
|
if (NotificationDll->pszKeyName != NULL)
|
||||||
|
RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszKeyName);
|
||||||
|
|
||||||
|
if (NotificationDll->pszDllName != NULL)
|
||||||
|
RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszDllName);
|
||||||
}
|
}
|
||||||
|
|
||||||
ListEntry = ListEntry->Flink;
|
ListEntry = ListEntry->Flink;
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue