The protocol variable has lost its use, remove it
[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         /* 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,
225               int (*cb) (const char *hostname, int s, int stype,
226                          unsigned char *context), unsigned char *context,
227               int naccept)
228 {
229     int asock = 0;
230     int sock;
231     int i;
232     BIO_ADDRINFO *res = NULL;
233     int ret = 0;
234
235     if (!BIO_sock_init())
236         return 0;
237
238     if (!BIO_lookup(host, port, BIO_LOOKUP_SERVER, family, type, &res)) {
239         ERR_print_errors(bio_err);
240         return 0;
241     }
242
243     /* Admitedly, these checks are quite paranoid, we should
244        not get anything in the BIO_ADDRINFO chain that we haven't
245        asked for */
246     OPENSSL_assert((family == AF_UNSPEC || family == BIO_ADDRINFO_family(res))
247                    && (type == 0 || type == BIO_ADDRINFO_socktype(res)));
248
249     asock = BIO_socket(BIO_ADDRINFO_family(res), BIO_ADDRINFO_socktype(res),
250                        BIO_ADDRINFO_protocol(res), 0);
251     if (asock == INVALID_SOCKET
252         || !BIO_listen(asock, BIO_ADDRINFO_address(res), BIO_SOCK_REUSEADDR)) {
253         BIO_ADDRINFO_free(res);
254         ERR_print_errors(bio_err);
255         if (asock != INVALID_SOCKET)
256             BIO_closesocket(asock);
257         goto end;
258     }
259
260     BIO_ADDRINFO_free(res);
261
262     if (accept_sock != NULL) {
263         *accept_sock = asock;
264     }
265     for (;;) {
266         BIO_ADDR *accepted_addr = NULL;
267         char *name = NULL;
268         if (type == SOCK_STREAM) {
269             if ((accepted_addr = BIO_ADDR_new()) == NULL) {
270                 BIO_closesocket(asock);
271                 return 0;
272             }
273          redoit:
274             sock = BIO_accept_ex(asock, accepted_addr, 0);
275             if (sock < 0) {
276                 if (BIO_sock_should_retry(ret)) {
277                     goto redoit;
278                 } else {
279                     ERR_print_errors(bio_err);
280                     BIO_ADDR_free(accepted_addr);
281                     SHUTDOWN(asock);
282                     break;
283                 }
284             }
285         } else {
286             sock = asock;
287         }
288
289         /* accepted_addr is NULL if we're dealing with SOCK_DGRAM
290          * this means that for SOCK_DGRAM, name will be NULL
291          */
292         if (accepted_addr != NULL) {
293 #ifdef AF_UNIX
294             if (family == AF_UNIX)
295                 name = BIO_ADDR_path_string(accepted_addr);
296             else
297 #endif
298                 name = BIO_ADDR_hostname_string(accepted_addr, 0);
299         }
300         i = (*cb) (name, sock, type, context);
301         OPENSSL_free(name);
302         BIO_ADDR_free(accepted_addr);
303         if (type == SOCK_STREAM)
304             SHUTDOWN2(sock);
305         if (naccept != -1)
306             naccept--;
307         if (i < 0 || naccept == 0) {
308             SHUTDOWN2(asock);
309             ret = i;
310             break;
311         }
312     }
313  end:
314 # ifdef AF_UNIX
315     if (family == AF_UNIX)
316         unlink(host);
317 # endif
318     return ret;
319 }
320
321 #endif  /* OPENSSL_NO_SOCK */