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