aboutsummaryrefslogtreecommitdiffstats
path: root/util/network/sock.h
diff options
context:
space:
mode:
authorAndrey Fomichev <andrey.fomichev@gmail.com>2022-02-10 16:49:15 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:49:15 +0300
commit1c61afbf3db63940d05e6fefa3104b03457788a1 (patch)
tree5d5cb817648f650d76cf1076100726fd9b8448e8 /util/network/sock.h
parente542cc14db4240643a06bb0dde87ecf361f101ab (diff)
downloadydb-1c61afbf3db63940d05e6fefa3104b03457788a1.tar.gz
Restoring authorship annotation for Andrey Fomichev <andrey.fomichev@gmail.com>. Commit 2 of 2.
Diffstat (limited to 'util/network/sock.h')
-rw-r--r--util/network/sock.h636
1 files changed, 318 insertions, 318 deletions
diff --git a/util/network/sock.h b/util/network/sock.h
index ae9018adba..b10be2f715 100644
--- a/util/network/sock.h
+++ b/util/network/sock.h
@@ -1,276 +1,276 @@
#pragma once
-
+
#include <util/folder/path.h>
-#include <util/system/defaults.h>
-#include <util/string/cast.h>
+#include <util/system/defaults.h>
+#include <util/string/cast.h>
#include <util/stream/output.h>
#include <util/system/sysstat.h>
-
+
#if defined(_win_) || defined(_cygwin_)
#include <util/system/file.h>
-#else
+#else
#include <sys/un.h>
#include <sys/stat.h>
-#endif //_win_
-
-#include "init.h"
-#include "ip.h"
-#include "socket.h"
-
+#endif //_win_
+
+#include "init.h"
+#include "ip.h"
+#include "socket.h"
+
constexpr ui16 DEF_LOCAL_SOCK_MODE = 00644;
-
-// Base abstract class for socket address
+
+// Base abstract class for socket address
struct ISockAddr {
virtual ~ISockAddr() = default;
- // Max size of the address that we can store (arg of recvfrom)
- virtual socklen_t Size() const = 0;
- // Real length of the address (arg of sendto)
- virtual socklen_t Len() const = 0;
- // cast to sockaddr* to pass to any syscall
- virtual sockaddr* SockAddr() = 0;
- virtual const sockaddr* SockAddr() const = 0;
- // address in human readable form
+ // Max size of the address that we can store (arg of recvfrom)
+ virtual socklen_t Size() const = 0;
+ // Real length of the address (arg of sendto)
+ virtual socklen_t Len() const = 0;
+ // cast to sockaddr* to pass to any syscall
+ virtual sockaddr* SockAddr() = 0;
+ virtual const sockaddr* SockAddr() const = 0;
+ // address in human readable form
virtual TString ToString() const = 0;
-
-protected:
- // below are the implemetation methods that can be called by T*Socket classes
- friend class TBaseSocket;
- friend class TDgramSocket;
- friend class TStreamSocket;
-
- virtual int ResolveAddr() const {
- // usually it's nothing to do here
- return 0;
- }
- virtual int Bind(SOCKET s, ui16 mode) const = 0;
-};
-
+
+protected:
+ // below are the implemetation methods that can be called by T*Socket classes
+ friend class TBaseSocket;
+ friend class TDgramSocket;
+ friend class TStreamSocket;
+
+ virtual int ResolveAddr() const {
+ // usually it's nothing to do here
+ return 0;
+ }
+ virtual int Bind(SOCKET s, ui16 mode) const = 0;
+};
+
#if defined(_win_) || defined(_cygwin_)
#define YAF_LOCAL AF_INET
struct TSockAddrLocal: public ISockAddr {
- TSockAddrLocal() {
- Clear();
- }
-
- TSockAddrLocal(const char* path) {
- Set(path);
- }
-
- socklen_t Size() const {
- return sizeof(sockaddr_in);
- }
-
- socklen_t Len() const {
- return Size();
- }
-
+ TSockAddrLocal() {
+ Clear();
+ }
+
+ TSockAddrLocal(const char* path) {
+ Set(path);
+ }
+
+ socklen_t Size() const {
+ return sizeof(sockaddr_in);
+ }
+
+ socklen_t Len() const {
+ return Size();
+ }
+
inline void Clear() noexcept {
- Zero(in);
- Zero(Path);
- }
-
+ Zero(in);
+ Zero(Path);
+ }
+
inline void Set(const char* path) noexcept {
- Clear();
- in.sin_family = AF_INET;
- in.sin_addr.s_addr = IpFromString("127.0.0.1");
- in.sin_port = 0;
- strlcpy(Path, path, PathSize);
- }
-
- sockaddr* SockAddr() {
- return (struct sockaddr*)(&in);
- }
-
- const sockaddr* SockAddr() const {
- return (const struct sockaddr*)(&in);
- }
-
+ Clear();
+ in.sin_family = AF_INET;
+ in.sin_addr.s_addr = IpFromString("127.0.0.1");
+ in.sin_port = 0;
+ strlcpy(Path, path, PathSize);
+ }
+
+ sockaddr* SockAddr() {
+ return (struct sockaddr*)(&in);
+ }
+
+ const sockaddr* SockAddr() const {
+ return (const struct sockaddr*)(&in);
+ }
+
TString ToString() const {
return TString(Path);
- }
-
+ }
+
TFsPath ToPath() const {
return TFsPath(Path);
}
- int ResolveAddr() const {
- if (in.sin_port == 0) {
- int ret = 0;
- // 1. open file
- TFileHandle f(Path, OpenExisting | RdOnly);
- if (!f.IsOpen())
- return -errno;
-
- // 2. read the port from file
- ret = f.Read(&in.sin_port, sizeof(in.sin_port));
- if (ret != sizeof(in.sin_port))
- return -(errno ? errno : EFAULT);
- }
-
- return 0;
- }
-
- int Bind(SOCKET s, ui16 mode) const {
+ int ResolveAddr() const {
+ if (in.sin_port == 0) {
+ int ret = 0;
+ // 1. open file
+ TFileHandle f(Path, OpenExisting | RdOnly);
+ if (!f.IsOpen())
+ return -errno;
+
+ // 2. read the port from file
+ ret = f.Read(&in.sin_port, sizeof(in.sin_port));
+ if (ret != sizeof(in.sin_port))
+ return -(errno ? errno : EFAULT);
+ }
+
+ return 0;
+ }
+
+ int Bind(SOCKET s, ui16 mode) const {
Y_UNUSED(mode);
int ret = 0;
// 1. open file
TFileHandle f(Path, CreateAlways | WrOnly);
if (!f.IsOpen())
return -errno;
-
+
// 2. find port and bind to it
in.sin_port = 0;
ret = bind(s, SockAddr(), Len());
if (ret != 0)
return -WSAGetLastError();
-
+
int size = Size();
ret = getsockname(s, (struct sockaddr*)(&in), &size);
if (ret != 0)
return -WSAGetLastError();
-
+
// 3. write port to file
ret = f.Write(&(in.sin_port), sizeof(in.sin_port));
if (ret != sizeof(in.sin_port))
return -errno;
-
+
return 0;
- }
-
+ }
+
static constexpr size_t PathSize = 128;
- mutable struct sockaddr_in in;
- char Path[PathSize];
-};
-#else
+ mutable struct sockaddr_in in;
+ char Path[PathSize];
+};
+#else
#define YAF_LOCAL AF_LOCAL
struct TSockAddrLocal: public sockaddr_un, public ISockAddr {
- TSockAddrLocal() {
- Clear();
- }
-
- TSockAddrLocal(const char* path) {
- Set(path);
- }
-
+ TSockAddrLocal() {
+ Clear();
+ }
+
+ TSockAddrLocal(const char* path) {
+ Set(path);
+ }
+
socklen_t Size() const override {
- return sizeof(sockaddr_un);
- }
-
+ return sizeof(sockaddr_un);
+ }
+
socklen_t Len() const override {
- return strlen(sun_path) + 2;
- }
-
+ return strlen(sun_path) + 2;
+ }
+
inline void Clear() noexcept {
- Zero(*(sockaddr_un*)this);
- }
-
+ Zero(*(sockaddr_un*)this);
+ }
+
inline void Set(const char* path) noexcept {
- Clear();
- sun_family = AF_UNIX;
- strlcpy(sun_path, path, sizeof(sun_path));
- }
-
+ Clear();
+ sun_family = AF_UNIX;
+ strlcpy(sun_path, path, sizeof(sun_path));
+ }
+
sockaddr* SockAddr() override {
return (struct sockaddr*)(struct sockaddr_un*)this;
- }
-
+ }
+
const sockaddr* SockAddr() const override {
return (const struct sockaddr*)(const struct sockaddr_un*)this;
- }
-
+ }
+
TString ToString() const override {
return TString(sun_path);
- }
-
+ }
+
TFsPath ToPath() const {
return TFsPath(sun_path);
}
int Bind(SOCKET s, ui16 mode) const override {
- (void)unlink(sun_path);
-
- int ret = bind(s, SockAddr(), Len());
- if (ret < 0)
- return -errno;
-
+ (void)unlink(sun_path);
+
+ int ret = bind(s, SockAddr(), Len());
+ if (ret < 0)
+ return -errno;
+
ret = Chmod(sun_path, mode);
- if (ret < 0)
- return -errno;
- return 0;
- }
-};
-#endif // _win_
-
+ if (ret < 0)
+ return -errno;
+ return 0;
+ }
+};
+#endif // _win_
+
struct TSockAddrInet: public sockaddr_in, public ISockAddr {
- TSockAddrInet() {
- Clear();
- }
-
- TSockAddrInet(TIpHost ip, TIpPort port) {
- Set(ip, port);
- }
-
+ TSockAddrInet() {
+ Clear();
+ }
+
+ TSockAddrInet(TIpHost ip, TIpPort port) {
+ Set(ip, port);
+ }
+
TSockAddrInet(const char* ip, TIpPort port) {
Set(IpFromString(ip), port);
}
socklen_t Size() const override {
- return sizeof(sockaddr_in);
- }
-
+ return sizeof(sockaddr_in);
+ }
+
socklen_t Len() const override {
- return Size();
- }
-
+ return Size();
+ }
+
inline void Clear() noexcept {
- Zero(*(sockaddr_in*)this);
- }
-
+ Zero(*(sockaddr_in*)this);
+ }
+
inline void Set(TIpHost ip, TIpPort port) noexcept {
- Clear();
- sin_family = AF_INET;
- sin_addr.s_addr = ip;
- sin_port = HostToInet(port);
- }
-
+ Clear();
+ sin_family = AF_INET;
+ sin_addr.s_addr = ip;
+ sin_port = HostToInet(port);
+ }
+
sockaddr* SockAddr() override {
return (struct sockaddr*)(struct sockaddr_in*)this;
- }
-
+ }
+
const sockaddr* SockAddr() const override {
return (const struct sockaddr*)(const struct sockaddr_in*)this;
- }
-
+ }
+
TString ToString() const override {
- return IpToString(sin_addr.s_addr) + ":" + ::ToString(InetToHost(sin_port));
- }
-
+ return IpToString(sin_addr.s_addr) + ":" + ::ToString(InetToHost(sin_port));
+ }
+
int Bind(SOCKET s, ui16 mode) const override {
Y_UNUSED(mode);
- int ret = bind(s, SockAddr(), Len());
- if (ret < 0)
- return -errno;
-
+ int ret = bind(s, SockAddr(), Len());
+ if (ret < 0)
+ return -errno;
+
socklen_t len = Len();
if (getsockname(s, (struct sockaddr*)(SockAddr()), &len) < 0)
return -WSAGetLastError();
- return 0;
- }
-
+ return 0;
+ }
+
TIpHost GetIp() const noexcept {
- return sin_addr.s_addr;
- }
-
+ return sin_addr.s_addr;
+ }
+
TIpPort GetPort() const noexcept {
- return InetToHost(sin_port);
- }
+ return InetToHost(sin_port);
+ }
void SetPort(TIpPort port) noexcept {
sin_port = HostToInet(port);
}
-};
-
+};
+
struct TSockAddrInet6: public sockaddr_in6, public ISockAddr {
TSockAddrInet6() {
Clear();
@@ -345,113 +345,113 @@ using TSockAddrInetStream = TSockAddrInet;
using TSockAddrInetDgram = TSockAddrInet;
using TSockAddrInet6Stream = TSockAddrInet6;
using TSockAddrInet6Dgram = TSockAddrInet6;
-
+
class TBaseSocket: public TSocketHolder {
-protected:
- TBaseSocket(SOCKET fd)
- : TSocketHolder(fd)
- {
- }
-
-public:
+protected:
+ TBaseSocket(SOCKET fd)
+ : TSocketHolder(fd)
+ {
+ }
+
+public:
int Bind(const ISockAddr* addr, ui16 mode = DEF_LOCAL_SOCK_MODE) {
return addr->Bind((SOCKET) * this, mode);
- }
-
- void CheckSock() {
+ }
+
+ void CheckSock() {
if ((SOCKET) * this == INVALID_SOCKET)
- ythrow TSystemError() << "no socket";
- }
-
- static ssize_t Check(ssize_t ret, const char* op = "") {
- if (ret < 0)
+ ythrow TSystemError() << "no socket";
+ }
+
+ static ssize_t Check(ssize_t ret, const char* op = "") {
+ if (ret < 0)
ythrow TSystemError(-(int)ret) << "socket operation " << op;
- return ret;
- }
-};
-
+ return ret;
+ }
+};
+
class TDgramSocket: public TBaseSocket {
-protected:
- TDgramSocket(SOCKET fd)
- : TBaseSocket(fd)
- {
- }
-
-public:
+protected:
+ TDgramSocket(SOCKET fd)
+ : TBaseSocket(fd)
+ {
+ }
+
+public:
ssize_t SendTo(const void* msg, size_t len, const ISockAddr* toAddr) {
- ssize_t ret = toAddr->ResolveAddr();
+ ssize_t ret = toAddr->ResolveAddr();
if (ret < 0) {
return -LastSystemError();
}
-
+
ret = sendto((SOCKET) * this, (const char*)msg, (int)len, 0, toAddr->SockAddr(), toAddr->Len());
if (ret < 0) {
return -LastSystemError();
}
-
- return ret;
- }
-
+
+ return ret;
+ }
+
ssize_t RecvFrom(void* buf, size_t len, ISockAddr* fromAddr) {
- socklen_t fromSize = fromAddr->Size();
+ socklen_t fromSize = fromAddr->Size();
const ssize_t ret = recvfrom((SOCKET) * this, (char*)buf, (int)len, 0, fromAddr->SockAddr(), &fromSize);
- if (ret < 0) {
+ if (ret < 0) {
return -LastSystemError();
- }
-
- return ret;
- }
-};
-
+ }
+
+ return ret;
+ }
+};
+
class TStreamSocket: public TBaseSocket {
-protected:
+protected:
explicit TStreamSocket(SOCKET fd)
- : TBaseSocket(fd)
- {
- }
-
-public:
+ : TBaseSocket(fd)
+ {
+ }
+
+public:
TStreamSocket()
: TBaseSocket(INVALID_SOCKET)
{
}
- ssize_t Send(const void* msg, size_t len, int flags = 0) {
+ ssize_t Send(const void* msg, size_t len, int flags = 0) {
const ssize_t ret = send((SOCKET) * this, (const char*)msg, (int)len, flags);
- if (ret < 0)
- return -errno;
-
- return ret;
- }
-
- ssize_t Recv(void* buf, size_t len, int flags = 0) {
+ if (ret < 0)
+ return -errno;
+
+ return ret;
+ }
+
+ ssize_t Recv(void* buf, size_t len, int flags = 0) {
const ssize_t ret = recv((SOCKET) * this, (char*)buf, (int)len, flags);
- if (ret < 0)
- return -errno;
-
- return ret;
- }
-
+ if (ret < 0)
+ return -errno;
+
+ return ret;
+ }
+
int Connect(const ISockAddr* addr) {
- int ret = addr->ResolveAddr();
- if (ret < 0)
- return -errno;
-
+ int ret = addr->ResolveAddr();
+ if (ret < 0)
+ return -errno;
+
ret = connect((SOCKET) * this, addr->SockAddr(), addr->Len());
- if (ret < 0)
- return -errno;
-
- return ret;
- }
-
- int Listen(int backlog) {
+ if (ret < 0)
+ return -errno;
+
+ return ret;
+ }
+
+ int Listen(int backlog) {
int ret = listen((SOCKET) * this, backlog);
- if (ret < 0)
- return -errno;
-
- return ret;
- }
-
+ if (ret < 0)
+ return -errno;
+
+ return ret;
+ }
+
int Accept(TStreamSocket* acceptedSock, ISockAddr* acceptedAddr = nullptr) {
SOCKET s = INVALID_SOCKET;
if (acceptedAddr) {
@@ -462,40 +462,40 @@ public:
}
if (s == INVALID_SOCKET)
- return -errno;
-
- TSocketHolder sock(s);
- acceptedSock->Swap(sock);
- return 0;
- }
-};
-
+ return -errno;
+
+ TSocketHolder sock(s);
+ acceptedSock->Swap(sock);
+ return 0;
+ }
+};
+
class TLocalDgramSocket: public TDgramSocket {
-public:
- TLocalDgramSocket(SOCKET fd)
- : TDgramSocket(fd)
- {
- }
-
- TLocalDgramSocket()
- : TDgramSocket(socket(YAF_LOCAL, SOCK_DGRAM, 0))
- {
- }
-};
-
+public:
+ TLocalDgramSocket(SOCKET fd)
+ : TDgramSocket(fd)
+ {
+ }
+
+ TLocalDgramSocket()
+ : TDgramSocket(socket(YAF_LOCAL, SOCK_DGRAM, 0))
+ {
+ }
+};
+
class TInetDgramSocket: public TDgramSocket {
-public:
- TInetDgramSocket(SOCKET fd)
- : TDgramSocket(fd)
- {
- }
-
- TInetDgramSocket()
- : TDgramSocket(socket(AF_INET, SOCK_DGRAM, 0))
- {
- }
-};
-
+public:
+ TInetDgramSocket(SOCKET fd)
+ : TDgramSocket(fd)
+ {
+ }
+
+ TInetDgramSocket()
+ : TDgramSocket(socket(AF_INET, SOCK_DGRAM, 0))
+ {
+ }
+};
+
class TInet6DgramSocket: public TDgramSocket {
public:
TInet6DgramSocket(SOCKET fd)
@@ -510,30 +510,30 @@ public:
};
class TLocalStreamSocket: public TStreamSocket {
-public:
- TLocalStreamSocket(SOCKET fd)
- : TStreamSocket(fd)
- {
- }
-
- TLocalStreamSocket()
- : TStreamSocket(socket(YAF_LOCAL, SOCK_STREAM, 0))
- {
- }
-};
-
+public:
+ TLocalStreamSocket(SOCKET fd)
+ : TStreamSocket(fd)
+ {
+ }
+
+ TLocalStreamSocket()
+ : TStreamSocket(socket(YAF_LOCAL, SOCK_STREAM, 0))
+ {
+ }
+};
+
class TInetStreamSocket: public TStreamSocket {
-public:
- TInetStreamSocket(SOCKET fd)
- : TStreamSocket(fd)
- {
- }
-
- TInetStreamSocket()
- : TStreamSocket(socket(AF_INET, SOCK_STREAM, 0))
- {
- }
-};
+public:
+ TInetStreamSocket(SOCKET fd)
+ : TStreamSocket(fd)
+ {
+ }
+
+ TInetStreamSocket()
+ : TStreamSocket(socket(AF_INET, SOCK_STREAM, 0))
+ {
+ }
+};
class TInet6StreamSocket: public TStreamSocket {
public: