#include "precomp.h"
#include <debug.h>

Include dependency graph for umpnpmgr.c:

Macros
#define	NDEBUG

Functions
static VOID	UpdateServiceStatus (_In_ DWORD dwState, _In_ DWORD dwCheckPoint)

static DWORD WINAPI	ServiceControlHandler (DWORD dwControl, DWORD dwEventType, LPVOID lpEventData, LPVOID lpContext)

static DWORD	GetBooleanRegValue (IN HKEY hKey, IN PCWSTR lpSubKey, IN PCWSTR lpValue, OUT PBOOL pValue)

BOOL	GetSuppressNewUIValue (VOID)

BOOL	RunningOnLiveMedium (VOID)

VOID WINAPI	ServiceMain (DWORD argc, LPTSTR *argv)

static DWORD	InitializePnPManager (VOID)

BOOL WINAPI	DllMain (HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)

Variables
static WCHAR	ServiceName [] = L"PlugPlay"

static SERVICE_STATUS_HANDLE	ServiceStatusHandle

static SERVICE_STATUS	ServiceStatus

HKEY	hEnumKey = NULL

HKEY	hClassKey = NULL

BOOL	g_IsUISuppressed = FALSE

BOOL	g_ShuttingDown = FALSE

BOOL	g_IsLiveMedium = FALSE

Macro Definition Documentation

◆ NDEBUG

#define NDEBUG

Definition at line 33 of file umpnpmgr.c.

Function Documentation

◆ DllMain()

BOOL WINAPI DllMain	(	HINSTANCE	hinstDLL,
		DWORD	fdwReason,
		LPVOID	lpvReserved
	)

Definition at line 406 of file umpnpmgr.c.

{
    switch (fdwReason)
    {
        case DLL_PROCESS_ATTACH:
            DisableThreadLibraryCalls(hinstDLL);
            InitializePnPManager();
            break;
 
        case DLL_PROCESS_DETACH:
            break;
    }
 
    return TRUE;
}

◆ GetBooleanRegValue()

static DWORD GetBooleanRegValue	(	IN HKEY	hKey,
		IN PCWSTR	lpSubKey,
		IN PCWSTR	lpValue,
		OUT PBOOL	pValue
	)

static

Definition at line 133 of file umpnpmgr.c.

{
    DWORD dwError, dwType, dwData;
    DWORD cbData = sizeof(dwData);
    HKEY hSubKey = NULL;
 
    /* Default value */
    *pValue = FALSE;
 
    dwError = RegOpenKeyExW(hKey,
                            lpSubKey,
                            0,
                            KEY_READ,
                            &hSubKey);
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT("GetBooleanRegValue(): RegOpenKeyExW() has failed to open '%S' key! (Error: %lu)\n",
               lpSubKey, dwError);
        return dwError;
    }
 
    dwError = RegQueryValueExW(hSubKey,
                               lpValue,
                               0,
                               &dwType,
                               (PBYTE)&dwData,
                               &cbData);
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT("GetBooleanRegValue(): RegQueryValueExW() has failed to query '%S' value! (Error: %lu)\n",
               lpValue, dwError);
        goto Cleanup;
    }
    if (dwType != REG_DWORD)
    {
        DPRINT("GetBooleanRegValue(): The value is not of REG_DWORD type!\n");
        goto Cleanup;
    }
 
    /* Return the value */
    *pValue = (dwData == 1);
 
Cleanup:
    RegCloseKey(hSubKey);
 
    return dwError;
}

Referenced by GetSuppressNewUIValue(), and InitializePnPManager().

◆ GetSuppressNewUIValue()

BOOL GetSuppressNewUIValue ( VOID )

Definition at line 186 of file umpnpmgr.c.

{
    BOOL bSuppressNewHWUI = FALSE;
 
    /*
     * Query the SuppressNewHWUI policy registry value. Don't cache it
     * as we want to update our behaviour in consequence.
     */
    GetBooleanRegValue(HKEY_LOCAL_MACHINE,
                       L"Software\\Policies\\Microsoft\\Windows\\DeviceInstall\\Settings",
                       L"SuppressNewHWUI",
                       &bSuppressNewHWUI);
    if (bSuppressNewHWUI)
        DPRINT("GetSuppressNewUIValue(): newdev.dll's wizard UI won't be shown!\n");
 
    return bSuppressNewHWUI;
}

Referenced by IsUISuppressionAllowed().

◆ InitializePnPManager()

static DWORD InitializePnPManager ( VOID )

static

Definition at line 320 of file umpnpmgr.c.

