mirror of
https://github.com/veracrypt/VeraCrypt.git
synced 2025-11-11 11:08:02 -06:00
102 lines
3.8 KiB
C++
102 lines
3.8 KiB
C++
/*
|
|
Derived from source code of TrueCrypt 7.1a, which is
|
|
Copyright (c) 2008-2012 TrueCrypt Developers Association and which is governed
|
|
by the TrueCrypt License 3.0.
|
|
|
|
Modifications and additions to the original source code (contained in this file)
|
|
and all other portions of this file are Copyright (c) 2013-2015 IDRIX
|
|
and are governed by the Apache License 2.0 the full text of which is
|
|
contained in the file License.txt included in VeraCrypt binary and source
|
|
code distribution packages.
|
|
*/
|
|
|
|
#define TC_MAIN_COM_VERSION_MAJOR 2
|
|
#define TC_MAIN_COM_VERSION_MINOR 7
|
|
|
|
#define TC_FORMAT_COM_VERSION_MAJOR 2
|
|
#define TC_FORMAT_COM_VERSION_MINOR 4
|
|
|
|
#include <atlbase.h>
|
|
#include <comdef.h>
|
|
#include <statreg.h>
|
|
#include <windows.h>
|
|
#include "ComSetup.h"
|
|
#include "Dlgcode.h"
|
|
#include "Resource.h"
|
|
#include "../Mount/MainCom_i.c"
|
|
#include "../Format/FormatCom_i.c"
|
|
|
|
|
|
extern "C" BOOL RegisterComServers (wchar_t *modulePath)
|
|
{
|
|
BOOL ret = TRUE;
|
|
wchar_t mainModule[1024], formatModule[1024];
|
|
CComPtr<ITypeLib> tl, tl2;
|
|
|
|
wsprintfW (mainModule, L"%sVeraCrypt.exe", modulePath);
|
|
wsprintfW (formatModule, L"%sVeraCrypt Format.exe", modulePath);
|
|
|
|
UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR, 0, SYS_WIN32);
|
|
UnRegisterTypeLib (LIBID_TrueCryptFormatCom, TC_FORMAT_COM_VERSION_MAJOR, TC_FORMAT_COM_VERSION_MINOR, 0, SYS_WIN32);
|
|
// unregister older versions that may still exist
|
|
UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-3, 0, SYS_WIN32);
|
|
UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-2, 0, SYS_WIN32);
|
|
UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-1, 0, SYS_WIN32);
|
|
|
|
wchar_t setupModule[MAX_PATH];
|
|
GetModuleFileNameW (NULL, setupModule, sizeof (setupModule) / sizeof (setupModule[0]));
|
|
|
|
CRegObject ro;
|
|
HRESULT r;
|
|
|
|
if (!SUCCEEDED (r = ro.FinalConstruct ())
|
|
|| !SUCCEEDED (r = ro.AddReplacement (L"MAIN_MODULE", mainModule))
|
|
|| !SUCCEEDED (r = ro.AddReplacement (L"FORMAT_MODULE", formatModule))
|
|
|| !SUCCEEDED (r = ro.ResourceRegister (setupModule, IDR_COMREG, L"REGISTRY"))
|
|
|| !SUCCEEDED (r = LoadTypeLib (mainModule, &tl))
|
|
|| !SUCCEEDED (r = RegisterTypeLib (tl, mainModule, 0))
|
|
|| !SUCCEEDED (r = LoadTypeLib (formatModule, &tl2))
|
|
|| !SUCCEEDED (r = RegisterTypeLib (tl2, formatModule, 0)))
|
|
{
|
|
MessageBox (MainDlg, _com_error (r).ErrorMessage(), _T(TC_APP_NAME), MB_ICONERROR);
|
|
ret = FALSE;
|
|
}
|
|
|
|
ro.FinalRelease ();
|
|
return ret;
|
|
}
|
|
|
|
|
|
extern "C" BOOL UnregisterComServers (wchar_t *modulePath)
|
|
{
|
|
BOOL ret;
|
|
|
|
if (UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR, 0, SYS_WIN32) != S_OK)
|
|
return FALSE;
|
|
if (UnRegisterTypeLib (LIBID_TrueCryptFormatCom, TC_FORMAT_COM_VERSION_MAJOR, TC_FORMAT_COM_VERSION_MINOR, 0, SYS_WIN32) != S_OK)
|
|
return FALSE;
|
|
|
|
// unregister older versions that may still exist
|
|
UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-3, 0, SYS_WIN32);
|
|
UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-2, 0, SYS_WIN32);
|
|
UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-1, 0, SYS_WIN32);
|
|
|
|
wchar_t module[1024];
|
|
CRegObject ro;
|
|
ro.FinalConstruct ();
|
|
|
|
wsprintfW (module, L"%sVeraCrypt.exe", modulePath);
|
|
ro.AddReplacement (L"MAIN_MODULE", module);
|
|
|
|
wsprintfW (module, L"%sVeraCrypt Format.exe", modulePath);
|
|
ro.AddReplacement (L"FORMAT_MODULE", module);
|
|
|
|
wchar_t setupModule[MAX_PATH];
|
|
GetModuleFileNameW (NULL, setupModule, sizeof (setupModule) / sizeof (setupModule[0]));
|
|
|
|
ret = ro.ResourceUnregister (setupModule, IDR_COMREG, L"REGISTRY") == S_OK;
|
|
|
|
ro.FinalRelease ();
|
|
return ret;
|
|
}
|