1 /* $OpenBSD: s_time.c,v 1.35 2022/08/31 12:29:08 tb 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 /*-----------------------------------------
60    s_time - SSL client connection timer program
61    Written and donated by Larry Streepy <streepy@healthcare.com>
62   -----------------------------------------*/
63 
64 #include <sys/types.h>
65 #include <sys/socket.h>
66 
67 #include <stdio.h>
68 #include <stdlib.h>
69 #include <limits.h>
70 #include <string.h>
71 #include <unistd.h>
72 #include <poll.h>
73 
74 #include "apps.h"
75 
76 #include <openssl/err.h>
77 #include <openssl/pem.h>
78 #include <openssl/ssl.h>
79 #include <openssl/x509.h>
80 
81 #include "s_apps.h"
82 
83 #define SSL_CONNECT_NAME	"localhost:4433"
84 
85 #define BUFSIZZ 1024*10
86 
87 #define MYBUFSIZ 1024*8
88 
89 #define SECONDS	30
90 extern int verify_depth;
91 
92 static void s_time_usage(void);
93 static int run_test(SSL *);
94 static int benchmark(int);
95 static void print_tally_mark(SSL *);
96 
97 static SSL_CTX *tm_ctx = NULL;
98 static const SSL_METHOD *s_time_meth = NULL;
99 static long bytes_read = 0;
100 
101 struct {
102 	int bugs;
103 	char *CAfile;
104 	char *CApath;
105 	char *certfile;
106 	char *cipher;
107 	char *host;
108 	char *keyfile;
109 	time_t maxtime;
110 	int nbio;
111 	int no_shutdown;
112 	int perform;
113 	int verify;
114 	int verify_depth;
115 	char *www_path;
116 } s_time_config;
117 
118 static const struct option s_time_options[] = {
119 	{
120 		.name = "bugs",
121 		.desc = "Enable workarounds for known SSL/TLS bugs",
122 		.type = OPTION_FLAG,
123 		.opt.flag = &s_time_config.bugs,
124 	},
125 	{
126 		.name = "CAfile",
127 		.argname = "file",
128 		.desc = "File containing trusted certificates in PEM format",
129 		.type = OPTION_ARG,
130 		.opt.arg = &s_time_config.CAfile,
131 	},
132 	{
133 		.name = "CApath",
134 		.argname = "path",
135 		.desc = "Directory containing trusted certificates",
136 		.type = OPTION_ARG,
137 		.opt.arg = &s_time_config.CApath,
138 	},
139 	{
140 		.name = "cert",
141 		.argname = "file",
142 		.desc = "Client certificate to use, if one is requested",
143 		.type = OPTION_ARG,
144 		.opt.arg = &s_time_config.certfile,
145 	},
146 	{
147 		.name = "cipher",
148 		.argname = "list",
149 		.desc = "List of cipher suites to send to the server",
150 		.type = OPTION_ARG,
151 		.opt.arg = &s_time_config.cipher,
152 	},
153 	{
154 		.name = "connect",
155 		.argname = "host:port",
156 		.desc = "Host and port to connect to (default "
157 		    SSL_CONNECT_NAME ")",
158 		.type = OPTION_ARG,
159 		.opt.arg = &s_time_config.host,
160 	},
161 	{
162 		.name = "key",
163 		.argname = "file",
164 		.desc = "Client private key to use, if one is required",
165 		.type = OPTION_ARG,
166 		.opt.arg = &s_time_config.keyfile,
167 	},
168 	{
169 		.name = "nbio",
170 		.desc = "Use non-blocking I/O",
171 		.type = OPTION_FLAG,
172 		.opt.flag = &s_time_config.nbio,
173 	},
174 	{
175 		.name = "new",
176 		.desc = "Use a new session ID for each connection",
177 		.type = OPTION_VALUE,
178 		.opt.value = &s_time_config.perform,
179 		.value = 1,
180 	},
181 	{
182 		.name = "no_shutdown",
183 		.desc = "Shut down the connection without notifying the server",
184 		.type = OPTION_FLAG,
185 		.opt.flag = &s_time_config.no_shutdown,
186 	},
187 	{
188 		.name = "reuse",
189 		.desc = "Reuse the same session ID for each connection",
190 		.type = OPTION_VALUE,
191 		.opt.value = &s_time_config.perform,
192 		.value = 2,
193 	},
194 	{
195 		.name = "time",
196 		.argname = "seconds",
197 		.desc = "Duration to perform timing tests for (default 30)",
198 		.type = OPTION_ARG_TIME,
199 		.opt.tvalue = &s_time_config.maxtime,
200 	},
201 	{
202 		.name = "verify",
203 		.argname = "depth",
204 		.desc = "Enable peer certificate verification with given depth",
205 		.type = OPTION_ARG_INT,
206 		.opt.value = &s_time_config.verify_depth,
207 	},
208 	{
209 		.name = "www",
210 		.argname = "page",
211 		.desc = "Page to GET from the server (default none)",
212 		.type = OPTION_ARG,
213 		.opt.arg = &s_time_config.www_path,
214 	},
215 	{ NULL },
216 };
217 
218 static void
219 s_time_usage(void)
220 {
221 	fprintf(stderr,
222 	    "usage: s_time "
223 	    "[-bugs] [-CAfile file] [-CApath directory] [-cert file]\n"
224 	    "    [-cipher cipherlist] [-connect host:port] [-key keyfile]\n"
225 	    "    [-nbio] [-new] [-no_shutdown] [-reuse] [-time seconds]\n"
226 	    "    [-verify depth] [-www page]\n\n");
227 	options_usage(s_time_options);
228 }
229 
230 /***********************************************************************
231  * MAIN - main processing area for client
232  *			real name depends on MONOLITH
233  */
234 int
235 s_time_main(int argc, char **argv)
236 {
237 	int ret = 1;
238 
239 	if (single_execution) {
240 		if (pledge("stdio rpath inet dns", NULL) == -1) {
241 			perror("pledge");
242 			exit(1);
243 		}
244 	}
245 
246 	s_time_meth = TLS_client_method();
247 
248 	verify_depth = 0;
249 
250 	memset(&s_time_config, 0, sizeof(s_time_config));
251 
252 	s_time_config.host = SSL_CONNECT_NAME;
253 	s_time_config.maxtime = SECONDS;
254 	s_time_config.perform = 3;
255 	s_time_config.verify = SSL_VERIFY_NONE;
256 	s_time_config.verify_depth = -1;
257 
258 	if (options_parse(argc, argv, s_time_options, NULL, NULL) != 0) {
259 		s_time_usage();
260 		goto end;
261 	}
262 
263 	if (s_time_config.verify_depth >= 0) {
264 		s_time_config.verify = SSL_VERIFY_PEER | SSL_VERIFY_CLIENT_ONCE;
265 		verify_depth = s_time_config.verify_depth;
266 		BIO_printf(bio_err, "verify depth is %d\n", verify_depth);
267 	}
268 
269 	if (s_time_config.www_path != NULL &&
270 	    strlen(s_time_config.www_path) > MYBUFSIZ - 100) {
271 		BIO_printf(bio_err, "-www option too long\n");
272 		goto end;
273 	}
274 
275 	if ((tm_ctx = SSL_CTX_new(s_time_meth)) == NULL)
276 		return (1);
277 
278 	SSL_CTX_set_quiet_shutdown(tm_ctx, 1);
279 
280 	if (s_time_config.bugs)
281 		SSL_CTX_set_options(tm_ctx, SSL_OP_ALL);
282 
283 	if (s_time_config.cipher != NULL) {
284 		if (!SSL_CTX_set_cipher_list(tm_ctx, s_time_config.cipher)) {
285 			BIO_printf(bio_err, "error setting cipher list\n");
286 			ERR_print_errors(bio_err);
287 			goto end;
288 		}
289 	}
290 
291 	SSL_CTX_set_verify(tm_ctx, s_time_config.verify, NULL);
292 
293 	if (!set_cert_stuff(tm_ctx, s_time_config.certfile,
294 	    s_time_config.keyfile))
295 		goto end;
296 
297 	if ((!SSL_CTX_load_verify_locations(tm_ctx, s_time_config.CAfile,
298 	    s_time_config.CApath)) ||
299 	    (!SSL_CTX_set_default_verify_paths(tm_ctx))) {
300 		/*
301 		 * BIO_printf(bio_err,"error setting default verify
302 		 * locations\n");
303 		 */
304 		ERR_print_errors(bio_err);
305 		/* goto end; */
306 	}
307 
308 	/* Loop and time how long it takes to make connections */
309 	if (s_time_config.perform & 1) {
310 		printf("Collecting connection statistics for %lld seconds\n",
311 		    (long long)s_time_config.maxtime);
312 		if (benchmark(0))
313 			goto end;
314 	}
315 	/*
316 	 * Now loop and time connections using the same session id over and
317 	 * over
318 	 */
319 	if (s_time_config.perform & 2) {
320 		printf("\n\nNow timing with session id reuse.\n");
321 		if (benchmark(1))
322 			goto end;
323 	}
324 	ret = 0;
325  end:
326 	if (tm_ctx != NULL) {
327 		SSL_CTX_free(tm_ctx);
328 		tm_ctx = NULL;
329 	}
330 
331 	return (ret);
332 }
333 
334 /***********************************************************************
335  * run_test - make a connection, get a file, and shut down the connection
336  *
337  * Args:
338  *		scon	= SSL connection
339  * Returns:
340  *		1 on success, 0 on error
341  */
342 static int
343 run_test(SSL *scon)
344 {
345 	char buf[1024 * 8];
346 	struct pollfd pfd[1];
347 	BIO *conn;
348 	long verify_error;
349 	int i, retval;
350 
351 	if ((conn = BIO_new(BIO_s_connect())) == NULL)
352 		return 0;
353 	BIO_set_conn_hostname(conn, s_time_config.host);
354 	SSL_set_connect_state(scon);
355 	SSL_set_bio(scon, conn, conn);
356 	for (;;) {
357 		i = SSL_connect(scon);
358 		if (BIO_sock_should_retry(i)) {
359 			BIO_printf(bio_err, "DELAY\n");
360 			pfd[0].fd = SSL_get_fd(scon);
361 			pfd[0].events = POLLIN;
362 			poll(pfd, 1, -1);
363 			continue;
364 		}
365 		break;
366 	}
367 	if (i <= 0) {
368 		BIO_printf(bio_err, "ERROR\n");
369 		verify_error = SSL_get_verify_result(scon);
370 		if (verify_error != X509_V_OK)
371 			BIO_printf(bio_err, "verify error:%s\n",
372 			    X509_verify_cert_error_string(verify_error));
373 		else
374 			ERR_print_errors(bio_err);
375 		return 0;
376 	}
377 	if (s_time_config.www_path != NULL) {
378 		retval = snprintf(buf, sizeof buf,
379 		    "GET %s HTTP/1.0\r\n\r\n", s_time_config.www_path);
380 		if (retval < 0 || retval >= sizeof buf) {
381 			fprintf(stderr, "URL too long\n");
382 			return 0;
383 		}
384 		if (SSL_write(scon, buf, retval) != retval)
385 			return 0;
386 		while ((i = SSL_read(scon, buf, sizeof(buf))) > 0)
387 			bytes_read += i;
388 	}
389 	if (s_time_config.no_shutdown)
390 		SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN |
391 		    SSL_RECEIVED_SHUTDOWN);
392 	else
393 		SSL_shutdown(scon);
394 	return 1;
395 }
396 
397 static void
398 print_tally_mark(SSL *scon)
399 {
400 	int ver;
401 
402 	if (SSL_session_reused(scon))
403 		ver = 'r';
404 	else {
405 		ver = SSL_version(scon);
406 		if (ver == TLS1_VERSION)
407 			ver = 't';
408 		else
409 			ver = '*';
410 	}
411 	fputc(ver, stdout);
412 	fflush(stdout);
413 }
414 
415 static int
416 benchmark(int reuse_session)
417 {
418 	double elapsed, totalTime;
419 	int nConn = 0;
420 	SSL *scon = NULL;
421 	int ret = 1;
422 
423 	if (reuse_session) {
424 		/* Get an SSL object so we can reuse the session id */
425 		if ((scon = SSL_new(tm_ctx)) == NULL)
426 			goto end;
427 		if (!run_test(scon)) {
428 			fprintf(stderr, "Unable to get connection\n");
429 			goto end;
430 		}
431 		printf("starting\n");
432 	}
433 
434 	nConn = 0;
435 	bytes_read = 0;
436 
437 	app_timer_real(TM_RESET);
438 	app_timer_user(TM_RESET);
439 	for (;;) {
440 		elapsed = app_timer_real(TM_GET);
441 		if (elapsed > s_time_config.maxtime)
442 			break;
443 		if (scon == NULL) {
444 			if ((scon = SSL_new(tm_ctx)) == NULL)
445 				goto end;
446 		}
447 		if (!run_test(scon))
448 			goto end;
449 		nConn += 1;
450 		print_tally_mark(scon);
451 		if (!reuse_session) {
452 			SSL_free(scon);
453 			scon = NULL;
454 		}
455 	}
456 	totalTime = app_timer_user(TM_GET);
457 
458 	printf("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n",
459 	    nConn, totalTime, ((double) nConn / totalTime), bytes_read);
460 	printf("%d connections in %.0f real seconds, %ld bytes read per connection\n",
461 	    nConn, elapsed, nConn > 0 ? bytes_read / nConn : 0);
462 
463 	ret = 0;
464  end:
465 	SSL_free(scon);
466 	return ret;
467 }
468