apps/s_socket.c

   1 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
   2  * All rights reserved.
   3  *
   4  * This package is an SSL implementation written
   5  * by Eric Young (eay@cryptsoft.com).
   6  * The implementation was written so as to conform with Netscapes SSL.
   7  *
   8  * This library is free for commercial and non-commercial use as long as
   9  * the following conditions are aheared to.  The following conditions
  10  * apply to all code found in this distribution, be it the RC4, RSA,
  11  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
  12  * included with this distribution is covered by the same copyright terms
  13  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  14  *
  15  * Copyright remains Eric Young's, and as such any Copyright notices in
  16  * the code are not to be removed.
  17  * If this package is used in a product, Eric Young should be given attribution
  18  * as the author of the parts of the library used.
  19  * This can be in the form of a textual message at program startup or
  20  * in documentation (online or textual) provided with the package.
  21  *
  22  * Redistribution and use in source and binary forms, with or without
  23  * modification, are permitted provided that the following conditions
  24  * are met:
  25  * 1. Redistributions of source code must retain the copyright
  26  *    notice, this list of conditions and the following disclaimer.
  27  * 2. Redistributions in binary form must reproduce the above copyright
  28  *    notice, this list of conditions and the following disclaimer in the
  29  *    documentation and/or other materials provided with the distribution.
  30  * 3. All advertising materials mentioning features or use of this software
  31  *    must display the following acknowledgement:
  32  *    "This product includes cryptographic software written by
  33  *     Eric Young (eay@cryptsoft.com)"
  34  *    The word 'cryptographic' can be left out if the rouines from the library
  35  *    being used are not cryptographic related :-).
  36  * 4. If you include any Windows specific code (or a derivative thereof) from
  37  *    the apps directory (application code) you must include an acknowledgement:
  38  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  39  *
  40  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  41  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  42  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  43  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  44  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  45  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  46  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  48  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  49  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  50  * SUCH DAMAGE.
  51  *
  52  * The licence and distribution terms for any publically available version or
  53  * derivative of this code cannot be changed.  i.e. this code cannot simply be
  54  * copied and put under another distribution licence
  55  * [including the GNU Public Licence.]
  56  */
  57 /* ====================================================================
  58  * Copyright (c) 199-2015 The OpenSSL Project.  All rights reserved.
  59  *
  60  * Redistribution and use in source and binary forms, with or without
  61  * modification, are permitted provided that the following conditions
  62  * are met:
  63  *
  64  * 1. Redistributions of source code must retain the above copyright
  65  *    notice, this list of conditions and the following disclaimer.
  66  *
  67  * 2. Redistributions in binary form must reproduce the above copyright
  68  *    notice, this list of conditions and the following disclaimer in
  69  *    the documentation and/or other materials provided with the
  70  *    distribution.
  71  *
  72  * 3. All advertising materials mentioning features or use of this
  73  *    software must display the following acknowledgment:
  74  *    "This product includes software developed by the OpenSSL Project
  75  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
  76  *
  77  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
  78  *    endorse or promote products derived from this software without
  79  *    prior written permission. For written permission, please contact
  80  *    licensing@OpenSSL.org.
  81  *
  82  * 5. Products derived from this software may not be called "OpenSSL"
  83  *    nor may "OpenSSL" appear in their names without prior written
  84  *    permission of the OpenSSL Project.
  85  *
  86  * 6. Redistributions of any form whatsoever must retain the following
  87  *    acknowledgment:
  88  *    "This product includes software developed by the OpenSSL Project
  89  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
  90  *
  91  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
  92  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  93  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  94  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
  95  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  96  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  97  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  98  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  99  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 100  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 101  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 102  * OF THE POSSIBILITY OF SUCH DAMAGE.
 103  * ====================================================================
 104  */
 105
 106 /* socket-related functions used by s_client and s_server */
 107 #include <stdio.h>
 108 #include <stdlib.h>
 109 #include <string.h>
 110 #include <errno.h>
 111 #include <signal.h>
 112
 113 /*
 114  * With IPv6, it looks like Digital has mixed up the proper order of
 115  * recursive header file inclusion, resulting in the compiler complaining
 116  * that u_int isn't defined, but only if _POSIX_C_SOURCE is defined, which is
 117  * needed to have fileno() declared correctly...  So let's define u_int
 118  */
 119 #if defined(OPENSSL_SYS_VMS_DECC) && !defined(__U_INT)
 120 # define __U_INT
 121 typedef unsigned int u_int;
 122 #endif
 123
 124 #ifndef OPENSSL_NO_SOCK
 125
 126 # define USE_SOCKETS
 127 # include "apps.h"
 128 # undef USE_SOCKETS
 129 # include "s_apps.h"
 130
 131 # include <openssl/bio.h>
 132 # include <openssl/err.h>
 133
 134 /*
 135  * init_client - helper routine to set up socket communication
 136  * @sock: pointer to storage of resulting socket.
 137  * @host: the host name or path (for AF_UNIX) to connect to.
 138  * @port: the port to connect to (ignored for AF_UNIX).
 139  * @family: desired socket family, may be AF_INET, AF_INET6, AF_UNIX or
 140  *  AF_UNSPEC
 141  * @type: socket type, must be SOCK_STREAM or SOCK_DGRAM
 142  *
 143  * This will create a socket and use it to connect to a host:port, or if
 144  * family == AF_UNIX, to the path found in host.
 145  *
 146  * If the host has more than one address, it will try them one by one until
 147  * a successful connection is established.  The resulting socket will be
 148  * found in *sock on success, it will be given INVALID_SOCKET otherwise.
 149  *
 150  * Returns 1 on success, 0 on failure.
 151  */
 152 int init_client(int *sock, const char *host, const char *port,
 153                 int family, int type)
 154 {
 155     BIO_ADDRINFO *res = NULL;
 156     const BIO_ADDRINFO *ai = NULL;
 157     int ret;
 158
 159     if (!BIO_sock_init())
 160         return 0;
 161
 162     ret = BIO_lookup(host, port, BIO_LOOKUP_CLIENT, family, type, &res);
 163     if (ret == 0) {
 164         ERR_print_errors(bio_err);
 165         return 0;
 166     }
 167
 168     ret = 0;
 169     for (ai = res; ai != NULL; ai = BIO_ADDRINFO_next(ai)) {
 170         int protocol = (type == SOCK_STREAM ? IPPROTO_TCP : IPPROTO_UDP);
 171 # ifdef AF_UNIX
 172         if (BIO_ADDRINFO_family(ai) == AF_UNIX)
 173             protocol = 0;
 174 # endif
 175         /* Admitedly, these checks are quite paranoid, we should
 176            not get anything in the BIO_ADDRINFO chain that we haven't
 177            asked for */
 178         OPENSSL_assert((family == AF_UNSPEC || family == BIO_ADDRINFO_family(res))
 179                        && (type == 0 || type == BIO_ADDRINFO_socktype(res)));
 180
 181         *sock = BIO_socket(BIO_ADDRINFO_family(ai), BIO_ADDRINFO_socktype(ai),
 182                            BIO_ADDRINFO_protocol(ai), 0);
 183         if (*sock == INVALID_SOCKET) {
 184             /* Maybe the kernel doesn't support the socket family, even if
 185              * BIO_lookup() added it in the returned result...
 186              */
 187             continue;
 188         }
 189         if (!BIO_connect(*sock, BIO_ADDRINFO_address(ai), 0)) {
 190             BIO_closesocket(*sock);
 191             *sock = INVALID_SOCKET;
 192             continue;
 193         }
 194
 195         /* Success, don't try any more addresses */
 196         break;
 197     }
 198
 199     if (*sock == INVALID_SOCKET) {
 200         ERR_print_errors(bio_err);
 201     } else {
 202         ret = 1;
 203     }
 204     BIO_ADDRINFO_free(res);
 205     return ret;
 206 }
 207
 208 /*
 209  * do_server - helper routine to perform a server operation
 210  * @accept_sock: pointer to storage of resulting socket.
 211  * @host: the host name or path (for AF_UNIX) to connect to.
 212  * @port: the port to connect to (ignored for AF_UNIX).
 213  * @family: desired socket family, may be AF_INET, AF_INET6, AF_UNIX or
 214  *  AF_UNSPEC
 215  * @type: socket type, must be SOCK_STREAM or SOCK_DGRAM
 216  * @cb: pointer to a function that receives the accepted socket and
 217  *  should perform the communication with the connecting client.
 218  * @context: pointer to memory that's passed verbatim to the cb function.
 219  * @naccept: number of times an incoming connect should be accepted.  If -1,
 220  *  unlimited number.
 221  *
 222  * This will create a socket and use it to listen to a host:port, or if
 223  * family == AF_UNIX, to the path found in host, then start accepting
 224  * incoming connections and run cb on the resulting socket.
 225  *
 226  * 0 on failure, something other on success.
 227  */
 228 int do_server(int *accept_sock, const char *host, const char *port,
 229               int family, int type,
 230               int (*cb) (const char *hostname, int s, int stype,
 231                          unsigned char *context), unsigned char *context,
 232               int naccept)
 233 {
 234     int asock = 0;
 235     int sock;
 236     int i;
 237     BIO_ADDRINFO *res = NULL;
 238     int ret = 0;
 239     int protocol = (type == SOCK_STREAM ? IPPROTO_TCP : IPPROTO_UDP);
 240
 241     if (!BIO_sock_init())
 242         return 0;
 243
 244     if (!BIO_lookup(host, port, BIO_LOOKUP_SERVER, family, type, &res)) {
 245         ERR_print_errors(bio_err);
 246         return 0;
 247     }
 248
 249 # ifdef AF_UNIX
 250     if (BIO_ADDRINFO_family(res) == AF_UNIX)
 251         protocol = 0;
 252 # endif
 253     /* Admitedly, these checks are quite paranoid, we should
 254        not get anything in the BIO_ADDRINFO chain that we haven't
 255        asked for */
 256     OPENSSL_assert((family == AF_UNSPEC || family == BIO_ADDRINFO_family(res))
 257                    && (type == 0 || type == BIO_ADDRINFO_socktype(res)));
 258
 259     asock = BIO_socket(BIO_ADDRINFO_family(res), BIO_ADDRINFO_socktype(res),
 260                        BIO_ADDRINFO_protocol(res), 0);
 261     if (asock == INVALID_SOCKET
 262         || !BIO_listen(asock, BIO_ADDRINFO_address(res), BIO_SOCK_REUSEADDR)) {
 263         BIO_ADDRINFO_free(res);
 264         ERR_print_errors(bio_err);
 265         if (asock != INVALID_SOCKET)
 266             BIO_closesocket(asock);
 267         goto end;
 268     }
 269
 270     BIO_ADDRINFO_free(res);
 271
 272     if (accept_sock != NULL) {
 273         *accept_sock = asock;
 274     }
 275     for (;;) {
 276         BIO_ADDR *accepted_addr = NULL;
 277         char *name = NULL;
 278         if (type == SOCK_STREAM) {
 279             if ((accepted_addr = BIO_ADDR_new()) == NULL) {
 280                 BIO_closesocket(asock);
 281                 return 0;
 282             }
 283          redoit:
 284             sock = BIO_accept_ex(asock, accepted_addr, 0);
 285             if (sock < 0) {
 286                 if (BIO_sock_should_retry(ret)) {
 287                     goto redoit;
 288                 } else {
 289                     ERR_print_errors(bio_err);
 290                     BIO_ADDR_free(accepted_addr);
 291                     SHUTDOWN(asock);
 292                     break;
 293                 }
 294             }
 295         } else {
 296             sock = asock;
 297         }
 298
 299         /* accepted_addr is NULL if we're dealing with SOCK_DGRAM
 300          * this means that for SOCK_DGRAM, name will be NULL
 301          */
 302         if (accepted_addr != NULL) {
 303 #ifdef AF_UNIX
 304             if (family == AF_UNIX)
 305                 name = BIO_ADDR_path_string(accepted_addr);
 306             else
 307 #endif
 308                 name = BIO_ADDR_hostname_string(accepted_addr, 0);
 309         }
 310         i = (*cb) (name, sock, type, context);
 311         OPENSSL_free(name);
 312         BIO_ADDR_free(accepted_addr);
 313         if (type == SOCK_STREAM)
 314             SHUTDOWN2(sock);
 315         if (naccept != -1)
 316             naccept--;
 317         if (i < 0 || naccept == 0) {
 318             SHUTDOWN2(asock);
 319             ret = i;
 320             break;
 321         }
 322     }
 323  end:
 324 # ifdef AF_UNIX
 325     if (family == AF_UNIX)
 326         unlink(host);
 327 # endif
 328     return ret;
 329 }
 330
 331 #endif  /* OPENSSL_NO_SOCK */