TCPSocket.cpp

Go to the documentation of this file.

#include "TCPSocket.h"
#include "TempFailure.h"
 
#include <sys/socket.h>
#include <netinet/in.h>
#include <poll.h>
 
using namespace NET;
 
TCPSocket::TCPSocket()
: InternetSocket( STREAM, IPPROTO_TCP)
{}
 
TCPSocket::TCPSocket( Handle handle)
: InternetSocket( handle.release() )
{}
 
int TCPSocket::sendAll( const void* buffer, size_t len)
{
    size_t sent = 0;
    while( sent != len)
    {
        const char* buf = static_cast<const char*>(buffer) + sent;
        int ret = send( buf, len - sent);
        if( ret < 0) return ret;
        sent += static_cast<unsigned>(ret);
    }
    return sent;
}
 
void TCPSocket::listen( int backlog /* = 5 */)
{
    int ret = ::listen( m_socket, backlog);
    if( ret < 0)
        throw SocketException("listen failed, most likely another socket is already listening on the same port");
}
 
TCPSocket::Handle TCPSocket::accept() const
{
    int ret = ::accept( m_socket, nullptr, nullptr);
    if( ret < 0)
        throw SocketException("TCPSocket::accept failed");
    return Handle(ret);
}
 
TCPSocket::Handle TCPSocket::timedAccept( int timeout) const
{
    struct pollfd poll;
    poll.fd = m_socket;
    poll.events = POLLIN;
 
    int ret = TEMP_FAILURE_RETRY (::poll( &poll, 1, timeout));
 
    if( ret == 0) return Handle();
    if( ret < 0) throw SocketException("timedAccept failed (poll)");
 
    ret = ::accept( m_socket, nullptr, nullptr);
    if( ret < 0)
        throw SocketException("TCPSocket::timedAccept failed");
    return Handle(ret);
}