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 "config.h"
  21 #include "wine/port.h"
  22
  23 #include <stdarg.h>
  24
  25 #define COBJMACROS
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "wingdi.h"
  29 #include "wine/library.h"
  30
  31 #include "ole2.h"
  32 #include "olectl.h"
  33 #include "rpcproxy.h"
  34 #include "propsys.h"
  35 #include "initguid.h"
  36 #include "propkeydef.h"
  37 #include "mmdeviceapi.h"
  38 #include "dshow.h"
  39 #include "dsound.h"
  40 #include "audioclient.h"
  41 #include "endpointvolume.h"
  42 #include "audiopolicy.h"
  43 #include "devpkey.h"
  44 #include "winreg.h"
  45
  46 #include "mmdevapi.h"
  47 #include "wine/debug.h"
  48 #include "wine/unicode.h"
  49
  50 WINE_DEFAULT_DEBUG_CHANNEL(mmdevapi);
  51
  52 static HINSTANCE instance;
  53
  54 DriverFuncs drvs;
  55
  56 static BOOL load_driver(const WCHAR *name)
  57 {
  58     WCHAR driver_module[264];
  59     static const WCHAR wineW[] = {'w','i','n','e',0};
  60     static const WCHAR dotdrvW[] = {'.','d','r','v',0};
  61
  62     lstrcpyW(driver_module, wineW);
  63     lstrcatW(driver_module, name);
  64     lstrcatW(driver_module, dotdrvW);
  65
  66     TRACE("Attempting to load %s\n", wine_dbgstr_w(driver_module));
  67
  68     drvs.module = LoadLibraryW(driver_module);
  69     if(!drvs.module){
  70         TRACE("Unable to load %s: %u\n", wine_dbgstr_w(driver_module),
  71                 GetLastError());
  72         return FALSE;
  73     }
  74
  75 #define LDFC(n) do { drvs.p##n = (void*)GetProcAddress(drvs.module, #n);\
  76         if(!drvs.p##n) { FreeLibrary(drvs.module); return FALSE; } } while(0)
  77     LDFC(GetEndpointIDs);
  78     LDFC(GetAudioEndpoint);
  79     LDFC(GetAudioSessionManager);
  80 #undef LDFC
  81
  82     lstrcpyW(drvs.module_name, driver_module);
  83     TRACE("Successfully loaded %s\n", wine_dbgstr_w(driver_module));
  84
  85     return TRUE;
  86 }
  87
  88 static BOOL init_driver(void)
  89 {
  90     static const WCHAR alsaW[] = {'a','l','s','a',0};
  91     static const WCHAR ossW[] = {'o','s','s',0};
  92     static const WCHAR coreaudioW[] = {'c','o','r','e','a','u','d','i','o',0};
  93     static const WCHAR *default_drivers[] = { alsaW, coreaudioW, ossW };
  94     static const WCHAR drv_key[] = {'S','o','f','t','w','a','r','e','\\',
  95         'W','i','n','e','\\','D','r','i','v','e','r','s',0};
  96     static const WCHAR drv_value[] = {'A','u','d','i','o',0};
  97     HKEY key;
  98     UINT i;
  99
 100     if(drvs.module)
 101         return TRUE;
 102
 103     if(RegOpenKeyW(HKEY_CURRENT_USER, drv_key, &key) == ERROR_SUCCESS){
 104         WCHAR driver_name[256], *p, *next;
 105         DWORD size = sizeof(driver_name);
 106
 107         if(RegQueryValueExW(key, drv_value, 0, NULL, (BYTE*)driver_name,
 108                     &size) == ERROR_SUCCESS){
 109             RegCloseKey(key);
 110
 111             if(driver_name[0] == '\0')
 112                 return TRUE;
 113
 114             for(next = p = driver_name; next; p = next + 1){
 115                 next = strchrW(p, ',');
 116                 if(next)
 117                     *next = '\0';
 118
 119                 if(load_driver(p))
 120                     return TRUE;
 121
 122                 TRACE("Failed to load driver: %s\n", wine_dbgstr_w(driver_name));
 123             }
 124
 125             ERR("No drivers in the registry loaded successfully!\n");
 126             return FALSE;
 127         }
 128
 129         RegCloseKey(key);
 130     }
 131
 132     for(i = 0; i < sizeof(default_drivers)/sizeof(*default_drivers); ++i)
 133         if(load_driver(default_drivers[i]))
 134             return TRUE;
 135
 136     return FALSE;
 137 }
 138
 139 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
 140 {
 141     TRACE("(0x%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved);
 142
 143     switch (fdwReason)
 144     {
 145         case DLL_PROCESS_ATTACH:
 146             instance = hinstDLL;
 147             DisableThreadLibraryCalls(hinstDLL);
 148             break;
 149         case DLL_PROCESS_DETACH:
 150             MMDevEnum_Free();
 151             break;
 152     }
 153
 154     return TRUE;
 155 }
 156
 157 HRESULT WINAPI DllCanUnloadNow(void)
 158 {
 159     return S_FALSE;
 160 }
 161
 162 typedef HRESULT (*FnCreateInstance)(REFIID riid, LPVOID *ppobj);
 163
 164 typedef struct {
 165     IClassFactory IClassFactory_iface;
 166     REFCLSID rclsid;
 167     FnCreateInstance pfnCreateInstance;
 168 } IClassFactoryImpl;
 169
 170 static inline IClassFactoryImpl *impl_from_IClassFactory(IClassFactory *iface)
 171 {
 172     return CONTAINING_RECORD(iface, IClassFactoryImpl, IClassFactory_iface);
 173 }
 174
 175 static HRESULT WINAPI
 176 MMCF_QueryInterface(LPCLASSFACTORY iface, REFIID riid, LPVOID *ppobj)
 177 {
 178     IClassFactoryImpl *This = impl_from_IClassFactory(iface);
 179     TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppobj);
 180     if (ppobj == NULL)
 181         return E_POINTER;
 182     if (IsEqualIID(riid, &IID_IUnknown) ||
 183         IsEqualIID(riid, &IID_IClassFactory))
 184     {
 185         *ppobj = iface;
 186         IUnknown_AddRef(iface);
 187         return S_OK;
 188     }
 189     *ppobj = NULL;
 190     return E_NOINTERFACE;
 191 }
 192
 193 static ULONG WINAPI MMCF_AddRef(LPCLASSFACTORY iface)
 194 {
 195     return 2;
 196 }
 197
 198 static ULONG WINAPI MMCF_Release(LPCLASSFACTORY iface)
 199 {
 200     /* static class, won't be freed */
 201     return 1;
 202 }
 203
 204 static HRESULT WINAPI MMCF_CreateInstance(
 205     LPCLASSFACTORY iface,
 206     LPUNKNOWN pOuter,
 207     REFIID riid,
 208     LPVOID *ppobj)
 209 {
 210     IClassFactoryImpl *This = impl_from_IClassFactory(iface);
 211     TRACE("(%p, %p, %s, %p)\n", This, pOuter, debugstr_guid(riid), ppobj);
 212
 213     if (pOuter)
 214         return CLASS_E_NOAGGREGATION;
 215
 216     if (ppobj == NULL) {
 217         WARN("invalid parameter\n");
 218         return E_POINTER;
 219     }
 220     *ppobj = NULL;
 221     return This->pfnCreateInstance(riid, ppobj);
 222 }
 223
 224 static HRESULT WINAPI MMCF_LockServer(LPCLASSFACTORY iface, BOOL dolock)
 225 {
 226     IClassFactoryImpl *This = impl_from_IClassFactory(iface);
 227     FIXME("(%p, %d) stub!\n", This, dolock);
 228     return S_OK;
 229 }
 230
 231 static const IClassFactoryVtbl MMCF_Vtbl = {
 232     MMCF_QueryInterface,
 233     MMCF_AddRef,
 234     MMCF_Release,
 235     MMCF_CreateInstance,
 236     MMCF_LockServer
 237 };
 238
 239 static IClassFactoryImpl MMDEVAPI_CF[] = {
 240     { { &MMCF_Vtbl }, &CLSID_MMDeviceEnumerator, (FnCreateInstance)MMDevEnum_Create }
 241 };
 242
 243 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
 244 {
 245     int i = 0;
 246     TRACE("(%s, %s, %p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 247
 248     if(!init_driver()){
 249         ERR("Driver initialization failed\n");
 250         return E_FAIL;
 251     }
 252
 253     if (ppv == NULL) {
 254         WARN("invalid parameter\n");
 255         return E_INVALIDARG;
 256     }
 257
 258     *ppv = NULL;
 259
 260     if (!IsEqualIID(riid, &IID_IClassFactory) &&
 261         !IsEqualIID(riid, &IID_IUnknown)) {
 262         WARN("no interface for %s\n", debugstr_guid(riid));
 263         return E_NOINTERFACE;
 264     }
 265
 266     for (i = 0; i < sizeof(MMDEVAPI_CF)/sizeof(MMDEVAPI_CF[0]); ++i)
 267     {
 268         if (IsEqualGUID(rclsid, MMDEVAPI_CF[i].rclsid)) {
 269             IUnknown_AddRef(&MMDEVAPI_CF[i].IClassFactory_iface);
 270             *ppv = &MMDEVAPI_CF[i];
 271             return S_OK;
 272         }
 273         i++;
 274     }
 275
 276     WARN("(%s, %s, %p): no class found.\n", debugstr_guid(rclsid),
 277          debugstr_guid(riid), ppv);
 278     return CLASS_E_CLASSNOTAVAILABLE;
 279 }
 280
 281 /***********************************************************************
 282  *              DllRegisterServer (MMDEVAPI.@)
 283  */
 284 HRESULT WINAPI DllRegisterServer(void)
 285 {
 286     return __wine_register_resources( instance );
 287 }
 288
 289 /***********************************************************************
 290  *              DllUnregisterServer (MMDEVAPI.@)
 291  */
 292 HRESULT WINAPI DllUnregisterServer(void)
 293 {
 294     return __wine_unregister_resources( instance );
 295 }