mirror of
https://github.com/reactos/reactos.git
synced 2024-11-02 21:09:15 +00:00
546 lines
17 KiB
C
546 lines
17 KiB
C
/*
|
|
* PnP Test
|
|
* ReactOS Device Interface functions implementation
|
|
*
|
|
* Copyright 2003, 2004 Filip Navara <xnavara@volny.cz>
|
|
* Copyright 2003, 2004 Matthew Brace <ismarc@austin.rr.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; see the file COPYING.LIB.
|
|
* If not, write to the Free Software Foundation,
|
|
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
/* INCLUDES *******************************************************************/
|
|
|
|
#include <ddk/ntddk.h>
|
|
#include "kmtest.h"
|
|
|
|
//#define NDEBUG
|
|
#include "debug.h"
|
|
|
|
/* PUBLIC FUNCTIONS ***********************************************************/
|
|
|
|
/*
|
|
* IoGetDeviceInterfaces
|
|
*
|
|
* Returns a list of device interfaces of a particular device interface class.
|
|
*
|
|
* Parameters
|
|
* InterfaceClassGuid
|
|
* Points to a class GUID specifying the device interface class.
|
|
*
|
|
* PhysicalDeviceObject
|
|
* Points to an optional PDO that narrows the search to only the
|
|
* device interfaces of the device represented by the PDO.
|
|
*
|
|
* Flags
|
|
* Specifies flags that modify the search for device interfaces. The
|
|
* DEVICE_INTERFACE_INCLUDE_NONACTIVE flag specifies that the list of
|
|
* returned symbolic links should contain also disabled device
|
|
* interfaces in addition to the enabled ones.
|
|
*
|
|
* SymbolicLinkList
|
|
* Points to a character pointer that is filled in on successful return
|
|
* with a list of unicode strings identifying the device interfaces
|
|
* that match the search criteria. The newly allocated buffer contains
|
|
* a list of symbolic link names. Each unicode string in the list is
|
|
* null-terminated; the end of the whole list is marked by an additional
|
|
* NULL. The caller is responsible for freeing the buffer (ExFreePool)
|
|
* when it is no longer needed.
|
|
* If no device interfaces match the search criteria, this routine
|
|
* returns STATUS_SUCCESS and the string contains a single NULL
|
|
* character.
|
|
*
|
|
* Status
|
|
* @unimplemented
|
|
*
|
|
* The parameters PhysicalDeviceObject and Flags aren't correctly
|
|
* processed. Rest of the cases was tested under Win XP and the
|
|
* function worked correctly.
|
|
*/
|
|
|
|
NTSTATUS STDCALL
|
|
ReactOS_IoGetDeviceInterfaces(
|
|
IN CONST GUID *InterfaceClassGuid,
|
|
IN PDEVICE_OBJECT PhysicalDeviceObject OPTIONAL,
|
|
IN ULONG Flags,
|
|
OUT PWSTR *SymbolicLinkList)
|
|
{
|
|
PWCHAR BaseKeyString = L"\\Registry\\Machine\\System\\CurrentControlSet\\Control\\DeviceClasses\\";
|
|
PWCHAR BaseInterfaceString = L"\\Registry\\Machine\\System\\CurrentControlSet\\Services\\";
|
|
UNICODE_STRING GuidString;
|
|
UNICODE_STRING BaseKeyName;
|
|
UNICODE_STRING AliasKeyName;
|
|
UNICODE_STRING SymbolicLink;
|
|
UNICODE_STRING Control;
|
|
UNICODE_STRING SubKeyName;
|
|
UNICODE_STRING SymbolicLinkKeyName;
|
|
UNICODE_STRING ControlKeyName;
|
|
UNICODE_STRING TempString;
|
|
HANDLE InterfaceKey;
|
|
HANDLE SubKey;
|
|
HANDLE SymbolicLinkKey;
|
|
PKEY_FULL_INFORMATION fip;
|
|
PKEY_FULL_INFORMATION bfip=NULL;
|
|
PKEY_BASIC_INFORMATION bip;
|
|
PKEY_VALUE_PARTIAL_INFORMATION vpip=NULL;
|
|
PWCHAR SymLinkList = NULL;
|
|
ULONG SymLinkListSize = 0;
|
|
NTSTATUS Status;
|
|
ULONG Size = 0;
|
|
ULONG i = 0;
|
|
ULONG j = 0;
|
|
OBJECT_ATTRIBUTES ObjectAttributes;
|
|
|
|
Status = RtlStringFromGUID(InterfaceClassGuid, &GuidString);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("RtlStringFromGUID() Failed.\n");
|
|
return STATUS_INVALID_HANDLE;
|
|
}
|
|
|
|
RtlInitUnicodeString(&AliasKeyName, BaseInterfaceString);
|
|
RtlInitUnicodeString(&SymbolicLink, L"SymbolicLink");
|
|
RtlInitUnicodeString(&Control, L"\\Control");
|
|
BaseKeyName.Length = wcslen(BaseKeyString) * sizeof(WCHAR);
|
|
BaseKeyName.MaximumLength = BaseKeyName.Length + (38 * sizeof(WCHAR));
|
|
BaseKeyName.Buffer = ExAllocatePool(
|
|
NonPagedPool,
|
|
BaseKeyName.MaximumLength);
|
|
ASSERT(BaseKeyName.Buffer != NULL);
|
|
wcscpy(BaseKeyName.Buffer, BaseKeyString);
|
|
RtlAppendUnicodeStringToString(&BaseKeyName, &GuidString);
|
|
|
|
if (PhysicalDeviceObject)
|
|
{
|
|
WCHAR GuidBuffer[40];
|
|
UNICODE_STRING PdoGuidString;
|
|
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
|
|
IoGetDeviceProperty(
|
|
PhysicalDeviceObject,
|
|
DevicePropertyClassGuid,
|
|
sizeof(GuidBuffer),
|
|
GuidBuffer,
|
|
&Size);
|
|
|
|
RtlInitUnicodeString(&PdoGuidString, GuidBuffer);
|
|
if (RtlCompareUnicodeString(&GuidString, &PdoGuidString, TRUE))
|
|
{
|
|
DPRINT("Inconsistent Guid's asked for in IoGetDeviceInterfaces()\n");
|
|
return STATUS_INVALID_HANDLE;
|
|
}
|
|
|
|
DPRINT("IoGetDeviceInterfaces() called with PDO, not implemented.\n");
|
|
return STATUS_NOT_IMPLEMENTED;
|
|
}
|
|
else
|
|
{
|
|
InitializeObjectAttributes(
|
|
&ObjectAttributes,
|
|
&BaseKeyName,
|
|
OBJ_CASE_INSENSITIVE,
|
|
NULL,
|
|
NULL);
|
|
|
|
Status = ZwOpenKey(
|
|
&InterfaceKey,
|
|
KEY_READ,
|
|
&ObjectAttributes);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("ZwOpenKey() Failed. (0x%X)\n", Status);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
return Status;
|
|
}
|
|
|
|
Status = ZwQueryKey(
|
|
InterfaceKey,
|
|
KeyFullInformation,
|
|
NULL,
|
|
0,
|
|
&Size);
|
|
|
|
if (Status != STATUS_BUFFER_TOO_SMALL)
|
|
{
|
|
DPRINT("ZwQueryKey() Failed. (0x%X)\n", Status);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
fip = (PKEY_FULL_INFORMATION)ExAllocatePool(NonPagedPool, Size);
|
|
ASSERT(fip != NULL);
|
|
|
|
Status = ZwQueryKey(
|
|
InterfaceKey,
|
|
KeyFullInformation,
|
|
fip,
|
|
Size,
|
|
&Size);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("ZwQueryKey() Failed. (0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
for (; i < fip->SubKeys; i++)
|
|
{
|
|
Status = ZwEnumerateKey(
|
|
InterfaceKey,
|
|
i,
|
|
KeyBasicInformation,
|
|
NULL,
|
|
0,
|
|
&Size);
|
|
|
|
if (Status != STATUS_BUFFER_TOO_SMALL)
|
|
{
|
|
DPRINT("ZwEnumerateKey() Failed.(0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
if (SymLinkList != NULL)
|
|
ExFreePool(SymLinkList);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
bip = (PKEY_BASIC_INFORMATION)ExAllocatePool(NonPagedPool, Size);
|
|
ASSERT(bip != NULL);
|
|
|
|
Status = ZwEnumerateKey(
|
|
InterfaceKey,
|
|
i,
|
|
KeyBasicInformation,
|
|
bip,
|
|
Size,
|
|
&Size);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("ZwEnumerateKey() Failed.(0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
ExFreePool(bip);
|
|
if (SymLinkList != NULL)
|
|
ExFreePool(SymLinkList);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
SubKeyName.Length = 0;
|
|
SubKeyName.MaximumLength = BaseKeyName.Length + bip->NameLength + sizeof(WCHAR);
|
|
SubKeyName.Buffer = ExAllocatePool(NonPagedPool, SubKeyName.MaximumLength);
|
|
ASSERT(SubKeyName.Buffer != NULL);
|
|
TempString.Length = TempString.MaximumLength = bip->NameLength;
|
|
TempString.Buffer = bip->Name;
|
|
RtlCopyUnicodeString(&SubKeyName, &BaseKeyName);
|
|
RtlAppendUnicodeToString(&SubKeyName, L"\\");
|
|
RtlAppendUnicodeStringToString(&SubKeyName, &TempString);
|
|
|
|
ExFreePool(bip);
|
|
|
|
InitializeObjectAttributes(
|
|
&ObjectAttributes,
|
|
&SubKeyName,
|
|
OBJ_CASE_INSENSITIVE,
|
|
NULL,
|
|
NULL);
|
|
|
|
Status = ZwOpenKey(
|
|
&SubKey,
|
|
KEY_READ,
|
|
&ObjectAttributes);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("ZwOpenKey() Failed. (0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
if (SymLinkList != NULL)
|
|
ExFreePool(SymLinkList);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
Status = ZwQueryKey(
|
|
SubKey,
|
|
KeyFullInformation,
|
|
NULL,
|
|
0,
|
|
&Size);
|
|
|
|
if (Status != STATUS_BUFFER_TOO_SMALL)
|
|
{
|
|
DPRINT("ZwQueryKey() Failed. (0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
ZwClose(SubKey);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
bfip = (PKEY_FULL_INFORMATION)ExAllocatePool(NonPagedPool, Size);
|
|
ASSERT(bfip != NULL);
|
|
|
|
Status = ZwQueryKey(
|
|
SubKey,
|
|
KeyFullInformation,
|
|
bfip,
|
|
Size,
|
|
&Size);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("ZwQueryKey() Failed. (0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(SubKey);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
for(j = 0; j < bfip->SubKeys; j++)
|
|
{
|
|
Status = ZwEnumerateKey(
|
|
SubKey,
|
|
j,
|
|
KeyBasicInformation,
|
|
NULL,
|
|
0,
|
|
&Size);
|
|
|
|
if (Status == STATUS_NO_MORE_ENTRIES)
|
|
continue;
|
|
|
|
if (Status != STATUS_BUFFER_TOO_SMALL)
|
|
{
|
|
DPRINT("ZwEnumerateKey() Failed.(0x%X)\n", Status);
|
|
ExFreePool(bfip);
|
|
ExFreePool(fip);
|
|
if (SymLinkList != NULL)
|
|
ExFreePool(SymLinkList);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(SubKey);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
bip = (PKEY_BASIC_INFORMATION)ExAllocatePool(NonPagedPool, Size);
|
|
ASSERT(bip != NULL);
|
|
|
|
Status = ZwEnumerateKey(
|
|
SubKey,
|
|
j,
|
|
KeyBasicInformation,
|
|
bip,
|
|
Size,
|
|
&Size);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("ZwEnumerateKey() Failed.(0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
ExFreePool(bfip);
|
|
ExFreePool(bip);
|
|
if (SymLinkList != NULL)
|
|
ExFreePool(SymLinkList);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(SubKey);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
if (!wcsncmp(bip->Name, L"Control", bip->NameLength))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
SymbolicLinkKeyName.Length = 0;
|
|
SymbolicLinkKeyName.MaximumLength = SubKeyName.Length + bip->NameLength + sizeof(WCHAR);
|
|
SymbolicLinkKeyName.Buffer = ExAllocatePool(NonPagedPool, SymbolicLinkKeyName.MaximumLength);
|
|
ASSERT(SymbolicLinkKeyName.Buffer != NULL);
|
|
TempString.Length = TempString.MaximumLength = bip->NameLength;
|
|
TempString.Buffer = bip->Name;
|
|
RtlCopyUnicodeString(&SymbolicLinkKeyName, &SubKeyName);
|
|
RtlAppendUnicodeToString(&SymbolicLinkKeyName, L"\\");
|
|
RtlAppendUnicodeStringToString(&SymbolicLinkKeyName, &TempString);
|
|
|
|
ControlKeyName.Length = 0;
|
|
ControlKeyName.MaximumLength = SymbolicLinkKeyName.Length + Control.Length + sizeof(WCHAR);
|
|
ControlKeyName.Buffer = ExAllocatePool(NonPagedPool, ControlKeyName.MaximumLength);
|
|
ASSERT(ControlKeyName.Buffer != NULL);
|
|
RtlCopyUnicodeString(&ControlKeyName, &SymbolicLinkKeyName);
|
|
RtlAppendUnicodeStringToString(&ControlKeyName, &Control);
|
|
|
|
ExFreePool(bip);
|
|
|
|
InitializeObjectAttributes(
|
|
&ObjectAttributes,
|
|
&SymbolicLinkKeyName,
|
|
OBJ_CASE_INSENSITIVE,
|
|
NULL,
|
|
NULL);
|
|
|
|
Status = ZwOpenKey(
|
|
&SymbolicLinkKey,
|
|
KEY_READ,
|
|
&ObjectAttributes);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("ZwOpenKey() Failed. (0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
ExFreePool(bfip);
|
|
if (SymLinkList != NULL)
|
|
ExFreePool(SymLinkList);
|
|
RtlFreeUnicodeString(&SymbolicLinkKeyName);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(SubKey);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
Status = ZwQueryValueKey(
|
|
SymbolicLinkKey,
|
|
&SymbolicLink,
|
|
KeyValuePartialInformation,
|
|
NULL,
|
|
0,
|
|
&Size);
|
|
|
|
if (Status == STATUS_OBJECT_NAME_NOT_FOUND)
|
|
continue;
|
|
|
|
if (Status != STATUS_BUFFER_TOO_SMALL)
|
|
{
|
|
DPRINT("ZwQueryValueKey() Failed.(0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
ExFreePool(bfip);
|
|
if (SymLinkList != NULL)
|
|
ExFreePool(SymLinkList);
|
|
RtlFreeUnicodeString(&SymbolicLinkKeyName);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(SymbolicLinkKey);
|
|
ZwClose(SubKey);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
vpip = (PKEY_VALUE_PARTIAL_INFORMATION)ExAllocatePool(NonPagedPool, Size);
|
|
ASSERT(vpip != NULL);
|
|
|
|
Status = ZwQueryValueKey(
|
|
SymbolicLinkKey,
|
|
&SymbolicLink,
|
|
KeyValuePartialInformation,
|
|
vpip,
|
|
Size,
|
|
&Size);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("ZwQueryValueKey() Failed.(0x%X)\n", Status);
|
|
ExFreePool(fip);
|
|
ExFreePool(bfip);
|
|
ExFreePool(vpip);
|
|
if (SymLinkList != NULL)
|
|
ExFreePool(SymLinkList);
|
|
RtlFreeUnicodeString(&SymbolicLinkKeyName);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(SymbolicLinkKey);
|
|
ZwClose(SubKey);
|
|
ZwClose(InterfaceKey);
|
|
return Status;
|
|
}
|
|
|
|
Status = RtlCheckRegistryKey(RTL_REGISTRY_ABSOLUTE, ControlKeyName.Buffer);
|
|
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
/* Put the name in the string here */
|
|
if (SymLinkList == NULL)
|
|
{
|
|
SymLinkListSize = vpip->DataLength;
|
|
SymLinkList = ExAllocatePool(NonPagedPool, SymLinkListSize + sizeof(WCHAR));
|
|
ASSERT(SymLinkList != NULL);
|
|
RtlCopyMemory(SymLinkList, vpip->Data, vpip->DataLength);
|
|
SymLinkList[vpip->DataLength / sizeof(WCHAR)] = 0;
|
|
SymLinkList[1] = '?';
|
|
}
|
|
else
|
|
{
|
|
PWCHAR OldSymLinkList;
|
|
ULONG OldSymLinkListSize;
|
|
PWCHAR SymLinkListPtr;
|
|
|
|
OldSymLinkList = SymLinkList;
|
|
OldSymLinkListSize = SymLinkListSize;
|
|
SymLinkListSize += vpip->DataLength;
|
|
SymLinkList = ExAllocatePool(NonPagedPool, SymLinkListSize + sizeof(WCHAR));
|
|
ASSERT(SymLinkList != NULL);
|
|
RtlCopyMemory(SymLinkList, OldSymLinkList, OldSymLinkListSize);
|
|
ExFreePool(OldSymLinkList);
|
|
SymLinkListPtr = SymLinkList + (OldSymLinkListSize / sizeof(WCHAR));
|
|
RtlCopyMemory(SymLinkListPtr, vpip->Data, vpip->DataLength);
|
|
SymLinkListPtr[vpip->DataLength / sizeof(WCHAR)] = 0;
|
|
SymLinkListPtr[1] = '?';
|
|
}
|
|
}
|
|
|
|
RtlFreeUnicodeString(&SymbolicLinkKeyName);
|
|
RtlFreeUnicodeString(&ControlKeyName);
|
|
ZwClose(SymbolicLinkKey);
|
|
}
|
|
|
|
ExFreePool(vpip);
|
|
RtlFreeUnicodeString(&SubKeyName);
|
|
ZwClose(SubKey);
|
|
}
|
|
|
|
if (SymLinkList != NULL)
|
|
{
|
|
SymLinkList[SymLinkListSize / sizeof(WCHAR)] = 0;
|
|
}
|
|
else
|
|
{
|
|
SymLinkList = ExAllocatePool(NonPagedPool, 2 * sizeof(WCHAR));
|
|
SymLinkList[0] = 0;
|
|
}
|
|
|
|
*SymbolicLinkList = SymLinkList;
|
|
|
|
RtlFreeUnicodeString(&BaseKeyName);
|
|
ZwClose(InterfaceKey);
|
|
ExFreePool(bfip);
|
|
ExFreePool(fip);
|
|
}
|
|
|
|
return STATUS_SUCCESS;
|
|
}
|