1 | /* |
2 | * Copyright 1995-2018 The OpenSSL Project Authors. All Rights Reserved. |
3 | * |
4 | * Licensed under the Apache License 2.0 (the "License"). You may not use |
5 | * this file except in compliance with the License. You can obtain a copy |
6 | * in the file LICENSE in the source distribution or at |
7 | * https://www.openssl.org/source/license.html |
8 | */ |
9 | |
10 | #include <stdio.h> |
11 | #include <stdlib.h> |
12 | #include <errno.h> |
13 | #include "bio_local.h" |
14 | #ifndef OPENSSL_NO_SOCK |
15 | # define SOCKET_PROTOCOL IPPROTO_TCP |
16 | # ifdef SO_MAXCONN |
17 | # define MAX_LISTEN SO_MAXCONN |
18 | # elif defined(SOMAXCONN) |
19 | # define MAX_LISTEN SOMAXCONN |
20 | # else |
21 | # define MAX_LISTEN 32 |
22 | # endif |
23 | # if defined(OPENSSL_SYS_WINDOWS) |
24 | static int wsa_init_done = 0; |
25 | # endif |
26 | |
27 | # ifndef OPENSSL_NO_DEPRECATED_1_1_0 |
28 | int BIO_get_host_ip(const char *str, unsigned char *ip) |
29 | { |
30 | BIO_ADDRINFO *res = NULL; |
31 | int ret = 0; |
32 | |
33 | if (BIO_sock_init() != 1) |
34 | return 0; /* don't generate another error code here */ |
35 | |
36 | if (BIO_lookup(str, NULL, BIO_LOOKUP_CLIENT, AF_INET, SOCK_STREAM, &res)) { |
37 | size_t l; |
38 | |
39 | if (BIO_ADDRINFO_family(res) != AF_INET) { |
40 | BIOerr(BIO_F_BIO_GET_HOST_IP, |
41 | BIO_R_GETHOSTBYNAME_ADDR_IS_NOT_AF_INET); |
42 | } else if (BIO_ADDR_rawaddress(BIO_ADDRINFO_address(res), NULL, &l)) { |
43 | /* |
44 | * Because only AF_INET addresses will reach this far, we can assert |
45 | * that l should be 4 |
46 | */ |
47 | if (ossl_assert(l == 4)) |
48 | ret = BIO_ADDR_rawaddress(BIO_ADDRINFO_address(res), ip, &l); |
49 | } |
50 | BIO_ADDRINFO_free(res); |
51 | } else { |
52 | ERR_add_error_data(2, "host=" , str); |
53 | } |
54 | |
55 | return ret; |
56 | } |
57 | |
58 | int BIO_get_port(const char *str, unsigned short *port_ptr) |
59 | { |
60 | BIO_ADDRINFO *res = NULL; |
61 | int ret = 0; |
62 | |
63 | if (str == NULL) { |
64 | BIOerr(BIO_F_BIO_GET_PORT, BIO_R_NO_PORT_DEFINED); |
65 | return 0; |
66 | } |
67 | |
68 | if (BIO_sock_init() != 1) |
69 | return 0; /* don't generate another error code here */ |
70 | |
71 | if (BIO_lookup(NULL, str, BIO_LOOKUP_CLIENT, AF_INET, SOCK_STREAM, &res)) { |
72 | if (BIO_ADDRINFO_family(res) != AF_INET) { |
73 | BIOerr(BIO_F_BIO_GET_PORT, |
74 | BIO_R_ADDRINFO_ADDR_IS_NOT_AF_INET); |
75 | } else { |
76 | *port_ptr = ntohs(BIO_ADDR_rawport(BIO_ADDRINFO_address(res))); |
77 | ret = 1; |
78 | } |
79 | BIO_ADDRINFO_free(res); |
80 | } else { |
81 | ERR_add_error_data(2, "host=" , str); |
82 | } |
83 | |
84 | return ret; |
85 | } |
86 | # endif |
87 | |
88 | int BIO_sock_error(int sock) |
89 | { |
90 | int j = 0, i; |
91 | socklen_t size = sizeof(j); |
92 | |
93 | /* |
94 | * Note: under Windows the third parameter is of type (char *) whereas |
95 | * under other systems it is (void *) if you don't have a cast it will |
96 | * choke the compiler: if you do have a cast then you can either go for |
97 | * (char *) or (void *). |
98 | */ |
99 | i = getsockopt(sock, SOL_SOCKET, SO_ERROR, (void *)&j, &size); |
100 | if (i < 0) |
101 | return get_last_socket_error(); |
102 | else |
103 | return j; |
104 | } |
105 | |
106 | # ifndef OPENSSL_NO_DEPRECATED_1_1_0 |
107 | struct hostent *BIO_gethostbyname(const char *name) |
108 | { |
109 | /* |
110 | * Caching gethostbyname() results forever is wrong, so we have to let |
111 | * the true gethostbyname() worry about this |
112 | */ |
113 | return gethostbyname(name); |
114 | } |
115 | # endif |
116 | |
117 | int BIO_sock_init(void) |
118 | { |
119 | # ifdef OPENSSL_SYS_WINDOWS |
120 | static struct WSAData wsa_state; |
121 | |
122 | if (!wsa_init_done) { |
123 | wsa_init_done = 1; |
124 | memset(&wsa_state, 0, sizeof(wsa_state)); |
125 | /* |
126 | * Not making wsa_state available to the rest of the code is formally |
127 | * wrong. But the structures we use are [believed to be] invariable |
128 | * among Winsock DLLs, while API availability is [expected to be] |
129 | * probed at run-time with DSO_global_lookup. |
130 | */ |
131 | if (WSAStartup(0x0202, &wsa_state) != 0) { |
132 | ERR_raise_data(ERR_LIB_SYS, get_last_socket_error(), |
133 | "calling wsastartup()" ); |
134 | BIOerr(BIO_F_BIO_SOCK_INIT, BIO_R_WSASTARTUP); |
135 | return -1; |
136 | } |
137 | } |
138 | # endif /* OPENSSL_SYS_WINDOWS */ |
139 | # ifdef WATT32 |
140 | extern int _watt_do_exit; |
141 | _watt_do_exit = 0; /* don't make sock_init() call exit() */ |
142 | if (sock_init()) |
143 | return -1; |
144 | # endif |
145 | |
146 | return 1; |
147 | } |
148 | |
149 | void bio_sock_cleanup_int(void) |
150 | { |
151 | # ifdef OPENSSL_SYS_WINDOWS |
152 | if (wsa_init_done) { |
153 | wsa_init_done = 0; |
154 | WSACleanup(); |
155 | } |
156 | # endif |
157 | } |
158 | |
159 | int BIO_socket_ioctl(int fd, long type, void *arg) |
160 | { |
161 | int i; |
162 | |
163 | # ifdef __DJGPP__ |
164 | i = ioctlsocket(fd, type, (char *)arg); |
165 | # else |
166 | # if defined(OPENSSL_SYS_VMS) |
167 | /*- |
168 | * 2011-02-18 SMS. |
169 | * VMS ioctl() can't tolerate a 64-bit "void *arg", but we |
170 | * observe that all the consumers pass in an "unsigned long *", |
171 | * so we arrange a local copy with a short pointer, and use |
172 | * that, instead. |
173 | */ |
174 | # if __INITIAL_POINTER_SIZE == 64 |
175 | # define ARG arg_32p |
176 | # pragma pointer_size save |
177 | # pragma pointer_size 32 |
178 | unsigned long arg_32; |
179 | unsigned long *arg_32p; |
180 | # pragma pointer_size restore |
181 | arg_32p = &arg_32; |
182 | arg_32 = *((unsigned long *)arg); |
183 | # else /* __INITIAL_POINTER_SIZE == 64 */ |
184 | # define ARG arg |
185 | # endif /* __INITIAL_POINTER_SIZE == 64 [else] */ |
186 | # else /* defined(OPENSSL_SYS_VMS) */ |
187 | # define ARG arg |
188 | # endif /* defined(OPENSSL_SYS_VMS) [else] */ |
189 | |
190 | i = ioctlsocket(fd, type, ARG); |
191 | # endif /* __DJGPP__ */ |
192 | if (i < 0) |
193 | ERR_raise_data(ERR_LIB_SYS, get_last_socket_error(), |
194 | "calling ioctlsocket()" ); |
195 | return i; |
196 | } |
197 | |
198 | # ifndef OPENSSL_NO_DEPRECATED_1_1_0 |
199 | int BIO_get_accept_socket(char *host, int bind_mode) |
200 | { |
201 | int s = INVALID_SOCKET; |
202 | char *h = NULL, *p = NULL; |
203 | BIO_ADDRINFO *res = NULL; |
204 | |
205 | if (!BIO_parse_hostserv(host, &h, &p, BIO_PARSE_PRIO_SERV)) |
206 | return INVALID_SOCKET; |
207 | |
208 | if (BIO_sock_init() != 1) |
209 | return INVALID_SOCKET; |
210 | |
211 | if (BIO_lookup(h, p, BIO_LOOKUP_SERVER, AF_UNSPEC, SOCK_STREAM, &res) != 0) |
212 | goto err; |
213 | |
214 | if ((s = BIO_socket(BIO_ADDRINFO_family(res), BIO_ADDRINFO_socktype(res), |
215 | BIO_ADDRINFO_protocol(res), 0)) == INVALID_SOCKET) { |
216 | s = INVALID_SOCKET; |
217 | goto err; |
218 | } |
219 | |
220 | if (!BIO_listen(s, BIO_ADDRINFO_address(res), |
221 | bind_mode ? BIO_SOCK_REUSEADDR : 0)) { |
222 | BIO_closesocket(s); |
223 | s = INVALID_SOCKET; |
224 | } |
225 | |
226 | err: |
227 | BIO_ADDRINFO_free(res); |
228 | OPENSSL_free(h); |
229 | OPENSSL_free(p); |
230 | |
231 | return s; |
232 | } |
233 | |
234 | int BIO_accept(int sock, char **ip_port) |
235 | { |
236 | BIO_ADDR res; |
237 | int ret = -1; |
238 | |
239 | ret = BIO_accept_ex(sock, &res, 0); |
240 | if (ret == (int)INVALID_SOCKET) { |
241 | if (BIO_sock_should_retry(ret)) { |
242 | ret = -2; |
243 | goto end; |
244 | } |
245 | ERR_raise_data(ERR_LIB_SYS, get_last_socket_error(), |
246 | "calling accept()" ); |
247 | BIOerr(BIO_F_BIO_ACCEPT, BIO_R_ACCEPT_ERROR); |
248 | goto end; |
249 | } |
250 | |
251 | if (ip_port != NULL) { |
252 | char *host = BIO_ADDR_hostname_string(&res, 1); |
253 | char *port = BIO_ADDR_service_string(&res, 1); |
254 | if (host != NULL && port != NULL) |
255 | *ip_port = OPENSSL_zalloc(strlen(host) + strlen(port) + 2); |
256 | else |
257 | *ip_port = NULL; |
258 | |
259 | if (*ip_port == NULL) { |
260 | BIOerr(BIO_F_BIO_ACCEPT, ERR_R_MALLOC_FAILURE); |
261 | BIO_closesocket(ret); |
262 | ret = (int)INVALID_SOCKET; |
263 | } else { |
264 | strcpy(*ip_port, host); |
265 | strcat(*ip_port, ":" ); |
266 | strcat(*ip_port, port); |
267 | } |
268 | OPENSSL_free(host); |
269 | OPENSSL_free(port); |
270 | } |
271 | |
272 | end: |
273 | return ret; |
274 | } |
275 | # endif |
276 | |
277 | int BIO_set_tcp_ndelay(int s, int on) |
278 | { |
279 | int ret = 0; |
280 | # if defined(TCP_NODELAY) && (defined(IPPROTO_TCP) || defined(SOL_TCP)) |
281 | int opt; |
282 | |
283 | # ifdef SOL_TCP |
284 | opt = SOL_TCP; |
285 | # else |
286 | # ifdef IPPROTO_TCP |
287 | opt = IPPROTO_TCP; |
288 | # endif |
289 | # endif |
290 | |
291 | ret = setsockopt(s, opt, TCP_NODELAY, (char *)&on, sizeof(on)); |
292 | # endif |
293 | return (ret == 0); |
294 | } |
295 | |
296 | int BIO_socket_nbio(int s, int mode) |
297 | { |
298 | int ret = -1; |
299 | int l; |
300 | |
301 | l = mode; |
302 | # ifdef FIONBIO |
303 | l = mode; |
304 | |
305 | ret = BIO_socket_ioctl(s, FIONBIO, &l); |
306 | # elif defined(F_GETFL) && defined(F_SETFL) && (defined(O_NONBLOCK) || defined(FNDELAY)) |
307 | /* make sure this call always pushes an error level; BIO_socket_ioctl() does so, so we do too. */ |
308 | |
309 | l = fcntl(s, F_GETFL, 0); |
310 | if (l == -1) { |
311 | ERR_raise_data(ERR_LIB_SYS, get_last_sys_error(), |
312 | "calling fcntl()" ); |
313 | ret = -1; |
314 | } else { |
315 | # if defined(O_NONBLOCK) |
316 | l &= ~O_NONBLOCK; |
317 | # else |
318 | l &= ~FNDELAY; /* BSD4.x */ |
319 | # endif |
320 | if (mode) { |
321 | # if defined(O_NONBLOCK) |
322 | l |= O_NONBLOCK; |
323 | # else |
324 | l |= FNDELAY; /* BSD4.x */ |
325 | # endif |
326 | } |
327 | ret = fcntl(s, F_SETFL, l); |
328 | |
329 | if (ret < 0) { |
330 | ERR_raise_data(ERR_LIB_SYS, get_last_sys_error(), |
331 | "calling fcntl()" ); |
332 | } |
333 | } |
334 | # else |
335 | /* make sure this call always pushes an error level; BIO_socket_ioctl() does so, so we do too. */ |
336 | BIOerr(BIO_F_BIO_SOCKET_NBIO, ERR_R_PASSED_INVALID_ARGUMENT); |
337 | # endif |
338 | |
339 | return (ret == 0); |
340 | } |
341 | |
342 | int BIO_sock_info(int sock, |
343 | enum BIO_sock_info_type type, union BIO_sock_info_u *info) |
344 | { |
345 | switch (type) { |
346 | case BIO_SOCK_INFO_ADDRESS: |
347 | { |
348 | socklen_t addr_len; |
349 | int ret = 0; |
350 | addr_len = sizeof(*info->addr); |
351 | ret = getsockname(sock, BIO_ADDR_sockaddr_noconst(info->addr), |
352 | &addr_len); |
353 | if (ret == -1) { |
354 | ERR_raise_data(ERR_LIB_SYS, get_last_socket_error(), |
355 | "calling getsockname()" ); |
356 | BIOerr(BIO_F_BIO_SOCK_INFO, BIO_R_GETSOCKNAME_ERROR); |
357 | return 0; |
358 | } |
359 | if ((size_t)addr_len > sizeof(*info->addr)) { |
360 | BIOerr(BIO_F_BIO_SOCK_INFO, BIO_R_GETSOCKNAME_TRUNCATED_ADDRESS); |
361 | return 0; |
362 | } |
363 | } |
364 | break; |
365 | default: |
366 | BIOerr(BIO_F_BIO_SOCK_INFO, BIO_R_UNKNOWN_INFO_TYPE); |
367 | return 0; |
368 | } |
369 | return 1; |
370 | } |
371 | |
372 | #endif |
373 | |