effec46fda45d9900d916d0823ebaae98a3ac90f
[ira/wip.git] / source4 / lib / ldb-samba / ldif_handlers.c
1 /* 
2    ldb database library - ldif handlers for Samba
3
4    Copyright (C) Andrew Tridgell 2005
5    Copyright (C) Andrew Bartlett 2006-2009
6    Copyright (C) Matthias Dieter Wallnöfer 2009
7      ** NOTE! The following LGPL license applies to the ldb
8      ** library. This does NOT imply that all of Samba is released
9      ** under the LGPL
10    
11    This library is free software; you can redistribute it and/or
12    modify it under the terms of the GNU Lesser General Public
13    License as published by the Free Software Foundation; either
14    version 3 of the License, or (at your option) any later version.
15
16    This library is distributed in the hope that it will be useful,
17    but WITHOUT ANY WARRANTY; without even the implied warranty of
18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19    Lesser General Public License for more details.
20
21    You should have received a copy of the GNU Lesser General Public
22    License along with this library; if not, see <http://www.gnu.org/licenses/>.
23 */
24
25 #include "includes.h"
26 #include "lib/ldb/include/ldb.h"
27 #include "lib/ldb/include/ldb_module.h"
28 #include "ldb_handlers.h"
29 #include "dsdb/samdb/samdb.h"
30 #include "librpc/gen_ndr/ndr_security.h"
31 #include "librpc/gen_ndr/ndr_misc.h"
32 #include "librpc/gen_ndr/ndr_drsblobs.h"
33 #include "librpc/ndr/libndr.h"
34 #include "libcli/security/security.h"
35 #include "param/param.h"
36 #include "../lib/util/asn1.h"
37
38 /*
39   use ndr_print_* to convert a NDR formatted blob to a ldif formatted blob
40
41   If mask_errors is true, then function succeeds but out data
42   is set to "<Unable to decode binary data>" message
43
44   \return 0 on success; -1 on error
45 */
46 static int ldif_write_NDR(struct ldb_context *ldb, void *mem_ctx,
47                           const struct ldb_val *in, struct ldb_val *out,
48                           size_t struct_size,
49                           ndr_pull_flags_fn_t pull_fn,
50                           ndr_print_fn_t print_fn,
51                           bool mask_errors)
52 {
53         uint8_t *p;
54         enum ndr_err_code err;
55         if (!(ldb_get_flags(ldb) & LDB_FLG_SHOW_BINARY)) {
56                 return ldb_handler_copy(ldb, mem_ctx, in, out);
57         }
58         p = talloc_size(mem_ctx, struct_size);
59         err = ndr_pull_struct_blob(in, mem_ctx, 
60                                    lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")), 
61                                    p, pull_fn);
62         if (err != NDR_ERR_SUCCESS) {
63                 /* fail in not in mask_error mode */
64                 if (!mask_errors) {
65                         return -1;
66                 }
67                 talloc_free(p);
68                 out->data = (uint8_t *)talloc_strdup(mem_ctx, "<Unable to decode binary data>");
69                 out->length = strlen((const char *)out->data);
70                 return 0;
71         }
72         out->data = (uint8_t *)ndr_print_struct_string(mem_ctx, print_fn, "NDR", p);
73         talloc_free(p);
74         if (out->data == NULL) {
75                 return ldb_handler_copy(ldb, mem_ctx, in, out);         
76         }
77         out->length = strlen((char *)out->data);
78         return 0;
79 }
80
81 /*
82   convert a ldif formatted objectSid to a NDR formatted blob
83 */
84 static int ldif_read_objectSid(struct ldb_context *ldb, void *mem_ctx,
85                                const struct ldb_val *in, struct ldb_val *out)
86 {
87         enum ndr_err_code ndr_err;
88         struct dom_sid *sid;
89         sid = dom_sid_parse_length(mem_ctx, in);
90         if (sid == NULL) {
91                 return -1;
92         }
93         ndr_err = ndr_push_struct_blob(out, mem_ctx, NULL, sid,
94                                        (ndr_push_flags_fn_t)ndr_push_dom_sid);
95         talloc_free(sid);
96         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
97                 return -1;
98         }
99         return 0;
100 }
101
102 /*
103   convert a NDR formatted blob to a ldif formatted objectSid
104 */
105 int ldif_write_objectSid(struct ldb_context *ldb, void *mem_ctx,
106                                 const struct ldb_val *in, struct ldb_val *out)
107 {
108         struct dom_sid *sid;
109         enum ndr_err_code ndr_err;
110
111         sid = talloc(mem_ctx, struct dom_sid);
112         if (sid == NULL) {
113                 return -1;
114         }
115         ndr_err = ndr_pull_struct_blob_all(in, sid, NULL, sid,
116                                            (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
117         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
118                 talloc_free(sid);
119                 return -1;
120         }
121         *out = data_blob_string_const(dom_sid_string(mem_ctx, sid));
122         talloc_free(sid);
123         if (out->data == NULL) {
124                 return -1;
125         }
126         return 0;
127 }
128
129 bool ldif_comparision_objectSid_isString(const struct ldb_val *v)
130 {
131         if (v->length < 3) {
132                 return false;
133         }
134
135         if (strncmp("S-", (const char *)v->data, 2) != 0) return false;
136         
137         return true;
138 }
139
140 /*
141   compare two objectSids
142 */
143 static int ldif_comparison_objectSid(struct ldb_context *ldb, void *mem_ctx,
144                                     const struct ldb_val *v1, const struct ldb_val *v2)
145 {
146         if (ldif_comparision_objectSid_isString(v1) && ldif_comparision_objectSid_isString(v2)) {
147                 return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
148         } else if (ldif_comparision_objectSid_isString(v1)
149                    && !ldif_comparision_objectSid_isString(v2)) {
150                 struct ldb_val v;
151                 int ret;
152                 if (ldif_read_objectSid(ldb, mem_ctx, v1, &v) != 0) {
153                         /* Perhaps not a string after all */
154                         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
155                 }
156                 ret = ldb_comparison_binary(ldb, mem_ctx, &v, v2);
157                 talloc_free(v.data);
158                 return ret;
159         } else if (!ldif_comparision_objectSid_isString(v1)
160                    && ldif_comparision_objectSid_isString(v2)) {
161                 struct ldb_val v;
162                 int ret;
163                 if (ldif_read_objectSid(ldb, mem_ctx, v2, &v) != 0) {
164                         /* Perhaps not a string after all */
165                         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
166                 }
167                 ret = ldb_comparison_binary(ldb, mem_ctx, v1, &v);
168                 talloc_free(v.data);
169                 return ret;
170         }
171         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
172 }
173
174 /*
175   canonicalise a objectSid
176 */
177 static int ldif_canonicalise_objectSid(struct ldb_context *ldb, void *mem_ctx,
178                                       const struct ldb_val *in, struct ldb_val *out)
179 {
180         if (ldif_comparision_objectSid_isString(in)) {
181                 if (ldif_read_objectSid(ldb, mem_ctx, in, out) != 0) {
182                         /* Perhaps not a string after all */
183                         return ldb_handler_copy(ldb, mem_ctx, in, out);
184                 }
185                 return 0;
186         }
187         return ldb_handler_copy(ldb, mem_ctx, in, out);
188 }
189
190 static int extended_dn_read_SID(struct ldb_context *ldb, void *mem_ctx,
191                               const struct ldb_val *in, struct ldb_val *out)
192 {
193         struct dom_sid sid;
194         enum ndr_err_code ndr_err;
195         if (ldif_comparision_objectSid_isString(in)) {
196                 if (ldif_read_objectSid(ldb, mem_ctx, in, out) == 0) {
197                         return 0;
198                 }
199         }
200         
201         /* Perhaps not a string after all */
202         *out = data_blob_talloc(mem_ctx, NULL, in->length/2+1);
203
204         if (!out->data) {
205                 return -1;
206         }
207
208         (*out).length = strhex_to_str((char *)out->data, out->length,
209                                      (const char *)in->data, in->length);
210
211         /* Check it looks like a SID */
212         ndr_err = ndr_pull_struct_blob_all(out, mem_ctx, NULL, &sid,
213                                            (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
214         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
215                 return -1;
216         }
217         return 0;
218 }
219
220 /*
221   convert a ldif formatted objectGUID to a NDR formatted blob
222 */
223 static int ldif_read_objectGUID(struct ldb_context *ldb, void *mem_ctx,
224                                 const struct ldb_val *in, struct ldb_val *out)
225 {
226         struct GUID guid;
227         NTSTATUS status;
228
229         status = GUID_from_data_blob(in, &guid);
230         if (!NT_STATUS_IS_OK(status)) {
231                 return -1;
232         }
233
234         status = GUID_to_ndr_blob(&guid, mem_ctx, out);
235         if (!NT_STATUS_IS_OK(status)) {
236                 return -1;
237         }
238         return 0;
239 }
240
241 /*
242   convert a NDR formatted blob to a ldif formatted objectGUID
243 */
244 static int ldif_write_objectGUID(struct ldb_context *ldb, void *mem_ctx,
245                                  const struct ldb_val *in, struct ldb_val *out)
246 {
247         struct GUID guid;
248         NTSTATUS status;
249
250         status = GUID_from_ndr_blob(in, &guid);
251         if (!NT_STATUS_IS_OK(status)) {
252                 return -1;
253         }
254         out->data = (uint8_t *)GUID_string(mem_ctx, &guid);
255         if (out->data == NULL) {
256                 return -1;
257         }
258         out->length = strlen((const char *)out->data);
259         return 0;
260 }
261
262 static bool ldif_comparision_objectGUID_isString(const struct ldb_val *v)
263 {
264         if (v->length != 36 && v->length != 38) return false;
265
266         /* Might be a GUID string, can't be a binary GUID (fixed 16 bytes) */
267         return true;
268 }
269
270 static int extended_dn_read_GUID(struct ldb_context *ldb, void *mem_ctx,
271                               const struct ldb_val *in, struct ldb_val *out)
272 {
273         struct GUID guid;
274         NTSTATUS status;
275
276         if (in->length == 36 && ldif_read_objectGUID(ldb, mem_ctx, in, out) == 0) {
277                 return 0;
278         }
279
280         /* Try as 'hex' form */
281         if (in->length != 32) {
282                 return -1;
283         }
284                 
285         *out = data_blob_talloc(mem_ctx, NULL, in->length/2+1);
286         
287         if (!out->data) {
288                 return -1;
289         }
290         
291         (*out).length = strhex_to_str((char *)out->data, out->length,
292                                       (const char *)in->data, in->length);
293         
294         /* Check it looks like a GUID */
295         status = GUID_from_ndr_blob(out, &guid);
296         if (!NT_STATUS_IS_OK(status)) {
297                 data_blob_free(out);
298                 return -1;
299         }
300         return 0;
301 }
302
303 /*
304   compare two objectGUIDs
305 */
306 static int ldif_comparison_objectGUID(struct ldb_context *ldb, void *mem_ctx,
307                                      const struct ldb_val *v1, const struct ldb_val *v2)
308 {
309         if (ldif_comparision_objectGUID_isString(v1) && ldif_comparision_objectGUID_isString(v2)) {
310                 return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
311         } else if (ldif_comparision_objectGUID_isString(v1)
312                    && !ldif_comparision_objectGUID_isString(v2)) {
313                 struct ldb_val v;
314                 int ret;
315                 if (ldif_read_objectGUID(ldb, mem_ctx, v1, &v) != 0) {
316                         /* Perhaps it wasn't a valid string after all */
317                         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
318                 }
319                 ret = ldb_comparison_binary(ldb, mem_ctx, &v, v2);
320                 talloc_free(v.data);
321                 return ret;
322         } else if (!ldif_comparision_objectGUID_isString(v1)
323                    && ldif_comparision_objectGUID_isString(v2)) {
324                 struct ldb_val v;
325                 int ret;
326                 if (ldif_read_objectGUID(ldb, mem_ctx, v2, &v) != 0) {
327                         /* Perhaps it wasn't a valid string after all */
328                         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
329                 }
330                 ret = ldb_comparison_binary(ldb, mem_ctx, v1, &v);
331                 talloc_free(v.data);
332                 return ret;
333         }
334         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
335 }
336
337 /*
338   canonicalise a objectGUID
339 */
340 static int ldif_canonicalise_objectGUID(struct ldb_context *ldb, void *mem_ctx,
341                                        const struct ldb_val *in, struct ldb_val *out)
342 {
343         if (ldif_comparision_objectGUID_isString(in)) {
344                 if (ldif_read_objectGUID(ldb, mem_ctx, in, out) != 0) {
345                         /* Perhaps it wasn't a valid string after all */
346                         return ldb_handler_copy(ldb, mem_ctx, in, out);
347                 }
348                 return 0;
349         }
350         return ldb_handler_copy(ldb, mem_ctx, in, out);
351 }
352
353
354 /*
355   convert a ldif (SDDL) formatted ntSecurityDescriptor to a NDR formatted blob
356 */
357 static int ldif_read_ntSecurityDescriptor(struct ldb_context *ldb, void *mem_ctx,
358                                           const struct ldb_val *in, struct ldb_val *out)
359 {
360         struct security_descriptor *sd;
361         enum ndr_err_code ndr_err;
362
363         sd = talloc(mem_ctx, struct security_descriptor);
364         if (sd == NULL) {
365                 return -1;
366         }
367
368         ndr_err = ndr_pull_struct_blob(in, sd, NULL, sd,
369                                        (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);
370         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
371                 /* If this does not parse, then it is probably SDDL, and we should try it that way */
372                 
373                 const struct dom_sid *sid = samdb_domain_sid(ldb);
374                 talloc_free(sd);
375                 sd = sddl_decode(mem_ctx, (const char *)in->data, sid);
376                 if (sd == NULL) {
377                         return -1;
378                 }
379         }
380
381         ndr_err = ndr_push_struct_blob(out, mem_ctx, NULL, sd,
382                                        (ndr_push_flags_fn_t)ndr_push_security_descriptor);
383         talloc_free(sd);
384         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
385                 return -1;
386         }
387
388         return 0;
389 }
390
391 /*
392   convert a NDR formatted blob to a ldif formatted ntSecurityDescriptor (SDDL format)
393 */
394 static int ldif_write_ntSecurityDescriptor(struct ldb_context *ldb, void *mem_ctx,
395                                            const struct ldb_val *in, struct ldb_val *out)
396 {
397         struct security_descriptor *sd;
398         enum ndr_err_code ndr_err;
399
400         if (ldb_get_flags(ldb) & LDB_FLG_SHOW_BINARY) {
401                 return ldif_write_NDR(ldb, mem_ctx, in, out, 
402                                       sizeof(struct security_descriptor),
403                                       (ndr_pull_flags_fn_t)ndr_pull_security_descriptor,
404                                       (ndr_print_fn_t)ndr_print_security_descriptor,
405                                       true);
406                                       
407         }
408
409         sd = talloc(mem_ctx, struct security_descriptor);
410         if (sd == NULL) {
411                 return -1;
412         }
413         /* We can't use ndr_pull_struct_blob_all because this contains relative pointers */
414         ndr_err = ndr_pull_struct_blob(in, sd, NULL, sd,
415                                            (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);
416         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
417                 talloc_free(sd);
418                 return -1;
419         }
420         out->data = (uint8_t *)sddl_encode(mem_ctx, sd, samdb_domain_sid_cache_only(ldb));
421         talloc_free(sd);
422         if (out->data == NULL) {
423                 return -1;
424         }
425         out->length = strlen((const char *)out->data);
426         return 0;
427 }
428
429 /* 
430    canonicalise an objectCategory.  We use the short form as the cannoical form:
431    cn=Person,cn=Schema,cn=Configuration,<basedn> becomes 'person'
432 */
433
434 static int ldif_canonicalise_objectCategory(struct ldb_context *ldb, void *mem_ctx,
435                                             const struct ldb_val *in, struct ldb_val *out)
436 {
437         struct ldb_dn *dn1 = NULL;
438         const struct dsdb_schema *schema = dsdb_get_schema(ldb);
439         const struct dsdb_class *sclass;
440         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
441         if (!tmp_ctx) {
442                 return LDB_ERR_OPERATIONS_ERROR;
443         }
444
445         if (!schema) {
446                 talloc_free(tmp_ctx);
447                 *out = data_blob_talloc(mem_ctx, in->data, in->length);
448                 if (in->data && !out->data) {
449                         return LDB_ERR_OPERATIONS_ERROR;
450                 }
451                 return LDB_SUCCESS;
452         }
453         dn1 = ldb_dn_from_ldb_val(tmp_ctx, ldb, in);
454         if ( ! ldb_dn_validate(dn1)) {
455                 const char *lDAPDisplayName = talloc_strndup(tmp_ctx, (char *)in->data, in->length);
456                 sclass = dsdb_class_by_lDAPDisplayName(schema, lDAPDisplayName);
457                 if (sclass) {
458                         struct ldb_dn *dn = ldb_dn_new(mem_ctx, ldb,  
459                                                        sclass->defaultObjectCategory);
460                         *out = data_blob_string_const(ldb_dn_alloc_casefold(mem_ctx, dn));
461                         talloc_free(tmp_ctx);
462
463                         if (!out->data) {
464                                 return LDB_ERR_OPERATIONS_ERROR;
465                         }
466                         return LDB_SUCCESS;
467                 } else {
468                         *out = data_blob_talloc(mem_ctx, in->data, in->length);
469                         talloc_free(tmp_ctx);
470
471                         if (in->data && !out->data) {
472                                 return LDB_ERR_OPERATIONS_ERROR;
473                         }
474                         return LDB_SUCCESS;
475                 }
476         }
477         *out = data_blob_string_const(ldb_dn_alloc_casefold(mem_ctx, dn1));
478         talloc_free(tmp_ctx);
479
480         if (!out->data) {
481                 return LDB_ERR_OPERATIONS_ERROR;
482         }
483         return LDB_SUCCESS;
484 }
485
486 static int ldif_comparison_objectCategory(struct ldb_context *ldb, void *mem_ctx,
487                                           const struct ldb_val *v1,
488                                           const struct ldb_val *v2)
489 {
490         return ldb_any_comparison(ldb, mem_ctx, ldif_canonicalise_objectCategory,
491                                   v1, v2);
492 }
493
494 /*
495   convert a ldif formatted prefixMap to a NDR formatted blob
496 */
497 static int ldif_read_prefixMap(struct ldb_context *ldb, void *mem_ctx,
498                                const struct ldb_val *in, struct ldb_val *out)
499 {
500         struct prefixMapBlob *blob;
501         enum ndr_err_code ndr_err;
502         char *string, *line, *p, *oid;
503         DATA_BLOB oid_blob;
504
505         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
506
507         if (tmp_ctx == NULL) {
508                 return -1;
509         }
510
511         blob = talloc_zero(tmp_ctx, struct prefixMapBlob);
512         if (blob == NULL) {
513                 talloc_free(blob);
514                 return -1;
515         }
516
517         blob->version = PREFIX_MAP_VERSION_DSDB;
518         
519         string = talloc_strndup(mem_ctx, (const char *)in->data, in->length);
520         if (string == NULL) {
521                 talloc_free(blob);
522                 return -1;
523         }
524
525         line = string;
526         while (line && line[0]) {
527                 p=strchr(line, ';');
528                 if (p) {
529                         p[0] = '\0';
530                 } else {
531                         p=strchr(line, '\n');
532                         if (p) {
533                                 p[0] = '\0';
534                         }
535                 }
536                 /* allow a trailing separator */
537                 if (line == p) {
538                         break;
539                 }
540                 
541                 blob->ctr.dsdb.mappings = talloc_realloc(blob, 
542                                                          blob->ctr.dsdb.mappings, 
543                                                          struct drsuapi_DsReplicaOIDMapping,
544                                                          blob->ctr.dsdb.num_mappings+1);
545                 if (!blob->ctr.dsdb.mappings) {
546                         talloc_free(tmp_ctx);
547                         return -1;
548                 }
549
550                 blob->ctr.dsdb.mappings[blob->ctr.dsdb.num_mappings].id_prefix = strtoul(line, &oid, 10);
551
552                 if (oid[0] != ':') {
553                         talloc_free(tmp_ctx);
554                         return -1;
555                 }
556
557                 /* we know there must be at least ":" */
558                 oid++;
559
560                 if (!ber_write_partial_OID_String(blob->ctr.dsdb.mappings, &oid_blob, oid)) {
561                         talloc_free(tmp_ctx);
562                         return -1;
563                 }
564                 blob->ctr.dsdb.mappings[blob->ctr.dsdb.num_mappings].oid.length = oid_blob.length;
565                 blob->ctr.dsdb.mappings[blob->ctr.dsdb.num_mappings].oid.binary_oid = oid_blob.data;
566
567                 blob->ctr.dsdb.num_mappings++;
568
569                 /* Now look past the terminator we added above */
570                 if (p) {
571                         line = p + 1;
572                 } else {
573                         line = NULL;
574                 }
575         }
576
577         ndr_err = ndr_push_struct_blob(out, mem_ctx, 
578                                        lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")), 
579                                        blob,
580                                        (ndr_push_flags_fn_t)ndr_push_prefixMapBlob);
581         talloc_free(tmp_ctx);
582         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
583                 return -1;
584         }
585         return 0;
586 }
587
588 /*
589   convert a NDR formatted blob to a ldif formatted prefixMap
590 */
591 static int ldif_write_prefixMap(struct ldb_context *ldb, void *mem_ctx,
592                                 const struct ldb_val *in, struct ldb_val *out)
593 {
594         struct prefixMapBlob *blob;
595         enum ndr_err_code ndr_err;
596         char *string;
597         uint32_t i;
598
599         if (ldb_get_flags(ldb) & LDB_FLG_SHOW_BINARY) {
600                 int err;
601                 /* try to decode the blob as S4 prefixMap */
602                 err = ldif_write_NDR(ldb, mem_ctx, in, out,
603                                      sizeof(struct prefixMapBlob),
604                                      (ndr_pull_flags_fn_t)ndr_pull_prefixMapBlob,
605                                      (ndr_print_fn_t)ndr_print_prefixMapBlob,
606                                      false);
607                 if (0 == err) {
608                         return err;
609                 }
610                 /* try parsing it as Windows PrefixMap value */
611                 return ldif_write_NDR(ldb, mem_ctx, in, out,
612                                       sizeof(struct drsuapi_MSPrefixMap_Ctr),
613                                       (ndr_pull_flags_fn_t)ndr_pull_drsuapi_MSPrefixMap_Ctr,
614                                       (ndr_print_fn_t)ndr_print_drsuapi_MSPrefixMap_Ctr,
615                                       true);
616         }
617
618         blob = talloc(mem_ctx, struct prefixMapBlob);
619         if (blob == NULL) {
620                 return -1;
621         }
622         ndr_err = ndr_pull_struct_blob_all(in, blob, 
623                                            lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")), 
624                                            blob,
625                                            (ndr_pull_flags_fn_t)ndr_pull_prefixMapBlob);
626         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
627                 goto failed;
628         }
629         if (blob->version != PREFIX_MAP_VERSION_DSDB) {
630                 goto failed;
631         }
632         string = talloc_strdup(mem_ctx, "");
633         if (string == NULL) {
634                 goto failed;
635         }
636
637         for (i=0; i < blob->ctr.dsdb.num_mappings; i++) {
638                 DATA_BLOB oid_blob;
639                 const char *partial_oid = NULL;
640
641                 if (i > 0) {
642                         string = talloc_asprintf_append(string, ";"); 
643                 }
644
645                 oid_blob = data_blob_const(blob->ctr.dsdb.mappings[i].oid.binary_oid,
646                                            blob->ctr.dsdb.mappings[i].oid.length);
647                 if (!ber_read_partial_OID_String(blob, oid_blob, &partial_oid)) {
648                         DEBUG(0, ("ber_read_partial_OID failed on prefixMap item with id: 0x%X",
649                                   blob->ctr.dsdb.mappings[i].id_prefix));
650                         goto failed;
651                 }
652                 string = talloc_asprintf_append(string, "%u:%s", 
653                                                    blob->ctr.dsdb.mappings[i].id_prefix,
654                                                    partial_oid);
655                 talloc_free(discard_const(partial_oid));
656                 if (string == NULL) {
657                         goto failed;
658                 }
659         }
660
661         talloc_free(blob);
662         *out = data_blob_string_const(string);
663         return 0;
664
665 failed:
666         talloc_free(blob);
667         return -1;
668 }
669
670 static bool ldif_comparision_prefixMap_isString(const struct ldb_val *v)
671 {
672         if (v->length < 4) {
673                 return true;
674         }
675
676         if (IVAL(v->data, 0) == PREFIX_MAP_VERSION_DSDB) {
677                 return false;
678         }
679         
680         return true;
681 }
682
683 /*
684   canonicalise a prefixMap
685 */
686 static int ldif_canonicalise_prefixMap(struct ldb_context *ldb, void *mem_ctx,
687                                        const struct ldb_val *in, struct ldb_val *out)
688 {
689         if (ldif_comparision_prefixMap_isString(in)) {
690                 return ldif_read_prefixMap(ldb, mem_ctx, in, out);
691         }
692         return ldb_handler_copy(ldb, mem_ctx, in, out);
693 }
694
695 static int ldif_comparison_prefixMap(struct ldb_context *ldb, void *mem_ctx,
696                                      const struct ldb_val *v1,
697                                      const struct ldb_val *v2)
698 {
699         return ldb_any_comparison(ldb, mem_ctx, ldif_canonicalise_prefixMap,
700                                   v1, v2);
701 }
702
703 /* length limited conversion of a ldb_val to a int32_t */
704 static int val_to_int32(const struct ldb_val *in, int32_t *v)
705 {
706         char *end;
707         char buf[64];
708
709         /* make sure we don't read past the end of the data */
710         if (in->length > sizeof(buf)-1) {
711                 return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
712         }
713         strncpy(buf, (char *)in->data, in->length);
714         buf[in->length] = 0;
715
716         /* We've to use "strtoll" here to have the intended overflows.
717          * Otherwise we may get "LONG_MAX" and the conversion is wrong. */
718         *v = (int32_t) strtoll(buf, &end, 0);
719         if (*end != 0) {
720                 return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
721         }
722         return LDB_SUCCESS;
723 }
724
725 /* length limited conversion of a ldb_val to a int64_t */
726 static int val_to_int64(const struct ldb_val *in, int64_t *v)
727 {
728         char *end;
729         char buf[64];
730
731         /* make sure we don't read past the end of the data */
732         if (in->length > sizeof(buf)-1) {
733                 return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
734         }
735         strncpy(buf, (char *)in->data, in->length);
736         buf[in->length] = 0;
737
738         *v = (int64_t) strtoll(buf, &end, 0);
739         if (*end != 0) {
740                 return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
741         }
742         return LDB_SUCCESS;
743 }
744
745 /* Canonicalisation of two 32-bit integers */
746 static int ldif_canonicalise_int32(struct ldb_context *ldb, void *mem_ctx,
747                         const struct ldb_val *in, struct ldb_val *out)
748 {
749         int32_t i;
750         int ret;
751
752         ret = val_to_int32(in, &i);
753         if (ret != LDB_SUCCESS) {
754                 return ret;
755         }
756         out->data = (uint8_t *) talloc_asprintf(mem_ctx, "%d", i);
757         if (out->data == NULL) {
758                 ldb_oom(ldb);
759                 return LDB_ERR_OPERATIONS_ERROR;
760         }
761         out->length = strlen((char *)out->data);
762         return 0;
763 }
764
765 /* Comparison of two 32-bit integers */
766 static int ldif_comparison_int32(struct ldb_context *ldb, void *mem_ctx,
767                                  const struct ldb_val *v1, const struct ldb_val *v2)
768 {
769         int32_t i1=0, i2=0;
770         val_to_int32(v1, &i1);
771         val_to_int32(v2, &i2);
772         if (i1 == i2) return 0;
773         return i1 > i2? 1 : -1;
774 }
775
776 /* Canonicalisation of two 64-bit integers */
777 static int ldif_canonicalise_int64(struct ldb_context *ldb, void *mem_ctx,
778                                    const struct ldb_val *in, struct ldb_val *out)
779 {
780         int64_t i;
781         int ret;
782
783         ret = val_to_int64(in, &i);
784         if (ret != LDB_SUCCESS) {
785                 return ret;
786         }
787         out->data = (uint8_t *) talloc_asprintf(mem_ctx, "%lld", (long long)i);
788         if (out->data == NULL) {
789                 ldb_oom(ldb);
790                 return LDB_ERR_OPERATIONS_ERROR;
791         }
792         out->length = strlen((char *)out->data);
793         return 0;
794 }
795
796 /* Comparison of two 64-bit integers */
797 static int ldif_comparison_int64(struct ldb_context *ldb, void *mem_ctx,
798                                  const struct ldb_val *v1, const struct ldb_val *v2)
799 {
800         int64_t i1=0, i2=0;
801         val_to_int64(v1, &i1);
802         val_to_int64(v2, &i2);
803         if (i1 == i2) return 0;
804         return i1 > i2? 1 : -1;
805 }
806
807 /*
808   convert a NDR formatted blob to a ldif formatted repsFromTo
809 */
810 static int ldif_write_repsFromTo(struct ldb_context *ldb, void *mem_ctx,
811                                  const struct ldb_val *in, struct ldb_val *out)
812 {
813         return ldif_write_NDR(ldb, mem_ctx, in, out, 
814                               sizeof(struct repsFromToBlob),
815                               (ndr_pull_flags_fn_t)ndr_pull_repsFromToBlob,
816                               (ndr_print_fn_t)ndr_print_repsFromToBlob,
817                               true);
818 }
819
820 /*
821   convert a NDR formatted blob to a ldif formatted replPropertyMetaData
822 */
823 static int ldif_write_replPropertyMetaData(struct ldb_context *ldb, void *mem_ctx,
824                                            const struct ldb_val *in, struct ldb_val *out)
825 {
826         return ldif_write_NDR(ldb, mem_ctx, in, out, 
827                               sizeof(struct replPropertyMetaDataBlob),
828                               (ndr_pull_flags_fn_t)ndr_pull_replPropertyMetaDataBlob,
829                               (ndr_print_fn_t)ndr_print_replPropertyMetaDataBlob,
830                               true);
831 }
832
833 /*
834   convert a NDR formatted blob to a ldif formatted replUpToDateVector
835 */
836 static int ldif_write_replUpToDateVector(struct ldb_context *ldb, void *mem_ctx,
837                                          const struct ldb_val *in, struct ldb_val *out)
838 {
839         return ldif_write_NDR(ldb, mem_ctx, in, out, 
840                               sizeof(struct replUpToDateVectorBlob),
841                               (ndr_pull_flags_fn_t)ndr_pull_replUpToDateVectorBlob,
842                               (ndr_print_fn_t)ndr_print_replUpToDateVectorBlob,
843                               true);
844 }
845
846
847 static int extended_dn_write_hex(struct ldb_context *ldb, void *mem_ctx,
848                                  const struct ldb_val *in, struct ldb_val *out)
849 {
850         *out = data_blob_string_const(data_blob_hex_string_lower(mem_ctx, in));
851         if (!out->data) {
852                 return -1;
853         }
854         return 0;
855 }
856
857
858 /*
859   write a 64 bit 2-part range
860 */
861 static int ldif_write_range64(struct ldb_context *ldb, void *mem_ctx,
862                               const struct ldb_val *in, struct ldb_val *out)
863 {
864         int64_t v;
865         int ret;
866         ret = val_to_int64(in, &v);
867         if (ret != LDB_SUCCESS) {
868                 return ret;
869         }
870         out->data = (uint8_t *)talloc_asprintf(mem_ctx, "%lu-%lu",
871                                                (unsigned long)(v&0xFFFFFFFF),
872                                                (unsigned long)(v>>32));
873         if (out->data == NULL) {
874                 ldb_oom(ldb);
875                 return LDB_ERR_OPERATIONS_ERROR;
876         }
877         out->length = strlen((char *)out->data);
878         return LDB_SUCCESS;
879 }
880
881 /*
882   read a 64 bit 2-part range
883 */
884 static int ldif_read_range64(struct ldb_context *ldb, void *mem_ctx,
885                               const struct ldb_val *in, struct ldb_val *out)
886 {
887         unsigned long high, low;
888         char buf[64];
889
890         if (memchr(in->data, '-', in->length) == NULL) {
891                 return ldb_handler_copy(ldb, mem_ctx, in, out);
892         }
893
894         if (in->length > sizeof(buf)-1) {
895                 return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
896         }
897         strncpy(buf, (const char *)in->data, in->length);
898         buf[in->length] = 0;
899
900         if (sscanf(buf, "%lu-%lu", &low, &high) != 2) {
901                 return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
902         }
903
904         out->data = (uint8_t *)talloc_asprintf(mem_ctx, "%llu",
905                                                (unsigned long long)(((uint64_t)high)<<32) | (low));
906
907         if (out->data == NULL) {
908                 ldb_oom(ldb);
909                 return LDB_ERR_OPERATIONS_ERROR;
910         }
911         out->length = strlen((char *)out->data);
912         return LDB_SUCCESS;
913 }
914
915 static const struct ldb_schema_syntax samba_syntaxes[] = {
916         {
917                 .name             = LDB_SYNTAX_SAMBA_SID,
918                 .ldif_read_fn     = ldif_read_objectSid,
919                 .ldif_write_fn    = ldif_write_objectSid,
920                 .canonicalise_fn  = ldif_canonicalise_objectSid,
921                 .comparison_fn    = ldif_comparison_objectSid
922         },{
923                 .name             = LDB_SYNTAX_SAMBA_SECURITY_DESCRIPTOR,
924                 .ldif_read_fn     = ldif_read_ntSecurityDescriptor,
925                 .ldif_write_fn    = ldif_write_ntSecurityDescriptor,
926                 .canonicalise_fn  = ldb_handler_copy,
927                 .comparison_fn    = ldb_comparison_binary
928         },{
929                 .name             = LDB_SYNTAX_SAMBA_GUID,
930                 .ldif_read_fn     = ldif_read_objectGUID,
931                 .ldif_write_fn    = ldif_write_objectGUID,
932                 .canonicalise_fn  = ldif_canonicalise_objectGUID,
933                 .comparison_fn    = ldif_comparison_objectGUID
934         },{
935                 .name             = LDB_SYNTAX_SAMBA_OBJECT_CATEGORY,
936                 .ldif_read_fn     = ldb_handler_copy,
937                 .ldif_write_fn    = ldb_handler_copy,
938                 .canonicalise_fn  = ldif_canonicalise_objectCategory,
939                 .comparison_fn    = ldif_comparison_objectCategory
940         },{
941                 .name             = LDB_SYNTAX_SAMBA_PREFIX_MAP,
942                 .ldif_read_fn     = ldif_read_prefixMap,
943                 .ldif_write_fn    = ldif_write_prefixMap,
944                 .canonicalise_fn  = ldif_canonicalise_prefixMap,
945                 .comparison_fn    = ldif_comparison_prefixMap
946         },{
947                 .name             = LDB_SYNTAX_SAMBA_INT32,
948                 .ldif_read_fn     = ldb_handler_copy,
949                 .ldif_write_fn    = ldb_handler_copy,
950                 .canonicalise_fn  = ldif_canonicalise_int32,
951                 .comparison_fn    = ldif_comparison_int32
952         },{
953                 .name             = LDB_SYNTAX_SAMBA_REPSFROMTO,
954                 .ldif_read_fn     = ldb_handler_copy,
955                 .ldif_write_fn    = ldif_write_repsFromTo,
956                 .canonicalise_fn  = ldb_handler_copy,
957                 .comparison_fn    = ldb_comparison_binary
958         },{
959                 .name             = LDB_SYNTAX_SAMBA_REPLPROPERTYMETADATA,
960                 .ldif_read_fn     = ldb_handler_copy,
961                 .ldif_write_fn    = ldif_write_replPropertyMetaData,
962                 .canonicalise_fn  = ldb_handler_copy,
963                 .comparison_fn    = ldb_comparison_binary
964         },{
965                 .name             = LDB_SYNTAX_SAMBA_REPLUPTODATEVECTOR,
966                 .ldif_read_fn     = ldb_handler_copy,
967                 .ldif_write_fn    = ldif_write_replUpToDateVector,
968                 .canonicalise_fn  = ldb_handler_copy,
969                 .comparison_fn    = ldb_comparison_binary
970         },{
971                 .name             = DSDB_SYNTAX_BINARY_DN,
972                 .ldif_read_fn     = ldb_handler_copy,
973                 .ldif_write_fn    = ldb_handler_copy,
974                 .canonicalise_fn  = dsdb_dn_binary_canonicalise,
975                 .comparison_fn    = dsdb_dn_binary_comparison
976         },{
977                 .name             = DSDB_SYNTAX_STRING_DN,
978                 .ldif_read_fn     = ldb_handler_copy,
979                 .ldif_write_fn    = ldb_handler_copy,
980                 .canonicalise_fn  = dsdb_dn_string_canonicalise,
981                 .comparison_fn    = dsdb_dn_string_comparison
982         },{
983                 .name             = LDB_SYNTAX_SAMBA_RANGE64,
984                 .ldif_read_fn     = ldif_read_range64,
985                 .ldif_write_fn    = ldif_write_range64,
986                 .canonicalise_fn  = ldif_canonicalise_int64,
987                 .comparison_fn    = ldif_comparison_int64
988         },
989 };
990
991 static const struct ldb_dn_extended_syntax samba_dn_syntax[] = {
992         {
993                 .name             = "SID",
994                 .read_fn          = extended_dn_read_SID,
995                 .write_clear_fn   = ldif_write_objectSid,
996                 .write_hex_fn     = extended_dn_write_hex
997         },{
998                 .name             = "GUID",
999                 .read_fn          = extended_dn_read_GUID,
1000                 .write_clear_fn   = ldif_write_objectGUID,
1001                 .write_hex_fn     = extended_dn_write_hex
1002         },{
1003                 .name             = "WKGUID",
1004                 .read_fn          = ldb_handler_copy,
1005                 .write_clear_fn   = ldb_handler_copy,
1006                 .write_hex_fn     = ldb_handler_copy
1007         },{
1008                 .name             = "RMD_INVOCID",
1009                 .read_fn          = extended_dn_read_GUID,
1010                 .write_clear_fn   = ldif_write_objectGUID,
1011                 .write_hex_fn     = extended_dn_write_hex
1012         },{
1013                 .name             = "RMD_FLAGS",
1014                 .read_fn          = ldb_handler_copy,
1015                 .write_clear_fn   = ldb_handler_copy,
1016                 .write_hex_fn     = ldb_handler_copy
1017         },{
1018                 .name             = "RMD_ADDTIME",
1019                 .read_fn          = ldb_handler_copy,
1020                 .write_clear_fn   = ldb_handler_copy,
1021                 .write_hex_fn     = ldb_handler_copy
1022         },{
1023                 .name             = "RMD_CHANGETIME",
1024                 .read_fn          = ldb_handler_copy,
1025                 .write_clear_fn   = ldb_handler_copy,
1026                 .write_hex_fn     = ldb_handler_copy
1027         },{
1028                 .name             = "RMD_LOCAL_USN",
1029                 .read_fn          = ldb_handler_copy,
1030                 .write_clear_fn   = ldb_handler_copy,
1031                 .write_hex_fn     = ldb_handler_copy
1032         },{
1033                 .name             = "RMD_ORIGINATING_USN",
1034                 .read_fn          = ldb_handler_copy,
1035                 .write_clear_fn   = ldb_handler_copy,
1036                 .write_hex_fn     = ldb_handler_copy
1037         },{
1038                 .name             = "RMD_VERSION",
1039                 .read_fn          = ldb_handler_copy,
1040                 .write_clear_fn   = ldb_handler_copy,
1041                 .write_hex_fn     = ldb_handler_copy
1042         }
1043 };
1044
1045 /* TODO: Should be dynamic at some point */
1046 static const struct {
1047         const char *name;
1048         const char *syntax;
1049 } samba_attributes[] = {
1050         { "objectSid",                  LDB_SYNTAX_SAMBA_SID },
1051         { "securityIdentifier",         LDB_SYNTAX_SAMBA_SID },
1052         { "ntSecurityDescriptor",       LDB_SYNTAX_SAMBA_SECURITY_DESCRIPTOR },
1053         { "objectGUID",                 LDB_SYNTAX_SAMBA_GUID },
1054         { "invocationId",               LDB_SYNTAX_SAMBA_GUID },
1055         { "schemaIDGUID",               LDB_SYNTAX_SAMBA_GUID },
1056         { "oMSyntax",                   LDB_SYNTAX_SAMBA_INT32 },
1057         { "attributeSecurityGUID",      LDB_SYNTAX_SAMBA_GUID },
1058         { "parentGUID",                 LDB_SYNTAX_SAMBA_GUID },
1059         { "siteGUID",                   LDB_SYNTAX_SAMBA_GUID },
1060         { "pKTGUID",                    LDB_SYNTAX_SAMBA_GUID },
1061         { "fRSVersionGUID",             LDB_SYNTAX_SAMBA_GUID },
1062         { "fRSReplicaSetGUID",          LDB_SYNTAX_SAMBA_GUID },
1063         { "netbootGUID",                LDB_SYNTAX_SAMBA_GUID },
1064         { "msDS-OptionalFeatureGUID",   LDB_SYNTAX_SAMBA_GUID },
1065         { "objectCategory",             LDB_SYNTAX_SAMBA_OBJECT_CATEGORY },
1066         { "prefixMap",                  LDB_SYNTAX_SAMBA_PREFIX_MAP },
1067         { "repsFrom",                   LDB_SYNTAX_SAMBA_REPSFROMTO },
1068         { "repsTo",                     LDB_SYNTAX_SAMBA_REPSFROMTO },
1069         { "replPropertyMetaData",       LDB_SYNTAX_SAMBA_REPLPROPERTYMETADATA },
1070         { "replUpToDateVector",         LDB_SYNTAX_SAMBA_REPLUPTODATEVECTOR },
1071         { "rIDAllocationPool",          LDB_SYNTAX_SAMBA_RANGE64 },
1072         { "rIDPreviousAllocationPool",  LDB_SYNTAX_SAMBA_RANGE64 },
1073         { "rIDAvailablePool",           LDB_SYNTAX_SAMBA_RANGE64 },
1074 };
1075
1076 const struct ldb_schema_syntax *ldb_samba_syntax_by_name(struct ldb_context *ldb, const char *name)
1077 {
1078         uint32_t j;
1079         const struct ldb_schema_syntax *s = NULL;
1080         
1081         for (j=0; j < ARRAY_SIZE(samba_syntaxes); j++) {
1082                 if (strcmp(name, samba_syntaxes[j].name) == 0) {
1083                         s = &samba_syntaxes[j];
1084                         break;
1085                 }
1086         }
1087         return s;
1088 }
1089
1090 const struct ldb_schema_syntax *ldb_samba_syntax_by_lDAPDisplayName(struct ldb_context *ldb, const char *name)
1091 {
1092         uint32_t j;
1093         const struct ldb_schema_syntax *s = NULL;
1094
1095         for (j=0; j < ARRAY_SIZE(samba_attributes); j++) {
1096                 if (strcmp(samba_attributes[j].name, name) == 0) {
1097                         s = ldb_samba_syntax_by_name(ldb, samba_attributes[j].syntax);
1098                         break;
1099                 }
1100         }
1101         
1102         return s;
1103 }
1104
1105 /*
1106   register the samba ldif handlers
1107 */
1108 int ldb_register_samba_handlers(struct ldb_context *ldb)
1109 {
1110         uint32_t i;
1111
1112         for (i=0; i < ARRAY_SIZE(samba_attributes); i++) {
1113                 int ret;
1114                 const struct ldb_schema_syntax *s = NULL;
1115
1116                 s = ldb_samba_syntax_by_name(ldb, samba_attributes[i].syntax);
1117
1118                 if (!s) {
1119                         s = ldb_standard_syntax_by_name(ldb, samba_attributes[i].syntax);
1120                 }
1121
1122                 if (!s) {
1123                         return -1;
1124                 }
1125
1126                 ret = ldb_schema_attribute_add_with_syntax(ldb, samba_attributes[i].name, LDB_ATTR_FLAG_FIXED, s);
1127                 if (ret != LDB_SUCCESS) {
1128                         return ret;
1129                 }
1130         }
1131
1132         for (i=0; i < ARRAY_SIZE(samba_dn_syntax); i++) {
1133                 int ret;
1134                 ret = ldb_dn_extended_add_syntax(ldb, LDB_ATTR_FLAG_FIXED, &samba_dn_syntax[i]);
1135                 if (ret != LDB_SUCCESS) {
1136                         return ret;
1137                 }
1138
1139                 
1140         }
1141
1142         return LDB_SUCCESS;
1143 }