libcruft-util/cruft/util/posix/socket.cpp

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Copyright 2017 Danny Robson <danny@nerdcruft.net>
 */

#include "socket.hpp"

#include <cruft/util/platform.hpp>
#include <cruft/util/posix/except.hpp>

#if !defined(PLATFORM_WIN32)
    #include <sys/types.h>
    #include <sys/socket.h>
    #include <netdb.h>
    #include <netinet/in.h>
    #include <netinet/tcp.h>
#else
    #include <ws2tcpip.h>
    #include <winsock2.h>
#endif

using cruft::posix::socket;


///////////////////////////////////////////////////////////////////////////////
class lookup {
public:
    lookup (const cruft::view<const char*> host, int port):
        m_addresses (nullptr, freeaddrinfo)
    {
        struct ::addrinfo hints {};
        hints.ai_family = AF_UNSPEC;
        hints.ai_socktype = SOCK_STREAM;

        const struct {
            std::string host;
            std::string port;
        } strings {
            .host = std::string (std::begin (host), std::end (host)),
            .port = std::to_string (port)
        };

        addrinfo* _addresses;
        cruft::posix::eai::try_code (
            getaddrinfo (
                strings.host.c_str (),
                strings.port.c_str (),
                &hints,
                &_addresses
            )
        );

        m_addresses.reset (_addresses);
    };

    // TODO: use a wrapper that supports the increment operator
    auto begin (void) const { return m_addresses.get (); }
    auto end   (void) const { return nullptr; }

private:
    std::unique_ptr<addrinfo,void(*)(addrinfo*)> m_addresses;
};


///////////////////////////////////////////////////////////////////////////////
static cruft::posix::fd
connect_host (const cruft::view<const char*> host, int port)
{
    lookup l {host,port};
    for (auto _info = l.begin (); _info; _info = _info->ai_next) {
        auto &info = *_info;
        cruft::posix::fd sock {
            socket (info.ai_family, info.ai_socktype, info.ai_protocol)
        };
        if (sock == -1)
            continue;

        if (connect (sock, info.ai_addr, info.ai_addrlen))
            continue;

        return sock;
    }

    throw std::runtime_error ("unable to connect");
}


///////////////////////////////////////////////////////////////////////////////
socket::socket (int _fd):
    fd (_fd)
{ ; }


//-----------------------------------------------------------------------------
socket::socket (int domain, int type):
    socket (domain, type, 0)
{ ; }


//-----------------------------------------------------------------------------
socket::socket (int domain, int type, int protocol):
    fd (::socket (domain, type, protocol))
{ ; }


//-----------------------------------------------------------------------------
socket::socket (cruft::view<const char*> host, int port):
    fd (connect_host (host, port))
{ ; }


//-----------------------------------------------------------------------------
socket::~socket ()
{ ; }


//-----------------------------------------------------------------------------
socket::socket (socket &&rhs) noexcept:
    fd (std::move (rhs))
{ ; }


//-----------------------------------------------------------------------------
class socket&
socket::operator= (socket &&rhs) noexcept
{
    fd::operator= (std::move (rhs));
    return *this;
}


///////////////////////////////////////////////////////////////////////////////
void
socket::bind (sockaddr const &addr)
{
    auto const size = addr.sa_family == AF_INET  ? sizeof (sockaddr_in )
                    : addr.sa_family == AF_INET6 ? sizeof (sockaddr_in6)
                    : throw std::runtime_error ("Unhandled sockaddr type");

    cruft::posix::error::try_call (
        ::bind,
        native (),
        &addr,
        size
    );
}


//-----------------------------------------------------------------------------
void
socket::bind (sockaddr_in const &addr)
{
    return bind (reinterpret_cast<sockaddr const&> (addr));
}


///////////////////////////////////////////////////////////////////////////////
void
socket::connect (sockaddr_in const &addr)
{
    cruft::posix::error::try_code (
        ::connect (native (), reinterpret_cast<sockaddr const*>(&addr), sizeof (addr))
    );
}


//-----------------------------------------------------------------------------
void
socket::connect (cruft::view<const char*> host, int port)
{
    const lookup l { host, port };
    for (auto cursor = l.begin (); cursor != l.end (); cursor = cursor->ai_next) {
        if (!::connect (*this, cursor->ai_addr, cursor->ai_addrlen))
            return;
    }

    throw std::runtime_error ("unable to reconnect");
}



///////////////////////////////////////////////////////////////////////////////
template <typename ByteT>
ssize_t
socket::sendto (cruft::view<const ByteT *> buffer, int flags, ::sockaddr_in const &dst)
{
    return cruft::posix::error::try_value (
        ::sendto (
            native (),
            buffer.data (),
            buffer.size (),
            flags,
            reinterpret_cast<sockaddr const*> (&dst),
            sizeof (dst)
        )
    );
}


template ssize_t socket::sendto (cruft::view<const char*>, int, sockaddr_in const&);
template ssize_t socket::sendto (cruft::view<const unsigned char*>, int, sockaddr_in const&);
template ssize_t socket::sendto (cruft::view<const std::byte*>, int, sockaddr_in const&);



//-----------------------------------------------------------------------------
template <typename ByteT>
cruft::view<ByteT*>
socket::recvfrom (cruft::view<ByteT*> buffer, int flags)
{
    auto res = cruft::posix::error::try_value (
        ::recvfrom (native (), buffer.data (), buffer.size (), flags, nullptr, nullptr)
    );

    return { buffer.begin (), res };
}


template cruft::view<char*> socket::recvfrom (cruft::view<char*> buffer, int flags);
template cruft::view<unsigned char*> socket::recvfrom (cruft::view<unsigned char*> buffer, int flags);
template cruft::view<std::byte*> socket::recvfrom (cruft::view<std::byte*> buffer, int flags);