| /************************************************************** |
| * |
| * 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(); |
| } |
| } |
| |
| |