]> git.sur5r.net Git - openldap/blob - libraries/libldap/tls.c
Sync with HEAD
[openldap] / libraries / libldap / tls.c
1 /* tls.c - Handle tls/ssl using SSLeay or OpenSSL. */
2 /* $OpenLDAP$ */
3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4  *
5  * Copyright 1998-2006 The OpenLDAP Foundation.
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted only as authorized by the OpenLDAP
10  * Public License.
11  *
12  * A copy of this license is available in the file LICENSE in the
13  * top-level directory of the distribution or, alternatively, at
14  * <http://www.OpenLDAP.org/license.html>.
15  */
16
17 #include "portable.h"
18 #include "ldap_config.h"
19
20 #include <stdio.h>
21
22 #include <ac/stdlib.h>
23 #include <ac/errno.h>
24 #include <ac/socket.h>
25 #include <ac/string.h>
26 #include <ac/ctype.h>
27 #include <ac/time.h>
28 #include <ac/unistd.h>
29 #include <ac/param.h>
30 #include <ac/dirent.h>
31
32 #include "ldap-int.h"
33
34 #ifdef HAVE_TLS
35
36 #ifdef LDAP_R_COMPILE
37 #include <ldap_pvt_thread.h>
38 #endif
39
40 #ifdef HAVE_OPENSSL_SSL_H
41 #include <openssl/ssl.h>
42 #include <openssl/x509v3.h>
43 #include <openssl/err.h>
44 #include <openssl/rand.h>
45 #include <openssl/safestack.h>
46 #elif defined( HAVE_SSL_H )
47 #include <ssl.h>
48 #endif
49
50 static int  tls_opt_trace = 1;
51 static char *tls_opt_randfile = NULL;
52
53 #define HAS_TLS( sb )   ber_sockbuf_ctrl( sb, LBER_SB_OPT_HAS_IO, \
54                                 (void *)&sb_tls_sbio )
55
56 static void tls_report_error( void );
57
58 static void tls_info_cb( const SSL *ssl, int where, int ret );
59 static int tls_verify_cb( int ok, X509_STORE_CTX *ctx );
60 static int tls_verify_ok( int ok, X509_STORE_CTX *ctx );
61 static RSA * tls_tmp_rsa_cb( SSL *ssl, int is_export, int key_length );
62 static STACK_OF(X509_NAME) * get_ca_list( char * bundle, char * dir );
63
64 static DH * tls_tmp_dh_cb( SSL *ssl, int is_export, int key_length );
65
66 typedef struct dhplist {
67         struct dhplist *next;
68         int keylength;
69         DH *param;
70 } dhplist;
71
72 static dhplist *dhparams;
73
74 static int tls_seed_PRNG( const char *randfile );
75
76 #ifdef LDAP_R_COMPILE
77 /*
78  * provide mutexes for the SSLeay library.
79  */
80 static ldap_pvt_thread_mutex_t  tls_mutexes[CRYPTO_NUM_LOCKS];
81
82 static void tls_locking_cb( int mode, int type, const char *file, int line )
83 {
84         if ( mode & CRYPTO_LOCK ) {
85                 ldap_pvt_thread_mutex_lock( &tls_mutexes[type] );
86         } else {
87                 ldap_pvt_thread_mutex_unlock( &tls_mutexes[type] );
88         }
89 }
90
91 /*
92  * an extra mutex for the default ctx.
93  */
94
95 static ldap_pvt_thread_mutex_t tls_def_ctx_mutex;
96
97 static void tls_init_threads( void )
98 {
99         int i;
100
101         for( i=0; i< CRYPTO_NUM_LOCKS ; i++ ) {
102                 ldap_pvt_thread_mutex_init( &tls_mutexes[i] );
103         }
104         CRYPTO_set_locking_callback( tls_locking_cb );
105         /* FIXME: the thread id should be added somehow... */
106
107         ldap_pvt_thread_mutex_init( &tls_def_ctx_mutex );
108 }
109 #endif /* LDAP_R_COMPILE */
110
111 void
112 ldap_int_tls_destroy( struct ldapoptions *lo )
113 {
114         if ( lo->ldo_tls_ctx ) {
115                 SSL_CTX_free( lo->ldo_tls_ctx );
116                 lo->ldo_tls_ctx = NULL;
117         }
118
119         if ( lo->ldo_tls_certfile ) {
120                 LDAP_FREE( lo->ldo_tls_certfile );
121                 lo->ldo_tls_certfile = NULL;
122         }
123         if ( lo->ldo_tls_keyfile ) {
124                 LDAP_FREE( lo->ldo_tls_keyfile );
125                 lo->ldo_tls_keyfile = NULL;
126         }
127         if ( lo->ldo_tls_dhfile ) {
128                 LDAP_FREE( lo->ldo_tls_dhfile );
129                 lo->ldo_tls_dhfile = NULL;
130         }
131         if ( lo->ldo_tls_cacertfile ) {
132                 LDAP_FREE( lo->ldo_tls_cacertfile );
133                 lo->ldo_tls_cacertfile = NULL;
134         }
135         if ( lo->ldo_tls_cacertdir ) {
136                 LDAP_FREE( lo->ldo_tls_cacertdir );
137                 lo->ldo_tls_cacertdir = NULL;
138         }
139         if ( lo->ldo_tls_ciphersuite ) {
140                 LDAP_FREE( lo->ldo_tls_ciphersuite );
141                 lo->ldo_tls_ciphersuite = NULL;
142         }
143 }
144
145 /*
146  * Tear down the TLS subsystem. Should only be called once.
147  */
148 void
149 ldap_pvt_tls_destroy( void )
150 {
151         struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();   
152
153         ldap_int_tls_destroy( lo );
154
155         EVP_cleanup();
156         ERR_remove_state(0);
157         ERR_free_strings();
158
159         if ( tls_opt_randfile ) {
160                 LDAP_FREE( tls_opt_randfile );
161                 tls_opt_randfile = NULL;
162         }
163 }
164
165 /*
166  * Initialize TLS subsystem. Should be called only once.
167  */
168 int
169 ldap_pvt_tls_init( void )
170 {
171         static int tls_initialized = 0;
172
173         if ( tls_initialized++ ) return 0;
174
175 #ifdef HAVE_EBCDIC
176         {
177                 char *file = LDAP_STRDUP( tls_opt_randfile );
178                 if ( file ) __atoe( file );
179                 (void) tls_seed_PRNG( file );
180                 LDAP_FREE( file );
181         }
182 #else
183         (void) tls_seed_PRNG( tls_opt_randfile );
184 #endif
185
186 #ifdef LDAP_R_COMPILE
187         tls_init_threads();
188 #endif
189
190         SSL_load_error_strings();
191         SSLeay_add_ssl_algorithms();
192
193         /* FIXME: mod_ssl does this */
194         X509V3_add_standard_extensions();
195         return 0;
196 }
197
198 /*
199  * initialize a new TLS context
200  */
201 static int
202 ldap_int_tls_init_ctx( struct ldapoptions *lo, int is_server )
203 {
204         STACK_OF(X509_NAME) *calist;
205         int i, rc = 0;
206         char *ciphersuite = lo->ldo_tls_ciphersuite;
207         char *cacertfile = lo->ldo_tls_cacertfile;
208         char *cacertdir = lo->ldo_tls_cacertdir;
209         char *certfile = lo->ldo_tls_certfile;
210         char *keyfile = lo->ldo_tls_keyfile;
211         char *dhfile = lo->ldo_tls_dhfile;
212
213         if ( lo->ldo_tls_ctx )
214                 return 0;
215
216         if ( is_server && !certfile && !keyfile && !cacertfile && !cacertdir ) {
217                 /* minimum configuration not provided */
218                 return LDAP_NOT_SUPPORTED;
219         }
220
221 #ifdef HAVE_EBCDIC
222         /* This ASCII/EBCDIC handling is a real pain! */
223         if ( ciphersuite ) {
224                 ciphersuite = LDAP_STRDUP( ciphersuite );
225                 __atoe( ciphersuite );
226         }
227         if ( cacertfile ) {
228                 cacertfile = LDAP_STRDUP( cacertfile );
229                 __atoe( cacertfile );
230         }
231         if ( cacertdir ) {
232                 cacertdir = LDAP_STRDUP( cacertdir );
233                 __atoe( cacertdir );
234         }
235         if ( certfile ) {
236                 certfile = LDAP_STRDUP( certfile );
237                 __atoe( certfile );
238         }
239         if ( keyfile ) {
240                 keyfile = LDAP_STRDUP( keyfile );
241                 __atoe( keyfile );
242         }
243         if ( dhfile ) {
244                 dhfile = LDAP_STRDUP( dhfile );
245                 __atoe( dhfile );
246         }
247 #endif
248         lo->ldo_tls_ctx = SSL_CTX_new( SSLv23_method() );
249         if ( lo->ldo_tls_ctx == NULL ) {
250                 Debug( LDAP_DEBUG_ANY,
251                    "TLS: could not allocate default ctx (%lu).\n",
252                         ERR_peek_error(),0,0);
253                 rc = -1;
254                 goto error_exit;
255         }
256
257         if ( is_server ) {
258                 SSL_CTX_set_session_id_context( lo->ldo_tls_ctx,
259                         (const unsigned char *) "OpenLDAP", sizeof("OpenLDAP")-1 );
260         }
261
262         if ( lo->ldo_tls_ciphersuite &&
263                 !SSL_CTX_set_cipher_list( lo->ldo_tls_ctx, ciphersuite ) )
264         {
265                 Debug( LDAP_DEBUG_ANY,
266                            "TLS: could not set cipher list %s.\n",
267                            lo->ldo_tls_ciphersuite, 0, 0 );
268                 tls_report_error();
269                 rc = -1;
270                 goto error_exit;
271         }
272
273         if (lo->ldo_tls_cacertfile != NULL || lo->ldo_tls_cacertdir != NULL) {
274                 if ( !SSL_CTX_load_verify_locations( lo->ldo_tls_ctx,
275                                 cacertfile, cacertdir ) ||
276                         !SSL_CTX_set_default_verify_paths( lo->ldo_tls_ctx ) )
277                 {
278                         Debug( LDAP_DEBUG_ANY, "TLS: "
279                                 "could not load verify locations (file:`%s',dir:`%s').\n",
280                                 lo->ldo_tls_cacertfile ? lo->ldo_tls_cacertfile : "",
281                                 lo->ldo_tls_cacertdir ? lo->ldo_tls_cacertdir : "",
282                                 0 );
283                         tls_report_error();
284                         rc = -1;
285                         goto error_exit;
286                 }
287
288                 if ( is_server ) {
289                         /* List of CA names to send to a client */
290                         calist = get_ca_list( cacertfile, cacertdir );
291                         if ( !calist ) {
292                                 Debug( LDAP_DEBUG_ANY, "TLS: "
293                                         "could not load client CA list (file:`%s',dir:`%s').\n",
294                                         lo->ldo_tls_cacertfile ? lo->ldo_tls_cacertfile : "",
295                                         lo->ldo_tls_cacertdir ? lo->ldo_tls_cacertdir : "",
296                                         0 );
297                                 tls_report_error();
298                                 rc = -1;
299                                 goto error_exit;
300                         }
301
302                         SSL_CTX_set_client_CA_list( lo->ldo_tls_ctx, calist );
303                 }
304         }
305
306         if ( lo->ldo_tls_certfile &&
307                 !SSL_CTX_use_certificate_file( lo->ldo_tls_ctx,
308                         certfile, SSL_FILETYPE_PEM ) )
309         {
310                 Debug( LDAP_DEBUG_ANY,
311                         "TLS: could not use certificate `%s'.\n",
312                         lo->ldo_tls_certfile,0,0);
313                 tls_report_error();
314                 rc = -1;
315                 goto error_exit;
316         }
317
318         /* Key validity is checked automatically if cert has already been set */
319         if ( lo->ldo_tls_keyfile &&
320                 !SSL_CTX_use_PrivateKey_file( lo->ldo_tls_ctx,
321                         keyfile, SSL_FILETYPE_PEM ) )
322         {
323                 Debug( LDAP_DEBUG_ANY,
324                         "TLS: could not use key file `%s'.\n",
325                         lo->ldo_tls_keyfile,0,0);
326                 tls_report_error();
327                 rc = -1;
328                 goto error_exit;
329         }
330
331         if ( lo->ldo_tls_dhfile ) {
332                 DH *dh = NULL;
333                 BIO *bio;
334                 dhplist *p;
335
336                 if (( bio=BIO_new_file( dhfile,"r" )) == NULL ) {
337                         Debug( LDAP_DEBUG_ANY,
338                                 "TLS: could not use DH parameters file `%s'.\n",
339                                 lo->ldo_tls_dhfile,0,0);
340                         tls_report_error();
341                         rc = -1;
342                         goto error_exit;
343                 }
344                 while (( dh=PEM_read_bio_DHparams( bio, NULL, NULL, NULL ))) {
345                         p = LDAP_MALLOC( sizeof(dhplist) );
346                         if ( p != NULL ) {
347                                 p->keylength = DH_size( dh ) * 8;
348                                 p->param = dh;
349                                 p->next = dhparams;
350                                 dhparams = p;
351                         }
352                 }
353                 BIO_free( bio );
354         }
355
356         if ( tls_opt_trace ) {
357                 SSL_CTX_set_info_callback( (SSL_CTX *)lo->ldo_tls_ctx, tls_info_cb );
358         }
359
360         i = SSL_VERIFY_NONE;
361         if ( lo->ldo_tls_require_cert ) {
362                 i = SSL_VERIFY_PEER;
363                 if ( lo->ldo_tls_require_cert == LDAP_OPT_X_TLS_DEMAND ||
364                          lo->ldo_tls_require_cert == LDAP_OPT_X_TLS_HARD ) {
365                         i |= SSL_VERIFY_FAIL_IF_NO_PEER_CERT;
366                 }
367         }
368
369         SSL_CTX_set_verify( lo->ldo_tls_ctx, i,
370                 lo->ldo_tls_require_cert == LDAP_OPT_X_TLS_ALLOW ?
371                 tls_verify_ok : tls_verify_cb );
372         SSL_CTX_set_tmp_rsa_callback( lo->ldo_tls_ctx, tls_tmp_rsa_cb );
373         if ( lo->ldo_tls_dhfile ) {
374                 SSL_CTX_set_tmp_dh_callback( lo->ldo_tls_ctx, tls_tmp_dh_cb );
375         }
376 #ifdef HAVE_OPENSSL_CRL
377         if ( lo->ldo_tls_crlcheck ) {
378                 X509_STORE *x509_s = SSL_CTX_get_cert_store( lo->ldo_tls_ctx );
379                 if ( lo->ldo_tls_crlcheck == LDAP_OPT_X_TLS_CRL_PEER ) {
380                         X509_STORE_set_flags( x509_s, X509_V_FLAG_CRL_CHECK );
381                 } else if ( lo->ldo_tls_crlcheck == LDAP_OPT_X_TLS_CRL_ALL ) {
382                         X509_STORE_set_flags( x509_s, 
383                                         X509_V_FLAG_CRL_CHECK | X509_V_FLAG_CRL_CHECK_ALL  );
384                 }
385         }
386 #endif
387
388 error_exit:
389         if ( rc == -1 && lo->ldo_tls_ctx != NULL ) {
390                 SSL_CTX_free( lo->ldo_tls_ctx );
391                 lo->ldo_tls_ctx = NULL;
392         }
393 #ifdef HAVE_EBCDIC
394         LDAP_FREE( ciphersuite );
395         LDAP_FREE( cacertfile );
396         LDAP_FREE( cacertdir );
397         LDAP_FREE( certfile );
398         LDAP_FREE( keyfile );
399         LDAP_FREE( dhfile );
400 #endif
401         return rc;
402 }
403
404 /*
405  * initialize the default context
406  */
407 int
408 ldap_pvt_tls_init_def_ctx( int is_server )
409 {
410         struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();   
411         int rc;
412 #ifdef LDAP_R_COMPILE
413         ldap_pvt_thread_mutex_lock( &tls_def_ctx_mutex );
414 #endif
415         rc = ldap_int_tls_init_ctx( lo, is_server );
416 #ifdef LDAP_R_COMPILE
417         ldap_pvt_thread_mutex_unlock( &tls_def_ctx_mutex );
418 #endif
419         return rc;
420 }
421
422 static STACK_OF(X509_NAME) *
423 get_ca_list( char * bundle, char * dir )
424 {
425         STACK_OF(X509_NAME) *ca_list = NULL;
426
427         if ( bundle ) {
428                 ca_list = SSL_load_client_CA_file( bundle );
429         }
430 #if defined(HAVE_DIRENT_H) || defined(dirent)
431         if ( dir ) {
432                 int freeit = 0;
433
434                 if ( !ca_list ) {
435                         ca_list = sk_X509_NAME_new_null();
436                         freeit = 1;
437                 }
438                 if ( !SSL_add_dir_cert_subjects_to_stack( ca_list, dir ) &&
439                         freeit ) {
440                         sk_X509_NAME_free( ca_list );
441                         ca_list = NULL;
442                 }
443         }
444 #endif
445         return ca_list;
446 }
447
448 static SSL *
449 alloc_handle( void *ctx_arg, int is_server )
450 {
451         SSL_CTX *ctx;
452         SSL     *ssl;
453
454         if ( ctx_arg ) {
455                 ctx = (SSL_CTX *) ctx_arg;
456         } else {
457                 struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();   
458                 if ( ldap_pvt_tls_init_def_ctx( is_server ) < 0 ) return NULL;
459                 ctx = lo->ldo_tls_ctx;
460         }
461
462         ssl = SSL_new( ctx );
463         if ( ssl == NULL ) {
464                 Debug( LDAP_DEBUG_ANY,"TLS: can't create ssl handle.\n",0,0,0);
465                 return NULL;
466         }
467         return ssl;
468 }
469
470 static int
471 update_flags( Sockbuf *sb, SSL * ssl, int rc )
472 {
473         int err = SSL_get_error(ssl, rc);
474
475         sb->sb_trans_needs_read  = 0;
476         sb->sb_trans_needs_write = 0;
477
478         if (err == SSL_ERROR_WANT_READ) {
479                 sb->sb_trans_needs_read  = 1;
480                 return 1;
481
482         } else if (err == SSL_ERROR_WANT_WRITE) {
483                 sb->sb_trans_needs_write = 1;
484                 return 1;
485
486         } else if (err == SSL_ERROR_WANT_CONNECT) {
487                 return 1;
488         }
489         return 0;
490 }
491
492 /*
493  * TLS support for LBER Sockbufs
494  */
495
496 struct tls_data {
497         SSL                     *ssl;
498         Sockbuf_IO_Desc         *sbiod;
499 };
500
501 static BIO_METHOD sb_tls_bio_method;
502
503 static int
504 sb_tls_setup( Sockbuf_IO_Desc *sbiod, void *arg )
505 {
506         struct tls_data         *p;
507         BIO                     *bio;
508
509         assert( sbiod != NULL );
510
511         p = LBER_MALLOC( sizeof( *p ) );
512         if ( p == NULL ) {
513                 return -1;
514         }
515         
516         p->ssl = (SSL *)arg;
517         p->sbiod = sbiod;
518         bio = BIO_new( &sb_tls_bio_method );
519         bio->ptr = (void *)p;
520         SSL_set_bio( p->ssl, bio, bio );
521         sbiod->sbiod_pvt = p;
522         return 0;
523 }
524
525 static int
526 sb_tls_remove( Sockbuf_IO_Desc *sbiod )
527 {
528         struct tls_data         *p;
529         
530         assert( sbiod != NULL );
531         assert( sbiod->sbiod_pvt != NULL );
532
533         p = (struct tls_data *)sbiod->sbiod_pvt;
534         SSL_free( p->ssl );
535         LBER_FREE( sbiod->sbiod_pvt );
536         sbiod->sbiod_pvt = NULL;
537         return 0;
538 }
539
540 static int
541 sb_tls_close( Sockbuf_IO_Desc *sbiod )
542 {
543         struct tls_data         *p;
544         
545         assert( sbiod != NULL );
546         assert( sbiod->sbiod_pvt != NULL );
547
548         p = (struct tls_data *)sbiod->sbiod_pvt;
549         SSL_shutdown( p->ssl );
550         return 0;
551 }
552
553 static int
554 sb_tls_ctrl( Sockbuf_IO_Desc *sbiod, int opt, void *arg )
555 {
556         struct tls_data         *p;
557         
558         assert( sbiod != NULL );
559         assert( sbiod->sbiod_pvt != NULL );
560
561         p = (struct tls_data *)sbiod->sbiod_pvt;
562         
563         if ( opt == LBER_SB_OPT_GET_SSL ) {
564                 *((SSL **)arg) = p->ssl;
565                 return 1;
566
567         } else if ( opt == LBER_SB_OPT_DATA_READY ) {
568                 if( SSL_pending( p->ssl ) > 0 ) {
569                         return 1;
570                 }
571         }
572         
573         return LBER_SBIOD_CTRL_NEXT( sbiod, opt, arg );
574 }
575
576 static ber_slen_t
577 sb_tls_read( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
578 {
579         struct tls_data         *p;
580         ber_slen_t              ret;
581         int                     err;
582
583         assert( sbiod != NULL );
584         assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
585
586         p = (struct tls_data *)sbiod->sbiod_pvt;
587
588         ret = SSL_read( p->ssl, (char *)buf, len );
589 #ifdef HAVE_WINSOCK
590         errno = WSAGetLastError();
591 #endif
592         err = SSL_get_error( p->ssl, ret );
593         if (err == SSL_ERROR_WANT_READ ) {
594                 sbiod->sbiod_sb->sb_trans_needs_read = 1;
595                 errno = EWOULDBLOCK;
596         }
597         else
598                 sbiod->sbiod_sb->sb_trans_needs_read = 0;
599         return ret;
600 }
601
602 static ber_slen_t
603 sb_tls_write( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
604 {
605         struct tls_data         *p;
606         ber_slen_t              ret;
607         int                     err;
608
609         assert( sbiod != NULL );
610         assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
611
612         p = (struct tls_data *)sbiod->sbiod_pvt;
613
614         ret = SSL_write( p->ssl, (char *)buf, len );
615 #ifdef HAVE_WINSOCK
616         errno = WSAGetLastError();
617 #endif
618         err = SSL_get_error( p->ssl, ret );
619         if (err == SSL_ERROR_WANT_WRITE ) {
620                 sbiod->sbiod_sb->sb_trans_needs_write = 1;
621                 errno = EWOULDBLOCK;
622
623         } else {
624                 sbiod->sbiod_sb->sb_trans_needs_write = 0;
625         }
626         return ret;
627 }
628
629 static Sockbuf_IO sb_tls_sbio =
630 {
631         sb_tls_setup,           /* sbi_setup */
632         sb_tls_remove,          /* sbi_remove */
633         sb_tls_ctrl,            /* sbi_ctrl */
634         sb_tls_read,            /* sbi_read */
635         sb_tls_write,           /* sbi_write */
636         sb_tls_close            /* sbi_close */
637 };
638
639 static int
640 sb_tls_bio_create( BIO *b ) {
641         b->init = 1;
642         b->num = 0;
643         b->ptr = NULL;
644         b->flags = 0;
645         return 1;
646 }
647
648 static int
649 sb_tls_bio_destroy( BIO *b )
650 {
651         if ( b == NULL ) return 0;
652
653         b->ptr = NULL;          /* sb_tls_remove() will free it */
654         b->init = 0;
655         b->flags = 0;
656         return 1;
657 }
658
659 static int
660 sb_tls_bio_read( BIO *b, char *buf, int len )
661 {
662         struct tls_data         *p;
663         int                     ret;
664                 
665         if ( buf == NULL || len <= 0 ) return 0;
666
667         p = (struct tls_data *)b->ptr;
668
669         if ( p == NULL || p->sbiod == NULL ) {
670                 return 0;
671         }
672
673         ret = LBER_SBIOD_READ_NEXT( p->sbiod, buf, len );
674
675         BIO_clear_retry_flags( b );
676         if ( ret < 0 ) {
677                 int err = errno;
678                 if ( err == EAGAIN || err == EWOULDBLOCK ) {
679                         BIO_set_retry_read( b );
680                 }
681         }
682
683         return ret;
684 }
685
686 static int
687 sb_tls_bio_write( BIO *b, const char *buf, int len )
688 {
689         struct tls_data         *p;
690         int                     ret;
691         
692         if ( buf == NULL || len <= 0 ) return 0;
693         
694         p = (struct tls_data *)b->ptr;
695
696         if ( p == NULL || p->sbiod == NULL ) {
697                 return 0;
698         }
699
700         ret = LBER_SBIOD_WRITE_NEXT( p->sbiod, (char *)buf, len );
701
702         BIO_clear_retry_flags( b );
703         if ( ret < 0 ) {
704                 int err = errno;
705                 if ( err == EAGAIN || err == EWOULDBLOCK ) {
706                         BIO_set_retry_write( b );
707                 }
708         }
709
710         return ret;
711 }
712
713 static long
714 sb_tls_bio_ctrl( BIO *b, int cmd, long num, void *ptr )
715 {
716         if ( cmd == BIO_CTRL_FLUSH ) {
717                 /* The OpenSSL library needs this */
718                 return 1;
719         }
720         return 0;
721 }
722
723 static int
724 sb_tls_bio_gets( BIO *b, char *buf, int len )
725 {
726         return -1;
727 }
728
729 static int
730 sb_tls_bio_puts( BIO *b, const char *str )
731 {
732         return sb_tls_bio_write( b, str, strlen( str ) );
733 }
734         
735 static BIO_METHOD sb_tls_bio_method =
736 {
737         ( 100 | 0x400 ),                /* it's a source/sink BIO */
738         "sockbuf glue",
739         sb_tls_bio_write,
740         sb_tls_bio_read,
741         sb_tls_bio_puts,
742         sb_tls_bio_gets,
743         sb_tls_bio_ctrl,
744         sb_tls_bio_create,
745         sb_tls_bio_destroy
746 };
747
748 /*
749  * Call this to do a TLS connect on a sockbuf. ctx_arg can be
750  * a SSL_CTX * or NULL, in which case the default ctx is used.
751  *
752  * Return value:
753  *
754  *  0 - Success. Connection is ready for communication.
755  * <0 - Error. Can't create a TLS stream.
756  * >0 - Partial success.
757  *        Do a select (using information from lber_pvt_sb_needs_{read,write}
758  *              and call again.
759  */
760
761 static int
762 ldap_int_tls_connect( LDAP *ld, LDAPConn *conn )
763 {
764         Sockbuf *sb = conn->lconn_sb;
765         int     err;
766         SSL     *ssl;
767
768         if ( HAS_TLS( sb ) ) {
769                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&ssl );
770
771         } else {
772                 struct ldapoptions *lo;
773                 SSL_CTX *ctx;
774
775                 ctx = ld->ld_options.ldo_tls_ctx;
776
777                 ssl = alloc_handle( ctx, 0 );
778
779                 if ( ssl == NULL ) return -1;
780
781 #ifdef LDAP_DEBUG
782                 ber_sockbuf_add_io( sb, &ber_sockbuf_io_debug,
783                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)"tls_" );
784 #endif
785                 ber_sockbuf_add_io( sb, &sb_tls_sbio,
786                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)ssl );
787
788                 lo = LDAP_INT_GLOBAL_OPT();   
789                 if( ctx == NULL ) {
790                         ctx = lo->ldo_tls_ctx;
791                         ld->ld_options.ldo_tls_ctx = ctx;
792                         CRYPTO_add( &ctx->references, 1, CRYPTO_LOCK_SSL_CTX );
793                 }
794                 if ( ld->ld_options.ldo_tls_connect_cb )
795                         ld->ld_options.ldo_tls_connect_cb( ld, ssl, ctx,
796                         ld->ld_options.ldo_tls_connect_arg );
797                 if ( lo && lo->ldo_tls_connect_cb && lo->ldo_tls_connect_cb !=
798                         ld->ld_options.ldo_tls_connect_cb )
799                         lo->ldo_tls_connect_cb( ld, ssl, ctx, lo->ldo_tls_connect_arg );
800         }
801
802         err = SSL_connect( ssl );
803
804 #ifdef HAVE_WINSOCK
805         errno = WSAGetLastError();
806 #endif
807
808         if ( err <= 0 ) {
809                 if ( update_flags( sb, ssl, err )) {
810                         return 1;
811                 }
812
813                 if ((err = ERR_peek_error())) {
814                         char buf[256];
815
816                         if ( ld->ld_error ) {
817                                 LDAP_FREE( ld->ld_error );
818                         }
819                         ld->ld_error = LDAP_STRDUP(ERR_error_string(err, buf));
820 #ifdef HAVE_EBCDIC
821                         if ( ld->ld_error ) __etoa(ld->ld_error);
822 #endif
823                 }
824
825                 Debug( LDAP_DEBUG_ANY,"TLS: can't connect: %s.\n",
826                         ld->ld_error ? ld->ld_error : "" ,0,0);
827
828                 ber_sockbuf_remove_io( sb, &sb_tls_sbio,
829                         LBER_SBIOD_LEVEL_TRANSPORT );
830 #ifdef LDAP_DEBUG
831                 ber_sockbuf_remove_io( sb, &ber_sockbuf_io_debug,
832                         LBER_SBIOD_LEVEL_TRANSPORT );
833 #endif
834                 return -1;
835         }
836
837         return 0;
838 }
839
840 /*
841  * Call this to do a TLS accept on a sockbuf.
842  * Everything else is the same as with tls_connect.
843  */
844 int
845 ldap_pvt_tls_accept( Sockbuf *sb, void *ctx_arg )
846 {
847         int     err;
848         SSL     *ssl;
849
850         if ( HAS_TLS( sb ) ) {
851                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&ssl );
852
853         } else {
854                 ssl = alloc_handle( ctx_arg, 1 );
855                 if ( ssl == NULL ) return -1;
856
857 #ifdef LDAP_DEBUG
858                 ber_sockbuf_add_io( sb, &ber_sockbuf_io_debug,
859                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)"tls_" );
860 #endif
861                 ber_sockbuf_add_io( sb, &sb_tls_sbio,
862                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)ssl );
863         }
864
865         err = SSL_accept( ssl );
866
867 #ifdef HAVE_WINSOCK
868         errno = WSAGetLastError();
869 #endif
870         if ( err <= 0 ) {
871                 if ( update_flags( sb, ssl, err )) return 1;
872
873                 Debug( LDAP_DEBUG_ANY,"TLS: can't accept.\n",0,0,0 );
874
875                 tls_report_error();
876                 ber_sockbuf_remove_io( sb, &sb_tls_sbio,
877                         LBER_SBIOD_LEVEL_TRANSPORT );
878 #ifdef LDAP_DEBUG
879                 ber_sockbuf_remove_io( sb, &ber_sockbuf_io_debug,
880                         LBER_SBIOD_LEVEL_TRANSPORT );
881 #endif
882                 return -1;
883         }
884
885         return 0;
886 }
887
888 int
889 ldap_pvt_tls_inplace ( Sockbuf *sb )
890 {
891         return HAS_TLS( sb ) ? 1 : 0;
892 }
893
894 int
895 ldap_tls_inplace( LDAP *ld )
896 {
897         Sockbuf         *sb = NULL;
898
899         if ( ld->ld_defconn && ld->ld_defconn->lconn_sb ) {
900                 sb = ld->ld_defconn->lconn_sb;
901
902         } else if ( ld->ld_sb ) {
903                 sb = ld->ld_sb;
904
905         } else {
906                 return 0;
907         }
908
909         return ldap_pvt_tls_inplace( sb );
910 }
911
912 static X509 *
913 tls_get_cert( SSL *s )
914 {
915         /* If peer cert was bad, treat as if no cert was given */
916         if (SSL_get_verify_result(s)) {
917                 /* If we can send an alert, do so */
918                 if (SSL_version(s) != SSL2_VERSION) {
919                         ssl3_send_alert(s,SSL3_AL_WARNING,SSL3_AD_BAD_CERTIFICATE);
920                 }
921                 return NULL;
922         }
923         return SSL_get_peer_certificate(s);
924 }
925
926 int
927 ldap_pvt_tls_get_peer_dn( void *s, struct berval *dn,
928         LDAPDN_rewrite_dummy *func, unsigned flags )
929 {
930         X509 *x;
931         X509_NAME *xn;
932         int rc;
933
934         x = tls_get_cert((SSL *)s);
935
936         if (!x) return LDAP_INVALID_CREDENTIALS;
937         
938         xn = X509_get_subject_name(x);
939         rc = ldap_X509dn2bv(xn, dn, (LDAPDN_rewrite_func *)func, flags);
940         X509_free(x);
941         return rc;
942 }
943
944 char *
945 ldap_pvt_tls_get_peer_hostname( void *s )
946 {
947         X509 *x;
948         X509_NAME *xn;
949         char buf[2048], *p;
950         int ret;
951
952         x = tls_get_cert((SSL *)s);
953         if (!x) return NULL;
954         
955         xn = X509_get_subject_name(x);
956
957         ret = X509_NAME_get_text_by_NID(xn, NID_commonName, buf, sizeof(buf));
958         if( ret == -1 ) {
959                 X509_free(x);
960                 return NULL;
961         }
962
963         p = LDAP_STRDUP(buf);
964         X509_free(x);
965         return p;
966 }
967
968 /* what kind of hostname were we given? */
969 #define IS_DNS  0
970 #define IS_IP4  1
971 #define IS_IP6  2
972
973 int
974 ldap_pvt_tls_check_hostname( LDAP *ld, void *s, const char *name_in )
975 {
976         int i, ret = LDAP_LOCAL_ERROR;
977         X509 *x;
978         const char *name;
979         char *ptr;
980         int ntype = IS_DNS;
981 #ifdef LDAP_PF_INET6
982         struct in6_addr addr;
983 #else
984         struct in_addr addr;
985 #endif
986
987         if( ldap_int_hostname &&
988                 ( !name_in || !strcasecmp( name_in, "localhost" ) ) )
989         {
990                 name = ldap_int_hostname;
991         } else {
992                 name = name_in;
993         }
994
995         x = tls_get_cert((SSL *)s);
996         if (!x) {
997                 Debug( LDAP_DEBUG_ANY,
998                         "TLS: unable to get peer certificate.\n",
999                         0, 0, 0 );
1000                 /* If this was a fatal condition, things would have
1001                  * aborted long before now.
1002                  */
1003                 return LDAP_SUCCESS;
1004         }
1005
1006 #ifdef LDAP_PF_INET6
1007         if (name[0] == '[' && strchr(name, ']')) {
1008                 char *n2 = ldap_strdup(name+1);
1009                 *strchr(n2, ']') = 2;
1010                 if (inet_pton(AF_INET6, n2, &addr))
1011                         ntype = IS_IP6;
1012                 LDAP_FREE(n2);
1013         } else 
1014 #endif
1015         if ((ptr = strrchr(name, '.')) && isdigit((unsigned char)ptr[1])) {
1016                 if (inet_aton(name, (struct in_addr *)&addr)) ntype = IS_IP4;
1017         }
1018         
1019         i = X509_get_ext_by_NID(x, NID_subject_alt_name, -1);
1020         if (i >= 0) {
1021                 X509_EXTENSION *ex;
1022                 STACK_OF(GENERAL_NAME) *alt;
1023
1024                 ex = X509_get_ext(x, i);
1025                 alt = X509V3_EXT_d2i(ex);
1026                 if (alt) {
1027                         int n, len1 = 0, len2 = 0;
1028                         char *domain = NULL;
1029                         GENERAL_NAME *gn;
1030
1031                         if (ntype == IS_DNS) {
1032                                 len1 = strlen(name);
1033                                 domain = strchr(name, '.');
1034                                 if (domain) {
1035                                         len2 = len1 - (domain-name);
1036                                 }
1037                         }
1038                         n = sk_GENERAL_NAME_num(alt);
1039                         for (i=0; i<n; i++) {
1040                                 char *sn;
1041                                 int sl;
1042                                 gn = sk_GENERAL_NAME_value(alt, i);
1043                                 if (gn->type == GEN_DNS) {
1044                                         if (ntype != IS_DNS) continue;
1045
1046                                         sn = (char *) ASN1_STRING_data(gn->d.ia5);
1047                                         sl = ASN1_STRING_length(gn->d.ia5);
1048
1049                                         /* ignore empty */
1050                                         if (sl == 0) continue;
1051
1052                                         /* Is this an exact match? */
1053                                         if ((len1 == sl) && !strncasecmp(name, sn, len1)) {
1054                                                 break;
1055                                         }
1056
1057                                         /* Is this a wildcard match? */
1058                                         if (domain && (sn[0] == '*') && (sn[1] == '.') &&
1059                                                 (len2 == sl-1) && !strncasecmp(domain, &sn[1], len2))
1060                                         {
1061                                                 break;
1062                                         }
1063
1064                                 } else if (gn->type == GEN_IPADD) {
1065                                         if (ntype == IS_DNS) continue;
1066
1067                                         sn = (char *) ASN1_STRING_data(gn->d.ia5);
1068                                         sl = ASN1_STRING_length(gn->d.ia5);
1069
1070 #ifdef LDAP_PF_INET6
1071                                         if (ntype == IS_IP6 && sl != sizeof(struct in6_addr)) {
1072                                                 continue;
1073                                         } else
1074 #endif
1075                                         if (ntype == IS_IP4 && sl != sizeof(struct in_addr)) {
1076                                                 continue;
1077                                         }
1078                                         if (!memcmp(sn, &addr, sl)) {
1079                                                 break;
1080                                         }
1081                                 }
1082                         }
1083
1084                         GENERAL_NAMES_free(alt);
1085                         if (i < n) {    /* Found a match */
1086                                 ret = LDAP_SUCCESS;
1087                         }
1088                 }
1089         }
1090
1091         if (ret != LDAP_SUCCESS) {
1092                 X509_NAME *xn;
1093                 char buf[2048];
1094                 buf[0] = '\0';
1095
1096                 xn = X509_get_subject_name(x);
1097                 if( X509_NAME_get_text_by_NID( xn, NID_commonName,
1098                         buf, sizeof(buf)) == -1)
1099                 {
1100                         Debug( LDAP_DEBUG_ANY,
1101                                 "TLS: unable to get common name from peer certificate.\n",
1102                                 0, 0, 0 );
1103                         ret = LDAP_CONNECT_ERROR;
1104                         if ( ld->ld_error ) {
1105                                 LDAP_FREE( ld->ld_error );
1106                         }
1107                         ld->ld_error = LDAP_STRDUP(
1108                                 _("TLS: unable to get CN from peer certificate"));
1109
1110                 } else if (strcasecmp(name, buf) == 0 ) {
1111                         ret = LDAP_SUCCESS;
1112
1113                 } else if (( buf[0] == '*' ) && ( buf[1] == '.' )) {
1114                         char *domain = strchr(name, '.');
1115                         if( domain ) {
1116                                 size_t dlen = 0;
1117                                 size_t sl;
1118
1119                                 sl = strlen(name);
1120                                 dlen = sl - (domain-name);
1121                                 sl = strlen(buf);
1122
1123                                 /* Is this a wildcard match? */
1124                                 if ((dlen == sl-1) && !strncasecmp(domain, &buf[1], dlen)) {
1125                                         ret = LDAP_SUCCESS;
1126                                 }
1127                         }
1128                 }
1129
1130                 if( ret == LDAP_LOCAL_ERROR ) {
1131                         Debug( LDAP_DEBUG_ANY, "TLS: hostname (%s) does not match "
1132                                 "common name in certificate (%s).\n", 
1133                                 name, buf, 0 );
1134                         ret = LDAP_CONNECT_ERROR;
1135                         if ( ld->ld_error ) {
1136                                 LDAP_FREE( ld->ld_error );
1137                         }
1138                         ld->ld_error = LDAP_STRDUP(
1139                                 _("TLS: hostname does not match CN in peer certificate"));
1140                 }
1141         }
1142         X509_free(x);
1143         return ret;
1144 }
1145
1146 const char *
1147 ldap_pvt_tls_get_peer_issuer( void *s )
1148 {
1149 #if 0   /* currently unused; see ldap_pvt_tls_get_peer_dn() if needed */
1150         X509 *x;
1151         X509_NAME *xn;
1152         char buf[2048], *p;
1153
1154         x = SSL_get_peer_certificate((SSL *)s);
1155
1156         if (!x) return NULL;
1157         
1158         xn = X509_get_issuer_name(x);
1159         p = LDAP_STRDUP(X509_NAME_oneline(xn, buf, sizeof(buf)));
1160         X509_free(x);
1161         return p;
1162 #else
1163         return NULL;
1164 #endif
1165 }
1166
1167 int
1168 ldap_int_tls_config( LDAP *ld, int option, const char *arg )
1169 {
1170         int i;
1171
1172         switch( option ) {
1173         case LDAP_OPT_X_TLS_CACERTFILE:
1174         case LDAP_OPT_X_TLS_CACERTDIR:
1175         case LDAP_OPT_X_TLS_CERTFILE:
1176         case LDAP_OPT_X_TLS_KEYFILE:
1177         case LDAP_OPT_X_TLS_RANDOM_FILE:
1178         case LDAP_OPT_X_TLS_CIPHER_SUITE:
1179         case LDAP_OPT_X_TLS_DHFILE:
1180                 return ldap_pvt_tls_set_option( ld, option, (void *) arg );
1181
1182         case LDAP_OPT_X_TLS_REQUIRE_CERT:
1183         case LDAP_OPT_X_TLS:
1184                 i = -1;
1185                 if ( strcasecmp( arg, "never" ) == 0 ) {
1186                         i = LDAP_OPT_X_TLS_NEVER ;
1187
1188                 } else if ( strcasecmp( arg, "demand" ) == 0 ) {
1189                         i = LDAP_OPT_X_TLS_DEMAND ;
1190
1191                 } else if ( strcasecmp( arg, "allow" ) == 0 ) {
1192                         i = LDAP_OPT_X_TLS_ALLOW ;
1193
1194                 } else if ( strcasecmp( arg, "try" ) == 0 ) {
1195                         i = LDAP_OPT_X_TLS_TRY ;
1196
1197                 } else if ( ( strcasecmp( arg, "hard" ) == 0 ) ||
1198                         ( strcasecmp( arg, "on" ) == 0 ) ||
1199                         ( strcasecmp( arg, "yes" ) == 0) ||
1200                         ( strcasecmp( arg, "true" ) == 0 ) )
1201                 {
1202                         i = LDAP_OPT_X_TLS_HARD ;
1203                 }
1204
1205                 if (i >= 0) {
1206                         return ldap_pvt_tls_set_option( ld, option, &i );
1207                 }
1208                 return -1;
1209 #ifdef HAVE_OPENSSL_CRL
1210         case LDAP_OPT_X_TLS_CRLCHECK:
1211                 i = -1;
1212                 if ( strcasecmp( arg, "none" ) == 0 ) {
1213                         i = LDAP_OPT_X_TLS_CRL_NONE ;
1214                 } else if ( strcasecmp( arg, "peer" ) == 0 ) {
1215                         i = LDAP_OPT_X_TLS_CRL_PEER ;
1216                 } else if ( strcasecmp( arg, "all" ) == 0 ) {
1217                         i = LDAP_OPT_X_TLS_CRL_ALL ;
1218                 }
1219                 if (i >= 0) {
1220                         return ldap_pvt_tls_set_option( ld, option, &i );
1221                 }
1222                 return -1;
1223 #endif
1224         }
1225         return -1;
1226 }
1227
1228 int
1229 ldap_pvt_tls_get_option( LDAP *ld, int option, void *arg )
1230 {
1231         struct ldapoptions *lo;
1232
1233         if( ld != NULL ) {
1234                 assert( LDAP_VALID( ld ) );
1235
1236                 if( !LDAP_VALID( ld ) ) {
1237                         return LDAP_OPT_ERROR;
1238                 }
1239
1240                 lo = &ld->ld_options;
1241
1242         } else {
1243                 /* Get pointer to global option structure */
1244                 lo = LDAP_INT_GLOBAL_OPT();   
1245                 if ( lo == NULL ) {
1246                         return LDAP_NO_MEMORY;
1247                 }
1248         }
1249
1250         switch( option ) {
1251         case LDAP_OPT_X_TLS:
1252                 *(int *)arg = lo->ldo_tls_mode;
1253                 break;
1254         case LDAP_OPT_X_TLS_CTX:
1255                 *(void **)arg = lo->ldo_tls_ctx;
1256                 if ( lo->ldo_tls_ctx ) {
1257                         SSL_CTX *ctx = lo->ldo_tls_ctx;
1258                         CRYPTO_add( &ctx->references, 1, CRYPTO_LOCK_SSL_CTX );
1259                 }
1260                 break;
1261         case LDAP_OPT_X_TLS_CACERTFILE:
1262                 *(char **)arg = lo->ldo_tls_cacertfile ?
1263                         LDAP_STRDUP( lo->ldo_tls_cacertfile ) : NULL;
1264                 break;
1265         case LDAP_OPT_X_TLS_CACERTDIR:
1266                 *(char **)arg = lo->ldo_tls_cacertdir ?
1267                         LDAP_STRDUP( lo->ldo_tls_cacertdir ) : NULL;
1268                 break;
1269         case LDAP_OPT_X_TLS_CERTFILE:
1270                 *(char **)arg = lo->ldo_tls_certfile ?
1271                         LDAP_STRDUP( lo->ldo_tls_certfile ) : NULL;
1272                 break;
1273         case LDAP_OPT_X_TLS_KEYFILE:
1274                 *(char **)arg = lo->ldo_tls_keyfile ?
1275                         LDAP_STRDUP( lo->ldo_tls_keyfile ) : NULL;
1276                 break;
1277         case LDAP_OPT_X_TLS_DHFILE:
1278                 *(char **)arg = lo->ldo_tls_dhfile ?
1279                         LDAP_STRDUP( lo->ldo_tls_dhfile ) : NULL;
1280                 break;
1281         case LDAP_OPT_X_TLS_REQUIRE_CERT:
1282                 *(int *)arg = lo->ldo_tls_require_cert;
1283                 break;
1284 #ifdef HAVE_OPENSSL_CRL
1285         case LDAP_OPT_X_TLS_CRLCHECK:
1286                 *(int *)arg = lo->ldo_tls_crlcheck;
1287                 break;
1288 #endif
1289         case LDAP_OPT_X_TLS_RANDOM_FILE:
1290                 *(char **)arg = tls_opt_randfile ?
1291                         LDAP_STRDUP( tls_opt_randfile ) : NULL;
1292                 break;
1293         case LDAP_OPT_X_TLS_SSL_CTX: {
1294                 void *retval = 0;
1295                 if ( ld != NULL ) {
1296                         LDAPConn *conn = ld->ld_defconn;
1297                         if ( conn != NULL ) {
1298                                 Sockbuf *sb = conn->lconn_sb;
1299                                 retval = ldap_pvt_tls_sb_ctx( sb );
1300                         }
1301                 }
1302                 *(void **)arg = retval;
1303                 break;
1304         }
1305         case LDAP_OPT_X_TLS_CONNECT_CB:
1306                 *(LDAP_TLS_CONNECT_CB **)arg = lo->ldo_tls_connect_cb;
1307                 break;
1308         case LDAP_OPT_X_TLS_CONNECT_ARG:
1309                 *(void **)arg = lo->ldo_tls_connect_arg;
1310                 break;
1311         default:
1312                 return -1;
1313         }
1314         return 0;
1315 }
1316
1317 int
1318 ldap_pvt_tls_set_option( LDAP *ld, int option, void *arg )
1319 {
1320         struct ldapoptions *lo;
1321
1322         if( ld != NULL ) {
1323                 assert( LDAP_VALID( ld ) );
1324
1325                 if( !LDAP_VALID( ld ) ) {
1326                         return LDAP_OPT_ERROR;
1327                 }
1328
1329                 lo = &ld->ld_options;
1330
1331         } else {
1332                 /* Get pointer to global option structure */
1333                 lo = LDAP_INT_GLOBAL_OPT();   
1334                 if ( lo == NULL ) {
1335                         return LDAP_NO_MEMORY;
1336                 }
1337         }
1338
1339         switch( option ) {
1340         case LDAP_OPT_X_TLS:
1341                 if ( !arg ) return -1;
1342
1343                 switch( *(int *) arg ) {
1344                 case LDAP_OPT_X_TLS_NEVER:
1345                 case LDAP_OPT_X_TLS_DEMAND:
1346                 case LDAP_OPT_X_TLS_ALLOW:
1347                 case LDAP_OPT_X_TLS_TRY:
1348                 case LDAP_OPT_X_TLS_HARD:
1349                         if (lo != NULL) {
1350                                 lo->ldo_tls_mode = *(int *)arg;
1351                         }
1352
1353                         return 0;
1354                 }
1355                 return -1;
1356
1357         case LDAP_OPT_X_TLS_CTX:
1358                 if ( lo->ldo_tls_ctx )
1359                         SSL_CTX_free( lo->ldo_tls_ctx );
1360                 lo->ldo_tls_ctx = arg;
1361                 return 0;
1362         case LDAP_OPT_X_TLS_CONNECT_CB:
1363                 lo->ldo_tls_connect_cb = (LDAP_TLS_CONNECT_CB *)arg;
1364                 return 0;
1365         case LDAP_OPT_X_TLS_CONNECT_ARG:
1366                 lo->ldo_tls_connect_arg = arg;
1367                 return 0;
1368         case LDAP_OPT_X_TLS_CACERTFILE:
1369                 if ( lo->ldo_tls_cacertfile ) LDAP_FREE( lo->ldo_tls_cacertfile );
1370                 lo->ldo_tls_cacertfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
1371                 return 0;
1372         case LDAP_OPT_X_TLS_CACERTDIR:
1373                 if ( lo->ldo_tls_cacertdir ) LDAP_FREE( lo->ldo_tls_cacertdir );
1374                 lo->ldo_tls_cacertdir = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
1375                 return 0;
1376         case LDAP_OPT_X_TLS_CERTFILE:
1377                 if ( lo->ldo_tls_certfile ) LDAP_FREE( lo->ldo_tls_certfile );
1378                 lo->ldo_tls_certfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
1379                 return 0;
1380         case LDAP_OPT_X_TLS_KEYFILE:
1381                 if ( lo->ldo_tls_keyfile ) LDAP_FREE( lo->ldo_tls_keyfile );
1382                 lo->ldo_tls_keyfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
1383                 return 0;
1384         case LDAP_OPT_X_TLS_DHFILE:
1385                 if ( lo->ldo_tls_dhfile ) LDAP_FREE( lo->ldo_tls_dhfile );
1386                 lo->ldo_tls_dhfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
1387                 return 0;
1388         case LDAP_OPT_X_TLS_REQUIRE_CERT:
1389                 if ( !arg ) return -1;
1390                 switch( *(int *) arg ) {
1391                 case LDAP_OPT_X_TLS_NEVER:
1392                 case LDAP_OPT_X_TLS_DEMAND:
1393                 case LDAP_OPT_X_TLS_ALLOW:
1394                 case LDAP_OPT_X_TLS_TRY:
1395                 case LDAP_OPT_X_TLS_HARD:
1396                         lo->ldo_tls_require_cert = * (int *) arg;
1397                         return 0;
1398                 }
1399                 return -1;
1400 #ifdef HAVE_OPENSSL_CRL
1401         case LDAP_OPT_X_TLS_CRLCHECK:
1402                 if ( !arg ) return -1;
1403                 switch( *(int *) arg ) {
1404                 case LDAP_OPT_X_TLS_CRL_NONE:
1405                 case LDAP_OPT_X_TLS_CRL_PEER:
1406                 case LDAP_OPT_X_TLS_CRL_ALL:
1407                         lo->ldo_tls_crlcheck = * (int *) arg;
1408                         return 0;
1409                 }
1410                 return -1;
1411 #endif
1412         case LDAP_OPT_X_TLS_CIPHER_SUITE:
1413                 if ( lo->ldo_tls_ciphersuite ) LDAP_FREE( lo->ldo_tls_ciphersuite );
1414                 lo->ldo_tls_ciphersuite = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
1415                 return 0;
1416
1417         case LDAP_OPT_X_TLS_RANDOM_FILE:
1418                 if ( ld != NULL )
1419                         return -1;
1420                 if (tls_opt_randfile ) LDAP_FREE (tls_opt_randfile );
1421                 tls_opt_randfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
1422                 break;
1423
1424         case LDAP_OPT_X_TLS_NEWCTX:
1425                 if ( !arg ) return -1;
1426                 if ( lo->ldo_tls_ctx )
1427                         SSL_CTX_free( lo->ldo_tls_ctx );
1428                 lo->ldo_tls_ctx = NULL;
1429                 return ldap_int_tls_init_ctx( lo, *(int *)arg );
1430         default:
1431                 return -1;
1432         }
1433         return 0;
1434 }
1435
1436 int
1437 ldap_int_tls_start ( LDAP *ld, LDAPConn *conn, LDAPURLDesc *srv )
1438 {
1439         Sockbuf *sb = conn->lconn_sb;
1440         char *host;
1441         void *ssl;
1442
1443         if( srv ) {
1444                 host = srv->lud_host;
1445         } else {
1446                 host = conn->lconn_server->lud_host;
1447         }
1448
1449         /* avoid NULL host */
1450         if( host == NULL ) {
1451                 host = "localhost";
1452         }
1453
1454         (void) ldap_pvt_tls_init();
1455
1456         /*
1457          * Fortunately, the lib uses blocking io...
1458          */
1459         if ( ldap_int_tls_connect( ld, conn ) < 0 ) {
1460                 ld->ld_errno = LDAP_CONNECT_ERROR;
1461                 return (ld->ld_errno);
1462         }
1463
1464         ssl = ldap_pvt_tls_sb_ctx( sb );
1465         assert( ssl != NULL );
1466
1467         /* 
1468          * compare host with name(s) in certificate
1469          */
1470         if (ld->ld_options.ldo_tls_require_cert != LDAP_OPT_X_TLS_NEVER) {
1471                 ld->ld_errno = ldap_pvt_tls_check_hostname( ld, ssl, host );
1472                 if (ld->ld_errno != LDAP_SUCCESS) {
1473                         return ld->ld_errno;
1474                 }
1475         }
1476
1477         return LDAP_SUCCESS;
1478 }
1479
1480 /* Derived from openssl/apps/s_cb.c */
1481 static void
1482 tls_info_cb( const SSL *ssl, int where, int ret )
1483 {
1484         int w;
1485         char *op;
1486         char *state = (char *) SSL_state_string_long( (SSL *)ssl );
1487
1488         w = where & ~SSL_ST_MASK;
1489         if ( w & SSL_ST_CONNECT ) {
1490                 op = "SSL_connect";
1491         } else if ( w & SSL_ST_ACCEPT ) {
1492                 op = "SSL_accept";
1493         } else {
1494                 op = "undefined";
1495         }
1496
1497 #ifdef HAVE_EBCDIC
1498         if ( state ) {
1499                 state = LDAP_STRDUP( state );
1500                 __etoa( state );
1501         }
1502 #endif
1503         if ( where & SSL_CB_LOOP ) {
1504                 Debug( LDAP_DEBUG_TRACE,
1505                            "TLS trace: %s:%s\n",
1506                            op, state, 0 );
1507
1508         } else if ( where & SSL_CB_ALERT ) {
1509                 char *atype = (char *) SSL_alert_type_string_long( ret );
1510                 char *adesc = (char *) SSL_alert_desc_string_long( ret );
1511                 op = ( where & SSL_CB_READ ) ? "read" : "write";
1512 #ifdef HAVE_EBCDIC
1513                 if ( atype ) {
1514                         atype = LDAP_STRDUP( atype );
1515                         __etoa( atype );
1516                 }
1517                 if ( adesc ) {
1518                         adesc = LDAP_STRDUP( adesc );
1519                         __etoa( adesc );
1520                 }
1521 #endif
1522                 Debug( LDAP_DEBUG_TRACE,
1523                            "TLS trace: SSL3 alert %s:%s:%s\n",
1524                            op, atype, adesc );
1525 #ifdef HAVE_EBCDIC
1526                 if ( atype ) LDAP_FREE( atype );
1527                 if ( adesc ) LDAP_FREE( adesc );
1528 #endif
1529         } else if ( where & SSL_CB_EXIT ) {
1530                 if ( ret == 0 ) {
1531                         Debug( LDAP_DEBUG_TRACE,
1532                                    "TLS trace: %s:failed in %s\n",
1533                                    op, state, 0 );
1534                 } else if ( ret < 0 ) {
1535                         Debug( LDAP_DEBUG_TRACE,
1536                                    "TLS trace: %s:error in %s\n",
1537                                    op, state, 0 );
1538                 }
1539         }
1540 #ifdef HAVE_EBCDIC
1541         if ( state ) LDAP_FREE( state );
1542 #endif
1543 }
1544
1545 static int
1546 tls_verify_cb( int ok, X509_STORE_CTX *ctx )
1547 {
1548         X509 *cert;
1549         int errnum;
1550         int errdepth;
1551         X509_NAME *subject;
1552         X509_NAME *issuer;
1553         char *sname;
1554         char *iname;
1555         char *certerr = NULL;
1556
1557         cert = X509_STORE_CTX_get_current_cert( ctx );
1558         errnum = X509_STORE_CTX_get_error( ctx );
1559         errdepth = X509_STORE_CTX_get_error_depth( ctx );
1560
1561         /*
1562          * X509_get_*_name return pointers to the internal copies of
1563          * those things requested.  So do not free them.
1564          */
1565         subject = X509_get_subject_name( cert );
1566         issuer = X509_get_issuer_name( cert );
1567         /* X509_NAME_oneline, if passed a NULL buf, allocate memomry */
1568         sname = X509_NAME_oneline( subject, NULL, 0 );
1569         iname = X509_NAME_oneline( issuer, NULL, 0 );
1570         if ( !ok ) certerr = (char *)X509_verify_cert_error_string( errnum );
1571 #ifdef HAVE_EBCDIC
1572         if ( sname ) __etoa( sname );
1573         if ( iname ) __etoa( iname );
1574         if ( certerr ) {
1575                 certerr = LDAP_STRDUP( certerr );
1576                 __etoa( certerr );
1577         }
1578 #endif
1579         Debug( LDAP_DEBUG_TRACE,
1580                    "TLS certificate verification: depth: %d, err: %d, subject: %s,",
1581                    errdepth, errnum,
1582                    sname ? sname : "-unknown-" );
1583         Debug( LDAP_DEBUG_TRACE, " issuer: %s\n", iname ? iname : "-unknown-", 0, 0 );
1584         if ( !ok ) {
1585                 Debug( LDAP_DEBUG_ANY,
1586                         "TLS certificate verification: Error, %s\n",
1587                         certerr, 0, 0 );
1588         }
1589         if ( sname )
1590                 CRYPTO_free ( sname );
1591         if ( iname )
1592                 CRYPTO_free ( iname );
1593 #ifdef HAVE_EBCDIC
1594         if ( certerr ) LDAP_FREE( certerr );
1595 #endif
1596         return ok;
1597 }
1598
1599 static int
1600 tls_verify_ok( int ok, X509_STORE_CTX *ctx )
1601 {
1602         (void) tls_verify_cb( ok, ctx );
1603         return 1;
1604 }
1605
1606 /* Inspired by ERR_print_errors in OpenSSL */
1607 static void
1608 tls_report_error( void )
1609 {
1610         unsigned long l;
1611         char buf[200];
1612         const char *file;
1613         int line;
1614
1615         while ( ( l = ERR_get_error_line( &file, &line ) ) != 0 ) {
1616                 ERR_error_string_n( l, buf, sizeof( buf ) );
1617 #ifdef HAVE_EBCDIC
1618                 if ( file ) {
1619                         file = LDAP_STRDUP( file );
1620                         __etoa( (char *)file );
1621                 }
1622                 __etoa( buf );
1623 #endif
1624                 Debug( LDAP_DEBUG_ANY, "TLS: %s %s:%d\n",
1625                         buf, file, line );
1626 #ifdef HAVE_EBCDIC
1627                 if ( file ) LDAP_FREE( (void *)file );
1628 #endif
1629         }
1630 }
1631
1632 static RSA *
1633 tls_tmp_rsa_cb( SSL *ssl, int is_export, int key_length )
1634 {
1635         RSA *tmp_rsa;
1636
1637         /* FIXME:  Pregenerate the key on startup */
1638         /* FIXME:  Who frees the key? */
1639         tmp_rsa = RSA_generate_key( key_length, RSA_F4, NULL, NULL );
1640
1641         if ( !tmp_rsa ) {
1642                 Debug( LDAP_DEBUG_ANY,
1643                         "TLS: Failed to generate temporary %d-bit %s RSA key\n",
1644                         key_length, is_export ? "export" : "domestic", 0 );
1645                 return NULL;
1646         }
1647         return tmp_rsa;
1648 }
1649
1650 static int
1651 tls_seed_PRNG( const char *randfile )
1652 {
1653 #ifndef URANDOM_DEVICE
1654         /* no /dev/urandom (or equiv) */
1655         long total=0;
1656         char buffer[MAXPATHLEN];
1657
1658         if (randfile == NULL) {
1659                 /* The seed file is $RANDFILE if defined, otherwise $HOME/.rnd.
1660                  * If $HOME is not set or buffer too small to hold the pathname,
1661                  * an error occurs.     - From RAND_file_name() man page.
1662                  * The fact is that when $HOME is NULL, .rnd is used.
1663                  */
1664                 randfile = RAND_file_name( buffer, sizeof( buffer ) );
1665
1666         } else if (RAND_egd(randfile) > 0) {
1667                 /* EGD socket */
1668                 return 0;
1669         }
1670
1671         if (randfile == NULL) {
1672                 Debug( LDAP_DEBUG_ANY,
1673                         "TLS: Use configuration file or $RANDFILE to define seed PRNG\n",
1674                         0, 0, 0);
1675                 return -1;
1676         }
1677
1678         total = RAND_load_file(randfile, -1);
1679
1680         if (RAND_status() == 0) {
1681                 Debug( LDAP_DEBUG_ANY,
1682                         "TLS: PRNG not been seeded with enough data\n",
1683                         0, 0, 0);
1684                 return -1;
1685         }
1686
1687         /* assume if there was enough bits to seed that it's okay
1688          * to write derived bits to the file
1689          */
1690         RAND_write_file(randfile);
1691
1692 #endif
1693
1694         return 0;
1695 }
1696
1697 struct dhinfo {
1698         int keylength;
1699         const char *pem;
1700         size_t size;
1701 };
1702
1703
1704 /* From the OpenSSL 0.9.7 distro */
1705 static const char dhpem512[] =
1706 "-----BEGIN DH PARAMETERS-----\n\
1707 MEYCQQDaWDwW2YUiidDkr3VvTMqS3UvlM7gE+w/tlO+cikQD7VdGUNNpmdsp13Yn\n\
1708 a6LT1BLiGPTdHghM9tgAPnxHdOgzAgEC\n\
1709 -----END DH PARAMETERS-----\n";
1710
1711 static const char dhpem1024[] =
1712 "-----BEGIN DH PARAMETERS-----\n\
1713 MIGHAoGBAJf2QmHKtQXdKCjhPx1ottPb0PMTBH9A6FbaWMsTuKG/K3g6TG1Z1fkq\n\
1714 /Gz/PWk/eLI9TzFgqVAuPvr3q14a1aZeVUMTgo2oO5/y2UHe6VaJ+trqCTat3xlx\n\
1715 /mNbIK9HA2RgPC3gWfVLZQrY+gz3ASHHR5nXWHEyvpuZm7m3h+irAgEC\n\
1716 -----END DH PARAMETERS-----\n";
1717
1718 static const char dhpem2048[] =
1719 "-----BEGIN DH PARAMETERS-----\n\
1720 MIIBCAKCAQEA7ZKJNYJFVcs7+6J2WmkEYb8h86tT0s0h2v94GRFS8Q7B4lW9aG9o\n\
1721 AFO5Imov5Jo0H2XMWTKKvbHbSe3fpxJmw/0hBHAY8H/W91hRGXKCeyKpNBgdL8sh\n\
1722 z22SrkO2qCnHJ6PLAMXy5fsKpFmFor2tRfCzrfnggTXu2YOzzK7q62bmqVdmufEo\n\
1723 pT8igNcLpvZxk5uBDvhakObMym9mX3rAEBoe8PwttggMYiiw7NuJKO4MqD1llGkW\n\
1724 aVM8U2ATsCun1IKHrRxynkE1/MJ86VHeYYX8GZt2YA8z+GuzylIOKcMH6JAWzMwA\n\
1725 Gbatw6QwizOhr9iMjZ0B26TE3X8LvW84wwIBAg==\n\
1726 -----END DH PARAMETERS-----\n";
1727
1728 static const char dhpem4096[] =
1729 "-----BEGIN DH PARAMETERS-----\n\
1730 MIICCAKCAgEA/urRnb6vkPYc/KEGXWnbCIOaKitq7ySIq9dTH7s+Ri59zs77zty7\n\
1731 vfVlSe6VFTBWgYjD2XKUFmtqq6CqXMhVX5ElUDoYDpAyTH85xqNFLzFC7nKrff/H\n\
1732 TFKNttp22cZE9V0IPpzedPfnQkE7aUdmF9JnDyv21Z/818O93u1B4r0szdnmEvEF\n\
1733 bKuIxEHX+bp0ZR7RqE1AeifXGJX3d6tsd2PMAObxwwsv55RGkn50vHO4QxtTARr1\n\
1734 rRUV5j3B3oPMgC7Offxx+98Xn45B1/G0Prp11anDsR1PGwtaCYipqsvMwQUSJtyE\n\
1735 EOQWk+yFkeMe4vWv367eEi0Sd/wnC+TSXBE3pYvpYerJ8n1MceI5GQTdarJ77OW9\n\
1736 bGTHmxRsLSCM1jpLdPja5jjb4siAa6EHc4qN9c/iFKS3PQPJEnX7pXKBRs5f7AF3\n\
1737 W3RIGt+G9IVNZfXaS7Z/iCpgzgvKCs0VeqN38QsJGtC1aIkwOeyjPNy2G6jJ4yqH\n\
1738 ovXYt/0mc00vCWeSNS1wren0pR2EiLxX0ypjjgsU1mk/Z3b/+zVf7fZSIB+nDLjb\n\
1739 NPtUlJCVGnAeBK1J1nG3TQicqowOXoM6ISkdaXj5GPJdXHab2+S7cqhKGv5qC7rR\n\
1740 jT6sx7RUr0CNTxzLI7muV2/a4tGmj0PSdXQdsZ7tw7gbXlaWT1+MM2MCAQI=\n\
1741 -----END DH PARAMETERS-----\n";
1742
1743 static const struct dhinfo dhpem[] = {
1744         { 512, dhpem512, sizeof(dhpem512) },
1745         { 1024, dhpem1024, sizeof(dhpem1024) },
1746         { 2048, dhpem2048, sizeof(dhpem2048) },
1747         { 4096, dhpem4096, sizeof(dhpem4096) },
1748         { 0, NULL, 0 }
1749 };
1750
1751 static DH *
1752 tls_tmp_dh_cb( SSL *ssl, int is_export, int key_length )
1753 {
1754         struct dhplist *p = NULL;
1755         BIO *b = NULL;
1756         DH *dh = NULL;
1757         int i;
1758
1759         /* Do we have params of this length already? */
1760 #ifdef LDAP_R_COMPILE
1761         ldap_pvt_thread_mutex_lock( &tls_def_ctx_mutex );
1762 #endif
1763         for ( p = dhparams; p; p=p->next ) {
1764                 if ( p->keylength == key_length ) {
1765 #ifdef LDAP_R_COMPILE
1766                         ldap_pvt_thread_mutex_unlock( &tls_def_ctx_mutex );
1767 #endif
1768                         return p->param;
1769                 }
1770         }
1771
1772         /* No - check for hardcoded params */
1773
1774         for (i=0; dhpem[i].keylength; i++) {
1775                 if ( dhpem[i].keylength == key_length ) {
1776                         b = BIO_new_mem_buf( (char *)dhpem[i].pem, dhpem[i].size );
1777                         break;
1778                 }
1779         }
1780
1781         if ( b ) {
1782                 dh = PEM_read_bio_DHparams( b, NULL, NULL, NULL );
1783                 BIO_free( b );
1784         }
1785
1786         /* Generating on the fly is expensive/slow... */
1787         if ( !dh ) {
1788                 dh = DH_generate_parameters( key_length, DH_GENERATOR_2, NULL, NULL );
1789         }
1790         if ( dh ) {
1791                 p = LDAP_MALLOC( sizeof(struct dhplist) );
1792                 if ( p != NULL ) {
1793                         p->keylength = key_length;
1794                         p->param = dh;
1795                         p->next = dhparams;
1796                         dhparams = p;
1797                 }
1798         }
1799
1800 #ifdef LDAP_R_COMPILE
1801         ldap_pvt_thread_mutex_unlock( &tls_def_ctx_mutex );
1802 #endif
1803         return dh;
1804 }
1805 #endif
1806
1807 void *
1808 ldap_pvt_tls_sb_ctx( Sockbuf *sb )
1809 {
1810 #ifdef HAVE_TLS
1811         void                    *p;
1812         
1813         if (HAS_TLS( sb )) {
1814                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&p );
1815                 return p;
1816         }
1817 #endif
1818
1819         return NULL;
1820 }
1821
1822 int
1823 ldap_pvt_tls_get_strength( void *s )
1824 {
1825 #ifdef HAVE_TLS
1826         SSL_CIPHER *c;
1827
1828         c = SSL_get_current_cipher((SSL *)s);
1829         return SSL_CIPHER_get_bits(c, NULL);
1830 #else
1831         return 0;
1832 #endif
1833 }
1834
1835
1836 int
1837 ldap_pvt_tls_get_my_dn( void *s, struct berval *dn, LDAPDN_rewrite_dummy *func, unsigned flags )
1838 {
1839 #ifdef HAVE_TLS
1840         X509 *x;
1841         X509_NAME *xn;
1842         int rc;
1843
1844         x = SSL_get_certificate((SSL *)s);
1845
1846         if (!x) return LDAP_INVALID_CREDENTIALS;
1847         
1848         xn = X509_get_subject_name(x);
1849         rc = ldap_X509dn2bv(xn, dn, (LDAPDN_rewrite_func *)func, flags );
1850         return rc;
1851 #else
1852         return LDAP_NOT_SUPPORTED;
1853 #endif
1854 }
1855
1856 int
1857 ldap_start_tls( LDAP *ld,
1858         LDAPControl **serverctrls,
1859         LDAPControl **clientctrls,
1860         int *msgidp )
1861 {
1862         return ldap_extended_operation( ld, LDAP_EXOP_START_TLS,
1863                 NULL, serverctrls, clientctrls, msgidp );
1864 }
1865
1866 int
1867 ldap_install_tls( LDAP *ld )
1868 {
1869 #ifndef HAVE_TLS
1870         return LDAP_NOT_SUPPORTED;
1871 #else
1872         if ( ldap_tls_inplace( ld ) ) {
1873                 return LDAP_LOCAL_ERROR;
1874         }
1875
1876         return ldap_int_tls_start( ld, ld->ld_defconn, NULL );
1877 #endif
1878 }
1879
1880 int
1881 ldap_start_tls_s ( LDAP *ld,
1882         LDAPControl **serverctrls,
1883         LDAPControl **clientctrls )
1884 {
1885 #ifndef HAVE_TLS
1886         return LDAP_NOT_SUPPORTED;
1887 #else
1888         int rc;
1889         char *rspoid = NULL;
1890         struct berval *rspdata = NULL;
1891
1892         /* XXYYZ: this initiates operation only on default connection! */
1893
1894         if ( ldap_tls_inplace( ld ) ) {
1895                 return LDAP_LOCAL_ERROR;
1896         }
1897
1898         rc = ldap_extended_operation_s( ld, LDAP_EXOP_START_TLS,
1899                 NULL, serverctrls, clientctrls, &rspoid, &rspdata );
1900
1901         if ( rspoid != NULL ) {
1902                 LDAP_FREE(rspoid);
1903         }
1904
1905         if ( rspdata != NULL ) {
1906                 ber_bvfree( rspdata );
1907         }
1908
1909         if ( rc == LDAP_SUCCESS ) {
1910                 rc = ldap_int_tls_start( ld, ld->ld_defconn, NULL );
1911         }
1912
1913         return rc;
1914 #endif
1915 }
1916