[2/3] winebus.sys: Implement IRP_MN_QUERY_ID. (v6)
Sebastian Lackner
sebastian at fds-team.de
Wed Sep 14 02:29:50 CDT 2016
From: Aric Stewart <aric at codeweavers.com>
Signed-off-by: Aric Stewart <aric at codeweavers.com>
Signed-off-by: Sebastian Lackner <sebastian at fds-team.de>
---
Rebasing was done by Aric.
emaining changes in v6:
* Remove unnecessary includes.
* Move code for compatible ids into a helper function.
* Fix leak of iid / did pointer.
* Set status to STATUS_NO_MEMORY when memory allocation fails.
* Add some more traces for the different IRP_MN_QUERY_ID types.
dlls/winebus.sys/main.c | 83 ++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 83 insertions(+)
diff --git a/dlls/winebus.sys/main.c b/dlls/winebus.sys/main.c
index 9c0f5aa..1364eab 100644
--- a/dlls/winebus.sys/main.c
+++ b/dlls/winebus.sys/main.c
@@ -111,6 +111,52 @@ static WCHAR *get_instance_id(DEVICE_OBJECT *device)
return dst;
}
+static WCHAR *get_device_id(DEVICE_OBJECT *device)
+{
+ static const WCHAR formatW[] = {'%','s','\\','V','i','d','_','%','0','4','x','&','P','i','d','_','%','0','4','x',0};
+ struct device_extension *ext = (struct device_extension *)device->DeviceExtension;
+ DWORD len = strlenW(ext->busid) + 19;
+ WCHAR *dst;
+
+ if ((dst = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR))))
+ sprintfW(dst, formatW, ext->busid, ext->vid, ext->pid);
+
+ return dst;
+}
+
+static WCHAR *get_compatible_ids(DEVICE_OBJECT *device)
+{
+ struct device_extension *ext = (struct device_extension *)device->DeviceExtension;
+ WCHAR *iid, *did, *dst, *ptr;
+ DWORD len;
+
+ if (!(iid = get_instance_id(device)))
+ return NULL;
+
+ if (!(did = get_device_id(device)))
+ {
+ HeapFree(GetProcessHeap(), 0, iid);
+ return NULL;
+ }
+
+ len = strlenW(iid) + strlenW(did) + strlenW(ext->busid) + 4;
+ if ((dst = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR))))
+ {
+ ptr = dst;
+ strcpyW(ptr, iid);
+ ptr += strlenW(iid) + 1;
+ strcpyW(ptr, did);
+ ptr += strlenW(did) + 1;
+ strcpyW(ptr, ext->busid);
+ ptr += strlenW(ext->busid) + 1;
+ *ptr = 0;
+ }
+
+ HeapFree(GetProcessHeap(), 0, iid);
+ HeapFree(GetProcessHeap(), 0, did);
+ return dst;
+}
+
DEVICE_OBJECT *bus_create_hid_device(DRIVER_OBJECT *driver, const WCHAR *busidW, void *native, WORD vid,
WORD pid, DWORD version, DWORD uid, const WCHAR *serialW, BOOL is_gamepad,
const GUID *class)
@@ -184,6 +230,41 @@ DEVICE_OBJECT *bus_create_hid_device(DRIVER_OBJECT *driver, const WCHAR *busidW,
return device;
}
+static NTSTATUS handle_IRP_MN_QUERY_ID(DEVICE_OBJECT *device, IRP *irp)
+{
+ NTSTATUS status = irp->IoStatus.u.Status;
+ IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp );
+ BUS_QUERY_ID_TYPE type = irpsp->Parameters.QueryId.IdType;
+
+ TRACE("(%p, %p)\n", device, irp);
+
+ switch (type)
+ {
+ case BusQueryHardwareIDs:
+ TRACE("BusQueryHardwareIDs\n");
+ irp->IoStatus.Information = (ULONG_PTR)get_compatible_ids(device);
+ break;
+ case BusQueryCompatibleIDs:
+ TRACE("BusQueryCompatibleIDs\n");
+ irp->IoStatus.Information = (ULONG_PTR)get_compatible_ids(device);
+ break;
+ case BusQueryDeviceID:
+ TRACE("BusQueryDeviceID\n");
+ irp->IoStatus.Information = (ULONG_PTR)get_device_id(device);
+ break;
+ case BusQueryInstanceID:
+ TRACE("BusQueryInstanceID\n");
+ irp->IoStatus.Information = (ULONG_PTR)get_instance_id(device);
+ break;
+ default:
+ FIXME("Unhandled type %08x\n", type);
+ return status;
+ }
+
+ status = irp->IoStatus.Information ? STATUS_SUCCESS : STATUS_NO_MEMORY;
+ return status;
+}
+
NTSTATUS WINAPI common_pnp_dispatch(DEVICE_OBJECT *device, IRP *irp)
{
NTSTATUS status = irp->IoStatus.u.Status;
@@ -196,6 +277,8 @@ NTSTATUS WINAPI common_pnp_dispatch(DEVICE_OBJECT *device, IRP *irp)
break;
case IRP_MN_QUERY_ID:
TRACE("IRP_MN_QUERY_ID\n");
+ status = handle_IRP_MN_QUERY_ID(device, irp);
+ irp->IoStatus.u.Status = status;
break;
case IRP_MN_QUERY_CAPABILITIES:
TRACE("IRP_MN_QUERY_CAPABILITIES\n");
--
2.9.0
More information about the wine-patches
mailing list