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-2005 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 #define PRINT_CONNTREE 0
  41
  42 /*
  43  * meta_back_conn_cmp
  44  *
  45  * compares two struct metaconn based on the value of the conn pointer;
  46  * used by avl stuff
  47  */
  48 int
  49 meta_back_conn_cmp(
  50         const void *c1,
  51         const void *c2 )
  52 {
  53         metaconn_t      *mc1 = ( metaconn_t * )c1;
  54         metaconn_t      *mc2 = ( metaconn_t * )c2;
  55
  56         return SLAP_PTRCMP( mc1->mc_conn, mc2->mc_conn );
  57 }
  58
  59 /*
  60  * meta_back_conn_dup
  61  *
  62  * returns -1 in case a duplicate struct metaconn has been inserted;
  63  * used by avl stuff
  64  */
  65 int
  66 meta_back_conn_dup(
  67         void *c1,
  68         void *c2 )
  69 {
  70         metaconn_t      *mc1 = ( metaconn_t * )c1;
  71         metaconn_t      *mc2 = ( metaconn_t * )c2;
  72
  73         return( ( mc1->mc_conn == mc2->mc_conn ) ? -1 : 0 );
  74 }
  75
  76 /*
  77  * Debug stuff (got it from libavl)
  78  */
  79 #if PRINT_CONNTREE > 0
  80 static void
  81 ravl_print( Avlnode *root, int depth )
  82 {
  83         int     i;
  84
  85         if ( root == 0 ) {
  86                 return;
  87         }
  88
  89         ravl_print( root->avl_right, depth + 1 );
  90
  91         for ( i = 0; i < depth; i++ ) {
  92                 printf( "    " );
  93         }
  94
  95         printf( "c(%d) %d\n", ( ( metaconn_t * )root->avl_data )->mc_conn->c_connid, root->avl_bf );
  96
  97         ravl_print( root->avl_left, depth + 1 );
  98 }
  99
 100 static void
 101 myprint( Avlnode *root )
 102 {
 103         printf( "********\n" );
 104
 105         if ( root == 0 ) {
 106                 printf( "\tNULL\n" );
 107         } else {
 108                 ravl_print( root, 0 );
 109         }
 110
 111         printf( "********\n" );
 112 }
 113 #endif /* PRINT_CONNTREE */
 114 /*
 115  * End of debug stuff
 116  */
 117
 118 /*
 119  * metaconn_alloc
 120  *
 121  * Allocates a connection structure, making room for all the referenced targets
 122  */
 123 static metaconn_t *
 124 metaconn_alloc(
 125         Operation               *op )
 126 {
 127         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 128         metaconn_t      *mc;
 129         int             i, ntargets = mi->mi_ntargets;
 130
 131         assert( ntargets > 0 );
 132
 133         /* malloc all in one */
 134         mc = ( metaconn_t * )ch_malloc( sizeof( metaconn_t )
 135                         + sizeof( metasingleconn_t ) * ntargets );
 136         if ( mc == NULL ) {
 137                 return NULL;
 138         }
 139
 140         for ( i = 0; i < ntargets; i++ ) {
 141                 mc->mc_conns[ i ].msc_ld = NULL;
 142                 BER_BVZERO( &mc->mc_conns[ i ].msc_bound_ndn );
 143                 BER_BVZERO( &mc->mc_conns[ i ].msc_cred );
 144                 mc->mc_conns[ i ].msc_bound = META_UNBOUND;
 145                 mc->mc_conns[ i ].msc_info = mi;
 146         }
 147
 148         mc->mc_auth_target = META_BOUND_NONE;
 149         ldap_pvt_thread_mutex_init( &mc->mc_mutex );
 150         mc->mc_refcnt = 1;
 151
 152         return mc;
 153 }
 154
 155 /*
 156  * meta_back_conn_free
 157  *
 158  * clears a metaconn
 159  */
 160
 161 void
 162 meta_back_conn_free(
 163         metaconn_t      *mc )
 164 {
 165         assert( mc != NULL );
 166         assert( mc->mc_refcnt == 0 );
 167
 168         ldap_pvt_thread_mutex_destroy( &mc->mc_mutex );
 169         free( mc );
 170 }
 171
 172 static void
 173 meta_back_freeconn(
 174         Operation       *op,
 175         metaconn_t      *mc )
 176 {
 177         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 178
 179         assert( mc != NULL );
 180
 181         ldap_pvt_thread_mutex_lock( &mi->mi_conn_mutex );
 182
 183         if ( --mc->mc_refcnt == 0 ) {
 184                 meta_back_conn_free( mc );
 185         }
 186
 187         ldap_pvt_thread_mutex_unlock( &mi->mi_conn_mutex );
 188 }
 189
 190 /*
 191  * meta_back_init_one_conn
 192  *
 193  * Initializes one connection
 194  */
 195 int
 196 meta_back_init_one_conn(
 197         Operation               *op,
 198         SlapReply               *rs,
 199         metatarget_t            *mt,
 200         metasingleconn_t        *msc,
 201         ldap_back_send_t        sendok )
 202 {
 203         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 204         int             vers;
 205         dncookie        dc;
 206
 207         /*
 208          * Already init'ed
 209          */
 210         if ( msc->msc_ld != NULL ) {
 211                 return rs->sr_err = LDAP_SUCCESS;
 212         }
 213
 214         /*
 215          * Attempts to initialize the connection to the target ds
 216          */
 217         rs->sr_err = ldap_initialize( &msc->msc_ld, mt->mt_uri );
 218         if ( rs->sr_err != LDAP_SUCCESS ) {
 219                 goto error_return;
 220         }
 221
 222         /*
 223          * Set LDAP version. This will always succeed: If the client
 224          * bound with a particular version, then so can we.
 225          */
 226         vers = op->o_conn->c_protocol;
 227         ldap_set_option( msc->msc_ld, LDAP_OPT_PROTOCOL_VERSION, &vers );
 228
 229         /* automatically chase referrals ("chase-referrals"/"dont-chase-referrals" statement) */
 230         if ( LDAP_BACK_CHASE_REFERRALS( mi ) ) {
 231                 ldap_set_option( msc->msc_ld, LDAP_OPT_REFERRALS, LDAP_OPT_ON );
 232         }
 233
 234 #ifdef HAVE_TLS
 235         /* start TLS ("tls [try-]{start|propagate}" statement) */
 236         if ( ( LDAP_BACK_USE_TLS( mi ) || ( op->o_conn->c_is_tls && LDAP_BACK_PROPAGATE_TLS( mi ) ) )
 237                         && !ldap_is_ldaps_url( mt->mt_uri ) )
 238         {
 239 #ifdef SLAP_STARTTLS_ASYNCHRONOUS
 240                 /*
 241                  * use asynchronous StartTLS; in case, chase referral
 242                  * FIXME: OpenLDAP does not return referral on StartTLS yet
 243                  */
 244                 int             msgid;
 245
 246                 rs->sr_err = ldap_start_tls( msc->msc_ld, NULL, NULL, &msgid );
 247                 if ( rs->sr_err == LDAP_SUCCESS ) {
 248                         LDAPMessage     *res = NULL;
 249                         int             rc, nretries = mt->mt_nretries;
 250                         struct timeval  tv = { 0, 0 };
 251
 252 retry:;
 253                         rc = ldap_result( msc->msc_ld, msgid, LDAP_MSG_ALL, &tv, &res );
 254                         if ( rc < 0 ) {
 255                                 rs->sr_err = LDAP_OTHER;
 256
 257                         } else if ( rc == 0 ) {
 258                                 if ( nretries != 0 ) {
 259                                         if ( nretries > 0 ) {
 260                                                 nretries--;
 261                                         }
 262                                         tv.tv_sec = 0;
 263                                         tv.tv_usec = 100000;
 264                                         goto retry;
 265                                 }
 266                                 rs->sr_err = LDAP_OTHER;
 267
 268                         } else if ( rc == LDAP_RES_EXTENDED ) {
 269                                 struct berval   *data = NULL;
 270
 271                                 rs->sr_err = ldap_parse_extended_result( msc->msc_ld, res,
 272                                                 NULL, &data, 0 );
 273                                 if ( rs->sr_err == LDAP_SUCCESS ) {
 274                                         rs->sr_err = ldap_result2error( msc->msc_ld, res, 1 );
 275                                         res = NULL;
 276
 277                                         /* FIXME: in case a referral
 278                                          * is returned, should we try
 279                                          * using it instead of the
 280                                          * configured URI? */
 281                                         if ( rs->sr_err == LDAP_SUCCESS ) {
 282                                                 ldap_install_tls( msc->msc_ld );
 283
 284                                         } else if ( rs->sr_err == LDAP_REFERRAL ) {
 285                                                 rs->sr_err = LDAP_OTHER;
 286                                                 rs->sr_text = "unwilling to chase referral returned by Start TLS exop";
 287                                         }
 288
 289                                         if ( data ) {
 290                                                 if ( data->bv_val ) {
 291                                                         ber_memfree( data->bv_val );
 292                                                 }
 293                                                 ber_memfree( data );
 294                                         }
 295                                 }
 296
 297                         } else {
 298                                 rs->sr_err = LDAP_OTHER;
 299                         }
 300
 301                         if ( res != NULL ) {
 302                                 ldap_msgfree( res );
 303                         }
 304                 }
 305 #else /* ! SLAP_STARTTLS_ASYNCHRONOUS */
 306                 /*
 307                  * use synchronous StartTLS
 308                  */
 309                 rs->sr_err = ldap_start_tls_s( msc->msc_ld, NULL, NULL );
 310 #endif /* ! SLAP_STARTTLS_ASYNCHRONOUS */
 311
 312                 /* if StartTLS is requested, only attempt it if the URL
 313                  * is not "ldaps://"; this may occur not only in case
 314                  * of misconfiguration, but also when used in the chain
 315                  * overlay, where the "uri" can be parsed out of a referral */
 316                 if ( rs->sr_err == LDAP_SERVER_DOWN
 317                                 || ( rs->sr_err != LDAP_SUCCESS && LDAP_BACK_TLS_CRITICAL( mi ) ) )
 318                 {
 319                         ldap_unbind_ext_s( msc->msc_ld, NULL, NULL );
 320                         goto error_return;
 321                 }
 322         }
 323 #endif /* HAVE_TLS */
 324
 325         /*
 326          * Set the network timeout if set
 327          */
 328         if ( mi->mi_network_timeout != 0 ) {
 329                 struct timeval  network_timeout;
 330
 331                 network_timeout.tv_usec = 0;
 332                 network_timeout.tv_sec = mi->mi_network_timeout;
 333
 334                 ldap_set_option( msc->msc_ld, LDAP_OPT_NETWORK_TIMEOUT,
 335                                 (void *)&network_timeout );
 336         }
 337
 338         /*
 339          * If the connection DN is not null, an attempt to rewrite it is made
 340          */
 341         if ( !BER_BVISEMPTY( &op->o_conn->c_dn ) ) {
 342                 dc.target = mt;
 343                 dc.conn = op->o_conn;
 344                 dc.rs = rs;
 345                 dc.ctx = "bindDN";
 346
 347                 /*
 348                  * Rewrite the bind dn if needed
 349                  */
 350                 if ( ldap_back_dn_massage( &dc, &op->o_conn->c_dn,
 351                                         &msc->msc_bound_ndn ) )
 352                 {
 353                         goto error_return;
 354                 }
 355
 356                 /* copy the DN idf needed */
 357                 if ( msc->msc_bound_ndn.bv_val == op->o_conn->c_dn.bv_val ) {
 358                         ber_dupbv( &msc->msc_bound_ndn, &op->o_conn->c_dn );
 359                 }
 360
 361                 assert( !BER_BVISNULL( &msc->msc_bound_ndn ) );
 362
 363         } else {
 364                 ber_str2bv( "", 0, 1, &msc->msc_bound_ndn );
 365         }
 366
 367         msc->msc_bound = META_UNBOUND;
 368
 369 error_return:;
 370         if ( rs->sr_err == LDAP_SUCCESS ) {
 371                 /*
 372                  * Sets a cookie for the rewrite session
 373                  */
 374                 ( void )rewrite_session_init( mt->mt_rwmap.rwm_rw, op->o_conn );
 375
 376         } else {
 377                 rs->sr_err = slap_map_api2result( rs );
 378                 if ( sendok & LDAP_BACK_SENDERR ) {
 379                         send_ldap_result( op, rs );
 380                         rs->sr_text = NULL;
 381                 }
 382         }
 383
 384         return rs->sr_err;
 385 }
 386
 387 /*
 388  * meta_back_retry
 389  *
 390  * Retries one connection
 391  */
 392 int
 393 meta_back_retry(
 394         Operation               *op,
 395         SlapReply               *rs,
 396         metaconn_t              *mc,
 397         int                     candidate,
 398         ldap_back_send_t        sendok )
 399 {
 400         metainfo_t              *mi = ( metainfo_t * )op->o_bd->be_private;
 401         metatarget_t            *mt = &mi->mi_targets[ candidate ];
 402         int                     rc = LDAP_UNAVAILABLE;
 403         metasingleconn_t        *msc = &mc->mc_conns[ candidate ];
 404
 405 retry_lock:;
 406         ldap_pvt_thread_mutex_lock( &mi->mi_conn_mutex );
 407
 408         assert( mc->mc_refcnt > 0 );
 409
 410         if ( mc->mc_refcnt == 1 ) {
 411                 while ( ldap_pvt_thread_mutex_trylock( &mc->mc_mutex ) ) {
 412                         ldap_pvt_thread_mutex_unlock( &mi->mi_conn_mutex );
 413                         ldap_pvt_thread_yield();
 414                         goto retry_lock;
 415                 }
 416
 417                 ldap_unbind_ext_s( msc->msc_ld, NULL, NULL );
 418                 msc->msc_ld = NULL;
 419                 msc->msc_bound = 0;
 420
 421                 ( void )rewrite_session_delete( mt->mt_rwmap.rwm_rw, op->o_conn );
 422
 423                 /* mc here must be the regular mc, reset and ready for init */
 424                 rc = meta_back_init_one_conn( op, rs, mt, msc, sendok );
 425
 426                 if ( rc == LDAP_SUCCESS ) {
 427                         rc = meta_back_single_dobind( op, rs, mc, candidate,
 428                                         sendok, mt->mt_nretries, 0 );
 429                 }
 430
 431                 ldap_pvt_thread_mutex_unlock( &mc->mc_mutex );
 432         }
 433
 434         ldap_pvt_thread_mutex_unlock( &mi->mi_conn_mutex );
 435
 436         return rc == LDAP_SUCCESS ? 1 : 0;
 437 }
 438
 439 /*
 440  * callback for unique candidate selection
 441  */
 442 static int
 443 meta_back_conn_cb( Operation *op, SlapReply *rs )
 444 {
 445         assert( op->o_tag == LDAP_REQ_SEARCH );
 446
 447         switch ( rs->sr_type ) {
 448         case REP_SEARCH:
 449                 ((int *)op->o_callback->sc_private)[0] = (int)op->o_private;
 450                 break;
 451
 452         case REP_SEARCHREF:
 453         case REP_RESULT:
 454                 break;
 455
 456         default:
 457                 return rs->sr_err;
 458         }
 459
 460         return 0;
 461 }
 462
 463
 464 static int
 465 meta_back_get_candidate(
 466         Operation       *op,
 467         SlapReply       *rs,
 468         struct berval   *ndn )
 469 {
 470         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 471         int             candidate;
 472
 473         /*
 474          * tries to get a unique candidate
 475          * (takes care of default target)
 476          */
 477         candidate = meta_back_select_unique_candidate( mi, ndn );
 478
 479         /*
 480          * if any is found, inits the connection
 481          */
 482         if ( candidate == META_TARGET_NONE ) {
 483                 rs->sr_err = LDAP_NO_SUCH_OBJECT;
 484                 rs->sr_text = "no suitable candidate target found";
 485
 486         } else if ( candidate == META_TARGET_MULTIPLE ) {
 487                 Filter          f = { 0 };
 488                 Operation       op2 = *op;
 489                 SlapReply       rs2 = { 0 };
 490                 slap_callback   cb2 = { 0 };
 491                 int             rc;
 492
 493                 /* try to get a unique match for the request ndn
 494                  * among the multiple candidates available */
 495                 op2.o_tag = LDAP_REQ_SEARCH;
 496                 op2.o_req_dn = *ndn;
 497                 op2.o_req_ndn = *ndn;
 498                 op2.ors_scope = LDAP_SCOPE_BASE;
 499                 op2.ors_deref = LDAP_DEREF_NEVER;
 500                 op2.ors_attrs = slap_anlist_no_attrs;
 501                 op2.ors_attrsonly = 0;
 502                 op2.ors_limit = NULL;
 503                 op2.ors_slimit = 1;
 504                 op2.ors_tlimit = SLAP_NO_LIMIT;
 505
 506                 f.f_choice = LDAP_FILTER_PRESENT;
 507                 f.f_desc = slap_schema.si_ad_objectClass;
 508                 op2.ors_filter = &f;
 509                 BER_BVSTR( &op2.ors_filterstr, "(objectClass=*)" );
 510
 511                 op2.o_callback = &cb2;
 512                 cb2.sc_response = meta_back_conn_cb;
 513                 cb2.sc_private = (void *)&candidate;
 514
 515                 rc = op->o_bd->be_search( &op2, &rs2 );
 516
 517                 switch ( rs2.sr_err ) {
 518                 case LDAP_SUCCESS:
 519                 default:
 520                         rs->sr_err = rs2.sr_err;
 521                         break;
 522
 523                 case LDAP_SIZELIMIT_EXCEEDED:
 524                         /* if multiple candidates can serve the operation,
 525                          * and a default target is defined, and it is
 526                          * a candidate, try using it (FIXME: YMMV) */
 527                         if ( mi->mi_defaulttarget != META_DEFAULT_TARGET_NONE
 528                                 && meta_back_is_candidate( &mi->mi_targets[ mi->mi_defaulttarget ].mt_nsuffix,
 529                                                 mi->mi_targets[ mi->mi_defaulttarget ].mt_scope,
 530                                                 ndn, op->o_tag == LDAP_REQ_SEARCH ? op->ors_scope : LDAP_SCOPE_BASE ) )
 531                         {
 532                                 candidate = mi->mi_defaulttarget;
 533                                 rs->sr_err = LDAP_SUCCESS;
 534                                 rs->sr_text = NULL;
 535
 536                         } else {
 537                                 rs->sr_err = LDAP_UNWILLING_TO_PERFORM;
 538                                 rs->sr_text = "cannot select unique candidate target";
 539                         }
 540                         break;
 541                 }
 542
 543         } else {
 544                 rs->sr_err = LDAP_SUCCESS;
 545         }
 546
 547         return candidate;
 548 }
 549
 550 static void     *meta_back_candidates_dummy;
 551
 552 static void
 553 meta_back_candidates_keyfree(
 554         void            *key,
 555         void            *data )
 556 {
 557         metacandidates_t        *mc = (metacandidates_t *)data;
 558
 559         ber_memfree_x( mc->mc_candidates, NULL );
 560         ber_memfree_x( data, NULL );
 561 }
 562
 563 SlapReply *
 564 meta_back_candidates_get( Operation *op )
 565 {
 566         metainfo_t              *mi = ( metainfo_t * )op->o_bd->be_private;
 567         metacandidates_t        *mc;
 568
 569         if ( op->o_threadctx ) {
 570                 void            *data = NULL;
 571
 572                 ldap_pvt_thread_pool_getkey( op->o_threadctx,
 573                                 &meta_back_candidates_dummy, &data, NULL );
 574                 mc = (metacandidates_t *)data;
 575
 576         } else {
 577                 mc = mi->mi_candidates;
 578         }
 579
 580         if ( mc == NULL ) {
 581                 mc = ch_calloc( sizeof( metacandidates_t ), 1 );
 582                 mc->mc_ntargets = mi->mi_ntargets;
 583                 mc->mc_candidates = ch_calloc( sizeof( SlapReply ), mc->mc_ntargets );
 584                 if ( op->o_threadctx ) {
 585                         void            *data = NULL;
 586
 587                         data = (void *)mc;
 588                         ldap_pvt_thread_pool_setkey( op->o_threadctx,
 589                                         &meta_back_candidates_dummy, data,
 590                                         meta_back_candidates_keyfree );
 591
 592                 } else {
 593                         mi->mi_candidates = mc;
 594                 }
 595
 596         } else if ( mc->mc_ntargets < mi->mi_ntargets ) {
 597                 /* NOTE: in the future, may want to allow back-config
 598                  * to add/remove targets from back-meta... */
 599                 mc->mc_ntargets = mi->mi_ntargets;
 600                 mc->mc_candidates = ch_realloc( mc->mc_candidates,
 601                                 sizeof( SlapReply ) * mc->mc_ntargets );
 602         }
 603
 604         return mc->mc_candidates;
 605 }
 606
 607 /*
 608  * meta_back_getconn
 609  *
 610  * Prepares the connection structure
 611  *
 612  * RATIONALE:
 613  *
 614  * - determine what DN is being requested:
 615  *
 616  *      op      requires candidate      checks
 617  *
 618  *      add     unique                  parent of o_req_ndn
 619  *      bind    unique^*[/all]          o_req_ndn [no check]
 620  *      compare unique^+                o_req_ndn
 621  *      delete  unique                  o_req_ndn
 622  *      modify  unique                  o_req_ndn
 623  *      search  any                     o_req_ndn
 624  *      modrdn  unique[, unique]        o_req_ndn[, orr_nnewSup]
 625  *
 626  * - for ops that require the candidate to be unique, in case of multiple
 627  *   occurrences an internal search with sizeLimit=1 is performed
 628  *   if a unique candidate can actually be determined.  If none is found,
 629  *   the operation aborts; if multiple are found, the default target
 630  *   is used if defined and candidate; otherwise the operation aborts.
 631  *
 632  * *^note: actually, the bind operation is handled much like a search;
 633  *   i.e. the bind is broadcast to all candidate targets.
 634  *
 635  * +^note: actually, the compare operation is handled much like a search;
 636  *   i.e. the compare is broadcast to all candidate targets, while checking
 637  *   that exactly none (noSuchObject) or one (TRUE/FALSE/UNDEFINED) is
 638  *   returned.
 639  */
 640 metaconn_t *
 641 meta_back_getconn(
 642         Operation               *op,
 643         SlapReply               *rs,
 644         int                     *candidate,
 645         ldap_back_send_t        sendok )
 646 {
 647         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
 648         metaconn_t      *mc = NULL,
 649                         mc_curr = { 0 };
 650         int             cached = META_TARGET_NONE,
 651                         i = META_TARGET_NONE,
 652                         err = LDAP_SUCCESS,
 653                         new_conn = 0,
 654                         ncandidates = 0;
 655
 656
 657         meta_op_type    op_type = META_OP_REQUIRE_SINGLE;
 658         int             parent = 0,
 659                         newparent = 0;
 660         struct berval   ndn = op->o_req_ndn,
 661                         pndn;
 662
 663         SlapReply       *candidates = meta_back_candidates_get( op );
 664
 665         /* Searches for a metaconn in the avl tree */
 666         mc_curr.mc_conn = op->o_conn;
 667         ldap_pvt_thread_mutex_lock( &mi->mi_conn_mutex );
 668         mc = (metaconn_t *)avl_find( mi->mi_conntree,
 669                 (caddr_t)&mc_curr, meta_back_conn_cmp );
 670         if ( mc ) {
 671                 mc->mc_refcnt++;
 672         }
 673         ldap_pvt_thread_mutex_unlock( &mi->mi_conn_mutex );
 674
 675         switch ( op->o_tag ) {
 676         case LDAP_REQ_ADD:
 677                 /* if we go to selection, the entry must not exist,
 678                  * and we must be able to resolve the parent */
 679                 parent = 1;
 680                 dnParent( &ndn, &pndn );
 681                 break;
 682
 683         case LDAP_REQ_MODRDN:
 684                 /* if nnewSuperior is not NULL, it must resolve
 685                  * to the same candidate as the req_ndn */
 686                 if ( op->orr_nnewSup ) {
 687                         newparent = 1;
 688                 }
 689                 break;
 690
 691         case LDAP_REQ_BIND:
 692                 /* if bound as rootdn, the backend must bind to all targets
 693                  * with the administrative identity */
 694                 if ( op->orb_method == LDAP_AUTH_SIMPLE && be_isroot_pw( op ) ) {
 695                         op_type = META_OP_REQUIRE_ALL;
 696                 }
 697                 break;
 698
 699         case LDAP_REQ_DELETE:
 700         case LDAP_REQ_MODIFY:
 701                 /* just a unique candidate */
 702                 break;
 703
 704         case LDAP_REQ_COMPARE:
 705         case LDAP_REQ_SEARCH:
 706                 /* allow multiple candidates for the searchBase */
 707                 op_type = META_OP_ALLOW_MULTIPLE;
 708                 break;
 709
 710         default:
 711                 /* right now, just break (exop?) */
 712                 break;
 713         }
 714
 715         /*
 716          * require all connections ...
 717          */
 718         if ( op_type == META_OP_REQUIRE_ALL ) {
 719
 720                 /* Looks like we didn't get a bind. Open a new session... */
 721                 if ( mc == NULL ) {
 722                         mc = metaconn_alloc( op );
 723                         mc->mc_conn = op->o_conn;
 724                         new_conn = 1;
 725                 }
 726
 727                 for ( i = 0; i < mi->mi_ntargets; i++ ) {
 728
 729                         /*
 730                          * The target is activated; if needed, it is
 731                          * also init'd
 732                          */
 733                         candidates[ i ].sr_err =
 734                                 meta_back_init_one_conn( op, rs,
 735                                                 &mi->mi_targets[ i ],
 736                                                 &mc->mc_conns[ i ], sendok );
 737                         if ( candidates[ i ].sr_err == LDAP_SUCCESS ) {
 738                                 candidates[ i ].sr_tag = META_CANDIDATE;
 739                                 ncandidates++;
 740
 741                         } else {
 742
 743                                 /*
 744                                  * FIXME: in case one target cannot
 745                                  * be init'd, should the other ones
 746                                  * be tried?
 747                                  */
 748                                 candidates[ i ].sr_tag = META_NOT_CANDIDATE;
 749                                 err = candidates[ i ].sr_err;
 750                                 continue;
 751                         }
 752                 }
 753
 754                 if ( ncandidates == 0 ) {
 755                         if ( new_conn ) {
 756                                 meta_back_freeconn( op, mc );
 757
 758                         } else {
 759                                 meta_back_release_conn( op, mc );
 760                         }
 761
 762                         rs->sr_err = LDAP_NO_SUCH_OBJECT;
 763                         rs->sr_text = "Unable to select valid candidates";
 764
 765                         if ( sendok & LDAP_BACK_SENDERR ) {
 766                                 if ( rs->sr_err == LDAP_NO_SUCH_OBJECT ) {
 767                                         rs->sr_matched = op->o_bd->be_suffix[ 0 ].bv_val;
 768                                 }
 769                                 send_ldap_result( op, rs );
 770                                 rs->sr_text = NULL;
 771                                 rs->sr_matched = NULL;
 772                         }
 773
 774                         return NULL;
 775                 }
 776
 777                 goto done;
 778         }
 779
 780         /*
 781          * looks in cache, if any
 782          */
 783         if ( mi->mi_cache.ttl != META_DNCACHE_DISABLED ) {
 784                 cached = i = meta_dncache_get_target( &mi->mi_cache, &op->o_req_ndn );
 785         }
 786
 787         if ( op_type == META_OP_REQUIRE_SINGLE ) {
 788                 int     j;
 789
 790                 for ( j = 0; j < mi->mi_ntargets; j++ ) {
 791                         candidates[ j ].sr_tag = META_NOT_CANDIDATE;
 792                 }
 793
 794                 /*
 795                  * tries to get a unique candidate
 796                  * (takes care of default target)
 797                  */
 798                 if ( i == META_TARGET_NONE ) {
 799                         i = meta_back_get_candidate( op, rs, &ndn );
 800
 801                         if ( rs->sr_err == LDAP_NO_SUCH_OBJECT && parent ) {
 802                                 i = meta_back_get_candidate( op, rs, &pndn );
 803                         }
 804
 805                         if ( rs->sr_err != LDAP_SUCCESS ) {
 806                                 if ( mc != NULL ) {
 807                                         meta_back_release_conn( op, mc );
 808                                 }
 809
 810                                 if ( sendok & LDAP_BACK_SENDERR ) {
 811                                         if ( rs->sr_err == LDAP_NO_SUCH_OBJECT ) {
 812                                                 rs->sr_matched = op->o_bd->be_suffix[ 0 ].bv_val;
 813                                         }
 814                                         send_ldap_result( op, rs );
 815                                         rs->sr_text = NULL;
 816                                         rs->sr_matched = NULL;
 817                                 }
 818
 819                                 return NULL;
 820                         }
 821                 }
 822
 823                 if ( newparent && meta_back_get_candidate( op, rs, op->orr_nnewSup ) != i )
 824                 {
 825                         if ( mc != NULL ) {
 826                                 meta_back_release_conn( op, mc );
 827                         }
 828
 829                         rs->sr_err = LDAP_UNWILLING_TO_PERFORM;
 830                         rs->sr_text = "cross-target rename not supported";
 831                         if ( sendok & LDAP_BACK_SENDERR ) {
 832                                 send_ldap_result( op, rs );
 833                                 rs->sr_text = NULL;
 834                         }
 835
 836                         return NULL;
 837                 }
 838
 839                 Debug( LDAP_DEBUG_TRACE,
 840         "==>meta_back_getconn: got target %d for ndn=\"%s\" from cache\n",
 841                                 i, op->o_req_ndn.bv_val, 0 );
 842
 843                 if ( mc == NULL ) {
 844                         /* Retries searching for a metaconn in the avl tree
 845                          * the reason is that the connection might have been
 846                          * created by meta_back_get_candidate() */
 847                         mc_curr.mc_conn = op->o_conn;
 848                         ldap_pvt_thread_mutex_lock( &mi->mi_conn_mutex );
 849                         mc = (metaconn_t *)avl_find( mi->mi_conntree,
 850                                 (caddr_t)&mc_curr, meta_back_conn_cmp );
 851                         if ( mc != NULL ) {
 852                                 mc->mc_refcnt++;
 853                         }
 854                         ldap_pvt_thread_mutex_unlock( &mi->mi_conn_mutex );
 855
 856                         /* Looks like we didn't get a bind. Open a new session... */
 857                         if ( mc == NULL ) {
 858                                 mc = metaconn_alloc( op );
 859                                 mc->mc_conn = op->o_conn;
 860                                 new_conn = 1;
 861                         }
 862                 }
 863
 864                 /*
 865                  * Clear all other candidates
 866                  */
 867                 ( void )meta_clear_unused_candidates( op, i );
 868
 869                 /*
 870                  * The target is activated; if needed, it is
 871                  * also init'd. In case of error, meta_back_init_one_conn
 872                  * sends the appropriate result.
 873                  */
 874                 err = meta_back_init_one_conn( op, rs, &mi->mi_targets[ i ],
 875                                 &mc->mc_conns[ i ], sendok );
 876                 if ( err != LDAP_SUCCESS ) {
 877                         /*
 878                          * FIXME: in case one target cannot
 879                          * be init'd, should the other ones
 880                          * be tried?
 881                          */
 882                         candidates[ i ].sr_tag = META_NOT_CANDIDATE;
 883                         if ( new_conn ) {
 884                                 (void)meta_clear_one_candidate( &mc->mc_conns[ i ] );
 885                                 meta_back_freeconn( op, mc );
 886
 887                         } else {
 888                                 meta_back_release_conn( op, mc );
 889                         }
 890                         return NULL;
 891                 }
 892
 893                 candidates[ i ].sr_err = LDAP_SUCCESS;
 894                 candidates[ i ].sr_tag = META_CANDIDATE;
 895                 ncandidates++;
 896
 897                 if ( candidate ) {
 898                         *candidate = i;
 899                 }
 900
 901         /*
 902          * if no unique candidate ...
 903          */
 904         } else {
 905
 906                 /* Looks like we didn't get a bind. Open a new session... */
 907                 if ( mc == NULL ) {
 908                         mc = metaconn_alloc( op );
 909                         mc->mc_conn = op->o_conn;
 910                         new_conn = 1;
 911                 }
 912
 913                 for ( i = 0; i < mi->mi_ntargets; i++ ) {
 914                         if ( i == cached
 915                                 || meta_back_is_candidate( &mi->mi_targets[ i ].mt_nsuffix,
 916                                                 mi->mi_targets[ i ].mt_scope,
 917                                                 &op->o_req_ndn, LDAP_SCOPE_SUBTREE ) )
 918                         {
 919
 920                                 /*
 921                                  * The target is activated; if needed, it is
 922                                  * also init'd
 923                                  */
 924                                 int lerr = meta_back_init_one_conn( op, rs,
 925                                                 &mi->mi_targets[ i ],
 926                                                 &mc->mc_conns[ i ], sendok );
 927                                 if ( lerr == LDAP_SUCCESS ) {
 928                                         candidates[ i ].sr_tag = META_CANDIDATE;
 929                                         candidates[ i ].sr_err = LDAP_SUCCESS;
 930                                         ncandidates++;
 931
 932                                         Debug( LDAP_DEBUG_TRACE, "%s: meta_back_init_one_conn(%d)\n",
 933                                                 op->o_log_prefix, i, 0 );
 934
 935                                 } else {
 936
 937                                         /*
 938                                          * FIXME: in case one target cannot
 939                                          * be init'd, should the other ones
 940                                          * be tried?
 941                                          */
 942                                         if ( new_conn ) {
 943                                                 ( void )meta_clear_one_candidate( &mc->mc_conns[ i ] );
 944                                         }
 945                                         /* leave the target candidate, but record the error for later use */
 946                                         candidates[ i ].sr_err = lerr;
 947                                         err = lerr;
 948
 949                                         Debug( LDAP_DEBUG_ANY, "%s: meta_back_init_one_conn(%d) failed: %d\n",
 950                                                 op->o_log_prefix, i, lerr );
 951
 952                                         continue;
 953                                 }
 954
 955                         } else {
 956                                 if ( new_conn ) {
 957                                         ( void )meta_clear_one_candidate( &mc->mc_conns[ i ] );
 958                                 }
 959                                 candidates[ i ].sr_tag = META_NOT_CANDIDATE;
 960                         }
 961                 }
 962
 963                 if ( ncandidates == 0 ) {
 964                         if ( new_conn ) {
 965                                 meta_back_freeconn( op, mc );
 966
 967                         } else {
 968                                 meta_back_release_conn( op, mc );
 969                         }
 970
 971                         rs->sr_err = LDAP_NO_SUCH_OBJECT;
 972                         rs->sr_text = "Unable to select valid candidates";
 973
 974                         if ( sendok & LDAP_BACK_SENDERR ) {
 975                                 if ( rs->sr_err == LDAP_NO_SUCH_OBJECT ) {
 976                                         rs->sr_matched = op->o_bd->be_suffix[ 0 ].bv_val;
 977                                 }
 978                                 send_ldap_result( op, rs );
 979                                 rs->sr_text = NULL;
 980                                 rs->sr_matched = NULL;
 981                         }
 982
 983                         return NULL;
 984                 }
 985         }
 986
 987 done:;
 988         /* clear out meta_back_init_one_conn non-fatal errors */
 989         rs->sr_err = LDAP_SUCCESS;
 990         rs->sr_text = NULL;
 991
 992         if ( new_conn ) {
 993
 994                 /*
 995                  * Inserts the newly created metaconn in the avl tree
 996                  */
 997                 ldap_pvt_thread_mutex_lock( &mi->mi_conn_mutex );
 998                 err = avl_insert( &mi->mi_conntree, ( caddr_t )mc,
 999                                 meta_back_conn_cmp, meta_back_conn_dup );
1000
1001 #if PRINT_CONNTREE > 0
1002                 myprint( mi->mi_conntree );
1003 #endif /* PRINT_CONNTREE */
1004
1005                 ldap_pvt_thread_mutex_unlock( &mi->mi_conn_mutex );
1006
1007                 /*
1008                  * Err could be -1 in case a duplicate metaconn is inserted
1009                  *
1010                  * FIXME: what if the same client issues more than one
1011                  * asynchronous operations?
1012                  */
1013                 if ( err != 0 ) {
1014                         Debug( LDAP_DEBUG_ANY,
1015                                 "%s meta_back_getconn: candidates=%d conn=%ld insert failed\n",
1016                                 op->o_log_prefix, ncandidates, mc->mc_conn->c_connid );
1017
1018                         rs->sr_err = LDAP_OTHER;
1019                         rs->sr_text = "Internal server error";
1020                         meta_back_freeconn( op, mc );
1021                         if ( sendok & LDAP_BACK_SENDERR ) {
1022                                 send_ldap_result( op, rs );
1023                                 rs->sr_text = NULL;
1024                         }
1025                         return NULL;
1026                 }
1027
1028                 Debug( LDAP_DEBUG_TRACE,
1029                         "%s meta_back_getconn: candidates=%d conn=%ld inserted\n",
1030                         op->o_log_prefix, ncandidates, mc->mc_conn->c_connid );
1031
1032         } else {
1033                 Debug( LDAP_DEBUG_TRACE,
1034                         "%s meta_back_getconn: candidates=%d conn=%ld fetched\n",
1035                         op->o_log_prefix, ncandidates, mc->mc_conn->c_connid );
1036         }
1037
1038         return mc;
1039 }
1040
1041 void
1042 meta_back_release_conn(
1043         Operation               *op,
1044         metaconn_t              *mc )
1045 {
1046         metainfo_t      *mi = ( metainfo_t * )op->o_bd->be_private;
1047
1048         assert( mc != NULL );
1049
1050         ldap_pvt_thread_mutex_lock( &mi->mi_conn_mutex );
1051         assert( mc->mc_refcnt > 0 );
1052         mc->mc_refcnt--;
1053         ldap_pvt_thread_mutex_unlock( &mi->mi_conn_mutex );
1054 }