notify.c File Reference

#include "winlogon.h"

Include dependency graph for notify.c:

Go to the source code of this file.

Classes
struct	_NOTIFICATION_ITEM

Typedefs
typedef VOID(WINAPI *	PWLX_NOTIFY_HANDLER) (PWLX_NOTIFICATION_INFO pInfo)
typedef struct _NOTIFICATION_ITEM	NOTIFICATION_ITEM
typedef struct _NOTIFICATION_ITEM *	PNOTIFICATION_ITEM

Functions
static VOID	AddSfcNotification (VOID)
static VOID	AddNotificationDll (HKEY hNotifyKey, PWSTR pszKeyName)
BOOL	InitNotifications (VOID)
static VOID	CallNotificationDll (HKEY hNotifyKey, PNOTIFICATION_ITEM NotificationDll, NOTIFICATION_TYPE Type, PWLX_NOTIFICATION_INFO pInfo)
VOID	CallNotificationDlls (PWLSESSION pSession, NOTIFICATION_TYPE Type)
VOID	CleanupNotifications (VOID)

Variables
static PSTR	FuncNames [LastHandler]
static LIST_ENTRY	NotificationDllListHead

Typedef Documentation

NOTIFICATION_ITEM

typedef struct _NOTIFICATION_ITEM NOTIFICATION_ITEM

PNOTIFICATION_ITEM

typedef struct _NOTIFICATION_ITEM * PNOTIFICATION_ITEM

PWLX_NOTIFY_HANDLER

typedef VOID(WINAPI * PWLX_NOTIFY_HANDLER) (PWLX_NOTIFICATION_INFO pInfo)

Definition at line 17 of file notify.c.

Function Documentation

AddNotificationDll()

static VOID AddNotificationDll ( HKEY  hNotifyKey, PWSTR  pszKeyName  )

static

Definition at line 109 of file notify.c.

{
    HKEY hDllKey = NULL;
    PNOTIFICATION_ITEM NotificationDll = NULL;
    DWORD dwSize, dwType;
    DWORD dwError;
 
    TRACE("AddNotificationDll(%p %S)\n", hNotifyKey, pszKeyName);
 
    dwError = RegOpenKeyExW(hNotifyKey,
                            pszKeyName,
                            0,
                            KEY_READ,
                            &hDllKey);
    if (dwError != ERROR_SUCCESS)
        return;
 
    NotificationDll = RtlAllocateHeap(RtlGetProcessHeap(),
                                      HEAP_ZERO_MEMORY,
                                      sizeof(NOTIFICATION_ITEM));
    if (NotificationDll == NULL)
    {
        dwError = ERROR_OUTOFMEMORY;
        goto done;
    }
 
    NotificationDll->pszKeyName = RtlAllocateHeap(RtlGetProcessHeap(),
                                                  HEAP_ZERO_MEMORY,
                                                  (wcslen(pszKeyName) + 1) * sizeof(WCHAR));
    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)
        {
            if (NotificationDll->pszKeyName != NULL)
                RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszKeyName);
 
            if (NotificationDll->pszDllName != NULL)
                RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszDllName);
 
            RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll);
        }
    }
 
    RegCloseKey(hDllKey);
}

Referenced by InitNotifications().

AddSfcNotification()

static VOID AddSfcNotification ( VOID  )

static

Definition at line 57 of file notify.c.

{
    WCHAR szSfcFileName[MAX_PATH];
    PNOTIFICATION_ITEM NotificationDll;
    DWORD dwError = ERROR_SUCCESS;
 
    ExpandEnvironmentStringsW(L"%windir%\\system32\\sfc.dll",
                              szSfcFileName,
                              ARRAYSIZE(szSfcFileName));
 
    NotificationDll = RtlAllocateHeap(RtlGetProcessHeap(),
                                      HEAP_ZERO_MEMORY,
                                      sizeof(NOTIFICATION_ITEM));
    if (NotificationDll == NULL)
    {
        dwError = ERROR_OUTOFMEMORY;
        goto done;
    }
 
    NotificationDll->pszDllName = RtlAllocateHeap(RtlGetProcessHeap(),
                                                  HEAP_ZERO_MEMORY,
                                                  (wcslen(szSfcFileName) + 1) * sizeof(WCHAR));
    if (NotificationDll->pszDllName == NULL)
    {
        dwError = ERROR_OUTOFMEMORY;
        goto done;
    }
 
    wcscpy(NotificationDll->pszDllName, szSfcFileName);
 
    NotificationDll->bEnabled = TRUE;
    NotificationDll->dwMaxWait = 30; /* FIXME: ??? */
    NotificationDll->bSfcNotification = TRUE;
 
    InsertHeadList(&NotificationDllListHead,
                   &NotificationDll->ListEntry);
 
done:
    if (dwError != ERROR_SUCCESS)
    {
        if (NotificationDll != NULL)
        {
            if (NotificationDll->pszDllName != NULL)
                RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszDllName);
 
            RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll);
        }
    }
}

Referenced by InitNotifications().

CallNotificationDll()

