source4/dsdb/samdb/ldb_modules/entryUUID.c

   1 /*
   2    ldb database module
   3
   4    LDAP semantics mapping module
   5
   6    Copyright (C) Jelmer Vernooij 2005
   7    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2006
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 /*
  24    This module relies on ldb_map to do all the real work, but performs
  25    some of the trivial mappings between AD semantics and that provided
  26    by OpenLDAP and similar servers.
  27 */
  28
  29 #include "includes.h"
  30 #include "ldb/include/ldb.h"
  31 #include "ldb/include/ldb_private.h"
  32 #include "ldb/include/ldb_errors.h"
  33 #include "ldb/ldb_map/ldb_map.h"
  34
  35 #include "librpc/gen_ndr/ndr_misc.h"
  36 #include "librpc/ndr/libndr.h"
  37
  38 struct entryUUID_private {
  39         struct ldb_dn **base_dns;
  40 };
  41
  42 static struct ldb_val encode_guid(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
  43 {
  44         struct GUID guid;
  45         NTSTATUS status = GUID_from_string((char *)val->data, &guid);
  46         enum ndr_err_code ndr_err;
  47         struct ldb_val out = data_blob(NULL, 0);
  48
  49         if (!NT_STATUS_IS_OK(status)) {
  50                 return out;
  51         }
  52         ndr_err = ndr_push_struct_blob(&out, ctx, &guid,
  53                                        (ndr_push_flags_fn_t)ndr_push_GUID);
  54         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
  55                 return out;
  56         }
  57
  58         return out;
  59 }
  60
  61 static struct ldb_val guid_always_string(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
  62 {
  63         struct GUID *guid;
  64         struct ldb_val out = data_blob(NULL, 0);
  65         if (val->length >= 32 && val->data[val->length] == '\0') {
  66                 ldb_handler_copy(module->ldb, ctx, val, &out);
  67         } else {
  68                 enum ndr_err_code ndr_err;
  69
  70                 guid = talloc(ctx, struct GUID);
  71                 if (guid == NULL) {
  72                         return out;
  73                 }
  74                 ndr_err = ndr_pull_struct_blob(val, guid, guid,
  75                                                (ndr_pull_flags_fn_t)ndr_pull_GUID);
  76                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
  77                         talloc_free(guid);
  78                         return out;
  79                 }
  80                 out = data_blob_string_const(GUID_string(ctx, guid));
  81                 talloc_free(guid);
  82         }
  83         return out;
  84 }
  85
  86 static struct ldb_val encode_ns_guid(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
  87 {
  88         struct GUID guid;
  89         NTSTATUS status = NS_GUID_from_string((char *)val->data, &guid);
  90         enum ndr_err_code ndr_err;
  91         struct ldb_val out = data_blob(NULL, 0);
  92
  93         if (!NT_STATUS_IS_OK(status)) {
  94                 return out;
  95         }
  96         ndr_err = ndr_push_struct_blob(&out, ctx, &guid,
  97                                        (ndr_push_flags_fn_t)ndr_push_GUID);
  98         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
  99                 return out;
 100         }
 101
 102         return out;
 103 }
 104
 105 static struct ldb_val guid_ns_string(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 106 {
 107         struct ldb_val out = data_blob(NULL, 0);
 108         if (val->length >= 32 && val->data[val->length] == '\0') {
 109                 struct GUID guid;
 110                 GUID_from_string((char *)val->data, &guid);
 111                 out = data_blob_string_const(NS_GUID_string(ctx, &guid));
 112         } else {
 113                 enum ndr_err_code ndr_err;
 114                 struct GUID *guid_p;
 115                 guid_p = talloc(ctx, struct GUID);
 116                 if (guid_p == NULL) {
 117                         return out;
 118                 }
 119                 ndr_err = ndr_pull_struct_blob(val, guid_p, guid_p,
 120                                                (ndr_pull_flags_fn_t)ndr_pull_GUID);
 121                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 122                         talloc_free(guid_p);
 123                         return out;
 124                 }
 125                 out = data_blob_string_const(NS_GUID_string(ctx, guid_p));
 126                 talloc_free(guid_p);
 127         }
 128         return out;
 129 }
 130
 131 /* The backend holds binary sids, so just copy them back */
 132 static struct ldb_val val_copy(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 133 {
 134         struct ldb_val out = data_blob(NULL, 0);
 135         ldb_handler_copy(module->ldb, ctx, val, &out);
 136
 137         return out;
 138 }
 139
 140 /* Ensure we always convert sids into binary, so the backend doesn't have to know about both forms */
 141 static struct ldb_val sid_always_binary(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 142 {
 143         struct ldb_val out = data_blob(NULL, 0);
 144         const struct ldb_schema_attribute *a = ldb_schema_attribute_by_name(module->ldb, "objectSid");
 145
 146         if (a->syntax->canonicalise_fn(module->ldb, ctx, val, &out) != LDB_SUCCESS) {
 147                 return data_blob(NULL, 0);
 148         }
 149
 150         return out;
 151 }
 152
 153 /* Ensure we always convert objectCategory into a DN */
 154 static struct ldb_val objectCategory_always_dn(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 155 {
 156         struct ldb_val out = data_blob(NULL, 0);
 157         const struct ldb_schema_attribute *a = ldb_schema_attribute_by_name(module->ldb, "objectCategory");
 158
 159         if (a->syntax->canonicalise_fn(module->ldb, ctx, val, &out) != LDB_SUCCESS) {
 160                 return data_blob(NULL, 0);
 161         }
 162
 163         return out;
 164 }
 165
 166 static struct ldb_val normalise_to_signed32(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 167 {
 168         long long int signed_ll = strtoll((const char *)val->data, NULL, 10);
 169         if (signed_ll >= 0x80000000LL) {
 170                 union {
 171                         int32_t signed_int;
 172                         uint32_t unsigned_int;
 173                 } u = {
 174                         .unsigned_int = strtoul((const char *)val->data, NULL, 10)
 175                 };
 176
 177                 struct ldb_val out = data_blob_string_const(talloc_asprintf(ctx, "%d", u.signed_int));
 178                 return out;
 179         }
 180         return val_copy(module, ctx, val);
 181 }
 182
 183 static struct ldb_val usn_to_entryCSN(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 184 {
 185         struct ldb_val out;
 186         unsigned long long usn = strtoull((const char *)val->data, NULL, 10);
 187         time_t t = (usn >> 24);
 188         out = data_blob_string_const(talloc_asprintf(ctx, "%s#%06x#00#000000", ldb_timestring(ctx, t), (unsigned int)(usn & 0xFFFFFF)));
 189         return out;
 190 }
 191
 192 static unsigned long long entryCSN_to_usn_int(TALLOC_CTX *ctx, const struct ldb_val *val)
 193 {
 194         char *entryCSN = talloc_strdup(ctx, (const char *)val->data);
 195         char *mod_per_sec;
 196         time_t t;
 197         unsigned long long usn;
 198         char *p;
 199         if (!entryCSN) {
 200                 return 0;
 201         }
 202         p = strchr(entryCSN, '#');
 203         if (!p) {
 204                 return 0;
 205         }
 206         p[0] = '\0';
 207         p++;
 208         mod_per_sec = p;
 209
 210         p = strchr(p, '#');
 211         if (!p) {
 212                 return 0;
 213         }
 214         p[0] = '\0';
 215         p++;
 216
 217         usn = strtol(mod_per_sec, NULL, 16);
 218
 219         t = ldb_string_to_time(entryCSN);
 220
 221         usn = usn | ((unsigned long long)t <<24);
 222         return usn;
 223 }
 224
 225 static struct ldb_val entryCSN_to_usn(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 226 {
 227         struct ldb_val out;
 228         unsigned long long usn = entryCSN_to_usn_int(ctx, val);
 229         out = data_blob_string_const(talloc_asprintf(ctx, "%lld", usn));
 230         return out;
 231 }
 232
 233 static struct ldb_val usn_to_timestamp(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 234 {
 235         struct ldb_val out;
 236         unsigned long long usn = strtoull((const char *)val->data, NULL, 10);
 237         time_t t = (usn >> 24);
 238         out = data_blob_string_const(ldb_timestring(ctx, t));
 239         return out;
 240 }
 241
 242 static struct ldb_val timestamp_to_usn(struct ldb_module *module, TALLOC_CTX *ctx, const struct ldb_val *val)
 243 {
 244         struct ldb_val out;
 245         time_t t;
 246         unsigned long long usn;
 247
 248         t = ldb_string_to_time((const char *)val->data);
 249
 250         usn = ((unsigned long long)t <<24);
 251
 252         out = data_blob_string_const(talloc_asprintf(ctx, "%lld", usn));
 253         return out;
 254 }
 255
 256
 257 static const struct ldb_map_attribute entryUUID_attributes[] =
 258 {
 259         /* objectGUID */
 260         {
 261                 .local_name = "objectGUID",
 262                 .type = MAP_CONVERT,
 263                 .u = {
 264                         .convert = {
 265                                 .remote_name = "entryUUID",
 266                                 .convert_local = guid_always_string,
 267                                 .convert_remote = encode_guid,
 268                         },
 269                 },
 270         },
 271         /* invocationId */
 272         {
 273                 .local_name = "invocationId",
 274                 .type = MAP_CONVERT,
 275                 .u = {
 276                         .convert = {
 277                                 .remote_name = "invocationId",
 278                                 .convert_local = guid_always_string,
 279                                 .convert_remote = encode_guid,
 280                         },
 281                 },
 282         },
 283         /* objectSid */
 284         {
 285                 .local_name = "objectSid",
 286                 .type = MAP_CONVERT,
 287                 .u = {
 288                         .convert = {
 289                                 .remote_name = "objectSid",
 290                                 .convert_local = sid_always_binary,
 291                                 .convert_remote = val_copy,
 292                         },
 293                 },
 294         },
 295         {
 296                 .local_name = "whenCreated",
 297                 .type = MAP_RENAME,
 298                 .u = {
 299                         .rename = {
 300                                  .remote_name = "createTimestamp"
 301                          }
 302                 }
 303         },
 304         {
 305                 .local_name = "whenChanged",
 306                 .type = MAP_RENAME,
 307                 .u = {
 308                         .rename = {
 309                                  .remote_name = "modifyTimestamp"
 310                          }
 311                 }
 312         },
 313         {
 314                 .local_name = "objectClasses",
 315                 .type = MAP_RENAME,
 316                 .u = {
 317                         .rename = {
 318                                  .remote_name = "samba4ObjectClasses"
 319                          }
 320                 }
 321         },
 322         {
 323                 .local_name = "dITContentRules",
 324                 .type = MAP_RENAME,
 325                 .u = {
 326                         .rename = {
 327                                  .remote_name = "samba4DITContentRules"
 328                          }
 329                 }
 330         },
 331         {
 332                 .local_name = "attributeTypes",
 333                 .type = MAP_RENAME,
 334                 .u = {
 335                         .rename = {
 336                                  .remote_name = "samba4AttributeTypes"
 337                          }
 338                 }
 339         },
 340         {
 341                 .local_name = "sambaPassword",
 342                 .type = MAP_RENAME,
 343                 .u = {
 344                         .rename = {
 345                                  .remote_name = "userPassword"
 346                          }
 347                 }
 348         },
 349         {
 350                 .local_name = "objectCategory",
 351                 .type = MAP_CONVERT,
 352                 .u = {
 353                         .convert = {
 354                                 .remote_name = "objectCategory",
 355                                 .convert_local = objectCategory_always_dn,
 356                                 .convert_remote = val_copy,
 357                         },
 358                 },
 359         },
 360         {
 361                 .local_name = "distinguishedName",
 362                 .type = MAP_RENAME,
 363                 .u = {
 364                         .rename = {
 365                                  .remote_name = "entryDN"
 366                          }
 367                 }
 368         },
 369         {
 370                 .local_name = "groupType",
 371                 .type = MAP_CONVERT,
 372                 .u = {
 373                         .convert = {
 374                                  .remote_name = "groupType",
 375                                  .convert_local = normalise_to_signed32,
 376                                  .convert_remote = val_copy,
 377                          },
 378                 }
 379         },
 380         {
 381                 .local_name = "sAMAccountType",
 382                 .type = MAP_CONVERT,
 383                 .u = {
 384                         .convert = {
 385                                  .remote_name = "sAMAccountType",
 386                                  .convert_local = normalise_to_signed32,
 387                                  .convert_remote = val_copy,
 388                          },
 389                 }
 390         },
 391         {
 392                 .local_name = "usnChanged",
 393                 .type = MAP_CONVERT,
 394                 .u = {
 395                         .convert = {
 396                                  .remote_name = "entryCSN",
 397                                  .convert_local = usn_to_entryCSN,
 398                                  .convert_remote = entryCSN_to_usn
 399                          },
 400                 },
 401         },
 402         {
 403                 .local_name = "usnCreated",
 404                 .type = MAP_CONVERT,
 405                 .u = {
 406                         .convert = {
 407                                  .remote_name = "createTimestamp",
 408                                  .convert_local = usn_to_timestamp,
 409                                  .convert_remote = timestamp_to_usn,
 410                          },
 411                 },
 412         },
 413         {
 414                 .local_name = "*",
 415                 .type = MAP_KEEP,
 416         },
 417         {
 418                 .local_name = NULL,
 419         }
 420 };
 421
 422 /* This objectClass conflicts with builtin classes on OpenLDAP */
 423 const struct ldb_map_objectclass entryUUID_objectclasses[] =
 424 {
 425         {
 426                 .local_name = "subSchema",
 427                 .remote_name = "samba4SubSchema"
 428         },
 429         {
 430                 .local_name = NULL
 431         }
 432 };
 433
 434 /* These things do not show up in wildcard searches in OpenLDAP, but
 435  * we need them to show up in the AD-like view */
 436 static const char * const entryUUID_wildcard_attributes[] = {
 437         "objectGUID",
 438         "whenCreated",
 439         "whenChanged",
 440         "usnCreated",
 441         "usnChanged",
 442         NULL
 443 };
 444
 445 static const struct ldb_map_attribute nsuniqueid_attributes[] =
 446 {
 447         /* objectGUID */
 448         {
 449                 .local_name = "objectGUID",
 450                 .type = MAP_CONVERT,
 451                 .u = {
 452                         .convert = {
 453                                 .remote_name = "nsuniqueid",
 454                                 .convert_local = guid_ns_string,
 455                                 .convert_remote = encode_ns_guid,
 456                         },
 457                 },
 458         },
 459         /* objectSid */
 460         {
 461                 .local_name = "objectSid",
 462                 .type = MAP_CONVERT,
 463                 .u = {
 464                         .convert = {
 465                                 .remote_name = "objectSid",
 466                                 .convert_local = sid_always_binary,
 467                                 .convert_remote = val_copy,
 468                         },
 469                 },
 470         },
 471         {
 472                 .local_name = "whenCreated",
 473                 .type = MAP_RENAME,
 474                 .u = {
 475                         .rename = {
 476                                  .remote_name = "createTimestamp"
 477                          }
 478                 }
 479         },
 480         {
 481                 .local_name = "whenChanged",
 482                 .type = MAP_RENAME,
 483                 .u = {
 484                         .rename = {
 485                                  .remote_name = "modifyTimestamp"
 486                          }
 487                 }
 488         },
 489         {
 490                 .local_name = "sambaPassword",
 491                 .type = MAP_RENAME,
 492                 .u = {
 493                         .rename = {
 494                                  .remote_name = "userPassword"
 495                          }
 496                 }
 497         },
 498         {
 499                 .local_name = "objectCategory",
 500                 .type = MAP_CONVERT,
 501                 .u = {
 502                         .convert = {
 503                                 .remote_name = "objectCategory",
 504                                 .convert_local = objectCategory_always_dn,
 505                                 .convert_remote = val_copy,
 506                         },
 507                 },
 508         },
 509         {
 510                 .local_name = "distinguishedName",
 511                 .type = MAP_RENAME,
 512                 .u = {
 513                         .rename = {
 514                                  .remote_name = "entryDN"
 515                          }
 516                 }
 517         },
 518         {
 519                 .local_name = "groupType",
 520                 .type = MAP_CONVERT,
 521                 .u = {
 522                         .convert = {
 523                                  .remote_name = "groupType",
 524                                  .convert_local = normalise_to_signed32,
 525                                  .convert_remote = val_copy,
 526                          },
 527                 }
 528         },
 529         {
 530                 .local_name = "sAMAccountType",
 531                 .type = MAP_CONVERT,
 532                 .u = {
 533                         .convert = {
 534                                  .remote_name = "sAMAccountType",
 535                                  .convert_local = normalise_to_signed32,
 536                                  .convert_remote = val_copy,
 537                          },
 538                 }
 539         },
 540         {
 541                 .local_name = "usnChanged",
 542                 .type = MAP_CONVERT,
 543                 .u = {
 544                         .convert = {
 545                                  .remote_name = "modifyTimestamp",
 546                                  .convert_local = usn_to_timestamp,
 547                                  .convert_remote = timestamp_to_usn,
 548                          },
 549                 },
 550         },
 551         {
 552                 .local_name = "usnCreated",
 553                 .type = MAP_CONVERT,
 554                 .u = {
 555                         .convert = {
 556                                  .remote_name = "createTimestamp",
 557                                  .convert_local = usn_to_timestamp,
 558                                  .convert_remote = timestamp_to_usn,
 559                          },
 560                 },
 561         },
 562         {
 563                 .local_name = "*",
 564                 .type = MAP_KEEP,
 565         },
 566         {
 567                 .local_name = NULL,
 568         }
 569 };
 570
 571 /* These things do not show up in wildcard searches in OpenLDAP, but
 572  * we need them to show up in the AD-like view */
 573 static const char * const nsuniqueid_wildcard_attributes[] = {
 574         "objectGUID",
 575         "whenCreated",
 576         "whenChanged",
 577         "usnCreated",
 578         "usnChanged",
 579         NULL
 580 };
 581
 582 static int get_remote_rootdse(struct ldb_context *ldb, void *context,
 583                        struct ldb_reply *ares)
 584 {
 585         struct entryUUID_private *entryUUID_private;
 586         entryUUID_private = talloc_get_type(context,
 587                                             struct entryUUID_private);
 588         if (ares->type == LDB_REPLY_ENTRY) {
 589                 int i;
 590                 struct ldb_message_element *el = ldb_msg_find_element(ares->message, "namingContexts");
 591                 entryUUID_private->base_dns = talloc_realloc(entryUUID_private, entryUUID_private->base_dns, struct ldb_dn *,
 592                                                              el->num_values + 1);
 593                 for (i=0; i < el->num_values; i++) {
 594                         if (!entryUUID_private->base_dns) {
 595                                 return LDB_ERR_OPERATIONS_ERROR;
 596                         }
 597                         entryUUID_private->base_dns[i] = ldb_dn_new(entryUUID_private->base_dns, ldb, (const char *)el->values[i].data);
 598                         if ( ! ldb_dn_validate(entryUUID_private->base_dns[i])) {
 599                                 return LDB_ERR_OPERATIONS_ERROR;
 600                         }
 601                 }
 602                 entryUUID_private->base_dns[i] = NULL;
 603         }
 604
 605         return LDB_SUCCESS;
 606 }
 607
 608 static int find_base_dns(struct ldb_module *module,
 609                           struct entryUUID_private *entryUUID_private)
 610 {
 611         int ret;
 612         struct ldb_request *req;
 613         const char *naming_context_attr[] = {
 614                 "namingContexts",
 615                 NULL
 616         };
 617         req = talloc(entryUUID_private, struct ldb_request);
 618         if (req == NULL) {
 619                 ldb_set_errstring(module->ldb, "Out of Memory");
 620                 return LDB_ERR_OPERATIONS_ERROR;
 621         }
 622
 623         req->operation = LDB_SEARCH;
 624         req->op.search.base = ldb_dn_new(req, module->ldb, NULL);
 625         req->op.search.scope = LDB_SCOPE_BASE;
 626
 627         req->op.search.tree = ldb_parse_tree(req, "objectClass=*");
 628         if (req->op.search.tree == NULL) {
 629                 ldb_set_errstring(module->ldb, "Unable to parse search expression");
 630                 talloc_free(req);
 631                 return LDB_ERR_OPERATIONS_ERROR;
 632         }
 633
 634         req->op.search.attrs = naming_context_attr;
 635         req->controls = NULL;
 636         req->context = entryUUID_private;
 637         req->callback = get_remote_rootdse;
 638         ldb_set_timeout(module->ldb, req, 0); /* use default timeout */
 639
 640         ret = ldb_next_request(module, req);
 641
 642         if (ret == LDB_SUCCESS) {
 643                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
 644         }
 645
 646         talloc_free(req);
 647         if (ret != LDB_SUCCESS) {
 648                 return ret;
 649         }
 650
 651         return LDB_SUCCESS;
 652 }
 653
 654 /* the context init function */
 655 static int entryUUID_init(struct ldb_module *module)
 656 {
 657         int ret;
 658         struct map_private *map_private;
 659         struct entryUUID_private *entryUUID_private;
 660
 661         ret = ldb_map_init(module, entryUUID_attributes, entryUUID_objectclasses, entryUUID_wildcard_attributes, NULL);
 662         if (ret != LDB_SUCCESS)
 663                 return ret;
 664
 665         map_private = talloc_get_type(module->private_data, struct map_private);
 666
 667         entryUUID_private = talloc_zero(map_private, struct entryUUID_private);
 668         map_private->caller_private = entryUUID_private;
 669
 670         ret = find_base_dns(module, entryUUID_private);
 671
 672         return ldb_next_init(module);
 673 }
 674
 675 /* the context init function */
 676 static int nsuniqueid_init(struct ldb_module *module)
 677 {
 678         int ret;
 679         struct map_private *map_private;
 680         struct entryUUID_private *entryUUID_private;
 681
 682         ret = ldb_map_init(module, nsuniqueid_attributes, NULL, nsuniqueid_wildcard_attributes, NULL);
 683         if (ret != LDB_SUCCESS)
 684                 return ret;
 685
 686         map_private = talloc_get_type(module->private_data, struct map_private);
 687
 688         entryUUID_private = talloc_zero(map_private, struct entryUUID_private);
 689         map_private->caller_private = entryUUID_private;
 690
 691         ret = find_base_dns(module, entryUUID_private);
 692
 693         return ldb_next_init(module);
 694 }
 695
 696 static int get_seq(struct ldb_context *ldb, void *context,
 697                    struct ldb_reply *ares)
 698 {
 699         unsigned long long *max_seq = (unsigned long long *)context;
 700         unsigned long long seq;
 701         if (ares->type == LDB_REPLY_ENTRY) {
 702                 struct ldb_message_element *el = ldb_msg_find_element(ares->message, "contextCSN");
 703                 if (el) {
 704                         seq = entryCSN_to_usn_int(ares, &el->values[0]);
 705                         *max_seq = MAX(seq, *max_seq);
 706                 }
 707         }
 708
 709         return LDB_SUCCESS;
 710 }
 711
 712 static int entryUUID_sequence_number(struct ldb_module *module, struct ldb_request *req)
 713 {
 714         int i, ret;
 715         struct map_private *map_private;
 716         struct entryUUID_private *entryUUID_private;
 717         unsigned long long max_seq = 0;
 718         struct ldb_request *search_req;
 719         map_private = talloc_get_type(module->private_data, struct map_private);
 720
 721         entryUUID_private = talloc_get_type(map_private->caller_private, struct entryUUID_private);
 722
 723         /* Search the baseDNs for a sequence number */
 724         for (i=0; entryUUID_private &&
 725                      entryUUID_private->base_dns &&
 726                      entryUUID_private->base_dns[i];
 727                 i++) {
 728                 static const char *contextCSN_attr[] = {
 729                         "contextCSN", NULL
 730                 };
 731                 search_req = talloc(req, struct ldb_request);
 732                 if (search_req == NULL) {
 733                         ldb_set_errstring(module->ldb, "Out of Memory");
 734                         return LDB_ERR_OPERATIONS_ERROR;
 735                 }
 736
 737                 search_req->operation = LDB_SEARCH;
 738                 search_req->op.search.base = entryUUID_private->base_dns[i];
 739                 search_req->op.search.scope = LDB_SCOPE_BASE;
 740
 741                 search_req->op.search.tree = ldb_parse_tree(search_req, "objectClass=*");
 742                 if (search_req->op.search.tree == NULL) {
 743                         ldb_set_errstring(module->ldb, "Unable to parse search expression");
 744                         talloc_free(search_req);
 745                         return LDB_ERR_OPERATIONS_ERROR;
 746                 }
 747
 748                 search_req->op.search.attrs = contextCSN_attr;
 749                 search_req->controls = NULL;
 750                 search_req->context = &max_seq;
 751                 search_req->callback = get_seq;
 752                 ldb_set_timeout(module->ldb, search_req, 0); /* use default timeout */
 753
 754                 ret = ldb_next_request(module, search_req);
 755
 756                 if (ret == LDB_SUCCESS) {
 757                         ret = ldb_wait(search_req->handle, LDB_WAIT_ALL);
 758                 }
 759
 760                 talloc_free(search_req);
 761                 if (ret != LDB_SUCCESS) {
 762                         return ret;
 763                 }
 764         }
 765
 766         switch (req->op.seq_num.type) {
 767         case LDB_SEQ_HIGHEST_SEQ:
 768                 req->op.seq_num.seq_num = max_seq;
 769                 break;
 770         case LDB_SEQ_NEXT:
 771                 req->op.seq_num.seq_num = max_seq;
 772                 req->op.seq_num.seq_num++;
 773                 break;
 774         case LDB_SEQ_HIGHEST_TIMESTAMP:
 775         {
 776                 req->op.seq_num.seq_num = (max_seq >> 24);
 777                 break;
 778         }
 779         }
 780         req->op.seq_num.flags = 0;
 781         req->op.seq_num.flags |= LDB_SEQ_TIMESTAMP_SEQUENCE;
 782         req->op.seq_num.flags |= LDB_SEQ_GLOBAL_SEQUENCE;
 783         return LDB_SUCCESS;
 784 }
 785
 786 static struct ldb_module_ops entryUUID_ops = {
 787         .name              = "entryUUID",
 788         .init_context      = entryUUID_init,
 789         .sequence_number   = entryUUID_sequence_number
 790 };
 791
 792 static struct ldb_module_ops nsuniqueid_ops = {
 793         .name              = "nsuniqueid",
 794         .init_context      = nsuniqueid_init,
 795         .sequence_number   = entryUUID_sequence_number
 796 };
 797
 798 /* the init function */
 799 int ldb_entryUUID_module_init(void)
 800 {
 801         int ret;
 802         struct ldb_module_ops ops = ldb_map_get_ops();
 803         entryUUID_ops.add       = ops.add;
 804         entryUUID_ops.modify    = ops.modify;
 805         entryUUID_ops.del       = ops.del;
 806         entryUUID_ops.rename    = ops.rename;
 807         entryUUID_ops.search    = ops.search;
 808         entryUUID_ops.wait      = ops.wait;
 809         ret = ldb_register_module(&entryUUID_ops);
 810
 811         if (ret) {
 812                 return ret;
 813         }
 814
 815         nsuniqueid_ops.add      = ops.add;
 816         nsuniqueid_ops.modify   = ops.modify;
 817         nsuniqueid_ops.del      = ops.del;
 818         nsuniqueid_ops.rename   = ops.rename;
 819         nsuniqueid_ops.search   = ops.search;
 820         nsuniqueid_ops.wait     = ops.wait;
 821         ret = ldb_register_module(&nsuniqueid_ops);
 822
 823         return ret;
 824 }