/**************************************************************
 * 
 * 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_extensions.hxx"
#include <sal/types.h>
#include <rtl/memory.h>
#ifndef _RTL_WSTRING_
#include <rtl/wstring>
#endif
#include <vos/macros.hxx>

#ifndef _USR_SMARTSERVICES_HXX_
#include <usr/smartservices.hxx>
#endif
#include <com/sun/star/lang/XMultiServiceFactory.hpp>
#include <com/sun/star/io/XInputStream.hpp>
#include <com/sun/star/io/XOutputStream.hpp>
#include <com/sun/star/pgp/RecipientsEvent.hpp>
#include <com/sun/star/pgp/SignatureEvent.hpp>
#include <com/sun/star/pgp/XPGPDecoder.hpp>
#include <com/sun/star/pgp/XPGPDecoderListener.hpp>
#include <com/sun/star/pgp/XPGPEncoder.hpp>
#include <com/sun/star/pgp/XPGPPreferences.hpp>
#include <com/sun/star/uno/XInterface.hpp>
#include <com/sun/star/uno/Any.h>
#include <com/sun/star/uno/Reference.h>
#include <com/sun/star/uno/Sequence.hxx>
#include <cppuhelper/weak.hxx>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <fcntl.h>
#include <io.h>

using namespace com::sun::star::lang;
using namespace com::sun::star::io;
using namespace com::sun::star::pgp;
using namespace com::sun::star::uno;

/*========================================================================
 *
 * DataSource_Impl interface.
 *
 *======================================================================*/
class DataSource_Impl :
	public OWeakObject,
	public XInputStream
{
	Sequence<sal_Int8> m_buffer;
	sal_Int32          m_position;
	int                m_fd;

public:
	DataSource_Impl (int fd = 0);
	virtual ~DataSource_Impl (void);

	void setBuffer (const Sequence<sal_Int8> &rBuffer);

	/** XInterface.
	 */
	virtual sal_Bool SAL_CALL queryInterface (
		const Uik &rUik, Any &rIfc) throw(RuntimeException);

	virtual void SAL_CALL acquire (void) throw(RuntimeException);

	virtual void SAL_CALL release (void) throw(RuntimeException);

	/** XInputStream.
	 */
	virtual sal_Int32 SAL_CALL readBytes (
		Sequence<sal_Int8> &rData, sal_Int32 nBytesToRead)
		throw (NotConnectedException,
			   BufferSizeExceededException,
			   IOException);

	virtual sal_Int32 SAL_CALL readSomeBytes (
		Sequence<sal_Int8> &rData, sal_Int32 nMaxBytesToRead)
		throw (NotConnectedException,
			   BufferSizeExceededException,
			   IOException);

	virtual void SAL_CALL skipBytes (sal_Int32 nBytesToSkip)
		throw (NotConnectedException,
			   BufferSizeExceededException,
			   IOException);

	virtual sal_Int32 SAL_CALL available (void)
		throw (NotConnectedException, IOException);

	virtual void SAL_CALL closeInput (void)
		throw (NotConnectedException, IOException);
};

/*========================================================================
 *
 * DataSink_Impl interface.
 *
 *======================================================================*/
class DataSink_Impl :
	public OWeakObject,
	public XOutputStream
{
	Sequence<sal_Int8> m_buffer;

public:
	DataSink_Impl (void);
	virtual ~DataSink_Impl (void);

	const Sequence<sal_Int8>& getBuffer (void) const { return m_buffer; }

	/** XInterface.
	 */
	virtual sal_Bool SAL_CALL queryInterface (
		const Uik &rUik, Any &rIfc) throw(RuntimeException);

	virtual void SAL_CALL acquire (void) throw(RuntimeException);
	virtual void SAL_CALL release (void) throw(RuntimeException);

	/** XOutputStream.
	 */
	virtual void SAL_CALL writeBytes (
		const Sequence<sal_Int8> &rBuffer)
		throw (NotConnectedException,
			   BufferSizeExceededException,
			   IOException);

	virtual void SAL_CALL flush (void)
		throw (NotConnectedException,
			   BufferSizeExceededException,
			   IOException);

	virtual void SAL_CALL closeOutput (void)
		throw (NotConnectedException,
			   BufferSizeExceededException,
			   IOException);
};

/*========================================================================
 *
 * DecoderListener_Impl interface.
 *
 *======================================================================*/
