CryptoFile.cpp

/*
		Project:		DocMan
		Module:			CryptoFile.n
		Description:	Encrypted files
		Author:			Martin Gäckler
		Address:		Hofmannsthalweg 14, A-4030 Linz
		Web:			https://www.gaeckler.at/

		Copyright:		(c) 1988-2025 Martin Gäckler

		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, version 3.

		You should have received a copy of the GNU General Public License 
		along with this program. If not, see <http://www.gnu.org/licenses/>.

		THIS SOFTWARE IS PROVIDED BY Martin Gäckler, Linz, Austria ``AS IS''
		AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
		TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
		PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR
		CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
		SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
		LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
		USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
		ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
		OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
		OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
		SUCH DAMAGE.
*/

// --------------------------------------------------------------------- //
// ----- switches ------------------------------------------------------ //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- includes ------------------------------------------------------ //
// --------------------------------------------------------------------- //

#include <gak/strFiles.h>
#include <gak/cryptoShared.h>

#include "CryptoFile.h"

// --------------------------------------------------------------------- //
// ----- imported datas ------------------------------------------------ //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- module switches ----------------------------------------------- //
// --------------------------------------------------------------------- //

#ifdef __BORLANDC__
#	pragma option -RT-
#	pragma option -b
#	ifdef __WIN32__
#		pragma option -a4
#		pragma option -pc
#	else
#		pragma option -po
#		pragma option -a2
#	endif
#endif

using namespace gak;

// --------------------------------------------------------------------- //
// ----- constants ----------------------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- macros -------------------------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- type definitions ---------------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- class definitions --------------------------------------------- //
// --------------------------------------------------------------------- //

class FACTORY_CRYPTO_FILE : public FACTORY_FILE
{
	virtual int getItemType( void ) const;
	virtual PTR_ITEM createItem( int id ) const;
	virtual const char *getName( void ) const;
};

class FACTORY_REMOTE_CRYPTO_FILE : public FACTORY_REMOTE_FILE
{
	virtual int getItemType( void ) const;
	virtual PTR_ITEM createItem( int id ) const;
	virtual const char *getName( void ) const;
};

// --------------------------------------------------------------------- //
// ----- exported datas ------------------------------------------------ //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- module static data -------------------------------------------- //
// --------------------------------------------------------------------- //

static FACTORY_CRYPTO_FILE theFactory;
static FACTORY_REMOTE_CRYPTO_FILE theRemoteFactory;

// --------------------------------------------------------------------- //
// ----- class static data --------------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- prototypes ---------------------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- module functions ---------------------------------------------- //
// --------------------------------------------------------------------- //

static template <typename ITEM>
STRING dowload( ITEM *theItem, int version, int flags, const STRING &dest )
{
	doEnterFunction("template <typename ITEM>::dowload");
	CryptoShared	theCrypto;
	STRING			cryptedFile = theItem->downloadCrypted( version, 0, dest );
	STRING			result = cryptedFile;

	try
	{
		readFromBinaryFile( cryptedFile, &theCrypto, cryptoMagic, cryptoVersion, false );

		if( result.endsWith( ".crypted" ) )
			result.cut( result.strlen()-8 );
		chmod( result, S_IREAD|S_IWRITE );
		theCrypto.decryptFile(
			DocManDataModule->getActUser()->userName,
			DocManDataModule->getPrivateKey(),
			result
		);

		if( flags&PROTECT_DOWNLOAD )
			chmod( result, S_IREAD );

		if( result != cryptedFile )
			strRemove( cryptedFile );
	}
	catch( ... )
	{
		strRemove( cryptedFile );
		throw;
	}

	return result;
}


