/* * Plug and Play * * Copyright 2016 Sebastian Lackner * Copyright 2016 Aric Stewart for CodeWeavers * Copyright 2019 Zebediah Figura * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include #define NONAMELESSUNION #include "ntstatus.h" #define WIN32_NO_STATUS #include "windef.h" #include "winbase.h" #include "winioctl.h" #include "winreg.h" #include "winuser.h" #include "winsvc.h" #include "winternl.h" #include "setupapi.h" #include "cfgmgr32.h" #include "dbt.h" #include "ddk/wdm.h" #include "ddk/ntifs.h" #include "wine/debug.h" #include "wine/heap.h" #include "wine/rbtree.h" #include "ntoskrnl_private.h" #include "initguid.h" DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0); WINE_DEFAULT_DEBUG_CHANNEL(plugplay); #define MAX_SERVICE_NAME 260 struct device_interface { struct wine_rb_entry entry; UNICODE_STRING symbolic_link; DEVICE_OBJECT *device; GUID interface_class; BOOL enabled; }; static int interface_rb_compare( const void *key, const struct wine_rb_entry *entry) { const struct device_interface *iface = WINE_RB_ENTRY_VALUE( entry, const struct device_interface, entry ); const UNICODE_STRING *k = key; return RtlCompareUnicodeString( k, &iface->symbolic_link, FALSE ); } static struct wine_rb_tree device_interfaces = { interface_rb_compare }; static NTSTATUS WINAPI internal_complete( DEVICE_OBJECT *device, IRP *irp, void *context ) { HANDLE event = context; SetEvent( event ); return STATUS_MORE_PROCESSING_REQUIRED; } static NTSTATUS send_device_irp( DEVICE_OBJECT *device, IRP *irp, ULONG_PTR *info ) { HANDLE event = CreateEventA( NULL, FALSE, FALSE, NULL ); DEVICE_OBJECT *toplevel_device; NTSTATUS status; irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED; IoSetCompletionRoutine( irp, internal_complete, event, TRUE, TRUE, TRUE ); toplevel_device = IoGetAttachedDeviceReference( device ); status = IoCallDriver( toplevel_device, irp ); if (status == STATUS_PENDING) WaitForSingleObject( event, INFINITE ); status = irp->IoStatus.u.Status; if (info) *info = irp->IoStatus.Information; IoCompleteRequest( irp, IO_NO_INCREMENT ); ObDereferenceObject( toplevel_device ); CloseHandle( event ); return status; } static NTSTATUS get_device_id( DEVICE_OBJECT *device, BUS_QUERY_ID_TYPE type, WCHAR **id ) { IO_STACK_LOCATION *irpsp; IO_STATUS_BLOCK irp_status; IRP *irp; device = IoGetAttachedDevice( device ); if (!(irp = IoBuildSynchronousFsdRequest( IRP_MJ_PNP, device, NULL, 0, NULL, NULL, &irp_status ))) return STATUS_NO_MEMORY; irpsp = IoGetNextIrpStackLocation( irp ); irpsp->MinorFunction = IRP_MN_QUERY_ID; irpsp->Parameters.QueryId.IdType = type; return send_device_irp( device, irp, (ULONG_PTR *)id ); } static NTSTATUS send_pnp_irp( DEVICE_OBJECT *device, UCHAR minor ) { IO_STACK_LOCATION *irpsp; IO_STATUS_BLOCK irp_status; IRP *irp; device = IoGetAttachedDevice( device ); if (!(irp = IoBuildSynchronousFsdRequest( IRP_MJ_PNP, device, NULL, 0, NULL, NULL, &irp_status ))) return STATUS_NO_MEMORY; irpsp = IoGetNextIrpStackLocation( irp ); irpsp->MinorFunction = minor; irpsp->Parameters.StartDevice.AllocatedResources = NULL; irpsp->Parameters.StartDevice.AllocatedResourcesTranslated = NULL; return send_device_irp( device, irp, NULL ); } static NTSTATUS get_device_instance_id( DEVICE_OBJECT *device, WCHAR *buffer ) { static const WCHAR backslashW[] = {'\\',0}; NTSTATUS status; WCHAR *id; if ((status = get_device_id( device, BusQueryDeviceID, &id ))) { ERR("Failed to get device ID, status %#x.\n", status); return status; } lstrcpyW( buffer, id ); ExFreePool( id ); if ((status = get_device_id( device, BusQueryInstanceID, &id ))) { ERR("Failed to get instance ID, status %#x.\n", status); return status; } lstrcatW( buffer, backslashW ); lstrcatW( buffer, id ); ExFreePool( id ); TRACE("Returning ID %s.\n", debugstr_w(buffer)); return STATUS_SUCCESS; } static NTSTATUS send_power_irp( DEVICE_OBJECT *device, DEVICE_POWER_STATE power ) { IO_STATUS_BLOCK irp_status; IO_STACK_LOCATION *irpsp; IRP *irp; device = IoGetAttachedDevice( device ); if (!(irp = IoBuildSynchronousFsdRequest( IRP_MJ_POWER, device, NULL, 0, NULL, NULL, &irp_status ))) return STATUS_NO_MEMORY; irpsp = IoGetNextIrpStackLocation( irp ); irpsp->MinorFunction = IRP_MN_SET_POWER; irpsp->Parameters.Power.Type = DevicePowerState; irpsp->Parameters.Power.State.DeviceState = power; return send_device_irp( device, irp, NULL ); } static void load_function_driver( DEVICE_OBJECT *device, HDEVINFO set, SP_DEVINFO_DATA *sp_device ) { static const WCHAR driverW[] = {'\\','D','r','i','v','e','r','\\',0}; WCHAR buffer[MAX_SERVICE_NAME + ARRAY_SIZE(servicesW)]; WCHAR driver[MAX_SERVICE_NAME] = {0}; DRIVER_OBJECT *driver_obj; UNICODE_STRING string; NTSTATUS status; if (!SetupDiGetDeviceRegistryPropertyW( set, sp_device, SPDRP_SERVICE, NULL, (BYTE *)driver, sizeof(driver), NULL )) { WARN("No driver registered for device %p.\n", device); return; } lstrcpyW( buffer, servicesW ); lstrcatW( buffer, driver ); RtlInitUnicodeString( &string, buffer ); status = ZwLoadDriver( &string ); if (status != STATUS_SUCCESS && status != STATUS_IMAGE_ALREADY_LOADED) { ERR("Failed to load driver %s, status %#x.\n", debugstr_w(driver), status); return; } lstrcpyW( buffer, driverW ); lstrcatW( buffer, driver ); RtlInitUnicodeString( &string, buffer ); if (ObReferenceObjectByName( &string, OBJ_CASE_INSENSITIVE, NULL, 0, NULL, KernelMode, NULL, (void **)&driver_obj ) != STATUS_SUCCESS) { ERR("Failed to locate loaded driver %s.\n", debugstr_w(driver)); return; } TRACE("Calling AddDevice routine %p.\n", driver_obj->DriverExtension->AddDevice); if (driver_obj->DriverExtension->AddDevice) status = driver_obj->DriverExtension->AddDevice( driver_obj, device ); else status = STATUS_NOT_IMPLEMENTED; TRACE("AddDevice routine %p returned %#x.\n", driver_obj->DriverExtension->AddDevice, status); ObDereferenceObject( driver_obj ); if (status != STATUS_SUCCESS) ERR("AddDevice failed for driver %s, status %#x.\n", debugstr_w(driver), status); } /* Return the total number of characters in a REG_MULTI_SZ string, including * the final terminating null. */ static size_t sizeof_multiszW( const WCHAR *str ) { const WCHAR *p; for (p = str; *p; p += lstrlenW(p) + 1); return p + 1 - str; } /* This does almost the same thing as UpdateDriverForPlugAndPlayDevices(), * except that we don't know the INF path beforehand. */ static BOOL install_device_driver( DEVICE_OBJECT *device, HDEVINFO set, SP_DEVINFO_DATA *sp_device ) { static const DWORD dif_list[] = { DIF_REGISTERDEVICE, DIF_SELECTBESTCOMPATDRV, DIF_ALLOW_INSTALL, DIF_INSTALLDEVICEFILES, DIF_REGISTER_COINSTALLERS, DIF_INSTALLINTERFACES, DIF_INSTALLDEVICE, DIF_NEWDEVICEWIZARD_FINISHINSTALL, }; NTSTATUS status; unsigned int i; WCHAR *ids; if ((status = get_device_id( device, BusQueryHardwareIDs, &ids )) || !ids) { ERR("Failed to get hardware IDs, status %#x.\n", status); return FALSE; } SetupDiSetDeviceRegistryPropertyW( set, sp_device, SPDRP_HARDWAREID, (BYTE *)ids, sizeof_multiszW( ids ) * sizeof(WCHAR) ); ExFreePool( ids ); if ((status = get_device_id( device, BusQueryCompatibleIDs, &ids )) || !ids) { ERR("Failed to get compatible IDs, status %#x.\n", status); return FALSE; } SetupDiSetDeviceRegistryPropertyW( set, sp_device, SPDRP_COMPATIBLEIDS, (BYTE *)ids, sizeof_multiszW( ids ) * sizeof(WCHAR) ); ExFreePool( ids ); if (!SetupDiBuildDriverInfoList( set, sp_device, SPDIT_COMPATDRIVER )) { ERR("Failed to build compatible driver list, error %#x.\n", GetLastError()); return FALSE; } for (i = 0; i < ARRAY_SIZE(dif_list); ++i) { if (!SetupDiCallClassInstaller(dif_list[i], set, sp_device) && GetLastError() != ERROR_DI_DO_DEFAULT) { ERR("Install function %#x failed, error %#x.\n", dif_list[i], GetLastError()); return FALSE; } } return TRUE; } /* Load the function driver for a newly created PDO, if one is present, and * send IRPs to start the device. */ static void start_device( DEVICE_OBJECT *device, HDEVINFO set, SP_DEVINFO_DATA *sp_device ) { load_function_driver( device, set, sp_device ); if (device->DriverObject) { send_pnp_irp( device, IRP_MN_START_DEVICE ); send_power_irp( device, PowerDeviceD0 ); } } static void handle_bus_relations( DEVICE_OBJECT *device ) { static const WCHAR infpathW[] = {'I','n','f','P','a','t','h',0}; SP_DEVINFO_DATA sp_device = {sizeof(sp_device)}; WCHAR device_instance_id[MAX_DEVICE_ID_LEN]; BOOL need_driver = TRUE; HDEVINFO set; HKEY key; /* We could (should?) do a full IRP_MN_QUERY_DEVICE_RELATIONS query, * but we don't have to, we have the DEVICE_OBJECT of the new device * so we can simply handle the process here */ if (get_device_instance_id( device, device_instance_id )) return; set = SetupDiCreateDeviceInfoList( NULL, NULL ); if (!SetupDiCreateDeviceInfoW( set, device_instance_id, &GUID_NULL, NULL, NULL, 0, &sp_device ) && !SetupDiOpenDeviceInfoW( set, device_instance_id, NULL, 0, &sp_device )) { ERR("Failed to create or open device %s, error %#x.\n", debugstr_w(device_instance_id), GetLastError()); SetupDiDestroyDeviceInfoList( set ); return; } TRACE("Creating new device %s.\n", debugstr_w(device_instance_id)); /* Check if the device already has a driver registered; if not, find one * and install it. */ key = SetupDiOpenDevRegKey( set, &sp_device, DICS_FLAG_GLOBAL, 0, DIREG_DRV, KEY_READ ); if (key != INVALID_HANDLE_VALUE) { if (!RegQueryValueExW( key, infpathW, NULL, NULL, NULL, NULL )) need_driver = FALSE; RegCloseKey( key ); } if (need_driver && !install_device_driver( device, set, &sp_device )) { SetupDiDestroyDeviceInfoList( set ); return; } start_device( device, set, &sp_device ); SetupDiDestroyDeviceInfoList( set ); } static void remove_device( DEVICE_OBJECT *device ) { TRACE("Removing device %p.\n", device); send_power_irp( device, PowerDeviceD3 ); send_pnp_irp( device, IRP_MN_SURPRISE_REMOVAL ); send_pnp_irp( device, IRP_MN_REMOVE_DEVICE ); } /*********************************************************************** * IoInvalidateDeviceRelations (NTOSKRNL.EXE.@) */ void WINAPI IoInvalidateDeviceRelations( DEVICE_OBJECT *device_object, DEVICE_RELATION_TYPE type ) { TRACE("device %p, type %#x.\n", device_object, type); switch (type) { case BusRelations: handle_bus_relations( device_object ); break; case RemovalRelations: remove_device( device_object ); break; default: FIXME("Unhandled relation %#x.\n", type); break; } } /*********************************************************************** * IoGetDeviceProperty (NTOSKRNL.EXE.@) */ NTSTATUS WINAPI IoGetDeviceProperty( DEVICE_OBJECT *device, DEVICE_REGISTRY_PROPERTY property, ULONG length, void *buffer, ULONG *needed ) { SP_DEVINFO_DATA sp_device = {sizeof(sp_device)}; WCHAR device_instance_id[MAX_DEVICE_ID_LEN]; DWORD sp_property = -1; NTSTATUS status; HDEVINFO set; TRACE("device %p, property %u, length %u, buffer %p, needed %p.\n", device, property, length, buffer, needed); switch (property) { case DevicePropertyEnumeratorName: { WCHAR *id, *ptr; status = get_device_id( device, BusQueryInstanceID, &id ); if (status != STATUS_SUCCESS) { ERR("Failed to get instance ID, status %#x.\n", status); break; } wcsupr( id ); ptr = wcschr( id, '\\' ); if (ptr) *ptr = 0; *needed = sizeof(WCHAR) * (lstrlenW(id) + 1); if (length >= *needed) memcpy( buffer, id, *needed ); else status = STATUS_BUFFER_TOO_SMALL; ExFreePool( id ); return status; } case DevicePropertyPhysicalDeviceObjectName: { ULONG used_len, len = length + sizeof(OBJECT_NAME_INFORMATION); OBJECT_NAME_INFORMATION *name = HeapAlloc(GetProcessHeap(), 0, len); HANDLE handle; status = ObOpenObjectByPointer( device, OBJ_KERNEL_HANDLE, NULL, 0, NULL, KernelMode, &handle ); if (!status) { status = NtQueryObject( handle, ObjectNameInformation, name, len, &used_len ); NtClose( handle ); } if (status == STATUS_SUCCESS) { /* Ensure room for NULL termination */ if (length >= name->Name.MaximumLength) memcpy(buffer, name->Name.Buffer, name->Name.MaximumLength); else status = STATUS_BUFFER_TOO_SMALL; *needed = name->Name.MaximumLength; } else { if (status == STATUS_INFO_LENGTH_MISMATCH || status == STATUS_BUFFER_OVERFLOW) { status = STATUS_BUFFER_TOO_SMALL; *needed = used_len - sizeof(OBJECT_NAME_INFORMATION); } else *needed = 0; } HeapFree(GetProcessHeap(), 0, name); return status; } case DevicePropertyDeviceDescription: sp_property = SPDRP_DEVICEDESC; break; case DevicePropertyHardwareID: sp_property = SPDRP_HARDWAREID; break; case DevicePropertyCompatibleIDs: sp_property = SPDRP_COMPATIBLEIDS; break; case DevicePropertyClassName: sp_property = SPDRP_CLASS; break; case DevicePropertyClassGuid: sp_property = SPDRP_CLASSGUID; break; case DevicePropertyManufacturer: sp_property = SPDRP_MFG; break; case DevicePropertyFriendlyName: sp_property = SPDRP_FRIENDLYNAME; break; case DevicePropertyLocationInformation: sp_property = SPDRP_LOCATION_INFORMATION; break; case DevicePropertyBusTypeGuid: sp_property = SPDRP_BUSTYPEGUID; break; case DevicePropertyLegacyBusType: sp_property = SPDRP_LEGACYBUSTYPE; break; case DevicePropertyBusNumber: sp_property = SPDRP_BUSNUMBER; break; case DevicePropertyAddress: sp_property = SPDRP_ADDRESS; break; case DevicePropertyUINumber: sp_property = SPDRP_UI_NUMBER; break; case DevicePropertyInstallState: sp_property = SPDRP_INSTALL_STATE; break; case DevicePropertyRemovalPolicy: sp_property = SPDRP_REMOVAL_POLICY; break; default: FIXME("Unhandled property %u.\n", property); return STATUS_NOT_IMPLEMENTED; } if ((status = get_device_instance_id( device, device_instance_id ))) return status; if ((set = SetupDiCreateDeviceInfoList( &GUID_NULL, NULL )) == INVALID_HANDLE_VALUE) { ERR("Failed to create device list, error %#x.\n", GetLastError()); return GetLastError(); } if (!SetupDiOpenDeviceInfoW( set, device_instance_id, NULL, 0, &sp_device)) { ERR("Failed to open device, error %#x.\n", GetLastError()); SetupDiDestroyDeviceInfoList( set ); return GetLastError(); } if (SetupDiGetDeviceRegistryPropertyW( set, &sp_device, sp_property, NULL, buffer, length, needed )) status = STATUS_SUCCESS; else status = GetLastError(); SetupDiDestroyDeviceInfoList( set ); return status; } static NTSTATUS create_device_symlink( DEVICE_OBJECT *device, UNICODE_STRING *symlink_name ) { UNICODE_STRING device_nameU; WCHAR *device_name; ULONG len = 0; NTSTATUS ret; ret = IoGetDeviceProperty( device, DevicePropertyPhysicalDeviceObjectName, 0, NULL, &len ); if (ret != STATUS_BUFFER_TOO_SMALL) return ret; device_name = heap_alloc( len ); ret = IoGetDeviceProperty( device, DevicePropertyPhysicalDeviceObjectName, len, device_name, &len ); if (ret) { heap_free( device_name ); return ret; } RtlInitUnicodeString( &device_nameU, device_name ); ret = IoCreateSymbolicLink( symlink_name, &device_nameU ); heap_free( device_name ); return ret; } /*********************************************************************** * IoSetDeviceInterfaceState (NTOSKRNL.EXE.@) */ NTSTATUS WINAPI IoSetDeviceInterfaceState( UNICODE_STRING *name, BOOLEAN enable ) { static const WCHAR DeviceClassesW[] = {'\\','R','E','G','I','S','T','R','Y','\\', 'M','a','c','h','i','n','e','\\','S','y','s','t','e','m','\\', 'C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t','\\', 'C','o','n','t','r','o','l','\\', 'D','e','v','i','c','e','C','l','a','s','s','e','s','\\',0}; static const WCHAR controlW[] = {'C','o','n','t','r','o','l',0}; static const WCHAR linkedW[] = {'L','i','n','k','e','d',0}; static const WCHAR slashW[] = {'\\',0}; static const WCHAR hashW[] = {'#',0}; size_t namelen = name->Length / sizeof(WCHAR); DEV_BROADCAST_DEVICEINTERFACE_W *broadcast; struct device_interface *iface; HANDLE iface_key, control_key; OBJECT_ATTRIBUTES attr = {0}; struct wine_rb_entry *entry; WCHAR *path, *refstr, *p; UNICODE_STRING string; DWORD data = enable; NTSTATUS ret; ULONG len; TRACE("device %s, enable %u.\n", debugstr_us(name), enable); entry = wine_rb_get( &device_interfaces, name ); if (!entry) return STATUS_OBJECT_NAME_NOT_FOUND; iface = WINE_RB_ENTRY_VALUE( entry, struct device_interface, entry ); if (!enable && !iface->enabled) return STATUS_OBJECT_NAME_NOT_FOUND; if (enable && iface->enabled) return STATUS_OBJECT_NAME_EXISTS; for (p = name->Buffer + 4, refstr = NULL; p < name->Buffer + namelen; p++) if (*p == '\\') refstr = p; if (!refstr) refstr = p; len = lstrlenW(DeviceClassesW) + 38 + 1 + namelen + 2 + 1; if (!(path = heap_alloc( len * sizeof(WCHAR) ))) return STATUS_NO_MEMORY; lstrcpyW( path, DeviceClassesW ); lstrcpynW( path + lstrlenW( path ), refstr - 38, 39 ); lstrcatW( path, slashW ); p = path + lstrlenW( path ); lstrcpynW( path + lstrlenW( path ), name->Buffer, (refstr - name->Buffer) + 1 ); p[0] = p[1] = p[3] = '#'; lstrcatW( path, slashW ); lstrcatW( path, hashW ); if (refstr < name->Buffer + namelen) lstrcpynW( path + lstrlenW( path ), refstr, name->Buffer + namelen - refstr + 1 ); attr.Length = sizeof(attr); attr.ObjectName = &string; RtlInitUnicodeString( &string, path ); ret = NtOpenKey( &iface_key, KEY_CREATE_SUB_KEY, &attr ); heap_free(path); if (ret) return ret; attr.RootDirectory = iface_key; RtlInitUnicodeString( &string, controlW ); ret = NtCreateKey( &control_key, KEY_SET_VALUE, &attr, 0, NULL, 0, NULL ); NtClose( iface_key ); if (ret) return ret; RtlInitUnicodeString( &string, linkedW ); ret = NtSetValueKey( control_key, &string, 0, REG_DWORD, &data, sizeof(data) ); if (ret) { NtClose( control_key ); return ret; } if (enable) ret = create_device_symlink( iface->device, name ); else ret = IoDeleteSymbolicLink( name ); if (ret) { NtDeleteValueKey( control_key, &string ); NtClose( control_key ); return ret; } iface->enabled = enable; len = offsetof(DEV_BROADCAST_DEVICEINTERFACE_W, dbcc_name[namelen + 1]); if ((broadcast = heap_alloc( len ))) { broadcast->dbcc_size = len; broadcast->dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE; broadcast->dbcc_classguid = iface->interface_class; lstrcpynW( broadcast->dbcc_name, name->Buffer, namelen + 1 ); BroadcastSystemMessageW( BSF_FORCEIFHUNG | BSF_QUERY, NULL, WM_DEVICECHANGE, enable ? DBT_DEVICEARRIVAL : DBT_DEVICEREMOVECOMPLETE, (LPARAM)broadcast ); heap_free( broadcast ); } return ret; } /*********************************************************************** * IoRegisterDeviceInterface (NTOSKRNL.EXE.@) */ NTSTATUS WINAPI IoRegisterDeviceInterface(DEVICE_OBJECT *device, const GUID *class_guid, UNICODE_STRING *refstr, UNICODE_STRING *symbolic_link) { SP_DEVICE_INTERFACE_DATA sp_iface = {sizeof(sp_iface)}; SP_DEVINFO_DATA sp_device = {sizeof(sp_device)}; WCHAR device_instance_id[MAX_DEVICE_ID_LEN]; SP_DEVICE_INTERFACE_DETAIL_DATA_W *data; NTSTATUS status = STATUS_SUCCESS; UNICODE_STRING device_path; struct device_interface *iface; struct wine_rb_entry *entry; DWORD required; HDEVINFO set; TRACE("device %p, class_guid %s, refstr %s, symbolic_link %p.\n", device, debugstr_guid(class_guid), debugstr_us(refstr), symbolic_link); if ((status = get_device_instance_id( device, device_instance_id ))) return status; set = SetupDiGetClassDevsW( class_guid, NULL, NULL, DIGCF_DEVICEINTERFACE ); if (set == INVALID_HANDLE_VALUE) return STATUS_UNSUCCESSFUL; if (!SetupDiCreateDeviceInfoW( set, device_instance_id, class_guid, NULL, NULL, 0, &sp_device ) && !SetupDiOpenDeviceInfoW( set, device_instance_id, NULL, 0, &sp_device )) { ERR("Failed to create device %s, error %#x.\n", debugstr_w(device_instance_id), GetLastError()); return GetLastError(); } if (!SetupDiCreateDeviceInterfaceW( set, &sp_device, class_guid, refstr ? refstr->Buffer : NULL, 0, &sp_iface )) return STATUS_UNSUCCESSFUL; required = 0; SetupDiGetDeviceInterfaceDetailW( set, &sp_iface, NULL, 0, &required, NULL ); if (required == 0) return STATUS_UNSUCCESSFUL; data = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, required ); data->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA_W); if (!SetupDiGetDeviceInterfaceDetailW( set, &sp_iface, data, required, NULL, NULL )) { HeapFree( GetProcessHeap(), 0, data ); return STATUS_UNSUCCESSFUL; } data->DevicePath[1] = '?'; TRACE("Returning path %s.\n", debugstr_w(data->DevicePath)); RtlCreateUnicodeString( &device_path, data->DevicePath); entry = wine_rb_get( &device_interfaces, &device_path ); if (entry) { iface = WINE_RB_ENTRY_VALUE( entry, struct device_interface, entry ); if (iface->enabled) ERR("Device interface %s is still enabled.\n", debugstr_us(&iface->symbolic_link)); } else { iface = heap_alloc_zero( sizeof(struct device_interface) ); RtlCreateUnicodeString(&iface->symbolic_link, data->DevicePath); if (wine_rb_put( &device_interfaces, &iface->symbolic_link, &iface->entry )) ERR("Failed to insert interface %s into tree.\n", debugstr_us(&iface->symbolic_link)); } iface->device = device; iface->interface_class = *class_guid; if (symbolic_link) RtlCreateUnicodeString( symbolic_link, data->DevicePath); HeapFree( GetProcessHeap(), 0, data ); RtlFreeUnicodeString( &device_path ); return status; } /*********************************************************************** * PoSetPowerState (NTOSKRNL.EXE.@) */ POWER_STATE WINAPI PoSetPowerState( DEVICE_OBJECT *device, POWER_STATE_TYPE type, POWER_STATE state) { FIXME("device %p, type %u, state %u, stub!\n", device, type, state.DeviceState); return state; } /***************************************************** * PoStartNextPowerIrp (NTOSKRNL.EXE.@) */ void WINAPI PoStartNextPowerIrp( IRP *irp ) { FIXME("irp %p, stub!\n", irp); } /***************************************************** * PoCallDriver (NTOSKRNL.EXE.@) */ NTSTATUS WINAPI PoCallDriver( DEVICE_OBJECT *device, IRP *irp ) { TRACE("device %p, irp %p.\n", device, irp); return IoCallDriver( device, irp ); } static DRIVER_OBJECT *pnp_manager; struct root_pnp_device { WCHAR id[MAX_DEVICE_ID_LEN]; struct wine_rb_entry entry; DEVICE_OBJECT *device; }; static int root_pnp_devices_rb_compare( const void *key, const struct wine_rb_entry *entry ) { const struct root_pnp_device *device = WINE_RB_ENTRY_VALUE( entry, const struct root_pnp_device, entry ); const WCHAR *k = key; return wcsicmp( k, device->id ); } static struct wine_rb_tree root_pnp_devices = { root_pnp_devices_rb_compare }; static NTSTATUS WINAPI pnp_manager_device_pnp( DEVICE_OBJECT *device, IRP *irp ) { IO_STACK_LOCATION *stack = IoGetCurrentIrpStackLocation( irp ); struct root_pnp_device *root_device = device->DeviceExtension; NTSTATUS status; TRACE("device %p, irp %p, minor function %#x.\n", device, irp, stack->MinorFunction); switch (stack->MinorFunction) { case IRP_MN_QUERY_DEVICE_RELATIONS: /* The FDO above already handled this, so return the same status. */ break; case IRP_MN_START_DEVICE: case IRP_MN_SURPRISE_REMOVAL: case IRP_MN_REMOVE_DEVICE: /* Nothing to do. */ irp->IoStatus.u.Status = STATUS_SUCCESS; break; case IRP_MN_QUERY_CAPABILITIES: irp->IoStatus.u.Status = STATUS_SUCCESS; break; case IRP_MN_QUERY_ID: { BUS_QUERY_ID_TYPE type = stack->Parameters.QueryId.IdType; WCHAR *id, *p; TRACE("Received IRP_MN_QUERY_ID, type %#x.\n", type); switch (type) { case BusQueryDeviceID: p = wcsrchr( root_device->id, '\\' ); if ((id = ExAllocatePool( NonPagedPool, (p - root_device->id + 1) * sizeof(WCHAR) ))) { memcpy( id, root_device->id, (p - root_device->id) * sizeof(WCHAR) ); id[p - root_device->id] = 0; irp->IoStatus.Information = (ULONG_PTR)id; irp->IoStatus.u.Status = STATUS_SUCCESS; } else { irp->IoStatus.Information = 0; irp->IoStatus.u.Status = STATUS_NO_MEMORY; } break; case BusQueryInstanceID: p = wcsrchr( root_device->id, '\\' ); if ((id = ExAllocatePool( NonPagedPool, (wcslen( p + 1 ) + 1) * sizeof(WCHAR) ))) { wcscpy( id, p + 1 ); irp->IoStatus.Information = (ULONG_PTR)id; irp->IoStatus.u.Status = STATUS_SUCCESS; } else { irp->IoStatus.Information = 0; irp->IoStatus.u.Status = STATUS_NO_MEMORY; } break; default: FIXME("Unhandled IRP_MN_QUERY_ID type %#x.\n", type); } break; } default: FIXME("Unhandled PnP request %#x.\n", stack->MinorFunction); } status = irp->IoStatus.u.Status; IoCompleteRequest( irp, IO_NO_INCREMENT ); return status; } static NTSTATUS WINAPI pnp_manager_driver_entry( DRIVER_OBJECT *driver, UNICODE_STRING *keypath ) { pnp_manager = driver; driver->MajorFunction[IRP_MJ_PNP] = pnp_manager_device_pnp; return STATUS_SUCCESS; } void pnp_manager_start(void) { static const WCHAR driver_nameW[] = {'\\','D','r','i','v','e','r','\\','P','n','p','M','a','n','a','g','e','r',0}; UNICODE_STRING driver_nameU; NTSTATUS status; RtlInitUnicodeString( &driver_nameU, driver_nameW ); if ((status = IoCreateDriver( &driver_nameU, pnp_manager_driver_entry ))) ERR("Failed to create PnP manager driver, status %#x.\n", status); } static void destroy_root_pnp_device( struct wine_rb_entry *entry, void *context ) { struct root_pnp_device *device = WINE_RB_ENTRY_VALUE(entry, struct root_pnp_device, entry); remove_device( device->device ); } void pnp_manager_stop(void) { wine_rb_destroy( &root_pnp_devices, destroy_root_pnp_device, NULL ); IoDeleteDriver( pnp_manager ); } void pnp_manager_enumerate_root_devices( const WCHAR *driver_name ) { static const WCHAR driverW[] = {'\\','D','r','i','v','e','r','\\',0}; static const WCHAR rootW[] = {'R','O','O','T',0}; WCHAR buffer[MAX_SERVICE_NAME + ARRAY_SIZE(driverW)], id[MAX_DEVICE_ID_LEN]; SP_DEVINFO_DATA sp_device = {sizeof(sp_device)}; struct root_pnp_device *pnp_device; DEVICE_OBJECT *device; NTSTATUS status; unsigned int i; HDEVINFO set; TRACE("Searching for new root-enumerated devices for driver %s.\n", debugstr_w(driver_name)); set = SetupDiGetClassDevsW( NULL, rootW, NULL, DIGCF_ALLCLASSES ); if (set == INVALID_HANDLE_VALUE) { ERR("Failed to build device set, error %#x.\n", GetLastError()); return; } for (i = 0; SetupDiEnumDeviceInfo( set, i, &sp_device ); ++i) { if (!SetupDiGetDeviceRegistryPropertyW( set, &sp_device, SPDRP_SERVICE, NULL, (BYTE *)buffer, sizeof(buffer), NULL ) || lstrcmpiW( buffer, driver_name )) { continue; } SetupDiGetDeviceInstanceIdW( set, &sp_device, id, ARRAY_SIZE(id), NULL ); if (wine_rb_get( &root_pnp_devices, id )) continue; TRACE("Adding new root-enumerated device %s.\n", debugstr_w(id)); if ((status = IoCreateDevice( pnp_manager, sizeof(struct root_pnp_device), NULL, FILE_DEVICE_CONTROLLER, FILE_AUTOGENERATED_DEVICE_NAME, FALSE, &device ))) { ERR("Failed to create root-enumerated PnP device %s, status %#x.\n", debugstr_w(id), status); continue; } pnp_device = device->DeviceExtension; wcscpy( pnp_device->id, id ); pnp_device->device = device; if (wine_rb_put( &root_pnp_devices, id, &pnp_device->entry )) { ERR("Failed to insert device %s into tree.\n", debugstr_w(id)); IoDeleteDevice( device ); continue; } start_device( device, set, &sp_device ); } SetupDiDestroyDeviceInfoList(set); }