kernel32: Remove superfluous heap reallocation calls in FormatMessageA/W.
[wine] / dlls / mmdevapi / tests / capture.c
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 /* This test is for audio capture specific mechanisms
20  * Tests:
21  * - IAudioClient with eCapture and IAudioCaptureClient
22  */
23
24 #include "wine/test.h"
25
26 #define CINTERFACE
27 #define COBJMACROS
28
29 #ifdef STANDALONE
30 #include "initguid.h"
31 #endif
32
33 #include "unknwn.h"
34 #include "uuids.h"
35 #include "mmdeviceapi.h"
36 #include "audioclient.h"
37
38 static void test_uninitialized(IAudioClient *ac)
39 {
40     HRESULT hr;
41     UINT32 num;
42     REFERENCE_TIME t1;
43
44     HANDLE handle = CreateEventW(NULL, FALSE, FALSE, NULL);
45     IUnknown *unk;
46
47     hr = IAudioClient_GetBufferSize(ac, &num);
48     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Uninitialized GetBufferSize call returns %08x\n", hr);
49
50     hr = IAudioClient_GetStreamLatency(ac, &t1);
51     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Uninitialized GetStreamLatency call returns %08x\n", hr);
52
53     hr = IAudioClient_GetCurrentPadding(ac, &num);
54     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Uninitialized GetCurrentPadding call returns %08x\n", hr);
55
56     hr = IAudioClient_Start(ac);
57     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Uninitialized Start call returns %08x\n", hr);
58
59     hr = IAudioClient_Stop(ac);
60     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Uninitialized Stop call returns %08x\n", hr);
61
62     hr = IAudioClient_Reset(ac);
63     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Uninitialized Reset call returns %08x\n", hr);
64
65     hr = IAudioClient_SetEventHandle(ac, handle);
66     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Uninitialized SetEventHandle call returns %08x\n", hr);
67
68     hr = IAudioClient_GetService(ac, &IID_IAudioStreamVolume, (void**)&unk);
69     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Uninitialized GetService call returns %08x\n", hr);
70
71     CloseHandle(handle);
72 }
73
74 static void test_capture(IAudioClient *ac, HANDLE handle, WAVEFORMATEX *wfx)
75 {
76     IAudioCaptureClient *acc;
77     HRESULT hr;
78     UINT32 frames = 0;
79     BYTE *data = NULL;
80     DWORD flags;
81     UINT64 devpos, qpcpos;
82
83     hr = IAudioClient_GetService(ac, &IID_IAudioCaptureClient, (void**)&acc);
84     ok(hr == S_OK, "IAudioClient_GetService(IID_IAudioCaptureClient) returns %08x\n", hr);
85     if (hr != S_OK)
86         return;
87
88     hr = IAudioCaptureClient_GetNextPacketSize(acc, NULL);
89     ok(hr == E_POINTER, "IAudioCaptureClient_GetNextPacketSize(NULL) returns %08x\n", hr);
90
91     ok(WaitForSingleObject(handle, 1000) == WAIT_OBJECT_0, "Waiting on event handle failed!\n");
92
93     /* frames can be 0 if no data is available yet.. */
94     hr = IAudioCaptureClient_GetNextPacketSize(acc, &frames);
95     ok(hr == S_OK, "IAudioCaptureClient_GetNextPacketSize returns %08x\n", hr);
96
97     data = (BYTE*)(DWORD_PTR)0xdeadbeef;
98     hr = IAudioCaptureClient_GetBuffer(acc, &data, NULL, NULL, NULL, NULL);
99     ok(hr == E_POINTER, "IAudioCaptureClient_GetBuffer(data, NULL, NULL) returns %08x\n", hr);
100     ok((DWORD_PTR)data == 0xdeadbeef, "data is reset to %p\n", data);
101
102     frames = 0xdeadbeef;
103     hr = IAudioCaptureClient_GetBuffer(acc, NULL, &frames, NULL, NULL, NULL);
104     ok(hr == E_POINTER, "IAudioCaptureClient_GetBuffer(NULL, &frames, NULL) returns %08x\n", hr);
105     ok(frames == 0xdeadbeef, "frames is reset to %08x\n", frames);
106
107     flags = 0xdeadbeef;
108     hr = IAudioCaptureClient_GetBuffer(acc, NULL, NULL, &flags, NULL, NULL);
109     ok(hr == E_POINTER, "IAudioCaptureClient_GetBuffer(NULL, NULL, &flags) returns %08x\n", hr);
110     ok(flags == 0xdeadbeef, "flags is reset to %08x\n", flags);
111
112     hr = IAudioCaptureClient_GetBuffer(acc, &data, &frames, NULL, NULL, NULL);
113     ok(hr == E_POINTER, "IAudioCaptureClient_GetBuffer(&ata, &frames, NULL) returns %08x\n", hr);
114
115     hr = IAudioCaptureClient_GetBuffer(acc, &data, &frames, &flags, &devpos, &qpcpos);
116     ok(hr == S_OK || hr == AUDCLNT_S_BUFFER_EMPTY, "Valid IAudioCaptureClient_GetBuffer returns %08x\n", hr);
117     if (hr == S_OK)
118         ok(frames, "Amount of frames locked is 0!\n");
119     else if (hr == AUDCLNT_S_BUFFER_EMPTY)
120         ok(!frames, "Amount of frames locked with empty buffer is %u!\n", frames);
121     trace("Device position is at %u, amount of frames locked: %u\n", (DWORD)devpos, frames);
122
123     if (frames) {
124         hr = IAudioCaptureClient_GetBuffer(acc, &data, &frames, &flags, &devpos, &qpcpos);
125         ok(hr == AUDCLNT_E_OUT_OF_ORDER, "Out of order IAudioCaptureClient_GetBuffer returns %08x\n", hr);
126     }
127
128     hr = IAudioCaptureClient_ReleaseBuffer(acc, frames);
129     ok(hr == S_OK, "Releasing buffer returns %08x\n", hr);
130
131     if (frames) {
132         hr = IAudioCaptureClient_ReleaseBuffer(acc, frames);
133         ok(hr == AUDCLNT_E_OUT_OF_ORDER, "Releasing buffer twice returns %08x\n", hr);
134     }
135
136     IUnknown_Release(acc);
137 }
138
139 static void test_audioclient(IAudioClient *ac)
140 {
141     IUnknown *unk;
142     HRESULT hr;
143     ULONG ref;
144     WAVEFORMATEX *pwfx, *pwfx2;
145     REFERENCE_TIME t1, t2;
146
147     HANDLE handle = CreateEventW(NULL, FALSE, FALSE, NULL);
148
149     hr = IAudioClient_QueryInterface(ac, &IID_IUnknown, NULL);
150     ok(hr == E_POINTER, "QueryInterface(NULL) returned %08x\n", hr);
151
152     unk = (void*)(LONG_PTR)0x12345678;
153     hr = IAudioClient_QueryInterface(ac, &IID_NULL, (void**)&unk);
154     ok(hr == E_NOINTERFACE, "QueryInterface(IID_NULL) returned %08x\n", hr);
155     ok(!unk, "QueryInterface(IID_NULL) returned non-null pointer %p\n", unk);
156
157     hr = IAudioClient_QueryInterface(ac, &IID_IUnknown, (void**)&unk);
158     ok(hr == S_OK, "QueryInterface(IID_IUnknown) returned %08x\n", hr);
159     if (unk)
160     {
161         ref = IUnknown_Release(unk);
162         ok(ref == 1, "Released count is %u\n", ref);
163     }
164
165     hr = IAudioClient_QueryInterface(ac, &IID_IAudioClient, (void**)&unk);
166     ok(hr == S_OK, "QueryInterface(IID_IAudioClient) returned %08x\n", hr);
167     if (unk)
168     {
169         ref = IUnknown_Release(unk);
170         ok(ref == 1, "Released count is %u\n", ref);
171     }
172
173     hr = IAudioClient_GetDevicePeriod(ac, NULL, NULL);
174     ok(hr == E_POINTER, "Invalid GetDevicePeriod call returns %08x\n", hr);
175
176     hr = IAudioClient_GetDevicePeriod(ac, &t1, NULL);
177     ok(hr == S_OK, "Valid GetDevicePeriod call returns %08x\n", hr);
178
179     hr = IAudioClient_GetDevicePeriod(ac, NULL, &t2);
180     ok(hr == S_OK, "Valid GetDevicePeriod call returns %08x\n", hr);
181
182     hr = IAudioClient_GetDevicePeriod(ac, &t1, &t2);
183     ok(hr == S_OK, "Valid GetDevicePeriod call returns %08x\n", hr);
184     trace("Returned periods: %u.%05u ms %u.%05u ms\n",
185           (UINT)(t1/10000), (UINT)(t1 % 10000),
186           (UINT)(t2/10000), (UINT)(t2 % 10000));
187
188     hr = IAudioClient_GetMixFormat(ac, NULL);
189     ok(hr == E_POINTER, "GetMixFormat returns %08x\n", hr);
190
191     hr = IAudioClient_GetMixFormat(ac, &pwfx);
192     ok(hr == S_OK, "Valid GetMixFormat returns %08x\n", hr);
193
194     trace("Tag: %04x\n", pwfx->wFormatTag);
195     trace("bits: %u\n", pwfx->wBitsPerSample);
196     trace("chan: %u\n", pwfx->nChannels);
197     trace("rate: %u\n", pwfx->nSamplesPerSec);
198     trace("align: %u\n", pwfx->nBlockAlign);
199     trace("extra: %u\n", pwfx->cbSize);
200     ok(pwfx->wFormatTag == WAVE_FORMAT_EXTENSIBLE, "wFormatTag is %x\n", pwfx->wFormatTag);
201     if (pwfx->wFormatTag == WAVE_FORMAT_EXTENSIBLE)
202     {
203         WAVEFORMATEXTENSIBLE *pwfxe = (void*)pwfx;
204         trace("Res: %u\n", pwfxe->Samples.wReserved);
205         trace("Mask: %x\n", pwfxe->dwChannelMask);
206         trace("Alg: %s\n",
207               IsEqualGUID(&pwfxe->SubFormat, &KSDATAFORMAT_SUBTYPE_PCM)?"PCM":
208               (IsEqualGUID(&pwfxe->SubFormat,
209                            &KSDATAFORMAT_SUBTYPE_IEEE_FLOAT)?"FLOAT":"Other"));
210     }
211
212     if (hr == S_OK)
213     {
214         hr = IAudioClient_IsFormatSupported(ac, AUDCLNT_SHAREMODE_SHARED, pwfx, &pwfx2);
215         ok(hr == S_OK, "Valid IsFormatSupported(Shared) call returns %08x\n", hr);
216         ok(pwfx2 == NULL, "pwfx2 is non-null\n");
217         CoTaskMemFree(pwfx2);
218
219         hr = IAudioClient_IsFormatSupported(ac, AUDCLNT_SHAREMODE_SHARED, NULL, NULL);
220         ok(hr == E_POINTER, "IsFormatSupported(NULL) call returns %08x\n", hr);
221
222         hr = IAudioClient_IsFormatSupported(ac, AUDCLNT_SHAREMODE_SHARED, pwfx, NULL);
223         ok(hr == E_POINTER, "IsFormatSupported(Shared,NULL) call returns %08x\n", hr);
224
225         hr = IAudioClient_IsFormatSupported(ac, AUDCLNT_SHAREMODE_EXCLUSIVE, pwfx, NULL);
226         ok(hr == S_OK || hr == AUDCLNT_E_UNSUPPORTED_FORMAT, "IsFormatSupported(Exclusive) call returns %08x\n", hr);
227
228         hr = IAudioClient_IsFormatSupported(ac, AUDCLNT_SHAREMODE_EXCLUSIVE, pwfx, &pwfx2);
229         ok(hr == S_OK || hr == AUDCLNT_E_UNSUPPORTED_FORMAT, "IsFormatSupported(Exclusive) call returns %08x\n", hr);
230         ok(pwfx2 == NULL, "pwfx2 non-null on exclusive IsFormatSupported\n");
231
232         hr = IAudioClient_IsFormatSupported(ac, 0xffffffff, pwfx, NULL);
233         ok(hr == E_INVALIDARG, "IsFormatSupported(0xffffffff) call returns %08x\n", hr);
234     }
235
236     test_uninitialized(ac);
237
238     hr = IAudioClient_Initialize(ac, 3, 0, 5000000, 0, pwfx, NULL);
239     ok(hr == AUDCLNT_E_NOT_INITIALIZED, "Initialize with invalid sharemode returns %08x\n", hr);
240
241     hr = IAudioClient_Initialize(ac, AUDCLNT_SHAREMODE_SHARED, 0xffffffff, 5000000, 0, pwfx, NULL);
242     ok(hr == E_INVALIDARG, "Initialize with invalid flags returns %08x\n", hr);
243
244     /* It seems that if length > 2s or periodicity != 0 the length is ignored and call succeeds
245      * Since we can only initialize succesfully once skip those tests
246      */
247     hr = IAudioClient_Initialize(ac, AUDCLNT_SHAREMODE_SHARED, 0, 5000000, 0, NULL, NULL);
248     ok(hr == E_POINTER, "Initialize with null format returns %08x\n", hr);
249
250     hr = IAudioClient_Initialize(ac, AUDCLNT_SHAREMODE_SHARED, AUDCLNT_STREAMFLAGS_EVENTCALLBACK, 5000000, 0, pwfx, NULL);
251     ok(hr == S_OK, "Valid Initialize returns %08x\n", hr);
252
253     if (hr != S_OK)
254     {
255         skip("Cannot initialize %08x, remainder of tests is useless\n", hr);
256         CoTaskMemFree(pwfx);
257         return;
258     }
259
260     hr = IAudioClient_GetStreamLatency(ac, NULL);
261     ok(hr == E_POINTER, "GetStreamLatency(NULL) call returns %08x\n", hr);
262
263     hr = IAudioClient_GetStreamLatency(ac, &t1);
264     ok(hr == S_OK, "Valid GetStreamLatency call returns %08x\n", hr);
265     trace("Returned latency: %u.%05u ms\n",
266           (UINT)(t1/10000), (UINT)(t1 % 10000));
267
268     hr = IAudioClient_Initialize(ac, AUDCLNT_SHAREMODE_SHARED, 0, 5000000, 0, pwfx, NULL);
269     ok(hr == AUDCLNT_E_ALREADY_INITIALIZED, "Calling Initialize twice returns %08x\n", hr);
270
271     hr = IAudioClient_SetEventHandle(ac, NULL);
272     ok(hr == E_INVALIDARG, "SetEventHandle(NULL) returns %08x\n", hr);
273
274     hr = IAudioClient_Start(ac);
275     ok(hr == AUDCLNT_E_EVENTHANDLE_NOT_SET, "Start before SetEventHandle returns %08x\n", hr);
276
277     hr = IAudioClient_SetEventHandle(ac, handle);
278     ok(hr == S_OK, "SetEventHandle returns %08x\n", hr);
279
280     hr = IAudioClient_Reset(ac);
281     ok(hr == S_OK, "Reset on a resetted stream returns %08x\n", hr);
282
283     hr = IAudioClient_Stop(ac);
284     ok(hr == S_FALSE, "Stop on a stopped stream returns %08x\n", hr);
285
286     hr = IAudioClient_Start(ac);
287     ok(hr == S_OK, "Start on a stopped stream returns %08x\n", hr);
288
289     test_capture(ac, handle, pwfx);
290
291     CloseHandle(handle);
292     CoTaskMemFree(pwfx);
293 }
294
295 START_TEST(capture)
296 {
297     HRESULT hr;
298     IMMDeviceEnumerator *mme = NULL;
299     IMMDevice *dev = NULL;
300     IAudioClient *ac = NULL;
301
302     CoInitializeEx(NULL, COINIT_MULTITHREADED);
303     hr = CoCreateInstance(&CLSID_MMDeviceEnumerator, NULL, CLSCTX_INPROC_SERVER, &IID_IMMDeviceEnumerator, (void**)&mme);
304     if (FAILED(hr))
305     {
306         skip("mmdevapi not available: 0x%08x\n", hr);
307         goto cleanup;
308     }
309
310     hr = IMMDeviceEnumerator_GetDefaultAudioEndpoint(mme, eCapture, eMultimedia, &dev);
311     ok(hr == S_OK || hr == E_NOTFOUND, "GetDefaultAudioEndpoint failed: 0x%08x\n", hr);
312     if (hr != S_OK || !dev)
313     {
314         if (hr == E_NOTFOUND)
315             skip("No sound card available\n");
316         else
317             skip("GetDefaultAudioEndpoint returns 0x%08x\n", hr);
318         goto cleanup;
319     }
320
321     hr = IMMDevice_Activate(dev, &IID_IAudioClient, CLSCTX_INPROC_SERVER, NULL, (void**)&ac);
322     todo_wine ok(hr == S_OK, "Activation failed with %08x\n", hr);
323     if (ac)
324     {
325         test_audioclient(ac);
326         IAudioClient_Release(ac);
327     }
328     IMMDevice_Release(dev);
329
330 cleanup:
331     if (mme)
332         IMMDeviceEnumerator_Release(mme);
333     CoUninitialize();
334 }