git.sur5r.net Git - openldap/blob - servers/slapd/back-meta/cache-merge.c

   1 /* Copyright (c) 2003 by International Business Machines, Inc.
   2  *
   3  * International Business Machines, Inc. (hereinafter called IBM) grants
   4  * permission under its copyrights to use, copy, modify, and distribute this
   5  * Software with or without fee, provided that the above copyright notice and
   6  * all paragraphs of this notice appear in all copies, and that the name of IBM
   7  * not be used in connection with the marketing of any product incorporating
   8  * the Software or modifications thereof, without specific, written prior
   9  * permission.
  10  *
  11  * THE SOFTWARE IS PROVIDED "AS IS", AND IBM DISCLAIMS ALL WARRANTIES,
  12  * INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  13  * PARTICULAR PURPOSE.  IN NO EVENT SHALL IBM BE LIABLE FOR ANY SPECIAL,
  14  * DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER ARISING
  15  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE, EVEN
  16  * IF IBM IS APPRISED OF THE POSSIBILITY OF SUCH DAMAGES.
  17  */
  18
  19 #include "portable.h"
  20
  21 #include <stdio.h>
  22
  23 #include <ac/socket.h>
  24 #include <ac/string.h>
  25 #include <ac/time.h>
  26
  27 #include "slap.h"
  28 #include "ldif.h"
  29 #include "../back-ldap/back-ldap.h"
  30 #include "back-meta.h"
  31 #include "ldap_pvt.h"
  32 #undef ldap_debug       /* silence a warning in ldap-int.h */
  33 #include "ldap_log.h"
  34 #include "../../../libraries/libldap/ldap-int.h"
  35 #include <sys/time.h>
  36
  37 #ifdef LDAP_CACHING
  38
  39 static struct berval bv_queryid_any = BER_BVC( "(queryid=*)" );
  40
  41 static Attribute*
  42 add_attribute(AttributeDescription *ad,
  43         Entry* e,
  44         BerVarray value_array
  45 );
  46
  47 static int
  48 null_response (
  49         Operation       *op,
  50         SlapReply       *rs
  51 );
  52
  53 static int
  54 normalize_values( Attribute* attr );
  55
  56 struct entry_info {
  57         int                     size_init;
  58         int                     size_final;
  59         int                     added;
  60         Entry*                  entry;
  61         struct berval*          uuid;
  62         struct timeval          tv;     /* time */
  63         enum type_of_result     err;
  64         Backend*                glue_be;
  65 };
  66
  67
  68 int
  69 get_entry_size(
  70         Entry* e,
  71         int size_init,
  72         struct exception* result )
  73 {
  74         Attribute       *a;
  75         struct berval   bv;
  76         int             i;
  77         int             tmplen;
  78         int             size=0;
  79
  80         if ( result )
  81                 result->type = SUCCESS;
  82
  83         if ( e->e_dn != NULL ) {
  84                 tmplen = strlen( e->e_dn );
  85                 size = LDIF_SIZE_NEEDED( 2, tmplen );
  86         }
  87
  88         for ( a = e->e_attrs; a != NULL; a = a->a_next ) {
  89                 for ( i = 0; a->a_vals[i].bv_val != NULL; i++ ) {
  90                         bv = a->a_vals[i];
  91                         tmplen = a->a_desc->ad_cname.bv_len;
  92                         size += LDIF_SIZE_NEEDED( tmplen, bv.bv_len);
  93                 }
  94         }
  95         if ((size < size_init) && result) {
  96                 result->type = SIZE_ERR;
  97         }
  98         return size;
  99 }
 100
 101 int
 102 merge_entry(
 103         Operation               *op,
 104         SlapReply               *rs,
 105         struct berval*          query_uuid,
 106         struct exception*       result )
 107 {
 108         struct entry_info info;
 109         int             rc;
 110         Modifications* modlist = NULL;
 111         const char*     text = NULL;
 112         BerVarray               value_array;
 113         Attribute               *uuid_attr, *attr;
 114         Entry                   *e;
 115
 116         SlapReply sreply = {REP_RESULT};
 117
 118         Operation op_tmp = *op;
 119         slap_callback cb;
 120
 121         sreply.sr_entry = NULL;
 122         sreply.sr_nentries = 0;
 123
 124         e = ( Entry * ) ch_calloc( 1, sizeof( Entry ));
 125         dnPrettyNormal(0, &rs->sr_entry->e_name, &e->e_name, &e->e_nname, op->o_tmpmemctx);
 126
 127         e->e_private = NULL;
 128         e->e_attrs = NULL;
 129         e->e_bv.bv_val = NULL;
 130
 131         /* add queryid attribute */
 132         value_array = (struct berval *)malloc(2 * sizeof( struct berval) );
 133         ber_dupbv(value_array, query_uuid);
 134         value_array[1].bv_val = NULL;
 135         value_array[1].bv_len = 0;
 136
 137         uuid_attr = add_attribute(slap_schema.si_ad_queryid, e, value_array);
 138
 139         /* append the attribute list from the fetched entry */
 140         uuid_attr->a_next = rs->sr_entry->e_attrs;
 141         rs->sr_entry->e_attrs = NULL;
 142
 143         for ( attr = e->e_attrs; attr; attr = attr->a_next ) {
 144                 if ( normalize_values( attr ) ) {
 145                         info.err = MERGE_ERR;
 146                         result->rc = info.err;
 147                         return 0;
 148                 }
 149         }
 150
 151         info.entry = e;
 152         info.uuid = query_uuid;
 153         info.size_init = get_entry_size( rs->sr_entry, 0, 0 );
 154         info.size_final = 0;
 155         info.added = 0;
 156         info.glue_be = op->o_bd;
 157         info.err = SUCCESS;
 158         cb.sc_private = &info;
 159         cb.sc_response = null_response;
 160
 161         op_tmp.o_tag = LDAP_REQ_ADD;
 162         op_tmp.o_protocol = LDAP_VERSION3;
 163         op_tmp.o_callback = &cb;
 164         op_tmp.o_time = slap_get_time();
 165         op_tmp.o_do_not_cache = 1;
 166
 167         op_tmp.ora_e = e;
 168         op_tmp.o_req_dn = e->e_name;
 169         op_tmp.o_req_ndn = e->e_nname;
 170         rc = op->o_bd->be_add( &op_tmp, &sreply );
 171
 172         if ( rc != LDAP_SUCCESS ) {
 173                 if ( rc == LDAP_ALREADY_EXISTS ) {
 174                         slap_entry2mods( e, &modlist, &text );
 175                         op_tmp.o_tag = LDAP_REQ_MODIFY;
 176                         op_tmp.orm_modlist = modlist;
 177                         op_tmp.o_req_dn = e->e_name;
 178                         op_tmp.o_req_ndn = e->e_nname;
 179                         rc = op->o_bd->be_modify( &op_tmp, &sreply );
 180                         result->rc = info.added;
 181                 } else if ( rc == LDAP_REFERRAL ||
 182                                         rc == LDAP_NO_SUCH_OBJECT ) {
 183                         slap_entry2mods( e, &modlist, &text );
 184                         syncrepl_add_glue( NULL, NULL, &op_tmp, e, modlist, 0, NULL, NULL );
 185                         result->rc = info.added;
 186                 } else {
 187                         result->rc = 0;
 188                 }
 189                 if ( modlist != NULL ) slap_mods_free( modlist );
 190         } else {
 191                 info.size_init = 0;
 192                 result->rc = info.added;
 193                 be_entry_release_w( &op_tmp, e );
 194         }
 195
 196         if ( result->rc )
 197                 info.size_final = get_entry_size( e, info.size_init, result );
 198         else
 199                 info.size_final = info.size_init;
 200
 201         return ( info.size_final - info.size_init );
 202 }
 203
 204 static Attribute*
 205 add_attribute(AttributeDescription *ad,
 206         Entry* e,
 207         BerVarray value_array)
 208 {
 209         Attribute* new_attr, *last_attr;
 210         const char* text;
 211
 212         if (e->e_attrs == NULL)
 213                 last_attr = NULL;
 214         else
 215                 for (last_attr = e->e_attrs; last_attr->a_next;
 216                                 last_attr = last_attr->a_next)
 217                         ;
 218
 219         new_attr = (Attribute*)malloc(sizeof(Attribute));
 220         if (last_attr)
 221                 last_attr->a_next = new_attr;
 222         else
 223                 e->e_attrs = new_attr;
 224
 225         new_attr->a_next = NULL;
 226         new_attr->a_desc = NULL;
 227         new_attr->a_vals = value_array;
 228         new_attr->a_desc = ad;
 229
 230         return new_attr;
 231 }
 232
 233 static int
 234 null_response (
 235         Operation       *op,
 236         SlapReply       *rs )
 237 {
 238         return 0;
 239 }
 240
 241 static int
 242 normalize_values( Attribute* attr )
 243 {
 244         int nvals, rc, i;
 245
 246         if (attr->a_vals == NULL) {
 247                 attr->a_nvals = NULL;
 248                 return 0;
 249         }
 250
 251         for ( nvals = 0; attr->a_vals[nvals].bv_val; nvals++ )
 252                 ;
 253
 254         attr->a_nvals = (struct berval*)ch_malloc((nvals+1)*sizeof(struct berval));
 255
 256         if ( attr->a_desc->ad_type->sat_equality &&
 257                                 attr->a_desc->ad_type->sat_equality->smr_normalize )
 258         {
 259                 for ( i = 0; i < nvals; i++ ) {
 260                         rc = attr->a_desc->ad_type->sat_equality->smr_normalize(
 261                                 0,
 262                                 attr->a_desc->ad_type->sat_syntax,
 263                                 attr->a_desc->ad_type->sat_equality,
 264                                 &attr->a_vals[i], &attr->a_nvals[i], NULL );
 265                         if ( rc ) {
 266 #ifdef NEW_LOGGING
 267                                 LDAP_LOG( OPERATION, DETAIL1,
 268                                         "Error in normalizing attribute %s value %d (%d)\n",
 269                                         attr->a_desc->ad_cname.bv_val, i, rc );
 270 #else
 271                                 Debug( LDAP_DEBUG_ANY,
 272                                         "Error in normalizing attribute %s value %d (%d)\n",
 273                                         attr->a_desc->ad_cname.bv_val, i, rc );
 274 #endif
 275                                 return rc;
 276                         }
 277                 }
 278         } else {
 279                 for ( i = 0; i < nvals; i++ ) {
 280                         ber_dupbv( &attr->a_nvals[i], &attr->a_vals[i] );
 281                 }
 282         }
 283
 284         attr->a_nvals[i].bv_val = NULL;
 285         attr->a_nvals[i].bv_len = 0;
 286
 287         return LDAP_SUCCESS;
 288 }
 289
 290 #endif /* LDAP_CACHING */