cosmetic changes
[openssl.git] / ssl / ssltest.c
1 /* ssl/ssltest.c */
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 #include <stdio.h>
60 #include <stdlib.h>
61 #include <string.h>
62 #include <errno.h>
63 #include <limits.h>
64
65 #include "openssl/e_os.h"
66
67 #include <openssl/bio.h>
68 #include <openssl/crypto.h>
69 #include <openssl/x509.h>
70 #include <openssl/ssl.h>
71 #include <openssl/err.h>
72 #ifdef WINDOWS
73 #include "../crypto/bio/bss_file.c"
74 #endif
75
76 #if defined(NO_RSA) && !defined(NO_SSL2)
77 #define NO_SSL2
78 #endif
79
80 #ifdef VMS
81 #  define TEST_SERVER_CERT "SYS$DISK:[-.APPS]SERVER.PEM"
82 #  define TEST_CLIENT_CERT "SYS$DISK:[-.APPS]CLIENT.PEM"
83 #else
84 #  define TEST_SERVER_CERT "../apps/server.pem"
85 #  define TEST_CLIENT_CERT "../apps/client.pem"
86 #endif
87
88 int MS_CALLBACK verify_callback(int ok, X509_STORE_CTX *ctx);
89 #ifndef NO_RSA
90 static RSA MS_CALLBACK *tmp_rsa_cb(SSL *s, int is_export,int keylength);
91 #endif
92 #ifndef NO_DH
93 static DH *get_dh512(void);
94 #endif
95 BIO *bio_err=NULL;
96 BIO *bio_stdout=NULL;
97
98 static char *cipher=NULL;
99 int verbose=0;
100 int debug=0;
101 #if 0
102 /* Not used yet. */
103 #ifdef FIONBIO
104 static int s_nbio=0;
105 #endif
106 #endif
107
108
109 int doit_biopair(SSL *s_ssl,SSL *c_ssl,long bytes);
110 int doit(SSL *s_ssl,SSL *c_ssl,long bytes);
111 static void sv_usage(void)
112         {
113         fprintf(stderr,"usage: ssltest [args ...]\n");
114         fprintf(stderr,"\n");
115         fprintf(stderr," -server_auth  - check server certificate\n");
116         fprintf(stderr," -client_auth  - do client authentication\n");
117         fprintf(stderr," -v            - more output\n");
118         fprintf(stderr," -d            - debug output\n");
119         fprintf(stderr," -reuse        - use session-id reuse\n");
120         fprintf(stderr," -num <val>    - number of connections to perform\n");
121         fprintf(stderr," -bytes <val>  - number of bytes to swap between client/server\n");
122 #ifndef NO_SSL2
123         fprintf(stderr," -ssl2         - use SSLv2\n");
124 #endif
125 #ifndef NO_SSL3
126         fprintf(stderr," -ssl3         - use SSLv3\n");
127 #endif
128 #ifndef NO_TLS1
129         fprintf(stderr," -tls1         - use TLSv1\n");
130 #endif
131         fprintf(stderr," -CApath arg   - PEM format directory of CA's\n");
132         fprintf(stderr," -CAfile arg   - PEM format file of CA's\n");
133         fprintf(stderr," -cert arg     - Certificate file\n");
134         fprintf(stderr," -s_cert arg   - Just the server certificate file\n");
135         fprintf(stderr," -c_cert arg   - Just the client certificate file\n");
136         fprintf(stderr," -cipher arg   - The cipher list\n");
137         fprintf(stderr," -bio_pair     - Use BIO pairs\n");
138         fprintf(stderr," -f            - Test even cases that can't work\n");
139         }
140
141 int main(int argc, char *argv[])
142         {
143         char *CApath=NULL,*CAfile=NULL;
144         int badop=0;
145         int bio_pair=0;
146         int force=0;
147         int tls1=0,ssl2=0,ssl3=0,ret=1;
148         int client_auth=0;
149         int server_auth=0,i;
150         char *server_cert=TEST_SERVER_CERT;
151         char *client_cert=TEST_CLIENT_CERT;
152         SSL_CTX *s_ctx=NULL;
153         SSL_CTX *c_ctx=NULL;
154         SSL_METHOD *meth=NULL;
155         SSL *c_ssl,*s_ssl;
156         int number=1,reuse=0;
157         long bytes=1L;
158         SSL_CIPHER *ciph;
159 #ifndef NO_DH
160         DH *dh;
161 #endif
162
163         bio_err=BIO_new_fp(stderr,BIO_NOCLOSE);
164         bio_stdout=BIO_new_fp(stdout,BIO_NOCLOSE);
165
166         CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
167
168         argc--;
169         argv++;
170
171         while (argc >= 1)
172                 {
173                 if      (strcmp(*argv,"-server_auth") == 0)
174                         server_auth=1;
175                 else if (strcmp(*argv,"-client_auth") == 0)
176                         client_auth=1;
177                 else if (strcmp(*argv,"-v") == 0)
178                         verbose=1;
179                 else if (strcmp(*argv,"-d") == 0)
180                         debug=1;
181                 else if (strcmp(*argv,"-reuse") == 0)
182                         reuse=1;
183                 else if (strcmp(*argv,"-ssl2") == 0)
184                         ssl2=1;
185                 else if (strcmp(*argv,"-tls1") == 0)
186                         tls1=1;
187                 else if (strcmp(*argv,"-ssl3") == 0)
188                         ssl3=1;
189                 else if (strncmp(*argv,"-num",4) == 0)
190                         {
191                         if (--argc < 1) goto bad;
192                         number= atoi(*(++argv));
193                         if (number == 0) number=1;
194                         }
195                 else if (strcmp(*argv,"-bytes") == 0)
196                         {
197                         if (--argc < 1) goto bad;
198                         bytes= atol(*(++argv));
199                         if (bytes == 0L) bytes=1L;
200                         i=strlen(argv[0]);
201                         if (argv[0][i-1] == 'k') bytes*=1024L;
202                         if (argv[0][i-1] == 'm') bytes*=1024L*1024L;
203                         }
204                 else if (strcmp(*argv,"-cert") == 0)
205                         {
206                         if (--argc < 1) goto bad;
207                         server_cert= *(++argv);
208                         }
209                 else if (strcmp(*argv,"-s_cert") == 0)
210                         {
211                         if (--argc < 1) goto bad;
212                         server_cert= *(++argv);
213                         }
214                 else if (strcmp(*argv,"-c_cert") == 0)
215                         {
216                         if (--argc < 1) goto bad;
217                         client_cert= *(++argv);
218                         }
219                 else if (strcmp(*argv,"-cipher") == 0)
220                         {
221                         if (--argc < 1) goto bad;
222                         cipher= *(++argv);
223                         }
224                 else if (strcmp(*argv,"-CApath") == 0)
225                         {
226                         if (--argc < 1) goto bad;
227                         CApath= *(++argv);
228                         }
229                 else if (strcmp(*argv,"-CAfile") == 0)
230                         {
231                         if (--argc < 1) goto bad;
232                         CAfile= *(++argv);
233                         }
234                 else if (strcmp(*argv,"-bio_pair") == 0)
235                         {
236                         bio_pair = 1;
237                         }
238                 else if (strcmp(*argv,"-f") == 0)
239                         {
240                         force = 1;
241                         }
242                 else
243                         {
244                         fprintf(stderr,"unknown option %s\n",*argv);
245                         badop=1;
246                         break;
247                         }
248                 argc--;
249                 argv++;
250                 }
251         if (badop)
252                 {
253 bad:
254                 sv_usage();
255                 goto end;
256                 }
257
258         if (!ssl2 && !ssl3 && !tls1 && number > 1 && !reuse && !force)
259                 {
260                 fprintf(stderr, "This case cannot work.  Use -f switch to perform "
261                         "the test anyway\n"
262                         "(and -d to see what happens, "
263                         "and -bio_pair to really make it happen :-)\n"
264                         "or add one of -ssl2, -ssl3, -tls1, -reuse to "
265                         "avoid protocol mismatch.\n");
266                 exit(1);
267                 }
268
269 /*      if (cipher == NULL) cipher=getenv("SSL_CIPHER"); */
270
271         SSL_library_init();
272         SSL_load_error_strings();
273
274 #if !defined(NO_SSL2) && !defined(NO_SSL3)
275         if (ssl2)
276                 meth=SSLv2_method();
277         else 
278         if (tls1)
279                 meth=TLSv1_method();
280         else
281         if (ssl3)
282                 meth=SSLv3_method();
283         else
284                 meth=SSLv23_method();
285 #else
286 #ifdef NO_SSL2
287         meth=SSLv3_method();
288 #else
289         meth=SSLv2_method();
290 #endif
291 #endif
292
293         c_ctx=SSL_CTX_new(meth);
294         s_ctx=SSL_CTX_new(meth);
295         if ((c_ctx == NULL) || (s_ctx == NULL))
296                 {
297                 ERR_print_errors(bio_err);
298                 goto end;
299                 }
300
301         if (cipher != NULL)
302                 {
303                 SSL_CTX_set_cipher_list(c_ctx,cipher);
304                 SSL_CTX_set_cipher_list(s_ctx,cipher);
305                 }
306
307 #ifndef NO_DH
308         dh=get_dh512();
309         SSL_CTX_set_tmp_dh(s_ctx,dh);
310         DH_free(dh);
311 #endif
312
313 #ifndef NO_RSA
314         SSL_CTX_set_tmp_rsa_callback(s_ctx,tmp_rsa_cb);
315 #endif
316
317         if (!SSL_CTX_use_certificate_file(s_ctx,server_cert,SSL_FILETYPE_PEM))
318                 {
319                 ERR_print_errors(bio_err);
320                 }
321         else if (!SSL_CTX_use_PrivateKey_file(s_ctx,server_cert,
322                 SSL_FILETYPE_PEM))
323                 {
324                 ERR_print_errors(bio_err);
325                 goto end;
326                 }
327
328         if (client_auth)
329                 {
330                 SSL_CTX_use_certificate_file(c_ctx,client_cert,
331                         SSL_FILETYPE_PEM);
332                 SSL_CTX_use_PrivateKey_file(c_ctx,client_cert,
333                         SSL_FILETYPE_PEM);
334                 }
335
336         if (    (!SSL_CTX_load_verify_locations(s_ctx,CAfile,CApath)) ||
337                 (!SSL_CTX_set_default_verify_paths(s_ctx)) ||
338                 (!SSL_CTX_load_verify_locations(c_ctx,CAfile,CApath)) ||
339                 (!SSL_CTX_set_default_verify_paths(c_ctx)))
340                 {
341                 /* fprintf(stderr,"SSL_load_verify_locations\n"); */
342                 ERR_print_errors(bio_err);
343                 /* goto end; */
344                 }
345
346         if (client_auth)
347                 {
348                 fprintf(stderr,"client authentication\n");
349                 SSL_CTX_set_verify(s_ctx,
350                         SSL_VERIFY_PEER|SSL_VERIFY_FAIL_IF_NO_PEER_CERT,
351                         verify_callback);
352                 }
353         if (server_auth)
354                 {
355                 fprintf(stderr,"server authentication\n");
356                 SSL_CTX_set_verify(c_ctx,SSL_VERIFY_PEER,
357                         verify_callback);
358                 }
359
360         c_ssl=SSL_new(c_ctx);
361         s_ssl=SSL_new(s_ctx);
362
363         for (i=0; i<number; i++)
364                 {
365                 if (!reuse) SSL_set_session(c_ssl,NULL);
366                 if (bio_pair)
367                         ret=doit_biopair(s_ssl,c_ssl,bytes);
368                 else
369                         ret=doit(s_ssl,c_ssl,bytes);
370                 }
371
372         if (!verbose)
373                 {
374                 ciph=SSL_get_current_cipher(c_ssl);
375                 fprintf(stdout,"Protocol %s, cipher %s, %s\n",
376                         SSL_get_version(c_ssl),
377                         SSL_CIPHER_get_version(ciph),
378                         SSL_CIPHER_get_name(ciph));
379                 }
380         if ((number > 1) || (bytes > 1L))
381                 printf("%d handshakes of %ld bytes done\n",number,bytes);
382
383         SSL_free(s_ssl);
384         SSL_free(c_ssl);
385
386 end:
387         if (s_ctx != NULL) SSL_CTX_free(s_ctx);
388         if (c_ctx != NULL) SSL_CTX_free(c_ctx);
389
390         if (bio_stdout != NULL) BIO_free(bio_stdout);
391
392         ERR_free_strings();
393         ERR_remove_state(0);
394         EVP_cleanup();
395         CRYPTO_mem_leaks(bio_err);
396         EXIT(ret);
397         }
398
399 int doit_biopair(SSL *s_ssl, SSL *c_ssl, long count)
400         {
401         long cw_num = count, cr_num = count, sw_num = count, sr_num = count;
402         BIO *s_ssl_bio = NULL, *c_ssl_bio = NULL;
403         BIO *server = NULL, *server_io = NULL, *client = NULL, *client_io = NULL;
404         SSL_CIPHER *ciph;
405         int ret = 1;
406         
407         size_t bufsiz = 256; /* small buffer for testing */
408
409         if (!BIO_new_bio_pair(&server, bufsiz, &server_io, bufsiz))
410                 goto err;
411         if (!BIO_new_bio_pair(&client, bufsiz, &client_io, bufsiz))
412                 goto err;
413         
414         s_ssl_bio = BIO_new(BIO_f_ssl());
415         if (!s_ssl_bio)
416                 goto err;
417
418         c_ssl_bio = BIO_new(BIO_f_ssl());
419         if (!c_ssl_bio)
420                 goto err;
421
422         SSL_set_connect_state(c_ssl);
423         SSL_set_bio(c_ssl, client, client);
424         (void)BIO_set_ssl(c_ssl_bio, c_ssl, BIO_NOCLOSE);
425
426         SSL_set_accept_state(s_ssl);
427         SSL_set_bio(s_ssl, server, server);
428         (void)BIO_set_ssl(s_ssl_bio, s_ssl, BIO_NOCLOSE);
429
430         do
431                 {
432                 /* c_ssl_bio:          SSL filter BIO
433                  *
434                  * client:             pseudo-I/O for SSL library
435                  *
436                  * client_io:          client's SSL communication; usually to be
437                  *                     relayed over some I/O facility, but in this
438                  *                     test program, we're the server, too:
439                  *
440                  * server_io:          server's SSL communication
441                  *
442                  * server:             pseudo-I/O for SSL library
443                  *
444                  * s_ssl_bio:          SSL filter BIO
445                  *
446                  * The client and the server each employ a "BIO pair":
447                  * client + client_io, server + server_io.
448                  * BIO pairs are symmetric.  A BIO pair behaves similar
449                  * to a non-blocking socketpair (but both endpoints must
450                  * be handled by the same thread).
451                  * [Here we could connect client and server to the ends
452                  * of a single BIO pair, but then this code would be less
453                  * suitable as an example for BIO pairs in general.]
454                  *
455                  * Useful functions for querying the state of BIO pair endpoints:
456                  *
457                  * BIO_ctrl_pending(bio)              number of bytes we can read now
458                  * BIO_ctrl_get_read_request(bio)     number of bytes needed to fulfil
459                  *                                      other side's read attempt
460                  * BIO_ctrl_get_write_gurantee(bio)   number of bytes we can write now
461                  *
462                  * ..._read_request is never more than ..._write_guarantee;
463                  * it depends on the application which one you should use.
464                  */
465
466                 /* We have non-blocking behaviour throughout this test program, but
467                  * can be sure that there is *some* progress in each iteration; so
468                  * we don't have to worry about ..._SHOULD_READ or ..._SHOULD_WRITE
469                  * -- we just try everything in each iteration
470                  */
471
472                         {
473                         /* CLIENT */
474                 
475                         MS_STATIC char cbuf[1024*8];
476                         int i, r;
477
478                         if (debug)
479                                 if (SSL_in_init(c_ssl))
480                                         printf("client waiting in SSL_connect - %s\n",
481                                                 SSL_state_string_long(c_ssl));
482
483                         if (cw_num > 0)
484                                 {
485                                 /* Write to server. */
486                                 
487                                 if (cw_num > (long)sizeof cbuf)
488                                         i = sizeof cbuf;
489                                 else
490                                         i = (int)cw_num;
491                                 r = BIO_write(c_ssl_bio, cbuf, i);
492                                 if (r == -1)
493                                         {
494                                         if (!BIO_should_retry(c_ssl_bio))
495                                                 {
496                                                 fprintf(stderr,"ERROR in CLIENT\n");
497                                                 goto err;
498                                                 }
499                                         /* BIO_should_retry(...) can just be ignored here.
500                                          * The library expects us to call BIO_write with
501                                          * the same arguments again, and that's what we will
502                                          * do in the next iteration. */
503                                         }
504                                 else if (r == 0)
505                                         {
506                                         fprintf(stderr,"SSL CLIENT STARTUP FAILED\n");
507                                         goto err;
508                                         }
509                                 else
510                                         {
511                                         if (debug)
512                                                 printf("client wrote %d\n", r);
513                                         cw_num -= r;                            
514                                         }
515                                 }
516
517                         if (cr_num > 0)
518                                 {
519                                 /* Read from server. */
520
521                                 r = BIO_read(c_ssl_bio, cbuf, sizeof(cbuf));
522                                 if (r < 0)
523                                         {
524                                         if (!BIO_should_retry(c_ssl_bio))
525                                                 {
526                                                 fprintf(stderr,"ERROR in CLIENT\n");
527                                                 goto err;
528                                                 }
529                                         /* Again, "BIO_should_retry" can be ignored. */
530                                         }
531                                 else if (r == 0)
532                                         {
533                                         fprintf(stderr,"SSL CLIENT STARTUP FAILED\n");
534                                         goto err;
535                                         }
536                                 else
537                                         {
538                                         if (debug)
539                                                 printf("client read %d\n", r);
540                                         cr_num -= r;
541                                         }
542                                 }
543                         }
544
545                         {
546                         /* SERVER */
547                 
548                         MS_STATIC char sbuf[1024*8];
549                         int i, r;
550
551                         if (debug)
552                                 if (SSL_in_init(s_ssl))
553                                         printf("server waiting in SSL_accept - %s\n",
554                                                 SSL_state_string_long(s_ssl));
555
556                         if (sw_num > 0)
557                                 {
558                                 /* Write to client. */
559                                 
560                                 if (sw_num > (long)sizeof sbuf)
561                                         i = sizeof sbuf;
562                                 else
563                                         i = (int)sw_num;
564                                 r = BIO_write(s_ssl_bio, sbuf, i);
565                                 if (r == -1)
566                                         {
567                                         if (!BIO_should_retry(s_ssl_bio))
568                                                 {
569                                                 fprintf(stderr,"ERROR in SERVER\n");
570                                                 goto err;
571                                                 }
572                                         /* Ignore "BIO_should_retry". */
573                                         }
574                                 else if (r == 0)
575                                         {
576                                         fprintf(stderr,"SSL SERVER STARTUP FAILED\n");
577                                         goto err;
578                                         }
579                                 else
580                                         {
581                                         if (debug)
582                                                 printf("server wrote %d\n", r);
583                                         sw_num -= r;                            
584                                         }
585                                 }
586
587                         if (sr_num > 0)
588                                 {
589                                 /* Read from client. */
590
591                                 r = BIO_read(s_ssl_bio, sbuf, sizeof(sbuf));
592                                 if (r < 0)
593                                         {
594                                         if (!BIO_should_retry(s_ssl_bio))
595                                                 {
596                                                 fprintf(stderr,"ERROR in SERVER\n");
597                                                 goto err;
598                                                 }
599                                         /* blah, blah */
600                                         }
601                                 else if (r == 0)
602                                         {
603                                         fprintf(stderr,"SSL SERVER STARTUP FAILED\n");
604                                         goto err;
605                                         }
606                                 else
607                                         {
608                                         if (debug)
609                                                 printf("server read %d\n", r);
610                                         sr_num -= r;
611                                         }
612                                 }
613                         }
614                         
615                         {
616                         /* "I/O" BETWEEN CLIENT AND SERVER. */
617
618 #define RELAYBUFSIZ 200
619                         static char buf[RELAYBUFSIZ];
620
621                         /* RELAYBUF is arbitrary.  When writing data over some real
622                          * network, use a buffer of the same size as in the BIO_pipe
623                          * and make that size large (for reading from the network
624                          * small buffers usually won't hurt).
625                          * Here sizes differ for testing. */
626
627                         size_t r1, r2;
628                         size_t num;
629                         int r;
630                         static int prev_progress = 1;
631                         int progress = 0;
632                         
633                         /* client to server */
634                         do
635                                 {
636                                 r1 = BIO_ctrl_pending(client_io);
637                                 r2 = BIO_ctrl_get_write_guarantee(server_io);
638
639                                 num = r1;
640                                 if (r2 < num)
641                                         num = r2;
642                                 if (num)
643                                         {
644                                         if (sizeof buf < num)
645                                                 num = sizeof buf;
646                                         if (INT_MAX < num) /* yeah, right */
647                                                 num = INT_MAX;
648                                         
649                                         r = BIO_read(client_io, buf, (int)num);
650                                         if (r != (int)num) /* can't happen */
651                                                 {
652                                                 fprintf(stderr, "ERROR: BIO_read could not read "
653                                                         "BIO_ctrl_pending() bytes");
654                                                 goto err;
655                                                 }
656                                         r = BIO_write(server_io, buf, (int)num);
657                                         if (r != (int)num) /* can't happen */
658                                                 {
659                                                 fprintf(stderr, "ERROR: BIO_write could not write "
660                                                         "BIO_ctrl_get_write_guarantee() bytes");
661                                                 goto err;
662                                                 }
663                                         progress = 1;
664
665                                         if (debug)
666                                                 printf("C->S relaying: %d bytes\n", (int)num);
667                                         }
668                                 }
669                         while (r1 && r2);
670
671                         /* server to client */
672                         do
673                                 {
674                                 r1 = BIO_ctrl_pending(server_io);
675                                 r2 = BIO_ctrl_get_write_guarantee(client_io);
676
677                                 num = r1;
678                                 if (r2 < num)
679                                         num = r2;
680                                 if (num)
681                                         {
682                                         if (sizeof buf < num)
683                                                 num = sizeof buf;
684                                         if (INT_MAX < num)
685                                                 num = INT_MAX;
686                                         
687                                         r = BIO_read(server_io, buf, (int)num);
688                                         if (r != (int)num) /* can't happen */
689                                                 {
690                                                 fprintf(stderr, "ERROR: BIO_read could not read "
691                                                         "BIO_ctrl_pending() bytes");
692                                                 goto err;
693                                                 }
694                                         r = BIO_write(client_io, buf, (int)num);
695                                         if (r != (int)num) /* can't happen */
696                                                 {
697                                                 fprintf(stderr, "ERROR: BIO_write could not write "
698                                                         "BIO_ctrl_get_write_guarantee() bytes");
699                                                 goto err;
700                                                 }
701                                         progress = 1;
702
703                                         if (debug)
704                                                 printf("S->C relaying: %d bytes\n", (int)num);
705                                         }
706                                 }
707                         while (r1 && r2);
708
709                         if (!progress && !prev_progress)
710                                 if (cw_num > 0 || cr_num > 0 || sw_num > 0 || sr_num > 0)
711                                         {
712                                         fprintf(stderr, "ERROR: got stuck\n");
713                                         if (strcmp("SSLv2", SSL_get_version(c_ssl)) == 0)
714                                                 {
715                                                 fprintf(stderr, "This can happen for SSL2 because "
716                                                         "CLIENT-FINISHED and SERVER-VERIFY are written \n"
717                                                         "concurrently ...");
718                                                 if (strncmp("2SCF", SSL_state_string(c_ssl), 4) == 0
719                                                         && strncmp("2SSV", SSL_state_string(s_ssl), 4) == 0)
720                                                         {
721                                                         fprintf(stderr, " ok.\n");
722                                                         goto end;
723                                                         }
724                                                 }
725                                         fprintf(stderr, " ERROR.\n");
726                                         goto err;
727                                         }
728                         prev_progress = progress;
729                         }
730                 }
731         while (cw_num > 0 || cr_num > 0 || sw_num > 0 || sr_num > 0);
732
733         ciph = SSL_get_current_cipher(c_ssl);
734         if (verbose)
735                 fprintf(stdout,"DONE via BIO pair, protocol %s, cipher %s, %s\n",
736                         SSL_get_version(c_ssl),
737                         SSL_CIPHER_get_version(ciph),
738                         SSL_CIPHER_get_name(ciph));
739  end:
740         ret = 0;
741
742  err:
743         ERR_print_errors(bio_err);
744         
745         if (server)
746                 BIO_free(server);
747         if (server_io)
748                 BIO_free(server_io);
749         if (client)
750                 BIO_free(client);
751         if (client_io)
752                 BIO_free(client_io);
753         if (s_ssl_bio)
754                 BIO_free(s_ssl_bio);
755         if (c_ssl_bio)
756                 BIO_free(c_ssl_bio);
757
758         return ret;
759         }
760
761
762 #define W_READ  1
763 #define W_WRITE 2
764 #define C_DONE  1
765 #define S_DONE  2
766
767 int doit(SSL *s_ssl, SSL *c_ssl, long count)
768         {
769         MS_STATIC char cbuf[1024*8],sbuf[1024*8];
770         long cw_num=count,cr_num=count;
771         long sw_num=count,sr_num=count;
772         int ret=1;
773         BIO *c_to_s=NULL;
774         BIO *s_to_c=NULL;
775         BIO *c_bio=NULL;
776         BIO *s_bio=NULL;
777         int c_r,c_w,s_r,s_w;
778         int c_want,s_want;
779         int i,j;
780         int done=0;
781         int c_write,s_write;
782         int do_server=0,do_client=0;
783         SSL_CIPHER *ciph;
784
785         c_to_s=BIO_new(BIO_s_mem());
786         s_to_c=BIO_new(BIO_s_mem());
787         if ((s_to_c == NULL) || (c_to_s == NULL))
788                 {
789                 ERR_print_errors(bio_err);
790                 goto err;
791                 }
792
793         c_bio=BIO_new(BIO_f_ssl());
794         s_bio=BIO_new(BIO_f_ssl());
795         if ((c_bio == NULL) || (s_bio == NULL))
796                 {
797                 ERR_print_errors(bio_err);
798                 goto err;
799                 }
800
801         SSL_set_connect_state(c_ssl);
802         SSL_set_bio(c_ssl,s_to_c,c_to_s);
803         BIO_set_ssl(c_bio,c_ssl,BIO_NOCLOSE);
804
805         SSL_set_accept_state(s_ssl);
806         SSL_set_bio(s_ssl,c_to_s,s_to_c);
807         BIO_set_ssl(s_bio,s_ssl,BIO_NOCLOSE);
808
809         c_r=0; s_r=1;
810         c_w=1; s_w=0;
811         c_want=W_WRITE;
812         s_want=0;
813         c_write=1,s_write=0;
814
815         /* We can always do writes */
816         for (;;)
817                 {
818                 do_server=0;
819                 do_client=0;
820
821                 i=(int)BIO_pending(s_bio);
822                 if ((i && s_r) || s_w) do_server=1;
823
824                 i=(int)BIO_pending(c_bio);
825                 if ((i && c_r) || c_w) do_client=1;
826
827                 if (do_server && debug)
828                         {
829                         if (SSL_in_init(s_ssl))
830                                 printf("server waiting in SSL_accept - %s\n",
831                                         SSL_state_string_long(s_ssl));
832 /*                      else if (s_write)
833                                 printf("server:SSL_write()\n");
834                         else
835                                 printf("server:SSL_read()\n"); */
836                         }
837
838                 if (do_client && debug)
839                         {
840                         if (SSL_in_init(c_ssl))
841                                 printf("client waiting in SSL_connect - %s\n",
842                                         SSL_state_string_long(c_ssl));
843 /*                      else if (c_write)
844                                 printf("client:SSL_write()\n");
845                         else
846                                 printf("client:SSL_read()\n"); */
847                         }
848
849                 if (!do_client && !do_server)
850                         {
851                         fprintf(stdout,"ERROR IN STARTUP\n");
852                         ERR_print_errors(bio_err);
853                         break;
854                         }
855                 if (do_client && !(done & C_DONE))
856                         {
857                         if (c_write)
858                                 {
859                                 j=(cw_num > (long)sizeof(cbuf))
860                                         ?sizeof(cbuf):(int)cw_num;
861                                 i=BIO_write(c_bio,cbuf,j);
862                                 if (i < 0)
863                                         {
864                                         c_r=0;
865                                         c_w=0;
866                                         if (BIO_should_retry(c_bio))
867                                                 {
868                                                 if (BIO_should_read(c_bio))
869                                                         c_r=1;
870                                                 if (BIO_should_write(c_bio))
871                                                         c_w=1;
872                                                 }
873                                         else
874                                                 {
875                                                 fprintf(stderr,"ERROR in CLIENT\n");
876                                                 ERR_print_errors(bio_err);
877                                                 goto err;
878                                                 }
879                                         }
880                                 else if (i == 0)
881                                         {
882                                         fprintf(stderr,"SSL CLIENT STARTUP FAILED\n");
883                                         goto err;
884                                         }
885                                 else
886                                         {
887                                         if (debug)
888                                                 printf("client wrote %d\n",i);
889                                         /* ok */
890                                         s_r=1;
891                                         c_write=0;
892                                         cw_num-=i;
893                                         }
894                                 }
895                         else
896                                 {
897                                 i=BIO_read(c_bio,cbuf,sizeof(cbuf));
898                                 if (i < 0)
899                                         {
900                                         c_r=0;
901                                         c_w=0;
902                                         if (BIO_should_retry(c_bio))
903                                                 {
904                                                 if (BIO_should_read(c_bio))
905                                                         c_r=1;
906                                                 if (BIO_should_write(c_bio))
907                                                         c_w=1;
908                                                 }
909                                         else
910                                                 {
911                                                 fprintf(stderr,"ERROR in CLIENT\n");
912                                                 ERR_print_errors(bio_err);
913                                                 goto err;
914                                                 }
915                                         }
916                                 else if (i == 0)
917                                         {
918                                         fprintf(stderr,"SSL CLIENT STARTUP FAILED\n");
919                                         goto err;
920                                         }
921                                 else
922                                         {
923                                         if (debug)
924                                                 printf("client read %d\n",i);
925                                         cr_num-=i;
926                                         if (sw_num > 0)
927                                                 {
928                                                 s_write=1;
929                                                 s_w=1;
930                                                 }
931                                         if (cr_num <= 0)
932                                                 {
933                                                 s_write=1;
934                                                 s_w=1;
935                                                 done=S_DONE|C_DONE;
936                                                 }
937                                         }
938                                 }
939                         }
940
941                 if (do_server && !(done & S_DONE))
942                         {
943                         if (!s_write)
944                                 {
945                                 i=BIO_read(s_bio,sbuf,sizeof(cbuf));
946                                 if (i < 0)
947                                         {
948                                         s_r=0;
949                                         s_w=0;
950                                         if (BIO_should_retry(s_bio))
951                                                 {
952                                                 if (BIO_should_read(s_bio))
953                                                         s_r=1;
954                                                 if (BIO_should_write(s_bio))
955                                                         s_w=1;
956                                                 }
957                                         else
958                                                 {
959                                                 fprintf(stderr,"ERROR in SERVER\n");
960                                                 ERR_print_errors(bio_err);
961                                                 goto err;
962                                                 }
963                                         }
964                                 else if (i == 0)
965                                         {
966                                         ERR_print_errors(bio_err);
967                                         fprintf(stderr,"SSL SERVER STARTUP FAILED in SSL_read\n");
968                                         goto err;
969                                         }
970                                 else
971                                         {
972                                         if (debug)
973                                                 printf("server read %d\n",i);
974                                         sr_num-=i;
975                                         if (cw_num > 0)
976                                                 {
977                                                 c_write=1;
978                                                 c_w=1;
979                                                 }
980                                         if (sr_num <= 0)
981                                                 {
982                                                 s_write=1;
983                                                 s_w=1;
984                                                 c_write=0;
985                                                 }
986                                         }
987                                 }
988                         else
989                                 {
990                                 j=(sw_num > (long)sizeof(sbuf))?
991                                         sizeof(sbuf):(int)sw_num;
992                                 i=BIO_write(s_bio,sbuf,j);
993                                 if (i < 0)
994                                         {
995                                         s_r=0;
996                                         s_w=0;
997                                         if (BIO_should_retry(s_bio))
998                                                 {
999                                                 if (BIO_should_read(s_bio))
1000                                                         s_r=1;
1001                                                 if (BIO_should_write(s_bio))
1002                                                         s_w=1;
1003                                                 }
1004                                         else
1005                                                 {
1006                                                 fprintf(stderr,"ERROR in SERVER\n");
1007                                                 ERR_print_errors(bio_err);
1008                                                 goto err;
1009                                                 }
1010                                         }
1011                                 else if (i == 0)
1012                                         {
1013                                         ERR_print_errors(bio_err);
1014                                         fprintf(stderr,"SSL SERVER STARTUP FAILED in SSL_write\n");
1015                                         goto err;
1016                                         }
1017                                 else
1018                                         {
1019                                         if (debug)
1020                                                 printf("server wrote %d\n",i);
1021                                         sw_num-=i;
1022                                         s_write=0;
1023                                         c_r=1;
1024                                         if (sw_num <= 0)
1025                                                 done|=S_DONE;
1026                                         }
1027                                 }
1028                         }
1029
1030                 if ((done & S_DONE) && (done & C_DONE)) break;
1031                 }
1032
1033         ciph=SSL_get_current_cipher(c_ssl);
1034         if (verbose)
1035                 fprintf(stdout,"DONE, protocol %s, cipher %s, %s\n",
1036                         SSL_get_version(c_ssl),
1037                         SSL_CIPHER_get_version(ciph),
1038                         SSL_CIPHER_get_name(ciph));
1039         ret=0;
1040 err:
1041         /* We have to set the BIO's to NULL otherwise they will be
1042          * Free()ed twice.  Once when th s_ssl is SSL_free()ed and
1043          * again when c_ssl is SSL_free()ed.
1044          * This is a hack required because s_ssl and c_ssl are sharing the same
1045          * BIO structure and SSL_set_bio() and SSL_free() automatically
1046          * BIO_free non NULL entries.
1047          * You should not normally do this or be required to do this */
1048         if (s_ssl != NULL)
1049                 {
1050                 s_ssl->rbio=NULL;
1051                 s_ssl->wbio=NULL;
1052                 }
1053         if (c_ssl != NULL)
1054                 {
1055                 c_ssl->rbio=NULL;
1056                 c_ssl->wbio=NULL;
1057                 }
1058
1059         if (c_to_s != NULL) BIO_free(c_to_s);
1060         if (s_to_c != NULL) BIO_free(s_to_c);
1061         if (c_bio != NULL) BIO_free_all(c_bio);
1062         if (s_bio != NULL) BIO_free_all(s_bio);
1063         return(ret);
1064         }
1065
1066 int MS_CALLBACK verify_callback(int ok, X509_STORE_CTX *ctx)
1067         {
1068         char *s,buf[256];
1069
1070         s=X509_NAME_oneline(X509_get_subject_name(ctx->current_cert),buf,256);
1071         if (s != NULL)
1072                 {
1073                 if (ok)
1074                         fprintf(stderr,"depth=%d %s\n",ctx->error_depth,buf);
1075                 else
1076                         fprintf(stderr,"depth=%d error=%d %s\n",
1077                                 ctx->error_depth,ctx->error,buf);
1078                 }
1079
1080         if (ok == 0)
1081                 {
1082                 switch (ctx->error)
1083                         {
1084                 case X509_V_ERR_CERT_NOT_YET_VALID:
1085                 case X509_V_ERR_CERT_HAS_EXPIRED:
1086                 case X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT:
1087                         ok=1;
1088                         }
1089                 }
1090
1091         return(ok);
1092         }
1093
1094 #ifndef NO_DH
1095 static unsigned char dh512_p[]={
1096         0xDA,0x58,0x3C,0x16,0xD9,0x85,0x22,0x89,0xD0,0xE4,0xAF,0x75,
1097         0x6F,0x4C,0xCA,0x92,0xDD,0x4B,0xE5,0x33,0xB8,0x04,0xFB,0x0F,
1098         0xED,0x94,0xEF,0x9C,0x8A,0x44,0x03,0xED,0x57,0x46,0x50,0xD3,
1099         0x69,0x99,0xDB,0x29,0xD7,0x76,0x27,0x6B,0xA2,0xD3,0xD4,0x12,
1100         0xE2,0x18,0xF4,0xDD,0x1E,0x08,0x4C,0xF6,0xD8,0x00,0x3E,0x7C,
1101         0x47,0x74,0xE8,0x33,
1102         };
1103 static unsigned char dh512_g[]={
1104         0x02,
1105         };
1106
1107 static DH *get_dh512(void)
1108         {
1109         DH *dh=NULL;
1110
1111         if ((dh=DH_new()) == NULL) return(NULL);
1112         dh->p=BN_bin2bn(dh512_p,sizeof(dh512_p),NULL);
1113         dh->g=BN_bin2bn(dh512_g,sizeof(dh512_g),NULL);
1114         if ((dh->p == NULL) || (dh->g == NULL))
1115                 return(NULL);
1116         return(dh);
1117         }
1118 #endif
1119
1120 #ifndef NO_RSA
1121 static RSA MS_CALLBACK *tmp_rsa_cb(SSL *s, int is_export, int keylength)
1122         {
1123         static RSA *rsa_tmp=NULL;
1124
1125         if (rsa_tmp == NULL)
1126                 {
1127                 BIO_printf(bio_err,"Generating temp (%d bit) RSA key...",keylength);
1128                 (void)BIO_flush(bio_err);
1129                 rsa_tmp=RSA_generate_key(keylength,RSA_F4,NULL,NULL);
1130                 BIO_printf(bio_err,"\n");
1131                 (void)BIO_flush(bio_err);
1132                 }
1133         return(rsa_tmp);
1134         }
1135 #endif