1 files changed, 114 insertions, 0 deletions
diff --git a/src/SetupDLL/ComSetup.cpp b/src/SetupDLL/ComSetup.cpp
new file mode 100644
index 00000000..07c3360a
--- /dev/null
+++ b/src/SetupDLL/ComSetup.cpp
@@ -0,0 +1,114 @@
+/*
+ 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-2017 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 11
+
+#define TC_FORMAT_COM_VERSION_MAJOR 2
+#define TC_FORMAT_COM_VERSION_MINOR 9
+
+#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"
+
+/* 
+ * Same as RegisterComServers() in Setup project, but 
+ * instead of using GetModuleFileNameW() to get this 
+ * DLL's path as setupModule which will not work because
+ * the DLL is embedded in the binary of the MSI,
+ * we ship the empty version of 'VeraCrypt Setup.exe' 
+ * as 'VeraCrypt COMReg.exe' and use it.
+ */
+extern "C" BOOL RegisterComServers (wchar_t *modulePath)
+{
+	BOOL ret = TRUE;
+	wchar_t mainModule[1024], formatModule[1024], setupModule[1024];
+	CComPtr<ITypeLib> tl, tl2;
+
+	wsprintfW (mainModule, L"%sVeraCrypt.exe", modulePath);
+	wsprintfW (formatModule, L"%sVeraCrypt Format.exe", modulePath);
+	wsprintfW (setupModule, L"%sVeraCrypt COMReg.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
+	for (WORD i = 7; i >= 1; i--)
+		UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-i, 0, SYS_WIN32);
+	for (WORD i = 5; i >= 1; i--)
+		UnRegisterTypeLib (LIBID_TrueCryptFormatCom, TC_FORMAT_COM_VERSION_MAJOR, TC_FORMAT_COM_VERSION_MINOR-i, 0, SYS_WIN32);
+
+	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;
+}
+
+/* 
+ * Same as UnregisterComServers() in Setup project, but 
+ * instead of using GetModuleFileNameW() to get this 
+ * DLL's path as setupModule which will not work because
+ * the DLL is embedded in the binary of the MSI,
+ * we ship the empty version of 'VeraCrypt Setup.exe' 
+ * as 'VeraCrypt COMReg.exe' and use it.
+ */
+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
+	for (WORD i = 7; i >= 1; i--)
+		UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-i, 0, SYS_WIN32);
+	for (WORD i = 5; i >= 1; i--)
+		UnRegisterTypeLib (LIBID_TrueCryptFormatCom, TC_FORMAT_COM_VERSION_MAJOR, TC_FORMAT_COM_VERSION_MINOR-i, 0, SYS_WIN32);
+
+	wchar_t module[1024];
+	HRESULT r;
+	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);
+
+	wsprintfW (module, L"%sVeraCrypt COMReg.exe", modulePath);
+	ret = SUCCEEDED(r = ro.ResourceUnregister (module, IDR_COMREG, L"REGISTRY"));
+
+	ro.FinalRelease ();
+	return ret;
+}