{
    BOOLEAN OldValue;
    DWORD dwError;
 
    DPRINT("UMPNPMGR: InitializePnPManager() started\n");
 
    /* We need this privilege for using CreateProcessAsUserW */
    RtlAdjustPrivilege(SE_ASSIGNPRIMARYTOKEN_PRIVILEGE, TRUE, FALSE, &OldValue);
 
    hInstallEvent = CreateEventW(NULL, TRUE, SetupIsActive()/*FALSE*/, NULL);
    if (hInstallEvent == NULL)
    {
        dwError = GetLastError();
        DPRINT1("Could not create the Install Event! (Error %lu)\n", dwError);
        return dwError;
    }
 
    hNoPendingInstalls = CreateEventW(NULL,
                                      TRUE,
                                      FALSE,
                                      L"Global\\PnP_No_Pending_Install_Events");
    if (hNoPendingInstalls == NULL)
    {
        dwError = GetLastError();
        DPRINT1("Could not create the Pending-Install Event! (Error %lu)\n", dwError);
        return dwError;
    }
 
    /*
     * Initialize the device-install event list
     */
 
    hDeviceInstallListNotEmpty = CreateEventW(NULL, FALSE, FALSE, NULL);
    if (hDeviceInstallListNotEmpty == NULL)
    {
        dwError = GetLastError();
        DPRINT1("Could not create the List Event! (Error %lu)\n", dwError);
        return dwError;
    }
 
    hDeviceInstallListMutex = CreateMutexW(NULL, FALSE, NULL);
    if (hDeviceInstallListMutex == NULL)
    {
        dwError = GetLastError();
        DPRINT1("Could not create the List Mutex! (Error %lu)\n", dwError);
        return dwError;
    }
    InitializeListHead(&DeviceInstallListHead);
 
    /* Query the SuppressUI registry value and cache it for our whole lifetime */
    GetBooleanRegValue(HKEY_LOCAL_MACHINE,
                       L"System\\CurrentControlSet\\Services\\PlugPlay\\Parameters",
                       L"SuppressUI",
                       &g_IsUISuppressed);
    if (g_IsUISuppressed)
        DPRINT("UMPNPMGR: newdev.dll's wizard UI won't be shown!\n");
 
    dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                            L"System\\CurrentControlSet\\Enum",
                            0,
                            KEY_ALL_ACCESS,
                            &hEnumKey);
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("Could not open the Enum Key! (Error %lu)\n", dwError);
        return dwError;
    }
 
    dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                            L"System\\CurrentControlSet\\Control\\Class",
                            0,
                            KEY_ALL_ACCESS,
                            &hClassKey);
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("Could not open the Class Key! (Error %lu)\n", dwError);
        return dwError;
    }
 
    DPRINT("UMPNPMGR: InitializePnPManager() done\n");
 
    return 0;
}

Referenced by DllMain().

◆ RunningOnLiveMedium()

BOOL RunningOnLiveMedium ( VOID )

Definition at line 205 of file umpnpmgr.c.

{
    WCHAR Options[MAX_PATH];
    PWSTR CurrentOption, NextOption;
    HKEY hControlKey;
    DWORD dwType;
    DWORD dwSize;
    DWORD dwError;
    BOOL LiveMedium = FALSE;
 
    DPRINT("RunningOnLiveMedium()\n");
 
    /* Open the Setup key */
    dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                            L"SYSTEM\\CurrentControlSet\\Control",
                            0,
                            KEY_QUERY_VALUE,
                            &hControlKey);
    if (dwError != ERROR_SUCCESS)
        goto done;
 
    /* Read the CmdLine value */
    dwSize = sizeof(Options);
    dwError = RegQueryValueExW(hControlKey,
                               L"SystemStartOptions",
                               NULL,
                               &dwType,
                               (LPBYTE)Options,
                               &dwSize);
    if ((dwError != ERROR_SUCCESS) || (dwType != REG_SZ))
        goto done;
 
    /* Check for the '-mini' option */
    CurrentOption = Options;
    while (CurrentOption)
    {
        NextOption = wcschr(CurrentOption, L' ');
        if (NextOption)
            *NextOption = L'\0';
        if (_wcsicmp(CurrentOption, L"MININT") == 0)
        {
            DPRINT("Found 'MININT' boot option\n");
            LiveMedium = TRUE;
            goto done;
        }
        CurrentOption = NextOption ? NextOption + 1 : NULL;
    }
 
done:
    RegCloseKey(hControlKey);
 
    return LiveMedium;
}

Referenced by ServiceMain().

◆ ServiceControlHandler()

static DWORD WINAPI ServiceControlHandler	(	DWORD	dwControl,
		DWORD	dwEventType,
		LPVOID	lpEventData,
		LPVOID	lpContext
	)

static

Definition at line 85 of file umpnpmgr.c.

