2 * Copyright 2017 Facebook, Inc.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
23 #include <arpa/inet.h>
24 #include <netinet/in.h>
25 #include <netinet/tcp.h>
26 #include <sys/socket.h>
30 #define FOLLY_HAVE_MSG_ERRQUEUE 1
31 /* for struct sock_extended_err*/
32 #include <linux/errqueue.h>
35 #ifndef SO_EE_ORIGIN_ZEROCOPY
36 #define SO_EE_ORIGIN_ZEROCOPY 5
39 #ifndef SO_EE_CODE_ZEROCOPY_COPIED
40 #define SO_EE_CODE_ZEROCOPY_COPIED 1
44 #define SO_ZEROCOPY 60
48 #define MSG_ZEROCOPY 0x4000000
52 #include <folly/portability/IOVec.h>
53 #include <folly/portability/SysTypes.h>
54 #include <folly/portability/Windows.h>
56 #include <WS2tcpip.h> // @manual
59 using sa_family_t = ADDRESS_FAMILY;
61 // these are not supported
62 #define SO_EE_ORIGIN_ZEROCOPY 0
64 #define MSG_ZEROCOPY 0x0
66 // We don't actually support either of these flags
68 #define MSG_DONTWAIT 0x1000
72 socklen_t msg_namelen;
73 struct iovec* msg_iov;
76 size_t msg_controllen;
81 sa_family_t sun_family;
85 #define SHUT_RD SD_RECEIVE
86 #define SHUT_WR SD_SEND
87 #define SHUT_RDWR SD_BOTH
89 // These are the same, but PF_LOCAL
90 // isn't defined by WinSock.
91 #define AF_LOCAL PF_UNIX
92 #define PF_LOCAL PF_UNIX
94 // This isn't defined by Windows, and we need to
95 // distinguish it from SO_REUSEADDR
96 #define SO_REUSEPORT 0x7001
98 // Someone thought it would be a good idea
99 // to define a field via a macro...
104 namespace portability {
125 // Some Windows specific helper functions.
126 bool is_fh_socket(int fh);
127 SOCKET fd_to_socket(int fd);
128 int socket_to_fd(SOCKET s);
129 int translate_wsa_error(int wsaErr);
131 // These aren't additional overloads, but rather other functions that
132 // are referenced that we need to wrap, or, in the case of inet_aton,
134 int accept(int s, struct sockaddr* addr, socklen_t* addrlen);
135 int inet_aton(const char* cp, struct in_addr* inp);
136 int socketpair(int domain, int type, int protocol, int sv[2]);
138 // Unless you have a case where you would normally have
139 // to reference the function as being explicitly in the
140 // global scope, then you shouldn't be calling these directly.
141 int bind(int s, const struct sockaddr* name, socklen_t namelen);
142 int connect(int s, const struct sockaddr* name, socklen_t namelen);
143 int getpeername(int s, struct sockaddr* name, socklen_t* namelen);
144 int getsockname(int s, struct sockaddr* name, socklen_t* namelen);
145 int getsockopt(int s, int level, int optname, void* optval, socklen_t* optlen);
146 const char* inet_ntop(int af, const void* src, char* dst, socklen_t size);
147 int listen(int s, int backlog);
148 int poll(struct pollfd fds[], nfds_t nfds, int timeout);
149 ssize_t recv(int s, void* buf, size_t len, int flags);
155 struct sockaddr* from,
157 ssize_t send(int s, const void* buf, size_t len, int flags);
165 ssize_t sendmsg(int socket, const struct msghdr* message, int flags);
172 int shutdown(int s, int how);
174 // This is the only function that _must_ be referenced via the namespace
175 // because there is no difference in parameter types to overload
177 int socket(int af, int type, int protocol);
179 // Windows needs a few extra overloads of some of the functions in order to
180 // resolve to our portability functions rather than the SOCKET accepting
182 int getsockopt(int s, int level, int optname, char* optval, socklen_t* optlen);
183 ssize_t recv(int s, char* buf, int len, int flags);
184 ssize_t recv(int s, void* buf, int len, int flags);
190 struct sockaddr* from,
197 struct sockaddr* from,
199 ssize_t recvmsg(int s, struct msghdr* message, int fl);
200 ssize_t send(int s, const char* buf, int len, int flags);
201 ssize_t send(int s, const void* buf, int len, int flags);
223 } // namespace sockets
224 } // namespace portability
228 // Add our helpers to the overload set.
230 using namespace folly::portability::sockets;