From c606f0866c3a2a5db3ef9bc41738ef33eb9612a9 Mon Sep 17 00:00:00 2001
From: Mounir IDRASSI <mounir.idrassi@idrix.fr>
Date: Sat, 22 Jun 2013 16:16:13 +0200
Subject: Add original TrueCrypt 7.1a sources

---
 src/Format/FormatCom.cpp | 222 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 222 insertions(+)
 create mode 100644 src/Format/FormatCom.cpp

(limited to 'src/Format/FormatCom.cpp')

diff --git a/src/Format/FormatCom.cpp b/src/Format/FormatCom.cpp
new file mode 100644
index 00000000..9ab57c86
--- /dev/null
+++ b/src/Format/FormatCom.cpp
@@ -0,0 +1,222 @@
+/*
+ Copyright (c) 2007-2008 TrueCrypt Developers Association. All rights reserved.
+
+ Governed by the TrueCrypt License 3.0 the full text of which is contained in
+ the file License.txt included in TrueCrypt binary and source code distribution
+ packages.
+*/
+
+#include <atlcomcli.h>
+#include <atlconv.h>
+#include <comutil.h>
+#include <windows.h>
+#include "BaseCom.h"
+#include "BootEncryption.h"
+#include "Dlgcode.h"
+#include "Format.h"
+#include "Progress.h"
+#include "TcFormat.h"
+#include "FormatCom.h"
+#include "FormatCom_h.h"
+#include "FormatCom_i.c"
+
+using namespace TrueCrypt;
+
+static volatile LONG ObjectCount = 0;
+
+class TrueCryptFormatCom : public ITrueCryptFormatCom
+{
+
+public:
+	TrueCryptFormatCom (DWORD messageThreadId) : RefCount (0),
+		MessageThreadId (messageThreadId),
+		CallBack (NULL)
+	{
+		InterlockedIncrement (&ObjectCount);
+	}
+
+	~TrueCryptFormatCom ()
+	{
+		if (InterlockedDecrement (&ObjectCount) == 0)
+			PostThreadMessage (MessageThreadId, WM_APP, 0, 0);
+	}
+
+	virtual ULONG STDMETHODCALLTYPE AddRef ()
+	{
+		return InterlockedIncrement (&RefCount);
+	}
+
+	virtual ULONG STDMETHODCALLTYPE Release ()
+	{
+		if (!InterlockedDecrement (&RefCount))
+		{
+			delete this;
+			return 0;
+		}
+
+		return RefCount;
+	}
+
+	virtual HRESULT STDMETHODCALLTYPE QueryInterface (REFIID riid, void **ppvObject)
+	{
+		if (riid == IID_IUnknown || riid == IID_ITrueCryptFormatCom)
+			*ppvObject = this;
+		else
+		{
+			*ppvObject = NULL;
+			return E_NOINTERFACE;
+		}
+
+		AddRef ();
+		return S_OK;
+	}
+	
+	virtual DWORD STDMETHODCALLTYPE CallDriver (DWORD ioctl, BSTR input, BSTR *output)
+	{
+		return BaseCom::CallDriver (ioctl, input, output);
+	}
+
+	virtual DWORD STDMETHODCALLTYPE CopyFile (BSTR sourceFile, BSTR destinationFile)
+	{
+		return BaseCom::CopyFile (sourceFile, destinationFile);
+	}
+
+	virtual DWORD STDMETHODCALLTYPE DeleteFile (BSTR file)
+	{
+		return BaseCom::DeleteFile (file);
+	}
+
+	virtual BOOL STDMETHODCALLTYPE FormatNtfs (int driveNo, int clusterSize)
+	{
+		return ::FormatNtfs (driveNo, clusterSize);
+	}
+
+	virtual int STDMETHODCALLTYPE AnalyzeHiddenVolumeHost (
+		LONG_PTR hwndDlg, int *driveNo, __int64 hiddenVolHostSize, int *realClusterSize, __int64 *nbrFreeClusters)
+	{
+		return ::AnalyzeHiddenVolumeHost (
+			(HWND) hwndDlg, driveNo, hiddenVolHostSize, realClusterSize, nbrFreeClusters);
+	}
+
+	virtual DWORD STDMETHODCALLTYPE ReadWriteFile (BOOL write, BOOL device, BSTR filePath, BSTR *bufferBstr, unsigned __int64 offset, unsigned __int32 size, DWORD *sizeDone)
+	{
+		return BaseCom::ReadWriteFile (write, device, filePath, bufferBstr, offset, size, sizeDone);
+	}
+
+	virtual DWORD STDMETHODCALLTYPE RegisterFilterDriver (BOOL registerDriver, int filterType)
+	{
+		return BaseCom::RegisterFilterDriver (registerDriver, filterType);
+	}
+
+	virtual DWORD STDMETHODCALLTYPE RegisterSystemFavoritesService (BOOL registerService)
+	{
+		return BaseCom::RegisterSystemFavoritesService (registerService);
+	}
+
+	virtual DWORD STDMETHODCALLTYPE SetDriverServiceStartType (DWORD startType)
+	{
+		return BaseCom::SetDriverServiceStartType (startType);
+	}
+
+	virtual BOOL STDMETHODCALLTYPE IsPagingFileActive (BOOL checkNonWindowsPartitionsOnly)
+	{
+		return BaseCom::IsPagingFileActive (checkNonWindowsPartitionsOnly);
+	}
+
+	virtual DWORD STDMETHODCALLTYPE WriteLocalMachineRegistryDwordValue (BSTR keyPath, BSTR valueName, DWORD value)
+	{
+		return BaseCom::WriteLocalMachineRegistryDwordValue (keyPath, valueName, value);
+	}
+
+protected:
+	DWORD MessageThreadId;
+	LONG RefCount;
+	ITrueCryptFormatCom *CallBack;
+};
+
+
+extern "C" BOOL ComServerFormat ()
+{
+	SetProcessShutdownParameters (0x100, 0);
+
+	TrueCryptFactory<TrueCryptFormatCom> factory (GetCurrentThreadId ());
+	DWORD cookie;
+
+	if (IsUacSupported ())
+		UacElevated = TRUE;
+
+	if (CoRegisterClassObject (CLSID_TrueCryptFormatCom, (LPUNKNOWN) &factory,
+		CLSCTX_LOCAL_SERVER, REGCLS_SINGLEUSE, &cookie) != S_OK)
+		return FALSE;
+
+	MSG msg;
+	while (int r = GetMessage (&msg, NULL, 0, 0))
+	{
+		if (r == -1)
+			return FALSE;
+
+		TranslateMessage (&msg);
+		DispatchMessage (&msg);
+
+		if (msg.message == WM_APP
+			&& ObjectCount < 1
+			&& !factory.IsServerLocked ())
+			break;
+	}
+	CoRevokeClassObject (cookie);
+
+	return TRUE;
+}
+
+
+static BOOL ComGetInstance (HWND hWnd, ITrueCryptFormatCom **tcServer)
+{
+	return ComGetInstanceBase (hWnd, CLSID_TrueCryptFormatCom, IID_ITrueCryptFormatCom, (void **) tcServer);
+}
+
+
+ITrueCryptFormatCom *GetElevatedInstance (HWND parent)
+{
+	ITrueCryptFormatCom *instance;
+
+	if (!ComGetInstance (parent, &instance))
+		throw UserAbort (SRC_POS);
+
+	return instance;
+}
+
+
+extern "C" int UacFormatNtfs (HWND hWnd, int driveNo, int clusterSize)
+{
+	CComPtr<ITrueCryptFormatCom> tc;
+	int r;
+
+	CoInitialize (NULL);
+
+	if (ComGetInstance (hWnd, &tc))
+		r = tc->FormatNtfs (driveNo, clusterSize);
+	else
+		r = 0;
+
+	CoUninitialize ();
+
+	return r;
+}
+
+
+extern "C" int UacAnalyzeHiddenVolumeHost (HWND hwndDlg, int *driveNo, __int64 hiddenVolHostSize, int *realClusterSize, __int64 *nbrFreeClusters)
+{
+	CComPtr<ITrueCryptFormatCom> tc;
+	int r;
+
+	CoInitialize (NULL);
+
+	if (ComGetInstance (hwndDlg, &tc))
+		r = tc->AnalyzeHiddenVolumeHost ((LONG_PTR) hwndDlg, driveNo, hiddenVolHostSize, realClusterSize, nbrFreeClusters);
+	else
+		r = 0;
+
+	CoUninitialize ();
+
+	return r;
+}
-- 
cgit v1.2.3