dlls/mmdevapi/main.c

   1 /*
   2  * Copyright 2009 Maarten Lankhorst
   3  * Copyright 2011 Andrew Eikum for CodeWeavers
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #include <stdarg.h>
  21
  22 #define COBJMACROS
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "wingdi.h"
  26
  27 #include "ole2.h"
  28 #include "olectl.h"
  29 #include "rpcproxy.h"
  30 #include "propsys.h"
  31 #include "propkeydef.h"
  32 #include "mmdeviceapi.h"
  33 #include "mmsystem.h"
  34 #include "dsound.h"
  35 #include "audioclient.h"
  36 #include "endpointvolume.h"
  37 #include "audiopolicy.h"
  38 #include "devpkey.h"
  39 #include "winreg.h"
  40
  41 #include "mmdevapi.h"
  42 #include "wine/debug.h"
  43
  44 WINE_DEFAULT_DEBUG_CHANNEL(mmdevapi);
  45
  46 static HINSTANCE instance;
  47
  48 DriverFuncs drvs;
  49
  50 const WCHAR drv_keyW[] = {'S','o','f','t','w','a','r','e','\\',
  51     'W','i','n','e','\\','D','r','i','v','e','r','s',0};
  52
  53 static const char *get_priority_string(int prio)
  54 {
  55     switch(prio){
  56     case Priority_Unavailable:
  57         return "Unavailable";
  58     case Priority_Low:
  59         return "Low";
  60     case Priority_Neutral:
  61         return "Neutral";
  62     case Priority_Preferred:
  63         return "Preferred";
  64     }
  65     return "Invalid";
  66 }
  67
  68 static BOOL load_driver(const WCHAR *name, DriverFuncs *driver)
  69 {
  70     WCHAR driver_module[264];
  71     static const WCHAR wineW[] = {'w','i','n','e',0};
  72     static const WCHAR dotdrvW[] = {'.','d','r','v',0};
  73
  74     lstrcpyW(driver_module, wineW);
  75     lstrcatW(driver_module, name);
  76     lstrcatW(driver_module, dotdrvW);
  77
  78     TRACE("Attempting to load %s\n", wine_dbgstr_w(driver_module));
  79
  80     driver->module = LoadLibraryW(driver_module);
  81     if(!driver->module){
  82         TRACE("Unable to load %s: %u\n", wine_dbgstr_w(driver_module),
  83                 GetLastError());
  84         return FALSE;
  85     }
  86
  87 #define LDFC(n) do { driver->p##n = (void*)GetProcAddress(driver->module, #n);\
  88         if(!driver->p##n) { FreeLibrary(driver->module); return FALSE; } } while(0)
  89     LDFC(GetPriority);
  90     LDFC(GetEndpointIDs);
  91     LDFC(GetAudioEndpoint);
  92     LDFC(GetAudioSessionManager);
  93 #undef LDFC
  94
  95     /* optional - do not fail if not found */
  96     driver->pGetPropValue = (void*)GetProcAddress(driver->module, "GetPropValue");
  97
  98     driver->priority = driver->pGetPriority();
  99     lstrcpyW(driver->module_name, driver_module);
 100
 101     TRACE("Successfully loaded %s with priority %s\n",
 102             wine_dbgstr_w(driver_module), get_priority_string(driver->priority));
 103
 104     return TRUE;
 105 }
 106
 107 static BOOL WINAPI init_driver(INIT_ONCE *once, void *param, void **context)
 108 {
 109     static const WCHAR drv_value[] = {'A','u','d','i','o',0};
 110
 111     static WCHAR default_list[] = {'p','u','l','s','e',',','a','l','s','a',',','o','s','s',',',
 112         'c','o','r','e','a','u','d','i','o',',','a','n','d','r','o','i','d',0};
 113
 114     DriverFuncs driver;
 115     HKEY key;
 116     WCHAR reg_list[256], *p, *next, *driver_list = default_list;
 117
 118     if(RegOpenKeyW(HKEY_CURRENT_USER, drv_keyW, &key) == ERROR_SUCCESS){
 119         DWORD size = sizeof(reg_list);
 120
 121         if(RegQueryValueExW(key, drv_value, 0, NULL, (BYTE*)reg_list,
 122                     &size) == ERROR_SUCCESS){
 123             if(reg_list[0] == '\0'){
 124                 TRACE("User explicitly chose no driver\n");
 125                 RegCloseKey(key);
 126                 return TRUE;
 127             }
 128
 129             driver_list = reg_list;
 130         }
 131
 132         RegCloseKey(key);
 133     }
 134
 135     TRACE("Loading driver list %s\n", wine_dbgstr_w(driver_list));
 136     for(next = p = driver_list; next; p = next + 1){
 137         next = wcschr(p, ',');
 138         if(next)
 139             *next = '\0';
 140
 141         driver.priority = Priority_Unavailable;
 142         if(load_driver(p, &driver)){
 143             if(driver.priority == Priority_Unavailable)
 144                 FreeLibrary(driver.module);
 145             else if(!drvs.module || driver.priority > drvs.priority){
 146                 TRACE("Selecting driver %s with priority %s\n",
 147                         wine_dbgstr_w(p), get_priority_string(driver.priority));
 148                 if(drvs.module)
 149                     FreeLibrary(drvs.module);
 150                 drvs = driver;
 151             }else
 152                 FreeLibrary(driver.module);
 153         }else
 154             TRACE("Failed to load driver %s\n", wine_dbgstr_w(p));
 155
 156         if(next)
 157             *next = ',';
 158     }
 159
 160     return drvs.module != 0;
 161 }
 162
 163 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
 164 {
 165     TRACE("(0x%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved);
 166
 167     switch (fdwReason)
 168     {
 169         case DLL_PROCESS_ATTACH:
 170             instance = hinstDLL;
 171             DisableThreadLibraryCalls(hinstDLL);
 172             break;
 173         case DLL_PROCESS_DETACH:
 174             if(lpvReserved)
 175                 break;
 176             MMDevEnum_Free();
 177             break;
 178     }
 179
 180     return TRUE;
 181 }
 182
 183 HRESULT WINAPI DllCanUnloadNow(void)
 184 {
 185     return S_FALSE;
 186 }
 187
 188 typedef HRESULT (*FnCreateInstance)(REFIID riid, LPVOID *ppobj);
 189
 190 typedef struct {
 191     IClassFactory IClassFactory_iface;
 192     REFCLSID rclsid;
 193     FnCreateInstance pfnCreateInstance;
 194 } IClassFactoryImpl;
 195
 196 static inline IClassFactoryImpl *impl_from_IClassFactory(IClassFactory *iface)
 197 {
 198     return CONTAINING_RECORD(iface, IClassFactoryImpl, IClassFactory_iface);
 199 }
 200
 201 static HRESULT WINAPI
 202 MMCF_QueryInterface(IClassFactory *iface, REFIID riid, void **ppobj)
 203 {
 204     IClassFactoryImpl *This = impl_from_IClassFactory(iface);
 205     TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppobj);
 206     if (ppobj == NULL)
 207         return E_POINTER;
 208     if (IsEqualIID(riid, &IID_IUnknown) ||
 209         IsEqualIID(riid, &IID_IClassFactory))
 210     {
 211         *ppobj = iface;
 212         IClassFactory_AddRef(iface);
 213         return S_OK;
 214     }
 215     *ppobj = NULL;
 216     return E_NOINTERFACE;
 217 }
 218
 219 static ULONG WINAPI MMCF_AddRef(LPCLASSFACTORY iface)
 220 {
 221     return 2;
 222 }
 223
 224 static ULONG WINAPI MMCF_Release(LPCLASSFACTORY iface)
 225 {
 226     /* static class, won't be freed */
 227     return 1;
 228 }
 229
 230 static HRESULT WINAPI MMCF_CreateInstance(
 231     LPCLASSFACTORY iface,
 232     LPUNKNOWN pOuter,
 233     REFIID riid,
 234     LPVOID *ppobj)
 235 {
 236     IClassFactoryImpl *This = impl_from_IClassFactory(iface);
 237     TRACE("(%p, %p, %s, %p)\n", This, pOuter, debugstr_guid(riid), ppobj);
 238
 239     if (pOuter)
 240         return CLASS_E_NOAGGREGATION;
 241
 242     if (ppobj == NULL) {
 243         WARN("invalid parameter\n");
 244         return E_POINTER;
 245     }
 246     *ppobj = NULL;
 247     return This->pfnCreateInstance(riid, ppobj);
 248 }
 249
 250 static HRESULT WINAPI MMCF_LockServer(LPCLASSFACTORY iface, BOOL dolock)
 251 {
 252     IClassFactoryImpl *This = impl_from_IClassFactory(iface);
 253     FIXME("(%p, %d) stub!\n", This, dolock);
 254     return S_OK;
 255 }
 256
 257 static const IClassFactoryVtbl MMCF_Vtbl = {
 258     MMCF_QueryInterface,
 259     MMCF_AddRef,
 260     MMCF_Release,
 261     MMCF_CreateInstance,
 262     MMCF_LockServer
 263 };
 264
 265 static IClassFactoryImpl MMDEVAPI_CF[] = {
 266     { { &MMCF_Vtbl }, &CLSID_MMDeviceEnumerator, (FnCreateInstance)MMDevEnum_Create }
 267 };
 268
 269 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
 270 {
 271     static INIT_ONCE init_once = INIT_ONCE_STATIC_INIT;
 272     unsigned int i = 0;
 273     TRACE("(%s, %s, %p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 274
 275     if(!InitOnceExecuteOnce(&init_once, init_driver, NULL, NULL)) {
 276         ERR("Driver initialization failed\n");
 277         return E_FAIL;
 278     }
 279
 280     if (ppv == NULL) {
 281         WARN("invalid parameter\n");
 282         return E_INVALIDARG;
 283     }
 284
 285     *ppv = NULL;
 286
 287     if (!IsEqualIID(riid, &IID_IClassFactory) &&
 288         !IsEqualIID(riid, &IID_IUnknown)) {
 289         WARN("no interface for %s\n", debugstr_guid(riid));
 290         return E_NOINTERFACE;
 291     }
 292
 293     for (i = 0; i < ARRAY_SIZE(MMDEVAPI_CF); ++i)
 294     {
 295         if (IsEqualGUID(rclsid, MMDEVAPI_CF[i].rclsid)) {
 296             IClassFactory_AddRef(&MMDEVAPI_CF[i].IClassFactory_iface);
 297             *ppv = &MMDEVAPI_CF[i];
 298             return S_OK;
 299         }
 300     }
 301
 302     WARN("(%s, %s, %p): no class found.\n", debugstr_guid(rclsid),
 303          debugstr_guid(riid), ppv);
 304     return CLASS_E_CLASSNOTAVAILABLE;
 305 }
 306
 307 /***********************************************************************
 308  *              DllRegisterServer (MMDEVAPI.@)
 309  */
 310 HRESULT WINAPI DllRegisterServer(void)
 311 {
 312     return __wine_register_resources( instance );
 313 }
 314
 315 /***********************************************************************
 316  *              DllUnregisterServer (MMDEVAPI.@)
 317  */
 318 HRESULT WINAPI DllUnregisterServer(void)
 319 {
 320     return __wine_unregister_resources( instance );
 321 }
 322
 323 struct activate_async_op {
 324     IActivateAudioInterfaceAsyncOperation IActivateAudioInterfaceAsyncOperation_iface;
 325     LONG ref;
 326
 327     IActivateAudioInterfaceCompletionHandler *callback;
 328     HRESULT result_hr;
 329     IUnknown *result_iface;
 330 };
 331
 332 static struct activate_async_op *impl_from_IActivateAudioInterfaceAsyncOperation(IActivateAudioInterfaceAsyncOperation *iface)
 333 {
 334     return CONTAINING_RECORD(iface, struct activate_async_op, IActivateAudioInterfaceAsyncOperation_iface);
 335 }
 336
 337 static HRESULT WINAPI activate_async_op_QueryInterface(IActivateAudioInterfaceAsyncOperation *iface,
 338         REFIID riid, void **ppv)
 339 {
 340     struct activate_async_op *This = impl_from_IActivateAudioInterfaceAsyncOperation(iface);
 341
 342     TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid), ppv);
 343
 344     if (!ppv)
 345         return E_POINTER;
 346
 347     if (IsEqualIID(riid, &IID_IUnknown) ||
 348             IsEqualIID(riid, &IID_IActivateAudioInterfaceAsyncOperation)) {
 349         *ppv = &This->IActivateAudioInterfaceAsyncOperation_iface;
 350     } else {
 351         *ppv = NULL;
 352         return E_NOINTERFACE;
 353     }
 354
 355     IUnknown_AddRef((IUnknown*)*ppv);
 356     return S_OK;
 357 }
 358
 359 static ULONG WINAPI activate_async_op_AddRef(IActivateAudioInterfaceAsyncOperation *iface)
 360 {
 361     struct activate_async_op *This = impl_from_IActivateAudioInterfaceAsyncOperation(iface);
 362     LONG ref = InterlockedIncrement(&This->ref);
 363     TRACE("(%p) refcount now %i\n", This, ref);
 364     return ref;
 365 }
 366
 367 static ULONG WINAPI activate_async_op_Release(IActivateAudioInterfaceAsyncOperation *iface)
 368 {
 369     struct activate_async_op *This = impl_from_IActivateAudioInterfaceAsyncOperation(iface);
 370     LONG ref = InterlockedDecrement(&This->ref);
 371     TRACE("(%p) refcount now %i\n", This, ref);
 372     if (!ref) {
 373         if(This->result_iface)
 374             IUnknown_Release(This->result_iface);
 375         IActivateAudioInterfaceCompletionHandler_Release(This->callback);
 376         HeapFree(GetProcessHeap(), 0, This);
 377     }
 378     return ref;
 379 }
 380
 381 static HRESULT WINAPI activate_async_op_GetActivateResult(IActivateAudioInterfaceAsyncOperation *iface,
 382         HRESULT *result_hr, IUnknown **result_iface)
 383 {
 384     struct activate_async_op *This = impl_from_IActivateAudioInterfaceAsyncOperation(iface);
 385
 386     TRACE("(%p)->(%p, %p)\n", This, result_hr, result_iface);
 387
 388     *result_hr = This->result_hr;
 389
 390     if(This->result_hr == S_OK){
 391         *result_iface = This->result_iface;
 392         IUnknown_AddRef(*result_iface);
 393     }
 394
 395     return S_OK;
 396 }
 397
 398 static IActivateAudioInterfaceAsyncOperationVtbl IActivateAudioInterfaceAsyncOperation_vtbl = {
 399     activate_async_op_QueryInterface,
 400     activate_async_op_AddRef,
 401     activate_async_op_Release,
 402     activate_async_op_GetActivateResult,
 403 };
 404
 405 static DWORD WINAPI activate_async_threadproc(void *user)
 406 {
 407     struct activate_async_op *op = user;
 408
 409     IActivateAudioInterfaceCompletionHandler_ActivateCompleted(op->callback, &op->IActivateAudioInterfaceAsyncOperation_iface);
 410
 411     IActivateAudioInterfaceAsyncOperation_Release(&op->IActivateAudioInterfaceAsyncOperation_iface);
 412
 413     return 0;
 414 }
 415
 416 static HRESULT get_mmdevice_by_activatepath(const WCHAR *path, IMMDevice **mmdev)
 417 {
 418     IMMDeviceEnumerator *devenum;
 419     HRESULT hr;
 420
 421     static const WCHAR DEVINTERFACE_AUDIO_RENDER_WSTR[] = L"{E6327CAD-DCEC-4949-AE8A-991E976A79D2}";
 422     static const WCHAR DEVINTERFACE_AUDIO_CAPTURE_WSTR[] = L"{2EEF81BE-33FA-4800-9670-1CD474972C3F}";
 423
 424     hr = MMDevEnum_Create(&IID_IMMDeviceEnumerator, (void**)&devenum);
 425     if (FAILED(hr)) {
 426         WARN("Failed to create MMDeviceEnumerator: %08x\n", hr);
 427         return hr;
 428     }
 429
 430     if (!lstrcmpiW(path, DEVINTERFACE_AUDIO_RENDER_WSTR)){
 431         hr = IMMDeviceEnumerator_GetDefaultAudioEndpoint(devenum, eRender, eMultimedia, mmdev);
 432     } else if (!lstrcmpiW(path, DEVINTERFACE_AUDIO_CAPTURE_WSTR)){
 433         hr = IMMDeviceEnumerator_GetDefaultAudioEndpoint(devenum, eCapture, eMultimedia, mmdev);
 434     } else {
 435         FIXME("How to map path to device id? %s\n", debugstr_w(path));
 436         hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
 437     }
 438
 439     if (FAILED(hr)) {
 440         WARN("Failed to get requested device (%s): %08x\n", debugstr_w(path), hr);
 441         *mmdev = NULL;
 442         hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
 443     }
 444
 445     IMMDeviceEnumerator_Release(devenum);
 446
 447     return hr;
 448 }
 449
 450 /***********************************************************************
 451  *              ActivateAudioInterfaceAsync (MMDEVAPI.17)
 452  */
 453 HRESULT WINAPI ActivateAudioInterfaceAsync(const WCHAR *path, REFIID riid,
 454         PROPVARIANT *params, IActivateAudioInterfaceCompletionHandler *done_handler,
 455         IActivateAudioInterfaceAsyncOperation **op_out)
 456 {
 457     struct activate_async_op *op;
 458     HANDLE ht;
 459     IMMDevice *mmdev;
 460
 461     TRACE("(%s, %s, %p, %p, %p)\n", debugstr_w(path), debugstr_guid(riid),
 462             params, done_handler, op_out);
 463
 464     op = HeapAlloc(GetProcessHeap(), 0, sizeof(*op));
 465     if (!op)
 466         return E_OUTOFMEMORY;
 467
 468     op->ref = 2; /* returned ref and threadproc ref */
 469     op->IActivateAudioInterfaceAsyncOperation_iface.lpVtbl = &IActivateAudioInterfaceAsyncOperation_vtbl;
 470     op->callback = done_handler;
 471     IActivateAudioInterfaceCompletionHandler_AddRef(done_handler);
 472
 473     op->result_hr = get_mmdevice_by_activatepath(path, &mmdev);
 474     if (SUCCEEDED(op->result_hr)) {
 475         op->result_hr = IMMDevice_Activate(mmdev, riid, CLSCTX_INPROC_SERVER, params, (void**)&op->result_iface);
 476         IMMDevice_Release(mmdev);
 477     }else
 478         op->result_iface = NULL;
 479
 480     ht = CreateThread(NULL, 0, &activate_async_threadproc, op, 0, NULL);
 481     CloseHandle(ht);
 482
 483     *op_out = &op->IActivateAudioInterfaceAsyncOperation_iface;
 484
 485     return S_OK;
 486 }