Mercurial > code
view C++/SocketTcp.h @ 316:4c0af1143fc4
Add wait operation (no tests yet)
author | David Demelier <markand@malikania.fr> |
---|---|
date | Tue, 03 Mar 2015 18:48:54 +0100 |
parents | c9356cb38c86 |
children | 68ae6d7dea1f |
line wrap: on
line source
/* * SocketTcp.h -- portable C++ socket wrappers * * Copyright (c) 2013, 2014 David Demelier <markand@malikania.fr> * * Permission to use, copy, modify, and/or distribute this software for any * purpose with or without fee is hereby granted, provided that the above * copyright notice and this permission notice appear in all copies. * * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */ #ifndef _SOCKET_TCP_NG_3_H_ #define _SOCKET_TCP_NG_3_H_ #include "Socket.h" /** * @class SocketAbstractTcp * @brief Base class for TCP sockets */ class SocketAbstractTcp : public Socket { public: using Socket::Socket; inline SocketAbstractTcp(int domain, int protocol) : Socket(domain, SOCK_STREAM, protocol) { } void listen(int max = 128); inline std::string recv(unsigned count) { std::string result; result.resize(count); auto n = recv(const_cast<char *>(result.data()), count); result.resize(n); return result; } inline std::string waitRecv(unsigned count, int timeout) { std::string result; result.resize(count); auto n = waitRecv(const_cast<char *>(result.data()), count, timeout); result.resize(n); return result; } inline unsigned send(const std::string &data) { return send(data.c_str(), data.size()); } inline unsigned waitSend(const std::string &data, int timeout) { return waitSend(data.c_str(), data.size(), timeout); } virtual unsigned recv(void *data, unsigned length) = 0; virtual unsigned waitRecv(void *data, unsigned length, int timeout) = 0; virtual unsigned send(const void *data, unsigned length) = 0; virtual unsigned waitSend(const void *data, unsigned length, int timeout) = 0; }; /** * @class SocketTcp * @brief End-user class for TCP sockets */ class SocketTcp final : public SocketAbstractTcp { public: using SocketAbstractTcp::SocketAbstractTcp; using SocketAbstractTcp::recv; using SocketAbstractTcp::send; SocketTcp accept(); SocketTcp accept(SocketAddress &info); SocketTcp waitAccept(int timeout); SocketTcp waitAccept(SocketAddress &info, int timeout); void connect(const SocketAddress &address); void waitConnect(const SocketAddress &address, int timeout); unsigned recv(void *data, unsigned length) override; unsigned waitRecv(void *data, unsigned length, int timeout) override; unsigned send(const void *data, unsigned length) override; unsigned waitSend(const void *data, unsigned length, int timeout) override; }; #endif // !_SOCKET_TCP_NG_3_H_