]> git.sur5r.net Git - openldap/blobdiff - servers/slapd/overlays/collect.c
Only return requested attrs in sssvlv response
[openldap] / servers / slapd / overlays / collect.c
index 680fb9b59f8309a55d00c46409eac3b39acccb89..2fc08ed4135d2d5c59502807bf36d998d245e061 100644 (file)
@@ -2,7 +2,7 @@
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 2003-2008 The OpenLDAP Foundation.
+ * Copyright 2003-2011 The OpenLDAP Foundation.
  * Portions Copyright 2003 Howard Chu.
  * All rights reserved.
  *
  * is no effect. If no attribute was configured, there is no effect.
  */
 
-/* Use C99 flexible array member if supported, C89 "struct hack" otherwise. */
-#define FLEXIBLE_STRUCT_SIZE(stype, mtype, mcount) \
-       (sizeof(stype) + sizeof(mtype) * ((mcount) - (0 FLEXIBLE_ARRAY_MEMBER)))
-#if __STDC_VERSION__>=199901L || (defined __GNUC__ && !defined __STRICT_ANSI__)
-#define FLEXIBLE_ARRAY_MEMBER
-#else
-#define FLEXIBLE_ARRAY_MEMBER +1
-#endif
-
 typedef struct collect_info {
        struct collect_info *ci_next;
        struct berval ci_dn;
        int ci_ad_num;
-       AttributeDescription *ci_ad[FLEXIBLE_ARRAY_MEMBER];
+       AttributeDescription *ci_ad[1];
 } collect_info;
 
+static int collect_cf( ConfigArgs *c );
+
+static ConfigTable collectcfg[] = {
+       { "collectinfo", "dn> <attribute", 3, 3, 0,
+         ARG_MAGIC, collect_cf,
+         "( OLcfgOvAt:19.1 NAME 'olcCollectInfo' "
+         "DESC 'DN of entry and attribute to distribute' "
+         "EQUALITY caseIgnoreMatch "
+         "SYNTAX OMsDirectoryString )", NULL, NULL },
+       { NULL, NULL, 0, 0, 0, ARG_IGNORED }
+};
+
+static ConfigOCs collectocs[] = {
+       { "( OLcfgOvOc:19.1 "
+         "NAME 'olcCollectConfig' "
+         "DESC 'Collective Attribute configuration' "
+         "SUP olcOverlayConfig "
+         "MAY olcCollectInfo )",
+         Cft_Overlay, collectcfg },
+       { NULL, 0, NULL }
+};
+
 /*
  * inserts a collect_info into on->on_bi.bi_private taking into account
  * order. this means longer dn's (i.e. more specific dn's) will be found
@@ -81,7 +94,7 @@ insert_ordered( slap_overinst *on, collect_info *ci ) {
                                ci->ci_next = NULL;
                        }
                        found = 1;
-               } else if (find->ci_dn.bv_len <= ci->ci_dn.bv_len) { 
+               } else if (find->ci_dn.bv_len < ci->ci_dn.bv_len) { 
                        /* insert into list here */
                        if (prev == NULL) {
                                /* entry is head of list */
@@ -160,7 +173,11 @@ collect_cf( ConfigArgs *c )
                        collect_info **cip, *ci;
                        int i;
                        cip = (collect_info **)&on->on_bi.bi_private;
-                       for ( i=0; i <= c->valx; i++, cip = &ci->ci_next ) ci = *cip;
+                       ci = *cip;
+                       for ( i=0; i < c->valx; i++ ) {
+                               cip = &ci->ci_next;
+                               ci = *cip;
+                       }
                        *cip = ci->ci_next;
                        ch_free( ci->ci_dn.bv_val );
                        ch_free( ci );
@@ -183,10 +200,6 @@ collect_cf( ConfigArgs *c )
                        arg = strtok(NULL, ",");
                }
 
-               /* allocate config info with room for attribute array */
-               ci = ch_malloc( FLEXIBLE_STRUCT_SIZE( collect_info,
-                       AttributeDescription *, count + 1 ));
-
                /* validate and normalize dn */
                ber_str2bv( c->argv[1], 0, 0, &bv );
                if ( dnNormalize( 0, NULL, NULL, &bv, &dn, NULL ) ) {
@@ -197,6 +210,30 @@ collect_cf( ConfigArgs *c )
                        return ARG_BAD_CONF;
                }
 