class DecoderListener_Impl :
	public OWeakObject,
	public XPGPDecoderListener
{
public:
	DecoderListener_Impl (void);
	virtual ~DecoderListener_Impl (void);

	/** XInterface.
	 */
	virtual sal_Bool SAL_CALL queryInterface (
		const Uik &rUik, Any &rIfc) throw(RuntimeException);

	virtual void SAL_CALL acquire (void) throw(RuntimeException);

	virtual void SAL_CALL release (void) throw(RuntimeException);

	/** XEventListener.
	 */
	virtual void SAL_CALL disposing (const EventObject &rSource);

	/** XPGPDecoderListener.
	 */
	virtual void SAL_CALL decrypted (const RecipientsEvent &rEvent);
	virtual void SAL_CALL verified  (const SignatureEvent &rEvent);
};

/*========================================================================
 *
 * DataSource_Impl implementation.
 *
 *======================================================================*/
/*
 * DataSource_Impl.
 */
DataSource_Impl::DataSource_Impl (int fd)
	: m_fd (fd), m_position (0)
{
}

/*
 * ~DataSource_Impl.
 */
DataSource_Impl::~DataSource_Impl (void)
{
}

/*
 * DataSource_Impl: setBuffer.
 */
void DataSource_Impl::setBuffer (const Sequence<sal_Int8> &rBuffer)
{
	m_buffer = rBuffer;
	if (!m_buffer.getLength())
	{
		// Fill buffer from file descriptor.
		char buffer[1024];
		rtl_zeroMemory (buffer, sizeof(buffer));

		int k = read (m_fd, buffer, sizeof(buffer));
		while (k > 0)
		{
			sal_Int32 n = m_buffer.getLength();
			m_buffer.realloc (n + k);

			rtl_copyMemory (m_buffer.getArray() + n, buffer, k);
			n += k;

			rtl_zeroMemory (buffer, k);
			k = read (m_fd, buffer, sizeof(buffer));
		}
	}
	m_position = 0;
}

/*
 * XInterface: queryInterface.
 */
sal_Bool SAL_CALL DataSource_Impl::queryInterface (
	const Uik &rUik, Any &rIfc) throw(RuntimeException)
{
	if (com::sun::star::uno::queryInterface (
		rUik, rIfc,
		SAL_STATIC_CAST (XInputStream*, this)))
		return sal_True;
	else
		return OWeakObject::queryInterface (rUik, rIfc);
}

/*
 * XInterface: acquire.
 */
void SAL_CALL DataSource_Impl::acquire (void) throw(RuntimeException)
{
	OWeakObject::acquire();
}

/*
 * XInterface: release.
 */
void SAL_CALL DataSource_Impl::release (void) throw(RuntimeException)
{
	OWeakObject::release();
}

/*
 * XInputStream: readBytes.
 */
sal_Int32 SAL_CALL DataSource_Impl::readBytes (
	Sequence<sal_Int8> &rData, sal_Int32 nBytesToRead)
	throw (NotConnectedException, BufferSizeExceededException, IOException)
{
	if (nBytesToRead < 0)
		throw IOException();

	sal_Int32 k = m_buffer.getLength() - m_position;
	k = VOS_BOUND(k, 0, nBytesToRead);
	if (k > 0)
	{
		rData.realloc(k);
		rtl_copyMemory (
			rData.getArray(), m_buffer.getConstArray() + m_position, k);
		m_position += k;
	}
	return k;
}

/*
 * XInputStream: readSomeBytes.
 */
sal_Int32 SAL_CALL DataSource_Impl::readSomeBytes (
	Sequence<sal_Int8> &rData, sal_Int32 nMaxBytesToRead)
	throw (NotConnectedException, BufferSizeExceededException, IOException)
{
	return readBytes (rData, nMaxBytesToRead);
}

/*
 * XInputStream: skipBytes.
 */
void SAL_CALL DataSource_Impl::skipBytes (sal_Int32 nBytesToSkip)
	throw (NotConnectedException, BufferSizeExceededException, IOException)
{
	if (nBytesToSkip < 0)
		throw IOException();

	m_position += nBytesToSkip;
}

/*
 * XInputStream: available.
 */
sal_Int32 SAL_CALL DataSource_Impl::available (void)
	throw (NotConnectedException, IOException)
{
	sal_Int32 k = m_buffer.getLength() - m_position;
	return ((k > 0) ? k : 0);
}

/*
 * XInputStream: closeInput.
 */