static template <typename ITEM>
void createVersion( ITEM *theItem, const STRING &filePath, const STRING &description )
{
	doEnterFunction("template <typename ITEM>::createVersion");

	struct stat statBuff;

	if( !strStat( filePath, &statBuff ) )
	{
		CryptoShared	theCrypto;
		STRING			actUser = DocManDataModule->getActUser()->userName;
		STRING			publicKeyFile = THE_FILE::getExternalStorageBase() + DIRECTORY_DELIMITER_STRING "keys" DIRECTORY_DELIMITER_STRING + actUser;

		try
		{
			STRING		cryptedFile = theItem->downloadCrypted( 0, 0, "" );

			readFromBinaryFile( cryptedFile, &theCrypto, cryptoMagic, cryptoVersion, false );
			theCrypto.decryptAesKey( actUser, DocManDataModule->getPrivateKey() );
			strRemove( cryptedFile );
		}
		catch( ... )
		{
			// ignore errors
		}

		STRING cryptedFile = filePath + ".crypted";

		theCrypto.encryptFile( filePath, actUser, publicKeyFile );
		writeToBinaryFile( cryptedFile, theCrypto, cryptoMagic, cryptoVersion, owmOverwrite );

		setModTime( cryptedFile, statBuff.st_mtime );
		theItem->createCryptedVersion( cryptedFile, description );

		strRemove( cryptedFile );
	}
	else
	{
/*@*/	throw Exception( "File not found" );
	}
}
// --------------------------------------------------------------------- //
// ----- class inlines ------------------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- class constructors/destructors -------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- class static functions ---------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- class privates ------------------------------------------------ //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- class protected ----------------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- class virtuals ------------------------------------------------ //
// --------------------------------------------------------------------- //
   
int FACTORY_CRYPTO_FILE::getItemType( void ) const
{
	return (int)TYPE_CRYPTO_FILE;
}

PTR_ITEM FACTORY_CRYPTO_FILE::createItem( int id ) const
{
	return new THE_CRYPTO_FILE( id, this );
}

const char *FACTORY_CRYPTO_FILE::getName( void ) const
{
	return "Encrypted File";
}

int FACTORY_REMOTE_CRYPTO_FILE::getItemType( void ) const
{
	return (int)TYPE_REMOTE_CRYPTO_FILE;
}

PTR_ITEM FACTORY_REMOTE_CRYPTO_FILE::createItem( int id ) const
{
	return new THE_REMOTE_CRYPTO_FILE( id, this );
}

const char *FACTORY_REMOTE_CRYPTO_FILE::getName( void ) const
{
	return "Remote Encrypted File";
}

void THE_CRYPTO_FILE::createVersion( const STRING &filePath, const STRING &description )
{
	::createVersion( this, filePath, description );
}

STRING THE_CRYPTO_FILE::download( int version, int flags, const STRING &dest )
{
	return ::dowload( this, version, flags, dest );
}

TGraphic *THE_CRYPTO_FILE::getItemPicture( void ) const
{
	static Graphics::TBitmap *thePic = NULL;

	if( !thePic )
	{
		thePic = new Graphics::TBitmap();
		thePic->LoadFromResourceName((unsigned int)HInstance, "CRYPTO_FILE_BITMAP");
	}
	return thePic;
}

STRING THE_REMOTE_CRYPTO_FILE::getVersionFileName( void )
{
	return getName() + ".crypted";
}

void THE_REMOTE_CRYPTO_FILE::createVersion( const STRING &filePath, const STRING &description )
{
	doEnterFunction("THE_REMOTE_CRYPTO_FILE::createVersion");
	doLogValue( filePath );
	::createVersion( this, filePath, description );
}

STRING THE_REMOTE_CRYPTO_FILE::download( int version, int flags, const STRING &iDest )
{
	STRING dest = ::dowload( this, version, flags, iDest );
	if( !version )
	{
		chmod( dest, S_IREAD|S_IWRITE );

		const DocManService::VersionRecord_t *theVersion = getLatestVersion();
		setModTime( dest, theVersion->modifiedDate );

		if( flags&PROTECT_DOWNLOAD )
			chmod( dest, S_IREAD );
	}

	return dest;
}

TGraphic *THE_REMOTE_CRYPTO_FILE::getItemPicture( void ) const
{
	static Graphics::TBitmap *thePic = NULL;

	if( !thePic )
	{
		thePic = new Graphics::TBitmap();
		thePic->LoadFromResourceName((unsigned int)HInstance, "CRYPTO_FILE_BITMAP");
	}
	return thePic;
}

// --------------------------------------------------------------------- //
// ----- class publics ------------------------------------------------- //
// --------------------------------------------------------------------- //

// --------------------------------------------------------------------- //
// ----- entry points -------------------------------------------------- //
// --------------------------------------------------------------------- //

#ifdef __BORLANDC__
#	pragma option -RT.
#	pragma option -b.
#	pragma option -a.
#	pragma option -p.
#endif