]> git.sur5r.net Git - openldap/blob - servers/slapd/back-ldbm/search.c
3731b92a6c8a913f2c9c7199047262b240b12688
[openldap] / servers / slapd / back-ldbm / search.c
1 /* search.c - ldbm backend search function */
2 /* $OpenLDAP$ */
3 /*
4  * Copyright 1998-1999 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
12 #include <ac/string.h>
13 #include <ac/socket.h>
14
15 #include "slap.h"
16 #include "back-ldbm.h"
17 #include "proto-back-ldbm.h"
18
19 static ID_BLOCK *base_candidate(
20         Backend *be, Entry *e );
21
22 static ID_BLOCK *search_candidates(
23         Backend *be, Entry *e, Filter *filter,
24         int scope, int deref, int manageDSAit );
25
26
27 int
28 ldbm_back_search(
29     Backend     *be,
30     Connection  *conn,
31     Operation   *op,
32     char        *base,
33     char        *nbase,
34     int         scope,
35     int         deref,
36     int         slimit,
37     int         tlimit,
38     Filter      *filter,
39     char        *filterstr,
40     char        **attrs,
41     int         attrsonly
42 )
43 {
44         struct ldbminfo *li = (struct ldbminfo *) be->be_private;
45         int             rc, err;
46         char *text;
47         time_t          stoptime;
48         ID_BLOCK                *candidates;
49         ID              id, cursor;
50         Entry           *e;
51         struct berval **v2refs = NULL;
52         Entry   *matched = NULL;
53         char    *realbase = NULL;
54         int             nentries = 0;
55         int             manageDSAit = get_manageDSAit( op );
56
57         Debug(LDAP_DEBUG_TRACE, "=> ldbm_back_search\n", 0, 0, 0);
58
59         /* get entry with reader lock */
60         if ( deref & LDAP_DEREF_FINDING ) {
61                 e = deref_dn_r( be, nbase, &err, &matched, &text );
62
63         } else {
64                 e = dn2entry_r( be, nbase, &matched );
65                 err = e != NULL ? LDAP_SUCCESS : LDAP_REFERRAL;
66                 text = NULL;
67         }
68
69         if ( e == NULL ) {
70                 char *matched_dn = NULL;
71                 struct berval **refs = NULL;
72
73                 if ( matched != NULL ) {
74                         matched_dn = ch_strdup( matched->e_dn );
75
76                         refs = is_entry_referral( matched )
77                                 ? get_entry_referrals( be, conn, op, matched )
78                                 : NULL;
79
80                         cache_return_entry_r( &li->li_cache, matched );
81                 } else {
82                         refs = default_referral;
83                 }
84
85                 send_ldap_result( conn, op, err,
86                         matched_dn, text, refs, NULL );
87
88                 if( matched != NULL ) {
89                         ber_bvecfree( refs );
90                         free( matched_dn );
91                 }
92
93                 return 1;
94         }
95
96         if (!manageDSAit && is_entry_referral( e ) ) {
97                 /* entry is a referral, don't allow add */
98                 char *matched_dn = ch_strdup( e->e_dn );
99                 struct berval **refs = get_entry_referrals( be,
100                         conn, op, e );
101
102                 cache_return_entry_r( &li->li_cache, e );
103
104                 Debug( LDAP_DEBUG_TRACE, "entry is referral\n", 0,
105                     0, 0 );
106
107                 send_ldap_result( conn, op, LDAP_REFERRAL,
108                     matched_dn, NULL, refs, NULL );
109
110                 ber_bvecfree( refs );
111                 free( matched_dn );
112
113                 return 1;
114         }
115
116         if ( tlimit == 0 && be_isroot( be, op->o_ndn ) ) {
117                 tlimit = -1;    /* allow root to set no limit */
118         } else {
119                 tlimit = (tlimit > be->be_timelimit || tlimit < 1) ?
120                     be->be_timelimit : tlimit;
121                 stoptime = op->o_time + tlimit;
122         }
123
124         if ( slimit == 0 && be_isroot( be, op->o_ndn ) ) {
125                 slimit = -1;    /* allow root to set no limit */
126         } else {
127                 slimit = (slimit > be->be_sizelimit || slimit < 1) ?
128                     be->be_sizelimit : slimit;
129         }
130
131         if ( scope == LDAP_SCOPE_BASE ) {
132                 candidates = base_candidate( be, e );
133
134         } else {
135                 candidates = search_candidates( be, e, filter,
136                     scope, deref, manageDSAit );
137         }
138
139         /* need normalized dn below */
140         realbase = ch_strdup( e->e_ndn );
141         cache_return_entry_r( &li->li_cache, e );
142
143         if ( candidates == NULL ) {
144                 /* no candidates */
145                 Debug( LDAP_DEBUG_TRACE, "no candidates\n", 0,
146                     0, 0 );
147
148                 send_search_result( conn, op,
149                         LDAP_SUCCESS,
150                         NULL, NULL, NULL, NULL, 0 );
151
152                 rc = 1;
153                 goto done;
154         }
155
156         for ( id = idl_firstid( candidates, &cursor ); id != NOID;
157             id = idl_nextid( candidates, &cursor ) )
158         {
159                 int             scopeok = 0;
160
161                 /* check for abandon */
162                 ldap_pvt_thread_mutex_lock( &op->o_abandonmutex );
163
164                 if ( op->o_abandon ) {
165                         ldap_pvt_thread_mutex_unlock( &op->o_abandonmutex );
166                         rc = 0;
167                         goto done;
168                 }
169
170                 ldap_pvt_thread_mutex_unlock( &op->o_abandonmutex );
171
172                 /* check time limit */
173                 if ( tlimit != -1 && slap_get_time() > stoptime ) {
174                         send_search_result( conn, op, LDAP_TIMELIMIT_EXCEEDED,
175                                 NULL, NULL, v2refs, NULL, nentries );
176                         rc = 0;
177                         goto done;
178                 }
179
180                 /* get the entry with reader lock */
181                 e = id2entry_r( be, id );
182
183                 if ( e == NULL ) {
184                         Debug( LDAP_DEBUG_ARGS, "search: candidate %ld not found\n",
185                                 id, 0, 0 );
186
187                         goto loop_continue;
188                 }
189
190                 if ( deref & LDAP_DEREF_SEARCHING && is_entry_alias( e ) ) {
191                         Entry *matched;
192                         int err;
193                         char *text;
194                         
195                         e = deref_entry_r( be, e, &err, &matched, &text );
196
197                         if( e == NULL ) {
198                                 e = matched;
199                                 goto loop_continue;
200                         }
201
202                         if( e->e_id == id ) {
203                                 /* circular loop */
204                                 goto loop_continue;
205                         }
206
207                         /* need to skip alias which deref into scope */
208                         if( scope & LDAP_SCOPE_ONELEVEL ) {
209                                 char *pdn = dn_parent( NULL, e->e_ndn );
210                                 if ( pdn != NULL ) {
211                                         if( strcmp( pdn, realbase ) ) {
212                                                 free( pdn );
213                                                 goto loop_continue;
214                                         }
215                                         free(pdn);
216                                 }
217
218                         } else if ( dn_issuffix( e->e_ndn, realbase ) ) {
219                                 /* alias is within scope */
220                                 Debug( LDAP_DEBUG_ARGS, "search: \"%s\" in subtree\n",
221                                         e->e_dn, 0, 0 );
222                                 goto loop_continue;
223                         }
224
225                         scopeok = 1;
226                 }
227
228                 /*
229                  * if it's a referral, add it to the list of referrals. only do
230                  * this for non-base searches, and don't check the filter
231                  * explicitly here since it's only a candidate anyway.
232                  */
233                 if ( !manageDSAit && scope != LDAP_SCOPE_BASE &&
234                         is_entry_referral( e ) )
235                 {
236                         struct berval **refs = get_entry_referrals(
237                                 be, conn, op, e );
238
239                         send_search_reference( be, conn, op,
240                                 e, refs, scope, NULL, &v2refs );
241
242                         ber_bvecfree( refs );
243
244                         goto loop_continue;
245                 }
246
247                 /* if it matches the filter and scope, send it */
248                 if ( test_filter( be, conn, op, e, filter ) == LDAP_COMPARE_TRUE ) {
249                         char    *dn;
250
251                         /* check scope */
252                         if ( !scopeok && scope == LDAP_SCOPE_ONELEVEL ) {
253                                 if ( (dn = dn_parent( be, e->e_ndn )) != NULL ) {
254                                         (void) dn_normalize( dn );
255                                         scopeok = (dn == realbase)
256                                                 ? 1
257                                                 : (strcmp( dn, realbase ) ? 0 : 1 );
258                                         free( dn );
259
260                                 } else {
261                                         scopeok = (realbase == NULL || *realbase == '\0');
262                                 }
263
264                         } else if ( !scopeok && scope == LDAP_SCOPE_SUBTREE ) {
265                                 dn = ch_strdup( e->e_ndn );
266                                 scopeok = dn_issuffix( dn, realbase );
267                                 free( dn );
268
269                         } else {
270                                 scopeok = 1;
271                         }
272
273                         if ( scopeok ) {
274                                 /* check size limit */
275                                 if ( --slimit == -1 ) {
276                                         cache_return_entry_r( &li->li_cache, e );
277                                         send_search_result( conn, op,
278                                                 LDAP_SIZELIMIT_EXCEEDED, NULL, NULL,
279                                                 v2refs, NULL, nentries );
280                                         rc = 0;
281                                         goto done;
282                                 }
283
284                                 if (e) {
285                                         /* Tack on subordinates attr */
286                                         int result;
287                                         ID_BLOCK *idl = NULL;
288                                         char CATTR_SUBS[] = "numsubordinates";
289
290                                         if (attrs &&
291                                             charray_inlist(attrs,
292                                                            CATTR_SUBS))
293                                         {
294                                             idl = dn2idl(be, e->e_ndn,
295                                                          DN_ONE_PREFIX);
296                                             if (idl)
297                                             {
298                                                 char buf[30];
299                                                 struct berval val, *vals[2];
300
301                                                 vals[0] = &val;
302                                                 vals[1] = NULL;
303
304                                                 sprintf(buf, "%lu",
305                                                         ID_BLOCK_NIDS(idl));
306
307                                                 val.bv_val = buf;
308                                                 val.bv_len = strlen(buf);
309
310                                                 attr_merge(e, CATTR_SUBS,
311                                                            vals);
312                                             }
313                                         }
314
315                                         result = send_search_entry(be,
316                                                                    conn,
317                                                                    op,
318                                                                    e,
319                                                                    attrs,
320                                                                    attrsonly,
321                                                                    NULL);
322                                         if (idl)
323                                         {
324                                             idl_free(idl);
325                                             attr_delete(&e->e_attrs,
326                                                         CATTR_SUBS);
327                                         }
328
329                                         switch (result)
330                                         {
331                                         case 0:         /* entry sent ok */
332                                                 nentries++;
333                                                 break;
334                                         case 1:         /* entry not sent */
335                                                 break;
336                                         case -1:        /* connection closed */
337                                                 cache_return_entry_r( &li->li_cache, e );
338                                                 rc = 0;
339                                                 goto done;
340                                         }
341                                 }
342                         } else {
343                                 Debug( LDAP_DEBUG_TRACE, "candidate %ld scope not okay\n",
344                                         id, 0, 0 );
345                         }
346                 } else {
347                         Debug( LDAP_DEBUG_TRACE, "candidate %ld does match filter\n",
348                                 id, 0, 0 );
349                 }
350
351 loop_continue:
352                 if( e != NULL ) {
353                         /* free reader lock */
354                         cache_return_entry_r( &li->li_cache, e );
355                 }
356
357                 ldap_pvt_thread_yield();
358         }
359         send_search_result( conn, op,
360                 v2refs == NULL ? LDAP_SUCCESS : LDAP_REFERRAL,
361                 NULL, NULL, v2refs, NULL, nentries );
362
363         rc = 0;
364
365 done:
366         if( candidates != NULL )
367                 idl_free( candidates );
368
369         ber_bvecfree( v2refs );
370         if( realbase ) free( realbase );
371
372         return rc;
373 }
374
375 static ID_BLOCK *
376 base_candidate(
377     Backend     *be,
378         Entry   *e
379 )
380 {
381         ID_BLOCK                *idl;
382
383         Debug(LDAP_DEBUG_TRACE, "base_candidates: base: \"%s\"\n",
384                 e->e_dn, 0, 0);
385
386         idl = idl_alloc( 1 );
387         idl_insert( &idl, e->e_id, 1 );
388
389         return( idl );
390 }
391
392 static ID_BLOCK *
393 search_candidates(
394     Backend     *be,
395     Entry       *e,
396     Filter      *filter,
397     int         scope,
398         int             deref,
399         int             manageDSAit
400 )
401 {
402         ID_BLOCK                *candidates;
403         Filter          *f, *rf, *af, *lf;
404
405         Debug(LDAP_DEBUG_TRACE, "search_candidates: base=\"%s\" s=%d d=%d\n",
406                 e->e_ndn, scope, deref );
407
408         f = NULL;
409
410         if( !manageDSAit ) {
411                 /* match referrals */
412                 rf = (Filter *) ch_malloc( sizeof(Filter) );
413                 rf->f_next = NULL;
414                 rf->f_choice = LDAP_FILTER_OR;
415 #ifndef SLAPD_SCHEMA_NOT_COMPAT
416                 rf->f_or = (Filter *) ch_malloc( sizeof(Filter) );
417                 rf->f_or->f_choice = LDAP_FILTER_EQUALITY;
418                 rf->f_or->f_avtype = ch_strdup( "objectclass" );
419                 rf->f_or->f_avvalue.bv_val = ch_strdup( "REFERRAL" );
420                 rf->f_or->f_avvalue.bv_len = sizeof("REFERRAL")-1;
421                 rf->f_or->f_next = filter;
422 #endif
423                 f = rf;
424         } else {
425                 rf = NULL;
426                 f = filter;
427         }
428
429         if( deref & LDAP_DEREF_SEARCHING ) {
430                 /* match aliases */
431                 af = (Filter *) ch_malloc( sizeof(Filter) );
432                 af->f_next = NULL;
433                 af->f_choice = LDAP_FILTER_OR;
434 #ifndef SLAPD_SCHEMA_NOT_COMPAT
435                 af->f_or = (Filter *) ch_malloc( sizeof(Filter) );
436                 af->f_or->f_choice = LDAP_FILTER_EQUALITY;
437                 af->f_or->f_avtype = ch_strdup( "objectclass" );
438                 af->f_or->f_avvalue.bv_val = ch_strdup( "ALIAS" );
439                 af->f_or->f_avvalue.bv_len = sizeof("ALIAS")-1;
440                 af->f_or->f_next = f;
441 #endif
442                 f = af;
443         } else {
444                 af = NULL;
445         }
446
447         if ( scope == LDAP_SCOPE_SUBTREE ) {
448                 lf = (Filter *) ch_malloc( sizeof(Filter) );
449                 lf->f_next = NULL;
450                 lf->f_choice = LDAP_FILTER_AND;
451 #ifndef SLAPD_SCHEMA_NOT_COMPAT
452                 lf->f_and = (Filter *) ch_malloc( sizeof(Filter) );
453
454                 lf->f_and->f_choice = SLAPD_FILTER_DN_SUBTREE;
455                 lf->f_and->f_dn = e->e_ndn;
456
457                 lf->f_and->f_next = f;
458 #endif
459                 f = lf;
460
461         } else if ( scope == LDAP_SCOPE_ONELEVEL ) {
462                 lf = (Filter *) ch_malloc( sizeof(Filter) );
463                 lf->f_next = NULL;
464                 lf->f_choice = LDAP_FILTER_AND;
465 #ifndef SLAPD_SCHEMA_NOT_COMPAT
466                 lf->f_and = (Filter *) ch_malloc( sizeof(Filter) );
467
468                 lf->f_and->f_choice = SLAPD_FILTER_DN_ONE;
469                 lf->f_and->f_dn = e->e_ndn;
470
471                 lf->f_and->f_next = f;
472 #endif
473                 f = lf;
474
475         } else {
476                 lf = NULL;
477         }
478
479         candidates = filter_candidates( be, f );
480
481         /* free up filter additions we allocated above */
482         if( lf != NULL ) {
483 #ifndef SLAPD_SCHEMA_NOT_COMPAT
484                 free( lf->f_and );
485 #endif
486                 free( lf );
487         }
488
489         if( af != NULL ) {
490 #ifndef SLAPD_SCHEMA_NOT_COMPAT
491                 af->f_or->f_next = NULL;
492 #endif
493                 filter_free( af );
494         }
495
496         if( rf != NULL ) {
497 #ifndef SLAPD_SCHEMA_NOT_COMPAT
498                 rf->f_or->f_next = NULL;
499 #endif
500                 filter_free( rf );
501         }
502
503         return( candidates );
504 }