socket.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <memory>
#include <vector>
 
#include "asio.hpp"
#include "util/result.hpp"
 
using namespace fur::util;
 
namespace fur::download::socket {
using timeout = std::chrono::steady_clock::duration;
 
enum class SocketError {
  Timeout,
  Other,
};
 
struct AsioEngine {
  asio::io_context ctx;
  asio::ip::tcp::socket socket{ctx};
};
 
class Socket {
 public:
  Outcome<SocketError> connect(uint32_t ip, uint16_t port, timeout timeout);
 
  bool is_open();
 
  Outcome<SocketError> write(const std::vector<uint8_t>& buf, timeout timeout);
  Result<std::vector<uint8_t>, SocketError> read(int64_t n, timeout timeout);
 
  Outcome<SocketError> close();
 
 private:
  std::unique_ptr<AsioEngine> engine = std::make_unique<AsioEngine>();
 
  void run(timeout timeout);
};
}  // namespace fur::download::socket