LCOV - code coverage report
Current view: top level - libreoffice/io/source/acceptor - acc_socket.cxx (source / functions) Hit Total Coverage
Test: libreoffice_filtered.info Lines: 0 144 0.0 %
Date: 2012-12-27 Functions: 0 23 0.0 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
       2             : /*
       3             :  * This file is part of the LibreOffice project.
       4             :  *
       5             :  * This Source Code Form is subject to the terms of the Mozilla Public
       6             :  * License, v. 2.0. If a copy of the MPL was not distributed with this
       7             :  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
       8             :  *
       9             :  * This file incorporates work covered by the following license notice:
      10             :  *
      11             :  *   Licensed to the Apache Software Foundation (ASF) under one or more
      12             :  *   contributor license agreements. See the NOTICE file distributed
      13             :  *   with this work for additional information regarding copyright
      14             :  *   ownership. The ASF licenses this file to you under the Apache
      15             :  *   License, Version 2.0 (the "License"); you may not use this file
      16             :  *   except in compliance with the License. You may obtain a copy of
      17             :  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
      18             :  */
      19             : 
      20             : #include "acceptor.hxx"
      21             : 
      22             : #include <boost/unordered_set.hpp>
      23             : #include <algorithm>
      24             : 
      25             : #include <rtl/ustrbuf.hxx>
      26             : #include <com/sun/star/connection/XConnectionBroadcaster.hpp>
      27             : #include <com/sun/star/connection/ConnectionSetupException.hpp>
      28             : 
      29             : #include <cppuhelper/implbase2.hxx>
      30             : 
      31             : using namespace ::osl;
      32             : using namespace ::rtl;
      33             : using namespace ::cppu;
      34             : using namespace ::com::sun::star::uno;
      35             : using namespace ::com::sun::star::io;
      36             : using namespace ::com::sun::star::connection;
      37             : 
      38             : 
      39             : namespace io_acceptor {
      40             :     template<class T>
      41             :     struct ReferenceHash
      42             :     {
      43           0 :         size_t operator () (const ::com::sun::star::uno::Reference<T> & ref) const
      44             :         {
      45           0 :             return (size_t)ref.get();
      46             :         }
      47             :     };
      48             : 
      49             :     template<class T>
      50             :     struct ReferenceEqual
      51             :     {
      52           0 :         sal_Bool operator () (const ::com::sun::star::uno::Reference<T> & op1,
      53             :                               const ::com::sun::star::uno::Reference<T> & op2) const
      54             :         {
      55           0 :             return op1.get() == op2.get();
      56             :         }
      57             :     };
      58             : 
      59             : 
      60             :     typedef ::boost::unordered_set< ::com::sun::star::uno::Reference< ::com::sun::star::io::XStreamListener>,
      61             :                              ReferenceHash< ::com::sun::star::io::XStreamListener>,
      62             :                              ReferenceEqual< ::com::sun::star::io::XStreamListener> >
      63             :             XStreamListener_hash_set;
      64             : 
      65             : 
      66             :     class SocketConnection : public ::cppu::WeakImplHelper2<
      67             :         ::com::sun::star::connection::XConnection,
      68             :         ::com::sun::star::connection::XConnectionBroadcaster>
      69             : 
      70             :     {
      71             :     public:
      72             :         SocketConnection( const OUString & sConnectionDescription );
      73             :         ~SocketConnection();
      74             : 
      75             :         virtual sal_Int32 SAL_CALL read( ::com::sun::star::uno::Sequence< sal_Int8 >& aReadBytes,
      76             :                                          sal_Int32 nBytesToRead )
      77             :             throw(::com::sun::star::io::IOException,
      78             :                   ::com::sun::star::uno::RuntimeException);
      79             :         virtual void SAL_CALL write( const ::com::sun::star::uno::Sequence< sal_Int8 >& aData )
      80             :             throw(::com::sun::star::io::IOException,
      81             :                   ::com::sun::star::uno::RuntimeException);
      82             :         virtual void SAL_CALL flush(  ) throw(
      83             :             ::com::sun::star::io::IOException,
      84             :             ::com::sun::star::uno::RuntimeException);
      85             :         virtual void SAL_CALL close(  )
      86             :             throw(::com::sun::star::io::IOException,
      87             :                   ::com::sun::star::uno::RuntimeException);
      88             :         virtual OUString SAL_CALL getDescription(  )
      89             :             throw(::com::sun::star::uno::RuntimeException);
      90             : 
      91             :         // XConnectionBroadcaster
      92             :         virtual void SAL_CALL addStreamListener(const ::com::sun::star::uno::Reference< ::com::sun::star::io::XStreamListener>& aListener)
      93             :             throw(::com::sun::star::uno::RuntimeException);
      94             :         virtual void SAL_CALL removeStreamListener(const ::com::sun::star::uno::Reference< ::com::sun::star::io::XStreamListener>& aListener)
      95             :             throw(::com::sun::star::uno::RuntimeException);
      96             : 
      97             :     public:
      98             :         void completeConnectionString();
      99             : 
     100             :         ::osl::StreamSocket m_socket;
     101             :         ::osl::SocketAddr m_addr;
     102             :         oslInterlockedCount m_nStatus;
     103             :         OUString m_sDescription;
     104             : 
     105             :         ::osl::Mutex _mutex;
     106             :         sal_Bool     _started;
     107             :         sal_Bool     _closed;
     108             :         sal_Bool     _error;
     109             :         XStreamListener_hash_set _listeners;
     110             :     };
     111             : 
     112             :     template<class T>
     113           0 :     void notifyListeners(SocketConnection * pCon, sal_Bool * notified, T t)
     114             :     {
     115           0 :           XStreamListener_hash_set listeners;
     116             : 
     117             :         {
     118           0 :             ::osl::MutexGuard guard(pCon->_mutex);
     119           0 :             if(!*notified)
     120             :             {
     121           0 :                 *notified = sal_True;
     122           0 :                 listeners = pCon->_listeners;
     123             :             }
     124             :         }
     125             : 
     126           0 :         ::std::for_each(listeners.begin(), listeners.end(), t);
     127           0 :     }
     128             : 
     129           0 :     static void callStarted(Reference<XStreamListener> xStreamListener)
     130             :     {
     131           0 :         xStreamListener->started();
     132           0 :     }
     133             : 
     134             :     struct callError {
     135             :         const Any & any;
     136             : 
     137             :         callError(const Any & any);
     138             : 
     139             :         void operator () (Reference<XStreamListener> xStreamListener);
     140             :     };
     141             : 
     142           0 :     callError::callError(const Any & aAny)
     143           0 :         : any(aAny)
     144             :     {
     145           0 :     }
     146             : 
     147           0 :     void callError::operator () (Reference<XStreamListener> xStreamListener)
     148             :     {
     149           0 :         xStreamListener->error(any);
     150           0 :     }
     151             : 
     152           0 :     static void callClosed(Reference<XStreamListener> xStreamListener)
     153             :     {
     154           0 :         xStreamListener->closed();
     155           0 :     }
     156             : 
     157             : 
     158           0 :     SocketConnection::SocketConnection( const OUString &sConnectionDescription) :
     159             :         m_nStatus( 0 ),
     160             :         m_sDescription( sConnectionDescription ),
     161             :         _started(sal_False),
     162             :         _closed(sal_False),
     163           0 :         _error(sal_False)
     164             :     {
     165           0 :         g_moduleCount.modCnt.acquire( &g_moduleCount.modCnt );
     166             :         // make it unique
     167           0 :         m_sDescription += OUString( ",uniqueValue=" ) ;
     168             :         m_sDescription += OUString::valueOf(
     169             :             sal::static_int_cast< sal_Int64 >(
     170             :                 reinterpret_cast< sal_IntPtr >(&m_socket)),
     171           0 :             10 );
     172           0 :     }
     173             : 
     174           0 :     SocketConnection::~SocketConnection()
     175             :     {
     176           0 :         g_moduleCount.modCnt.release( &g_moduleCount.modCnt );
     177           0 :     }
     178             : 
     179           0 :     void SocketConnection::completeConnectionString()
     180             :     {
     181           0 :         OUStringBuffer buf( 256 );
     182           0 :         buf.appendAscii( ",peerPort=" );
     183           0 :         buf.append( (sal_Int32) m_socket.getPeerPort() );
     184           0 :         buf.appendAscii( ",peerHost=" );
     185           0 :         buf.append( m_socket.getPeerHost( ) );
     186             : 
     187           0 :         buf.appendAscii( ",localPort=" );
     188           0 :         buf.append( (sal_Int32) m_socket.getLocalPort() );
     189           0 :         buf.appendAscii( ",localHost=" );
     190           0 :         buf.append( m_socket.getLocalHost() );
     191             : 
     192           0 :         m_sDescription += buf.makeStringAndClear();
     193           0 :     }
     194             : 
     195           0 :     sal_Int32 SocketConnection::read( Sequence < sal_Int8 > & aReadBytes , sal_Int32 nBytesToRead )
     196             :             throw(::com::sun::star::io::IOException,
     197             :                   ::com::sun::star::uno::RuntimeException)
     198             :     {
     199           0 :         if( ! m_nStatus )
     200             :         {
     201           0 :             notifyListeners(this, &_started, callStarted);
     202             : 
     203           0 :             if( aReadBytes.getLength() != nBytesToRead )
     204             :             {
     205           0 :                 aReadBytes.realloc( nBytesToRead );
     206             :             }
     207             : 
     208           0 :             sal_Int32 i = 0;
     209           0 :             i = m_socket.read( aReadBytes.getArray()  , aReadBytes.getLength() );
     210             : 
     211           0 :             if(i != nBytesToRead)
     212             :             {
     213           0 :                 OUString message("acc_socket.cxx:SocketConnection::read: error - ");
     214           0 :                 message +=  m_socket.getErrorAsString();
     215             : 
     216           0 :                 IOException ioException(message, Reference<XInterface>(static_cast<XConnection *>(this)));
     217             : 
     218           0 :                 Any any;
     219           0 :                 any <<= ioException;
     220             : 
     221           0 :                 notifyListeners(this, &_error, callError(any));
     222             : 
     223           0 :                 throw ioException;
     224             :             }
     225             : 
     226           0 :             return i;
     227             :         }
     228             :         else
     229             :         {
     230           0 :             OUString message("acc_socket.cxx:SocketConnection::read: error - connection already closed");
     231             : 
     232           0 :             IOException ioException(message, Reference<XInterface>(static_cast<XConnection *>(this)));
     233             : 
     234           0 :             Any any;
     235           0 :             any <<= ioException;
     236             : 
     237           0 :             notifyListeners(this, &_error, callError(any));
     238             : 
     239           0 :             throw ioException;
     240             :         }
     241             :     }
     242             : 
     243           0 :     void SocketConnection::write( const Sequence < sal_Int8 > &seq )
     244             :             throw(::com::sun::star::io::IOException,
     245             :                   ::com::sun::star::uno::RuntimeException)
     246             :     {
     247           0 :         if( ! m_nStatus )
     248             :         {
     249           0 :             if( m_socket.write( seq.getConstArray() , seq.getLength() ) != seq.getLength() )
     250             :             {
     251           0 :                 OUString message("acc_socket.cxx:SocketConnection::write: error - ");
     252           0 :                 message += m_socket.getErrorAsString();
     253             : 
     254           0 :                 IOException ioException(message, Reference<XInterface>(static_cast<XConnection *>(this)));
     255             : 
     256           0 :                 Any any;
     257           0 :                 any <<= ioException;
     258             : 
     259           0 :                 notifyListeners(this, &_error, callError(any));
     260             : 
     261           0 :                 throw ioException;
     262             :             }
     263             :         }
     264             :         else
     265             :         {
     266           0 :             OUString message("acc_socket.cxx:SocketConnection::write: error - connection already closed");
     267             : 
     268           0 :             IOException ioException(message, Reference<XInterface>(static_cast<XConnection *>(this)));
     269             : 
     270           0 :             Any any;
     271           0 :             any <<= ioException;
     272             : 
     273           0 :             notifyListeners(this, &_error, callError(any));
     274             : 
     275           0 :             throw ioException;
     276             :         }
     277           0 :     }
     278             : 
     279           0 :     void SocketConnection::flush( )
     280             :             throw(::com::sun::star::io::IOException,
     281             :                   ::com::sun::star::uno::RuntimeException)
     282             :     {
     283             : 
     284           0 :     }
     285             : 
     286           0 :     void SocketConnection::close()
     287             :             throw(::com::sun::star::io::IOException,
     288             :                   ::com::sun::star::uno::RuntimeException)
     289             :     {
     290             :         // enshure close is called only once
     291           0 :         if(  1 == osl_atomic_increment( (&m_nStatus) ) )
     292             :         {
     293           0 :             m_socket.shutdown();
     294           0 :             notifyListeners(this, &_closed, callClosed);
     295             :         }
     296           0 :     }
     297             : 
     298           0 :     OUString SocketConnection::getDescription()
     299             :             throw( ::com::sun::star::uno::RuntimeException)
     300             :     {
     301           0 :         return m_sDescription;
     302             :     }
     303             : 
     304             : 
     305             :     // XConnectionBroadcaster
     306           0 :     void SAL_CALL SocketConnection::addStreamListener(const Reference<XStreamListener> & aListener) throw(RuntimeException)
     307             :     {
     308           0 :         MutexGuard guard(_mutex);
     309             : 
     310           0 :         _listeners.insert(aListener);
     311           0 :     }
     312             : 
     313           0 :     void SAL_CALL SocketConnection::removeStreamListener(const Reference<XStreamListener> & aListener) throw(RuntimeException)
     314             :     {
     315           0 :         MutexGuard guard(_mutex);
     316             : 
     317           0 :         _listeners.erase(aListener);
     318           0 :     }
     319             : 
     320           0 :     SocketAcceptor::SocketAcceptor( const OUString &sSocketName,
     321             :                                     sal_uInt16 nPort,
     322             :                                     sal_Bool bTcpNoDelay,
     323             :                                     const OUString &sConnectionDescription) :
     324             :         m_sSocketName( sSocketName ),
     325             :         m_sConnectionDescription( sConnectionDescription ),
     326             :         m_nPort( nPort ),
     327             :         m_bTcpNoDelay( bTcpNoDelay ),
     328           0 :         m_bClosed( sal_False )
     329             :     {
     330           0 :     }
     331             : 
     332             : 
     333           0 :     void SocketAcceptor::init()
     334             :     {
     335           0 :         if( ! m_addr.setPort( m_nPort ) )
     336             :         {
     337           0 :             OUStringBuffer message( 128 );
     338           0 :             message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - invalid tcp/ip port " );
     339           0 :             message.append( (sal_Int32) m_nPort );
     340             :             throw ConnectionSetupException(
     341           0 :                 message.makeStringAndClear() , Reference< XInterface> () );
     342             :         }
     343           0 :         if( ! m_addr.setHostname( m_sSocketName.pData ) )
     344             :         {
     345           0 :             OUStringBuffer message( 128 );
     346           0 :             message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - invalid host " );
     347           0 :             message.append( m_sSocketName );
     348             :             throw ConnectionSetupException(
     349           0 :                 message.makeStringAndClear(), Reference< XInterface > () );
     350             :         }
     351           0 :         m_socket.setOption( osl_Socket_OptionReuseAddr, 1);
     352             : 
     353           0 :         if(! m_socket.bind(m_addr) )
     354             :         {
     355           0 :             OUStringBuffer message( 128 );
     356           0 :             message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - couldn't bind on " );
     357           0 :             message.append( m_sSocketName ).appendAscii( ":" ).append((sal_Int32)m_nPort);
     358             :             throw ConnectionSetupException(
     359             :                 message.makeStringAndClear(),
     360           0 :                 Reference<XInterface>());
     361             :         }
     362             : 
     363           0 :         if(! m_socket.listen() )
     364             :         {
     365           0 :             OUStringBuffer message( 128 );
     366           0 :             message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - can't listen on " );
     367           0 :             message.append( m_sSocketName ).appendAscii( ":" ).append( (sal_Int32) m_nPort);
     368           0 :             throw ConnectionSetupException( message.makeStringAndClear(),Reference<XInterface>() );
     369             :         }
     370           0 :     }
     371             : 
     372           0 :     Reference< XConnection > SocketAcceptor::accept( )
     373             :     {
     374           0 :         SocketConnection *pConn = new SocketConnection( m_sConnectionDescription );
     375             : 
     376           0 :         if( m_socket.acceptConnection( pConn->m_socket )!= osl_Socket_Ok )
     377             :         {
     378             :             // stopAccepting was called
     379           0 :             delete pConn;
     380           0 :             return Reference < XConnection > ();
     381             :         }
     382           0 :         if( m_bClosed )
     383             :         {
     384           0 :             delete pConn;
     385           0 :             return Reference < XConnection > ();
     386             :         }
     387             : 
     388           0 :         pConn->completeConnectionString();
     389           0 :         if( m_bTcpNoDelay )
     390             :         {
     391           0 :             sal_Int32 nTcpNoDelay = sal_True;
     392             :             pConn->m_socket.setOption( osl_Socket_OptionTcpNoDelay , &nTcpNoDelay,
     393           0 :                                        sizeof( nTcpNoDelay ) , osl_Socket_LevelTcp );
     394             :         }
     395             : 
     396           0 :         return Reference < XConnection > ( (XConnection * ) pConn );
     397             :     }
     398             : 
     399           0 :     void SocketAcceptor::stopAccepting()
     400             :     {
     401           0 :         m_bClosed = sal_True;
     402           0 :         m_socket.close();
     403           0 :     }
     404             : }
     405             : 
     406             : 
     407             : /* vim:set shiftwidth=4 softtabstop=4 expandtab: */

Generated by: LCOV version 1.10