1 /*****************************************************************************
2 * main.cpp: ActiveX control for VLC
3 *****************************************************************************
4 * Copyright (C) 2005 VideoLAN
6 * Authors: Damien Fouilleul <Damien.Fouilleul@laposte.net>
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
21 *****************************************************************************/
31 #define COMPANY_STR "VideoLAN"
32 #define PROGRAM_STR "VLCPlugin"
33 #define VERSION_MAJOR_STR "1"
34 #define VERSION_MINOR_STR "0"
35 #define DESCRIPTION "VideoLAN VLC ActiveX Plugin"
37 #define THREADING_MODEL "Apartment"
38 #define MISC_STATUS "131473"
40 #define PROGID_STR COMPANY_STR"."PROGRAM_STR
41 #define VERS_PROGID_STR COMPANY_STR"."PROGRAM_STR"."VERSION_MAJOR_STR
42 #define VERSION_STR VERSION_MAJOR_STR"."VERSION_MINOR_STR
44 #define GUID_STRLEN 39
47 ** MingW headers do not declare those
49 extern const CATID CATID_SafeForInitializing;
50 extern const CATID CATID_SafeForScripting;
52 static LONG i_class_ref= 0;
53 static HINSTANCE h_instance= 0;
55 STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
57 HRESULT hr = CLASS_E_CLASSNOTAVAILABLE;
61 if( CLSID_VLCPlugin == rclsid )
63 VLCPluginClass *plugin = new VLCPluginClass(&i_class_ref, h_instance);
64 hr = plugin->QueryInterface(riid, ppv);
70 STDAPI DllCanUnloadNow(VOID)
72 return (0 == i_class_ref) ? S_OK: S_FALSE;
75 static LPCTSTR TStrFromGUID(REFGUID clsid)
79 if( FAILED(StringFromIID(clsid, &oleStr)) )
82 //check whether TCHAR and OLECHAR are both either ANSI or UNICODE
83 if( sizeof(TCHAR) == sizeof(OLECHAR) )
84 return (LPCTSTR)oleStr;
86 LPTSTR pct_CLSID = NULL;
88 size_t len = WideCharToMultiByte(CP_ACP, 0, oleStr, -1, NULL, 0, NULL, NULL);
91 pct_CLSID = (char *)CoTaskMemAlloc(len);
92 WideCharToMultiByte(CP_ACP, 0, oleStr, -1, pct_CLSID, len, NULL, NULL);
95 size_t len = MutiByteToWideChar(CP_ACP, 0, oleStr, -1, NULL, 0);
98 clsidStr = (wchar_t *)CoTaskMemAlloc(len*sizeof(wchar_t));
99 WideCharToMultiByte(CP_ACP, 0, oleStr, -1, pct_CLSID, len);
102 CoTaskMemFree(oleStr);
106 static HKEY keyCreate(HKEY parentKey, LPCTSTR keyName)
109 if( ERROR_SUCCESS == RegCreateKeyEx(parentKey, keyName, 0, NULL,
110 REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &childKey, NULL) )
117 STDAPI DllUnregisterServer(VOID)
119 // unregister type lib from the registry
120 UnRegisterTypeLib(LIBID_AXVLC, 1, 0, LOCALE_NEUTRAL, SYS_WIN32);
122 // remove component categories we supports
124 if( SUCCEEDED(CoCreateInstance(CLSID_StdComponentCategoriesMgr,
125 NULL, CLSCTX_INPROC_SERVER, IID_ICatRegister, (void**)&pcr)) ) {
126 CATID implCategories[] = {
128 CATID_PersistsToPropertyBag,
129 CATID_SafeForInitializing,
130 CATID_SafeForScripting,
133 pcr->UnRegisterClassImplCategories(CLSID_VLCPlugin,
134 sizeof(implCategories)/sizeof(CATID), implCategories);
138 SHDeleteKey(HKEY_CLASSES_ROOT, TEXT(VERS_PROGID_STR));
139 SHDeleteKey(HKEY_CLASSES_ROOT, TEXT(PROGID_STR));
141 LPCTSTR psz_CLSID = TStrFromGUID(CLSID_VLCPlugin);
143 if( NULL == psz_CLSID )
144 return E_OUTOFMEMORY;
147 if( ERROR_SUCCESS == RegOpenKeyEx(HKEY_CLASSES_ROOT, TEXT("CLSID"), 0, KEY_WRITE, &hClsIDKey) )
149 SHDeleteKey(hClsIDKey, psz_CLSID);
150 RegCloseKey(hClsIDKey);
152 CoTaskMemFree((void *)psz_CLSID);
157 STDAPI DllRegisterServer(VOID)
159 DllUnregisterServer();
161 char DllPath[MAX_PATH];
162 DWORD DllPathLen= GetModuleFileName(h_instance, DllPath, sizeof(DllPath)) ;
163 if( 0 == DllPathLen )
166 LPCTSTR psz_CLSID = TStrFromGUID(CLSID_VLCPlugin);
168 if( NULL == psz_CLSID )
169 return E_OUTOFMEMORY;
173 if( ERROR_SUCCESS != RegOpenKeyEx(HKEY_CLASSES_ROOT, TEXT("CLSID"), 0, KEY_CREATE_SUB_KEY, &hBaseKey) )
174 return SELFREG_E_CLASS;
176 HKEY hClassKey = keyCreate(hBaseKey, psz_CLSID);
177 if( NULL != hClassKey )
182 RegSetValueEx(hClassKey, NULL, 0, REG_SZ,
183 (const BYTE*)DESCRIPTION, sizeof(DESCRIPTION));
186 hSubKey = keyCreate(hClassKey, TEXT("Control"));
187 RegCloseKey(hSubKey);
189 #ifdef BUILD_LOCALSERVER
190 // LocalServer32 key value
191 hSubKey = keyCreate(hClassKey, TEXT("LocalServer32"));
192 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
193 (const BYTE*)DllPath, DllPathLen);
194 RegCloseKey(hSubKey);
196 // InprocServer32 key value
197 hSubKey = keyCreate(hClassKey, TEXT("InprocServer32"));
198 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
199 (const BYTE*)DllPath, DllPathLen);
200 RegSetValueEx(hSubKey, TEXT("ThreadingModel"), 0, REG_SZ,
201 (const BYTE*)THREADING_MODEL, sizeof(THREADING_MODEL));
202 RegCloseKey(hSubKey);
205 // MiscStatus key value
206 hSubKey = keyCreate(hClassKey, TEXT("MiscStatus\\1"));
207 RegSetValueEx(hSubKey, NULL, 0, REG_SZ, (const BYTE*)MISC_STATUS, sizeof(MISC_STATUS));
208 RegCloseKey(hSubKey);
210 // Programmable key value
211 hSubKey = keyCreate(hClassKey, TEXT("Programmable"));
212 RegCloseKey(hSubKey);
215 hSubKey = keyCreate(hClassKey, TEXT("ProgID"));
216 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
217 (const BYTE*)VERS_PROGID_STR, sizeof(VERS_PROGID_STR));
218 RegCloseKey(hSubKey);
220 // VersionIndependentProgID key value
221 hSubKey = keyCreate(hClassKey, TEXT("VersionIndependentProgID"));
222 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
223 (const BYTE*)PROGID_STR, sizeof(PROGID_STR));
224 RegCloseKey(hSubKey);
227 hSubKey = keyCreate(hClassKey, TEXT("Version"));
228 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
229 (const BYTE*)VERSION_STR, sizeof(VERSION_STR));
230 RegCloseKey(hSubKey);
233 LPCTSTR psz_LIBID = TStrFromGUID(LIBID_AXVLC);
234 if( NULL != psz_LIBID )
236 hSubKey = keyCreate(hClassKey, TEXT("TypeLib"));
237 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
238 (const BYTE*)psz_LIBID, sizeof(TCHAR)*GUID_STRLEN);
239 RegCloseKey(hSubKey);
241 RegCloseKey(hClassKey);
243 RegCloseKey(hBaseKey);
245 hBaseKey = keyCreate(HKEY_CLASSES_ROOT, TEXT(PROGID_STR));
246 if( NULL != hBaseKey )
249 RegSetValueEx(hBaseKey, NULL, 0, REG_SZ,
250 (const BYTE*)DESCRIPTION, sizeof(DESCRIPTION));
252 HKEY hSubKey = keyCreate(hBaseKey, TEXT("CLSID"));
253 if( NULL != hSubKey )
256 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
257 (const BYTE*)psz_CLSID, sizeof(TCHAR)*GUID_STRLEN);
259 RegCloseKey(hSubKey);
261 hSubKey = keyCreate(hBaseKey, TEXT("CurVer"));
262 if( NULL != hSubKey )
265 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
266 (const BYTE*)VERS_PROGID_STR, sizeof(VERS_PROGID_STR));
268 RegCloseKey(hSubKey);
270 RegCloseKey(hBaseKey);
273 hBaseKey = keyCreate(HKEY_CLASSES_ROOT, TEXT(VERS_PROGID_STR));
274 if( NULL != hBaseKey )
277 RegSetValueEx(hBaseKey, NULL, 0, REG_SZ,
278 (const BYTE*)DESCRIPTION, sizeof(DESCRIPTION));
280 HKEY hSubKey = keyCreate(hBaseKey, TEXT("CLSID"));
281 if( NULL != hSubKey )
284 RegSetValueEx(hSubKey, NULL, 0, REG_SZ,
285 (const BYTE*)psz_CLSID, sizeof(TCHAR)*GUID_STRLEN);
287 RegCloseKey(hSubKey);
289 RegCloseKey(hBaseKey);
292 // indicate which component categories we support
294 if( SUCCEEDED(CoCreateInstance(CLSID_StdComponentCategoriesMgr,
295 NULL, CLSCTX_INPROC_SERVER, IID_ICatRegister, (void**)&pcr)) ) {
296 CATID implCategories[] = {
298 CATID_PersistsToPropertyBag,
299 CATID_SafeForInitializing,
300 CATID_SafeForScripting,
303 pcr->RegisterClassImplCategories(CLSID_VLCPlugin,
304 sizeof(implCategories)/sizeof(CATID), implCategories);
308 // register type lib into the registry
311 #ifdef BUILD_LOCALSERVER
312 // replace .exe by .tlb
313 strcpy(DllPath+DllPathLen-4, ".tlb");
317 size_t typeLibPathLen = MultiByteToWideChar(CP_ACP, 0, DllPath, -1, NULL, 0);
318 if( typeLibPathLen > 0 )
320 LPOLESTR typeLibPath = (LPOLESTR)CoTaskMemAlloc(typeLibPathLen*sizeof(wchar_t));
321 MultiByteToWideChar(CP_ACP, 0, DllPath, DllPathLen, typeLibPath, typeLibPathLen);
322 if( FAILED(LoadTypeLibEx(typeLibPath, REGKIND_REGISTER, &typeLib)) )
323 #ifndef BUILD_LOCALSERVER
324 return SELFREG_E_TYPELIB;
327 CoTaskMemFree((void *)typeLibPath);
330 if( FAILED(LoadTypeLibEx((LPOLESTR)DllPath, REGKIND_REGISTER, &typeLib)) )
331 return SELFREG_E_TYPELIB;
335 CoTaskMemFree((void *)psz_CLSID);
340 #ifdef BUILD_LOCALSERVER
343 ** easier to debug an application than a DLL on cygwin GDB :)
347 STDAPI_(int) WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR szCmdLine, int sw)
351 if( FAILED(OleInitialize(NULL)) )
353 cerr << "cannot initialize OLE" << endl;
359 if( FAILED(DllRegisterServer()) )
361 cerr << "cannot register Local Server" << endl;
365 IUnknown *classProc = NULL;
367 if( FAILED(DllGetClassObject(CLSID_VLCPlugin, IID_IUnknown, (LPVOID *)&classProc)) )
370 DWORD dwRegisterClassObject;
372 if( FAILED(CoRegisterClassObject(CLSID_VLCPlugin, classProc,
373 CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE, &dwRegisterClassObject)) )
376 DWORD dwRegisterActiveObject;
378 if( FAILED(RegisterActiveObject(classProc, CLSID_VLCPlugin,
379 ACTIVEOBJECT_WEAK, &dwRegisterActiveObject)) )
382 classProc->Release();
385 * Polling messages from event queue
387 while( S_FALSE == DllCanUnloadNow() )
389 while( PeekMessage(&msg, NULL, 0, 0, PM_REMOVE) )
391 if( msg.message == WM_QUIT )
392 break; // Leave the PeekMessage while() loop
394 /*if(TranslateAccelerator(ghwndApp, ghAccel, &msg))
397 TranslateMessage(&msg);
398 DispatchMessage(&msg);
401 if(msg.message == WM_QUIT)
402 break; // Leave the for() loop
407 if( SUCCEEDED(RevokeActiveObject(dwRegisterActiveObject, NULL)) )
408 CoRevokeClassObject(dwRegisterClassObject);
410 // Reached on WM_QUIT message
412 return ((int) msg.wParam);
417 STDAPI_(BOOL) DllMain(HANDLE hModule, DWORD fdwReason, LPVOID lpReserved )
421 case DLL_PROCESS_ATTACH:
422 h_instance = (HINSTANCE)hModule;