/**************************************************************
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 *
 *************************************************************/


// MARKER(update_precomp.py): autogen include statement, do not remove
#include "precompiled_xmlsecurity.hxx"

#include "signatureengine.hxx"
#include <com/sun/star/xml/crypto/XXMLSignatureTemplate.hpp>
#include <com/sun/star/xml/wrapper/XXMLElementWrapper.hpp>
#include <com/sun/star/lang/XMultiServiceFactory.hpp>

namespace cssu = com::sun::star::uno;
namespace cssl = com::sun::star::lang;
namespace cssxc = com::sun::star::xml::crypto;
namespace cssxw = com::sun::star::xml::wrapper;

#define SIGNATURE_TEMPLATE "com.sun.star.xml.crypto.XMLSignatureTemplate"

#define	DECLARE_ASCII( SASCIIVALUE )																			\
	rtl::OUString( RTL_CONSTASCII_USTRINGPARAM( SASCIIVALUE ) )

SignatureEngine::SignatureEngine( )
	:m_nTotalReferenceNumber(-1)
{
}

bool SignatureEngine::checkReady() const
/****** SignatureEngine/checkReady *******************************************
 *
 *   NAME
 *	checkReady -- checks the conditions for the main operation.
 *
 *   SYNOPSIS
 *	bReady = checkReady( );
 *
 *   FUNCTION
 *	checks whether all following conditions are satisfied:
 *	1. the main operation has't begun yet;
 *	2. the key material is known;
 *	3. the amount of reference is known;
 *	4. all of referenced elements, the key element and the signature
 *	   template are bufferred.
 *
 *   INPUTS
 *	empty
 *
 *   RESULT
 *	bReady - true if all conditions are satisfied, false otherwise
 *
 *   HISTORY
 *	05.01.2004 -	implemented
 *
 *   AUTHOR
 *	Michael Mi
 *	Email: michael.mi@sun.com
 ******************************************************************************/
{
	bool rc = true;

	sal_Int32 nKeyInc = 0;
	if (m_nIdOfKeyEC != 0)
	{
		nKeyInc = 1;
	}

	if (m_bMissionDone ||
	    m_nIdOfKeyEC == -1 ||
	    m_nTotalReferenceNumber == -1 ||
	    m_nTotalReferenceNumber+1+nKeyInc > m_nNumOfResolvedReferences)
	{
		rc = false;
	}

	return rc;
}

void SignatureEngine::tryToPerform( )
    	throw (cssu::Exception, cssu::RuntimeException)
/****** SignatureEngine/tryToPerform *****************************************
 *
 *   NAME
 *	tryToPerform -- tries to perform the signature operation.
 *
 *   SYNOPSIS
 *	tryToPerform( );
 *
 *   FUNCTION
 *	if the situation is ready, perform following operations.
 *	1. prepares a signature template;
 *	2. calls the signature bridge component;
 *	3. clears up all used resources;
 *	4. notifies the result listener;
 *	5. sets the "accomplishment" flag.
 *
 *   INPUTS
 *	empty
 *
 *   RESULT
 *	empty
 *
 *   HISTORY
 *	05.01.2004 -	implemented
 *
 *   AUTHOR
 *	Michael Mi
 *	Email: michael.mi@sun.com
 ******************************************************************************/
{
	if (checkReady())
	{
		const rtl::OUString ouSignatureTemplate (
			RTL_CONSTASCII_USTRINGPARAM( SIGNATURE_TEMPLATE ) );
		cssu::Reference < cssxc::XXMLSignatureTemplate >
			xSignatureTemplate( mxMSF->createInstance( ouSignatureTemplate ), cssu::UNO_QUERY );

		OSL_ASSERT( xSignatureTemplate.is() );

		cssu::Reference< cssxw::XXMLElementWrapper >
			xXMLElement = m_xSAXEventKeeper->getElement( m_nIdOfTemplateEC );

		xSignatureTemplate->setTemplate(xXMLElement);

		std::vector< sal_Int32 >::const_iterator ii = m_vReferenceIds.begin();

		for( ; ii != m_vReferenceIds.end() ; ++ii )
		{
			xXMLElement = m_xSAXEventKeeper->getElement( *ii );
			xSignatureTemplate->setTarget(xXMLElement);
		}

		/*
		 * set the Uri binding
		 */
		xSignatureTemplate->setBinding( this );

		startEngine( xSignatureTemplate );

		/*
		 * done
		 */
		clearUp( );

		notifyResultListener();

		m_bMissionDone = true;
	}
}