{
    DPRINT1("ServiceControlHandler() called\n");
 
    switch (dwControl)
    {
        case SERVICE_CONTROL_STOP:
            DPRINT1("  SERVICE_CONTROL_STOP received\n");
            UpdateServiceStatus(SERVICE_STOP_PENDING, 1);
            /* Stop listening to RPC Messages */
            RpcMgmtStopServerListening(NULL);
            UpdateServiceStatus(SERVICE_STOPPED, 0);
            return ERROR_SUCCESS;
 
        case SERVICE_CONTROL_PAUSE:
            DPRINT1("  SERVICE_CONTROL_PAUSE received\n");
            UpdateServiceStatus(SERVICE_PAUSED, 0);
            return ERROR_SUCCESS;
 
        case SERVICE_CONTROL_CONTINUE:
            DPRINT1("  SERVICE_CONTROL_CONTINUE received\n");
            UpdateServiceStatus(SERVICE_RUNNING, 0);
            return ERROR_SUCCESS;
 
        case SERVICE_CONTROL_INTERROGATE:
            DPRINT1("  SERVICE_CONTROL_INTERROGATE received\n");
            SetServiceStatus(ServiceStatusHandle,
                             &ServiceStatus);
            return ERROR_SUCCESS;
 
        case SERVICE_CONTROL_SHUTDOWN:
            DPRINT1("  SERVICE_CONTROL_SHUTDOWN received\n");
            UpdateServiceStatus(SERVICE_STOP_PENDING, 1);
            /* Stop listening to RPC Messages */
            RpcMgmtStopServerListening(NULL);
            UpdateServiceStatus(SERVICE_STOPPED, 0);
            return ERROR_SUCCESS;
 
        default :
            DPRINT1("  Control %lu received\n", dwControl);
            return ERROR_CALL_NOT_IMPLEMENTED;
    }
}

Referenced by ServiceMain().

◆ ServiceMain()

VOID WINAPI ServiceMain	(	DWORD	argc,
		LPTSTR *	argv
	)

Definition at line 260 of file umpnpmgr.c.

{
    HANDLE hThread;
    DWORD dwThreadId;
 
    UNREFERENCED_PARAMETER(argc);
    UNREFERENCED_PARAMETER(argv);
 
    DPRINT("ServiceMain() called\n");
 
    g_IsLiveMedium = RunningOnLiveMedium();
 
    ServiceStatusHandle = RegisterServiceCtrlHandlerExW(ServiceName,
                                                        ServiceControlHandler,
                                                        NULL);
    if (!ServiceStatusHandle)
    {
        DPRINT1("RegisterServiceCtrlHandlerExW() failed! (Error %lu)\n", GetLastError());
        return;
    }
 
    UpdateServiceStatus(SERVICE_START_PENDING, 1);
 
    hThread = CreateThread(NULL,
                           0,
                           PnpEventThread,
                           NULL,
                           0,
                           &dwThreadId);
    if (hThread != NULL)
        CloseHandle(hThread);
 
    UpdateServiceStatus(SERVICE_START_PENDING, 2);
 
    hThread = CreateThread(NULL,
                           0,
                           RpcServerThread,
                           NULL,
                           0,
                           &dwThreadId);
    if (hThread != NULL)
        CloseHandle(hThread);
 
    UpdateServiceStatus(SERVICE_START_PENDING, 3);
 
    hThread = CreateThread(NULL,
                           0,
                           DeviceInstallThread,
                           NULL,
                           0,
                           &dwThreadId);
    if (hThread != NULL)
        CloseHandle(hThread);
 
    UpdateServiceStatus(SERVICE_RUNNING, 0);
 
    DPRINT("ServiceMain() done\n");
}

◆ UpdateServiceStatus()

static VOID UpdateServiceStatus	(	_In_ DWORD	dwState,
		_In_ DWORD	dwCheckPoint
	)

static

Definition at line 53 of file umpnpmgr.c.

{
    if ((dwState == SERVICE_STOPPED) || (dwState == SERVICE_STOP_PENDING))
        g_ShuttingDown = TRUE;
 
    ServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
    ServiceStatus.dwCurrentState = dwState;
    ServiceStatus.dwWin32ExitCode = 0;
    ServiceStatus.dwServiceSpecificExitCode = 0;
    ServiceStatus.dwCheckPoint = dwCheckPoint;
 
    if (dwState == SERVICE_RUNNING)
        ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_SHUTDOWN;
    else
        ServiceStatus.dwControlsAccepted = 0;
 
    if (dwState == SERVICE_START_PENDING ||
        dwState == SERVICE_STOP_PENDING ||
        dwState == SERVICE_PAUSE_PENDING ||
        dwState == SERVICE_CONTINUE_PENDING)
        ServiceStatus.dwWaitHint = 10000;
    else
        ServiceStatus.dwWaitHint = 0;
 
    SetServiceStatus(ServiceStatusHandle,
                     &ServiceStatus);
}