diff options
Diffstat (limited to 'io/source/acceptor/acc_socket.cxx')
| -rw-r--r-- | io/source/acceptor/acc_socket.cxx | 416 |
1 files changed, 416 insertions, 0 deletions
diff --git a/io/source/acceptor/acc_socket.cxx b/io/source/acceptor/acc_socket.cxx new file mode 100644 index 000000000000..ecdf59495f3d --- /dev/null +++ b/io/source/acceptor/acc_socket.cxx @@ -0,0 +1,416 @@ +/************************************************************************* + * + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * Copyright 2000, 2010 Oracle and/or its affiliates. + * + * OpenOffice.org - a multi-platform office productivity suite + * + * This file is part of OpenOffice.org. + * + * OpenOffice.org is free software: you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License version 3 + * only, as published by the Free Software Foundation. + * + * OpenOffice.org is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Lesser General Public License version 3 for more details + * (a copy is included in the LICENSE file that accompanied this code). + * + * You should have received a copy of the GNU Lesser General Public License + * version 3 along with OpenOffice.org. If not, see + * <http://www.openoffice.org/license.html> + * for a copy of the LGPLv3 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(); + } +} + + |