2018-12-31 20:04:05 +01:00
|
|
|
//
|
2018-12-31 23:02:34 +01:00
|
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2019
|
2018-12-31 20:04:05 +01:00
|
|
|
//
|
|
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
|
|
//
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include "td/utils/port/config.h"
|
|
|
|
|
2018-09-07 02:41:21 +02:00
|
|
|
#include "td/utils/port/detail/NativeFd.h"
|
2018-08-13 19:15:09 +02:00
|
|
|
#include "td/utils/port/detail/PollableFd.h"
|
2018-12-31 20:04:05 +01:00
|
|
|
#include "td/utils/port/IPAddress.h"
|
2019-07-06 13:29:15 +02:00
|
|
|
#include "td/utils/port/IoSlice.h"
|
2018-09-07 02:41:21 +02:00
|
|
|
#include "td/utils/Slice.h"
|
2019-07-06 13:29:15 +02:00
|
|
|
#include "td/utils/Span.h"
|
2018-12-31 20:04:05 +01:00
|
|
|
#include "td/utils/Status.h"
|
|
|
|
|
2018-09-27 03:19:03 +02:00
|
|
|
#include <memory>
|
|
|
|
|
2018-12-31 20:04:05 +01:00
|
|
|
namespace td {
|
|
|
|
|
2018-08-13 19:15:09 +02:00
|
|
|
namespace detail {
|
|
|
|
class SocketFdImpl;
|
|
|
|
class SocketFdImplDeleter {
|
|
|
|
public:
|
|
|
|
void operator()(SocketFdImpl *impl);
|
|
|
|
};
|
|
|
|
} // namespace detail
|
2018-09-07 02:41:21 +02:00
|
|
|
|
2018-12-31 20:04:05 +01:00
|
|
|
class SocketFd {
|
|
|
|
public:
|
2018-08-13 19:15:09 +02:00
|
|
|
SocketFd();
|
2018-12-31 20:04:05 +01:00
|
|
|
SocketFd(const SocketFd &) = delete;
|
|
|
|
SocketFd &operator=(const SocketFd &) = delete;
|
2018-08-13 19:15:09 +02:00
|
|
|
SocketFd(SocketFd &&);
|
|
|
|
SocketFd &operator=(SocketFd &&);
|
|
|
|
~SocketFd();
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
static Result<SocketFd> open(const IPAddress &address) TD_WARN_UNUSED_RESULT;
|
|
|
|
|
2018-08-13 19:15:09 +02:00
|
|
|
PollableFdInfo &get_poll_info();
|
|
|
|
const PollableFdInfo &get_poll_info() const;
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
Status get_pending_error() TD_WARN_UNUSED_RESULT;
|
|
|
|
|
|
|
|
Result<size_t> write(Slice slice) TD_WARN_UNUSED_RESULT;
|
2019-07-06 13:29:15 +02:00
|
|
|
Result<size_t> writev(Span<IoSlice> slices) TD_WARN_UNUSED_RESULT;
|
2018-12-31 20:04:05 +01:00
|
|
|
Result<size_t> read(MutableSlice slice) TD_WARN_UNUSED_RESULT;
|
|
|
|
|
2018-08-13 19:15:09 +02:00
|
|
|
const NativeFd &get_native_fd() const;
|
|
|
|
static Result<SocketFd> from_native_fd(NativeFd fd);
|
|
|
|
|
2018-12-31 20:04:05 +01:00
|
|
|
void close();
|
|
|
|
bool empty() const;
|
|
|
|
|
|
|
|
private:
|
2018-08-13 19:15:09 +02:00
|
|
|
std::unique_ptr<detail::SocketFdImpl, detail::SocketFdImplDeleter> impl_;
|
2018-09-27 03:19:03 +02:00
|
|
|
explicit SocketFd(unique_ptr<detail::SocketFdImpl> impl);
|
2018-08-13 19:15:09 +02:00
|
|
|
};
|
2018-12-31 20:04:05 +01:00
|
|
|
|
2018-09-10 03:08:15 +02:00
|
|
|
namespace detail {
|
2018-12-17 17:12:47 +01:00
|
|
|
#if TD_PORT_POSIX
|
2018-08-13 19:15:09 +02:00
|
|
|
Status get_socket_pending_error(const NativeFd &fd);
|
2018-12-17 17:12:47 +01:00
|
|
|
#elif TD_PORT_WINDOWS
|
|
|
|
Status get_socket_pending_error(const NativeFd &fd, WSAOVERLAPPED *overlapped, Status iocp_error);
|
2018-09-10 03:08:15 +02:00
|
|
|
#endif
|
2018-12-17 17:12:47 +01:00
|
|
|
} // namespace detail
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
} // namespace td
|