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>
29 #include <folly/portability/IOVec.h>
30 #include <folly/portability/SysTypes.h>
31 #include <folly/portability/Windows.h>
36 using sa_family_t = ADDRESS_FAMILY;
38 // We don't actually support either of these flags
40 #define MSG_DONTWAIT 0x1000
44 socklen_t msg_namelen;
45 struct iovec* msg_iov;
48 size_t msg_controllen;
53 sa_family_t sun_family;
57 #define SHUT_RD SD_RECEIVE
58 #define SHUT_WR SD_SEND
59 #define SHUT_RDWR SD_BOTH
61 // These are the same, but PF_LOCAL
62 // isn't defined by WinSock.
63 #define AF_LOCAL PF_UNIX
64 #define PF_LOCAL PF_UNIX
66 // This isn't defined by Windows, and we need to
67 // distinguish it from SO_REUSEADDR
68 #define SO_REUSEPORT 0x7001
70 // Someone thought it would be a good idea
71 // to define a field via a macro...
76 namespace portability {
97 // Some Windows specific helper functions.
98 bool is_fh_socket(int fh);
99 SOCKET fd_to_socket(int fd);
100 int socket_to_fd(SOCKET s);
101 int translate_wsa_error(int wsaErr);
103 // These aren't additional overloads, but rather other functions that
104 // are referenced that we need to wrap, or, in the case of inet_aton,
106 int accept(int s, struct sockaddr* addr, socklen_t* addrlen);
107 int inet_aton(const char* cp, struct in_addr* inp);
108 int socketpair(int domain, int type, int protocol, int sv[2]);
110 // Unless you have a case where you would normally have
111 // to reference the function as being explicitly in the
112 // global scope, then you shouldn't be calling these directly.
113 int bind(int s, const struct sockaddr* name, socklen_t namelen);
114 int connect(int s, const struct sockaddr* name, socklen_t namelen);
115 int getpeername(int s, struct sockaddr* name, socklen_t* namelen);
116 int getsockname(int s, struct sockaddr* name, socklen_t* namelen);
117 int getsockopt(int s, int level, int optname, void* optval, socklen_t* optlen);
118 const char* inet_ntop(int af, const void* src, char* dst, socklen_t size);
119 int listen(int s, int backlog);
120 int poll(struct pollfd fds[], nfds_t nfds, int timeout);
121 ssize_t recv(int s, void* buf, size_t len, int flags);
127 struct sockaddr* from,
129 ssize_t send(int s, const void* buf, size_t len, int flags);
137 ssize_t sendmsg(int socket, const struct msghdr* message, int flags);
144 int shutdown(int s, int how);
146 // This is the only function that _must_ be referenced via the namespace
147 // because there is no difference in parameter types to overload
149 int socket(int af, int type, int protocol);
151 // Windows needs a few extra overloads of some of the functions in order to
152 // resolve to our portability functions rather than the SOCKET accepting
154 int getsockopt(int s, int level, int optname, char* optval, socklen_t* optlen);
155 ssize_t recv(int s, char* buf, int len, int flags);
156 ssize_t recv(int s, void* buf, int len, int flags);
162 struct sockaddr* from,
169 struct sockaddr* from,
171 ssize_t recvmsg(int s, struct msghdr* message, int fl);
172 ssize_t send(int s, const char* buf, int len, int flags);
173 ssize_t send(int s, const void* buf, int len, int flags);
200 // Add our helpers to the overload set.
201 /* using override */ using namespace folly::portability::sockets;