1 /*
2  * Copyright 2010 Maarten Lankhorst for CodeWeavers
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 #define NONAMELESSUNION
20 #include "wine/test.h"
21 
22 #define COBJMACROS
23 
24 #ifdef STANDALONE
25 #include "initguid.h"
26 #endif
27 
28 #include "unknwn.h"
29 #include "uuids.h"
30 #include "mmdeviceapi.h"
31 #include "devpkey.h"
32 
33 static BOOL (WINAPI *pIsWow64Process)(HANDLE, BOOL *);
34 
35 static const WCHAR software_renderW[] =
36     { 'S','o','f','t','w','a','r','e','\\',
37       'M','i','c','r','o','s','o','f','t','\\',
38       'W','i','n','d','o','w','s','\\',
39       'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
40       'M','M','D','e','v','i','c','e','s','\\',
41       'A','u','d','i','o','\\',
42       'R','e','n','d','e','r',0 };
43 static const WCHAR propertiesW[] = {'P','r','o','p','e','r','t','i','e','s',0};
44 
45 
46 static void test_propertystore(IPropertyStore *store)
47 {
48     HRESULT hr;
49     PROPVARIANT pv;
50     char temp[128];
51     temp[sizeof(temp)-1] = 0;
52 
53     pv.vt = VT_EMPTY;
54     hr = IPropertyStore_GetValue(store, &PKEY_AudioEndpoint_GUID, &pv);
55     ok(hr == S_OK, "Failed with %08x\n", hr);
56     ok(pv.vt == VT_LPWSTR, "Value should be %i, is %i\n", VT_LPWSTR, pv.vt);
57     if (hr == S_OK && pv.vt == VT_LPWSTR)
58     {
59         WideCharToMultiByte(CP_ACP, 0, pv.u.pwszVal, -1, temp, sizeof(temp)-1, NULL, NULL);
60         trace("guid: %s\n", temp);
61         PropVariantClear(&pv);
62     }
63 
64     pv.vt = VT_EMPTY;
65     hr = IPropertyStore_GetValue(store, (const PROPERTYKEY*)&DEVPKEY_DeviceInterface_FriendlyName, &pv);
66     ok(hr == S_OK, "Failed with %08x\n", hr);
67     ok(pv.vt == VT_LPWSTR && pv.u.pwszVal, "FriendlyName value had wrong type: 0x%x or was NULL\n", pv.vt);
68     PropVariantClear(&pv);
69 
70     pv.vt = VT_EMPTY;
71     hr = IPropertyStore_GetValue(store, (const PROPERTYKEY*)&DEVPKEY_DeviceInterface_Enabled, &pv);
72     ok(hr == S_OK, "Failed with %08x\n", hr);
73     ok(pv.vt == VT_EMPTY, "Key should not be found\n");
74     PropVariantClear(&pv);
75 
76     pv.vt = VT_EMPTY;
77     hr = IPropertyStore_GetValue(store, (const PROPERTYKEY*)&DEVPKEY_DeviceInterface_ClassGuid, &pv);
78     ok(hr == S_OK, "Failed with %08x\n", hr);
79     ok(pv.vt == VT_EMPTY, "Key should not be found\n");
80     PropVariantClear(&pv);
81 }
82 
83 static void test_deviceinterface(IPropertyStore *store)
84 {
85     HRESULT hr;
86     PROPVARIANT pv;
87 
88     static const PROPERTYKEY deviceinterface_key = {
89         {0x233164c8, 0x1b2c, 0x4c7d, {0xbc, 0x68, 0xb6, 0x71, 0x68, 0x7a, 0x25, 0x67}}, 1
90     };
91 
92     pv.vt = VT_EMPTY;
93     hr = IPropertyStore_GetValue(store, &deviceinterface_key, &pv);
94     ok(hr == S_OK, "GetValue failed: %08x\n", hr);
95     ok(pv.vt == VT_LPWSTR, "Got wrong variant type: 0x%x\n", pv.vt);
96     trace("device interface: %s\n", wine_dbgstr_w(pv.u.pwszVal));
97     PropVariantClear(&pv);
98 }
99 
100 static void test_getat(IPropertyStore *store)
101 {
102     HRESULT hr;
103     DWORD propcount;
104     DWORD prop;
105     PROPERTYKEY pkey;
106     BOOL found_name = FALSE;
107     BOOL found_desc = FALSE;
108     char temp[128];
109     temp[sizeof(temp)-1] = 0;
110 
111     hr = IPropertyStore_GetCount(store, &propcount);
112 
113     ok(hr == S_OK, "Failed with %08x\n", hr);
114     ok(propcount > 0, "Propcount %d should be greater than zero\n", propcount);
115 
116     for (prop = 0; prop < propcount; prop++) {
117 	hr = IPropertyStore_GetAt(store, prop, &pkey);
118 	ok(hr == S_OK, "Failed with %08x\n", hr);
119 	if (IsEqualPropertyKey(pkey, DEVPKEY_Device_FriendlyName))
120 	    found_name = TRUE;
121 	if (IsEqualPropertyKey(pkey, DEVPKEY_Device_DeviceDesc))
122 	    found_desc = TRUE;
123     }
124     ok(found_name ||
125             broken(!found_name) /* vista */, "DEVPKEY_Device_FriendlyName not found\n");
126     ok(found_desc, "DEVPKEY_Device_DeviceDesc not found\n");
127 }
128 
129 static void test_setvalue_on_wow64(IPropertyStore *store)
130 {
131     PROPVARIANT pv;
132     HRESULT hr;
133     LONG ret;
134     WCHAR *guidW;
135     HKEY root, props, devkey;
136     DWORD type, regval, size;
137 
138     static const PROPERTYKEY PKEY_Bogus = {
139         {0x1da5d803, 0xd492, 0x4edd, {0x8c, 0x23, 0xe0, 0xc0, 0xff, 0xee, 0x7f, 0x00}}, 0x7f
140     };
141     static const WCHAR bogusW[] = {'{','1','D','A','5','D','8','0','3','-','D','4','9','2','-','4','E','D','D','-','8','C','2','3','-','E','0','C','0','F','F','E','E','7','F','0','0','}',',','1','2','7',0};
142 
143     PropVariantInit(&pv);
144 
145     pv.vt = VT_EMPTY;
146     hr = IPropertyStore_GetValue(store, &PKEY_AudioEndpoint_GUID, &pv);
147     ok(hr == S_OK, "Failed to get Endpoint GUID: %08x\n", hr);
148 
149     guidW = pv.u.pwszVal;
150 
151     pv.vt = VT_UI4;
152     pv.u.ulVal = 0xAB;
153 
154     hr = IPropertyStore_SetValue(store, &PKEY_Bogus, &pv);
155     ok(hr == S_OK || hr == E_ACCESSDENIED, "SetValue failed: %08x\n", hr);
156     if (hr != S_OK)
157     {
158         win_skip("Missing permission to write to registry\n");
159         return;
160     }
161 
162     pv.u.ulVal = 0x00;
163 
164     hr = IPropertyStore_GetValue(store, &PKEY_Bogus, &pv);
165     ok(hr == S_OK, "GetValue failed: %08x\n", hr);
166     ok(pv.u.ulVal == 0xAB, "Got wrong value: 0x%x\n", pv.u.ulVal);
167 
168     /* should find the key in 64-bit view */
169     ret = RegOpenKeyExW(HKEY_LOCAL_MACHINE, software_renderW, 0, KEY_READ|KEY_WOW64_64KEY, &root);
170     ok(ret == ERROR_SUCCESS, "Couldn't open mmdevices Render key: %u\n", ret);
171 
172     ret = RegOpenKeyExW(root, guidW, 0, KEY_READ|KEY_WOW64_64KEY, &devkey);
173     ok(ret == ERROR_SUCCESS, "Couldn't open mmdevice guid key: %u\n", ret);
174 
175     ret = RegOpenKeyExW(devkey, propertiesW, 0, KEY_READ|KEY_WOW64_64KEY, &props);
176     ok(ret == ERROR_SUCCESS, "Couldn't open mmdevice property key: %u\n", ret);
177 
178     /* Note: the registry key exists even without calling IPropStore::Commit */
179     size = sizeof(regval);
180     ret = RegQueryValueExW(props, bogusW, NULL, &type, (LPBYTE)&regval, &size);
181     ok(ret == ERROR_SUCCESS, "Couldn't get bogus propertykey value: %u\n", ret);
182     ok(type == REG_DWORD, "Got wrong value type: %u\n", type);
183     ok(regval == 0xAB, "Got wrong value: 0x%x\n", regval);
184 
185     RegCloseKey(props);
186     RegCloseKey(devkey);
187     RegCloseKey(root);
188 
189     CoTaskMemFree(guidW);
190 
191     /* should NOT find the key in 32-bit view */
192     ret = RegOpenKeyExW(HKEY_LOCAL_MACHINE, software_renderW, 0, KEY_READ, &root);
193     ok(ret == ERROR_FILE_NOT_FOUND, "Wrong error when opening mmdevices Render key: %u\n", ret);
194 }
195 
196 START_TEST(propstore)
197 {
198     HRESULT hr;
199     IMMDeviceEnumerator *mme = NULL;
200     IMMDevice *dev = NULL;
201     IPropertyStore *store;
202     BOOL is_wow64 = FALSE;
203     HMODULE hk32 = GetModuleHandleA("kernel32.dll");
204 
205     pIsWow64Process = (void *)GetProcAddress(hk32, "IsWow64Process");
206 
207     if (pIsWow64Process)
208         pIsWow64Process(GetCurrentProcess(), &is_wow64);
209 
210     CoInitializeEx(NULL, COINIT_MULTITHREADED);
211     hr = CoCreateInstance(&CLSID_MMDeviceEnumerator, NULL, CLSCTX_INPROC_SERVER, &IID_IMMDeviceEnumerator, (void**)&mme);
212     if (FAILED(hr))
213     {
214         skip("mmdevapi not available: 0x%08x\n", hr);
215         goto cleanup;
216     }
217 
218     hr = IMMDeviceEnumerator_GetDefaultAudioEndpoint(mme, eRender, eMultimedia, &dev);
219     ok(hr == S_OK || hr == E_NOTFOUND, "GetDefaultAudioEndpoint failed: 0x%08x\n", hr);
220     if (hr != S_OK)
221     {
222         if (hr == E_NOTFOUND)
223             skip("No sound card available\n");
224         else
225             skip("GetDefaultAudioEndpoint returns 0x%08x\n", hr);
226         goto cleanup;
227     }
228     store = NULL;
229     hr = IMMDevice_OpenPropertyStore(dev, 3, &store);
230     ok(hr == E_INVALIDARG, "Wrong hr returned: %08x\n", hr);
231     if (hr != S_OK)
232         /* It seems on windows returning with E_INVALIDARG doesn't
233          * set store to NULL, so just don't set store to non-null
234          * before calling this function
235          */
236         ok(!store, "Store set to non-NULL on failure: %p/%08x\n", store, hr);
237     else if (store)
238         IPropertyStore_Release(store);
239     hr = IMMDevice_OpenPropertyStore(dev, STGM_READ, NULL);
240     ok(hr == E_POINTER, "Wrong hr returned: %08x\n", hr);
241 
242     store = NULL;
243     hr = IMMDevice_OpenPropertyStore(dev, STGM_READWRITE, &store);
244     if(hr == E_ACCESSDENIED)
245         hr = IMMDevice_OpenPropertyStore(dev, STGM_READ, &store);
246     ok(hr == S_OK, "Opening valid store returned %08x\n", hr);
247     if (store)
248     {
249         test_propertystore(store);
250         test_deviceinterface(store);
251         test_getat(store);
252         if (is_wow64)
253             test_setvalue_on_wow64(store);
254         IPropertyStore_Release(store);
255     }
256     IMMDevice_Release(dev);
257 cleanup:
258     if (mme)
259         IMMDeviceEnumerator_Release(mme);
260     CoUninitialize();
261 }
262