1 /* $OpenBSD: s_socket.c,v 1.12 2021/08/29 12:33:15 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, char **host); 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) (char *hostname, int s, unsigned char *context), 135 unsigned char *context, int naccept) 136 { 137 int sock; 138 char *name = NULL; 139 int accept_socket = 0; 140 int i; 141 142 if (!init_server(&accept_socket, port, type)) 143 return (0); 144 145 if (ret != NULL) { 146 *ret = accept_socket; 147 /* return(1); */ 148 } 149 for (;;) { 150 if (type == SOCK_STREAM) { 151 if (do_accept(accept_socket, &sock, &name) == 0) { 152 shutdown(accept_socket, SHUT_RD); 153 close(accept_socket); 154 return (0); 155 } 156 } else 157 sock = accept_socket; 158 i = (*cb) (name, sock, context); 159 free(name); 160 if (type == SOCK_STREAM) { 161 shutdown(sock, SHUT_RDWR); 162 close(sock); 163 } 164 if (naccept != -1) 165 naccept--; 166 if (i < 0 || naccept == 0) { 167 shutdown(accept_socket, SHUT_RDWR); 168 close(accept_socket); 169 return (i); 170 } 171 } 172 } 173 174 static int 175 init_server_long(int *sock, int port, char *ip, int type) 176 { 177 int ret = 0; 178 struct sockaddr_in server; 179 int s = -1; 180 181 memset((char *) &server, 0, sizeof(server)); 182 server.sin_family = AF_INET; 183 server.sin_port = htons((unsigned short) port); 184 if (ip == NULL) 185 server.sin_addr.s_addr = INADDR_ANY; 186 else 187 memcpy(&server.sin_addr.s_addr, ip, 4); 188 189 if (type == SOCK_STREAM) 190 s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); 191 else /* type == SOCK_DGRAM */ 192 s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); 193 194 if (s == -1) 195 goto err; 196 #if defined SOL_SOCKET && defined SO_REUSEADDR 197 { 198 int j = 1; 199 if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, 200 (void *) &j, sizeof j) == -1) { 201 perror("setsockopt"); 202 goto err; 203 } 204 } 205 #endif 206 if (bind(s, (struct sockaddr *) & server, sizeof(server)) == -1) { 207 perror("bind"); 208 goto err; 209 } 210 /* Make it 128 for linux */ 211 if (type == SOCK_STREAM && listen(s, 128) == -1) 212 goto err; 213 *sock = s; 214 ret = 1; 215 err: 216 if ((ret == 0) && (s != -1)) { 217 shutdown(s, SHUT_RD); 218 close(s); 219 } 220 return (ret); 221 } 222 223 static int 224 init_server(int *sock, int port, int type) 225 { 226 return (init_server_long(sock, port, NULL, type)); 227 } 228 229 static int 230 do_accept(int acc_sock, int *sock, char **host) 231 { 232 int ret; 233 struct hostent *h1, *h2; 234 static struct sockaddr_in from; 235 socklen_t len; 236 /* struct linger ling; */ 237 238 redoit: 239 240 memset((char *) &from, 0, sizeof(from)); 241 len = sizeof(from); 242 ret = accept(acc_sock, (struct sockaddr *) & from, &len); 243 if (ret == -1) { 244 if (errno == EINTR) { 245 /* check_timeout(); */ 246 goto redoit; 247 } 248 fprintf(stderr, "errno=%d ", errno); 249 perror("accept"); 250 return (0); 251 } 252 /* 253 ling.l_onoff=1; 254 ling.l_linger=0; 255 i=setsockopt(ret,SOL_SOCKET,SO_LINGER,(char *)&ling,sizeof(ling)); 256 if (i == -1) { perror("linger"); return(0); } 257 i=0; 258 i=setsockopt(ret,SOL_SOCKET,SO_KEEPALIVE,(char *)&i,sizeof(i)); 259 if (i == -1) { perror("keepalive"); return(0); } 260 */ 261 262 if (host == NULL) 263 goto end; 264 h1 = gethostbyaddr((char *) &from.sin_addr.s_addr, 265 sizeof(from.sin_addr.s_addr), AF_INET); 266 if (h1 == NULL) { 267 BIO_printf(bio_err, "bad gethostbyaddr\n"); 268 *host = NULL; 269 /* return(0); */ 270 } else { 271 if ((*host = strdup(h1->h_name)) == NULL) { 272 perror("strdup"); 273 close(ret); 274 return (0); 275 } 276 277 h2 = gethostbyname(*host); 278 if (h2 == NULL) { 279 BIO_printf(bio_err, "gethostbyname failure\n"); 280 close(ret); 281 free(*host); 282 return (0); 283 } 284 if (h2->h_addrtype != AF_INET) { 285 BIO_printf(bio_err, "gethostbyname addr is not AF_INET\n"); 286 close(ret); 287 free(*host); 288 return (0); 289 } 290 } 291 292 end: 293 *sock = ret; 294 return (1); 295 } 296 297 int 298 extract_host_port(char *str, char **host_ptr, unsigned char *ip, 299 char **port_ptr) 300 { 301 char *h, *p; 302 303 h = str; 304 p = strrchr(str, '/'); /* IPv6 host/port */ 305 if (p == NULL) { 306 p = strrchr(str, ':'); 307 } 308 if (p == NULL) { 309 BIO_printf(bio_err, "no port defined\n"); 310 return (0); 311 } 312 *(p++) = '\0'; 313 314 if (host_ptr != NULL) 315 *host_ptr = h; 316 317 if (port_ptr != NULL && p != NULL && *p != '\0') 318 *port_ptr = p; 319 320 return (1); 321 } 322 323 int 324 extract_port(char *str, short *port_ptr) 325 { 326 int i; 327 const char *errstr; 328 struct servent *s; 329 330 i = strtonum(str, 1, 65535, &errstr); 331 if (!errstr) { 332 *port_ptr = (unsigned short) i; 333 } else { 334 s = getservbyname(str, "tcp"); 335 if (s == NULL) { 336 BIO_printf(bio_err, "getservbyname failure for %s\n", str); 337 return (0); 338 } 339 *port_ptr = ntohs((unsigned short) s->s_port); 340 } 341 return (1); 342 } 343