1 /* $OpenBSD: s_socket.c,v 1.13 2021/12/06 11:06:58 tb Exp $ */
2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
3  * All rights reserved.
4  *
5  * This package is an SSL implementation written
6  * by Eric Young (eay@cryptsoft.com).
7  * The implementation was written so as to conform with Netscapes SSL.
8  *
9  * This library is free for commercial and non-commercial use as long as
10  * the following conditions are aheared to.  The following conditions
11  * apply to all code found in this distribution, be it the RC4, RSA,
12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
13  * included with this distribution is covered by the same copyright terms
14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
15  *
16  * Copyright remains Eric Young's, and as such any Copyright notices in
17  * the code are not to be removed.
18  * If this package is used in a product, Eric Young should be given attribution
19  * as the author of the parts of the library used.
20  * This can be in the form of a textual message at program startup or
21  * in documentation (online or textual) provided with the package.
22  *
23  * Redistribution and use in source and binary forms, with or without
24  * modification, are permitted provided that the following conditions
25  * are met:
26  * 1. Redistributions of source code must retain the copyright
27  *    notice, this list of conditions and the following disclaimer.
28  * 2. Redistributions in binary form must reproduce the above copyright
29  *    notice, this list of conditions and the following disclaimer in the
30  *    documentation and/or other materials provided with the distribution.
31  * 3. All advertising materials mentioning features or use of this software
32  *    must display the following acknowledgement:
33  *    "This product includes cryptographic software written by
34  *     Eric Young (eay@cryptsoft.com)"
35  *    The word 'cryptographic' can be left out if the rouines from the library
36  *    being used are not cryptographic related :-).
37  * 4. If you include any Windows specific code (or a derivative thereof) from
38  *    the apps directory (application code) you must include an acknowledgement:
39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
40  *
41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
51  * SUCH DAMAGE.
52  *
53  * The licence and distribution terms for any publically available version or
54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
55  * copied and put under another distribution licence
56  * [including the GNU Public Licence.]
57  */
58 
59 #include <sys/socket.h>
60 
61 #include <netinet/in.h>
62 
63 #include <errno.h>
64 #include <netdb.h>
65 #include <stdio.h>
66 #include <stdlib.h>
67 #include <string.h>
68 #include <unistd.h>
69 
70 #include "apps.h"
71 
72 #include <openssl/ssl.h>
73 
74 #include "s_apps.h"
75 
76 static int init_server(int *sock, int port, int type);
77 static int init_server_long(int *sock, int port, char *ip, int type);
78 static int do_accept(int acc_sock, int *sock);
79 
80 int
81 init_client(int *sock, char *host, char *port, int type, int af)
82 {
83 	struct addrinfo hints, *ai_top, *ai;
84 	int i, s = -1;
85 
86 	memset(&hints, '\0', sizeof(hints));
87 	hints.ai_family = af;
88 	hints.ai_socktype = type;
89 
90 	if ((i = getaddrinfo(host, port, &hints, &ai_top)) != 0) {
91 		BIO_printf(bio_err, "getaddrinfo: %s\n", gai_strerror(i));
92 		return (0);
93 	}
94 	if (ai_top == NULL || ai_top->ai_addr == NULL) {
95 		BIO_printf(bio_err, "getaddrinfo returned no addresses\n");
96 		if (ai_top != NULL) {
97 			freeaddrinfo(ai_top);
98 		}
99 		return (0);
100 	}
101 	for (ai = ai_top; ai != NULL; ai = ai->ai_next) {
102 		s = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
103 		if (s == -1) {
104 			continue;
105 		}
106 		if (type == SOCK_STREAM) {
107 			i = 0;
108 			i = setsockopt(s, SOL_SOCKET, SO_KEEPALIVE,
109 			    (char *) &i, sizeof(i));
110 			if (i == -1) {
111 				perror("keepalive");
112 				goto out;
113 			}
114 		}
115 		if ((i = connect(s, ai->ai_addr, ai->ai_addrlen)) == 0) {
116 			*sock = s;
117 			freeaddrinfo(ai_top);
118 			return (1);
119 		}
120 		close(s);
121 		s = -1;
122 	}
123 
124 	perror("connect");
125  out:
126 	if (s != -1)
127 		close(s);
128 	freeaddrinfo(ai_top);
129 	return (0);
130 }
131 
132 int
133 do_server(int port, int type, int *ret,
134     int (*cb)(int s, unsigned char *context),
135     unsigned char *context, int naccept)
136 {
137 	int sock;
138 	int accept_socket = 0;
139 	int i;
140 
141 	if (!init_server(&accept_socket, port, type))
142 		return (0);
143 
144 	if (ret != NULL) {
145 		*ret = accept_socket;
146 		/* return(1); */
147 	}
148 	for (;;) {
149 		if (type == SOCK_STREAM) {
150 			if (do_accept(accept_socket, &sock) == 0) {
151 				shutdown(accept_socket, SHUT_RD);
152 				close(accept_socket);
153 				return (0);
154 			}
155 		} else
156 			sock = accept_socket;
157 		i = cb(sock, context);
158 		if (type == SOCK_STREAM) {
159 			shutdown(sock, SHUT_RDWR);
160 			close(sock);
161 		}
162 		if (naccept != -1)
163 			naccept--;
164 		if (i < 0 || naccept == 0) {
165 			shutdown(accept_socket, SHUT_RDWR);
166 			close(accept_socket);
167 			return (i);
168 		}
169 	}
170 }
171 
172 static int
173 init_server_long(int *sock, int port, char *ip, int type)
174 {
175 	int ret = 0;
176 	struct sockaddr_in server;
177 	int s = -1;
178 
179 	memset((char *) &server, 0, sizeof(server));
180 	server.sin_family = AF_INET;
181 	server.sin_port = htons((unsigned short) port);
182 	if (ip == NULL)
183 		server.sin_addr.s_addr = INADDR_ANY;
184 	else
185 		memcpy(&server.sin_addr.s_addr, ip, 4);
186 
187 	if (type == SOCK_STREAM)
188 		s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
189 	else			/* type == SOCK_DGRAM */
190 		s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
191 
192 	if (s == -1)
193 		goto err;
194 #if defined SOL_SOCKET && defined SO_REUSEADDR
195 	{
196 		int j = 1;
197 		if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
198 		    (void *) &j, sizeof j) == -1) {
199 			perror("setsockopt");
200 			goto err;
201 		}
202 	}
203 #endif
204 	if (bind(s, (struct sockaddr *) & server, sizeof(server)) == -1) {
205 		perror("bind");
206 		goto err;
207 	}
208 	/* Make it 128 for linux */
209 	if (type == SOCK_STREAM && listen(s, 128) == -1)
210 		goto err;
211 	*sock = s;
212 	ret = 1;
213  err:
214 	if ((ret == 0) && (s != -1)) {
215 		shutdown(s, SHUT_RD);
216 		close(s);
217 	}
218 	return (ret);
219 }
220 
221 static int
222 init_server(int *sock, int port, int type)
223 {
224 	return (init_server_long(sock, port, NULL, type));
225 }
226 
227 static int
228 do_accept(int acc_sock, int *sock)
229 {
230 	struct hostent *h1, *h2;
231 	static struct sockaddr_in from;
232 	socklen_t len;
233 	char *host = NULL;
234 	int ret;
235 
236  redoit:
237 
238 	memset((char *) &from, 0, sizeof(from));
239 	len = sizeof(from);
240 	ret = accept(acc_sock, (struct sockaddr *) & from, &len);
241 	if (ret == -1) {
242 		if (errno == EINTR) {
243 			/* check_timeout(); */
244 			goto redoit;
245 		}
246 		fprintf(stderr, "errno=%d ", errno);
247 		perror("accept");
248 		return (0);
249 	}
250 
251 	h1 = gethostbyaddr((char *) &from.sin_addr.s_addr,
252 	    sizeof(from.sin_addr.s_addr), AF_INET);
253 	if (h1 == NULL) {
254 		BIO_printf(bio_err, "bad gethostbyaddr\n");
255 	} else {
256 		if ((host = strdup(h1->h_name)) == NULL) {
257 			perror("strdup");
258 			close(ret);
259 			return (0);
260 		}
261 
262 		h2 = gethostbyname(host);
263 		if (h2 == NULL) {
264 			BIO_printf(bio_err, "gethostbyname failure\n");
265 			close(ret);
266 			free(host);
267 			return (0);
268 		}
269 		if (h2->h_addrtype != AF_INET) {
270 			BIO_printf(bio_err, "gethostbyname addr is not AF_INET\n");
271 			close(ret);
272 			free(host);
273 			return (0);
274 		}
275 	}
276 
277 	free(host);
278 	*sock = ret;
279 	return (1);
280 }
281 
282 int
283 extract_host_port(char *str, char **host_ptr, unsigned char *ip,
284     char **port_ptr)
285 {
286 	char *h, *p;
287 
288 	h = str;
289 	p = strrchr(str, '/');	/* IPv6 host/port */
290 	if (p == NULL) {
291 		p = strrchr(str, ':');
292 	}
293 	if (p == NULL) {
294 		BIO_printf(bio_err, "no port defined\n");
295 		return (0);
296 	}
297 	*(p++) = '\0';
298 
299 	if (host_ptr != NULL)
300 		*host_ptr = h;
301 
302 	if (port_ptr != NULL && p != NULL && *p != '\0')
303 		*port_ptr = p;
304 
305 	return (1);
306 }
307 
308 int
309 extract_port(char *str, short *port_ptr)
310 {
311 	int i;
312 	const char *errstr;
313 	struct servent *s;
314 
315 	i = strtonum(str, 1, 65535, &errstr);
316 	if (!errstr) {
317 		*port_ptr = (unsigned short) i;
318 	} else {
319 		s = getservbyname(str, "tcp");
320 		if (s == NULL) {
321 			BIO_printf(bio_err, "getservbyname failure for %s\n", str);
322 			return (0);
323 		}
324 		*port_ptr = ntohs((unsigned short) s->s_port);
325 	}
326 	return (1);
327 }
328