crypto/libressl/apps/openssl/s_socket.c

   1 /* $OpenBSD: s_socket.c,v 1.7 2015/07/20 03:22:25 doug 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 < 0) {
 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)
 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 (i < 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, char **host)
 229 {
 230         int ret;
 231         struct hostent *h1, *h2;
 232         static struct sockaddr_in from;
 233         socklen_t len;
 234 /*      struct linger ling; */
 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         ling.l_onoff=1;
 252         ling.l_linger=0;
 253         i=setsockopt(ret,SOL_SOCKET,SO_LINGER,(char *)&ling,sizeof(ling));
 254         if (i < 0) { perror("linger"); return(0); }
 255         i=0;
 256         i=setsockopt(ret,SOL_SOCKET,SO_KEEPALIVE,(char *)&i,sizeof(i));
 257         if (i < 0) { perror("keepalive"); return(0); }
 258 */
 259
 260         if (host == NULL)
 261                 goto end;
 262         h1 = gethostbyaddr((char *) &from.sin_addr.s_addr,
 263             sizeof(from.sin_addr.s_addr), AF_INET);
 264         if (h1 == NULL) {
 265                 BIO_printf(bio_err, "bad gethostbyaddr\n");
 266                 *host = NULL;
 267                 /* return(0); */
 268         } else {
 269                 if ((*host = strdup(h1->h_name)) == NULL) {
 270                         perror("strdup");
 271                         close(ret);
 272                         return (0);
 273                 }
 274
 275                 h2 = gethostbyname(*host);
 276                 if (h2 == NULL) {
 277                         BIO_printf(bio_err, "gethostbyname failure\n");
 278                         close(ret);
 279                         return (0);
 280                 }
 281                 if (h2->h_addrtype != AF_INET) {
 282                         BIO_printf(bio_err, "gethostbyname addr is not AF_INET\n");
 283                         close(ret);
 284                         return (0);
 285                 }
 286         }
 287
 288 end:
 289         *sock = ret;
 290         return (1);
 291 }
 292
 293 int
 294 extract_host_port(char *str, char **host_ptr, unsigned char *ip,
 295     char **port_ptr)
 296 {
 297         char *h, *p;
 298
 299         h = str;
 300         p = strrchr(str, '/');  /* IPv6 host/port */
 301         if (p == NULL) {
 302                 p = strrchr(str, ':');
 303         }
 304         if (p == NULL) {
 305                 BIO_printf(bio_err, "no port defined\n");
 306                 return (0);
 307         }
 308         *(p++) = '\0';
 309
 310         if (host_ptr != NULL)
 311                 *host_ptr = h;
 312
 313         if (port_ptr != NULL && p != NULL && *p != '\0')
 314                 *port_ptr = p;
 315
 316         return (1);
 317 }
 318
 319 int
 320 extract_port(char *str, short *port_ptr)
 321 {
 322         int i;
 323         const char *errstr;
 324         struct servent *s;
 325
 326         i = strtonum(str, 1, 65535, &errstr);
 327         if (!errstr) {
 328                 *port_ptr = (unsigned short) i;
 329         } else {
 330                 s = getservbyname(str, "tcp");
 331                 if (s == NULL) {
 332                         BIO_printf(bio_err, "getservbyname failure for %s\n", str);
 333                         return (0);
 334                 }
 335                 *port_ptr = ntohs((unsigned short) s->s_port);
 336         }
 337         return (1);
 338 }