/*
  KeePass Password Safe - The Open-Source Password Manager
  Copyright (C) 2003-2019 Dominik Reichl <dominik.reichl@t-online.de>

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#ifndef ___FILE_TRANSACTION_EX_H___
#define ___FILE_TRANSACTION_EX_H___

#pragma once

#include <vector>
#include <string>
#include <tchar.h>

typedef std::basic_string<TCHAR> std_string;

typedef HANDLE(WINAPI *LPCREATETRANSACTION)(LPSECURITY_ATTRIBUTES lpTransactionAttributes,
	LPGUID UOW, DWORD CreateOptions, DWORD IsolationLevel, DWORD IsolationFlags,
	DWORD Timeout, LPWSTR Description);
typedef BOOL(WINAPI *LPMOVEFILETRANSACTED)(LPCTSTR lpExistingFileName,
	LPCTSTR lpNewFileName, LPPROGRESS_ROUTINE lpProgressRoutine,
	LPVOID lpData, DWORD dwFlags, HANDLE hTransaction);
typedef BOOL(WINAPI *LPCOMMITTRANSACTION)(HANDLE TransactionHandle);

class CFileTransactionEx
{
public:
	CFileTransactionEx(LPCTSTR lpBaseFile, bool bTransacted);
	virtual ~CFileTransactionEx();

	bool OpenWrite(std_string& strOutBufferFile);
	bool CommitWrite();

private:
	bool CommitWriteTransaction();

	static std_string TxfCreateID();
	static bool TxfIsSupported(TCHAR chDriveLetter);
	static void TxfClosePrsv(HANDLE hTx);

	void TxfPrepare();
	bool TxfMove(bool& bFatalError);
	bool TxfMoveWithTx();

	bool m_bTransacted;
	std_string m_strBase;
	std_string m_strTemp;
	std_string m_strTxfMidFallback;

	bool m_bMadeUnhidden;
	PSECURITY_DESCRIPTOR m_pSec;

	HMODULE m_hKernel32;
	HMODULE m_hKtmW32;
	LPCREATETRANSACTION m_fnCreateTransaction;
	LPMOVEFILETRANSACTED m_fnMoveFileTransacted;
	LPCOMMITTRANSACTION m_fnCommitTransaction;

	std::vector<std_string> m_vToDelete;
};

#endif // ___FILE_TRANSACTION_EX_H___

By viewing downloads associated with this article you agree to the Terms of Service and the article's licence.

If a file you wish to view isn't highlighted, and is a text file (not binary), please let us know and we'll add colourisation support for it.

License

This article, along with any associated source code and files, is licensed under The GNU General Public License (GPLv3)

About the Author

Dominik Reichl

	Software Developer
	Unknown

Dominik started programming in Omikron Basic, a programming language for the good old Atari ST. After this, there was some short period of QBasic programming on the PC, but soon he began learning C++, which is his favorite language up to now.

Today, his programming experience includes C / C++ / [Visual] C++ [MFC], C#/.NET, Java, JavaScript, PHP and HTML and the basics of pure assembler.

He is interested in almost everything that has to do with computing; his special interests are security, cryptography and data compression.

You can find his latest freeware, open source projects and articles on his website: https://www.dominik-reichl.de/.

Tagged as

Stats

KeePass Password Safe

License

Share

About the Author

You may also be interested in...