void SignatureEngine::clearUp( ) const
/****** SignatureEngine/clearUp **********************************************
 *
 *   NAME
 *	clearUp -- clear up all resources used by this operation.
 *
 *   SYNOPSIS
 *	clearUp( );
 *
 *   FUNCTION
 *	cleaning resources up includes:
 *	1. releases the ElementCollector for the signature template element;
 *	2. releases ElementCollectors for referenced elements;
 *	3. releases the ElementCollector for the key element, if there is one.
 *
 *   INPUTS
 *	empty
 *
 *   RESULT
 *	empty
 *
 *   HISTORY
 *	05.01.2004 -	implemented
 *
 *   AUTHOR
 *	Michael Mi
 *	Email: michael.mi@sun.com
 ******************************************************************************/
{
	cssu::Reference < cssxc::sax::XReferenceResolvedBroadcaster >
		xReferenceResolvedBroadcaster( m_xSAXEventKeeper, cssu::UNO_QUERY );
	xReferenceResolvedBroadcaster->removeReferenceResolvedListener(
		m_nIdOfTemplateEC,
		(const cssu::Reference < cssxc::sax::XReferenceResolvedListener >)((SecurityEngine *)this));

	m_xSAXEventKeeper->removeElementCollector(m_nIdOfTemplateEC);

	std::vector< sal_Int32 >::const_iterator ii = m_vReferenceIds.begin();

	for( ; ii != m_vReferenceIds.end() ; ++ii )
	{
		xReferenceResolvedBroadcaster->removeReferenceResolvedListener(
			*ii,
			(const cssu::Reference < cssxc::sax::XReferenceResolvedListener >)((SecurityEngine *)this));
		m_xSAXEventKeeper->removeElementCollector(*ii);
	}

	if (m_nIdOfKeyEC != 0 && m_nIdOfKeyEC != -1)
	{
		m_xSAXEventKeeper->removeElementCollector(m_nIdOfKeyEC);
	}
}

/* XReferenceCollector */
void SAL_CALL SignatureEngine::setReferenceCount( sal_Int32 count )
	throw (cssu::Exception, cssu::RuntimeException)
{
	m_nTotalReferenceNumber = count;
	tryToPerform();
}

void SAL_CALL SignatureEngine::setReferenceId( sal_Int32 id )
	throw (cssu::Exception, cssu::RuntimeException)
{
	m_vReferenceIds.push_back( id );
}

/* XUriBinding */
void SAL_CALL SignatureEngine::setUriBinding(
	const rtl::OUString& uri,
	const cssu::Reference< com::sun::star::io::XInputStream >& aInputStream )
	throw (cssu::Exception, cssu::RuntimeException)
{
	m_vUris.push_back(uri);
	m_vXInputStreams.push_back(aInputStream);
}

cssu::Reference< com::sun::star::io::XInputStream > SAL_CALL SignatureEngine::getUriBinding( const rtl::OUString& uri )
	throw (cssu::Exception, cssu::RuntimeException)
{
	cssu::Reference< com::sun::star::io::XInputStream > xInputStream;

	int size = m_vUris.size();

	for( int i=0; i<size; ++i)
	{
		if (m_vUris[i] == uri)
		{
			xInputStream = m_vXInputStreams[i];
			break;
		}
	}

	return xInputStream;
}