/**************************************************************
 * 
 * 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_io.hxx"
#include "acceptor.hxx"

#include <hash_set>
#include <algorithm>

#include <rtl/ustrbuf.hxx>
#include <com/sun/star/connection/XConnectionBroadcaster.hpp>
#include <com/sun/star/connection/ConnectionSetupException.hpp>

#include <cppuhelper/implbase2.hxx>

using namespace ::osl;
using namespace ::rtl;
using namespace ::cppu;
using namespace ::com::sun::star::uno;
using namespace ::com::sun::star::io;
using namespace ::com::sun::star::connection;


namespace io_acceptor {
	template<class T>
	struct ReferenceHash
	{
		size_t operator () (const ::com::sun::star::uno::Reference<T> & ref) const
        {
			return (size_t)ref.get();
		}
	};

	template<class T>
	struct ReferenceEqual
	{
		sal_Bool operator () (const ::com::sun::star::uno::Reference<T> & op1, 
							  const ::com::sun::star::uno::Reference<T> & op2) const
        {
			return op1.get() == op2.get();
		}
	};


	typedef ::std::hash_set< ::com::sun::star::uno::Reference< ::com::sun::star::io::XStreamListener>, 
                             ReferenceHash< ::com::sun::star::io::XStreamListener>, 
                             ReferenceEqual< ::com::sun::star::io::XStreamListener> >
	        XStreamListener_hash_set;


	class SocketConnection : public ::cppu::WeakImplHelper2< 
        ::com::sun::star::connection::XConnection,
		::com::sun::star::connection::XConnectionBroadcaster>
	
	{
	public:
		SocketConnection( const OUString & sConnectionDescription );
		~SocketConnection();
		
		virtual sal_Int32 SAL_CALL read( ::com::sun::star::uno::Sequence< sal_Int8 >& aReadBytes,
										 sal_Int32 nBytesToRead )
			throw(::com::sun::star::io::IOException,
				  ::com::sun::star::uno::RuntimeException);
		virtual void SAL_CALL write( const ::com::sun::star::uno::Sequence< sal_Int8 >& aData )
			throw(::com::sun::star::io::IOException,
				  ::com::sun::star::uno::RuntimeException);
		virtual void SAL_CALL flush(  ) throw(
			::com::sun::star::io::IOException,
			::com::sun::star::uno::RuntimeException);
		virtual void SAL_CALL close(  )
			throw(::com::sun::star::io::IOException,
				  ::com::sun::star::uno::RuntimeException);
		virtual ::rtl::OUString SAL_CALL getDescription(  )
			throw(::com::sun::star::uno::RuntimeException);

		// XConnectionBroadcaster
		virtual void SAL_CALL addStreamListener(const ::com::sun::star::uno::Reference< ::com::sun::star::io::XStreamListener>& aListener) 
			throw(::com::sun::star::uno::RuntimeException);
		virtual void SAL_CALL removeStreamListener(const ::com::sun::star::uno::Reference< ::com::sun::star::io::XStreamListener>& aListener) 
			throw(::com::sun::star::uno::RuntimeException);

	public:
		void completeConnectionString();
		
		::osl::StreamSocket m_socket;
		::osl::SocketAddr m_addr;
		oslInterlockedCount m_nStatus;
		::rtl::OUString m_sDescription;

		::osl::Mutex _mutex;
		sal_Bool     _started;
		sal_Bool     _closed;
		sal_Bool     _error;
		XStreamListener_hash_set _listeners;
	};

	template<class T>
	void notifyListeners(SocketConnection * pCon, sal_Bool * notified, T t)
	{
  		XStreamListener_hash_set listeners;

		{
			::osl::MutexGuard guard(pCon->_mutex);
			if(!*notified) 
			{
				*notified = sal_True;
				listeners = pCon->_listeners;
			}
		}

		::std::for_each(listeners.begin(), listeners.end(), t);
	}

	static void callStarted(Reference<XStreamListener> xStreamListener)
	{
		xStreamListener->started();
	}

	struct callError {
		const Any & any;

		callError(const Any & any);

		void operator () (Reference<XStreamListener> xStreamListener);
	};

	callError::callError(const Any & aAny)
		: any(aAny)
	{
	}

	void callError::operator () (Reference<XStreamListener> xStreamListener)
	{
		xStreamListener->error(any);
	}

	static void callClosed(Reference<XStreamListener> xStreamListener)
	{
		xStreamListener->closed();
	}

	
	SocketConnection::SocketConnection( const OUString &sConnectionDescription) :
		m_nStatus( 0 ),
		m_sDescription( sConnectionDescription ),
		_started(sal_False),
		_closed(sal_False),
		_error(sal_False)
	{
		g_moduleCount.modCnt.acquire( &g_moduleCount.modCnt );
		// make it unique
		m_sDescription += OUString( RTL_CONSTASCII_USTRINGPARAM( ",uniqueValue=" ) );
		m_sDescription += OUString::valueOf(
            sal::static_int_cast< sal_Int64 >(
                reinterpret_cast< sal_IntPtr >(&m_socket)),
            10 );
	}

	SocketConnection::~SocketConnection()
	{
		g_moduleCount.modCnt.release( &g_moduleCount.modCnt );
	}

	void SocketConnection::completeConnectionString()
	{
		OUStringBuffer buf( 256 );
		buf.appendAscii( ",peerPort=" );
		buf.append( (sal_Int32) m_socket.getPeerPort() );
		buf.appendAscii( ",peerHost=" );
		buf.append( m_socket.getPeerHost( ) );
		
		buf.appendAscii( ",localPort=" );
		buf.append( (sal_Int32) m_socket.getLocalPort() );
		buf.appendAscii( ",localHost=" );
		buf.append( m_socket.getLocalHost() );

		m_sDescription += buf.makeStringAndClear();
	}

	sal_Int32 SocketConnection::read( Sequence < sal_Int8 > & aReadBytes , sal_Int32 nBytesToRead )
			throw(::com::sun::star::io::IOException,
				  ::com::sun::star::uno::RuntimeException)
	{
		if( ! m_nStatus )
		{
			notifyListeners(this, &_started, callStarted);

			if( aReadBytes.getLength() != nBytesToRead )
			{
				aReadBytes.realloc( nBytesToRead );
			}

			sal_Int32 i = 0;
			i = m_socket.read( aReadBytes.getArray()  , aReadBytes.getLength() );

			if(i != nBytesToRead)
			{
				OUString message(RTL_CONSTASCII_USTRINGPARAM("acc_socket.cxx:SocketConnection::read: error - "));
				message +=	m_socket.getErrorAsString();
				
				IOException ioException(message, Reference<XInterface>(static_cast<XConnection *>(this)));
				
				Any any;
				any <<= ioException;
				
				notifyListeners(this, &_error, callError(any));
				
				throw ioException;
			}

			return i;
		}
		else
		{
			OUString message(RTL_CONSTASCII_USTRINGPARAM("acc_socket.cxx:SocketConnection::read: error - connection already closed"));

			IOException ioException(message, Reference<XInterface>(static_cast<XConnection *>(this)));

			Any any;
			any <<= ioException;

			notifyListeners(this, &_error, callError(any));

			throw ioException;
		}
	}

	void SocketConnection::write( const Sequence < sal_Int8 > &seq )
			throw(::com::sun::star::io::IOException,
				  ::com::sun::star::uno::RuntimeException)
	{
		if( ! m_nStatus )
		{
			if( m_socket.write( seq.getConstArray() , seq.getLength() ) != seq.getLength() )
			{
				OUString message(RTL_CONSTASCII_USTRINGPARAM("acc_socket.cxx:SocketConnection::write: error - "));
				message += m_socket.getErrorAsString();
				
				IOException ioException(message, Reference<XInterface>(static_cast<XConnection *>(this)));
				
				Any any;
				any <<= ioException;
				
				notifyListeners(this, &_error, callError(any));

				throw ioException;
			}
		}
		else
		{
			OUString message(RTL_CONSTASCII_USTRINGPARAM("acc_socket.cxx:SocketConnection::write: error - connection already closed"));

			IOException ioException(message, Reference<XInterface>(static_cast<XConnection *>(this)));

			Any any;
			any <<= ioException;

			notifyListeners(this, &_error, callError(any));

			throw ioException;
		}
	}

	void SocketConnection::flush( )
			throw(::com::sun::star::io::IOException,
				  ::com::sun::star::uno::RuntimeException)
	{

	}

	void SocketConnection::close()
			throw(::com::sun::star::io::IOException,
				  ::com::sun::star::uno::RuntimeException)
	{
		// enshure close is called only once
		if(  1 == osl_incrementInterlockedCount( (&m_nStatus) ) )
		{
			m_socket.shutdown();
			notifyListeners(this, &_closed, callClosed);
		}
	}

	OUString SocketConnection::getDescription()
			throw( ::com::sun::star::uno::RuntimeException)
	{
		return m_sDescription;
	}


	// XConnectionBroadcaster
	void SAL_CALL SocketConnection::addStreamListener(const Reference<XStreamListener> & aListener) throw(RuntimeException)
	{
		MutexGuard guard(_mutex);

		_listeners.insert(aListener);
	}

	void SAL_CALL SocketConnection::removeStreamListener(const Reference<XStreamListener> & aListener) throw(RuntimeException)
	{
		MutexGuard guard(_mutex);

		_listeners.erase(aListener);
	}

	SocketAcceptor::SocketAcceptor( const OUString &sSocketName,
									sal_uInt16 nPort,
									sal_Bool bTcpNoDelay,
									const OUString &sConnectionDescription) :
		m_sSocketName( sSocketName ),
		m_sConnectionDescription( sConnectionDescription ),
		m_nPort( nPort ),
		m_bTcpNoDelay( bTcpNoDelay ),
		m_bClosed( sal_False )
	{
	}


	void SocketAcceptor::init()
	{
		if( ! m_addr.setPort( m_nPort ) )
		{
			OUStringBuffer message( 128 );
			message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - invalid tcp/ip port " );
			message.append( (sal_Int32) m_nPort );
			throw ConnectionSetupException(
				message.makeStringAndClear() , Reference< XInterface> () );
		}
		if( ! m_addr.setHostname( m_sSocketName.pData ) )
		{
			OUStringBuffer message( 128 );
			message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - invalid host " );
			message.append( m_sSocketName );
			throw ConnectionSetupException(
				message.makeStringAndClear(), Reference< XInterface > () );
		}
		m_socket.setOption( osl_Socket_OptionReuseAddr, 1);
		
		if(! m_socket.bind(m_addr) )
		{
			OUStringBuffer message( 128 );
			message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - couldn't bind on " );
			message.append( m_sSocketName ).appendAscii( ":" ).append((sal_Int32)m_nPort);
			throw ConnectionSetupException(
				message.makeStringAndClear(),
				Reference<XInterface>());
		}
		
		if(! m_socket.listen() )
		{
			OUStringBuffer message( 128 );
			message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - can't listen on " );
			message.append( m_sSocketName ).appendAscii( ":" ).append( (sal_Int32) m_nPort);
			throw ConnectionSetupException(	message.makeStringAndClear(),Reference<XInterface>() );
		}
	}

	Reference< XConnection > SocketAcceptor::accept( )
	{
		SocketConnection *pConn = new SocketConnection( m_sConnectionDescription );

		if( m_socket.acceptConnection( pConn->m_socket )!= osl_Socket_Ok )
		{
			// stopAccepting was called
			delete pConn;
			return Reference < XConnection > ();
		}
		if( m_bClosed )
		{
			delete pConn;
			return Reference < XConnection > ();
		}

		pConn->completeConnectionString();
		if( m_bTcpNoDelay )
		{
			sal_Int32 nTcpNoDelay = sal_True;
			pConn->m_socket.setOption( osl_Socket_OptionTcpNoDelay , &nTcpNoDelay,
									   sizeof( nTcpNoDelay ) , osl_Socket_LevelTcp );
		}

		return Reference < XConnection > ( (XConnection * ) pConn );
	}

	void SocketAcceptor::stopAccepting()
	{
		m_bClosed = sal_True;
		m_socket.close();
	}
}