void SAL_CALL DataSource_Impl::closeInput (void)
	throw (NotConnectedException, IOException)
{
}

/*========================================================================
 *
 * DataSink_Impl implementation.
 *
 *======================================================================*/
/*
 * DataSink_Impl.
 */
DataSink_Impl::DataSink_Impl (void)
{
}

/*
 * ~DataSink_Impl.
 */
DataSink_Impl::~DataSink_Impl (void)
{
}

/*
 * XInterface: queryInterface.
 */
sal_Bool SAL_CALL DataSink_Impl::queryInterface (
	const Uik &rUik, Any &rIfc) throw(RuntimeException)
{
	if (com::sun::star::uno::queryInterface (
		rUik, rIfc,
		SAL_STATIC_CAST (XOutputStream*, this)))
		return sal_True;
	else
		return OWeakObject::queryInterface (rUik, rIfc);
}

/*
 * XInterface: acquire.
 */
void SAL_CALL DataSink_Impl::acquire (void) throw(RuntimeException)
{
	OWeakObject::acquire();
}

/*
 * XInterface: release.
 */
void SAL_CALL DataSink_Impl::release (void) throw(RuntimeException)
{
	OWeakObject::release();
}

/*
 * XOutputStream: writeBytes.
 */
void SAL_CALL DataSink_Impl::writeBytes (const Sequence<sal_Int8> &rBuffer)
	throw (NotConnectedException, BufferSizeExceededException, IOException)
{
	if (rBuffer.getLength())
	{
		sal_Int32 n = m_buffer.getLength();
		m_buffer.realloc (n + rBuffer.getLength());

		rtl_copyMemory (
			m_buffer.getArray() + n,
			rBuffer.getConstArray(),
			rBuffer.getLength());
	}
}

/*
 * XOutputStream: flush.
 */
void SAL_CALL DataSink_Impl::flush (void)
	throw (NotConnectedException, BufferSizeExceededException, IOException)
{
	if (m_buffer.getLength())
	{
		// Write data to stdout.
		const sal_Int8 *pData = m_buffer.getConstArray();
		sal_Int32       nData = m_buffer.getLength();

		int prev = ::setmode (1, O_BINARY);
		if (!(prev < 0))
		{
			int k = ::write (1, pData, nData);
			if (k > 0)
				::write (1, "\n", 1);
			::setmode (1, prev);
		}
	}
}

/*
 * XOutputStream: closeOutput.
 */
void SAL_CALL DataSink_Impl::closeOutput (void)
	throw (NotConnectedException, BufferSizeExceededException, IOException)
{
	flush();
}

/*========================================================================
 *
 * DecoderListener_Impl implementation.
 *
 *======================================================================*/
/*
 * DecoderListener_Impl.
 */
DecoderListener_Impl::DecoderListener_Impl (void)
{
}

/*
 * ~DecoderListener_Impl.
 */
DecoderListener_Impl::~DecoderListener_Impl (void)
{
}

/*
 * XInterface: queryInterface.
 */
sal_Bool SAL_CALL DecoderListener_Impl::queryInterface (
	const Uik &rUik, Any &rIfc) throw(RuntimeException)
{
	if (com::sun::star::uno::queryInterface (
		rUik, rIfc,
		SAL_STATIC_CAST (XEventListener*, this),
		SAL_STATIC_CAST (XPGPDecoderListener*, this)))
		return sal_True;
	else
		return OWeakObject::queryInterface (rUik, rIfc);
}

/*
 * XInterface: acquire.
 */
void SAL_CALL DecoderListener_Impl::acquire (void) throw(RuntimeException)
{
	OWeakObject::acquire();
}

/*
 * XInterface: release.
 */
void SAL_CALL DecoderListener_Impl::release (void) throw(RuntimeException)
{
	OWeakObject::release();
}

/*
 * XEventListener: disposing.
 */
void SAL_CALL DecoderListener_Impl::disposing (const EventObject &rSource)
{
}

/*
 * XPGPDecoderListener: decrypted.
 */
void SAL_CALL DecoderListener_Impl::decrypted (const RecipientsEvent &rEvent)
{
}

/*
 * XPGPDecoderListener: verified.
 */
void SAL_CALL DecoderListener_Impl::verified (const SignatureEvent &rEvent)
{
}

/*========================================================================
 *
 * Main.
 *
 *======================================================================*/
inline rtl::OWString S2U (const sal_Char *ascii)
{
	return rtl::OWString::createFromAscii (ascii);
}

