[UMPNPMGR][USETUP] Fix the way device-install events are queued and dequeued. Fixes CORE-16103.

Dedicated to the hard work of Joachim Henze! xD

This reverts part of commit 043a98dd (see also commit b2aeafca).

Contrary to what I assumed in commit 043a98dd (and was also assumed in
the older commit b2aeafca), we cannot use the singled-linked lists to
queue and dequeue the PnP device-install events, because:

- the events must be treated from the oldest to the newest ones, for
  consistency, otherwise this creates problems, as shown by e.g. CORE-16103;

- the system singled-linked lists only offer access to the top of the
  list (like a stack) instead of to both the top and the bottom of the
  list, as would be required for a queue. Using the SLISTs would mean
  that only the newest-received events would be treated first, while the
  oldest (which were the first received) events would be treated last,
  and this is wrong.

Therefore one must use e.g. the standard doubly-linked list. Also, using
locked operations (insertion & removal) on the list of device-install
events is necessary, because these operations are done concurrently by
two different threads: PnpEventThread() and DeviceInstallThread().
Since the interlocked linked list functions are not available in user-mode,
we need to use instead locking access through e.g. a mutex.
This commit is contained in:
Hermès Bélusca-Maïto 2020-09-28 22:38:55 +02:00
parent 828a65f470
commit ffc96d26ec
No known key found for this signature in database
GPG key ID: 3B2539C65E7B93D0
4 changed files with 93 additions and 42 deletions

View file

