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 <algorithm>
23 : #include <unordered_set>
24 :
25 : #include <osl/mutex.hxx>
26 : #include <rtl/ustrbuf.hxx>
27 : #include <com/sun/star/connection/XConnectionBroadcaster.hpp>
28 : #include <com/sun/star/connection/ConnectionSetupException.hpp>
29 :
30 : #include <cppuhelper/implbase2.hxx>
31 :
32 : using namespace ::osl;
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 reinterpret_cast<size_t>(ref.get());
46 : }
47 : };
48 :
49 : template<class T>
50 : struct ReferenceEqual
51 : {
52 0 : 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 std::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 : virtual ~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, std::exception) SAL_OVERRIDE;
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, std::exception) SAL_OVERRIDE;
82 : virtual void SAL_CALL flush( ) throw(
83 : ::com::sun::star::io::IOException,
84 : ::com::sun::star::uno::RuntimeException, std::exception) SAL_OVERRIDE;
85 : virtual void SAL_CALL close( )
86 : throw(::com::sun::star::io::IOException,
87 : ::com::sun::star::uno::RuntimeException, std::exception) SAL_OVERRIDE;
88 : virtual OUString SAL_CALL getDescription( )
89 : throw(::com::sun::star::uno::RuntimeException, std::exception) SAL_OVERRIDE;
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, std::exception) SAL_OVERRIDE;
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, std::exception) SAL_OVERRIDE;
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 : bool _started;
107 : bool _closed;
108 : bool _error;
109 : XStreamListener_hash_set _listeners;
110 : };
111 :
112 : template<class T>
113 0 : void notifyListeners(SocketConnection * pCon, 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 = true;
122 0 : listeners = pCon->_listeners;
123 0 : }
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(false),
162 : _closed(false),
163 0 : _error(false)
164 : {
165 : // make it unique
166 0 : m_sDescription += ",uniqueValue=" ;
167 0 : m_sDescription += OUString::number(
168 : sal::static_int_cast< sal_Int64 >(
169 : reinterpret_cast< sal_IntPtr >(&m_socket)),
170 0 : 10 );
171 0 : }
172 :
173 0 : SocketConnection::~SocketConnection()
174 : {
175 0 : }
176 :
177 0 : void SocketConnection::completeConnectionString()
178 : {
179 0 : OUStringBuffer buf( 256 );
180 0 : buf.appendAscii( ",peerPort=" );
181 0 : buf.append( (sal_Int32) m_socket.getPeerPort() );
182 0 : buf.appendAscii( ",peerHost=" );
183 0 : buf.append( m_socket.getPeerHost( ) );
184 :
185 0 : buf.appendAscii( ",localPort=" );
186 0 : buf.append( (sal_Int32) m_socket.getLocalPort() );
187 0 : buf.appendAscii( ",localHost=" );
188 0 : buf.append( m_socket.getLocalHost() );
189 :
190 0 : m_sDescription += buf.makeStringAndClear();
191 0 : }
192 :
193 0 : sal_Int32 SocketConnection::read( Sequence < sal_Int8 > & aReadBytes , sal_Int32 nBytesToRead )
194 : throw(::com::sun::star::io::IOException,
195 : ::com::sun::star::uno::RuntimeException, std::exception)
196 : {
197 0 : if( ! m_nStatus )
198 : {
199 0 : notifyListeners(this, &_started, callStarted);
200 :
201 0 : if( aReadBytes.getLength() != nBytesToRead )
202 : {
203 0 : aReadBytes.realloc( nBytesToRead );
204 : }
205 :
206 0 : sal_Int32 i = 0;
207 0 : i = m_socket.read( aReadBytes.getArray() , aReadBytes.getLength() );
208 :
209 0 : if(i != nBytesToRead)
210 : {
211 0 : OUString message("acc_socket.cxx:SocketConnection::read: error - ");
212 0 : message += m_socket.getErrorAsString();
213 :
214 0 : IOException ioException(message, static_cast<XConnection *>(this));
215 :
216 0 : Any any;
217 0 : any <<= ioException;
218 :
219 0 : notifyListeners(this, &_error, callError(any));
220 :
221 0 : throw ioException;
222 : }
223 :
224 0 : return i;
225 : }
226 : else
227 : {
228 0 : OUString message("acc_socket.cxx:SocketConnection::read: error - connection already closed");
229 :
230 0 : IOException ioException(message, static_cast<XConnection *>(this));
231 :
232 0 : Any any;
233 0 : any <<= ioException;
234 :
235 0 : notifyListeners(this, &_error, callError(any));
236 :
237 0 : throw ioException;
238 : }
239 : }
240 :
241 0 : void SocketConnection::write( const Sequence < sal_Int8 > &seq )
242 : throw(::com::sun::star::io::IOException,
243 : ::com::sun::star::uno::RuntimeException, std::exception)
244 : {
245 0 : if( ! m_nStatus )
246 : {
247 0 : if( m_socket.write( seq.getConstArray() , seq.getLength() ) != seq.getLength() )
248 : {
249 0 : OUString message("acc_socket.cxx:SocketConnection::write: error - ");
250 0 : message += m_socket.getErrorAsString();
251 :
252 0 : IOException ioException(message, static_cast<XConnection *>(this));
253 :
254 0 : Any any;
255 0 : any <<= ioException;
256 :
257 0 : notifyListeners(this, &_error, callError(any));
258 :
259 0 : throw ioException;
260 : }
261 : }
262 : else
263 : {
264 0 : OUString message("acc_socket.cxx:SocketConnection::write: error - connection already closed");
265 :
266 0 : IOException ioException(message, static_cast<XConnection *>(this));
267 :
268 0 : Any any;
269 0 : any <<= ioException;
270 :
271 0 : notifyListeners(this, &_error, callError(any));
272 :
273 0 : throw ioException;
274 : }
275 0 : }
276 :
277 0 : void SocketConnection::flush( )
278 : throw(::com::sun::star::io::IOException,
279 : ::com::sun::star::uno::RuntimeException, std::exception)
280 : {
281 :
282 0 : }
283 :
284 0 : void SocketConnection::close()
285 : throw(::com::sun::star::io::IOException,
286 : ::com::sun::star::uno::RuntimeException, std::exception)
287 : {
288 : // ensure close is called only once
289 0 : if( 1 == osl_atomic_increment( (&m_nStatus) ) )
290 : {
291 0 : m_socket.shutdown();
292 0 : notifyListeners(this, &_closed, callClosed);
293 : }
294 0 : }
295 :
296 0 : OUString SocketConnection::getDescription()
297 : throw( ::com::sun::star::uno::RuntimeException, std::exception)
298 : {
299 0 : return m_sDescription;
300 : }
301 :
302 :
303 : // XConnectionBroadcaster
304 0 : void SAL_CALL SocketConnection::addStreamListener(const Reference<XStreamListener> & aListener) throw(RuntimeException, std::exception)
305 : {
306 0 : MutexGuard guard(_mutex);
307 :
308 0 : _listeners.insert(aListener);
309 0 : }
310 :
311 0 : void SAL_CALL SocketConnection::removeStreamListener(const Reference<XStreamListener> & aListener) throw(RuntimeException, std::exception)
312 : {
313 0 : MutexGuard guard(_mutex);
314 :
315 0 : _listeners.erase(aListener);
316 0 : }
317 :
318 0 : SocketAcceptor::SocketAcceptor( const OUString &sSocketName,
319 : sal_uInt16 nPort,
320 : bool bTcpNoDelay,
321 : const OUString &sConnectionDescription) :
322 : m_sSocketName( sSocketName ),
323 : m_sConnectionDescription( sConnectionDescription ),
324 : m_nPort( nPort ),
325 : m_bTcpNoDelay( bTcpNoDelay ),
326 0 : m_bClosed( false )
327 : {
328 0 : }
329 :
330 :
331 0 : void SocketAcceptor::init()
332 : {
333 0 : if( ! m_addr.setPort( m_nPort ) )
334 : {
335 0 : OUStringBuffer message( 128 );
336 0 : message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - invalid tcp/ip port " );
337 0 : message.append( (sal_Int32) m_nPort );
338 0 : throw ConnectionSetupException( message.makeStringAndClear() );
339 : }
340 0 : if( ! m_addr.setHostname( m_sSocketName.pData ) )
341 : {
342 0 : OUStringBuffer message( 128 );
343 0 : message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - invalid host " );
344 0 : message.append( m_sSocketName );
345 0 : throw ConnectionSetupException( message.makeStringAndClear() );
346 : }
347 0 : m_socket.setOption( osl_Socket_OptionReuseAddr, 1);
348 :
349 0 : if(! m_socket.bind(m_addr) )
350 : {
351 0 : OUStringBuffer message( 128 );
352 0 : message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - couldn't bind on " );
353 0 : message.append( m_sSocketName ).appendAscii( ":" ).append((sal_Int32)m_nPort);
354 0 : throw ConnectionSetupException( message.makeStringAndClear() );
355 : }
356 :
357 0 : if(! m_socket.listen() )
358 : {
359 0 : OUStringBuffer message( 128 );
360 0 : message.appendAscii( "acc_socket.cxx:SocketAcceptor::init - error - can't listen on " );
361 0 : message.append( m_sSocketName ).appendAscii( ":" ).append( (sal_Int32) m_nPort);
362 0 : throw ConnectionSetupException( message.makeStringAndClear() );
363 : }
364 0 : }
365 :
366 0 : Reference< XConnection > SocketAcceptor::accept( )
367 : {
368 0 : SocketConnection *pConn = new SocketConnection( m_sConnectionDescription );
369 :
370 0 : if( m_socket.acceptConnection( pConn->m_socket )!= osl_Socket_Ok )
371 : {
372 : // stopAccepting was called
373 0 : delete pConn;
374 0 : return Reference < XConnection > ();
375 : }
376 0 : if( m_bClosed )
377 : {
378 0 : delete pConn;
379 0 : return Reference < XConnection > ();
380 : }
381 :
382 0 : pConn->completeConnectionString();
383 0 : OUString remoteHostname = pConn->m_addr.getHostname();
384 : // we enable tcpNoDelay for loopback connections because
385 : // it can make a significant speed difference on linux boxes.
386 0 : if( m_bTcpNoDelay || remoteHostname == "localhost" ||
387 0 : remoteHostname.startsWith("127.0.0.") )
388 : {
389 0 : sal_Int32 nTcpNoDelay = sal_Int32(true);
390 : pConn->m_socket.setOption( osl_Socket_OptionTcpNoDelay , &nTcpNoDelay,
391 0 : sizeof( nTcpNoDelay ) , osl_Socket_LevelTcp );
392 : }
393 :
394 0 : return Reference < XConnection > ( static_cast<XConnection *>(pConn) );
395 : }
396 :
397 0 : void SocketAcceptor::stopAccepting()
398 : {
399 0 : m_bClosed = true;
400 0 : m_socket.close();
401 0 : }
402 : }
403 :
404 :
405 : /* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|
