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-2007
   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
  37 /*
  38   use ndr_print_* to convert a NDR formatted blob to a ldif formatted blob
  39 */
  40 static int ldif_write_NDR(struct ldb_context *ldb, void *mem_ctx,
  41                           const struct ldb_val *in, struct ldb_val *out,
  42                           size_t struct_size,
  43                           ndr_pull_flags_fn_t pull_fn,
  44                           ndr_print_fn_t print_fn)
  45 {
  46         uint8_t *p;
  47         enum ndr_err_code err;
  48         if (!(ldb_get_flags(ldb) & LDB_FLG_SHOW_BINARY)) {
  49                 return ldb_handler_copy(ldb, mem_ctx, in, out);
  50         }
  51         p = talloc_size(mem_ctx, struct_size);
  52         err = ndr_pull_struct_blob(in, mem_ctx,
  53                                    lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")),
  54                                    p, pull_fn);
  55         if (err != NDR_ERR_SUCCESS) {
  56                 talloc_free(p);
  57                 out->data = (uint8_t *)talloc_strdup(mem_ctx, "<Unable to decode binary data>");
  58                 out->length = strlen((const char *)out->data);
  59                 return 0;
  60         }
  61         out->data = (uint8_t *)ndr_print_struct_string(mem_ctx, print_fn, "NDR", p);
  62         talloc_free(p);
  63         if (out->data == NULL) {
  64                 return ldb_handler_copy(ldb, mem_ctx, in, out);
  65         }
  66         out->length = strlen((char *)out->data);
  67         return 0;
  68 }
  69
  70 /*
  71   convert a ldif formatted objectSid to a NDR formatted blob
  72 */
  73 static int ldif_read_objectSid(struct ldb_context *ldb, void *mem_ctx,
  74                                const struct ldb_val *in, struct ldb_val *out)
  75 {
  76         enum ndr_err_code ndr_err;
  77         struct dom_sid *sid;
  78         sid = dom_sid_parse_length(mem_ctx, in);
  79         if (sid == NULL) {
  80                 return -1;
  81         }
  82         ndr_err = ndr_push_struct_blob(out, mem_ctx, NULL, sid,
  83                                        (ndr_push_flags_fn_t)ndr_push_dom_sid);
  84         talloc_free(sid);
  85         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
  86                 return -1;
  87         }
  88         return 0;
  89 }
  90
  91 /*
  92   convert a NDR formatted blob to a ldif formatted objectSid
  93 */
  94 static int ldif_write_objectSid(struct ldb_context *ldb, void *mem_ctx,
  95                                 const struct ldb_val *in, struct ldb_val *out)
  96 {
  97         struct dom_sid *sid;
  98         enum ndr_err_code ndr_err;
  99
 100         sid = talloc(mem_ctx, struct dom_sid);
 101         if (sid == NULL) {
 102                 return -1;
 103         }
 104         ndr_err = ndr_pull_struct_blob_all(in, sid, NULL, sid,
 105                                            (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
 106         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 107                 talloc_free(sid);
 108                 return -1;
 109         }
 110         *out = data_blob_string_const(dom_sid_string(mem_ctx, sid));
 111         talloc_free(sid);
 112         if (out->data == NULL) {
 113                 return -1;
 114         }
 115         return 0;
 116 }
 117
 118 static bool ldif_comparision_objectSid_isString(const struct ldb_val *v)
 119 {
 120         if (v->length < 3) {
 121                 return false;
 122         }
 123
 124         if (strncmp("S-", (const char *)v->data, 2) != 0) return false;
 125
 126         return true;
 127 }
 128
 129 /*
 130   compare two objectSids
 131 */
 132 static int ldif_comparison_objectSid(struct ldb_context *ldb, void *mem_ctx,
 133                                     const struct ldb_val *v1, const struct ldb_val *v2)
 134 {
 135         if (ldif_comparision_objectSid_isString(v1) && ldif_comparision_objectSid_isString(v2)) {
 136                 return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
 137         } else if (ldif_comparision_objectSid_isString(v1)
 138                    && !ldif_comparision_objectSid_isString(v2)) {
 139                 struct ldb_val v;
 140                 int ret;
 141                 if (ldif_read_objectSid(ldb, mem_ctx, v1, &v) != 0) {
 142                         /* Perhaps not a string after all */
 143                         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
 144                 }
 145                 ret = ldb_comparison_binary(ldb, mem_ctx, &v, v2);
 146                 talloc_free(v.data);
 147                 return ret;
 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, v2, &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, v1, &v);
 157                 talloc_free(v.data);
 158                 return ret;
 159         }
 160         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
 161 }
 162
 163 /*
 164   canonicalise a objectSid
 165 */
 166 static int ldif_canonicalise_objectSid(struct ldb_context *ldb, void *mem_ctx,
 167                                       const struct ldb_val *in, struct ldb_val *out)
 168 {
 169         if (ldif_comparision_objectSid_isString(in)) {
 170                 if (ldif_read_objectSid(ldb, mem_ctx, in, out) != 0) {
 171                         /* Perhaps not a string after all */
 172                         return ldb_handler_copy(ldb, mem_ctx, in, out);
 173                 }
 174                 return 0;
 175         }
 176         return ldb_handler_copy(ldb, mem_ctx, in, out);
 177 }
 178
 179 static int extended_dn_read_SID(struct ldb_context *ldb, void *mem_ctx,
 180                               const struct ldb_val *in, struct ldb_val *out)
 181 {
 182         struct dom_sid sid;
 183         enum ndr_err_code ndr_err;
 184         TALLOC_CTX *tmp_ctx;
 185         if (ldif_comparision_objectSid_isString(in)) {
 186                 if (ldif_read_objectSid(ldb, mem_ctx, in, out) == 0) {
 187                         return 0;
 188                 }
 189         }
 190
 191         tmp_ctx = talloc_new(mem_ctx);
 192         if (!tmp_ctx) {
 193                 return -1;
 194         }
 195
 196         /* Perhaps not a string after all */
 197         *out = data_blob_talloc(tmp_ctx, NULL, in->length/2+1);
 198
 199         if (!out->data) {
 200                 talloc_free(tmp_ctx);
 201                 return -1;
 202         }
 203
 204         out->length = strhex_to_str((char *)out->data, out->length,
 205                                     (const char *)in->data, in->length);
 206
 207         /* Check it looks like a SID */
 208         ndr_err = ndr_pull_struct_blob_all(out, tmp_ctx, NULL, &sid,
 209                                            (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
 210         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 211                 talloc_free(tmp_ctx);
 212                 return -1;
 213         }
 214         talloc_steal(mem_ctx, out->data);
 215         talloc_free(tmp_ctx);
 216         return 0;
 217 }
 218
 219 /*
 220   convert a ldif formatted objectGUID to a NDR formatted blob
 221 */
 222 static int ldif_read_objectGUID(struct ldb_context *ldb, void *mem_ctx,
 223                                 const struct ldb_val *in, struct ldb_val *out)
 224 {
 225         struct GUID guid;
 226         NTSTATUS status;
 227         enum ndr_err_code ndr_err;
 228         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 229         if (!tmp_ctx) {
 230                 return -1;
 231         }
 232
 233         status = GUID_from_data_blob(in, &guid);
 234         if (!NT_STATUS_IS_OK(status)) {
 235                 return -1;
 236         }
 237
 238         ndr_err = ndr_push_struct_blob(out, tmp_ctx, NULL, &guid,
 239                                        (ndr_push_flags_fn_t)ndr_push_GUID);
 240         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 241                 talloc_free(tmp_ctx);
 242                 return -1;
 243         }
 244         talloc_steal(mem_ctx, out->data);
 245         talloc_free(tmp_ctx);
 246         return 0;
 247 }
 248
 249 /*
 250   convert a NDR formatted blob to a ldif formatted objectGUID
 251 */
 252 static int ldif_write_objectGUID(struct ldb_context *ldb, void *mem_ctx,
 253                                  const struct ldb_val *in, struct ldb_val *out)
 254 {
 255         struct GUID guid;
 256         enum ndr_err_code ndr_err;
 257         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 258         if (!tmp_ctx) {
 259                 return -1;
 260         }
 261         ndr_err = ndr_pull_struct_blob_all(in, tmp_ctx, NULL, &guid,
 262                                            (ndr_pull_flags_fn_t)ndr_pull_GUID);
 263         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 264                 talloc_free(tmp_ctx);
 265                 return -1;
 266         }
 267         out->data = (uint8_t *)GUID_string(mem_ctx, &guid);
 268         talloc_free(tmp_ctx);
 269         if (out->data == NULL) {
 270                 return -1;
 271         }
 272         out->length = strlen((const char *)out->data);
 273         return 0;
 274 }
 275
 276 static bool ldif_comparision_objectGUID_isString(const struct ldb_val *v)
 277 {
 278         if (v->length != 36 && v->length != 38) return false;
 279
 280         /* Might be a GUID string, can't be a binary GUID (fixed 16 bytes) */
 281         return true;
 282 }
 283
 284 static int extended_dn_read_GUID(struct ldb_context *ldb, void *mem_ctx,
 285                               const struct ldb_val *in, struct ldb_val *out)
 286 {
 287         struct GUID guid;
 288         enum ndr_err_code ndr_err;
 289         TALLOC_CTX *tmp_ctx;
 290         if (in->length == 36 && ldif_read_objectGUID(ldb, mem_ctx, in, out) == 0) {
 291                 return 0;
 292         }
 293
 294         /* Try as 'hex' form */
 295         if (in->length != 32) {
 296                 return -1;
 297         }
 298
 299         tmp_ctx = talloc_new(mem_ctx);
 300         if (!tmp_ctx) {
 301                 return -1;
 302         }
 303
 304         *out = data_blob_talloc(tmp_ctx, NULL, in->length/2+1);
 305
 306         if (!out->data) {
 307                 talloc_free(tmp_ctx);
 308                 return -1;
 309         }
 310
 311         out->length = strhex_to_str((char *)out->data, out->length,
 312                                     (const char *)in->data, in->length);
 313
 314         /* Check it looks like a GUID */
 315         ndr_err = ndr_pull_struct_blob_all(out, tmp_ctx, NULL, &guid,
 316                                            (ndr_pull_flags_fn_t)ndr_pull_GUID);
 317         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 318                 talloc_free(tmp_ctx);
 319                 return -1;
 320         }
 321         talloc_steal(mem_ctx, out->data);
 322         talloc_free(tmp_ctx);
 323         return 0;
 324 }
 325
 326 /*
 327   compare two objectGUIDs
 328 */
 329 static int ldif_comparison_objectGUID(struct ldb_context *ldb, void *mem_ctx,
 330                                      const struct ldb_val *v1, const struct ldb_val *v2)
 331 {
 332         if (ldif_comparision_objectGUID_isString(v1) && ldif_comparision_objectGUID_isString(v2)) {
 333                 return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
 334         } else if (ldif_comparision_objectGUID_isString(v1)
 335                    && !ldif_comparision_objectGUID_isString(v2)) {
 336                 struct ldb_val v;
 337                 int ret;
 338                 if (ldif_read_objectGUID(ldb, mem_ctx, v1, &v) != 0) {
 339                         /* Perhaps it wasn't a valid string after all */
 340                         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
 341                 }
 342                 ret = ldb_comparison_binary(ldb, mem_ctx, &v, v2);
 343                 talloc_free(v.data);
 344                 return ret;
 345         } else if (!ldif_comparision_objectGUID_isString(v1)
 346                    && ldif_comparision_objectGUID_isString(v2)) {
 347                 struct ldb_val v;
 348                 int ret;
 349                 if (ldif_read_objectGUID(ldb, mem_ctx, v2, &v) != 0) {
 350                         /* Perhaps it wasn't a valid string after all */
 351                         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
 352                 }
 353                 ret = ldb_comparison_binary(ldb, mem_ctx, v1, &v);
 354                 talloc_free(v.data);
 355                 return ret;
 356         }
 357         return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
 358 }
 359
 360 /*
 361   canonicalise a objectGUID
 362 */
 363 static int ldif_canonicalise_objectGUID(struct ldb_context *ldb, void *mem_ctx,
 364                                        const struct ldb_val *in, struct ldb_val *out)
 365 {
 366         if (ldif_comparision_objectGUID_isString(in)) {
 367                 if (ldif_read_objectGUID(ldb, mem_ctx, in, out) != 0) {
 368                         /* Perhaps it wasn't a valid string after all */
 369                         return ldb_handler_copy(ldb, mem_ctx, in, out);
 370                 }
 371                 return 0;
 372         }
 373         return ldb_handler_copy(ldb, mem_ctx, in, out);
 374 }
 375
 376
 377 /*
 378   convert a ldif (SDDL) formatted ntSecurityDescriptor to a NDR formatted blob
 379 */
 380 static int ldif_read_ntSecurityDescriptor(struct ldb_context *ldb, void *mem_ctx,
 381                                           const struct ldb_val *in, struct ldb_val *out)
 382 {
 383         struct security_descriptor *sd;
 384         enum ndr_err_code ndr_err;
 385
 386         sd = talloc(mem_ctx, struct security_descriptor);
 387         if (sd == NULL) {
 388                 return -1;
 389         }
 390
 391         ndr_err = ndr_pull_struct_blob(in, sd, NULL, sd,
 392                                        (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);
 393         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 394                 /* If this does not parse, then it is probably SDDL, and we should try it that way */
 395
 396                 const struct dom_sid *sid = samdb_domain_sid(ldb);
 397                 talloc_free(sd);
 398                 sd = sddl_decode(mem_ctx, (const char *)in->data, sid);
 399                 if (sd == NULL) {
 400                         return -1;
 401                 }
 402         }
 403
 404         ndr_err = ndr_push_struct_blob(out, sd, NULL, sd,
 405                                        (ndr_push_flags_fn_t)ndr_push_security_descriptor);
 406         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 407                 talloc_free(sd);
 408                 return -1;
 409         }
 410         talloc_steal(mem_ctx, out->data);
 411         talloc_free(sd);
 412
 413         return 0;
 414 }
 415
 416 /*
 417   convert a NDR formatted blob to a ldif formatted ntSecurityDescriptor (SDDL format)
 418 */
 419 static int ldif_write_ntSecurityDescriptor(struct ldb_context *ldb, void *mem_ctx,
 420                                            const struct ldb_val *in, struct ldb_val *out)
 421 {
 422         struct security_descriptor *sd;
 423         enum ndr_err_code ndr_err;
 424
 425         if (ldb_get_flags(ldb) & LDB_FLG_SHOW_BINARY) {
 426                 return ldif_write_NDR(ldb, mem_ctx, in, out,
 427                                       sizeof(struct security_descriptor),
 428                                       (ndr_pull_flags_fn_t)ndr_pull_security_descriptor,
 429                                       (ndr_print_fn_t)ndr_print_security_descriptor);
 430
 431         }
 432
 433         sd = talloc(mem_ctx, struct security_descriptor);
 434         if (sd == NULL) {
 435                 return -1;
 436         }
 437         /* We can't use ndr_pull_struct_blob_all because this contains relative pointers */
 438         ndr_err = ndr_pull_struct_blob(in, sd, NULL, sd,
 439                                            (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);
 440         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 441                 talloc_free(sd);
 442                 return -1;
 443         }
 444         out->data = (uint8_t *)sddl_encode(mem_ctx, sd, NULL);
 445         talloc_free(sd);
 446         if (out->data == NULL) {
 447                 return -1;
 448         }
 449         out->length = strlen((const char *)out->data);
 450         return 0;
 451 }
 452
 453 /*
 454    canonicalise an objectCategory.  We use the short form as the cannoical form:
 455    cn=Person,cn=Schema,cn=Configuration,<basedn> becomes 'person'
 456 */
 457
 458 static int ldif_canonicalise_objectCategory(struct ldb_context *ldb, void *mem_ctx,
 459                                             const struct ldb_val *in, struct ldb_val *out)
 460 {
 461         struct ldb_dn *dn1 = NULL;
 462         const struct dsdb_schema *schema = dsdb_get_schema(ldb);
 463         const struct dsdb_class *sclass;
 464         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 465         if (!tmp_ctx) {
 466                 return LDB_ERR_OPERATIONS_ERROR;
 467         }
 468
 469         if (!schema) {
 470                 talloc_free(tmp_ctx);
 471                 *out = data_blob_talloc(mem_ctx, in->data, in->length);
 472                 if (in->data && !out->data) {
 473                         return LDB_ERR_OPERATIONS_ERROR;
 474                 }
 475                 return LDB_SUCCESS;
 476         }
 477         dn1 = ldb_dn_from_ldb_val(tmp_ctx, ldb, in);
 478         if ( ! ldb_dn_validate(dn1)) {
 479                 const char *lDAPDisplayName = talloc_strndup(tmp_ctx, (char *)in->data, in->length);
 480                 sclass = dsdb_class_by_lDAPDisplayName(schema, lDAPDisplayName);
 481                 if (sclass) {
 482                         struct ldb_dn *dn = ldb_dn_new(mem_ctx, ldb,
 483                                                        sclass->defaultObjectCategory);
 484                         *out = data_blob_string_const(ldb_dn_alloc_casefold(mem_ctx, dn));
 485                         talloc_free(tmp_ctx);
 486
 487                         if (!out->data) {
 488                                 return LDB_ERR_OPERATIONS_ERROR;
 489                         }
 490                         return LDB_SUCCESS;
 491                 } else {
 492                         *out = data_blob_talloc(mem_ctx, in->data, in->length);
 493                         talloc_free(tmp_ctx);
 494
 495                         if (in->data && !out->data) {
 496                                 return LDB_ERR_OPERATIONS_ERROR;
 497                         }
 498                         return LDB_SUCCESS;
 499                 }
 500         }
 501         *out = data_blob_string_const(ldb_dn_alloc_casefold(mem_ctx, dn1));
 502         talloc_free(tmp_ctx);
 503
 504         if (!out->data) {
 505                 return LDB_ERR_OPERATIONS_ERROR;
 506         }
 507         return LDB_SUCCESS;
 508 }
 509
 510 static int ldif_comparison_objectCategory(struct ldb_context *ldb, void *mem_ctx,
 511                                           const struct ldb_val *v1,
 512                                           const struct ldb_val *v2)
 513 {
 514
 515         int ret, ret1, ret2;
 516         struct ldb_val v1_canon, v2_canon;
 517         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 518
 519         /* I could try and bail if tmp_ctx was NULL, but what return
 520          * value would I use?
 521          *
 522          * It seems easier to continue on the NULL context
 523          */
 524         ret1 = ldif_canonicalise_objectCategory(ldb, tmp_ctx, v1, &v1_canon);
 525         ret2 = ldif_canonicalise_objectCategory(ldb, tmp_ctx, v2, &v2_canon);
 526
 527         if (ret1 == LDB_SUCCESS && ret2 == LDB_SUCCESS) {
 528                 ret = data_blob_cmp(&v1_canon, &v2_canon);
 529         } else {
 530                 ret = data_blob_cmp(v1, v2);
 531         }
 532         talloc_free(tmp_ctx);
 533         return ret;
 534 }
 535
 536 /*
 537   convert a ldif formatted prefixMap to a NDR formatted blob
 538 */
 539 static int ldif_read_prefixMap(struct ldb_context *ldb, void *mem_ctx,
 540                                const struct ldb_val *in, struct ldb_val *out)
 541 {
 542         struct prefixMapBlob *blob;
 543         enum ndr_err_code ndr_err;
 544         char *string, *line, *p, *oid;
 545
 546         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 547
 548         if (tmp_ctx == NULL) {
 549                 return -1;
 550         }
 551
 552         blob = talloc_zero(tmp_ctx, struct prefixMapBlob);
 553         if (blob == NULL) {
 554                 talloc_free(blob);
 555                 return -1;
 556         }
 557
 558         blob->version = PREFIX_MAP_VERSION_DSDB;
 559
 560         string = talloc_strndup(mem_ctx, (const char *)in->data, in->length);
 561         if (string == NULL) {
 562                 talloc_free(blob);
 563                 return -1;
 564         }
 565
 566         line = string;
 567         while (line && line[0]) {
 568                 p=strchr(line, ';');
 569                 if (p) {
 570                         p[0] = '\0';
 571                 } else {
 572                         p=strchr(line, '\n');
 573                         if (p) {
 574                                 p[0] = '\0';
 575                         }
 576                 }
 577                 /* allow a traling seperator */
 578                 if (line == p) {
 579                         break;
 580                 }
 581
 582                 blob->ctr.dsdb.mappings = talloc_realloc(blob,
 583                                                          blob->ctr.dsdb.mappings,
 584                                                          struct drsuapi_DsReplicaOIDMapping,
 585                                                          blob->ctr.dsdb.num_mappings+1);
 586                 if (!blob->ctr.dsdb.mappings) {
 587                         talloc_free(tmp_ctx);
 588                         return -1;
 589                 }
 590
 591                 blob->ctr.dsdb.mappings[blob->ctr.dsdb.num_mappings].id_prefix = strtoul(line, &oid, 10);
 592
 593                 if (oid[0] != ':') {
 594                         talloc_free(tmp_ctx);
 595                         return -1;
 596                 }
 597
 598                 /* we know there must be at least ":" */
 599                 oid++;
 600
 601                 blob->ctr.dsdb.mappings[blob->ctr.dsdb.num_mappings].oid.oid
 602                         = talloc_strdup(blob->ctr.dsdb.mappings, oid);
 603
 604                 blob->ctr.dsdb.num_mappings++;
 605
 606                 /* Now look past the terminator we added above */
 607                 if (p) {
 608                         line = p + 1;
 609                 } else {
 610                         line = NULL;
 611                 }
 612         }
 613
 614         ndr_err = ndr_push_struct_blob(out, mem_ctx,
 615                                        lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")),
 616                                        blob,
 617                                        (ndr_push_flags_fn_t)ndr_push_prefixMapBlob);
 618         talloc_free(tmp_ctx);
 619         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 620                 return -1;
 621         }
 622         return 0;
 623 }
 624
 625 /*
 626   convert a NDR formatted blob to a ldif formatted prefixMap
 627 */
 628 static int ldif_write_prefixMap(struct ldb_context *ldb, void *mem_ctx,
 629                                 const struct ldb_val *in, struct ldb_val *out)
 630 {
 631         struct prefixMapBlob *blob;
 632         enum ndr_err_code ndr_err;
 633         char *string;
 634         uint32_t i;
 635
 636         if (ldb_get_flags(ldb) & LDB_FLG_SHOW_BINARY) {
 637                 return ldif_write_NDR(ldb, mem_ctx, in, out,
 638                                       sizeof(struct prefixMapBlob),
 639                                       (ndr_pull_flags_fn_t)ndr_pull_prefixMapBlob,
 640                                       (ndr_print_fn_t)ndr_print_prefixMapBlob);
 641
 642         }
 643
 644         blob = talloc(mem_ctx, struct prefixMapBlob);
 645         if (blob == NULL) {
 646                 return -1;
 647         }
 648         ndr_err = ndr_pull_struct_blob_all(in, blob,
 649                                            lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")),
 650                                            blob,
 651                                            (ndr_pull_flags_fn_t)ndr_pull_prefixMapBlob);
 652         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 653                 talloc_free(blob);
 654                 return -1;
 655         }
 656         if (blob->version != PREFIX_MAP_VERSION_DSDB) {
 657                 return -1;
 658         }
 659         string = talloc_strdup(mem_ctx, "");
 660         if (string == NULL) {
 661                 return -1;
 662         }
 663
 664         for (i=0; i < blob->ctr.dsdb.num_mappings; i++) {
 665                 if (i > 0) {
 666                         string = talloc_asprintf_append(string, ";");
 667                 }
 668                 string = talloc_asprintf_append(string, "%u:%s",
 669                                                    blob->ctr.dsdb.mappings[i].id_prefix,
 670                                                    blob->ctr.dsdb.mappings[i].oid.oid);
 671                 if (string == NULL) {
 672                         return -1;
 673                 }
 674         }
 675
 676         talloc_free(blob);
 677         *out = data_blob_string_const(string);
 678         return 0;
 679 }
 680
 681 static bool ldif_comparision_prefixMap_isString(const struct ldb_val *v)
 682 {
 683         if (v->length < 4) {
 684                 return true;
 685         }
 686
 687         if (IVAL(v->data, 0) == PREFIX_MAP_VERSION_DSDB) {
 688                 return false;
 689         }
 690
 691         return true;
 692 }
 693
 694 /*
 695   canonicalise a prefixMap
 696 */
 697 static int ldif_canonicalise_prefixMap(struct ldb_context *ldb, void *mem_ctx,
 698                                        const struct ldb_val *in, struct ldb_val *out)
 699 {
 700         if (ldif_comparision_prefixMap_isString(in)) {
 701                 return ldif_read_prefixMap(ldb, mem_ctx, in, out);
 702         }
 703         return ldb_handler_copy(ldb, mem_ctx, in, out);
 704 }
 705
 706 static int ldif_comparison_prefixMap(struct ldb_context *ldb, void *mem_ctx,
 707                                      const struct ldb_val *v1,
 708                                      const struct ldb_val *v2)
 709 {
 710
 711         int ret, ret1, ret2;
 712         struct ldb_val v1_canon, v2_canon;
 713         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 714
 715         /* I could try and bail if tmp_ctx was NULL, but what return
 716          * value would I use?
 717          *
 718          * It seems easier to continue on the NULL context
 719          */
 720         ret1 = ldif_canonicalise_prefixMap(ldb, tmp_ctx, v1, &v1_canon);
 721         ret2 = ldif_canonicalise_prefixMap(ldb, tmp_ctx, v2, &v2_canon);
 722
 723         if (ret1 == LDB_SUCCESS && ret2 == LDB_SUCCESS) {
 724                 ret = data_blob_cmp(&v1_canon, &v2_canon);
 725         } else {
 726                 ret = data_blob_cmp(v1, v2);
 727         }
 728         talloc_free(tmp_ctx);
 729         return ret;
 730 }
 731
 732 /* Canonicalisation of two 32-bit integers */
 733 static int ldif_canonicalise_int32(struct ldb_context *ldb, void *mem_ctx,
 734                         const struct ldb_val *in, struct ldb_val *out)
 735 {
 736         char *end;
 737         /* We've to use "strtoll" here to have the intended overflows.
 738          * Otherwise we may get "LONG_MAX" and the conversion is wrong. */
 739         int32_t i = (int32_t) strtoll((char *)in->data, &end, 0);
 740         if (*end != 0) {
 741                 return -1;
 742         }
 743         out->data = (uint8_t *) talloc_asprintf(mem_ctx, "%d", i);
 744         if (out->data == NULL) {
 745                 return -1;
 746         }
 747         out->length = strlen((char *)out->data);
 748         return 0;
 749 }
 750
 751 /* Comparison of two 32-bit integers */
 752 static int ldif_comparison_int32(struct ldb_context *ldb, void *mem_ctx,
 753                         const struct ldb_val *v1, const struct ldb_val *v2)
 754 {
 755         /* We've to use "strtoll" here to have the intended overflows.
 756          * Otherwise we may get "LONG_MAX" and the conversion is wrong. */
 757         return (int32_t) strtoll((char *)v1->data, NULL, 0)
 758          - (int32_t) strtoll((char *)v2->data, NULL, 0);
 759 }
 760
 761 /*
 762   convert a NDR formatted blob to a ldif formatted repsFromTo
 763 */
 764 static int ldif_write_repsFromTo(struct ldb_context *ldb, void *mem_ctx,
 765                                  const struct ldb_val *in, struct ldb_val *out)
 766 {
 767         return ldif_write_NDR(ldb, mem_ctx, in, out,
 768                               sizeof(struct repsFromToBlob),
 769                               (ndr_pull_flags_fn_t)ndr_pull_repsFromToBlob,
 770                               (ndr_print_fn_t)ndr_print_repsFromToBlob);
 771 }
 772
 773 /*
 774   convert a NDR formatted blob to a ldif formatted replPropertyMetaData
 775 */
 776 static int ldif_write_replPropertyMetaData(struct ldb_context *ldb, void *mem_ctx,
 777                                            const struct ldb_val *in, struct ldb_val *out)
 778 {
 779         return ldif_write_NDR(ldb, mem_ctx, in, out,
 780                               sizeof(struct replPropertyMetaDataBlob),
 781                               (ndr_pull_flags_fn_t)ndr_pull_replPropertyMetaDataBlob,
 782                               (ndr_print_fn_t)ndr_print_replPropertyMetaDataBlob);
 783 }
 784
 785 /*
 786   convert a NDR formatted blob to a ldif formatted replUpToDateVector
 787 */
 788 static int ldif_write_replUpToDateVector(struct ldb_context *ldb, void *mem_ctx,
 789                                          const struct ldb_val *in, struct ldb_val *out)
 790 {
 791         return ldif_write_NDR(ldb, mem_ctx, in, out,
 792                               sizeof(struct replUpToDateVectorBlob),
 793                               (ndr_pull_flags_fn_t)ndr_pull_replUpToDateVectorBlob,
 794                               (ndr_print_fn_t)ndr_print_replPropertyMetaDataBlob);
 795 }
 796
 797
 798 static int extended_dn_write_hex(struct ldb_context *ldb, void *mem_ctx,
 799                                  const struct ldb_val *in, struct ldb_val *out)
 800 {
 801         *out = data_blob_string_const(data_blob_hex_string(mem_ctx, in));
 802         if (!out->data) {
 803                 return -1;
 804         }
 805         return 0;
 806 }
 807
 808 static const struct ldb_schema_syntax samba_syntaxes[] = {
 809         {
 810                 .name             = LDB_SYNTAX_SAMBA_SID,
 811                 .ldif_read_fn     = ldif_read_objectSid,
 812                 .ldif_write_fn    = ldif_write_objectSid,
 813                 .canonicalise_fn  = ldif_canonicalise_objectSid,
 814                 .comparison_fn    = ldif_comparison_objectSid
 815         },{
 816                 .name             = LDB_SYNTAX_SAMBA_SECURITY_DESCRIPTOR,
 817                 .ldif_read_fn     = ldif_read_ntSecurityDescriptor,
 818                 .ldif_write_fn    = ldif_write_ntSecurityDescriptor,
 819                 .canonicalise_fn  = ldb_handler_copy,
 820                 .comparison_fn    = ldb_comparison_binary
 821         },{
 822                 .name             = LDB_SYNTAX_SAMBA_GUID,
 823                 .ldif_read_fn     = ldif_read_objectGUID,
 824                 .ldif_write_fn    = ldif_write_objectGUID,
 825                 .canonicalise_fn  = ldif_canonicalise_objectGUID,
 826                 .comparison_fn    = ldif_comparison_objectGUID
 827         },{
 828                 .name             = LDB_SYNTAX_SAMBA_OBJECT_CATEGORY,
 829                 .ldif_read_fn     = ldb_handler_copy,
 830                 .ldif_write_fn    = ldb_handler_copy,
 831                 .canonicalise_fn  = ldif_canonicalise_objectCategory,
 832                 .comparison_fn    = ldif_comparison_objectCategory
 833         },{
 834                 .name             = LDB_SYNTAX_SAMBA_PREFIX_MAP,
 835                 .ldif_read_fn     = ldif_read_prefixMap,
 836                 .ldif_write_fn    = ldif_write_prefixMap,
 837                 .canonicalise_fn  = ldif_canonicalise_prefixMap,
 838                 .comparison_fn    = ldif_comparison_prefixMap
 839         },{
 840                 .name             = LDB_SYNTAX_SAMBA_INT32,
 841                 .ldif_read_fn     = ldb_handler_copy,
 842                 .ldif_write_fn    = ldb_handler_copy,
 843                 .canonicalise_fn  = ldif_canonicalise_int32,
 844                 .comparison_fn    = ldif_comparison_int32
 845         },{
 846                 .name             = LDB_SYNTAX_SAMBA_REPSFROMTO,
 847                 .ldif_read_fn     = ldb_handler_copy,
 848                 .ldif_write_fn    = ldif_write_repsFromTo,
 849                 .canonicalise_fn  = ldb_handler_copy,
 850                 .comparison_fn    = ldb_comparison_binary
 851         },{
 852                 .name             = LDB_SYNTAX_SAMBA_REPLPROPERTYMETADATA,
 853                 .ldif_read_fn     = ldb_handler_copy,
 854                 .ldif_write_fn    = ldif_write_replPropertyMetaData,
 855                 .canonicalise_fn  = ldb_handler_copy,
 856                 .comparison_fn    = ldb_comparison_binary
 857         },{
 858                 .name             = LDB_SYNTAX_SAMBA_REPLUPTODATEVECTOR,
 859                 .ldif_read_fn     = ldb_handler_copy,
 860                 .ldif_write_fn    = ldif_write_replUpToDateVector,
 861                 .canonicalise_fn  = ldb_handler_copy,
 862                 .comparison_fn    = ldb_comparison_binary
 863         },
 864 };
 865
 866 static const struct ldb_dn_extended_syntax samba_dn_syntax[] = {
 867         {
 868                 .name             = "SID",
 869                 .read_fn          = extended_dn_read_SID,
 870                 .write_clear_fn   = ldif_write_objectSid,
 871                 .write_hex_fn     = extended_dn_write_hex
 872         },{
 873                 .name             = "GUID",
 874                 .read_fn          = extended_dn_read_GUID,
 875                 .write_clear_fn   = ldif_write_objectGUID,
 876                 .write_hex_fn     = extended_dn_write_hex
 877         },{
 878                 .name             = "WKGUID",
 879                 .read_fn          = ldb_handler_copy,
 880                 .write_clear_fn   = ldb_handler_copy,
 881                 .write_hex_fn     = ldb_handler_copy
 882         }
 883 };
 884
 885 /* TODO: Should be dynamic at some point */
 886 static const struct {
 887         const char *name;
 888         const char *syntax;
 889 } samba_attributes[] = {
 890         { "objectSid",                  LDB_SYNTAX_SAMBA_SID },
 891         { "securityIdentifier",         LDB_SYNTAX_SAMBA_SID },
 892         { "ntSecurityDescriptor",       LDB_SYNTAX_SAMBA_SECURITY_DESCRIPTOR },
 893         { "objectGUID",                 LDB_SYNTAX_SAMBA_GUID },
 894         { "invocationId",               LDB_SYNTAX_SAMBA_GUID },
 895         { "schemaIDGUID",               LDB_SYNTAX_SAMBA_GUID },
 896         { "attributeSecurityGUID",      LDB_SYNTAX_SAMBA_GUID },
 897         { "parentGUID",                 LDB_SYNTAX_SAMBA_GUID },
 898         { "siteGUID",                   LDB_SYNTAX_SAMBA_GUID },
 899         { "pKTGUID",                    LDB_SYNTAX_SAMBA_GUID },
 900         { "fRSVersionGUID",             LDB_SYNTAX_SAMBA_GUID },
 901         { "fRSReplicaSetGUID",          LDB_SYNTAX_SAMBA_GUID },
 902         { "netbootGUID",                LDB_SYNTAX_SAMBA_GUID },
 903         { "objectCategory",             LDB_SYNTAX_SAMBA_OBJECT_CATEGORY },
 904         { "prefixMap",                  LDB_SYNTAX_SAMBA_PREFIX_MAP },
 905         { "repsFrom",                   LDB_SYNTAX_SAMBA_REPSFROMTO },
 906         { "repsTo",                     LDB_SYNTAX_SAMBA_REPSFROMTO },
 907         { "replPropertyMetaData",       LDB_SYNTAX_SAMBA_REPLPROPERTYMETADATA },
 908         { "replUpToDateVector",         LDB_SYNTAX_SAMBA_REPLUPTODATEVECTOR },
 909 };
 910
 911 const struct ldb_schema_syntax *ldb_samba_syntax_by_name(struct ldb_context *ldb, const char *name)
 912 {
 913         uint32_t j;
 914         const struct ldb_schema_syntax *s = NULL;
 915
 916         for (j=0; j < ARRAY_SIZE(samba_syntaxes); j++) {
 917                 if (strcmp(name, samba_syntaxes[j].name) == 0) {
 918                         s = &samba_syntaxes[j];
 919                         break;
 920                 }
 921         }
 922         return s;
 923 }
 924
 925 const struct ldb_schema_syntax *ldb_samba_syntax_by_lDAPDisplayName(struct ldb_context *ldb, const char *name)
 926 {
 927         uint32_t j;
 928         const struct ldb_schema_syntax *s = NULL;
 929
 930         for (j=0; j < ARRAY_SIZE(samba_attributes); j++) {
 931                 if (strcmp(samba_attributes[j].name, name) == 0) {
 932                         s = ldb_samba_syntax_by_name(ldb, samba_attributes[j].syntax);
 933                         break;
 934                 }
 935         }
 936
 937         return s;
 938 }
 939
 940 /*
 941   register the samba ldif handlers
 942 */
 943 int ldb_register_samba_handlers(struct ldb_context *ldb)
 944 {
 945         uint32_t i;
 946
 947         for (i=0; i < ARRAY_SIZE(samba_attributes); i++) {
 948                 int ret;
 949                 const struct ldb_schema_syntax *s = NULL;
 950
 951                 s = ldb_samba_syntax_by_name(ldb, samba_attributes[i].syntax);
 952
 953                 if (!s) {
 954                         s = ldb_standard_syntax_by_name(ldb, samba_attributes[i].syntax);
 955                 }
 956
 957                 if (!s) {
 958                         return -1;
 959                 }
 960
 961                 ret = ldb_schema_attribute_add_with_syntax(ldb, samba_attributes[i].name, LDB_ATTR_FLAG_FIXED, s);
 962                 if (ret != LDB_SUCCESS) {
 963                         return ret;
 964                 }
 965         }
 966
 967         for (i=0; i < ARRAY_SIZE(samba_dn_syntax); i++) {
 968                 int ret;
 969                 ret = ldb_dn_extended_add_syntax(ldb, LDB_ATTR_FLAG_FIXED, &samba_dn_syntax[i]);
 970                 if (ret != LDB_SUCCESS) {
 971                         return ret;
 972                 }
 973
 974
 975         }
 976
 977         return LDB_SUCCESS;
 978 }