dlls/devenum/devenum_main.c

   1 /*
   2  *      exported dll functions for devenum.dll
   3  *
   4  * Copyright (C) 2002 John K. Hohm
   5  * Copyright (C) 2002 Robert Shearman
   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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20  */
  21
  22 #include "devenum_private.h"
  23 #include "wine/debug.h"
  24 #include "winreg.h"
  25
  26 WINE_DEFAULT_DEBUG_CHANNEL(devenum);
  27
  28 LONG dll_refs;
  29 HINSTANCE DEVENUM_hInstance;
  30
  31 typedef struct
  32 {
  33     REFCLSID clsid;
  34     LPCWSTR friendly_name;
  35     BOOL instance;
  36 } register_info;
  37
  38 static HRESULT register_clsids(int count, const register_info * pRegInfo, LPCWSTR pszThreadingModel);
  39 static void DEVENUM_RegisterQuartz(void);
  40
  41 /***********************************************************************
  42  *              Global string constant definitions
  43  */
  44 const WCHAR clsid_keyname[6] = { 'C', 'L', 'S', 'I', 'D', 0 };
  45
  46 /***********************************************************************
  47  *              DllEntryPoint
  48  */
  49 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID fImpLoad)
  50 {
  51     TRACE("%p 0x%lx %p\n", hinstDLL, fdwReason, fImpLoad);
  52
  53     switch(fdwReason) {
  54     case DLL_PROCESS_ATTACH:
  55         DEVENUM_hInstance = hinstDLL;
  56         DisableThreadLibraryCalls(hinstDLL);
  57         break;
  58
  59     case DLL_PROCESS_DETACH:
  60         DEVENUM_hInstance = 0;
  61         break;
  62     }
  63     return TRUE;
  64 }
  65
  66 /***********************************************************************
  67  *              DllGetClassObject (DEVENUM.@)
  68  */
  69 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID iid, LPVOID *ppv)
  70 {
  71     TRACE("(%s, %s, %p)\n", debugstr_guid(rclsid), debugstr_guid(iid), ppv);
  72
  73     *ppv = NULL;
  74
  75     /* FIXME: we should really have two class factories.
  76      * Oh well - works just fine as it is */
  77     if (IsEqualGUID(rclsid, &CLSID_SystemDeviceEnum) ||
  78         IsEqualGUID(rclsid, &CLSID_CDeviceMoniker))
  79         return IClassFactory_QueryInterface((IClassFactory*)&DEVENUM_ClassFactory, iid, ppv);
  80
  81     FIXME("\n\tCLSID:\t%s,\n\tIID:\t%s\n",debugstr_guid(rclsid),debugstr_guid(iid));
  82     return CLASS_E_CLASSNOTAVAILABLE;
  83 }
  84
  85 /***********************************************************************
  86  *              DllCanUnloadNow (DEVENUM.@)
  87  */
  88 HRESULT WINAPI DllCanUnloadNow(void)
  89 {
  90     return dll_refs != 0 ? S_FALSE : S_OK;
  91 }
  92
  93 /***********************************************************************
  94  *              DllRegisterServer (DEVENUM.@)
  95  */
  96 HRESULT WINAPI DllRegisterServer(void)
  97 {
  98     HRESULT res;
  99     HKEY hkeyClsid = NULL;
 100     HKEY hkey1 = NULL;
 101     HKEY hkey2 = NULL;
 102     LPOLESTR pszClsidDevMon = NULL;
 103     IFilterMapper2 * pMapper = NULL;
 104     LPVOID mapvptr;
 105     static const WCHAR threadingModel[] = {'B','o','t','h',0};
 106     static const WCHAR sysdevenum[] = {'S','y','s','t','e','m',' ','D','e','v','i','c','e',' ','E','n','u','m',0};
 107     static const WCHAR devmon[] = {'D','e','v','i','c','e','M','o','n','i','k','e','r',0};
 108     static const WCHAR acmcat[] = {'A','C','M',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 109     static const WCHAR vidcat[] = {'I','C','M',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 110     static const WCHAR filtcat[] = {'A','c','t','i','v','e','M','o','v','i','e',' ','F','i','l','t','e','r',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 111     static const WCHAR vfwcat[] = {'V','F','W',' ','C','a','p','t','u','r','e',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 112     static const WCHAR wavein[] = {'W','a','v','e','I','n',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r', 0};
 113     static const WCHAR waveout[] = {'W','a','v','e','O','u','t',' ','a','n','d',' ','D','S','o','u','n','d',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 114     static const WCHAR midiout[] = {'M','i','d','i','O','u','t',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 115     static const WCHAR amcat[] = {'A','c','t','i','v','e','M','o','v','i','e',' ','F','i','l','t','e','r',' ','C','a','t','e','g','o','r','i','e','s',0};
 116     static const WCHAR device[] = {'d','e','v','i','c','e',0};
 117     static const WCHAR device_1[] = {'d','e','v','i','c','e','.','1',0};
 118     static const register_info ri[] =
 119     {
 120         {&CLSID_SystemDeviceEnum, sysdevenum, FALSE},
 121         {&CLSID_CDeviceMoniker, devmon, FALSE},
 122         {&CLSID_AudioCompressorCategory, acmcat, TRUE},
 123         {&CLSID_VideoCompressorCategory, vidcat, TRUE},
 124         {&CLSID_LegacyAmFilterCategory, filtcat, TRUE},
 125         {&CLSID_VideoInputDeviceCategory, vfwcat, FALSE},
 126         {&CLSID_AudioInputDeviceCategory, wavein, FALSE},
 127         {&CLSID_AudioRendererCategory, waveout, FALSE},
 128         {&CLSID_MidiRendererCategory, midiout, FALSE},
 129         {&CLSID_ActiveMovieCategories, amcat, TRUE}
 130     };
 131
 132     TRACE("\n");
 133
 134     res = register_clsids(sizeof(ri) / sizeof(register_info), ri, threadingModel);
 135
 136     /* Quartz is needed for IFilterMapper2 */
 137     DEVENUM_RegisterQuartz();
 138
 139 /*** ActiveMovieFilter Categories ***/
 140
 141     CoInitialize(NULL);
 142
 143     res = CoCreateInstance(&CLSID_FilterMapper2, NULL, CLSCTX_INPROC,
 144                            &IID_IFilterMapper2,  &mapvptr);
 145     if (SUCCEEDED(res))
 146     {
 147         static const WCHAR friendlyvidcap[] = {'V','i','d','e','o',' ','C','a','p','t','u','r','e',' ','S','o','u','r','c','e','s',0};
 148         static const WCHAR friendlydshow[] = {'D','i','r','e','c','t','S','h','o','w',' ','F','i','l','t','e','r','s',0};
 149         static const WCHAR friendlyvidcomp[] = {'V','i','d','e','o',' ','C','o','m','p','r','e','s','s','o','r','s',0};
 150         static const WCHAR friendlyaudcap[] = {'A','u','d','i','o',' ','C','a','p','t','u','r','e',' ','S','o','u','r','c','e','s',0};
 151         static const WCHAR friendlyaudcomp[] = {'A','u','d','i','o',' ','C','o','m','p','r','e','s','s','o','r','s',0};
 152         static const WCHAR friendlyaudrend[] = {'A','u','d','i','o',' ','R','e','n','d','e','r','e','r','s',0};
 153         static const WCHAR friendlymidirend[] = {'M','i','d','i',' ','R','e','n','d','e','r','e','r','s',0};
 154         static const WCHAR friendlyextrend[] = {'E','x','t','e','r','n','a','l',' ','R','e','n','d','e','r','e','r','s',0};
 155         static const WCHAR friendlydevctrl[] = {'D','e','v','i','c','e',' ','C','o','n','t','r','o','l',' ','F','i','l','t','e','r','s',0};
 156
 157         pMapper = (IFilterMapper2*)mapvptr;
 158
 159         IFilterMapper2_CreateCategory(pMapper, &CLSID_VideoInputDeviceCategory, MERIT_DO_NOT_USE, friendlyvidcap);
 160         IFilterMapper2_CreateCategory(pMapper, &CLSID_LegacyAmFilterCategory, MERIT_NORMAL, friendlydshow);
 161         IFilterMapper2_CreateCategory(pMapper, &CLSID_VideoCompressorCategory, MERIT_DO_NOT_USE, friendlyvidcomp);
 162         IFilterMapper2_CreateCategory(pMapper, &CLSID_AudioInputDeviceCategory, MERIT_DO_NOT_USE, friendlyaudcap);
 163         IFilterMapper2_CreateCategory(pMapper, &CLSID_AudioCompressorCategory, MERIT_DO_NOT_USE, friendlyaudcomp);
 164         IFilterMapper2_CreateCategory(pMapper, &CLSID_AudioRendererCategory, MERIT_NORMAL, friendlyaudrend);
 165         IFilterMapper2_CreateCategory(pMapper, &CLSID_MidiRendererCategory, MERIT_NORMAL, friendlymidirend);
 166         IFilterMapper2_CreateCategory(pMapper, &CLSID_TransmitCategory, MERIT_DO_NOT_USE, friendlyextrend);
 167         IFilterMapper2_CreateCategory(pMapper, &CLSID_DeviceControlCategory, MERIT_DO_NOT_USE, friendlydevctrl);
 168
 169         IFilterMapper2_Release(pMapper);
 170     }
 171
 172 /*** CDeviceMoniker ***/
 173     if (SUCCEEDED(res))
 174     {
 175         res = StringFromCLSID(&CLSID_CDeviceMoniker, &pszClsidDevMon);
 176     }
 177     if (SUCCEEDED(res))
 178     {
 179         res = RegOpenKeyW(HKEY_CLASSES_ROOT, clsid_keyname, &hkeyClsid)
 180               == ERROR_SUCCESS ? S_OK : E_FAIL;
 181     }
 182     if (SUCCEEDED(res))
 183     {
 184         res = RegOpenKeyW(hkeyClsid, pszClsidDevMon, &hkey1)
 185                == ERROR_SUCCESS ? S_OK : E_FAIL;
 186     }
 187     if (SUCCEEDED(res))
 188     {
 189         static const WCHAR wszProgID[] = {'P','r','o','g','I','D',0};
 190         res = RegCreateKeyW(hkey1, wszProgID, &hkey2)
 191               == ERROR_SUCCESS ? S_OK : E_FAIL;
 192     }
 193     if (SUCCEEDED(res))
 194     {
 195         res = RegSetValueW(hkey2, NULL, REG_SZ, device_1, (lstrlenW(device_1) + 1) * sizeof(WCHAR))
 196               == ERROR_SUCCESS ? S_OK : E_FAIL;
 197     }
 198
 199     if (hkey2)
 200     {
 201         RegCloseKey(hkey2);
 202         hkey2 = NULL;
 203     }
 204
 205     if (SUCCEEDED(res))
 206     {
 207         static const WCHAR wszVProgID[] = {'V','e','r','s','i','o','n','I','n','d','e','p','e','d','e','n','t','P','r','o','g','I','D',0};
 208         res = RegCreateKeyW(hkey1, wszVProgID, &hkey2)
 209               == ERROR_SUCCESS ? S_OK : E_FAIL;
 210     }
 211     if (SUCCEEDED(res))
 212     {
 213         res = RegSetValueW(hkey2, NULL, REG_SZ, device, (lstrlenW(device) + 1) * sizeof(WCHAR))
 214               == ERROR_SUCCESS ? S_OK : E_FAIL;
 215     }
 216
 217     if (hkey2)
 218     {
 219         RegCloseKey(hkey2);
 220         hkey2 = NULL;
 221     }
 222
 223     if (hkey1)
 224     {
 225         RegCloseKey(hkey1);
 226         hkey1 = NULL;
 227     }
 228
 229     if (SUCCEEDED(res))
 230     {
 231         res = RegCreateKeyW(HKEY_CLASSES_ROOT, device, &hkey1)
 232               == ERROR_SUCCESS ? S_OK : E_FAIL;
 233     }
 234     if (SUCCEEDED(res))
 235     {
 236         res = RegCreateKeyW(hkey1, clsid_keyname, &hkey2)
 237               == ERROR_SUCCESS ? S_OK : E_FAIL;
 238     }
 239     if (SUCCEEDED(res))
 240     {
 241         res = RegSetValueW(hkey2, NULL, REG_SZ, pszClsidDevMon, (lstrlenW(pszClsidDevMon) + 1) * sizeof(WCHAR))
 242               == ERROR_SUCCESS ? S_OK : E_FAIL;
 243     }
 244     if (hkey2)
 245     {
 246         RegCloseKey(hkey2);
 247         hkey2 = NULL;
 248     }
 249
 250     if (hkey1)
 251     {
 252         RegCloseKey(hkey1);
 253         hkey1 = NULL;
 254     }
 255
 256     if (SUCCEEDED(res))
 257     {
 258         res = RegCreateKeyW(HKEY_CLASSES_ROOT, device_1, &hkey1)
 259               == ERROR_SUCCESS ? S_OK : E_FAIL;
 260     }
 261     if (SUCCEEDED(res))
 262     {
 263         res = RegCreateKeyW(hkey1, clsid_keyname, &hkey2)
 264               == ERROR_SUCCESS ? S_OK : E_FAIL;
 265     }
 266     if (SUCCEEDED(res))
 267     {
 268         res = RegSetValueW(hkey2, NULL, REG_SZ, pszClsidDevMon, (lstrlenW(pszClsidDevMon) + 1) * sizeof(WCHAR))
 269               == ERROR_SUCCESS ? S_OK : E_FAIL;
 270     }
 271
 272     if (hkey2)
 273         RegCloseKey(hkey2);
 274
 275     if (hkey1)
 276         RegCloseKey(hkey1);
 277
 278     if (hkeyClsid)
 279         RegCloseKey(hkeyClsid);
 280
 281     if (pszClsidDevMon)
 282         CoTaskMemFree(pszClsidDevMon);
 283
 284     CoUninitialize();
 285
 286     return res;
 287 }
 288
 289 /***********************************************************************
 290  *              DllUnregisterServer (DEVENUM.@)
 291  */
 292 HRESULT WINAPI DllUnregisterServer(void)
 293 {
 294         FIXME("stub!\n");
 295         return E_FAIL;
 296 }
 297
 298 static HRESULT register_clsids(int count, const register_info * pRegInfo, LPCWSTR pszThreadingModel)
 299 {
 300     HRESULT res = S_OK;
 301     WCHAR dll_module[MAX_PATH];
 302     LPOLESTR clsidString;
 303     HKEY hkeyClsid;
 304     HKEY hkeySub;
 305     HKEY hkeyInproc32;
 306     HKEY hkeyInstance = NULL;
 307     int i;
 308     static const WCHAR wcszInproc32[] = {'I','n','p','r','o','c','S','e','r','v','e','r','3','2',0};
 309     static const WCHAR wcszThreadingModel[] = {'T','h','r','e','a','d','i','n','g','M','o','d','e','l',0};
 310
 311     res = RegOpenKeyW(HKEY_CLASSES_ROOT, clsid_keyname, &hkeyClsid)
 312           == ERROR_SUCCESS ? S_OK : E_FAIL;
 313
 314     TRACE("HModule = %p\n", DEVENUM_hInstance);
 315     i = GetModuleFileNameW(DEVENUM_hInstance, dll_module,
 316                            sizeof(dll_module) / sizeof(WCHAR));
 317     if (!i)
 318         return HRESULT_FROM_WIN32(GetLastError());
 319     if (i >= sizeof(dll_module) / sizeof(WCHAR))
 320         return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
 321
 322     for (i = 0; i < count; i++)
 323     {
 324         if (SUCCEEDED(res))
 325         {
 326             res = StringFromCLSID(pRegInfo[i].clsid, &clsidString);
 327         }
 328         if (SUCCEEDED(res))
 329         {
 330             res = RegCreateKeyW(hkeyClsid, clsidString, &hkeySub)
 331                   == ERROR_SUCCESS ? S_OK : E_FAIL;
 332         }
 333         if (pRegInfo[i].instance && SUCCEEDED(res))
 334         {
 335             res = RegCreateKeyW(hkeySub, wszInstanceKeyName, &hkeyInstance)
 336                   == ERROR_SUCCESS ? S_OK : E_FAIL;
 337             RegCloseKey(hkeyInstance);
 338         }
 339         if (SUCCEEDED(res))
 340         {
 341             RegSetValueW(hkeySub,
 342                          NULL,
 343                          REG_SZ,
 344                          pRegInfo->friendly_name ? pRegInfo[i].friendly_name : clsidString,
 345                          (lstrlenW(pRegInfo[i].friendly_name ? pRegInfo->friendly_name : clsidString) + 1) * sizeof(WCHAR));
 346             res = RegCreateKeyW(hkeySub, wcszInproc32, &hkeyInproc32)
 347                   == ERROR_SUCCESS ? S_OK : E_FAIL;
 348         }
 349         if (SUCCEEDED(res))
 350         {
 351             RegSetValueW(hkeyInproc32,
 352                          NULL,
 353                          REG_SZ,
 354                          dll_module,
 355                          (lstrlenW(dll_module) + 1) * sizeof(WCHAR));
 356             RegSetValueExW(hkeyInproc32,
 357                            wcszThreadingModel,
 358                            0,
 359                            REG_SZ,
 360                            (LPCVOID)pszThreadingModel,
 361                            (lstrlenW(pszThreadingModel) + 1) * sizeof(WCHAR));
 362             RegCloseKey(hkeyInproc32);
 363         }
 364         RegCloseKey(hkeySub);
 365         CoTaskMemFree(clsidString);
 366         clsidString = NULL;
 367     }
 368
 369     RegCloseKey(hkeyClsid);
 370
 371     return res;
 372 }
 373
 374 typedef HRESULT (WINAPI *DllRegisterServer_func)(void);
 375
 376 /* calls DllRegisterServer() for the Quartz DLL */
 377 static void DEVENUM_RegisterQuartz()
 378 {
 379     HANDLE hDLL = LoadLibraryA("quartz.dll");
 380     DllRegisterServer_func pDllRegisterServer = NULL;
 381     if (hDLL)
 382         pDllRegisterServer = (DllRegisterServer_func)GetProcAddress(hDLL, "DllRegisterServer");
 383     if (pDllRegisterServer)
 384     {
 385         HRESULT hr = pDllRegisterServer();
 386         if (FAILED(hr))
 387             ERR("Failed to register Quartz. Error was 0x%lx)\n", hr);
 388     }
 389 }