#if 0  /* OLD */

/*
 * queryModuleActivator.
 */
BOOL queryModuleActivator (
	const XServiceManagerRef &rxManager,
	XServiceActivatorRef     &rxActivator)
{
	XServiceProviderRef xProv;
	XInterfaceRef       xProvInst;

	xProv = rxManager->queryServiceProvider (
		L"stardiv.uno.ServiceActivator.module");
	if (!xProv.is())
	{
		printf ("Error: no ServiceActivator service.\n");
		return FALSE;
	}

	xProvInst = xProv->createInstance();
	if (!xProvInst.is())
	{
		printf ("Error: no ServiceActivator instance.\n");
		return FALSE;
	}

	return xProvInst->queryInterface (
		XServiceActivator::getSmartUik(), rxActivator);
}

/*
 * install.
 */
BOOL install (
	const XServiceActivatorRef &rxActivator,
	const char                 *prefix)
{
	String aModule ("module://");
	char   pBuffer[1024];

	vos::ORealDynamicLoader::computeModuleName (
		prefix, pBuffer, sizeof(pBuffer));
	aModule += pBuffer;

	return rxActivator->install (
		StringToUString (aModule, CHARSET_SYSTEM));
}

/*
 * uninstall.
 */
BOOL uninstall (
	const XServiceActivatorRef &rxActivator,
	const char                 *prefix)
{
	String aModule ("module://");
	char   pBuffer[1024];

	vos::ORealDynamicLoader::computeModuleName (
		prefix, pBuffer, sizeof(pBuffer));
	aModule += pBuffer;

	return rxActivator->deinstall (
		StringToUString (aModule, CHARSET_SYSTEM));
}

#endif /* OLD */

/*
 * main.
 */
