socket.hh

Go to the documentation of this file.

#ifndef UTILMM_SOCKET_HH
#define UTILMM_SOCKET_HH

#include <boost/noncopyable.hpp>
#include <string>
#include <vector>
#include <sys/types.h>
#include <stdint.h>

namespace utilmm
{
    class base_socket : boost::noncopyable
    {
    public:
        enum Domain { Unix, Inet };
        enum Type { Stream, Datagram };
        enum Wait { WaitRead = 1, WaitWrite = 2, WaitException = 4};

    private:
        static int to_unix(Domain d);
        static int to_unix(Type t);

        struct bad_address : public std::exception { };

        int     m_fd;
        Domain  m_domain;
        Type    m_type;

        int wait(int what, timeval* tv) const;

    protected:
        std::vector<uint8_t> to_sockaddr(std::string const& to) const;

    public:
        base_socket(int fd);
        base_socket(Domain domain, Type type);

        ~base_socket();

        int fd() const;

        bool try_wait(int what) const;

        void wait(int what) const;

        void flush() const;
    };

    class socket : public base_socket
    {
        friend class server_socket;

    public:
        socket(int fd);

        socket(Domain domain, Type type, std::string const& connect_to);

        void connect(std::string const& to);

        int read(void* buf, size_t size) const;

        int write(void const* buf, size_t size) const;
    };

    class server_socket : public base_socket
    {
    public:
        server_socket(Domain domain, Type type, std::string const& bind_to, int backlog = 256);

        void bind(std::string const& to);

        socket* accept() const;

        bool try_wait() const;

        void wait() const;
    };
}

#endif