socket.cpp

Go to the documentation of this file.

/*
 * This file is part of the dodo library (https://github.com/jmspit/dodo).
 * Copyright (c) 2019 Jan-Marten Spit.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
 
/**
 * @file socket.cpp
 * Implements the dodo::network::Socket class.
 */
 
#include <cstring>
#include <iostream>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
 
#include "network/socket.hpp"
#include "common/exception.hpp"
 
namespace dodo::network {
 
  Socket Socket::SocketInvalid = Socket();
 
  common::SystemError Socket::send( const void* buf, ssize_t len, bool more ) {
    int flags = MSG_NOSIGNAL;
    if ( more ) flags = flags | MSG_MORE;
    ssize_t left = len;
    ssize_t rc = 0;
    const char* cp = (const char*)buf;
    while ( left > 0 ) {
      rc = ::send( socket_, cp, left, flags );
      if ( rc < 0 ) break;
      cp += rc;
      left -= rc;
    }
    if ( rc < 0 ) {
      switch ( errno ) {
        case common::SystemError::ecEAGAIN :
        case common::SystemError::ecECONNRESET :
          return errno;
        default: throw_SystemExceptionObject( "Socket::send failed", errno, this );
      };
    } else if ( left ) throw_SystemExceptionObject( "Socket::send not all bytes send", errno, this );
    else return common::SystemError::ecOK;
  }
 
  common::SystemError Socket::sendTo( const Address& address, const void* buf, ssize_t len ) {
    int flags = 0;
    ssize_t left = len;
    ssize_t rc = 0;
    const char* cp = (const char*)buf;
    while ( left > 0 ) {
      rc = ::sendto( socket_, cp, left, flags, (const sockaddr*)&(address.addr_), sizeof(address.addr_) );
      if ( rc < 0 ) break;
      cp += rc;
      left -= rc;
    }
    if ( rc < 0 ) {
      switch ( errno ) {
        case common::SystemError::ecEAGAIN :
        case common::SystemError::ecECONNRESET :
          return errno;
        default: throw_SystemExceptionObject( "Socket::send failed", errno, this );
      };
    } else if ( left ) throw_SystemExceptionObject( "Socket::send not all bytes send", errno, this );
    else return common::SystemError::ecOK;
  }
 
  common::SystemError Socket::receive( void* buf, ssize_t request, ssize_t &received ) {
    received = 0;
    ssize_t rc = ::recv( socket_, buf, request, 0 );
    if ( rc < 0 ) {
      switch ( errno ) {
        case common::SystemError::ecEAGAIN :
        case common::SystemError::ecECONNREFUSED :
        case common::SystemError::ecENOTCONN :
          return errno;
        default: throw_SystemExceptionObject( "Socket::receive failed", errno, this );
      };
    } else {
      received = rc;
      return common::SystemError::ecOK;
    }
  }
 
  Socket* Socket::accept() {
    Socket* ret = new Socket();
    *ret = Socket::SocketInvalid;
    int rc = ::accept( socket_, nullptr, nullptr );
    if ( rc == -1 ) {
      if ( errno != EAGAIN && errno != EWOULDBLOCK )
        throw_SystemExceptionObject( "Socket::accept failed", errno, this );
      else
        *ret = rc;
    } else *ret = rc;
    return ret;
  }
 
}