@ -22,7 +22,7 @@
* FILE: base/services/umpnpmgr/install.c * FILE: base/services/umpnpmgr/install.c
* PURPOSE: Device installer * PURPOSE: Device installer
* PROGRAMMER: Eric Kohl (eric.kohl@reactos.org) * PROGRAMMER: Eric Kohl (eric.kohl@reactos.org)
* Hervé Poussineau (hpoussin@reactos.org) * Hervé Poussineau (hpoussin@reactos.org)
* Colin Finck (colin@reactos.org) * Colin Finck (colin@reactos.org)
*/ */
@ -40,7 +40,9 @@ HANDLE hUserToken = NULL;
HANDLE hInstallEvent = NULL; HANDLE hInstallEvent = NULL;
HANDLE hNoPendingInstalls = NULL; HANDLE hNoPendingInstalls = NULL;
SLIST_HEADER DeviceInstallListHead; /* Device-install event list */
HANDLE hDeviceInstallListMutex;
LIST_ENTRY DeviceInstallListHead;
HANDLE hDeviceInstallListNotEmpty; HANDLE hDeviceInstallListNotEmpty;
@ -354,7 +356,7 @@ DWORD
WINAPI WINAPI
DeviceInstallThread(LPVOID lpParameter) DeviceInstallThread(LPVOID lpParameter)
{ {
PSLIST_ENTRY ListEntry; PLIST_ENTRY ListEntry;
DeviceInstallParams* Params; DeviceInstallParams* Params;
BOOL showWizard; BOOL showWizard;
@ -366,7 +368,11 @@ DeviceInstallThread(LPVOID lpParameter)
while (TRUE) while (TRUE)
{ {
ListEntry = InterlockedPopEntrySList(&DeviceInstallListHead); /* Dequeue the next oldest device-install event */
WaitForSingleObject(hDeviceInstallListMutex, INFINITE);
ListEntry = (IsListEmpty(&DeviceInstallListHead)
? NULL : RemoveHeadList(&DeviceInstallListHead));
ReleaseMutex(hDeviceInstallListMutex);
if (ListEntry == NULL) if (ListEntry == NULL)
{ {

View file

@ -35,8 +35,8 @@
typedef struct typedef struct
{ {
SLIST_ENTRY ListEntry; LIST_ENTRY ListEntry;
WCHAR DeviceIds[1]; WCHAR DeviceIds[ANYSIZE_ARRAY];
} DeviceInstallParams; } DeviceInstallParams;
/* install.c */ /* install.c */
@ -45,7 +45,9 @@ extern HANDLE hUserToken;
extern HANDLE hInstallEvent; extern HANDLE hInstallEvent;
extern HANDLE hNoPendingInstalls; extern HANDLE hNoPendingInstalls;
extern SLIST_HEADER DeviceInstallListHead; /* Device-install event list */
extern HANDLE hDeviceInstallListMutex;
extern LIST_ENTRY DeviceInstallListHead;
extern HANDLE hDeviceInstallListNotEmpty; extern HANDLE hDeviceInstallListNotEmpty;
BOOL BOOL

View file

@ -104,13 +104,18 @@ PnpEventThread(LPVOID lpParameter)
DeviceIdLength = lstrlenW(PnpEvent->TargetDevice.DeviceIds); DeviceIdLength = lstrlenW(PnpEvent->TargetDevice.DeviceIds);
if (DeviceIdLength) if (DeviceIdLength)
{ {
/* Queue device install (will be dequeued by DeviceInstallThread) */ /* Allocate a new device-install event */
len = FIELD_OFFSET(DeviceInstallParams, DeviceIds) + (DeviceIdLength + 1) * sizeof(WCHAR); len = FIELD_OFFSET(DeviceInstallParams, DeviceIds) + (DeviceIdLength + 1) * sizeof(WCHAR);
Params = HeapAlloc(GetProcessHeap(), 0, len); Params = HeapAlloc(GetProcessHeap(), 0, len);
if (Params) if (Params)
{ {
wcscpy(Params->DeviceIds, PnpEvent->TargetDevice.DeviceIds); wcscpy(Params->DeviceIds, PnpEvent->TargetDevice.DeviceIds);
InterlockedPushEntrySList(&DeviceInstallListHead, &Params->ListEntry);
/* Queue the event (will be dequeued by DeviceInstallThread) */
WaitForSingleObject(hDeviceInstallListMutex, INFINITE);
InsertTailList(&DeviceInstallListHead, &Params->ListEntry);
ReleaseMutex(hDeviceInstallListMutex);
SetEvent(hDeviceInstallListNotEmpty); SetEvent(hDeviceInstallListNotEmpty);
} }
} }
@ -413,14 +418,6 @@ InitializePnPManager(VOID)
return dwError; return dwError;
} }
hDeviceInstallListNotEmpty = CreateEventW(NULL, FALSE, FALSE, NULL);
if (hDeviceInstallListNotEmpty == NULL)
{
dwError = GetLastError();
DPRINT1("Could not create the Event! (Error %lu)\n", dwError);
return dwError;
}
hNoPendingInstalls = CreateEventW(NULL, hNoPendingInstalls = CreateEventW(NULL,
TRUE, TRUE,
FALSE, FALSE,
@ -428,11 +425,30 @@ InitializePnPManager(VOID)
if (hNoPendingInstalls == NULL) if (hNoPendingInstalls == NULL)
{ {
dwError = GetLastError(); dwError = GetLastError();
DPRINT1("Could not create the Event! (Error %lu)\n", dwError); DPRINT1("Could not create the Pending-Install Event! (Error %lu)\n", dwError);
return dwError; return dwError;
} }
InitializeSListHead(&DeviceInstallListHead); /*
* 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 */ /* Query the SuppressUI registry value and cache it for our whole lifetime */
GetBooleanRegValue(HKEY_LOCAL_MACHINE, GetBooleanRegValue(HKEY_LOCAL_MACHINE,

View file

@ -25,12 +25,14 @@ static HANDLE hNoPendingInstalls = NULL;
static HANDLE hPnpThread = NULL; static HANDLE hPnpThread = NULL;
static HANDLE hDeviceInstallThread = NULL; static HANDLE hDeviceInstallThread = NULL;
static SLIST_HEADER DeviceInstallListHead; /* Device-install event list */
static HANDLE hDeviceInstallListMutex = NULL;
static LIST_ENTRY DeviceInstallListHead;
static HANDLE hDeviceInstallListNotEmpty = NULL; static HANDLE hDeviceInstallListNotEmpty = NULL;
typedef struct typedef struct
{ {
SLIST_ENTRY ListEntry; LIST_ENTRY ListEntry;
WCHAR DeviceIds[ANYSIZE_ARRAY]; WCHAR DeviceIds[ANYSIZE_ARRAY];
} DeviceInstallParams; } DeviceInstallParams;
@ -363,13 +365,17 @@ static ULONG NTAPI
DeviceInstallThread(IN PVOID Parameter) DeviceInstallThread(IN PVOID Parameter)
{ {
HINF hSetupInf = *(HINF*)Parameter; HINF hSetupInf = *(HINF*)Parameter;
PSLIST_ENTRY ListEntry; PLIST_ENTRY ListEntry;
DeviceInstallParams* Params; DeviceInstallParams* Params;
LARGE_INTEGER Timeout; LARGE_INTEGER Timeout;
for (;;) for (;;)
{ {
ListEntry = RtlInterlockedPopEntrySList(&DeviceInstallListHead); /* Dequeue the next oldest device-install event */
NtWaitForSingleObject(hDeviceInstallListMutex, FALSE, NULL);
ListEntry = (IsListEmpty(&DeviceInstallListHead)
? NULL : RemoveHeadList(&DeviceInstallListHead));
NtReleaseMutant(hDeviceInstallListMutex, NULL);
if (ListEntry == NULL) if (ListEntry == NULL)
{ {
@ -454,18 +460,23 @@ PnpEventThread(IN PVOID Parameter)
ULONG len; ULONG len;
ULONG DeviceIdLength; ULONG DeviceIdLength;
DPRINT("Device enumerated event: %S\n", PnpEvent->TargetDevice.DeviceIds); DPRINT("Device enumerated: %S\n", PnpEvent->TargetDevice.DeviceIds);
DeviceIdLength = wcslen(PnpEvent->TargetDevice.DeviceIds); DeviceIdLength = wcslen(PnpEvent->TargetDevice.DeviceIds);
if (DeviceIdLength) if (DeviceIdLength)
{ {
/* Queue device install (will be dequeued by DeviceInstallThread) */ /* Allocate a new device-install event */
len = FIELD_OFFSET(DeviceInstallParams, DeviceIds) + (DeviceIdLength + 1) * sizeof(WCHAR); len = FIELD_OFFSET(DeviceInstallParams, DeviceIds) + (DeviceIdLength + 1) * sizeof(WCHAR);
Params = RtlAllocateHeap(ProcessHeap, 0, len); Params = RtlAllocateHeap(ProcessHeap, 0, len);
if (Params) if (Params)
{ {
wcscpy(Params->DeviceIds, PnpEvent->TargetDevice.DeviceIds); wcscpy(Params->DeviceIds, PnpEvent->TargetDevice.DeviceIds);
RtlInterlockedPushEntrySList(&DeviceInstallListHead, &Params->ListEntry);
/* Queue the event (will be dequeued by DeviceInstallThread) */
NtWaitForSingleObject(hDeviceInstallListMutex, FALSE, NULL);
InsertTailList(&DeviceInstallListHead, &Params->ListEntry);
NtReleaseMutant(hDeviceInstallListMutex, NULL);
NtSetEvent(hDeviceInstallListNotEmpty, NULL); NtSetEvent(hDeviceInstallListNotEmpty, NULL);
} }
else else
@ -559,17 +570,6 @@ InitializeUserModePnpManager(
UNICODE_STRING EnumU = RTL_CONSTANT_STRING(L"\\Registry\\Machine\\SYSTEM\\CurrentControlSet\\Enum"); UNICODE_STRING EnumU = RTL_CONSTANT_STRING(L"\\Registry\\Machine\\SYSTEM\\CurrentControlSet\\Enum");
UNICODE_STRING ServicesU = RTL_CONSTANT_STRING(L"\\Registry\\Machine\\SYSTEM\\CurrentControlSet\\Services"); UNICODE_STRING ServicesU = RTL_CONSTANT_STRING(L"\\Registry\\Machine\\SYSTEM\\CurrentControlSet\\Services");
Status = NtCreateEvent(&hDeviceInstallListNotEmpty,
EVENT_ALL_ACCESS,
NULL,
SynchronizationEvent,
FALSE);
if (!NT_SUCCESS(Status))
{
DPRINT1("Could not create the event! (Status 0x%08lx)\n", Status);
goto Failure;
}
Status = NtCreateEvent(&hNoPendingInstalls, Status = NtCreateEvent(&hNoPendingInstalls,
EVENT_ALL_ACCESS, EVENT_ALL_ACCESS,
NULL, NULL,
@ -577,11 +577,34 @@ InitializeUserModePnpManager(
FALSE); FALSE);
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
DPRINT1("Could not create the event! (Status 0x%08lx)\n", Status); DPRINT1("Could not create the Pending-Install Event! (Status 0x%08lx)\n", Status);
goto Failure; goto Failure;
} }
RtlInitializeSListHead(&DeviceInstallListHead); /*
* Initialize the device-install event list
*/
Status = NtCreateEvent(&hDeviceInstallListNotEmpty,
EVENT_ALL_ACCESS,
NULL,
SynchronizationEvent,
FALSE);
if (!NT_SUCCESS(Status))
{
DPRINT1("Could not create the List Event! (Status 0x%08lx)\n", Status);
goto Failure;
}
Status = NtCreateMutant(&hDeviceInstallListMutex,
MUTANT_ALL_ACCESS,
NULL, FALSE);
if (!NT_SUCCESS(Status))
{
DPRINT1("Could not create the List Mutex! (Status 0x%08lx)\n", Status);
goto Failure;
}
InitializeListHead(&DeviceInstallListHead);
InitializeObjectAttributes(&ObjectAttributes, &EnumU, OBJ_CASE_INSENSITIVE, NULL, NULL); InitializeObjectAttributes(&ObjectAttributes, &EnumU, OBJ_CASE_INSENSITIVE, NULL, NULL);
Status = NtOpenKey(&hEnumKey, KEY_QUERY_VALUE, &ObjectAttributes); Status = NtOpenKey(&hEnumKey, KEY_QUERY_VALUE, &ObjectAttributes);
@ -653,14 +676,18 @@ Failure:
NtClose(hEnumKey); NtClose(hEnumKey);
hEnumKey = NULL; hEnumKey = NULL;
if (hNoPendingInstalls) if (hDeviceInstallListMutex)
NtClose(hNoPendingInstalls); NtClose(hDeviceInstallListMutex);
hNoPendingInstalls = NULL; hDeviceInstallListMutex = NULL;
if (hDeviceInstallListNotEmpty) if (hDeviceInstallListNotEmpty)
NtClose(hDeviceInstallListNotEmpty); NtClose(hDeviceInstallListNotEmpty);
hDeviceInstallListNotEmpty = NULL; hDeviceInstallListNotEmpty = NULL;
if (hNoPendingInstalls)
NtClose(hNoPendingInstalls);
hNoPendingInstalls = NULL;
return Status; return Status;
} }