]> git.sur5r.net Git - openldap/blob - servers/slapd/back-ldap/bind.c
ITS#5411
[openldap] / servers / slapd / back-ldap / bind.c
1 /* bind.c - ldap backend bind function */
2 /* $OpenLDAP$ */
3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4  *
5  * Copyright 1999-2008 The OpenLDAP Foundation.
6  * Portions Copyright 2000-2003 Pierangelo Masarati.
7  * Portions Copyright 1999-2003 Howard Chu.
8  * All rights reserved.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted only as authorized by the OpenLDAP
12  * Public License.
13  *
14  * A copy of this license is available in the file LICENSE in the
15  * top-level directory of the distribution or, alternatively, at
16  * <http://www.OpenLDAP.org/license.html>.
17  */
18 /* ACKNOWLEDGEMENTS:
19  * This work was initially developed by Howard Chu for inclusion
20  * in OpenLDAP Software and subsequently enhanced by Pierangelo
21  * Masarati.
22  */
23
24 #include "portable.h"
25
26 #include <stdio.h>
27
28 #include <ac/errno.h>
29 #include <ac/socket.h>
30 #include <ac/string.h>
31
32 #define AVL_INTERNAL
33 #include "slap.h"
34 #include "back-ldap.h"
35 #undef ldap_debug       /* silence a warning in ldap-int.h */
36 #include "../../../libraries/libldap/ldap-int.h"
37
38 #include "lutil_ldap.h"
39
40 #define LDAP_CONTROL_OBSOLETE_PROXY_AUTHZ       "2.16.840.1.113730.3.4.12"
41
42 #if LDAP_BACK_PRINT_CONNTREE > 0
43 static void
44 ldap_back_ravl_print( Avlnode *root, int depth )
45 {
46         int             i;
47         ldapconn_t      *lc;
48         
49         if ( root == 0 ) {
50                 return;
51         }
52         
53         ldap_back_ravl_print( root->avl_right, depth+1 );
54         
55         for ( i = 0; i < depth; i++ ) {
56                 fprintf( stderr, "-" );
57         }
58
59         lc = root->avl_data;
60         fprintf( stderr, "lc=%p local=\"%s\" conn=%p %s refcnt=%d flags=0x%08x\n",
61                 (void *)lc,
62                 lc->lc_local_ndn.bv_val ? lc->lc_local_ndn.bv_val : "",
63                 (void *)lc->lc_conn,
64                 avl_bf2str( root->avl_bf ), lc->lc_refcnt, lc->lc_lcflags );
65         
66         ldap_back_ravl_print( root->avl_left, depth+1 );
67 }
68
69 static char* priv2str[] = {
70         "privileged",
71         "privileged/TLS",
72         "anonymous",
73         "anonymous/TLS",
74         "bind",
75         "bind/TLS",
76         NULL
77 };
78
79 void
80 ldap_back_print_conntree( ldapinfo_t *li, char *msg )
81 {
82         int     c;
83
84         fprintf( stderr, "========> %s\n", msg );
85
86         for ( c = LDAP_BACK_PCONN_FIRST; c < LDAP_BACK_PCONN_LAST; c++ ) {
87                 int             i = 0;
88                 ldapconn_t      *lc;
89
90                 fprintf( stderr, "  %s[%d]\n", priv2str[ c ], li->li_conn_priv[ c ].lic_num );
91
92                 LDAP_TAILQ_FOREACH( lc, &li->li_conn_priv[ c ].lic_priv, lc_q )
93                 {
94                         fprintf( stderr, "    [%d] lc=%p local=\"%s\" conn=%p refcnt=%d flags=0x%08x\n",
95                                 i,
96                                 (void *)lc,
97                                 lc->lc_local_ndn.bv_val ? lc->lc_local_ndn.bv_val : "",
98                                 (void *)lc->lc_conn, lc->lc_refcnt, lc->lc_lcflags );
99                         i++;
100                 }
101         }
102         
103         if ( li->li_conninfo.lai_tree == 0 ) {
104                 fprintf( stderr, "\t(empty)\n" );
105
106         } else {
107                 ldap_back_ravl_print( li->li_conninfo.lai_tree, 0 );
108         }
109         
110         fprintf( stderr, "<======== %s\n", msg );
111 }
112 #endif /* LDAP_BACK_PRINT_CONNTREE */
113
114 static int
115 ldap_back_freeconn( ldapinfo_t *li, ldapconn_t *lc, int dolock );
116
117 static ldapconn_t *
118 ldap_back_getconn( Operation *op, SlapReply *rs, ldap_back_send_t sendok,
119         struct berval *binddn, struct berval *bindcred );
120
121 static int
122 ldap_back_is_proxy_authz( Operation *op, SlapReply *rs, ldap_back_send_t sendok,
123         struct berval *binddn, struct berval *bindcred );
124
125 static int
126 ldap_back_proxy_authz_bind( ldapconn_t *lc, Operation *op, SlapReply *rs,
127         ldap_back_send_t sendok, struct berval *binddn, struct berval *bindcred );
128
129 static int
130 ldap_back_prepare_conn( ldapconn_t *lc, Operation *op, SlapReply *rs,
131         ldap_back_send_t sendok );
132
133 static int
134 ldap_back_conndnlc_cmp( const void *c1, const void *c2 );
135
136 ldapconn_t *
137 ldap_back_conn_delete( ldapinfo_t *li, ldapconn_t *lc )
138 {
139         if ( LDAP_BACK_PCONN_ISPRIV( lc ) ) {
140                 if ( LDAP_BACK_CONN_CACHED( lc ) ) {
141                         assert( lc->lc_q.tqe_prev != NULL );
142                         assert( li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_num > 0 );
143                         li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_num--;
144                         LDAP_TAILQ_REMOVE( &li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_priv, lc, lc_q );
145                         LDAP_TAILQ_ENTRY_INIT( lc, lc_q );
146                         LDAP_BACK_CONN_CACHED_CLEAR( lc );
147
148                 } else {
149                         assert( LDAP_BACK_CONN_TAINTED( lc ) );
150                         assert( lc->lc_q.tqe_prev == NULL );
151                 }
152
153         } else {
154                 ldapconn_t      *tmplc = NULL;
155
156                 if ( LDAP_BACK_CONN_CACHED( lc ) ) {
157                         assert( !LDAP_BACK_CONN_TAINTED( lc ) );
158                         tmplc = avl_delete( &li->li_conninfo.lai_tree, (caddr_t)lc,
159                                 ldap_back_conndnlc_cmp );
160                         assert( tmplc == lc );
161                         LDAP_BACK_CONN_CACHED_CLEAR( lc );
162                 }
163
164                 assert( LDAP_BACK_CONN_TAINTED( lc ) || tmplc == lc );
165         }
166
167         return lc;
168 }
169
170 int
171 ldap_back_bind( Operation *op, SlapReply *rs )
172 {
173         ldapinfo_t              *li = (ldapinfo_t *) op->o_bd->be_private;
174         ldapconn_t              *lc;
175
176         LDAPControl             **ctrls = NULL;
177         struct berval           save_o_dn;
178         int                     save_o_do_not_cache,
179                                 rc = 0;
180         ber_int_t               msgid;
181         ldap_back_send_t        retrying = LDAP_BACK_RETRYING;
182
183         /* allow rootdn as a means to auth without the need to actually
184          * contact the proxied DSA */
185         switch ( be_rootdn_bind( op, rs ) ) {
186         case SLAP_CB_CONTINUE:
187                 break;
188
189         default:
190                 return rs->sr_err;
191         }
192
193         lc = ldap_back_getconn( op, rs, LDAP_BACK_BIND_SERR, NULL, NULL );
194         if ( !lc ) {
195                 return rs->sr_err;
196         }
197
198         /* we can do (almost) whatever we want with this conn,
199          * because either it's temporary, or it's marked as binding */
200         if ( !BER_BVISNULL( &lc->lc_bound_ndn ) ) {
201                 ch_free( lc->lc_bound_ndn.bv_val );
202                 BER_BVZERO( &lc->lc_bound_ndn );
203         }
204         if ( !BER_BVISNULL( &lc->lc_cred ) ) {
205                 memset( lc->lc_cred.bv_val, 0, lc->lc_cred.bv_len );
206                 ch_free( lc->lc_cred.bv_val );
207                 BER_BVZERO( &lc->lc_cred );
208         }
209         LDAP_BACK_CONN_ISBOUND_CLEAR( lc );
210
211         /* don't add proxyAuthz; set the bindDN */
212         save_o_dn = op->o_dn;
213         save_o_do_not_cache = op->o_do_not_cache;
214         op->o_dn = op->o_req_dn;
215         op->o_do_not_cache = 1;
216
217         ctrls = op->o_ctrls;
218         rc = ldap_back_controls_add( op, rs, lc, &ctrls );
219         op->o_dn = save_o_dn;
220         op->o_do_not_cache = save_o_do_not_cache;
221         if ( rc != LDAP_SUCCESS ) {
222                 send_ldap_result( op, rs );
223                 ldap_back_release_conn( li, lc );
224                 return( rc );
225         }
226
227 retry:;
228         /* method is always LDAP_AUTH_SIMPLE if we got here */
229         rs->sr_err = ldap_sasl_bind( lc->lc_ld, op->o_req_dn.bv_val,
230                         LDAP_SASL_SIMPLE,
231                         &op->orb_cred, ctrls, NULL, &msgid );
232         /* FIXME: should we always retry, or only when piping the bind
233          * in the "override" connection pool? */
234         rc = ldap_back_op_result( lc, op, rs, msgid,
235                 li->li_timeout[ SLAP_OP_BIND ],
236                 LDAP_BACK_BIND_SERR | retrying );
237         if ( rc == LDAP_UNAVAILABLE && retrying ) {
238                 retrying &= ~LDAP_BACK_RETRYING;
239                 if ( ldap_back_retry( &lc, op, rs, LDAP_BACK_BIND_SERR ) ) {
240                         goto retry;
241                 }
242         }
243
244         ldap_back_controls_free( op, rs, &ctrls );
245
246         if ( rc == LDAP_SUCCESS ) {
247                 /* If defined, proxyAuthz will be used also when
248                  * back-ldap is the authorizing backend; for this
249                  * purpose, after a successful bind the connection
250                  * is left for further binds, and further operations 
251                  * on this client connection will use a default
252                  * connection with identity assertion */
253                 /* NOTE: use with care */
254                 if ( li->li_idassert_flags & LDAP_BACK_AUTH_OVERRIDE ) {
255                         ldap_back_release_conn( li, lc );
256                         return( rc );
257                 }
258
259                 /* rebind is now done inside ldap_back_proxy_authz_bind()
260                  * in case of success */
261                 LDAP_BACK_CONN_ISBOUND_SET( lc );
262                 ber_dupbv( &lc->lc_bound_ndn, &op->o_req_ndn );
263
264                 if ( !BER_BVISNULL( &lc->lc_cred ) ) {
265                         memset( lc->lc_cred.bv_val, 0,
266                                         lc->lc_cred.bv_len );
267                 }
268
269                 if ( LDAP_BACK_SAVECRED( li ) ) {
270                         ber_bvreplace( &lc->lc_cred, &op->orb_cred );
271                         ldap_set_rebind_proc( lc->lc_ld, li->li_rebind_f, lc );
272
273                 } else {
274                         lc->lc_cred.bv_len = 0;
275                 }
276         }
277
278         /* must re-insert if local DN changed as result of bind */
279         if ( !LDAP_BACK_CONN_ISBOUND( lc )
280                 || ( !dn_match( &op->o_req_ndn, &lc->lc_local_ndn )
281                         && !LDAP_BACK_PCONN_ISPRIV( lc ) ) )
282         {
283                 int             lerr = -1;
284                 ldapconn_t      *tmplc;
285
286                 /* wait for all other ops to release the connection */
287 retry_lock:;
288                 ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
289                 if ( lc->lc_refcnt > 1 ) {
290                         ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
291                         ldap_pvt_thread_yield();
292                         goto retry_lock;
293                 }
294
295 #if LDAP_BACK_PRINT_CONNTREE > 0
296                 ldap_back_print_conntree( li, ">>> ldap_back_bind" );
297 #endif /* LDAP_BACK_PRINT_CONNTREE */
298
299                 assert( lc->lc_refcnt == 1 );
300                 ldap_back_conn_delete( li, lc );
301
302                 /* delete all cached connections with the current connection */
303                 if ( LDAP_BACK_SINGLECONN( li ) ) {
304                         while ( ( tmplc = avl_delete( &li->li_conninfo.lai_tree, (caddr_t)lc, ldap_back_conn_cmp ) ) != NULL )
305                         {
306                                 Debug( LDAP_DEBUG_TRACE,
307                                         "=>ldap_back_bind: destroying conn %ld (refcnt=%u)\n",
308                                         LDAP_BACK_PCONN_ID( lc ), lc->lc_refcnt, 0 );
309
310                                 if ( tmplc->lc_refcnt != 0 ) {
311                                         /* taint it */
312                                         LDAP_BACK_CONN_TAINTED_SET( tmplc );
313                                         LDAP_BACK_CONN_CACHED_CLEAR( tmplc );
314
315                                 } else {
316                                         /*
317                                          * Needs a test because the handler may be corrupted,
318                                          * and calling ldap_unbind on a corrupted header results
319                                          * in a segmentation fault
320                                          */
321                                         ldap_back_conn_free( tmplc );
322                                 }
323                         }
324                 }
325
326                 if ( LDAP_BACK_CONN_ISBOUND( lc ) ) {
327                         ber_bvreplace( &lc->lc_local_ndn, &op->o_req_ndn );
328                         if ( be_isroot_dn( op->o_bd, &op->o_req_ndn ) ) {
329                                 LDAP_BACK_PCONN_ROOTDN_SET( lc, op );
330                         }
331                         lerr = avl_insert( &li->li_conninfo.lai_tree, (caddr_t)lc,
332                                 ldap_back_conndn_cmp, ldap_back_conndn_dup );
333                 }
334
335 #if LDAP_BACK_PRINT_CONNTREE > 0
336                 ldap_back_print_conntree( li, "<<< ldap_back_bind" );
337 #endif /* LDAP_BACK_PRINT_CONNTREE */
338         
339                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
340                 switch ( lerr ) {
341                 case 0:
342                         LDAP_BACK_CONN_CACHED_SET( lc );
343                         break;
344
345                 case -1:
346                         /* duplicate; someone else successfully bound
347                          * on the same connection with the same identity;
348                          * we can do this because lc_refcnt == 1 */
349                         ldap_back_conn_free( lc );
350                         lc = NULL;
351                 }
352         }
353
354         if ( lc != NULL ) {
355                 ldap_back_release_conn( li, lc );
356         }
357
358         return( rc );
359 }
360
361 /*
362  * ldap_back_conndn_cmp
363  *
364  * compares two ldapconn_t based on the value of the conn pointer
365  * and of the local DN; used by avl stuff for insert, lookup
366  * and direct delete
367  */
368 int
369 ldap_back_conndn_cmp( const void *c1, const void *c2 )
370 {
371         const ldapconn_t        *lc1 = (const ldapconn_t *)c1;
372         const ldapconn_t        *lc2 = (const ldapconn_t *)c2;
373         int rc;
374
375         /* If local DNs don't match, it is definitely not a match */
376         /* For shared sessions, conn is NULL. Only explicitly
377          * bound sessions will have non-NULL conn.
378          */
379         rc = SLAP_PTRCMP( lc1->lc_conn, lc2->lc_conn );
380         if ( rc == 0 ) {
381                 rc = ber_bvcmp( &lc1->lc_local_ndn, &lc2->lc_local_ndn );
382         }
383
384         return rc;
385 }
386
387 /*
388  * ldap_back_conndnlc_cmp
389  *
390  * compares two ldapconn_t based on the value of the conn pointer,
391  * the local DN and the lc pointer; used by avl stuff for insert, lookup
392  * and direct delete
393  */
394 static int
395 ldap_back_conndnlc_cmp( const void *c1, const void *c2 )
396 {
397         const ldapconn_t        *lc1 = (const ldapconn_t *)c1;
398         const ldapconn_t        *lc2 = (const ldapconn_t *)c2;
399         int rc;
400
401         /* If local DNs don't match, it is definitely not a match */
402         /* For shared sessions, conn is NULL. Only explicitly
403          * bound sessions will have non-NULL conn.
404          */
405         rc = SLAP_PTRCMP( lc1->lc_conn, lc2->lc_conn );
406         if ( rc == 0 ) {
407                 rc = ber_bvcmp( &lc1->lc_local_ndn, &lc2->lc_local_ndn );
408                 if ( rc == 0 ) {
409                         rc = SLAP_PTRCMP( lc1, lc2 );
410                 }
411         }
412
413         return rc;
414 }
415
416 /*
417  * ldap_back_conn_cmp
418  *
419  * compares two ldapconn_t based on the value of the conn pointer;
420  * used by avl stuff for delete of all conns with the same connid
421  */
422 int
423 ldap_back_conn_cmp( const void *c1, const void *c2 )
424 {
425         const ldapconn_t        *lc1 = (const ldapconn_t *)c1;
426         const ldapconn_t        *lc2 = (const ldapconn_t *)c2;
427
428         /* For shared sessions, conn is NULL. Only explicitly
429          * bound sessions will have non-NULL conn.
430          */
431         return SLAP_PTRCMP( lc1->lc_conn, lc2->lc_conn );
432 }
433
434 /*
435  * ldap_back_conndn_dup
436  *
437  * returns -1 in case a duplicate ldapconn_t has been inserted;
438  * used by avl stuff
439  */
440 int
441 ldap_back_conndn_dup( void *c1, void *c2 )
442 {
443         ldapconn_t      *lc1 = (ldapconn_t *)c1;
444         ldapconn_t      *lc2 = (ldapconn_t *)c2;
445
446         /* Cannot have more than one shared session with same DN */
447         if ( lc1->lc_conn == lc2->lc_conn &&
448                 dn_match( &lc1->lc_local_ndn, &lc2->lc_local_ndn ) )
449         {
450                 return -1;
451         }
452                 
453         return 0;
454 }
455
456 static int
457 ldap_back_freeconn( ldapinfo_t *li, ldapconn_t *lc, int dolock )
458 {
459         if ( dolock ) {
460                 ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
461         }
462
463 #if LDAP_BACK_PRINT_CONNTREE > 0
464         ldap_back_print_conntree( li, ">>> ldap_back_freeconn" );
465 #endif /* LDAP_BACK_PRINT_CONNTREE */
466
467         (void)ldap_back_conn_delete( li, lc );
468
469         if ( lc->lc_refcnt == 0 ) {
470                 ldap_back_conn_free( (void *)lc );
471         }
472
473 #if LDAP_BACK_PRINT_CONNTREE > 0
474         ldap_back_print_conntree( li, "<<< ldap_back_freeconn" );
475 #endif /* LDAP_BACK_PRINT_CONNTREE */
476
477         if ( dolock ) {
478                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
479         }
480
481         return 0;
482 }
483
484 #ifdef HAVE_TLS
485 static int
486 ldap_back_start_tls(
487         LDAP            *ld,
488         int             protocol,
489         int             *is_tls,
490         const char      *url,
491         unsigned        flags,
492         int             retries,
493         const char      **text )
494 {
495         int             rc = LDAP_SUCCESS;
496
497         /* start TLS ("tls-[try-]{start,propagate}" statements) */
498         if ( ( LDAP_BACK_USE_TLS_F( flags ) || ( *is_tls && LDAP_BACK_PROPAGATE_TLS_F( flags ) ) )
499                                 && !ldap_is_ldaps_url( url ) )
500         {
501 #ifdef SLAP_STARTTLS_ASYNCHRONOUS
502                 /*
503                  * use asynchronous StartTLS
504                  * in case, chase referral (not implemented yet)
505                  */
506                 int             msgid;
507
508                 if ( protocol == 0 ) {
509                         ldap_get_option( ld, LDAP_OPT_PROTOCOL_VERSION,
510                                         (void *)&protocol );
511                 }
512
513                 if ( protocol < LDAP_VERSION3 ) {
514                         /* we should rather bail out... */
515                         rc = LDAP_UNWILLING_TO_PERFORM;
516                         *text = "invalid protocol version";
517                 }
518
519                 if ( rc == LDAP_SUCCESS ) {
520                         rc = ldap_start_tls( ld, NULL, NULL, &msgid );
521                 }
522
523                 if ( rc == LDAP_SUCCESS ) {
524                         LDAPMessage     *res = NULL;
525                         struct timeval  tv;
526
527                         LDAP_BACK_TV_SET( &tv );
528
529 retry:;
530                         rc = ldap_result( ld, msgid, LDAP_MSG_ALL, &tv, &res );
531                         if ( rc < 0 ) {
532                                 rc = LDAP_UNAVAILABLE;
533
534                         } else if ( rc == 0 ) {
535                                 if ( retries != LDAP_BACK_RETRY_NEVER ) {
536                                         ldap_pvt_thread_yield();
537                                         if ( retries > 0 ) {
538                                                 retries--;
539                                         }
540                                         LDAP_BACK_TV_SET( &tv );
541                                         goto retry;
542                                 }
543                                 rc = LDAP_UNAVAILABLE;
544
545                         } else if ( rc == LDAP_RES_EXTENDED ) {
546                                 struct berval   *data = NULL;
547
548                                 rc = ldap_parse_extended_result( ld, res,
549                                                 NULL, &data, 0 );
550                                 if ( rc == LDAP_SUCCESS ) {
551                                         int err;
552                                         rc = ldap_parse_result( ld, res, &err,
553                                                 NULL, NULL, NULL, NULL, 1 );
554                                         if ( rc == LDAP_SUCCESS ) {
555                                                 rc = err;
556                                         }
557                                         res = NULL;
558                                         
559                                         /* FIXME: in case a referral 
560                                          * is returned, should we try
561                                          * using it instead of the 
562                                          * configured URI? */
563                                         if ( rc == LDAP_SUCCESS ) {
564                                                 rc = ldap_install_tls( ld );
565
566                                         } else if ( rc == LDAP_REFERRAL ) {
567                                                 rc = LDAP_UNWILLING_TO_PERFORM;
568                                                 *text = "unwilling to chase referral returned by Start TLS exop";
569                                         }
570
571                                         if ( data ) {
572                                                 if ( data->bv_val ) {
573                                                         ber_memfree( data->bv_val );
574                                                 }
575                                                 ber_memfree( data );
576                                         }
577                                 }
578
579                         } else {
580                                 rc = LDAP_OTHER;
581                         }
582
583                         if ( res != NULL ) {
584                                 ldap_msgfree( res );
585                         }
586                 }
587 #else /* ! SLAP_STARTTLS_ASYNCHRONOUS */
588                 /*
589                  * use synchronous StartTLS
590                  */
591                 rc = ldap_start_tls_s( ld, NULL, NULL );
592 #endif /* ! SLAP_STARTTLS_ASYNCHRONOUS */
593
594                 /* if StartTLS is requested, only attempt it if the URL
595                  * is not "ldaps://"; this may occur not only in case
596                  * of misconfiguration, but also when used in the chain 
597                  * overlay, where the "uri" can be parsed out of a referral */
598                 switch ( rc ) {
599                 case LDAP_SUCCESS:
600                         *is_tls = 1;
601                         break;
602
603                 case LDAP_SERVER_DOWN:
604                         break;
605
606                 default:
607                         if ( LDAP_BACK_TLS_CRITICAL_F( flags ) ) {
608                                 *text = "could not start TLS";
609                                 break;
610                         }
611
612                         /* in case Start TLS is not critical */
613                         *is_tls = 0;
614                         rc = LDAP_SUCCESS;
615                         break;
616                 }
617
618         } else {
619                 *is_tls = 0;
620         }
621
622         return rc;
623 }
624 #endif /* HAVE_TLS */
625
626 static int
627 ldap_back_prepare_conn( ldapconn_t *lc, Operation *op, SlapReply *rs, ldap_back_send_t sendok )
628 {
629         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
630         int             version;
631         LDAP            *ld = NULL;
632 #ifdef HAVE_TLS
633         int             is_tls = op->o_conn->c_is_tls;
634         time_t          lc_time = (time_t)(-1);
635         slap_bindconf *sb;
636 #endif /* HAVE_TLS */
637
638         ldap_pvt_thread_mutex_lock( &li->li_uri_mutex );
639         rs->sr_err = ldap_initialize( &ld, li->li_uri );
640         ldap_pvt_thread_mutex_unlock( &li->li_uri_mutex );
641         if ( rs->sr_err != LDAP_SUCCESS ) {
642                 goto error_return;
643         }
644
645         if ( li->li_urllist_f ) {
646                 ldap_set_urllist_proc( ld, li->li_urllist_f, li->li_urllist_p );
647         }
648
649         /* Set LDAP version. This will always succeed: If the client
650          * bound with a particular version, then so can we.
651          */
652         if ( li->li_version != 0 ) {
653                 version = li->li_version;
654
655         } else if ( op->o_protocol != 0 ) {
656                 version = op->o_protocol;
657
658         } else {
659                 /* assume it's an internal op; set to LDAPv3 */
660                 version = LDAP_VERSION3;
661         }
662         ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION, (const void *)&version );
663
664         /* automatically chase referrals ("chase-referrals [{yes|no}]" statement) */
665         ldap_set_option( ld, LDAP_OPT_REFERRALS,
666                 LDAP_BACK_CHASE_REFERRALS( li ) ? LDAP_OPT_ON : LDAP_OPT_OFF );
667
668         if ( li->li_network_timeout > 0 ) {
669                 struct timeval          tv;
670
671                 tv.tv_sec = li->li_network_timeout;
672                 tv.tv_usec = 0;
673                 ldap_set_option( ld, LDAP_OPT_NETWORK_TIMEOUT, (const void *)&tv );
674         }
675
676 #ifdef HAVE_TLS
677         if ( LDAP_BACK_CONN_ISPRIV( lc ) ) {
678                 sb = &li->li_acl;
679
680         } else if ( LDAP_BACK_CONN_ISIDASSERT( lc ) ) {
681                 sb = &li->li_idassert.si_bc;
682
683         } else {
684                 sb = &li->li_tls;
685         }
686
687         if ( sb->sb_tls_do_init ) {
688                 bindconf_tls_set( sb, ld );
689         } else if ( sb->sb_tls_ctx ) {
690                 ldap_set_option( ld, LDAP_OPT_X_TLS_CTX, sb->sb_tls_ctx );
691         }
692
693         ldap_pvt_thread_mutex_lock( &li->li_uri_mutex );
694         rs->sr_err = ldap_back_start_tls( ld, op->o_protocol, &is_tls,
695                         li->li_uri, li->li_flags, li->li_nretries, &rs->sr_text );
696         ldap_pvt_thread_mutex_unlock( &li->li_uri_mutex );
697         if ( rs->sr_err != LDAP_SUCCESS ) {
698                 ldap_unbind_ext( ld, NULL, NULL );
699                 rs->sr_text = "Start TLS failed";
700                 goto error_return;
701
702         } else if ( li->li_idle_timeout ) {
703                 /* only touch when activity actually took place... */
704                 lc_time = op->o_time;
705         }
706 #endif /* HAVE_TLS */
707
708         lc->lc_ld = ld;
709         lc->lc_refcnt = 1;
710 #ifdef HAVE_TLS
711         if ( is_tls ) {
712                 LDAP_BACK_CONN_ISTLS_SET( lc );
713         } else {
714                 LDAP_BACK_CONN_ISTLS_CLEAR( lc );
715         }
716         if ( lc_time != (time_t)(-1) ) {
717                 lc->lc_time = lc_time;
718         }
719 #endif /* HAVE_TLS */
720
721 error_return:;
722         if ( rs->sr_err != LDAP_SUCCESS ) {
723                 rs->sr_err = slap_map_api2result( rs );
724                 if ( sendok & LDAP_BACK_SENDERR ) {
725                         if ( rs->sr_text == NULL ) {
726                                 rs->sr_text = "Proxy connection initialization failed";
727                         }
728                         send_ldap_result( op, rs );
729                 }
730
731         } else {
732                 if ( li->li_conn_ttl > 0 ) {
733                         lc->lc_create_time = op->o_time;
734                 }
735         }
736
737         return rs->sr_err;
738 }
739
740 static ldapconn_t *
741 ldap_back_getconn(
742         Operation               *op,
743         SlapReply               *rs,
744         ldap_back_send_t        sendok,
745         struct berval           *binddn,
746         struct berval           *bindcred )
747 {
748         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
749         ldapconn_t      *lc = NULL,
750                         lc_curr = { 0 };
751         int             refcnt = 1,
752                         lookupconn = !( sendok & LDAP_BACK_BINDING );
753
754         /* if the server is quarantined, and
755          * - the current interval did not expire yet, or
756          * - no more retries should occur,
757          * don't return the connection */
758         if ( li->li_isquarantined ) {
759                 slap_retry_info_t       *ri = &li->li_quarantine;
760                 int                     dont_retry = 1;
761
762                 if ( li->li_quarantine.ri_interval ) {
763                         ldap_pvt_thread_mutex_lock( &li->li_quarantine_mutex );
764                         if ( li->li_isquarantined == LDAP_BACK_FQ_YES ) {
765                                 dont_retry = ( ri->ri_num[ ri->ri_idx ] == SLAP_RETRYNUM_TAIL
766                                         || slap_get_time() < ri->ri_last + ri->ri_interval[ ri->ri_idx ] );
767                                 if ( !dont_retry ) {
768                                         Debug( LDAP_DEBUG_ANY,
769                                                 "%s: ldap_back_getconn quarantine "
770                                                 "retry block #%d try #%d.\n",
771                                                 op->o_log_prefix, ri->ri_idx, ri->ri_count );
772                                         li->li_isquarantined = LDAP_BACK_FQ_RETRYING;
773                                 }
774                         }
775                         ldap_pvt_thread_mutex_unlock( &li->li_quarantine_mutex );
776                 }
777
778                 if ( dont_retry ) {
779                         rs->sr_err = LDAP_UNAVAILABLE;
780                         if ( op->o_conn && ( sendok & LDAP_BACK_SENDERR ) ) {
781                                 rs->sr_text = "Target is quarantined";
782                                 send_ldap_result( op, rs );
783                         }
784                         return NULL;
785                 }
786         }
787
788         /* Internal searches are privileged and shared. So is root. */
789         if ( op->o_do_not_cache || be_isroot( op ) ) {
790                 LDAP_BACK_CONN_ISPRIV_SET( &lc_curr );
791                 lc_curr.lc_local_ndn = op->o_bd->be_rootndn;
792                 LDAP_BACK_PCONN_ROOTDN_SET( &lc_curr, op );
793
794         } else {
795                 struct berval   tmpbinddn,
796                                 tmpbindcred,
797                                 save_o_dn,
798                                 save_o_ndn;
799                 int             isproxyauthz;
800
801                 /* need cleanup */
802                 if ( binddn == NULL ) {
803                         binddn = &tmpbinddn;
804                 }       
805                 if ( bindcred == NULL ) {
806                         bindcred = &tmpbindcred;
807                 }
808                 if ( op->o_tag == LDAP_REQ_BIND ) {
809                         save_o_dn = op->o_dn;
810                         save_o_ndn = op->o_ndn;
811                         op->o_dn = op->o_req_dn;
812                         op->o_ndn = op->o_req_ndn;
813                 }
814                 isproxyauthz = ldap_back_is_proxy_authz( op, rs, sendok, binddn, bindcred );
815                 if ( op->o_tag == LDAP_REQ_BIND ) {
816                         op->o_dn = save_o_dn;
817                         op->o_ndn = save_o_ndn;
818                 }
819                 if ( isproxyauthz == -1 ) {
820                         return NULL;
821                 }
822
823                 lc_curr.lc_local_ndn = op->o_ndn;
824                 /* Explicit binds must not be shared;
825                  * however, explicit binds are piped in a special connection
826                  * when idassert is to occur with "override" set */
827                 if ( op->o_tag == LDAP_REQ_BIND && !isproxyauthz ) {
828                         lc_curr.lc_conn = op->o_conn;
829
830                 } else {
831                         if ( isproxyauthz && !( sendok & LDAP_BACK_BINDING ) ) {
832                                 lc_curr.lc_local_ndn = *binddn;
833                                 LDAP_BACK_PCONN_ROOTDN_SET( &lc_curr, op );
834                                 LDAP_BACK_CONN_ISIDASSERT_SET( &lc_curr );
835
836                         } else if ( isproxyauthz && ( li->li_idassert_flags & LDAP_BACK_AUTH_OVERRIDE ) ) {
837                                 lc_curr.lc_local_ndn = slap_empty_bv;
838                                 LDAP_BACK_PCONN_BIND_SET( &lc_curr, op );
839                                 LDAP_BACK_CONN_ISIDASSERT_SET( &lc_curr );
840                                 lookupconn = 1;
841
842                         } else if ( SLAP_IS_AUTHZ_BACKEND( op ) ) {
843                                 lc_curr.lc_conn = op->o_conn;
844
845                         } else {
846                                 LDAP_BACK_PCONN_ANON_SET( &lc_curr, op );
847                         }
848                 }
849         }
850
851         /* Explicit Bind requests always get their own conn */
852         if ( lookupconn ) {
853 retry_lock:
854                 ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
855                 if ( LDAP_BACK_PCONN_ISPRIV( &lc_curr ) ) {
856                         /* lookup a conn that's not binding */
857                         LDAP_TAILQ_FOREACH( lc,
858                                 &li->li_conn_priv[ LDAP_BACK_CONN2PRIV( &lc_curr ) ].lic_priv,
859                                 lc_q )
860                         {
861                                 if ( !LDAP_BACK_CONN_BINDING( lc ) && lc->lc_refcnt == 0 ) {
862                                         break;
863                                 }
864                         }
865
866                         if ( lc != NULL ) {
867                                 if ( lc != LDAP_TAILQ_LAST( &li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_priv,
868                                         ldapconn_t, lc_q ) )
869                                 {
870                                         LDAP_TAILQ_REMOVE( &li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_priv,
871                                                 lc, lc_q );
872                                         LDAP_TAILQ_ENTRY_INIT( lc, lc_q );
873                                         LDAP_TAILQ_INSERT_TAIL( &li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_priv,
874                                                 lc, lc_q );
875                                 }
876
877                         } else if ( !LDAP_BACK_USE_TEMPORARIES( li )
878                                 && li->li_conn_priv[ LDAP_BACK_CONN2PRIV( &lc_curr ) ].lic_num == li->li_conn_priv_max )
879                         {
880                                 lc = LDAP_TAILQ_FIRST( &li->li_conn_priv[ LDAP_BACK_CONN2PRIV( &lc_curr ) ].lic_priv );
881                         }
882                         
883                 } else {
884
885                         /* Searches for a ldapconn in the avl tree */
886                         lc = (ldapconn_t *)avl_find( li->li_conninfo.lai_tree, 
887                                         (caddr_t)&lc_curr, ldap_back_conndn_cmp );
888                 }
889
890                 if ( lc != NULL ) {
891                         /* Don't reuse connections while they're still binding */
892                         if ( LDAP_BACK_CONN_BINDING( lc ) ) {
893                                 if ( !LDAP_BACK_USE_TEMPORARIES( li ) ) {
894                                         ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
895
896                                         ldap_pvt_thread_yield();
897                                         goto retry_lock;
898                                 }
899                                 lc = NULL;
900                         }
901
902                         if ( lc != NULL ) {
903                                 if ( op->o_tag == LDAP_REQ_BIND ) {
904                                         /* right now, this is the only possible case */
905                                         assert( ( li->li_idassert_flags & LDAP_BACK_AUTH_OVERRIDE ) );
906                                         LDAP_BACK_CONN_BINDING_SET( lc );
907                                 }
908
909                                 refcnt = ++lc->lc_refcnt;
910                         }
911                 }
912                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
913         }
914
915         /* Looks like we didn't get a bind. Open a new session... */
916         if ( lc == NULL ) {
917                 lc = (ldapconn_t *)ch_calloc( 1, sizeof( ldapconn_t ) );
918                 lc->lc_flags = li->li_flags;
919                 lc->lc_lcflags = lc_curr.lc_lcflags;
920                 if ( ldap_back_prepare_conn( lc, op, rs, sendok ) != LDAP_SUCCESS ) {
921                         ch_free( lc );
922                         return NULL;
923                 }
924
925                 if ( sendok & LDAP_BACK_BINDING ) {
926                         LDAP_BACK_CONN_BINDING_SET( lc );
927                 }
928
929                 lc->lc_conn = lc_curr.lc_conn;
930                 ber_dupbv( &lc->lc_local_ndn, &lc_curr.lc_local_ndn );
931
932                 /*
933                  * the rationale is: connections as the rootdn are privileged,
934                  * so acl_authcDN is to be used; however, in some cases
935                  * one already configured identity assertion with a highly
936                  * privileged idassert_authcDN, so if acl_authcDN is NULL
937                  * and idassert_authcDN is not, use the second instead.
938                  *
939                  * might change in the future, because it's preferable
940                  * to make clear what identity is being used, since
941                  * the only drawback is that one risks to configure
942                  * the same identity twice...
943                  */
944                 if ( LDAP_BACK_CONN_ISPRIV( &lc_curr ) ) {
945                         if ( BER_BVISNULL( &li->li_acl_authcDN ) && !BER_BVISNULL( &li->li_idassert_authcDN ) ) {
946                                 ber_dupbv( &lc->lc_bound_ndn, &li->li_idassert_authcDN );
947                                 ber_dupbv( &lc->lc_cred, &li->li_idassert_passwd );
948
949                         } else {
950                                 ber_dupbv( &lc->lc_bound_ndn, &li->li_acl_authcDN );
951                                 ber_dupbv( &lc->lc_cred, &li->li_acl_passwd );
952                         }
953                         LDAP_BACK_CONN_ISPRIV_SET( lc );
954
955                 } else if ( LDAP_BACK_CONN_ISIDASSERT( &lc_curr ) ) {
956                         if ( !LDAP_BACK_PCONN_ISBIND( &lc_curr ) ) {
957                                 ber_dupbv( &lc->lc_bound_ndn, &li->li_idassert_authcDN );
958                                 ber_dupbv( &lc->lc_cred, &li->li_idassert_passwd );
959                         }
960                         LDAP_BACK_CONN_ISIDASSERT_SET( lc );
961
962                 } else {
963                         BER_BVZERO( &lc->lc_cred );
964                         BER_BVZERO( &lc->lc_bound_ndn );
965                         if ( !BER_BVISEMPTY( &op->o_ndn )
966                                 && SLAP_IS_AUTHZ_BACKEND( op ) )
967                         {
968                                 ber_dupbv( &lc->lc_bound_ndn, &op->o_ndn );
969                         }
970                 }
971
972 #ifdef HAVE_TLS
973                 /* if start TLS failed but it was not mandatory,
974                  * check if the non-TLS connection was already
975                  * in cache; in case, destroy the newly created
976                  * connection and use the existing one */
977                 if ( LDAP_BACK_PCONN_ISTLS( lc ) 
978                                 && !ldap_tls_inplace( lc->lc_ld ) )
979                 {
980                         ldapconn_t      *tmplc = NULL;
981                         int             idx = LDAP_BACK_CONN2PRIV( &lc_curr ) - 1;
982                         
983                         ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
984                         LDAP_TAILQ_FOREACH( tmplc,
985                                 &li->li_conn_priv[ idx ].lic_priv,
986                                 lc_q )
987                         {
988                                 if ( !LDAP_BACK_CONN_BINDING( tmplc ) ) {
989                                         break;
990                                 }
991                         }
992
993                         if ( tmplc != NULL ) {
994                                 refcnt = ++tmplc->lc_refcnt;
995                                 ldap_back_conn_free( lc );
996                                 lc = tmplc;
997                         }
998                         ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
999
1000                         if ( tmplc != NULL ) {
1001                                 goto done;
1002                         }
1003                 }
1004 #endif /* HAVE_TLS */
1005
1006                 /* Inserts the newly created ldapconn in the avl tree */
1007                 ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
1008
1009                 LDAP_BACK_CONN_ISBOUND_CLEAR( lc );
1010
1011                 assert( lc->lc_refcnt == 1 );
1012
1013 #if LDAP_BACK_PRINT_CONNTREE > 0
1014                 ldap_back_print_conntree( li, ">>> ldap_back_getconn(insert)" );
1015 #endif /* LDAP_BACK_PRINT_CONNTREE */
1016         
1017                 if ( LDAP_BACK_PCONN_ISPRIV( lc ) ) {
1018                         if ( li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_num < li->li_conn_priv_max ) {
1019                                 LDAP_TAILQ_INSERT_TAIL( &li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_priv, lc, lc_q );
1020                                 li->li_conn_priv[ LDAP_BACK_CONN2PRIV( lc ) ].lic_num++;
1021                                 LDAP_BACK_CONN_CACHED_SET( lc );
1022
1023                         } else {
1024                                 LDAP_BACK_CONN_TAINTED_SET( lc );
1025                         }
1026                         rs->sr_err = 0;
1027
1028                 } else {
1029                         rs->sr_err = avl_insert( &li->li_conninfo.lai_tree, (caddr_t)lc,
1030                                 ldap_back_conndn_cmp, ldap_back_conndn_dup );
1031                         LDAP_BACK_CONN_CACHED_SET( lc );
1032                 }
1033
1034 #if LDAP_BACK_PRINT_CONNTREE > 0
1035                 ldap_back_print_conntree( li, "<<< ldap_back_getconn(insert)" );
1036 #endif /* LDAP_BACK_PRINT_CONNTREE */
1037         
1038                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1039
1040                 if ( LogTest( LDAP_DEBUG_TRACE ) ) {
1041                         char    buf[ SLAP_TEXT_BUFLEN ];
1042
1043                         snprintf( buf, sizeof( buf ),
1044                                 "lc=%p inserted refcnt=%u rc=%d",
1045                                 (void *)lc, refcnt, rs->sr_err );
1046                                 
1047                         Debug( LDAP_DEBUG_TRACE,
1048                                 "=>ldap_back_getconn: %s: %s\n",
1049                                 op->o_log_prefix, buf, 0 );
1050                 }
1051         
1052                 if ( !LDAP_BACK_PCONN_ISPRIV( lc ) ) {
1053                         /* Err could be -1 in case a duplicate ldapconn is inserted */
1054                         switch ( rs->sr_err ) {
1055                         case 0:
1056                                 break;
1057
1058                         case -1:
1059                                 LDAP_BACK_CONN_CACHED_CLEAR( lc );
1060                                 if ( !( sendok & LDAP_BACK_BINDING ) && !LDAP_BACK_USE_TEMPORARIES( li ) ) {
1061                                         /* duplicate: free and try to get the newly created one */
1062                                         ldap_back_conn_free( lc );
1063                                         lc = NULL;
1064                                         goto retry_lock;
1065                                 }
1066
1067                                 /* taint connection, so that it'll be freed when released */
1068                                 LDAP_BACK_CONN_TAINTED_SET( lc );
1069                                 break;
1070
1071                         default:
1072                                 LDAP_BACK_CONN_CACHED_CLEAR( lc );
1073                                 ldap_back_conn_free( lc );
1074                                 rs->sr_err = LDAP_OTHER;
1075                                 rs->sr_text = "Proxy bind collision";
1076                                 if ( op->o_conn && ( sendok & LDAP_BACK_SENDERR ) ) {
1077                                         send_ldap_result( op, rs );
1078                                 }
1079                                 return NULL;
1080                         }
1081                 }
1082
1083         } else {
1084                 int     expiring = 0;
1085
1086                 if ( ( li->li_idle_timeout != 0 && op->o_time > lc->lc_time + li->li_idle_timeout )
1087                         || ( li->li_conn_ttl != 0 && op->o_time > lc->lc_create_time + li->li_conn_ttl ) )
1088                 {
1089                         expiring = 1;
1090
1091                         /* let it be used, but taint/delete it so that 
1092                          * no-one else can look it up any further */
1093                         ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
1094
1095 #if LDAP_BACK_PRINT_CONNTREE > 0
1096                         ldap_back_print_conntree( li, ">>> ldap_back_getconn(timeout)" );
1097 #endif /* LDAP_BACK_PRINT_CONNTREE */
1098
1099                         (void)ldap_back_conn_delete( li, lc );
1100                         LDAP_BACK_CONN_TAINTED_SET( lc );
1101
1102 #if LDAP_BACK_PRINT_CONNTREE > 0
1103                         ldap_back_print_conntree( li, "<<< ldap_back_getconn(timeout)" );
1104 #endif /* LDAP_BACK_PRINT_CONNTREE */
1105
1106                         ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1107                 }
1108
1109                 if ( LogTest( LDAP_DEBUG_TRACE ) ) {
1110                         char    buf[ SLAP_TEXT_BUFLEN ];
1111
1112                         snprintf( buf, sizeof( buf ),
1113                                 "conn %p fetched refcnt=%u%s",
1114                                 (void *)lc, refcnt,
1115                                 expiring ? " expiring" : "" );
1116                         Debug( LDAP_DEBUG_TRACE,
1117                                 "=>ldap_back_getconn: %s.\n", buf, 0, 0 );
1118                 }
1119         }
1120
1121 #ifdef HAVE_TLS
1122 done:;
1123 #endif /* HAVE_TLS */
1124
1125         return lc;
1126 }
1127
1128 void
1129 ldap_back_release_conn_lock(
1130         ldapinfo_t              *li,
1131         ldapconn_t              **lcp,
1132         int                     dolock )
1133 {
1134
1135         ldapconn_t      *lc = *lcp;
1136
1137         if ( dolock ) {
1138                 ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
1139         }
1140         assert( lc->lc_refcnt > 0 );
1141         LDAP_BACK_CONN_BINDING_CLEAR( lc );
1142         lc->lc_refcnt--;
1143         if ( LDAP_BACK_CONN_TAINTED( lc ) ) {
1144                 ldap_back_freeconn( li, lc, 0 );
1145                 *lcp = NULL;
1146         }
1147         if ( dolock ) {
1148                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1149         }
1150 }
1151
1152 void
1153 ldap_back_quarantine(
1154         Operation       *op,
1155         SlapReply       *rs )
1156 {
1157         ldapinfo_t              *li = (ldapinfo_t *)op->o_bd->be_private;
1158
1159         slap_retry_info_t       *ri = &li->li_quarantine;
1160
1161         ldap_pvt_thread_mutex_lock( &li->li_quarantine_mutex );
1162
1163         if ( rs->sr_err == LDAP_UNAVAILABLE ) {
1164                 time_t          new_last = slap_get_time();
1165
1166                 switch ( li->li_isquarantined ) {
1167                 case LDAP_BACK_FQ_NO:
1168                         if ( ri->ri_last == new_last ) {
1169                                 goto done;
1170                         }
1171
1172                         Debug( LDAP_DEBUG_ANY,
1173                                 "%s: ldap_back_quarantine enter.\n",
1174                                 op->o_log_prefix, 0, 0 );
1175
1176                         ri->ri_idx = 0;
1177                         ri->ri_count = 0;
1178                         break;
1179
1180                 case LDAP_BACK_FQ_RETRYING:
1181                         Debug( LDAP_DEBUG_ANY,
1182                                 "%s: ldap_back_quarantine block #%d try #%d failed.\n",
1183                                 op->o_log_prefix, ri->ri_idx, ri->ri_count );
1184
1185                         ++ri->ri_count;
1186                         if ( ri->ri_num[ ri->ri_idx ] != SLAP_RETRYNUM_FOREVER
1187                                 && ri->ri_count == ri->ri_num[ ri->ri_idx ] )
1188                         {
1189                                 ri->ri_count = 0;
1190                                 ++ri->ri_idx;
1191                         }
1192                         break;
1193
1194                 default:
1195                         break;
1196                 }
1197
1198                 li->li_isquarantined = LDAP_BACK_FQ_YES;
1199                 ri->ri_last = new_last;
1200
1201         } else if ( li->li_isquarantined != LDAP_BACK_FQ_NO ) {
1202                 if ( ri->ri_last == slap_get_time() ) {
1203                         goto done;
1204                 }
1205
1206                 Debug( LDAP_DEBUG_ANY,
1207                         "%s: ldap_back_quarantine exit (%d) err=%d.\n",
1208                         op->o_log_prefix, li->li_isquarantined, rs->sr_err );
1209
1210                 if ( li->li_quarantine_f ) {
1211                         (void)li->li_quarantine_f( li, li->li_quarantine_p );
1212                 }
1213
1214                 ri->ri_count = 0;
1215                 ri->ri_idx = 0;
1216                 li->li_isquarantined = LDAP_BACK_FQ_NO;
1217         }
1218
1219 done:;
1220         ldap_pvt_thread_mutex_unlock( &li->li_quarantine_mutex );
1221 }
1222
1223 static int
1224 ldap_back_dobind_cb(
1225         Operation *op,
1226         SlapReply *rs
1227 )
1228 {
1229         ber_tag_t *tptr = op->o_callback->sc_private;
1230         op->o_tag = *tptr;
1231         rs->sr_tag = slap_req2res( op->o_tag );
1232
1233         return SLAP_CB_CONTINUE;
1234 }
1235
1236 /*
1237  * ldap_back_dobind_int
1238  *
1239  * Note: dolock indicates whether li->li_conninfo.lai_mutex must be locked or not
1240  */
1241 static int
1242 ldap_back_dobind_int(
1243         ldapconn_t              **lcp,
1244         Operation               *op,
1245         SlapReply               *rs,
1246         ldap_back_send_t        sendok,
1247         int                     retries,
1248         int                     dolock )
1249 {       
1250         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
1251
1252         ldapconn_t      *lc;
1253         struct berval   binddn = slap_empty_bv,
1254                         bindcred = slap_empty_bv;
1255
1256         int             rc = 0,
1257                         isbound,
1258                         binding = 0;
1259         ber_int_t       msgid;
1260         ber_tag_t       o_tag = op->o_tag;
1261         slap_callback cb = {0};
1262
1263         assert( lcp != NULL );
1264         assert( retries >= 0 );
1265
1266         if ( sendok & LDAP_BACK_GETCONN ) {
1267                 assert( *lcp == NULL );
1268
1269                 lc = ldap_back_getconn( op, rs, sendok, &binddn, &bindcred );
1270                 if ( lc == NULL ) {
1271                         return 0;
1272                 }
1273                 *lcp = lc;
1274
1275         } else {
1276                 lc = *lcp;
1277         }
1278
1279         assert( lc != NULL );
1280
1281 retry_lock:;
1282         if ( dolock ) {
1283                 ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
1284         }
1285
1286         if ( binding == 0 ) {
1287                 /* check if already bound */
1288                 rc = isbound = LDAP_BACK_CONN_ISBOUND( lc );
1289                 if ( isbound ) {
1290                         if ( dolock ) {
1291                                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1292                         }
1293                         return rc;
1294                 }
1295
1296                 if ( LDAP_BACK_CONN_BINDING( lc ) ) {
1297                         /* if someone else is about to bind it, give up and retry */
1298                         if ( dolock ) {
1299                                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1300                         }
1301                         ldap_pvt_thread_yield();
1302                         goto retry_lock;
1303
1304                 } else {
1305                         /* otherwise this thread will bind it */
1306                         LDAP_BACK_CONN_BINDING_SET( lc );
1307                         binding = 1;
1308                 }
1309         }
1310
1311         if ( dolock ) {
1312                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1313         }
1314
1315         /*
1316          * FIXME: we need to let clients use proxyAuthz
1317          * otherwise we cannot do symmetric pools of servers;
1318          * we have to live with the fact that a user can
1319          * authorize itself as any ID that is allowed
1320          * by the authzTo directive of the "proxyauthzdn".
1321          */
1322         /*
1323          * NOTE: current Proxy Authorization specification
1324          * and implementation do not allow proxy authorization
1325          * control to be provided with Bind requests
1326          */
1327         /*
1328          * if no bind took place yet, but the connection is bound
1329          * and the "idassert-authcDN" (or other ID) is set, 
1330          * then bind as the asserting identity and explicitly 
1331          * add the proxyAuthz control to every operation with the
1332          * dn bound to the connection as control value.
1333          * This is done also if this is the authorizing backend,
1334          * but the "override" flag is given to idassert.
1335          * It allows to use SASL bind and yet proxyAuthz users
1336          */
1337         op->o_tag = LDAP_REQ_BIND;
1338         cb.sc_next = op->o_callback;
1339         cb.sc_private = &o_tag;
1340         cb.sc_response = ldap_back_dobind_cb;
1341         op->o_callback = &cb;
1342
1343         if ( LDAP_BACK_CONN_ISIDASSERT( lc ) ) {
1344                 if ( BER_BVISEMPTY( &binddn ) && BER_BVISEMPTY( &bindcred ) ) {
1345                         /* if we got here, it shouldn't return result */
1346                         rc = ldap_back_is_proxy_authz( op, rs,
1347                                 LDAP_BACK_DONTSEND, &binddn, &bindcred );
1348                         assert( rc == 1 );
1349                 }
1350                 rc = ldap_back_proxy_authz_bind( lc, op, rs, sendok, &binddn, &bindcred );
1351                 goto done;
1352         }
1353
1354 #ifdef HAVE_CYRUS_SASL
1355         if ( LDAP_BACK_CONN_ISPRIV( lc )
1356                 && li->li_acl_authmethod == LDAP_AUTH_SASL )
1357         {
1358                 void            *defaults = NULL;
1359
1360                 if ( li->li_acl_secprops != NULL ) {
1361                         rc = ldap_set_option( lc->lc_ld,
1362                                 LDAP_OPT_X_SASL_SECPROPS, li->li_acl_secprops );
1363
1364                         if ( rc != LDAP_OPT_SUCCESS ) {
1365                                 Debug( LDAP_DEBUG_ANY, "Error: ldap_set_option "
1366                                         "(SECPROPS,\"%s\") failed!\n",
1367                                         li->li_acl_secprops, 0, 0 );
1368                                 goto done;
1369                         }
1370                 }
1371
1372                 defaults = lutil_sasl_defaults( lc->lc_ld,
1373                                 li->li_acl_sasl_mech.bv_val,
1374                                 li->li_acl_sasl_realm.bv_val,
1375                                 li->li_acl_authcID.bv_val,
1376                                 li->li_acl_passwd.bv_val,
1377                                 NULL );
1378                 if ( defaults == NULL ) {
1379                         rs->sr_err = LDAP_OTHER;
1380                         LDAP_BACK_CONN_ISBOUND_CLEAR( lc );
1381                         if ( sendok & LDAP_BACK_SENDERR ) {
1382                                 send_ldap_result( op, rs );
1383                         }
1384                         goto done;
1385                 }
1386
1387                 rs->sr_err = ldap_sasl_interactive_bind_s( lc->lc_ld,
1388                                 li->li_acl_authcDN.bv_val,
1389                                 li->li_acl_sasl_mech.bv_val, NULL, NULL,
1390                                 LDAP_SASL_QUIET, lutil_sasl_interact,
1391                                 defaults );
1392
1393                 lutil_sasl_freedefs( defaults );
1394
1395                 switch ( rs->sr_err ) {
1396                 case LDAP_SUCCESS:
1397                         LDAP_BACK_CONN_ISBOUND_SET( lc );
1398                         break;
1399
1400                 case LDAP_LOCAL_ERROR:
1401                         /* list client API error codes that require
1402                          * to taint the connection */
1403                         /* FIXME: should actually retry? */
1404                         LDAP_BACK_CONN_TAINTED_SET( lc );
1405
1406                         /* fallthru */
1407
1408                 default:
1409                         LDAP_BACK_CONN_ISBOUND_CLEAR( lc );
1410                         rs->sr_err = slap_map_api2result( rs );
1411                         if ( sendok & LDAP_BACK_SENDERR ) {
1412                                 send_ldap_result( op, rs );
1413                         }
1414                         break;
1415                 }
1416
1417                 if ( LDAP_BACK_QUARANTINE( li ) ) {
1418                         ldap_back_quarantine( op, rs );
1419                 }
1420
1421                 goto done;
1422         }
1423 #endif /* HAVE_CYRUS_SASL */
1424
1425 retry:;
1426         rs->sr_err = ldap_sasl_bind( lc->lc_ld,
1427                         BER_BVISNULL( &lc->lc_cred ) ? "" : lc->lc_bound_ndn.bv_val,
1428                         LDAP_SASL_SIMPLE, &lc->lc_cred,
1429                         NULL, NULL, &msgid );
1430
1431         if ( rs->sr_err == LDAP_SERVER_DOWN ) {
1432                 if ( retries != LDAP_BACK_RETRY_NEVER ) {
1433                         if ( dolock ) {
1434                                 ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
1435                         }
1436
1437                         assert( lc->lc_refcnt > 0 );
1438                         if ( lc->lc_refcnt == 1 ) {
1439                                 ldap_unbind_ext( lc->lc_ld, NULL, NULL );
1440                                 lc->lc_ld = NULL;
1441
1442                                 /* lc here must be the regular lc, reset and ready for init */
1443                                 rs->sr_err = ldap_back_prepare_conn( lc, op, rs, sendok );
1444                                 if ( rs->sr_err != LDAP_SUCCESS ) {
1445                                         sendok &= ~LDAP_BACK_SENDERR;
1446                                         lc->lc_refcnt = 0;
1447                                 }
1448                         }
1449
1450                         if ( dolock ) {
1451                                 ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1452                         }
1453
1454                         if ( rs->sr_err == LDAP_SUCCESS ) {
1455                                 if ( retries > 0 ) {
1456                                         retries--;
1457                                 }
1458                                 goto retry;
1459                         }
1460                 }
1461
1462                 assert( lc->lc_refcnt == 1 );
1463                 lc->lc_refcnt = 0;
1464                 ldap_back_freeconn( li, lc, dolock );
1465                 *lcp = NULL;
1466                 rs->sr_err = slap_map_api2result( rs );
1467
1468                 if ( LDAP_BACK_QUARANTINE( li ) ) {
1469                         ldap_back_quarantine( op, rs );
1470                 }
1471
1472                 if ( rs->sr_err != LDAP_SUCCESS &&
1473                         ( sendok & LDAP_BACK_SENDERR ) )
1474                 {
1475                         if ( op->o_callback == &cb )
1476                                 op->o_callback = cb.sc_next;
1477                         op->o_tag = o_tag;
1478                         rs->sr_text = "Proxy can't contact remote server";
1479                         send_ldap_result( op, rs );
1480                 }
1481
1482                 rc = 0;
1483                 goto func_leave;
1484         }
1485
1486         rc = ldap_back_op_result( lc, op, rs, msgid,
1487                 -1, ( sendok | LDAP_BACK_BINDING ) );
1488         if ( rc == LDAP_SUCCESS ) {
1489                 LDAP_BACK_CONN_ISBOUND_SET( lc );
1490         }
1491
1492 done:;
1493         LDAP_BACK_CONN_BINDING_CLEAR( lc );
1494         rc = LDAP_BACK_CONN_ISBOUND( lc );
1495         if ( !rc ) {
1496                 ldap_back_release_conn_lock( li, lcp, dolock );
1497
1498         } else if ( LDAP_BACK_SAVECRED( li ) ) {
1499                 ldap_set_rebind_proc( lc->lc_ld, li->li_rebind_f, lc );
1500         }
1501
1502 func_leave:;
1503         if ( op->o_callback == &cb )
1504                 op->o_callback = cb.sc_next;
1505         op->o_tag = o_tag;
1506
1507         return rc;
1508 }
1509
1510 /*
1511  * ldap_back_dobind
1512  *
1513  * Note: dolock indicates whether li->li_conninfo.lai_mutex must be locked or not
1514  */
1515 int
1516 ldap_back_dobind( ldapconn_t **lcp, Operation *op, SlapReply *rs, ldap_back_send_t sendok )
1517 {
1518         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
1519
1520         return ldap_back_dobind_int( lcp, op, rs,
1521                 ( sendok | LDAP_BACK_GETCONN ), li->li_nretries, 1 );
1522 }
1523
1524 /*
1525  * ldap_back_default_rebind
1526  *
1527  * This is a callback used for chasing referrals using the same
1528  * credentials as the original user on this session.
1529  */
1530 int 
1531 ldap_back_default_rebind( LDAP *ld, LDAP_CONST char *url, ber_tag_t request,
1532         ber_int_t msgid, void *params )
1533 {
1534         ldapconn_t      *lc = (ldapconn_t *)params;
1535
1536 #ifdef HAVE_TLS
1537         /* ... otherwise we couldn't get here */
1538         assert( lc != NULL );
1539
1540         if ( !ldap_tls_inplace( ld ) ) {
1541                 int             is_tls = LDAP_BACK_CONN_ISTLS( lc ),
1542                                 rc;
1543                 const char      *text = NULL;
1544
1545                 rc = ldap_back_start_tls( ld, 0, &is_tls, url, lc->lc_flags,
1546                         LDAP_BACK_RETRY_DEFAULT, &text );
1547                 if ( rc != LDAP_SUCCESS ) {
1548                         return rc;
1549                 }
1550         }
1551 #endif /* HAVE_TLS */
1552
1553         /* FIXME: add checks on the URL/identity? */
1554
1555         return ldap_sasl_bind_s( ld,
1556                         BER_BVISNULL( &lc->lc_cred ) ? "" : lc->lc_bound_ndn.bv_val,
1557                         LDAP_SASL_SIMPLE, &lc->lc_cred, NULL, NULL, NULL );
1558 }
1559
1560 /*
1561  * ldap_back_default_urllist
1562  */
1563 int 
1564 ldap_back_default_urllist(
1565         LDAP            *ld,
1566         LDAPURLDesc     **urllist,
1567         LDAPURLDesc     **url,
1568         void            *params )
1569 {
1570         ldapinfo_t      *li = (ldapinfo_t *)params;
1571         LDAPURLDesc     **urltail;
1572
1573         if ( urllist == url ) {
1574                 return LDAP_SUCCESS;
1575         }
1576
1577         for ( urltail = &(*url)->lud_next; *urltail; urltail = &(*urltail)->lud_next )
1578                 /* count */ ;
1579
1580         *urltail = *urllist;
1581         *urllist = *url;
1582         *url = NULL;
1583
1584         ldap_pvt_thread_mutex_lock( &li->li_uri_mutex );
1585         if ( li->li_uri ) {
1586                 ch_free( li->li_uri );
1587         }
1588
1589         ldap_get_option( ld, LDAP_OPT_URI, (void *)&li->li_uri );
1590         ldap_pvt_thread_mutex_unlock( &li->li_uri_mutex );
1591
1592         return LDAP_SUCCESS;
1593 }
1594
1595 int
1596 ldap_back_cancel(
1597                 ldapconn_t              *lc,
1598                 Operation               *op,
1599                 SlapReply               *rs,
1600                 ber_int_t               msgid,
1601                 ldap_back_send_t        sendok )
1602 {
1603         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
1604
1605         /* default behavior */
1606         if ( LDAP_BACK_ABANDON( li ) ) {
1607                 return ldap_abandon_ext( lc->lc_ld, msgid, NULL, NULL );
1608         }
1609
1610         if ( LDAP_BACK_IGNORE( li ) ) {
1611                 return ldap_pvt_discard( lc->lc_ld, msgid );
1612         }
1613
1614         if ( LDAP_BACK_CANCEL( li ) ) {
1615                 /* FIXME: asynchronous? */
1616                 return ldap_cancel_s( lc->lc_ld, msgid, NULL, NULL );
1617         }
1618
1619         assert( 0 );
1620
1621         return LDAP_OTHER;
1622 }
1623
1624 int
1625 ldap_back_op_result(
1626                 ldapconn_t              *lc,
1627                 Operation               *op,
1628                 SlapReply               *rs,
1629                 ber_int_t               msgid,
1630                 time_t                  timeout,
1631                 ldap_back_send_t        sendok )
1632 {
1633         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
1634
1635         char            *match = NULL;
1636         char            *text = NULL;
1637         char            **refs = NULL;
1638         LDAPControl     **ctrls = NULL;
1639
1640 #define ERR_OK(err) ((err) == LDAP_SUCCESS || (err) == LDAP_COMPARE_FALSE || (err) == LDAP_COMPARE_TRUE)
1641
1642         rs->sr_text = NULL;
1643         rs->sr_matched = NULL;
1644         rs->sr_ref = NULL;
1645         rs->sr_ctrls = NULL;
1646
1647         /* if the error recorded in the reply corresponds
1648          * to a successful state, get the error from the
1649          * remote server response */
1650         if ( ERR_OK( rs->sr_err ) ) {
1651                 int             rc;
1652                 struct timeval  tv;
1653                 LDAPMessage     *res = NULL;
1654                 time_t          stoptime = (time_t)(-1);
1655                 int             timeout_err = op->o_protocol >= LDAP_VERSION3 ?
1656                                         LDAP_ADMINLIMIT_EXCEEDED : LDAP_OTHER;
1657                 const char      *timeout_text = "Operation timed out";
1658
1659                 /* if timeout is not specified, compute and use
1660                  * the one specific to the ongoing operation */
1661                 if ( timeout == (time_t)(-1) ) {
1662                         slap_op_t       opidx = slap_req2op( op->o_tag );
1663
1664                         if ( opidx == SLAP_OP_SEARCH ) {
1665                                 if ( op->ors_tlimit <= 0 ) {
1666                                         timeout = 0;
1667
1668                                 } else {
1669                                         timeout = op->ors_tlimit;
1670                                         timeout_err = LDAP_TIMELIMIT_EXCEEDED;
1671                                         timeout_text = NULL;
1672                                 }
1673
1674                         } else {
1675                                 timeout = li->li_timeout[ opidx ];
1676                         }
1677                 }
1678
1679                 /* better than nothing :) */
1680                 if ( timeout == 0 ) {
1681                         if ( li->li_idle_timeout ) {
1682                                 timeout = li->li_idle_timeout;
1683
1684                         } else if ( li->li_conn_ttl ) {
1685                                 timeout = li->li_conn_ttl;
1686                         }
1687                 }
1688
1689                 if ( timeout ) {
1690                         stoptime = op->o_time + timeout;
1691                 }
1692
1693                 LDAP_BACK_TV_SET( &tv );
1694
1695 retry:;
1696                 /* if result parsing fails, note the failure reason */
1697                 rc = ldap_result( lc->lc_ld, msgid, LDAP_MSG_ALL, &tv, &res );
1698                 switch ( rc ) {
1699                 case 0:
1700                         if ( timeout && slap_get_time() > stoptime ) {
1701                                 if ( sendok & LDAP_BACK_BINDING ) {
1702                                         ldap_unbind_ext( lc->lc_ld, NULL, NULL );
1703                                         lc->lc_ld = NULL;
1704
1705                                         /* let it be used, but taint/delete it so that 
1706                                          * no-one else can look it up any further */
1707                                         ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
1708
1709 #if LDAP_BACK_PRINT_CONNTREE > 0
1710                                         ldap_back_print_conntree( li, ">>> ldap_back_getconn(timeout)" );
1711 #endif /* LDAP_BACK_PRINT_CONNTREE */
1712
1713                                         (void)ldap_back_conn_delete( li, lc );
1714                                         LDAP_BACK_CONN_TAINTED_SET( lc );
1715
1716 #if LDAP_BACK_PRINT_CONNTREE > 0
1717                                         ldap_back_print_conntree( li, "<<< ldap_back_getconn(timeout)" );
1718 #endif /* LDAP_BACK_PRINT_CONNTREE */
1719                                         ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1720
1721                                 } else {
1722                                         (void)ldap_back_cancel( lc, op, rs, msgid, sendok );
1723                                 }
1724                                 rs->sr_err = timeout_err;
1725                                 rs->sr_text = timeout_text;
1726                                 break;
1727                         }
1728
1729                         /* timeout == 0 */
1730                         LDAP_BACK_TV_SET( &tv );
1731                         ldap_pvt_thread_yield();
1732                         goto retry;
1733
1734                 case -1:
1735                         ldap_get_option( lc->lc_ld, LDAP_OPT_ERROR_NUMBER,
1736                                         &rs->sr_err );
1737                         break;
1738
1739
1740                 /* otherwise get the result; if it is not
1741                  * LDAP_SUCCESS, record it in the reply
1742                  * structure (this includes 
1743                  * LDAP_COMPARE_{TRUE|FALSE}) */
1744                 default:
1745                         /* only touch when activity actually took place... */
1746                         if ( li->li_idle_timeout && lc ) {
1747                                 lc->lc_time = op->o_time;
1748                         }
1749
1750                         rc = ldap_parse_result( lc->lc_ld, res, &rs->sr_err,
1751                                         &match, &text, &refs, &ctrls, 1 );
1752                         rs->sr_text = text;
1753                         if ( rc != LDAP_SUCCESS ) {
1754                                 rs->sr_err = rc;
1755                         }
1756
1757                         /* RFC 4511: referrals can only appear
1758                          * if result code is LDAP_REFERRAL */
1759                         if ( refs != NULL
1760                                 && refs[ 0 ] != NULL
1761                                 && refs[ 0 ][ 0 ] != '\0' )
1762                         {
1763                                 if ( rs->sr_err != LDAP_REFERRAL ) {
1764                                         Debug( LDAP_DEBUG_ANY,
1765                                                 "%s ldap_back_op_result: "
1766                                                 "got referrals with err=%d\n",
1767                                                 op->o_log_prefix,
1768                                                 rs->sr_err, 0 );
1769
1770                                 } else {
1771                                         int     i;
1772
1773                                         for ( i = 0; refs[ i ] != NULL; i++ )
1774                                                 /* count */ ;
1775                                         rs->sr_ref = op->o_tmpalloc( sizeof( struct berval ) * ( i + 1 ),
1776                                                 op->o_tmpmemctx );
1777                                         for ( i = 0; refs[ i ] != NULL; i++ ) {
1778                                                 ber_str2bv( refs[ i ], 0, 0, &rs->sr_ref[ i ] );
1779                                         }
1780                                         BER_BVZERO( &rs->sr_ref[ i ] );
1781                                 }
1782
1783                         } else if ( rs->sr_err == LDAP_REFERRAL ) {
1784                                 Debug( LDAP_DEBUG_ANY,
1785                                         "%s ldap_back_op_result: "
1786                                         "got err=%d with null "
1787                                         "or empty referrals\n",
1788                                         op->o_log_prefix,
1789                                         rs->sr_err, 0 );
1790
1791                                 rs->sr_err = LDAP_NO_SUCH_OBJECT;
1792                         }
1793
1794                         if ( ctrls != NULL ) {
1795                                 rs->sr_ctrls = ctrls;
1796                         }
1797                 }
1798         }
1799
1800         /* if the error in the reply structure is not
1801          * LDAP_SUCCESS, try to map it from client 
1802          * to server error */
1803         if ( !ERR_OK( rs->sr_err ) ) {
1804                 rs->sr_err = slap_map_api2result( rs );
1805
1806                 /* internal ops ( op->o_conn == NULL ) 
1807                  * must not reply to client */
1808                 if ( op->o_conn && !op->o_do_not_cache && match ) {
1809
1810                         /* record the (massaged) matched
1811                          * DN into the reply structure */
1812                         rs->sr_matched = match;
1813                 }
1814         }
1815
1816         if ( rs->sr_err == LDAP_UNAVAILABLE ) {
1817                 if ( !( sendok & LDAP_BACK_RETRYING ) ) {
1818                         if ( LDAP_BACK_QUARANTINE( li ) ) {
1819                                 ldap_back_quarantine( op, rs );
1820                         }
1821                         if ( op->o_conn && ( sendok & LDAP_BACK_SENDERR ) ) {
1822                                 if ( rs->sr_text == NULL ) rs->sr_text = "Proxy operation retry failed";
1823                                 send_ldap_result( op, rs );
1824                         }
1825                 }
1826
1827         } else if ( op->o_conn &&
1828                 ( ( ( sendok & LDAP_BACK_SENDOK ) && ERR_OK( rs->sr_err ) )
1829                         || ( ( sendok & LDAP_BACK_SENDERR ) && rs->sr_err != LDAP_SUCCESS ) ) )
1830         {
1831                 send_ldap_result( op, rs );
1832         }
1833
1834         if ( match ) {
1835                 if ( rs->sr_matched != match ) {
1836                         free( (char *)rs->sr_matched );
1837                 }
1838                 rs->sr_matched = NULL;
1839                 ldap_memfree( match );
1840         }
1841
1842         if ( text ) {
1843                 ldap_memfree( text );
1844         }
1845         rs->sr_text = NULL;
1846
1847         if ( rs->sr_ref ) {
1848                 op->o_tmpfree( rs->sr_ref, op->o_tmpmemctx );
1849                 rs->sr_ref = NULL;
1850         }
1851
1852         if ( refs ) {
1853                 ber_memvfree( (void **)refs );
1854         }
1855
1856         if ( ctrls ) {
1857                 assert( rs->sr_ctrls != NULL );
1858                 ldap_controls_free( ctrls );
1859                 rs->sr_ctrls = NULL;
1860         }
1861
1862         return( ERR_OK( rs->sr_err ) ? LDAP_SUCCESS : rs->sr_err );
1863 }
1864
1865 /* return true if bound, false if failed */
1866 int
1867 ldap_back_retry( ldapconn_t **lcp, Operation *op, SlapReply *rs, ldap_back_send_t sendok )
1868 {
1869         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
1870         int             rc = 0;
1871
1872         assert( lcp != NULL );
1873         assert( *lcp != NULL );
1874
1875         ldap_pvt_thread_mutex_lock( &li->li_conninfo.lai_mutex );
1876
1877         if ( (*lcp)->lc_refcnt == 1 ) {
1878                 int binding = LDAP_BACK_CONN_BINDING( *lcp );
1879
1880                 ldap_pvt_thread_mutex_lock( &li->li_uri_mutex );
1881                 Debug( LDAP_DEBUG_ANY,
1882                         "%s ldap_back_retry: retrying URI=\"%s\" DN=\"%s\"\n",
1883                         op->o_log_prefix, li->li_uri,
1884                         BER_BVISNULL( &(*lcp)->lc_bound_ndn ) ?
1885                                 "" : (*lcp)->lc_bound_ndn.bv_val );
1886                 ldap_pvt_thread_mutex_unlock( &li->li_uri_mutex );
1887
1888                 ldap_unbind_ext( (*lcp)->lc_ld, NULL, NULL );
1889                 (*lcp)->lc_ld = NULL;
1890                 LDAP_BACK_CONN_ISBOUND_CLEAR( (*lcp) );
1891
1892                 /* lc here must be the regular lc, reset and ready for init */
1893                 rc = ldap_back_prepare_conn( *lcp, op, rs, sendok );
1894                 if ( rc != LDAP_SUCCESS ) {
1895                         /* freeit, because lc_refcnt == 1 */
1896                         (*lcp)->lc_refcnt = 0;
1897                         (void)ldap_back_freeconn( li, *lcp, 0 );
1898                         *lcp = NULL;
1899                         rc = 0;
1900
1901                 } else if ( ( sendok & LDAP_BACK_BINDING ) ) {
1902                         if ( binding ) {
1903                                 LDAP_BACK_CONN_BINDING_SET( *lcp );
1904                         }
1905                         rc = 1;
1906
1907                 } else {
1908                         rc = ldap_back_dobind_int( lcp, op, rs, sendok, 0, 0 );
1909                         if ( rc == 0 && *lcp != NULL ) {
1910                                 /* freeit, because lc_refcnt == 1 */
1911                                 (*lcp)->lc_refcnt = 0;
1912                                 LDAP_BACK_CONN_TAINTED_SET( *lcp );
1913                                 (void)ldap_back_freeconn( li, *lcp, 0 );
1914                                 *lcp = NULL;
1915                         }
1916                 }
1917
1918         } else {
1919                 Debug( LDAP_DEBUG_TRACE,
1920                         "ldap_back_retry: conn %p refcnt=%u unable to retry.\n",
1921                         (void *)(*lcp), (*lcp)->lc_refcnt, 0 );
1922
1923                 LDAP_BACK_CONN_TAINTED_SET( *lcp );
1924                 ldap_back_release_conn_lock( li, lcp, 0 );
1925                 assert( *lcp == NULL );
1926
1927                 if ( sendok & LDAP_BACK_SENDERR ) {
1928                         rs->sr_err = LDAP_UNAVAILABLE;
1929                         rs->sr_text = "Unable to retry";
1930                         send_ldap_result( op, rs );
1931                 }
1932         }
1933
1934         ldap_pvt_thread_mutex_unlock( &li->li_conninfo.lai_mutex );
1935
1936         return rc;
1937 }
1938
1939 static int
1940 ldap_back_is_proxy_authz( Operation *op, SlapReply *rs, ldap_back_send_t sendok,
1941         struct berval *binddn, struct berval *bindcred )
1942 {
1943         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
1944         struct berval   ndn;
1945         int             dobind = 0;
1946
1947         if ( op->o_conn == NULL || op->o_do_not_cache ) {
1948                 goto done;
1949         }
1950
1951         /* don't proxyAuthz if protocol is not LDAPv3 */
1952         switch ( li->li_version ) {
1953         case LDAP_VERSION3:
1954                 break;
1955
1956         case 0:
1957                 if ( op->o_protocol == 0 || op->o_protocol == LDAP_VERSION3 ) {
1958                         break;
1959                 }
1960                 /* fall thru */
1961
1962         default:
1963                 rs->sr_err = LDAP_UNWILLING_TO_PERFORM;
1964                 if ( sendok & LDAP_BACK_SENDERR ) {
1965                         send_ldap_result( op, rs );
1966                         dobind = -1;
1967                 }
1968                 goto done;
1969         }
1970
1971         /* safe default */
1972         *binddn = slap_empty_bv;
1973         *bindcred = slap_empty_bv;
1974
1975         if ( !BER_BVISNULL( &op->o_conn->c_ndn ) ) {
1976                 ndn = op->o_conn->c_ndn;
1977
1978         } else {
1979                 ndn = op->o_ndn;
1980         }
1981
1982         switch ( li->li_idassert_mode ) {
1983         case LDAP_BACK_IDASSERT_LEGACY:
1984                 if ( !BER_BVISNULL( &ndn ) && !BER_BVISEMPTY( &ndn ) ) {
1985                         if ( !BER_BVISNULL( &li->li_idassert_authcDN ) && !BER_BVISEMPTY( &li->li_idassert_authcDN ) )
1986                         {
1987                                 *binddn = li->li_idassert_authcDN;
1988                                 *bindcred = li->li_idassert_passwd;
1989                                 dobind = 1;
1990                         }
1991                 }
1992                 break;
1993
1994         default:
1995                 /* NOTE: rootdn can always idassert */
1996                 if ( BER_BVISNULL( &ndn )
1997                         && li->li_idassert_authz == NULL
1998                         && !( li->li_idassert_flags & LDAP_BACK_AUTH_AUTHZ_ALL ) )
1999                 {
2000                         if ( li->li_idassert_flags & LDAP_BACK_AUTH_PRESCRIPTIVE ) {
2001                                 rs->sr_err = LDAP_INAPPROPRIATE_AUTH;
2002                                 if ( sendok & LDAP_BACK_SENDERR ) {
2003                                         send_ldap_result( op, rs );
2004                                         dobind = -1;
2005                                 }
2006
2007                         } else {
2008                                 rs->sr_err = LDAP_SUCCESS;
2009                                 *binddn = slap_empty_bv;
2010                                 *bindcred = slap_empty_bv;
2011                                 break;
2012                         }
2013
2014                         goto done;
2015
2016                 } else if ( li->li_idassert_authz && !be_isroot( op ) ) {
2017                         struct berval authcDN;
2018
2019                         if ( BER_BVISNULL( &ndn ) ) {
2020                                 authcDN = slap_empty_bv;
2021
2022                         } else {
2023                                 authcDN = ndn;
2024                         }       
2025                         rs->sr_err = slap_sasl_matches( op, li->li_idassert_authz,
2026                                         &authcDN, &authcDN );
2027                         if ( rs->sr_err != LDAP_SUCCESS ) {
2028                                 if ( li->li_idassert_flags & LDAP_BACK_AUTH_PRESCRIPTIVE ) {
2029                                         if ( sendok & LDAP_BACK_SENDERR ) {
2030                                                 send_ldap_result( op, rs );
2031                                                 dobind = -1;
2032                                         }
2033
2034                                 } else {
2035                                         rs->sr_err = LDAP_SUCCESS;
2036                                         *binddn = slap_empty_bv;
2037                                         *bindcred = slap_empty_bv;
2038                                         break;
2039                                 }
2040
2041                                 goto done;
2042                         }
2043                 }
2044
2045                 *binddn = li->li_idassert_authcDN;
2046                 *bindcred = li->li_idassert_passwd;
2047                 dobind = 1;
2048                 break;
2049         }
2050
2051 done:;
2052         return dobind;
2053 }
2054
2055 static int
2056 ldap_back_proxy_authz_bind(
2057         ldapconn_t              *lc,
2058         Operation               *op,
2059         SlapReply               *rs,
2060         ldap_back_send_t        sendok,
2061         struct berval           *binddn,
2062         struct berval           *bindcred )
2063 {
2064         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
2065         struct berval   ndn;
2066         int             msgid;
2067         int             rc;
2068
2069         if ( !BER_BVISNULL( &op->o_conn->c_ndn ) ) {
2070                 ndn = op->o_conn->c_ndn;
2071
2072         } else {
2073                 ndn = op->o_ndn;
2074         }
2075
2076         if ( li->li_idassert_authmethod == LDAP_AUTH_SASL ) {
2077 #ifdef HAVE_CYRUS_SASL
2078                 void            *defaults = NULL;
2079                 struct berval   authzID = BER_BVNULL;
2080                 int             freeauthz = 0;
2081
2082                 /* if SASL supports native authz, prepare for it */
2083                 if ( ( !op->o_do_not_cache || !op->o_is_auth_check ) &&
2084                                 ( li->li_idassert_flags & LDAP_BACK_AUTH_NATIVE_AUTHZ ) )
2085                 {
2086                         switch ( li->li_idassert_mode ) {
2087                         case LDAP_BACK_IDASSERT_OTHERID:
2088                         case LDAP_BACK_IDASSERT_OTHERDN:
2089                                 authzID = li->li_idassert_authzID;
2090                                 break;
2091
2092                         case LDAP_BACK_IDASSERT_ANONYMOUS:
2093                                 BER_BVSTR( &authzID, "dn:" );
2094                                 break;
2095
2096                         case LDAP_BACK_IDASSERT_SELF:
2097                                 if ( BER_BVISNULL( &ndn ) ) {
2098                                         /* connection is not authc'd, so don't idassert */
2099                                         BER_BVSTR( &authzID, "dn:" );
2100                                         break;
2101                                 }
2102                                 authzID.bv_len = STRLENOF( "dn:" ) + ndn.bv_len;
2103                                 authzID.bv_val = slap_sl_malloc( authzID.bv_len + 1, op->o_tmpmemctx );
2104                                 AC_MEMCPY( authzID.bv_val, "dn:", STRLENOF( "dn:" ) );
2105                                 AC_MEMCPY( authzID.bv_val + STRLENOF( "dn:" ),
2106                                                 ndn.bv_val, ndn.bv_len + 1 );
2107                                 freeauthz = 1;
2108                                 break;
2109
2110                         default:
2111                                 break;
2112                         }
2113                 }
2114
2115                 if ( li->li_idassert_secprops != NULL ) {
2116                         rs->sr_err = ldap_set_option( lc->lc_ld,
2117                                 LDAP_OPT_X_SASL_SECPROPS,
2118                                 (void *)li->li_idassert_secprops );
2119
2120                         if ( rs->sr_err != LDAP_OPT_SUCCESS ) {
2121                                 rs->sr_err = LDAP_OTHER;
2122                                 if ( sendok & LDAP_BACK_SENDERR ) {
2123                                         send_ldap_result( op, rs );
2124                                 }
2125                                 LDAP_BACK_CONN_ISBOUND_CLEAR( lc );
2126                                 goto done;
2127                         }
2128                 }
2129
2130                 defaults = lutil_sasl_defaults( lc->lc_ld,
2131                                 li->li_idassert_sasl_mech.bv_val,
2132                                 li->li_idassert_sasl_realm.bv_val,
2133                                 li->li_idassert_authcID.bv_val,
2134                                 li->li_idassert_passwd.bv_val,
2135                                 authzID.bv_val );
2136                 if ( defaults == NULL ) {
2137                         rs->sr_err = LDAP_OTHER;
2138                         LDAP_BACK_CONN_ISBOUND_CLEAR( lc );
2139                         if ( sendok & LDAP_BACK_SENDERR ) {
2140                                 send_ldap_result( op, rs );
2141                         }
2142                         goto done;
2143                 }
2144
2145                 rs->sr_err = ldap_sasl_interactive_bind_s( lc->lc_ld, binddn->bv_val,
2146                                 li->li_idassert_sasl_mech.bv_val, NULL, NULL,
2147                                 LDAP_SASL_QUIET, lutil_sasl_interact,
2148                                 defaults );
2149
2150                 switch ( rs->sr_err ) {
2151                 case LDAP_SUCCESS:
2152                         LDAP_BACK_CONN_ISBOUND_SET( lc );
2153                         break;
2154
2155                 case LDAP_LOCAL_ERROR:
2156                         /* list client API error codes that require
2157                          * to taint the connection */
2158                         /* FIXME: should actually retry? */
2159                         LDAP_BACK_CONN_TAINTED_SET( lc );
2160
2161                         /* fallthru */
2162
2163                 default:
2164                         LDAP_BACK_CONN_ISBOUND_CLEAR( lc );
2165                         rs->sr_err = slap_map_api2result( rs );
2166                         if ( sendok & LDAP_BACK_SENDERR ) {
2167                                 send_ldap_result( op, rs );
2168                         }
2169                         break;
2170                 }
2171
2172                 lutil_sasl_freedefs( defaults );
2173                 if ( freeauthz ) {
2174                         slap_sl_free( authzID.bv_val, op->o_tmpmemctx );
2175                 }
2176
2177                 goto done;
2178 #endif /* HAVE_CYRUS_SASL */
2179         }
2180
2181         switch ( li->li_idassert_authmethod ) {
2182         case LDAP_AUTH_NONE:
2183                 /* FIXME: do we really need this? */
2184                 BER_BVSTR( binddn, "" );
2185                 BER_BVSTR( bindcred, "" );
2186                 /* fallthru */
2187
2188         case LDAP_AUTH_SIMPLE:
2189                 rs->sr_err = ldap_sasl_bind( lc->lc_ld,
2190                                 binddn->bv_val, LDAP_SASL_SIMPLE,
2191                                 bindcred, NULL, NULL, &msgid );
2192                 rc = ldap_back_op_result( lc, op, rs, msgid,
2193                         -1, ( sendok | LDAP_BACK_BINDING ) );
2194                 break;
2195
2196         default:
2197                 /* unsupported! */
2198                 LDAP_BACK_CONN_ISBOUND_CLEAR( lc );
2199                 rs->sr_err = LDAP_AUTH_METHOD_NOT_SUPPORTED;
2200                 if ( sendok & LDAP_BACK_SENDERR ) {
2201                         send_ldap_result( op, rs );
2202                 }
2203                 goto done;
2204         }
2205
2206         if ( rc == LDAP_SUCCESS ) {
2207                 /* set rebind stuff in case of successful proxyAuthz bind,
2208                  * so that referral chasing is attempted using the right
2209                  * identity */
2210                 LDAP_BACK_CONN_ISBOUND_SET( lc );
2211                 ber_bvreplace( &lc->lc_bound_ndn, binddn );
2212
2213                 if ( !BER_BVISNULL( &lc->lc_cred ) ) {
2214                         memset( lc->lc_cred.bv_val, 0,
2215                                         lc->lc_cred.bv_len );
2216                 }
2217
2218                 if ( LDAP_BACK_SAVECRED( li ) ) {
2219                         ber_bvreplace( &lc->lc_cred, bindcred );
2220                         ldap_set_rebind_proc( lc->lc_ld, li->li_rebind_f, lc );
2221
2222                 } else {
2223                         lc->lc_cred.bv_len = 0;
2224                 }
2225         }
2226 done:;
2227         return LDAP_BACK_CONN_ISBOUND( lc );
2228 }
2229
2230 /*
2231  * ldap_back_proxy_authz_ctrl() prepends a proxyAuthz control
2232  * to existing server-side controls if required; if not,
2233  * the existing server-side controls are placed in *pctrls.
2234  * The caller, after using the controls in client API 
2235  * operations, if ( *pctrls != op->o_ctrls ), should
2236  * free( (*pctrls)[ 0 ] ) and free( *pctrls ).
2237  * The function returns success if the control could
2238  * be added if required, or if it did nothing; in the future,
2239  * it might return some error if it failed.
2240  * 
2241  * if no bind took place yet, but the connection is bound
2242  * and the "proxyauthzdn" is set, then bind as "proxyauthzdn" 
2243  * and explicitly add proxyAuthz the control to every operation
2244  * with the dn bound to the connection as control value.
2245  *
2246  * If no server-side controls are defined for the operation,
2247  * simply add the proxyAuthz control; otherwise, if the
2248  * proxyAuthz control is not already set, add it as
2249  * the first one
2250  *
2251  * FIXME: is controls order significant for security?
2252  * ANSWER: controls ordering and interoperability
2253  * must be indicated by the specs of each control; if none
2254  * is specified, the order is irrelevant.
2255  */
2256 int
2257 ldap_back_proxy_authz_ctrl(
2258                 Operation       *op,
2259                 SlapReply       *rs,
2260                 struct berval   *bound_ndn,
2261                 int             version,
2262                 slap_idassert_t *si,
2263                 LDAPControl     *ctrl )
2264 {
2265         slap_idassert_mode_t    mode;
2266         struct berval           assertedID,
2267                                 ndn;
2268         int                     isroot = 0;
2269
2270         rs->sr_err = SLAP_CB_CONTINUE;
2271
2272         /* FIXME: SASL/EXTERNAL over ldapi:// doesn't honor the authcID,
2273          * but if it is not set this test fails.  We need a different
2274          * means to detect if idassert is enabled */
2275         if ( ( BER_BVISNULL( &si->si_bc.sb_authcId ) || BER_BVISEMPTY( &si->si_bc.sb_authcId ) )
2276                 && ( BER_BVISNULL( &si->si_bc.sb_binddn ) || BER_BVISEMPTY( &si->si_bc.sb_binddn ) )
2277                 && BER_BVISNULL( &si->si_bc.sb_saslmech ) )
2278         {
2279                 goto done;
2280         }
2281
2282         if ( !op->o_conn || op->o_do_not_cache || ( isroot = be_isroot( op ) ) ) {
2283                 goto done;
2284         }
2285
2286         if ( op->o_tag == LDAP_REQ_BIND ) {
2287                 ndn = op->o_req_ndn;
2288
2289         } else if ( !BER_BVISNULL( &op->o_conn->c_ndn ) ) {
2290                 ndn = op->o_conn->c_ndn;
2291
2292         } else {
2293                 ndn = op->o_ndn;
2294         }
2295
2296         if ( si->si_mode == LDAP_BACK_IDASSERT_LEGACY ) {
2297                 if ( op->o_proxy_authz ) {
2298                         /*
2299                          * FIXME: we do not want to perform proxyAuthz
2300                          * on behalf of the client, because this would
2301                          * be performed with "proxyauthzdn" privileges.
2302                          *
2303                          * This might actually be too strict, since
2304                          * the "proxyauthzdn" authzTo, and each entry's
2305                          * authzFrom attributes may be crafted
2306                          * to avoid unwanted proxyAuthz to take place.
2307                          */
2308 #if 0
2309                         rs->sr_err = LDAP_UNWILLING_TO_PERFORM;
2310                         rs->sr_text = "proxyAuthz not allowed within namingContext";
2311 #endif
2312                         goto done;
2313                 }
2314
2315                 if ( !BER_BVISNULL( bound_ndn ) ) {
2316                         goto done;
2317                 }
2318
2319                 if ( BER_BVISNULL( &ndn ) ) {
2320                         goto done;
2321                 }
2322
2323                 if ( BER_BVISNULL( &si->si_bc.sb_binddn ) ) {
2324                         goto done;
2325                 }
2326
2327         } else if ( si->si_bc.sb_method == LDAP_AUTH_SASL ) {
2328                 if ( ( si->si_flags & LDAP_BACK_AUTH_NATIVE_AUTHZ ) )
2329                 {
2330                         /* already asserted in SASL via native authz */
2331                         goto done;
2332                 }
2333
2334         } else if ( si->si_authz && !isroot ) {
2335                 int             rc;
2336                 struct berval authcDN;
2337
2338                 if ( BER_BVISNULL( &ndn ) ) {
2339                         authcDN = slap_empty_bv;
2340                 } else {
2341                         authcDN = ndn;
2342                 }
2343                 rc = slap_sasl_matches( op, si->si_authz,
2344                                 &authcDN, &authcDN );
2345                 if ( rc != LDAP_SUCCESS ) {
2346                         if ( si->si_flags & LDAP_BACK_AUTH_PRESCRIPTIVE ) {
2347                                 /* ndn is not authorized
2348                                  * to use idassert */
2349                                 rs->sr_err = rc;
2350                         }
2351                         goto done;
2352                 }
2353         }
2354
2355         if ( op->o_proxy_authz ) {
2356                 /*
2357                  * FIXME: we can:
2358                  * 1) ignore the already set proxyAuthz control
2359                  * 2) leave it in place, and don't set ours
2360                  * 3) add both
2361                  * 4) reject the operation
2362                  *
2363                  * option (4) is very drastic
2364                  * option (3) will make the remote server reject
2365                  * the operation, thus being equivalent to (4)
2366                  * option (2) will likely break the idassert
2367                  * assumptions, so we cannot accept it;
2368                  * option (1) means that we are contradicting
2369                  * the client's reques.
2370                  *
2371                  * I think (4) is the only correct choice.
2372                  */
2373                 rs->sr_err = LDAP_UNWILLING_TO_PERFORM;
2374                 rs->sr_text = "proxyAuthz not allowed within namingContext";
2375         }
2376
2377         if ( op->o_is_auth_check ) {
2378                 mode = LDAP_BACK_IDASSERT_NOASSERT;
2379
2380         } else {
2381                 mode = si->si_mode;
2382         }
2383
2384         switch ( mode ) {
2385         case LDAP_BACK_IDASSERT_LEGACY:
2386                 /* original behavior:
2387                  * assert the client's identity */
2388         case LDAP_BACK_IDASSERT_SELF:
2389                 assertedID = ndn;
2390                 break;
2391
2392         case LDAP_BACK_IDASSERT_ANONYMOUS:
2393                 /* assert "anonymous" */
2394                 assertedID = slap_empty_bv;
2395                 break;
2396
2397         case LDAP_BACK_IDASSERT_NOASSERT:
2398                 /* don't assert; bind as proxyauthzdn */
2399                 goto done;
2400
2401         case LDAP_BACK_IDASSERT_OTHERID:
2402         case LDAP_BACK_IDASSERT_OTHERDN:
2403                 /* assert idassert DN */
2404                 assertedID = si->si_bc.sb_authzId;
2405                 break;
2406
2407         default:
2408                 assert( 0 );
2409         }
2410
2411         /* if we got here, "" is allowed to proxyAuthz */
2412         if ( BER_BVISNULL( &assertedID ) ) {
2413                 assertedID = slap_empty_bv;
2414         }
2415
2416         /* don't idassert the bound DN (ITS#4497) */
2417         if ( dn_match( &assertedID, bound_ndn ) ) {
2418                 goto done;
2419         }
2420
2421         ctrl->ldctl_oid = LDAP_CONTROL_PROXY_AUTHZ;
2422
2423         switch ( si->si_mode ) {
2424         /* already in u:ID or dn:DN form */
2425         case LDAP_BACK_IDASSERT_OTHERID:
2426         case LDAP_BACK_IDASSERT_OTHERDN:
2427                 ber_dupbv_x( &ctrl->ldctl_value, &assertedID, op->o_tmpmemctx );
2428                 rs->sr_err = LDAP_SUCCESS;
2429                 break;
2430
2431         /* needs the dn: prefix */
2432         default:
2433                 ctrl->ldctl_value.bv_len = assertedID.bv_len + STRLENOF( "dn:" );
2434                 ctrl->ldctl_value.bv_val = op->o_tmpalloc( ctrl->ldctl_value.bv_len + 1,
2435                                 op->o_tmpmemctx );
2436                 AC_MEMCPY( ctrl->ldctl_value.bv_val, "dn:", STRLENOF( "dn:" ) );
2437                 AC_MEMCPY( &ctrl->ldctl_value.bv_val[ STRLENOF( "dn:" ) ],
2438                                 assertedID.bv_val, assertedID.bv_len + 1 );
2439                 rs->sr_err = LDAP_SUCCESS;
2440                 break;
2441         }
2442
2443         /* Older versions of <draft-weltman-ldapv3-proxy> required
2444          * to encode the value of the authzID (and called it proxyDN);
2445          * this hack provides compatibility with those DSAs that
2446          * implement it this way */
2447         if ( si->si_flags & LDAP_BACK_AUTH_OBSOLETE_ENCODING_WORKAROUND ) {
2448                 struct berval           authzID = ctrl->ldctl_value;
2449                 BerElementBuffer        berbuf;
2450                 BerElement              *ber = (BerElement *)&berbuf;
2451                 ber_tag_t               tag;
2452
2453                 ber_init2( ber, 0, LBER_USE_DER );
2454                 ber_set_option( ber, LBER_OPT_BER_MEMCTX, &op->o_tmpmemctx );
2455
2456                 tag = ber_printf( ber, "O", &authzID );
2457                 if ( tag == LBER_ERROR ) {
2458                         rs->sr_err = LDAP_OTHER;
2459                         goto free_ber;
2460                 }
2461
2462                 if ( ber_flatten2( ber, &ctrl->ldctl_value, 1 ) == -1 ) {
2463                         rs->sr_err = LDAP_OTHER;
2464                         goto free_ber;
2465                 }
2466
2467                 rs->sr_err = LDAP_SUCCESS;
2468
2469 free_ber:;
2470                 op->o_tmpfree( authzID.bv_val, op->o_tmpmemctx );
2471                 ber_free_buf( ber );
2472
2473                 if ( rs->sr_err != LDAP_SUCCESS ) {
2474                         goto done;
2475                 }
2476
2477         } else if ( si->si_flags & LDAP_BACK_AUTH_OBSOLETE_PROXY_AUTHZ ) {
2478                 struct berval           authzID = ctrl->ldctl_value,
2479                                         tmp;
2480                 BerElementBuffer        berbuf;
2481                 BerElement              *ber = (BerElement *)&berbuf;
2482                 ber_tag_t               tag;
2483
2484                 if ( strncasecmp( authzID.bv_val, "dn:", STRLENOF( "dn:" ) ) != 0 ) {
2485                         rs->sr_err = LDAP_PROTOCOL_ERROR;
2486                         goto done;
2487                 }
2488
2489                 tmp = authzID;
2490                 tmp.bv_val += STRLENOF( "dn:" );
2491                 tmp.bv_len -= STRLENOF( "dn:" );
2492
2493                 ber_init2( ber, 0, LBER_USE_DER );
2494                 ber_set_option( ber, LBER_OPT_BER_MEMCTX, &op->o_tmpmemctx );
2495
2496                 /* apparently, Mozilla API encodes this
2497                  * as "SEQUENCE { LDAPDN }" */
2498                 tag = ber_printf( ber, "{O}", &tmp );
2499                 if ( tag == LBER_ERROR ) {
2500                         rs->sr_err = LDAP_OTHER;
2501                         goto free_ber2;
2502                 }
2503
2504                 if ( ber_flatten2( ber, &ctrl->ldctl_value, 1 ) == -1 ) {
2505                         rs->sr_err = LDAP_OTHER;
2506                         goto free_ber2;
2507                 }
2508
2509                 ctrl->ldctl_oid = LDAP_CONTROL_OBSOLETE_PROXY_AUTHZ;
2510                 rs->sr_err = LDAP_SUCCESS;
2511
2512 free_ber2:;
2513                 op->o_tmpfree( authzID.bv_val, op->o_tmpmemctx );
2514                 ber_free_buf( ber );
2515
2516                 if ( rs->sr_err != LDAP_SUCCESS ) {
2517                         goto done;
2518                 }
2519         }
2520
2521 done:;
2522
2523         return rs->sr_err;
2524 }
2525
2526 /*
2527  * Add controls;
2528  *
2529  * if any needs to be added, it is prepended to existing ones,
2530  * in a newly allocated array.  The companion function
2531  * ldap_back_controls_free() must be used to restore the original
2532  * status of op->o_ctrls.
2533  */
2534 int
2535 ldap_back_controls_add(
2536                 Operation       *op,
2537                 SlapReply       *rs,
2538                 ldapconn_t      *lc,
2539                 LDAPControl     ***pctrls )
2540 {
2541         ldapinfo_t      *li = (ldapinfo_t *)op->o_bd->be_private;
2542
2543         LDAPControl     **ctrls = NULL;
2544         /* set to the maximum number of controls this backend can add */
2545         LDAPControl     c[ 2 ] = { { 0 } };
2546         int             n = 0, i, j1 = 0, j2 = 0;
2547
2548         *pctrls = NULL;
2549
2550         rs->sr_err = LDAP_SUCCESS;
2551
2552         /* don't add controls if protocol is not LDAPv3 */
2553         switch ( li->li_version ) {
2554         case LDAP_VERSION3:
2555                 break;
2556
2557         case 0:
2558                 if ( op->o_protocol == 0 || op->o_protocol == LDAP_VERSION3 ) {
2559                         break;
2560                 }
2561                 /* fall thru */
2562
2563         default:
2564                 goto done;
2565         }
2566
2567         /* put controls that go __before__ existing ones here */
2568
2569         /* proxyAuthz for identity assertion */
2570         switch ( ldap_back_proxy_authz_ctrl( op, rs, &lc->lc_bound_ndn,
2571                 li->li_version, &li->li_idassert, &c[ j1 ] ) )
2572         {
2573         case SLAP_CB_CONTINUE:
2574                 break;
2575
2576         case LDAP_SUCCESS:
2577                 j1++;
2578                 break;
2579
2580         default:
2581                 goto done;
2582         }
2583
2584         /* put controls that go __after__ existing ones here */
2585
2586 #ifdef SLAP_CONTROL_X_SESSION_TRACKING
2587         /* FIXME: according to <draft-wahl-ldap-session>, 
2588          * the server should check if the control can be added
2589          * based on the identity of the client and so */
2590
2591         /* session tracking */
2592         if ( LDAP_BACK_ST_REQUEST( li ) ) {
2593                 switch ( slap_ctrl_session_tracking_request_add( op, rs, &c[ j1 + j2 ] ) ) {
2594                 case SLAP_CB_CONTINUE:
2595                         break;
2596
2597                 case LDAP_SUCCESS:
2598                         j2++;
2599                         break;
2600
2601                 default:
2602                         goto done;
2603                 }
2604         }
2605 #endif /* SLAP_CONTROL_X_SESSION_TRACKING */
2606
2607         if ( rs->sr_err == SLAP_CB_CONTINUE ) {
2608                 rs->sr_err = LDAP_SUCCESS;
2609         }
2610
2611         /* if nothing to do, just bail out */
2612         if ( j1 == 0 && j2 == 0 ) {
2613                 goto done;
2614         }
2615
2616         assert( j1 + j1 <= sizeof( c )/sizeof(LDAPControl) );
2617
2618         if ( op->o_ctrls ) {
2619                 for ( n = 0; op->o_ctrls[ n ]; n++ )
2620                         /* just count ctrls */ ;
2621         }
2622
2623         ctrls = op->o_tmpalloc( (n + j1 + j2 + 1) * sizeof( LDAPControl * ) + ( j1 + j2 ) * sizeof( LDAPControl ),
2624                         op->o_tmpmemctx );
2625         if ( j1 ) {
2626                 ctrls[ 0 ] = (LDAPControl *)&ctrls[ n + j1 + j2 + 1 ];
2627                 *ctrls[ 0 ] = c[ 0 ];
2628                 for ( i = 1; i < j1; i++ ) {
2629                         ctrls[ i ] = &ctrls[ 0 ][ i ];
2630                         *ctrls[ i ] = c[ i ];
2631                 }
2632         }
2633
2634         i = 0;
2635         if ( op->o_ctrls ) {
2636                 for ( i = 0; op->o_ctrls[ i ]; i++ ) {
2637                         ctrls[ i + j1 ] = op->o_ctrls[ i ];
2638                 }
2639         }
2640
2641         n += j1;
2642         if ( j2 ) {
2643                 ctrls[ n ] = (LDAPControl *)&ctrls[ n + j2 + 1 ] + j1;
2644                 *ctrls[ n ] = c[ j1 ];
2645                 for ( i = 1; i < j2; i++ ) {
2646                         ctrls[ n + i ] = &ctrls[ n ][ i ];
2647                         *ctrls[ n + i ] = c[ i ];
2648                 }
2649         }
2650
2651         ctrls[ n + j2 ] = NULL;
2652
2653 done:;
2654         if ( ctrls == NULL ) {
2655                 ctrls = op->o_ctrls;
2656         }
2657
2658         *pctrls = ctrls;
2659         
2660         return rs->sr_err;
2661 }
2662
2663 int
2664 ldap_back_controls_free( Operation *op, SlapReply *rs, LDAPControl ***pctrls )
2665 {
2666         LDAPControl     **ctrls = *pctrls;
2667
2668         /* we assume that the controls added by the proxy come first,
2669          * so as soon as we find op->o_ctrls[ 0 ] we can stop */
2670         if ( ctrls && ctrls != op->o_ctrls ) {
2671                 int             i = 0, n = 0, n_added;
2672                 LDAPControl     *lower, *upper;
2673
2674                 assert( ctrls[ 0 ] != NULL );
2675
2676                 for ( n = 0; ctrls[ n ] != NULL; n++ )
2677                         /* count 'em */ ;
2678
2679                 if ( op->o_ctrls ) {
2680                         for ( i = 0; op->o_ctrls[ i ] != NULL; i++ )
2681                                 /* count 'em */ ;
2682                 }
2683
2684                 n_added = n - i;
2685                 lower = (LDAPControl *)&ctrls[ n ];
2686                 upper = &lower[ n_added ];
2687
2688                 for ( i = 0; ctrls[ i ] != NULL; i++ ) {
2689                         if ( ctrls[ i ] < lower || ctrls[ i ] >= upper ) {
2690                                 /* original; don't touch */
2691                                 continue;
2692                         }
2693
2694                         if ( !BER_BVISNULL( &ctrls[ i ]->ldctl_value ) ) {
2695                                 op->o_tmpfree( ctrls[ i ]->ldctl_value.bv_val, op->o_tmpmemctx );
2696                         }
2697                 }
2698
2699                 op->o_tmpfree( ctrls, op->o_tmpmemctx );
2700         } 
2701
2702         *pctrls = NULL;
2703
2704         return 0;
2705 }