/** -- C++ Source File -- **/

// MARKER(update_precomp.py): autogen include statement, do not remove
#include "precompiled_xmlsecurity.hxx"
#include <stdio.h>
#include "helper.hxx"

#include "libxml/tree.h"
#include "libxml/parser.h"
#ifndef XMLSEC_NO_XSLT
#include "libxslt/xslt.h"
#endif

#include "securityenvironment_mscryptimpl.hxx"
#include "xmlelementwrapper_xmlsecimpl.hxx"

#include "xmlsec/strings.h"
#include "xmlsec/mscrypto/app.h"
#include "xmlsec/xmltree.h"

#include <rtl/ustring.hxx>
#include <cppuhelper/servicefactory.hxx>

#include <com/sun/star/lang/XComponent.hpp>
#include <com/sun/star/beans/PropertyValue.hpp>
#include <com/sun/star/xml/wrapper/XXMLElementWrapper.hpp>
#include <com/sun/star/xml/wrapper/XXMLDocumentWrapper.hpp>
#include <com/sun/star/xml/crypto/XXMLEncryption.hpp>
#include <com/sun/star/xml/crypto/XXMLEncryptionTemplate.hpp>
#include <com/sun/star/xml/crypto/XXMLSecurityContext.hpp>
#include <com/sun/star/xml/crypto/XSecurityEnvironment.hpp>

using namespace ::rtl ;
using namespace ::cppu ;
using namespace ::com::sun::star::uno ;
using namespace ::com::sun::star::io ;
using namespace ::com::sun::star::ucb ;
using namespace ::com::sun::star::beans ;
using namespace ::com::sun::star::document ;
using namespace ::com::sun::star::lang ;
using namespace ::com::sun::star::registry ;
using namespace ::com::sun::star::xml::wrapper ;
using namespace ::com::sun::star::xml::crypto ;