int SAL_CALL main (int argc, char **argv)
{
	enum Option
	{
		OPTION_INSTALL   = 0x01,
		OPTION_UNINSTALL = 0x02,

		OPTION_DECRYPT   = 0x04,
		OPTION_ENCRYPT   = 0x08,
		OPTION_SIGN      = 0x10,

		OPTION_FILE      = 0x20,
		OPTION_HELP      = 0x40
	};

	int fd = 0;
	unsigned long nOptions = 0;

	for (int i = 1; i < argc; i++)
	{
		const char *opt = argv[i];
		if (opt[0] == '-')
		{
			switch (opt[1])
			{
				case 'i':
					nOptions |= OPTION_INSTALL;
					break;

				case 'u':
					nOptions |= OPTION_UNINSTALL;
					break;

				case 'd':
					nOptions |= OPTION_DECRYPT;
					break;

				case 'e':
					nOptions |= OPTION_ENCRYPT;
					break;

				case 's':
					nOptions |= OPTION_SIGN;
					break;

				case 'f':
					nOptions |= OPTION_FILE;
					break;

				case 'h':
				default:
					nOptions |= OPTION_HELP;
					break;
			}
		}
		else
		{
			if (nOptions & OPTION_FILE)
			{
				if ((fd = open (argv[i], O_RDONLY | O_BINARY)) < 0)
				{
					printf ("Error: can't open file: %s\n", argv[i]);
					exit (0);
				}
			}
		}
	}

	if ((nOptions == 0) || (nOptions & OPTION_HELP))
	{
		printf ("Options:\n");
		printf ("-i\tinstall module\n");
		printf ("-u\tuninstall module\n");
		printf ("-d\tdecrypt and verify\n");
		printf ("-e\tencrypt test pattern\n");
		printf ("-s\tsign test pattern\n");
		printf ("-h\thelp\n");
		exit (0);
	}

	Reference<XMultiServiceFactory> xManager (
		usr::createDefaultSmartRegistryServiceFactory());
	if (!xManager.is())
	{
		printf ("Error: no ProcessServiceManager.\n");
		exit (1);
	}
	usr::setProcessServiceFactory (xManager);

	if (nOptions & OPTION_INSTALL)
	{
#if 0  /* OLD */
		XServiceActivatorRef xActivator;
		if (queryModuleActivator (xManager, xActivator))
		{
			if (install (xActivator, "pgp"))
				printf ("Module PGP installed.\n");
			else
				printf ("Error: module PGP not installed.\n");
		}
		nOptions &= ~OPTION_INSTALL;
#endif /* OLD */
	}

	if (nOptions & (OPTION_DECRYPT | OPTION_ENCRYPT | OPTION_SIGN))
	{
		Reference<XMultiServiceFactory> xProv (
			xManager->createInstance (
				S2U("com.sun.star.pgp.PGPFactory")),
			UNO_QUERY);
		if (!xProv.is())
		{
			printf ("Error: no PGPFactory service.\n");
			exit (1);
		}

		Reference<XInterface> xProvInst (
			xProv->createInstance (
				S2U("com.sun.star.pgp.SimplePGPMailer")));
		if (!xProvInst.is())
		{
			printf ("Error: no SimplePGPMailer service.\n");
			exit (2);
		}

		Reference<XPGPPreferences> xPrefs (xProvInst, UNO_QUERY);
		if (xPrefs.is())
		{
			unsigned long nDefaults = 0;

			if (xPrefs->getEncryptByDefault())
				nDefaults |= OPTION_ENCRYPT;
			if (xPrefs->getSignByDefault())
				nDefaults |= OPTION_SIGN;
			if (xPrefs->getAutoDecrypt())
				nDefaults |= OPTION_DECRYPT;

			if (nDefaults)
			{
			}
		}

		static const sal_Int8 pData[] = "" /* "Hello PGP World." */;
		Sequence<sal_Int8> buffer (pData, sizeof (pData) - 1);

		if (nOptions & (OPTION_ENCRYPT | OPTION_SIGN))
		{
			Reference<XPGPEncoder> xEncoder (xProvInst, UNO_QUERY);
			if (!xEncoder.is())
			{
				printf ("Error: no PGPEncoder interface.\n");
				exit (4);
			}
			
			DataSource_Impl *source = new DataSource_Impl (fd);
			source->setBuffer (buffer);

			DataSink_Impl *sink = new DataSink_Impl;

			Reference<XInputStream>  xPlainText  (source);
			Reference<XOutputStream> xCipherText (sink);

			if (nOptions & OPTION_ENCRYPT)
			{
				rtl::OWString aRecipients[] =
				{
					S2U("er@stardiv.de"),
					// L"mhu@stardivision.de",
					S2U("mhu@rabbit")
				};

				sal_Int32 nRecipients =
					sizeof(aRecipients) / sizeof(aRecipients[0]);

				if (nOptions & OPTION_SIGN)
				{
					xEncoder->encryptAndSign (
						Sequence<rtl::OWString>(aRecipients, nRecipients),
						xPlainText,
						xCipherText);
					nOptions &= ~OPTION_SIGN;
				}
				else
				{
					xEncoder->encrypt (
						Sequence<rtl::OWString>(aRecipients, nRecipients),
						xPlainText,
						xCipherText);
				}
				nOptions &= ~OPTION_ENCRYPT;
			}

			if (nOptions & OPTION_SIGN)
			{
				sal_Bool bDataIsAscii = (fd == 0); // stdin.

				xEncoder->sign (
					bDataIsAscii,
					xPlainText,
					xCipherText);
				nOptions &= ~OPTION_SIGN;
			}

			buffer = sink->getBuffer();
		}

		if (nOptions & OPTION_DECRYPT)
		{
			Reference<XPGPDecoder> xDecoder (xProvInst, UNO_QUERY);
			if (!xDecoder.is())
			{
				printf ("Error: no PGPDecoder interface.\n");
				exit (5);
			}

			DataSource_Impl *source = new DataSource_Impl;
			source->setBuffer (buffer);

			DataSink_Impl *sink = new DataSink_Impl;

			Reference<XInputStream>  xCipherText (source);
			Reference<XOutputStream> xPlainText  (sink);

			Reference<XPGPDecoderListener> xListener (
				new DecoderListener_Impl);
			xDecoder->addDecoderListener (xListener);

			xDecoder->decryptAndVerify (
				xCipherText,
				xPlainText);
			nOptions &= ~OPTION_DECRYPT;

			xDecoder->removeDecoderListener (xListener);

			buffer = sink->getBuffer();
		}
	}

	if (nOptions & OPTION_UNINSTALL)
	{
#if 0  /* OLD */
		XServiceActivatorRef xActivator;
		if (queryModuleActivator (xManager, xActivator))
		{
			if (uninstall (xActivator, "pgp"))
				printf ("Module PGP uninstalled.\n");
			else
				printf ("Error: module PGP not uninstalled.\n");
		}
		nOptions &= ~OPTION_UNINSTALL;
#endif /* OLD */
	}

	return 0;
}