Fixed some small bugs.
[BearSSL] / tools / client.c
1 /*
2 * Copyright (c) 2016 Thomas Pornin <pornin@bolet.org>
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining
5 * a copy of this software and associated documentation files (the
6 * "Software"), to deal in the Software without restriction, including
7 * without limitation the rights to use, copy, modify, merge, publish,
8 * distribute, sublicense, and/or sell copies of the Software, and to
9 * permit persons to whom the Software is furnished to do so, subject to
10 * the following conditions:
11 *
12 * The above copyright notice and this permission notice shall be
13 * included in all copies or substantial portions of the Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
19 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
20 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
21 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <stdint.h>
29 #include <errno.h>
30
31 #include <sys/types.h>
32 #include <sys/socket.h>
33 #include <netdb.h>
34 #include <netinet/in.h>
35 #include <arpa/inet.h>
36 #include <unistd.h>
37 #include <fcntl.h>
38
39 #include "brssl.h"
40 #include "bearssl.h"
41
42 static int
43 host_connect(const char *host, const char *port, int verbose)
44 {
45 struct addrinfo hints, *si, *p;
46 int fd;
47 int err;
48
49 memset(&hints, 0, sizeof hints);
50 hints.ai_family = PF_UNSPEC;
51 hints.ai_socktype = SOCK_STREAM;
52 err = getaddrinfo(host, port, &hints, &si);
53 if (err != 0) {
54 fprintf(stderr, "ERROR: getaddrinfo(): %s\n",
55 gai_strerror(err));
56 return -1;
57 }
58 fd = -1;
59 for (p = si; p != NULL; p = p->ai_next) {
60 struct sockaddr *sa;
61 void *addr;
62 char tmp[INET6_ADDRSTRLEN + 50];
63
64 sa = (struct sockaddr *)p->ai_addr;
65 if (sa->sa_family == AF_INET) {
66 addr = &((struct sockaddr_in *)sa)->sin_addr;
67 } else if (sa->sa_family == AF_INET6) {
68 addr = &((struct sockaddr_in6 *)sa)->sin6_addr;
69 } else {
70 addr = NULL;
71 }
72 if (addr != NULL) {
73 inet_ntop(p->ai_family, addr, tmp, sizeof tmp);
74 } else {
75 sprintf(tmp, "<unknown family: %d>",
76 (int)sa->sa_family);
77 }
78 if (verbose) {
79 fprintf(stderr, "connecting to: %s\n", tmp);
80 }
81 fd = socket(p->ai_family, p->ai_socktype, p->ai_protocol);
82 if (fd < 0) {
83 if (verbose) {
84 perror("socket()");
85 }
86 continue;
87 }
88 if (connect(fd, p->ai_addr, p->ai_addrlen) < 0) {
89 if (verbose) {
90 perror("connect()");
91 }
92 close(fd);
93 continue;
94 }
95 break;
96 }
97 if (p == NULL) {
98 freeaddrinfo(si);
99 fprintf(stderr, "ERROR: failed to connect\n");
100 return -1;
101 }
102 freeaddrinfo(si);
103 if (verbose) {
104 fprintf(stderr, "connected.\n");
105 }
106
107 /*
108 * We make the socket non-blocking, since we are going to use
109 * poll() to organise I/O.
110 */
111 fcntl(fd, F_SETFL, O_NONBLOCK);
112 return fd;
113 }
114
115 static void
116 usage_client(void)
117 {
118 fprintf(stderr,
119 "usage: brssl client server[:port] [ options ]\n");
120 fprintf(stderr,
121 "options:\n");
122 fprintf(stderr,
123 " -q suppress verbose messages\n");
124 fprintf(stderr,
125 " -trace activate extra debug messages (dump of all packets)\n");
126 fprintf(stderr,
127 " -sni name use this specific name for SNI\n");
128 fprintf(stderr,
129 " -nosni do not send any SNI\n");
130 fprintf(stderr,
131 " -mono use monodirectional buffering\n");
132 fprintf(stderr,
133 " -buf length set the I/O buffer length (in bytes)\n");
134 fprintf(stderr,
135 " -CA file add certificates in 'file' to trust anchors\n");
136 fprintf(stderr,
137 " -list list supported names (protocols, algorithms...)\n");
138 fprintf(stderr,
139 " -vmin name set minimum supported version (default: TLS-1.0)\n");
140 fprintf(stderr,
141 " -vmax name set maximum supported version (default: TLS-1.2)\n");
142 fprintf(stderr,
143 " -cs names set list of supported cipher suites (comma-separated)\n");
144 fprintf(stderr,
145 " -hf names add support for some hash functions (comma-separated)\n");
146 }
147
148 /* see brssl.h */
149 int
150 do_client(int argc, char *argv[])
151 {
152 int retcode;
153 int verbose;
154 int trace;
155 int i, bidi;
156 const char *server_name;
157 char *host;
158 char *port;
159 const char *sni;
160 anchor_list anchors = VEC_INIT;
161 unsigned vmin, vmax;
162 cipher_suite *suites;
163 size_t num_suites;
164 uint16_t *suite_ids;
165 unsigned hfuns;
166 size_t u;
167 br_ssl_client_context cc;
168 br_x509_minimal_context xc;
169 x509_noanchor_context xwc;
170 const br_hash_class *dnhash;
171 unsigned char *iobuf;
172 size_t iobuf_len;
173 int fd;
174
175 retcode = 0;
176 verbose = 1;
177 trace = 0;
178 server_name = NULL;
179 host = NULL;
180 port = NULL;
181 sni = NULL;
182 bidi = 1;
183 vmin = 0;
184 vmax = 0;
185 suites = NULL;
186 num_suites = 0;
187 hfuns = 0;
188 suite_ids = NULL;
189 iobuf = NULL;
190 iobuf_len = 0;
191 fd = -1;
192 for (i = 0; i < argc; i ++) {
193 const char *arg;
194
195 arg = argv[i];
196 if (arg[0] != '-') {
197 if (server_name != NULL) {
198 fprintf(stderr,
199 "ERROR: duplicate server name\n");
200 usage_client();
201 goto client_exit_error;
202 }
203 server_name = arg;
204 continue;
205 }
206 if (eqstr(arg, "-v") || eqstr(arg, "-verbose")) {
207 verbose = 1;
208 } else if (eqstr(arg, "-q") || eqstr(arg, "-quiet")) {
209 verbose = 0;
210 } else if (eqstr(arg, "-trace")) {
211 trace = 1;
212 } else if (eqstr(arg, "-sni")) {
213 if (++ i >= argc) {
214 fprintf(stderr,
215 "ERROR: no argument for '-sni'\n");
216 usage_client();
217 goto client_exit_error;
218 }
219 if (sni != NULL) {
220 fprintf(stderr, "ERROR: duplicate SNI\n");
221 usage_client();
222 goto client_exit_error;
223 }
224 sni = argv[i];
225 } else if (eqstr(arg, "-nosni")) {
226 if (sni != NULL) {
227 fprintf(stderr, "ERROR: duplicate SNI\n");
228 usage_client();
229 goto client_exit_error;
230 }
231 sni = "";
232 } else if (eqstr(arg, "-mono")) {
233 bidi = 0;
234 } else if (eqstr(arg, "-buf")) {
235 if (++ i >= argc) {
236 fprintf(stderr,
237 "ERROR: no argument for '-buf'\n");
238 usage_client();
239 goto client_exit_error;
240 }
241 arg = argv[i];
242 if (iobuf_len != 0) {
243 fprintf(stderr,
244 "ERROR: duplicate I/O buffer length\n");
245 usage_client();
246 goto client_exit_error;
247 }
248 iobuf_len = strtoul(arg, 0, 10);
249 } else if (eqstr(arg, "-CA")) {
250 if (++ i >= argc) {
251 fprintf(stderr,
252 "ERROR: no argument for '-CA'\n");
253 usage_client();
254 goto client_exit_error;
255 }
256 arg = argv[i];
257 if (read_trust_anchors(&anchors, arg) == 0) {
258 usage_client();
259 goto client_exit_error;
260 }
261 } else if (eqstr(arg, "-list")) {
262 list_names();
263 goto client_exit;
264 } else if (eqstr(arg, "-vmin")) {
265 if (++ i >= argc) {
266 fprintf(stderr,
267 "ERROR: no argument for '-vmin'\n");
268 usage_client();
269 goto client_exit_error;
270 }
271 arg = argv[i];
272 if (vmin != 0) {
273 fprintf(stderr,
274 "ERROR: duplicate minimum version\n");
275 usage_client();
276 goto client_exit_error;
277 }
278 vmin = parse_version(arg, strlen(arg));
279 if (vmin == 0) {
280 fprintf(stderr,
281 "ERROR: unrecognised version '%s'\n",
282 arg);
283 usage_client();
284 goto client_exit_error;
285 }
286 } else if (eqstr(arg, "-vmax")) {
287 if (++ i >= argc) {
288 fprintf(stderr,
289 "ERROR: no argument for '-vmax'\n");
290 usage_client();
291 goto client_exit_error;
292 }
293 arg = argv[i];
294 if (vmax != 0) {
295 fprintf(stderr,
296 "ERROR: duplicate maximum version\n");
297 usage_client();
298 goto client_exit_error;
299 }
300 vmax = parse_version(arg, strlen(arg));
301 if (vmax == 0) {
302 fprintf(stderr,
303 "ERROR: unrecognised version '%s'\n",
304 arg);
305 usage_client();
306 goto client_exit_error;
307 }
308 } else if (eqstr(arg, "-cs")) {
309 if (++ i >= argc) {
310 fprintf(stderr,
311 "ERROR: no argument for '-cs'\n");
312 usage_client();
313 goto client_exit_error;
314 }
315 arg = argv[i];
316 if (suites != NULL) {
317 fprintf(stderr, "ERROR: duplicate list"
318 " of cipher suites\n");
319 usage_client();
320 goto client_exit_error;
321 }
322 suites = parse_suites(arg, &num_suites);
323 if (suites == NULL) {
324 usage_client();
325 goto client_exit_error;
326 }
327 } else if (eqstr(arg, "-hf")) {
328 unsigned x;
329
330 if (++ i >= argc) {
331 fprintf(stderr,
332 "ERROR: no argument for '-hf'\n");
333 usage_client();
334 goto client_exit_error;
335 }
336 arg = argv[i];
337 x = parse_hash_functions(arg);
338 if (x == 0) {
339 usage_client();
340 goto client_exit_error;
341 }
342 hfuns |= x;
343 } else {
344 fprintf(stderr, "ERROR: unknown option: '%s'\n", arg);
345 usage_client();
346 goto client_exit_error;
347 }
348 }
349 if (server_name == NULL) {
350 fprintf(stderr, "ERROR: no server name/address provided\n");
351 usage_client();
352 goto client_exit_error;
353 }
354 for (u = strlen(server_name); u > 0; u --) {
355 int c = server_name[u - 1];
356 if (c == ':') {
357 break;
358 }
359 if (c < '0' || c > '9') {
360 u = 0;
361 break;
362 }
363 }
364 if (u == 0) {
365 host = xstrdup(server_name);
366 port = "443";
367 } else {
368 port = xstrdup(server_name + u);
369 host = xmalloc(u);
370 memcpy(host, server_name, u - 1);
371 host[u - 1] = 0;
372 }
373 if (sni == NULL) {
374 sni = host;
375 }
376
377 if (vmin == 0) {
378 vmin = BR_TLS10;
379 }
380 if (vmax == 0) {
381 vmax = BR_TLS12;
382 }
383 if (vmax < vmin) {
384 fprintf(stderr, "ERROR: impossible minimum/maximum protocol"
385 " version combination\n");
386 usage_client();
387 goto client_exit_error;
388 }
389 if (suites == NULL) {
390 num_suites = 0;
391
392 for (u = 0; cipher_suites[u].name; u ++) {
393 if ((cipher_suites[u].req & REQ_TLS12) == 0
394 || vmax >= BR_TLS12)
395 {
396 num_suites ++;
397 }
398 }
399 suites = xmalloc(num_suites * sizeof *suites);
400 num_suites = 0;
401 for (u = 0; cipher_suites[u].name; u ++) {
402 if ((cipher_suites[u].req & REQ_TLS12) == 0
403 || vmax >= BR_TLS12)
404 {
405 suites[num_suites ++] = cipher_suites[u];
406 }
407 }
408 }
409 if (hfuns == 0) {
410 hfuns = (unsigned)-1;
411 }
412 if (iobuf_len == 0) {
413 if (bidi) {
414 iobuf_len = BR_SSL_BUFSIZE_BIDI;
415 } else {
416 iobuf_len = BR_SSL_BUFSIZE_MONO;
417 }
418 }
419 iobuf = xmalloc(iobuf_len);
420
421 /*
422 * Compute implementation requirements and inject implementations.
423 */
424 suite_ids = xmalloc(num_suites * sizeof *suite_ids);
425 br_ssl_client_zero(&cc);
426 br_ssl_engine_set_versions(&cc.eng, vmin, vmax);
427 dnhash = NULL;
428 for (u = 0; hash_functions[u].name; u ++) {
429 const br_hash_class *hc;
430 int id;
431
432 hc = hash_functions[u].hclass;
433 id = (hc->desc >> BR_HASHDESC_ID_OFF) & BR_HASHDESC_ID_MASK;
434 if ((hfuns & ((unsigned)1 << id)) != 0) {
435 dnhash = hc;
436 }
437 }
438 if (dnhash == NULL) {
439 fprintf(stderr, "ERROR: no supported hash function\n");
440 goto client_exit_error;
441 }
442 br_x509_minimal_init(&xc, dnhash,
443 &VEC_ELT(anchors, 0), VEC_LEN(anchors));
444 if (vmin <= BR_TLS11) {
445 if (!(hfuns & (1 << br_md5_ID))) {
446 fprintf(stderr, "ERROR: TLS 1.0 and 1.1 need MD5\n");
447 goto client_exit_error;
448 }
449 if (!(hfuns & (1 << br_sha1_ID))) {
450 fprintf(stderr, "ERROR: TLS 1.0 and 1.1 need SHA-1\n");
451 goto client_exit_error;
452 }
453 }
454 for (u = 0; u < num_suites; u ++) {
455 unsigned req;
456
457 req = suites[u].req;
458 suite_ids[u] = suites[u].suite;
459 if ((req & REQ_TLS12) != 0 && vmax < BR_TLS12) {
460 fprintf(stderr,
461 "ERROR: cipher suite %s requires TLS 1.2\n",
462 suites[u].name);
463 goto client_exit_error;
464 }
465 if ((req & REQ_SHA1) != 0 && !(hfuns & (1 << br_sha1_ID))) {
466 fprintf(stderr,
467 "ERROR: cipher suite %s requires SHA-1\n",
468 suites[u].name);
469 goto client_exit_error;
470 }
471 if ((req & REQ_SHA256) != 0 && !(hfuns & (1 << br_sha256_ID))) {
472 fprintf(stderr,
473 "ERROR: cipher suite %s requires SHA-256\n",
474 suites[u].name);
475 goto client_exit_error;
476 }
477 if ((req & REQ_SHA384) != 0 && !(hfuns & (1 << br_sha384_ID))) {
478 fprintf(stderr,
479 "ERROR: cipher suite %s requires SHA-384\n",
480 suites[u].name);
481 goto client_exit_error;
482 }
483 /* TODO: algorithm implementation selection */
484 if ((req & REQ_AESCBC) != 0) {
485 br_ssl_engine_set_aes_cbc(&cc.eng,
486 &br_aes_ct_cbcenc_vtable,
487 &br_aes_ct_cbcdec_vtable);
488 br_ssl_engine_set_cbc(&cc.eng,
489 &br_sslrec_in_cbc_vtable,
490 &br_sslrec_out_cbc_vtable);
491 }
492 if ((req & REQ_AESGCM) != 0) {
493 br_ssl_engine_set_aes_ctr(&cc.eng,
494 &br_aes_ct_ctr_vtable);
495 br_ssl_engine_set_ghash(&cc.eng,
496 &br_ghash_ctmul);
497 br_ssl_engine_set_gcm(&cc.eng,
498 &br_sslrec_in_gcm_vtable,
499 &br_sslrec_out_gcm_vtable);
500 }
501 if ((req & REQ_3DESCBC) != 0) {
502 br_ssl_engine_set_des_cbc(&cc.eng,
503 &br_des_ct_cbcenc_vtable,
504 &br_des_ct_cbcdec_vtable);
505 br_ssl_engine_set_cbc(&cc.eng,
506 &br_sslrec_in_cbc_vtable,
507 &br_sslrec_out_cbc_vtable);
508 }
509 if ((req & REQ_RSAKEYX) != 0) {
510 br_ssl_client_set_rsapub(&cc, &br_rsa_i31_public);
511 }
512 if ((req & REQ_ECDHE_RSA) != 0) {
513 br_ssl_engine_set_ec(&cc.eng, &br_ec_prime_i31);
514 br_ssl_client_set_rsavrfy(&cc, &br_rsa_i31_pkcs1_vrfy);
515 }
516 if ((req & REQ_ECDHE_ECDSA) != 0) {
517 br_ssl_engine_set_ec(&cc.eng, &br_ec_prime_i31);
518 br_ssl_client_set_ecdsa(&cc, &br_ecdsa_i31_vrfy_asn1);
519 }
520 if ((req & REQ_ECDH) != 0) {
521 br_ssl_engine_set_ec(&cc.eng, &br_ec_prime_i31);
522 }
523 }
524 br_ssl_engine_set_suites(&cc.eng, suite_ids, num_suites);
525
526 for (u = 0; hash_functions[u].name; u ++) {
527 const br_hash_class *hc;
528 int id;
529
530 hc = hash_functions[u].hclass;
531 id = (hc->desc >> BR_HASHDESC_ID_OFF) & BR_HASHDESC_ID_MASK;
532 if ((hfuns & ((unsigned)1 << id)) != 0) {
533 br_ssl_engine_set_hash(&cc.eng, id, hc);
534 br_x509_minimal_set_hash(&xc, id, hc);
535 }
536 }
537 if (vmin <= BR_TLS11) {
538 br_ssl_engine_set_prf10(&cc.eng, &br_tls10_prf);
539 }
540 if (vmax >= BR_TLS12) {
541 if ((hfuns & ((unsigned)1 << br_sha256_ID)) != 0) {
542 br_ssl_engine_set_prf_sha256(&cc.eng,
543 &br_tls12_sha256_prf);
544 }
545 if ((hfuns & ((unsigned)1 << br_sha384_ID)) != 0) {
546 br_ssl_engine_set_prf_sha384(&cc.eng,
547 &br_tls12_sha384_prf);
548 }
549 }
550 br_x509_minimal_set_rsa(&xc, &br_rsa_i31_pkcs1_vrfy);
551 br_x509_minimal_set_ecdsa(&xc,
552 &br_ec_prime_i31, &br_ecdsa_i31_vrfy_asn1);
553
554 /*
555 * If there is no provided trust anchor, then certificate validation
556 * will always fail. In that situation, we use our custom wrapper
557 * that tolerates unknown anchors.
558 */
559 if (VEC_LEN(anchors) == 0) {
560 if (verbose) {
561 fprintf(stderr,
562 "WARNING: no configured trust anchor\n");
563 }
564 x509_noanchor_init(&xwc, &xc.vtable);
565 br_ssl_engine_set_x509(&cc.eng, &xwc.vtable);
566 } else {
567 br_ssl_engine_set_x509(&cc.eng, &xc.vtable);
568 }
569
570 br_ssl_engine_set_buffer(&cc.eng, iobuf, iobuf_len, bidi);
571 br_ssl_client_reset(&cc, sni, 0);
572
573 /*
574 * Connect to the peer.
575 */
576 fd = host_connect(host, port, verbose);
577 if (fd < 0) {
578 goto client_exit_error;
579 }
580
581 /*
582 * Run the engine until completion.
583 */
584 if (run_ssl_engine(&cc.eng, fd,
585 (verbose ? RUN_ENGINE_VERBOSE : 0)
586 | (trace ? RUN_ENGINE_TRACE : 0)) != 0)
587 {
588 goto client_exit_error;
589 } else {
590 goto client_exit;
591 }
592
593 /*
594 * Release allocated structures.
595 */
596 client_exit:
597 xfree(host);
598 xfree(suites);
599 xfree(suite_ids);
600 VEC_CLEAREXT(anchors, &free_ta_contents);
601 xfree(iobuf);
602 if (fd >= 0) {
603 close(fd);
604 }
605 return retcode;
606
607 client_exit_error:
608 retcode = -1;
609 goto client_exit;
610 }