dlls/mmdevapi/tests/mmdevenum.c

   1 /*
   2  * Copyright 2009 Maarten Lankhorst
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "wine/test.h"
  20
  21 #define COBJMACROS
  22
  23 #include "initguid.h"
  24 #include "endpointvolume.h"
  25 #include "mmdeviceapi.h"
  26 #include "audioclient.h"
  27 #include "audiopolicy.h"
  28 #include "dshow.h"
  29 #include "dsound.h"
  30 #include "devpkey.h"
  31
  32 DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0);
  33
  34 /* Some of the QueryInterface tests are really just to check if I got the IIDs right :) */
  35
  36 /* IMMDeviceCollection appears to have no QueryInterface method and instead forwards to mme */
  37 static void test_collection(IMMDeviceEnumerator *mme, IMMDeviceCollection *col)
  38 {
  39     IMMDeviceCollection *col2;
  40     IMMDeviceEnumerator *mme2;
  41     IUnknown *unk;
  42     HRESULT hr;
  43     ULONG ref;
  44     UINT numdev;
  45     IMMDevice *dev;
  46
  47     /* collection doesn't keep a ref on parent */
  48     IMMDeviceEnumerator_AddRef(mme);
  49     ref = IMMDeviceEnumerator_Release(mme);
  50     ok(ref == 2, "Reference count on parent is %u\n", ref);
  51
  52     ref = IMMDeviceCollection_AddRef(col);
  53     IMMDeviceCollection_Release(col);
  54     ok(ref == 2, "Invalid reference count %u on collection\n", ref);
  55
  56     hr = IMMDeviceCollection_QueryInterface(col, &IID_IUnknown, NULL);
  57     ok(hr == E_POINTER, "Null ppv returns %08x\n", hr);
  58
  59     hr = IMMDeviceCollection_QueryInterface(col, &IID_IUnknown, (void**)&unk);
  60     ok(hr == S_OK, "Cannot query for IID_IUnknown: 0x%08x\n", hr);
  61     if (hr == S_OK)
  62     {
  63         ok((IUnknown*)col == unk, "Pointers are not identical %p/%p/%p\n", col, unk, mme);
  64         IUnknown_Release(unk);
  65     }
  66
  67     hr = IMMDeviceCollection_QueryInterface(col, &IID_IMMDeviceCollection, (void**)&col2);
  68     ok(hr == S_OK, "Cannot query for IID_IMMDeviceCollection: 0x%08x\n", hr);
  69     if (hr == S_OK)
  70         IMMDeviceCollection_Release(col2);
  71
  72     hr = IMMDeviceCollection_QueryInterface(col, &IID_IMMDeviceEnumerator, (void**)&mme2);
  73     ok(hr == E_NOINTERFACE, "Query for IID_IMMDeviceEnumerator returned: 0x%08x\n", hr);
  74     if (hr == S_OK)
  75         IMMDeviceEnumerator_Release(mme2);
  76
  77     hr = IMMDeviceCollection_GetCount(col, NULL);
  78     ok(hr == E_POINTER, "GetCount returned 0x%08x\n", hr);
  79
  80     hr = IMMDeviceCollection_GetCount(col, &numdev);
  81     ok(hr == S_OK, "GetCount returned 0x%08x\n", hr);
  82
  83     dev = (void*)(LONG_PTR)0x12345678;
  84     hr = IMMDeviceCollection_Item(col, numdev, &dev);
  85     ok(hr == E_INVALIDARG, "Asking for too high device returned 0x%08x\n", hr);
  86     ok(dev == NULL, "Returned non-null device\n");
  87
  88     if (numdev)
  89     {
  90         hr = IMMDeviceCollection_Item(col, 0, NULL);
  91         ok(hr == E_POINTER, "Query with null pointer returned 0x%08x\n", hr);
  92
  93         hr = IMMDeviceCollection_Item(col, 0, &dev);
  94         ok(hr == S_OK, "Valid Item returned 0x%08x\n", hr);
  95         ok(dev != NULL, "Device is null!\n");
  96         if (dev != NULL)
  97         {
  98             char temp[128];
  99             WCHAR *id = NULL;
 100             if (IMMDevice_GetId(dev, &id) == S_OK)
 101             {
 102                 IMMDevice *dev2;
 103
 104                 temp[sizeof(temp)-1] = 0;
 105                 WideCharToMultiByte(CP_ACP, 0, id, -1, temp, sizeof(temp)-1, NULL, NULL);
 106                 trace("Device found: %s\n", temp);
 107
 108                 hr = IMMDeviceEnumerator_GetDevice(mme, id, &dev2);
 109                 ok(hr == S_OK, "GetDevice failed: %08x\n", hr);
 110
 111                 IMMDevice_Release(dev2);
 112
 113                 CoTaskMemFree(id);
 114             }
 115         }
 116         if (dev)
 117             IMMDevice_Release(dev);
 118     }
 119     IMMDeviceCollection_Release(col);
 120 }
 121
 122 static HRESULT WINAPI notif_QueryInterface(IMMNotificationClient *iface,
 123         const GUID *riid, void **obj)
 124 {
 125     ok(0, "Unexpected QueryInterface call\n");
 126     return E_NOTIMPL;
 127 }
 128
 129 static ULONG WINAPI notif_AddRef(IMMNotificationClient *iface)
 130 {
 131     ok(0, "Unexpected AddRef call\n");
 132     return 2;
 133 }
 134
 135 static ULONG WINAPI notif_Release(IMMNotificationClient *iface)
 136 {
 137     ok(0, "Unexpected Release call\n");
 138     return 1;
 139 }
 140
 141 static HRESULT WINAPI notif_OnDeviceStateChanged(IMMNotificationClient *iface,
 142         const WCHAR *device_id, DWORD new_state)
 143 {
 144     ok(0, "Unexpected OnDeviceStateChanged call\n");
 145     return E_NOTIMPL;
 146 }
 147
 148 static HRESULT WINAPI notif_OnDeviceAdded(IMMNotificationClient *iface,
 149         const WCHAR *device_id)
 150 {
 151     ok(0, "Unexpected OnDeviceAdded call\n");
 152     return E_NOTIMPL;
 153 }
 154
 155 static HRESULT WINAPI notif_OnDeviceRemoved(IMMNotificationClient *iface,
 156         const WCHAR *device_id)
 157 {
 158     ok(0, "Unexpected OnDeviceRemoved call\n");
 159     return E_NOTIMPL;
 160 }
 161
 162 static HRESULT WINAPI notif_OnDefaultDeviceChanged(IMMNotificationClient *iface,
 163         EDataFlow flow, ERole role, const WCHAR *device_id)
 164 {
 165     ok(0, "Unexpected OnDefaultDeviceChanged call\n");
 166     return E_NOTIMPL;
 167 }
 168
 169 static HRESULT WINAPI notif_OnPropertyValueChanged(IMMNotificationClient *iface,
 170         const WCHAR *device_id, const PROPERTYKEY key)
 171 {
 172     ok(0, "Unexpected OnPropertyValueChanged call\n");
 173     return E_NOTIMPL;
 174 }
 175
 176 static IMMNotificationClientVtbl notif_vtbl = {
 177     notif_QueryInterface,
 178     notif_AddRef,
 179     notif_Release,
 180     notif_OnDeviceStateChanged,
 181     notif_OnDeviceAdded,
 182     notif_OnDeviceRemoved,
 183     notif_OnDefaultDeviceChanged,
 184     notif_OnPropertyValueChanged
 185 };
 186
 187 static IMMNotificationClient notif = { &notif_vtbl };
 188
 189 /* Only do parameter tests here, the actual MMDevice testing should be a separate test */
 190 START_TEST(mmdevenum)
 191 {
 192     static const WCHAR not_a_deviceW[] = {'n','o','t','a','d','e','v','i','c','e',0};
 193
 194     HRESULT hr;
 195     IUnknown *unk = NULL;
 196     IMMDeviceEnumerator *mme, *mme2;
 197     ULONG ref;
 198     IMMDeviceCollection *col;
 199     IMMDevice *dev;
 200
 201     CoInitializeEx(NULL, COINIT_MULTITHREADED);
 202     hr = CoCreateInstance(&CLSID_MMDeviceEnumerator, NULL, CLSCTX_INPROC_SERVER, &IID_IMMDeviceEnumerator, (void**)&mme);
 203     if (FAILED(hr))
 204     {
 205         skip("mmdevapi not available: 0x%08x\n", hr);
 206         return;
 207     }
 208
 209     /* Odd behavior.. bug? */
 210     ref = IMMDeviceEnumerator_AddRef(mme);
 211     ok(ref == 3, "Invalid reference count after incrementing: %u\n", ref);
 212     IMMDeviceEnumerator_Release(mme);
 213
 214     hr = IMMDeviceEnumerator_QueryInterface(mme, &IID_IUnknown, (void**)&unk);
 215     ok(hr == S_OK, "returned 0x%08x\n", hr);
 216     if (hr != S_OK) return;
 217
 218     ok( (LONG_PTR)mme == (LONG_PTR)unk, "Pointers are unequal %p/%p\n", unk, mme);
 219     IUnknown_Release(unk);
 220
 221     /* Proving that it is static.. */
 222     hr = CoCreateInstance(&CLSID_MMDeviceEnumerator, NULL, CLSCTX_INPROC_SERVER, &IID_IMMDeviceEnumerator, (void**)&mme2);
 223     ok(hr == S_OK, "CoCreateInstance failed: 0x%08x\n", hr);
 224     IMMDeviceEnumerator_Release(mme2);
 225     ok(mme == mme2, "Pointers are not equal!\n");
 226
 227     hr = IMMDeviceEnumerator_QueryInterface(mme, &IID_IUnknown, NULL);
 228     ok(hr == E_POINTER, "Null pointer on QueryInterface returned %08x\n", hr);
 229
 230     hr = IMMDeviceEnumerator_QueryInterface(mme, &GUID_NULL, (void**)&unk);
 231     ok(!unk, "Unk not reset to null after invalid QI\n");
 232     ok(hr == E_NOINTERFACE, "Invalid hr %08x returned on IID_NULL\n", hr);
 233
 234     hr = IMMDeviceEnumerator_GetDevice(mme, not_a_deviceW, NULL);
 235     ok(hr == E_POINTER, "GetDevice gave wrong error: %08x\n", hr);
 236
 237     hr = IMMDeviceEnumerator_GetDevice(mme, NULL, &dev);
 238     ok(hr == E_POINTER, "GetDevice gave wrong error: %08x\n", hr);
 239
 240     hr = IMMDeviceEnumerator_GetDevice(mme, not_a_deviceW, &dev);
 241     ok(hr == E_INVALIDARG, "GetDevice gave wrong error: %08x\n", hr);
 242
 243     col = (void*)(LONG_PTR)0x12345678;
 244     hr = IMMDeviceEnumerator_EnumAudioEndpoints(mme, 0xffff, DEVICE_STATEMASK_ALL, &col);
 245     ok(hr == E_INVALIDARG, "Setting invalid data flow returned 0x%08x\n", hr);
 246     ok(col == NULL, "Collection pointer non-null on failure\n");
 247
 248     hr = IMMDeviceEnumerator_EnumAudioEndpoints(mme, eAll, DEVICE_STATEMASK_ALL+1, &col);
 249     ok(hr == E_INVALIDARG, "Setting invalid mask returned 0x%08x\n", hr);
 250
 251     hr = IMMDeviceEnumerator_EnumAudioEndpoints(mme, eAll, DEVICE_STATEMASK_ALL, NULL);
 252     ok(hr == E_POINTER, "Invalid pointer returned: 0x%08x\n", hr);
 253
 254     hr = IMMDeviceEnumerator_EnumAudioEndpoints(mme, eAll, DEVICE_STATEMASK_ALL, &col);
 255     ok(hr == S_OK, "Valid EnumAudioEndpoints returned 0x%08x\n", hr);
 256     if (hr == S_OK)
 257     {
 258         ok(!!col, "Returned null pointer\n");
 259         if (col)
 260             test_collection(mme, col);
 261     }
 262
 263     hr = IMMDeviceEnumerator_RegisterEndpointNotificationCallback(mme, NULL);
 264     ok(hr == E_POINTER, "RegisterEndpointNotificationCallback failed: %08x\n", hr);
 265
 266     hr = IMMDeviceEnumerator_RegisterEndpointNotificationCallback(mme, &notif);
 267     ok(hr == S_OK, "RegisterEndpointNotificationCallback failed: %08x\n", hr);
 268
 269     hr = IMMDeviceEnumerator_RegisterEndpointNotificationCallback(mme, &notif);
 270     ok(hr == S_OK, "RegisterEndpointNotificationCallback failed: %08x\n", hr);
 271
 272     hr = IMMDeviceEnumerator_UnregisterEndpointNotificationCallback(mme, NULL);
 273     ok(hr == E_POINTER, "UnregisterEndpointNotificationCallback failed: %08x\n", hr);
 274
 275     hr = IMMDeviceEnumerator_UnregisterEndpointNotificationCallback(mme, (IMMNotificationClient*)0xdeadbeef);
 276     ok(hr == E_NOTFOUND, "UnregisterEndpointNotificationCallback failed: %08x\n", hr);
 277
 278     hr = IMMDeviceEnumerator_UnregisterEndpointNotificationCallback(mme, &notif);
 279     ok(hr == S_OK, "UnregisterEndpointNotificationCallback failed: %08x\n", hr);
 280
 281     hr = IMMDeviceEnumerator_UnregisterEndpointNotificationCallback(mme, &notif);
 282     ok(hr == S_OK, "UnregisterEndpointNotificationCallback failed: %08x\n", hr);
 283
 284     hr = IMMDeviceEnumerator_UnregisterEndpointNotificationCallback(mme, &notif);
 285     ok(hr == E_NOTFOUND, "UnregisterEndpointNotificationCallback failed: %08x\n", hr);
 286
 287     IMMDeviceEnumerator_Release(mme);
 288 }