Signed-off-by: Andrew Eikum <aeikum(a)codeweavers.com>
---
dlls/wbemprox/Makefile.in | 2 +-
dlls/wbemprox/builtin.c | 59 +++++++++++++++++++++++++++++++++++++
dlls/wbemprox/tests/query.c | 51 ++++++++++++++++++++++++++++++++
3 files changed, 111 insertions(+), 1 deletion(-)
diff --git a/dlls/wbemprox/Makefile.in b/dlls/wbemprox/Makefile.in
index f411898dcd..a86893a45c 100644
--- a/dlls/wbemprox/Makefile.in
+++ b/dlls/wbemprox/Makefile.in
@@ -1,5 +1,5 @@
MODULE = wbemprox.dll
-IMPORTS = winspool version iphlpapi dxgi oleaut32 ole32 advapi32 user32 gdi32 ws2_32 rpcrt4
+IMPORTS = winspool version iphlpapi dxgi oleaut32 ole32 advapi32 user32 gdi32 ws2_32 rpcrt4 setupapi
C_SRCS = \
builtin.c \
diff --git a/dlls/wbemprox/builtin.c b/dlls/wbemprox/builtin.c
index 73f4192779..2c472edeec 100644
--- a/dlls/wbemprox/builtin.c
+++ b/dlls/wbemprox/builtin.c
@@ -58,6 +58,7 @@
#include "sddl.h"
#include "ntsecapi.h"
#include "winspool.h"
+#include "setupapi.h"
#include "wine/debug.h"
#include "wbemprox_private.h"
@@ -103,6 +104,8 @@ static const WCHAR class_physicalmediaW[] =
{'W','i','n','3','2','_','P','h','y','s','i','c','a','l','M','e','d','i','a',0};
static const WCHAR class_physicalmemoryW[] =
{'W','i','n','3','2','_','P','h','y','s','i','c','a','l','M','e','m','o','r','y',0};
+static const WCHAR class_pnpentityW[] =
+ {'W','i','n','3','2','_','P','n','P','E','n','t','i','t','y',0};
static const WCHAR class_printerW[] =
{'W','i','n','3','2','_','P','r','i','n','t','e','r',0};
static const WCHAR class_process_getowner_outW[] =
@@ -605,6 +608,10 @@ static const struct column col_physicalmemory[] =
{ prop_capacityW, CIM_UINT64 },
{ prop_memorytypeW, CIM_UINT16, VT_I4 }
};
+static const struct column col_pnpentity[] =
+{
+ { prop_deviceidW, CIM_STRING|COL_FLAG_DYNAMIC },
+};
static const struct column col_printer[] =
{
{ prop_attributesW, CIM_UINT32 },
@@ -1022,6 +1029,10 @@ struct record_physicalmemory
UINT64 capacity;
UINT16 memorytype;
};
+struct record_pnpentity
+{
+ const WCHAR *device_id;
+};
struct record_printer
{
UINT32 attributes;
@@ -2513,6 +2524,53 @@ static enum fill_status fill_physicalmemory( struct table *table, const struct e
return status;
}
+static enum fill_status fill_pnpentity( struct table *table, const struct expr *cond )
+{
+ struct record_pnpentity *rec;
+ enum fill_status status = FILL_STATUS_UNFILTERED;
+ HDEVINFO device_info_set;
+ SP_DEVINFO_DATA devinfo = {0};
+ DWORD idx;
+
+ device_info_set = SetupDiGetClassDevsW( NULL, NULL, NULL, DIGCF_ALLCLASSES|DIGCF_PRESENT );
+
+ devinfo.cbSize = sizeof(devinfo);
+
+ idx = 0;
+ while (SetupDiEnumDeviceInfo( device_info_set, idx++, &devinfo ))
+ {
+ /* noop */
+ }
+
+ resize_table( table, idx, sizeof(*rec) );
+ table->num_rows = 0;
+ rec = (struct record_pnpentity *)table->data;
+
+ idx = 0;
+ while (SetupDiEnumDeviceInfo( device_info_set, idx++, &devinfo ))
+ {
+ WCHAR device_id[MAX_PATH];
+ if (SetupDiGetDeviceInstanceIdW( device_info_set, &devinfo, device_id,
+ ARRAY_SIZE(device_id), NULL ))
+ {
+ rec->device_id = heap_strdupW( device_id );
+
+ table->num_rows++;
+ if (!match_row( table, table->num_rows - 1, cond, &status ))
+ {
+ free_row_values( table, table->num_rows - 1 );
+ table->num_rows--;
+ }
+ else
+ rec++;
+ }
+ }
+
+ SetupDiDestroyDeviceInfoList( device_info_set );
+
+ return status;
+}
+
static enum fill_status fill_printer( struct table *table, const struct expr *cond )
{
static const WCHAR fmtW[] = {'P','r','i','n','t','e','r','%','d',0};
@@ -3382,6 +3440,7 @@ static struct table builtin_classes[] =
{ class_paramsW, SIZEOF(col_param), col_param, SIZEOF(data_param), 0, (BYTE *)data_param },
{ class_physicalmediaW, SIZEOF(col_physicalmedia), col_physicalmedia, SIZEOF(data_physicalmedia), 0, (BYTE *)data_physicalmedia },
{ class_physicalmemoryW, SIZEOF(col_physicalmemory), col_physicalmemory, 0, 0, NULL, fill_physicalmemory },
+ { class_pnpentityW, SIZEOF(col_pnpentity), col_pnpentity, 0, 0, NULL, fill_pnpentity },
{ class_printerW, SIZEOF(col_printer), col_printer, 0, 0, NULL, fill_printer },
{ class_processW, SIZEOF(col_process), col_process, 0, 0, NULL, fill_process },
{ class_processorW, SIZEOF(col_processor), col_processor, 0, 0, NULL, fill_processor },
diff --git a/dlls/wbemprox/tests/query.c b/dlls/wbemprox/tests/query.c
index 87b42daeca..39b6ba9c95 100644
--- a/dlls/wbemprox/tests/query.c
+++ b/dlls/wbemprox/tests/query.c
@@ -1709,6 +1709,56 @@ static void test_Win32_Printer( IWbemServices *services )
SysFreeString( wql );
}
+static void test_Win32_PnPEntity( IWbemServices *services )
+{
+ HRESULT hr;
+ IEnumWbemClassObject *enm;
+ IWbemClassObject *obj;
+ VARIANT val;
+ CIMTYPE type;
+ ULONG count, i;
+ BSTR bstr;
+
+ static WCHAR win32_pnpentityW[] = {'W','i','n','3','2','_','P','n','P','E','n','t','i','t','y',0};
+ static const WCHAR deviceidW[] = {'D','e','v','i','c','e','I','d',0};
+
+ bstr = SysAllocString( win32_pnpentityW );
+
+ hr = IWbemServices_CreateInstanceEnum( services, bstr, 0, NULL, &enm );
+ ok( hr == S_OK, "got %08x\n", hr );
+
+ SysFreeString( bstr );
+ bstr = SysAllocString( deviceidW );
+
+ while (1)
+ {
+ hr = IEnumWbemClassObject_Next( enm, 1000, 1, &obj, &count );
+ ok( (count == 1 && (hr == WBEM_S_FALSE || hr == WBEM_S_NO_ERROR)) ||
+ (count == 0 && (hr == WBEM_S_FALSE || hr == WBEM_S_TIMEDOUT)),
+ "got %08x with %u objects returned\n", hr, count );
+
+ if (count == 0)
+ break;
+
+ for (i = 0; i < count; ++i)
+ {
+ hr = IWbemClassObject_Get( obj, bstr, 0, &val, &type, NULL );
+ ok( hr == S_OK, "got %08x\n", hr );
+
+ if (SUCCEEDED( hr ))
+ {
+ ok( V_VT( &val ) == VT_BSTR, "unexpected variant type 0x%x\n", V_VT( &val ) );
+ ok( type == CIM_STRING, "unexpected type 0x%x\n", type );
+ VariantClear( &val );
+ }
+ }
+ }
+
+ SysFreeString( bstr );
+
+ IEnumWbemClassObject_Release( enm );
+}
+
START_TEST(query)
{
static const WCHAR cimv2W[] = {'R','O','O','T','\\','C','I','M','V','2',0};
@@ -1753,6 +1803,7 @@ START_TEST(query)
test_Win32_Processor( services );
test_Win32_VideoController( services );
test_Win32_Printer( services );
+ test_Win32_PnPEntity( services );
SysFreeString( path );
IWbemServices_Release( services );
--
2.18.0