]> git.sur5r.net Git - openldap/blob - servers/slapd/back-ldap/distproc.c
Merge remote-tracking branch 'origin/mdb.RE/0.9'
[openldap] / servers / slapd / back-ldap / distproc.c
1 /* distproc.c - implement distributed procedures */
2 /* $OpenLDAP$ */
3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4  *
5  * Copyright 2005-2014 The OpenLDAP Foundation.
6  * Portions Copyright 2003 Howard Chu.
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted only as authorized by the OpenLDAP
11  * Public License.
12  *
13  * A copy of this license is available in the file LICENSE in the
14  * top-level directory of the distribution or, alternatively, at
15  * <http://www.OpenLDAP.org/license.html>.
16  */
17 /* ACKNOWLEDGEMENTS:
18  * This work was initially developed by Pierangelo Masarati for inclusion
19  * in OpenLDAP Software.
20  * Based on back-ldap and slapo-chain, developed by Howard Chu
21  */
22
23 #include "portable.h"
24
25 #include <stdio.h>
26
27 #include <ac/string.h>
28 #include <ac/socket.h>
29
30 #include "slap.h"
31
32 #ifdef SLAP_DISTPROC
33
34 #include "back-ldap.h"
35
36 #include "config.h"
37
38 /*
39  * From <draft-sermersheim-ldap-distproc>
40  *
41
42       ContinuationReference ::= SET {
43          referralURI      [0] SET SIZE (1..MAX) OF URI,
44          localReference   [2] LDAPDN,
45          referenceType    [3] ReferenceType,
46          remainingName    [4] RelativeLDAPDN OPTIONAL,
47          searchScope      [5] SearchScope OPTIONAL,
48          searchedSubtrees [6] SearchedSubtrees OPTIONAL,
49          failedName       [7] LDAPDN OPTIONAL,
50          ...  }
51
52       ReferenceType ::= ENUMERATED {
53          superior               (0),
54          subordinate            (1),
55          cross                  (2),
56          nonSpecificSubordinate (3),
57          supplier               (4),
58          master                 (5),
59          immediateSuperior      (6),
60          self                   (7),
61          ...  }
62
63       SearchScope ::= ENUMERATED {
64          baseObject         (0),
65          singleLevel        (1),
66          wholeSubtree       (2),
67          subordinateSubtree (3),
68          ...  }
69
70    SearchedSubtrees ::= SET OF RelativeLDAPDN
71
72    LDAPDN, RelativeLDAPDN, and LDAPString, are defined in [RFC2251].
73
74  */
75
76 typedef enum ReferenceType_t {
77         LDAP_DP_RT_UNKNOWN                      = -1,
78         LDAP_DP_RT_SUPERIOR                     = 0,
79         LDAP_DP_RT_SUBORDINATE                  = 1,
80         LDAP_DP_RT_CROSS                        = 2,
81         LDAP_DP_RT_NONSPECIFICSUBORDINATE       = 3,
82         LDAP_DP_RT_SUPPLIER                     = 4,
83         LDAP_DP_RT_MASTER                       = 5,
84         LDAP_DP_RT_IMMEDIATESUPERIOR            = 6,
85         LDAP_DP_RT_SELF                         = 7,
86         LDAP_DP_RT_LAST
87 } ReferenceType_t;
88
89 typedef enum SearchScope_t {
90         LDAP_DP_SS_UNKNOWN                      = -1,
91         LDAP_DP_SS_BASEOBJECT                   = 0,
92         LDAP_DP_SS_SINGLELEVEL                  = 1,
93         LDAP_DP_SS_WHOLESUBTREE                 = 2,
94         LDAP_DP_SS_SUBORDINATESUBTREE           = 3,
95         LDAP_DP_SS_LAST
96 } SearchScope_t;
97
98 typedef struct ContinuationReference_t {
99         BerVarray               cr_referralURI;
100         /* ?                    [1] ? */
101         struct berval           cr_localReference;
102         ReferenceType_t         cr_referenceType;
103         struct berval           cr_remainingName;
104         SearchScope_t           cr_searchScope;
105         BerVarray               cr_searchedSubtrees;
106         struct berval           cr_failedName;
107 } ContinuationReference_t;
108 #define CR_INIT         { NULL, BER_BVNULL, LDAP_DP_RT_UNKNOWN, BER_BVNULL, LDAP_DP_SS_UNKNOWN, NULL, BER_BVNULL }
109
110 #ifdef unused
111 static struct berval    bv2rt[] = {
112         BER_BVC( "superior" ),
113         BER_BVC( "subordinate" ),
114         BER_BVC( "cross" ),
115         BER_BVC( "nonSpecificSubordinate" ),
116         BER_BVC( "supplier" ),
117         BER_BVC( "master" ),
118         BER_BVC( "immediateSuperior" ),
119         BER_BVC( "self" ),
120         BER_BVNULL
121 };
122
123 static struct berval    bv2ss[] = {
124         BER_BVC( "baseObject" ),
125         BER_BVC( "singleLevel" ),
126         BER_BVC( "wholeSubtree" ),
127         BER_BVC( "subordinateSubtree" ),
128         BER_BVNULL
129 };
130
131 static struct berval *
132 ldap_distproc_rt2bv( ReferenceType_t rt )
133 {
134         return &bv2rt[ rt ];
135 }
136
137 static const char *
138 ldap_distproc_rt2str( ReferenceType_t rt )
139 {
140         return bv2rt[ rt ].bv_val;
141 }
142
143 static ReferenceType_t
144 ldap_distproc_bv2rt( struct berval *bv )
145 {
146         ReferenceType_t         rt;
147
148         for ( rt = 0; !BER_BVISNULL( &bv2rt[ rt ] ); rt++ ) {
149                 if ( ber_bvstrcasecmp( bv, &bv2rt[ rt ] ) == 0 ) {
150                         return rt;
151                 }
152         }
153
154         return LDAP_DP_RT_UNKNOWN;
155 }
156
157 static ReferenceType_t
158 ldap_distproc_str2rt( const char *s )
159 {
160         struct berval   bv;
161
162         ber_str2bv( s, 0, 0, &bv );
163         return ldap_distproc_bv2rt( &bv );
164 }
165
166 static struct berval *
167 ldap_distproc_ss2bv( SearchScope_t ss )
168 {
169         return &bv2ss[ ss ];
170 }
171
172 static const char *
173 ldap_distproc_ss2str( SearchScope_t ss )
174 {
175         return bv2ss[ ss ].bv_val;
176 }
177
178 static SearchScope_t
179 ldap_distproc_bv2ss( struct berval *bv )
180 {
181         ReferenceType_t         ss;
182
183         for ( ss = 0; !BER_BVISNULL( &bv2ss[ ss ] ); ss++ ) {
184                 if ( ber_bvstrcasecmp( bv, &bv2ss[ ss ] ) == 0 ) {
185                         return ss;
186                 }
187         }
188
189         return LDAP_DP_SS_UNKNOWN;
190 }
191
192 static SearchScope_t
193 ldap_distproc_str2ss( const char *s )
194 {
195         struct berval   bv;
196
197         ber_str2bv( s, 0, 0, &bv );
198         return ldap_distproc_bv2ss( &bv );
199 }
200 #endif /* unused */
201
202 /*
203  * NOTE: this overlay assumes that the chainingBehavior control
204  * is registered by the chain overlay; it may move here some time.
205  * This overlay provides support for that control as well.
206  */
207
208
209 static int              sc_returnContRef;
210 #define o_returnContRef                 o_ctrlflag[sc_returnContRef]
211 #define get_returnContRef(op)           ((op)->o_returnContRef & SLAP_CONTROL_MASK)
212
213 static struct berval    slap_EXOP_CHAINEDREQUEST = BER_BVC( LDAP_EXOP_X_CHAINEDREQUEST );
214 static struct berval    slap_FEATURE_CANCHAINOPS = BER_BVC( LDAP_FEATURE_X_CANCHAINOPS );
215
216 static BackendInfo      *lback;
217
218 typedef struct ldap_distproc_t {
219         /* "common" configuration info (anything occurring before an "uri") */
220         ldapinfo_t              *lc_common_li;
221
222         /* current configuration info */
223         ldapinfo_t              *lc_cfg_li;
224
225         /* tree of configured[/generated?] "uri" info */
226         ldap_avl_info_t         lc_lai;
227
228         unsigned                lc_flags;
229 #define LDAP_DISTPROC_F_NONE            (0x00U)
230 #define LDAP_DISTPROC_F_CHAINING        (0x01U)
231 #define LDAP_DISTPROC_F_CACHE_URI       (0x10U)
232
233 #define LDAP_DISTPROC_CHAINING( lc )    ( ( (lc)->lc_flags & LDAP_DISTPROC_F_CHAINING ) == LDAP_DISTPROC_F_CHAINING )
234 #define LDAP_DISTPROC_CACHE_URI( lc )   ( ( (lc)->lc_flags & LDAP_DISTPROC_F_CACHE_URI ) == LDAP_DISTPROC_F_CACHE_URI )
235
236 } ldap_distproc_t;
237
238 static int ldap_distproc_db_init_common( BackendDB      *be );
239 static int ldap_distproc_db_init_one( BackendDB *be );
240 #define ldap_distproc_db_open_one(be)           (lback)->bi_db_open( (be) )
241 #define ldap_distproc_db_close_one(be)          (0)
242 #define ldap_distproc_db_destroy_one(be, ca)    (lback)->bi_db_destroy( (be), (ca) )
243
244 static int
245 ldap_distproc_uri_cmp( const void *c1, const void *c2 )
246 {
247         const ldapinfo_t        *li1 = (const ldapinfo_t *)c1;
248         const ldapinfo_t        *li2 = (const ldapinfo_t *)c2;
249
250         assert( li1->li_bvuri != NULL );
251         assert( !BER_BVISNULL( &li1->li_bvuri[ 0 ] ) );
252         assert( BER_BVISNULL( &li1->li_bvuri[ 1 ] ) );
253
254         assert( li2->li_bvuri != NULL );
255         assert( !BER_BVISNULL( &li2->li_bvuri[ 0 ] ) );
256         assert( BER_BVISNULL( &li2->li_bvuri[ 1 ] ) );
257
258         /* If local DNs don't match, it is definitely not a match */
259         return ber_bvcmp( &li1->li_bvuri[ 0 ], &li2->li_bvuri[ 0 ] );
260 }
261
262 static int
263 ldap_distproc_uri_dup( void *c1, void *c2 )
264 {
265         ldapinfo_t      *li1 = (ldapinfo_t *)c1;
266         ldapinfo_t      *li2 = (ldapinfo_t *)c2;
267
268         assert( li1->li_bvuri != NULL );
269         assert( !BER_BVISNULL( &li1->li_bvuri[ 0 ] ) );
270         assert( BER_BVISNULL( &li1->li_bvuri[ 1 ] ) );
271
272         assert( li2->li_bvuri != NULL );
273         assert( !BER_BVISNULL( &li2->li_bvuri[ 0 ] ) );
274         assert( BER_BVISNULL( &li2->li_bvuri[ 1 ] ) );
275
276         /* Cannot have more than one shared session with same DN */
277         if ( ber_bvcmp( &li1->li_bvuri[ 0 ], &li2->li_bvuri[ 0 ] ) == 0 ) {
278                 return -1;
279         }
280                 
281         return 0;
282 }
283
284 static int
285 ldap_distproc_operational( Operation *op, SlapReply *rs )
286 {
287         /* Trap entries generated by back-ldap.
288          * 
289          * FIXME: we need a better way to recognize them; a cleaner
290          * solution would be to be able to intercept the response
291          * of be_operational(), so that we can divert only those
292          * calls that fail because operational attributes were
293          * requested for entries that do not belong to the underlying
294          * database.  This fix is likely to intercept also entries
295          * generated by back-perl and so. */
296         if ( rs->sr_entry->e_private == NULL ) {
297                 return LDAP_SUCCESS;
298         }
299
300         return SLAP_CB_CONTINUE;
301 }
302
303 static int
304 ldap_distproc_response( Operation *op, SlapReply *rs )
305 {
306         return SLAP_CB_CONTINUE;
307 }
308
309 /*
310  * configuration...
311  */
312
313 enum {
314         /* NOTE: the chaining behavior control is registered
315          * by the chain overlay; it may move here some time */
316         DP_CHAINING = 1,
317         DP_CACHE_URI,
318
319         DP_LAST
320 };
321
322 static ConfigDriver distproc_cfgen;
323 static ConfigCfAdd distproc_cfadd;
324 static ConfigLDAPadd distproc_ldadd;
325
326 static ConfigTable distproc_cfg[] = {
327         { "distproc-chaining", "args",
328                 2, 4, 0, ARG_MAGIC|ARG_BERVAL|DP_CHAINING, distproc_cfgen,
329                 /* NOTE: using the same attributeTypes defined
330                  * for the "chain" overlay */
331                 "( OLcfgOvAt:3.1 NAME 'olcChainingBehavior' "
332                         "DESC 'Chaining behavior control parameters (draft-sermersheim-ldap-chaining)' "
333                         "SYNTAX OMsDirectoryString SINGLE-VALUE )", NULL, NULL },
334         { "distproc-cache-uri", "TRUE/FALSE",
335                 2, 2, 0, ARG_MAGIC|ARG_ON_OFF|DP_CACHE_URI, distproc_cfgen,
336                 "( OLcfgOvAt:3.2 NAME 'olcChainCacheURI' "
337                         "DESC 'Enables caching of URIs not present in configuration' "
338                         "SYNTAX OMsBoolean "
339                         "SINGLE-VALUE )", NULL, NULL },
340         { NULL, NULL, 0, 0, 0, ARG_IGNORED }
341 };
342
343 static ConfigOCs distproc_ocs[] = {
344         { "( OLcfgOvOc:7.1 "
345                 "NAME 'olcDistProcConfig' "
346                 "DESC 'Distributed procedures <draft-sermersheim-ldap-distproc> configuration' "
347                 "SUP olcOverlayConfig "
348                 "MAY ( "
349                         "olcChainingBehavior $ "
350                         "olcChainCacheURI "
351                         ") )",
352                 Cft_Overlay, distproc_cfg, NULL, distproc_cfadd },
353         { "( OLcfgOvOc:7.2 "
354                 "NAME 'olcDistProcDatabase' "
355                 "DESC 'Distributed procedure remote server configuration' "
356                 "AUXILIARY )",
357                 Cft_Misc, distproc_cfg, distproc_ldadd },
358         { NULL, 0, NULL }
359 };
360
361 static int
362 distproc_ldadd( CfEntryInfo *p, Entry *e, ConfigArgs *ca )
363 {
364         slap_overinst           *on;
365         ldap_distproc_t         *lc;
366
367         ldapinfo_t              *li;
368
369         AttributeDescription    *ad = NULL;
370         Attribute               *at;
371         const char              *text;
372
373         int                     rc;
374
375         if ( p->ce_type != Cft_Overlay
376                 || !p->ce_bi
377                 || p->ce_bi->bi_cf_ocs != distproc_ocs )
378         {
379                 return LDAP_CONSTRAINT_VIOLATION;
380         }
381
382         on = (slap_overinst *)p->ce_bi;
383         lc = (ldap_distproc_t *)on->on_bi.bi_private;
384
385         assert( ca->be == NULL );
386         ca->be = (BackendDB *)ch_calloc( 1, sizeof( BackendDB ) );
387
388         ca->be->bd_info = (BackendInfo *)on;
389
390         rc = slap_str2ad( "olcDbURI", &ad, &text );
391         assert( rc == LDAP_SUCCESS );
392
393         at = attr_find( e->e_attrs, ad );
394         if ( lc->lc_common_li == NULL && at != NULL ) {
395                 /* FIXME: we should generate an empty default entry
396                  * if none is supplied */
397                 Debug( LDAP_DEBUG_ANY, "slapd-distproc: "
398                         "first underlying database \"%s\" "
399                         "cannot contain attribute \"%s\".\n",
400                         e->e_name.bv_val, ad->ad_cname.bv_val, 0 );
401                 rc = LDAP_CONSTRAINT_VIOLATION;
402                 goto done;
403
404         } else if ( lc->lc_common_li != NULL && at == NULL ) {
405                 /* FIXME: we should generate an empty default entry
406                  * if none is supplied */
407                 Debug( LDAP_DEBUG_ANY, "slapd-distproc: "
408                         "subsequent underlying database \"%s\" "
409                         "must contain attribute \"%s\".\n",
410                         e->e_name.bv_val, ad->ad_cname.bv_val, 0 );
411                 rc = LDAP_CONSTRAINT_VIOLATION;
412                 goto done;
413         }
414
415         if ( lc->lc_common_li == NULL ) {
416                 rc = ldap_distproc_db_init_common( ca->be );
417
418         } else {
419                 rc = ldap_distproc_db_init_one( ca->be );
420         }
421
422         if ( rc != 0 ) {
423                 Debug( LDAP_DEBUG_ANY, "slapd-distproc: "
424                         "unable to init %sunderlying database \"%s\".\n",
425                         lc->lc_common_li == NULL ? "common " : "", e->e_name.bv_val, 0 );
426                 return LDAP_CONSTRAINT_VIOLATION;
427         }
428
429         li = ca->be->be_private;
430
431         if ( lc->lc_common_li == NULL ) {
432                 lc->lc_common_li = li;
433
434         } else if ( avl_insert( &lc->lc_lai.lai_tree, (caddr_t)li,
435                 ldap_distproc_uri_cmp, ldap_distproc_uri_dup ) )
436         {
437                 Debug( LDAP_DEBUG_ANY, "slapd-distproc: "
438                         "database \"%s\" insert failed.\n",
439                         e->e_name.bv_val, 0, 0 );
440                 rc = LDAP_CONSTRAINT_VIOLATION;
441                 goto done;
442         }
443
444 done:;
445         if ( rc != LDAP_SUCCESS ) {
446                 (void)ldap_distproc_db_destroy_one( ca->be, NULL );
447                 ch_free( ca->be );
448                 ca->be = NULL;
449         }
450
451         return rc;
452 }
453
454 typedef struct ldap_distproc_cfadd_apply_t {
455         Operation       *op;
456         SlapReply       *rs;
457         Entry           *p;
458         ConfigArgs      *ca;
459         int             count;
460 } ldap_distproc_cfadd_apply_t;
461
462 static int
463 ldap_distproc_cfadd_apply( void *datum, void *arg )
464 {
465         ldapinfo_t                      *li = (ldapinfo_t *)datum;
466         ldap_distproc_cfadd_apply_t     *lca = (ldap_distproc_cfadd_apply_t *)arg;
467
468         struct berval                   bv;
469
470         /* FIXME: should not hardcode "olcDatabase" here */
471         bv.bv_len = snprintf( lca->ca->cr_msg, sizeof( lca->ca->cr_msg ),
472                 "olcDatabase={%d}%s", lca->count, lback->bi_type );
473         bv.bv_val = lca->ca->cr_msg;
474
475         lca->ca->be->be_private = (void *)li;
476         config_build_entry( lca->op, lca->rs, lca->p->e_private, lca->ca,
477                 &bv, lback->bi_cf_ocs, &distproc_ocs[ 1 ] );
478
479         lca->count++;
480
481         return 0;
482 }
483
484 static int
485 distproc_cfadd( Operation *op, SlapReply *rs, Entry *p, ConfigArgs *ca )
486 {
487         CfEntryInfo     *pe = p->e_private;
488         slap_overinst   *on = (slap_overinst *)pe->ce_bi;
489         ldap_distproc_t *lc = (ldap_distproc_t *)on->on_bi.bi_private;
490         void            *priv = (void *)ca->be->be_private;
491
492         if ( lback->bi_cf_ocs ) {
493                 ldap_distproc_cfadd_apply_t     lca = { 0 };
494
495                 lca.op = op;
496                 lca.rs = rs;
497                 lca.p = p;
498                 lca.ca = ca;
499                 lca.count = 0;
500
501                 (void)ldap_distproc_cfadd_apply( (void *)lc->lc_common_li, (void *)&lca );
502
503                 (void)avl_apply( lc->lc_lai.lai_tree, ldap_distproc_cfadd_apply,
504                         &lca, 1, AVL_INORDER );
505
506                 ca->be->be_private = priv;
507         }
508
509         return 0;
510 }
511
512 static int
513 distproc_cfgen( ConfigArgs *c )
514 {
515         slap_overinst   *on = (slap_overinst *)c->bi;
516         ldap_distproc_t *lc = (ldap_distproc_t *)on->on_bi.bi_private;
517
518         int             rc = 0;
519
520         if ( c->op == SLAP_CONFIG_EMIT ) {
521                 switch( c->type ) {
522                 case DP_CACHE_URI:
523                         c->value_int = LDAP_DISTPROC_CACHE_URI( lc );
524                         break;
525
526                 default:
527                         assert( 0 );
528                         rc = 1;
529                 }
530                 return rc;
531
532         } else if ( c->op == LDAP_MOD_DELETE ) {
533                 switch( c->type ) {
534                 case DP_CHAINING:
535                         return 1;
536
537                 case DP_CACHE_URI:
538                         lc->lc_flags &= ~LDAP_DISTPROC_F_CACHE_URI;
539                         break;
540
541                 default:
542                         return 1;
543                 }
544                 return rc;
545         }
546
547         switch( c->type ) {
548         case DP_CACHE_URI:
549                 if ( c->value_int ) {
550                         lc->lc_flags |= LDAP_DISTPROC_F_CACHE_URI;
551                 } else {
552                         lc->lc_flags &= ~LDAP_DISTPROC_F_CACHE_URI;
553                 }
554                 break;
555
556         default:
557                 assert( 0 );
558                 return 1;
559         }
560
561         return rc;
562 }
563
564 static int
565 ldap_distproc_db_init(
566         BackendDB *be,
567         ConfigReply *cr )
568 {
569         slap_overinst   *on = (slap_overinst *)be->bd_info;
570         ldap_distproc_t *lc = NULL;
571
572         if ( lback == NULL ) {
573                 lback = backend_info( "ldap" );
574
575                 if ( lback == NULL ) {
576                         return 1;
577                 }
578         }
579
580         lc = ch_malloc( sizeof( ldap_distproc_t ) );
581         if ( lc == NULL ) {
582                 return 1;
583         }
584         memset( lc, 0, sizeof( ldap_distproc_t ) );
585         ldap_pvt_thread_mutex_init( &lc->lc_lai.lai_mutex );
586
587         on->on_bi.bi_private = (void *)lc;
588
589         return 0;
590 }
591
592 static int
593 ldap_distproc_db_config(
594         BackendDB       *be,
595         const char      *fname,
596         int             lineno,
597         int             argc,
598         char            **argv )
599 {
600         slap_overinst   *on = (slap_overinst *)be->bd_info;
601         ldap_distproc_t *lc = (ldap_distproc_t *)on->on_bi.bi_private;
602
603         int             rc = SLAP_CONF_UNKNOWN;
604                 
605         if ( lc->lc_common_li == NULL ) {
606                 void    *be_private = be->be_private;
607                 ldap_distproc_db_init_common( be );
608                 lc->lc_common_li = lc->lc_cfg_li = (ldapinfo_t *)be->be_private;
609                 be->be_private = be_private;
610         }
611
612         /* Something for the distproc database? */
613         if ( strncasecmp( argv[ 0 ], "distproc-", STRLENOF( "distproc-" ) ) == 0 ) {
614                 char            *save_argv0 = argv[ 0 ];
615                 BackendInfo     *bd_info = be->bd_info;
616                 void            *be_private = be->be_private;
617                 ConfigOCs       *be_cf_ocs = be->be_cf_ocs;
618                 int             is_uri = 0;
619
620                 argv[ 0 ] += STRLENOF( "distproc-" );
621
622                 if ( strcasecmp( argv[ 0 ], "uri" ) == 0 ) {
623                         rc = ldap_distproc_db_init_one( be );
624                         if ( rc != 0 ) {
625                                 Debug( LDAP_DEBUG_ANY, "%s: line %d: "
626                                         "underlying slapd-ldap initialization failed.\n.",
627                                         fname, lineno, 0 );
628                                 return 1;
629                         }
630                         lc->lc_cfg_li = be->be_private;
631                         is_uri = 1;
632                 }
633
634                 /* TODO: add checks on what other slapd-ldap(5) args
635                  * should be put in the template; this is not quite
636                  * harmful, because attributes that shouldn't don't
637                  * get actually used, but the user should at least
638                  * be warned.
639                  */
640
641                 be->bd_info = lback;
642                 be->be_private = (void *)lc->lc_cfg_li;
643                 be->be_cf_ocs = lback->bi_cf_ocs;
644
645                 rc = config_generic_wrapper( be, fname, lineno, argc, argv );
646
647                 argv[ 0 ] = save_argv0;
648                 be->be_cf_ocs = be_cf_ocs;
649                 be->be_private = be_private;
650                 be->bd_info = bd_info;
651
652                 if ( is_uri ) {
653 private_destroy:;
654                         if ( rc != 0 ) {
655                                 BackendDB               db = *be;
656
657                                 db.bd_info = lback;
658                                 db.be_private = (void *)lc->lc_cfg_li;
659                                 ldap_distproc_db_destroy_one( &db, NULL );
660                                 lc->lc_cfg_li = NULL;
661
662                         } else {
663                                 if ( lc->lc_cfg_li->li_bvuri == NULL
664                                         || BER_BVISNULL( &lc->lc_cfg_li->li_bvuri[ 0 ] )
665                                         || !BER_BVISNULL( &lc->lc_cfg_li->li_bvuri[ 1 ] ) )
666                                 {
667                                         Debug( LDAP_DEBUG_ANY, "%s: line %d: "
668                                                 "no URI list allowed in slapo-distproc.\n",
669                                                 fname, lineno, 0 );
670                                         rc = 1;
671                                         goto private_destroy;
672                                 }
673
674                                 if ( avl_insert( &lc->lc_lai.lai_tree,
675                                         (caddr_t)lc->lc_cfg_li,
676                                         ldap_distproc_uri_cmp, ldap_distproc_uri_dup ) )
677                                 {
678                                         Debug( LDAP_DEBUG_ANY, "%s: line %d: "
679                                                 "duplicate URI in slapo-distproc.\n",
680                                                 fname, lineno, 0 );
681                                         rc = 1;
682                                         goto private_destroy;
683                                 }
684                         }
685                 }
686         }
687         
688         return rc;
689 }
690
691 enum db_which {
692         db_open = 0,
693         db_close,
694         db_destroy,
695
696         db_last
697 };
698
699 typedef struct ldap_distproc_db_apply_t {
700         BackendDB       *be;
701         BI_db_func      *func;
702 } ldap_distproc_db_apply_t;
703
704 static int
705 ldap_distproc_db_apply( void *datum, void *arg )
706 {
707         ldapinfo_t              *li = (ldapinfo_t *)datum;
708         ldap_distproc_db_apply_t        *lca = (ldap_distproc_db_apply_t *)arg;
709
710         lca->be->be_private = (void *)li;
711
712         return lca->func( lca->be, NULL );
713 }
714
715 static int
716 ldap_distproc_db_func(
717         BackendDB *be,
718         enum db_which which
719 )
720 {
721         slap_overinst   *on = (slap_overinst *)be->bd_info;
722         ldap_distproc_t *lc = (ldap_distproc_t *)on->on_bi.bi_private;
723
724         int             rc = 0;
725
726         if ( lc ) {
727                 BI_db_func      *func = (&lback->bi_db_open)[ which ];
728
729                 if ( func != NULL && lc->lc_common_li != NULL ) {
730                         BackendDB               db = *be;
731
732                         db.bd_info = lback;
733                         db.be_private = lc->lc_common_li;
734
735                         rc = func( &db, NULL );
736
737                         if ( rc != 0 ) {
738                                 return rc;
739                         }
740
741                         if ( lc->lc_lai.lai_tree != NULL ) {
742                                 ldap_distproc_db_apply_t        lca;
743
744                                 lca.be = &db;
745                                 lca.func = func;
746
747                                 rc = avl_apply( lc->lc_lai.lai_tree,
748                                         ldap_distproc_db_apply, (void *)&lca,
749                                         1, AVL_INORDER ) != AVL_NOMORE;
750                         }
751                 }
752         }
753
754         return rc;
755 }
756
757 static int
758 ldap_distproc_db_open(
759         BackendDB       *be,
760         ConfigReply     *cr )
761 {
762         return ldap_distproc_db_func( be, db_open );
763 }
764
765 static int
766 ldap_distproc_db_close(
767         BackendDB       *be,
768         ConfigReply     *cr )
769 {
770         return ldap_distproc_db_func( be, db_close );
771 }
772
773 static int
774 ldap_distproc_db_destroy(
775         BackendDB       *be,
776         ConfigReply     *cr )
777 {
778         slap_overinst   *on = (slap_overinst *) be->bd_info;
779         ldap_distproc_t *lc = (ldap_distproc_t *)on->on_bi.bi_private;
780
781         int             rc;
782
783         rc = ldap_distproc_db_func( be, db_destroy );
784
785         if ( lc ) {
786                 avl_free( lc->lc_lai.lai_tree, NULL );
787                 ldap_pvt_thread_mutex_destroy( &lc->lc_lai.lai_mutex );
788                 ch_free( lc );
789         }
790
791         return rc;
792 }
793
794 /*
795  * inits one instance of the slapd-ldap backend, and stores
796  * the private info in be_private of the arg
797  */
798 static int
799 ldap_distproc_db_init_common(
800         BackendDB       *be )
801 {
802         BackendInfo     *bi = be->bd_info;
803         int             t;
804
805         be->bd_info = lback;
806         be->be_private = NULL;
807         t = lback->bi_db_init( be, NULL );
808         if ( t != 0 ) {
809                 return t;
810         }
811         be->bd_info = bi;
812
813         return 0;
814 }
815
816 /*
817  * inits one instance of the slapd-ldap backend, stores
818  * the private info in be_private of the arg and fills
819  * selected fields with data from the template.
820  *
821  * NOTE: add checks about the other fields of the template,
822  * which are ignored and SHOULD NOT be configured by the user.
823  */
824 static int
825 ldap_distproc_db_init_one(
826         BackendDB       *be )
827 {
828         slap_overinst   *on = (slap_overinst *)be->bd_info;
829         ldap_distproc_t *lc = (ldap_distproc_t *)on->on_bi.bi_private;
830
831         BackendInfo     *bi = be->bd_info;
832         ldapinfo_t      *li;
833
834         slap_op_t       t;
835
836         be->bd_info = lback;
837         be->be_private = NULL;
838         t = lback->bi_db_init( be, NULL );
839         if ( t != 0 ) {
840                 return t;
841         }
842         li = (ldapinfo_t *)be->be_private;
843
844         /* copy common data */
845         li->li_nretries = lc->lc_common_li->li_nretries;
846         li->li_flags = lc->lc_common_li->li_flags;
847         li->li_version = lc->lc_common_li->li_version;
848         for ( t = 0; t < SLAP_OP_LAST; t++ ) {
849                 li->li_timeout[ t ] = lc->lc_common_li->li_timeout[ t ];
850         }
851         be->bd_info = bi;
852
853         return 0;
854 }
855
856 typedef struct ldap_distproc_conn_apply_t {
857         BackendDB       *be;
858         Connection      *conn;
859 } ldap_distproc_conn_apply_t;
860
861 static int
862 ldap_distproc_conn_apply( void *datum, void *arg )
863 {
864         ldapinfo_t              *li = (ldapinfo_t *)datum;
865         ldap_distproc_conn_apply_t      *lca = (ldap_distproc_conn_apply_t *)arg;
866
867         lca->be->be_private = (void *)li;
868
869         return lback->bi_connection_destroy( lca->be, lca->conn );
870 }
871
872 static int
873 ldap_distproc_connection_destroy(
874         BackendDB *be,
875         Connection *conn
876 )
877 {
878         slap_overinst           *on = (slap_overinst *) be->bd_info;
879         ldap_distproc_t         *lc = (ldap_distproc_t *)on->on_bi.bi_private;
880         void                    *private = be->be_private;
881         ldap_distproc_conn_apply_t      lca;
882         int                     rc;
883
884         be->be_private = NULL;
885         lca.be = be;
886         lca.conn = conn;
887         ldap_pvt_thread_mutex_lock( &lc->lc_lai.lai_mutex );
888         rc = avl_apply( lc->lc_lai.lai_tree, ldap_distproc_conn_apply,
889                 (void *)&lca, 1, AVL_INORDER ) != AVL_NOMORE;
890         ldap_pvt_thread_mutex_unlock( &lc->lc_lai.lai_mutex );
891         be->be_private = private;
892
893         return rc;
894 }
895
896 static int
897 ldap_distproc_parse_returnContRef_ctrl(
898         Operation       *op,
899         SlapReply       *rs,
900         LDAPControl     *ctrl )
901 {
902         if ( get_returnContRef( op ) != SLAP_CONTROL_NONE ) {
903                 rs->sr_text = "returnContinuationReference control specified multiple times";
904                 return LDAP_PROTOCOL_ERROR;
905         }
906
907         if ( op->o_pagedresults != SLAP_CONTROL_NONE ) {
908                 rs->sr_text = "returnContinuationReference control specified with pagedResults control";
909                 return LDAP_PROTOCOL_ERROR;
910         }
911
912         if ( !BER_BVISEMPTY( &ctrl->ldctl_value ) ) {
913                 rs->sr_text = "returnContinuationReference control: value must be NULL";
914                 return LDAP_PROTOCOL_ERROR;
915         }
916
917         op->o_returnContRef = ctrl->ldctl_iscritical ? SLAP_CONTROL_CRITICAL : SLAP_CONTROL_NONCRITICAL;
918
919         return LDAP_SUCCESS;
920 }
921
922 static int
923 ldap_exop_chained_request(
924                 Operation       *op,
925                 SlapReply       *rs )
926 {
927         Statslog( LDAP_DEBUG_STATS, "%s CHAINED REQUEST\n",
928             op->o_log_prefix, 0, 0, 0, 0 );
929
930         rs->sr_err = backend_check_restrictions( op, rs, 
931                         (struct berval *)&slap_EXOP_CHAINEDREQUEST );
932         if ( rs->sr_err != LDAP_SUCCESS ) {
933                 return rs->sr_err;
934         }
935
936         /* by now, just reject requests */
937         rs->sr_text = "under development";
938         return LDAP_UNWILLING_TO_PERFORM;
939 }
940
941
942 static slap_overinst distproc;
943
944 int
945 distproc_initialize( void )
946 {
947         int     rc;
948
949         /* Make sure we don't exceed the bits reserved for userland */
950         config_check_userland( DP_LAST );
951
952         rc = load_extop( (struct berval *)&slap_EXOP_CHAINEDREQUEST,
953                 SLAP_EXOP_HIDE, ldap_exop_chained_request );
954         if ( rc != LDAP_SUCCESS ) {
955                 Debug( LDAP_DEBUG_ANY, "slapd-distproc: "
956                         "unable to register chainedRequest exop: %d.\n",
957                         rc, 0, 0 );
958                 return rc;
959         }
960
961         rc = supported_feature_load( &slap_FEATURE_CANCHAINOPS );
962         if ( rc != LDAP_SUCCESS ) {
963                 Debug( LDAP_DEBUG_ANY, "slapd-distproc: "
964                         "unable to register canChainOperations supported feature: %d.\n",
965                         rc, 0, 0 );
966                 return rc;
967         }
968
969         rc = register_supported_control( LDAP_CONTROL_X_RETURNCONTREF,
970                         SLAP_CTRL_GLOBAL|SLAP_CTRL_ACCESS|SLAP_CTRL_HIDE, NULL,
971                         ldap_distproc_parse_returnContRef_ctrl, &sc_returnContRef );
972         if ( rc != LDAP_SUCCESS ) {
973                 Debug( LDAP_DEBUG_ANY, "slapd-distproc: "
974                         "unable to register returnContinuationReference control: %d.\n",
975                         rc, 0, 0 );
976                 return rc;
977         }
978
979         distproc.on_bi.bi_type = "distproc";
980         distproc.on_bi.bi_db_init = ldap_distproc_db_init;
981         distproc.on_bi.bi_db_config = ldap_distproc_db_config;
982         distproc.on_bi.bi_db_open = ldap_distproc_db_open;
983         distproc.on_bi.bi_db_close = ldap_distproc_db_close;
984         distproc.on_bi.bi_db_destroy = ldap_distproc_db_destroy;
985
986         /* ... otherwise the underlying backend's function would be called,
987          * likely passing an invalid entry; on the contrary, the requested
988          * operational attributes should have been returned while chasing
989          * the referrals.  This all in all is a bit messy, because part
990          * of the operational attributes are generated by the backend;
991          * part by the frontend; back-ldap should receive all the available
992          * ones from the remote server, but then, on its own, it strips those
993          * it assumes will be (re)generated by the frontend (e.g.
994          * subschemaSubentry, entryDN, ...) */
995         distproc.on_bi.bi_operational = ldap_distproc_operational;
996         
997         distproc.on_bi.bi_connection_destroy = ldap_distproc_connection_destroy;
998
999         distproc.on_response = ldap_distproc_response;
1000
1001         distproc.on_bi.bi_cf_ocs = distproc_ocs;
1002
1003         rc = config_register_schema( distproc_cfg, distproc_ocs );
1004         if ( rc ) {
1005                 return rc;
1006         }
1007
1008         return overlay_register( &distproc );
1009 }
1010
1011 #endif /* SLAP_DISTPROC */