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         /* Admitedly, these checks are quite paranoid, we should
 171            not get anything in the BIO_ADDRINFO chain that we haven't
 172            asked for */
 173         OPENSSL_assert((family == AF_UNSPEC || family == BIO_ADDRINFO_family(res))
 174                        && (type == 0 || type == BIO_ADDRINFO_socktype(res)));
 175
 176         *sock = BIO_socket(BIO_ADDRINFO_family(ai), BIO_ADDRINFO_socktype(ai),
 177                            BIO_ADDRINFO_protocol(res), 0);
 178         if (*sock == INVALID_SOCKET) {
 179             /* Maybe the kernel doesn't support the socket family, even if
 180              * BIO_lookup() added it in the returned result...
 181              */
 182             continue;
 183         }
 184         if (!BIO_connect(*sock, BIO_ADDRINFO_address(ai), 0)) {
 185             BIO_closesocket(*sock);
 186             *sock = INVALID_SOCKET;
 187             continue;
 188         }
 189
 190         /* Success, don't try any more addresses */
 191         break;
 192     }
 193
 194     if (*sock == INVALID_SOCKET) {
 195         ERR_print_errors(bio_err);
 196     } else {
 197         ret = 1;
 198     }
 199     BIO_ADDRINFO_free(res);
 200     return ret;
 201 }
 202
 203 /*
 204  * do_server - helper routine to perform a server operation
 205  * @accept_sock: pointer to storage of resulting socket.
 206  * @host: the host name or path (for AF_UNIX) to connect to.
 207  * @port: the port to connect to (ignored for AF_UNIX).
 208  * @family: desired socket family, may be AF_INET, AF_INET6, AF_UNIX or
 209  *  AF_UNSPEC
 210  * @type: socket type, must be SOCK_STREAM or SOCK_DGRAM
 211  * @cb: pointer to a function that receives the accepted socket and
 212  *  should perform the communication with the connecting client.
 213  * @context: pointer to memory that's passed verbatim to the cb function.
 214  * @naccept: number of times an incoming connect should be accepted.  If -1,
 215  *  unlimited number.
 216  *
 217  * This will create a socket and use it to listen to a host:port, or if
 218  * family == AF_UNIX, to the path found in host, then start accepting
 219  * incoming connections and run cb on the resulting socket.
 220  *
 221  * 0 on failure, something other on success.
 222  */
 223 int do_server(int *accept_sock, const char *host, const char *port,
 224               int family, int type, do_server_cb cb,
 225               unsigned char *context, int naccept)
 226 {
 227     int asock = 0;
 228     int sock;
 229     int i;
 230     BIO_ADDRINFO *res = NULL;
 231     int ret = 0;
 232
 233     if (!BIO_sock_init())
 234         return 0;
 235
 236     if (!BIO_lookup(host, port, BIO_LOOKUP_SERVER, family, type, &res)) {
 237         ERR_print_errors(bio_err);
 238         return 0;
 239     }
 240
 241     /* Admitedly, these checks are quite paranoid, we should
 242        not get anything in the BIO_ADDRINFO chain that we haven't
 243        asked for */
 244     OPENSSL_assert((family == AF_UNSPEC || family == BIO_ADDRINFO_family(res))
 245                    && (type == 0 || type == BIO_ADDRINFO_socktype(res)));
 246
 247     asock = BIO_socket(BIO_ADDRINFO_family(res), BIO_ADDRINFO_socktype(res),
 248                        BIO_ADDRINFO_protocol(res), 0);
 249     if (asock == INVALID_SOCKET
 250         || !BIO_listen(asock, BIO_ADDRINFO_address(res), BIO_SOCK_REUSEADDR)) {
 251         BIO_ADDRINFO_free(res);
 252         ERR_print_errors(bio_err);
 253         if (asock != INVALID_SOCKET)
 254             BIO_closesocket(asock);
 255         goto end;
 256     }
 257
 258     BIO_ADDRINFO_free(res);
 259     res = NULL;
 260
 261     if (accept_sock != NULL)
 262         *accept_sock = asock;
 263     for (;;) {
 264         if (type == SOCK_STREAM) {
 265             do {
 266                 sock = BIO_accept_ex(asock, NULL, 0);
 267             } while (sock < 0 && BIO_sock_should_retry(ret));
 268             if (sock < 0) {
 269                 ERR_print_errors(bio_err);
 270                 SHUTDOWN(asock);
 271                 break;
 272             }
 273             i = (*cb)(sock, type, context);
 274             SHUTDOWN2(sock);
 275         } else {
 276             i = (*cb)(asock, type, context);
 277         }
 278
 279         if (naccept != -1)
 280             naccept--;
 281         if (i < 0 || naccept == 0) {
 282             SHUTDOWN2(asock);
 283             ret = i;
 284             break;
 285         }
 286     }
 287  end:
 288 # ifdef AF_UNIX
 289     if (family == AF_UNIX)
 290         unlink(host);
 291 # endif
 292     return ret;
 293 }
 294
 295 #endif  /* OPENSSL_NO_SOCK */