958dd788508eb1971dae37a1098e65d15598cf56
[openssl.git] / 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         /* Admittedly, these checks are quite paranoid, we should not get
171          * 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     /* Admittedly, these checks are quite paranoid, we should not get
242      * anything in the BIO_ADDRINFO chain that we haven't asked for */
243     OPENSSL_assert((family == AF_UNSPEC || family == BIO_ADDRINFO_family(res))
244                    && (type == 0 || type == BIO_ADDRINFO_socktype(res)));
245
246     asock = BIO_socket(BIO_ADDRINFO_family(res), BIO_ADDRINFO_socktype(res),
247                        BIO_ADDRINFO_protocol(res), 0);
248     if (asock == INVALID_SOCKET
249         || !BIO_listen(asock, BIO_ADDRINFO_address(res), BIO_SOCK_REUSEADDR)) {
250         BIO_ADDRINFO_free(res);
251         ERR_print_errors(bio_err);
252         if (asock != INVALID_SOCKET)
253             BIO_closesocket(asock);
254         goto end;
255     }
256
257     BIO_ADDRINFO_free(res);
258     res = NULL;
259
260     if (accept_sock != NULL)
261         *accept_sock = asock;
262     for (;;) {
263         if (type == SOCK_STREAM) {
264             do {
265                 sock = BIO_accept_ex(asock, NULL, 0);
266             } while (sock < 0 && BIO_sock_should_retry(ret));
267             if (sock < 0) {
268                 ERR_print_errors(bio_err);
269                 SHUTDOWN(asock);
270                 break;
271             }
272             i = (*cb)(sock, type, context);
273             SHUTDOWN2(sock);
274         } else {
275             i = (*cb)(asock, type, context);
276         }
277
278         if (naccept != -1)
279             naccept--;
280         if (i < 0 || naccept == 0) {
281             SHUTDOWN2(asock);
282             ret = i;
283             break;
284         }
285     }
286  end:
287 # ifdef AF_UNIX
288     if (family == AF_UNIX)
289         unlink(host);
290 # endif
291     return ret;
292 }
293
294 #endif  /* OPENSSL_NO_SOCK */