+               /* check for duplicate DNs */
+               for ( ci = (collect_info *)on->on_bi.bi_private; ci;
+                       ci = ci->ci_next ) {
+                       /* If new DN is longest, there are no possible matches */
+                       if ( dn.bv_len > ci->ci_dn.bv_len ) {
+                               ci = NULL;
+                               break;
+                       }
+                       if ( bvmatch( &dn, &ci->ci_dn )) {
+                               break;
+                       }
+               }
+               if ( ci ) {
+                       snprintf( c->cr_msg, sizeof( c->cr_msg ), "%s DN already configured: \"%s\"",
+                               c->argv[0], c->argv[1] );
+                       Debug( LDAP_DEBUG_CONFIG|LDAP_DEBUG_NONE,
+                               "%s: %s\n", c->log, c->cr_msg, 0 );
+                       return ARG_BAD_CONF;
+               }
+
+               /* allocate config info with room for attribute array */
+               ci = ch_malloc( sizeof( collect_info ) +
+                       sizeof( AttributeDescription * ) * count );
+
                /* load attribute description for attribute list */
                arg = c->argv[2];
                for( idx=0; idx<count; idx++) {
@@ -208,6 +245,7 @@ collect_cf( ConfigArgs *c )
                                        c->argv[0], arg);
                                Debug( LDAP_DEBUG_CONFIG|LDAP_DEBUG_NONE,
                                        "%s: %s\n", c->log, c->cr_msg, 0 );
+                               ch_free( ci );
                                return ARG_BAD_CONF;
                        }
                        while(*arg!='\0') {
@@ -228,31 +266,34 @@ collect_cf( ConfigArgs *c )
                /* creates list of ci's ordered by dn length */ 
                insert_ordered ( on, ci );
 
+               /* New ci wasn't simply appended to end, adjust its
+                * position in the config entry's a_vals
+                */
+               if ( c->ca_entry && ci->ci_next ) {
+                       Attribute *a = attr_find( c->ca_entry->e_attrs,
+                               collectcfg[0].ad );
+                       if ( a ) {
+                               struct berval bv, nbv;
+                               collect_info *c2 = (collect_info *)on->on_bi.bi_private;
+                               int i, j;
+                               for ( i=0; c2 != ci; i++, c2 = c2->ci_next );
+                               bv = a->a_vals[a->a_numvals-1];
+                               nbv = a->a_nvals[a->a_numvals-1];
+                               for ( j=a->a_numvals-1; j>i; j-- ) {
+                                       a->a_vals[j] = a->a_vals[j-1];
+                                       a->a_nvals[j] = a->a_nvals[j-1];
+                               }
+                               a->a_vals[j] = bv;
+                               a->a_nvals[j] = nbv;
+                       }
+               }
+
                rc = 0;
                }
        }
        return rc;
 }
 
-static ConfigTable collectcfg[] = {
-       { "collectinfo", "dn> <attribute", 3, 3, 0,
-         ARG_MAGIC, collect_cf,
-         "( OLcfgOvAt:19.1 NAME 'olcCollectInfo' "
-         "DESC 'DN of entry and attribute to distribute' "
-         "SYNTAX OMsDirectoryString )", NULL, NULL },
-       { NULL, NULL, 0, 0, 0, ARG_IGNORED }
-};
-
-static ConfigOCs collectocs[] = {
-       { "( OLcfgOvOc:19.1 "
-         "NAME 'olcCollectConfig' "
-         "DESC 'Collective Attribute configuration' "
-         "SUP olcOverlayConfig "
-         "MAY olcCollectInfo )",
-         Cft_Overlay, collectcfg },
-       { NULL, 0, NULL }
-};
-
 static int
 collect_destroy(
        BackendDB *be,
@@ -345,11 +386,7 @@ collect_response( Operation *op, SlapReply *rs )
                        * don't modify it directly. Make a copy and
                        * work with that instead.
                        */
-                       if ( !( rs->sr_flags & REP_ENTRY_MODIFIABLE )) {
-                               rs->sr_entry = entry_dup( rs->sr_entry );
-                               rs->sr_flags |= REP_ENTRY_MODIFIABLE |
-                                       REP_ENTRY_MUSTBEFREED;
-                       }
+                       rs_entry2modifiable( op, rs, on );
 
                        /* Loop for each attribute in this collectinfo */
                        for(idx=0; idx<ci->ci_ad_num; idx++) {