git.oblomov.eu Git - wine/blob - 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 DWORD dll_ref = 0;
  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
  40 /***********************************************************************
  41  *              Global string constant definitions
  42  */
  43 const WCHAR clsid_keyname[6] = { 'C', 'L', 'S', 'I', 'D', 0 };
  44
  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 DEVENUM_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((LPCLASSFACTORY)&DEVENUM_ClassFactory, iid, ppv);
  80     FIXME("\n\tCLSID:\t%s,\n\tIID:\t%s\n",debugstr_guid(rclsid),debugstr_guid(iid));
  81     return CLASS_E_CLASSNOTAVAILABLE;
  82 }
  83
  84 /***********************************************************************
  85  *              DllCanUnloadNow (DEVENUM.@)
  86  */
  87 HRESULT WINAPI DEVENUM_DllCanUnloadNow(void)
  88 {
  89     return dll_ref != 0 ? S_FALSE : S_OK;
  90 }
  91
  92 /***********************************************************************
  93  *              DllRegisterServer (DEVENUM.@)
  94  */
  95 HRESULT WINAPI DEVENUM_DllRegisterServer(void)
  96 {
  97     HRESULT res;
  98     HKEY hkeyClsid = NULL;
  99     HKEY hkey1 = NULL;
 100     HKEY hkey2 = NULL;
 101     LPOLESTR pszClsidDevMon = NULL;
 102     IFilterMapper2 * pMapper = NULL;
 103     const WCHAR threadingModel[] = {'B','o','t','h',0};
 104     const WCHAR sysdevenum[] = {'S','y','s','t','e','m',' ','D','e','v','i','c','e',' ','E','n','u','m',0};
 105     const WCHAR devmon[] = {'D','e','v','i','c','e','M','o','n','i','k','e','r',0};
 106     const WCHAR acmcat[] = {'A','C','M',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 107     const WCHAR vidcat[] = {'I','C','M',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 108     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};
 109     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};
 110     const WCHAR wavein[] = {'W','a','v','e','I','n',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r', 0};
 111     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};
 112     const WCHAR midiout[] = {'M','i','d','i','O','u','t',' ','C','l','a','s','s',' ','M','a','n','a','g','e','r',0};
 113     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};
 114     const WCHAR device[] = {'d','e','v','i','c','e',0};
 115     const WCHAR device_1[] = {'d','e','v','i','c','e','.','1',0};
 116     const register_info ri[] =
 117     {
 118         {&CLSID_SystemDeviceEnum, sysdevenum, FALSE},
 119         {&CLSID_CDeviceMoniker, devmon, FALSE},
 120         {&CLSID_AudioCompressorCategory, acmcat, TRUE},
 121         {&CLSID_VideoCompressorCategory, vidcat, TRUE},
 122         {&CLSID_LegacyAmFilterCategory, filtcat, TRUE},
 123         {&CLSID_VideoInputDeviceCategory, vfwcat, FALSE},
 124         {&CLSID_AudioInputDeviceCategory, wavein, FALSE},
 125         {&CLSID_AudioRendererCategory, waveout, FALSE},
 126         {&CLSID_MidiRendererCategory, midiout, FALSE},
 127         {&CLSID_ActiveMovieCategories, amcat, TRUE}
 128     };
 129
 130     TRACE("\n");
 131
 132     res = register_clsids(sizeof(ri) / sizeof(register_info), ri, threadingModel);
 133
 134 /*** ActiveMovieFilter Categories ***/
 135     {
 136     const WCHAR friendlyvidcap[] = {'V','i','d','e','o',' ','C','a','p','t','u','r','e',' ','S','o','u','r','c','e','s',0};
 137     const WCHAR friendlydshow[] = {'D','i','r','e','c','t','S','h','o','w',' ','F','i','l','t','e','r','s',0};
 138     const WCHAR friendlyvidcomp[] = {'V','i','d','e','o',' ','C','o','m','p','r','e','s','s','o','r','s',0};
 139     const WCHAR friendlyaudcap[] = {'A','u','d','i','o',' ','C','a','p','t','u','r','e',' ','S','o','u','r','c','e','s',0};
 140     const WCHAR friendlyaudcomp[] = {'A','u','d','i','o',' ','C','o','m','p','r','e','s','s','o','r','s',0};
 141     const WCHAR friendlyaudrend[] = {'A','u','d','i','o',' ','R','e','n','d','e','r','e','r','s',0};
 142     const WCHAR friendlymidirend[] = {'M','i','d','i',' ','R','e','n','d','e','r','e','r','s',0};
 143     const WCHAR friendlyextrend[] = {'E','x','t','e','r','n','a','l',' ','R','e','n','d','e','r','e','r','s',0};
 144     const WCHAR friendlydevctrl[] = {'D','e','v','i','c','e',' ','C','o','n','t','r','o','l',' ','F','i','l','t','e','r','s',0};
 145     CoInitialize(NULL);
 146     res = CoCreateInstance(&CLSID_FilterMapper2, NULL, CLSCTX_INPROC,
 147                            &IID_IFilterMapper2, (void **) &pMapper);
 148
 149     IFilterMapper2_CreateCategory(pMapper, &CLSID_VideoInputDeviceCategory, MERIT_DO_NOT_USE, friendlyvidcap);
 150     IFilterMapper2_CreateCategory(pMapper, &CLSID_LegacyAmFilterCategory, MERIT_NORMAL, friendlydshow);
 151     IFilterMapper2_CreateCategory(pMapper, &CLSID_VideoCompressorCategory, MERIT_DO_NOT_USE, friendlyvidcomp);
 152     IFilterMapper2_CreateCategory(pMapper, &CLSID_AudioInputDeviceCategory, MERIT_DO_NOT_USE, friendlyaudcap);
 153     IFilterMapper2_CreateCategory(pMapper, &CLSID_AudioCompressorCategory, MERIT_DO_NOT_USE, friendlyaudcomp);
 154     IFilterMapper2_CreateCategory(pMapper, &CLSID_AudioRendererCategory, MERIT_NORMAL, friendlyaudrend);
 155     IFilterMapper2_CreateCategory(pMapper, &CLSID_MidiRendererCategory, MERIT_NORMAL, friendlymidirend);
 156     IFilterMapper2_CreateCategory(pMapper, &CLSID_TransmitCategory, MERIT_DO_NOT_USE, friendlyextrend);
 157     IFilterMapper2_CreateCategory(pMapper, &CLSID_DeviceControlCategory, MERIT_DO_NOT_USE, friendlydevctrl);
 158
 159     IFilterMapper2_Release(pMapper);
 160     CoUninitialize();
 161     }
 162
 163 /*** CDeviceMoniker ***/
 164     if (SUCCEEDED(res))
 165     {
 166         res = StringFromCLSID(&CLSID_CDeviceMoniker, &pszClsidDevMon);
 167     }
 168     if (SUCCEEDED(res))
 169     {
 170         res = RegOpenKeyW(HKEY_CLASSES_ROOT, clsid_keyname, &hkeyClsid)
 171               == ERROR_SUCCESS ? S_OK : E_FAIL;
 172     }
 173     if (SUCCEEDED(res))
 174     {
 175         res = RegOpenKeyW(hkeyClsid, pszClsidDevMon, &hkey1)
 176                == ERROR_SUCCESS ? S_OK : E_FAIL;
 177     }
 178     if (SUCCEEDED(res))
 179     {
 180         const WCHAR wszProgID[] = {'P','r','o','g','I','D',0};
 181         res = RegCreateKeyW(hkey1, wszProgID, &hkey2)
 182               == ERROR_SUCCESS ? S_OK : E_FAIL;
 183     }
 184     if (SUCCEEDED(res))
 185     {
 186         res = RegSetValueW(hkey2, NULL, REG_SZ, device_1, (lstrlenW(device_1) + 1) * sizeof(WCHAR))
 187               == ERROR_SUCCESS ? S_OK : E_FAIL;
 188     }
 189     RegCloseKey(hkey2);
 190     if (SUCCEEDED(res))
 191     {
 192         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};
 193         res = RegCreateKeyW(hkey1, wszVProgID, &hkey2)
 194               == ERROR_SUCCESS ? S_OK : E_FAIL;
 195     }
 196     if (SUCCEEDED(res))
 197     {
 198         res = RegSetValueW(hkey2, NULL, REG_SZ, device, (lstrlenW(device) + 1) * sizeof(WCHAR))
 199               == ERROR_SUCCESS ? S_OK : E_FAIL;
 200     }
 201     RegCloseKey(hkey2);
 202     RegCloseKey(hkey1);
 203     if (SUCCEEDED(res))
 204     {
 205         res = RegCreateKeyW(HKEY_CLASSES_ROOT, device, &hkey1)
 206               == ERROR_SUCCESS ? S_OK : E_FAIL;
 207     }
 208     if (SUCCEEDED(res))
 209     {
 210         res = RegCreateKeyW(hkey1, clsid_keyname, &hkey2)
 211               == ERROR_SUCCESS ? S_OK : E_FAIL;
 212     }
 213     if (SUCCEEDED(res))
 214     {
 215         res = RegSetValueW(hkey2, NULL, REG_SZ, pszClsidDevMon, (lstrlenW(pszClsidDevMon) + 1) * sizeof(WCHAR))
 216               == ERROR_SUCCESS ? S_OK : E_FAIL;
 217     }
 218     RegCloseKey(hkey2);
 219     RegCloseKey(hkey1);
 220
 221     if (SUCCEEDED(res))
 222     {
 223         res = RegCreateKeyW(HKEY_CLASSES_ROOT, device_1, &hkey1)
 224               == ERROR_SUCCESS ? S_OK : E_FAIL;
 225     }
 226     if (SUCCEEDED(res))
 227     {
 228         res = RegCreateKeyW(hkey1, clsid_keyname, &hkey2)
 229               == ERROR_SUCCESS ? S_OK : E_FAIL;
 230     }
 231     if (SUCCEEDED(res))
 232     {
 233         res = RegSetValueW(hkey2, NULL, REG_SZ, pszClsidDevMon, (lstrlenW(pszClsidDevMon) + 1) * sizeof(WCHAR))
 234               == ERROR_SUCCESS ? S_OK : E_FAIL;
 235     }
 236     RegCloseKey(hkey2);
 237     RegCloseKey(hkey1);
 238
 239     RegCloseKey(hkeyClsid);
 240
 241     if (pszClsidDevMon)
 242         CoTaskMemFree(pszClsidDevMon);
 243
 244     return res;
 245 }
 246
 247 /***********************************************************************
 248  *              DllUnregisterServer (DEVENUM.@)
 249  */
 250 HRESULT WINAPI DEVENUM_DllUnregisterServer(void)
 251 {
 252         FIXME("stub!\n");
 253         return E_FAIL;
 254 }
 255
 256 static HRESULT register_clsids(int count, const register_info * pRegInfo, LPCWSTR pszThreadingModel)
 257 {
 258     HRESULT res = S_OK;
 259     WCHAR dll_module[MAX_PATH];
 260     LPOLESTR clsidString;
 261     HKEY hkeyClsid;
 262     HKEY hkeySub;
 263     HKEY hkeyInproc32;
 264     HKEY hkeyInstance = NULL;
 265     int i;
 266     const WCHAR wcszInproc32[] = {'I','n','p','r','o','c','S','e','r','v','e','r','3','2',0};
 267     const WCHAR wcszThreadingModel[] = {'T','h','r','e','a','d','i','n','g','M','o','d','e','l',0};
 268
 269     res = RegOpenKeyW(HKEY_CLASSES_ROOT, clsid_keyname, &hkeyClsid)
 270           == ERROR_SUCCESS ? S_OK : E_FAIL;
 271
 272     TRACE("HModule = %p\n", DEVENUM_hInstance);
 273     if (!GetModuleFileNameW(DEVENUM_hInstance, dll_module,
 274                             sizeof(dll_module) / sizeof(WCHAR)))
 275         return HRESULT_FROM_WIN32(GetLastError());
 276
 277     for (i = 0; i < count; i++)
 278     {
 279         if (SUCCEEDED(res))
 280         {
 281             res = StringFromCLSID(pRegInfo[i].clsid, &clsidString);
 282         }
 283         if (SUCCEEDED(res))
 284         {
 285             res = RegCreateKeyW(hkeyClsid, clsidString, &hkeySub)
 286                   == ERROR_SUCCESS ? S_OK : E_FAIL;
 287         }
 288         if (pRegInfo[i].instance && SUCCEEDED(res))
 289         {
 290             res = RegCreateKeyW(hkeySub, wszInstanceKeyName, &hkeyInstance)
 291                   == ERROR_SUCCESS ? S_OK : E_FAIL;
 292             RegCloseKey(hkeyInstance);
 293         }
 294         if (SUCCEEDED(res))
 295         {
 296             RegSetValueW(hkeySub,
 297                          NULL,
 298                          REG_SZ,
 299                          pRegInfo->friendly_name ? pRegInfo[i].friendly_name : clsidString,
 300                          (lstrlenW(pRegInfo[i].friendly_name ? pRegInfo->friendly_name : clsidString) + 1) * sizeof(WCHAR));
 301             res = RegCreateKeyW(hkeySub, wcszInproc32, &hkeyInproc32)
 302                   == ERROR_SUCCESS ? S_OK : E_FAIL;
 303         }
 304         if (SUCCEEDED(res))
 305         {
 306             RegSetValueW(hkeyInproc32,
 307                          NULL,
 308                          REG_SZ,
 309                          dll_module,
 310                          (lstrlenW(dll_module) + 1) * sizeof(WCHAR));
 311             RegSetValueExW(hkeyInproc32,
 312                            wcszThreadingModel,
 313                            0,
 314                            REG_SZ,
 315                            (LPVOID)pszThreadingModel,
 316                            (lstrlenW(pszThreadingModel) + 1) * sizeof(WCHAR));
 317             RegCloseKey(hkeyInproc32);
 318         }
 319         RegCloseKey(hkeySub);
 320         CoTaskMemFree(clsidString);
 321         clsidString = NULL;
 322     }
 323
 324     RegCloseKey(hkeyClsid);
 325
 326     return res;
 327 }