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