int SAL_CALL main( int argc, char **argv )
{
	const char* 		n_pCertStore ;
	HCERTSTORE			n_hStoreHandle ;

	xmlDocPtr			doc = NULL ;
	xmlNodePtr			tplNode ;
	xmlNodePtr			tarNode ;
	FILE*				dstFile = NULL ;

	HCRYPTPROV			hCryptProv = NULL ;
	HCRYPTKEY			symKey = NULL ;

	if( argc != 6 && argc != 7 ) {
		fprintf( stderr, "Usage: %s <file_url of template> <file_url of result> <target element name> <target element namespace> <rdb file>\n\n" , argv[0] ) ;
		fprintf( stderr, "Usage: %s <file_url of template> <file_url of result> <target element name> <target element namespace> <rdb file> < Cert Store Name >\n\n" , argv[0] ) ;
		return 1 ;
	}

	//Init libxml and libxslt libraries
	xmlInitParser();
	LIBXML_TEST_VERSION
	xmlLoadExtDtdDefaultValue = XML_DETECT_IDS | XML_COMPLETE_ATTRS;
	xmlSubstituteEntitiesDefault(1);

	#ifndef XMLSEC_NO_XSLT
	xmlIndentTreeOutput = 1; 
	#endif // XMLSEC_NO_XSLT

	//Initialize the crypto engine
	if( argc == 7 ) {
		n_pCertStore = argv[6] ;
		n_hStoreHandle = CertOpenSystemStore( NULL, n_pCertStore ) ;
		if( n_hStoreHandle == NULL ) {
			fprintf( stderr, "Can not open the system cert store %s\n", n_pCertStore ) ;
			return 1 ;
		}
	} else {
		n_pCertStore = NULL ;
		n_hStoreHandle = NULL ;
	}
	xmlSecMSCryptoAppInit( n_pCertStore ) ;

	//Create encryption key.
	//CryptAcquireContext( &hCryptProv , NULL , NULL , PROV_RSA_FULL , CRYPT_DELETEKEYSET ) ;
	//CryptAcquireContext( &hCryptProv , "MyTempKeyContainer" , NULL , PROV_RSA_FULL , CRYPT_DELETEKEYSET ) ;

	if( !CryptAcquireContext( &hCryptProv , NULL , NULL , PROV_RSA_FULL , CRYPT_VERIFYCONTEXT ) ) {
		fprintf( stderr, "### cannot get crypto provider context!\n" );
		goto done ;
	}

	if( !CryptGenKey( hCryptProv, CALG_RC4, 0x00800000 | CRYPT_EXPORTABLE, &symKey ) ) {
		fprintf( stderr , "### cannot create symmetric key!\n" ) ;
		goto done ;
	}

	//Load XML document
	doc = xmlParseFile( argv[1] ) ;
	if( doc == NULL || xmlDocGetRootElement( doc ) == NULL ) {
		fprintf( stderr , "### Cannot load template xml document!\n" ) ;
		goto done ;
	}

	//Find the encryption template
	tplNode = xmlSecFindNode( xmlDocGetRootElement( doc ), xmlSecNodeEncryptedData, xmlSecEncNs ) ;
	if( tplNode == NULL ) {
		fprintf( stderr , "### Cannot find the encryption template!\n" ) ;
		goto done ;
	}

	//Find the encryption template
	tarNode = xmlSecFindNode( xmlDocGetRootElement( doc ), ( const unsigned char*)argv[3], ( const unsigned char*)argv[4] ) ;
	if( tarNode == NULL ) {
		fprintf( stderr , "### Cannot find the encryption target!\n" ) ;
		goto done ;
	}

	try {
		Reference< XMultiComponentFactory > xManager = NULL ;
		Reference< XComponentContext > xContext = NULL ;

		xManager = serviceManager( xContext , OUString::createFromAscii( "local" ), OUString::createFromAscii( argv[5] ) ) ;

		//Create encryption template
		Reference< XInterface > tplElement =
			xManager->createInstanceWithContext( OUString::createFromAscii( "com.sun.star.xml.security.bridge.xmlsec.XMLElementWrapper_XmlSecImpl" ) , xContext ) ;
		OSL_ENSURE( tplElement.is() ,
			"Encryptor - "
			"Cannot get service instance of \"xsec.XMLElementWrapper\"" ) ;

		Reference< XXMLElementWrapper > xTplElement( tplElement , UNO_QUERY ) ;
		OSL_ENSURE( xTplElement.is() ,
			"Encryptor - "
			"Cannot get interface of \"XXMLElementWrapper\" from service \"xsec.XMLElementWrapper\"" ) ;

		Reference< XUnoTunnel > xTplEleTunnel( xTplElement , UNO_QUERY ) ;
		OSL_ENSURE( xTplEleTunnel.is() ,
			"Encryptor - "
			"Cannot get interface of \"XUnoTunnel\" from service \"xsec.XMLElementWrapper\"" ) ;

		XMLElementWrapper_XmlSecImpl* pTplElement = ( XMLElementWrapper_XmlSecImpl* )xTplEleTunnel->getSomething( XMLElementWrapper_XmlSecImpl::getUnoTunnelImplementationId() ) ;
		OSL_ENSURE( pTplElement != NULL ,
			"Encryptor - "
			"Cannot get implementation of \"xsec.XMLElementWrapper\"" ) ;

		pTplElement->setNativeElement( tplNode ) ;

		//Create encryption target element
		Reference< XInterface > tarElement =
			xManager->createInstanceWithContext( OUString::createFromAscii( "com.sun.star.xml.security.bridge.xmlsec.XMLElementWrapper_XmlSecImpl" ) , xContext ) ;
		OSL_ENSURE( tarElement.is() ,
			"Encryptor - "
			"Cannot get service instance of \"xsec.XMLElementWrapper\"" ) ;

		Reference< XXMLElementWrapper > xTarElement( tarElement , UNO_QUERY ) ;
		OSL_ENSURE( xTarElement.is() ,
			"Encryptor - "
			"Cannot get interface of \"XXMLElementWrapper\" from service \"xsec.XMLElementWrapper\"" ) ;

		Reference< XUnoTunnel > xTarEleTunnel( xTarElement , UNO_QUERY ) ;
		OSL_ENSURE( xTarEleTunnel.is() ,
			"Encryptor - "
			"Cannot get interface of \"XUnoTunnel\" from service \"xsec.XMLElementWrapper\"" ) ;

		XMLElementWrapper_XmlSecImpl* pTarElement = ( XMLElementWrapper_XmlSecImpl* )xTarEleTunnel->getSomething( XMLElementWrapper_XmlSecImpl::getUnoTunnelImplementationId() ) ;
		OSL_ENSURE( pTarElement != NULL ,
			"Encryptor - "
			"Cannot get implementation of \"xsec.XMLElementWrapper\"" ) ;

		pTarElement->setNativeElement( tarNode ) ;


		//Build XML Encryption template
		Reference< XInterface > enctpl =
			xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.crypto.XMLEncryptionTemplate"), xContext ) ;
		OSL_ENSURE( enctpl.is() ,
			"Encryptor - "
			"Cannot get service instance of \"xsec.XMLEncryptionTemplate\"" ) ;

		Reference< XXMLEncryptionTemplate > xTemplate( enctpl , UNO_QUERY ) ;
		OSL_ENSURE( xTemplate.is() ,
			"Encryptor - "
			"Cannot get interface of \"XXMLEncryptionTemplate\" from service \"xsec.XMLEncryptionTemplate\"" ) ;

		//Import the encryption template
		xTemplate->setTemplate( xTplElement ) ;
		xTemplate->setTarget( xTarElement ) ;

		//Create security environment
		//Build Security Environment
		Reference< XInterface > xsecenv =
			xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.security.bridge.xmlsec.SecurityEnvironment_MSCryptImpl"), xContext ) ;
		OSL_ENSURE( xsecenv.is() ,
			"Encryptor - "
			"Cannot get service instance of \"xsec.SecurityEnvironment\"" ) ;

		Reference< XSecurityEnvironment > xSecEnv( xsecenv , UNO_QUERY ) ;
		OSL_ENSURE( xSecEnv.is() ,
			"Encryptor - "
			"Cannot get interface of \"XSecurityEnvironment\" from service \"xsec.SecurityEnvironment\"" ) ;

		//Setup key slot and certDb
		Reference< XUnoTunnel > xEnvTunnel( xsecenv , UNO_QUERY ) ;
		OSL_ENSURE( xEnvTunnel.is() ,
			"Encryptor - "
			"Cannot get interface of \"XUnoTunnel\" from service \"xsec.SecurityEnvironment\"" ) ;

		SecurityEnvironment_MSCryptImpl* pSecEnv = ( SecurityEnvironment_MSCryptImpl* )xEnvTunnel->getSomething( SecurityEnvironment_MSCryptImpl::getUnoTunnelId() ) ;
		OSL_ENSURE( pSecEnv != NULL ,
			"Encryptor - "
			"Cannot get implementation of \"xsec.SecurityEnvironment\"" ) ;

		//Setup key slot and certDb
		if( n_hStoreHandle != NULL ) {
			pSecEnv->setCryptoSlot( n_hStoreHandle ) ;
			pSecEnv->setCertDb( n_hStoreHandle ) ;
		} else {
			pSecEnv->enableDefaultCrypt( sal_True ) ;
		}

		pSecEnv->adoptSymKey( symKey ) ;


		//Build XML Security Context
		Reference< XInterface > xmlsecctx =
			xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.security.bridge.xmlsec.XMLSecurityContext_MSCryptImpl"), xContext ) ;
		OSL_ENSURE( xmlsecctx.is() ,
			"Encryptor - "
			"Cannot get service instance of \"xsec.XMLSecurityContext\"" ) ;

		Reference< XXMLSecurityContext > xSecCtx( xmlsecctx , UNO_QUERY ) ;
		OSL_ENSURE( xSecCtx.is() ,
			"Encryptor - "
			"Cannot get interface of \"XXMLSecurityContext\" from service \"xsec.XMLSecurityContext\"" ) ;

		xSecCtx->addSecurityEnvironment( xSecEnv ) ;

		//Get encrypter
		Reference< XInterface > xmlencrypter =
			xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.security.bridge.xmlsec.XMLEncryption_MSCryptImpl"), xContext ) ;
		OSL_ENSURE( xmlencrypter.is() ,
			"Encryptor - "
			"Cannot get service instance of \"xsec.XMLEncryption\"" ) ;

		Reference< XXMLEncryption > xEncrypter( xmlencrypter , UNO_QUERY ) ;
		OSL_ENSURE( xEncrypter.is() ,
			"Encryptor - "
			"Cannot get interface of \"XXMLEncryption\" from service \"xsec.XMLEncryption\"" ) ;

		//perform encryption
		xTemplate = xEncrypter->encrypt( xTemplate , xSecEnv ) ;
		OSL_ENSURE( xTemplate.is() ,
			"Encryptor - "
			"Cannot encrypt the xml document" ) ;
			
			
		com::sun::star::xml::crypto::SecurityOperationStatus m_nStatus = xTemplate->getStatus();
		if (m_nStatus == SecurityOperationStatus_OPERATION_SUCCEEDED)
		{
			fprintf( stdout, "Operation succeeds.\n") ;
		}
		else
		{
			fprintf( stdout, "Operation fails.\n") ;
		}
	} catch( Exception& e ) {
		fprintf( stderr , "Error Message: %s\n" , OUStringToOString( e.Message , RTL_TEXTENCODING_ASCII_US ).getStr() ) ;
		goto done ;
	}

	dstFile = fopen( argv[2], "w" ) ;
	if( dstFile == NULL ) {
		fprintf( stderr , "### Can not open file %s\n", argv[2] ) ;
		goto done ;
	}

	//Save result
	xmlDocDump( dstFile, doc ) ;

done:
	if( dstFile != NULL )
		fclose( dstFile ) ;

	if( symKey != NULL ) {
		CryptDestroyKey( symKey ) ;
	}

	if( hCryptProv != NULL ) {
		CryptReleaseContext( hCryptProv, 0 ) ;
	}

	if( n_hStoreHandle != NULL )
		CertCloseStore( n_hStoreHandle, CERT_CLOSE_STORE_FORCE_FLAG ) ;

	/* Shutdown libxslt/libxml */
	#ifndef XMLSEC_NO_XSLT
	xsltCleanupGlobals();            
	#endif /* XMLSEC_NO_XSLT */
	xmlCleanupParser();

	return 0;
}

