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