TCPTransmitterSocket.cc

#include "otsdaq/NetworkUtilities/TCPTransmitterSocket.h"
#include <string.h>
#include <sys/socket.h>
#include <unistd.h>
#include <stdexcept>
#include "otsdaq/NetworkUtilities/TCPPacket.h"
//#include <iostream>

using namespace ots;

//==============================================================================
TCPTransmitterSocket::TCPTransmitterSocket(int socketId) : TCPSocket(socketId) {}

//==============================================================================
TCPTransmitterSocket::~TCPTransmitterSocket(void) {}

//==============================================================================
void TCPTransmitterSocket::sendPacket(char const* buffer, std::size_t size) { send(TCPPacket::encode(buffer, size)); }

//==============================================================================
void TCPTransmitterSocket::sendPacket(const std::string& buffer) { send(TCPPacket::encode(buffer)); }

//==============================================================================
void TCPTransmitterSocket::send(char const* buffer, std::size_t size)
{
    std::size_t sentBytes = ::send(getSocketId(), buffer, size, MSG_NOSIGNAL);
    if(sentBytes == static_cast<std::size_t>(-1))
    {
        switch(errno)
        {
        // case EINVAL:
        // case EBADF:
        // case ECONNRESET:
        // case ENXIO:
        case EPIPE:
        {
            // Fatal error. Programming bug
            throw std::runtime_error(std::string("Write: critical error: ") + strerror(errno));
        }
        // case EDQUOT:
        // case EFBIG:
        // case EIO:
        // case ENETDOWN:
        // case ENETUNREACH:
        case ENOSPC:
        {
            // Resource acquisition failure or device error
            throw std::runtime_error(std::string("Write: resource failure: ") + strerror(errno));
        }
        case EINTR:
            // TODO: Check for user interrupt flags.
            //       Beyond the scope of this project
            //       so continue normal operations.
        case EAGAIN:
        {
            // Temporary error.
            throw std::runtime_error(std::string("Write: temporary error: ") + strerror(errno));
        }
        default:
        {
            throw std::runtime_error(std::string("Write: returned -1: ") + strerror(errno));
        }
        }
    }
}

//==============================================================================
void TCPTransmitterSocket::send(const std::string& buffer)
{
    if(buffer.size() == 0)
    {
        std::cout << __PRETTY_FUNCTION__ << "I am sorry but I won't send an empty packet!" << std::endl;
        return;
    }
    send(&buffer.at(0), buffer.size());
}

//==============================================================================
void TCPTransmitterSocket::send(const std::vector<char>& buffer)
{
    if(buffer.size() == 0)
    {
        std::cout << __PRETTY_FUNCTION__ << "I am sorry but I won't send an empty packet!" << std::endl;
        return;
    }
    send(&buffer.at(0), buffer.size());
}

//==============================================================================
void TCPTransmitterSocket::send(const std::vector<uint16_t>& buffer)
{
    if(buffer.size() == 0)
    {
        std::cout << __PRETTY_FUNCTION__ << "I am sorry but I won't send an empty packet!" << std::endl;
        return;
    }
    send((const char*)&buffer.at(0), buffer.size());
}

//==============================================================================
void TCPTransmitterSocket::setSendTimeout(unsigned int timeoutSeconds, unsigned int timeoutMicroSeconds)
{
    struct timeval tv;
    tv.tv_sec  = timeoutSeconds;
    tv.tv_usec = timeoutMicroSeconds;
    setsockopt(getSocketId(), SOL_SOCKET, SO_SNDTIMEO, (const char*)&tv, sizeof tv);
}