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