tcp.h

Go to the documentation of this file.

// uvco (c) 2024 Lewin Bormann. See LICENSE for specific terms.
 
#pragma once
 
#include <uv.h>
 
#include <boost/assert.hpp>
#include <fmt/format.h>
 
#include "uvco/internal/internal_utils.h"
#include "uvco/name_resolution.h"
#include "uvco/promise/promise.h"
#include "uvco/run.h"
#include "uvco/stream_server_base.h"
#include "uvco/tcp_stream.h"
 
#include <coroutine>
#include <cstdint>
#include <optional>
#include <string>
#include <sys/socket.h>
 
namespace uvco {
 
 
class TcpClient {
public:
  TcpClient(const Loop &loop, std::string target_host_address,
            uint16_t target_host_port, int af_hint = AF_UNSPEC);
  TcpClient(const Loop &loop, AddressHandle address);
 
  TcpClient(TcpClient &&other) noexcept;
  TcpClient(const TcpClient &) = delete;
  TcpClient &operator=(TcpClient &&other) noexcept;
  TcpClient &operator=(const TcpClient &) = delete;
  ~TcpClient() = default;
 
  Promise<TcpStream> connect();
 
private:
  const Loop *loop_;
 
  // May be a name or address; resolved upon connect().
  std::string host_;
  int af_hint_;
  uint16_t port_;
 
  static void onConnect(uv_connect_t *req, uv_status status);
 
  struct ConnectAwaiter_ {
    [[nodiscard]] bool await_ready() const;
    bool await_suspend(std::coroutine_handle<> handle);
    uv_status await_resume();
 
    void onConnect(uv_status status);
 
    std::optional<std::coroutine_handle<>> handle_;
    std::optional<uv_status> status_;
  };
};
 
class TcpServer : public StreamServerBase<uv_tcp_t, TcpStream> {
public:
  TcpServer(const Loop &loop, AddressHandle bindAddress, bool ipv6Only = false);
 
  TcpServer(const TcpServer &) = delete;
  TcpServer(TcpServer &&) = default;
  TcpServer &operator=(const TcpServer &) = delete;
  TcpServer &operator=(TcpServer &&) = default;
  ~TcpServer() = default;
 
private:
  void bind(const struct sockaddr *addr, int flags);
};
 
 
} // namespace uvco