]> git.sur5r.net Git - openldap/blob - libraries/libldap/tls.c
make openssl builds working again
[openldap] / libraries / libldap / tls.c
1 /* tls.c - Handle tls/ssl using SSLeay, OpenSSL or GNUTLS. */
2 /* $OpenLDAP$ */
3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4  *
5  * Copyright 1998-2007 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 /* ACKNOWLEDGEMENTS: GNUTLS support by Howard Chu
17  */
18
19 #include "portable.h"
20 #include "ldap_config.h"
21
22 #include <stdio.h>
23
24 #include <ac/stdlib.h>
25 #include <ac/errno.h>
26 #include <ac/socket.h>
27 #include <ac/string.h>
28 #include <ac/ctype.h>
29 #include <ac/time.h>
30 #include <ac/unistd.h>
31 #include <ac/param.h>
32 #include <ac/dirent.h>
33
34 #include "ldap-int.h"
35
36 #ifdef HAVE_TLS
37
38 #ifdef LDAP_R_COMPILE
39 #include <ldap_pvt_thread.h>
40 #endif
41
42 #ifdef HAVE_GNUTLS
43 #include <gnutls/gnutls.h>
44 #include <gnutls/x509.h>
45 #include <gcrypt.h>
46
47 #define DH_BITS (1024)
48
49 #else
50 #ifdef HAVE_OPENSSL_SSL_H
51 #include <openssl/ssl.h>
52 #include <openssl/x509v3.h>
53 #include <openssl/err.h>
54 #include <openssl/rand.h>
55 #include <openssl/safestack.h>
56 #elif defined( HAVE_SSL_H )
57 #include <ssl.h>
58 #endif
59 #endif
60
61 #define HAS_TLS( sb )   ber_sockbuf_ctrl( sb, LBER_SB_OPT_HAS_IO, \
62                                 (void *)&sb_tls_sbio )
63
64 /* RFC2459 minimum required set of supported attribute types
65  * in a certificate DN
66  */
67 typedef struct oid_name {
68         struct berval oid;
69         struct berval name;
70 } oid_name;
71
72 #define CN_OID  oids[0].oid.bv_val
73
74 static oid_name oids[] = {
75         { BER_BVC("2.5.4.3"), BER_BVC("cn") },
76         { BER_BVC("2.5.4.4"), BER_BVC("sn") },
77         { BER_BVC("2.5.4.6"), BER_BVC("c") },
78         { BER_BVC("2.5.4.7"), BER_BVC("l") },
79         { BER_BVC("2.5.4.8"), BER_BVC("st") },
80         { BER_BVC("2.5.4.10"), BER_BVC("o") },
81         { BER_BVC("2.5.4.11"), BER_BVC("ou") },
82         { BER_BVC("2.5.4.12"), BER_BVC("title") },
83         { BER_BVC("2.5.4.41"), BER_BVC("name") },
84         { BER_BVC("2.5.4.42"), BER_BVC("givenName") },
85         { BER_BVC("2.5.4.43"), BER_BVC("initials") },
86         { BER_BVC("2.5.4.44"), BER_BVC("generationQualifier") },
87         { BER_BVC("2.5.4.46"), BER_BVC("dnQualifier") },
88         { BER_BVC("1.2.840.113549.1.9.1"), BER_BVC("email") },
89         { BER_BVC("0.9.2342.19200300.100.1.25"), BER_BVC("dc") },
90         { BER_BVNULL, BER_BVNULL }
91 };
92
93 #ifdef HAVE_GNUTLS
94
95 typedef struct tls_cipher_suite {
96         const char *name;
97         gnutls_kx_algorithm_t kx;
98         gnutls_cipher_algorithm_t cipher;
99         gnutls_mac_algorithm_t mac;
100         gnutls_protocol_t version;
101 } tls_cipher_suite;
102
103 static tls_cipher_suite *ciphers;
104 static int n_ciphers;
105
106 /* sorta replacing SSL_CTX */
107 typedef struct tls_ctx {
108         struct ldapoptions *lo;
109         gnutls_certificate_credentials_t cred;
110         gnutls_dh_params_t dh_params;
111         unsigned long verify_depth;
112         int refcount;
113         int *kx_list;
114         int *cipher_list;
115         int *mac_list;
116 #ifdef LDAP_R_COMPILE
117         ldap_pvt_thread_mutex_t ref_mutex;
118 #endif
119 } tls_ctx;
120
121 /* sorta replacing SSL */
122 typedef struct tls_session {
123         tls_ctx *ctx;
124         gnutls_session_t session;
125         struct berval peer_der_dn;
126 } tls_session;
127
128 #ifdef LDAP_R_COMPILE
129
130 static int
131 ldap_pvt_gcry_mutex_init( void **priv )
132 {
133         int err = 0;
134         ldap_pvt_thread_mutex_t *lock = LDAP_MALLOC( sizeof( ldap_pvt_thread_mutex_t ));
135
136         if ( !lock )
137                 err = ENOMEM;
138         if ( !err ) {
139                 err = ldap_pvt_thread_mutex_init( lock );
140                 if ( err )
141                         LDAP_FREE( lock );
142                 else
143                         *priv = lock;
144         }
145         return err;
146 }
147 static int
148 ldap_pvt_gcry_mutex_destroy( void **lock )
149 {
150         int err = ldap_pvt_thread_mutex_destroy( *lock );
151         LDAP_FREE( *lock );
152         return err;
153 }
154 static int
155 ldap_pvt_gcry_mutex_lock( void **lock )
156 {
157         return ldap_pvt_thread_mutex_lock( *lock );
158 }
159 static int
160 ldap_pvt_gcry_mutex_unlock( void **lock )
161 {
162         return ldap_pvt_thread_mutex_unlock( *lock );
163 }
164
165 static struct gcry_thread_cbs ldap_generic_thread_cbs = {
166         GCRY_THREAD_OPTION_USER,
167         NULL,
168         ldap_pvt_gcry_mutex_init,
169         ldap_pvt_gcry_mutex_destroy,
170         ldap_pvt_gcry_mutex_lock,
171         ldap_pvt_gcry_mutex_unlock,
172         NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL
173 };
174
175 static void
176 tls_init_threads( void )
177 {
178         gcry_control (GCRYCTL_SET_THREAD_CBS, &ldap_generic_thread_cbs);
179 }
180 #endif /* LDAP_R_COMPILE */
181
182 void
183 ldap_pvt_tls_ctx_free ( void *c )
184 {
185         int refcount;
186         tls_ctx *ctx = c;
187
188         if ( !ctx ) return;
189
190 #ifdef LDAP_R_COMPILE
191         ldap_pvt_thread_mutex_lock( &ctx->ref_mutex );
192 #endif
193         refcount = --ctx->refcount;
194 #ifdef LDAP_R_COMPILE
195         ldap_pvt_thread_mutex_unlock( &ctx->ref_mutex );
196 #endif
197         if ( refcount )
198                 return;
199         LDAP_FREE( ctx->kx_list );
200         gnutls_certificate_free_credentials( ctx->cred );
201         ber_memfree ( ctx );
202 }
203
204 static void *
205 tls_ctx_new ( struct ldapoptions *lo )
206 {
207         tls_ctx *ctx;
208
209         ctx = ber_memcalloc ( 1, sizeof (*ctx) );
210         if ( ctx ) {
211                 ctx->lo = lo;
212                 if ( gnutls_certificate_allocate_credentials( &ctx->cred )) {
213                         ber_memfree( ctx );
214                         return NULL;
215                 }
216                 ctx->refcount = 1;
217 #ifdef LDAP_R_COMPILE
218                 ldap_pvt_thread_mutex_init( &ctx->ref_mutex );
219 #endif
220         }
221         return ctx;
222 }
223
224 static void
225 tls_ctx_ref( tls_ctx *ctx )
226 {
227 #ifdef LDAP_R_COMPILE
228         ldap_pvt_thread_mutex_lock( &ctx->ref_mutex );
229 #endif
230         ctx->refcount++;
231 #ifdef LDAP_R_COMPILE
232         ldap_pvt_thread_mutex_unlock( &ctx->ref_mutex );
233 #endif
234 }
235
236 tls_session *
237 tls_session_new ( tls_ctx * ctx, int is_server )
238 {
239         tls_session *session;
240
241         session = ber_memcalloc ( 1, sizeof (*session) );
242         if ( !session )
243                 return NULL;
244
245         session->ctx = ctx;
246         gnutls_init( &session->session, is_server ? GNUTLS_SERVER : GNUTLS_CLIENT );
247         gnutls_set_default_priority( session->session );
248         if ( ctx->kx_list ) {
249                 gnutls_kx_set_priority( session->session, ctx->kx_list );
250                 gnutls_cipher_set_priority( session->session, ctx->cipher_list );
251                 gnutls_mac_set_priority( session->session, ctx->mac_list );
252         }
253         if ( ctx->cred )
254                 gnutls_credentials_set( session->session, GNUTLS_CRD_CERTIFICATE, ctx->cred );
255         
256         if ( is_server ) {
257                 int flag = 0;
258                 if ( ctx->lo->ldo_tls_require_cert ) {
259                         flag = GNUTLS_CERT_REQUEST;
260                         if ( ctx->lo->ldo_tls_require_cert == LDAP_OPT_X_TLS_DEMAND ||
261                                 ctx->lo->ldo_tls_require_cert == LDAP_OPT_X_TLS_HARD )
262                                 flag = GNUTLS_CERT_REQUIRE;
263                         gnutls_certificate_server_set_request( session->session, flag );
264                 }
265         }
266         return session;
267
268
269 void
270 tls_session_free ( tls_session * session )
271 {
272         ber_memfree ( session );
273 }
274
275 #define tls_session_connect( ssl )      gnutls_handshake( ssl->session )
276 #define tls_session_accept( ssl )       gnutls_handshake( ssl->session )
277
278 /* suites is a string of colon-separated cipher suite names. */
279 static int
280 tls_parse_ciphers( tls_ctx *ctx, char *suites )
281 {
282         char *ptr, *end;
283         int i, j, len, num;
284         int *list, nkx = 0, ncipher = 0, nmac = 0;
285         int *kx, *cipher, *mac;
286
287         num = 0;
288         ptr = suites;
289         do {
290                 end = strchr(ptr, ':');
291                 if ( end )
292                         len = end - ptr;
293                 else
294                         len = strlen(ptr);
295                 for (i=0; i<n_ciphers; i++) {
296                         if ( !strncasecmp( ciphers[i].name, ptr, len )) {
297                                 num++;
298                                 break;
299                         }
300                 }
301                 if ( i == n_ciphers ) {
302                         /* unrecognized cipher suite */
303                         return -1;
304                 }
305         } while (end);
306
307         /* Space for all 3 lists */
308         list = LDAP_MALLOC( (num+1) * sizeof(int) * 3 );
309         if ( !list )
310                 return -1;
311         kx = list;
312         cipher = kx+num+1;
313         mac = cipher+num+1;
314
315         ptr = suites;
316         do {
317                 end = strchr(ptr, ':');
318                 if ( end )
319                         len = end - ptr;
320                 else
321                         len = strlen(ptr);
322                 for (i=0; i<n_ciphers; i++) {
323                         /* For each cipher suite, insert its algorithms into
324                          * their respective priority lists. Make sure they
325                          * only appear once in each list.
326                          */
327                         if ( !strncasecmp( ciphers[i].name, ptr, len )) {
328                                 for (j=0; j<nkx; j++)
329                                         if ( kx[j] == ciphers[i].kx )
330                                                 break;
331                                 if ( j == nkx )
332                                         kx[nkx++] = ciphers[i].kx;
333                                 for (j=0; j<ncipher; j++)
334                                         if ( cipher[j] == ciphers[i].cipher )
335                                                 break;
336                                 if ( j == ncipher ) 
337                                         cipher[ncipher++] = ciphers[i].cipher;
338                                 for (j=0; j<nmac; j++)
339                                         if ( mac[j] == ciphers[i].mac )
340                                                 break;
341                                 if ( j == nmac )
342                                         mac[nmac++] = ciphers[i].mac;
343                                 break;
344                         }
345                 }
346         } while (end);
347         kx[nkx] = 0;
348         cipher[ncipher] = 0;
349         mac[nmac] = 0;
350         ctx->kx_list = kx;
351         ctx->cipher_list = cipher;
352         ctx->mac_list = mac;
353         return 0;
354 }
355
356 #else /* OpenSSL */
357
358 typedef SSL_CTX tls_ctx;
359 typedef SSL tls_session;
360
361 static int  tls_opt_trace = 1;
362 static char *tls_opt_randfile = NULL;
363
364 static void tls_report_error( void );
365
366 static void tls_info_cb( const SSL *ssl, int where, int ret );
367 static int tls_verify_cb( int ok, X509_STORE_CTX *ctx );
368 static int tls_verify_ok( int ok, X509_STORE_CTX *ctx );
369 static RSA * tls_tmp_rsa_cb( SSL *ssl, int is_export, int key_length );
370
371 static DH * tls_tmp_dh_cb( SSL *ssl, int is_export, int key_length );
372
373 typedef struct dhplist {
374         struct dhplist *next;
375         int keylength;
376         DH *param;
377 } dhplist;
378
379 static dhplist *dhparams;
380
381 static int tls_seed_PRNG( const char *randfile );
382
383 #ifdef LDAP_R_COMPILE
384 /*
385  * provide mutexes for the SSLeay library.
386  */
387 static ldap_pvt_thread_mutex_t  tls_mutexes[CRYPTO_NUM_LOCKS];
388
389 static void tls_locking_cb( int mode, int type, const char *file, int line )
390 {
391         if ( mode & CRYPTO_LOCK ) {
392                 ldap_pvt_thread_mutex_lock( &tls_mutexes[type] );
393         } else {
394                 ldap_pvt_thread_mutex_unlock( &tls_mutexes[type] );
395         }
396 }
397
398 static void tls_init_threads( void )
399 {
400         int i;
401
402         for( i=0; i< CRYPTO_NUM_LOCKS ; i++ ) {
403                 ldap_pvt_thread_mutex_init( &tls_mutexes[i] );
404         }
405         CRYPTO_set_locking_callback( tls_locking_cb );
406         CRYPTO_set_id_callback( ldap_pvt_thread_self );
407         /* FIXME: CRYPTO_set_id_callback only works when ldap_pvt_thread_t
408          * is an integral type that fits in an unsigned long
409          */
410 }
411 #endif /* LDAP_R_COMPILE */
412
413 void
414 ldap_pvt_tls_ctx_free ( void *c )
415 {
416
417         SSL_CTX_free( c );
418 }
419
420 static void *
421 tls_ctx_new( struct ldapoptions *lo )
422 {
423         return SSL_CTX_new( SSLv23_method() );
424 }
425
426 static void
427 tls_ctx_ref( void *c )
428 {
429         SSL_CTX *ctx = c;
430         CRYPTO_add( &ctx->references, 1, CRYPTO_LOCK_SSL_CTX );
431 }
432
433 static tls_session *
434 tls_session_new( tls_ctx *ctx, int is_server )
435 {
436         return SSL_new( ctx );
437 }
438
439 #define tls_session_connect( ssl )      SSL_connect( ssl )
440 #define tls_session_accept( ssl )       SSL_accept( ssl )
441
442 static STACK_OF(X509_NAME) *
443 get_ca_list( char * bundle, char * dir )
444 {
445         STACK_OF(X509_NAME) *ca_list = NULL;
446
447         if ( bundle ) {
448                 ca_list = SSL_load_client_CA_file( bundle );
449         }
450 #if defined(HAVE_DIRENT_H) || defined(dirent)
451         if ( dir ) {
452                 int freeit = 0;
453
454                 if ( !ca_list ) {
455                         ca_list = sk_X509_NAME_new_null();
456                         freeit = 1;
457                 }
458                 if ( !SSL_add_dir_cert_subjects_to_stack( ca_list, dir ) &&
459                         freeit ) {
460                         sk_X509_NAME_free( ca_list );
461                         ca_list = NULL;
462                 }
463         }
464 #endif
465         return ca_list;
466 }
467
468 #endif /* HAVE_GNUTLS */
469
470 #ifdef LDAP_R_COMPILE
471 /*
472  * an extra mutex for the default ctx.
473  */
474 static ldap_pvt_thread_mutex_t tls_def_ctx_mutex;
475 #endif
476
477 void
478 ldap_int_tls_destroy( struct ldapoptions *lo )
479 {
480         if ( lo->ldo_tls_ctx ) {
481                 ldap_pvt_tls_ctx_free( lo->ldo_tls_ctx );
482                 lo->ldo_tls_ctx = NULL;
483         }
484
485         if ( lo->ldo_tls_certfile ) {
486                 LDAP_FREE( lo->ldo_tls_certfile );
487                 lo->ldo_tls_certfile = NULL;
488         }
489         if ( lo->ldo_tls_keyfile ) {
490                 LDAP_FREE( lo->ldo_tls_keyfile );
491                 lo->ldo_tls_keyfile = NULL;
492         }
493         if ( lo->ldo_tls_dhfile ) {
494                 LDAP_FREE( lo->ldo_tls_dhfile );
495                 lo->ldo_tls_dhfile = NULL;
496         }
497         if ( lo->ldo_tls_cacertfile ) {
498                 LDAP_FREE( lo->ldo_tls_cacertfile );
499                 lo->ldo_tls_cacertfile = NULL;
500         }
501         if ( lo->ldo_tls_cacertdir ) {
502                 LDAP_FREE( lo->ldo_tls_cacertdir );
503                 lo->ldo_tls_cacertdir = NULL;
504         }
505         if ( lo->ldo_tls_ciphersuite ) {
506                 LDAP_FREE( lo->ldo_tls_ciphersuite );
507                 lo->ldo_tls_ciphersuite = NULL;
508         }
509 #ifdef HAVE_GNUTLS
510         if ( lo->ldo_tls_crlfile ) {
511                 LDAP_FREE( lo->ldo_tls_crlfile );
512                 lo->ldo_tls_crlfile = NULL;
513         }
514 #endif
515 }
516
517 /*
518  * Tear down the TLS subsystem. Should only be called once.
519  */
520 void
521 ldap_pvt_tls_destroy( void )
522 {
523         struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();   
524
525         ldap_int_tls_destroy( lo );
526
527 #ifdef HAVE_GNUTLS
528         LDAP_FREE( ciphers );
529         ciphers = NULL;
530
531         gnutls_global_deinit();
532 #else
533         EVP_cleanup();
534         ERR_remove_state(0);
535         ERR_free_strings();
536
537         if ( tls_opt_randfile ) {
538                 LDAP_FREE( tls_opt_randfile );
539                 tls_opt_randfile = NULL;
540         }
541 #endif
542 }
543
544 /*
545  * Initialize TLS subsystem. Should be called only once.
546  */
547 int
548 ldap_pvt_tls_init( void )
549 {
550         static int tls_initialized = 0;
551
552         if ( tls_initialized++ ) return 0;
553
554 #ifdef LDAP_R_COMPILE
555         tls_init_threads();
556         ldap_pvt_thread_mutex_init( &tls_def_ctx_mutex );
557 #endif
558
559 #ifdef HAVE_GNUTLS
560         gnutls_global_init ();
561
562         /* GNUtls cipher suite handling: The library ought to parse suite
563          * names for us, but it doesn't. It will return a list of suite names
564          * that it supports, so we can do parsing ourselves. It ought to tell
565          * us how long the list is, but it doesn't do that either, so we just
566          * have to count it manually...
567          */
568         {
569                 int i = 0;
570                 tls_cipher_suite *ptr, tmp;
571                 char cs_id[2];
572
573                 while ( gnutls_cipher_suite_info( i, cs_id, &tmp.kx, &tmp.cipher,
574                         &tmp.mac, &tmp.version ))
575                         i++;
576                 n_ciphers = i;
577
578                 /* Store a copy */
579                 ciphers = LDAP_MALLOC(n_ciphers * sizeof(tls_cipher_suite));
580                 if ( !ciphers )
581                         return -1;
582                 for ( i=0; i<n_ciphers; i++ ) {
583                         ciphers[i].name = gnutls_cipher_suite_info( i, cs_id,
584                                 &ciphers[i].kx, &ciphers[i].cipher, &ciphers[i].mac,
585                                 &ciphers[i].version );
586                 }
587         }
588
589 #else /* !HAVE_GNUTLS */
590
591 #ifdef HAVE_EBCDIC
592         {
593                 char *file = LDAP_STRDUP( tls_opt_randfile );
594                 if ( file ) __atoe( file );
595                 (void) tls_seed_PRNG( file );
596                 LDAP_FREE( file );
597         }
598 #else
599         (void) tls_seed_PRNG( tls_opt_randfile );
600 #endif
601
602         SSL_load_error_strings();
603         SSLeay_add_ssl_algorithms();
604
605         /* FIXME: mod_ssl does this */
606         X509V3_add_standard_extensions();
607
608 #endif /* HAVE_GNUTLS */
609         return 0;
610 }
611
612 /*
613  * initialize a new TLS context
614  */
615 static int
616 ldap_int_tls_init_ctx( struct ldapoptions *lo, int is_server )
617 {
618         int i, rc = 0;
619         char *ciphersuite = lo->ldo_tls_ciphersuite;
620         char *cacertfile = lo->ldo_tls_cacertfile;
621         char *cacertdir = lo->ldo_tls_cacertdir;
622         char *certfile = lo->ldo_tls_certfile;
623         char *keyfile = lo->ldo_tls_keyfile;
624 #ifdef HAVE_GNUTLS
625         char *crlfile = lo->ldo_tls_crlfile;
626 #else
627         char *dhfile = lo->ldo_tls_dhfile;
628 #endif
629
630         if ( lo->ldo_tls_ctx )
631                 return 0;
632
633         ldap_pvt_tls_init();
634
635         if ( is_server && !certfile && !keyfile && !cacertfile && !cacertdir ) {
636                 /* minimum configuration not provided */
637                 return LDAP_NOT_SUPPORTED;
638         }
639
640 #ifdef HAVE_EBCDIC
641         /* This ASCII/EBCDIC handling is a real pain! */
642         if ( ciphersuite ) {
643                 ciphersuite = LDAP_STRDUP( ciphersuite );
644                 __atoe( ciphersuite );
645         }
646         if ( cacertfile ) {
647                 cacertfile = LDAP_STRDUP( cacertfile );
648                 __atoe( cacertfile );
649         }
650         if ( certfile ) {
651                 certfile = LDAP_STRDUP( certfile );
652                 __atoe( certfile );
653         }
654         if ( keyfile ) {
655                 keyfile = LDAP_STRDUP( keyfile );
656                 __atoe( keyfile );
657         }
658 #ifdef HAVE_GNUTLS
659         if ( crlfile ) {
660                 crlfile = LDAP_STRDUP( crlfile );
661                 __atoe( crlfile );
662         }
663 #else
664         if ( cacertdir ) {
665                 cacertdir = LDAP_STRDUP( cacertdir );
666                 __atoe( cacertdir );
667         }
668         if ( dhfile ) {
669                 dhfile = LDAP_STRDUP( dhfile );
670                 __atoe( dhfile );
671         }
672 #endif
673 #endif
674         lo->ldo_tls_ctx = tls_ctx_new( lo );
675         if ( lo->ldo_tls_ctx == NULL ) {
676 #ifdef HAVE_GNUTLS
677                 Debug( LDAP_DEBUG_ANY,
678                    "TLS: could not allocate default ctx.\n",
679                         0,0,0);
680 #else
681                 Debug( LDAP_DEBUG_ANY,
682                    "TLS: could not allocate default ctx (%lu).\n",
683                         ERR_peek_error(),0,0);
684 #endif
685                 rc = -1;
686                 goto error_exit;
687         }
688
689 #ifdef HAVE_GNUTLS
690         if ( lo->ldo_tls_ciphersuite &&
691                 tls_parse_ciphers( lo->ldo_tls_ctx,
692                         ciphersuite )) {
693                 Debug( LDAP_DEBUG_ANY,
694                            "TLS: could not set cipher list %s.\n",
695                            lo->ldo_tls_ciphersuite, 0, 0 );
696                 rc = -1;
697                 goto error_exit;
698         }
699
700         if (lo->ldo_tls_cacertdir != NULL) {
701                 Debug( LDAP_DEBUG_ANY, 
702                        "TLS: warning: cacertdir not implemented for gnutls\n",
703                        NULL, NULL, NULL );
704         }
705
706         if (lo->ldo_tls_cacertfile != NULL) {
707                 rc = gnutls_certificate_set_x509_trust_file( 
708                         ((tls_ctx*) lo->ldo_tls_ctx)->cred,
709                         cacertfile,
710                         GNUTLS_X509_FMT_PEM );
711                 if ( rc < 0 ) goto error_exit;
712         }
713
714         if ( lo->ldo_tls_certfile && lo->ldo_tls_keyfile ) {
715                 rc = gnutls_certificate_set_x509_key_file( 
716                         ((tls_ctx*) lo->ldo_tls_ctx)->cred,
717                         certfile,
718                         keyfile,
719                         GNUTLS_X509_FMT_PEM );
720                 if ( rc ) goto error_exit;
721         } else if ( lo->ldo_tls_certfile || lo->ldo_tls_keyfile ) {
722                 Debug( LDAP_DEBUG_ANY, 
723                        "TLS: only one of certfile and keyfile specified\n",
724                        NULL, NULL, NULL );
725                 rc = 1;
726                 goto error_exit;
727         }
728
729         if ( lo->ldo_tls_dhfile ) {
730                 Debug( LDAP_DEBUG_ANY, 
731                        "TLS: warning: ignoring dhfile\n", 
732                        NULL, NULL, NULL );
733         }
734
735         if ( lo->ldo_tls_crlfile ) {
736                 rc = gnutls_certificate_set_x509_crl_file( 
737                         ((tls_ctx*) lo->ldo_tls_ctx)->cred,
738                         crlfile,
739                         GNUTLS_X509_FMT_PEM );
740                 if ( rc < 0 ) goto error_exit;
741         }
742         if ( is_server ) {
743                 gnutls_dh_params_init (&((tls_ctx*) 
744                                         lo->ldo_tls_ctx)->dh_params);
745                 gnutls_dh_params_generate2 (((tls_ctx*) 
746                                                  lo->ldo_tls_ctx)->dh_params, 
747                                                 DH_BITS);
748         }
749
750 #else /* !HAVE_GNUTLS */
751
752         if ( is_server ) {
753                 SSL_CTX_set_session_id_context( lo->ldo_tls_ctx,
754                         (const unsigned char *) "OpenLDAP", sizeof("OpenLDAP")-1 );
755         }
756
757         if ( lo->ldo_tls_ciphersuite &&
758                 !SSL_CTX_set_cipher_list( lo->ldo_tls_ctx, ciphersuite ) )
759         {
760                 Debug( LDAP_DEBUG_ANY,
761                            "TLS: could not set cipher list %s.\n",
762                            lo->ldo_tls_ciphersuite, 0, 0 );
763                 tls_report_error();
764                 rc = -1;
765                 goto error_exit;
766         }
767
768         if (lo->ldo_tls_cacertfile != NULL || lo->ldo_tls_cacertdir != NULL) {
769                 if ( !SSL_CTX_load_verify_locations( lo->ldo_tls_ctx,
770                                 cacertfile, cacertdir ) ||
771                         !SSL_CTX_set_default_verify_paths( lo->ldo_tls_ctx ) )
772                 {
773                         Debug( LDAP_DEBUG_ANY, "TLS: "
774                                 "could not load verify locations (file:`%s',dir:`%s').\n",
775                                 lo->ldo_tls_cacertfile ? lo->ldo_tls_cacertfile : "",
776                                 lo->ldo_tls_cacertdir ? lo->ldo_tls_cacertdir : "",
777                                 0 );
778                         tls_report_error();
779                         rc = -1;
780                         goto error_exit;
781                 }
782
783                 if ( is_server ) {
784                         STACK_OF(X509_NAME) *calist;
785                         /* List of CA names to send to a client */
786                         calist = get_ca_list( cacertfile, cacertdir );
787                         if ( !calist ) {
788                                 Debug( LDAP_DEBUG_ANY, "TLS: "
789                                         "could not load client CA list (file:`%s',dir:`%s').\n",
790                                         lo->ldo_tls_cacertfile ? lo->ldo_tls_cacertfile : "",
791                                         lo->ldo_tls_cacertdir ? lo->ldo_tls_cacertdir : "",
792                                         0 );
793                                 tls_report_error();
794                                 rc = -1;
795                                 goto error_exit;
796                         }
797
798                         SSL_CTX_set_client_CA_list( lo->ldo_tls_ctx, calist );
799                 }
800         }
801
802         if ( lo->ldo_tls_certfile &&
803                 !SSL_CTX_use_certificate_file( lo->ldo_tls_ctx,
804                         certfile, SSL_FILETYPE_PEM ) )
805         {
806                 Debug( LDAP_DEBUG_ANY,
807                         "TLS: could not use certificate `%s'.\n",
808                         lo->ldo_tls_certfile,0,0);
809                 tls_report_error();
810                 rc = -1;
811                 goto error_exit;
812         }
813
814         /* Key validity is checked automatically if cert has already been set */
815         if ( lo->ldo_tls_keyfile &&
816                 !SSL_CTX_use_PrivateKey_file( lo->ldo_tls_ctx,
817                         keyfile, SSL_FILETYPE_PEM ) )
818         {
819                 Debug( LDAP_DEBUG_ANY,
820                         "TLS: could not use key file `%s'.\n",
821                         lo->ldo_tls_keyfile,0,0);
822                 tls_report_error();
823                 rc = -1;
824                 goto error_exit;
825         }
826
827         if ( lo->ldo_tls_dhfile ) {
828                 DH *dh = NULL;
829                 BIO *bio;
830                 dhplist *p;
831
832                 if (( bio=BIO_new_file( dhfile,"r" )) == NULL ) {
833                         Debug( LDAP_DEBUG_ANY,
834                                 "TLS: could not use DH parameters file `%s'.\n",
835                                 lo->ldo_tls_dhfile,0,0);
836                         tls_report_error();
837                         rc = -1;
838                         goto error_exit;
839                 }
840                 while (( dh=PEM_read_bio_DHparams( bio, NULL, NULL, NULL ))) {
841                         p = LDAP_MALLOC( sizeof(dhplist) );
842                         if ( p != NULL ) {
843                                 p->keylength = DH_size( dh ) * 8;
844                                 p->param = dh;
845                                 p->next = dhparams;
846                                 dhparams = p;
847                         }
848                 }
849                 BIO_free( bio );
850         }
851
852         if ( tls_opt_trace ) {
853                 SSL_CTX_set_info_callback( (SSL_CTX *)lo->ldo_tls_ctx, tls_info_cb );
854         }
855
856         i = SSL_VERIFY_NONE;
857         if ( lo->ldo_tls_require_cert ) {
858                 i = SSL_VERIFY_PEER;
859                 if ( lo->ldo_tls_require_cert == LDAP_OPT_X_TLS_DEMAND ||
860                          lo->ldo_tls_require_cert == LDAP_OPT_X_TLS_HARD ) {
861                         i |= SSL_VERIFY_FAIL_IF_NO_PEER_CERT;
862                 }
863         }
864
865         SSL_CTX_set_verify( lo->ldo_tls_ctx, i,
866                 lo->ldo_tls_require_cert == LDAP_OPT_X_TLS_ALLOW ?
867                 tls_verify_ok : tls_verify_cb );
868         SSL_CTX_set_tmp_rsa_callback( lo->ldo_tls_ctx, tls_tmp_rsa_cb );
869         if ( lo->ldo_tls_dhfile ) {
870                 SSL_CTX_set_tmp_dh_callback( lo->ldo_tls_ctx, tls_tmp_dh_cb );
871         }
872 #ifdef HAVE_OPENSSL_CRL
873         if ( lo->ldo_tls_crlcheck ) {
874                 X509_STORE *x509_s = SSL_CTX_get_cert_store( lo->ldo_tls_ctx );
875                 if ( lo->ldo_tls_crlcheck == LDAP_OPT_X_TLS_CRL_PEER ) {
876                         X509_STORE_set_flags( x509_s, X509_V_FLAG_CRL_CHECK );
877                 } else if ( lo->ldo_tls_crlcheck == LDAP_OPT_X_TLS_CRL_ALL ) {
878                         X509_STORE_set_flags( x509_s, 
879                                         X509_V_FLAG_CRL_CHECK | X509_V_FLAG_CRL_CHECK_ALL  );
880                 }
881         }
882 #endif
883
884 #endif /* HAVE_GNUTLS */
885
886 error_exit:
887         if ( rc == -1 && lo->ldo_tls_ctx != NULL ) {
888                 ldap_pvt_tls_ctx_free( lo->ldo_tls_ctx );
889                 lo->ldo_tls_ctx = NULL;
890         }
891 #ifdef HAVE_EBCDIC
892         LDAP_FREE( ciphersuite );
893         LDAP_FREE( cacertfile );
894         LDAP_FREE( certfile );
895         LDAP_FREE( keyfile );
896 #ifdef HAVE_GNUTLS
897         LDAP_FREE( crlfile );
898 #else
899         LDAP_FREE( cacertdir );
900         LDAP_FREE( dhfile );
901 #endif
902 #endif
903         return rc;
904 }
905
906 /*
907  * initialize the default context
908  */
909 int
910 ldap_pvt_tls_init_def_ctx( int is_server )
911 {
912         struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();   
913         int rc;
914 #ifdef LDAP_R_COMPILE
915         ldap_pvt_thread_mutex_lock( &tls_def_ctx_mutex );
916 #endif
917         rc = ldap_int_tls_init_ctx( lo, is_server );
918 #ifdef LDAP_R_COMPILE
919         ldap_pvt_thread_mutex_unlock( &tls_def_ctx_mutex );
920 #endif
921         return rc;
922 }
923
924 static tls_session *
925 alloc_handle( void *ctx_arg, int is_server )
926 {
927         tls_ctx *ctx;
928         tls_session     *ssl;
929
930         if ( ctx_arg ) {
931                 ctx = ctx_arg;
932         } else {
933                 struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();   
934                 if ( ldap_pvt_tls_init_def_ctx( is_server ) < 0 ) return NULL;
935                 ctx = lo->ldo_tls_ctx;
936         }
937
938         ssl = tls_session_new( ctx, is_server );
939         if ( ssl == NULL ) {
940                 Debug( LDAP_DEBUG_ANY,"TLS: can't create ssl handle.\n",0,0,0);
941                 return NULL;
942         }
943         return ssl;
944 }
945
946 static int
947 update_flags( Sockbuf *sb, tls_session * ssl, int rc )
948 {
949         sb->sb_trans_needs_read  = 0;
950         sb->sb_trans_needs_write = 0;
951
952 #ifdef HAVE_GNUTLS
953         if ( rc != GNUTLS_E_INTERRUPTED && rc != GNUTLS_E_AGAIN )
954                 return 0;
955
956         switch (gnutls_record_get_direction (ssl->session)) {
957         case 0: 
958                 sb->sb_trans_needs_read = 1;
959                 return 1;
960         case 1:
961                 sb->sb_trans_needs_write = 1;
962                 return 1;
963         }
964 #else /* !HAVE_GNUTLS */
965         rc = SSL_get_error(ssl, rc);
966         if (rc == SSL_ERROR_WANT_READ) {
967                 sb->sb_trans_needs_read  = 1;
968                 return 1;
969
970         } else if (rc == SSL_ERROR_WANT_WRITE) {
971                 sb->sb_trans_needs_write = 1;
972                 return 1;
973
974         } else if (rc == SSL_ERROR_WANT_CONNECT) {
975                 return 1;
976         }
977 #endif /* HAVE_GNUTLS */
978         return 0;
979 }
980
981 /*
982  * TLS support for LBER Sockbufs
983  */
984
985 struct tls_data {
986         tls_session                     *ssl;
987         Sockbuf_IO_Desc         *sbiod;
988 };
989
990 #ifdef HAVE_GNUTLS
991
992 static ssize_t
993 sb_gtls_recv( gnutls_transport_ptr_t ptr, void *buf, size_t len )
994 {
995         struct tls_data         *p;
996
997         if ( buf == NULL || len <= 0 ) return 0;
998
999         p = (struct tls_data *)ptr;
1000
1001         if ( p == NULL || p->sbiod == NULL ) {
1002                 return 0;
1003         }
1004
1005         return LBER_SBIOD_READ_NEXT( p->sbiod, buf, len );
1006 }
1007
1008 static ssize_t
1009 sb_gtls_send( gnutls_transport_ptr_t ptr, const void *buf, size_t len )
1010 {
1011         struct tls_data         *p;
1012         
1013         if ( buf == NULL || len <= 0 ) return 0;
1014         
1015         p = (struct tls_data *)ptr;
1016
1017         if ( p == NULL || p->sbiod == NULL ) {
1018                 return 0;
1019         }
1020
1021         return LBER_SBIOD_WRITE_NEXT( p->sbiod, (char *)buf, len );
1022 }
1023
1024 static int
1025 sb_tls_setup( Sockbuf_IO_Desc *sbiod, void *arg )
1026 {
1027         struct tls_data         *p;
1028         tls_session     *session = arg;
1029
1030         assert( sbiod != NULL );
1031
1032         p = LBER_MALLOC( sizeof( *p ) );
1033         if ( p == NULL ) {
1034                 return -1;
1035         }
1036         
1037         gnutls_transport_set_ptr( session->session, (gnutls_transport_ptr)p );
1038         gnutls_transport_set_pull_function( session->session, sb_gtls_recv );
1039         gnutls_transport_set_push_function( session->session, sb_gtls_send );
1040         p->ssl = arg;
1041         p->sbiod = sbiod;
1042         sbiod->sbiod_pvt = p;
1043         return 0;
1044 }
1045
1046 static int
1047 sb_tls_remove( Sockbuf_IO_Desc *sbiod )
1048 {
1049         struct tls_data         *p;
1050         
1051         assert( sbiod != NULL );
1052         assert( sbiod->sbiod_pvt != NULL );
1053
1054         p = (struct tls_data *)sbiod->sbiod_pvt;
1055         gnutls_deinit ( p->ssl->session );
1056         LBER_FREE( p->ssl );
1057         LBER_FREE( sbiod->sbiod_pvt );
1058         sbiod->sbiod_pvt = NULL;
1059         return 0;
1060 }
1061
1062 static int
1063 sb_tls_close( Sockbuf_IO_Desc *sbiod )
1064 {
1065         struct tls_data         *p;
1066         
1067         assert( sbiod != NULL );
1068         assert( sbiod->sbiod_pvt != NULL );
1069
1070         p = (struct tls_data *)sbiod->sbiod_pvt;
1071         gnutls_bye ( p->ssl->session, GNUTLS_SHUT_RDWR );
1072         return 0;
1073 }
1074
1075 static int
1076 sb_tls_ctrl( Sockbuf_IO_Desc *sbiod, int opt, void *arg )
1077 {
1078         struct tls_data         *p;
1079         
1080         assert( sbiod != NULL );
1081         assert( sbiod->sbiod_pvt != NULL );
1082
1083         p = (struct tls_data *)sbiod->sbiod_pvt;
1084         
1085         if ( opt == LBER_SB_OPT_GET_SSL ) {
1086                 *((tls_session **)arg) = p->ssl;
1087                 return 1;
1088                 
1089         } else if ( opt == LBER_SB_OPT_DATA_READY ) {
1090                 if( gnutls_record_check_pending( p->ssl->session ) > 0 ) {
1091                         return 1;
1092                 }
1093         }
1094         
1095         return LBER_SBIOD_CTRL_NEXT( sbiod, opt, arg );
1096 }
1097
1098 static ber_slen_t
1099 sb_tls_read( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
1100 {
1101         struct tls_data         *p;
1102         ber_slen_t              ret;
1103         int                     err;
1104
1105         assert( sbiod != NULL );
1106         assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
1107
1108         p = (struct tls_data *)sbiod->sbiod_pvt;
1109
1110         ret = gnutls_record_recv ( p->ssl->session, buf, len );
1111         switch (ret) {
1112         case GNUTLS_E_INTERRUPTED:
1113         case GNUTLS_E_AGAIN:
1114                 sbiod->sbiod_sb->sb_trans_needs_read = 1;
1115                 sock_errset(EWOULDBLOCK);
1116                 ret = 0;
1117                 break;
1118         case GNUTLS_E_REHANDSHAKE:
1119                 for ( ret = gnutls_handshake ( p->ssl->session );
1120                       ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN;
1121                       ret = gnutls_handshake ( p->ssl->session ) );
1122                 sbiod->sbiod_sb->sb_trans_needs_read = 1;
1123                 ret = 0;
1124                 break;
1125         default:
1126                 sbiod->sbiod_sb->sb_trans_needs_read = 0;
1127         }
1128         return ret;
1129 }
1130
1131 static ber_slen_t
1132 sb_tls_write( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
1133 {
1134         struct tls_data         *p;
1135         ber_slen_t              ret;
1136         int                     err;
1137
1138         assert( sbiod != NULL );
1139         assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
1140
1141         p = (struct tls_data *)sbiod->sbiod_pvt;
1142
1143         ret = gnutls_record_send ( p->ssl->session, (char *)buf, len );
1144
1145         if ( ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN ) {
1146                 sbiod->sbiod_sb->sb_trans_needs_write = 1;
1147                 sock_errset(EWOULDBLOCK);
1148                 ret = 0;
1149         } else {
1150                 sbiod->sbiod_sb->sb_trans_needs_write = 0;
1151         }
1152         return ret;
1153 }
1154
1155 #else /* !HAVE_GNUTLS */
1156
1157 static int
1158 sb_tls_bio_create( BIO *b ) {
1159         b->init = 1;
1160         b->num = 0;
1161         b->ptr = NULL;
1162         b->flags = 0;
1163         return 1;
1164 }
1165
1166 static int
1167 sb_tls_bio_destroy( BIO *b )
1168 {
1169         if ( b == NULL ) return 0;
1170
1171         b->ptr = NULL;          /* sb_tls_remove() will free it */
1172         b->init = 0;
1173         b->flags = 0;
1174         return 1;
1175 }
1176
1177 static int
1178 sb_tls_bio_read( BIO *b, char *buf, int len )
1179 {
1180         struct tls_data         *p;
1181         int                     ret;
1182                 
1183         if ( buf == NULL || len <= 0 ) return 0;
1184
1185         p = (struct tls_data *)b->ptr;
1186
1187         if ( p == NULL || p->sbiod == NULL ) {
1188                 return 0;
1189         }
1190
1191         ret = LBER_SBIOD_READ_NEXT( p->sbiod, buf, len );
1192
1193         BIO_clear_retry_flags( b );
1194         if ( ret < 0 ) {
1195                 int err = sock_errno();
1196                 if ( err == EAGAIN || err == EWOULDBLOCK ) {
1197                         BIO_set_retry_read( b );
1198                 }
1199         }
1200
1201         return ret;
1202 }
1203
1204 static int
1205 sb_tls_bio_write( BIO *b, const char *buf, int len )
1206 {
1207         struct tls_data         *p;
1208         int                     ret;
1209         
1210         if ( buf == NULL || len <= 0 ) return 0;
1211         
1212         p = (struct tls_data *)b->ptr;
1213
1214         if ( p == NULL || p->sbiod == NULL ) {
1215                 return 0;
1216         }
1217
1218         ret = LBER_SBIOD_WRITE_NEXT( p->sbiod, (char *)buf, len );
1219
1220         BIO_clear_retry_flags( b );
1221         if ( ret < 0 ) {
1222                 int err = sock_errno();
1223                 if ( err == EAGAIN || err == EWOULDBLOCK ) {
1224                         BIO_set_retry_write( b );
1225                 }
1226         }
1227
1228         return ret;
1229 }
1230
1231 static long
1232 sb_tls_bio_ctrl( BIO *b, int cmd, long num, void *ptr )
1233 {
1234         if ( cmd == BIO_CTRL_FLUSH ) {
1235                 /* The OpenSSL library needs this */
1236                 return 1;
1237         }
1238         return 0;
1239 }
1240
1241 static int
1242 sb_tls_bio_gets( BIO *b, char *buf, int len )
1243 {
1244         return -1;
1245 }
1246
1247 static int
1248 sb_tls_bio_puts( BIO *b, const char *str )
1249 {
1250         return sb_tls_bio_write( b, str, strlen( str ) );
1251 }
1252         
1253 static BIO_METHOD sb_tls_bio_method =
1254 {
1255         ( 100 | 0x400 ),                /* it's a source/sink BIO */
1256         "sockbuf glue",
1257         sb_tls_bio_write,
1258         sb_tls_bio_read,
1259         sb_tls_bio_puts,
1260         sb_tls_bio_gets,
1261         sb_tls_bio_ctrl,
1262         sb_tls_bio_create,
1263         sb_tls_bio_destroy
1264 };
1265
1266 static int
1267 sb_tls_setup( Sockbuf_IO_Desc *sbiod, void *arg )
1268 {
1269         struct tls_data         *p;
1270         BIO                     *bio;
1271
1272         assert( sbiod != NULL );
1273
1274         p = LBER_MALLOC( sizeof( *p ) );
1275         if ( p == NULL ) {
1276                 return -1;
1277         }
1278         
1279         p->ssl = (SSL *)arg;
1280         p->sbiod = sbiod;
1281         bio = BIO_new( &sb_tls_bio_method );
1282         bio->ptr = (void *)p;
1283         SSL_set_bio( p->ssl, bio, bio );
1284         sbiod->sbiod_pvt = p;
1285         return 0;
1286 }
1287
1288 static int
1289 sb_tls_remove( Sockbuf_IO_Desc *sbiod )
1290 {
1291         struct tls_data         *p;
1292         
1293         assert( sbiod != NULL );
1294         assert( sbiod->sbiod_pvt != NULL );
1295
1296         p = (struct tls_data *)sbiod->sbiod_pvt;
1297         SSL_free( p->ssl );
1298         LBER_FREE( sbiod->sbiod_pvt );
1299         sbiod->sbiod_pvt = NULL;
1300         return 0;
1301 }
1302
1303 static int
1304 sb_tls_close( Sockbuf_IO_Desc *sbiod )
1305 {
1306         struct tls_data         *p;
1307         
1308         assert( sbiod != NULL );
1309         assert( sbiod->sbiod_pvt != NULL );
1310
1311         p = (struct tls_data *)sbiod->sbiod_pvt;
1312         SSL_shutdown( p->ssl );
1313         return 0;
1314 }
1315
1316 static int
1317 sb_tls_ctrl( Sockbuf_IO_Desc *sbiod, int opt, void *arg )
1318 {
1319         struct tls_data         *p;
1320         
1321         assert( sbiod != NULL );
1322         assert( sbiod->sbiod_pvt != NULL );
1323
1324         p = (struct tls_data *)sbiod->sbiod_pvt;
1325         
1326         if ( opt == LBER_SB_OPT_GET_SSL ) {
1327                 *((SSL **)arg) = p->ssl;
1328                 return 1;
1329
1330         } else if ( opt == LBER_SB_OPT_DATA_READY ) {
1331                 if( SSL_pending( p->ssl ) > 0 ) {
1332                         return 1;
1333                 }
1334         }
1335         
1336         return LBER_SBIOD_CTRL_NEXT( sbiod, opt, arg );
1337 }
1338
1339 static ber_slen_t
1340 sb_tls_read( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
1341 {
1342         struct tls_data         *p;
1343         ber_slen_t              ret;
1344         int                     err;
1345
1346         assert( sbiod != NULL );
1347         assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
1348
1349         p = (struct tls_data *)sbiod->sbiod_pvt;
1350
1351         ret = SSL_read( p->ssl, (char *)buf, len );
1352 #ifdef HAVE_WINSOCK
1353         errno = WSAGetLastError();
1354 #endif
1355         err = SSL_get_error( p->ssl, ret );
1356         if (err == SSL_ERROR_WANT_READ ) {
1357                 sbiod->sbiod_sb->sb_trans_needs_read = 1;
1358                 sock_errset(EWOULDBLOCK);
1359         }
1360         else
1361                 sbiod->sbiod_sb->sb_trans_needs_read = 0;
1362         return ret;
1363 }
1364
1365 static ber_slen_t
1366 sb_tls_write( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
1367 {
1368         struct tls_data         *p;
1369         ber_slen_t              ret;
1370         int                     err;
1371
1372         assert( sbiod != NULL );
1373         assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
1374
1375         p = (struct tls_data *)sbiod->sbiod_pvt;
1376
1377         ret = SSL_write( p->ssl, (char *)buf, len );
1378 #ifdef HAVE_WINSOCK
1379         errno = WSAGetLastError();
1380 #endif
1381         err = SSL_get_error( p->ssl, ret );
1382         if (err == SSL_ERROR_WANT_WRITE ) {
1383                 sbiod->sbiod_sb->sb_trans_needs_write = 1;
1384                 sock_errset(EWOULDBLOCK);
1385
1386         } else {
1387                 sbiod->sbiod_sb->sb_trans_needs_write = 0;
1388         }
1389         return ret;
1390 }
1391
1392 #endif
1393
1394 static Sockbuf_IO sb_tls_sbio =
1395 {
1396         sb_tls_setup,           /* sbi_setup */
1397         sb_tls_remove,          /* sbi_remove */
1398         sb_tls_ctrl,            /* sbi_ctrl */
1399         sb_tls_read,            /* sbi_read */
1400         sb_tls_write,           /* sbi_write */
1401         sb_tls_close            /* sbi_close */
1402 };
1403
1404 #ifdef HAVE_GNUTLS
1405 /* Certs are not automatically varified during the handshake */
1406 static int
1407 tls_cert_verify( tls_session *ssl )
1408 {
1409         unsigned int status = 0;
1410         int err;
1411         time_t now = time(0);
1412
1413         err = gnutls_certificate_verify_peers2( ssl->session, &status );
1414         if ( err < 0 ) {
1415                 Debug( LDAP_DEBUG_ANY,"TLS: gnutls_certificate_verify_peers2 failed %d\n",
1416                         err,0,0 );
1417                 return -1;
1418         }
1419         if ( status ) {
1420                 Debug( LDAP_DEBUG_TRACE,"TLS: peer cert untrusted or revoked (0x%x)\n",
1421                         status, 0,0 );
1422                 return -1;
1423         }
1424         if ( gnutls_certificate_expiration_time_peers( ssl->session ) < now ) {
1425                 Debug( LDAP_DEBUG_ANY, "TLS: peer certificate is expired\n",
1426                         0, 0, 0 );
1427                 return -1;
1428         }
1429         if ( gnutls_certificate_activation_time_peers( ssl->session ) > now ) {
1430                 Debug( LDAP_DEBUG_ANY, "TLS: peer certificate not yet active\n",
1431                         0, 0, 0 );
1432                 return -1;
1433         }
1434         return 0;
1435 }
1436 #endif /* HAVE_GNUTLS */
1437
1438 /*
1439  * Call this to do a TLS connect on a sockbuf. ctx_arg can be
1440  * a SSL_CTX * or NULL, in which case the default ctx is used.
1441  *
1442  * Return value:
1443  *
1444  *  0 - Success. Connection is ready for communication.
1445  * <0 - Error. Can't create a TLS stream.
1446  * >0 - Partial success.
1447  *        Do a select (using information from lber_pvt_sb_needs_{read,write}
1448  *              and call again.
1449  */
1450
1451 static int
1452 ldap_int_tls_connect( LDAP *ld, LDAPConn *conn )
1453 {
1454         Sockbuf *sb = conn->lconn_sb;
1455         int     err;
1456         tls_session     *ssl;
1457
1458         if ( HAS_TLS( sb ) ) {
1459                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&ssl );
1460
1461         } else {
1462                 struct ldapoptions *lo;
1463                 tls_ctx *ctx;
1464
1465                 ctx = ld->ld_options.ldo_tls_ctx;
1466
1467                 ssl = alloc_handle( ctx, 0 );
1468
1469                 if ( ssl == NULL ) return -1;
1470
1471 #ifdef LDAP_DEBUG
1472                 ber_sockbuf_add_io( sb, &ber_sockbuf_io_debug,
1473                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)"tls_" );
1474 #endif
1475                 ber_sockbuf_add_io( sb, &sb_tls_sbio,
1476                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)ssl );
1477
1478                 lo = LDAP_INT_GLOBAL_OPT();   
1479                 if( ctx == NULL ) {
1480                         ctx = lo->ldo_tls_ctx;
1481                         ld->ld_options.ldo_tls_ctx = ctx;
1482                         tls_ctx_ref( ctx );
1483                 }
1484                 if ( ld->ld_options.ldo_tls_connect_cb )
1485                         ld->ld_options.ldo_tls_connect_cb( ld, ssl, ctx,
1486                         ld->ld_options.ldo_tls_connect_arg );
1487                 if ( lo && lo->ldo_tls_connect_cb && lo->ldo_tls_connect_cb !=
1488                         ld->ld_options.ldo_tls_connect_cb )
1489                         lo->ldo_tls_connect_cb( ld, ssl, ctx, lo->ldo_tls_connect_arg );
1490         }
1491
1492         err = tls_session_connect( ssl );
1493
1494 #ifdef HAVE_WINSOCK
1495         errno = WSAGetLastError();
1496 #endif
1497
1498 #ifdef HAVE_GNUTLS
1499         if ( err < 0 )
1500 #else
1501         if ( err <= 0 )
1502 #endif
1503         {
1504                 if ( update_flags( sb, ssl, err )) {
1505                         return 1;
1506                 }
1507
1508 #ifndef HAVE_GNUTLS
1509                 if ((err = ERR_peek_error()))
1510 #endif
1511                 {
1512                         if ( ld->ld_error ) {
1513                                 LDAP_FREE( ld->ld_error );
1514                         }
1515 #ifdef HAVE_GNUTLS
1516                         ld->ld_error = LDAP_STRDUP(gnutls_strerror( err ));
1517 #else
1518                         {
1519                                 char buf[256];
1520                                 ld->ld_error = LDAP_STRDUP(ERR_error_string(err, buf));
1521                         }
1522 #endif
1523 #ifdef HAVE_EBCDIC
1524                         if ( ld->ld_error ) __etoa(ld->ld_error);
1525 #endif
1526                 }
1527
1528                 Debug( LDAP_DEBUG_ANY,"TLS: can't connect: %s.\n",
1529                         ld->ld_error ? ld->ld_error : "" ,0,0);
1530
1531                 ber_sockbuf_remove_io( sb, &sb_tls_sbio,
1532                         LBER_SBIOD_LEVEL_TRANSPORT );
1533 #ifdef LDAP_DEBUG
1534                 ber_sockbuf_remove_io( sb, &ber_sockbuf_io_debug,
1535                         LBER_SBIOD_LEVEL_TRANSPORT );
1536 #endif
1537                 return -1;
1538         }
1539
1540 #ifdef HAVE_GNUTLS
1541         if ( ld->ld_options.ldo_tls_require_cert != LDAP_OPT_X_TLS_NEVER ) {
1542                 err = tls_cert_verify( ssl );
1543                 if ( err && ld->ld_options.ldo_tls_require_cert != LDAP_OPT_X_TLS_ALLOW )
1544                         return err;
1545         }
1546 #endif
1547
1548         return 0;
1549 }
1550
1551 /*
1552  * Call this to do a TLS accept on a sockbuf.
1553  * Everything else is the same as with tls_connect.
1554  */
1555 int
1556 ldap_pvt_tls_accept( Sockbuf *sb, void *ctx_arg )
1557 {
1558         int     err;
1559         tls_session     *ssl;
1560
1561         if ( HAS_TLS( sb ) ) {
1562                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&ssl );
1563
1564         } else {
1565                 ssl = alloc_handle( ctx_arg, 1 );
1566                 if ( ssl == NULL ) return -1;
1567
1568 #ifdef LDAP_DEBUG
1569                 ber_sockbuf_add_io( sb, &ber_sockbuf_io_debug,
1570                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)"tls_" );
1571 #endif
1572                 ber_sockbuf_add_io( sb, &sb_tls_sbio,
1573                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)ssl );
1574         }
1575
1576         err = tls_session_accept( ssl );
1577
1578 #ifdef HAVE_WINSOCK
1579         errno = WSAGetLastError();
1580 #endif
1581
1582 #ifdef HAVE_GNUTLS
1583         if ( err < 0 )
1584 #else
1585         if ( err <= 0 )
1586 #endif
1587         {
1588                 if ( update_flags( sb, ssl, err )) return 1;
1589
1590 #ifdef HAVE_GNUTLS
1591                 Debug( LDAP_DEBUG_ANY,"TLS: can't accept: %s.\n",
1592                         gnutls_strerror( err ),0,0 );
1593 #else
1594                 Debug( LDAP_DEBUG_ANY,"TLS: can't accept.\n",0,0,0 );
1595                 tls_report_error();
1596 #endif
1597                 ber_sockbuf_remove_io( sb, &sb_tls_sbio,
1598                         LBER_SBIOD_LEVEL_TRANSPORT );
1599 #ifdef LDAP_DEBUG
1600                 ber_sockbuf_remove_io( sb, &ber_sockbuf_io_debug,
1601                         LBER_SBIOD_LEVEL_TRANSPORT );
1602 #endif
1603                 return -1;
1604         }
1605
1606 #ifdef HAVE_GNUTLS
1607         if ( ssl->ctx->lo->ldo_tls_require_cert != LDAP_OPT_X_TLS_NEVER ) {
1608                 err = tls_cert_verify( ssl );
1609                 if ( err && ssl->ctx->lo->ldo_tls_require_cert != LDAP_OPT_X_TLS_ALLOW )
1610                         return err;
1611         }
1612 #endif
1613         return 0;
1614 }
1615
1616 int
1617 ldap_pvt_tls_inplace ( Sockbuf *sb )
1618 {
1619         return HAS_TLS( sb ) ? 1 : 0;
1620 }
1621
1622 int
1623 ldap_tls_inplace( LDAP *ld )
1624 {
1625         Sockbuf         *sb = NULL;
1626
1627         if ( ld->ld_defconn && ld->ld_defconn->lconn_sb ) {
1628                 sb = ld->ld_defconn->lconn_sb;
1629
1630         } else if ( ld->ld_sb ) {
1631                 sb = ld->ld_sb;
1632
1633         } else {
1634                 return 0;
1635         }
1636
1637         return ldap_pvt_tls_inplace( sb );
1638 }
1639
1640 static void
1641 x509_cert_get_dn( struct berval *cert, struct berval *dn, int get_subject )
1642 {
1643         BerElementBuffer berbuf;
1644         BerElement *ber = (BerElement *)&berbuf;
1645         ber_tag_t tag;
1646         ber_len_t len;
1647         ber_int_t i;
1648
1649         ber_init2( ber, cert, LBER_USE_DER );
1650         tag = ber_skip_tag( ber, &len );        /* Sequence */
1651         tag = ber_skip_tag( ber, &len );        /* Sequence */
1652         tag = ber_skip_tag( ber, &len );        /* Context + Constructed (version) */
1653         if ( tag == 0xa0 )      /* Version is optional */
1654                 tag = ber_get_int( ber, &i );   /* Int: Version */
1655         tag = ber_get_int( ber, &i );   /* Int: Serial */
1656         tag = ber_skip_tag( ber, &len );        /* Sequence: Signature */
1657         ber_skip_data( ber, len );
1658         if ( !get_subject ) {
1659                 tag = ber_peek_tag( ber, &len );        /* Sequence: Issuer DN */
1660         } else {
1661                 tag = ber_skip_tag( ber, &len );
1662                 ber_skip_data( ber, len );
1663                 tag = ber_skip_tag( ber, &len );        /* Sequence: Validity */
1664                 ber_skip_data( ber, len );
1665                 tag = ber_peek_tag( ber, &len );        /* Sequence: Subject DN */
1666         }
1667         len = ber_ptrlen( ber );
1668         dn->bv_val = cert->bv_val + len;
1669         dn->bv_len = cert->bv_len - len;
1670 }
1671
1672 #ifdef HAVE_GNUTLS
1673 static int
1674 tls_get_cert_dn( tls_session *session, struct berval *dnbv )
1675 {
1676         if ( !session->peer_der_dn.bv_val ) {
1677                 const gnutls_datum_t *peer_cert_list;
1678                 int list_size;
1679                 struct berval bv;
1680
1681                 peer_cert_list = gnutls_certificate_get_peers( session->session, 
1682                                                         &list_size );
1683                 if ( !peer_cert_list ) return LDAP_INVALID_CREDENTIALS;
1684
1685                 bv.bv_len = peer_cert_list->size;
1686                 bv.bv_val = peer_cert_list->data;
1687
1688                 x509_cert_get_dn( &bv, &session->peer_der_dn, 1 );
1689                 *dnbv = session->peer_der_dn;
1690         }
1691         return 0;
1692 }
1693 #else
1694 static X509 *
1695 tls_get_cert( SSL *s )
1696 {
1697         /* If peer cert was bad, treat as if no cert was given */
1698         if (SSL_get_verify_result(s)) {
1699                 /* If we can send an alert, do so */
1700                 if (SSL_version(s) != SSL2_VERSION) {
1701                         ssl3_send_alert(s,SSL3_AL_WARNING,SSL3_AD_BAD_CERTIFICATE);
1702                 }
1703                 return NULL;
1704         }
1705         return SSL_get_peer_certificate(s);
1706 }
1707
1708 static int
1709 tls_get_cert_dn( tls_session *session, struct berval *dnbv )
1710 {
1711         X509_NAME *xn;
1712         X509 *x = tls_get_cert( session );
1713         int len;
1714
1715         if ( !x )
1716                 return LDAP_INVALID_CREDENTIALS;
1717
1718         xn = X509_get_subject_name(x);
1719         dnbv->bv_len = i2d_X509_NAME( xn, NULL );
1720         dnbv->bv_val = xn->bytes->data;
1721         return 0;
1722 }
1723 #endif /* HAVE_GNUTLS */
1724
1725 int
1726 ldap_pvt_tls_get_peer_dn( void *s, struct berval *dn,
1727         LDAPDN_rewrite_dummy *func, unsigned flags )
1728 {
1729         tls_session *session = s;
1730         struct berval bvdn;
1731         int rc;
1732
1733         rc = tls_get_cert_dn( session, &bvdn );
1734         if ( rc ) return rc;
1735
1736         rc = ldap_X509dn2bv( &bvdn, dn, 
1737                             (LDAPDN_rewrite_func *)func, flags);
1738         return rc;
1739 }
1740
1741 /* what kind of hostname were we given? */
1742 #define IS_DNS  0
1743 #define IS_IP4  1
1744 #define IS_IP6  2
1745
1746 #ifdef HAVE_GNUTLS
1747
1748 int
1749 ldap_pvt_tls_check_hostname( LDAP *ld, void *s, const char *name_in )
1750 {
1751         tls_session *session = s;
1752         int i, ret;
1753         const gnutls_datum_t *peer_cert_list;
1754         int list_size;
1755         struct berval bv;
1756         char altname[NI_MAXHOST];
1757         size_t altnamesize;
1758
1759         gnutls_x509_crt_t cert;
1760         gnutls_datum_t *x;
1761         const char *name;
1762         char *ptr;
1763         char *domain = NULL;
1764 #ifdef LDAP_PF_INET6
1765         struct in6_addr addr;
1766 #else
1767         struct in_addr addr;
1768 #endif
1769         int n, len1 = 0, len2 = 0;
1770         int ntype = IS_DNS;
1771         time_t now = time(0);
1772
1773         if( ldap_int_hostname &&
1774                 ( !name_in || !strcasecmp( name_in, "localhost" ) ) )
1775         {
1776                 name = ldap_int_hostname;
1777         } else {
1778                 name = name_in;
1779         }
1780
1781         peer_cert_list = gnutls_certificate_get_peers( session->session, 
1782                                                 &list_size );
1783         if ( !peer_cert_list ) {
1784                 Debug( LDAP_DEBUG_ANY,
1785                         "TLS: unable to get peer certificate.\n",
1786                         0, 0, 0 );
1787                 /* If this was a fatal condition, things would have
1788                  * aborted long before now.
1789                  */
1790                 return LDAP_SUCCESS;
1791         }
1792         ret = gnutls_x509_crt_init( &cert );
1793         if ( ret < 0 )
1794                 return LDAP_LOCAL_ERROR;
1795         ret = gnutls_x509_crt_import( cert, peer_cert_list, GNUTLS_X509_FMT_DER );
1796         if ( ret ) {
1797                 gnutls_x509_crt_deinit( cert );
1798                 return LDAP_LOCAL_ERROR;
1799         }
1800
1801 #ifdef LDAP_PF_INET6
1802         if (name[0] == '[' && strchr(name, ']')) {
1803                 char *n2 = ldap_strdup(name+1);
1804                 *strchr(n2, ']') = 2;
1805                 if (inet_pton(AF_INET6, n2, &addr))
1806                         ntype = IS_IP6;
1807                 LDAP_FREE(n2);
1808         } else 
1809 #endif
1810         if ((ptr = strrchr(name, '.')) && isdigit((unsigned char)ptr[1])) {
1811                 if (inet_aton(name, (struct in_addr *)&addr)) ntype = IS_IP4;
1812         }
1813         
1814         if (ntype == IS_DNS) {
1815                 len1 = strlen(name);
1816                 domain = strchr(name, '.');
1817                 if (domain) {
1818                         len2 = len1 - (domain-name);
1819                 }
1820         }
1821
1822         for ( i=0, ret=0; ret >= 0; i++ ) {
1823                 altnamesize = sizeof(altname);
1824                 ret = gnutls_x509_crt_get_subject_alt_name( cert, i, 
1825                         altname, &altnamesize, NULL );
1826                 if ( ret < 0 ) break;
1827
1828                 /* ignore empty */
1829                 if ( altnamesize == 0 ) continue;
1830
1831                 if ( ret == GNUTLS_SAN_DNSNAME ) {
1832                         if (ntype != IS_DNS) continue;
1833         
1834                         /* Is this an exact match? */
1835                         if ((len1 == altnamesize) && !strncasecmp(name, altname, len1)) {
1836                                 break;
1837                         }
1838
1839                         /* Is this a wildcard match? */
1840                         if (domain && (altname[0] == '*') && (altname[1] == '.') &&
1841                                 (len2 == altnamesize-1) && !strncasecmp(domain, &altname[1], len2))
1842                         {
1843                                 break;
1844                         }
1845                 } else if ( ret == GNUTLS_SAN_IPADDRESS ) {
1846                         if (ntype == IS_DNS) continue;
1847
1848 #ifdef LDAP_PF_INET6
1849                         if (ntype == IS_IP6 && altnamesize != sizeof(struct in6_addr)) {
1850                                 continue;
1851                         } else
1852 #endif
1853                         if (ntype == IS_IP4 && altnamesize != sizeof(struct in_addr)) {
1854                                 continue;
1855                         }
1856                         if (!memcmp(altname, &addr, altnamesize)) {
1857                                 break;
1858                         }
1859                 }
1860         }
1861         if ( ret >= 0 ) {
1862                 ret = LDAP_SUCCESS;
1863         } else {
1864                 altnamesize = sizeof(altname);
1865                 ret = gnutls_x509_crt_get_dn_by_oid( cert, CN_OID,
1866                         0, 0, altname, &altnamesize );
1867                 if ( ret < 0 ) {
1868                         Debug( LDAP_DEBUG_ANY,
1869                                 "TLS: unable to get common name from peer certificate.\n",
1870                                 0, 0, 0 );
1871                         ret = LDAP_CONNECT_ERROR;
1872                         if ( ld->ld_error ) {
1873                                 LDAP_FREE( ld->ld_error );
1874                         }
1875                         ld->ld_error = LDAP_STRDUP(
1876                                 _("TLS: unable to get CN from peer certificate"));
1877
1878                 } else {
1879                         ret = LDAP_LOCAL_ERROR;
1880                         if ( len1 == altnamesize && strncasecmp(name, altname, altnamesize) == 0 ) {
1881                                 ret = LDAP_SUCCESS;
1882
1883                         } else if (( altname[0] == '*' ) && ( altname[1] == '.' )) {
1884                                         /* Is this a wildcard match? */
1885                                 if( domain &&
1886                                         (len2 == altnamesize-1) && !strncasecmp(domain, &altname[1], len2)) {
1887                                         ret = LDAP_SUCCESS;
1888                                 }
1889                         }
1890                 }
1891
1892                 if( ret == LDAP_LOCAL_ERROR ) {
1893                         altname[altnamesize] = '\0';
1894                         Debug( LDAP_DEBUG_ANY, "TLS: hostname (%s) does not match "
1895                                 "common name in certificate (%s).\n", 
1896                                 name, altname, 0 );
1897                         ret = LDAP_CONNECT_ERROR;
1898                         if ( ld->ld_error ) {
1899                                 LDAP_FREE( ld->ld_error );
1900                         }
1901                         ld->ld_error = LDAP_STRDUP(
1902                                 _("TLS: hostname does not match CN in peer certificate"));
1903                 }
1904         }
1905         gnutls_x509_crt_deinit( cert );
1906         return ret;
1907 }
1908
1909 #else /* !HAVE_GNUTLS */
1910
1911 int
1912 ldap_pvt_tls_check_hostname( LDAP *ld, void *s, const char *name_in )
1913 {
1914         int i, ret = LDAP_LOCAL_ERROR;
1915         X509 *x;
1916         const char *name;
1917         char *ptr;
1918         int ntype = IS_DNS;
1919 #ifdef LDAP_PF_INET6
1920         struct in6_addr addr;
1921 #else
1922         struct in_addr addr;
1923 #endif
1924
1925         if( ldap_int_hostname &&
1926                 ( !name_in || !strcasecmp( name_in, "localhost" ) ) )
1927         {
1928                 name = ldap_int_hostname;
1929         } else {
1930                 name = name_in;
1931         }
1932
1933         x = tls_get_cert((SSL *)s);
1934         if (!x) {
1935                 Debug( LDAP_DEBUG_ANY,
1936                         "TLS: unable to get peer certificate.\n",
1937                         0, 0, 0 );
1938                 /* If this was a fatal condition, things would have
1939                  * aborted long before now.
1940                  */
1941                 return LDAP_SUCCESS;
1942         }
1943
1944 #ifdef LDAP_PF_INET6
1945         if (name[0] == '[' && strchr(name, ']')) {
1946                 char *n2 = ldap_strdup(name+1);
1947                 *strchr(n2, ']') = 2;
1948                 if (inet_pton(AF_INET6, n2, &addr))
1949                         ntype = IS_IP6;
1950                 LDAP_FREE(n2);
1951         } else 
1952 #endif
1953         if ((ptr = strrchr(name, '.')) && isdigit((unsigned char)ptr[1])) {
1954                 if (inet_aton(name, (struct in_addr *)&addr)) ntype = IS_IP4;
1955         }
1956         
1957         i = X509_get_ext_by_NID(x, NID_subject_alt_name, -1);
1958         if (i >= 0) {
1959                 X509_EXTENSION *ex;
1960                 STACK_OF(GENERAL_NAME) *alt;
1961
1962                 ex = X509_get_ext(x, i);
1963                 alt = X509V3_EXT_d2i(ex);
1964                 if (alt) {
1965                         int n, len1 = 0, len2 = 0;
1966                         char *domain = NULL;
1967                         GENERAL_NAME *gn;
1968
1969                         if (ntype == IS_DNS) {
1970                                 len1 = strlen(name);
1971                                 domain = strchr(name, '.');
1972                                 if (domain) {
1973                                         len2 = len1 - (domain-name);
1974                                 }
1975                         }
1976                         n = sk_GENERAL_NAME_num(alt);
1977                         for (i=0; i<n; i++) {
1978                                 char *sn;
1979                                 int sl;
1980                                 gn = sk_GENERAL_NAME_value(alt, i);
1981                                 if (gn->type == GEN_DNS) {
1982                                         if (ntype != IS_DNS) continue;
1983
1984                                         sn = (char *) ASN1_STRING_data(gn->d.ia5);
1985                                         sl = ASN1_STRING_length(gn->d.ia5);
1986
1987                                         /* ignore empty */
1988                                         if (sl == 0) continue;
1989
1990                                         /* Is this an exact match? */
1991                                         if ((len1 == sl) && !strncasecmp(name, sn, len1)) {
1992                                                 break;
1993                                         }
1994
1995                                         /* Is this a wildcard match? */
1996                                         if (domain && (sn[0] == '*') && (sn[1] == '.') &&
1997                                                 (len2 == sl-1) && !strncasecmp(domain, &sn[1], len2))
1998                                         {
1999                                                 break;
2000                                         }
2001
2002                                 } else if (gn->type == GEN_IPADD) {
2003                                         if (ntype == IS_DNS) continue;
2004
2005                                         sn = (char *) ASN1_STRING_data(gn->d.ia5);
2006                                         sl = ASN1_STRING_length(gn->d.ia5);
2007
2008 #ifdef LDAP_PF_INET6
2009                                         if (ntype == IS_IP6 && sl != sizeof(struct in6_addr)) {
2010                                                 continue;
2011                                         } else
2012 #endif
2013                                         if (ntype == IS_IP4 && sl != sizeof(struct in_addr)) {
2014                                                 continue;
2015                                         }
2016                                         if (!memcmp(sn, &addr, sl)) {
2017                                                 break;
2018                                         }
2019                                 }
2020                         }
2021
2022                         GENERAL_NAMES_free(alt);
2023                         if (i < n) {    /* Found a match */
2024                                 ret = LDAP_SUCCESS;
2025                         }
2026                 }
2027         }
2028
2029         if (ret != LDAP_SUCCESS) {
2030                 X509_NAME *xn;
2031                 char buf[2048];
2032                 buf[0] = '\0';
2033
2034                 xn = X509_get_subject_name(x);
2035                 if( X509_NAME_get_text_by_NID( xn, NID_commonName,
2036                         buf, sizeof(buf)) == -1)
2037                 {
2038                         Debug( LDAP_DEBUG_ANY,
2039                                 "TLS: unable to get common name from peer certificate.\n",
2040                                 0, 0, 0 );
2041                         ret = LDAP_CONNECT_ERROR;
2042                         if ( ld->ld_error ) {
2043                                 LDAP_FREE( ld->ld_error );
2044                         }
2045                         ld->ld_error = LDAP_STRDUP(
2046                                 _("TLS: unable to get CN from peer certificate"));
2047
2048                 } else if (strcasecmp(name, buf) == 0 ) {
2049                         ret = LDAP_SUCCESS;
2050
2051                 } else if (( buf[0] == '*' ) && ( buf[1] == '.' )) {
2052                         char *domain = strchr(name, '.');
2053                         if( domain ) {
2054                                 size_t dlen = 0;
2055                                 size_t sl;
2056
2057                                 sl = strlen(name);
2058                                 dlen = sl - (domain-name);
2059                                 sl = strlen(buf);
2060
2061                                 /* Is this a wildcard match? */
2062                                 if ((dlen == sl-1) && !strncasecmp(domain, &buf[1], dlen)) {
2063                                         ret = LDAP_SUCCESS;
2064                                 }
2065                         }
2066                 }
2067
2068                 if( ret == LDAP_LOCAL_ERROR ) {
2069                         Debug( LDAP_DEBUG_ANY, "TLS: hostname (%s) does not match "
2070                                 "common name in certificate (%s).\n", 
2071                                 name, buf, 0 );
2072                         ret = LDAP_CONNECT_ERROR;
2073                         if ( ld->ld_error ) {
2074                                 LDAP_FREE( ld->ld_error );
2075                         }
2076                         ld->ld_error = LDAP_STRDUP(
2077                                 _("TLS: hostname does not match CN in peer certificate"));
2078                 }
2079         }
2080         X509_free(x);
2081         return ret;
2082 }
2083 #endif
2084
2085 int
2086 ldap_int_tls_config( LDAP *ld, int option, const char *arg )
2087 {
2088         int i;
2089
2090         switch( option ) {
2091         case LDAP_OPT_X_TLS_CACERTFILE:
2092         case LDAP_OPT_X_TLS_CACERTDIR:
2093         case LDAP_OPT_X_TLS_CERTFILE:
2094         case LDAP_OPT_X_TLS_KEYFILE:
2095         case LDAP_OPT_X_TLS_RANDOM_FILE:
2096         case LDAP_OPT_X_TLS_CIPHER_SUITE:
2097         case LDAP_OPT_X_TLS_DHFILE:
2098 #ifdef HAVE_GNUTLS
2099         case LDAP_OPT_X_TLS_CRLFILE:
2100 #endif
2101                 return ldap_pvt_tls_set_option( ld, option, (void *) arg );
2102
2103         case LDAP_OPT_X_TLS_REQUIRE_CERT:
2104         case LDAP_OPT_X_TLS:
2105                 i = -1;
2106                 if ( strcasecmp( arg, "never" ) == 0 ) {
2107                         i = LDAP_OPT_X_TLS_NEVER ;
2108
2109                 } else if ( strcasecmp( arg, "demand" ) == 0 ) {
2110                         i = LDAP_OPT_X_TLS_DEMAND ;
2111
2112                 } else if ( strcasecmp( arg, "allow" ) == 0 ) {
2113                         i = LDAP_OPT_X_TLS_ALLOW ;
2114
2115                 } else if ( strcasecmp( arg, "try" ) == 0 ) {
2116                         i = LDAP_OPT_X_TLS_TRY ;
2117
2118                 } else if ( ( strcasecmp( arg, "hard" ) == 0 ) ||
2119                         ( strcasecmp( arg, "on" ) == 0 ) ||
2120                         ( strcasecmp( arg, "yes" ) == 0) ||
2121                         ( strcasecmp( arg, "true" ) == 0 ) )
2122                 {
2123                         i = LDAP_OPT_X_TLS_HARD ;
2124                 }
2125
2126                 if (i >= 0) {
2127                         return ldap_pvt_tls_set_option( ld, option, &i );
2128                 }
2129                 return -1;
2130 #ifdef HAVE_OPENSSL_CRL
2131         case LDAP_OPT_X_TLS_CRLCHECK:
2132                 i = -1;
2133                 if ( strcasecmp( arg, "none" ) == 0 ) {
2134                         i = LDAP_OPT_X_TLS_CRL_NONE ;
2135                 } else if ( strcasecmp( arg, "peer" ) == 0 ) {
2136                         i = LDAP_OPT_X_TLS_CRL_PEER ;
2137                 } else if ( strcasecmp( arg, "all" ) == 0 ) {
2138                         i = LDAP_OPT_X_TLS_CRL_ALL ;
2139                 }
2140                 if (i >= 0) {
2141                         return ldap_pvt_tls_set_option( ld, option, &i );
2142                 }
2143                 return -1;
2144 #endif
2145         }
2146         return -1;
2147 }
2148
2149 int
2150 ldap_pvt_tls_get_option( LDAP *ld, int option, void *arg )
2151 {
2152         struct ldapoptions *lo;
2153
2154         if( ld != NULL ) {
2155                 assert( LDAP_VALID( ld ) );
2156
2157                 if( !LDAP_VALID( ld ) ) {
2158                         return LDAP_OPT_ERROR;
2159                 }
2160
2161                 lo = &ld->ld_options;
2162
2163         } else {
2164                 /* Get pointer to global option structure */
2165                 lo = LDAP_INT_GLOBAL_OPT();   
2166                 if ( lo == NULL ) {
2167                         return LDAP_NO_MEMORY;
2168                 }
2169         }
2170
2171         switch( option ) {
2172         case LDAP_OPT_X_TLS:
2173                 *(int *)arg = lo->ldo_tls_mode;
2174                 break;
2175         case LDAP_OPT_X_TLS_CTX:
2176                 *(void **)arg = lo->ldo_tls_ctx;
2177                 if ( lo->ldo_tls_ctx ) {
2178                         tls_ctx_ref( lo->ldo_tls_ctx );
2179                 }
2180                 break;
2181         case LDAP_OPT_X_TLS_CACERTFILE:
2182                 *(char **)arg = lo->ldo_tls_cacertfile ?
2183                         LDAP_STRDUP( lo->ldo_tls_cacertfile ) : NULL;
2184                 break;
2185         case LDAP_OPT_X_TLS_CACERTDIR:
2186                 *(char **)arg = lo->ldo_tls_cacertdir ?
2187                         LDAP_STRDUP( lo->ldo_tls_cacertdir ) : NULL;
2188                 break;
2189         case LDAP_OPT_X_TLS_CERTFILE:
2190                 *(char **)arg = lo->ldo_tls_certfile ?
2191                         LDAP_STRDUP( lo->ldo_tls_certfile ) : NULL;
2192                 break;
2193         case LDAP_OPT_X_TLS_KEYFILE:
2194                 *(char **)arg = lo->ldo_tls_keyfile ?
2195                         LDAP_STRDUP( lo->ldo_tls_keyfile ) : NULL;
2196                 break;
2197         case LDAP_OPT_X_TLS_DHFILE:
2198                 *(char **)arg = lo->ldo_tls_dhfile ?
2199                         LDAP_STRDUP( lo->ldo_tls_dhfile ) : NULL;
2200                 break;
2201 #ifdef HAVE_GNUTLS
2202         case LDAP_OPT_X_TLS_CRLFILE:
2203                 *(char **)arg = lo->ldo_tls_crlfile ?
2204                         LDAP_STRDUP( lo->ldo_tls_crlfile ) : NULL;
2205                 break;
2206 #endif
2207         case LDAP_OPT_X_TLS_REQUIRE_CERT:
2208                 *(int *)arg = lo->ldo_tls_require_cert;
2209                 break;
2210 #ifdef HAVE_OPENSSL_CRL
2211         case LDAP_OPT_X_TLS_CRLCHECK:
2212                 *(int *)arg = lo->ldo_tls_crlcheck;
2213                 break;
2214 #endif
2215         case LDAP_OPT_X_TLS_CIPHER_SUITE:
2216                 *(char **)arg = lo->ldo_tls_ciphersuite ?
2217                         LDAP_STRDUP( lo->ldo_tls_ciphersuite ) : NULL;
2218                 break;
2219         case LDAP_OPT_X_TLS_RANDOM_FILE:
2220 #ifdef HAVE_OPENSSL
2221                 *(char **)arg = tls_opt_randfile ?
2222                         LDAP_STRDUP( tls_opt_randfile ) : NULL;
2223 #else
2224                 *(char **)arg = NULL;
2225 #endif
2226                 break;
2227         case LDAP_OPT_X_TLS_SSL_CTX: {
2228                 void *retval = 0;
2229                 if ( ld != NULL ) {
2230                         LDAPConn *conn = ld->ld_defconn;
2231                         if ( conn != NULL ) {
2232                                 Sockbuf *sb = conn->lconn_sb;
2233                                 retval = ldap_pvt_tls_sb_ctx( sb );
2234                         }
2235                 }
2236                 *(void **)arg = retval;
2237                 break;
2238         }
2239         case LDAP_OPT_X_TLS_CONNECT_CB:
2240                 *(LDAP_TLS_CONNECT_CB **)arg = lo->ldo_tls_connect_cb;
2241                 break;
2242         case LDAP_OPT_X_TLS_CONNECT_ARG:
2243                 *(void **)arg = lo->ldo_tls_connect_arg;
2244                 break;
2245         default:
2246                 return -1;
2247         }
2248         return 0;
2249 }
2250
2251 int
2252 ldap_pvt_tls_set_option( LDAP *ld, int option, void *arg )
2253 {
2254         struct ldapoptions *lo;
2255
2256         if( ld != NULL ) {
2257                 assert( LDAP_VALID( ld ) );
2258
2259                 if( !LDAP_VALID( ld ) ) {
2260                         return LDAP_OPT_ERROR;
2261                 }
2262
2263                 lo = &ld->ld_options;
2264
2265         } else {
2266                 /* Get pointer to global option structure */
2267                 lo = LDAP_INT_GLOBAL_OPT();   
2268                 if ( lo == NULL ) {
2269                         return LDAP_NO_MEMORY;
2270                 }
2271         }
2272
2273         switch( option ) {
2274         case LDAP_OPT_X_TLS:
2275                 if ( !arg ) return -1;
2276
2277                 switch( *(int *) arg ) {
2278                 case LDAP_OPT_X_TLS_NEVER:
2279                 case LDAP_OPT_X_TLS_DEMAND:
2280                 case LDAP_OPT_X_TLS_ALLOW:
2281                 case LDAP_OPT_X_TLS_TRY:
2282                 case LDAP_OPT_X_TLS_HARD:
2283                         if (lo != NULL) {
2284                                 lo->ldo_tls_mode = *(int *)arg;
2285                         }
2286
2287                         return 0;
2288                 }
2289                 return -1;
2290
2291         case LDAP_OPT_X_TLS_CTX:
2292                 if ( lo->ldo_tls_ctx )
2293                         ldap_pvt_tls_ctx_free( lo->ldo_tls_ctx );
2294                 lo->ldo_tls_ctx = arg;
2295                 tls_ctx_ref( lo->ldo_tls_ctx );
2296                 return 0;
2297         case LDAP_OPT_X_TLS_CONNECT_CB:
2298                 lo->ldo_tls_connect_cb = (LDAP_TLS_CONNECT_CB *)arg;
2299                 return 0;
2300         case LDAP_OPT_X_TLS_CONNECT_ARG:
2301                 lo->ldo_tls_connect_arg = arg;
2302                 return 0;
2303         case LDAP_OPT_X_TLS_CACERTFILE:
2304                 if ( lo->ldo_tls_cacertfile ) LDAP_FREE( lo->ldo_tls_cacertfile );
2305                 lo->ldo_tls_cacertfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
2306                 return 0;
2307         case LDAP_OPT_X_TLS_CACERTDIR:
2308                 if ( lo->ldo_tls_cacertdir ) LDAP_FREE( lo->ldo_tls_cacertdir );
2309                 lo->ldo_tls_cacertdir = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
2310                 return 0;
2311         case LDAP_OPT_X_TLS_CERTFILE:
2312                 if ( lo->ldo_tls_certfile ) LDAP_FREE( lo->ldo_tls_certfile );
2313                 lo->ldo_tls_certfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
2314                 return 0;
2315         case LDAP_OPT_X_TLS_KEYFILE:
2316                 if ( lo->ldo_tls_keyfile ) LDAP_FREE( lo->ldo_tls_keyfile );
2317                 lo->ldo_tls_keyfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
2318                 return 0;
2319         case LDAP_OPT_X_TLS_DHFILE:
2320                 if ( lo->ldo_tls_dhfile ) LDAP_FREE( lo->ldo_tls_dhfile );
2321                 lo->ldo_tls_dhfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
2322                 return 0;
2323 #ifdef HAVE_GNUTLS
2324         case LDAP_OPT_X_TLS_CRLFILE:
2325                 if ( lo->ldo_tls_crlfile ) LDAP_FREE( lo->ldo_tls_crlfile );
2326                 lo->ldo_tls_crlfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
2327                 return 0;
2328 #endif
2329         case LDAP_OPT_X_TLS_REQUIRE_CERT:
2330                 if ( !arg ) return -1;
2331                 switch( *(int *) arg ) {
2332                 case LDAP_OPT_X_TLS_NEVER:
2333                 case LDAP_OPT_X_TLS_DEMAND:
2334                 case LDAP_OPT_X_TLS_ALLOW:
2335                 case LDAP_OPT_X_TLS_TRY:
2336                 case LDAP_OPT_X_TLS_HARD:
2337                         lo->ldo_tls_require_cert = * (int *) arg;
2338                         return 0;
2339                 }
2340                 return -1;
2341 #ifdef HAVE_OPENSSL_CRL
2342         case LDAP_OPT_X_TLS_CRLCHECK:
2343                 if ( !arg ) return -1;
2344                 switch( *(int *) arg ) {
2345                 case LDAP_OPT_X_TLS_CRL_NONE:
2346                 case LDAP_OPT_X_TLS_CRL_PEER:
2347                 case LDAP_OPT_X_TLS_CRL_ALL:
2348                         lo->ldo_tls_crlcheck = * (int *) arg;
2349                         return 0;
2350                 }
2351                 return -1;
2352 #endif
2353         case LDAP_OPT_X_TLS_CIPHER_SUITE:
2354                 if ( lo->ldo_tls_ciphersuite ) LDAP_FREE( lo->ldo_tls_ciphersuite );
2355                 lo->ldo_tls_ciphersuite = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
2356                 return 0;
2357
2358         case LDAP_OPT_X_TLS_RANDOM_FILE:
2359                 if ( ld != NULL )
2360                         return -1;
2361 #ifdef HAVE_OPENSSL
2362                 if (tls_opt_randfile ) LDAP_FREE (tls_opt_randfile );
2363                 tls_opt_randfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
2364 #endif
2365                 break;
2366
2367         case LDAP_OPT_X_TLS_NEWCTX:
2368                 if ( !arg ) return -1;
2369                 if ( lo->ldo_tls_ctx )
2370                         ldap_pvt_tls_ctx_free( lo->ldo_tls_ctx );
2371                 lo->ldo_tls_ctx = NULL;
2372                 return ldap_int_tls_init_ctx( lo, *(int *)arg );
2373         default:
2374                 return -1;
2375         }
2376         return 0;
2377 }
2378
2379 int
2380 ldap_int_tls_start ( LDAP *ld, LDAPConn *conn, LDAPURLDesc *srv )
2381 {
2382         Sockbuf *sb = conn->lconn_sb;
2383         char *host;
2384         void *ssl;
2385
2386         if( srv ) {
2387                 host = srv->lud_host;
2388         } else {
2389                 host = conn->lconn_server->lud_host;
2390         }
2391
2392         /* avoid NULL host */
2393         if( host == NULL ) {
2394                 host = "localhost";
2395         }
2396
2397         (void) ldap_pvt_tls_init();
2398
2399         /*
2400          * Fortunately, the lib uses blocking io...
2401          */
2402         if ( ldap_int_tls_connect( ld, conn ) < 0 ) {
2403                 ld->ld_errno = LDAP_CONNECT_ERROR;
2404                 return (ld->ld_errno);
2405         }
2406
2407         ssl = ldap_pvt_tls_sb_ctx( sb );
2408         assert( ssl != NULL );
2409
2410         /* 
2411          * compare host with name(s) in certificate
2412          */
2413         if (ld->ld_options.ldo_tls_require_cert != LDAP_OPT_X_TLS_NEVER) {
2414                 ld->ld_errno = ldap_pvt_tls_check_hostname( ld, ssl, host );
2415                 if (ld->ld_errno != LDAP_SUCCESS) {
2416                         return ld->ld_errno;
2417                 }
2418         }
2419
2420         return LDAP_SUCCESS;
2421 }
2422
2423 #ifdef HAVE_OPENSSL
2424 /* Derived from openssl/apps/s_cb.c */
2425 static void
2426 tls_info_cb( const SSL *ssl, int where, int ret )
2427 {
2428         int w;
2429         char *op;
2430         char *state = (char *) SSL_state_string_long( (SSL *)ssl );
2431
2432         w = where & ~SSL_ST_MASK;
2433         if ( w & SSL_ST_CONNECT ) {
2434                 op = "SSL_connect";
2435         } else if ( w & SSL_ST_ACCEPT ) {
2436                 op = "SSL_accept";
2437         } else {
2438                 op = "undefined";
2439         }
2440
2441 #ifdef HAVE_EBCDIC
2442         if ( state ) {
2443                 state = LDAP_STRDUP( state );
2444                 __etoa( state );
2445         }
2446 #endif
2447         if ( where & SSL_CB_LOOP ) {
2448                 Debug( LDAP_DEBUG_TRACE,
2449                            "TLS trace: %s:%s\n",
2450                            op, state, 0 );
2451
2452         } else if ( where & SSL_CB_ALERT ) {
2453                 char *atype = (char *) SSL_alert_type_string_long( ret );
2454                 char *adesc = (char *) SSL_alert_desc_string_long( ret );
2455                 op = ( where & SSL_CB_READ ) ? "read" : "write";
2456 #ifdef HAVE_EBCDIC
2457                 if ( atype ) {
2458                         atype = LDAP_STRDUP( atype );
2459                         __etoa( atype );
2460                 }
2461                 if ( adesc ) {
2462                         adesc = LDAP_STRDUP( adesc );
2463                         __etoa( adesc );
2464                 }
2465 #endif
2466                 Debug( LDAP_DEBUG_TRACE,
2467                            "TLS trace: SSL3 alert %s:%s:%s\n",
2468                            op, atype, adesc );
2469 #ifdef HAVE_EBCDIC
2470                 if ( atype ) LDAP_FREE( atype );
2471                 if ( adesc ) LDAP_FREE( adesc );
2472 #endif
2473         } else if ( where & SSL_CB_EXIT ) {
2474                 if ( ret == 0 ) {
2475                         Debug( LDAP_DEBUG_TRACE,
2476                                    "TLS trace: %s:failed in %s\n",
2477                                    op, state, 0 );
2478                 } else if ( ret < 0 ) {
2479                         Debug( LDAP_DEBUG_TRACE,
2480                                    "TLS trace: %s:error in %s\n",
2481                                    op, state, 0 );
2482                 }
2483         }
2484 #ifdef HAVE_EBCDIC
2485         if ( state ) LDAP_FREE( state );
2486 #endif
2487 }
2488
2489 static int
2490 tls_verify_cb( int ok, X509_STORE_CTX *ctx )
2491 {
2492         X509 *cert;
2493         int errnum;
2494         int errdepth;
2495         X509_NAME *subject;
2496         X509_NAME *issuer;
2497         char *sname;
2498         char *iname;
2499         char *certerr = NULL;
2500
2501         cert = X509_STORE_CTX_get_current_cert( ctx );
2502         errnum = X509_STORE_CTX_get_error( ctx );
2503         errdepth = X509_STORE_CTX_get_error_depth( ctx );
2504
2505         /*
2506          * X509_get_*_name return pointers to the internal copies of
2507          * those things requested.  So do not free them.
2508          */
2509         subject = X509_get_subject_name( cert );
2510         issuer = X509_get_issuer_name( cert );
2511         /* X509_NAME_oneline, if passed a NULL buf, allocate memomry */
2512         sname = X509_NAME_oneline( subject, NULL, 0 );
2513         iname = X509_NAME_oneline( issuer, NULL, 0 );
2514         if ( !ok ) certerr = (char *)X509_verify_cert_error_string( errnum );
2515 #ifdef HAVE_EBCDIC
2516         if ( sname ) __etoa( sname );
2517         if ( iname ) __etoa( iname );
2518         if ( certerr ) {
2519                 certerr = LDAP_STRDUP( certerr );
2520                 __etoa( certerr );
2521         }
2522 #endif
2523         Debug( LDAP_DEBUG_TRACE,
2524                    "TLS certificate verification: depth: %d, err: %d, subject: %s,",
2525                    errdepth, errnum,
2526                    sname ? sname : "-unknown-" );
2527         Debug( LDAP_DEBUG_TRACE, " issuer: %s\n", iname ? iname : "-unknown-", 0, 0 );
2528         if ( !ok ) {
2529                 Debug( LDAP_DEBUG_ANY,
2530                         "TLS certificate verification: Error, %s\n",
2531                         certerr, 0, 0 );
2532         }
2533         if ( sname )
2534                 CRYPTO_free ( sname );
2535         if ( iname )
2536                 CRYPTO_free ( iname );
2537 #ifdef HAVE_EBCDIC
2538         if ( certerr ) LDAP_FREE( certerr );
2539 #endif
2540         return ok;
2541 }
2542
2543 static int
2544 tls_verify_ok( int ok, X509_STORE_CTX *ctx )
2545 {
2546         (void) tls_verify_cb( ok, ctx );
2547         return 1;
2548 }
2549
2550 /* Inspired by ERR_print_errors in OpenSSL */
2551 static void
2552 tls_report_error( void )
2553 {
2554         unsigned long l;
2555         char buf[200];
2556         const char *file;
2557         int line;
2558
2559         while ( ( l = ERR_get_error_line( &file, &line ) ) != 0 ) {
2560                 ERR_error_string_n( l, buf, sizeof( buf ) );
2561 #ifdef HAVE_EBCDIC
2562                 if ( file ) {
2563                         file = LDAP_STRDUP( file );
2564                         __etoa( (char *)file );
2565                 }
2566                 __etoa( buf );
2567 #endif
2568                 Debug( LDAP_DEBUG_ANY, "TLS: %s %s:%d\n",
2569                         buf, file, line );
2570 #ifdef HAVE_EBCDIC
2571                 if ( file ) LDAP_FREE( (void *)file );
2572 #endif
2573         }
2574 }
2575
2576 static RSA *
2577 tls_tmp_rsa_cb( SSL *ssl, int is_export, int key_length )
2578 {
2579         RSA *tmp_rsa;
2580
2581         /* FIXME:  Pregenerate the key on startup */
2582         /* FIXME:  Who frees the key? */
2583         tmp_rsa = RSA_generate_key( key_length, RSA_F4, NULL, NULL );
2584
2585         if ( !tmp_rsa ) {
2586                 Debug( LDAP_DEBUG_ANY,
2587                         "TLS: Failed to generate temporary %d-bit %s RSA key\n",
2588                         key_length, is_export ? "export" : "domestic", 0 );
2589                 return NULL;
2590         }
2591         return tmp_rsa;
2592 }
2593
2594 static int
2595 tls_seed_PRNG( const char *randfile )
2596 {
2597 #ifndef URANDOM_DEVICE
2598         /* no /dev/urandom (or equiv) */
2599         long total=0;
2600         char buffer[MAXPATHLEN];
2601
2602         if (randfile == NULL) {
2603                 /* The seed file is $RANDFILE if defined, otherwise $HOME/.rnd.
2604                  * If $HOME is not set or buffer too small to hold the pathname,
2605                  * an error occurs.     - From RAND_file_name() man page.
2606                  * The fact is that when $HOME is NULL, .rnd is used.
2607                  */
2608                 randfile = RAND_file_name( buffer, sizeof( buffer ) );
2609
2610         } else if (RAND_egd(randfile) > 0) {
2611                 /* EGD socket */
2612                 return 0;
2613         }
2614
2615         if (randfile == NULL) {
2616                 Debug( LDAP_DEBUG_ANY,
2617                         "TLS: Use configuration file or $RANDFILE to define seed PRNG\n",
2618                         0, 0, 0);
2619                 return -1;
2620         }
2621
2622         total = RAND_load_file(randfile, -1);
2623
2624         if (RAND_status() == 0) {
2625                 Debug( LDAP_DEBUG_ANY,
2626                         "TLS: PRNG not been seeded with enough data\n",
2627                         0, 0, 0);
2628                 return -1;
2629         }
2630
2631         /* assume if there was enough bits to seed that it's okay
2632          * to write derived bits to the file
2633          */
2634         RAND_write_file(randfile);
2635
2636 #endif
2637
2638         return 0;
2639 }
2640
2641 struct dhinfo {
2642         int keylength;
2643         const char *pem;
2644         size_t size;
2645 };
2646
2647
2648 /* From the OpenSSL 0.9.7 distro */
2649 static const char dhpem512[] =
2650 "-----BEGIN DH PARAMETERS-----\n\
2651 MEYCQQDaWDwW2YUiidDkr3VvTMqS3UvlM7gE+w/tlO+cikQD7VdGUNNpmdsp13Yn\n\
2652 a6LT1BLiGPTdHghM9tgAPnxHdOgzAgEC\n\
2653 -----END DH PARAMETERS-----\n";
2654
2655 static const char dhpem1024[] =
2656 "-----BEGIN DH PARAMETERS-----\n\
2657 MIGHAoGBAJf2QmHKtQXdKCjhPx1ottPb0PMTBH9A6FbaWMsTuKG/K3g6TG1Z1fkq\n\
2658 /Gz/PWk/eLI9TzFgqVAuPvr3q14a1aZeVUMTgo2oO5/y2UHe6VaJ+trqCTat3xlx\n\
2659 /mNbIK9HA2RgPC3gWfVLZQrY+gz3ASHHR5nXWHEyvpuZm7m3h+irAgEC\n\
2660 -----END DH PARAMETERS-----\n";
2661
2662 static const char dhpem2048[] =
2663 "-----BEGIN DH PARAMETERS-----\n\
2664 MIIBCAKCAQEA7ZKJNYJFVcs7+6J2WmkEYb8h86tT0s0h2v94GRFS8Q7B4lW9aG9o\n\
2665 AFO5Imov5Jo0H2XMWTKKvbHbSe3fpxJmw/0hBHAY8H/W91hRGXKCeyKpNBgdL8sh\n\
2666 z22SrkO2qCnHJ6PLAMXy5fsKpFmFor2tRfCzrfnggTXu2YOzzK7q62bmqVdmufEo\n\
2667 pT8igNcLpvZxk5uBDvhakObMym9mX3rAEBoe8PwttggMYiiw7NuJKO4MqD1llGkW\n\
2668 aVM8U2ATsCun1IKHrRxynkE1/MJ86VHeYYX8GZt2YA8z+GuzylIOKcMH6JAWzMwA\n\
2669 Gbatw6QwizOhr9iMjZ0B26TE3X8LvW84wwIBAg==\n\
2670 -----END DH PARAMETERS-----\n";
2671
2672 static const char dhpem4096[] =
2673 "-----BEGIN DH PARAMETERS-----\n\
2674 MIICCAKCAgEA/urRnb6vkPYc/KEGXWnbCIOaKitq7ySIq9dTH7s+Ri59zs77zty7\n\
2675 vfVlSe6VFTBWgYjD2XKUFmtqq6CqXMhVX5ElUDoYDpAyTH85xqNFLzFC7nKrff/H\n\
2676 TFKNttp22cZE9V0IPpzedPfnQkE7aUdmF9JnDyv21Z/818O93u1B4r0szdnmEvEF\n\
2677 bKuIxEHX+bp0ZR7RqE1AeifXGJX3d6tsd2PMAObxwwsv55RGkn50vHO4QxtTARr1\n\
2678 rRUV5j3B3oPMgC7Offxx+98Xn45B1/G0Prp11anDsR1PGwtaCYipqsvMwQUSJtyE\n\
2679 EOQWk+yFkeMe4vWv367eEi0Sd/wnC+TSXBE3pYvpYerJ8n1MceI5GQTdarJ77OW9\n\
2680 bGTHmxRsLSCM1jpLdPja5jjb4siAa6EHc4qN9c/iFKS3PQPJEnX7pXKBRs5f7AF3\n\
2681 W3RIGt+G9IVNZfXaS7Z/iCpgzgvKCs0VeqN38QsJGtC1aIkwOeyjPNy2G6jJ4yqH\n\
2682 ovXYt/0mc00vCWeSNS1wren0pR2EiLxX0ypjjgsU1mk/Z3b/+zVf7fZSIB+nDLjb\n\
2683 NPtUlJCVGnAeBK1J1nG3TQicqowOXoM6ISkdaXj5GPJdXHab2+S7cqhKGv5qC7rR\n\
2684 jT6sx7RUr0CNTxzLI7muV2/a4tGmj0PSdXQdsZ7tw7gbXlaWT1+MM2MCAQI=\n\
2685 -----END DH PARAMETERS-----\n";
2686
2687 static const struct dhinfo dhpem[] = {
2688         { 512, dhpem512, sizeof(dhpem512) },
2689         { 1024, dhpem1024, sizeof(dhpem1024) },
2690         { 2048, dhpem2048, sizeof(dhpem2048) },
2691         { 4096, dhpem4096, sizeof(dhpem4096) },
2692         { 0, NULL, 0 }
2693 };
2694
2695 static DH *
2696 tls_tmp_dh_cb( SSL *ssl, int is_export, int key_length )
2697 {
2698         struct dhplist *p = NULL;
2699         BIO *b = NULL;
2700         DH *dh = NULL;
2701         int i;
2702
2703         /* Do we have params of this length already? */
2704 #ifdef LDAP_R_COMPILE
2705         ldap_pvt_thread_mutex_lock( &tls_def_ctx_mutex );
2706 #endif
2707         for ( p = dhparams; p; p=p->next ) {
2708                 if ( p->keylength == key_length ) {
2709 #ifdef LDAP_R_COMPILE
2710                         ldap_pvt_thread_mutex_unlock( &tls_def_ctx_mutex );
2711 #endif
2712                         return p->param;
2713                 }
2714         }
2715
2716         /* No - check for hardcoded params */
2717
2718         for (i=0; dhpem[i].keylength; i++) {
2719                 if ( dhpem[i].keylength == key_length ) {
2720                         b = BIO_new_mem_buf( (char *)dhpem[i].pem, dhpem[i].size );
2721                         break;
2722                 }
2723         }
2724
2725         if ( b ) {
2726                 dh = PEM_read_bio_DHparams( b, NULL, NULL, NULL );
2727                 BIO_free( b );
2728         }
2729
2730         /* Generating on the fly is expensive/slow... */
2731         if ( !dh ) {
2732                 dh = DH_generate_parameters( key_length, DH_GENERATOR_2, NULL, NULL );
2733         }
2734         if ( dh ) {
2735                 p = LDAP_MALLOC( sizeof(struct dhplist) );
2736                 if ( p != NULL ) {
2737                         p->keylength = key_length;
2738                         p->param = dh;
2739                         p->next = dhparams;
2740                         dhparams = p;
2741                 }
2742         }
2743
2744 #ifdef LDAP_R_COMPILE
2745         ldap_pvt_thread_mutex_unlock( &tls_def_ctx_mutex );
2746 #endif
2747         return dh;
2748 }
2749 #endif
2750
2751 #endif /* HAVE_OPENSSL */
2752
2753 void *
2754 ldap_pvt_tls_sb_ctx( Sockbuf *sb )
2755 {
2756 #ifdef HAVE_TLS
2757         void                    *p;
2758         
2759         if (HAS_TLS( sb )) {
2760                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&p );
2761                 return p;
2762         }
2763 #endif
2764
2765         return NULL;
2766 }
2767
2768 int
2769 ldap_pvt_tls_get_strength( void *s )
2770 {
2771 #ifdef HAVE_OPENSSL
2772         SSL_CIPHER *c;
2773
2774         c = SSL_get_current_cipher((SSL *)s);
2775         return SSL_CIPHER_get_bits(c, NULL);
2776 #elif defined(HAVE_GNUTLS)
2777         tls_session *session = s;
2778         gnutls_cipher_algorithm_t c;
2779
2780         c = gnutls_cipher_get( session->session );
2781         return gnutls_cipher_get_key_size( c );
2782 #else
2783         return 0;
2784 #endif
2785 }
2786
2787
2788 int
2789 ldap_pvt_tls_get_my_dn( void *s, struct berval *dn, LDAPDN_rewrite_dummy *func, unsigned flags )
2790 {
2791 #ifdef HAVE_TLS
2792         struct berval der_dn;
2793         int rc;
2794 #ifdef HAVE_OPENSSL
2795         X509 *x;
2796         X509_NAME *xn;
2797
2798         x = SSL_get_certificate((SSL *)s);
2799
2800         if (!x) return LDAP_INVALID_CREDENTIALS;
2801         
2802         xn = X509_get_subject_name(x);
2803         der_dn.bv_len = i2d_X509_NAME( xn, NULL );
2804         der_dn.bv_val = xn->bytes->data;
2805 #elif defined(HAVE_GNUTLS)
2806         tls_session *session = s;
2807         const gnutls_datum_t *x;
2808         struct berval bv;
2809
2810         x = gnutls_certificate_get_ours( session->session );
2811
2812         if (!x) return LDAP_INVALID_CREDENTIALS;
2813         
2814         bv.bv_val = x->data;
2815         bv.bv_len = x->size;
2816
2817         x509_cert_get_dn( &bv, &der_dn, 1 );
2818 #endif
2819         rc = ldap_X509dn2bv(&der_dn, dn, (LDAPDN_rewrite_func *)func, flags );
2820         return rc;
2821 #else /* !HAVE_TLS */
2822         return LDAP_NOT_SUPPORTED;
2823 #endif
2824 }
2825
2826 int
2827 ldap_start_tls( LDAP *ld,
2828         LDAPControl **serverctrls,
2829         LDAPControl **clientctrls,
2830         int *msgidp )
2831 {
2832         return ldap_extended_operation( ld, LDAP_EXOP_START_TLS,
2833                 NULL, serverctrls, clientctrls, msgidp );
2834 }
2835
2836 int
2837 ldap_install_tls( LDAP *ld )
2838 {
2839 #ifndef HAVE_TLS
2840         return LDAP_NOT_SUPPORTED;
2841 #else
2842         if ( ldap_tls_inplace( ld ) ) {
2843                 return LDAP_LOCAL_ERROR;
2844         }
2845
2846         return ldap_int_tls_start( ld, ld->ld_defconn, NULL );
2847 #endif
2848 }
2849
2850 int
2851 ldap_start_tls_s ( LDAP *ld,
2852         LDAPControl **serverctrls,
2853         LDAPControl **clientctrls )
2854 {
2855 #ifndef HAVE_TLS
2856         return LDAP_NOT_SUPPORTED;
2857 #else
2858         int rc;
2859         char *rspoid = NULL;
2860         struct berval *rspdata = NULL;
2861
2862         /* XXYYZ: this initiates operation only on default connection! */
2863
2864         if ( ldap_tls_inplace( ld ) ) {
2865                 return LDAP_LOCAL_ERROR;
2866         }
2867
2868         rc = ldap_extended_operation_s( ld, LDAP_EXOP_START_TLS,
2869                 NULL, serverctrls, clientctrls, &rspoid, &rspdata );
2870
2871         if ( rspoid != NULL ) {
2872                 LDAP_FREE(rspoid);
2873         }
2874
2875         if ( rspdata != NULL ) {
2876                 ber_bvfree( rspdata );
2877         }
2878
2879         if ( rc == LDAP_SUCCESS ) {
2880                 rc = ldap_int_tls_start( ld, ld->ld_defconn, NULL );
2881         }
2882
2883         return rc;
2884 #endif
2885 }
2886
2887 /* These tags probably all belong in lber.h, but they're
2888  * not normally encountered when processing LDAP, so maybe
2889  * they belong somewhere else instead.
2890  */
2891
2892 #define LBER_TAG_OID            ((ber_tag_t) 0x06UL)
2893
2894 /* Tags for string types used in a DirectoryString.
2895  *
2896  * Note that IA5string is not one of the defined choices for
2897  * DirectoryString in X.520, but it gets used for email AVAs.
2898  */
2899 #define LBER_TAG_UTF8           ((ber_tag_t) 0x0cUL)
2900 #define LBER_TAG_PRINTABLE      ((ber_tag_t) 0x13UL)
2901 #define LBER_TAG_TELETEX        ((ber_tag_t) 0x14UL)
2902 #define LBER_TAG_IA5            ((ber_tag_t) 0x16UL)
2903 #define LBER_TAG_UNIVERSAL      ((ber_tag_t) 0x1cUL)
2904 #define LBER_TAG_BMP            ((ber_tag_t) 0x1eUL)
2905
2906 static oid_name *
2907 find_oid( struct berval *oid )
2908 {
2909         int i;
2910
2911         for ( i=0; !BER_BVISNULL( &oids[i].oid ); i++ ) {
2912                 if ( oids[i].oid.bv_len != oid->bv_len ) continue;
2913                 if ( !strcmp( oids[i].oid.bv_val, oid->bv_val ))
2914                         return &oids[i];
2915         }
2916         return NULL;
2917 }
2918
2919 /* Convert a structured DN from an X.509 certificate into an LDAPV3 DN.
2920  * x509_name must be raw DER. If func is non-NULL, the
2921  * constructed DN will use numeric OIDs to identify attributeTypes,
2922  * and the func() will be invoked to rewrite the DN with the given
2923  * flags.
2924  *
2925  * Otherwise the DN will use shortNames from a hardcoded table.
2926  */
2927 int
2928 ldap_X509dn2bv( void *x509_name, struct berval *bv, LDAPDN_rewrite_func *func,
2929         unsigned flags )
2930 {
2931         LDAPDN  newDN;
2932         LDAPRDN newRDN;
2933         LDAPAVA *newAVA, *baseAVA;
2934         BerElementBuffer berbuf;
2935         BerElement *ber = (BerElement *)&berbuf;
2936         char oids[8192], *oidptr = oids, *oidbuf = NULL;
2937         void *ptrs[2048];
2938         char *dn_end, *rdn_end;
2939         int i, navas, nrdns, rc = LDAP_SUCCESS;
2940         size_t dnsize, oidrem = sizeof(oids), oidsize = 0;
2941         int csize;
2942         ber_tag_t tag;
2943         ber_len_t len;
2944         oid_name *oidname;
2945
2946         struct berval   Oid, Val, oid2, *in = x509_name;
2947
2948         assert( bv != NULL );
2949
2950         bv->bv_len = 0;
2951         bv->bv_val = NULL;
2952
2953         navas = 0;
2954         nrdns = 0;
2955
2956         /* A DN is a SEQUENCE of RDNs. An RDN is a SET of AVAs.
2957          * An AVA is a SEQUENCE of attr and value.
2958          * Count the number of AVAs and RDNs
2959          */
2960         ber_init2( ber, in, LBER_USE_DER );
2961         tag = ber_peek_tag( ber, &len );
2962         if ( tag != LBER_SEQUENCE )
2963                 return LDAP_DECODING_ERROR;
2964
2965         for ( tag = ber_first_element( ber, &len, &dn_end );
2966                 tag == LBER_SET;
2967                 tag = ber_next_element( ber, &len, dn_end )) {
2968                 nrdns++;
2969                 for ( tag = ber_first_element( ber, &len, &rdn_end );
2970                         tag == LBER_SEQUENCE;
2971                         tag = ber_next_element( ber, &len, rdn_end )) {
2972                         tag = ber_skip_tag( ber, &len );
2973                         ber_skip_data( ber, len );
2974                         navas++;
2975                 }
2976         }
2977
2978         /* Allocate the DN/RDN/AVA stuff as a single block */    
2979         dnsize = sizeof(LDAPRDN) * (nrdns+1);
2980         dnsize += sizeof(LDAPAVA *) * (navas+nrdns);
2981         dnsize += sizeof(LDAPAVA) * navas;
2982         if (dnsize > sizeof(ptrs)) {
2983                 newDN = (LDAPDN)LDAP_MALLOC( dnsize );
2984                 if ( newDN == NULL )
2985                         return LDAP_NO_MEMORY;
2986         } else {
2987                 newDN = (LDAPDN)(char *)ptrs;
2988         }
2989         
2990         newDN[nrdns] = NULL;
2991         newRDN = (LDAPRDN)(newDN + nrdns+1);
2992         newAVA = (LDAPAVA *)(newRDN + navas + nrdns);
2993         baseAVA = newAVA;
2994
2995         /* Rewind and start extracting */
2996         ber_rewind( ber );
2997
2998         tag = ber_first_element( ber, &len, &dn_end );
2999         for ( i = nrdns - 1; i >= 0; i-- ) {
3000                 newDN[i] = newRDN;
3001
3002                 for ( tag = ber_first_element( ber, &len, &rdn_end );
3003                         tag == LBER_SEQUENCE;
3004                         tag = ber_next_element( ber, &len, rdn_end )) {
3005
3006                         *newRDN++ = newAVA;
3007                         tag = ber_skip_tag( ber, &len );
3008                         tag = ber_get_stringbv( ber, &Oid, LBER_BV_NOTERM );
3009                         if ( tag != LBER_TAG_OID ) {
3010                                 rc = LDAP_DECODING_ERROR;
3011                                 goto nomem;
3012                         }
3013
3014                         oid2.bv_val = oidptr;
3015                         oid2.bv_len = oidrem;
3016                         ber_decode_oid( &Oid, &oid2 );
3017                         oidname = find_oid( &oid2 );
3018                         if ( !oidname ) {
3019                                 newAVA->la_attr = oid2;
3020                                 oidptr += oid2.bv_len + 1;
3021                                 oidrem -= oid2.bv_len + 1;
3022
3023                                 /* Running out of OID buffer space? */
3024                                 if (oidrem < 128) {
3025                                         if ( oidsize == 0 ) {
3026                                                 oidsize = sizeof(oids) * 2;
3027                                                 oidrem = oidsize;
3028                                                 oidbuf = LDAP_MALLOC( oidsize );
3029                                                 if ( oidbuf == NULL ) goto nomem;
3030                                                 oidptr = oidbuf;
3031                                         } else {
3032                                                 char *old = oidbuf;
3033                                                 oidbuf = LDAP_REALLOC( oidbuf, oidsize*2 );
3034                                                 if ( oidbuf == NULL ) goto nomem;
3035                                                 /* Buffer moved! Fix AVA pointers */
3036                                                 if ( old != oidbuf ) {
3037                                                         LDAPAVA *a;
3038                                                         long dif = oidbuf - old;
3039
3040                                                         for (a=baseAVA; a<=newAVA; a++){
3041                                                                 if (a->la_attr.bv_val >= old &&
3042                                                                         a->la_attr.bv_val <= (old + oidsize))
3043                                                                         a->la_attr.bv_val += dif;
3044                                                         }
3045                                                 }
3046                                                 oidptr = oidbuf + oidsize - oidrem;
3047                                                 oidrem += oidsize;
3048                                                 oidsize *= 2;
3049                                         }
3050                                 }
3051                         } else {
3052                                 if ( func ) {
3053                                         newAVA->la_attr = oidname->oid;
3054                                 } else {
3055                                         newAVA->la_attr = oidname->name;
3056                                 }
3057                         }
3058                         tag = ber_get_stringbv( ber, &Val, LBER_BV_NOTERM );
3059                         switch(tag) {
3060                         case LBER_TAG_UNIVERSAL:
3061                                 /* This uses 32-bit ISO 10646-1 */
3062                                 csize = 4; goto to_utf8;
3063                         case LBER_TAG_BMP:
3064                                 /* This uses 16-bit ISO 10646-1 */
3065                                 csize = 2; goto to_utf8;
3066                         case LBER_TAG_TELETEX:
3067                                 /* This uses 8-bit, assume ISO 8859-1 */
3068                                 csize = 1;
3069 to_utf8:                rc = ldap_ucs_to_utf8s( &Val, csize, &newAVA->la_value );
3070                                 newAVA->la_flags |= LDAP_AVA_FREE_VALUE;
3071                                 if (rc != LDAP_SUCCESS) goto nomem;
3072                                 newAVA->la_flags = LDAP_AVA_NONPRINTABLE;
3073                                 break;
3074                         case LBER_TAG_UTF8:
3075                                 newAVA->la_flags = LDAP_AVA_NONPRINTABLE;
3076                                 /* This is already in UTF-8 encoding */
3077                         case LBER_TAG_IA5:
3078                         case LBER_TAG_PRINTABLE:
3079                                 /* These are always 7-bit strings */
3080                                 newAVA->la_value = Val;
3081                         default:
3082                                 ;
3083                         }
3084                         newAVA->la_private = NULL;
3085                         newAVA->la_flags = LDAP_AVA_STRING;
3086                         newAVA++;
3087                 }
3088                 *newRDN++ = NULL;
3089                 tag = ber_next_element( ber, &len, dn_end );
3090         }
3091                 
3092         if ( func ) {
3093                 rc = func( newDN, flags, NULL );
3094                 if ( rc != LDAP_SUCCESS )
3095                         goto nomem;
3096         }
3097
3098         rc = ldap_dn2bv_x( newDN, bv, LDAP_DN_FORMAT_LDAPV3, NULL );
3099
3100 nomem:
3101         for (;baseAVA < newAVA; baseAVA++) {
3102                 if (baseAVA->la_flags & LDAP_AVA_FREE_ATTR)
3103                         LDAP_FREE( baseAVA->la_attr.bv_val );
3104                 if (baseAVA->la_flags & LDAP_AVA_FREE_VALUE)
3105                         LDAP_FREE( baseAVA->la_value.bv_val );
3106         }
3107
3108         if ( oidsize != 0 )
3109                 LDAP_FREE( oidbuf );
3110         if ( newDN != (LDAPDN)(char *) ptrs )
3111                 LDAP_FREE( newDN );
3112         return rc;
3113 }
3114