git.sur5r.net Git - openldap/blob - libraries/libldap/tls2.c

   1 /* tls.c - Handle tls/ssl. */
   2 /* $OpenLDAP$ */
   3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   4  *
   5  * Copyright 1998-2017 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: restructured by Howard Chu.
  17  */
  18
  19 #include "portable.h"
  20 #include "ldap_config.h"
  21
  22 #include <stdio.h>
  23
  24 #include <ac/stdlib.h>
  25 #include <ac/errno.h>
  26 #include <ac/socket.h>
  27 #include <ac/string.h>
  28 #include <ac/ctype.h>
  29 #include <ac/time.h>
  30 #include <ac/unistd.h>
  31 #include <ac/param.h>
  32 #include <ac/dirent.h>
  33
  34 #include "ldap-int.h"
  35
  36 #ifdef HAVE_TLS
  37
  38 #include "ldap-tls.h"
  39
  40 static tls_impl *tls_imp = &ldap_int_tls_impl;
  41 #define HAS_TLS( sb )   ber_sockbuf_ctrl( sb, LBER_SB_OPT_HAS_IO, \
  42                                 (void *)tls_imp->ti_sbio )
  43
  44 #endif /* HAVE_TLS */
  45
  46 #ifndef HAVE_MOZNSS
  47 #define LDAP_USE_NON_BLOCKING_TLS
  48 #endif
  49
  50 /* RFC2459 minimum required set of supported attribute types
  51  * in a certificate DN
  52  */
  53 typedef struct oid_name {
  54         struct berval oid;
  55         struct berval name;
  56 } oid_name;
  57
  58 static oid_name oids[] = {
  59         { BER_BVC("2.5.4.3"), BER_BVC("cn") },
  60         { BER_BVC("2.5.4.4"), BER_BVC("sn") },
  61         { BER_BVC("2.5.4.6"), BER_BVC("c") },
  62         { BER_BVC("2.5.4.7"), BER_BVC("l") },
  63         { BER_BVC("2.5.4.8"), BER_BVC("st") },
  64         { BER_BVC("2.5.4.10"), BER_BVC("o") },
  65         { BER_BVC("2.5.4.11"), BER_BVC("ou") },
  66         { BER_BVC("2.5.4.12"), BER_BVC("title") },
  67         { BER_BVC("2.5.4.41"), BER_BVC("name") },
  68         { BER_BVC("2.5.4.42"), BER_BVC("givenName") },
  69         { BER_BVC("2.5.4.43"), BER_BVC("initials") },
  70         { BER_BVC("2.5.4.44"), BER_BVC("generationQualifier") },
  71         { BER_BVC("2.5.4.46"), BER_BVC("dnQualifier") },
  72         { BER_BVC("1.2.840.113549.1.9.1"), BER_BVC("email") },
  73         { BER_BVC("0.9.2342.19200300.100.1.25"), BER_BVC("dc") },
  74         { BER_BVNULL, BER_BVNULL }
  75 };
  76
  77 #ifdef HAVE_TLS
  78
  79 void
  80 ldap_pvt_tls_ctx_free ( void *c )
  81 {
  82         if ( !c ) return;
  83         tls_imp->ti_ctx_free( c );
  84 }
  85
  86 static void
  87 tls_ctx_ref( tls_ctx *ctx )
  88 {
  89         if ( !ctx ) return;
  90
  91         tls_imp->ti_ctx_ref( ctx );
  92 }
  93
  94 #ifdef LDAP_R_COMPILE
  95 /*
  96  * an extra mutex for the default ctx.
  97  */
  98 static ldap_pvt_thread_mutex_t tls_def_ctx_mutex;
  99 #endif
 100
 101 void
 102 ldap_int_tls_destroy( struct ldapoptions *lo )
 103 {
 104         if ( lo->ldo_tls_ctx ) {
 105                 ldap_pvt_tls_ctx_free( lo->ldo_tls_ctx );
 106                 lo->ldo_tls_ctx = NULL;
 107         }
 108
 109         if ( lo->ldo_tls_certfile ) {
 110                 LDAP_FREE( lo->ldo_tls_certfile );
 111                 lo->ldo_tls_certfile = NULL;
 112         }
 113         if ( lo->ldo_tls_keyfile ) {
 114                 LDAP_FREE( lo->ldo_tls_keyfile );
 115                 lo->ldo_tls_keyfile = NULL;
 116         }
 117         if ( lo->ldo_tls_dhfile ) {
 118                 LDAP_FREE( lo->ldo_tls_dhfile );
 119                 lo->ldo_tls_dhfile = NULL;
 120         }
 121         if ( lo->ldo_tls_ecname ) {
 122                 LDAP_FREE( lo->ldo_tls_ecname );
 123                 lo->ldo_tls_ecname = NULL;
 124         }
 125         if ( lo->ldo_tls_cacertfile ) {
 126                 LDAP_FREE( lo->ldo_tls_cacertfile );
 127                 lo->ldo_tls_cacertfile = NULL;
 128         }
 129         if ( lo->ldo_tls_cacertdir ) {
 130                 LDAP_FREE( lo->ldo_tls_cacertdir );
 131                 lo->ldo_tls_cacertdir = NULL;
 132         }
 133         if ( lo->ldo_tls_ciphersuite ) {
 134                 LDAP_FREE( lo->ldo_tls_ciphersuite );
 135                 lo->ldo_tls_ciphersuite = NULL;
 136         }
 137         if ( lo->ldo_tls_crlfile ) {
 138                 LDAP_FREE( lo->ldo_tls_crlfile );
 139                 lo->ldo_tls_crlfile = NULL;
 140         }
 141         /* tls_pin_hashalg and tls_pin share the same buffer */
 142         if ( lo->ldo_tls_pin_hashalg ) {
 143                 LDAP_FREE( lo->ldo_tls_pin_hashalg );
 144                 lo->ldo_tls_pin_hashalg = NULL;
 145         } else {
 146                 LDAP_FREE( lo->ldo_tls_pin.bv_val );
 147         }
 148         BER_BVZERO( &lo->ldo_tls_pin );
 149 }
 150
 151 /*
 152  * Tear down the TLS subsystem. Should only be called once.
 153  */
 154 void
 155 ldap_pvt_tls_destroy( void )
 156 {
 157         struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();
 158
 159         ldap_int_tls_destroy( lo );
 160
 161         tls_imp->ti_tls_destroy();
 162 }
 163
 164 /*
 165  * Initialize a particular TLS implementation.
 166  * Called once per implementation.
 167  */
 168 static int
 169 tls_init(tls_impl *impl )
 170 {
 171         static int tls_initialized = 0;
 172
 173         if ( !tls_initialized++ ) {
 174 #ifdef LDAP_R_COMPILE
 175                 ldap_pvt_thread_mutex_init( &tls_def_ctx_mutex );
 176 #endif
 177         }
 178
 179         if ( impl->ti_inited++ ) return 0;
 180
 181 #ifdef LDAP_R_COMPILE
 182         impl->ti_thr_init();
 183 #endif
 184         return impl->ti_tls_init();
 185 }
 186
 187 /*
 188  * Initialize TLS subsystem. Called once per implementation.
 189  */
 190 int
 191 ldap_pvt_tls_init( void )
 192 {
 193         return tls_init( tls_imp );
 194 }
 195
 196 /*
 197  * initialize a new TLS context
 198  */
 199 static int
 200 ldap_int_tls_init_ctx( struct ldapoptions *lo, int is_server )
 201 {
 202         int rc = 0;
 203         tls_impl *ti = tls_imp;
 204         struct ldaptls lts = lo->ldo_tls_info;
 205
 206         if ( lo->ldo_tls_ctx )
 207                 return 0;
 208
 209         tls_init( ti );
 210
 211         if ( is_server && !lts.lt_certfile && !lts.lt_keyfile &&
 212                 !lts.lt_cacertfile && !lts.lt_cacertdir &&
 213                 !lts.lt_cacert.bv_val && !lts.lt_cert.bv_val &&
 214                 !lts.lt_key.bv_val ) {
 215                 /* minimum configuration not provided */
 216                 return LDAP_NOT_SUPPORTED;
 217         }
 218
 219 #ifdef HAVE_EBCDIC
 220         /* This ASCII/EBCDIC handling is a real pain! */
 221         if ( lts.lt_ciphersuite ) {
 222                 lts.lt_ciphersuite = LDAP_STRDUP( lts.lt_ciphersuite );
 223                 __atoe( lts.lt_ciphersuite );
 224         }
 225         if ( lts.lt_cacertfile ) {
 226                 lts.lt_cacertfile = LDAP_STRDUP( lts.lt_cacertfile );
 227                 __atoe( lts.lt_cacertfile );
 228         }
 229         if ( lts.lt_certfile ) {
 230                 lts.lt_certfile = LDAP_STRDUP( lts.lt_certfile );
 231                 __atoe( lts.lt_certfile );
 232         }
 233         if ( lts.lt_keyfile ) {
 234                 lts.lt_keyfile = LDAP_STRDUP( lts.lt_keyfile );
 235                 __atoe( lts.lt_keyfile );
 236         }
 237         if ( lts.lt_crlfile ) {
 238                 lts.lt_crlfile = LDAP_STRDUP( lts.lt_crlfile );
 239                 __atoe( lts.lt_crlfile );
 240         }
 241         if ( lts.lt_cacertdir ) {
 242                 lts.lt_cacertdir = LDAP_STRDUP( lts.lt_cacertdir );
 243                 __atoe( lts.lt_cacertdir );
 244         }
 245         if ( lts.lt_dhfile ) {
 246                 lts.lt_dhfile = LDAP_STRDUP( lts.lt_dhfile );
 247                 __atoe( lts.lt_dhfile );
 248         }
 249         if ( lts.lt_ecname ) {
 250                 lts.lt_ecname = LDAP_STRDUP( lts.lt_ecname );
 251                 __atoe( lts.lt_ecname );
 252         }
 253 #endif
 254         lo->ldo_tls_ctx = ti->ti_ctx_new( lo );
 255         if ( lo->ldo_tls_ctx == NULL ) {
 256                 Debug( LDAP_DEBUG_ANY,
 257                    "TLS: could not allocate default ctx.\n",
 258                         0,0,0);
 259                 rc = -1;
 260                 goto error_exit;
 261         }
 262
 263         rc = ti->ti_ctx_init( lo, &lts, is_server );
 264
 265 error_exit:
 266         if ( rc < 0 && lo->ldo_tls_ctx != NULL ) {
 267                 ldap_pvt_tls_ctx_free( lo->ldo_tls_ctx );
 268                 lo->ldo_tls_ctx = NULL;
 269         }
 270 #ifdef HAVE_EBCDIC
 271         LDAP_FREE( lts.lt_ciphersuite );
 272         LDAP_FREE( lts.lt_cacertfile );
 273         LDAP_FREE( lts.lt_certfile );
 274         LDAP_FREE( lts.lt_keyfile );
 275         LDAP_FREE( lts.lt_crlfile );
 276         LDAP_FREE( lts.lt_cacertdir );
 277         LDAP_FREE( lts.lt_dhfile );
 278         LDAP_FREE( lts.lt_ecname );
 279 #endif
 280         return rc;
 281 }
 282
 283 /*
 284  * initialize the default context
 285  */
 286 int
 287 ldap_pvt_tls_init_def_ctx( int is_server )
 288 {
 289         struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();
 290         int rc;
 291         LDAP_MUTEX_LOCK( &tls_def_ctx_mutex );
 292         rc = ldap_int_tls_init_ctx( lo, is_server );
 293         LDAP_MUTEX_UNLOCK( &tls_def_ctx_mutex );
 294         return rc;
 295 }
 296
 297 static tls_session *
 298 alloc_handle( void *ctx_arg, int is_server )
 299 {
 300         tls_ctx *ctx;
 301         tls_session     *ssl;
 302
 303         if ( ctx_arg ) {
 304                 ctx = ctx_arg;
 305         } else {
 306                 struct ldapoptions *lo = LDAP_INT_GLOBAL_OPT();
 307                 if ( ldap_pvt_tls_init_def_ctx( is_server ) < 0 ) return NULL;
 308                 ctx = lo->ldo_tls_ctx;
 309         }
 310
 311         ssl = tls_imp->ti_session_new( ctx, is_server );
 312         if ( ssl == NULL ) {
 313                 Debug( LDAP_DEBUG_ANY,"TLS: can't create ssl handle.\n",0,0,0);
 314                 return NULL;
 315         }
 316         return ssl;
 317 }
 318
 319 static int
 320 update_flags( Sockbuf *sb, tls_session * ssl, int rc )
 321 {
 322         sb->sb_trans_needs_read  = 0;
 323         sb->sb_trans_needs_write = 0;
 324
 325         return tls_imp->ti_session_upflags( sb, ssl, rc );
 326 }
 327
 328 /*
 329  * Call this to do a TLS connect on a sockbuf. ctx_arg can be
 330  * a SSL_CTX * or NULL, in which case the default ctx is used.
 331  *
 332  * Return value:
 333  *
 334  *  0 - Success. Connection is ready for communication.
 335  * <0 - Error. Can't create a TLS stream.
 336  * >0 - Partial success.
 337  *        Do a select (using information from lber_pvt_sb_needs_{read,write}
 338  *              and call again.
 339  */
 340
 341 static int
 342 ldap_int_tls_connect( LDAP *ld, LDAPConn *conn, const char *host )
 343 {
 344         Sockbuf *sb = conn->lconn_sb;
 345         int     err;
 346         tls_session     *ssl = NULL;
 347
 348         if ( HAS_TLS( sb )) {
 349                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&ssl );
 350         } else {
 351                 struct ldapoptions *lo;
 352                 tls_ctx *ctx;
 353
 354                 ctx = ld->ld_options.ldo_tls_ctx;
 355
 356                 ssl = alloc_handle( ctx, 0 );
 357
 358                 if ( ssl == NULL ) return -1;
 359
 360 #ifdef LDAP_DEBUG
 361                 ber_sockbuf_add_io( sb, &ber_sockbuf_io_debug,
 362                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)"tls_" );
 363 #endif
 364                 ber_sockbuf_add_io( sb, tls_imp->ti_sbio,
 365                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)ssl );
 366
 367                 lo = LDAP_INT_GLOBAL_OPT();
 368                 if( ctx == NULL ) {
 369                         ctx = lo->ldo_tls_ctx;
 370                         ld->ld_options.ldo_tls_ctx = ctx;
 371                         tls_ctx_ref( ctx );
 372                 }
 373                 if ( ld->ld_options.ldo_tls_connect_cb )
 374                         ld->ld_options.ldo_tls_connect_cb( ld, ssl, ctx,
 375                         ld->ld_options.ldo_tls_connect_arg );
 376                 if ( lo && lo->ldo_tls_connect_cb && lo->ldo_tls_connect_cb !=
 377                         ld->ld_options.ldo_tls_connect_cb )
 378                         lo->ldo_tls_connect_cb( ld, ssl, ctx, lo->ldo_tls_connect_arg );
 379         }
 380
 381         err = tls_imp->ti_session_connect( ld, ssl );
 382
 383 #ifdef HAVE_WINSOCK
 384         errno = WSAGetLastError();
 385 #endif
 386
 387         if ( err == 0 ) {
 388                 err = ldap_pvt_tls_check_hostname( ld, ssl, host );
 389         }
 390
 391         if ( err < 0 )
 392         {
 393                 char buf[256], *msg;
 394                 if ( update_flags( sb, ssl, err )) {
 395                         return 1;
 396                 }
 397
 398                 msg = tls_imp->ti_session_errmsg( ssl, err, buf, sizeof(buf) );
 399                 if ( msg ) {
 400                         if ( ld->ld_error ) {
 401                                 LDAP_FREE( ld->ld_error );
 402                         }
 403                         ld->ld_error = LDAP_STRDUP( msg );
 404 #ifdef HAVE_EBCDIC
 405                         if ( ld->ld_error ) __etoa(ld->ld_error);
 406 #endif
 407                 }
 408
 409                 Debug( LDAP_DEBUG_ANY,"TLS: can't connect: %s.\n",
 410                         ld->ld_error ? ld->ld_error : "" ,0,0);
 411
 412                 ber_sockbuf_remove_io( sb, tls_imp->ti_sbio,
 413                         LBER_SBIOD_LEVEL_TRANSPORT );
 414 #ifdef LDAP_DEBUG
 415                 ber_sockbuf_remove_io( sb, &ber_sockbuf_io_debug,
 416                         LBER_SBIOD_LEVEL_TRANSPORT );
 417 #endif
 418                 return -1;
 419         }
 420
 421         return 0;
 422 }
 423
 424 /*
 425  * Call this to do a TLS accept on a sockbuf.
 426  * Everything else is the same as with tls_connect.
 427  */
 428 int
 429 ldap_pvt_tls_accept( Sockbuf *sb, void *ctx_arg )
 430 {
 431         int     err;
 432         tls_session     *ssl = NULL;
 433
 434         if ( HAS_TLS( sb )) {
 435                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&ssl );
 436         } else {
 437                 ssl = alloc_handle( ctx_arg, 1 );
 438                 if ( ssl == NULL ) return -1;
 439
 440 #ifdef LDAP_DEBUG
 441                 ber_sockbuf_add_io( sb, &ber_sockbuf_io_debug,
 442                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)"tls_" );
 443 #endif
 444                 ber_sockbuf_add_io( sb, tls_imp->ti_sbio,
 445                         LBER_SBIOD_LEVEL_TRANSPORT, (void *)ssl );
 446         }
 447
 448         err = tls_imp->ti_session_accept( ssl );
 449
 450 #ifdef HAVE_WINSOCK
 451         errno = WSAGetLastError();
 452 #endif
 453
 454         if ( err < 0 )
 455         {
 456                 if ( update_flags( sb, ssl, err )) return 1;
 457
 458                 if ( DebugTest( LDAP_DEBUG_ANY ) ) {
 459                         char buf[256], *msg;
 460                         msg = tls_imp->ti_session_errmsg( ssl, err, buf, sizeof(buf) );
 461                         Debug( LDAP_DEBUG_ANY,"TLS: can't accept: %s.\n",
 462                                 msg ? msg : "(unknown)", 0, 0 );
 463                 }
 464
 465                 ber_sockbuf_remove_io( sb, tls_imp->ti_sbio,
 466                         LBER_SBIOD_LEVEL_TRANSPORT );
 467 #ifdef LDAP_DEBUG
 468                 ber_sockbuf_remove_io( sb, &ber_sockbuf_io_debug,
 469                         LBER_SBIOD_LEVEL_TRANSPORT );
 470 #endif
 471                 return -1;
 472         }
 473         return 0;
 474 }
 475
 476 int
 477 ldap_pvt_tls_inplace ( Sockbuf *sb )
 478 {
 479         return HAS_TLS( sb ) ? 1 : 0;
 480 }
 481
 482 int
 483 ldap_tls_inplace( LDAP *ld )
 484 {
 485         Sockbuf         *sb = NULL;
 486
 487         if ( ld->ld_defconn && ld->ld_defconn->lconn_sb ) {
 488                 sb = ld->ld_defconn->lconn_sb;
 489
 490         } else if ( ld->ld_sb ) {
 491                 sb = ld->ld_sb;
 492
 493         } else {
 494                 return 0;
 495         }
 496
 497         return ldap_pvt_tls_inplace( sb );
 498 }
 499
 500 int
 501 ldap_pvt_tls_get_peer_dn( void *s, struct berval *dn,
 502         LDAPDN_rewrite_dummy *func, unsigned flags )
 503 {
 504         tls_session *session = s;
 505         struct berval bvdn;
 506         int rc;
 507
 508         rc = tls_imp->ti_session_peer_dn( session, &bvdn );
 509         if ( rc ) return rc;
 510
 511         rc = ldap_X509dn2bv( &bvdn, dn,
 512                             (LDAPDN_rewrite_func *)func, flags);
 513         return rc;
 514 }
 515
 516 int
 517 ldap_pvt_tls_check_hostname( LDAP *ld, void *s, const char *name_in )
 518 {
 519         tls_session *session = s;
 520
 521         if (ld->ld_options.ldo_tls_require_cert != LDAP_OPT_X_TLS_NEVER &&
 522             ld->ld_options.ldo_tls_require_cert != LDAP_OPT_X_TLS_ALLOW) {
 523                 ld->ld_errno = tls_imp->ti_session_chkhost( ld, session, name_in );
 524                 if (ld->ld_errno != LDAP_SUCCESS) {
 525                         return ld->ld_errno;
 526                 }
 527         }
 528
 529         /*
 530          * If instructed to do pinning, do it now
 531          */
 532         if ( !BER_BVISNULL( &ld->ld_options.ldo_tls_pin ) ) {
 533                 ld->ld_errno = tls_imp->ti_session_pinning( ld, s,
 534                                 ld->ld_options.ldo_tls_pin_hashalg,
 535                                 &ld->ld_options.ldo_tls_pin );
 536                 if (ld->ld_errno != LDAP_SUCCESS) {
 537                         return ld->ld_errno;
 538                 }
 539         }
 540
 541         return LDAP_SUCCESS;
 542 }
 543
 544 int
 545 ldap_pvt_tls_config( LDAP *ld, int option, const char *arg )
 546 {
 547         int i;
 548
 549         switch( option ) {
 550         case LDAP_OPT_X_TLS_CACERTFILE:
 551         case LDAP_OPT_X_TLS_CACERTDIR:
 552         case LDAP_OPT_X_TLS_CERTFILE:
 553         case LDAP_OPT_X_TLS_KEYFILE:
 554         case LDAP_OPT_X_TLS_RANDOM_FILE:
 555         case LDAP_OPT_X_TLS_CIPHER_SUITE:
 556         case LDAP_OPT_X_TLS_DHFILE:
 557         case LDAP_OPT_X_TLS_PEERKEY_HASH:
 558         case LDAP_OPT_X_TLS_CRLFILE:    /* GnuTLS only */
 559                 return ldap_pvt_tls_set_option( ld, option, (void *) arg );
 560
 561         case LDAP_OPT_X_TLS_REQUIRE_CERT:
 562         case LDAP_OPT_X_TLS:
 563                 i = -1;
 564                 if ( strcasecmp( arg, "never" ) == 0 ) {
 565                         i = LDAP_OPT_X_TLS_NEVER ;
 566
 567                 } else if ( strcasecmp( arg, "demand" ) == 0 ) {
 568                         i = LDAP_OPT_X_TLS_DEMAND ;
 569
 570                 } else if ( strcasecmp( arg, "allow" ) == 0 ) {
 571                         i = LDAP_OPT_X_TLS_ALLOW ;
 572
 573                 } else if ( strcasecmp( arg, "try" ) == 0 ) {
 574                         i = LDAP_OPT_X_TLS_TRY ;
 575
 576                 } else if ( ( strcasecmp( arg, "hard" ) == 0 ) ||
 577                         ( strcasecmp( arg, "on" ) == 0 ) ||
 578                         ( strcasecmp( arg, "yes" ) == 0) ||
 579                         ( strcasecmp( arg, "true" ) == 0 ) )
 580                 {
 581                         i = LDAP_OPT_X_TLS_HARD ;
 582                 }
 583
 584                 if (i >= 0) {
 585                         return ldap_pvt_tls_set_option( ld, option, &i );
 586                 }
 587                 return -1;
 588         case LDAP_OPT_X_TLS_PROTOCOL_MIN: {
 589                 char *next;
 590                 long l;
 591                 l = strtol( arg, &next, 10 );
 592                 if ( l < 0 || l > 0xff || next == arg ||
 593                         ( *next != '\0' && *next != '.' ) )
 594                         return -1;
 595                 i = l << 8;
 596                 if (*next == '.') {
 597                         arg = next + 1;
 598                         l = strtol( arg, &next, 10 );
 599                         if ( l < 0 || l > 0xff || next == arg || *next != '\0' )
 600                                 return -1;
 601                         i += l;
 602                 }
 603                 return ldap_pvt_tls_set_option( ld, option, &i );
 604                 }
 605 #ifdef HAVE_OPENSSL_CRL
 606         case LDAP_OPT_X_TLS_CRLCHECK:   /* OpenSSL only */
 607                 i = -1;
 608                 if ( strcasecmp( arg, "none" ) == 0 ) {
 609                         i = LDAP_OPT_X_TLS_CRL_NONE ;
 610                 } else if ( strcasecmp( arg, "peer" ) == 0 ) {
 611                         i = LDAP_OPT_X_TLS_CRL_PEER ;
 612                 } else if ( strcasecmp( arg, "all" ) == 0 ) {
 613                         i = LDAP_OPT_X_TLS_CRL_ALL ;
 614                 }
 615                 if (i >= 0) {
 616                         return ldap_pvt_tls_set_option( ld, option, &i );
 617                 }
 618                 return -1;
 619 #endif
 620         }
 621         return -1;
 622 }
 623
 624 int
 625 ldap_pvt_tls_get_option( LDAP *ld, int option, void *arg )
 626 {
 627         struct ldapoptions *lo;
 628
 629         if( option == LDAP_OPT_X_TLS_PACKAGE ) {
 630                 *(char **)arg = LDAP_STRDUP( tls_imp->ti_name );
 631                 return 0;
 632         }
 633
 634         if( ld != NULL ) {
 635                 assert( LDAP_VALID( ld ) );
 636
 637                 if( !LDAP_VALID( ld ) ) {
 638                         return LDAP_OPT_ERROR;
 639                 }
 640
 641                 lo = &ld->ld_options;
 642
 643         } else {
 644                 /* Get pointer to global option structure */
 645                 lo = LDAP_INT_GLOBAL_OPT();
 646                 if ( lo == NULL ) {
 647                         return LDAP_NO_MEMORY;
 648                 }
 649         }
 650
 651         switch( option ) {
 652         case LDAP_OPT_X_TLS:
 653                 *(int *)arg = lo->ldo_tls_mode;
 654                 break;
 655         case LDAP_OPT_X_TLS_CTX:
 656                 *(void **)arg = lo->ldo_tls_ctx;
 657                 if ( lo->ldo_tls_ctx ) {
 658                         tls_ctx_ref( lo->ldo_tls_ctx );
 659                 }
 660                 break;
 661         case LDAP_OPT_X_TLS_CACERTFILE:
 662                 *(char **)arg = lo->ldo_tls_cacertfile ?
 663                         LDAP_STRDUP( lo->ldo_tls_cacertfile ) : NULL;
 664                 break;
 665         case LDAP_OPT_X_TLS_CACERTDIR:
 666                 *(char **)arg = lo->ldo_tls_cacertdir ?
 667                         LDAP_STRDUP( lo->ldo_tls_cacertdir ) : NULL;
 668                 break;
 669         case LDAP_OPT_X_TLS_CERTFILE:
 670                 *(char **)arg = lo->ldo_tls_certfile ?
 671                         LDAP_STRDUP( lo->ldo_tls_certfile ) : NULL;
 672                 break;
 673         case LDAP_OPT_X_TLS_KEYFILE:
 674                 *(char **)arg = lo->ldo_tls_keyfile ?
 675                         LDAP_STRDUP( lo->ldo_tls_keyfile ) : NULL;
 676                 break;
 677         case LDAP_OPT_X_TLS_DHFILE:
 678                 *(char **)arg = lo->ldo_tls_dhfile ?
 679                         LDAP_STRDUP( lo->ldo_tls_dhfile ) : NULL;
 680                 break;
 681         case LDAP_OPT_X_TLS_ECNAME:
 682                 *(char **)arg = lo->ldo_tls_ecname ?
 683                         LDAP_STRDUP( lo->ldo_tls_ecname ) : NULL;
 684                 break;
 685         case LDAP_OPT_X_TLS_CRLFILE:    /* GnuTLS only */
 686                 *(char **)arg = lo->ldo_tls_crlfile ?
 687                         LDAP_STRDUP( lo->ldo_tls_crlfile ) : NULL;
 688                 break;
 689         case LDAP_OPT_X_TLS_REQUIRE_CERT:
 690                 *(int *)arg = lo->ldo_tls_require_cert;
 691                 break;
 692 #ifdef HAVE_OPENSSL_CRL
 693         case LDAP_OPT_X_TLS_CRLCHECK:   /* OpenSSL only */
 694                 *(int *)arg = lo->ldo_tls_crlcheck;
 695                 break;
 696 #endif
 697         case LDAP_OPT_X_TLS_CIPHER_SUITE:
 698                 *(char **)arg = lo->ldo_tls_ciphersuite ?
 699                         LDAP_STRDUP( lo->ldo_tls_ciphersuite ) : NULL;
 700                 break;
 701         case LDAP_OPT_X_TLS_PROTOCOL_MIN:
 702                 *(int *)arg = lo->ldo_tls_protocol_min;
 703                 break;
 704         case LDAP_OPT_X_TLS_RANDOM_FILE:
 705                 *(char **)arg = lo->ldo_tls_randfile ?
 706                         LDAP_STRDUP( lo->ldo_tls_randfile ) : NULL;
 707                 break;
 708         case LDAP_OPT_X_TLS_SSL_CTX: {
 709                 void *retval = 0;
 710                 if ( ld != NULL ) {
 711                         LDAPConn *conn = ld->ld_defconn;
 712                         if ( conn != NULL ) {
 713                                 Sockbuf *sb = conn->lconn_sb;
 714                                 retval = ldap_pvt_tls_sb_ctx( sb );
 715                         }
 716                 }
 717                 *(void **)arg = retval;
 718                 break;
 719         }
 720         case LDAP_OPT_X_TLS_CONNECT_CB:
 721                 *(LDAP_TLS_CONNECT_CB **)arg = lo->ldo_tls_connect_cb;
 722                 break;
 723         case LDAP_OPT_X_TLS_CONNECT_ARG:
 724                 *(void **)arg = lo->ldo_tls_connect_arg;
 725                 break;
 726         case LDAP_OPT_X_TLS_VERSION: {
 727                 void *sess = NULL;
 728                 const char *retval = NULL;
 729                 if ( ld != NULL ) {
 730                         LDAPConn *conn = ld->ld_defconn;
 731                         if ( conn != NULL ) {
 732                                 Sockbuf *sb = conn->lconn_sb;
 733                                 sess = ldap_pvt_tls_sb_ctx( sb );
 734                                 if ( sess != NULL )
 735                                         retval = ldap_pvt_tls_get_version( sess );
 736                         }
 737                 }
 738                 *(char **)arg = retval ? LDAP_STRDUP( retval ) : NULL;
 739                 break;
 740         }
 741         case LDAP_OPT_X_TLS_CIPHER: {
 742                 void *sess = NULL;
 743                 const char *retval = NULL;
 744                 if ( ld != NULL ) {
 745                         LDAPConn *conn = ld->ld_defconn;
 746                         if ( conn != NULL ) {
 747                                 Sockbuf *sb = conn->lconn_sb;
 748                                 sess = ldap_pvt_tls_sb_ctx( sb );
 749                                 if ( sess != NULL )
 750                                         retval = ldap_pvt_tls_get_cipher( sess );
 751                         }
 752                 }
 753                 *(char **)arg = retval ? LDAP_STRDUP( retval ) : NULL;
 754                 break;
 755         }
 756         case LDAP_OPT_X_TLS_PEERCERT: {
 757                 void *sess = NULL;
 758                 struct berval *bv = arg;
 759                 bv->bv_len = 0;
 760                 bv->bv_val = NULL;
 761                 if ( ld != NULL ) {
 762                         LDAPConn *conn = ld->ld_defconn;
 763                         if ( conn != NULL ) {
 764                                 Sockbuf *sb = conn->lconn_sb;
 765                                 sess = ldap_pvt_tls_sb_ctx( sb );
 766                                 if ( sess != NULL )
 767                                         return ldap_pvt_tls_get_peercert( sess, bv );
 768                         }
 769                 }
 770                 break;
 771         }
 772         case LDAP_OPT_X_TLS_CACERT: {
 773                 struct berval *bv = arg;
 774                 if ( lo->ldo_tls_cacert.bv_val ) {
 775                         ber_dupbv( bv, &lo->ldo_tls_cacert );
 776                 } else {
 777                         BER_BVZERO( bv );
 778                 }
 779                 break;
 780         }
 781         case LDAP_OPT_X_TLS_CERT: {
 782                 struct berval *bv = arg;
 783                 if ( lo->ldo_tls_cert.bv_val ) {
 784                         ber_dupbv( bv, &lo->ldo_tls_cert );
 785                 } else {
 786                         BER_BVZERO( bv );
 787                 }
 788                 break;
 789         }
 790         case LDAP_OPT_X_TLS_KEY: {
 791                 struct berval *bv = arg;
 792                 if ( lo->ldo_tls_key.bv_val ) {
 793                         ber_dupbv( bv, &lo->ldo_tls_key );
 794                 } else {
 795                         BER_BVZERO( bv );
 796                 }
 797                 break;
 798         }
 799
 800         default:
 801                 return -1;
 802         }
 803         return 0;
 804 }
 805
 806 int
 807 ldap_pvt_tls_set_option( LDAP *ld, int option, void *arg )
 808 {
 809         struct ldapoptions *lo;
 810
 811         if( ld != NULL ) {
 812                 assert( LDAP_VALID( ld ) );
 813
 814                 if( !LDAP_VALID( ld ) ) {
 815                         return LDAP_OPT_ERROR;
 816                 }
 817
 818                 lo = &ld->ld_options;
 819
 820         } else {
 821                 /* Get pointer to global option structure */
 822                 lo = LDAP_INT_GLOBAL_OPT();
 823                 if ( lo == NULL ) {
 824                         return LDAP_NO_MEMORY;
 825                 }
 826         }
 827
 828         switch( option ) {
 829         case LDAP_OPT_X_TLS:
 830                 if ( !arg ) return -1;
 831
 832                 switch( *(int *) arg ) {
 833                 case LDAP_OPT_X_TLS_NEVER:
 834                 case LDAP_OPT_X_TLS_DEMAND:
 835                 case LDAP_OPT_X_TLS_ALLOW:
 836                 case LDAP_OPT_X_TLS_TRY:
 837                 case LDAP_OPT_X_TLS_HARD:
 838                         if (lo != NULL) {
 839                                 lo->ldo_tls_mode = *(int *)arg;
 840                         }
 841
 842                         return 0;
 843                 }
 844                 return -1;
 845
 846         case LDAP_OPT_X_TLS_CTX:
 847                 if ( lo->ldo_tls_ctx )
 848                         ldap_pvt_tls_ctx_free( lo->ldo_tls_ctx );
 849                 lo->ldo_tls_ctx = arg;
 850                 tls_ctx_ref( lo->ldo_tls_ctx );
 851                 return 0;
 852         case LDAP_OPT_X_TLS_CONNECT_CB:
 853                 lo->ldo_tls_connect_cb = (LDAP_TLS_CONNECT_CB *)arg;
 854                 return 0;
 855         case LDAP_OPT_X_TLS_CONNECT_ARG:
 856                 lo->ldo_tls_connect_arg = arg;
 857                 return 0;
 858         case LDAP_OPT_X_TLS_CACERTFILE:
 859                 if ( lo->ldo_tls_cacertfile ) LDAP_FREE( lo->ldo_tls_cacertfile );
 860                 lo->ldo_tls_cacertfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 861                 return 0;
 862         case LDAP_OPT_X_TLS_CACERTDIR:
 863                 if ( lo->ldo_tls_cacertdir ) LDAP_FREE( lo->ldo_tls_cacertdir );
 864                 lo->ldo_tls_cacertdir = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 865                 return 0;
 866         case LDAP_OPT_X_TLS_CERTFILE:
 867                 if ( lo->ldo_tls_certfile ) LDAP_FREE( lo->ldo_tls_certfile );
 868                 lo->ldo_tls_certfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 869                 return 0;
 870         case LDAP_OPT_X_TLS_KEYFILE:
 871                 if ( lo->ldo_tls_keyfile ) LDAP_FREE( lo->ldo_tls_keyfile );
 872                 lo->ldo_tls_keyfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 873                 return 0;
 874         case LDAP_OPT_X_TLS_DHFILE:
 875                 if ( lo->ldo_tls_dhfile ) LDAP_FREE( lo->ldo_tls_dhfile );
 876                 lo->ldo_tls_dhfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 877                 return 0;
 878         case LDAP_OPT_X_TLS_ECNAME:
 879                 if ( lo->ldo_tls_ecname ) LDAP_FREE( lo->ldo_tls_ecname );
 880                 lo->ldo_tls_ecname = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 881                 return 0;
 882         case LDAP_OPT_X_TLS_CRLFILE:    /* GnuTLS only */
 883                 if ( lo->ldo_tls_crlfile ) LDAP_FREE( lo->ldo_tls_crlfile );
 884                 lo->ldo_tls_crlfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 885                 return 0;
 886         case LDAP_OPT_X_TLS_REQUIRE_CERT:
 887                 if ( !arg ) return -1;
 888                 switch( *(int *) arg ) {
 889                 case LDAP_OPT_X_TLS_NEVER:
 890                 case LDAP_OPT_X_TLS_DEMAND:
 891                 case LDAP_OPT_X_TLS_ALLOW:
 892                 case LDAP_OPT_X_TLS_TRY:
 893                 case LDAP_OPT_X_TLS_HARD:
 894                         lo->ldo_tls_require_cert = * (int *) arg;
 895                         return 0;
 896                 }
 897                 return -1;
 898 #ifdef HAVE_OPENSSL_CRL
 899         case LDAP_OPT_X_TLS_CRLCHECK:   /* OpenSSL only */
 900                 if ( !arg ) return -1;
 901                 switch( *(int *) arg ) {
 902                 case LDAP_OPT_X_TLS_CRL_NONE:
 903                 case LDAP_OPT_X_TLS_CRL_PEER:
 904                 case LDAP_OPT_X_TLS_CRL_ALL:
 905                         lo->ldo_tls_crlcheck = * (int *) arg;
 906                         return 0;
 907                 }
 908                 return -1;
 909 #endif
 910         case LDAP_OPT_X_TLS_CIPHER_SUITE:
 911                 if ( lo->ldo_tls_ciphersuite ) LDAP_FREE( lo->ldo_tls_ciphersuite );
 912                 lo->ldo_tls_ciphersuite = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 913                 return 0;
 914
 915         case LDAP_OPT_X_TLS_PROTOCOL_MIN:
 916                 if ( !arg ) return -1;
 917                 lo->ldo_tls_protocol_min = *(int *)arg;
 918                 return 0;
 919         case LDAP_OPT_X_TLS_RANDOM_FILE:
 920                 if ( ld != NULL )
 921                         return -1;
 922                 if ( lo->ldo_tls_randfile ) LDAP_FREE (lo->ldo_tls_randfile );
 923                 lo->ldo_tls_randfile = arg ? LDAP_STRDUP( (char *) arg ) : NULL;
 924                 break;
 925         case LDAP_OPT_X_TLS_NEWCTX:
 926                 if ( !arg ) return -1;
 927                 if ( lo->ldo_tls_ctx )
 928                         ldap_pvt_tls_ctx_free( lo->ldo_tls_ctx );
 929                 lo->ldo_tls_ctx = NULL;
 930                 return ldap_int_tls_init_ctx( lo, *(int *)arg );
 931         case LDAP_OPT_X_TLS_CACERT:
 932                 if ( lo->ldo_tls_cacert.bv_val )
 933                         LDAP_FREE( lo->ldo_tls_cacert.bv_val );
 934                 if ( arg ) {
 935                         lo->ldo_tls_cacert.bv_len = ((struct berval *)arg)->bv_len;
 936                         lo->ldo_tls_cacert.bv_val = LDAP_MALLOC( lo->ldo_tls_cacert.bv_len );
 937                         if ( !lo->ldo_tls_cacert.bv_val )
 938                                 return -1;
 939                         AC_MEMCPY( lo->ldo_tls_cacert.bv_val, ((struct berval *)arg)->bv_val, lo->ldo_tls_cacert.bv_len );
 940                 } else {
 941                         BER_BVZERO( &lo->ldo_tls_cacert );
 942                 }
 943                 break;
 944         case LDAP_OPT_X_TLS_CERT:
 945                 if ( lo->ldo_tls_cert.bv_val )
 946                         LDAP_FREE( lo->ldo_tls_cert.bv_val );
 947                 if ( arg ) {
 948                         lo->ldo_tls_cert.bv_len = ((struct berval *)arg)->bv_len;
 949                         lo->ldo_tls_cert.bv_val = LDAP_MALLOC( lo->ldo_tls_cert.bv_len );
 950                         if ( !lo->ldo_tls_cert.bv_val )
 951                                 return -1;
 952                         AC_MEMCPY( lo->ldo_tls_cert.bv_val, ((struct berval *)arg)->bv_val, lo->ldo_tls_cert.bv_len );
 953                 } else {
 954                         BER_BVZERO( &lo->ldo_tls_cert );
 955                 }
 956                 break;
 957         case LDAP_OPT_X_TLS_KEY:
 958                 if ( lo->ldo_tls_key.bv_val )
 959                         LDAP_FREE( lo->ldo_tls_key.bv_val );
 960                 if ( arg ) {
 961                         lo->ldo_tls_key.bv_len = ((struct berval *)arg)->bv_len;
 962                         lo->ldo_tls_key.bv_val = LDAP_MALLOC( lo->ldo_tls_key.bv_len );
 963                         if ( !lo->ldo_tls_key.bv_val )
 964                                 return -1;
 965                         AC_MEMCPY( lo->ldo_tls_key.bv_val, ((struct berval *)arg)->bv_val, lo->ldo_tls_key.bv_len );
 966                 } else {
 967                         BER_BVZERO( &lo->ldo_tls_key );
 968                 }
 969                 break;
 970         case LDAP_OPT_X_TLS_PEERKEY_HASH: {
 971                 /* arg = "[hashalg:]pubkey_hash" */
 972                 struct berval bv;
 973                 char *p, *pin = arg;
 974                 int rc = LDAP_SUCCESS;
 975
 976                 if ( !tls_imp->ti_session_pinning ) return -1;
 977
 978                 if ( !pin ) {
 979                         if ( lo->ldo_tls_pin_hashalg ) {
 980                                 LDAP_FREE( lo->ldo_tls_pin_hashalg );
 981                         } else if ( lo->ldo_tls_pin.bv_val ) {
 982                                 LDAP_FREE( lo->ldo_tls_pin.bv_val );
 983                         }
 984                         lo->ldo_tls_pin_hashalg = NULL;
 985                         BER_BVZERO( &lo->ldo_tls_pin );
 986                         return rc;
 987                 }
 988
 989                 pin = LDAP_STRDUP( pin );
 990                 p = strchr( pin, ':' );
 991
 992                 /* pubkey (its hash) goes in bv, alg in p */
 993                 if ( p ) {
 994                         *p = '\0';
 995                         bv.bv_val = p+1;
 996                         p = pin;
 997                 } else {
 998                         bv.bv_val = pin;
 999                 }
1000
1001                 bv.bv_len = strlen(bv.bv_val);
1002                 if ( ldap_int_decode_b64_inplace( &bv ) ) {
1003                         LDAP_FREE( pin );
1004                         return -1;
1005                 }
1006
1007                 if ( ld != NULL ) {
1008                         LDAPConn *conn = ld->ld_defconn;
1009                         if ( conn != NULL ) {
1010                                 Sockbuf *sb = conn->lconn_sb;
1011                                 void *sess = ldap_pvt_tls_sb_ctx( sb );
1012                                 if ( sess != NULL ) {
1013                                         rc = tls_imp->ti_session_pinning( ld, sess, p, &bv );
1014                                 }
1015                         }
1016                 }
1017
1018                 if ( rc == LDAP_SUCCESS ) {
1019                         if ( lo->ldo_tls_pin_hashalg ) {
1020                                 LDAP_FREE( lo->ldo_tls_pin_hashalg );
1021                         } else if ( lo->ldo_tls_pin.bv_val ) {
1022                                 LDAP_FREE( lo->ldo_tls_pin.bv_val );
1023                         }
1024                         lo->ldo_tls_pin_hashalg = p;
1025                         lo->ldo_tls_pin = bv;
1026                 } else {
1027                         LDAP_FREE( pin );
1028                 }
1029
1030                 return rc;
1031         }
1032         default:
1033                 return -1;
1034         }
1035         return 0;
1036 }
1037
1038 int
1039 ldap_int_tls_start ( LDAP *ld, LDAPConn *conn, LDAPURLDesc *srv )
1040 {
1041         Sockbuf *sb;
1042         char *host;
1043         void *ssl;
1044         int ret;
1045 #ifdef LDAP_USE_NON_BLOCKING_TLS
1046         struct timeval start_time_tv, tv, tv0;
1047         ber_socket_t    sd = AC_SOCKET_ERROR;
1048 #endif /* LDAP_USE_NON_BLOCKING_TLS */
1049
1050         if ( !conn )
1051                 return LDAP_PARAM_ERROR;
1052
1053         sb = conn->lconn_sb;
1054         if( srv ) {
1055                 host = srv->lud_host;
1056         } else {
1057                 host = conn->lconn_server->lud_host;
1058         }
1059
1060         /* avoid NULL host */
1061         if( host == NULL ) {
1062                 host = "localhost";
1063         }
1064
1065         (void) tls_init( tls_imp );
1066
1067 #ifdef LDAP_USE_NON_BLOCKING_TLS
1068         /*
1069          * Use non-blocking io during SSL Handshake when a timeout is configured
1070          */
1071         if ( ld->ld_options.ldo_tm_net.tv_sec >= 0 ) {
1072                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_SET_NONBLOCK, (void*)1 );
1073                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_FD, &sd );
1074                 tv = ld->ld_options.ldo_tm_net;
1075                 tv0 = tv;
1076 #ifdef HAVE_GETTIMEOFDAY
1077                 gettimeofday( &start_time_tv, NULL );
1078 #else /* ! HAVE_GETTIMEOFDAY */
1079                 time( &start_time_tv.tv_sec );
1080                 start_time_tv.tv_usec = 0;
1081 #endif /* ! HAVE_GETTIMEOFDAY */
1082         }
1083
1084 #endif /* LDAP_USE_NON_BLOCKING_TLS */
1085
1086         ld->ld_errno = LDAP_SUCCESS;
1087         ret = ldap_int_tls_connect( ld, conn, host );
1088
1089 #ifdef LDAP_USE_NON_BLOCKING_TLS
1090         while ( ret > 0 ) { /* this should only happen for non-blocking io */
1091                 int wr=0;
1092
1093                 if ( sb->sb_trans_needs_read ) {
1094                         wr=0;
1095                 } else if ( sb->sb_trans_needs_write ) {
1096                         wr=1;
1097                 }
1098                 Debug( LDAP_DEBUG_TRACE, "ldap_int_tls_start: ldap_int_tls_connect needs %s\n",
1099                                 wr ? "write": "read", 0, 0);
1100
1101                 ret = ldap_int_poll( ld, sd, &tv, wr);
1102                 if ( ret < 0 ) {
1103                         ld->ld_errno = LDAP_TIMEOUT;
1104                         break;
1105                 } else {
1106                         /* ldap_int_poll called ldap_pvt_ndelay_off */
1107                         ber_sockbuf_ctrl( sb, LBER_SB_OPT_SET_NONBLOCK, (void*)1 );
1108                         ret = ldap_int_tls_connect( ld, conn, host );
1109                         if ( ret > 0 ) { /* need to call tls_connect once more */
1110                                 struct timeval curr_time_tv, delta_tv;
1111
1112                                 /* This is mostly copied from result.c:wait4msg(), should
1113                                  * probably be moved into a separate function */
1114 #ifdef HAVE_GETTIMEOFDAY
1115                                 gettimeofday( &curr_time_tv, NULL );
1116 #else /* ! HAVE_GETTIMEOFDAY */
1117                                 time( &curr_time_tv.tv_sec );
1118                                 curr_time_tv.tv_usec = 0;
1119 #endif /* ! HAVE_GETTIMEOFDAY */
1120
1121                                 /* delta = curr - start */
1122                                 delta_tv.tv_sec = curr_time_tv.tv_sec - start_time_tv.tv_sec;
1123                                 delta_tv.tv_usec = curr_time_tv.tv_usec - start_time_tv.tv_usec;
1124                                 if ( delta_tv.tv_usec < 0 ) {
1125                                         delta_tv.tv_sec--;
1126                                         delta_tv.tv_usec += 1000000;
1127                                 }
1128
1129                                 /* tv0 < delta ? */
1130                                 if ( ( tv0.tv_sec < delta_tv.tv_sec ) ||
1131                                          ( ( tv0.tv_sec == delta_tv.tv_sec ) &&
1132                                            ( tv0.tv_usec < delta_tv.tv_usec ) ) )
1133                                 {
1134                                         ret = -1;
1135                                         ld->ld_errno = LDAP_TIMEOUT;
1136                                         break;
1137                                 } else {
1138                                         /* timeout -= delta_time */
1139                                         tv0.tv_sec -= delta_tv.tv_sec;
1140                                         tv0.tv_usec -= delta_tv.tv_usec;
1141                                         if ( tv0.tv_usec < 0 ) {
1142                                                 tv0.tv_sec--;
1143                                                 tv0.tv_usec += 1000000;
1144                                         }
1145                                         start_time_tv.tv_sec = curr_time_tv.tv_sec;
1146                                         start_time_tv.tv_usec = curr_time_tv.tv_usec;
1147                                 }
1148                                 tv = tv0;
1149                                 Debug( LDAP_DEBUG_TRACE, "ldap_int_tls_start: ld %p %ld s %ld us to go\n",
1150                                         (void *)ld, (long) tv.tv_sec, (long) tv.tv_usec );
1151                         }
1152                 }
1153         }
1154         if ( ld->ld_options.ldo_tm_net.tv_sec >= 0 ) {
1155                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_SET_NONBLOCK, NULL );
1156         }
1157 #endif /* LDAP_USE_NON_BLOCKING_TLS */
1158
1159         if ( ret < 0 ) {
1160                 if ( ld->ld_errno == LDAP_SUCCESS )
1161                         ld->ld_errno = LDAP_CONNECT_ERROR;
1162                 return (ld->ld_errno);
1163         }
1164
1165         return LDAP_SUCCESS;
1166 }
1167
1168 void *
1169 ldap_pvt_tls_sb_ctx( Sockbuf *sb )
1170 {
1171         void                    *p = NULL;
1172
1173         ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_SSL, (void *)&p );
1174         return p;
1175 }
1176
1177 int
1178 ldap_pvt_tls_get_strength( void *s )
1179 {
1180         tls_session *session = s;
1181
1182         return tls_imp->ti_session_strength( session );
1183 }
1184
1185 int
1186 ldap_pvt_tls_get_my_dn( void *s, struct berval *dn, LDAPDN_rewrite_dummy *func, unsigned flags )
1187 {
1188         tls_session *session = s;
1189         struct berval der_dn;
1190         int rc;
1191
1192         rc = tls_imp->ti_session_my_dn( session, &der_dn );
1193         if ( rc == LDAP_SUCCESS )
1194                 rc = ldap_X509dn2bv(&der_dn, dn, (LDAPDN_rewrite_func *)func, flags );
1195         return rc;
1196 }
1197
1198 int
1199 ldap_pvt_tls_get_unique( void *s, struct berval *buf, int is_server )
1200 {
1201         tls_session *session = s;
1202         return tls_imp->ti_session_unique( session, buf, is_server );
1203 }
1204
1205 const char *
1206 ldap_pvt_tls_get_version( void *s )
1207 {
1208         tls_session *session = s;
1209         return tls_imp->ti_session_version( session );
1210 }
1211
1212 const char *
1213 ldap_pvt_tls_get_cipher( void *s )
1214 {
1215         tls_session *session = s;
1216         return tls_imp->ti_session_cipher( session );
1217 }
1218
1219 int
1220 ldap_pvt_tls_get_peercert( void *s, struct berval *der )
1221 {
1222         tls_session *session = s;
1223         return tls_imp->ti_session_peercert( session, der );
1224 }
1225 #endif /* HAVE_TLS */
1226
1227 int
1228 ldap_start_tls( LDAP *ld,
1229         LDAPControl **serverctrls,
1230         LDAPControl **clientctrls,
1231         int *msgidp )
1232 {
1233         return ldap_extended_operation( ld, LDAP_EXOP_START_TLS,
1234                 NULL, serverctrls, clientctrls, msgidp );
1235 }
1236
1237 int
1238 ldap_install_tls( LDAP *ld )
1239 {
1240 #ifndef HAVE_TLS
1241         return LDAP_NOT_SUPPORTED;
1242 #else
1243         if ( ldap_tls_inplace( ld ) ) {
1244                 return LDAP_LOCAL_ERROR;
1245         }
1246
1247         return ldap_int_tls_start( ld, ld->ld_defconn, NULL );
1248 #endif
1249 }
1250
1251 int
1252 ldap_start_tls_s ( LDAP *ld,
1253         LDAPControl **serverctrls,
1254         LDAPControl **clientctrls )
1255 {
1256 #ifndef HAVE_TLS
1257         return LDAP_NOT_SUPPORTED;
1258 #else
1259         int rc;
1260         char *rspoid = NULL;
1261         struct berval *rspdata = NULL;
1262
1263         /* XXYYZ: this initiates operation only on default connection! */
1264
1265         if ( ldap_tls_inplace( ld ) ) {
1266                 return LDAP_LOCAL_ERROR;
1267         }
1268
1269         rc = ldap_extended_operation_s( ld, LDAP_EXOP_START_TLS,
1270                 NULL, serverctrls, clientctrls, &rspoid, &rspdata );
1271
1272         if ( rspoid != NULL ) {
1273                 LDAP_FREE(rspoid);
1274         }
1275
1276         if ( rspdata != NULL ) {
1277                 ber_bvfree( rspdata );
1278         }
1279
1280         if ( rc == LDAP_SUCCESS ) {
1281                 rc = ldap_int_tls_start( ld, ld->ld_defconn, NULL );
1282         }
1283
1284         return rc;
1285 #endif
1286 }
1287
1288 /* These tags probably all belong in lber.h, but they're
1289  * not normally encountered when processing LDAP, so maybe
1290  * they belong somewhere else instead.
1291  */
1292
1293 #define LBER_TAG_OID            ((ber_tag_t) 0x06UL)
1294
1295 /* Tags for string types used in a DirectoryString.
1296  *
1297  * Note that IA5string is not one of the defined choices for
1298  * DirectoryString in X.520, but it gets used for email AVAs.
1299  */
1300 #define LBER_TAG_UTF8           ((ber_tag_t) 0x0cUL)
1301 #define LBER_TAG_PRINTABLE      ((ber_tag_t) 0x13UL)
1302 #define LBER_TAG_TELETEX        ((ber_tag_t) 0x14UL)
1303 #define LBER_TAG_IA5            ((ber_tag_t) 0x16UL)
1304 #define LBER_TAG_UNIVERSAL      ((ber_tag_t) 0x1cUL)
1305 #define LBER_TAG_BMP            ((ber_tag_t) 0x1eUL)
1306
1307 static oid_name *
1308 find_oid( struct berval *oid )
1309 {
1310         int i;
1311
1312         for ( i=0; !BER_BVISNULL( &oids[i].oid ); i++ ) {
1313                 if ( oids[i].oid.bv_len != oid->bv_len ) continue;
1314                 if ( !strcmp( oids[i].oid.bv_val, oid->bv_val ))
1315                         return &oids[i];
1316         }
1317         return NULL;
1318 }
1319
1320 /* Converts BER Bitstring value to LDAP BitString value (RFC4517)
1321  *
1322  * berValue    : IN
1323  * rfc4517Value: OUT
1324  *
1325  * berValue and ldapValue should not be NULL
1326  */
1327
1328 #define BITS_PER_BYTE   8
1329 #define SQUOTE_LENGTH   1
1330 #define B_CHAR_LENGTH   1
1331 #define STR_OVERHEAD    (2*SQUOTE_LENGTH + B_CHAR_LENGTH)
1332
1333 static int
1334 der_to_ldap_BitString (struct berval *berValue,
1335                                    struct berval *ldapValue)
1336 {
1337         ber_len_t bitPadding=0;
1338         ber_len_t bits, maxBits;
1339         char *tmpStr;
1340         unsigned char byte;
1341         ber_len_t bitLength;
1342         ber_len_t valLen;
1343         unsigned char* valPtr;
1344
1345         ldapValue->bv_len=0;
1346         ldapValue->bv_val=NULL;
1347
1348         /* Gets padding and points to binary data */
1349         valLen=berValue->bv_len;
1350         valPtr=(unsigned char*)berValue->bv_val;
1351         if (valLen) {
1352                 bitPadding=(ber_len_t)(valPtr[0]);
1353                 valLen--;
1354                 valPtr++;
1355         }
1356         /* If Block is non DER encoding fixes to DER encoding */
1357         if (bitPadding >= BITS_PER_BYTE) {
1358                 if (valLen*BITS_PER_BYTE > bitPadding ) {
1359                         valLen-=(bitPadding/BITS_PER_BYTE);
1360                         bitPadding%=BITS_PER_BYTE;
1361                 } else {
1362                         valLen=0;
1363                         bitPadding=0;
1364                 }
1365         }
1366         /* Just in case bad encoding */
1367         if (valLen*BITS_PER_BYTE < bitPadding ) {
1368                 bitPadding=0;
1369                 valLen=0;
1370         }
1371
1372         /* Gets buffer to hold RFC4517 Bit String format */
1373         bitLength=valLen*BITS_PER_BYTE-bitPadding;
1374         tmpStr=LDAP_MALLOC(bitLength + STR_OVERHEAD + 1);
1375
1376         if (!tmpStr)
1377                 return LDAP_NO_MEMORY;
1378
1379         ldapValue->bv_val=tmpStr;
1380         ldapValue->bv_len=bitLength + STR_OVERHEAD;
1381
1382         /* Formatting in '*binary-digit'B format */
1383         maxBits=BITS_PER_BYTE;
1384         *tmpStr++ ='\'';
1385         while(valLen) {
1386                 byte=*valPtr;
1387                 if (valLen==1)
1388                         maxBits-=bitPadding;
1389                 for (bits=0; bits<maxBits; bits++) {
1390                         if (0x80 & byte)
1391                                 *tmpStr='1';
1392                         else
1393                                 *tmpStr='0';
1394                         tmpStr++;
1395                         byte<<=1;
1396                 }
1397                 valPtr++;
1398                 valLen--;
1399         }
1400         *tmpStr++ ='\'';
1401         *tmpStr++ ='B';
1402         *tmpStr=0;
1403
1404         return LDAP_SUCCESS;
1405 }
1406
1407 /* Convert a structured DN from an X.509 certificate into an LDAPV3 DN.
1408  * x509_name must be raw DER. If func is non-NULL, the
1409  * constructed DN will use numeric OIDs to identify attributeTypes,
1410  * and the func() will be invoked to rewrite the DN with the given
1411  * flags.
1412  *
1413  * Otherwise the DN will use shortNames from a hardcoded table.
1414  */
1415 int
1416 ldap_X509dn2bv( void *x509_name, struct berval *bv, LDAPDN_rewrite_func *func,
1417         unsigned flags )
1418 {
1419         LDAPDN  newDN;
1420         LDAPRDN newRDN;
1421         LDAPAVA *newAVA, *baseAVA;
1422         BerElementBuffer berbuf;
1423         BerElement *ber = (BerElement *)&berbuf;
1424         char oids[8192], *oidptr = oids, *oidbuf = NULL;
1425         void *ptrs[2048];
1426         char *dn_end, *rdn_end;
1427         int i, navas, nrdns, rc = LDAP_SUCCESS;
1428         size_t dnsize, oidrem = sizeof(oids), oidsize = 0;
1429         int csize;
1430         ber_tag_t tag;
1431         ber_len_t len;
1432         oid_name *oidname;
1433
1434         struct berval   Oid, Val, oid2, *in = x509_name;
1435
1436         assert( bv != NULL );
1437
1438         bv->bv_len = 0;
1439         bv->bv_val = NULL;
1440
1441         navas = 0;
1442         nrdns = 0;
1443
1444         /* A DN is a SEQUENCE of RDNs. An RDN is a SET of AVAs.
1445          * An AVA is a SEQUENCE of attr and value.
1446          * Count the number of AVAs and RDNs
1447          */
1448         ber_init2( ber, in, LBER_USE_DER );
1449         tag = ber_peek_tag( ber, &len );
1450         if ( tag != LBER_SEQUENCE )
1451                 return LDAP_DECODING_ERROR;
1452
1453         for ( tag = ber_first_element( ber, &len, &dn_end );
1454                 tag == LBER_SET;
1455                 tag = ber_next_element( ber, &len, dn_end )) {
1456                 nrdns++;
1457                 for ( tag = ber_first_element( ber, &len, &rdn_end );
1458                         tag == LBER_SEQUENCE;
1459                         tag = ber_next_element( ber, &len, rdn_end )) {
1460                         tag = ber_skip_tag( ber, &len );
1461                         ber_skip_data( ber, len );
1462                         navas++;
1463                 }
1464         }
1465
1466         /* Allocate the DN/RDN/AVA stuff as a single block */
1467         dnsize = sizeof(LDAPRDN) * (nrdns+1);
1468         dnsize += sizeof(LDAPAVA *) * (navas+nrdns);
1469         dnsize += sizeof(LDAPAVA) * navas;
1470         if (dnsize > sizeof(ptrs)) {
1471                 newDN = (LDAPDN)LDAP_MALLOC( dnsize );
1472                 if ( newDN == NULL )
1473                         return LDAP_NO_MEMORY;
1474         } else {
1475                 newDN = (LDAPDN)(char *)ptrs;
1476         }
1477
1478         newDN[nrdns] = NULL;
1479         newRDN = (LDAPRDN)(newDN + nrdns+1);
1480         newAVA = (LDAPAVA *)(newRDN + navas + nrdns);
1481         baseAVA = newAVA;
1482
1483         /* Rewind and start extracting */
1484         ber_rewind( ber );
1485
1486         tag = ber_first_element( ber, &len, &dn_end );
1487         for ( i = nrdns - 1; i >= 0; i-- ) {
1488                 newDN[i] = newRDN;
1489
1490                 for ( tag = ber_first_element( ber, &len, &rdn_end );
1491                         tag == LBER_SEQUENCE;
1492                         tag = ber_next_element( ber, &len, rdn_end )) {
1493
1494                         *newRDN++ = newAVA;
1495                         tag = ber_skip_tag( ber, &len );
1496                         tag = ber_get_stringbv( ber, &Oid, LBER_BV_NOTERM );
1497                         if ( tag != LBER_TAG_OID ) {
1498                                 rc = LDAP_DECODING_ERROR;
1499                                 goto nomem;
1500                         }
1501
1502                         oid2.bv_val = oidptr;
1503                         oid2.bv_len = oidrem;
1504                         if ( ber_decode_oid( &Oid, &oid2 ) < 0 ) {
1505                                 rc = LDAP_DECODING_ERROR;
1506                                 goto nomem;
1507                         }
1508                         oidname = find_oid( &oid2 );
1509                         if ( !oidname ) {
1510                                 newAVA->la_attr = oid2;
1511                                 oidptr += oid2.bv_len + 1;
1512                                 oidrem -= oid2.bv_len + 1;
1513
1514                                 /* Running out of OID buffer space? */
1515                                 if (oidrem < 128) {
1516                                         if ( oidsize == 0 ) {
1517                                                 oidsize = sizeof(oids) * 2;
1518                                                 oidrem = oidsize;
1519                                                 oidbuf = LDAP_MALLOC( oidsize );
1520                                                 if ( oidbuf == NULL ) goto nomem;
1521                                                 oidptr = oidbuf;
1522                                         } else {
1523                                                 char *old = oidbuf;
1524                                                 oidbuf = LDAP_REALLOC( oidbuf, oidsize*2 );
1525                                                 if ( oidbuf == NULL ) goto nomem;
1526                                                 /* Buffer moved! Fix AVA pointers */
1527                                                 if ( old != oidbuf ) {
1528                                                         LDAPAVA *a;
1529                                                         long dif = oidbuf - old;
1530
1531                                                         for (a=baseAVA; a<=newAVA; a++){
1532                                                                 if (a->la_attr.bv_val >= old &&
1533                                                                         a->la_attr.bv_val <= (old + oidsize))
1534                                                                         a->la_attr.bv_val += dif;
1535                                                         }
1536                                                 }
1537                                                 oidptr = oidbuf + oidsize - oidrem;
1538                                                 oidrem += oidsize;
1539                                                 oidsize *= 2;
1540                                         }
1541                                 }
1542                         } else {
1543                                 if ( func ) {
1544                                         newAVA->la_attr = oidname->oid;
1545                                 } else {
1546                                         newAVA->la_attr = oidname->name;
1547                                 }
1548                         }
1549                         newAVA->la_private = NULL;
1550                         newAVA->la_flags = LDAP_AVA_STRING;
1551                         tag = ber_get_stringbv( ber, &Val, LBER_BV_NOTERM );
1552                         switch(tag) {
1553                         case LBER_TAG_UNIVERSAL:
1554                                 /* This uses 32-bit ISO 10646-1 */
1555                                 csize = 4; goto to_utf8;
1556                         case LBER_TAG_BMP:
1557                                 /* This uses 16-bit ISO 10646-1 */
1558                                 csize = 2; goto to_utf8;
1559                         case LBER_TAG_TELETEX:
1560                                 /* This uses 8-bit, assume ISO 8859-1 */
1561                                 csize = 1;
1562 to_utf8:                rc = ldap_ucs_to_utf8s( &Val, csize, &newAVA->la_value );
1563                                 newAVA->la_flags |= LDAP_AVA_NONPRINTABLE;
1564 allocd:
1565                                 newAVA->la_flags |= LDAP_AVA_FREE_VALUE;
1566                                 if (rc != LDAP_SUCCESS) goto nomem;
1567                                 break;
1568                         case LBER_TAG_UTF8:
1569                                 newAVA->la_flags |= LDAP_AVA_NONPRINTABLE;
1570                                 /* This is already in UTF-8 encoding */
1571                         case LBER_TAG_IA5:
1572                         case LBER_TAG_PRINTABLE:
1573                                 /* These are always 7-bit strings */
1574                                 newAVA->la_value = Val;
1575                                 break;
1576                         case LBER_BITSTRING:
1577                                 /* X.690 bitString value converted to RFC4517 Bit String */
1578                                 rc = der_to_ldap_BitString( &Val, &newAVA->la_value );
1579                                 goto allocd;
1580                         default:
1581                                 /* Not a string type at all */
1582                                 newAVA->la_flags = 0;
1583                                 newAVA->la_value = Val;
1584                                 break;
1585                         }
1586                         newAVA++;
1587                 }
1588                 *newRDN++ = NULL;
1589                 tag = ber_next_element( ber, &len, dn_end );
1590         }
1591
1592         if ( func ) {
1593                 rc = func( newDN, flags, NULL );
1594                 if ( rc != LDAP_SUCCESS )
1595                         goto nomem;
1596         }
1597
1598         rc = ldap_dn2bv_x( newDN, bv, LDAP_DN_FORMAT_LDAPV3, NULL );
1599
1600 nomem:
1601         for (;baseAVA < newAVA; baseAVA++) {
1602                 if (baseAVA->la_flags & LDAP_AVA_FREE_ATTR)
1603                         LDAP_FREE( baseAVA->la_attr.bv_val );
1604                 if (baseAVA->la_flags & LDAP_AVA_FREE_VALUE)
1605                         LDAP_FREE( baseAVA->la_value.bv_val );
1606         }
1607
1608         if ( oidsize != 0 )
1609                 LDAP_FREE( oidbuf );
1610         if ( newDN != (LDAPDN)(char *) ptrs )
1611                 LDAP_FREE( newDN );
1612         return rc;
1613 }
1614