git.sur5r.net Git - openldap/blob - servers/slapd/back-meta/conn.c

   1 /* $OpenLDAP$ */
   2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   3  *
   4  * Copyright 1999-2006 The OpenLDAP Foundation.
   5  * Portions Copyright 2001-2003 Pierangelo Masarati.
   6  * Portions Copyright 1999-2003 Howard Chu.
   7  * All rights reserved.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted only as authorized by the OpenLDAP
  11  * Public License.
  12  *
  13  * A copy of this license is available in the file LICENSE in the
  14  * top-level directory of the distribution or, alternatively, at
  15  * <http://www.OpenLDAP.org/license.html>.
  16  */
  17 /* ACKNOWLEDGEMENTS:
  18  * This work was initially developed by the Howard Chu for inclusion
  19  * in OpenLDAP Software and subsequently enhanced by Pierangelo
  20  * Masarati.
  21  */
  22
  23 #include "portable.h"
  24
  25 #include <stdio.h>
  26
  27 #include <ac/errno.h>
  28 #include <ac/socket.h>
  29 #include <ac/string.h>
  30
  31
  32 #define AVL_INTERNAL
  33 #include "slap.h"
  34 #include "../back-ldap/back-ldap.h"
  35 #include "back-meta.h"
  36
  37 /*
  38  * Set PRINT_CONNTREE larger than 0 to dump the connection tree (debug only)
  39  */
  40 #ifndef PRINT_CONNTREE
  41 #define PRINT_CONNTREE 0
  42 #endif /* !PRINT_CONNTREE */
  43
  44 /*
  45  * meta_back_conndn_cmp
  46  *
  47  * compares two struct metaconn based on the value of the conn pointer
  48  * and of the local DN; used by avl stuff
  49  */
  50 int
  51 meta_back_conndn_cmp(
  52         const void *c1,
  53         const void *c2 )
  54 {
  55         metaconn_t      *mc1 = ( metaconn_t * )c1;
  56         metaconn_t      *mc2 = ( metaconn_t * )c2;
  57         int             rc;
  58
  59         /* If local DNs don't match, it is definitely not a match */
  60         /* For shared sessions, conn is NULL. Only explicitly
  61          * bound sessions will have non-NULL conn.
  62          */
  63         rc = SLAP_PTRCMP( mc1->mc_conn, mc2->mc_conn );
  64         if ( rc == 0 ) {
  65                 rc = ber_bvcmp( &mc1->mc_local_ndn, &mc2->mc_local_ndn );
  66         }
  67
  68         return rc;
  69 }
  70
  71 /*
  72  * meta_back_conndnmc_cmp
  73  *
  74  * compares two struct metaconn based on the value of the conn pointer,
  75  * the local DN and the struct pointer; used by avl stuff
  76  */
  77 static int
  78 meta_back_conndnmc_cmp(
  79         const void *c1,
  80         const void *c2 )
  81 {
  82         metaconn_t      *mc1 = ( metaconn_t * )c1;
  83         metaconn_t      *mc2 = ( metaconn_t * )c2;
  84         int             rc;
  85
  86         /* If local DNs don't match, it is definitely not a match */
  87         /* For shared sessions, conn is NULL. Only explicitly
  88          * bound sessions will have non-NULL conn.
  89          */
  90         rc = SLAP_PTRCMP( mc1->mc_conn, mc2->mc_conn );
  91         if ( rc == 0 ) {
  92                 rc = ber_bvcmp( &mc1->mc_local_ndn, &mc2->mc_local_ndn );
  93                 if ( rc == 0 ) {
  94                         rc = SLAP_PTRCMP( mc1, mc2 );
  95                 }
  96         }
  97
  98         return rc;
  99 }
 100
 101 /*
 102  * meta_back_conn_cmp
 103  *
 104  * compares two struct metaconn based on the value of the conn pointer;
 105  * used by avl stuff
 106  */
 107 int
 108 meta_back_conn_cmp(
 109         const void *c1,
 110         const void *c2 )
 111 {
 112         metaconn_t      *mc1 = ( metaconn_t * )c1;
 113         metaconn_t      *mc2 = ( metaconn_t * )c2;
 114
 115         /* For shared sessions, conn is NULL. Only explicitly
 116          * bound sessions will have non-NULL conn.
 117          */
 118         return SLAP_PTRCMP( mc1->mc_conn, mc2->mc_conn );
 119 }
 120
 121 /*
 122  * meta_back_conndn_dup
 123  *
 124  * returns -1 in case a duplicate struct metaconn has been inserted;
 125  * used by avl stuff
 126  */
 127 int
 128 meta_back_conndn_dup(
 129         void *c1,
 130         void *c2 )
 131 {
 132         metaconn_t      *mc1 = ( metaconn_t * )c1;
 133         metaconn_t      *mc2 = ( metaconn_t * )c2;
 134
 135         /* Cannot have more than one shared session with same DN */
 136         if ( mc1->mc_conn == mc2->mc_conn &&
 137                 dn_match( &mc1->mc_local_ndn, &mc2->mc_local_ndn ) )
 138         {
 139                 return -1;
 140         }
 141
 142         return 0;
 143 }
 144
 145 /*
 146  * Debug stuff (got it from libavl)
 147  */
 148 #if PRINT_CONNTREE > 0
 149 static void
 150 ravl_print( Avlnode *root, int depth )
 151 {
 152         int             i;
 153         metaconn_t      *mc;
 154
 155         if ( root == 0 ) {
 156                 return;
 157         }
 158
 159         ravl_print( root->avl_right, depth + 1 );
 160
 161         for ( i = 0; i < depth; i++ ) {
 162                 fprintf( stderr, "-" );
 163         }
 164
 165         mc = (metaconn_t *)root->avl_data;
 166         fprintf( stderr, "mc=%p local=\"%s\" conn=%p %s refcnt=%d\n",
 167                 (void *)mc,
 168                 mc->mc_local_ndn.bv_val ? mc->mc_local_ndn.bv_val : "",
 169                 (void *)mc->mc_conn,
 170                 avl_bf2str( root->avl_bf ), mc->mc_refcnt );
 171
 172         ravl_print( root->avl_left, depth + 1 );
 173 }
 174
 175 static void
 176 myprint( Avlnode *root, char *msg )
 177 {
 178         fprintf( stderr, "========> %s\n", msg );
 179
 180         if ( root == 0 ) {
 181                 fprintf( stderr, "\tNULL\n" );
 182
 183         } else {
 184                 ravl_print( root, 0 );
 185         }
 186
 187         fprintf( stderr, "<======== %s\n", msg );
 188 }
 189 #endif /* PRINT_CONNTREE */
 190 /*
 191  * End of debug stuff
 192  */
 193
 194 /*
 195  * metaconn_alloc
 196  *
 197  * Allocates a connection structure, making room for all the referenced targets
 198  */
 199 static metaconn_t *
 200 metaconn_alloc(
 201         Operation               *op )
 202 {
 203         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 204         metaconn_t      *mc;
 205         int             i, ntargets = mi->mi_ntargets;
 206
 207         assert( ntargets > 0 );
 208
 209         /* malloc all in one */
 210         mc = ( metaconn_t * )ch_malloc( sizeof( metaconn_t )
 211                         + sizeof( metasingleconn_t ) * ntargets );
 212         if ( mc == NULL ) {
 213                 return NULL;
 214         }
 215
 216         for ( i = 0; i < ntargets; i++ ) {
 217                 mc->mc_conns[ i ].msc_ld = NULL;
 218                 BER_BVZERO( &mc->mc_conns[ i ].msc_bound_ndn );
 219                 BER_BVZERO( &mc->mc_conns[ i ].msc_cred );
 220                 mc->mc_conns[ i ].msc_mscflags = 0;
 221                 mc->mc_conns[ i ].msc_info = mi;
 222         }
 223
 224         BER_BVZERO( &mc->mc_local_ndn );
 225         mc->msc_mscflags = 0;
 226         mc->mc_authz_target = META_BOUND_NONE;
 227         mc->mc_refcnt = 1;
 228
 229         return mc;
 230 }
 231
 232 static void
 233 meta_back_freeconn(
 234         Operation       *op,
 235         metaconn_t      *mc )
 236 {
 237         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 238
 239         assert( mc != NULL );
 240
 241         ldap_pvt_thread_mutex_lock( &mi->mi_conninfo.lai_mutex );
 242
 243         if ( --mc->mc_refcnt == 0 ) {
 244                 meta_back_conn_free( mc );
 245         }
 246
 247         ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
 248 }
 249
 250 /*
 251  * meta_back_init_one_conn
 252  *
 253  * Initializes one connection
 254  */
 255 int
 256 meta_back_init_one_conn(
 257         Operation               *op,
 258         SlapReply               *rs,
 259         metaconn_t              *mc,
 260         int                     candidate,
 261         int                     ispriv,
 262         ldap_back_send_t        sendok )
 263 {
 264         metainfo_t              *mi = ( metainfo_t * )op->o_bd->be_private;
 265         metatarget_t            *mt = mi->mi_targets[ candidate ];
 266         metasingleconn_t        *msc = &mc->mc_conns[ candidate ];
 267         int                     version;
 268         dncookie                dc;
 269         int                     isauthz = ( candidate == mc->mc_authz_target );
 270 #ifdef HAVE_TLS
 271         int                     is_ldaps = 0;
 272 #endif /* HAVE_TLS */
 273
 274         /* if the server is quarantined, and
 275          * - the current interval did not expire yet, or
 276          * - no more retries should occur,
 277          * don't return the connection */
 278         if ( mt->mt_isquarantined ) {
 279                 slap_retry_info_t       *ri = &mt->mt_quarantine;
 280                 int                     dont_retry = 1;
 281
 282                 if ( mt->mt_isquarantined == LDAP_BACK_FQ_YES ) {
 283                         dont_retry = ( ri->ri_num[ ri->ri_idx ] == SLAP_RETRYNUM_TAIL
 284                                 || slap_get_time() < ri->ri_last + ri->ri_interval[ ri->ri_idx ] );
 285                         if ( !dont_retry ) {
 286                                 Debug( LDAP_DEBUG_ANY,
 287                                         "%s: meta_back_init_one_conn quarantine "
 288                                         "retry block #%d try #%d.\n",
 289                                         op->o_log_prefix, ri->ri_idx, ri->ri_count );
 290                                 mt->mt_isquarantined = LDAP_BACK_FQ_RETRYING;
 291                         }
 292                 }
 293
 294                 if ( dont_retry ) {
 295                         rs->sr_err = LDAP_UNAVAILABLE;
 296                         if ( op->o_conn && ( sendok & LDAP_BACK_SENDERR ) ) {
 297                                 send_ldap_result( op, rs );
 298                         }
 299                         return rs->sr_err;
 300                 }
 301         }
 302
 303         /*
 304          * Already init'ed
 305          */
 306         if ( msc->msc_ld != NULL ) {
 307                 return rs->sr_err = LDAP_SUCCESS;
 308         }
 309
 310         msc->msc_mscflags = 0;
 311
 312         /*
 313          * Attempts to initialize the connection to the target ds
 314          */
 315         ldap_pvt_thread_mutex_lock( &mt->mt_uri_mutex );
 316         rs->sr_err = ldap_initialize( &msc->msc_ld, mt->mt_uri );
 317 #ifdef HAVE_TLS
 318         is_ldaps = ldap_is_ldaps_url( mt->mt_uri );
 319 #endif /* HAVE_TLS */
 320         ldap_pvt_thread_mutex_unlock( &mt->mt_uri_mutex );
 321         if ( rs->sr_err != LDAP_SUCCESS ) {
 322                 goto error_return;
 323         }
 324
 325         /*
 326          * Set LDAP version. This will always succeed: If the client
 327          * bound with a particular version, then so can we.
 328          */
 329         if ( mt->mt_version != 0 ) {
 330                 version = mt->mt_version;
 331
 332         } else if ( op->o_conn->c_protocol != 0 ) {
 333                 version = op->o_conn->c_protocol;
 334
 335         } else {
 336                 version = LDAP_VERSION3;
 337         }
 338         ldap_set_option( msc->msc_ld, LDAP_OPT_PROTOCOL_VERSION, &version );
 339         ldap_set_urllist_proc( msc->msc_ld, mt->mt_urllist_f, mt->mt_urllist_p );
 340
 341         /* automatically chase referrals ("chase-referrals [{yes|no}]" statement) */
 342         ldap_set_option( msc->msc_ld, LDAP_OPT_REFERRALS,
 343                 LDAP_BACK_CHASE_REFERRALS( mi ) ? LDAP_OPT_ON : LDAP_OPT_OFF );
 344
 345 #ifdef HAVE_TLS
 346         /* start TLS ("tls [try-]{start|propagate}" statement) */
 347         if ( ( LDAP_BACK_USE_TLS( mi ) || ( op->o_conn->c_is_tls && LDAP_BACK_PROPAGATE_TLS( mi ) ) )
 348                         && !is_ldaps )
 349         {
 350 #ifdef SLAP_STARTTLS_ASYNCHRONOUS
 351                 /*
 352                  * use asynchronous StartTLS; in case, chase referral
 353                  * FIXME: OpenLDAP does not return referral on StartTLS yet
 354                  */
 355                 int             msgid;
 356
 357                 rs->sr_err = ldap_start_tls( msc->msc_ld, NULL, NULL, &msgid );
 358                 if ( rs->sr_err == LDAP_SUCCESS ) {
 359                         LDAPMessage     *res = NULL;
 360                         int             rc, nretries = mt->mt_nretries;
 361                         struct timeval  tv;
 362
 363                         LDAP_BACK_TV_SET( &tv );
 364
 365 retry:;
 366                         rc = ldap_result( msc->msc_ld, msgid, LDAP_MSG_ALL, &tv, &res );
 367                         if ( rc < 0 ) {
 368                                 rs->sr_err = LDAP_OTHER;
 369
 370                         } else if ( rc == 0 ) {
 371                                 if ( nretries != 0 ) {
 372                                         if ( nretries > 0 ) {
 373                                                 nretries--;
 374                                         }
 375                                         LDAP_BACK_TV_SET( &tv );
 376                                         goto retry;
 377                                 }
 378                                 rs->sr_err = LDAP_OTHER;
 379
 380                         } else if ( rc == LDAP_RES_EXTENDED ) {
 381                                 struct berval   *data = NULL;
 382
 383                                 rs->sr_err = ldap_parse_extended_result( msc->msc_ld, res,
 384                                                 NULL, &data, 0 );
 385                                 if ( rs->sr_err == LDAP_SUCCESS ) {
 386                                         int             err;
 387
 388                                         rs->sr_err = ldap_parse_result( msc->msc_ld, res,
 389                                                 &err, NULL, NULL, NULL, NULL, 1 );
 390                                         res = NULL;
 391
 392                                         if ( rs->sr_err == LDAP_SUCCESS ) {
 393                                                 rs->sr_err = err;
 394                                         }
 395
 396                                         /* FIXME: in case a referral
 397                                          * is returned, should we try
 398                                          * using it instead of the
 399                                          * configured URI? */
 400                                         if ( rs->sr_err == LDAP_SUCCESS ) {
 401                                                 ldap_install_tls( msc->msc_ld );
 402
 403                                         } else if ( rs->sr_err == LDAP_REFERRAL ) {
 404                                                 rs->sr_err = LDAP_OTHER;
 405                                                 rs->sr_text = "unwilling to chase referral returned by Start TLS exop";
 406                                         }
 407
 408                                         if ( data ) {
 409                                                 if ( data->bv_val ) {
 410                                                         ber_memfree( data->bv_val );
 411                                                 }
 412                                                 ber_memfree( data );
 413                                         }
 414                                 }
 415
 416                         } else {
 417                                 rs->sr_err = LDAP_OTHER;
 418                         }
 419
 420                         if ( res != NULL ) {
 421                                 ldap_msgfree( res );
 422                         }
 423                 }
 424 #else /* ! SLAP_STARTTLS_ASYNCHRONOUS */
 425                 /*
 426                  * use synchronous StartTLS
 427                  */
 428                 rs->sr_err = ldap_start_tls_s( msc->msc_ld, NULL, NULL );
 429 #endif /* ! SLAP_STARTTLS_ASYNCHRONOUS */
 430
 431                 /* if StartTLS is requested, only attempt it if the URL
 432                  * is not "ldaps://"; this may occur not only in case
 433                  * of misconfiguration, but also when used in the chain
 434                  * overlay, where the "uri" can be parsed out of a referral */
 435                 if ( rs->sr_err == LDAP_SERVER_DOWN
 436                                 || ( rs->sr_err != LDAP_SUCCESS && LDAP_BACK_TLS_CRITICAL( mi ) ) )
 437                 {
 438                         ldap_unbind_ext( msc->msc_ld, NULL, NULL );
 439                         msc->msc_ld = NULL;
 440                         goto error_return;
 441                 }
 442         }
 443 #endif /* HAVE_TLS */
 444
 445         /*
 446          * Set the network timeout if set
 447          */
 448         if ( mt->mt_network_timeout != 0 ) {
 449                 struct timeval  network_timeout;
 450
 451                 network_timeout.tv_usec = 0;
 452                 network_timeout.tv_sec = mt->mt_network_timeout;
 453
 454                 ldap_set_option( msc->msc_ld, LDAP_OPT_NETWORK_TIMEOUT,
 455                                 (void *)&network_timeout );
 456         }
 457
 458         /*
 459          * If the connection DN is not null, an attempt to rewrite it is made
 460          */
 461
 462         if ( ispriv ) {
 463                 if ( !BER_BVISNULL( &mt->mt_pseudorootdn ) ) {
 464                         ber_dupbv( &msc->msc_bound_ndn, &mt->mt_pseudorootdn );
 465                         if ( !BER_BVISNULL( &mt->mt_pseudorootpw ) ) {
 466                                 ber_dupbv( &msc->msc_cred, &mt->mt_pseudorootpw );
 467                         }
 468
 469                 } else {
 470                         ber_str2bv( "", 0, 1, &msc->msc_bound_ndn );
 471                 }
 472
 473                 LDAP_BACK_CONN_ISPRIV_SET( msc );
 474
 475         } else {
 476                 BER_BVZERO( &msc->msc_cred );
 477                 BER_BVZERO( &msc->msc_bound_ndn );
 478                 if ( !BER_BVISEMPTY( &op->o_ndn )
 479                         && SLAP_IS_AUTHZ_BACKEND( op )
 480                         && isauthz )
 481                 {
 482                         dc.target = mt;
 483                         dc.conn = op->o_conn;
 484                         dc.rs = rs;
 485                         dc.ctx = "bindDN";
 486
 487                         /*
 488                          * Rewrite the bind dn if needed
 489                          */
 490                         if ( ldap_back_dn_massage( &dc, &op->o_conn->c_dn,
 491                                                 &msc->msc_bound_ndn ) )
 492                         {
 493                                 ldap_unbind_ext( msc->msc_ld, NULL, NULL );
 494                                 msc->msc_ld = NULL;
 495                                 goto error_return;
 496                         }
 497
 498                         /* copy the DN idf needed */
 499                         if ( msc->msc_bound_ndn.bv_val == op->o_conn->c_dn.bv_val ) {
 500                                 ber_dupbv( &msc->msc_bound_ndn, &op->o_conn->c_dn );
 501                         }
 502
 503                 } else {
 504                         ber_str2bv( "", 0, 1, &msc->msc_bound_ndn );
 505                 }
 506         }
 507
 508         assert( !BER_BVISNULL( &msc->msc_bound_ndn ) );
 509
 510 error_return:;
 511         if ( rs->sr_err == LDAP_SUCCESS ) {
 512                 /*
 513                  * Sets a cookie for the rewrite session
 514                  */
 515                 ( void )rewrite_session_init( mt->mt_rwmap.rwm_rw, op->o_conn );
 516
 517         } else {
 518                 rs->sr_err = slap_map_api2result( rs );
 519                 if ( sendok & LDAP_BACK_SENDERR ) {
 520                         send_ldap_result( op, rs );
 521                         rs->sr_text = NULL;
 522                 }
 523         }
 524
 525         return rs->sr_err;
 526 }
 527
 528 /*
 529  * meta_back_retry
 530  *
 531  * Retries one connection
 532  */
 533 int
 534 meta_back_retry(
 535         Operation               *op,
 536         SlapReply               *rs,
 537         metaconn_t              **mcp,
 538         int                     candidate,
 539         ldap_back_send_t        sendok )
 540 {
 541         metainfo_t              *mi = ( metainfo_t * )op->o_bd->be_private;
 542         metatarget_t            *mt = mi->mi_targets[ candidate ];
 543         metaconn_t              *mc = *mcp;
 544         metasingleconn_t        *msc = &mc->mc_conns[ candidate ];
 545         int                     rc = LDAP_UNAVAILABLE,
 546                                 binding = LDAP_BACK_CONN_BINDING( msc );
 547
 548         ldap_pvt_thread_mutex_lock( &mi->mi_conninfo.lai_mutex );
 549
 550         assert( mc->mc_refcnt > 0 );
 551         if ( mc->mc_refcnt == 1 ) {
 552                 if ( LogTest( LDAP_DEBUG_ANY ) ) {
 553                         char    buf[ SLAP_TEXT_BUFLEN ];
 554
 555                         ldap_pvt_thread_mutex_lock( &mt->mt_uri_mutex );
 556                         snprintf( buf, sizeof( buf ),
 557                                 "retrying URI=\"%s\" DN=\"%s\"",
 558                                 mt->mt_uri,
 559                                 BER_BVISNULL( &msc->msc_bound_ndn ) ?
 560                                         "" : msc->msc_bound_ndn.bv_val );
 561                         ldap_pvt_thread_mutex_unlock( &mt->mt_uri_mutex );
 562
 563                         Debug( LDAP_DEBUG_ANY,
 564                                 "%s meta_back_retry[%d]: %s.\n",
 565                                 op->o_log_prefix, candidate, buf );
 566                 }
 567
 568                 meta_clear_one_candidate( msc );
 569                 LDAP_BACK_CONN_ISBOUND_CLEAR( msc );
 570
 571                 ( void )rewrite_session_delete( mt->mt_rwmap.rwm_rw, op->o_conn );
 572
 573                 /* mc here must be the regular mc, reset and ready for init */
 574                 rc = meta_back_init_one_conn( op, rs, mc, candidate,
 575                         LDAP_BACK_CONN_ISPRIV( mc ), sendok );
 576                 if ( binding ) {
 577                         LDAP_BACK_CONN_BINDING_SET( msc );
 578                 }
 579
 580                 if ( rc == LDAP_SUCCESS ) {
 581                         rc = meta_back_single_dobind( op, rs, mcp, candidate,
 582                                 sendok, mt->mt_nretries, 0 );
 583
 584                         Debug( LDAP_DEBUG_ANY,
 585                                 "%s meta_back_retry[%d]: "
 586                                 "meta_back_single_dobind=%d\n",
 587                                 op->o_log_prefix, candidate, rc );
 588                         if ( rc == LDAP_SUCCESS ) {
 589                                 if ( be_isroot( op )  ) {
 590                                         LDAP_BACK_CONN_ISBOUND_SET( msc );
 591                                 } else {
 592                                         LDAP_BACK_CONN_ISANON_SET( msc );
 593                                 }
 594                         }
 595                 }
 596         }
 597
 598         if ( rc != LDAP_SUCCESS ) {
 599                 SlapReply               *candidates = meta_back_candidates_get( op );
 600
 601                 candidates[ candidate ].sr_err = rc;
 602
 603                 if ( *mcp != NULL ) {
 604                         if ( binding ) {
 605                                 LDAP_BACK_CONN_BINDING_CLEAR( msc );
 606                         }
 607                         LDAP_BACK_CONN_TAINTED_SET( mc );
 608                         /* only release if mandatory; otherwise
 609                          * let the caller do what's best before
 610                          * releasing */
 611                         if ( META_BACK_ONERR_STOP( mi ) ) {
 612                                 meta_back_release_conn_lock( op, mc, 0 );
 613                                 *mcp = NULL;
 614                         }
 615                 }
 616
 617                 if ( sendok ) {
 618                         rs->sr_err = rc;
 619                         rs->sr_text = NULL;
 620                         send_ldap_result( op, rs );
 621                 }
 622         }
 623
 624         if ( META_BACK_QUARANTINE( mi ) ) {
 625                 meta_back_quarantine( op, rs, candidate, 0 );
 626         }
 627
 628         ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
 629
 630         return rc == LDAP_SUCCESS ? 1 : 0;
 631 }
 632
 633 /*
 634  * callback for unique candidate selection
 635  */
 636 static int
 637 meta_back_conn_cb( Operation *op, SlapReply *rs )
 638 {
 639         assert( op->o_tag == LDAP_REQ_SEARCH );
 640
 641         switch ( rs->sr_type ) {
 642         case REP_SEARCH:
 643                 ((long *)op->o_callback->sc_private)[0] = (long)op->o_private;
 644                 break;
 645
 646         case REP_SEARCHREF:
 647         case REP_RESULT:
 648                 break;
 649
 650         default:
 651                 return rs->sr_err;
 652         }
 653
 654         return 0;
 655 }
 656
 657
 658 static int
 659 meta_back_get_candidate(
 660         Operation       *op,
 661         SlapReply       *rs,
 662         struct berval   *ndn )
 663 {
 664         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 665         long            candidate;
 666
 667         /*
 668          * tries to get a unique candidate
 669          * (takes care of default target)
 670          */
 671         candidate = meta_back_select_unique_candidate( mi, ndn );
 672
 673         /*
 674          * if any is found, inits the connection
 675          */
 676         if ( candidate == META_TARGET_NONE ) {
 677                 rs->sr_err = LDAP_NO_SUCH_OBJECT;
 678                 rs->sr_text = "no suitable candidate target found";
 679
 680         } else if ( candidate == META_TARGET_MULTIPLE ) {
 681                 Filter          f = { 0 };
 682                 Operation       op2 = *op;
 683                 SlapReply       rs2 = { 0 };
 684                 slap_callback   cb2 = { 0 };
 685                 int             rc;
 686
 687                 /* try to get a unique match for the request ndn
 688                  * among the multiple candidates available */
 689                 op2.o_tag = LDAP_REQ_SEARCH;
 690                 op2.o_req_dn = *ndn;
 691                 op2.o_req_ndn = *ndn;
 692                 op2.ors_scope = LDAP_SCOPE_BASE;
 693                 op2.ors_deref = LDAP_DEREF_NEVER;
 694                 op2.ors_attrs = slap_anlist_no_attrs;
 695                 op2.ors_attrsonly = 0;
 696                 op2.ors_limit = NULL;
 697                 op2.ors_slimit = 1;
 698                 op2.ors_tlimit = SLAP_NO_LIMIT;
 699
 700                 f.f_choice = LDAP_FILTER_PRESENT;
 701                 f.f_desc = slap_schema.si_ad_objectClass;
 702                 op2.ors_filter = &f;
 703                 BER_BVSTR( &op2.ors_filterstr, "(objectClass=*)" );
 704
 705                 op2.o_callback = &cb2;
 706                 cb2.sc_response = meta_back_conn_cb;
 707                 cb2.sc_private = (void *)&candidate;
 708
 709                 rc = op->o_bd->be_search( &op2, &rs2 );
 710
 711                 switch ( rs2.sr_err ) {
 712                 case LDAP_SUCCESS:
 713                 default:
 714                         rs->sr_err = rs2.sr_err;
 715                         break;
 716
 717                 case LDAP_SIZELIMIT_EXCEEDED:
 718                         /* if multiple candidates can serve the operation,
 719                          * and a default target is defined, and it is
 720                          * a candidate, try using it (FIXME: YMMV) */
 721                         if ( mi->mi_defaulttarget != META_DEFAULT_TARGET_NONE
 722                                 && meta_back_is_candidate( &mi->mi_targets[ mi->mi_defaulttarget ]->mt_nsuffix,
 723                                                 mi->mi_targets[ mi->mi_defaulttarget ]->mt_scope,
 724                                                 mi->mi_targets[ mi->mi_defaulttarget ]->mt_subtree_exclude,
 725                                                 ndn, op->o_tag == LDAP_REQ_SEARCH ? op->ors_scope : LDAP_SCOPE_BASE ) )
 726                         {
 727                                 candidate = mi->mi_defaulttarget;
 728                                 rs->sr_err = LDAP_SUCCESS;
 729                                 rs->sr_text = NULL;
 730
 731                         } else {
 732                                 rs->sr_err = LDAP_UNWILLING_TO_PERFORM;
 733                                 rs->sr_text = "cannot select unique candidate target";
 734                         }
 735                         break;
 736                 }
 737
 738         } else {
 739                 rs->sr_err = LDAP_SUCCESS;
 740         }
 741
 742         return candidate;
 743 }
 744
 745 static void     *meta_back_candidates_dummy;
 746
 747 static void
 748 meta_back_candidates_keyfree(
 749         void            *key,
 750         void            *data )
 751 {
 752         metacandidates_t        *mc = (metacandidates_t *)data;
 753
 754         ber_memfree_x( mc->mc_candidates, NULL );
 755         ber_memfree_x( data, NULL );
 756 }
 757
 758 SlapReply *
 759 meta_back_candidates_get( Operation *op )
 760 {
 761         metainfo_t              *mi = ( metainfo_t * )op->o_bd->be_private;
 762         metacandidates_t        *mc;
 763
 764         if ( op->o_threadctx ) {
 765                 void            *data = NULL;
 766
 767                 ldap_pvt_thread_pool_getkey( op->o_threadctx,
 768                                 &meta_back_candidates_dummy, &data, NULL );
 769                 mc = (metacandidates_t *)data;
 770
 771         } else {
 772                 mc = mi->mi_candidates;
 773         }
 774
 775         if ( mc == NULL ) {
 776                 mc = ch_calloc( sizeof( metacandidates_t ), 1 );
 777                 mc->mc_ntargets = mi->mi_ntargets;
 778                 mc->mc_candidates = ch_calloc( sizeof( SlapReply ), mc->mc_ntargets );
 779                 if ( op->o_threadctx ) {
 780                         void            *data = NULL;
 781
 782                         data = (void *)mc;
 783                         ldap_pvt_thread_pool_setkey( op->o_threadctx,
 784                                         &meta_back_candidates_dummy, data,
 785                                         meta_back_candidates_keyfree );
 786
 787                 } else {
 788                         mi->mi_candidates = mc;
 789                 }
 790
 791         } else if ( mc->mc_ntargets < mi->mi_ntargets ) {
 792                 /* NOTE: in the future, may want to allow back-config
 793                  * to add/remove targets from back-meta... */
 794                 mc->mc_candidates = ch_realloc( mc->mc_candidates,
 795                                 sizeof( SlapReply ) * mi->mi_ntargets );
 796                 memset( &mc->mc_candidates[ mc->mc_ntargets ], 0,
 797                         sizeof( SlapReply ) * ( mi->mi_ntargets - mc->mc_ntargets ) );
 798                 mc->mc_ntargets = mi->mi_ntargets;
 799         }
 800
 801         return mc->mc_candidates;
 802 }
 803
 804 /*
 805  * meta_back_getconn
 806  *
 807  * Prepares the connection structure
 808  *
 809  * RATIONALE:
 810  *
 811  * - determine what DN is being requested:
 812  *
 813  *      op      requires candidate      checks
 814  *
 815  *      add     unique                  parent of o_req_ndn
 816  *      bind    unique^*[/all]          o_req_ndn [no check]
 817  *      compare unique^+                o_req_ndn
 818  *      delete  unique                  o_req_ndn
 819  *      modify  unique                  o_req_ndn
 820  *      search  any                     o_req_ndn
 821  *      modrdn  unique[, unique]        o_req_ndn[, orr_nnewSup]
 822  *
 823  * - for ops that require the candidate to be unique, in case of multiple
 824  *   occurrences an internal search with sizeLimit=1 is performed
 825  *   if a unique candidate can actually be determined.  If none is found,
 826  *   the operation aborts; if multiple are found, the default target
 827  *   is used if defined and candidate; otherwise the operation aborts.
 828  *
 829  * *^note: actually, the bind operation is handled much like a search;
 830  *   i.e. the bind is broadcast to all candidate targets.
 831  *
 832  * +^note: actually, the compare operation is handled much like a search;
 833  *   i.e. the compare is broadcast to all candidate targets, while checking
 834  *   that exactly none (noSuchObject) or one (TRUE/FALSE/UNDEFINED) is
 835  *   returned.
 836  */
 837 metaconn_t *
 838 meta_back_getconn(
 839         Operation               *op,
 840         SlapReply               *rs,
 841         int                     *candidate,
 842         ldap_back_send_t        sendok )
 843 {
 844         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 845         metaconn_t      *mc = NULL,
 846                         mc_curr = { 0 };
 847         int             cached = META_TARGET_NONE,
 848                         i = META_TARGET_NONE,
 849                         err = LDAP_SUCCESS,
 850                         new_conn = 0,
 851                         ncandidates = 0;
 852
 853
 854         meta_op_type    op_type = META_OP_REQUIRE_SINGLE;
 855         enum            {
 856                 META_DNTYPE_ENTRY,
 857                 META_DNTYPE_PARENT,
 858                 META_DNTYPE_NEWPARENT
 859                         } dn_type = META_DNTYPE_ENTRY;
 860         struct berval   ndn = op->o_req_ndn,
 861                         pndn;
 862
 863         SlapReply       *candidates = meta_back_candidates_get( op );
 864
 865         /* Internal searches are privileged and shared. So is root. */
 866         /* FIXME: there seem to be concurrency issues */
 867         if ( op->o_do_not_cache || be_isroot( op ) ) {
 868                 mc_curr.mc_local_ndn = op->o_bd->be_rootndn;
 869                 LDAP_BACK_CONN_ISPRIV_SET( &mc_curr );
 870                 mc_curr.mc_conn = LDAP_BACK_PCONN_SET( op );
 871
 872         } else {
 873                 mc_curr.mc_local_ndn = op->o_ndn;
 874
 875                 /* Explicit binds must not be shared */
 876                 if ( op->o_tag == LDAP_REQ_BIND || SLAP_IS_AUTHZ_BACKEND( op ) ) {
 877                         mc_curr.mc_conn = op->o_conn;
 878
 879                 } else {
 880                         mc_curr.mc_conn = LDAP_BACK_PCONN_SET( op );
 881                 }
 882         }
 883
 884         /* Explicit Bind requests always get their own conn */
 885         if ( !( sendok & LDAP_BACK_BINDING ) ) {
 886                 /* Searches for a metaconn in the avl tree */
 887 retry_lock:
 888                 ldap_pvt_thread_mutex_lock( &mi->mi_conninfo.lai_mutex );
 889                 mc = (metaconn_t *)avl_find( mi->mi_conninfo.lai_tree,
 890                         (caddr_t)&mc_curr, meta_back_conndn_cmp );
 891                 if ( mc ) {
 892                         if ( ( mi->mi_conn_ttl != 0 && op->o_time > mc->mc_create_time + mi->mi_conn_ttl )
 893                                 || ( mi->mi_idle_timeout != 0 && op->o_time > mc->mc_time + mi->mi_idle_timeout ) )
 894                         {
 895                                 /* don't let anyone else use this expired connection */
 896                                 (void)avl_delete( &mi->mi_conninfo.lai_tree,
 897                                         (caddr_t)mc, meta_back_conndnmc_cmp );
 898                                 LDAP_BACK_CONN_TAINTED_SET( mc );
 899
 900                                 Debug( LDAP_DEBUG_TRACE, "%s meta_back_getconn: mc=%p conn=%ld expired.\n",
 901                                         op->o_log_prefix, (void *)mc, LDAP_BACK_PCONN_ID( mc->mc_conn ) );
 902                         }
 903
 904                         /* Don't reuse connections while they're still binding */
 905                         if ( LDAP_BACK_CONN_BINDING( mc ) ) {
 906                                 ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
 907                                 ldap_pvt_thread_yield();
 908                                 goto retry_lock;
 909                         }
 910
 911                         mc->mc_refcnt++;
 912                 }
 913                 ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
 914         }
 915
 916         switch ( op->o_tag ) {
 917         case LDAP_REQ_ADD:
 918                 /* if we go to selection, the entry must not exist,
 919                  * and we must be able to resolve the parent */
 920                 dn_type = META_DNTYPE_PARENT;
 921                 dnParent( &ndn, &pndn );
 922                 break;
 923
 924         case LDAP_REQ_MODRDN:
 925                 /* if nnewSuperior is not NULL, it must resolve
 926                  * to the same candidate as the req_ndn */
 927                 if ( op->orr_nnewSup ) {
 928                         dn_type = META_DNTYPE_NEWPARENT;
 929                 }
 930                 break;
 931
 932         case LDAP_REQ_BIND:
 933                 /* if bound as rootdn, the backend must bind to all targets
 934                  * with the administrative identity */
 935                 if ( op->orb_method == LDAP_AUTH_SIMPLE && be_isroot_pw( op ) ) {
 936                         op_type = META_OP_REQUIRE_ALL;
 937                 }
 938                 break;
 939
 940         case LDAP_REQ_DELETE:
 941         case LDAP_REQ_MODIFY:
 942                 /* just a unique candidate */
 943                 break;
 944
 945         case LDAP_REQ_COMPARE:
 946         case LDAP_REQ_SEARCH:
 947                 /* allow multiple candidates for the searchBase */
 948                 op_type = META_OP_ALLOW_MULTIPLE;
 949                 break;
 950
 951         default:
 952                 /* right now, just break (exop?) */
 953                 break;
 954         }
 955
 956         /*
 957          * require all connections ...
 958          */
 959         if ( op_type == META_OP_REQUIRE_ALL ) {
 960
 961                 /* Looks like we didn't get a bind. Open a new session... */
 962                 if ( mc == NULL ) {
 963                         mc = metaconn_alloc( op );
 964                         mc->mc_conn = mc_curr.mc_conn;
 965                         ber_dupbv( &mc->mc_local_ndn, &mc_curr.mc_local_ndn );
 966                         new_conn = 1;
 967                         if ( sendok & LDAP_BACK_BINDING ) {
 968                                 LDAP_BACK_CONN_BINDING_SET( mc );
 969                         }
 970                 }
 971
 972                 for ( i = 0; i < mi->mi_ntargets; i++ ) {
 973                         /*
 974                          * The target is activated; if needed, it is
 975                          * also init'd
 976                          */
 977                         candidates[ i ].sr_err = meta_back_init_one_conn( op,
 978                                 rs, mc, i, LDAP_BACK_CONN_ISPRIV( &mc_curr ),
 979                                 sendok );
 980                         if ( candidates[ i ].sr_err == LDAP_SUCCESS ) {
 981                                 candidates[ i ].sr_tag = META_CANDIDATE;
 982                                 ncandidates++;
 983
 984                         } else {
 985
 986                                 /*
 987                                  * FIXME: in case one target cannot
 988                                  * be init'd, should the other ones
 989                                  * be tried?
 990                                  */
 991                                 candidates[ i ].sr_tag = META_NOT_CANDIDATE;
 992                                 err = candidates[ i ].sr_err;
 993                                 continue;
 994                         }
 995                 }
 996
 997                 if ( ncandidates == 0 ) {
 998                         if ( new_conn ) {
 999                                 meta_back_freeconn( op, mc );
1000
1001                         } else {
1002                                 meta_back_release_conn( op, mc );
1003                         }
1004
1005                         rs->sr_err = LDAP_NO_SUCH_OBJECT;
1006                         rs->sr_text = "Unable to select valid candidates";
1007
1008                         if ( sendok & LDAP_BACK_SENDERR ) {
1009                                 if ( rs->sr_err == LDAP_NO_SUCH_OBJECT ) {
1010                                         rs->sr_matched = op->o_bd->be_suffix[ 0 ].bv_val;
1011                                 }
1012                                 send_ldap_result( op, rs );
1013                                 rs->sr_text = NULL;
1014                                 rs->sr_matched = NULL;
1015                         }
1016
1017                         return NULL;
1018                 }
1019
1020                 goto done;
1021         }
1022
1023         /*
1024          * looks in cache, if any
1025          */
1026         if ( mi->mi_cache.ttl != META_DNCACHE_DISABLED ) {
1027                 cached = i = meta_dncache_get_target( &mi->mi_cache, &op->o_req_ndn );
1028         }
1029
1030         if ( op_type == META_OP_REQUIRE_SINGLE ) {
1031                 metatarget_t            *mt = NULL;
1032                 metasingleconn_t        *msc = NULL;
1033
1034                 int                     j;
1035
1036                 for ( j = 0; j < mi->mi_ntargets; j++ ) {
1037                         candidates[ j ].sr_tag = META_NOT_CANDIDATE;
1038                 }
1039
1040                 /*
1041                  * tries to get a unique candidate
1042                  * (takes care of default target)
1043                  */
1044                 if ( i == META_TARGET_NONE ) {
1045                         i = meta_back_get_candidate( op, rs, &ndn );
1046
1047                         if ( rs->sr_err == LDAP_NO_SUCH_OBJECT && dn_type == META_DNTYPE_PARENT ) {
1048                                 i = meta_back_get_candidate( op, rs, &pndn );
1049                         }
1050
1051                         if ( i < 0 || rs->sr_err != LDAP_SUCCESS ) {
1052                                 if ( mc != NULL ) {
1053                                         meta_back_release_conn( op, mc );
1054                                 }
1055
1056                                 if ( sendok & LDAP_BACK_SENDERR ) {
1057                                         if ( rs->sr_err == LDAP_NO_SUCH_OBJECT ) {
1058                                                 rs->sr_matched = op->o_bd->be_suffix[ 0 ].bv_val;
1059                                         }
1060                                         send_ldap_result( op, rs );
1061                                         rs->sr_text = NULL;
1062                                         rs->sr_matched = NULL;
1063                                 }
1064
1065                                 return NULL;
1066                         }
1067                 }
1068
1069                 if ( dn_type == META_DNTYPE_NEWPARENT && meta_back_get_candidate( op, rs, op->orr_nnewSup ) != i )
1070                 {
1071                         if ( mc != NULL ) {
1072                                 meta_back_release_conn( op, mc );
1073                         }
1074
1075                         rs->sr_err = LDAP_UNWILLING_TO_PERFORM;
1076                         rs->sr_text = "cross-target rename not supported";
1077                         if ( sendok & LDAP_BACK_SENDERR ) {
1078                                 send_ldap_result( op, rs );
1079                                 rs->sr_text = NULL;
1080                         }
1081
1082                         return NULL;
1083                 }
1084
1085                 Debug( LDAP_DEBUG_TRACE,
1086         "==>meta_back_getconn: got target=%d for ndn=\"%s\" from cache\n",
1087                                 i, op->o_req_ndn.bv_val, 0 );
1088
1089                 if ( mc == NULL ) {
1090                         /* Retries searching for a metaconn in the avl tree
1091                          * the reason is that the connection might have been
1092                          * created by meta_back_get_candidate() */
1093                         if ( !( sendok & LDAP_BACK_BINDING ) ) {
1094 retry_lock2:;
1095                                 ldap_pvt_thread_mutex_lock( &mi->mi_conninfo.lai_mutex );
1096                                 mc = (metaconn_t *)avl_find( mi->mi_conninfo.lai_tree,
1097                                         (caddr_t)&mc_curr, meta_back_conndn_cmp );
1098                                 if ( mc != NULL ) {
1099                                         /* Don't reuse connections while they're still binding */
1100                                         if ( LDAP_BACK_CONN_BINDING( mc ) ) {
1101                                                 ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
1102                                                 ldap_pvt_thread_yield();
1103                                                 goto retry_lock2;
1104                                         }
1105
1106                                         mc->mc_refcnt++;
1107                                 }
1108                                 ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
1109                         }
1110
1111                         /* Looks like we didn't get a bind. Open a new session... */
1112                         if ( mc == NULL ) {
1113                                 mc = metaconn_alloc( op );
1114                                 mc->mc_conn = mc_curr.mc_conn;
1115                                 ber_dupbv( &mc->mc_local_ndn, &mc_curr.mc_local_ndn );
1116                                 new_conn = 1;
1117                                 if ( sendok & LDAP_BACK_BINDING ) {
1118                                         LDAP_BACK_CONN_BINDING_SET( mc );
1119                                 }
1120                         }
1121                 }
1122
1123                 /*
1124                  * Clear all other candidates
1125                  */
1126                 ( void )meta_clear_unused_candidates( op, i );
1127
1128                 mt = mi->mi_targets[ i ];
1129                 msc = &mc->mc_conns[ i ];
1130
1131                 /*
1132                  * The target is activated; if needed, it is
1133                  * also init'd. In case of error, meta_back_init_one_conn
1134                  * sends the appropriate result.
1135                  */
1136                 err = meta_back_init_one_conn( op, rs, mc, i,
1137                         LDAP_BACK_CONN_ISPRIV( &mc_curr ), sendok );
1138                 if ( err != LDAP_SUCCESS ) {
1139                         /*
1140                          * FIXME: in case one target cannot
1141                          * be init'd, should the other ones
1142                          * be tried?
1143                          */
1144                         candidates[ i ].sr_tag = META_NOT_CANDIDATE;
1145                         if ( new_conn ) {
1146                                 (void)meta_clear_one_candidate( msc );
1147                                 meta_back_freeconn( op, mc );
1148
1149                         } else {
1150                                 meta_back_release_conn( op, mc );
1151                         }
1152                         return NULL;
1153                 }
1154
1155                 candidates[ i ].sr_err = LDAP_SUCCESS;
1156                 candidates[ i ].sr_tag = META_CANDIDATE;
1157                 ncandidates++;
1158
1159                 if ( candidate ) {
1160                         *candidate = i;
1161                 }
1162
1163         /*
1164          * if no unique candidate ...
1165          */
1166         } else {
1167
1168                 /* Looks like we didn't get a bind. Open a new session... */
1169                 if ( mc == NULL ) {
1170                         mc = metaconn_alloc( op );
1171                         mc->mc_conn = mc_curr.mc_conn;
1172                         ber_dupbv( &mc->mc_local_ndn, &mc_curr.mc_local_ndn );
1173                         new_conn = 1;
1174                         if ( sendok & LDAP_BACK_BINDING ) {
1175                                 LDAP_BACK_CONN_BINDING_SET( mc );
1176                         }
1177                 }
1178
1179                 for ( i = 0; i < mi->mi_ntargets; i++ ) {
1180                         metatarget_t            *mt = mi->mi_targets[ i ];
1181                         metasingleconn_t        *msc = &mc->mc_conns[ i ];
1182
1183                         if ( i == cached
1184                                 || meta_back_is_candidate( &mt->mt_nsuffix,
1185                                                 mt->mt_scope,
1186                                                 mt->mt_subtree_exclude,
1187                                                 &op->o_req_ndn,
1188                                                 LDAP_SCOPE_SUBTREE ) )
1189                         {
1190
1191                                 /*
1192                                  * The target is activated; if needed, it is
1193                                  * also init'd
1194                                  */
1195                                 int lerr = meta_back_init_one_conn( op, rs, mc, i,
1196                                         LDAP_BACK_CONN_ISPRIV( &mc_curr ), LDAP_BACK_DONTSEND );
1197                                 if ( lerr == LDAP_SUCCESS ) {
1198                                         candidates[ i ].sr_tag = META_CANDIDATE;
1199                                         candidates[ i ].sr_err = LDAP_SUCCESS;
1200                                         ncandidates++;
1201
1202                                         Debug( LDAP_DEBUG_TRACE, "%s: meta_back_getconn[%d]\n",
1203                                                 op->o_log_prefix, i, 0 );
1204
1205                                 } else {
1206
1207                                         /*
1208                                          * FIXME: in case one target cannot
1209                                          * be init'd, should the other ones
1210                                          * be tried?
1211                                          */
1212                                         if ( new_conn ) {
1213                                                 ( void )meta_clear_one_candidate( msc );
1214                                         }
1215                                         /* leave the target candidate, but record the error for later use */
1216                                         candidates[ i ].sr_err = lerr;
1217                                         err = lerr;
1218
1219                                         Debug( LDAP_DEBUG_ANY, "%s: meta_back_getconn[%d] failed: %d\n",
1220                                                 op->o_log_prefix, i, lerr );
1221
1222                                         if ( META_BACK_ONERR_STOP( mi ) ) {
1223                                                 if ( sendok & LDAP_BACK_SENDERR ) {
1224                                                         send_ldap_result( op, rs );
1225                                                         rs->sr_text = NULL;
1226                                                 }
1227                                                 if ( new_conn ) {
1228                                                         meta_back_freeconn( op, mc );
1229
1230                                                 } else {
1231                                                         meta_back_release_conn( op, mc );
1232                                                 }
1233
1234                                                 return NULL;
1235                                         }
1236
1237                                         rs->sr_text = NULL;
1238                                         continue;
1239                                 }
1240
1241                         } else {
1242                                 if ( new_conn ) {
1243                                         ( void )meta_clear_one_candidate( msc );
1244                                 }
1245                                 candidates[ i ].sr_tag = META_NOT_CANDIDATE;
1246                         }
1247                 }
1248
1249                 if ( ncandidates == 0 ) {
1250                         if ( new_conn ) {
1251                                 meta_back_freeconn( op, mc );
1252
1253                         } else {
1254                                 meta_back_release_conn( op, mc );
1255                         }
1256
1257                         rs->sr_err = LDAP_NO_SUCH_OBJECT;
1258                         rs->sr_text = "Unable to select valid candidates";
1259
1260                         if ( sendok & LDAP_BACK_SENDERR ) {
1261                                 if ( rs->sr_err == LDAP_NO_SUCH_OBJECT ) {
1262                                         rs->sr_matched = op->o_bd->be_suffix[ 0 ].bv_val;
1263                                 }
1264                                 send_ldap_result( op, rs );
1265                                 rs->sr_text = NULL;
1266                                 rs->sr_matched = NULL;
1267                         }
1268
1269                         return NULL;
1270                 }
1271         }
1272
1273 done:;
1274         /* clear out meta_back_init_one_conn non-fatal errors */
1275         rs->sr_err = LDAP_SUCCESS;
1276         rs->sr_text = NULL;
1277
1278         /* touch the timestamp */
1279         if ( mi->mi_idle_timeout != 0 ) {
1280                 mc->mc_time = op->o_time;
1281         }
1282
1283         if ( new_conn ) {
1284                 if ( mi->mi_conn_ttl ) {
1285                         mc->mc_create_time = op->o_time;
1286                 }
1287
1288                 /*
1289                  * Inserts the newly created metaconn in the avl tree
1290                  */
1291                 ldap_pvt_thread_mutex_lock( &mi->mi_conninfo.lai_mutex );
1292                 err = avl_insert( &mi->mi_conninfo.lai_tree, ( caddr_t )mc,
1293                                 meta_back_conndn_cmp, meta_back_conndn_dup );
1294
1295 #if PRINT_CONNTREE > 0
1296                 myprint( mi->mi_conninfo.lai_tree, "meta_back_getconn" );
1297 #endif /* PRINT_CONNTREE */
1298
1299                 ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
1300
1301                 /*
1302                  * Err could be -1 in case a duplicate metaconn is inserted
1303                  */
1304                 switch ( err ) {
1305                 case 0:
1306                         break;
1307
1308                 case -1:
1309                         if ( !( sendok & LDAP_BACK_BINDING ) ) {
1310                                 /* duplicate: free and try to get the newly created one */
1311                                 goto retry_lock;
1312                         }
1313                         LDAP_BACK_CONN_TAINTED_SET( mc );
1314                         break;
1315
1316                 default:
1317                         Debug( LDAP_DEBUG_ANY,
1318                                 "%s meta_back_getconn: candidates=%d conn=%ld insert failed\n",
1319                                 op->o_log_prefix, ncandidates,
1320                                 LDAP_BACK_PCONN_ID( mc->mc_conn ) );
1321
1322                         meta_back_freeconn( op, mc );
1323
1324                         rs->sr_err = LDAP_OTHER;
1325                         rs->sr_text = "proxy bind collision";
1326                         if ( sendok & LDAP_BACK_SENDERR ) {
1327                                 send_ldap_result( op, rs );
1328                                 rs->sr_text = NULL;
1329                         }
1330                         return NULL;
1331                 }
1332
1333                 Debug( LDAP_DEBUG_TRACE,
1334                         "%s meta_back_getconn: candidates=%d conn=%ld inserted\n",
1335                         op->o_log_prefix, ncandidates,
1336                         LDAP_BACK_PCONN_ID( mc->mc_conn ) );
1337
1338         } else {
1339                 Debug( LDAP_DEBUG_TRACE,
1340                         "%s meta_back_getconn: candidates=%d conn=%ld fetched\n",
1341                         op->o_log_prefix, ncandidates,
1342                         LDAP_BACK_PCONN_ID( mc->mc_conn ) );
1343         }
1344
1345         return mc;
1346 }
1347
1348 void
1349 meta_back_release_conn_lock(
1350         Operation               *op,
1351         metaconn_t              *mc,
1352         int                     dolock )
1353 {
1354         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
1355
1356         assert( mc != NULL );
1357
1358         if ( dolock ) {
1359                 ldap_pvt_thread_mutex_lock( &mi->mi_conninfo.lai_mutex );
1360         }
1361         assert( mc->mc_refcnt > 0 );
1362         mc->mc_refcnt--;
1363         LDAP_BACK_CONN_BINDING_CLEAR( mc );
1364         if ( LDAP_BACK_CONN_TAINTED( mc ) ) {
1365                 Debug( LDAP_DEBUG_TRACE, "%s meta_back_release_conn: mc=%p conn=%ld expired.\n",
1366                         op->o_log_prefix, (void *)mc, LDAP_BACK_PCONN_ID( mc->mc_conn ) );
1367                 (void)avl_delete( &mi->mi_conninfo.lai_tree,
1368                         ( caddr_t )mc, meta_back_conndnmc_cmp );
1369                 if ( mc->mc_refcnt == 0 ) {
1370                         meta_clear_candidates( op, mc );
1371                         meta_back_conn_free( mc );
1372                 }
1373         }
1374         if ( dolock ) {
1375                 ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
1376         }
1377 }
1378
1379 void
1380 meta_back_quarantine(
1381         Operation       *op,
1382         SlapReply       *rs,
1383         int             candidate,
1384         int             dolock )
1385 {
1386         metainfo_t              *mi = (metainfo_t *)op->o_bd->be_private;
1387         metatarget_t            *mt = mi->mi_targets[ candidate ];
1388
1389         slap_retry_info_t       *ri = &mt->mt_quarantine;
1390
1391         if ( dolock ) {
1392                 ldap_pvt_thread_mutex_lock( &mi->mi_conninfo.lai_mutex );
1393         }
1394
1395         if ( rs->sr_err == LDAP_UNAVAILABLE ) {
1396                 switch ( mt->mt_isquarantined ) {
1397                 case LDAP_BACK_FQ_NO:
1398                         Debug( LDAP_DEBUG_ANY,
1399                                 "%s: meta_back_quarantine enter.\n",
1400                                 op->o_log_prefix, 0, 0 );
1401
1402                         ri->ri_idx = 0;
1403                         ri->ri_count = 0;
1404                         break;
1405
1406                 case LDAP_BACK_FQ_RETRYING:
1407                         Debug( LDAP_DEBUG_ANY,
1408                                 "%s: meta_back_quarantine block #%d try #%d failed.\n",
1409                                 op->o_log_prefix, ri->ri_idx, ri->ri_count );
1410
1411                         ++ri->ri_count;
1412                         if ( ri->ri_num[ ri->ri_idx ] != SLAP_RETRYNUM_FOREVER
1413                                 && ri->ri_count == ri->ri_num[ ri->ri_idx ] )
1414                         {
1415                                 ri->ri_count = 0;
1416                                 ++ri->ri_idx;
1417                         }
1418                         break;
1419
1420                 default:
1421                         break;
1422                 }
1423
1424                 mt->mt_isquarantined = LDAP_BACK_FQ_YES;
1425                 ri->ri_last = slap_get_time();
1426
1427         } else if ( mt->mt_isquarantined != LDAP_BACK_FQ_NO ) {
1428                 Debug( LDAP_DEBUG_ANY,
1429                         "%s: meta_back_quarantine exit.\n",
1430                         op->o_log_prefix, ri->ri_idx, ri->ri_count );
1431
1432                 ri->ri_count = 0;
1433                 ri->ri_idx = 0;
1434                 mt->mt_isquarantined = LDAP_BACK_FQ_NO;
1435         }
1436
1437         if ( dolock ) {
1438                 ldap_pvt_thread_mutex_unlock( &mi->mi_conninfo.lai_mutex );
1439         }
1440 }
1441