git.sur5r.net Git - openldap/blob - servers/slapd/overlays/syncprov.c

   1 /* $OpenLDAP$ */
   2 /* syncprov.c - syncrepl provider */
   3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   4  *
   5  * Copyright 2004-2006 The OpenLDAP Foundation.
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted only as authorized by the OpenLDAP
  10  * Public License.
  11  *
  12  * A copy of this license is available in the file LICENSE in the
  13  * top-level directory of the distribution or, alternatively, at
  14  * <http://www.OpenLDAP.org/license.html>.
  15  */
  16 /* ACKNOWLEDGEMENTS:
  17  * This work was initially developed by Howard Chu for inclusion in
  18  * OpenLDAP Software.
  19  */
  20
  21 #include "portable.h"
  22
  23 #ifdef SLAPD_OVER_SYNCPROV
  24
  25 #include <ac/string.h>
  26 #include "lutil.h"
  27 #include "slap.h"
  28 #include "config.h"
  29 #include "ldap_rq.h"
  30
  31 /* A modify request on a particular entry */
  32 typedef struct modinst {
  33         struct modinst *mi_next;
  34         Operation *mi_op;
  35 } modinst;
  36
  37 typedef struct modtarget {
  38         struct modinst *mt_mods;
  39         struct modinst *mt_tail;
  40         Operation *mt_op;
  41         ldap_pvt_thread_mutex_t mt_mutex;
  42 } modtarget;
  43
  44 /* A queued result of a persistent search */
  45 typedef struct syncres {
  46         struct syncres *s_next;
  47         struct berval s_dn;
  48         struct berval s_ndn;
  49         struct berval s_uuid;
  50         struct berval s_csn;
  51         char s_mode;
  52         char s_isreference;
  53 } syncres;
  54
  55 /* Record of a persistent search */
  56 typedef struct syncops {
  57         struct syncops *s_next;
  58         struct berval   s_base;         /* ndn of search base */
  59         ID              s_eid;          /* entryID of search base */
  60         Operation       *s_op;          /* search op */
  61         int             s_rid;
  62         struct berval s_filterstr;
  63         int             s_flags;        /* search status */
  64 #define PS_IS_REFRESHING        0x01
  65 #define PS_IS_DETACHED          0x02
  66 #define PS_WROTE_BASE           0x04
  67 #define PS_FIND_BASE            0x08
  68 #define PS_FIX_FILTER           0x10
  69
  70         int             s_inuse;        /* reference count */
  71         struct syncres *s_res;
  72         struct syncres *s_restail;
  73         struct re_s     *s_qtask;       /* task for playing psearch responses */
  74 #define RUNQ_INTERVAL   36000   /* a long time */
  75         ldap_pvt_thread_mutex_t s_mutex;
  76 } syncops;
  77
  78 /* A received sync control */
  79 typedef struct sync_control {
  80         struct sync_cookie sr_state;
  81         int sr_rhint;
  82 } sync_control;
  83
  84 #if 0 /* moved back to slap.h */
  85 #define o_sync  o_ctrlflag[slap_cids.sc_LDAPsync]
  86 #endif
  87 /* o_sync_mode uses data bits of o_sync */
  88 #define o_sync_mode     o_ctrlflag[slap_cids.sc_LDAPsync]
  89
  90 #define SLAP_SYNC_NONE                                  (LDAP_SYNC_NONE<<SLAP_CONTROL_SHIFT)
  91 #define SLAP_SYNC_REFRESH                               (LDAP_SYNC_REFRESH_ONLY<<SLAP_CONTROL_SHIFT)
  92 #define SLAP_SYNC_PERSIST                               (LDAP_SYNC_RESERVED<<SLAP_CONTROL_SHIFT)
  93 #define SLAP_SYNC_REFRESH_AND_PERSIST   (LDAP_SYNC_REFRESH_AND_PERSIST<<SLAP_CONTROL_SHIFT)
  94
  95 /* Record of which searches matched at premodify step */
  96 typedef struct syncmatches {
  97         struct syncmatches *sm_next;
  98         syncops *sm_op;
  99 } syncmatches;
 100
 101 /* Session log data */
 102 typedef struct slog_entry {
 103         struct slog_entry *se_next;
 104         struct berval se_uuid;
 105         struct berval se_csn;
 106         ber_tag_t       se_tag;
 107 } slog_entry;
 108
 109 typedef struct sessionlog {
 110         struct berval   sl_mincsn;
 111         int             sl_num;
 112         int             sl_size;
 113         slog_entry *sl_head;
 114         slog_entry *sl_tail;
 115         ldap_pvt_thread_mutex_t sl_mutex;
 116 } sessionlog;
 117
 118 /* The main state for this overlay */
 119 typedef struct syncprov_info_t {
 120         syncops         *si_ops;
 121         struct berval   si_ctxcsn;      /* ldapsync context */
 122         int             si_chkops;      /* checkpointing info */
 123         int             si_chktime;
 124         int             si_numops;      /* number of ops since last checkpoint */
 125         int             si_nopres;      /* Skip present phase */
 126         int             si_usehint;     /* use reload hint */
 127         time_t  si_chklast;     /* time of last checkpoint */
 128         Avlnode *si_mods;       /* entries being modified */
 129         sessionlog      *si_logs;
 130         ldap_pvt_thread_mutex_t si_csn_mutex;
 131         ldap_pvt_thread_mutex_t si_ops_mutex;
 132         ldap_pvt_thread_mutex_t si_mods_mutex;
 133         char            si_ctxcsnbuf[LDAP_LUTIL_CSNSTR_BUFSIZE];
 134 } syncprov_info_t;
 135
 136 typedef struct opcookie {
 137         slap_overinst *son;
 138         syncmatches *smatches;
 139         struct berval sdn;      /* DN of entry, for deletes */
 140         struct berval sndn;
 141         struct berval suuid;    /* UUID of entry */
 142         struct berval sctxcsn;
 143         int sreference; /* Is the entry a reference? */
 144 } opcookie;
 145
 146 typedef struct fbase_cookie {
 147         struct berval *fdn;     /* DN of a modified entry, for scope testing */
 148         syncops *fss;   /* persistent search we're testing against */
 149         int fbase;      /* if TRUE we found the search base and it's still valid */
 150         int fscope;     /* if TRUE then fdn is within the psearch scope */
 151 } fbase_cookie;
 152
 153 static AttributeName csn_anlist[3];
 154 static AttributeName uuid_anlist[2];
 155
 156 /* Build a LDAPsync intermediate state control */
 157 static int
 158 syncprov_state_ctrl(
 159         Operation       *op,
 160         SlapReply       *rs,
 161         Entry           *e,
 162         int             entry_sync_state,
 163         LDAPControl     **ctrls,
 164         int             num_ctrls,
 165         int             send_cookie,
 166         struct berval   *cookie )
 167 {
 168         Attribute* a;
 169         int ret;
 170
 171         BerElementBuffer berbuf;
 172         BerElement *ber = (BerElement *)&berbuf;
 173
 174         struct berval   entryuuid_bv = BER_BVNULL;
 175
 176         ber_init2( ber, 0, LBER_USE_DER );
 177         ber_set_option( ber, LBER_OPT_BER_MEMCTX, &op->o_tmpmemctx );
 178
 179         ctrls[num_ctrls] = op->o_tmpalloc( sizeof ( LDAPControl ), op->o_tmpmemctx );
 180
 181         for ( a = e->e_attrs; a != NULL; a = a->a_next ) {
 182                 AttributeDescription *desc = a->a_desc;
 183                 if ( desc == slap_schema.si_ad_entryUUID ) {
 184                         entryuuid_bv = a->a_nvals[0];
 185                         break;
 186                 }
 187         }
 188
 189         /* FIXME: what if entryuuid is NULL or empty ? */
 190
 191         if ( send_cookie && cookie ) {
 192                 ber_printf( ber, "{eOON}",
 193                         entry_sync_state, &entryuuid_bv, cookie );
 194         } else {
 195                 ber_printf( ber, "{eON}",
 196                         entry_sync_state, &entryuuid_bv );
 197         }
 198
 199         ctrls[num_ctrls]->ldctl_oid = LDAP_CONTROL_SYNC_STATE;
 200         ctrls[num_ctrls]->ldctl_iscritical = (op->o_sync == SLAP_CONTROL_CRITICAL);
 201         ret = ber_flatten2( ber, &ctrls[num_ctrls]->ldctl_value, 1 );
 202
 203         ber_free_buf( ber );
 204
 205         if ( ret < 0 ) {
 206                 Debug( LDAP_DEBUG_TRACE,
 207                         "slap_build_sync_ctrl: ber_flatten2 failed\n",
 208                         0, 0, 0 );
 209                 send_ldap_error( op, rs, LDAP_OTHER, "internal error" );
 210                 return ret;
 211         }
 212
 213         return LDAP_SUCCESS;
 214 }
 215
 216 /* Build a LDAPsync final state control */
 217 static int
 218 syncprov_done_ctrl(
 219         Operation       *op,
 220         SlapReply       *rs,
 221         LDAPControl     **ctrls,
 222         int                     num_ctrls,
 223         int                     send_cookie,
 224         struct berval *cookie,
 225         int                     refreshDeletes )
 226 {
 227         int ret;
 228         BerElementBuffer berbuf;
 229         BerElement *ber = (BerElement *)&berbuf;
 230
 231         ber_init2( ber, NULL, LBER_USE_DER );
 232         ber_set_option( ber, LBER_OPT_BER_MEMCTX, &op->o_tmpmemctx );
 233
 234         ctrls[num_ctrls] = op->o_tmpalloc( sizeof ( LDAPControl ), op->o_tmpmemctx );
 235
 236         ber_printf( ber, "{" );
 237         if ( send_cookie && cookie ) {
 238                 ber_printf( ber, "O", cookie );
 239         }
 240         if ( refreshDeletes == LDAP_SYNC_REFRESH_DELETES ) {
 241                 ber_printf( ber, "b", refreshDeletes );
 242         }
 243         ber_printf( ber, "N}" );
 244
 245         ctrls[num_ctrls]->ldctl_oid = LDAP_CONTROL_SYNC_DONE;
 246         ctrls[num_ctrls]->ldctl_iscritical = (op->o_sync == SLAP_CONTROL_CRITICAL);
 247         ret = ber_flatten2( ber, &ctrls[num_ctrls]->ldctl_value, 1 );
 248
 249         ber_free_buf( ber );
 250
 251         if ( ret < 0 ) {
 252                 Debug( LDAP_DEBUG_TRACE,
 253                         "syncprov_done_ctrl: ber_flatten2 failed\n",
 254                         0, 0, 0 );
 255                 send_ldap_error( op, rs, LDAP_OTHER, "internal error" );
 256                 return ret;
 257         }
 258
 259         return LDAP_SUCCESS;
 260 }
 261
 262 static int
 263 syncprov_sendinfo(
 264         Operation       *op,
 265         SlapReply       *rs,
 266         int                     type,
 267         struct berval *cookie,
 268         int                     refreshDone,
 269         BerVarray       syncUUIDs,
 270         int                     refreshDeletes )
 271 {
 272         BerElementBuffer berbuf;
 273         BerElement *ber = (BerElement *)&berbuf;
 274         struct berval rspdata;
 275
 276         int ret;
 277
 278         ber_init2( ber, NULL, LBER_USE_DER );
 279         ber_set_option( ber, LBER_OPT_BER_MEMCTX, &op->o_tmpmemctx );
 280
 281         if ( type ) {
 282                 switch ( type ) {
 283                 case LDAP_TAG_SYNC_NEW_COOKIE:
 284                         ber_printf( ber, "tO", type, cookie );
 285                         break;
 286                 case LDAP_TAG_SYNC_REFRESH_DELETE:
 287                 case LDAP_TAG_SYNC_REFRESH_PRESENT:
 288                         ber_printf( ber, "t{", type );
 289                         if ( cookie ) {
 290                                 ber_printf( ber, "O", cookie );
 291                         }
 292                         if ( refreshDone == 0 ) {
 293                                 ber_printf( ber, "b", refreshDone );
 294                         }
 295                         ber_printf( ber, "N}" );
 296                         break;
 297                 case LDAP_TAG_SYNC_ID_SET:
 298                         ber_printf( ber, "t{", type );
 299                         if ( cookie ) {
 300                                 ber_printf( ber, "O", cookie );
 301                         }
 302                         if ( refreshDeletes == 1 ) {
 303                                 ber_printf( ber, "b", refreshDeletes );
 304                         }
 305                         ber_printf( ber, "[W]", syncUUIDs );
 306                         ber_printf( ber, "N}" );
 307                         break;
 308                 default:
 309                         Debug( LDAP_DEBUG_TRACE,
 310                                 "syncprov_sendinfo: invalid syncinfo type (%d)\n",
 311                                 type, 0, 0 );
 312                         return LDAP_OTHER;
 313                 }
 314         }
 315
 316         ret = ber_flatten2( ber, &rspdata, 0 );
 317
 318         if ( ret < 0 ) {
 319                 Debug( LDAP_DEBUG_TRACE,
 320                         "syncprov_sendinfo: ber_flatten2 failed\n",
 321                         0, 0, 0 );
 322                 send_ldap_error( op, rs, LDAP_OTHER, "internal error" );
 323                 return ret;
 324         }
 325
 326         rs->sr_rspoid = LDAP_SYNC_INFO;
 327         rs->sr_rspdata = &rspdata;
 328         send_ldap_intermediate( op, rs );
 329         rs->sr_rspdata = NULL;
 330         ber_free_buf( ber );
 331
 332         return LDAP_SUCCESS;
 333 }
 334
 335 /* Find a modtarget in an AVL tree */
 336 static int
 337 sp_avl_cmp( const void *c1, const void *c2 )
 338 {
 339         const modtarget *m1, *m2;
 340         int rc;
 341
 342         m1 = c1; m2 = c2;
 343         rc = m1->mt_op->o_req_ndn.bv_len - m2->mt_op->o_req_ndn.bv_len;
 344
 345         if ( rc ) return rc;
 346         return ber_bvcmp( &m1->mt_op->o_req_ndn, &m2->mt_op->o_req_ndn );
 347 }
 348
 349 /* syncprov_findbase:
 350  *   finds the true DN of the base of a search (with alias dereferencing) and
 351  * checks to make sure the base entry doesn't get replaced with a different
 352  * entry (e.g., swapping trees via ModDN, or retargeting an alias). If a
 353  * change is detected, any persistent search on this base must be terminated /
 354  * reloaded.
 355  *   On the first call, we just save the DN and entryID. On subsequent calls
 356  * we compare the DN and entryID with the saved values.
 357  */
 358 static int
 359 findbase_cb( Operation *op, SlapReply *rs )
 360 {
 361         slap_callback *sc = op->o_callback;
 362
 363         if ( rs->sr_type == REP_SEARCH && rs->sr_err == LDAP_SUCCESS ) {
 364                 fbase_cookie *fc = sc->sc_private;
 365
 366                 /* If no entryID, we're looking for the first time.
 367                  * Just store whatever we got.
 368                  */
 369                 if ( fc->fss->s_eid == NOID ) {
 370                         fc->fbase = 2;
 371                         fc->fss->s_eid = rs->sr_entry->e_id;
 372                         ber_dupbv( &fc->fss->s_base, &rs->sr_entry->e_nname );
 373
 374                 } else if ( rs->sr_entry->e_id == fc->fss->s_eid &&
 375                         dn_match( &rs->sr_entry->e_nname, &fc->fss->s_base )) {
 376
 377                 /* OK, the DN is the same and the entryID is the same. */
 378                         fc->fbase = 1;
 379                 }
 380         }
 381         if ( rs->sr_err != LDAP_SUCCESS ) {
 382                 Debug( LDAP_DEBUG_ANY, "findbase failed! %d\n", rs->sr_err,0,0 );
 383         }
 384         return LDAP_SUCCESS;
 385 }
 386
 387 static Filter generic_filter = { LDAP_FILTER_PRESENT, { 0 }, NULL };
 388 static struct berval generic_filterstr = BER_BVC("(objectclass=*)");
 389
 390 static int
 391 syncprov_findbase( Operation *op, fbase_cookie *fc )
 392 {
 393         opcookie *opc = op->o_callback->sc_private;
 394         slap_overinst *on = opc->son;
 395
 396         /* Use basic parameters from syncrepl search, but use
 397          * current op's threadctx / tmpmemctx
 398          */
 399         ldap_pvt_thread_mutex_lock( &fc->fss->s_mutex );
 400         if ( fc->fss->s_flags & PS_FIND_BASE ) {
 401                 slap_callback cb = {0};
 402                 Operation fop;
 403                 SlapReply frs = { REP_RESULT };
 404                 int rc;
 405
 406                 fc->fss->s_flags ^= PS_FIND_BASE;
 407                 ldap_pvt_thread_mutex_unlock( &fc->fss->s_mutex );
 408
 409                 fop = *fc->fss->s_op;
 410
 411                 fop.o_hdr = op->o_hdr;
 412                 fop.o_bd = op->o_bd;
 413                 fop.o_time = op->o_time;
 414                 fop.o_tincr = op->o_tincr;
 415
 416                 cb.sc_response = findbase_cb;
 417                 cb.sc_private = fc;
 418
 419                 fop.o_sync_mode = 0;    /* turn off sync mode */
 420                 fop.o_managedsait = SLAP_CONTROL_CRITICAL;
 421                 fop.o_callback = &cb;
 422                 fop.o_tag = LDAP_REQ_SEARCH;
 423                 fop.ors_scope = LDAP_SCOPE_BASE;
 424                 fop.ors_limit = NULL;
 425                 fop.ors_slimit = 1;
 426                 fop.ors_tlimit = SLAP_NO_LIMIT;
 427                 fop.ors_attrs = slap_anlist_no_attrs;
 428                 fop.ors_attrsonly = 1;
 429                 fop.ors_filter = &generic_filter;
 430                 fop.ors_filterstr = generic_filterstr;
 431
 432                 fop.o_bd->bd_info = on->on_info->oi_orig;
 433                 rc = fop.o_bd->be_search( &fop, &frs );
 434                 fop.o_bd->bd_info = (BackendInfo *)on;
 435         } else {
 436                 ldap_pvt_thread_mutex_unlock( &fc->fss->s_mutex );
 437                 fc->fbase = 1;
 438         }
 439
 440         /* After the first call, see if the fdn resides in the scope */
 441         if ( fc->fbase == 1 ) {
 442                 switch ( fc->fss->s_op->ors_scope ) {
 443                 case LDAP_SCOPE_BASE:
 444                         fc->fscope = dn_match( fc->fdn, &fc->fss->s_base );
 445                         break;
 446                 case LDAP_SCOPE_ONELEVEL: {
 447                         struct berval pdn;
 448                         dnParent( fc->fdn, &pdn );
 449                         fc->fscope = dn_match( &pdn, &fc->fss->s_base );
 450                         break; }
 451                 case LDAP_SCOPE_SUBTREE:
 452                         fc->fscope = dnIsSuffix( fc->fdn, &fc->fss->s_base );
 453                         break;
 454                 case LDAP_SCOPE_SUBORDINATE:
 455                         fc->fscope = dnIsSuffix( fc->fdn, &fc->fss->s_base ) &&
 456                                 !dn_match( fc->fdn, &fc->fss->s_base );
 457                         break;
 458                 }
 459         }
 460
 461         if ( fc->fbase )
 462                 return LDAP_SUCCESS;
 463
 464         /* If entryID has changed, then the base of this search has
 465          * changed. Invalidate the psearch.
 466          */
 467         return LDAP_NO_SUCH_OBJECT;
 468 }
 469
 470 /* syncprov_findcsn:
 471  *   This function has three different purposes, but they all use a search
 472  * that filters on entryCSN so they're combined here.
 473  * 1: at startup time, after a contextCSN has been read from the database,
 474  * we search for all entries with CSN >= contextCSN in case the contextCSN
 475  * was not checkpointed at the previous shutdown.
 476  *
 477  * 2: when the current contextCSN is known and we have a sync cookie, we search
 478  * for one entry with CSN = the cookie CSN. If not found, try <= cookie CSN.
 479  * If an entry is found, the cookie CSN is valid, otherwise it is stale.
 480  *
 481  * 3: during a refresh phase, we search for all entries with CSN <= the cookie
 482  * CSN, and generate Present records for them. We always collect this result
 483  * in SyncID sets, even if there's only one match.
 484  */
 485 typedef enum find_csn_t {
 486         FIND_MAXCSN     = 1,
 487         FIND_CSN        = 2,
 488         FIND_PRESENT    = 3
 489 } find_csn_t;
 490
 491 static int
 492 findmax_cb( Operation *op, SlapReply *rs )
 493 {
 494         if ( rs->sr_type == REP_SEARCH && rs->sr_err == LDAP_SUCCESS ) {
 495                 struct berval *maxcsn = op->o_callback->sc_private;
 496                 Attribute *a = attr_find( rs->sr_entry->e_attrs,
 497                         slap_schema.si_ad_entryCSN );
 498
 499                 if ( a && ber_bvcmp( &a->a_vals[0], maxcsn ) > 0 ) {
 500                         maxcsn->bv_len = a->a_vals[0].bv_len;
 501                         strcpy( maxcsn->bv_val, a->a_vals[0].bv_val );
 502                 }
 503         }
 504         return LDAP_SUCCESS;
 505 }
 506
 507 static int
 508 findcsn_cb( Operation *op, SlapReply *rs )
 509 {
 510         slap_callback *sc = op->o_callback;
 511
 512         /* We just want to know that at least one exists, so it's OK if
 513          * we exceed the unchecked limit.
 514          */
 515         if ( rs->sr_err == LDAP_ADMINLIMIT_EXCEEDED ||
 516                 (rs->sr_type == REP_SEARCH && rs->sr_err == LDAP_SUCCESS )) {
 517                 sc->sc_private = (void *)1;
 518         }
 519         return LDAP_SUCCESS;
 520 }
 521
 522 /* Build a list of entryUUIDs for sending in a SyncID set */
 523
 524 #define UUID_LEN        16
 525
 526 typedef struct fpres_cookie {
 527         int num;
 528         BerVarray uuids;
 529         char *last;
 530 } fpres_cookie;
 531
 532 static int
 533 findpres_cb( Operation *op, SlapReply *rs )
 534 {
 535         slap_callback *sc = op->o_callback;
 536         fpres_cookie *pc = sc->sc_private;
 537         Attribute *a;
 538         int ret = SLAP_CB_CONTINUE;
 539
 540         switch ( rs->sr_type ) {
 541         case REP_SEARCH:
 542                 a = attr_find( rs->sr_entry->e_attrs, slap_schema.si_ad_entryUUID );
 543                 if ( a ) {
 544                         pc->uuids[pc->num].bv_val = pc->last;
 545                         AC_MEMCPY( pc->uuids[pc->num].bv_val, a->a_nvals[0].bv_val,
 546                                 pc->uuids[pc->num].bv_len );
 547                         pc->num++;
 548                         pc->last = pc->uuids[pc->num].bv_val;
 549                         pc->uuids[pc->num].bv_val = NULL;
 550                 }
 551                 ret = LDAP_SUCCESS;
 552                 if ( pc->num != SLAP_SYNCUUID_SET_SIZE )
 553                         break;
 554                 /* FALLTHRU */
 555         case REP_RESULT:
 556                 ret = rs->sr_err;
 557                 if ( pc->num ) {
 558                         ret = syncprov_sendinfo( op, rs, LDAP_TAG_SYNC_ID_SET, NULL,
 559                                 0, pc->uuids, 0 );
 560                         pc->uuids[pc->num].bv_val = pc->last;
 561                         pc->num = 0;
 562                         pc->last = pc->uuids[0].bv_val;
 563                 }
 564                 break;
 565         default:
 566                 break;
 567         }
 568         return ret;
 569 }
 570
 571 static int
 572 syncprov_findcsn( Operation *op, find_csn_t mode )
 573 {
 574         slap_overinst           *on = (slap_overinst *)op->o_bd->bd_info;
 575         syncprov_info_t         *si = on->on_bi.bi_private;
 576
 577         slap_callback cb = {0};
 578         Operation fop;
 579         SlapReply frs = { REP_RESULT };
 580         char buf[LDAP_LUTIL_CSNSTR_BUFSIZE + STRLENOF("(entryCSN<=)")];
 581         char cbuf[LDAP_LUTIL_CSNSTR_BUFSIZE];
 582         struct berval maxcsn;
 583         Filter cf, af;
 584 #ifdef LDAP_COMP_MATCH
 585         AttributeAssertion eq = { NULL, BER_BVNULL, NULL };
 586 #else
 587         AttributeAssertion eq = { NULL, BER_BVNULL };
 588 #endif
 589         fpres_cookie pcookie;
 590         sync_control *srs = NULL;
 591         struct slap_limits_set fc_limits;
 592         int i, rc = LDAP_SUCCESS, findcsn_retry = 1;
 593
 594         if ( mode != FIND_MAXCSN ) {
 595                 srs = op->o_controls[slap_cids.sc_LDAPsync];
 596
 597                 if ( srs->sr_state.ctxcsn.bv_len >= LDAP_LUTIL_CSNSTR_BUFSIZE ) {
 598                         return LDAP_OTHER;
 599                 }
 600         }
 601
 602         fop = *op;
 603         fop.o_sync_mode &= SLAP_CONTROL_MASK;   /* turn off sync_mode */
 604         /* We want pure entries, not referrals */
 605         fop.o_managedsait = SLAP_CONTROL_CRITICAL;
 606
 607         cf.f_ava = &eq;
 608         cf.f_av_desc = slap_schema.si_ad_entryCSN;
 609         cf.f_next = NULL;
 610
 611         fop.o_callback = &cb;
 612         fop.ors_limit = NULL;
 613         fop.ors_tlimit = SLAP_NO_LIMIT;
 614         fop.ors_filter = &cf;
 615         fop.ors_filterstr.bv_val = buf;
 616
 617 again:
 618         switch( mode ) {
 619         case FIND_MAXCSN:
 620                 cf.f_choice = LDAP_FILTER_GE;
 621                 cf.f_av_value = si->si_ctxcsn;
 622                 fop.ors_filterstr.bv_len = sprintf( buf, "(entryCSN>=%s)",
 623                         cf.f_av_value.bv_val );
 624                 fop.ors_attrsonly = 0;
 625                 fop.ors_attrs = csn_anlist;
 626                 fop.ors_slimit = SLAP_NO_LIMIT;
 627                 cb.sc_private = &maxcsn;
 628                 cb.sc_response = findmax_cb;
 629                 strcpy( cbuf, si->si_ctxcsn.bv_val );
 630                 maxcsn.bv_val = cbuf;
 631                 maxcsn.bv_len = si->si_ctxcsn.bv_len;
 632                 break;
 633         case FIND_CSN:
 634                 cf.f_av_value = srs->sr_state.ctxcsn;
 635                 /* Look for exact match the first time */
 636                 if ( findcsn_retry ) {
 637                         cf.f_choice = LDAP_FILTER_EQUALITY;
 638                         fop.ors_filterstr.bv_len = sprintf( buf, "(entryCSN=%s)",
 639                                 cf.f_av_value.bv_val );
 640                 /* On retry, look for <= */
 641                 } else {
 642                         cf.f_choice = LDAP_FILTER_LE;
 643                         fop.ors_limit = &fc_limits;
 644                         fc_limits.lms_s_unchecked = 1;
 645                         fop.ors_filterstr.bv_len = sprintf( buf, "(entryCSN<=%s)",
 646                                 cf.f_av_value.bv_val );
 647                 }
 648                 fop.ors_attrsonly = 1;
 649                 fop.ors_attrs = slap_anlist_no_attrs;
 650                 fop.ors_slimit = 1;
 651                 cb.sc_private = NULL;
 652                 cb.sc_response = findcsn_cb;
 653                 break;
 654         case FIND_PRESENT:
 655                 af.f_choice = LDAP_FILTER_AND;
 656                 af.f_next = NULL;
 657                 af.f_and = &cf;
 658                 cf.f_choice = LDAP_FILTER_LE;
 659                 cf.f_av_value = srs->sr_state.ctxcsn;
 660                 cf.f_next = op->ors_filter;
 661                 fop.ors_filter = &af;
 662                 filter2bv_x( &fop, fop.ors_filter, &fop.ors_filterstr );
 663                 fop.ors_attrsonly = 0;
 664                 fop.ors_attrs = uuid_anlist;
 665                 fop.ors_slimit = SLAP_NO_LIMIT;
 666                 cb.sc_private = &pcookie;
 667                 cb.sc_response = findpres_cb;
 668                 pcookie.num = 0;
 669
 670                 /* preallocate storage for a full set */
 671                 pcookie.uuids = op->o_tmpalloc( (SLAP_SYNCUUID_SET_SIZE+1) *
 672                         sizeof(struct berval) + SLAP_SYNCUUID_SET_SIZE * UUID_LEN,
 673                         op->o_tmpmemctx );
 674                 pcookie.last = (char *)(pcookie.uuids + SLAP_SYNCUUID_SET_SIZE+1);
 675                 pcookie.uuids[0].bv_val = pcookie.last;
 676                 pcookie.uuids[0].bv_len = UUID_LEN;
 677                 for (i=1; i<SLAP_SYNCUUID_SET_SIZE; i++) {
 678                         pcookie.uuids[i].bv_val = pcookie.uuids[i-1].bv_val + UUID_LEN;
 679                         pcookie.uuids[i].bv_len = UUID_LEN;
 680                 }
 681                 break;
 682         }
 683
 684         fop.o_bd->bd_info = on->on_info->oi_orig;
 685         fop.o_bd->be_search( &fop, &frs );
 686         fop.o_bd->bd_info = (BackendInfo *)on;
 687
 688         switch( mode ) {
 689         case FIND_MAXCSN:
 690                 strcpy( si->si_ctxcsnbuf, maxcsn.bv_val );
 691                 si->si_ctxcsn.bv_len = maxcsn.bv_len;
 692                 break;
 693         case FIND_CSN:
 694                 /* If matching CSN was not found, invalidate the context. */
 695                 if ( !cb.sc_private ) {
 696                         /* If we didn't find an exact match, then try for <= */
 697                         if ( findcsn_retry ) {
 698                                 findcsn_retry = 0;
 699                                 goto again;
 700                         }
 701                         rc = LDAP_NO_SUCH_OBJECT;
 702                 }
 703                 break;
 704         case FIND_PRESENT:
 705                 op->o_tmpfree( pcookie.uuids, op->o_tmpmemctx );
 706                 op->o_tmpfree( fop.ors_filterstr.bv_val, op->o_tmpmemctx );
 707                 break;
 708         }
 709
 710         return rc;
 711 }
 712
 713 static void
 714 syncprov_free_syncop( syncops *so )
 715 {
 716         syncres *sr, *srnext;
 717         GroupAssertion *ga, *gnext;
 718
 719         ldap_pvt_thread_mutex_lock( &so->s_mutex );
 720         if ( --so->s_inuse > 0 ) {
 721                 ldap_pvt_thread_mutex_unlock( &so->s_mutex );
 722                 return;
 723         }
 724         ldap_pvt_thread_mutex_unlock( &so->s_mutex );
 725         if ( so->s_flags & PS_IS_DETACHED ) {
 726                 filter_free( so->s_op->ors_filter );
 727                 for ( ga = so->s_op->o_groups; ga; ga=gnext ) {
 728                         gnext = ga->ga_next;
 729                         ch_free( ga );
 730                 }
 731                 ch_free( so->s_op );
 732         }
 733         ch_free( so->s_base.bv_val );
 734         for ( sr=so->s_res; sr; sr=srnext ) {
 735                 srnext = sr->s_next;
 736                 ch_free( sr );
 737         }
 738         ldap_pvt_thread_mutex_destroy( &so->s_mutex );
 739         ch_free( so );
 740 }
 741
 742 /* Send a persistent search response */
 743 static int
 744 syncprov_sendresp( Operation *op, opcookie *opc, syncops *so,
 745         Entry **e, int mode )
 746 {
 747         slap_overinst *on = opc->son;
 748
 749         SlapReply rs = { REP_SEARCH };
 750         LDAPControl *ctrls[2];
 751         struct berval cookie;
 752         Entry e_uuid = {0};
 753         Attribute a_uuid = {0};
 754
 755         if ( so->s_op->o_abandon )
 756                 return SLAPD_ABANDON;
 757
 758         ctrls[1] = NULL;
 759         slap_compose_sync_cookie( op, &cookie, &opc->sctxcsn, so->s_rid );
 760
 761         e_uuid.e_attrs = &a_uuid;
 762         a_uuid.a_desc = slap_schema.si_ad_entryUUID;
 763         a_uuid.a_nvals = &opc->suuid;
 764         rs.sr_err = syncprov_state_ctrl( op, &rs, &e_uuid,
 765                 mode, ctrls, 0, 1, &cookie );
 766         op->o_tmpfree( cookie.bv_val, op->o_tmpmemctx );
 767
 768         rs.sr_ctrls = ctrls;
 769         op->o_bd->bd_info = (BackendInfo *)on->on_info;
 770         switch( mode ) {
 771         case LDAP_SYNC_ADD:
 772                 rs.sr_entry = *e;
 773                 if ( rs.sr_entry->e_private )
 774                         rs.sr_flags = REP_ENTRY_MUSTRELEASE;
 775                 if ( opc->sreference ) {
 776                         rs.sr_ref = get_entry_referrals( op, rs.sr_entry );
 777                         rs.sr_err = send_search_reference( op, &rs );
 778                         ber_bvarray_free( rs.sr_ref );
 779                         if ( !rs.sr_entry )
 780                                 *e = NULL;
 781                         break;
 782                 }
 783                 /* fallthru */
 784         case LDAP_SYNC_MODIFY:
 785                 rs.sr_entry = *e;
 786                 if ( rs.sr_entry->e_private )
 787                         rs.sr_flags = REP_ENTRY_MUSTRELEASE;
 788                 rs.sr_attrs = op->ors_attrs;
 789                 rs.sr_err = send_search_entry( op, &rs );
 790                 if ( !rs.sr_entry )
 791                         *e = NULL;
 792                 break;
 793         case LDAP_SYNC_DELETE:
 794                 e_uuid.e_attrs = NULL;
 795                 e_uuid.e_name = opc->sdn;
 796                 e_uuid.e_nname = opc->sndn;
 797                 rs.sr_entry = &e_uuid;
 798                 if ( opc->sreference ) {
 799                         struct berval bv = BER_BVNULL;
 800                         rs.sr_ref = &bv;
 801                         rs.sr_err = send_search_reference( op, &rs );
 802                 } else {
 803                         rs.sr_err = send_search_entry( op, &rs );
 804                 }
 805                 break;
 806         default:
 807                 assert(0);
 808         }
 809         /* In case someone else freed it already? */
 810         if ( rs.sr_ctrls ) {
 811                 op->o_tmpfree( rs.sr_ctrls[0], op->o_tmpmemctx );
 812                 rs.sr_ctrls = NULL;
 813         }
 814
 815         return rs.sr_err;
 816 }
 817
 818 /* Play back queued responses */
 819 static int
 820 syncprov_qplay( Operation *op, slap_overinst *on, syncops *so )
 821 {
 822         syncres *sr;
 823         Entry *e;
 824         opcookie opc;
 825         int rc = 0;
 826
 827         opc.son = on;
 828         op->o_bd->bd_info = (BackendInfo *)on->on_info;
 829
 830         for (;;) {
 831                 ldap_pvt_thread_mutex_lock( &so->s_mutex );
 832                 sr = so->s_res;
 833                 if ( sr )
 834                         so->s_res = sr->s_next;
 835                 if ( !so->s_res )
 836                         so->s_restail = NULL;
 837                 ldap_pvt_thread_mutex_unlock( &so->s_mutex );
 838
 839                 if ( !sr || so->s_op->o_abandon )
 840                         break;
 841
 842                 opc.sdn = sr->s_dn;
 843                 opc.sndn = sr->s_ndn;
 844                 opc.suuid = sr->s_uuid;
 845                 opc.sctxcsn = sr->s_csn;
 846                 opc.sreference = sr->s_isreference;
 847                 e = NULL;
 848
 849                 if ( sr->s_mode != LDAP_SYNC_DELETE ) {
 850                         rc = be_entry_get_rw( op, &opc.sndn, NULL, NULL, 0, &e );
 851                         if ( rc ) {
 852                                 ch_free( sr );
 853                                 continue;
 854                         }
 855                 }
 856                 rc = syncprov_sendresp( op, &opc, so, &e, sr->s_mode );
 857
 858                 if ( e ) {
 859                         be_entry_release_rw( op, e, 0 );
 860                 }
 861
 862                 ch_free( sr );
 863
 864                 if ( rc )
 865                         break;
 866         }
 867         op->o_bd->bd_info = (BackendInfo *)on;
 868         return rc;
 869 }
 870
 871 /* runqueue task for playing back queued responses */
 872 static void *
 873 syncprov_qtask( void *ctx, void *arg )
 874 {
 875         struct re_s *rtask = arg;
 876         syncops *so = rtask->arg;
 877         slap_overinst *on = so->s_op->o_private;
 878         OperationBuffer opbuf;
 879         Operation *op;
 880         BackendDB be;
 881         int rc;
 882
 883         op = (Operation *) &opbuf;
 884         *op = *so->s_op;
 885         op->o_hdr = (Opheader *)(op+1);
 886         op->o_controls = (void **)(op->o_hdr+1);
 887         memset( op->o_controls, 0, SLAP_MAX_CIDS * sizeof(void *));
 888
 889         *op->o_hdr = *so->s_op->o_hdr;
 890
 891         op->o_tmpmemctx = slap_sl_mem_create(SLAP_SLAB_SIZE, SLAP_SLAB_STACK, ctx);
 892         op->o_tmpmfuncs = &slap_sl_mfuncs;
 893         op->o_threadctx = ctx;
 894
 895         /* syncprov_qplay expects a fake db */
 896         be = *so->s_op->o_bd;
 897         be.be_flags |= SLAP_DBFLAG_OVERLAY;
 898         op->o_bd = &be;
 899         op->o_private = NULL;
 900         op->o_callback = NULL;
 901
 902         rc = syncprov_qplay( op, on, so );
 903
 904         /* decrement use count... */
 905         syncprov_free_syncop( so );
 906
 907         /* wait until we get explicitly scheduled again */
 908         ldap_pvt_thread_mutex_lock( &slapd_rq.rq_mutex );
 909         ldap_pvt_runqueue_stoptask( &slapd_rq, rtask );
 910         if ( rc == 0 ) {
 911                 ldap_pvt_runqueue_resched( &slapd_rq, rtask, 1 );
 912         } else {
 913                 /* bail out on any error */
 914                 ldap_pvt_runqueue_remove( &slapd_rq, rtask );
 915         }
 916         ldap_pvt_thread_mutex_unlock( &slapd_rq.rq_mutex );
 917
 918 #if 0   /* FIXME: connection_close isn't exported from slapd.
 919                  * should it be?
 920                  */
 921         if ( rc ) {
 922                 ldap_pvt_thread_mutex_lock( &op->o_conn->c_mutex );
 923                 if ( connection_state_closing( op->o_conn )) {
 924                         connection_close( op->o_conn );
 925                 }
 926                 ldap_pvt_thread_mutex_unlock( &op->o_conn->c_mutex );
 927         }
 928 #endif
 929         return NULL;
 930 }
 931
 932 /* Start the task to play back queued psearch responses */
 933 static void
 934 syncprov_qstart( syncops *so )
 935 {
 936         int wake=0;
 937         ldap_pvt_thread_mutex_lock( &slapd_rq.rq_mutex );
 938         if ( !so->s_qtask ) {
 939                 so->s_qtask = ldap_pvt_runqueue_insert( &slapd_rq, RUNQ_INTERVAL,
 940                         syncprov_qtask, so, "syncprov_qtask",
 941                         so->s_op->o_conn->c_peer_name.bv_val );
 942                 ++so->s_inuse;
 943                 wake = 1;
 944         } else {
 945                 if (!ldap_pvt_runqueue_isrunning( &slapd_rq, so->s_qtask ) &&
 946                         !so->s_qtask->next_sched.tv_sec ) {
 947                         so->s_qtask->interval.tv_sec = 0;
 948                         ldap_pvt_runqueue_resched( &slapd_rq, so->s_qtask, 0 );
 949                         so->s_qtask->interval.tv_sec = RUNQ_INTERVAL;
 950                         ++so->s_inuse;
 951                         wake = 1;
 952                 }
 953         }
 954         ldap_pvt_thread_mutex_unlock( &slapd_rq.rq_mutex );
 955         if ( wake )
 956                 slap_wake_listener();
 957 }
 958
 959 /* Queue a persistent search response */
 960 static int
 961 syncprov_qresp( opcookie *opc, syncops *so, int mode )
 962 {
 963         syncres *sr;
 964
 965         sr = ch_malloc(sizeof(syncres) + opc->suuid.bv_len + 1 +
 966                 opc->sdn.bv_len + 1 + opc->sndn.bv_len + 1 + opc->sctxcsn.bv_len + 1 );
 967         sr->s_next = NULL;
 968         sr->s_dn.bv_val = (char *)(sr + 1);
 969         sr->s_dn.bv_len = opc->sdn.bv_len;
 970         sr->s_mode = mode;
 971         sr->s_isreference = opc->sreference;
 972         sr->s_ndn.bv_val = lutil_strcopy( sr->s_dn.bv_val,
 973                  opc->sdn.bv_val ) + 1;
 974         sr->s_ndn.bv_len = opc->sndn.bv_len;
 975         sr->s_uuid.bv_val = lutil_strcopy( sr->s_ndn.bv_val,
 976                  opc->sndn.bv_val ) + 1;
 977         sr->s_uuid.bv_len = opc->suuid.bv_len;
 978         AC_MEMCPY( sr->s_uuid.bv_val, opc->suuid.bv_val, opc->suuid.bv_len );
 979         sr->s_csn.bv_val = sr->s_uuid.bv_val + sr->s_uuid.bv_len + 1;
 980         sr->s_csn.bv_len = opc->sctxcsn.bv_len;
 981         strcpy( sr->s_csn.bv_val, opc->sctxcsn.bv_val );
 982
 983         ldap_pvt_thread_mutex_lock( &so->s_mutex );
 984         if ( !so->s_res ) {
 985                 so->s_res = sr;
 986         } else {
 987                 so->s_restail->s_next = sr;
 988         }
 989         so->s_restail = sr;
 990
 991         /* If the base of the psearch was modified, check it next time round */
 992         if ( so->s_flags & PS_WROTE_BASE ) {
 993                 so->s_flags ^= PS_WROTE_BASE;
 994                 so->s_flags |= PS_FIND_BASE;
 995         }
 996         if ( so->s_flags & PS_IS_DETACHED ) {
 997                 syncprov_qstart( so );
 998         }
 999         ldap_pvt_thread_mutex_unlock( &so->s_mutex );
1000         return LDAP_SUCCESS;
1001 }
1002
1003 static int
1004 syncprov_drop_psearch( syncops *so, int lock )
1005 {
1006         if ( so->s_flags & PS_IS_DETACHED ) {
1007                 if ( lock )
1008                         ldap_pvt_thread_mutex_lock( &so->s_op->o_conn->c_mutex );
1009                 so->s_op->o_conn->c_n_ops_executing--;
1010                 so->s_op->o_conn->c_n_ops_completed++;
1011                 LDAP_STAILQ_REMOVE( &so->s_op->o_conn->c_ops, so->s_op, slap_op,
1012                         o_next );
1013                 if ( lock )
1014                         ldap_pvt_thread_mutex_unlock( &so->s_op->o_conn->c_mutex );
1015         }
1016         syncprov_free_syncop( so );
1017
1018         return 0;
1019 }
1020
1021 static int
1022 syncprov_ab_cleanup( Operation *op, SlapReply *rs )
1023 {
1024         slap_callback *sc = op->o_callback;
1025         op->o_callback = sc->sc_next;
1026         syncprov_drop_psearch( op->o_callback->sc_private, 0 );
1027         op->o_tmpfree( sc, op->o_tmpmemctx );
1028         return 0;
1029 }
1030
1031 static int
1032 syncprov_op_abandon( Operation *op, SlapReply *rs )
1033 {
1034         slap_overinst           *on = (slap_overinst *)op->o_bd->bd_info;
1035         syncprov_info_t         *si = on->on_bi.bi_private;
1036         syncops *so, *soprev;
1037
1038         ldap_pvt_thread_mutex_lock( &si->si_ops_mutex );
1039         for ( so=si->si_ops, soprev = (syncops *)&si->si_ops; so;
1040                 soprev=so, so=so->s_next ) {
1041                 if ( so->s_op->o_connid == op->o_connid &&
1042                         so->s_op->o_msgid == op->orn_msgid ) {
1043                                 so->s_op->o_abandon = 1;
1044                                 soprev->s_next = so->s_next;
1045                                 break;
1046                 }
1047         }
1048         ldap_pvt_thread_mutex_unlock( &si->si_ops_mutex );
1049         if ( so ) {
1050                 /* Is this really a Cancel exop? */
1051                 if ( op->o_tag != LDAP_REQ_ABANDON ) {
1052                         so->s_op->o_cancel = SLAP_CANCEL_ACK;
1053                         rs->sr_err = LDAP_CANCELLED;
1054                         send_ldap_result( so->s_op, rs );
1055                         if ( so->s_flags & PS_IS_DETACHED ) {
1056                                 slap_callback *cb;
1057                                 cb = op->o_tmpcalloc( 1, sizeof(slap_callback), op->o_tmpmemctx );
1058                                 cb->sc_cleanup = syncprov_ab_cleanup;
1059                                 cb->sc_next = op->o_callback;
1060                                 cb->sc_private = so;
1061                                 return SLAP_CB_CONTINUE;
1062                         }
1063                 }
1064                 syncprov_drop_psearch( so, 0 );
1065         }
1066         return SLAP_CB_CONTINUE;
1067 }
1068
1069 /* Find which persistent searches are affected by this operation */
1070 static void
1071 syncprov_matchops( Operation *op, opcookie *opc, int saveit )
1072 {
1073         slap_overinst *on = opc->son;
1074         syncprov_info_t         *si = on->on_bi.bi_private;
1075
1076         fbase_cookie fc;
1077         syncops *ss, *sprev, *snext;
1078         Entry *e;
1079         Attribute *a;
1080         int rc;
1081         struct berval newdn;
1082         int freefdn = 0;
1083         BackendDB *b0 = op->o_bd, db;
1084
1085         fc.fdn = &op->o_req_ndn;
1086         /* compute new DN */
1087         if ( op->o_tag == LDAP_REQ_MODRDN && !saveit ) {
1088                 struct berval pdn;
1089                 if ( op->orr_nnewSup ) pdn = *op->orr_nnewSup;
1090                 else dnParent( fc.fdn, &pdn );
1091                 build_new_dn( &newdn, &pdn, &op->orr_nnewrdn, op->o_tmpmemctx );
1092                 fc.fdn = &newdn;
1093                 freefdn = 1;
1094         }
1095         if ( op->o_tag != LDAP_REQ_ADD ) {
1096                 if ( !SLAP_ISOVERLAY( op->o_bd )) {
1097                         db = *op->o_bd;
1098                         op->o_bd = &db;
1099                 }
1100                 op->o_bd->bd_info = (BackendInfo *)on->on_info;
1101                 rc = be_entry_get_rw( op, fc.fdn, NULL, NULL, 0, &e );
1102                 /* If we're sending responses now, make a copy and unlock the DB */
1103                 if ( e && !saveit ) {
1104                         Entry *e2 = entry_dup( e );
1105                         be_entry_release_rw( op, e, 0 );
1106                         e = e2;
1107                 }
1108                 op->o_bd->bd_info = (BackendInfo *)on;
1109                 if ( rc ) {
1110                         op->o_bd = b0;
1111                         return;
1112                 }
1113         } else {
1114                 e = op->ora_e;
1115         }
1116
1117         if ( saveit || op->o_tag == LDAP_REQ_ADD ) {
1118                 ber_dupbv_x( &opc->sdn, &e->e_name, op->o_tmpmemctx );
1119                 ber_dupbv_x( &opc->sndn, &e->e_nname, op->o_tmpmemctx );
1120                 opc->sreference = is_entry_referral( e );
1121                 a = attr_find( e->e_attrs, slap_schema.si_ad_entryUUID );
1122                 if ( a )
1123                         ber_dupbv_x( &opc->suuid, &a->a_nvals[0], op->o_tmpmemctx );
1124         } else if ( op->o_tag == LDAP_REQ_MODRDN && !saveit ) {
1125                 op->o_tmpfree( opc->sndn.bv_val, op->o_tmpmemctx );
1126                 op->o_tmpfree( opc->sdn.bv_val, op->o_tmpmemctx );
1127                 ber_dupbv_x( &opc->sdn, &e->e_name, op->o_tmpmemctx );
1128                 ber_dupbv_x( &opc->sndn, &e->e_nname, op->o_tmpmemctx );
1129         }
1130
1131         ldap_pvt_thread_mutex_lock( &si->si_ops_mutex );
1132         for (ss = si->si_ops, sprev = (syncops *)&si->si_ops; ss;
1133                 sprev = ss, ss=snext)
1134         {
1135                 syncmatches *sm;
1136                 int found = 0;
1137
1138                 snext = ss->s_next;
1139                 /* validate base */
1140                 fc.fss = ss;
1141                 fc.fbase = 0;
1142                 fc.fscope = 0;
1143
1144                 /* If the base of the search is missing, signal a refresh */
1145                 rc = syncprov_findbase( op, &fc );
1146                 if ( rc != LDAP_SUCCESS ) {
1147                         SlapReply rs = {REP_RESULT};
1148                         send_ldap_error( ss->s_op, &rs, LDAP_SYNC_REFRESH_REQUIRED,
1149                                 "search base has changed" );
1150                         sprev->s_next = snext;
1151                         syncprov_drop_psearch( ss, 1 );
1152                         ss = sprev;
1153                         continue;
1154                 }
1155
1156
1157                 /* If we're sending results now, look for this op in old matches */
1158                 if ( !saveit ) {
1159                         syncmatches *old;
1160
1161                         /* Did we modify the search base? */
1162                         if ( dn_match( &op->o_req_ndn, &ss->s_base )) {
1163                                 ldap_pvt_thread_mutex_lock( &ss->s_mutex );
1164                                 ss->s_flags |= PS_WROTE_BASE;
1165                                 ldap_pvt_thread_mutex_unlock( &ss->s_mutex );
1166                         }
1167
1168                         for ( sm=opc->smatches, old=(syncmatches *)&opc->smatches; sm;
1169                                 old=sm, sm=sm->sm_next ) {
1170                                 if ( sm->sm_op == ss ) {
1171                                         found = 1;
1172                                         old->sm_next = sm->sm_next;
1173                                         op->o_tmpfree( sm, op->o_tmpmemctx );
1174                                         break;
1175                                 }
1176                         }
1177                 }
1178
1179                 /* check if current o_req_dn is in scope and matches filter */
1180                 if ( fc.fscope && test_filter( op, e, ss->s_op->ors_filter ) ==
1181                         LDAP_COMPARE_TRUE ) {
1182                         if ( saveit ) {
1183                                 sm = op->o_tmpalloc( sizeof(syncmatches), op->o_tmpmemctx );
1184                                 sm->sm_next = opc->smatches;
1185                                 sm->sm_op = ss;
1186                                 ldap_pvt_thread_mutex_lock( &ss->s_mutex );
1187                                 ++ss->s_inuse;
1188                                 ldap_pvt_thread_mutex_unlock( &ss->s_mutex );
1189                                 opc->smatches = sm;
1190                         } else {
1191                                 /* if found send UPDATE else send ADD */
1192                                 syncprov_qresp( opc, ss,
1193                                         found ? LDAP_SYNC_MODIFY : LDAP_SYNC_ADD );
1194                         }
1195                 } else if ( !saveit && found ) {
1196                         /* send DELETE */
1197                         syncprov_qresp( opc, ss, LDAP_SYNC_DELETE );
1198                 }
1199         }
1200         ldap_pvt_thread_mutex_unlock( &si->si_ops_mutex );
1201
1202         if ( op->o_tag != LDAP_REQ_ADD && e ) {
1203                 op->o_bd->bd_info = (BackendInfo *)on->on_info;
1204                 be_entry_release_rw( op, e, 0 );
1205                 op->o_bd->bd_info = (BackendInfo *)on;
1206         }
1207         if ( freefdn ) {
1208                 op->o_tmpfree( fc.fdn->bv_val, op->o_tmpmemctx );
1209         }
1210         op->o_bd = b0;
1211 }
1212
1213 static int
1214 syncprov_op_cleanup( Operation *op, SlapReply *rs )
1215 {
1216         slap_callback *cb = op->o_callback;
1217         opcookie *opc = cb->sc_private;
1218         slap_overinst *on = opc->son;
1219         syncprov_info_t         *si = on->on_bi.bi_private;
1220         syncmatches *sm, *snext;
1221         modtarget *mt, mtdummy;
1222
1223         for (sm = opc->smatches; sm; sm=snext) {
1224                 snext = sm->sm_next;
1225                 syncprov_free_syncop( sm->sm_op );
1226                 op->o_tmpfree( sm, op->o_tmpmemctx );
1227         }
1228
1229         /* Remove op from lock table */
1230         mtdummy.mt_op = op;
1231         ldap_pvt_thread_mutex_lock( &si->si_mods_mutex );
1232         mt = avl_find( si->si_mods, &mtdummy, sp_avl_cmp );
1233         if ( mt ) {
1234                 modinst *mi = mt->mt_mods;
1235
1236                 /* If there are more, promote the next one */
1237                 ldap_pvt_thread_mutex_lock( &mt->mt_mutex );
1238                 if ( mi->mi_next ) {
1239                         mt->mt_mods = mi->mi_next;
1240                         mt->mt_op = mt->mt_mods->mi_op;
1241                         ldap_pvt_thread_mutex_unlock( &mt->mt_mutex );
1242                 } else {
1243                         avl_delete( &si->si_mods, mt, sp_avl_cmp );
1244                         ldap_pvt_thread_mutex_unlock( &mt->mt_mutex );
1245                         ldap_pvt_thread_mutex_destroy( &mt->mt_mutex );
1246                         ch_free( mt );
1247                 }
1248         }
1249         ldap_pvt_thread_mutex_unlock( &si->si_mods_mutex );
1250         if ( !BER_BVISNULL( &opc->suuid ))
1251                 op->o_tmpfree( opc->suuid.bv_val, op->o_tmpmemctx );
1252         if ( !BER_BVISNULL( &opc->sndn ))
1253                 op->o_tmpfree( opc->sndn.bv_val, op->o_tmpmemctx );
1254         if ( !BER_BVISNULL( &opc->sdn ))
1255                 op->o_tmpfree( opc->sdn.bv_val, op->o_tmpmemctx );
1256         op->o_callback = cb->sc_next;
1257         op->o_tmpfree(cb, op->o_tmpmemctx);
1258
1259         return 0;
1260 }
1261
1262 static void
1263 syncprov_checkpoint( Operation *op, SlapReply *rs, slap_overinst *on )
1264 {
1265         syncprov_info_t         *si = on->on_bi.bi_private;
1266         Modifications mod;
1267         Operation opm;
1268         SlapReply rsm = { 0 };
1269         struct berval bv[2];
1270         slap_callback cb = {0};
1271
1272         /* If ctxcsn is empty, delete it */
1273         if ( BER_BVISEMPTY( &si->si_ctxcsn )) {
1274                 mod.sml_values = NULL;
1275         } else {
1276                 mod.sml_values = bv;
1277                 bv[1].bv_val = NULL;
1278                 bv[0] = si->si_ctxcsn;
1279         }
1280         mod.sml_nvalues = NULL;
1281         mod.sml_desc = slap_schema.si_ad_contextCSN;
1282         mod.sml_op = LDAP_MOD_REPLACE;
1283         mod.sml_flags = 0;
1284         mod.sml_next = NULL;
1285
1286         cb.sc_response = slap_null_cb;
1287         opm = *op;
1288         opm.o_tag = LDAP_REQ_MODIFY;
1289         opm.o_callback = &cb;
1290         opm.orm_modlist = &mod;
1291         opm.o_req_dn = op->o_bd->be_suffix[0];
1292         opm.o_req_ndn = op->o_bd->be_nsuffix[0];
1293         opm.o_bd->bd_info = on->on_info->oi_orig;
1294         opm.o_managedsait = SLAP_CONTROL_NONCRITICAL;
1295         SLAP_DBFLAGS( opm.o_bd ) |= SLAP_DBFLAG_NOLASTMOD;
1296         opm.o_bd->be_modify( &opm, &rsm );
1297         SLAP_DBFLAGS( opm.o_bd ) ^= SLAP_DBFLAG_NOLASTMOD;
1298         if ( mod.sml_next != NULL ) {
1299                 slap_mods_free( mod.sml_next, 1 );
1300         }
1301 }
1302
1303 static void
1304 syncprov_add_slog( Operation *op )
1305 {
1306         opcookie *opc = op->o_callback->sc_private;
1307         slap_overinst *on = opc->son;
1308         syncprov_info_t         *si = on->on_bi.bi_private;
1309         sessionlog *sl;
1310         slog_entry *se;
1311
1312         sl = si->si_logs;
1313         {
1314                 /* Allocate a record. UUIDs are not NUL-terminated. */
1315                 se = ch_malloc( sizeof( slog_entry ) + opc->suuid.bv_len +
1316                         op->o_csn.bv_len + 1 );
1317                 se->se_next = NULL;
1318                 se->se_tag = op->o_tag;
1319
1320                 se->se_uuid.bv_val = (char *)(&se[1]);
1321                 AC_MEMCPY( se->se_uuid.bv_val, opc->suuid.bv_val, opc->suuid.bv_len );
1322                 se->se_uuid.bv_len = opc->suuid.bv_len;
1323
1324                 se->se_csn.bv_val = se->se_uuid.bv_val + opc->suuid.bv_len;
1325                 AC_MEMCPY( se->se_csn.bv_val, op->o_csn.bv_val, op->o_csn.bv_len );
1326                 se->se_csn.bv_val[op->o_csn.bv_len] = '\0';
1327                 se->se_csn.bv_len = op->o_csn.bv_len;
1328
1329                 ldap_pvt_thread_mutex_lock( &sl->sl_mutex );
1330                 if ( sl->sl_head ) {
1331                         sl->sl_tail->se_next = se;
1332                 } else {
1333                         sl->sl_head = se;
1334                 }
1335                 sl->sl_tail = se;
1336                 sl->sl_num++;
1337                 while ( sl->sl_num > sl->sl_size ) {
1338                         se = sl->sl_head;
1339                         sl->sl_head = se->se_next;
1340                         strcpy( sl->sl_mincsn.bv_val, se->se_csn.bv_val );
1341                         sl->sl_mincsn.bv_len = se->se_csn.bv_len;
1342                         ch_free( se );
1343                         sl->sl_num--;
1344                 }
1345                 ldap_pvt_thread_mutex_unlock( &sl->sl_mutex );
1346         }
1347 }
1348
1349 /* Just set a flag if we found the matching entry */
1350 static int
1351 playlog_cb( Operation *op, SlapReply *rs )
1352 {
1353         if ( rs->sr_type == REP_SEARCH ) {
1354                 op->o_callback->sc_private = (void *)1;
1355         }
1356         return rs->sr_err;
1357 }
1358
1359 /* enter with sl->sl_mutex locked, release before returning */
1360 static void
1361 syncprov_playlog( Operation *op, SlapReply *rs, sessionlog *sl,
1362         sync_control *srs, struct berval *ctxcsn )
1363 {
1364         slap_overinst           *on = (slap_overinst *)op->o_bd->bd_info;
1365         slog_entry *se;
1366         int i, j, ndel, num, nmods, mmods;
1367         char cbuf[LDAP_LUTIL_CSNSTR_BUFSIZE];
1368         BerVarray uuids;
1369         struct berval delcsn;
1370
1371         if ( !sl->sl_num ) {
1372                 ldap_pvt_thread_mutex_unlock( &sl->sl_mutex );
1373                 return;
1374         }
1375
1376         num = sl->sl_num;
1377         i = 0;
1378         nmods = 0;
1379
1380         uuids = op->o_tmpalloc( (num+1) * sizeof( struct berval ) +
1381                 num * UUID_LEN, op->o_tmpmemctx );
1382         uuids[0].bv_val = (char *)(uuids + num + 1);
1383
1384         delcsn.bv_len = 0;
1385         delcsn.bv_val = cbuf;
1386
1387         /* Make a copy of the relevant UUIDs. Put the Deletes up front
1388          * and everything else at the end. Do this first so we can
1389          * unlock the list mutex.
1390          */
1391         for ( se=sl->sl_head; se; se=se->se_next ) {
1392                 if ( ber_bvcmp( &se->se_csn, &srs->sr_state.ctxcsn ) <= 0 ) continue;
1393                 if ( ber_bvcmp( &se->se_csn, ctxcsn ) > 0 ) break;
1394                 if ( se->se_tag == LDAP_REQ_DELETE ) {
1395                         j = i;
1396                         i++;
1397                         AC_MEMCPY( cbuf, se->se_csn.bv_val, se->se_csn.bv_len );
1398                         delcsn.bv_len = se->se_csn.bv_len;
1399                         delcsn.bv_val[delcsn.bv_len] = '\0';
1400                 } else {
1401                         nmods++;
1402                         j = num - nmods;
1403                 }
1404                 uuids[j].bv_val = uuids[0].bv_val + (j * UUID_LEN);
1405                 AC_MEMCPY(uuids[j].bv_val, se->se_uuid.bv_val, UUID_LEN);
1406                 uuids[j].bv_len = UUID_LEN;
1407         }
1408         ldap_pvt_thread_mutex_unlock( &sl->sl_mutex );
1409
1410         ndel = i;
1411
1412         /* Zero out unused slots */
1413         for ( i=ndel; i < num - nmods; i++ )
1414                 uuids[i].bv_len = 0;
1415
1416         /* Mods must be validated to see if they belong in this delete set.
1417          */
1418
1419         mmods = nmods;
1420         /* Strip any duplicates */
1421         for ( i=0; i<nmods; i++ ) {
1422                 for ( j=0; j<ndel; j++ ) {
1423                         if ( bvmatch( &uuids[j], &uuids[num - 1 - i] )) {
1424                                 uuids[num - 1 - i].bv_len = 0;
1425                                 mmods --;
1426                                 break;
1427                         }
1428                 }
1429                 if ( uuids[num - 1 - i].bv_len == 0 ) continue;
1430                 for ( j=0; j<i; j++ ) {
1431                         if ( bvmatch( &uuids[num - 1 - j], &uuids[num - 1 - i] )) {
1432                                 uuids[num - 1 - i].bv_len = 0;
1433                                 mmods --;
1434                                 break;
1435                         }
1436                 }
1437         }
1438
1439         if ( mmods ) {
1440                 Operation fop;
1441                 SlapReply frs = { REP_RESULT };
1442                 int rc;
1443                 Filter mf, af;
1444 #ifdef LDAP_COMP_MATCH
1445                 AttributeAssertion eq = { NULL, BER_BVNULL, NULL };
1446 #else
1447                 AttributeAssertion eq;
1448 #endif
1449                 slap_callback cb = {0};
1450
1451                 fop = *op;
1452
1453                 fop.o_sync_mode = 0;
1454                 fop.o_callback = &cb;
1455                 fop.ors_limit = NULL;
1456                 fop.ors_tlimit = SLAP_NO_LIMIT;
1457                 fop.ors_attrs = slap_anlist_all_attributes;
1458                 fop.ors_attrsonly = 0;
1459                 fop.o_managedsait = SLAP_CONTROL_CRITICAL;
1460
1461                 af.f_choice = LDAP_FILTER_AND;
1462                 af.f_next = NULL;
1463                 af.f_and = &mf;
1464                 mf.f_choice = LDAP_FILTER_EQUALITY;
1465                 mf.f_ava = &eq;
1466                 mf.f_av_desc = slap_schema.si_ad_entryUUID;
1467                 mf.f_next = fop.ors_filter;
1468
1469                 fop.ors_filter = &af;
1470
1471                 cb.sc_response = playlog_cb;
1472                 fop.o_bd->bd_info = on->on_info->oi_orig;
1473
1474                 for ( i=ndel; i<num; i++ ) {
1475                         if ( uuids[i].bv_len == 0 ) continue;
1476
1477                         mf.f_av_value = uuids[i];
1478                         cb.sc_private = NULL;
1479                         fop.ors_slimit = 1;
1480                         frs.sr_nentries = 0;
1481                         rc = fop.o_bd->be_search( &fop, &frs );
1482
1483                         /* If entry was not found, add to delete list */
1484                         if ( !cb.sc_private ) {
1485                                 uuids[ndel++] = uuids[i];
1486                         }
1487                 }
1488                 fop.o_bd->bd_info = (BackendInfo *)on;
1489         }
1490         if ( ndel ) {
1491                 struct berval cookie;
1492
1493                 slap_compose_sync_cookie( op, &cookie, &delcsn, srs->sr_state.rid );
1494                 uuids[ndel].bv_val = NULL;
1495                 syncprov_sendinfo( op, rs, LDAP_TAG_SYNC_ID_SET, &cookie, 0, uuids, 1 );
1496                 op->o_tmpfree( cookie.bv_val, op->o_tmpmemctx );
1497         }
1498         op->o_tmpfree( uuids, op->o_tmpmemctx );
1499 }
1500
1501 static int
1502 syncprov_op_response( Operation *op, SlapReply *rs )
1503 {
1504         opcookie *opc = op->o_callback->sc_private;
1505         slap_overinst *on = opc->son;
1506         syncprov_info_t         *si = on->on_bi.bi_private;
1507         syncmatches *sm;
1508
1509         if ( rs->sr_err == LDAP_SUCCESS )
1510         {
1511                 struct berval maxcsn = BER_BVNULL;
1512                 char cbuf[LDAP_LUTIL_CSNSTR_BUFSIZE];
1513
1514                 /* Update our context CSN */
1515                 cbuf[0] = '\0';
1516                 ldap_pvt_thread_mutex_lock( &si->si_csn_mutex );
1517                 slap_get_commit_csn( op, &maxcsn );
1518                 if ( !BER_BVISNULL( &maxcsn ) ) {
1519                         strcpy( cbuf, maxcsn.bv_val );
1520                         if ( ber_bvcmp( &maxcsn, &si->si_ctxcsn ) > 0 ) {
1521                                 strcpy( si->si_ctxcsnbuf, cbuf );
1522                                 si->si_ctxcsn.bv_len = maxcsn.bv_len;
1523                         }
1524                 }
1525
1526                 /* Don't do any processing for consumer contextCSN updates */
1527                 if ( SLAP_SYNC_SHADOW( op->o_bd ) &&
1528                         op->o_msgid == SLAP_SYNC_UPDATE_MSGID ) {
1529                         ldap_pvt_thread_mutex_unlock( &si->si_csn_mutex );
1530                         return SLAP_CB_CONTINUE;
1531                 }
1532
1533                 si->si_numops++;
1534                 if ( si->si_chkops || si->si_chktime ) {
1535                         int do_check=0;
1536                         if ( si->si_chkops && si->si_numops >= si->si_chkops ) {
1537                                 do_check = 1;
1538                                 si->si_numops = 0;
1539                         }
1540                         if ( si->si_chktime &&
1541                                 (op->o_time - si->si_chklast >= si->si_chktime )) {
1542                                 do_check = 1;
1543                                 si->si_chklast = op->o_time;
1544                         }
1545                         if ( do_check ) {
1546                                 syncprov_checkpoint( op, rs, on );
1547                         }
1548                 }
1549                 ldap_pvt_thread_mutex_unlock( &si->si_csn_mutex );
1550
1551                 opc->sctxcsn.bv_len = maxcsn.bv_len;
1552                 opc->sctxcsn.bv_val = cbuf;
1553
1554                 /* Handle any persistent searches */
1555                 if ( si->si_ops ) {
1556                         switch(op->o_tag) {
1557                         case LDAP_REQ_ADD:
1558                         case LDAP_REQ_MODIFY:
1559                         case LDAP_REQ_MODRDN:
1560                         case LDAP_REQ_EXTENDED:
1561                                 syncprov_matchops( op, opc, 0 );
1562                                 break;
1563                         case LDAP_REQ_DELETE:
1564                                 /* for each match in opc->smatches:
1565                                  *   send DELETE msg
1566                                  */
1567                                 for ( sm = opc->smatches; sm; sm=sm->sm_next ) {
1568                                         if ( sm->sm_op->s_op->o_abandon )
1569                                                 continue;
1570                                         syncprov_qresp( opc, sm->sm_op, LDAP_SYNC_DELETE );
1571                                 }
1572                                 break;
1573                         }
1574                 }
1575
1576                 /* Add any log records */
1577                 if ( si->si_logs && op->o_tag != LDAP_REQ_ADD ) {
1578                         syncprov_add_slog( op );
1579                 }
1580
1581         }
1582         return SLAP_CB_CONTINUE;
1583 }
1584
1585 /* We don't use a subentry to store the context CSN any more.
1586  * We expose the current context CSN as an operational attribute
1587  * of the suffix entry.
1588  */
1589 static int
1590 syncprov_op_compare( Operation *op, SlapReply *rs )
1591 {
1592         slap_overinst           *on = (slap_overinst *)op->o_bd->bd_info;
1593         syncprov_info_t         *si = on->on_bi.bi_private;
1594         int rc = SLAP_CB_CONTINUE;
1595
1596         if ( dn_match( &op->o_req_ndn, op->o_bd->be_nsuffix ) &&
1597                 op->oq_compare.rs_ava->aa_desc == slap_schema.si_ad_contextCSN )
1598         {
1599                 Entry e = {0};
1600                 Attribute a = {0};
1601                 struct berval bv[2];
1602
1603                 e.e_name = op->o_bd->be_suffix[0];
1604                 e.e_nname = op->o_bd->be_nsuffix[0];
1605
1606                 BER_BVZERO( &bv[1] );
1607                 bv[0] = si->si_ctxcsn;
1608
1609                 a.a_desc = slap_schema.si_ad_contextCSN;
1610                 a.a_vals = bv;
1611                 a.a_nvals = a.a_vals;
1612
1613                 ldap_pvt_thread_mutex_lock( &si->si_csn_mutex );
1614
1615                 rs->sr_err = access_allowed( op, &e, op->oq_compare.rs_ava->aa_desc,
1616                         &op->oq_compare.rs_ava->aa_value, ACL_COMPARE, NULL );
1617                 if ( ! rs->sr_err ) {
1618                         rs->sr_err = LDAP_INSUFFICIENT_ACCESS;
1619                         goto return_results;
1620                 }
1621
1622                 if ( get_assert( op ) &&
1623                         ( test_filter( op, &e, get_assertion( op ) ) != LDAP_COMPARE_TRUE ) )
1624                 {
1625                         rs->sr_err = LDAP_ASSERTION_FAILED;
1626                         goto return_results;
1627                 }
1628
1629
1630                 rs->sr_err = LDAP_COMPARE_FALSE;
1631
1632                 if ( value_find_ex( op->oq_compare.rs_ava->aa_desc,
1633                         SLAP_MR_ATTRIBUTE_VALUE_NORMALIZED_MATCH |
1634                                 SLAP_MR_ASSERTED_VALUE_NORMALIZED_MATCH,
1635                                 a.a_nvals, &op->oq_compare.rs_ava->aa_value, op->o_tmpmemctx ) == 0 )
1636                 {
1637                         rs->sr_err = LDAP_COMPARE_TRUE;
1638                 }
1639
1640 return_results:;
1641
1642                 ldap_pvt_thread_mutex_unlock( &si->si_csn_mutex );
1643
1644                 send_ldap_result( op, rs );
1645
1646                 if( rs->sr_err == LDAP_COMPARE_FALSE || rs->sr_err == LDAP_COMPARE_TRUE ) {
1647                         rs->sr_err = LDAP_SUCCESS;
1648                 }
1649                 rc = rs->sr_err;
1650         }
1651
1652         return rc;
1653 }
1654
1655 static int
1656 syncprov_op_mod( Operation *op, SlapReply *rs )
1657 {
1658         slap_overinst           *on = (slap_overinst *)op->o_bd->bd_info;
1659         syncprov_info_t         *si = on->on_bi.bi_private;
1660
1661         slap_callback *cb = op->o_tmpcalloc(1, sizeof(slap_callback)+
1662                 sizeof(opcookie) +
1663                 (si->si_ops ? sizeof(modinst) : 0 ),
1664                 op->o_tmpmemctx);
1665         opcookie *opc = (opcookie *)(cb+1);
1666         opc->son = on;
1667         cb->sc_response = syncprov_op_response;
1668         cb->sc_cleanup = syncprov_op_cleanup;
1669         cb->sc_private = opc;
1670         cb->sc_next = op->o_callback;
1671         op->o_callback = cb;
1672
1673         /* If there are active persistent searches, lock this operation.
1674          * See seqmod.c for the locking logic on its own.
1675          */
1676         if ( si->si_ops ) {
1677                 modtarget *mt, mtdummy;
1678                 modinst *mi;
1679
1680                 mi = (modinst *)(opc+1);
1681                 mi->mi_op = op;
1682
1683                 /* See if we're already modifying this entry... */
1684                 mtdummy.mt_op = op;
1685                 ldap_pvt_thread_mutex_lock( &si->si_mods_mutex );
1686                 mt = avl_find( si->si_mods, &mtdummy, sp_avl_cmp );
1687                 if ( mt ) {
1688                         ldap_pvt_thread_mutex_lock( &mt->mt_mutex );
1689                         ldap_pvt_thread_mutex_unlock( &si->si_mods_mutex );
1690                         mt->mt_tail->mi_next = mi;
1691                         mt->mt_tail = mi;
1692                         /* wait for this op to get to head of list */
1693                         while ( mt->mt_mods != mi ) {
1694                                 ldap_pvt_thread_mutex_unlock( &mt->mt_mutex );
1695                                 ldap_pvt_thread_yield();
1696                                 ldap_pvt_thread_mutex_lock( &mt->mt_mutex );
1697
1698                                 /* clean up if the caller is giving up */
1699                                 if ( op->o_abandon ) {
1700                                         modinst *m2;
1701                                         for ( m2 = mt->mt_mods; m2->mi_next != mi;
1702                                                 m2 = m2->mi_next );
1703                                         m2->mi_next = mi->mi_next;
1704                                         if ( mt->mt_tail == mi ) mt->mt_tail = m2;
1705                                         op->o_tmpfree( cb, op->o_tmpmemctx );
1706                                         ldap_pvt_thread_mutex_unlock( &mt->mt_mutex );
1707                                         return SLAPD_ABANDON;
1708                                 }
1709                         }
1710                         ldap_pvt_thread_mutex_unlock( &mt->mt_mutex );
1711                 } else {
1712                         /* Record that we're modifying this entry now */
1713                         mt = ch_malloc( sizeof(modtarget) );
1714                         mt->mt_mods = mi;
1715                         mt->mt_tail = mi;
1716                         mt->mt_op = mi->mi_op;
1717                         ldap_pvt_thread_mutex_init( &mt->mt_mutex );
1718                         avl_insert( &si->si_mods, mt, sp_avl_cmp, avl_dup_error );
1719                         ldap_pvt_thread_mutex_unlock( &si->si_mods_mutex );
1720                 }
1721         }
1722
1723         if (( si->si_ops || si->si_logs ) && op->o_tag != LDAP_REQ_ADD )
1724                 syncprov_matchops( op, opc, 1 );
1725
1726         return SLAP_CB_CONTINUE;
1727 }
1728
1729 static int
1730 syncprov_op_extended( Operation *op, SlapReply *rs )
1731 {
1732         if ( exop_is_write( op ))
1733                 return syncprov_op_mod( op, rs );
1734
1735         return SLAP_CB_CONTINUE;
1736 }
1737
1738 typedef struct searchstate {
1739         slap_overinst *ss_on;
1740         syncops *ss_so;
1741         int ss_present;
1742         struct berval ss_ctxcsn;
1743         char ss_csnbuf[LDAP_LUTIL_CSNSTR_BUFSIZE];
1744 } searchstate;
1745
1746 static int
1747 syncprov_search_cleanup( Operation *op, SlapReply *rs )
1748 {
1749         if ( rs->sr_ctrls ) {
1750                 op->o_tmpfree( rs->sr_ctrls[0], op->o_tmpmemctx );
1751                 op->o_tmpfree( rs->sr_ctrls, op->o_tmpmemctx );
1752                 rs->sr_ctrls = NULL;
1753         }
1754         return 0;
1755 }
1756
1757 static void
1758 syncprov_detach_op( Operation *op, syncops *so, slap_overinst *on )
1759 {
1760         Operation *op2;
1761         int i, alen = 0;
1762         size_t size;
1763         char *ptr;
1764         GroupAssertion *g1, *g2;
1765
1766         /* count the search attrs */
1767         for (i=0; op->ors_attrs && !BER_BVISNULL( &op->ors_attrs[i].an_name ); i++) {
1768                 alen += op->ors_attrs[i].an_name.bv_len + 1;
1769         }
1770         /* Make a new copy of the operation */
1771         size = sizeof(Operation) + sizeof(Opheader) +
1772                 (i ? ( (i+1) * sizeof(AttributeName) + alen) : 0) +
1773                 op->o_req_dn.bv_len + 1 +
1774                 op->o_req_ndn.bv_len + 1 +
1775                 op->o_ndn.bv_len + 1 +
1776                 so->s_filterstr.bv_len + 1;
1777         op2 = (Operation *)ch_calloc( 1, size );
1778         op2->o_hdr = (Opheader *)(op2+1);
1779
1780         /* Copy the fields we care about explicitly, leave the rest alone */
1781         *op2->o_hdr = *op->o_hdr;
1782         op2->o_tag = op->o_tag;
1783         op2->o_time = op->o_time;
1784         op2->o_bd = on->on_info->oi_origdb;
1785         op2->o_request = op->o_request;
1786         op2->o_private = on;
1787
1788         if ( i ) {
1789                 op2->ors_attrs = (AttributeName *)(op2->o_hdr + 1);
1790                 ptr = (char *)(op2->ors_attrs+i+1);
1791                 for (i=0; !BER_BVISNULL( &op->ors_attrs[i].an_name ); i++) {
1792                         op2->ors_attrs[i] = op->ors_attrs[i];
1793                         op2->ors_attrs[i].an_name.bv_val = ptr;
1794                         ptr = lutil_strcopy( ptr, op->ors_attrs[i].an_name.bv_val ) + 1;
1795                 }
1796                 BER_BVZERO( &op2->ors_attrs[i].an_name );
1797         } else {
1798                 ptr = (char *)(op2->o_hdr + 1);
1799         }
1800         op2->o_authz = op->o_authz;
1801         op2->o_ndn.bv_val = ptr;
1802         ptr = lutil_strcopy(ptr, op->o_ndn.bv_val) + 1;
1803         op2->o_dn = op2->o_ndn;
1804         op2->o_req_dn.bv_len = op->o_req_dn.bv_len;
1805         op2->o_req_dn.bv_val = ptr;
1806         ptr = lutil_strcopy(ptr, op->o_req_dn.bv_val) + 1;
1807         op2->o_req_ndn.bv_len = op->o_req_ndn.bv_len;
1808         op2->o_req_ndn.bv_val = ptr;
1809         ptr = lutil_strcopy(ptr, op->o_req_ndn.bv_val) + 1;
1810         op2->ors_filterstr.bv_val = ptr;
1811         strcpy( ptr, so->s_filterstr.bv_val );
1812         op2->ors_filterstr.bv_len = so->s_filterstr.bv_len;
1813
1814         /* Skip the AND/GE clause that we stuck on in front */
1815         if ( so->s_flags & PS_FIX_FILTER ) {
1816                 op2->ors_filter = op->ors_filter->f_and->f_next;
1817                 so->s_flags ^= PS_FIX_FILTER;
1818         } else {
1819                 op2->ors_filter = op->ors_filter;
1820         }
1821         op2->ors_filter = filter_dup( op2->ors_filter, NULL );
1822         so->s_op = op2;
1823
1824         /* Copy any cached group ACLs individually */
1825         op2->o_groups = NULL;
1826         for ( g1=op->o_groups; g1; g1=g1->ga_next ) {
1827                 g2 = ch_malloc( sizeof(GroupAssertion) + g1->ga_len );
1828                 *g2 = *g1;
1829                 strcpy( g2->ga_ndn, g1->ga_ndn );
1830                 g2->ga_next = op2->o_groups;
1831                 op2->o_groups = g2;
1832         }
1833         /* Don't allow any further group caching */
1834         op2->o_do_not_cache = 1;
1835
1836         /* Add op2 to conn so abandon will find us */
1837         ldap_pvt_thread_mutex_lock( &op->o_conn->c_mutex );
1838         op->o_conn->c_n_ops_executing++;
1839         op->o_conn->c_n_ops_completed--;
1840         LDAP_STAILQ_INSERT_TAIL( &op->o_conn->c_ops, op2, o_next );
1841         so->s_flags |= PS_IS_DETACHED;
1842         ldap_pvt_thread_mutex_unlock( &op->o_conn->c_mutex );
1843
1844         /* Prevent anyone else from trying to send a result for this op */
1845         op->o_abandon = 1;
1846 }
1847
1848 static int
1849 syncprov_search_response( Operation *op, SlapReply *rs )
1850 {
1851         searchstate *ss = op->o_callback->sc_private;
1852         slap_overinst *on = ss->ss_on;
1853         sync_control *srs = op->o_controls[slap_cids.sc_LDAPsync];
1854
1855         if ( rs->sr_type == REP_SEARCH || rs->sr_type == REP_SEARCHREF ) {
1856                 Attribute *a;
1857                 /* If we got a referral without a referral object, there's
1858                  * something missing that we cannot replicate. Just ignore it.
1859                  * The consumer will abort because we didn't send the expected
1860                  * control.
1861                  */
1862                 if ( !rs->sr_entry ) {
1863                         assert( rs->sr_entry != NULL );
1864                         Debug( LDAP_DEBUG_ANY, "bogus referral in context\n",0,0,0 );
1865                         return SLAP_CB_CONTINUE;
1866                 }
1867                 a = attr_find( rs->sr_entry->e_attrs, slap_schema.si_ad_entryCSN );
1868                 if ( a == NULL && rs->sr_operational_attrs != NULL ) {
1869                         a = attr_find( rs->sr_operational_attrs, slap_schema.si_ad_entryCSN );
1870                 }
1871                 if ( a ) {
1872                         /* Make sure entry is less than the snapshot'd contextCSN */
1873                         if ( ber_bvcmp( &a->a_nvals[0], &ss->ss_ctxcsn ) > 0 ) {
1874                                 Debug( LDAP_DEBUG_SYNC, "Entry %s CSN %s greater than snapshot %s\n",
1875                                         rs->sr_entry->e_name.bv_val,
1876                                         a->a_nvals[0].bv_val,
1877                                         ss->ss_ctxcsn.bv_val );
1878                                 return LDAP_SUCCESS;
1879                         }
1880
1881                         /* Don't send the ctx entry twice */
1882                         if ( !BER_BVISNULL( &srs->sr_state.ctxcsn ) &&
1883                                 bvmatch( &a->a_nvals[0], &srs->sr_state.ctxcsn ) ) {
1884                                 Debug( LDAP_DEBUG_SYNC, "Entry %s CSN %s matches ctx %s\n",
1885                                         rs->sr_entry->e_name.bv_val,
1886                                         a->a_nvals[0].bv_val,
1887                                         srs->sr_state.ctxcsn.bv_val );
1888                                 return LDAP_SUCCESS;
1889                         }
1890                 }
1891                 rs->sr_ctrls = op->o_tmpalloc( sizeof(LDAPControl *)*2,
1892                         op->o_tmpmemctx );
1893                 rs->sr_ctrls[1] = NULL;
1894                 rs->sr_err = syncprov_state_ctrl( op, rs, rs->sr_entry,
1895                         LDAP_SYNC_ADD, rs->sr_ctrls, 0, 0, NULL );
1896         } else if ( rs->sr_type == REP_RESULT && rs->sr_err == LDAP_SUCCESS ) {
1897                 struct berval cookie;
1898
1899                 slap_compose_sync_cookie( op, &cookie, &ss->ss_ctxcsn,
1900                         srs->sr_state.rid );
1901
1902                 /* Is this a regular refresh? */
1903                 if ( !ss->ss_so ) {
1904                         rs->sr_ctrls = op->o_tmpalloc( sizeof(LDAPControl *)*2,
1905                                 op->o_tmpmemctx );
1906                         rs->sr_ctrls[1] = NULL;
1907                         rs->sr_err = syncprov_done_ctrl( op, rs, rs->sr_ctrls,
1908                                 0, 1, &cookie, ss->ss_present ?  LDAP_SYNC_REFRESH_PRESENTS :
1909                                         LDAP_SYNC_REFRESH_DELETES );
1910                         op->o_tmpfree( cookie.bv_val, op->o_tmpmemctx );
1911                 } else {
1912                 /* It's RefreshAndPersist, transition to Persist phase */
1913                         syncprov_sendinfo( op, rs, ss->ss_present ?
1914                                 LDAP_TAG_SYNC_REFRESH_PRESENT : LDAP_TAG_SYNC_REFRESH_DELETE,
1915                                 &cookie, 1, NULL, 0 );
1916                         op->o_tmpfree( cookie.bv_val, op->o_tmpmemctx );
1917
1918                         /* Detach this Op from frontend control */
1919                         ldap_pvt_thread_mutex_lock( &ss->ss_so->s_mutex );
1920
1921                         /* Turn off the refreshing flag */
1922                         ss->ss_so->s_flags ^= PS_IS_REFRESHING;
1923
1924                         syncprov_detach_op( op, ss->ss_so, on );
1925
1926                         /* If there are queued responses, fire them off */
1927                         if ( ss->ss_so->s_res )
1928                                 syncprov_qstart( ss->ss_so );
1929                         ldap_pvt_thread_mutex_unlock( &ss->ss_so->s_mutex );
1930
1931                         return LDAP_SUCCESS;
1932                 }
1933         }
1934
1935         return SLAP_CB_CONTINUE;
1936 }
1937
1938 static int
1939 syncprov_op_search( Operation *op, SlapReply *rs )
1940 {
1941         slap_overinst           *on = (slap_overinst *)op->o_bd->bd_info;
1942         syncprov_info_t         *si = (syncprov_info_t *)on->on_bi.bi_private;
1943         slap_callback   *cb;
1944         int gotstate = 0, nochange = 0, do_present;
1945         syncops *sop = NULL;
1946         searchstate *ss;
1947         sync_control *srs;
1948         struct berval ctxcsn;
1949         char csnbuf[LDAP_LUTIL_CSNSTR_BUFSIZE];
1950
1951         if ( !(op->o_sync_mode & SLAP_SYNC_REFRESH) ) return SLAP_CB_CONTINUE;
1952
1953         if ( op->ors_deref & LDAP_DEREF_SEARCHING ) {
1954                 send_ldap_error( op, rs, LDAP_PROTOCOL_ERROR, "illegal value for derefAliases" );
1955                 return rs->sr_err;
1956         }
1957
1958         do_present = si->si_nopres ? 0 : 1;
1959
1960         srs = op->o_controls[slap_cids.sc_LDAPsync];
1961         op->o_managedsait = SLAP_CONTROL_NONCRITICAL;
1962
1963         /* If this is a persistent search, set it up right away */
1964         if ( op->o_sync_mode & SLAP_SYNC_PERSIST ) {
1965                 syncops so = {0};
1966                 fbase_cookie fc;
1967                 opcookie opc;
1968                 slap_callback sc;
1969
1970                 fc.fss = &so;
1971                 fc.fbase = 0;
1972                 so.s_eid = NOID;
1973                 so.s_op = op;
1974                 so.s_flags = PS_IS_REFRESHING | PS_FIND_BASE;
1975                 /* syncprov_findbase expects to be called as a callback... */
1976                 sc.sc_private = &opc;
1977                 opc.son = on;
1978                 ldap_pvt_thread_mutex_init( &so.s_mutex );
1979                 cb = op->o_callback;
1980                 op->o_callback = &sc;
1981                 rs->sr_err = syncprov_findbase( op, &fc );
1982                 op->o_callback = cb;
1983                 ldap_pvt_thread_mutex_destroy( &so.s_mutex );
1984
1985                 if ( rs->sr_err != LDAP_SUCCESS ) {
1986                         send_ldap_result( op, rs );
1987                         return rs->sr_err;
1988                 }
1989                 sop = ch_malloc( sizeof( syncops ));
1990                 *sop = so;
1991                 ldap_pvt_thread_mutex_init( &sop->s_mutex );
1992                 sop->s_rid = srs->sr_state.rid;
1993                 sop->s_inuse = 1;
1994
1995                 ldap_pvt_thread_mutex_lock( &si->si_ops_mutex );
1996                 sop->s_next = si->si_ops;
1997                 si->si_ops = sop;
1998                 ldap_pvt_thread_mutex_unlock( &si->si_ops_mutex );
1999         }
2000
2001         /* snapshot the ctxcsn */
2002         ldap_pvt_thread_mutex_lock( &si->si_csn_mutex );
2003         strcpy( csnbuf, si->si_ctxcsnbuf );
2004         ctxcsn.bv_len = si->si_ctxcsn.bv_len;
2005         ldap_pvt_thread_mutex_unlock( &si->si_csn_mutex );
2006         ctxcsn.bv_val = csnbuf;
2007
2008         /* If we have a cookie, handle the PRESENT lookups */
2009         if ( !BER_BVISNULL( &srs->sr_state.ctxcsn )) {
2010                 sessionlog *sl;
2011
2012                 /* The cookie was validated when it was parsed, just use it */
2013
2014                 /* If just Refreshing and nothing has changed, shortcut it */
2015                 if ( bvmatch( &srs->sr_state.ctxcsn, &ctxcsn )) {
2016                         nochange = 1;
2017                         if ( !(op->o_sync_mode & SLAP_SYNC_PERSIST) ) {
2018                                 LDAPControl     *ctrls[2];
2019
2020                                 ctrls[0] = NULL;
2021                                 ctrls[1] = NULL;
2022                                 syncprov_done_ctrl( op, rs, ctrls, 0, 0,
2023                                         NULL, LDAP_SYNC_REFRESH_DELETES );
2024                                 rs->sr_ctrls = ctrls;
2025                                 rs->sr_err = LDAP_SUCCESS;
2026                                 send_ldap_result( op, rs );
2027                                 rs->sr_ctrls = NULL;
2028                                 return rs->sr_err;
2029                         }
2030                         goto shortcut;
2031                 }
2032                 /* Do we have a sessionlog for this search? */
2033                 sl=si->si_logs;
2034                 if ( sl ) {
2035                         ldap_pvt_thread_mutex_lock( &sl->sl_mutex );
2036                         /* Are there any log entries, and is the consumer state
2037                          * present in the session log?
2038                          */
2039                         if ( sl->sl_num > 0 && ber_bvcmp( &srs->sr_state.ctxcsn, &sl->sl_mincsn ) >= 0 ) {
2040                                 do_present = 0;
2041                                 /* mutex is unlocked in playlog */
2042                                 syncprov_playlog( op, rs, sl, srs, &ctxcsn );
2043                         } else {
2044                                 ldap_pvt_thread_mutex_unlock( &sl->sl_mutex );
2045                         }
2046                 }
2047                 /* Is the CSN still present in the database? */
2048                 if ( syncprov_findcsn( op, FIND_CSN ) != LDAP_SUCCESS ) {
2049                         /* No, so a reload is required */
2050                         /* the 2.2 consumer doesn't send this hint */
2051                         if ( si->si_usehint && srs->sr_rhint == 0 ) {
2052                                 send_ldap_error( op, rs, LDAP_SYNC_REFRESH_REQUIRED, "sync cookie is stale" );
2053                                 return rs->sr_err;
2054                         }
2055                 } else {
2056                         gotstate = 1;
2057                         /* If changed and doing Present lookup, send Present UUIDs */
2058                         if ( do_present && syncprov_findcsn( op, FIND_PRESENT ) !=
2059                                 LDAP_SUCCESS ) {
2060                                 send_ldap_result( op, rs );
2061                                 return rs->sr_err;
2062                         }
2063                 }
2064         }
2065
2066 shortcut:
2067         /* Append CSN range to search filter, save original filter
2068          * for persistent search evaluation
2069          */
2070         if ( sop ) {
2071                 sop->s_filterstr= op->ors_filterstr;
2072         }
2073
2074         /* If something changed, find the changes */
2075         if ( gotstate && !nochange ) {
2076                 Filter *fand, *fava;
2077
2078                 fand = op->o_tmpalloc( sizeof(Filter), op->o_tmpmemctx );
2079                 fand->f_choice = LDAP_FILTER_AND;
2080                 fand->f_next = NULL;
2081                 fava = op->o_tmpalloc( sizeof(Filter), op->o_tmpmemctx );
2082                 fand->f_and = fava;
2083                 fava->f_choice = LDAP_FILTER_GE;
2084                 fava->f_ava = op->o_tmpalloc( sizeof(AttributeAssertion), op->o_tmpmemctx );
2085                 fava->f_ava->aa_desc = slap_schema.si_ad_entryCSN;
2086 #ifdef LDAP_COMP_MATCH
2087                 fava->f_ava->aa_cf = NULL;
2088 #endif
2089                 ber_dupbv_x( &fava->f_ava->aa_value, &srs->sr_state.ctxcsn, op->o_tmpmemctx );
2090                 fava->f_next = op->ors_filter;
2091                 op->ors_filter = fand;
2092                 filter2bv_x( op, op->ors_filter, &op->ors_filterstr );
2093                 if ( sop )
2094                         sop->s_flags |= PS_FIX_FILTER;
2095         }
2096
2097         /* Let our callback add needed info to returned entries */
2098         cb = op->o_tmpcalloc(1, sizeof(slap_callback)+sizeof(searchstate), op->o_tmpmemctx);
2099         ss = (searchstate *)(cb+1);
2100         ss->ss_on = on;
2101         ss->ss_so = sop;
2102         ss->ss_present = do_present;
2103         ss->ss_ctxcsn.bv_len = ctxcsn.bv_len;
2104         ss->ss_ctxcsn.bv_val = ss->ss_csnbuf;
2105         strcpy( ss->ss_ctxcsn.bv_val, ctxcsn.bv_val );
2106         cb->sc_response = syncprov_search_response;
2107         cb->sc_cleanup = syncprov_search_cleanup;
2108         cb->sc_private = ss;
2109         cb->sc_next = op->o_callback;
2110         op->o_callback = cb;
2111
2112 #if 0   /* I don't think we need to shortcircuit back-bdb any more */
2113         op->o_sync_mode &= SLAP_CONTROL_MASK;
2114 #endif
2115
2116         /* If this is a persistent search and no changes were reported during
2117          * the refresh phase, just invoke the response callback to transition
2118          * us into persist phase
2119          */
2120         if ( nochange ) {
2121                 rs->sr_err = LDAP_SUCCESS;
2122                 rs->sr_nentries = 0;
2123                 send_ldap_result( op, rs );
2124                 return rs->sr_err;
2125         }
2126         return SLAP_CB_CONTINUE;
2127 }
2128
2129 static int
2130 syncprov_operational(
2131         Operation *op,
2132         SlapReply *rs )
2133 {
2134         slap_overinst           *on = (slap_overinst *)op->o_bd->bd_info;
2135         syncprov_info_t         *si = (syncprov_info_t *)on->on_bi.bi_private;
2136
2137         if ( rs->sr_entry &&
2138                 dn_match( &rs->sr_entry->e_nname, op->o_bd->be_nsuffix )) {
2139
2140                 if ( SLAP_OPATTRS( rs->sr_attr_flags ) ||
2141                         ad_inlist( slap_schema.si_ad_contextCSN, rs->sr_attrs )) {
2142                         Attribute *a, **ap = NULL;
2143
2144                         for ( a=rs->sr_entry->e_attrs; a; a=a->a_next ) {
2145                                 if ( a->a_desc == slap_schema.si_ad_contextCSN )
2146                                         break;
2147                         }
2148
2149                         if ( !a ) {
2150                                 for ( ap = &rs->sr_operational_attrs; *ap; ap=&(*ap)->a_next );
2151
2152                                 a = attr_alloc( slap_schema.si_ad_contextCSN );
2153                                 a->a_vals = ch_malloc( 2 * sizeof(struct berval));
2154                                 a->a_vals[1].bv_val = NULL;
2155                                 a->a_nvals = a->a_vals;
2156                                 *ap = a;
2157                         }
2158
2159                         ldap_pvt_thread_mutex_lock( &si->si_csn_mutex );
2160                         if ( !ap ) {
2161                                 strcpy( a->a_vals[0].bv_val, si->si_ctxcsnbuf );
2162                         } else {
2163                                 ber_dupbv( &a->a_vals[0], &si->si_ctxcsn );
2164                         }
2165                         ldap_pvt_thread_mutex_unlock( &si->si_csn_mutex );
2166                 }
2167         }
2168         return SLAP_CB_CONTINUE;
2169 }
2170
2171 enum {
2172         SP_CHKPT = 1,
2173         SP_SESSL,
2174         SP_NOPRES,
2175         SP_USEHINT
2176 };
2177
2178 static ConfigDriver sp_cf_gen;
2179
2180 static ConfigTable spcfg[] = {
2181         { "syncprov-checkpoint", "ops> <minutes", 3, 3, 0, ARG_MAGIC|SP_CHKPT,
2182                 sp_cf_gen, "( OLcfgOvAt:1.1 NAME 'olcSpCheckpoint' "
2183                         "DESC 'ContextCSN checkpoint interval in ops and minutes' "
2184                         "SYNTAX OMsDirectoryString SINGLE-VALUE )", NULL, NULL },
2185         { "syncprov-sessionlog", "ops", 2, 2, 0, ARG_INT|ARG_MAGIC|SP_SESSL,
2186                 sp_cf_gen, "( OLcfgOvAt:1.2 NAME 'olcSpSessionlog' "
2187                         "DESC 'Session log size in ops' "
2188                         "SYNTAX OMsInteger SINGLE-VALUE )", NULL, NULL },
2189         { "syncprov-nopresent", NULL, 2, 2, 0, ARG_ON_OFF|ARG_MAGIC|SP_NOPRES,
2190                 sp_cf_gen, "( OLcfgOvAt:1.3 NAME 'olcSpNoPresent' "
2191                         "DESC 'Omit Present phase processing' "
2192                         "SYNTAX OMsBoolean SINGLE-VALUE )", NULL, NULL },
2193         { "syncprov-reloadhint", NULL, 2, 2, 0, ARG_ON_OFF|ARG_MAGIC|SP_USEHINT,
2194                 sp_cf_gen, "( OLcfgOvAt:1.4 NAME 'olcSpReloadHint' "
2195                         "DESC 'Observe Reload Hint in Request control' "
2196                         "SYNTAX OMsBoolean SINGLE-VALUE )", NULL, NULL },
2197         { NULL, NULL, 0, 0, 0, ARG_IGNORED }
2198 };
2199
2200 static ConfigOCs spocs[] = {
2201         { "( OLcfgOvOc:1.1 "
2202                 "NAME 'olcSyncProvConfig' "
2203                 "DESC 'SyncRepl Provider configuration' "
2204                 "SUP olcOverlayConfig "
2205                 "MAY ( olcSpCheckpoint $ olcSpSessionlog $ olcSpNoPresent ) )",
2206                         Cft_Overlay, spcfg },
2207         { NULL, 0, NULL }
2208 };
2209
2210 static int
2211 sp_cf_gen(ConfigArgs *c)
2212 {
2213         slap_overinst           *on = (slap_overinst *)c->bi;
2214         syncprov_info_t         *si = (syncprov_info_t *)on->on_bi.bi_private;
2215         int rc = 0;
2216
2217         if ( c->op == SLAP_CONFIG_EMIT ) {
2218                 switch ( c->type ) {
2219                 case SP_CHKPT:
2220                         if ( si->si_chkops || si->si_chktime ) {
2221                                 struct berval bv;
2222                                 bv.bv_len = sprintf( c->msg, "%d %d",
2223                                         si->si_chkops, si->si_chktime );
2224                                 bv.bv_val = c->msg;
2225                                 value_add_one( &c->rvalue_vals, &bv );
2226                         } else {
2227                                 rc = 1;
2228                         }
2229                         break;
2230                 case SP_SESSL:
2231                         if ( si->si_logs ) {
2232                                 c->value_int = si->si_logs->sl_size;
2233                         } else {
2234                                 rc = 1;
2235                         }
2236                         break;
2237                 case SP_NOPRES:
2238                         if ( si->si_nopres ) {
2239                                 c->value_int = 1;
2240                         } else {
2241                                 rc = 1;
2242                         }
2243                         break;
2244                 case SP_USEHINT:
2245                         if ( si->si_usehint ) {
2246                                 c->value_int = 1;
2247                         } else {
2248                                 rc = 1;
2249                         }
2250                         break;
2251                 }
2252                 return rc;
2253         } else if ( c->op == LDAP_MOD_DELETE ) {
2254                 switch ( c->type ) {
2255                 case SP_CHKPT:
2256                         si->si_chkops = 0;
2257                         si->si_chktime = 0;
2258                         break;
2259                 case SP_SESSL:
2260                         if ( si->si_logs )
2261                                 si->si_logs->sl_size = 0;
2262                         else
2263                                 rc = LDAP_NO_SUCH_ATTRIBUTE;
2264                         break;
2265                 case SP_NOPRES:
2266                         if ( si->si_nopres )
2267                                 si->si_nopres = 0;
2268                         else
2269                                 rc = LDAP_NO_SUCH_ATTRIBUTE;
2270                         break;
2271                 case SP_USEHINT:
2272                         if ( si->si_usehint )
2273                                 si->si_usehint = 0;
2274                         else
2275                                 rc = LDAP_NO_SUCH_ATTRIBUTE;
2276                         break;
2277                 }
2278                 return rc;
2279         }
2280         switch ( c->type ) {
2281         case SP_CHKPT:
2282                 if ( lutil_atoi( &si->si_chkops, c->argv[1] ) != 0 ) {
2283                         sprintf( c->msg, "%s unable to parse checkpoint ops # \"%s\"",
2284                                 c->argv[0], c->argv[1] );
2285                         Debug( LDAP_DEBUG_CONFIG, "%s: %s\n", c->log, c->msg, 0 );
2286                         return ARG_BAD_CONF;
2287                 }
2288                 if ( si->si_chkops <= 0 ) {
2289                         sprintf( c->msg, "%s invalid checkpoint ops # \"%d\"",
2290                                 c->argv[0], si->si_chkops );
2291                         Debug( LDAP_DEBUG_CONFIG, "%s: %s\n", c->log, c->msg, 0 );
2292                         return ARG_BAD_CONF;
2293                 }
2294                 if ( lutil_atoi( &si->si_chktime, c->argv[2] ) != 0 ) {
2295                         sprintf( c->msg, "%s unable to parse checkpoint time \"%s\"",
2296                                 c->argv[0], c->argv[1] );
2297                         Debug( LDAP_DEBUG_CONFIG, "%s: %s\n", c->log, c->msg, 0 );
2298                         return ARG_BAD_CONF;
2299                 }
2300                 if ( si->si_chktime <= 0 ) {
2301                         sprintf( c->msg, "%s invalid checkpoint time \"%d\"",
2302                                 c->argv[0], si->si_chkops );
2303                         Debug( LDAP_DEBUG_CONFIG, "%s: %s\n", c->log, c->msg, 0 );
2304                         return ARG_BAD_CONF;
2305                 }
2306                 si->si_chktime *= 60;
2307                 break;
2308         case SP_SESSL: {
2309                 sessionlog *sl;
2310                 int size = c->value_int;
2311
2312                 if ( size < 0 ) {
2313                         sprintf( c->msg, "%s size %d is negative",
2314                                 c->argv[0], size );
2315                         Debug( LDAP_DEBUG_CONFIG, "%s: %s\n", c->log, c->msg, 0 );
2316                         return ARG_BAD_CONF;
2317                 }
2318                 sl = si->si_logs;
2319                 if ( !sl ) {
2320                         sl = ch_malloc( sizeof( sessionlog ) + LDAP_LUTIL_CSNSTR_BUFSIZE );
2321                         sl->sl_mincsn.bv_val = (char *)(sl+1);
2322                         sl->sl_mincsn.bv_len = 0;
2323                         sl->sl_num = 0;
2324                         sl->sl_head = sl->sl_tail = NULL;
2325                         ldap_pvt_thread_mutex_init( &sl->sl_mutex );
2326                         si->si_logs = sl;
2327                 }
2328                 sl->sl_size = size;
2329                 }
2330                 break;
2331         case SP_NOPRES:
2332                 si->si_nopres = c->value_int;
2333                 break;
2334         case SP_USEHINT:
2335                 si->si_usehint = c->value_int;
2336                 break;
2337         }
2338         return rc;
2339 }
2340
2341 /* ITS#3456 we cannot run this search on the main thread, must use a
2342  * child thread in order to insure we have a big enough stack.
2343  */
2344 static void *
2345 syncprov_db_otask(
2346         void *ptr
2347 )
2348 {
2349         syncprov_findcsn( ptr, FIND_MAXCSN );
2350         return NULL;
2351 }
2352
2353 /* Read any existing contextCSN from the underlying db.
2354  * Then search for any entries newer than that. If no value exists,
2355  * just generate it. Cache whatever result.
2356  */
2357 static int
2358 syncprov_db_open(
2359     BackendDB *be
2360 )
2361 {
2362         slap_overinst   *on = (slap_overinst *) be->bd_info;
2363         syncprov_info_t *si = (syncprov_info_t *)on->on_bi.bi_private;
2364
2365         Connection conn = { 0 };
2366         OperationBuffer opbuf = { 0 };
2367         char ctxcsnbuf[LDAP_LUTIL_CSNSTR_BUFSIZE];
2368         Operation *op = (Operation *) &opbuf;
2369         Entry *e;
2370         Attribute *a;
2371         int rc;
2372         void *thrctx = NULL;
2373
2374         if ( !SLAP_LASTMOD( be )) {
2375                 Debug( LDAP_DEBUG_ANY,
2376                         "syncprov_db_open: invalid config, lastmod must be enabled\n", 0, 0, 0 );
2377                 return -1;
2378         }
2379
2380         if ( slapMode & SLAP_TOOL_MODE ) {
2381                 return 0;
2382         }
2383
2384         rc = overlay_register_control( be, LDAP_CONTROL_SYNC );
2385         if ( rc ) {
2386                 return rc;
2387         }
2388
2389         thrctx = ldap_pvt_thread_pool_context();
2390         connection_fake_init( &conn, op, thrctx );
2391         op->o_bd = be;
2392         op->o_dn = be->be_rootdn;
2393         op->o_ndn = be->be_rootndn;
2394
2395         ctxcsnbuf[0] = '\0';
2396
2397         op->o_bd->bd_info = on->on_info->oi_orig;
2398         rc = be_entry_get_rw( op, be->be_nsuffix, NULL,
2399                 slap_schema.si_ad_contextCSN, 0, &e );
2400
2401         if ( e ) {
2402                 ldap_pvt_thread_t tid;
2403
2404                 a = attr_find( e->e_attrs, slap_schema.si_ad_contextCSN );
2405                 if ( a ) {
2406                         si->si_ctxcsn.bv_len = a->a_nvals[0].bv_len;
2407                         if ( si->si_ctxcsn.bv_len >= sizeof(si->si_ctxcsnbuf ))
2408                                 si->si_ctxcsn.bv_len = sizeof(si->si_ctxcsnbuf)-1;
2409                         strncpy( si->si_ctxcsnbuf, a->a_nvals[0].bv_val,
2410                                 si->si_ctxcsn.bv_len );
2411                         si->si_ctxcsnbuf[si->si_ctxcsn.bv_len] = '\0';
2412                         strcpy( ctxcsnbuf, si->si_ctxcsnbuf );
2413                 }
2414                 be_entry_release_rw( op, e, 0 );
2415                 if ( !BER_BVISEMPTY( &si->si_ctxcsn ) ) {
2416                         op->o_bd->bd_info = (BackendInfo *)on;
2417                         op->o_req_dn = be->be_suffix[0];
2418                         op->o_req_ndn = be->be_nsuffix[0];
2419                         op->ors_scope = LDAP_SCOPE_SUBTREE;
2420                         ldap_pvt_thread_create( &tid, 0, syncprov_db_otask, op );
2421                         ldap_pvt_thread_join( tid, NULL );
2422                 }
2423         }
2424
2425         if ( BER_BVISEMPTY( &si->si_ctxcsn ) ) {
2426                 if ( SLAP_SYNC_SHADOW( op->o_bd )) {
2427                 /* If we're also a consumer, and we didn't get a contextCSN,
2428                  * then don't generate anything, wait for our provider to send it
2429                  * to us.
2430                  */
2431                         goto out;
2432                 }
2433                 si->si_ctxcsn.bv_len = sizeof( si->si_ctxcsnbuf );
2434                 slap_get_csn( op, &si->si_ctxcsn, 0 );
2435         }
2436
2437         /* If our ctxcsn is different from what was read from the root
2438          * entry, make sure we do a checkpoint on close
2439          */
2440         if ( strcmp( si->si_ctxcsnbuf, ctxcsnbuf )) {
2441                 si->si_numops++;
2442         }
2443
2444 out:
2445         op->o_bd->bd_info = (BackendInfo *)on;
2446         ldap_pvt_thread_pool_context_reset( thrctx );
2447         return 0;
2448 }
2449
2450 /* Write the current contextCSN into the underlying db.
2451  */
2452 static int
2453 syncprov_db_close(
2454     BackendDB *be
2455 )
2456 {
2457     slap_overinst   *on = (slap_overinst *) be->bd_info;
2458     syncprov_info_t *si = (syncprov_info_t *)on->on_bi.bi_private;
2459
2460         if ( slapMode & SLAP_TOOL_MODE ) {
2461                 return 0;
2462         }
2463         if ( si->si_numops ) {
2464                 Connection conn;
2465                 OperationBuffer opbuf;
2466                 Operation *op = (Operation *) &opbuf;
2467                 SlapReply rs = {REP_RESULT};
2468                 void *thrctx;
2469
2470                 thrctx = ldap_pvt_thread_pool_context();
2471                 connection_fake_init( &conn, op, thrctx );
2472                 op->o_bd = be;
2473                 op->o_dn = be->be_rootdn;
2474                 op->o_ndn = be->be_rootndn;
2475                 syncprov_checkpoint( op, &rs, on );
2476                 ldap_pvt_thread_pool_context_reset( thrctx );
2477         }
2478
2479     return 0;
2480 }
2481
2482 static int
2483 syncprov_db_init(
2484         BackendDB *be
2485 )
2486 {
2487         slap_overinst   *on = (slap_overinst *)be->bd_info;
2488         syncprov_info_t *si;
2489
2490         si = ch_calloc(1, sizeof(syncprov_info_t));
2491         on->on_bi.bi_private = si;
2492         ldap_pvt_thread_mutex_init( &si->si_csn_mutex );
2493         ldap_pvt_thread_mutex_init( &si->si_ops_mutex );
2494         ldap_pvt_thread_mutex_init( &si->si_mods_mutex );
2495         si->si_ctxcsn.bv_val = si->si_ctxcsnbuf;
2496
2497         csn_anlist[0].an_desc = slap_schema.si_ad_entryCSN;
2498         csn_anlist[0].an_name = slap_schema.si_ad_entryCSN->ad_cname;
2499         csn_anlist[1].an_desc = slap_schema.si_ad_entryUUID;
2500         csn_anlist[1].an_name = slap_schema.si_ad_entryUUID->ad_cname;
2501
2502         uuid_anlist[0].an_desc = slap_schema.si_ad_entryUUID;
2503         uuid_anlist[0].an_name = slap_schema.si_ad_entryUUID->ad_cname;
2504
2505         return 0;
2506 }
2507
2508 static int
2509 syncprov_db_destroy(
2510         BackendDB *be
2511 )
2512 {
2513         slap_overinst   *on = (slap_overinst *)be->bd_info;
2514         syncprov_info_t *si = (syncprov_info_t *)on->on_bi.bi_private;
2515
2516         if ( si ) {
2517                 if ( si->si_logs ) {
2518                         slog_entry *se = si->si_logs->sl_head;
2519
2520                         while ( se ) {
2521                                 slog_entry *se_next = se->se_next;
2522                                 ch_free( se );
2523                                 se = se_next;
2524                         }
2525
2526                         ch_free( si->si_logs );
2527                 }
2528                 ldap_pvt_thread_mutex_destroy( &si->si_mods_mutex );
2529                 ldap_pvt_thread_mutex_destroy( &si->si_ops_mutex );
2530                 ldap_pvt_thread_mutex_destroy( &si->si_csn_mutex );
2531                 ch_free( si );
2532         }
2533
2534         return 0;
2535 }
2536
2537 static int syncprov_parseCtrl (
2538         Operation *op,
2539         SlapReply *rs,
2540         LDAPControl *ctrl )
2541 {
2542         ber_tag_t tag;
2543         BerElementBuffer berbuf;
2544         BerElement *ber = (BerElement *)&berbuf;
2545         ber_int_t mode;
2546         ber_len_t len;
2547         struct berval cookie = BER_BVNULL;
2548         sync_control *sr;
2549         int rhint = 0;
2550
2551         if ( op->o_sync != SLAP_CONTROL_NONE ) {
2552                 rs->sr_text = "Sync control specified multiple times";
2553                 return LDAP_PROTOCOL_ERROR;
2554         }
2555
2556         if ( op->o_pagedresults != SLAP_CONTROL_NONE ) {
2557                 rs->sr_text = "Sync control specified with pagedResults control";
2558                 return LDAP_PROTOCOL_ERROR;
2559         }
2560
2561         if ( BER_BVISEMPTY( &ctrl->ldctl_value ) ) {
2562                 rs->sr_text = "Sync control value is empty (or absent)";
2563                 return LDAP_PROTOCOL_ERROR;
2564         }
2565
2566         /* Parse the control value
2567          *      syncRequestValue ::= SEQUENCE {
2568          *              mode   ENUMERATED {
2569          *                      -- 0 unused
2570          *                      refreshOnly             (1),
2571          *                      -- 2 reserved
2572          *                      refreshAndPersist       (3)
2573          *              },
2574          *              cookie  syncCookie OPTIONAL
2575          *      }
2576          */
2577
2578         ber_init2( ber, &ctrl->ldctl_value, 0 );
2579
2580         if ( (tag = ber_scanf( ber, "{i" /*}*/, &mode )) == LBER_ERROR ) {
2581                 rs->sr_text = "Sync control : mode decoding error";
2582                 return LDAP_PROTOCOL_ERROR;
2583         }
2584
2585         switch( mode ) {
2586         case LDAP_SYNC_REFRESH_ONLY:
2587                 mode = SLAP_SYNC_REFRESH;
2588                 break;
2589         case LDAP_SYNC_REFRESH_AND_PERSIST:
2590                 mode = SLAP_SYNC_REFRESH_AND_PERSIST;
2591                 break;
2592         default:
2593                 rs->sr_text = "Sync control : unknown update mode";
2594                 return LDAP_PROTOCOL_ERROR;
2595         }
2596
2597         tag = ber_peek_tag( ber, &len );
2598
2599         if ( tag == LDAP_TAG_SYNC_COOKIE ) {
2600                 if (( ber_scanf( ber, /*{*/ "m", &cookie )) == LBER_ERROR ) {
2601                         rs->sr_text = "Sync control : cookie decoding error";
2602                         return LDAP_PROTOCOL_ERROR;
2603                 }
2604                 tag = ber_peek_tag( ber, &len );
2605         }
2606         if ( tag == LDAP_TAG_RELOAD_HINT ) {
2607                 if (( ber_scanf( ber, /*{*/ "b", &rhint )) == LBER_ERROR ) {
2608                         rs->sr_text = "Sync control : rhint decoding error";
2609                         return LDAP_PROTOCOL_ERROR;
2610                 }
2611         }
2612         if (( ber_scanf( ber, /*{*/ "}")) == LBER_ERROR ) {
2613                         rs->sr_text = "Sync control : decoding error";
2614                         return LDAP_PROTOCOL_ERROR;
2615         }
2616         sr = op->o_tmpcalloc( 1, sizeof(struct sync_control), op->o_tmpmemctx );
2617         sr->sr_rhint = rhint;
2618         if (!BER_BVISNULL(&cookie)) {
2619                 ber_dupbv_x( &sr->sr_state.octet_str, &cookie, op->o_tmpmemctx );
2620                 slap_parse_sync_cookie( &sr->sr_state, op->o_tmpmemctx );
2621                 if ( sr->sr_state.rid == -1 ) {
2622                         rs->sr_text = "Sync control : cookie parsing error";
2623                         return LDAP_PROTOCOL_ERROR;
2624                 }
2625         }
2626
2627         op->o_controls[slap_cids.sc_LDAPsync] = sr;
2628
2629         op->o_sync = ctrl->ldctl_iscritical
2630                 ? SLAP_CONTROL_CRITICAL
2631                 : SLAP_CONTROL_NONCRITICAL;
2632
2633         op->o_sync_mode |= mode;        /* o_sync_mode shares o_sync */
2634
2635         return LDAP_SUCCESS;
2636 }
2637
2638 /* This overlay is set up for dynamic loading via moduleload. For static
2639  * configuration, you'll need to arrange for the slap_overinst to be
2640  * initialized and registered by some other function inside slapd.
2641  */
2642
2643 static slap_overinst            syncprov;
2644
2645 int
2646 syncprov_initialize()
2647 {
2648         int rc;
2649
2650         rc = register_supported_control( LDAP_CONTROL_SYNC,
2651                 SLAP_CTRL_SEARCH, NULL,
2652                 syncprov_parseCtrl, &slap_cids.sc_LDAPsync );
2653         if ( rc != LDAP_SUCCESS ) {
2654                 Debug( LDAP_DEBUG_ANY,
2655                         "syncprov_init: Failed to register control %d\n", rc, 0, 0 );
2656                 return rc;
2657         }
2658
2659         syncprov.on_bi.bi_type = "syncprov";
2660         syncprov.on_bi.bi_db_init = syncprov_db_init;
2661         syncprov.on_bi.bi_db_destroy = syncprov_db_destroy;
2662         syncprov.on_bi.bi_db_open = syncprov_db_open;
2663         syncprov.on_bi.bi_db_close = syncprov_db_close;
2664
2665         syncprov.on_bi.bi_op_abandon = syncprov_op_abandon;
2666         syncprov.on_bi.bi_op_cancel = syncprov_op_abandon;
2667
2668         syncprov.on_bi.bi_op_add = syncprov_op_mod;
2669         syncprov.on_bi.bi_op_compare = syncprov_op_compare;
2670         syncprov.on_bi.bi_op_delete = syncprov_op_mod;
2671         syncprov.on_bi.bi_op_modify = syncprov_op_mod;
2672         syncprov.on_bi.bi_op_modrdn = syncprov_op_mod;
2673         syncprov.on_bi.bi_op_search = syncprov_op_search;
2674         syncprov.on_bi.bi_extended = syncprov_op_extended;
2675         syncprov.on_bi.bi_operational = syncprov_operational;
2676
2677         syncprov.on_bi.bi_cf_ocs = spocs;
2678
2679         generic_filter.f_desc = slap_schema.si_ad_objectClass;
2680
2681         rc = config_register_schema( spcfg, spocs );
2682         if ( rc ) return rc;
2683
2684         return overlay_register( &syncprov );
2685 }
2686
2687 #if SLAPD_OVER_SYNCPROV == SLAPD_MOD_DYNAMIC
2688 int
2689 init_module( int argc, char *argv[] )
2690 {
2691         return syncprov_initialize();
2692 }
2693 #endif /* SLAPD_OVER_SYNCPROV == SLAPD_MOD_DYNAMIC */
2694
2695 #endif /* defined(SLAPD_OVER_SYNCPROV) */