]> git.sur5r.net Git - openldap/blob - servers/slapd/ldapsync.c
Merge remote-tracking branch 'origin/mdb.master' into OPENLDAP_REL_ENG_2_5
[openldap] / servers / slapd / ldapsync.c
1 /* ldapsync.c -- LDAP Content Sync Routines */
2 /* $OpenLDAP$ */
3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4  *
5  * Copyright 2003-2014 The OpenLDAP Foundation.
6  * Portions Copyright 2003 IBM Corporation.
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted only as authorized by the OpenLDAP
11  * Public License.
12  *
13  * A copy of this license is available in the file LICENSE in the
14  * top-level directory of the distribution or, alternatively, at
15  * <http://www.OpenLDAP.org/license.html>.
16  */
17
18 #include "portable.h"
19
20 #include <stdio.h>
21
22 #include <ac/string.h>
23 #include <ac/socket.h>
24
25 #include "lutil.h"
26 #include "slap.h"
27 #include "../../libraries/liblber/lber-int.h" /* get ber_strndup() */
28 #include "lutil_ldap.h"
29
30 struct slap_sync_cookie_s slap_sync_cookie =
31         LDAP_STAILQ_HEAD_INITIALIZER( slap_sync_cookie );
32
33 void
34 slap_compose_sync_cookie(
35         Operation *op,
36         struct berval *cookie,
37         BerVarray csn,
38         int rid,
39         int sid )
40 {
41         int len, numcsn = 0;
42
43         if ( csn ) {
44                 for (; !BER_BVISNULL( &csn[numcsn] ); numcsn++);
45         }
46
47         if ( numcsn == 0 || rid == -1 ) {
48                 char cookiestr[ LDAP_PVT_CSNSTR_BUFSIZE + 20 ];
49                 if ( rid == -1 ) {
50                         cookiestr[0] = '\0';
51                         len = 0;
52                 } else {
53                         len = snprintf( cookiestr, sizeof( cookiestr ),
54                                         "rid=%03d", rid );
55                         if ( sid >= 0 ) {
56                                 len += sprintf( cookiestr+len, ",sid=%03x", sid );
57                         }
58                 }
59                 ber_str2bv_x( cookiestr, len, 1, cookie, 
60                         op ? op->o_tmpmemctx : NULL );
61         } else {
62                 char *ptr;
63                 int i;
64
65                 len = 0;
66                 for ( i=0; i<numcsn; i++)
67                         len += csn[i].bv_len + 1;
68
69                 len += STRLENOF("rid=123,csn=");
70                 if ( sid >= 0 )
71                         len += STRLENOF("sid=xxx,");
72
73                 cookie->bv_val = slap_sl_malloc( len, op ? op->o_tmpmemctx : NULL );
74
75                 len = sprintf( cookie->bv_val, "rid=%03d,", rid );
76                 ptr = cookie->bv_val + len;
77                 if ( sid >= 0 ) {
78                         ptr += sprintf( ptr, "sid=%03x,", sid );
79                 }
80                 ptr = lutil_strcopy( ptr, "csn=" );
81                 for ( i=0; i<numcsn; i++) {
82                         ptr = lutil_strncopy( ptr, csn[i].bv_val, csn[i].bv_len );
83                         *ptr++ = ';';
84                 }
85                 ptr--;
86                 *ptr = '\0';
87                 cookie->bv_len = ptr - cookie->bv_val;
88         }
89 }
90
91 void
92 slap_sync_cookie_free(
93         struct sync_cookie *cookie,
94         int free_cookie
95 )
96 {
97         if ( cookie == NULL )
98                 return;
99
100         if ( cookie->sids ) {
101                 ch_free( cookie->sids );
102                 cookie->sids = NULL;
103         }
104
105         if ( cookie->ctxcsn ) {
106                 ber_bvarray_free( cookie->ctxcsn );
107                 cookie->ctxcsn = NULL;
108         }
109         cookie->numcsns = 0;
110         if ( !BER_BVISNULL( &cookie->octet_str )) {
111                 ch_free( cookie->octet_str.bv_val );
112                 BER_BVZERO( &cookie->octet_str );
113         }
114
115         if ( free_cookie ) {
116                 ch_free( cookie );
117         }
118
119         return;
120 }
121
122 int
123 slap_parse_csn_sid( struct berval *csnp )
124 {
125         char *p, *q;
126         struct berval csn = *csnp;
127         int i;
128
129         p = ber_bvchr( &csn, '#' );
130         if ( !p )
131                 return -1;
132         p++;
133         csn.bv_len -= p - csn.bv_val;
134         csn.bv_val = p;
135
136         p = ber_bvchr( &csn, '#' );
137         if ( !p )
138                 return -1;
139         p++;
140         csn.bv_len -= p - csn.bv_val;
141         csn.bv_val = p;
142
143         q = ber_bvchr( &csn, '#' );
144         if ( !q )
145                 return -1;
146
147         csn.bv_len = q - p;
148
149         i = strtol( p, &q, 16 );
150         if ( p == q || q != p + csn.bv_len || i < 0 || i > SLAP_SYNC_SID_MAX ) {
151                 i = -1;
152         }
153
154         return i;
155 }
156
157 int *
158 slap_parse_csn_sids( BerVarray csns, int numcsns, void *memctx )
159 {
160         int i, *ret;
161
162         ret = slap_sl_malloc( numcsns * sizeof(int), memctx );
163         for ( i=0; i<numcsns; i++ ) {
164                 ret[i] = slap_parse_csn_sid( &csns[i] );
165         }
166         return ret;
167 }
168
169 static slap_mr_match_func sidsort_cmp;
170
171 static const MatchingRule sidsort_mr = {
172         { 0 },
173         NULL,
174         { 0 },
175         { 0 },
176         0,
177         NULL, NULL, NULL, sidsort_cmp
178 };
179 static const AttributeType sidsort_at = {
180         { 0 },
181         { 0 },
182         NULL, NULL, (MatchingRule *)&sidsort_mr,
183         NULL, NULL, NULL, NULL, NULL, NULL, NULL, SLAP_AT_SORTED_VAL
184 };
185 static const AttributeDescription sidsort_ad = {
186         NULL,
187         (AttributeType *)&sidsort_at
188 };
189
190 static int
191 sidsort_cmp(
192         int *matchp,
193         slap_mask_t flags,
194         Syntax *syntax,
195         MatchingRule *mr,
196         struct berval *b1,
197         void *v2 )
198 {
199         struct berval *b2 = v2;
200         *matchp = b1->bv_len - b2->bv_len;
201         return LDAP_SUCCESS;
202 }
203
204 /* sort CSNs by SID. Use a fake Attribute with our own
205  * syntax and matching rule, which sorts the nvals by
206  * bv_len order. Stuff our sids into the bv_len.
207  */
208 int
209 slap_sort_csn_sids( BerVarray csns, int *sids, int numcsns, void *memctx )
210 {
211         Attribute a;
212         const char *text;
213         int i, rc;
214
215         a.a_desc = (AttributeDescription *)&sidsort_ad;
216         a.a_nvals = slap_sl_malloc( numcsns * sizeof(struct berval), memctx );
217         for ( i=0; i<numcsns; i++ ) {
218                 a.a_nvals[i].bv_len = sids[i];
219                 a.a_nvals[i].bv_val = NULL;
220         }
221         a.a_vals = csns;
222         a.a_numvals = numcsns;
223         a.a_flags = 0;
224         rc = slap_sort_vals( (Modifications *)&a, &text, &i, memctx );
225         for ( i=0; i<numcsns; i++ )
226                 sids[i] = a.a_nvals[i].bv_len;
227         slap_sl_free( a.a_nvals, memctx );
228         return rc;
229 }
230
231 void
232 slap_insert_csn_sids(
233         struct sync_cookie *ck,
234         int pos,
235         int sid,
236         struct berval *csn
237 )
238 {
239         int i;
240         ck->numcsns++;
241         ck->ctxcsn = ch_realloc( ck->ctxcsn,
242                 (ck->numcsns+1) * sizeof(struct berval));
243         BER_BVZERO( &ck->ctxcsn[ck->numcsns] );
244         ck->sids = ch_realloc( ck->sids, ck->numcsns * sizeof(int));
245         for ( i = ck->numcsns-1; i > pos; i-- ) {
246                 ck->ctxcsn[i] = ck->ctxcsn[i-1];
247                 ck->sids[i] = ck->sids[i-1];
248         }
249         ck->sids[i] = sid;
250         ber_dupbv( &ck->ctxcsn[i], csn );
251 }
252
253 int
254 slap_parse_sync_cookie(
255         struct sync_cookie *cookie,
256         void *memctx
257 )
258 {
259         char *csn_ptr;
260         char *csn_str;
261         char *cval;
262         char *next, *end;
263         AttributeDescription *ad = slap_schema.si_ad_entryCSN;
264
265         if ( cookie == NULL )
266                 return -1;
267
268         if ( cookie->octet_str.bv_len <= STRLENOF( "rid=" ) )
269                 return -1;
270
271         cookie->rid = -1;
272         cookie->sid = -1;
273         cookie->ctxcsn = NULL;
274         cookie->sids = NULL;
275         cookie->numcsns = 0;
276
277         end = cookie->octet_str.bv_val + cookie->octet_str.bv_len;
278
279         for ( next=cookie->octet_str.bv_val; next < end; ) {
280                 if ( !strncmp( next, "rid=", STRLENOF("rid=") )) {
281                         char *rid_ptr = next;
282                         cookie->rid = strtol( &rid_ptr[ STRLENOF( "rid=" ) ], &next, 10 );
283                         if ( next == rid_ptr ||
284                                 next > end ||
285                                 ( *next && *next != ',' ) ||
286                                 cookie->rid < 0 ||
287                                 cookie->rid > SLAP_SYNC_RID_MAX )
288                         {
289                                 return -1;
290                         }
291                         if ( *next == ',' ) {
292                                 next++;
293                         }
294                         if ( !ad ) {
295                                 break;
296                         }
297                         continue;
298                 }
299                 if ( !strncmp( next, "sid=", STRLENOF("sid=") )) {
300                         char *sid_ptr = next;
301                         sid_ptr = next;
302                         cookie->sid = strtol( &sid_ptr[ STRLENOF( "sid=" ) ], &next, 16 );
303                         if ( next == sid_ptr ||
304                                 next > end ||
305                                 ( *next && *next != ',' ) ||
306                                 cookie->sid < 0 ||
307                                 cookie->sid > SLAP_SYNC_SID_MAX )
308                         {
309                                 return -1;
310                         }
311                         if ( *next == ',' ) {
312                                 next++;
313                         }
314                         continue;
315                 }
316                 if ( !strncmp( next, "csn=", STRLENOF("csn=") )) {
317                         struct berval stamp;
318
319                         next += STRLENOF("csn=");
320                         while ( next < end ) {
321                                 csn_str = next;
322                                 csn_ptr = strchr( csn_str, '#' );
323                                 if ( !csn_ptr || csn_ptr > end )
324                                         break;
325                                 /* ad will be NULL when called from main. we just
326                                  * want to parse the rid then. But we still iterate
327                                  * through the string to find the end.
328                                  */
329                                 cval = strchr( csn_ptr, ';' );
330                                 if ( !cval )
331                                         cval = strchr(csn_ptr, ',' );
332                                 if ( cval )
333                                         stamp.bv_len = cval - csn_str;
334                                 else
335                                         stamp.bv_len = end - csn_str;
336                                 if ( ad ) {
337                                         struct berval bv;
338                                         stamp.bv_val = csn_str;
339                                         if ( ad->ad_type->sat_syntax->ssyn_validate(
340                                                 ad->ad_type->sat_syntax, &stamp ) != LDAP_SUCCESS )
341                                                 break;
342                                         if ( ad->ad_type->sat_equality->smr_normalize(
343                                                 SLAP_MR_VALUE_OF_ATTRIBUTE_SYNTAX,
344                                                 ad->ad_type->sat_syntax,
345                                                 ad->ad_type->sat_equality,
346                                                 &stamp, &bv, memctx ) != LDAP_SUCCESS )
347                                                 break;
348                                         ber_bvarray_add_x( &cookie->ctxcsn, &bv, memctx );
349                                         cookie->numcsns++;
350                                 }
351                                 if ( cval ) {
352                                         next = cval + 1;
353                                         if ( *cval != ';' )
354                                                 break;
355                                 } else {
356                                         next = end;
357                                         break;
358                                 }
359                         }
360                         continue;
361                 }
362                 next++;
363         }
364         if ( cookie->numcsns ) {
365                 cookie->sids = slap_parse_csn_sids( cookie->ctxcsn, cookie->numcsns,
366                         memctx );
367                 if ( cookie->numcsns > 1 )
368                         slap_sort_csn_sids( cookie->ctxcsn, cookie->sids, cookie->numcsns, memctx );
369         }
370         return 0;
371 }
372
373 /* count the numcsns and regenerate the list of SIDs in a recomposed cookie */
374 void
375 slap_reparse_sync_cookie(
376         struct sync_cookie *cookie,
377         void *memctx )
378 {
379         if ( cookie->ctxcsn ) {
380                 for (; !BER_BVISNULL( &cookie->ctxcsn[cookie->numcsns] ); cookie->numcsns++);
381         }
382         if ( cookie->numcsns ) {
383                 cookie->sids = slap_parse_csn_sids( cookie->ctxcsn, cookie->numcsns, NULL );
384                 if ( cookie->numcsns > 1 )
385                         slap_sort_csn_sids( cookie->ctxcsn, cookie->sids, cookie->numcsns, memctx );
386         }
387 }
388
389 int
390 slap_init_sync_cookie_ctxcsn(
391         struct sync_cookie *cookie
392 )
393 {
394         char csnbuf[ LDAP_PVT_CSNSTR_BUFSIZE + 4 ];
395         struct berval octet_str = BER_BVNULL;
396         struct berval ctxcsn = BER_BVNULL;
397
398         if ( cookie == NULL )
399                 return -1;
400
401         octet_str.bv_len = snprintf( csnbuf, LDAP_PVT_CSNSTR_BUFSIZE + 4,
402                                         "csn=%4d%02d%02d%02d%02d%02dZ#%06x#%02x#%06x",
403                                         1900, 1, 1, 0, 0, 0, 0, 0, 0 );
404         octet_str.bv_val = csnbuf;
405         ch_free( cookie->octet_str.bv_val );
406         ber_dupbv( &cookie->octet_str, &octet_str );
407
408         ctxcsn.bv_val = octet_str.bv_val + 4;
409         ctxcsn.bv_len = octet_str.bv_len - 4;
410         cookie->ctxcsn = NULL;
411         value_add_one( &cookie->ctxcsn, &ctxcsn );
412         cookie->numcsns = 1;
413         cookie->sid = -1;
414
415         return 0;
416 }
417
418 struct sync_cookie *
419 slap_dup_sync_cookie(
420         struct sync_cookie *dst,
421         struct sync_cookie *src
422 )
423 {
424         struct sync_cookie *new;
425         int i;
426
427         if ( src == NULL )
428                 return NULL;
429
430         if ( dst ) {
431                 ber_bvarray_free( dst->ctxcsn );
432                 dst->ctxcsn = NULL;
433                 dst->sids = NULL;
434                 ch_free( dst->octet_str.bv_val );
435                 BER_BVZERO( &dst->octet_str );
436                 new = dst;
437         } else {
438                 new = ( struct sync_cookie * )
439                                 ch_calloc( 1, sizeof( struct sync_cookie ));
440         }
441
442         new->rid = src->rid;
443         new->sid = src->sid;
444         new->numcsns = src->numcsns;
445
446         if ( src->numcsns ) {
447                 if ( ber_bvarray_dup_x( &new->ctxcsn, src->ctxcsn, NULL )) {
448                         if ( !dst ) {
449                                 ch_free( new );
450                         }
451                         return NULL;
452                 }
453                 new->sids = ch_malloc( src->numcsns * sizeof(int) );
454                 for (i=0; i<src->numcsns; i++)
455                         new->sids[i] = src->sids[i];
456         }
457
458         if ( !BER_BVISNULL( &src->octet_str )) {
459                 ber_dupbv( &new->octet_str, &src->octet_str );
460         }
461
462         return new;
463 }
464