dlls/dsound/capture.c

   1 /*              DirectSoundCapture
   2  *
   3  * Copyright 1998 Marcus Meissner
   4  * Copyright 1998 Rob Riggs
   5  * Copyright 2000-2001 TransGaming Technologies, Inc.
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21 /*
  22  * TODO:
  23  *      Implement FX support.
  24  *      Implement both IDirectSoundCaptureBuffer and IDirectSoundCaptureBuffer8
  25  *      Make DirectSoundCaptureCreate and DirectSoundCaptureCreate8 behave differently
  26  */
  27
  28 #include <stdarg.h>
  29
  30 #define COBJMACROS
  31
  32 #include "windef.h"
  33 #include "winbase.h"
  34 #include "winuser.h"
  35 #include "mmsystem.h"
  36 #include "mmddk.h"
  37 #include "winnls.h"
  38 #include "wine/debug.h"
  39 #include "dsound.h"
  40 #include "dsound_private.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(dsound);
  43
  44 typedef struct DirectSoundCaptureDevice      DirectSoundCaptureDevice;
  45
  46 /* IDirectSoundCaptureBuffer implementation structure */
  47 typedef struct IDirectSoundCaptureBufferImpl
  48 {
  49     IDirectSoundCaptureBuffer8          IDirectSoundCaptureBuffer8_iface;
  50     IDirectSoundNotify                  IDirectSoundNotify_iface;
  51     LONG                                numIfaces; /* "in use interfaces" refcount */
  52     LONG                                ref, refn, has_dsc8;
  53     /* IDirectSoundCaptureBuffer fields */
  54     DirectSoundCaptureDevice            *device;
  55     DSCBUFFERDESC                       *pdscbd;
  56     DWORD                               flags;
  57     /* IDirectSoundNotify fields */
  58     DSBPOSITIONNOTIFY                   *notifies;
  59     int                                 nrofnotifies;
  60     HANDLE thread;
  61     HANDLE sleepev;
  62 } IDirectSoundCaptureBufferImpl;
  63
  64 /* DirectSoundCaptureDevice implementation structure */
  65 struct DirectSoundCaptureDevice
  66 {
  67     GUID                          guid;
  68     LONG                          ref;
  69     DSCCAPS                       drvcaps;
  70     BYTE                          *buffer;
  71     DWORD                         buflen, write_pos_bytes;
  72     WAVEFORMATEX                  *pwfx;
  73     IDirectSoundCaptureBufferImpl *capture_buffer;
  74     DWORD                         state;
  75     CRITICAL_SECTION              lock;
  76     IMMDevice                     *mmdevice;
  77     IAudioClient                  *client;
  78     IAudioCaptureClient           *capture;
  79     struct list                   entry;
  80 };
  81
  82 static DWORD WINAPI DSOUND_capture_thread(void *user);
  83
  84 static void capturebuffer_destroy(IDirectSoundCaptureBufferImpl *This)
  85 {
  86     if (This->device->state == STATE_CAPTURING)
  87         This->device->state = STATE_STOPPING;
  88
  89     if(This->thread){
  90         SetEvent(This->sleepev);
  91         WaitForSingleObject(This->thread, INFINITE);
  92         CloseHandle(This->thread);
  93     }
  94     CloseHandle(This->sleepev);
  95
  96     HeapFree(GetProcessHeap(),0, This->pdscbd);
  97
  98     if (This->device->client) {
  99         IAudioClient_Release(This->device->client);
 100         This->device->client = NULL;
 101     }
 102
 103     if (This->device->capture) {
 104         IAudioCaptureClient_Release(This->device->capture);
 105         This->device->capture = NULL;
 106     }
 107
 108     /* remove from DirectSoundCaptureDevice */
 109     This->device->capture_buffer = NULL;
 110
 111     HeapFree(GetProcessHeap(), 0, This->notifies);
 112     HeapFree(GetProcessHeap(), 0, This);
 113     TRACE("(%p) released\n", This);
 114 }
 115
 116 /*******************************************************************************
 117  * IDirectSoundNotify
 118  */
 119 static inline struct IDirectSoundCaptureBufferImpl *impl_from_IDirectSoundNotify(IDirectSoundNotify *iface)
 120 {
 121     return CONTAINING_RECORD(iface, IDirectSoundCaptureBufferImpl, IDirectSoundNotify_iface);
 122 }
 123
 124 static HRESULT WINAPI IDirectSoundNotifyImpl_QueryInterface(IDirectSoundNotify *iface, REFIID riid,
 125         void **ppobj)
 126 {
 127     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundNotify(iface);
 128
 129     TRACE("(%p,%s,%p)\n", This, debugstr_guid(riid), ppobj);
 130
 131     return IDirectSoundCaptureBuffer8_QueryInterface(&This->IDirectSoundCaptureBuffer8_iface, riid, ppobj);
 132 }
 133
 134 static ULONG WINAPI IDirectSoundNotifyImpl_AddRef(IDirectSoundNotify *iface)
 135 {
 136     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundNotify(iface);
 137     ULONG ref = InterlockedIncrement(&This->refn);
 138
 139     TRACE("(%p) ref %d\n", This, ref);
 140
 141     if(ref == 1)
 142         InterlockedIncrement(&This->numIfaces);
 143
 144     return ref;
 145 }
 146
 147 static ULONG WINAPI IDirectSoundNotifyImpl_Release(IDirectSoundNotify *iface)
 148 {
 149     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundNotify(iface);
 150     ULONG ref = InterlockedDecrement(&This->refn);
 151
 152     TRACE("(%p) ref %d\n", This, ref);
 153
 154     if (!ref && !InterlockedDecrement(&This->numIfaces))
 155         capturebuffer_destroy(This);
 156
 157     return ref;
 158 }
 159
 160 static HRESULT WINAPI IDirectSoundNotifyImpl_SetNotificationPositions(IDirectSoundNotify *iface,
 161         DWORD howmuch, const DSBPOSITIONNOTIFY *notify)
 162 {
 163     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundNotify(iface);
 164     TRACE("(%p,0x%08x,%p)\n",This,howmuch,notify);
 165
 166     if (howmuch > 0 && notify == NULL) {
 167         WARN("invalid parameter: notify == NULL\n");
 168         return DSERR_INVALIDPARAM;
 169     }
 170
 171     if (TRACE_ON(dsound)) {
 172         unsigned int i;
 173         for (i=0;i<howmuch;i++)
 174             TRACE("notify at %d to %p\n",
 175             notify[i].dwOffset,notify[i].hEventNotify);
 176     }
 177
 178     if (howmuch > 0) {
 179         /* Make an internal copy of the caller-supplied array.
 180          * Replace the existing copy if one is already present. */
 181         if (This->notifies)
 182             This->notifies = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, This->notifies,
 183                     howmuch * sizeof(DSBPOSITIONNOTIFY));
 184         else
 185             This->notifies = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
 186                     howmuch * sizeof(DSBPOSITIONNOTIFY));
 187
 188         if (!This->notifies) {
 189             WARN("out of memory\n");
 190             return DSERR_OUTOFMEMORY;
 191         }
 192         CopyMemory(This->notifies, notify, howmuch * sizeof(DSBPOSITIONNOTIFY));
 193         This->nrofnotifies = howmuch;
 194     } else {
 195         HeapFree(GetProcessHeap(), 0, This->notifies);
 196         This->notifies = NULL;
 197         This->nrofnotifies = 0;
 198     }
 199
 200     return S_OK;
 201 }
 202
 203 static const IDirectSoundNotifyVtbl dscnvt =
 204 {
 205     IDirectSoundNotifyImpl_QueryInterface,
 206     IDirectSoundNotifyImpl_AddRef,
 207     IDirectSoundNotifyImpl_Release,
 208     IDirectSoundNotifyImpl_SetNotificationPositions
 209 };
 210
 211
 212 static const char * const captureStateString[] = {
 213     "STATE_STOPPED",
 214     "STATE_STARTING",
 215     "STATE_CAPTURING",
 216     "STATE_STOPPING"
 217 };
 218
 219
 220 /*******************************************************************************
 221  * IDirectSoundCaptureBuffer
 222  */
 223 static inline IDirectSoundCaptureBufferImpl *impl_from_IDirectSoundCaptureBuffer8(IDirectSoundCaptureBuffer8 *iface)
 224 {
 225     return CONTAINING_RECORD(iface, IDirectSoundCaptureBufferImpl, IDirectSoundCaptureBuffer8_iface);
 226 }
 227
 228 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_QueryInterface(IDirectSoundCaptureBuffer8 *iface,
 229         REFIID riid, void **ppobj)
 230 {
 231     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 232
 233     TRACE( "(%p,%s,%p)\n", This, debugstr_guid(riid), ppobj );
 234
 235     if (ppobj == NULL) {
 236         WARN("invalid parameter\n");
 237         return E_INVALIDARG;
 238     }
 239
 240     *ppobj = NULL;
 241
 242     if ( IsEqualIID( &IID_IUnknown, riid ) ||
 243          IsEqualIID( &IID_IDirectSoundCaptureBuffer, riid ) ||
 244          (This->has_dsc8 && IsEqualIID( &IID_IDirectSoundCaptureBuffer8, riid )) ) {
 245         IDirectSoundCaptureBuffer8_AddRef(iface);
 246         *ppobj = iface;
 247         return S_OK;
 248     }
 249
 250     if ( IsEqualGUID( &IID_IDirectSoundNotify, riid ) ) {
 251         IDirectSoundNotify_AddRef(&This->IDirectSoundNotify_iface);
 252         *ppobj = &This->IDirectSoundNotify_iface;
 253         return S_OK;
 254     }
 255
 256     FIXME("(%p,%s,%p) unsupported GUID\n", This, debugstr_guid(riid), ppobj);
 257     return E_NOINTERFACE;
 258 }
 259
 260 static ULONG WINAPI IDirectSoundCaptureBufferImpl_AddRef(IDirectSoundCaptureBuffer8 *iface)
 261 {
 262     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 263     ULONG ref = InterlockedIncrement(&This->ref);
 264
 265     TRACE("(%p) ref %d\n", This, ref);
 266
 267     if(ref == 1)
 268         InterlockedIncrement(&This->numIfaces);
 269
 270     return ref;
 271 }
 272
 273 static ULONG WINAPI IDirectSoundCaptureBufferImpl_Release(IDirectSoundCaptureBuffer8 *iface)
 274 {
 275     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 276     ULONG ref = InterlockedDecrement(&This->ref);
 277
 278     TRACE("(%p) ref %d\n", This, ref);
 279
 280     if (!ref && !InterlockedDecrement(&This->numIfaces))
 281         capturebuffer_destroy(This);
 282
 283     return ref;
 284 }
 285
 286 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_GetCaps(IDirectSoundCaptureBuffer8 *iface,
 287         DSCBCAPS *lpDSCBCaps)
 288 {
 289     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 290     TRACE( "(%p,%p)\n", This, lpDSCBCaps );
 291
 292     if (lpDSCBCaps == NULL) {
 293         WARN("invalid parameter: lpDSCBCaps == NULL\n");
 294         return DSERR_INVALIDPARAM;
 295     }
 296
 297     if (lpDSCBCaps->dwSize < sizeof(DSCBCAPS)) {
 298         WARN("invalid parameter: lpDSCBCaps->dwSize = %d\n", lpDSCBCaps->dwSize);
 299         return DSERR_INVALIDPARAM;
 300     }
 301
 302     if (This->device == NULL) {
 303         WARN("invalid parameter: This->device == NULL\n");
 304         return DSERR_INVALIDPARAM;
 305     }
 306
 307     lpDSCBCaps->dwSize = sizeof(DSCBCAPS);
 308     lpDSCBCaps->dwFlags = This->flags;
 309     lpDSCBCaps->dwBufferBytes = This->pdscbd->dwBufferBytes;
 310     lpDSCBCaps->dwReserved = 0;
 311
 312     TRACE("returning DS_OK\n");
 313     return DS_OK;
 314 }
 315
 316 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_GetCurrentPosition(IDirectSoundCaptureBuffer8 *iface,
 317         DWORD *lpdwCapturePosition, DWORD *lpdwReadPosition)
 318 {
 319     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 320
 321     TRACE( "(%p,%p,%p)\n", This, lpdwCapturePosition, lpdwReadPosition );
 322
 323     if (This->device == NULL) {
 324         WARN("invalid parameter: This->device == NULL\n");
 325         return DSERR_INVALIDPARAM;
 326     }
 327
 328     EnterCriticalSection(&This->device->lock);
 329
 330     if (!This->device->client) {
 331         LeaveCriticalSection(&This->device->lock);
 332         WARN("no driver\n");
 333         return DSERR_NODRIVER;
 334     }
 335
 336     if(lpdwCapturePosition)
 337         *lpdwCapturePosition = This->device->write_pos_bytes;
 338
 339     if(lpdwReadPosition)
 340         *lpdwReadPosition = This->device->write_pos_bytes;
 341
 342     LeaveCriticalSection(&This->device->lock);
 343
 344     TRACE("cappos=%d readpos=%d\n", (lpdwCapturePosition?*lpdwCapturePosition:-1), (lpdwReadPosition?*lpdwReadPosition:-1));
 345     TRACE("returning DS_OK\n");
 346
 347     return DS_OK;
 348 }
 349
 350 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_GetFormat(IDirectSoundCaptureBuffer8 *iface,
 351         WAVEFORMATEX *lpwfxFormat, DWORD dwSizeAllocated, DWORD *lpdwSizeWritten)
 352 {
 353     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 354     HRESULT hres = DS_OK;
 355
 356     TRACE("(%p,%p,0x%08x,%p)\n", This, lpwfxFormat, dwSizeAllocated, lpdwSizeWritten);
 357
 358     if (This->device == NULL) {
 359         WARN("invalid parameter: This->device == NULL\n");
 360         return DSERR_INVALIDPARAM;
 361     }
 362
 363     if (dwSizeAllocated > (sizeof(WAVEFORMATEX) + This->device->pwfx->cbSize))
 364         dwSizeAllocated = sizeof(WAVEFORMATEX) + This->device->pwfx->cbSize;
 365
 366     if (lpwfxFormat) { /* NULL is valid (just want size) */
 367         CopyMemory(lpwfxFormat, This->device->pwfx, dwSizeAllocated);
 368         if (lpdwSizeWritten)
 369             *lpdwSizeWritten = dwSizeAllocated;
 370     } else {
 371         if (lpdwSizeWritten)
 372             *lpdwSizeWritten = sizeof(WAVEFORMATEX) + This->device->pwfx->cbSize;
 373         else {
 374             TRACE("invalid parameter: lpdwSizeWritten = NULL\n");
 375             hres = DSERR_INVALIDPARAM;
 376         }
 377     }
 378
 379     TRACE("returning %08x\n", hres);
 380     return hres;
 381 }
 382
 383 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_GetStatus(IDirectSoundCaptureBuffer8 *iface,
 384         DWORD *lpdwStatus)
 385 {
 386     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 387
 388     TRACE( "(%p, %p)\n", This, lpdwStatus );
 389
 390     if (This->device == NULL) {
 391         WARN("invalid parameter: This->device == NULL\n");
 392         return DSERR_INVALIDPARAM;
 393     }
 394
 395     if (lpdwStatus == NULL) {
 396         WARN("invalid parameter: lpdwStatus == NULL\n");
 397         return DSERR_INVALIDPARAM;
 398     }
 399
 400     *lpdwStatus = 0;
 401     EnterCriticalSection(&(This->device->lock));
 402
 403     TRACE("old This->device->state=%s, old lpdwStatus=%08x\n",
 404         captureStateString[This->device->state],*lpdwStatus);
 405     if ((This->device->state == STATE_STARTING) ||
 406         (This->device->state == STATE_CAPTURING)) {
 407         *lpdwStatus |= DSCBSTATUS_CAPTURING;
 408         if (This->flags & DSCBSTART_LOOPING)
 409             *lpdwStatus |= DSCBSTATUS_LOOPING;
 410     }
 411     TRACE("new This->device->state=%s, new lpdwStatus=%08x\n",
 412         captureStateString[This->device->state],*lpdwStatus);
 413     LeaveCriticalSection(&(This->device->lock));
 414
 415     TRACE("status=%x\n", *lpdwStatus);
 416     TRACE("returning DS_OK\n");
 417     return DS_OK;
 418 }
 419
 420 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_Initialize(IDirectSoundCaptureBuffer8 *iface,
 421         IDirectSoundCapture *lpDSC, const DSCBUFFERDESC *lpcDSCBDesc)
 422 {
 423     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 424
 425     FIXME( "(%p,%p,%p): stub\n", This, lpDSC, lpcDSCBDesc );
 426
 427     return DS_OK;
 428 }
 429
 430 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_Lock(IDirectSoundCaptureBuffer8 *iface,
 431         DWORD dwReadCusor, DWORD dwReadBytes, void **lplpvAudioPtr1, DWORD *lpdwAudioBytes1,
 432         void **lplpvAudioPtr2, DWORD *lpdwAudioBytes2, DWORD dwFlags)
 433 {
 434     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 435     HRESULT hres = DS_OK;
 436
 437     TRACE( "(%p,%08u,%08u,%p,%p,%p,%p,0x%08x) at %d\n", This, dwReadCusor,
 438         dwReadBytes, lplpvAudioPtr1, lpdwAudioBytes1, lplpvAudioPtr2,
 439         lpdwAudioBytes2, dwFlags, GetTickCount() );
 440
 441     if (This->device == NULL) {
 442         WARN("invalid parameter: This->device == NULL\n");
 443         return DSERR_INVALIDPARAM;
 444     }
 445
 446     if (lplpvAudioPtr1 == NULL) {
 447         WARN("invalid parameter: lplpvAudioPtr1 == NULL\n");
 448         return DSERR_INVALIDPARAM;
 449     }
 450
 451     if (lpdwAudioBytes1 == NULL) {
 452         WARN("invalid parameter: lpdwAudioBytes1 == NULL\n");
 453         return DSERR_INVALIDPARAM;
 454     }
 455
 456     EnterCriticalSection(&(This->device->lock));
 457
 458     if (This->device->client) {
 459         *lplpvAudioPtr1 = This->device->buffer + dwReadCusor;
 460         if ( (dwReadCusor + dwReadBytes) > This->device->buflen) {
 461             *lpdwAudioBytes1 = This->device->buflen - dwReadCusor;
 462             if (lplpvAudioPtr2)
 463                 *lplpvAudioPtr2 = This->device->buffer;
 464             if (lpdwAudioBytes2)
 465                 *lpdwAudioBytes2 = dwReadBytes - *lpdwAudioBytes1;
 466         } else {
 467             *lpdwAudioBytes1 = dwReadBytes;
 468             if (lplpvAudioPtr2)
 469                 *lplpvAudioPtr2 = 0;
 470             if (lpdwAudioBytes2)
 471                 *lpdwAudioBytes2 = 0;
 472         }
 473     } else {
 474         TRACE("invalid call\n");
 475         hres = DSERR_INVALIDCALL;   /* DSERR_NODRIVER ? */
 476     }
 477
 478     LeaveCriticalSection(&(This->device->lock));
 479
 480     TRACE("returning %08x\n", hres);
 481     return hres;
 482 }
 483
 484 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_Start(IDirectSoundCaptureBuffer8 *iface,
 485         DWORD dwFlags)
 486 {
 487     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 488     HRESULT hres;
 489
 490     TRACE( "(%p,0x%08x)\n", This, dwFlags );
 491
 492     if (This->device == NULL) {
 493         WARN("invalid parameter: This->device == NULL\n");
 494         return DSERR_INVALIDPARAM;
 495     }
 496
 497     if ( !This->device->client ) {
 498         WARN("no driver\n");
 499         return DSERR_NODRIVER;
 500     }
 501
 502     EnterCriticalSection(&(This->device->lock));
 503
 504     if (This->device->state == STATE_STOPPED)
 505         This->device->state = STATE_STARTING;
 506     else if (This->device->state == STATE_STOPPING)
 507         This->device->state = STATE_CAPTURING;
 508     else
 509         goto out;
 510     TRACE("new This->device->state=%s\n",captureStateString[This->device->state]);
 511     This->flags = dwFlags;
 512
 513     if (This->device->buffer)
 514         FillMemory(This->device->buffer, This->device->buflen, (This->device->pwfx->wBitsPerSample == 8) ? 128 : 0);
 515
 516     hres = IAudioClient_Start(This->device->client);
 517     if(FAILED(hres)){
 518         WARN("Start failed: %08x\n", hres);
 519         LeaveCriticalSection(&This->device->lock);
 520         return hres;
 521     }
 522
 523 out:
 524     LeaveCriticalSection(&This->device->lock);
 525
 526     TRACE("returning DS_OK\n");
 527     return DS_OK;
 528 }
 529
 530 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_Stop(IDirectSoundCaptureBuffer8 *iface)
 531 {
 532     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 533     HRESULT hres;
 534
 535     TRACE("(%p)\n", This);
 536
 537     if (This->device == NULL) {
 538         WARN("invalid parameter: This->device == NULL\n");
 539         return DSERR_INVALIDPARAM;
 540     }
 541
 542     EnterCriticalSection(&(This->device->lock));
 543
 544     TRACE("old This->device->state=%s\n",captureStateString[This->device->state]);
 545     if (This->device->state == STATE_CAPTURING)
 546         This->device->state = STATE_STOPPING;
 547     else if (This->device->state == STATE_STARTING)
 548         This->device->state = STATE_STOPPED;
 549     TRACE("new This->device->state=%s\n",captureStateString[This->device->state]);
 550
 551     if(This->device->client){
 552         hres = IAudioClient_Stop(This->device->client);
 553         if(FAILED(hres)){
 554             LeaveCriticalSection(&This->device->lock);
 555             return hres;
 556         }
 557     }
 558
 559     LeaveCriticalSection(&(This->device->lock));
 560
 561     TRACE("returning DS_OK\n");
 562     return DS_OK;
 563 }
 564
 565 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_Unlock(IDirectSoundCaptureBuffer8 *iface,
 566         void *lpvAudioPtr1, DWORD dwAudioBytes1, void *lpvAudioPtr2, DWORD dwAudioBytes2)
 567 {
 568     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 569     HRESULT hres = DS_OK;
 570
 571     TRACE( "(%p,%p,%08u,%p,%08u)\n", This, lpvAudioPtr1, dwAudioBytes1,
 572         lpvAudioPtr2, dwAudioBytes2 );
 573
 574     if (lpvAudioPtr1 == NULL) {
 575         WARN("invalid parameter: lpvAudioPtr1 == NULL\n");
 576         return DSERR_INVALIDPARAM;
 577     }
 578
 579     if (!This->device->client) {
 580         WARN("invalid call\n");
 581         hres = DSERR_INVALIDCALL;
 582     }
 583
 584     TRACE("returning %08x\n", hres);
 585     return hres;
 586 }
 587
 588 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_GetObjectInPath(IDirectSoundCaptureBuffer8 *iface,
 589         REFGUID rguidObject, DWORD dwIndex, REFGUID rguidInterface, void **ppObject)
 590 {
 591     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 592
 593     FIXME( "(%p,%s,%u,%s,%p): stub\n", This, debugstr_guid(rguidObject),
 594         dwIndex, debugstr_guid(rguidInterface), ppObject );
 595
 596     if (!ppObject)
 597         return DSERR_INVALIDPARAM;
 598
 599     *ppObject = NULL;
 600     return DSERR_CONTROLUNAVAIL;
 601 }
 602
 603 static HRESULT WINAPI IDirectSoundCaptureBufferImpl_GetFXStatus(IDirectSoundCaptureBuffer8 *iface,
 604         DWORD dwFXCount, DWORD *pdwFXStatus)
 605 {
 606     IDirectSoundCaptureBufferImpl *This = impl_from_IDirectSoundCaptureBuffer8(iface);
 607
 608     FIXME( "(%p,%u,%p): stub\n", This, dwFXCount, pdwFXStatus );
 609
 610     return DS_OK;
 611 }
 612
 613 static const IDirectSoundCaptureBuffer8Vtbl dscbvt =
 614 {
 615     /* IUnknown methods */
 616     IDirectSoundCaptureBufferImpl_QueryInterface,
 617     IDirectSoundCaptureBufferImpl_AddRef,
 618     IDirectSoundCaptureBufferImpl_Release,
 619
 620     /* IDirectSoundCaptureBuffer methods */
 621     IDirectSoundCaptureBufferImpl_GetCaps,
 622     IDirectSoundCaptureBufferImpl_GetCurrentPosition,
 623     IDirectSoundCaptureBufferImpl_GetFormat,
 624     IDirectSoundCaptureBufferImpl_GetStatus,
 625     IDirectSoundCaptureBufferImpl_Initialize,
 626     IDirectSoundCaptureBufferImpl_Lock,
 627     IDirectSoundCaptureBufferImpl_Start,
 628     IDirectSoundCaptureBufferImpl_Stop,
 629     IDirectSoundCaptureBufferImpl_Unlock,
 630
 631     /* IDirectSoundCaptureBuffer methods */
 632     IDirectSoundCaptureBufferImpl_GetObjectInPath,
 633     IDirectSoundCaptureBufferImpl_GetFXStatus
 634 };
 635
 636 static void capture_CheckNotify(IDirectSoundCaptureBufferImpl *This, DWORD from, DWORD len)
 637 {
 638     int i;
 639     for (i = 0; i < This->nrofnotifies; ++i) {
 640         LPDSBPOSITIONNOTIFY event = This->notifies + i;
 641         DWORD offset = event->dwOffset;
 642         TRACE("checking %d, position %d, event = %p\n", i, offset, event->hEventNotify);
 643
 644         if (offset == DSBPN_OFFSETSTOP) {
 645             if (!from && !len) {
 646                 SetEvent(event->hEventNotify);
 647                 TRACE("signalled event %p (%d)\n", event->hEventNotify, i);
 648                 return;
 649             }
 650             else return;
 651         }
 652
 653         if (offset >= from && offset < (from + len))
 654         {
 655             TRACE("signalled event %p (%d)\n", event->hEventNotify, i);
 656             SetEvent(event->hEventNotify);
 657         }
 658     }
 659 }
 660
 661 static HRESULT IDirectSoundCaptureBufferImpl_Create(
 662     DirectSoundCaptureDevice *device,
 663     IDirectSoundCaptureBufferImpl ** ppobj,
 664     LPCDSCBUFFERDESC lpcDSCBufferDesc)
 665 {
 666     LPWAVEFORMATEX  wfex;
 667     IDirectSoundCaptureBufferImpl *This;
 668     TRACE( "(%p,%p,%p)\n", device, ppobj, lpcDSCBufferDesc);
 669
 670     if (ppobj == NULL) {
 671         WARN("invalid parameter: ppobj == NULL\n");
 672         return DSERR_INVALIDPARAM;
 673     }
 674
 675     *ppobj = NULL;
 676
 677     if (!device) {
 678         WARN("not initialized\n");
 679         return DSERR_UNINITIALIZED;
 680     }
 681
 682     if (lpcDSCBufferDesc == NULL) {
 683         WARN("invalid parameter: lpcDSCBufferDesc == NULL\n");
 684         return DSERR_INVALIDPARAM;
 685     }
 686
 687     if ( ((lpcDSCBufferDesc->dwSize != sizeof(DSCBUFFERDESC)) &&
 688           (lpcDSCBufferDesc->dwSize != sizeof(DSCBUFFERDESC1))) ||
 689         (lpcDSCBufferDesc->dwBufferBytes == 0) ||
 690         (lpcDSCBufferDesc->lpwfxFormat == NULL) ) { /* FIXME: DSERR_BADFORMAT ? */
 691         WARN("invalid lpcDSCBufferDesc\n");
 692         return DSERR_INVALIDPARAM;
 693     }
 694
 695     wfex = lpcDSCBufferDesc->lpwfxFormat;
 696
 697     TRACE("(formattag=0x%04x,chans=%d,samplerate=%d,"
 698         "bytespersec=%d,blockalign=%d,bitspersamp=%d,cbSize=%d)\n",
 699         wfex->wFormatTag, wfex->nChannels, wfex->nSamplesPerSec,
 700         wfex->nAvgBytesPerSec, wfex->nBlockAlign,
 701         wfex->wBitsPerSample, wfex->cbSize);
 702
 703     device->pwfx = DSOUND_CopyFormat(wfex);
 704     if ( device->pwfx == NULL )
 705         return DSERR_OUTOFMEMORY;
 706
 707     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,
 708         sizeof(IDirectSoundCaptureBufferImpl));
 709
 710     if ( This == NULL ) {
 711         WARN("out of memory\n");
 712         return DSERR_OUTOFMEMORY;
 713     } else {
 714         HRESULT err = DS_OK;
 715         LPBYTE newbuf;
 716         DWORD buflen;
 717
 718         This->numIfaces = 0;
 719         This->ref = 0;
 720         This->refn = 0;
 721         This->device = device;
 722         This->device->capture_buffer = This;
 723         This->nrofnotifies = 0;
 724
 725         This->pdscbd = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,
 726             lpcDSCBufferDesc->dwSize);
 727         if (This->pdscbd)
 728             CopyMemory(This->pdscbd, lpcDSCBufferDesc, lpcDSCBufferDesc->dwSize);
 729         else {
 730             WARN("no memory\n");
 731             This->device->capture_buffer = 0;
 732             HeapFree( GetProcessHeap(), 0, This );
 733             return DSERR_OUTOFMEMORY;
 734         }
 735
 736         This->IDirectSoundCaptureBuffer8_iface.lpVtbl = &dscbvt;
 737         This->IDirectSoundNotify_iface.lpVtbl = &dscnvt;
 738
 739         err = IMMDevice_Activate(device->mmdevice, &IID_IAudioClient,
 740                 CLSCTX_INPROC_SERVER, NULL, (void**)&device->client);
 741         if(FAILED(err)){
 742             WARN("Activate failed: %08x\n", err);
 743             HeapFree(GetProcessHeap(), 0, This->pdscbd);
 744             This->device->capture_buffer = 0;
 745             HeapFree( GetProcessHeap(), 0, This );
 746             return err;
 747         }
 748
 749         err = IAudioClient_Initialize(device->client,
 750                 AUDCLNT_SHAREMODE_SHARED, AUDCLNT_STREAMFLAGS_NOPERSIST | AUDCLNT_STREAMFLAGS_EVENTCALLBACK,
 751                 200 * 100000, 0, device->pwfx, NULL);
 752         if(FAILED(err)){
 753             WARN("Initialize failed: %08x\n", err);
 754             IAudioClient_Release(device->client);
 755             device->client = NULL;
 756             HeapFree(GetProcessHeap(), 0, This->pdscbd);
 757             This->device->capture_buffer = 0;
 758             HeapFree( GetProcessHeap(), 0, This );
 759             if(err == AUDCLNT_E_UNSUPPORTED_FORMAT)
 760                 return DSERR_BADFORMAT;
 761             return err;
 762         }
 763
 764         This->sleepev = CreateEventW(NULL, 0, 0, NULL);
 765
 766         err = IAudioClient_SetEventHandle(device->client, This->sleepev);
 767         if(FAILED(err)){
 768             WARN("SetEventHandle failed: %08x\n", err);
 769             IAudioClient_Release(device->client);
 770             device->client = NULL;
 771             CloseHandle(This->sleepev);
 772             HeapFree(GetProcessHeap(), 0, This->pdscbd);
 773             This->device->capture_buffer = 0;
 774             HeapFree( GetProcessHeap(), 0, This );
 775             return err;
 776         }
 777
 778         err = IAudioClient_GetService(device->client, &IID_IAudioCaptureClient,
 779                 (void**)&device->capture);
 780         if(FAILED(err)){
 781             WARN("GetService failed: %08x\n", err);
 782             IAudioClient_Release(device->client);
 783             device->client = NULL;
 784             CloseHandle(This->sleepev);
 785             HeapFree(GetProcessHeap(), 0, This->pdscbd);
 786             This->device->capture_buffer = 0;
 787             HeapFree( GetProcessHeap(), 0, This );
 788             return err;
 789         }
 790
 791         buflen = lpcDSCBufferDesc->dwBufferBytes;
 792         TRACE("desired buflen=%d, old buffer=%p\n", buflen, device->buffer);
 793         if (device->buffer)
 794             newbuf = HeapReAlloc(GetProcessHeap(),0,device->buffer,buflen);
 795         else
 796             newbuf = HeapAlloc(GetProcessHeap(),0,buflen);
 797         if (newbuf == NULL) {
 798             IAudioClient_Release(device->client);
 799             device->client = NULL;
 800             IAudioCaptureClient_Release(device->capture);
 801             device->capture = NULL;
 802             CloseHandle(This->sleepev);
 803             HeapFree(GetProcessHeap(), 0, This->pdscbd);
 804             This->device->capture_buffer = 0;
 805             HeapFree( GetProcessHeap(), 0, This );
 806             return DSERR_OUTOFMEMORY;
 807         }
 808         device->buffer = newbuf;
 809         device->buflen = buflen;
 810         This->thread = CreateThread(NULL, 0, DSOUND_capture_thread, This, 0, NULL);
 811     }
 812
 813     IDirectSoundCaptureBuffer8_AddRef(&This->IDirectSoundCaptureBuffer8_iface);
 814     *ppobj = This;
 815
 816     TRACE("returning DS_OK\n");
 817     return DS_OK;
 818 }
 819
 820
 821 /*******************************************************************************
 822  * DirectSoundCaptureDevice
 823  */
 824 static HRESULT DirectSoundCaptureDevice_Create(
 825     DirectSoundCaptureDevice ** ppDevice)
 826 {
 827     DirectSoundCaptureDevice * device;
 828     TRACE("(%p)\n", ppDevice);
 829
 830     /* Allocate memory */
 831     device = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DirectSoundCaptureDevice));
 832
 833     if (device == NULL) {
 834         WARN("out of memory\n");
 835         return DSERR_OUTOFMEMORY;
 836     }
 837
 838     device->ref = 1;
 839     device->state = STATE_STOPPED;
 840
 841     InitializeCriticalSection( &(device->lock) );
 842     device->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": DirectSoundCaptureDevice.lock");
 843
 844     *ppDevice = device;
 845
 846     return DS_OK;
 847 }
 848
 849 static ULONG DirectSoundCaptureDevice_Release(
 850     DirectSoundCaptureDevice * device)
 851 {
 852     ULONG ref = InterlockedDecrement(&(device->ref));
 853     TRACE("(%p) ref %d\n", device, ref);
 854
 855     if (!ref) {
 856         TRACE("deleting object\n");
 857
 858         EnterCriticalSection(&DSOUND_capturers_lock);
 859         list_remove(&device->entry);
 860         LeaveCriticalSection(&DSOUND_capturers_lock);
 861
 862         if (device->capture_buffer)
 863             IDirectSoundCaptureBufferImpl_Release(&device->capture_buffer->IDirectSoundCaptureBuffer8_iface);
 864
 865         if(device->mmdevice)
 866             IMMDevice_Release(device->mmdevice);
 867         HeapFree(GetProcessHeap(), 0, device->pwfx);
 868         device->lock.DebugInfo->Spare[0] = 0;
 869         DeleteCriticalSection( &(device->lock) );
 870         HeapFree(GetProcessHeap(), 0, device);
 871         TRACE("(%p) released\n", device);
 872     }
 873     return ref;
 874 }
 875
 876 static HRESULT DSOUND_capture_data(DirectSoundCaptureDevice *device)
 877 {
 878     if(!device->capture_buffer || device->state == STATE_STOPPED)
 879         return S_FALSE;
 880
 881     if(device->state == STATE_STOPPING){
 882         device->state = STATE_STOPPED;
 883         return S_FALSE;
 884     }
 885
 886     if(device->state == STATE_STARTING)
 887         device->state = STATE_CAPTURING;
 888
 889     while(1){
 890         HRESULT hr;
 891         UINT32 packet_frames, packet_bytes, avail_bytes, skip_bytes = 0;
 892         DWORD flags;
 893         BYTE *buf;
 894
 895         hr = IAudioCaptureClient_GetBuffer(device->capture, &buf, &packet_frames,
 896                 &flags, NULL, NULL);
 897         if(FAILED(hr)){
 898             WARN("GetBuffer failed: %08x\n", hr);
 899             return hr;
 900         }
 901         if(hr == AUDCLNT_S_BUFFER_EMPTY)
 902             break;
 903
 904         packet_bytes = packet_frames * device->pwfx->nBlockAlign;
 905         if(packet_bytes > device->buflen){
 906             TRACE("audio glitch: dsound buffer too small for data\n");
 907             skip_bytes = packet_bytes - device->buflen;
 908             packet_bytes = device->buflen;
 909         }
 910
 911         avail_bytes = device->buflen - device->write_pos_bytes;
 912         if(avail_bytes > packet_bytes)
 913             avail_bytes = packet_bytes;
 914
 915         memcpy(device->buffer + device->write_pos_bytes, buf + skip_bytes, avail_bytes);
 916         capture_CheckNotify(device->capture_buffer, device->write_pos_bytes, avail_bytes);
 917
 918         packet_bytes -= avail_bytes;
 919         if(packet_bytes > 0){
 920             if(device->capture_buffer->flags & DSCBSTART_LOOPING){
 921                 memcpy(device->buffer, buf + skip_bytes + avail_bytes, packet_bytes);
 922                 capture_CheckNotify(device->capture_buffer, 0, packet_bytes);
 923             }else{
 924                 device->state = STATE_STOPPED;
 925                 capture_CheckNotify(device->capture_buffer, 0, 0);
 926             }
 927         }
 928
 929         device->write_pos_bytes += avail_bytes + packet_bytes;
 930         device->write_pos_bytes %= device->buflen;
 931
 932         hr = IAudioCaptureClient_ReleaseBuffer(device->capture, packet_frames);
 933         if(FAILED(hr)){
 934             WARN("ReleaseBuffer failed: %08x\n", hr);
 935             return hr;
 936         }
 937     }
 938
 939     return S_OK;
 940 }
 941
 942 static DWORD WINAPI DSOUND_capture_thread(void *user)
 943 {
 944     IDirectSoundCaptureBufferImpl *buffer = user;
 945     HRESULT hr;
 946     DWORD ret, wait_ms;
 947     REFERENCE_TIME period;
 948
 949     hr = IAudioClient_GetDevicePeriod(buffer->device->client, &period, NULL);
 950     if(FAILED(hr)){
 951         WARN("GetDevicePeriod failed: %08x\n", hr);
 952         wait_ms = 5;
 953     }else
 954         wait_ms = MulDiv(5, period, 10000);
 955
 956     while(buffer->ref){
 957         ret = WaitForSingleObject(buffer->sleepev, wait_ms);
 958
 959         if(!buffer->device->ref)
 960             break;
 961
 962         if(ret == WAIT_OBJECT_0){
 963             EnterCriticalSection(&buffer->device->lock);
 964
 965             DSOUND_capture_data(buffer->device);
 966
 967             LeaveCriticalSection(&buffer->device->lock);
 968         }else if(ret != WAIT_TIMEOUT)
 969             WARN("WaitForSingleObject failed: %u\n", GetLastError());
 970     }
 971
 972     return 0;
 973 }
 974
 975 static struct _TestFormat {
 976     DWORD flag;
 977     DWORD rate;
 978     DWORD depth;
 979     WORD channels;
 980 } formats_to_test[] = {
 981     { WAVE_FORMAT_1M08, 11025, 8, 1 },
 982     { WAVE_FORMAT_1M16, 11025, 16, 1 },
 983     { WAVE_FORMAT_1S08, 11025, 8, 2 },
 984     { WAVE_FORMAT_1S16, 11025, 16, 2 },
 985     { WAVE_FORMAT_2M08, 22050, 8, 1 },
 986     { WAVE_FORMAT_2M16, 22050, 16, 1 },
 987     { WAVE_FORMAT_2S08, 22050, 8, 2 },
 988     { WAVE_FORMAT_2S16, 22050, 16, 2 },
 989     { WAVE_FORMAT_4M08, 44100, 8, 1 },
 990     { WAVE_FORMAT_4M16, 44100, 16, 1 },
 991     { WAVE_FORMAT_4S08, 44100, 8, 2 },
 992     { WAVE_FORMAT_4S16, 44100, 16, 2 },
 993     { WAVE_FORMAT_48M08, 48000, 8, 1 },
 994     { WAVE_FORMAT_48M16, 48000, 16, 1 },
 995     { WAVE_FORMAT_48S08, 48000, 8, 2 },
 996     { WAVE_FORMAT_48S16, 48000, 16, 2 },
 997     { WAVE_FORMAT_96M08, 96000, 8, 1 },
 998     { WAVE_FORMAT_96M16, 96000, 16, 1 },
 999     { WAVE_FORMAT_96S08, 96000, 8, 2 },
1000     { WAVE_FORMAT_96S16, 96000, 16, 2 },
1001     {0}
1002 };
1003
1004 static HRESULT DirectSoundCaptureDevice_Initialize(
1005     DirectSoundCaptureDevice ** ppDevice,
1006     LPCGUID lpcGUID)
1007 {
1008     HRESULT hr;
1009     GUID devGUID;
1010     IMMDevice *mmdevice;
1011     struct _TestFormat *fmt;
1012     DirectSoundCaptureDevice *device;
1013     IAudioClient *client;
1014
1015     TRACE("(%p, %s)\n", ppDevice, debugstr_guid(lpcGUID));
1016
1017     /* Default device? */
1018     if ( !lpcGUID || IsEqualGUID(lpcGUID, &GUID_NULL) )
1019         lpcGUID = &DSDEVID_DefaultCapture;
1020
1021     if(IsEqualGUID(lpcGUID, &DSDEVID_DefaultPlayback) ||
1022             IsEqualGUID(lpcGUID, &DSDEVID_DefaultVoicePlayback))
1023         return DSERR_NODRIVER;
1024
1025     if (GetDeviceID(lpcGUID, &devGUID) != DS_OK) {
1026         WARN("invalid parameter: lpcGUID\n");
1027         return DSERR_INVALIDPARAM;
1028     }
1029
1030     hr = get_mmdevice(eCapture, &devGUID, &mmdevice);
1031     if(FAILED(hr))
1032         return hr;
1033
1034     EnterCriticalSection(&DSOUND_capturers_lock);
1035
1036     hr = DirectSoundCaptureDevice_Create(&device);
1037     if (hr != DS_OK) {
1038         WARN("DirectSoundCaptureDevice_Create failed\n");
1039         LeaveCriticalSection(&DSOUND_capturers_lock);
1040         return hr;
1041     }
1042
1043     device->guid = devGUID;
1044
1045     device->mmdevice = mmdevice;
1046
1047     device->drvcaps.dwFlags = 0;
1048
1049     device->drvcaps.dwFormats = 0;
1050     device->drvcaps.dwChannels = 0;
1051     hr = IMMDevice_Activate(mmdevice, &IID_IAudioClient,
1052             CLSCTX_INPROC_SERVER, NULL, (void**)&client);
1053     if(FAILED(hr)){
1054         device->lock.DebugInfo->Spare[0] = 0;
1055         DeleteCriticalSection(&device->lock);
1056         HeapFree(GetProcessHeap(), 0, device);
1057         LeaveCriticalSection(&DSOUND_capturers_lock);
1058         return DSERR_NODRIVER;
1059     }
1060
1061     for(fmt = formats_to_test; fmt->flag; ++fmt){
1062         if(DSOUND_check_supported(client, fmt->rate, fmt->depth, fmt->channels)){
1063             device->drvcaps.dwFormats |= fmt->flag;
1064             if(fmt->channels > device->drvcaps.dwChannels)
1065                 device->drvcaps.dwChannels = fmt->channels;
1066         }
1067     }
1068     IAudioClient_Release(client);
1069
1070     list_add_tail(&DSOUND_capturers, &device->entry);
1071
1072     *ppDevice = device;
1073
1074     LeaveCriticalSection(&DSOUND_capturers_lock);
1075
1076     return S_OK;
1077 }
1078
1079
1080 /*****************************************************************************
1081  * IDirectSoundCapture implementation structure
1082  */
1083 typedef struct IDirectSoundCaptureImpl
1084 {
1085     IUnknown                 IUnknown_inner;
1086     IDirectSoundCapture      IDirectSoundCapture_iface;
1087     LONG                     ref, refdsc, numIfaces;
1088     IUnknown                 *outer_unk;        /* internal */
1089     DirectSoundCaptureDevice *device;
1090     BOOL                     has_dsc8;
1091 } IDirectSoundCaptureImpl;
1092
1093 static void capture_destroy(IDirectSoundCaptureImpl *This)
1094 {
1095     if (This->device)
1096         DirectSoundCaptureDevice_Release(This->device);
1097     HeapFree(GetProcessHeap(),0,This);
1098     TRACE("(%p) released\n", This);
1099 }
1100
1101 /*******************************************************************************
1102  *      IUnknown Implementation for DirectSoundCapture
1103  */
1104 static inline IDirectSoundCaptureImpl *impl_from_IUnknown(IUnknown *iface)
1105 {
1106     return CONTAINING_RECORD(iface, IDirectSoundCaptureImpl, IUnknown_inner);
1107 }
1108
1109 static HRESULT WINAPI IUnknownImpl_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
1110 {
1111     IDirectSoundCaptureImpl *This = impl_from_IUnknown(iface);
1112
1113     TRACE("(%p,%s,%p)\n", This, debugstr_guid(riid), ppv);
1114
1115     if (!ppv) {
1116         WARN("invalid parameter\n");
1117         return E_INVALIDARG;
1118     }
1119     *ppv = NULL;
1120
1121     if (IsEqualIID(riid, &IID_IUnknown))
1122         *ppv = &This->IUnknown_inner;
1123     else if (IsEqualIID(riid, &IID_IDirectSoundCapture))
1124         *ppv = &This->IDirectSoundCapture_iface;
1125     else {
1126         WARN("unknown IID %s\n", debugstr_guid(riid));
1127         return E_NOINTERFACE;
1128     }
1129
1130     IUnknown_AddRef((IUnknown*)*ppv);
1131     return S_OK;
1132 }
1133
1134 static ULONG WINAPI IUnknownImpl_AddRef(IUnknown *iface)
1135 {
1136     IDirectSoundCaptureImpl *This = impl_from_IUnknown(iface);
1137     ULONG ref = InterlockedIncrement(&This->ref);
1138
1139     TRACE("(%p) ref=%d\n", This, ref);
1140
1141     if(ref == 1)
1142         InterlockedIncrement(&This->numIfaces);
1143     return ref;
1144 }
1145
1146 static ULONG WINAPI IUnknownImpl_Release(IUnknown *iface)
1147 {
1148     IDirectSoundCaptureImpl *This = impl_from_IUnknown(iface);
1149     ULONG ref = InterlockedDecrement(&This->ref);
1150
1151     TRACE("(%p) ref=%d\n", This, ref);
1152
1153     if (!ref && !InterlockedDecrement(&This->numIfaces))
1154         capture_destroy(This);
1155     return ref;
1156 }
1157
1158 static const IUnknownVtbl unk_vtbl =
1159 {
1160     IUnknownImpl_QueryInterface,
1161     IUnknownImpl_AddRef,
1162     IUnknownImpl_Release
1163 };
1164
1165 /***************************************************************************
1166  * IDirectSoundCaptureImpl
1167  */
1168 static inline struct IDirectSoundCaptureImpl *impl_from_IDirectSoundCapture(IDirectSoundCapture *iface)
1169 {
1170     return CONTAINING_RECORD(iface, struct IDirectSoundCaptureImpl, IDirectSoundCapture_iface);
1171 }
1172
1173 static HRESULT WINAPI IDirectSoundCaptureImpl_QueryInterface(IDirectSoundCapture *iface,
1174         REFIID riid, void **ppv)
1175 {
1176     IDirectSoundCaptureImpl *This = impl_from_IDirectSoundCapture(iface);
1177     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(riid), ppv);
1178     return IUnknown_QueryInterface(This->outer_unk, riid, ppv);
1179 }
1180
1181 static ULONG WINAPI IDirectSoundCaptureImpl_AddRef(IDirectSoundCapture *iface)
1182 {
1183     IDirectSoundCaptureImpl *This = impl_from_IDirectSoundCapture(iface);
1184     ULONG ref = InterlockedIncrement(&This->refdsc);
1185
1186     TRACE("(%p) ref=%d\n", This, ref);
1187
1188     if(ref == 1)
1189         InterlockedIncrement(&This->numIfaces);
1190     return ref;
1191 }
1192
1193 static ULONG WINAPI IDirectSoundCaptureImpl_Release(IDirectSoundCapture *iface)
1194 {
1195     IDirectSoundCaptureImpl *This = impl_from_IDirectSoundCapture(iface);
1196     ULONG ref = InterlockedDecrement(&This->refdsc);
1197
1198     TRACE("(%p) ref=%d\n", This, ref);
1199
1200     if (!ref && !InterlockedDecrement(&This->numIfaces))
1201         capture_destroy(This);
1202     return ref;
1203 }
1204
1205 static HRESULT WINAPI IDirectSoundCaptureImpl_CreateCaptureBuffer(IDirectSoundCapture *iface,
1206         LPCDSCBUFFERDESC lpcDSCBufferDesc, IDirectSoundCaptureBuffer **lplpDSCaptureBuffer,
1207         IUnknown *pUnk)
1208 {
1209     IDirectSoundCaptureImpl *This = impl_from_IDirectSoundCapture(iface);
1210     HRESULT hr;
1211
1212     TRACE( "(%p,%p,%p,%p)\n",iface,lpcDSCBufferDesc,lplpDSCaptureBuffer,pUnk);
1213
1214     if (pUnk) {
1215         WARN("invalid parameter: pUnk != NULL\n");
1216         return DSERR_NOAGGREGATION;
1217     }
1218
1219     if (lpcDSCBufferDesc == NULL) {
1220         WARN("invalid parameter: lpcDSCBufferDesc == NULL)\n");
1221         return DSERR_INVALIDPARAM;
1222     }
1223
1224     if (lplpDSCaptureBuffer == NULL) {
1225         WARN("invalid parameter: lplpDSCaptureBuffer == NULL\n");
1226         return DSERR_INVALIDPARAM;
1227     }
1228
1229     /* FIXME: We can only have one buffer so what do we do here? */
1230     if (This->device->capture_buffer) {
1231         WARN("invalid parameter: already has buffer\n");
1232         return DSERR_INVALIDPARAM;    /* DSERR_GENERIC ? */
1233     }
1234
1235     hr = IDirectSoundCaptureBufferImpl_Create(This->device,
1236         (IDirectSoundCaptureBufferImpl **)lplpDSCaptureBuffer, lpcDSCBufferDesc);
1237
1238     if (hr != DS_OK)
1239         WARN("IDirectSoundCaptureBufferImpl_Create failed\n");
1240     else
1241         This->device->capture_buffer->has_dsc8 = This->has_dsc8;
1242
1243     return hr;
1244 }
1245
1246 static HRESULT WINAPI IDirectSoundCaptureImpl_GetCaps(IDirectSoundCapture *iface,
1247         LPDSCCAPS lpDSCCaps)
1248 {
1249     IDirectSoundCaptureImpl *This = impl_from_IDirectSoundCapture(iface);
1250
1251     TRACE("(%p,%p)\n",This,lpDSCCaps);
1252
1253     if (This->device == NULL) {
1254         WARN("not initialized\n");
1255         return DSERR_UNINITIALIZED;
1256     }
1257
1258     if (lpDSCCaps== NULL) {
1259         WARN("invalid parameter: lpDSCCaps== NULL\n");
1260         return DSERR_INVALIDPARAM;
1261     }
1262
1263     if (lpDSCCaps->dwSize < sizeof(*lpDSCCaps)) {
1264         WARN("invalid parameter: lpDSCCaps->dwSize = %d\n", lpDSCCaps->dwSize);
1265         return DSERR_INVALIDPARAM;
1266     }
1267
1268     lpDSCCaps->dwFlags = This->device->drvcaps.dwFlags;
1269     lpDSCCaps->dwFormats = This->device->drvcaps.dwFormats;
1270     lpDSCCaps->dwChannels = This->device->drvcaps.dwChannels;
1271
1272     TRACE("(flags=0x%08x,format=0x%08x,channels=%d)\n",lpDSCCaps->dwFlags,
1273         lpDSCCaps->dwFormats, lpDSCCaps->dwChannels);
1274
1275     return DS_OK;
1276 }
1277
1278 static HRESULT WINAPI IDirectSoundCaptureImpl_Initialize(IDirectSoundCapture *iface,
1279         LPCGUID lpcGUID)
1280 {
1281     IDirectSoundCaptureImpl *This = impl_from_IDirectSoundCapture(iface);
1282
1283     TRACE("(%p,%s)\n", This, debugstr_guid(lpcGUID));
1284
1285     if (This->device != NULL) {
1286         WARN("already initialized\n");
1287         return DSERR_ALREADYINITIALIZED;
1288     }
1289     return DirectSoundCaptureDevice_Initialize(&This->device, lpcGUID);
1290 }
1291
1292 static const IDirectSoundCaptureVtbl dscvt =
1293 {
1294     /* IUnknown methods */
1295     IDirectSoundCaptureImpl_QueryInterface,
1296     IDirectSoundCaptureImpl_AddRef,
1297     IDirectSoundCaptureImpl_Release,
1298
1299     /* IDirectSoundCapture methods */
1300     IDirectSoundCaptureImpl_CreateCaptureBuffer,
1301     IDirectSoundCaptureImpl_GetCaps,
1302     IDirectSoundCaptureImpl_Initialize
1303 };
1304
1305 HRESULT IDirectSoundCaptureImpl_Create(IUnknown *outer_unk, REFIID riid, void **ppv, BOOL has_dsc8)
1306 {
1307     IDirectSoundCaptureImpl *obj;
1308     HRESULT hr;
1309
1310     TRACE("(%s, %p)\n", debugstr_guid(riid), ppv);
1311
1312     *ppv = NULL;
1313     obj = HeapAlloc(GetProcessHeap(), 0, sizeof(*obj));
1314     if (obj == NULL) {
1315         WARN("out of memory\n");
1316         return DSERR_OUTOFMEMORY;
1317     }
1318
1319     setup_dsound_options();
1320
1321     obj->IUnknown_inner.lpVtbl = &unk_vtbl;
1322     obj->IDirectSoundCapture_iface.lpVtbl = &dscvt;
1323     obj->ref = 1;
1324     obj->refdsc = 0;
1325     obj->numIfaces = 1;
1326     obj->device = NULL;
1327     obj->has_dsc8 = has_dsc8;
1328
1329     /* COM aggregation supported only internally */
1330     if (outer_unk)
1331         obj->outer_unk = outer_unk;
1332     else
1333         obj->outer_unk = &obj->IUnknown_inner;
1334
1335     hr = IUnknown_QueryInterface(&obj->IUnknown_inner, riid, ppv);
1336     IUnknown_Release(&obj->IUnknown_inner);
1337
1338     return hr;
1339 }
1340
1341 HRESULT DSOUND_CaptureCreate(REFIID riid, void **ppv)
1342 {
1343     return IDirectSoundCaptureImpl_Create(NULL, riid, ppv, FALSE);
1344 }
1345
1346 HRESULT DSOUND_CaptureCreate8(REFIID riid, void **ppv)
1347 {
1348     return IDirectSoundCaptureImpl_Create(NULL, riid, ppv, TRUE);
1349 }
1350
1351 /***************************************************************************
1352  * DirectSoundCaptureCreate [DSOUND.6]
1353  *
1354  * Create and initialize a DirectSoundCapture interface.
1355  *
1356  * PARAMS
1357  *    lpcGUID   [I] Address of the GUID that identifies the sound capture device.
1358  *    lplpDSC   [O] Address of a variable to receive the interface pointer.
1359  *    pUnkOuter [I] Must be NULL.
1360  *
1361  * RETURNS
1362  *    Success: DS_OK
1363  *    Failure: DSERR_NOAGGREGATION, DSERR_ALLOCATED, DSERR_INVALIDPARAM,
1364  *             DSERR_OUTOFMEMORY
1365  *
1366  * NOTES
1367  *    lpcGUID must be one of the values returned from DirectSoundCaptureEnumerate
1368  *    or NULL for the default device or DSDEVID_DefaultCapture or
1369  *    DSDEVID_DefaultVoiceCapture.
1370  *
1371  *    DSERR_ALLOCATED is returned for sound devices that do not support full duplex.
1372  */
1373 HRESULT WINAPI DirectSoundCaptureCreate(LPCGUID lpcGUID, IDirectSoundCapture **ppDSC,
1374         IUnknown *pUnkOuter)
1375 {
1376     HRESULT hr;
1377     IDirectSoundCapture *pDSC;
1378
1379     TRACE("(%s,%p,%p)\n", debugstr_guid(lpcGUID), ppDSC, pUnkOuter);
1380
1381     if (ppDSC == NULL) {
1382         WARN("invalid parameter: ppDSC == NULL\n");
1383         return DSERR_INVALIDPARAM;
1384     }
1385
1386     if (pUnkOuter) {
1387         WARN("invalid parameter: pUnkOuter != NULL\n");
1388         return DSERR_NOAGGREGATION;
1389     }
1390
1391     hr = DSOUND_CaptureCreate(&IID_IDirectSoundCapture, (void**)&pDSC);
1392     if (hr == DS_OK) {
1393         hr = IDirectSoundCapture_Initialize(pDSC, lpcGUID);
1394         if (hr != DS_OK) {
1395             IDirectSoundCapture_Release(pDSC);
1396             pDSC = 0;
1397         }
1398     }
1399
1400     *ppDSC = pDSC;
1401
1402     return hr;
1403 }
1404
1405 /***************************************************************************
1406  * DirectSoundCaptureCreate8 [DSOUND.12]
1407  *
1408  * Create and initialize a DirectSoundCapture interface.
1409  *
1410  * PARAMS
1411  *    lpcGUID   [I] Address of the GUID that identifies the sound capture device.
1412  *    lplpDSC   [O] Address of a variable to receive the interface pointer.
1413  *    pUnkOuter [I] Must be NULL.
1414  *
1415  * RETURNS
1416  *    Success: DS_OK
1417  *    Failure: DSERR_NOAGGREGATION, DSERR_ALLOCATED, DSERR_INVALIDPARAM,
1418  *             DSERR_OUTOFMEMORY
1419  *
1420  * NOTES
1421  *    lpcGUID must be one of the values returned from DirectSoundCaptureEnumerate
1422  *    or NULL for the default device or DSDEVID_DefaultCapture or
1423  *    DSDEVID_DefaultVoiceCapture.
1424  *
1425  *    DSERR_ALLOCATED is returned for sound devices that do not support full duplex.
1426  */
1427 HRESULT WINAPI DirectSoundCaptureCreate8(
1428     LPCGUID lpcGUID,
1429     LPDIRECTSOUNDCAPTURE8 *ppDSC8,
1430     LPUNKNOWN pUnkOuter)
1431 {
1432     HRESULT hr;
1433     LPDIRECTSOUNDCAPTURE8 pDSC8;
1434     TRACE("(%s,%p,%p)\n", debugstr_guid(lpcGUID), ppDSC8, pUnkOuter);
1435
1436     if (ppDSC8 == NULL) {
1437         WARN("invalid parameter: ppDSC8 == NULL\n");
1438         return DSERR_INVALIDPARAM;
1439     }
1440
1441     if (pUnkOuter) {
1442         WARN("invalid parameter: pUnkOuter != NULL\n");
1443         *ppDSC8 = NULL;
1444         return DSERR_NOAGGREGATION;
1445     }
1446
1447     hr = DSOUND_CaptureCreate8(&IID_IDirectSoundCapture8, (void**)&pDSC8);
1448     if (hr == DS_OK) {
1449         hr = IDirectSoundCapture_Initialize(pDSC8, lpcGUID);
1450         if (hr != DS_OK) {
1451             IDirectSoundCapture_Release(pDSC8);
1452             pDSC8 = 0;
1453         }
1454     }
1455
1456     *ppDSC8 = pDSC8;
1457
1458     return hr;
1459 }