]> git.sur5r.net Git - openldap/blob - servers/slapd/overlays/rwmmap.c
plug leaks; fix map initialization
[openldap] / servers / slapd / overlays / rwmmap.c
1 /* rwmmap.c - rewrite/mapping routines */
2 /* $OpenLDAP$ */
3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4  *
5  * Copyright 1999-2005 The OpenLDAP Foundation.
6  * Portions Copyright 1999-2003 Howard Chu.
7  * Portions Copyright 2000-2003 Pierangelo Masarati.
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 the Howard Chu for inclusion
20  * in OpenLDAP Software and subsequently enhanced by Pierangelo
21  * Masarati.
22  */
23
24 #include "portable.h"
25
26 #ifdef SLAPD_OVER_RWM
27
28 #include <stdio.h>
29
30 #include <ac/string.h>
31 #include <ac/socket.h>
32
33 #include "slap.h"
34 #include "rwm.h"
35
36 #undef ldap_debug       /* silence a warning in ldap-int.h */
37 #include "../../../libraries/libldap/ldap-int.h"
38
39 int
40 rwm_mapping_cmp( const void *c1, const void *c2 )
41 {
42         struct ldapmapping *map1 = (struct ldapmapping *)c1;
43         struct ldapmapping *map2 = (struct ldapmapping *)c2;
44         int rc = map1->m_src.bv_len - map2->m_src.bv_len;
45         
46         if ( rc ) {
47                 return rc;
48         }
49
50         return strcasecmp( map1->m_src.bv_val, map2->m_src.bv_val );
51 }
52
53 int
54 rwm_mapping_dup( void *c1, void *c2 )
55 {
56         struct ldapmapping *map1 = (struct ldapmapping *)c1;
57         struct ldapmapping *map2 = (struct ldapmapping *)c2;
58         int rc = map1->m_src.bv_len - map2->m_src.bv_len;
59
60         if ( rc ) {
61                 return 0;
62         }
63
64         return ( ( strcasecmp( map1->m_src.bv_val, map2->m_src.bv_val ) == 0 ) ? -1 : 0 );
65 }
66
67 int
68 rwm_map_init( struct ldapmap *lm, struct ldapmapping **m )
69 {
70         struct ldapmapping      *mapping;
71         const char              *text;
72         int                     rc;
73
74         assert( m != NULL );
75
76         *m = NULL;
77         
78         mapping = (struct ldapmapping *)ch_calloc( 2, 
79                         sizeof( struct ldapmapping ) );
80         if ( mapping == NULL ) {
81                 return LDAP_NO_MEMORY;
82         }
83
84         /* FIXME: I don't think this is needed any more... */
85         rc = slap_str2ad( "objectClass", &mapping[0].m_src_ad, &text );
86         if ( rc != LDAP_SUCCESS ) {
87                 return rc;
88         }
89
90         mapping[0].m_dst_ad = mapping[0].m_src_ad;
91         ber_dupbv( &mapping[0].m_src, &mapping[0].m_src_ad->ad_cname );
92         ber_dupbv( &mapping[0].m_dst, &mapping[0].m_src );
93
94         mapping[1].m_src = mapping[0].m_src;
95         mapping[1].m_dst = mapping[0].m_dst;
96         mapping[1].m_src_ad = mapping[0].m_src_ad;
97         mapping[1].m_dst_ad = mapping[1].m_src_ad;
98
99         avl_insert( &lm->map, (caddr_t)&mapping[0], 
100                         rwm_mapping_cmp, rwm_mapping_dup );
101         avl_insert( &lm->remap, (caddr_t)&mapping[1], 
102                         rwm_mapping_cmp, rwm_mapping_dup );
103
104         *m = mapping;
105
106         return rc;
107 }
108
109 int
110 rwm_mapping( struct ldapmap *map, struct berval *s, struct ldapmapping **m, int remap )
111 {
112         Avlnode *tree;
113         struct ldapmapping fmapping;
114
115         assert( m != NULL );
116
117         if ( remap == RWM_REMAP ) {
118                 tree = map->remap;
119
120         } else {
121                 tree = map->map;
122         }
123
124         fmapping.m_src = *s;
125         *m = (struct ldapmapping *)avl_find( tree, (caddr_t)&fmapping,
126                         rwm_mapping_cmp );
127
128         if ( *m == NULL ) {
129                 return map->drop_missing;
130         }
131
132         return 0;
133 }
134
135 void
136 rwm_map( struct ldapmap *map, struct berval *s, struct berval *bv, int remap )
137 {
138         struct ldapmapping *mapping;
139
140         BER_BVZERO( bv );
141         ( void )rwm_mapping( map, s, &mapping, remap );
142         if ( mapping != NULL ) {
143                 if ( !BER_BVISNULL( &mapping->m_dst ) ) {
144                         *bv = mapping->m_dst;
145                 }
146                 return;
147         }
148
149         if ( !map->drop_missing ) {
150                 *bv = *s;
151         }
152 }
153
154 /*
155  * Map attribute names in place
156  */
157 int
158 rwm_map_attrnames(
159                 struct ldapmap *at_map,
160                 struct ldapmap *oc_map,
161                 AttributeName *an,
162                 AttributeName **anp,
163                 int remap
164 )
165 {
166         int             i, j;
167
168         assert( anp != NULL );
169
170         *anp = NULL;
171
172         if ( an == NULL ) {
173                 return LDAP_SUCCESS;
174         }
175
176         for ( i = 0; !BER_BVISNULL( &an[i].an_name ); i++ )
177                 /* just count */ ;
178         *anp = ch_malloc( ( i + 1 )* sizeof( AttributeName ) );
179         if ( *anp == NULL ) {
180                 return LDAP_NO_MEMORY;
181         }
182
183         for ( i = 0, j = 0; !BER_BVISNULL( &an[i].an_name ); i++ ) {
184                 struct ldapmapping      *m;
185                 int                     at_drop_missing = 0,
186                                         oc_drop_missing = 0;
187
188                 if ( an[i].an_desc ) {
189                         if ( !at_map ) {
190                                 /* FIXME: better leave as is? */
191                                 continue;
192                         }
193                                 
194                         at_drop_missing = rwm_mapping( at_map, &an[i].an_name, &m, remap );
195                         if ( at_drop_missing || ( m && BER_BVISNULL( &m->m_dst ) ) ) {
196                                 continue;
197                         }
198
199                         if ( !m ) {
200                                 (*anp)[j] = an[i];
201                                 j++;
202                                 continue;
203                         }
204
205                         (*anp)[j] = an[i];
206                         if ( remap == RWM_MAP ) {
207                                 (*anp)[j].an_name = m->m_dst;
208                                 (*anp)[j].an_desc = m->m_dst_ad;
209                         } else {
210                                 (*anp)[j].an_name = m->m_src;
211                                 (*anp)[j].an_desc = m->m_src_ad;
212
213                         }
214
215                         j++;
216                         continue;
217
218                 } else if ( an[i].an_oc ) {
219                         if ( !oc_map ) {
220                                 /* FIXME: better leave as is? */
221                                 continue;
222                         }
223
224                         oc_drop_missing = rwm_mapping( oc_map, &an[i].an_name, &m, remap );
225
226                         if ( oc_drop_missing || ( m && BER_BVISNULL( &m->m_dst ) ) ) {
227                                 continue;
228                         }
229
230                         if ( !m ) {
231                                 (*anp)[j] = an[i];
232                                 j++;
233                                 continue;
234                         }
235
236                         (*anp)[j] = an[i];
237                         if ( remap == RWM_MAP ) {
238                                 (*anp)[j].an_name = m->m_dst;
239                                 (*anp)[j].an_oc = m->m_dst_oc;
240                         } else {
241                                 (*anp)[j].an_name = m->m_src;
242                                 (*anp)[j].an_oc = m->m_src_oc;
243                         }
244
245                 } else {
246                         at_drop_missing = rwm_mapping( at_map, &an[i].an_name, &m, remap );
247                 
248                         if ( at_drop_missing || !m ) {
249
250                                 oc_drop_missing = rwm_mapping( oc_map, &an[i].an_name, &m, remap );
251
252                                 /* if both at_map and oc_map required to drop missing,
253                                  * then do it */
254                                 if ( oc_drop_missing && at_drop_missing ) {
255                                         continue;
256                                 }
257
258                                 /* if no oc_map mapping was found and at_map required
259                                  * to drop missing, then do it; otherwise, at_map wins
260                                  * and an is considered an attr and is left unchanged */
261                                 if ( !m ) {
262                                         if ( at_drop_missing ) {
263                                                 continue;
264                                         }
265                                         (*anp)[j] = an[i];
266                                         j++;
267                                         continue;
268                                 }
269         
270                                 if ( BER_BVISNULL( &m->m_dst ) ) {
271                                         continue;
272                                 }
273
274                                 (*anp)[j] = an[i];
275                                 if ( remap == RWM_MAP ) {
276                                         (*anp)[j].an_name = m->m_dst;
277                                         (*anp)[j].an_oc = m->m_dst_oc;
278                                 } else {
279                                         (*anp)[j].an_name = m->m_src;
280                                         (*anp)[j].an_oc = m->m_src_oc;
281                                 }
282                                 j++;
283                                 continue;
284                         }
285
286                         if ( !BER_BVISNULL( &m->m_dst ) ) {
287                                 (*anp)[j] = an[i];
288                                 if ( remap == RWM_MAP ) {
289                                         (*anp)[j].an_name = m->m_dst;
290                                         (*anp)[j].an_desc = m->m_dst_ad;
291                                 } else {
292                                         (*anp)[j].an_name = m->m_src;
293                                         (*anp)[j].an_desc = m->m_src_ad;
294                                 }
295                                 j++;
296                                 continue;
297                         }
298                 }
299         }
300
301         if ( j == 0 && i != 0 ) {
302                 memset( &(*anp)[0], 0, sizeof( AttributeName ) );
303                 BER_BVSTR( &(*anp)[0].an_name, LDAP_NO_ATTRS );
304         }
305         memset( &(*anp)[j], 0, sizeof( AttributeName ) );
306
307         return LDAP_SUCCESS;
308 }
309
310 int
311 rwm_map_attrs(
312                 struct ldapmap *at_map,
313                 AttributeName *an,
314                 int remap,
315                 char ***mapped_attrs
316 )
317 {
318         int i, j;
319         char **na;
320
321         if ( an == NULL ) {
322                 *mapped_attrs = NULL;
323                 return LDAP_SUCCESS;
324         }
325
326         for ( i = 0; !BER_BVISNULL( &an[ i ].an_name ); i++ ) {
327                 /*  */
328         }
329
330         na = (char **)ch_calloc( i + 1, sizeof( char * ) );
331         if ( na == NULL ) {
332                 *mapped_attrs = NULL;
333                 return LDAP_NO_MEMORY;
334         }
335
336         for ( i = j = 0; !BER_BVISNULL( &an[i].an_name ); i++ ) {
337                 struct ldapmapping      *mapping;
338                 
339                 if ( rwm_mapping( at_map, &an[i].an_name, &mapping, remap ) ) {
340                         continue;
341                 }
342
343                 if ( !mapping ) {
344                         na[ j++ ] = an[ i ].an_name.bv_val;
345                         
346                 } else if ( !BER_BVISNULL( &mapping->m_dst ) ) {
347                         na[ j++ ] = mapping->m_dst.bv_val;
348                 }
349         }
350
351         if ( j == 0 && i != 0 ) {
352                 na[ j++ ] = LDAP_NO_ATTRS;
353         }
354
355         na[ j ] = NULL;
356
357         *mapped_attrs = na;
358
359         return LDAP_SUCCESS;
360 }
361
362 static int
363 map_attr_value(
364                 dncookie                *dc,
365                 AttributeDescription    **adp,
366                 struct berval           *mapped_attr,
367                 struct berval           *value,
368                 struct berval           *mapped_value,
369                 int                     remap )
370 {
371         struct berval           vtmp = BER_BVNULL;
372         int                     freeval = 0;
373         AttributeDescription    *ad = *adp;
374         struct ldapmapping      *mapping = NULL;
375
376         rwm_mapping( &dc->rwmap->rwm_at, &ad->ad_cname, &mapping, remap );
377         if ( mapping == NULL ) {
378                 if ( dc->rwmap->rwm_at.drop_missing ) {
379                         return -1;
380                 }
381
382                 *mapped_attr = ad->ad_cname;
383
384         } else {
385                 *mapped_attr = mapping->m_dst;
386         }
387
388         if ( value != NULL ) {
389                 assert( mapped_value != NULL );
390
391                 if ( ad->ad_type->sat_syntax == slap_schema.si_syn_distinguishedName
392                                 || ( mapping != NULL && mapping->m_dst_ad->ad_type->sat_syntax == slap_schema.si_syn_distinguishedName ) )
393                 {
394                         dncookie        fdc = *dc;
395                         int             rc;
396
397 #ifdef ENABLE_REWRITE
398                         fdc.ctx = "searchFilterAttrDN";
399 #endif /* ENABLE_REWRITE */
400
401                         vtmp = *value;
402                         rc = rwm_dn_massage_normalize( &fdc, value, &vtmp );
403                         switch ( rc ) {
404                         case LDAP_SUCCESS:
405                                 if ( vtmp.bv_val != value->bv_val ) {
406                                         freeval = 1;
407                                 }
408                                 break;
409                 
410                         case LDAP_UNWILLING_TO_PERFORM:
411                         case LDAP_OTHER:
412                         default:
413                                 return -1;
414                         }
415
416                 } else if ( ad == slap_schema.si_ad_objectClass
417                                 || ad == slap_schema.si_ad_structuralObjectClass )
418                 {
419                         rwm_map( &dc->rwmap->rwm_oc, value, &vtmp, remap );
420                         if ( BER_BVISNULL( &vtmp ) || BER_BVISEMPTY( &vtmp ) ) {
421                                 vtmp = *value;
422                         }
423                 
424                 } else {
425                         vtmp = *value;
426                 }
427
428                 filter_escape_value( &vtmp, mapped_value );
429
430                 if ( freeval ) {
431                         ch_free( vtmp.bv_val );
432                 }
433         }
434         
435         if ( mapping != NULL ) {
436                 assert( mapping->m_dst_ad != NULL );
437                 *adp = mapping->m_dst_ad;
438         }
439
440         return 0;
441 }
442
443 static int
444 rwm_int_filter_map_rewrite(
445                 dncookie                *dc,
446                 Filter                  *f,
447                 struct berval           *fstr )
448 {
449         int             i;
450         Filter          *p;
451         struct berval   atmp,
452                         vtmp,
453                         tmp;
454         static struct berval
455                         /* better than nothing... */
456                         ber_bvfalse = BER_BVC( "(!(objectClass=*))" ),
457                         ber_bvtf_false = BER_BVC( "(|)" ),
458                         /* better than nothing... */
459                         ber_bvtrue = BER_BVC( "(objectClass=*)" ),
460                         ber_bvtf_true = BER_BVC( "(&)" ),
461 #if 0
462                         /* no longer needed; preserved for completeness */
463                         ber_bvundefined = BER_BVC( "(?=undefined)" ),
464 #endif
465                         ber_bverror = BER_BVC( "(?=error)" ),
466                         ber_bvunknown = BER_BVC( "(?=unknown)" ),
467                         ber_bvnone = BER_BVC( "(?=none)" );
468         ber_len_t       len;
469
470         if ( f == NULL ) {
471                 ber_dupbv( fstr, &ber_bvnone );
472                 return -1;
473         }
474
475         switch ( f->f_choice ) {
476         case LDAP_FILTER_EQUALITY:
477                 if ( map_attr_value( dc, &f->f_av_desc, &atmp,
478                                         &f->f_av_value, &vtmp, RWM_MAP ) )
479                 {
480                         return -1;
481                 }
482
483                 fstr->bv_len = atmp.bv_len + vtmp.bv_len + STRLENOF( "(=)" );
484                 fstr->bv_val = ch_malloc( fstr->bv_len + 1 );
485
486                 snprintf( fstr->bv_val, fstr->bv_len + 1, "(%s=%s)",
487                         atmp.bv_val, vtmp.bv_val );
488
489                 ch_free( vtmp.bv_val );
490                 break;
491
492         case LDAP_FILTER_GE:
493                 if ( map_attr_value( dc, &f->f_av_desc, &atmp,
494                                         &f->f_av_value, &vtmp, RWM_MAP ) )
495                 {
496                         return -1;
497                 }
498
499                 fstr->bv_len = atmp.bv_len + vtmp.bv_len + STRLENOF( "(>=)" );
500                 fstr->bv_val = ch_malloc( fstr->bv_len + 1 );
501
502                 snprintf( fstr->bv_val, fstr->bv_len + 1, "(%s>=%s)",
503                         atmp.bv_val, vtmp.bv_val );
504
505                 ch_free( vtmp.bv_val );
506                 break;
507
508         case LDAP_FILTER_LE:
509                 if ( map_attr_value( dc, &f->f_av_desc, &atmp,
510                                         &f->f_av_value, &vtmp, RWM_MAP ) )
511                 {
512                         return -1;
513                 }
514
515                 fstr->bv_len = atmp.bv_len + vtmp.bv_len + STRLENOF( "(<=)" );
516                 fstr->bv_val = ch_malloc( fstr->bv_len + 1 );
517
518                 snprintf( fstr->bv_val, fstr->bv_len + 1, "(%s<=%s)",
519                         atmp.bv_val, vtmp.bv_val );
520
521                 ch_free( vtmp.bv_val );
522                 break;
523
524         case LDAP_FILTER_APPROX:
525                 if ( map_attr_value( dc, &f->f_av_desc, &atmp,
526                                         &f->f_av_value, &vtmp, RWM_MAP ) )
527                 {
528                         return -1;
529                 }
530
531                 fstr->bv_len = atmp.bv_len + vtmp.bv_len + STRLENOF( "(~=)" );
532                 fstr->bv_val = ch_malloc( fstr->bv_len + 1 );
533
534                 snprintf( fstr->bv_val, fstr->bv_len + 1, "(%s~=%s)",
535                         atmp.bv_val, vtmp.bv_val );
536
537                 ch_free( vtmp.bv_val );
538                 break;
539
540         case LDAP_FILTER_SUBSTRINGS:
541                 if ( map_attr_value( dc, &f->f_sub_desc, &atmp,
542                                         NULL, NULL, RWM_MAP ) )
543                 {
544                         return -1;
545                 }
546
547                 /* cannot be a DN ... */
548
549                 fstr->bv_len = atmp.bv_len + STRLENOF( "(=*)" );
550                 fstr->bv_val = ch_malloc( fstr->bv_len + 128 );
551
552                 snprintf( fstr->bv_val, fstr->bv_len + 1, "(%s=*)",
553                         atmp.bv_val );
554
555                 if ( !BER_BVISNULL( &f->f_sub_initial ) ) {
556                         len = fstr->bv_len;
557
558                         filter_escape_value( &f->f_sub_initial, &vtmp );
559
560                         fstr->bv_len += vtmp.bv_len;
561                         fstr->bv_val = ch_realloc( fstr->bv_val, fstr->bv_len + 1 );
562
563                         snprintf( &fstr->bv_val[len - 2], vtmp.bv_len + 3,
564                                 /* "(attr=" */ "%s*)",
565                                 vtmp.bv_val );
566
567                         ch_free( vtmp.bv_val );
568                 }
569
570                 if ( f->f_sub_any != NULL ) {
571                         for ( i = 0; !BER_BVISNULL( &f->f_sub_any[i] ); i++ ) {
572                                 len = fstr->bv_len;
573                                 filter_escape_value( &f->f_sub_any[i], &vtmp );
574
575                                 fstr->bv_len += vtmp.bv_len + 1;
576                                 fstr->bv_val = ch_realloc( fstr->bv_val, fstr->bv_len + 1 );
577
578                                 snprintf( &fstr->bv_val[len - 1], vtmp.bv_len + 3,
579                                         /* "(attr=[init]*[any*]" */ "%s*)",
580                                         vtmp.bv_val );
581                                 ch_free( vtmp.bv_val );
582                         }
583                 }
584
585                 if ( !BER_BVISNULL( &f->f_sub_final ) ) {
586                         len = fstr->bv_len;
587
588                         filter_escape_value( &f->f_sub_final, &vtmp );
589
590                         fstr->bv_len += vtmp.bv_len;
591                         fstr->bv_val = ch_realloc( fstr->bv_val, fstr->bv_len + 1 );
592
593                         snprintf( &fstr->bv_val[len - 1], vtmp.bv_len + 3,
594                                 /* "(attr=[init*][any*]" */ "%s)",
595                                 vtmp.bv_val );
596
597                         ch_free( vtmp.bv_val );
598                 }
599
600                 break;
601
602         case LDAP_FILTER_PRESENT:
603                 if ( map_attr_value( dc, &f->f_desc, &atmp,
604                                         NULL, NULL, RWM_MAP ) )
605                 {
606                         return -1;
607                 }
608
609                 fstr->bv_len = atmp.bv_len + STRLENOF( "(=*)" );
610                 fstr->bv_val = ch_malloc( fstr->bv_len + 1 );
611
612                 snprintf( fstr->bv_val, fstr->bv_len + 1, "(%s=*)",
613                         atmp.bv_val );
614                 break;
615
616         case LDAP_FILTER_AND:
617         case LDAP_FILTER_OR:
618         case LDAP_FILTER_NOT:
619                 fstr->bv_len = STRLENOF( "(%)" );
620                 fstr->bv_val = ch_malloc( fstr->bv_len + 128 );
621
622                 snprintf( fstr->bv_val, fstr->bv_len + 1, "(%c)",
623                         f->f_choice == LDAP_FILTER_AND ? '&' :
624                         f->f_choice == LDAP_FILTER_OR ? '|' : '!' );
625
626                 for ( p = f->f_list; p != NULL; p = p->f_next ) {
627                         len = fstr->bv_len;
628
629                         if ( rwm_int_filter_map_rewrite( dc, p, &vtmp ) )
630                         {
631                                 return -1;
632                         }
633                         
634                         fstr->bv_len += vtmp.bv_len;
635                         fstr->bv_val = ch_realloc( fstr->bv_val, fstr->bv_len + 1 );
636
637                         snprintf( &fstr->bv_val[len-1], vtmp.bv_len + 2, 
638                                 /*"("*/ "%s)", vtmp.bv_val );
639
640                         ch_free( vtmp.bv_val );
641                 }
642
643                 break;
644
645         case LDAP_FILTER_EXT: {
646                 if ( f->f_mr_desc ) {
647                         if ( map_attr_value( dc, &f->f_mr_desc, &atmp,
648                                                 &f->f_mr_value, &vtmp, RWM_MAP ) )
649                         {
650                                 return -1;
651                         }
652
653                 } else {
654                         BER_BVSTR( &atmp, "" );
655                         filter_escape_value( &f->f_mr_value, &vtmp );
656                 }
657                         
658
659                 fstr->bv_len = atmp.bv_len +
660                         ( f->f_mr_dnattrs ? STRLENOF( ":dn" ) : 0 ) +
661                         ( f->f_mr_rule_text.bv_len ? f->f_mr_rule_text.bv_len + 1 : 0 ) +
662                         vtmp.bv_len + STRLENOF( "(:=)" );
663                 fstr->bv_val = ch_malloc( fstr->bv_len + 1 );
664
665                 snprintf( fstr->bv_val, fstr->bv_len + 1, "(%s%s%s%s:=%s)",
666                         atmp.bv_val,
667                         f->f_mr_dnattrs ? ":dn" : "",
668                         !BER_BVISEMPTY( &f->f_mr_rule_text ) ? ":" : "",
669                         !BER_BVISEMPTY( &f->f_mr_rule_text ) ? f->f_mr_rule_text.bv_val : "",
670                         vtmp.bv_val );
671                 ch_free( vtmp.bv_val );
672                 break;
673         }
674
675         case SLAPD_FILTER_COMPUTED:
676                 switch ( f->f_result ) {
677                 case LDAP_COMPARE_FALSE:
678                         if ( dc->rwmap->rwm_flags & RWM_F_SUPPORT_T_F ) {
679                                 tmp = ber_bvtf_false;
680                                 break;
681                         }
682                         /* fallthru */
683
684                 case SLAPD_COMPARE_UNDEFINED:
685                         tmp = ber_bvfalse;
686                         break;
687
688                 case LDAP_COMPARE_TRUE:
689                         if ( dc->rwmap->rwm_flags & RWM_F_SUPPORT_T_F ) {
690                                 tmp = ber_bvtf_true;
691                         } else {
692                                 tmp = ber_bvtrue;
693                         }
694                         break;
695                         
696                 default:
697                         tmp = ber_bverror;
698                         break;
699                 }
700
701                 ber_dupbv( fstr, &tmp );
702                 break;
703                 
704         default:
705                 ber_dupbv( fstr, &ber_bvunknown );
706                 break;
707         }
708
709         return 0;
710 }
711
712 int
713 rwm_filter_map_rewrite(
714                 dncookie                *dc,
715                 Filter                  *f,
716                 struct berval           *fstr )
717 {
718         int             rc;
719         dncookie        fdc;
720         struct berval   ftmp;
721
722         rc = rwm_int_filter_map_rewrite( dc, f, fstr );
723
724 #ifdef ENABLE_REWRITE
725         if ( rc != LDAP_SUCCESS ) {
726                 return rc;
727         }
728
729         fdc = *dc;
730         ftmp = *fstr;
731
732         fdc.ctx = "searchFilter";
733
734         switch ( rewrite_session( fdc.rwmap->rwm_rw, fdc.ctx, 
735                                 ( !BER_BVISEMPTY( &ftmp ) ? ftmp.bv_val : "" ), 
736                                 fdc.conn, &fstr->bv_val ) )
737         {
738         case REWRITE_REGEXEC_OK:
739                 if ( !BER_BVISNULL( fstr ) ) {
740                         fstr->bv_len = strlen( fstr->bv_val );
741                         ch_free( ftmp.bv_val );
742
743                 } else {
744                         *fstr = ftmp;
745                 }
746
747                 Debug( LDAP_DEBUG_ARGS,
748                         "[rw] %s: \"%s\" -> \"%s\"\n",
749                         fdc.ctx, ftmp.bv_val, fstr->bv_val );           
750                 rc = LDAP_SUCCESS;
751                 break;
752                 
753         case REWRITE_REGEXEC_UNWILLING:
754                 if ( fdc.rs ) {
755                         fdc.rs->sr_err = LDAP_UNWILLING_TO_PERFORM;
756                         fdc.rs->sr_text = "Operation not allowed";
757                 }
758                 rc = LDAP_UNWILLING_TO_PERFORM;
759                 break;
760                 
761         case REWRITE_REGEXEC_ERR:
762                 if ( fdc.rs ) {
763                         fdc.rs->sr_err = LDAP_OTHER;
764                         fdc.rs->sr_text = "Rewrite error";
765                 }
766                 rc = LDAP_OTHER;
767                 break;
768         }
769
770 #endif /* ENABLE_REWRITE */
771         return rc;
772 }
773
774 /*
775  * I don't like this much, but we need two different
776  * functions because different heap managers may be
777  * in use in back-ldap/meta to reduce the amount of
778  * calls to malloc routines, and some of the free()
779  * routines may be macros with args
780  */
781 int
782 rwm_referral_rewrite(
783         Operation               *op,
784         SlapReply               *rs,
785         void                    *cookie,
786         BerVarray               a_vals,
787         BerVarray               *pa_nvals )
788 {
789         slap_overinst           *on = (slap_overinst *) op->o_bd->bd_info;
790         struct ldaprwmap        *rwmap = 
791                         (struct ldaprwmap *)on->on_bi.bi_private;
792
793         int                     i, last;
794
795         dncookie                dc;
796         struct berval           dn = BER_BVNULL,
797                                 ndn = BER_BVNULL;
798
799         assert( a_vals != NULL );
800
801         /*
802          * Rewrite the dn if needed
803          */
804         dc.rwmap = rwmap;
805 #ifdef ENABLE_REWRITE
806         dc.conn = op->o_conn;
807         dc.rs = rs;
808         dc.ctx = (char *)cookie;
809 #else /* ! ENABLE_REWRITE */
810         dc.tofrom = ((int *)cookie)[0];
811         dc.normalized = 0;
812 #endif /* ! ENABLE_REWRITE */
813
814         for ( last = 0; !BER_BVISNULL( &a_vals[last] ); last++ )
815                 ;
816         last--;
817         
818         if ( pa_nvals != NULL ) {
819                 if ( *pa_nvals == NULL ) {
820                         *pa_nvals = ch_malloc( ( last + 2 ) * sizeof(struct berval) );
821                         memset( *pa_nvals, 0, ( last + 2 ) * sizeof(struct berval) );
822                 }
823         }
824
825         for ( i = 0; !BER_BVISNULL( &a_vals[i] ); i++ ) {
826                 struct berval   olddn = BER_BVNULL,
827                                 oldval;
828                 int             rc;
829                 LDAPURLDesc     *ludp;
830
831                 oldval = a_vals[i];
832                 rc = ldap_url_parse( oldval.bv_val, &ludp );
833                 if ( rc != LDAP_URL_SUCCESS ) {
834                         /* leave attr untouched if massage failed */
835                         if ( pa_nvals && BER_BVISNULL( &(*pa_nvals)[i] ) ) {
836                                 ber_dupbv( &(*pa_nvals)[i], &oldval );
837                         }
838                         continue;
839                 }
840
841                 /* FIXME: URLs like "ldap:///dc=suffix" if passed
842                  * thru ldap_url_parse() and ldap_url_desc2str() 
843                  * get rewritten as "ldap:///dc=suffix??base";
844                  * we don't want this to occur... */
845                 if ( ludp->lud_scope == LDAP_SCOPE_BASE ) {
846                         ludp->lud_scope = LDAP_SCOPE_DEFAULT;
847                 }
848
849                 ber_str2bv( ludp->lud_dn, 0, 0, &olddn );
850
851                 dn = olddn;
852                 if ( pa_nvals ) {
853                         ndn = olddn;
854                         rc = rwm_dn_massage_pretty_normalize( &dc, &olddn,
855                                         &dn, &ndn );
856                 } else {
857                         rc = rwm_dn_massage_pretty( &dc, &olddn, &dn );
858                 }
859
860                 switch ( rc ) {
861                 case LDAP_UNWILLING_TO_PERFORM:
862                         /*
863                          * FIXME: need to check if it may be considered 
864                          * legal to trim values when adding/modifying;
865                          * it should be when searching (e.g. ACLs).
866                          */
867                         ch_free( a_vals[i].bv_val );
868                         if (last > i ) {
869                                 a_vals[i] = a_vals[last];
870                                 if ( pa_nvals ) {
871                                         (*pa_nvals)[i] = (*pa_nvals)[last];
872                                 }
873                         }
874                         BER_BVZERO( &a_vals[last] );
875                         if ( pa_nvals ) {
876                                 BER_BVZERO( &(*pa_nvals)[last] );
877                         }
878                         last--;
879                         break;
880                 
881                 case LDAP_SUCCESS:
882                         if ( !BER_BVISNULL( &dn ) && dn.bv_val != olddn.bv_val ) {
883                                 char    *newurl;
884
885                                 ludp->lud_dn = dn.bv_val;
886                                 newurl = ldap_url_desc2str( ludp );
887                                 ludp->lud_dn = olddn.bv_val;
888                                 ch_free( dn.bv_val );
889                                 if ( newurl == NULL ) {
890                                         /* FIXME: leave attr untouched
891                                          * even if ldap_url_desc2str failed...
892                                          */
893                                         break;
894                                 }
895
896                                 ber_str2bv( newurl, 0, 1, &a_vals[i] );
897                                 LDAP_FREE( newurl );
898
899                                 if ( pa_nvals ) {
900                                         ludp->lud_dn = ndn.bv_val;
901                                         newurl = ldap_url_desc2str( ludp );
902                                         ludp->lud_dn = olddn.bv_val;
903                                         ch_free( ndn.bv_val );
904                                         if ( newurl == NULL ) {
905                                                 /* FIXME: leave attr untouched
906                                                  * even if ldap_url_desc2str failed...
907                                                  */
908                                                 ch_free( a_vals[i].bv_val );
909                                                 a_vals[i] = oldval;
910                                                 break;
911                                         }
912
913                                         if ( !BER_BVISNULL( &(*pa_nvals)[i] ) ) {
914                                                 ch_free( (*pa_nvals)[i].bv_val );
915                                         }
916                                         ber_str2bv( newurl, 0, 1, &(*pa_nvals)[i] );
917                                         LDAP_FREE( newurl );
918                                 }
919
920                                 ch_free( oldval.bv_val );
921                                 ludp->lud_dn = olddn.bv_val;
922                         }
923                         break;
924
925                 default:
926                         /* leave attr untouched if massage failed */
927                         if ( pa_nvals && BER_BVISNULL( &(*pa_nvals)[i] ) ) {
928                                 ber_dupbv( &(*pa_nvals)[i], &a_vals[i] );
929                         }
930                         break;
931                 }
932                 ldap_free_urldesc( ludp );
933         }
934         
935         return 0;
936 }
937
938 /*
939  * I don't like this much, but we need two different
940  * functions because different heap managers may be
941  * in use in back-ldap/meta to reduce the amount of
942  * calls to malloc routines, and some of the free()
943  * routines may be macros with args
944  */
945 int
946 rwm_dnattr_rewrite(
947         Operation               *op,
948         SlapReply               *rs,
949         void                    *cookie,
950         BerVarray               a_vals,
951         BerVarray               *pa_nvals )
952 {
953         slap_overinst           *on = (slap_overinst *) op->o_bd->bd_info;
954         struct ldaprwmap        *rwmap = 
955                         (struct ldaprwmap *)on->on_bi.bi_private;
956
957         int                     i, last;
958
959         dncookie                dc;
960         struct berval           dn = BER_BVNULL,
961                                 ndn = BER_BVNULL;
962         BerVarray               in;
963
964         if ( a_vals ) {
965                 in = a_vals;
966
967         } else {
968                 if ( pa_nvals == NULL || *pa_nvals == NULL ) {
969                         return LDAP_OTHER;
970                 }
971                 in = *pa_nvals;
972         }
973
974         /*
975          * Rewrite the dn if needed
976          */
977         dc.rwmap = rwmap;
978 #ifdef ENABLE_REWRITE
979         dc.conn = op->o_conn;
980         dc.rs = rs;
981         dc.ctx = (char *)cookie;
982 #else /* ! ENABLE_REWRITE */
983         dc.tofrom = ((int *)cookie)[0];
984         dc.normalized = 0;
985 #endif /* ! ENABLE_REWRITE */
986
987         for ( last = 0; !BER_BVISNULL( &in[last] ); last++ );
988         last--;
989         if ( pa_nvals != NULL ) {
990                 if ( *pa_nvals == NULL ) {
991                         *pa_nvals = ch_malloc( ( last + 2 ) * sizeof(struct berval) );
992                         memset( *pa_nvals, 0, ( last + 2 ) * sizeof(struct berval) );
993                 }
994         }
995
996         for ( i = 0; !BER_BVISNULL( &in[i] ); i++ ) {
997                 int             rc;
998
999                 if ( a_vals ) {
1000                         dn = in[i];
1001                         if ( pa_nvals ) {
1002                                 ndn = (*pa_nvals)[i];
1003                                 rc = rwm_dn_massage_pretty_normalize( &dc, &in[i], &dn, &ndn );
1004                         } else {
1005                                 rc = rwm_dn_massage_pretty( &dc, &in[i], &dn );
1006                         }
1007                 } else {
1008                         ndn = in[i];
1009                         rc = rwm_dn_massage_normalize( &dc, &in[i], &ndn );
1010                 }
1011
1012                 switch ( rc ) {
1013                 case LDAP_UNWILLING_TO_PERFORM:
1014                         /*
1015                          * FIXME: need to check if it may be considered 
1016                          * legal to trim values when adding/modifying;
1017                          * it should be when searching (e.g. ACLs).
1018                          */
1019                         ch_free( in[i].bv_val );
1020                         if (last > i ) {
1021                                 in[i] = in[last];
1022                                 if ( a_vals && pa_nvals ) {
1023                                         (*pa_nvals)[i] = (*pa_nvals)[last];
1024                                 }
1025                         }
1026                         BER_BVZERO( &in[last] );
1027                         if ( a_vals && pa_nvals ) {
1028                                 BER_BVZERO( &(*pa_nvals)[last] );
1029                         }
1030                         last--;
1031                         break;
1032                 
1033                 case LDAP_SUCCESS:
1034                         if ( a_vals ) {
1035                                 if ( !BER_BVISNULL( &dn ) && dn.bv_val != a_vals[i].bv_val ) {
1036                                         ch_free( a_vals[i].bv_val );
1037                                         a_vals[i] = dn;
1038
1039                                         if ( pa_nvals ) {
1040                                                 if ( !BER_BVISNULL( &(*pa_nvals)[i] ) ) {
1041                                                         ch_free( (*pa_nvals)[i].bv_val );
1042                                                 }
1043                                                 (*pa_nvals)[i] = ndn;
1044                                         }
1045                                 }
1046                                 
1047                         } else {
1048                                 if ( !BER_BVISNULL( &ndn ) && ndn.bv_val != (*pa_nvals)[i].bv_val ) {
1049                                         ch_free( (*pa_nvals)[i].bv_val );
1050                                         (*pa_nvals)[i] = ndn;
1051                                 }
1052                         }
1053                         break;
1054
1055                 default:
1056                         /* leave attr untouched if massage failed */
1057                         if ( a_vals && pa_nvals && BER_BVISNULL( &(*pa_nvals)[i] ) ) {
1058                                 dnNormalize( 0, NULL, NULL, &a_vals[i], &(*pa_nvals)[i], NULL );
1059                         }
1060                         break;
1061                 }
1062         }
1063         
1064         return 0;
1065 }
1066
1067 int
1068 rwm_referral_result_rewrite(
1069         dncookie                *dc,
1070         BerVarray               a_vals
1071 )
1072 {
1073         int             i, last;
1074
1075         for ( last = 0; !BER_BVISNULL( &a_vals[last] ); last++ );
1076         last--;
1077
1078         for ( i = 0; !BER_BVISNULL( &a_vals[i] ); i++ ) {
1079                 struct berval   dn,
1080                                 olddn = BER_BVNULL;
1081                 int             rc;
1082                 LDAPURLDesc     *ludp;
1083
1084                 rc = ldap_url_parse( a_vals[i].bv_val, &ludp );
1085                 if ( rc != LDAP_URL_SUCCESS ) {
1086                         /* leave attr untouched if massage failed */
1087                         continue;
1088                 }
1089
1090                 /* FIXME: URLs like "ldap:///dc=suffix" if passed
1091                  * thru ldap_url_parse() and ldap_url_desc2str()
1092                  * get rewritten as "ldap:///dc=suffix??base";
1093                  * we don't want this to occur... */
1094                 if ( ludp->lud_scope == LDAP_SCOPE_BASE ) {
1095                         ludp->lud_scope = LDAP_SCOPE_DEFAULT;
1096                 }
1097
1098                 ber_str2bv( ludp->lud_dn, 0, 0, &olddn );
1099
1100                 dn = olddn;
1101                 rc = rwm_dn_massage_pretty( dc, &olddn, &dn );
1102                 switch ( rc ) {
1103                 case LDAP_UNWILLING_TO_PERFORM:
1104                         /*
1105                          * FIXME: need to check if it may be considered 
1106                          * legal to trim values when adding/modifying;
1107                          * it should be when searching (e.g. ACLs).
1108                          */
1109                         ch_free( a_vals[i].bv_val );
1110                         if ( last > i ) {
1111                                 a_vals[i] = a_vals[last];
1112                         }
1113                         BER_BVZERO( &a_vals[last] );
1114                         last--;
1115                         i--;
1116                         break;
1117
1118                 default:
1119                         /* leave attr untouched if massage failed */
1120                         if ( !BER_BVISNULL( &dn ) && olddn.bv_val != dn.bv_val ) {
1121                                 char    *newurl;
1122
1123                                 ludp->lud_dn = dn.bv_val;
1124                                 newurl = ldap_url_desc2str( ludp );
1125                                 if ( newurl == NULL ) {
1126                                         /* FIXME: leave attr untouched
1127                                          * even if ldap_url_desc2str failed...
1128                                          */
1129                                         break;
1130                                 }
1131
1132                                 ch_free( a_vals[i].bv_val );
1133                                 ber_str2bv( newurl, 0, 1, &a_vals[i] );
1134                                 LDAP_FREE( newurl );
1135                                 ludp->lud_dn = olddn.bv_val;
1136                         }
1137                         break;
1138                 }
1139
1140                 ldap_free_urldesc( ludp );
1141         }
1142
1143         return 0;
1144 }
1145
1146 int
1147 rwm_dnattr_result_rewrite(
1148         dncookie                *dc,
1149         BerVarray               a_vals
1150 )
1151 {
1152         int             i, last;
1153
1154         for ( last = 0; !BER_BVISNULL( &a_vals[last] ); last++ );
1155         last--;
1156
1157         for ( i = 0; !BER_BVISNULL( &a_vals[i] ); i++ ) {
1158                 struct berval   dn;
1159                 int             rc;
1160                 
1161                 dn = a_vals[i];
1162                 rc = rwm_dn_massage_pretty( dc, &a_vals[i], &dn );
1163                 switch ( rc ) {
1164                 case LDAP_UNWILLING_TO_PERFORM:
1165                         /*
1166                          * FIXME: need to check if it may be considered 
1167                          * legal to trim values when adding/modifying;
1168                          * it should be when searching (e.g. ACLs).
1169                          */
1170                         ch_free( a_vals[i].bv_val );
1171                         if ( last > i ) {
1172                                 a_vals[i] = a_vals[last];
1173                         }
1174                         BER_BVZERO( &a_vals[last] );
1175                         last--;
1176                         break;
1177
1178                 default:
1179                         /* leave attr untouched if massage failed */
1180                         if ( !BER_BVISNULL( &dn ) && a_vals[i].bv_val != dn.bv_val ) {
1181                                 ch_free( a_vals[i].bv_val );
1182                                 a_vals[i] = dn;
1183                         }
1184                         break;
1185                 }
1186         }
1187
1188         return 0;
1189 }
1190
1191 void
1192 rwm_mapping_free( void *v_mapping )
1193 {
1194         struct ldapmapping *mapping = v_mapping;
1195
1196         if ( !BER_BVISNULL( &mapping[0].m_src ) ) {
1197                 ch_free( mapping[0].m_src.bv_val );
1198         }
1199
1200         if ( mapping[0].m_flags & RWMMAP_F_FREE_SRC ) {
1201                 if ( mapping[0].m_flags & RWMMAP_F_IS_OC ) {
1202                         if ( mapping[0].m_src_oc ) {
1203                                 ch_free( mapping[0].m_src_oc );
1204                         }
1205
1206                 } else {
1207                         if ( mapping[0].m_src_ad ) {
1208                                 ch_free( mapping[0].m_src_ad );
1209                         }
1210                 }
1211         }
1212
1213         if ( !BER_BVISNULL( &mapping[0].m_dst ) ) {
1214                 ch_free( mapping[0].m_dst.bv_val );
1215         }
1216
1217         if ( mapping[0].m_flags & RWMMAP_F_FREE_DST ) {
1218                 if ( mapping[0].m_flags & RWMMAP_F_IS_OC ) {
1219                         if ( mapping[0].m_dst_oc ) {
1220                                 ch_free( mapping[0].m_dst_oc );
1221                         }
1222
1223                 } else {
1224                         if ( mapping[0].m_dst_ad ) {
1225                                 ch_free( mapping[0].m_dst_ad );
1226                         }
1227                 }
1228         }
1229
1230         ch_free( mapping );
1231
1232 }
1233
1234 #endif /* SLAPD_OVER_RWM */