1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#pragma once
#include <util/system/platform.h>
#include <util/generic/noncopyable.h>
#include <util/generic/ptr.h>
#include <util/network/init.h>
#include <util/system/defaults.h>
#include <util/system/hp_timer.h>
#include "udp_recv_packet.h"
#include "protocols.h"
#include <sys/uio.h>
namespace NNetlibaSocket {
typedef iovec TIoVec;
#ifdef _win32_
struct TMsgHdr {
void* msg_name; /* optional address */
int msg_namelen; /* size of address */
TIoVec* msg_iov; /* scatter/gather array */
int msg_iovlen; /* # elements in msg_iov */
int Tos; // netlib_socket extension
};
#else
#include <sys/socket.h>
typedef msghdr TMsgHdr;
#endif
// equal to glibc 2.14 mmsghdr definition, defined for windows and darwin compatibility
struct TMMsgHdr {
TMsgHdr msg_hdr;
unsigned int msg_len;
};
#if defined(_linux_)
#include <linux/version.h>
#include <features.h>
// sendmmsg was added in glibc 2.14 and linux 3.0
#if __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 14 && LINUX_VERSION_CODE >= KERNEL_VERSION(3, 0, 0)
#include <sys/socket.h> // sendmmsg
static_assert(sizeof(TMMsgHdr) == sizeof(mmsghdr), "expect sizeof(TMMsgHdr) == sizeof(mmsghdr)");
#endif
#endif
#ifdef _win32_
const size_t TOS_BUFFER_SIZE = sizeof(int);
const size_t CTRL_BUFFER_SIZE = 32;
#else
#if defined(_darwin_)
#define Y_DARWIN_ALIGN32(p) ((__darwin_size_t)((__darwin_size_t)(p) + __DARWIN_ALIGNBYTES32) & ~__DARWIN_ALIGNBYTES32)
#define Y_CMSG_SPACE(l) (Y_DARWIN_ALIGN32(sizeof(struct cmsghdr)) + Y_DARWIN_ALIGN32(l))
#else
#define Y_CMSG_SPACE(l) CMSG_SPACE(l)
#endif
constexpr size_t TOS_BUFFER_SIZE = Y_CMSG_SPACE(sizeof(int));
constexpr size_t CTRL_BUFFER_SIZE = Y_CMSG_SPACE(sizeof(int)) + Y_CMSG_SPACE(sizeof(struct in6_pktinfo));
#endif
///////////////////////////////////////////////////////////////////////////////
// Warning: every variable (tosBuffer, data, addr, iov) passed and returned from these functions must exist until actual send!!!
void* CreateTos(const ui8 tos, void* tosBuffer);
TIoVec CreateIoVec(char* data, const size_t dataSize);
TMsgHdr CreateSendMsgHdr(const sockaddr_in6& addr, const TIoVec& iov, void* tosBuffer);
TMsgHdr CreateRecvMsgHdr(sockaddr_in6* addrBuf, const TIoVec& iov, void* ctrlBuffer = nullptr);
TMsgHdr* AddSockAuxData(TMsgHdr* header, const ui8 tos, const sockaddr_in6& addr, void* buffer, size_t bufferSize);
///////////////////////////////////////////////////////////////////////////////
//returns false if TOS wasn't readed and do not touch *tos
bool ReadTos(const TMsgHdr& msgHdr, ui8* tos);
bool ExtractDestinationAddress(TMsgHdr& msgHdr, sockaddr_in6* addrBuf);
///////////////////////////////////////////////////////////////////////////////
// currently netliba v6 version id is any number which's not equal to NETLIBA_V12_VERSION
enum { NETLIBA_ANY_VERSION = -1,
NETLIBA_V12_VERSION = 112 };
enum EFragFlag {
FF_ALLOW_FRAG,
FF_DONT_FRAG
};
///////////////////////////////////////////////////////////////////////////////
class ISocket: public TNonCopyable, public TThrRefBase {
public:
~ISocket() override {
}
virtual int Open(int port) = 0;
virtual void Close() = 0;
virtual bool IsValid() const = 0;
virtual const sockaddr_in6& GetSelfAddress() const = 0;
virtual int GetNetworkOrderPort() const = 0;
virtual int GetPort() const = 0;
virtual int GetSockOpt(int level, int option_name, void* option_value, socklen_t* option_len) = 0;
// send all packets to this and only this address by default
virtual int Connect(const struct sockaddr* address, socklen_t address_len) = 0;
virtual void Wait(float timeoutSec, int netlibaVersion = NETLIBA_ANY_VERSION) const = 0;
virtual void CancelWait(int netlibaVersion = NETLIBA_ANY_VERSION) = 0;
virtual void CancelWaitHost(const sockaddr_in6 address) = 0;
virtual bool IsSendMMsgSupported() const = 0;
virtual int SendMMsg(struct TMMsgHdr* msgvec, unsigned int vlen, unsigned int flags) = 0;
virtual ssize_t SendMsg(const TMsgHdr* hdr, int flags, const EFragFlag frag) = 0;
virtual bool IsRecvMsgSupported() const = 0;
virtual ssize_t RecvMsg(TMsgHdr* hdr, int flags) = 0;
virtual TUdpRecvPacket* Recv(sockaddr_in6* srcAddr, sockaddr_in6* dstAddr, int netlibaVersion = NETLIBA_ANY_VERSION) = 0;
virtual bool IncreaseSendBuff() = 0;
virtual int GetSendSysSocketSize() = 0;
virtual void SetRecvLagTime(NHPTimer::STime time) = 0;
};
TIntrusivePtr<ISocket> CreateSocket(); // not thread safe!
TIntrusivePtr<ISocket> CreateDualStackSocket(); // has thread safe send/recv methods
// this function was added mostly for testing
TIntrusivePtr<ISocket> CreateBestRecvSocket();
}
|