static VOID CallNotificationDll ( HKEY  hNotifyKey, PNOTIFICATION_ITEM  NotificationDll, NOTIFICATION_TYPE  Type, PWLX_NOTIFICATION_INFO  pInfo  )

static

Definition at line 309 of file notify.c.

{
    HMODULE hModule;
    CHAR szFuncBuffer[128];
    DWORD dwError = ERROR_SUCCESS;
    PWLX_NOTIFY_HANDLER pNotifyHandler;
 
    if (NotificationDll->bSfcNotification)
    {
        switch (Type)
        {
            case LogonHandler:
                strcpy(szFuncBuffer, "SfcWLEventLogon");
                break;
 
            case LogoffHandler:
                strcpy(szFuncBuffer, "SfcWLEventLogoff");
                break;
 
            default:
                return;
        }
    }
    else
    {
        HKEY hDllKey;
        DWORD dwSize;
        DWORD dwType;
 
        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);
 
        RegCloseKey(hDllKey);
    }
 
    if (dwError != ERROR_SUCCESS)
        return;
 
    hModule = LoadLibraryW(NotificationDll->pszDllName);
    if (!hModule)
        return;
 
    pNotifyHandler = (PWLX_NOTIFY_HANDLER)GetProcAddress(hModule, szFuncBuffer);
 
    _SEH2_TRY
    {
        if (pNotifyHandler)
            pNotifyHandler(pInfo);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        ERR("WL: Exception while running notification %S!%s, Status 0x%08lx\n",
            NotificationDll->pszDllName, szFuncBuffer, _SEH2_GetExceptionCode());
    }
    _SEH2_END;
 
    FreeLibrary(hModule);
}

Referenced by CallNotificationDlls().

CallNotificationDlls()

VOID CallNotificationDlls	(	PWLSESSION	pSession,
		NOTIFICATION_TYPE	Type
	)

Definition at line 390 of file notify.c.

{
    PLIST_ENTRY ListEntry;
    PNOTIFICATION_ITEM NotificationDll;
    WLX_NOTIFICATION_INFO Info;
    HKEY hNotifyKey = NULL;
    DWORD dwError;
 
    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);
 
    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)
            CallNotificationDll(hNotifyKey, NotificationDll, Type, &Info);
 
        ListEntry = ListEntry->Flink;
    }
 
    RegCloseKey(hNotifyKey);
}

Referenced by DoGenericAction(), HandleLogoff(), HandleLogon(), HandleShutdown(), StartScreenSaver(), and WinMain().

CleanupNotifications()

VOID CleanupNotifications

(

VOID

)

Definition at line 470 of file notify.c.

{
    PLIST_ENTRY ListEntry;
    PNOTIFICATION_ITEM NotificationDll;
 
    ListEntry = NotificationDllListHead.Flink;
    while (ListEntry != &NotificationDllListHead)
    {
        NotificationDll = CONTAINING_RECORD(ListEntry,
                                            NOTIFICATION_ITEM,
                                            ListEntry);
        if (NotificationDll != NULL)
        {
            if (NotificationDll->pszKeyName != NULL)
                RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszKeyName);
 
            if (NotificationDll->pszDllName != NULL)
                RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll->pszDllName);
        }
 
        ListEntry = ListEntry->Flink;
 
        RemoveEntryList(&NotificationDll->ListEntry);
 
        RtlFreeHeap(RtlGetProcessHeap(), 0, NotificationDll);
    }
}

Referenced by WinMain().

InitNotifications()

BOOL InitNotifications

(

VOID

)

Definition at line 253 of file notify.c.

{
    HKEY hNotifyKey = NULL;
    LONG lError;
    DWORD dwIndex;
    WCHAR szKeyName[80];
    DWORD dwKeyName;
 
    TRACE("InitNotifications()\n");
 
    InitializeListHead(&NotificationDllListHead);
 
    AddSfcNotification();
 
    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 = ARRAYSIZE(szKeyName);
        lError = RegEnumKeyExW(hNotifyKey,
                               dwIndex,
                               szKeyName,
                               &dwKeyName,
                               NULL,
                               NULL,
                               NULL,
                               NULL);
        if (lError != ERROR_SUCCESS)
            break;
 
        TRACE("Notification DLL: %S\n", szKeyName);
        AddNotificationDll(hNotifyKey, szKeyName);
 
        dwIndex++;
    }
 
    RegCloseKey(hNotifyKey);
 
    TRACE("InitNotifications() done\n");
 
    return TRUE;
}

Referenced by WinMain().

Variable Documentation

FuncNames

PSTR FuncNames[LastHandler]

static

Initial value:

=
{
    "Logon",
    "Logoff",
    "Lock",
    "Unlock",
    "Startup",
    "Shutdown",
    "StartScreenSaver",
    "StopScreenSaver",
    "Disconnect",
    "Reconnect",
    "StartShell",
    "PostShell"
}

Definition at line 19 of file notify.c.

Referenced by CallNotificationDll().

NotificationDllListHead

LIST_ENTRY NotificationDllListHead

static

Definition at line 50 of file notify.c.

Referenced by AddNotificationDll(), AddSfcNotification(), CallNotificationDlls(), CleanupNotifications(), and InitNotifications().