Source/Base/Network/Socket/OSGPointSockPipeline.cpp

   1 /*---------------------------------------------------------------------------*\
   2  *                                OpenSG                                     *
   3  *                                                                           *
   4  *                                                                           *
   5  *             Copyright (C) 2000-2002 by the OpenSG Forum                   *
   6  *                                                                           *
   7  *                            www.opensg.org                                 *
   8  *                                                                           *
   9  *   contact: dirk@opensg.org, gerrit.voss@vossg.org, jbehr@zgdv.de          *
  10  *                                                                           *
  11 \*---------------------------------------------------------------------------*/
  12 /*---------------------------------------------------------------------------*\
  13  *                                License                                    *
  14  *                                                                           *
  15  * This library is free software; you can redistribute it and/or modify it   *
  16  * under the terms of the GNU Library General Public License as published    *
  17  * by the Free Software Foundation, version 2.                               *
  18  *                                                                           *
  19  * This library is distributed in the hope that it will be useful, but       *
  20  * WITHOUT ANY WARRANTY; without even the implied warranty of                *
  21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU         *
  22  * Library General Public License for more details.                          *
  23  *                                                                           *
  24  * You should have received a copy of the GNU Library General Public         *
  25  * License along with this library; if not, write to the Free Software       *
  26  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.                 *
  27  *                                                                           *
  28 \*---------------------------------------------------------------------------*/
  29 /*---------------------------------------------------------------------------*\
  30  *                                Changes                                    *
  31  *                                                                           *
  32  *                                                                           *
  33  *                                                                           *
  34  *                                                                           *
  35  *                                                                           *
  36  *                                                                           *
  37 \*---------------------------------------------------------------------------*/
  38
  39 //---------------------------------------------------------------------------
  40 //  Includes
  41 //---------------------------------------------------------------------------
  42 #include <cstdlib>
  43 #include <cstdio>
  44 #include <cassert>
  45
  46 #include "OSGConfig.h"
  47 #include "OSGLog.h"
  48 #include "OSGBaseThread.h"
  49 #include "OSGSocketSelection.h"
  50 #include "OSGBinaryMessage.h"
  51 #include "OSGPointSockPipeline.h"
  52 #include "OSGGroupSockPipeline.h"
  53 #include "OSGConnectionType.h"
  54 #include "OSGBinaryMessage.h"
  55
  56 OSG_USING_NAMESPACE
  57
  58 /** \class OSG::PointSockPipeline
  59  **/
  60
  61 /*-------------------------------------------------------------------------*/
  62 /*                            constructor destructor                       */
  63
  64 /*! Constructor
  65  */
  66
  67 PointSockPipeline::PointSockPipeline():
  68      Inherited  (     ),
  69     _next       (     ),
  70     _prev       (     ),
  71     _last       (false),
  72     _initialized(false)
  73 {
  74     _prev.open();
  75     _next.open();
  76 }
  77
  78 /*! Destructor
  79  */
  80 PointSockPipeline::~PointSockPipeline(void)
  81 {
  82     _prev.close();
  83     _next.close();
  84 }
  85
  86 /*! get connection type
  87  */
  88 const ConnectionType *PointSockPipeline::getType()
  89 {
  90     return &_type;
  91 }
  92
  93 /*-------------------------------------------------------------------------*/
  94 /*                            connection                                   */
  95
  96 /*! connect to the given group. If timeout is reached, -1 is
  97     returned
  98  */
  99 Connection::Channel PointSockPipeline::connectGroup(
 100     const std::string &address,
 101     Time               timeout)
 102 {
 103     Channel channel = Inherited::connectGroup(address,timeout);
 104     return channel;
 105 }
 106
 107 /*! disconnect the given channel
 108  */
 109 void PointSockPipeline::disconnect(void)
 110 {
 111     _socket.close();
 112 }
 113
 114 /*! accept an icomming grop connection. If timeout is reached,
 115     -1 is returned. If timeout is -1 then wait without timeout
 116  */
 117 Connection::Channel PointSockPipeline::acceptGroup(Time timeout)
 118 {
 119     Channel channel = Inherited::acceptGroup(timeout);
 120     return channel;
 121 }
 122
 123 /*-------------------------------------------------------------------------*/
 124 /*                              channel handling                           */
 125
 126 /*! select the next channel for reading. If timeout is not -1
 127     then -1 is returned if timeout is reached
 128 */
 129 Connection::Channel PointSockPipeline::selectChannel(Time timeout)
 130     OSG_THROW (ReadError)
 131 {
 132     if(!_initialized)
 133         initialize();
 134     try
 135     {
 136         if(_prev.waitReadable(timeout))
 137             return 0;
 138     }
 139     catch(SocketError &e)
 140     {
 141         throw ReadError(e.what());
 142     }
 143     return -1;
 144 }
 145
 146 /*-------------------------- create ---------------------------------------*/
 147
 148 /** \brief create conneciton
 149  */
 150
 151 PointConnection *PointSockPipeline::create(void)
 152 {
 153     return new PointSockPipeline();
 154 }
 155
 156 /*-------------------------------------------------------------------------*/
 157 /*                              read write                                 */
 158
 159 /** Read data into given memory
 160  *
 161  * Read data form the current read socket. The read socket is that
 162  * socket, that was selectet in selectChannel.
 163  *
 164  **/
 165
 166 void PointSockPipeline::read(MemoryHandle mem,UInt32 size)
 167 {
 168     int len;
 169
 170     if(!_initialized)
 171         initialize();
 172     // read data
 173     len=_prev.recv(mem,size);
 174     if(len==0)
 175     {
 176         throw ReadError("read got 0 bytes!");
 177     }
 178     // send to next in chain
 179     if(!_last)
 180         _next.send(mem,size);
 181 }
 182
 183 /** Read next data block
 184  *
 185  * The stream connection uses only BinaryDataHandler buffer. If more
 186  * then one buffer is present, then this methode must be changed!
 187  *
 188  */
 189
 190 void PointSockPipeline::readBuffer() OSG_THROW (ReadError)
 191 {
 192     int size;
 193     int len;
 194
 195     if(!_initialized)
 196         initialize();
 197
 198     // read buffer header
 199     len=_prev.recv(&_socketReadBuffer[0],sizeof(SocketBufferHeader));
 200
 201     if(len==0)
 202         throw ReadError("peek got 0 bytes!");
 203     // read remaining data
 204     size=osgNetToHost<UInt32>((reinterpret_cast<SocketBufferHeader*>(&_socketReadBuffer[0]))->size);
 205     len=_prev.recv(&_socketReadBuffer[sizeof(SocketBufferHeader)],
 206                    size);
 207
 208     if(len==0)
 209         throw ReadError("read got 0 bytes!");
 210     readBufBegin()->setDataSize(size);
 211     // send to next in chain
 212     if(!_last)
 213         _next.send(&_socketReadBuffer[0],
 214                    sizeof(SocketBufferHeader)+size);
 215 }
 216
 217 /*-------------------------------------------------------------------------*/
 218 /*                              private helpers                            */
 219
 220 /*! initialize pipeline
 221  */
 222 void PointSockPipeline::initialize(void)
 223 {
 224     BinaryMessage message;
 225     StreamSocket  sock;
 226     UInt32        nextPort;
 227     std::string   nextHost;
 228     UInt32        len;
 229     char          localhost[256];
 230     std::string   interf;
 231
 232     // get local host name
 233     osgGetHostname(localhost,255);
 234     if(!getInterface().empty())
 235         interf = getInterface();
 236     else
 237         interf = localhost;
 238
 239     sock.open();
 240     sock.bind(SocketAddress(interf.c_str(),0));
 241     sock.listen();
 242
 243     // send my own address
 244     message.putString(interf);
 245     message.putUInt32(sock.getAddress().getPort());
 246     _socket.send(message);
 247     // accept prev
 248     _prev = sock.accept();
 249     sock.close();
 250
 251     len = _socket.recv(message);
 252     if(len == 0)
 253         throw ReadError("Channel closed\n");
 254     _last = message.getUInt32() != 0;
 255
 256     if(!_last)
 257     {
 258         nextHost = message.getString();
 259         nextPort = message.getUInt32();
 260         for(;;)
 261         {
 262             try
 263             {
 264                 _next.connect(SocketAddress(nextHost.c_str(),
 265                                             nextPort));
 266                 break;
 267             }
 268             catch(...)
 269             {
 270             }
 271         }
 272     }
 273
 274     _initialized = true;
 275 }
 276
 277 /*-------------------------------------------------------------------------*/
 278 /*                              static type                                */
 279
 280 ConnectionType PointSockPipeline::_type(
 281     &PointSockPipeline::create,
 282     "SockPipeline");