cabal_tcpsocket.cpp

/*
   cabal_tcpsocket.cpp
   Basic TCP socket object for cabal

   $Id: cabal_tcpsocket.cpp,v 1.2 2004/04/10 15:28:16 jonnymind Exp $
---------------------------------------------
   Begin      : Thu, 11 Mar 2004 03:05:18 +0100
   Author     : Giancarlo Niccolai

   Last modified because:

*/

#include <cabal_tcpsocket.h>

namespace Cabal {

bool TCPSocket::connect()
{
   
   if( ! Net::connect( m_skCom, m_remote.address() ) && ! Net::inProgress() )
   {
      m_osError = Net::errorCode();
      return false;
   }


   if(! selectFor( true ) )  //select for write
   {
      /* Timed out */
      m_osError = 0;
      m_opType = OT_CONNECTING;
      m_bTout = true;
      return false;
   }

   //Connection has been completed with a failure or a success
   
   int iErrval;
   Net::getSocketOption( m_skCom, SO_ERROR, iErrval ); 
   
   if ( iErrval ) {
      m_osError = iErrval;
      return false;
   }

   m_bTout = false;
   m_osError = 0;
   return true;
}

void TCPSocket::buildSocket( const Address *local, const Address *remote ) 
   throw( InitError )
{
   m_skCom = Net::createTCPSocket();
   if ( ! Net::validSocket( m_skCom ) )
      throw InitError( "Low level socket not created", Net::errorCode() );

   Net::setNonBlocking( m_skCom );
   Net::setTCPDefaults( m_skCom );

   // if we have a local address, we must bind.
   if ( local != 0 )
   {
      m_local = *local;
      if ( ! Net::bind( m_skCom, m_local.address() ) )
      {
         Net::closeSocket( m_skCom );
         Net::invalidate( m_skCom );
         throw InitError( "Low level socket can't bind", Net::errorCode() );
      }
   }

   // if we have a remote address, we must connect
   if ( remote != 0 )
   {
      m_remote = *remote;
   }
}

std::string TCPSocket::recvLine( int maxCount )
{
   int recvd = 0;
   int r;
   int pos = 0;
   std::string ret = "";
   char buf[ 512 ];

   while ( (maxCount == 0 || recvd < maxCount ) )
   {
      r = recv( buf + pos, 1 );
      if ( r < 0 ) return "";
      if ( r == 0 ) {
         if ( pos > 0 ) {
            buf[pos] = '\0';
            ret+=buf;
         }

         return ret;
      }

      if ( pos >=1 ) {
         if ( buf[pos-1] == '\r' && buf[pos] == '\n' ) {
            buf[pos-1] = '\0';
            pos = 0;
            ret += buf;
            break;
         }
      }

      pos ++;
      recvd++;

      if ( (pos == 510 && buf[509] != '\r') || pos == 511) {
         buf[pos] = '\0';
         ret+=buf;
         pos = 0;
      }
   }

   if ( pos > 0 ) {
      buf[pos] = '\0';
      ret+=buf;
   }

   return ret;
}


int TCPSocket::sendAll( const void *buffer, const int len )
{
   int sent;
   int sentTotal =0;

   while ( sentTotal < len ) {
      sent = rawSend( static_cast<const void *>(static_cast<const char *>(buffer) + sentTotal), len - sentTotal );
      if ( sent < 0 || hasTimedOut() ) {
         m_opType = OT_SENDINGALL;
         m_buffer = buffer;
         m_buflen = len;
         m_bufdone = sentTotal;
         return sentTotal;
      }
      sentTotal += sent;
   }
   return sentTotal;
}


int TCPSocket::recvAll( void *buffer, const int len )
{
   int recvd;
   int recvTotal = 0;

   while ( recvTotal < len ) {
      recvd = recv( static_cast<char *>(buffer) + recvTotal, len - recvTotal );
      if ( recvd < 0 || hasTimedOut() ) {
         m_opType = OT_RECEIVINGALL;
         m_buffer = buffer;
         m_buflen = len;
         m_bufdone = recvTotal;
         return recvTotal;
      }
      recvTotal += recvd;
   }
   return recvTotal;
}


bool TCPSocket::continueOp()
{
   switch( m_opType ) {
      case OT_CONNECTING:
         m_osError = 0;
         if(! selectFor( true ) )  //select for write
         {
            /* Timed out */
            m_bTout = true;
            return false;
         }

         m_bTout = false;
      return true;
      // others not yet implemented
   }

   return false;
}
 
}

/* end of cabal_tcpsocket.h */

---