]> git.sur5r.net Git - openldap/blob - servers/slapd/back-bdb/dn2id.c
comment cleanup
[openldap] / servers / slapd / back-bdb / dn2id.c
1 /* dn2id.c - routines to deal with the dn2id index */
2 /* $OpenLDAP$ */
3 /*
4  * Copyright 1998-2003 The OpenLDAP Foundation, All Rights Reserved.
5  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
6  */
7
8 #include "portable.h"
9
10 #include <stdio.h>
11 #include <ac/string.h>
12
13 #include "back-bdb.h"
14 #include "idl.h"
15 #include "lutil.h"
16
17 #ifndef BDB_HIER
18 int
19 bdb_dn2id_add(
20         BackendDB       *be,
21         DB_TXN *txn,
22         EntryInfo *eip,
23         Entry           *e,
24         void *ctx )
25 {
26         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
27         DB *db = bdb->bi_dn2id->bdi_db;
28         int             rc;
29         DBT             key, data;
30         char            *buf;
31         struct berval   ptr, pdn;
32
33 #ifdef NEW_LOGGING
34         LDAP_LOG ( INDEX, ARGS, "bdb_dn2id_add( \"%s\", 0x%08lx )\n",
35                 e->e_ndn, (long) e->e_id, 0 );
36 #else
37         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_add( \"%s\", 0x%08lx )\n",
38                 e->e_ndn, (long) e->e_id, 0 );
39 #endif
40         assert( e->e_id != NOID );
41
42         DBTzero( &key );
43         key.size = e->e_nname.bv_len + 2;
44         key.ulen = key.size;
45         key.flags = DB_DBT_USERMEM;
46         buf = sl_malloc( key.size, ctx );
47         key.data = buf;
48         buf[0] = DN_BASE_PREFIX;
49         ptr.bv_val = buf + 1;
50         ptr.bv_len = e->e_nname.bv_len;
51         AC_MEMCPY( ptr.bv_val, e->e_nname.bv_val, e->e_nname.bv_len );
52         ptr.bv_val[ptr.bv_len] = '\0';
53
54         DBTzero( &data );
55         data.data = (char *) &e->e_id;
56         data.size = sizeof( e->e_id );
57
58         /* store it -- don't override */
59         rc = db->put( db, txn, &key, &data, DB_NOOVERWRITE );
60         if( rc != 0 ) {
61 #ifdef NEW_LOGGING
62                 LDAP_LOG ( INDEX, ERR, "bdb_dn2id_add: put failed: %s %d\n",
63                         db_strerror(rc), rc, 0 );
64 #else
65                 Debug( LDAP_DEBUG_ANY, "=> bdb_dn2id_add: put failed: %s %d\n",
66                         db_strerror(rc), rc, 0 );
67 #endif
68                 goto done;
69         }
70
71 #ifndef BDB_MULTIPLE_SUFFIXES
72         if( !be_issuffix( be, &ptr )) {
73 #endif
74                 buf[0] = DN_SUBTREE_PREFIX;
75                 rc = db->put( db, txn, &key, &data, DB_NOOVERWRITE );
76                 if( rc != 0 ) {
77 #ifdef NEW_LOGGING
78                         LDAP_LOG ( INDEX, ERR, 
79                                 "=> bdb_dn2id_add: subtree (%s) put failed: %d\n",
80                                 ptr.bv_val, rc, 0 );
81 #else
82                         Debug( LDAP_DEBUG_ANY,
83                         "=> bdb_dn2id_add: subtree (%s) put failed: %d\n",
84                         ptr.bv_val, rc, 0 );
85 #endif
86                         goto done;
87                 }
88                 
89 #ifdef BDB_MULTIPLE_SUFFIXES
90         if( !be_issuffix( be, &ptr )) {
91 #endif
92                 dnParent( &ptr, &pdn );
93         
94                 key.size = pdn.bv_len + 2;
95                 key.ulen = key.size;
96                 pdn.bv_val[-1] = DN_ONE_PREFIX;
97                 key.data = pdn.bv_val-1;
98                 ptr = pdn;
99
100                 rc = bdb_idl_insert_key( be, db, txn, &key, e->e_id );
101
102                 if( rc != 0 ) {
103 #ifdef NEW_LOGGING
104                         LDAP_LOG ( INDEX, ERR, 
105                                 "=> bdb_dn2id_add: parent (%s) insert failed: %d\n",
106                                 ptr.bv_val, rc, 0 );
107 #else
108                         Debug( LDAP_DEBUG_ANY,
109                                 "=> bdb_dn2id_add: parent (%s) insert failed: %d\n",
110                                         ptr.bv_val, rc, 0 );
111 #endif
112                         goto done;
113                 }
114 #ifndef BDB_MULTIPLE_SUFFIXES
115         }
116
117         while( !be_issuffix( be, &ptr )) {
118 #else
119         for (;;) {
120 #endif
121                 ptr.bv_val[-1] = DN_SUBTREE_PREFIX;
122
123                 rc = bdb_idl_insert_key( be, db, txn, &key, e->e_id );
124
125                 if( rc != 0 ) {
126 #ifdef NEW_LOGGING
127                         LDAP_LOG ( INDEX, ERR, 
128                                 "=> bdb_dn2id_add: subtree (%s) insert failed: %d\n",
129                                 ptr.bv_val, rc, 0 );
130 #else
131                         Debug( LDAP_DEBUG_ANY,
132                                 "=> bdb_dn2id_add: subtree (%s) insert failed: %d\n",
133                                         ptr.bv_val, rc, 0 );
134 #endif
135                         break;
136                 }
137 #ifdef BDB_MULTIPLE_SUFFIXES
138                 if( be_issuffix( be, &ptr )) break;
139 #endif
140                 dnParent( &ptr, &pdn );
141
142                 key.size = pdn.bv_len + 2;
143                 key.ulen = key.size;
144                 key.data = pdn.bv_val - 1;
145                 ptr = pdn;
146         }
147 #ifdef BDB_MULTIPLE_SUFFIXES
148         }
149 #endif
150
151 done:
152         sl_free( buf, ctx );
153 #ifdef NEW_LOGGING
154         LDAP_LOG ( INDEX, RESULTS, "<= bdb_dn2id_add: %d\n", rc, 0, 0 );
155 #else
156         Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_add: %d\n", rc, 0, 0 );
157 #endif
158         return rc;
159 }
160
161 int
162 bdb_dn2id_delete(
163         BackendDB       *be,
164         DB_TXN *txn,
165         EntryInfo       *eip,
166         Entry           *e,
167         void *ctx )
168 {
169         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
170         DB *db = bdb->bi_dn2id->bdi_db;
171         int             rc;
172         DBT             key;
173         char            *buf;
174         struct berval   pdn, ptr;
175
176 #ifdef NEW_LOGGING
177         LDAP_LOG ( INDEX, ARGS, 
178                 "=> bdb_dn2id_delete ( \"%s\", 0x%08lx )\n", e->e_ndn, e->e_id, 0);
179 #else
180         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_delete( \"%s\", 0x%08lx )\n",
181                 e->e_ndn, e->e_id, 0 );
182 #endif
183
184         DBTzero( &key );
185         key.size = e->e_nname.bv_len + 2;
186         buf = sl_malloc( key.size, ctx );
187         key.data = buf;
188         key.flags = DB_DBT_USERMEM;
189         buf[0] = DN_BASE_PREFIX;
190         ptr.bv_val = buf+1;
191         ptr.bv_len = e->e_nname.bv_len;
192         AC_MEMCPY( ptr.bv_val, e->e_nname.bv_val, e->e_nname.bv_len );
193         ptr.bv_val[ptr.bv_len] = '\0';
194
195         /* delete it */
196         rc = db->del( db, txn, &key, 0 );
197         if( rc != 0 ) {
198 #ifdef NEW_LOGGING
199                 LDAP_LOG ( INDEX, ERR, 
200                         "=> bdb_dn2id_delete: delete failed: %s %d\n", 
201                         db_strerror(rc), rc, 0 );
202 #else
203                 Debug( LDAP_DEBUG_ANY, "=> bdb_dn2id_delete: delete failed: %s %d\n",
204                         db_strerror(rc), rc, 0 );
205 #endif
206                 goto done;
207         }
208
209 #ifndef BDB_MULTIPLE_SUFFIXES
210         if( !be_issuffix( be, &ptr )) {
211 #endif
212                 buf[0] = DN_SUBTREE_PREFIX;
213                 rc = db->del( db, txn, &key, 0 );
214                 if( rc != 0 ) {
215 #ifdef NEW_LOGGING
216                         LDAP_LOG ( INDEX, ERR, 
217                                 "=> bdb_dn2id_delete: subtree (%s) delete failed: %d\n", 
218                                 ptr.bv_val, rc, 0 );
219 #else
220                         Debug( LDAP_DEBUG_ANY,
221                         "=> bdb_dn2id_delete: subtree (%s) delete failed: %d\n",
222                         ptr.bv_val, rc, 0 );
223 #endif
224                         goto done;
225                 }
226
227 #ifdef BDB_MULTIPLE_SUFFIXES
228         if( !be_issuffix( be, &ptr )) {
229 #endif
230                 dnParent( &ptr, &pdn );
231
232                 key.size = pdn.bv_len + 2;
233                 key.ulen = key.size;
234                 pdn.bv_val[-1] = DN_ONE_PREFIX;
235                 key.data = pdn.bv_val - 1;
236                 ptr = pdn;
237
238                 rc = bdb_idl_delete_key( be, db, txn, &key, e->e_id );
239
240                 if( rc != 0 ) {
241 #ifdef NEW_LOGGING
242                         LDAP_LOG ( INDEX, ERR, 
243                                 "=> bdb_dn2id_delete: parent (%s) delete failed: %d\n", 
244                                 ptr.bv_val, rc, 0 );
245 #else
246                         Debug( LDAP_DEBUG_ANY,
247                                 "=> bdb_dn2id_delete: parent (%s) delete failed: %d\n",
248                                 ptr.bv_val, rc, 0 );
249 #endif
250                         goto done;
251                 }
252 #ifndef BDB_MULTIPLE_SUFFIXES
253         }
254
255         while( !be_issuffix( be, &ptr )) {
256 #else
257         for (;;) {
258 #endif
259                 ptr.bv_val[-1] = DN_SUBTREE_PREFIX;
260
261                 rc = bdb_idl_delete_key( be, db, txn, &key, e->e_id );
262                 if( rc != 0 ) {
263 #ifdef NEW_LOGGING
264                         LDAP_LOG ( INDEX, ERR, 
265                                 "=> bdb_dn2id_delete: subtree (%s) delete failed: %d\n", 
266                                 ptr.bv_val, rc, 0 );
267 #else
268                         Debug( LDAP_DEBUG_ANY,
269                                 "=> bdb_dn2id_delete: subtree (%s) delete failed: %d\n",
270                                 ptr.bv_val, rc, 0 );
271 #endif
272                         goto done;
273                 }
274 #ifdef BDB_MULTIPLE_SUFFIXES
275                 if( be_issuffix( be, &ptr )) break;
276 #endif
277                 dnParent( &ptr, &pdn );
278
279                 key.size = pdn.bv_len + 2;
280                 key.ulen = key.size;
281                 key.data = pdn.bv_val - 1;
282                 ptr = pdn;
283         }
284 #ifdef BDB_MULTIPLE_SUFFIXES
285         }
286 #endif
287
288 done:
289         sl_free( buf, ctx );
290 #ifdef NEW_LOGGING
291         LDAP_LOG ( INDEX, RESULTS, "<= bdb_dn2id_delete %d\n", rc, 0, 0 );
292 #else
293         Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_delete %d\n", rc, 0, 0 );
294 #endif
295         return rc;
296 }
297
298 int
299 bdb_dn2id(
300         BackendDB       *be,
301         DB_TXN *txn,
302         struct berval   *dn,
303         EntryInfo *ei,
304         void *ctx )
305 {
306         int             rc;
307         DBT             key, data;
308         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
309         DB *db = bdb->bi_dn2id->bdi_db;
310
311 #ifdef NEW_LOGGING
312         LDAP_LOG ( INDEX, ARGS, "=> bdb_dn2id( \"%s\" )\n", dn->bv_val, 0, 0 );
313 #else
314         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id( \"%s\" )\n", dn->bv_val, 0, 0 );
315 #endif
316         DBTzero( &key );
317         key.size = dn->bv_len + 2;
318         key.data = sl_malloc( key.size, ctx );
319         ((char *)key.data)[0] = DN_BASE_PREFIX;
320         AC_MEMCPY( &((char *)key.data)[1], dn->bv_val, key.size - 1 );
321
322         /* store the ID */
323         DBTzero( &data );
324         data.data = &ei->bei_id;
325         data.ulen = sizeof(ID);
326         data.flags = DB_DBT_USERMEM;
327
328         /* fetch it */
329         rc = db->get( db, txn, &key, &data, bdb->bi_db_opflags );
330
331         if( rc != 0 ) {
332 #ifdef NEW_LOGGING
333                 LDAP_LOG ( INDEX, ERR, "<= bdb_dn2id: get failed %s (%d)\n", 
334                         db_strerror(rc), rc, 0 );
335 #else
336                 Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id: get failed: %s (%d)\n",
337                         db_strerror( rc ), rc, 0 );
338 #endif
339         } else {
340 #ifdef NEW_LOGGING
341                 LDAP_LOG ( INDEX, RESULTS, 
342                         "<= bdb_dn2id: got id=0x%08lx\n", ei->bei_id, 0, 0 );
343 #else
344                 Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id: got id=0x%08lx\n",
345                         ei->bei_id, 0, 0 );
346 #endif
347         }
348
349         sl_free( key.data, ctx );
350         return rc;
351 }
352
353 int
354 bdb_dn2id_children(
355         Operation *op,
356         DB_TXN *txn,
357         Entry *e )
358 {
359         DBT             key, data;
360         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
361         DB *db = bdb->bi_dn2id->bdi_db;
362         ID              id;
363         int             rc;
364
365 #ifdef NEW_LOGGING
366         LDAP_LOG ( INDEX, ARGS, 
367                 "=> bdb_dn2id_children( %s )\n", e->e_nname.bv_val, 0, 0 );
368 #else
369         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_children( %s )\n",
370                 e->e_nname.bv_val, 0, 0 );
371 #endif
372         DBTzero( &key );
373         key.size = e->e_nname.bv_len + 2;
374         key.data = sl_malloc( key.size, op->o_tmpmemctx );
375         ((char *)key.data)[0] = DN_ONE_PREFIX;
376         AC_MEMCPY( &((char *)key.data)[1], e->e_nname.bv_val, key.size - 1 );
377
378 #ifdef SLAP_IDL_CACHE
379         if ( bdb->bi_idl_cache_size ) {
380                 rc = bdb_idl_cache_get( bdb, db, &key, NULL );
381                 if ( rc != LDAP_NO_SUCH_OBJECT ) {
382                         sl_free( key.data, op->o_tmpmemctx );
383                         return rc;
384                 }
385         }
386 #endif
387         /* we actually could do a empty get... */
388         DBTzero( &data );
389         data.data = &id;
390         data.ulen = sizeof(id);
391         data.flags = DB_DBT_USERMEM;
392         data.doff = 0;
393         data.dlen = sizeof(id);
394
395         rc = db->get( db, txn, &key, &data, bdb->bi_db_opflags );
396         sl_free( key.data, op->o_tmpmemctx );
397
398 #ifdef NEW_LOGGING
399         LDAP_LOG ( INDEX, DETAIL1, 
400                 "<= bdb_dn2id_children( %s ): %s (%d)\n", 
401                 e->e_nname.bv_val, rc == 0 ? "" : ( rc == DB_NOTFOUND ? "no " :
402                 db_strerror(rc)), rc );
403 #else
404         Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_children( %s ): %s (%d)\n",
405                 e->e_nname.bv_val,
406                 rc == 0 ? "" : ( rc == DB_NOTFOUND ? "no " :
407                         db_strerror(rc) ), rc );
408 #endif
409
410         return rc;
411 }
412
413 int
414 bdb_dn2idl(
415         BackendDB       *be,
416         struct berval   *dn,
417         int prefix,
418         ID *ids,
419         ID *stack,
420         void *ctx )
421 {
422         int             rc;
423         DBT             key;
424         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
425         DB *db = bdb->bi_dn2id->bdi_db;
426
427 #ifdef NEW_LOGGING
428         LDAP_LOG ( INDEX, ARGS, 
429                 "=> bdb_dn2ididl( \"%s\" )\n", dn->bv_val, 0, 0 );
430 #else
431         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2idl( \"%s\" )\n", dn->bv_val, 0, 0 );
432 #endif
433
434 #ifndef BDB_MULTIPLE_SUFFIXES
435         if (prefix == DN_SUBTREE_PREFIX && be_issuffix(be, dn))
436         {
437                 BDB_IDL_ALL(bdb, ids);
438                 return 0;
439         }
440 #endif
441
442         DBTzero( &key );
443         key.size = dn->bv_len + 2;
444         key.ulen = key.size;
445         key.flags = DB_DBT_USERMEM;
446         key.data = sl_malloc( key.size, ctx );
447         ((char *)key.data)[0] = prefix;
448         AC_MEMCPY( &((char *)key.data)[1], dn->bv_val, key.size - 1 );
449
450         rc = bdb_idl_fetch_key( be, db, NULL, &key, ids );
451
452         if( rc != 0 ) {
453 #ifdef NEW_LOGGING
454                 LDAP_LOG ( INDEX, ERR, 
455                         "<= bdb_dn2ididl: get failed: %s (%d)\n", db_strerror(rc), rc, 0 );
456 #else
457                 Debug( LDAP_DEBUG_TRACE,
458                         "<= bdb_dn2idl: get failed: %s (%d)\n",
459                         db_strerror( rc ), rc, 0 );
460 #endif
461
462         } else {
463 #ifdef NEW_LOGGING
464                 LDAP_LOG ( INDEX, RESULTS, 
465                         "<= bdb_dn2ididl: id=%ld first=%ld last=%ld\n", 
466                         (long) ids[0], (long) BDB_IDL_FIRST( ids ), 
467                         (long) BDB_IDL_LAST( ids ) );
468 #else
469                 Debug( LDAP_DEBUG_TRACE,
470                         "<= bdb_dn2idl: id=%ld first=%ld last=%ld\n",
471                         (long) ids[0],
472                         (long) BDB_IDL_FIRST( ids ), (long) BDB_IDL_LAST( ids ) );
473 #endif
474         }
475
476         sl_free( key.data, ctx );
477         return rc;
478 }
479 #else   /* BDB_HIER */
480
481 /* Experimental management routines for a hierarchically structured database.
482  *
483  * Unsupported! Use at your own risk!
484  * -- Howard Chu, Symas Corp. 2003.
485  *
486  * Instead of a ldbm-style dn2id database, we use a hierarchical one. Each
487  * entry in this database is a struct diskNode, keyed by entryID and with
488  * the data containing the RDN and entryID of the node's children. We use
489  * a B-Tree with sorted duplicates to store all the children of a node under
490  * the same key. Also, the first item under the key contains the entry's own
491  * rdn and the ID of the node's parent, to allow bottom-up tree traversal as
492  * well as top-down. To keep this info first in the list, the nrdnlen is set
493  * to the negative of its value.
494  *
495  * The diskNode is a variable length structure. This definition is not
496  * directly usable for in-memory manipulation.
497  */
498 typedef struct diskNode {
499         ID entryID;
500         short nrdnlen;
501         char nrdn[1];
502         char rdn[1];
503 } diskNode;
504
505 /* Sort function for the sorted duplicate data items of a dn2id key.
506  * Sorts based on normalized RDN, in length order.
507  */
508 int
509 bdb_dup_compare(
510         DB *db, 
511         const DBT *usrkey,
512         const DBT *curkey
513 )
514 {
515         char *u = (char *)&(((diskNode *)(usrkey->data))->nrdnlen);
516         char *c = (char *)&(((diskNode *)(curkey->data))->nrdnlen);
517         int rc, i;
518
519         /* data is not aligned, cannot compare directly */
520 #ifdef WORDS_BIGENDIAN
521         for( i = 0; i < (int)sizeof(short); i++)
522 #else
523         for( i = sizeof(short)-1; i >= 0; i--)
524 #endif
525         {
526                 rc = u[i] - c[i];
527                 if( rc ) return rc;
528         }
529         return strcmp( u+sizeof(short), c+sizeof(short) );
530 }
531
532 /* This function constructs a full DN for a given entry.
533  */
534 int bdb_fix_dn(
535         Entry *e,
536         int checkit
537 )
538 {
539         EntryInfo *ei;
540         int rlen = 0, nrlen = 0;
541         char *ptr, *nptr;
542         int max = 0;
543         
544         for ( ei = BEI(e); ei && ei->bei_id; ei=ei->bei_parent ) {
545                 rlen += ei->bei_rdn.bv_len + 1;
546                 nrlen += ei->bei_nrdn.bv_len + 1;
547                 if (ei->bei_modrdns > max) max = ei->bei_modrdns;
548         }
549
550         /* See if the entry DN was invalidated by a subtree rename */
551         if ( checkit ) {
552                 if ( BEI(e)->bei_modrdns >= max ) {
553                         return 0;
554                 }
555                 /* We found a mismatch, tell the caller to lock it */
556                 if ( checkit == 1 ) {
557                         return 1;
558                 }
559                 /* checkit == 2. do the fix. */
560                 free( e->e_name.bv_val );
561         }
562
563         e->e_name.bv_len = rlen - 1;
564         e->e_nname.bv_len = nrlen - 1;
565         e->e_name.bv_val = ch_malloc(rlen + nrlen);
566         e->e_nname.bv_val = e->e_name.bv_val + rlen;
567         ptr = e->e_name.bv_val;
568         nptr = e->e_nname.bv_val;
569         for ( ei = BEI(e); ei && ei->bei_id; ei=ei->bei_parent ) {
570                 ptr = lutil_strcopy(ptr, ei->bei_rdn.bv_val);
571                 nptr = lutil_strcopy(nptr, ei->bei_nrdn.bv_val);
572                 if ( ei->bei_parent ) {
573                         *ptr++ = ',';
574                         *nptr++ = ',';
575                 }
576         }
577         BEI(e)->bei_modrdns = max;
578         ptr[-1] = '\0';
579         nptr[-1] = '\0';
580
581         return 0;
582 }
583
584 /* We add two elements to the DN2ID database - a data item under the parent's
585  * entryID containing the child's RDN and entryID, and an item under the
586  * child's entryID containing the parent's entryID.
587  */
588 int
589 bdb_dn2id_add(
590         BackendDB       *be,
591         DB_TXN *txn,
592         EntryInfo       *eip,
593         Entry           *e,
594         void *ctx )
595 {
596         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
597         DB *db = bdb->bi_dn2id->bdi_db;
598         DBT             key, data;
599         int             rc, rlen, nrlen;
600         diskNode *d;
601         char *ptr;
602
603         nrlen = dn_rdnlen( be, &e->e_nname );
604         if (nrlen) {
605                 rlen = dn_rdnlen( be, &e->e_name );
606         } else {
607                 nrlen = e->e_nname.bv_len;
608                 rlen = e->e_name.bv_len;
609         }
610
611         d = sl_malloc(sizeof(diskNode) + rlen + nrlen, ctx);
612         d->entryID = e->e_id;
613         d->nrdnlen = nrlen;
614         ptr = lutil_strncopy( d->nrdn, e->e_nname.bv_val, nrlen );
615         *ptr++ = '\0';
616         ptr = lutil_strncopy( ptr, e->e_name.bv_val, rlen );
617         *ptr = '\0';
618
619         DBTzero(&key);
620         DBTzero(&data);
621         key.data = &eip->bei_id;
622         key.size = sizeof(ID);
623         key.flags = DB_DBT_USERMEM;
624
625 #ifdef SLAP_IDL_CACHE
626         if ( bdb->bi_idl_cache_size ) {
627                 bdb_idl_cache_del( bdb, db, &key );
628         }
629 #endif
630         data.data = d;
631         data.size = sizeof(diskNode) + rlen + nrlen;
632         data.flags = DB_DBT_USERMEM;
633
634         rc = db->put( db, txn, &key, &data, DB_NODUPDATA );
635
636         if (rc == 0) {
637                 key.data = &e->e_id;
638                 d->entryID = eip->bei_id;
639                 d->nrdnlen = 0 - nrlen;
640
641                 rc = db->put( db, txn, &key, &data, DB_NODUPDATA );
642         }
643
644         sl_free( d, ctx );
645
646         return rc;
647 }
648
649 int
650 bdb_dn2id_delete(
651         BackendDB       *be,
652         DB_TXN *txn,
653         EntryInfo       *eip,
654         Entry   *e,
655         void    *ctx )
656 {
657         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
658         DB *db = bdb->bi_dn2id->bdi_db;
659         DBT             key, data;
660         DBC     *cursor;
661         diskNode *d;
662         int rc, nrlen;
663
664         DBTzero(&key);
665         key.size = sizeof(ID);
666         key.ulen = key.size;
667         key.data = &eip->bei_id;
668         key.flags = DB_DBT_USERMEM;
669
670         DBTzero(&data);
671         data.size = sizeof(diskNode) + BEI(e)->bei_nrdn.bv_len;
672         data.ulen = data.size;
673         data.dlen = data.size;
674         data.flags = DB_DBT_USERMEM | DB_DBT_PARTIAL;
675
676 #ifdef SLAP_IDL_CACHE
677         if ( bdb->bi_idl_cache_size ) {
678                 bdb_idl_cache_del( bdb, db, &key );
679         }
680 #endif
681         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
682         if ( rc ) return rc;
683
684         d = sl_malloc( data.size, ctx );
685         d->entryID = e->e_id;
686         d->nrdnlen = BEI(e)->bei_nrdn.bv_len;
687         strcpy( d->nrdn, BEI(e)->bei_nrdn.bv_val );
688         data.data = d;
689
690         /* Delete our ID from the parent's list */
691         rc = cursor->c_get( cursor, &key, &data, DB_GET_BOTH | DB_RMW );
692         if ( rc == 0 )
693                 rc = cursor->c_del( cursor, 0 );
694
695         /* Delete our ID from the tree. With sorted duplicates, this
696          * will leave any child nodes still hanging around. This is OK
697          * for modrdn, which will add our info back in later.
698          */
699         if ( rc == 0 ) {
700                 key.data = &e->e_id;
701                 rc = cursor->c_get( cursor, &key, &data, DB_SET | DB_RMW );
702                 if ( rc == 0 )
703                         rc = cursor->c_del( cursor, 0 );
704         }
705         cursor->c_close( cursor );
706         sl_free( d, ctx );
707
708         return rc;
709 }
710
711 int
712 bdb_dn2id(
713         BackendDB       *be,
714         DB_TXN *txn,
715         struct berval   *in,
716         EntryInfo       *ei,
717         void *ctx )
718 {
719         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
720         DB *db = bdb->bi_dn2id->bdi_db;
721         DBT             key, data;
722         DBC     *cursor;
723         int             rc = 0, nrlen;
724         diskNode *d;
725         char    *ptr;
726         ID idp = ei->bei_parent->bei_id;
727
728         nrlen = dn_rdnlen( be, in );
729         if (!nrlen) nrlen = in->bv_len;
730
731         DBTzero(&key);
732         key.size = sizeof(ID);
733         key.data = &idp;
734         key.ulen = sizeof(ID);
735         key.flags = DB_DBT_USERMEM;
736
737         DBTzero(&data);
738         data.size = sizeof(diskNode) + nrlen;
739         data.ulen = data.size * 3;
740         data.flags = DB_DBT_USERMEM;
741
742         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
743         if ( rc ) return rc;
744
745         d = sl_malloc( data.size * 3, ctx );
746         d->nrdnlen = nrlen;
747         ptr = lutil_strncopy( d->nrdn, in->bv_val, nrlen );
748         *ptr = '\0';
749         data.data = d;
750
751         rc = cursor->c_get( cursor, &key, &data, DB_GET_BOTH );
752         cursor->c_close( cursor );
753
754         if ( rc == 0 ) {
755                 AC_MEMCPY( &ei->bei_id, &d->entryID, sizeof(ID) );
756                 ei->bei_rdn.bv_len = data.size - sizeof(diskNode) - nrlen;
757                 ptr = d->nrdn + nrlen + 1;
758                 ei->bei_rdn.bv_val = ch_malloc( ei->bei_rdn.bv_len + 1 );
759                 strcpy( ei->bei_rdn.bv_val, ptr );
760         }
761         sl_free( d, ctx );
762
763         return rc;
764 }
765
766 int
767 bdb_dn2id_parent(
768         Backend *be,
769         DB_TXN *txn,
770         EntryInfo *ei,
771         ID *idp,
772         void *ctx )
773 {
774         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
775         DB *db = bdb->bi_dn2id->bdi_db;
776         DBT             key, data;
777         DBC     *cursor;
778         int             rc = 0;
779         diskNode *d;
780         char    *ptr;
781         unsigned char *pt2;
782
783         DBTzero(&key);
784         key.size = sizeof(ID);
785         key.data = &ei->bei_id;
786         key.ulen = sizeof(ID);
787         key.flags = DB_DBT_USERMEM;
788
789         DBTzero(&data);
790         data.flags = DB_DBT_USERMEM;
791
792         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
793         if ( rc ) return rc;
794
795         data.ulen = sizeof(diskNode) + (SLAP_LDAPDN_MAXLEN * 2);
796         d = sl_malloc( data.ulen, ctx );
797         data.data = d;
798
799         rc = cursor->c_get( cursor, &key, &data, DB_SET );
800         cursor->c_close( cursor );
801         if ( rc == 0 ) {
802                 if (d->nrdnlen >= 0) {
803                         return LDAP_OTHER;
804                 }
805                 AC_MEMCPY( idp, &d->entryID, sizeof(ID) );
806                 ei->bei_nrdn.bv_len = 0 - d->nrdnlen;
807                 ber_str2bv( d->nrdn, ei->bei_nrdn.bv_len, 1, &ei->bei_nrdn );
808                 ei->bei_rdn.bv_len = data.size - sizeof(diskNode) -
809                         ei->bei_nrdn.bv_len;
810                 ptr = d->nrdn + ei->bei_nrdn.bv_len + 1;
811                 ber_str2bv( ptr, ei->bei_rdn.bv_len, 1, &ei->bei_rdn );
812         }
813         sl_free( d, ctx );
814         return rc;
815 }
816
817 int
818 bdb_dn2id_children(
819         Operation *op,
820         DB_TXN *txn,
821         Entry *e )
822 {
823         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
824         DB *db = bdb->bi_dn2id->bdi_db;
825         DBT             key, data;
826         DBC             *cursor;
827         int             rc;
828         ID              id;
829         diskNode d;
830
831         DBTzero(&key);
832         key.size = sizeof(ID);
833         key.data = &e->e_id;
834         key.flags = DB_DBT_USERMEM;
835
836 #ifdef SLAP_IDL_CACHE
837         if ( bdb->bi_idl_cache_size ) {
838                 rc = bdb_idl_cache_get( bdb, db, &key, NULL );
839                 if ( rc != LDAP_NO_SUCH_OBJECT ) {
840                         return rc;
841                 }
842         }
843 #endif
844         DBTzero(&data);
845         data.data = &d;
846         data.ulen = sizeof(d);
847         data.flags = DB_DBT_USERMEM | DB_DBT_PARTIAL;
848         data.dlen = sizeof(d);
849
850         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
851         if ( rc ) return rc;
852
853         rc = cursor->c_get( cursor, &key, &data, DB_SET );
854         if ( rc == 0 ) {
855                 rc = cursor->c_get( cursor, &key, &data, DB_NEXT_DUP );
856         }
857         cursor->c_close( cursor );
858         return rc;
859 }
860
861 /* bdb_dn2idl:
862  * We can't just use bdb_idl_fetch_key because
863  * 1 - our data items are longer than just an entry ID
864  * 2 - our data items are sorted alphabetically by nrdn, not by ID.
865  *
866  * We descend the tree recursively, so we define this cookie
867  * to hold our necessary state information. The bdb_dn2idl_internal
868  * function uses this cookie when calling itself.
869  */
870
871 struct dn2id_cookie {
872         struct bdb_info *bdb;
873         DB *db;
874         int prefix;
875         int rc;
876         ID id;
877         ID dbuf;
878         ID *ids;
879         void *ptr;
880         ID tmp[BDB_IDL_DB_SIZE];
881         ID *buf;
882         DBT key;
883         DBT data;
884         DBC *dbc;
885         void *ctx;
886 };
887
888 static int
889 bdb_dn2idl_internal(
890         struct dn2id_cookie *cx
891 )
892 {
893 #ifdef SLAP_IDL_CACHE
894         if ( cx->bdb->bi_idl_cache_size ) {
895                 cx->rc = bdb_idl_cache_get(cx->bdb, cx->db, &cx->key, cx->tmp);
896                 if ( cx->rc == DB_NOTFOUND ) {
897                         return cx->rc;
898                 }
899                 if ( cx->rc == LDAP_SUCCESS ) {
900                         goto saveit;
901                 }
902         }
903 #endif
904
905         cx->rc = cx->db->cursor( cx->db, NULL, &cx->dbc,
906                 cx->bdb->bi_db_opflags );
907         if ( cx->rc ) return cx->rc;
908         BDB_IDL_ZERO( cx->tmp );
909
910         cx->data.data = &cx->dbuf;
911         cx->data.ulen = sizeof(ID);
912         cx->data.dlen = sizeof(ID);
913         cx->data.flags = DB_DBT_USERMEM | DB_DBT_PARTIAL;
914
915         /* The first item holds the parent ID. Ignore it. */
916         cx->rc = cx->dbc->c_get( cx->dbc, &cx->key, &cx->data, DB_SET );
917         if ( cx->rc == DB_NOTFOUND ) goto saveit;
918         if ( cx->rc ) return cx->rc;
919
920         cx->data.data = cx->buf;
921         cx->data.ulen = BDB_IDL_UM_SIZE * sizeof(ID);
922         cx->data.flags = DB_DBT_USERMEM;
923
924         /* Fetch the rest of the IDs in a loop... */
925         while ( (cx->rc = cx->dbc->c_get( cx->dbc, &cx->key, &cx->data,
926                 DB_MULTIPLE | DB_NEXT_DUP )) == 0 ) {
927                 u_int8_t *j;
928                 size_t len;
929                 DB_MULTIPLE_INIT( cx->ptr, &cx->data );
930                 while (cx->ptr) {
931                         DB_MULTIPLE_NEXT( cx->ptr, &cx->data, j, len );
932                         if (j) {
933                                 AC_MEMCPY( &cx->dbuf, j, sizeof(ID) );
934                                 bdb_idl_insert( cx->tmp, cx->dbuf );
935                         }
936                 }
937         }
938         cx->dbc->c_close( cx->dbc );
939
940         /* If we got some records, treat as success */
941         if (!BDB_IDL_IS_ZERO(cx->tmp)) {
942                 cx->rc = 0;
943         }
944
945 saveit:
946 #ifdef SLAP_IDL_CACHE
947         if ( cx->bdb->bi_idl_cache_max_size ) {
948                 bdb_idl_cache_put( cx->bdb, cx->db, &cx->key, cx->tmp, cx->rc );
949         }
950 #endif
951         if ( cx->rc == 0 ) {
952                 if ( cx->prefix == DN_SUBTREE_PREFIX ) {
953                         ID *save, idcurs;
954
955                         save = sl_malloc( BDB_IDL_SIZEOF( cx->tmp ), cx->ctx );
956                         BDB_IDL_CPY( save, cx->tmp );
957                         bdb_idl_union( cx->ids, cx->tmp );
958         
959                         idcurs = 0;
960                         for ( cx->id = bdb_idl_first( save, &idcurs );
961                                 cx->id != NOID;
962                                 cx->id = bdb_idl_next( save, &idcurs )) {
963                                 bdb_dn2idl_internal( cx );
964                         }
965                         sl_free( save, cx->ctx );
966                         cx->rc = 0;
967                 } else {
968                         BDB_IDL_CPY( cx->ids, cx->tmp );
969                 }
970         }
971         return cx->rc;
972 }
973
974 int
975 bdb_dn2idl(
976         BackendDB       *be,
977         struct berval   *dn,
978         int prefix,
979         ID *ids,
980         ID *stack,
981         void *ctx )
982 {
983         struct dn2id_cookie cx;
984         EntryInfo *ei = (EntryInfo *)dn;
985
986 #ifndef BDB_MULTIPLE_SUFFIXES
987         if ( ei->bei_parent->bei_id == 0 ) {
988                 struct bdb_info *bdb = (struct bdb_info *)be->be_private;
989                 BDB_IDL_ALL( bdb, ids );
990                 return 0;
991         }
992 #endif
993
994         cx.id = ei->bei_id;
995         cx.bdb = (struct bdb_info *)be->be_private;
996         cx.db = cx.bdb->bi_dn2id->bdi_db;
997         cx.prefix = prefix;
998         cx.ids = ids;
999         cx.buf = stack;
1000         cx.ctx = ctx;
1001
1002         BDB_IDL_ZERO( ids );
1003         if ( prefix == DN_SUBTREE_PREFIX ) {
1004                 bdb_idl_insert( ids, cx.id );
1005         }
1006
1007         DBTzero(&cx.key);
1008         cx.key.data = &cx.id;
1009         cx.key.ulen = sizeof(ID);
1010         cx.key.size = sizeof(ID);
1011         cx.key.flags = DB_DBT_USERMEM;
1012
1013         DBTzero(&cx.data);
1014
1015         return bdb_dn2idl_internal(&cx);
1016 }
1017 #endif  /* BDB_HIER */