mirror of
https://github.com/XTXMarkets/ternfs.git
synced 2026-01-04 01:49:42 -06:00
8a0ea10cde
* core: UDPSocketPair and use IpPort AddrsInfo everywhere * Refactor UDPSocketPair a bit * ci: kmod always delete img before create * shuckle: fix scripts/json marshal --------- Co-authored-by: Francesco Mazzoli <francesco.mazzoli@xtxmarkets.com>
43 lines
1.2 KiB
C++
43 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <utility>
|
|
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
|
|
#include "Assert.hpp"
|
|
#include "Time.hpp"
|
|
|
|
class Sock {
|
|
public:
|
|
static Sock SockError(int errnum) { ALWAYS_ASSERT(errnum > 0); return {-errnum}; }
|
|
static Sock TCPSock() { return {socket(AF_INET, SOCK_STREAM, 0)}; }
|
|
static Sock UDPSock() { return {socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)}; }
|
|
|
|
Sock(): _fd(-1) {}
|
|
Sock(const Sock&) = delete;
|
|
Sock(Sock&& s) : _fd(s._fd) { s._fd = -1; }
|
|
~Sock() { if (_fd > 0) { close(_fd); _fd = -1; } }
|
|
Sock& operator=(Sock&& s) { this->~Sock(); this->_fd = s.release(); return *this; }
|
|
|
|
bool error() const { return _fd < 0; }
|
|
|
|
int getErrno() const { ALWAYS_ASSERT(_fd < 0); return -_fd; }
|
|
int get() const { ALWAYS_ASSERT(_fd > 0); return _fd; }
|
|
int release() { int fd = _fd; _fd = -1; return fd; }
|
|
private:
|
|
Sock(int fd) : _fd(fd) {}
|
|
int _fd;
|
|
};
|
|
|
|
// For errors that we probably shouldn't retry, an exception is thrown. Otherwise
|
|
// -errno fd and error string.
|
|
std::pair<Sock, std::string> connectToHost(
|
|
const std::string& host,
|
|
uint16_t port,
|
|
Duration timeout = 10_sec
|
|
);
|