source4/dsdb/common/util.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4
   5    Copyright (C) Andrew Tridgell 2004
   6    Copyright (C) Volker Lendecke 2004
   7    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2006
   8    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
   9
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 3 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20    You should have received a copy of the GNU General Public License
  21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #include "includes.h"
  25 #include "events/events.h"
  26 #include "ldb.h"
  27 #include "ldb_errors.h"
  28 #include "../lib/util/util_ldb.h"
  29 #include "../lib/crypto/crypto.h"
  30 #include "dsdb/samdb/samdb.h"
  31 #include "libcli/security/security.h"
  32 #include "librpc/gen_ndr/ndr_security.h"
  33 #include "librpc/gen_ndr/ndr_misc.h"
  34 #include "../libds/common/flags.h"
  35 #include "dsdb/common/proto.h"
  36 #include "libcli/ldap/ldap_ndr.h"
  37 #include "param/param.h"
  38 #include "libcli/auth/libcli_auth.h"
  39 #include "librpc/gen_ndr/ndr_drsblobs.h"
  40 #include "system/locale.h"
  41
  42 /*
  43   search the sam for the specified attributes in a specific domain, filter on
  44   objectSid being in domain_sid.
  45 */
  46 int samdb_search_domain(struct ldb_context *sam_ldb,
  47                         TALLOC_CTX *mem_ctx,
  48                         struct ldb_dn *basedn,
  49                         struct ldb_message ***res,
  50                         const char * const *attrs,
  51                         const struct dom_sid *domain_sid,
  52                         const char *format, ...)  _PRINTF_ATTRIBUTE(7,8)
  53 {
  54         va_list ap;
  55         int i, count;
  56
  57         va_start(ap, format);
  58         count = gendb_search_v(sam_ldb, mem_ctx, basedn,
  59                                res, attrs, format, ap);
  60         va_end(ap);
  61
  62         i=0;
  63
  64         while (i<count) {
  65                 struct dom_sid *entry_sid;
  66
  67                 entry_sid = samdb_result_dom_sid(mem_ctx, (*res)[i], "objectSid");
  68
  69                 if ((entry_sid == NULL) ||
  70                     (!dom_sid_in_domain(domain_sid, entry_sid))) {
  71                         /* Delete that entry from the result set */
  72                         (*res)[i] = (*res)[count-1];
  73                         count -= 1;
  74                         talloc_free(entry_sid);
  75                         continue;
  76                 }
  77                 talloc_free(entry_sid);
  78                 i += 1;
  79         }
  80
  81         return count;
  82 }
  83
  84 /*
  85   search the sam for a single string attribute in exactly 1 record
  86 */
  87 const char *samdb_search_string_v(struct ldb_context *sam_ldb,
  88                                   TALLOC_CTX *mem_ctx,
  89                                   struct ldb_dn *basedn,
  90                                   const char *attr_name,
  91                                   const char *format, va_list ap) _PRINTF_ATTRIBUTE(5,0)
  92 {
  93         int count;
  94         const char *attrs[2] = { NULL, NULL };
  95         struct ldb_message **res = NULL;
  96
  97         attrs[0] = attr_name;
  98
  99         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
 100         if (count > 1) {
 101                 DEBUG(1,("samdb: search for %s %s not single valued (count=%d)\n",
 102                          attr_name, format, count));
 103         }
 104         if (count != 1) {
 105                 talloc_free(res);
 106                 return NULL;
 107         }
 108
 109         return samdb_result_string(res[0], attr_name, NULL);
 110 }
 111
 112 /*
 113   search the sam for a single string attribute in exactly 1 record
 114 */
 115 const char *samdb_search_string(struct ldb_context *sam_ldb,
 116                                 TALLOC_CTX *mem_ctx,
 117                                 struct ldb_dn *basedn,
 118                                 const char *attr_name,
 119                                 const char *format, ...) _PRINTF_ATTRIBUTE(5,6)
 120 {
 121         va_list ap;
 122         const char *str;
 123
 124         va_start(ap, format);
 125         str = samdb_search_string_v(sam_ldb, mem_ctx, basedn, attr_name, format, ap);
 126         va_end(ap);
 127
 128         return str;
 129 }
 130
 131 struct ldb_dn *samdb_search_dn(struct ldb_context *sam_ldb,
 132                                TALLOC_CTX *mem_ctx,
 133                                struct ldb_dn *basedn,
 134                                const char *format, ...) _PRINTF_ATTRIBUTE(4,5)
 135 {
 136         va_list ap;
 137         struct ldb_dn *ret;
 138         struct ldb_message **res = NULL;
 139         int count;
 140
 141         va_start(ap, format);
 142         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, NULL, format, ap);
 143         va_end(ap);
 144
 145         if (count != 1) return NULL;
 146
 147         ret = talloc_steal(mem_ctx, res[0]->dn);
 148         talloc_free(res);
 149
 150         return ret;
 151 }
 152
 153 /*
 154   search the sam for a dom_sid attribute in exactly 1 record
 155 */
 156 struct dom_sid *samdb_search_dom_sid(struct ldb_context *sam_ldb,
 157                                      TALLOC_CTX *mem_ctx,
 158                                      struct ldb_dn *basedn,
 159                                      const char *attr_name,
 160                                      const char *format, ...) _PRINTF_ATTRIBUTE(5,6)
 161 {
 162         va_list ap;
 163         int count;
 164         struct ldb_message **res;
 165         const char *attrs[2] = { NULL, NULL };
 166         struct dom_sid *sid;
 167
 168         attrs[0] = attr_name;
 169
 170         va_start(ap, format);
 171         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
 172         va_end(ap);
 173         if (count > 1) {
 174                 DEBUG(1,("samdb: search for %s %s not single valued (count=%d)\n",
 175                          attr_name, format, count));
 176         }
 177         if (count != 1) {
 178                 talloc_free(res);
 179                 return NULL;
 180         }
 181         sid = samdb_result_dom_sid(mem_ctx, res[0], attr_name);
 182         talloc_free(res);
 183         return sid;
 184 }
 185
 186 /*
 187   return the count of the number of records in the sam matching the query
 188 */
 189 int samdb_search_count(struct ldb_context *sam_ldb,
 190                        struct ldb_dn *basedn,
 191                        const char *format, ...) _PRINTF_ATTRIBUTE(3,4)
 192 {
 193         va_list ap;
 194         struct ldb_message **res;
 195         const char *attrs[] = { NULL };
 196         int ret;
 197         TALLOC_CTX *tmp_ctx = talloc_new(sam_ldb);
 198
 199         va_start(ap, format);
 200         ret = gendb_search_v(sam_ldb, tmp_ctx, basedn, &res, attrs, format, ap);
 201         va_end(ap);
 202         talloc_free(tmp_ctx);
 203
 204         return ret;
 205 }
 206
 207
 208 /*
 209   search the sam for a single integer attribute in exactly 1 record
 210 */
 211 uint_t samdb_search_uint(struct ldb_context *sam_ldb,
 212                          TALLOC_CTX *mem_ctx,
 213                          uint_t default_value,
 214                          struct ldb_dn *basedn,
 215                          const char *attr_name,
 216                          const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
 217 {
 218         va_list ap;
 219         int count;
 220         struct ldb_message **res;
 221         const char *attrs[2] = { NULL, NULL };
 222
 223         attrs[0] = attr_name;
 224
 225         va_start(ap, format);
 226         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
 227         va_end(ap);
 228
 229         if (count != 1) {
 230                 return default_value;
 231         }
 232
 233         return samdb_result_uint(res[0], attr_name, default_value);
 234 }
 235
 236 /*
 237   search the sam for a single signed 64 bit integer attribute in exactly 1 record
 238 */
 239 int64_t samdb_search_int64(struct ldb_context *sam_ldb,
 240                            TALLOC_CTX *mem_ctx,
 241                            int64_t default_value,
 242                            struct ldb_dn *basedn,
 243                            const char *attr_name,
 244                            const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
 245 {
 246         va_list ap;
 247         int count;
 248         struct ldb_message **res;
 249         const char *attrs[2] = { NULL, NULL };
 250
 251         attrs[0] = attr_name;
 252
 253         va_start(ap, format);
 254         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
 255         va_end(ap);
 256
 257         if (count != 1) {
 258                 return default_value;
 259         }
 260
 261         return samdb_result_int64(res[0], attr_name, default_value);
 262 }
 263
 264 /*
 265   search the sam for multipe records each giving a single string attribute
 266   return the number of matches, or -1 on error
 267 */
 268 int samdb_search_string_multiple(struct ldb_context *sam_ldb,
 269                                  TALLOC_CTX *mem_ctx,
 270                                  struct ldb_dn *basedn,
 271                                  const char ***strs,
 272                                  const char *attr_name,
 273                                  const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
 274 {
 275         va_list ap;
 276         int count, i;
 277         const char *attrs[2] = { NULL, NULL };
 278         struct ldb_message **res = NULL;
 279
 280         attrs[0] = attr_name;
 281
 282         va_start(ap, format);
 283         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
 284         va_end(ap);
 285
 286         if (count <= 0) {
 287                 return count;
 288         }
 289
 290         /* make sure its single valued */
 291         for (i=0;i<count;i++) {
 292                 if (res[i]->num_elements != 1) {
 293                         DEBUG(1,("samdb: search for %s %s not single valued\n",
 294                                  attr_name, format));
 295                         talloc_free(res);
 296                         return -1;
 297                 }
 298         }
 299
 300         *strs = talloc_array(mem_ctx, const char *, count+1);
 301         if (! *strs) {
 302                 talloc_free(res);
 303                 return -1;
 304         }
 305
 306         for (i=0;i<count;i++) {
 307                 (*strs)[i] = samdb_result_string(res[i], attr_name, NULL);
 308         }
 309         (*strs)[count] = NULL;
 310
 311         return count;
 312 }
 313
 314 /*
 315   pull a uint from a result set.
 316 */
 317 uint_t samdb_result_uint(const struct ldb_message *msg, const char *attr, uint_t default_value)
 318 {
 319         return ldb_msg_find_attr_as_uint(msg, attr, default_value);
 320 }
 321
 322 /*
 323   pull a (signed) int64 from a result set.
 324 */
 325 int64_t samdb_result_int64(const struct ldb_message *msg, const char *attr, int64_t default_value)
 326 {
 327         return ldb_msg_find_attr_as_int64(msg, attr, default_value);
 328 }
 329
 330 /*
 331   pull a string from a result set.
 332 */
 333 const char *samdb_result_string(const struct ldb_message *msg, const char *attr,
 334                                 const char *default_value)
 335 {
 336         return ldb_msg_find_attr_as_string(msg, attr, default_value);
 337 }
 338
 339 struct ldb_dn *samdb_result_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
 340                                const char *attr, struct ldb_dn *default_value)
 341 {
 342         struct ldb_dn *ret_dn = ldb_msg_find_attr_as_dn(ldb, mem_ctx, msg, attr);
 343         if (!ret_dn) {
 344                 return default_value;
 345         }
 346         return ret_dn;
 347 }
 348
 349 /*
 350   pull a rid from a objectSid in a result set.
 351 */
 352 uint32_t samdb_result_rid_from_sid(TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
 353                                    const char *attr, uint32_t default_value)
 354 {
 355         struct dom_sid *sid;
 356         uint32_t rid;
 357
 358         sid = samdb_result_dom_sid(mem_ctx, msg, attr);
 359         if (sid == NULL) {
 360                 return default_value;
 361         }
 362         rid = sid->sub_auths[sid->num_auths-1];
 363         talloc_free(sid);
 364         return rid;
 365 }
 366
 367 /*
 368   pull a dom_sid structure from a objectSid in a result set.
 369 */
 370 struct dom_sid *samdb_result_dom_sid(TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
 371                                      const char *attr)
 372 {
 373         const struct ldb_val *v;
 374         struct dom_sid *sid;
 375         enum ndr_err_code ndr_err;
 376         v = ldb_msg_find_ldb_val(msg, attr);
 377         if (v == NULL) {
 378                 return NULL;
 379         }
 380         sid = talloc(mem_ctx, struct dom_sid);
 381         if (sid == NULL) {
 382                 return NULL;
 383         }
 384         ndr_err = ndr_pull_struct_blob(v, sid, NULL, sid,
 385                                        (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
 386         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 387                 talloc_free(sid);
 388                 return NULL;
 389         }
 390         return sid;
 391 }
 392
 393 /*
 394   pull a guid structure from a objectGUID in a result set.
 395 */
 396 struct GUID samdb_result_guid(const struct ldb_message *msg, const char *attr)
 397 {
 398         const struct ldb_val *v;
 399         struct GUID guid;
 400         NTSTATUS status;
 401
 402         v = ldb_msg_find_ldb_val(msg, attr);
 403         if (!v) return GUID_zero();
 404
 405         status = GUID_from_ndr_blob(v, &guid);
 406         if (!NT_STATUS_IS_OK(status)) {
 407                 return GUID_zero();
 408         }
 409
 410         return guid;
 411 }
 412
 413 /*
 414   pull a sid prefix from a objectSid in a result set.
 415   this is used to find the domain sid for a user
 416 */
 417 struct dom_sid *samdb_result_sid_prefix(TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
 418                                         const char *attr)
 419 {
 420         struct dom_sid *sid = samdb_result_dom_sid(mem_ctx, msg, attr);
 421         if (!sid || sid->num_auths < 1) return NULL;
 422         sid->num_auths--;
 423         return sid;
 424 }
 425
 426 /*
 427   pull a NTTIME in a result set.
 428 */
 429 NTTIME samdb_result_nttime(const struct ldb_message *msg, const char *attr,
 430                            NTTIME default_value)
 431 {
 432         return ldb_msg_find_attr_as_uint64(msg, attr, default_value);
 433 }
 434
 435 /*
 436  * Windows stores 0 for lastLogoff.
 437  * But when a MS DC return the lastLogoff (as Logoff Time)
 438  * it returns 0x7FFFFFFFFFFFFFFF, not returning this value in this case
 439  * cause windows 2008 and newer version to fail for SMB requests
 440  */
 441 NTTIME samdb_result_last_logoff(const struct ldb_message *msg)
 442 {
 443         NTTIME ret = ldb_msg_find_attr_as_uint64(msg, "lastLogoff",0);
 444
 445         if (ret == 0)
 446                 ret = 0x7FFFFFFFFFFFFFFFULL;
 447
 448         return ret;
 449 }
 450
 451 /*
 452  * Windows uses both 0 and 9223372036854775807 (0x7FFFFFFFFFFFFFFFULL) to
 453  * indicate an account doesn't expire.
 454  *
 455  * When Windows initially creates an account, it sets
 456  * accountExpires = 9223372036854775807 (0x7FFFFFFFFFFFFFFF).  However,
 457  * when changing from an account having a specific expiration date to
 458  * that account never expiring, it sets accountExpires = 0.
 459  *
 460  * Consolidate that logic here to allow clearer logic for account expiry in
 461  * the rest of the code.
 462  */
 463 NTTIME samdb_result_account_expires(const struct ldb_message *msg)
 464 {
 465         NTTIME ret = ldb_msg_find_attr_as_uint64(msg, "accountExpires",
 466                                                  0);
 467
 468         if (ret == 0)
 469                 ret = 0x7FFFFFFFFFFFFFFFULL;
 470
 471         return ret;
 472 }
 473
 474 /*
 475   pull a uint64_t from a result set.
 476 */
 477 uint64_t samdb_result_uint64(const struct ldb_message *msg, const char *attr,
 478                              uint64_t default_value)
 479 {
 480         return ldb_msg_find_attr_as_uint64(msg, attr, default_value);
 481 }
 482
 483
 484 /*
 485   construct the allow_password_change field from the PwdLastSet attribute and the
 486   domain password settings
 487 */
 488 NTTIME samdb_result_allow_password_change(struct ldb_context *sam_ldb,
 489                                           TALLOC_CTX *mem_ctx,
 490                                           struct ldb_dn *domain_dn,
 491                                           struct ldb_message *msg,
 492                                           const char *attr)
 493 {
 494         uint64_t attr_time = samdb_result_uint64(msg, attr, 0);
 495         int64_t minPwdAge;
 496
 497         if (attr_time == 0) {
 498                 return 0;
 499         }
 500
 501         minPwdAge = samdb_search_int64(sam_ldb, mem_ctx, 0, domain_dn, "minPwdAge", NULL);
 502
 503         /* yes, this is a -= not a += as minPwdAge is stored as the negative
 504            of the number of 100-nano-seconds */
 505         attr_time -= minPwdAge;
 506
 507         return attr_time;
 508 }
 509
 510 /*
 511   construct the force_password_change field from the PwdLastSet
 512   attribute, the userAccountControl and the domain password settings
 513 */
 514 NTTIME samdb_result_force_password_change(struct ldb_context *sam_ldb,
 515                                           TALLOC_CTX *mem_ctx,
 516                                           struct ldb_dn *domain_dn,
 517                                           struct ldb_message *msg)
 518 {
 519         uint64_t attr_time = samdb_result_uint64(msg, "pwdLastSet", 0);
 520         uint32_t userAccountControl = samdb_result_uint64(msg, "userAccountControl", 0);
 521         int64_t maxPwdAge;
 522
 523         /* Machine accounts don't expire, and there is a flag for 'no expiry' */
 524         if (!(userAccountControl & UF_NORMAL_ACCOUNT)
 525             || (userAccountControl & UF_DONT_EXPIRE_PASSWD)) {
 526                 return 0x7FFFFFFFFFFFFFFFULL;
 527         }
 528
 529         if (attr_time == 0) {
 530                 return 0;
 531         }
 532
 533         maxPwdAge = samdb_search_int64(sam_ldb, mem_ctx, 0, domain_dn, "maxPwdAge", NULL);
 534         if (maxPwdAge == 0) {
 535                 return 0x7FFFFFFFFFFFFFFFULL;
 536         } else {
 537                 attr_time -= maxPwdAge;
 538         }
 539
 540         return attr_time;
 541 }
 542
 543 /*
 544   pull a samr_Password structutre from a result set.
 545 */
 546 struct samr_Password *samdb_result_hash(TALLOC_CTX *mem_ctx, const struct ldb_message *msg, const char *attr)
 547 {
 548         struct samr_Password *hash = NULL;
 549         const struct ldb_val *val = ldb_msg_find_ldb_val(msg, attr);
 550         if (val && (val->length >= sizeof(hash->hash))) {
 551                 hash = talloc(mem_ctx, struct samr_Password);
 552                 memcpy(hash->hash, val->data, MIN(val->length, sizeof(hash->hash)));
 553         }
 554         return hash;
 555 }
 556
 557 /*
 558   pull an array of samr_Password structutres from a result set.
 559 */
 560 uint_t samdb_result_hashes(TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
 561                            const char *attr, struct samr_Password **hashes)
 562 {
 563         uint_t count, i;
 564         const struct ldb_val *val = ldb_msg_find_ldb_val(msg, attr);
 565
 566         *hashes = NULL;
 567         if (!val) {
 568                 return 0;
 569         }
 570         count = val->length / 16;
 571         if (count == 0) {
 572                 return 0;
 573         }
 574
 575         *hashes = talloc_array(mem_ctx, struct samr_Password, count);
 576         if (! *hashes) {
 577                 return 0;
 578         }
 579
 580         for (i=0;i<count;i++) {
 581                 memcpy((*hashes)[i].hash, (i*16)+(char *)val->data, 16);
 582         }
 583
 584         return count;
 585 }
 586
 587 NTSTATUS samdb_result_passwords(TALLOC_CTX *mem_ctx, struct loadparm_context *lp_ctx, struct ldb_message *msg,
 588                                 struct samr_Password **lm_pwd, struct samr_Password **nt_pwd)
 589 {
 590         struct samr_Password *lmPwdHash, *ntPwdHash;
 591         if (nt_pwd) {
 592                 int num_nt;
 593                 num_nt = samdb_result_hashes(mem_ctx, msg, "unicodePwd", &ntPwdHash);
 594                 if (num_nt == 0) {
 595                         *nt_pwd = NULL;
 596                 } else if (num_nt > 1) {
 597                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
 598                 } else {
 599                         *nt_pwd = &ntPwdHash[0];
 600                 }
 601         }
 602         if (lm_pwd) {
 603                 /* Ensure that if we have turned off LM
 604                  * authentication, that we never use the LM hash, even
 605                  * if we store it */
 606                 if (lp_lanman_auth(lp_ctx)) {
 607                         int num_lm;
 608                         num_lm = samdb_result_hashes(mem_ctx, msg, "dBCSPwd", &lmPwdHash);
 609                         if (num_lm == 0) {
 610                                 *lm_pwd = NULL;
 611                         } else if (num_lm > 1) {
 612                                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
 613                         } else {
 614                                 *lm_pwd = &lmPwdHash[0];
 615                         }
 616                 } else {
 617                         *lm_pwd = NULL;
 618                 }
 619         }
 620         return NT_STATUS_OK;
 621 }
 622
 623 /*
 624   pull a samr_LogonHours structutre from a result set.
 625 */
 626 struct samr_LogonHours samdb_result_logon_hours(TALLOC_CTX *mem_ctx, struct ldb_message *msg, const char *attr)
 627 {
 628         struct samr_LogonHours hours;
 629         const int units_per_week = 168;
 630         const struct ldb_val *val = ldb_msg_find_ldb_val(msg, attr);
 631         ZERO_STRUCT(hours);
 632         hours.bits = talloc_array(mem_ctx, uint8_t, units_per_week);
 633         if (!hours.bits) {
 634                 return hours;
 635         }
 636         hours.units_per_week = units_per_week;
 637         memset(hours.bits, 0xFF, units_per_week);
 638         if (val) {
 639                 memcpy(hours.bits, val->data, MIN(val->length, units_per_week));
 640         }
 641         return hours;
 642 }
 643
 644 /*
 645   pull a set of account_flags from a result set.
 646
 647   This requires that the attributes:
 648    pwdLastSet
 649    userAccountControl
 650   be included in 'msg'
 651 */
 652 uint32_t samdb_result_acct_flags(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
 653                                  struct ldb_message *msg, struct ldb_dn *domain_dn)
 654 {
 655         uint32_t userAccountControl = ldb_msg_find_attr_as_uint(msg, "userAccountControl", 0);
 656         uint32_t acct_flags = ds_uf2acb(userAccountControl);
 657         NTTIME must_change_time;
 658         NTTIME now;
 659
 660         must_change_time = samdb_result_force_password_change(sam_ctx, mem_ctx,
 661                                                               domain_dn, msg);
 662
 663         /* Test account expire time */
 664         unix_to_nt_time(&now, time(NULL));
 665         /* check for expired password */
 666         if (must_change_time < now) {
 667                 acct_flags |= ACB_PW_EXPIRED;
 668         }
 669         return acct_flags;
 670 }
 671
 672 struct lsa_BinaryString samdb_result_parameters(TALLOC_CTX *mem_ctx,
 673                                                 struct ldb_message *msg,
 674                                                 const char *attr)
 675 {
 676         struct lsa_BinaryString s;
 677         const struct ldb_val *val = ldb_msg_find_ldb_val(msg, attr);
 678
 679         ZERO_STRUCT(s);
 680
 681         if (!val) {
 682                 return s;
 683         }
 684
 685         s.array = talloc_array(mem_ctx, uint16_t, val->length/2);
 686         if (!s.array) {
 687                 return s;
 688         }
 689         s.length = s.size = val->length/2;
 690         memcpy(s.array, val->data, val->length);
 691
 692         return s;
 693 }
 694
 695 /* Find an attribute, with a particular value */
 696
 697 /* The current callers of this function expect a very specific
 698  * behaviour: In particular, objectClass subclass equivilance is not
 699  * wanted.  This means that we should not lookup the schema for the
 700  * comparison function */
 701 struct ldb_message_element *samdb_find_attribute(struct ldb_context *ldb,
 702                                                  const struct ldb_message *msg,
 703                                                  const char *name, const char *value)
 704 {
 705         int i;
 706         struct ldb_message_element *el = ldb_msg_find_element(msg, name);
 707
 708         if (!el) {
 709                 return NULL;
 710         }
 711
 712         for (i=0;i<el->num_values;i++) {
 713                 if (ldb_attr_cmp(value, (char *)el->values[i].data) == 0) {
 714                         return el;
 715                 }
 716         }
 717
 718         return NULL;
 719 }
 720
 721 int samdb_find_or_add_value(struct ldb_context *ldb, struct ldb_message *msg, const char *name, const char *set_value)
 722 {
 723         if (samdb_find_attribute(ldb, msg, name, set_value) == NULL) {
 724                 return samdb_msg_add_string(ldb, msg, msg, name, set_value);
 725         }
 726         return LDB_SUCCESS;
 727 }
 728
 729 int samdb_find_or_add_attribute(struct ldb_context *ldb, struct ldb_message *msg, const char *name, const char *set_value)
 730 {
 731         struct ldb_message_element *el;
 732
 733         el = ldb_msg_find_element(msg, name);
 734         if (el) {
 735                 return LDB_SUCCESS;
 736         }
 737
 738         return samdb_msg_add_string(ldb, msg, msg, name, set_value);
 739 }
 740
 741
 742
 743 /*
 744   add a string element to a message
 745 */
 746 int samdb_msg_add_string(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 747                          const char *attr_name, const char *str)
 748 {
 749         char *s = talloc_strdup(mem_ctx, str);
 750         char *a = talloc_strdup(mem_ctx, attr_name);
 751         if (s == NULL || a == NULL) {
 752                 return LDB_ERR_OPERATIONS_ERROR;
 753         }
 754         return ldb_msg_add_string(msg, a, s);
 755 }
 756
 757 /*
 758   add a dom_sid element to a message
 759 */
 760 int samdb_msg_add_dom_sid(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 761                          const char *attr_name, struct dom_sid *sid)
 762 {
 763         struct ldb_val v;
 764         enum ndr_err_code ndr_err;
 765
 766         ndr_err = ndr_push_struct_blob(&v, mem_ctx,
 767                                        lp_iconv_convenience(ldb_get_opaque(sam_ldb, "loadparm")),
 768                                        sid,
 769                                        (ndr_push_flags_fn_t)ndr_push_dom_sid);
 770         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 771                 return -1;
 772         }
 773         return ldb_msg_add_value(msg, attr_name, &v, NULL);
 774 }
 775
 776
 777 /*
 778   add a delete element operation to a message
 779 */
 780 int samdb_msg_add_delete(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 781                          const char *attr_name)
 782 {
 783         /* we use an empty replace rather than a delete, as it allows for
 784            samdb_replace() to be used everywhere */
 785         return ldb_msg_add_empty(msg, attr_name, LDB_FLAG_MOD_REPLACE, NULL);
 786 }
 787
 788 /*
 789   add a add attribute value to a message
 790 */
 791 int samdb_msg_add_addval(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 792                          const char *attr_name, const char *value)
 793 {
 794         struct ldb_message_element *el;
 795         char *a, *v;
 796         int ret;
 797         a = talloc_strdup(mem_ctx, attr_name);
 798         if (a == NULL)
 799                 return -1;
 800         v = talloc_strdup(mem_ctx, value);
 801         if (v == NULL)
 802                 return -1;
 803         ret = ldb_msg_add_string(msg, a, v);
 804         if (ret != 0)
 805                 return ret;
 806         el = ldb_msg_find_element(msg, a);
 807         if (el == NULL)
 808                 return -1;
 809         el->flags = LDB_FLAG_MOD_ADD;
 810         return 0;
 811 }
 812
 813 /*
 814   add a delete attribute value to a message
 815 */
 816 int samdb_msg_add_delval(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 817                          const char *attr_name, const char *value)
 818 {
 819         struct ldb_message_element *el;
 820         char *a, *v;
 821         int ret;
 822         a = talloc_strdup(mem_ctx, attr_name);
 823         if (a == NULL)
 824                 return -1;
 825         v = talloc_strdup(mem_ctx, value);
 826         if (v == NULL)
 827                 return -1;
 828         ret = ldb_msg_add_string(msg, a, v);
 829         if (ret != 0)
 830                 return ret;
 831         el = ldb_msg_find_element(msg, a);
 832         if (el == NULL)
 833                 return -1;
 834         el->flags = LDB_FLAG_MOD_DELETE;
 835         return 0;
 836 }
 837
 838 /*
 839   add a int element to a message
 840 */
 841 int samdb_msg_add_int(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 842                        const char *attr_name, int v)
 843 {
 844         const char *s = talloc_asprintf(mem_ctx, "%d", v);
 845         return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, s);
 846 }
 847
 848 /*
 849   add a uint_t element to a message
 850 */
 851 int samdb_msg_add_uint(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 852                        const char *attr_name, uint_t v)
 853 {
 854         const char *s = talloc_asprintf(mem_ctx, "%u", v);
 855         return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, s);
 856 }
 857
 858 /*
 859   add a (signed) int64_t element to a message
 860 */
 861 int samdb_msg_add_int64(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 862                         const char *attr_name, int64_t v)
 863 {
 864         const char *s = talloc_asprintf(mem_ctx, "%lld", (long long)v);
 865         return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, s);
 866 }
 867
 868 /*
 869   add a uint64_t element to a message
 870 */
 871 int samdb_msg_add_uint64(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 872                         const char *attr_name, uint64_t v)
 873 {
 874         const char *s = talloc_asprintf(mem_ctx, "%llu", (unsigned long long)v);
 875         return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, s);
 876 }
 877
 878 /*
 879   add a samr_Password element to a message
 880 */
 881 int samdb_msg_add_hash(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 882                        const char *attr_name, struct samr_Password *hash)
 883 {
 884         struct ldb_val val;
 885         val.data = talloc_memdup(mem_ctx, hash->hash, 16);
 886         if (!val.data) {
 887                 return -1;
 888         }
 889         val.length = 16;
 890         return ldb_msg_add_value(msg, attr_name, &val, NULL);
 891 }
 892
 893 /*
 894   add a samr_Password array to a message
 895 */
 896 int samdb_msg_add_hashes(TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 897                          const char *attr_name, struct samr_Password *hashes, uint_t count)
 898 {
 899         struct ldb_val val;
 900         int i;
 901         val.data = talloc_array_size(mem_ctx, 16, count);
 902         val.length = count*16;
 903         if (!val.data) {
 904                 return -1;
 905         }
 906         for (i=0;i<count;i++) {
 907                 memcpy(i*16 + (char *)val.data, hashes[i].hash, 16);
 908         }
 909         return ldb_msg_add_value(msg, attr_name, &val, NULL);
 910 }
 911
 912 /*
 913   add a acct_flags element to a message
 914 */
 915 int samdb_msg_add_acct_flags(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 916                              const char *attr_name, uint32_t v)
 917 {
 918         return samdb_msg_add_uint(sam_ldb, mem_ctx, msg, attr_name, ds_acb2uf(v));
 919 }
 920
 921 /*
 922   add a logon_hours element to a message
 923 */
 924 int samdb_msg_add_logon_hours(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 925                               const char *attr_name, struct samr_LogonHours *hours)
 926 {
 927         struct ldb_val val;
 928         val.length = hours->units_per_week / 8;
 929         val.data = hours->bits;
 930         return ldb_msg_add_value(msg, attr_name, &val, NULL);
 931 }
 932
 933 /*
 934   add a parameters element to a message
 935 */
 936 int samdb_msg_add_parameters(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 937                              const char *attr_name, struct lsa_BinaryString *parameters)
 938 {
 939         struct ldb_val val;
 940         val.length = parameters->length * 2;
 941         val.data = (uint8_t *)parameters->array;
 942         return ldb_msg_add_value(msg, attr_name, &val, NULL);
 943 }
 944 /*
 945   add a general value element to a message
 946 */
 947 int samdb_msg_add_value(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 948                               const char *attr_name, const struct ldb_val *val)
 949 {
 950         return ldb_msg_add_value(msg, attr_name, val, NULL);
 951 }
 952
 953 /*
 954   sets a general value element to a message
 955 */
 956 int samdb_msg_set_value(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 957                         const char *attr_name, const struct ldb_val *val)
 958 {
 959         struct ldb_message_element *el;
 960
 961         el = ldb_msg_find_element(msg, attr_name);
 962         if (el) {
 963                 el->num_values = 0;
 964         }
 965         return ldb_msg_add_value(msg, attr_name, val, NULL);
 966 }
 967
 968 /*
 969   set a string element in a message
 970 */
 971 int samdb_msg_set_string(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 972                          const char *attr_name, const char *str)
 973 {
 974         struct ldb_message_element *el;
 975
 976         el = ldb_msg_find_element(msg, attr_name);
 977         if (el) {
 978                 el->num_values = 0;
 979         }
 980         return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, str);
 981 }
 982
 983 /*
 984   replace elements in a record
 985 */
 986 int samdb_replace(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg)
 987 {
 988         int i;
 989
 990         /* mark all the message elements as LDB_FLAG_MOD_REPLACE */
 991         for (i=0;i<msg->num_elements;i++) {
 992                 msg->elements[i].flags = LDB_FLAG_MOD_REPLACE;
 993         }
 994
 995         /* modify the samdb record */
 996         return ldb_modify(sam_ldb, msg);
 997 }
 998
 999 /*
1000  * Handle ldb_request in transaction
1001  */
1002 static int dsdb_autotransaction_request(struct ldb_context *sam_ldb,
1003                                  struct ldb_request *req)
1004 {
1005         int ret;
1006
1007         ret = ldb_transaction_start(sam_ldb);
1008         if (ret != LDB_SUCCESS) {
1009                 return ret;
1010         }
1011
1012         ret = ldb_request(sam_ldb, req);
1013         if (ret == LDB_SUCCESS) {
1014                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
1015         }
1016
1017         if (ret == LDB_SUCCESS) {
1018                 return ldb_transaction_commit(sam_ldb);
1019         }
1020         ldb_transaction_cancel(sam_ldb);
1021
1022         return ret;
1023 }
1024
1025 /*
1026  * replace elements in a record using LDB_CONTROL_AS_SYSTEM
1027  * used to skip access checks on operations
1028  * that are performed by the system
1029  */
1030 int samdb_replace_as_system(struct ldb_context *sam_ldb,
1031                             TALLOC_CTX *mem_ctx,
1032                             struct ldb_message *msg)
1033 {
1034         int i;
1035         int ldb_ret;
1036         struct ldb_request *req = NULL;
1037
1038         /* mark all the message elements as LDB_FLAG_MOD_REPLACE */
1039         for (i=0;i<msg->num_elements;i++) {
1040                 msg->elements[i].flags = LDB_FLAG_MOD_REPLACE;
1041         }
1042
1043
1044         ldb_ret = ldb_msg_sanity_check(sam_ldb, msg);
1045         if (ldb_ret != LDB_SUCCESS) {
1046                 return ldb_ret;
1047         }
1048
1049         ldb_ret = ldb_build_mod_req(&req, sam_ldb, mem_ctx,
1050                                     msg,
1051                                     NULL,
1052                                     NULL,
1053                                     ldb_op_default_callback,
1054                                     NULL);
1055
1056         if (ldb_ret != LDB_SUCCESS) {
1057                 talloc_free(req);
1058                 return ldb_ret;
1059         }
1060
1061         ldb_ret = ldb_request_add_control(req, LDB_CONTROL_AS_SYSTEM_OID, false, NULL);
1062         if (ldb_ret != LDB_SUCCESS) {
1063                 talloc_free(req);
1064                 return ldb_ret;
1065         }
1066
1067         /* do request and auto start a transaction */
1068         ldb_ret = dsdb_autotransaction_request(sam_ldb, req);
1069
1070         talloc_free(req);
1071         return ldb_ret;
1072 }
1073
1074 /*
1075   return a default security descriptor
1076 */
1077 struct security_descriptor *samdb_default_security_descriptor(TALLOC_CTX *mem_ctx)
1078 {
1079         struct security_descriptor *sd;
1080
1081         sd = security_descriptor_initialise(mem_ctx);
1082
1083         return sd;
1084 }
1085
1086 struct ldb_dn *samdb_base_dn(struct ldb_context *sam_ctx)
1087 {
1088         return ldb_get_default_basedn(sam_ctx);
1089 }
1090
1091 struct ldb_dn *samdb_config_dn(struct ldb_context *sam_ctx)
1092 {
1093         return ldb_get_config_basedn(sam_ctx);
1094 }
1095
1096 struct ldb_dn *samdb_schema_dn(struct ldb_context *sam_ctx)
1097 {
1098         return ldb_get_schema_basedn(sam_ctx);
1099 }
1100
1101 struct ldb_dn *samdb_aggregate_schema_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
1102 {
1103         struct ldb_dn *schema_dn = ldb_get_schema_basedn(sam_ctx);
1104         struct ldb_dn *aggregate_dn;
1105         if (!schema_dn) {
1106                 return NULL;
1107         }
1108
1109         aggregate_dn = ldb_dn_copy(mem_ctx, schema_dn);
1110         if (!aggregate_dn) {
1111                 return NULL;
1112         }
1113         if (!ldb_dn_add_child_fmt(aggregate_dn, "CN=Aggregate")) {
1114                 return NULL;
1115         }
1116         return aggregate_dn;
1117 }
1118
1119 struct ldb_dn *samdb_root_dn(struct ldb_context *sam_ctx)
1120 {
1121         return ldb_get_root_basedn(sam_ctx);
1122 }
1123
1124 struct ldb_dn *samdb_partitions_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
1125 {
1126         struct ldb_dn *new_dn;
1127
1128         new_dn = ldb_dn_copy(mem_ctx, samdb_config_dn(sam_ctx));
1129         if ( ! ldb_dn_add_child_fmt(new_dn, "CN=Partitions")) {
1130                 talloc_free(new_dn);
1131                 return NULL;
1132         }
1133         return new_dn;
1134 }
1135
1136 struct ldb_dn *samdb_sites_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
1137 {
1138         struct ldb_dn *new_dn;
1139
1140         new_dn = ldb_dn_copy(mem_ctx, samdb_config_dn(sam_ctx));
1141         if ( ! ldb_dn_add_child_fmt(new_dn, "CN=Sites")) {
1142                 talloc_free(new_dn);
1143                 return NULL;
1144         }
1145         return new_dn;
1146 }
1147
1148 /*
1149   work out the domain sid for the current open ldb
1150 */
1151 const struct dom_sid *samdb_domain_sid(struct ldb_context *ldb)
1152 {
1153         TALLOC_CTX *tmp_ctx;
1154         const struct dom_sid *domain_sid;
1155         const char *attrs[] = {
1156                 "objectSid",
1157                 NULL
1158         };
1159         struct ldb_result *res;
1160         int ret;
1161
1162         /* see if we have a cached copy */
1163         domain_sid = (struct dom_sid *)ldb_get_opaque(ldb, "cache.domain_sid");
1164         if (domain_sid) {
1165                 return domain_sid;
1166         }
1167
1168         tmp_ctx = talloc_new(ldb);
1169         if (tmp_ctx == NULL) {
1170                 goto failed;
1171         }
1172
1173         ret = ldb_search(ldb, tmp_ctx, &res, ldb_get_default_basedn(ldb), LDB_SCOPE_BASE, attrs, "objectSid=*");
1174
1175         if (ret != LDB_SUCCESS) {
1176                 goto failed;
1177         }
1178
1179         if (res->count != 1) {
1180                 goto failed;
1181         }
1182
1183         domain_sid = samdb_result_dom_sid(tmp_ctx, res->msgs[0], "objectSid");
1184         if (domain_sid == NULL) {
1185                 goto failed;
1186         }
1187
1188         /* cache the domain_sid in the ldb */
1189         if (ldb_set_opaque(ldb, "cache.domain_sid", discard_const_p(struct dom_sid, domain_sid)) != LDB_SUCCESS) {
1190                 goto failed;
1191         }
1192
1193         talloc_steal(ldb, domain_sid);
1194         talloc_free(tmp_ctx);
1195
1196         return domain_sid;
1197
1198 failed:
1199         DEBUG(1,("Failed to find domain_sid for open ldb\n"));
1200         talloc_free(tmp_ctx);
1201         return NULL;
1202 }
1203
1204 bool samdb_set_domain_sid(struct ldb_context *ldb, const struct dom_sid *dom_sid_in)
1205 {
1206         TALLOC_CTX *tmp_ctx;
1207         struct dom_sid *dom_sid_new;
1208         struct dom_sid *dom_sid_old;
1209
1210         /* see if we have a cached copy */
1211         dom_sid_old = talloc_get_type(ldb_get_opaque(ldb,
1212                                                      "cache.domain_sid"), struct dom_sid);
1213
1214         tmp_ctx = talloc_new(ldb);
1215         if (tmp_ctx == NULL) {
1216                 goto failed;
1217         }
1218
1219         dom_sid_new = dom_sid_dup(tmp_ctx, dom_sid_in);
1220         if (!dom_sid_new) {
1221                 goto failed;
1222         }
1223
1224         /* cache the domain_sid in the ldb */
1225         if (ldb_set_opaque(ldb, "cache.domain_sid", dom_sid_new) != LDB_SUCCESS) {
1226                 goto failed;
1227         }
1228
1229         talloc_steal(ldb, dom_sid_new);
1230         talloc_free(tmp_ctx);
1231         talloc_free(dom_sid_old);
1232
1233         return true;
1234
1235 failed:
1236         DEBUG(1,("Failed to set our own cached domain SID in the ldb!\n"));
1237         talloc_free(tmp_ctx);
1238         return false;
1239 }
1240
1241 /* Obtain the short name of the flexible single master operator
1242  * (FSMO), such as the PDC Emulator */
1243 const char *samdb_result_fsmo_name(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
1244                              const char *attr)
1245 {
1246         /* Format is cn=NTDS Settings,cn=<NETBIOS name of FSMO>,.... */
1247         struct ldb_dn *fsmo_dn = ldb_msg_find_attr_as_dn(ldb, mem_ctx, msg, attr);
1248         const struct ldb_val *val = ldb_dn_get_component_val(fsmo_dn, 1);
1249         const char *name = ldb_dn_get_component_name(fsmo_dn, 1);
1250
1251         if (!name || (ldb_attr_cmp(name, "cn") != 0)) {
1252                 /* Ensure this matches the format.  This gives us a
1253                  * bit more confidence that a 'cn' value will be a
1254                  * ascii string */
1255                 return NULL;
1256         }
1257         if (val) {
1258                 return (char *)val->data;
1259         }
1260         return NULL;
1261 }
1262
1263 /*
1264   work out the ntds settings dn for the current open ldb
1265 */
1266 struct ldb_dn *samdb_ntds_settings_dn(struct ldb_context *ldb)
1267 {
1268         TALLOC_CTX *tmp_ctx;
1269         const char *root_attrs[] = { "dsServiceName", NULL };
1270         int ret;
1271         struct ldb_result *root_res;
1272         struct ldb_dn *settings_dn;
1273
1274         /* see if we have a cached copy */
1275         settings_dn = (struct ldb_dn *)ldb_get_opaque(ldb, "cache.settings_dn");
1276         if (settings_dn) {
1277                 return settings_dn;
1278         }
1279
1280         tmp_ctx = talloc_new(ldb);
1281         if (tmp_ctx == NULL) {
1282                 goto failed;
1283         }
1284
1285         ret = ldb_search(ldb, tmp_ctx, &root_res, ldb_dn_new(tmp_ctx, ldb, ""), LDB_SCOPE_BASE, root_attrs, NULL);
1286         if (ret) {
1287                 DEBUG(1,("Searching for dsServiceName in rootDSE failed: %s\n",
1288                          ldb_errstring(ldb)));
1289                 goto failed;
1290         }
1291
1292         if (root_res->count != 1) {
1293                 goto failed;
1294         }
1295
1296         settings_dn = ldb_msg_find_attr_as_dn(ldb, tmp_ctx, root_res->msgs[0], "dsServiceName");
1297
1298         /* cache the domain_sid in the ldb */
1299         if (ldb_set_opaque(ldb, "cache.settings_dn", settings_dn) != LDB_SUCCESS) {
1300                 goto failed;
1301         }
1302
1303         talloc_steal(ldb, settings_dn);
1304         talloc_free(tmp_ctx);
1305
1306         return settings_dn;
1307
1308 failed:
1309         DEBUG(1,("Failed to find our own NTDS Settings DN in the ldb!\n"));
1310         talloc_free(tmp_ctx);
1311         return NULL;
1312 }
1313
1314 /*
1315   work out the ntds settings invocationId for the current open ldb
1316 */
1317 const struct GUID *samdb_ntds_invocation_id(struct ldb_context *ldb)
1318 {
1319         TALLOC_CTX *tmp_ctx;
1320         const char *attrs[] = { "invocationId", NULL };
1321         int ret;
1322         struct ldb_result *res;
1323         struct GUID *invocation_id;
1324
1325         /* see if we have a cached copy */
1326         invocation_id = (struct GUID *)ldb_get_opaque(ldb, "cache.invocation_id");
1327         if (invocation_id) {
1328                 return invocation_id;
1329         }
1330
1331         tmp_ctx = talloc_new(ldb);
1332         if (tmp_ctx == NULL) {
1333                 goto failed;
1334         }
1335
1336         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1337         if (ret) {
1338                 goto failed;
1339         }
1340
1341         if (res->count != 1) {
1342                 goto failed;
1343         }
1344
1345         invocation_id = talloc(tmp_ctx, struct GUID);
1346         if (!invocation_id) {
1347                 goto failed;
1348         }
1349
1350         *invocation_id = samdb_result_guid(res->msgs[0], "invocationId");
1351
1352         /* cache the domain_sid in the ldb */
1353         if (ldb_set_opaque(ldb, "cache.invocation_id", invocation_id) != LDB_SUCCESS) {
1354                 goto failed;
1355         }
1356
1357         talloc_steal(ldb, invocation_id);
1358         talloc_free(tmp_ctx);
1359
1360         return invocation_id;
1361
1362 failed:
1363         DEBUG(1,("Failed to find our own NTDS Settings invocationId in the ldb!\n"));
1364         talloc_free(tmp_ctx);
1365         return NULL;
1366 }
1367
1368 bool samdb_set_ntds_invocation_id(struct ldb_context *ldb, const struct GUID *invocation_id_in)
1369 {
1370         TALLOC_CTX *tmp_ctx;
1371         struct GUID *invocation_id_new;
1372         struct GUID *invocation_id_old;
1373
1374         /* see if we have a cached copy */
1375         invocation_id_old = (struct GUID *)ldb_get_opaque(ldb,
1376                                                          "cache.invocation_id");
1377
1378         tmp_ctx = talloc_new(ldb);
1379         if (tmp_ctx == NULL) {
1380                 goto failed;
1381         }
1382
1383         invocation_id_new = talloc(tmp_ctx, struct GUID);
1384         if (!invocation_id_new) {
1385                 goto failed;
1386         }
1387
1388         *invocation_id_new = *invocation_id_in;
1389
1390         /* cache the domain_sid in the ldb */
1391         if (ldb_set_opaque(ldb, "cache.invocation_id", invocation_id_new) != LDB_SUCCESS) {
1392                 goto failed;
1393         }
1394
1395         talloc_steal(ldb, invocation_id_new);
1396         talloc_free(tmp_ctx);
1397         talloc_free(invocation_id_old);
1398
1399         return true;
1400
1401 failed:
1402         DEBUG(1,("Failed to set our own cached invocationId in the ldb!\n"));
1403         talloc_free(tmp_ctx);
1404         return false;
1405 }
1406
1407 /*
1408   work out the ntds settings objectGUID for the current open ldb
1409 */
1410 const struct GUID *samdb_ntds_objectGUID(struct ldb_context *ldb)
1411 {
1412         TALLOC_CTX *tmp_ctx;
1413         const char *attrs[] = { "objectGUID", NULL };
1414         int ret;
1415         struct ldb_result *res;
1416         struct GUID *ntds_guid;
1417
1418         /* see if we have a cached copy */
1419         ntds_guid = (struct GUID *)ldb_get_opaque(ldb, "cache.ntds_guid");
1420         if (ntds_guid) {
1421                 return ntds_guid;
1422         }
1423
1424         tmp_ctx = talloc_new(ldb);
1425         if (tmp_ctx == NULL) {
1426                 goto failed;
1427         }
1428
1429         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1430         if (ret) {
1431                 goto failed;
1432         }
1433
1434         if (res->count != 1) {
1435                 goto failed;
1436         }
1437
1438         ntds_guid = talloc(tmp_ctx, struct GUID);
1439         if (!ntds_guid) {
1440                 goto failed;
1441         }
1442
1443         *ntds_guid = samdb_result_guid(res->msgs[0], "objectGUID");
1444
1445         /* cache the domain_sid in the ldb */
1446         if (ldb_set_opaque(ldb, "cache.ntds_guid", ntds_guid) != LDB_SUCCESS) {
1447                 goto failed;
1448         }
1449
1450         talloc_steal(ldb, ntds_guid);
1451         talloc_free(tmp_ctx);
1452
1453         return ntds_guid;
1454
1455 failed:
1456         DEBUG(1,("Failed to find our own NTDS Settings objectGUID in the ldb!\n"));
1457         talloc_free(tmp_ctx);
1458         return NULL;
1459 }
1460
1461 bool samdb_set_ntds_objectGUID(struct ldb_context *ldb, const struct GUID *ntds_guid_in)
1462 {
1463         TALLOC_CTX *tmp_ctx;
1464         struct GUID *ntds_guid_new;
1465         struct GUID *ntds_guid_old;
1466
1467         /* see if we have a cached copy */
1468         ntds_guid_old = (struct GUID *)ldb_get_opaque(ldb, "cache.ntds_guid");
1469
1470         tmp_ctx = talloc_new(ldb);
1471         if (tmp_ctx == NULL) {
1472                 goto failed;
1473         }
1474
1475         ntds_guid_new = talloc(tmp_ctx, struct GUID);
1476         if (!ntds_guid_new) {
1477                 goto failed;
1478         }
1479
1480         *ntds_guid_new = *ntds_guid_in;
1481
1482         /* cache the domain_sid in the ldb */
1483         if (ldb_set_opaque(ldb, "cache.ntds_guid", ntds_guid_new) != LDB_SUCCESS) {
1484                 goto failed;
1485         }
1486
1487         talloc_steal(ldb, ntds_guid_new);
1488         talloc_free(tmp_ctx);
1489         talloc_free(ntds_guid_old);
1490
1491         return true;
1492
1493 failed:
1494         DEBUG(1,("Failed to set our own cached invocationId in the ldb!\n"));
1495         talloc_free(tmp_ctx);
1496         return false;
1497 }
1498
1499 /*
1500   work out the server dn for the current open ldb
1501 */
1502 struct ldb_dn *samdb_server_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1503 {
1504         return ldb_dn_get_parent(mem_ctx, samdb_ntds_settings_dn(ldb));
1505 }
1506
1507 /*
1508   work out the server dn for the current open ldb
1509 */
1510 struct ldb_dn *samdb_server_site_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1511 {
1512         struct ldb_dn *server_dn;
1513         struct ldb_dn *server_site_dn;
1514
1515         server_dn = samdb_server_dn(ldb, mem_ctx);
1516         if (!server_dn) return NULL;
1517
1518         server_site_dn = ldb_dn_get_parent(mem_ctx, server_dn);
1519
1520         talloc_free(server_dn);
1521         return server_site_dn;
1522 }
1523
1524 const char *samdb_server_site_name(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1525 {
1526         const struct ldb_val *val = ldb_dn_get_rdn_val(samdb_server_site_dn(ldb, mem_ctx));
1527
1528         if (val != NULL)
1529                 return (const char *) val->data;
1530         else
1531                 return NULL;
1532 }
1533
1534 /*
1535   work out if we are the PDC for the domain of the current open ldb
1536 */
1537 bool samdb_is_pdc(struct ldb_context *ldb)
1538 {
1539         const char *dom_attrs[] = { "fSMORoleOwner", NULL };
1540         int ret;
1541         struct ldb_result *dom_res;
1542         TALLOC_CTX *tmp_ctx;
1543         bool is_pdc;
1544         struct ldb_dn *pdc;
1545
1546         tmp_ctx = talloc_new(ldb);
1547         if (tmp_ctx == NULL) {
1548                 DEBUG(1, ("talloc_new failed in samdb_is_pdc"));
1549                 return false;
1550         }
1551
1552         ret = ldb_search(ldb, tmp_ctx, &dom_res, ldb_get_default_basedn(ldb), LDB_SCOPE_BASE, dom_attrs, NULL);
1553         if (ret) {
1554                 DEBUG(1,("Searching for fSMORoleOwner in %s failed: %s\n",
1555                          ldb_dn_get_linearized(ldb_get_default_basedn(ldb)),
1556                          ldb_errstring(ldb)));
1557                 goto failed;
1558         }
1559         if (dom_res->count != 1) {
1560                 goto failed;
1561         }
1562
1563         pdc = ldb_msg_find_attr_as_dn(ldb, tmp_ctx, dom_res->msgs[0], "fSMORoleOwner");
1564
1565         if (ldb_dn_compare(samdb_ntds_settings_dn(ldb), pdc) == 0) {
1566                 is_pdc = true;
1567         } else {
1568                 is_pdc = false;
1569         }
1570
1571         talloc_free(tmp_ctx);
1572
1573         return is_pdc;
1574
1575 failed:
1576         DEBUG(1,("Failed to find if we are the PDC for this ldb\n"));
1577         talloc_free(tmp_ctx);
1578         return false;
1579 }
1580
1581 /*
1582   work out if we are a Global Catalog server for the domain of the current open ldb
1583 */
1584 bool samdb_is_gc(struct ldb_context *ldb)
1585 {
1586         const char *attrs[] = { "options", NULL };
1587         int ret, options;
1588         struct ldb_result *res;
1589         TALLOC_CTX *tmp_ctx;
1590
1591         tmp_ctx = talloc_new(ldb);
1592         if (tmp_ctx == NULL) {
1593                 DEBUG(1, ("talloc_new failed in samdb_is_pdc"));
1594                 return false;
1595         }
1596
1597         /* Query cn=ntds settings,.... */
1598         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1599         if (ret) {
1600                 talloc_free(tmp_ctx);
1601                 return false;
1602         }
1603         if (res->count != 1) {
1604                 talloc_free(tmp_ctx);
1605                 return false;
1606         }
1607
1608         options = ldb_msg_find_attr_as_int(res->msgs[0], "options", 0);
1609         talloc_free(tmp_ctx);
1610
1611         /* if options attribute has the 0x00000001 flag set, then enable the global catlog */
1612         if (options & 0x000000001) {
1613                 return true;
1614         }
1615         return false;
1616 }
1617
1618 /* Find a domain object in the parents of a particular DN.  */
1619 int samdb_search_for_parent_domain(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
1620                                    struct ldb_dn **parent_dn, const char **errstring)
1621 {
1622         TALLOC_CTX *local_ctx;
1623         struct ldb_dn *sdn = dn;
1624         struct ldb_result *res = NULL;
1625         int ret = 0;
1626         const char *attrs[] = { NULL };
1627
1628         local_ctx = talloc_new(mem_ctx);
1629         if (local_ctx == NULL) return LDB_ERR_OPERATIONS_ERROR;
1630
1631         while ((sdn = ldb_dn_get_parent(local_ctx, sdn))) {
1632                 ret = ldb_search(ldb, local_ctx, &res, sdn, LDB_SCOPE_BASE, attrs,
1633                                  "(|(objectClass=domain)(objectClass=builtinDomain))");
1634                 if (ret == LDB_SUCCESS) {
1635                         if (res->count == 1) {
1636                                 break;
1637                         }
1638                 } else {
1639                         break;
1640                 }
1641         }
1642
1643         if (ret != LDB_SUCCESS) {
1644                 *errstring = talloc_asprintf(mem_ctx, "Error searching for parent domain of %s, failed searching for %s: %s",
1645                                              ldb_dn_get_linearized(dn),
1646                                              ldb_dn_get_linearized(sdn),
1647                                              ldb_errstring(ldb));
1648                 talloc_free(local_ctx);
1649                 return ret;
1650         }
1651         if (res->count != 1) {
1652                 *errstring = talloc_asprintf(mem_ctx, "Invalid dn (%s), not child of a domain object",
1653                                              ldb_dn_get_linearized(dn));
1654                 DEBUG(0,(__location__ ": %s\n", *errstring));
1655                 talloc_free(local_ctx);
1656                 return LDB_ERR_CONSTRAINT_VIOLATION;
1657         }
1658
1659         *parent_dn = talloc_steal(mem_ctx, res->msgs[0]->dn);
1660         talloc_free(local_ctx);
1661         return ret;
1662 }
1663
1664
1665 /*
1666  * Performs checks on a user password (plaintext UNIX format - attribute
1667  * "password"). The remaining parameters have to be extracted from the domain
1668  * object in the AD.
1669  *
1670  * Result codes from "enum samr_ValidationStatus" (consider "samr.idl")
1671  */
1672 enum samr_ValidationStatus samdb_check_password(const DATA_BLOB *password,
1673                                                 const uint32_t pwdProperties,
1674                                                 const uint32_t minPwdLength)
1675 {
1676         /* checks if the "minPwdLength" property is satisfied */
1677         if (minPwdLength > password->length)
1678                 return SAMR_VALIDATION_STATUS_PWD_TOO_SHORT;
1679
1680         /* checks the password complexity */
1681         if (((pwdProperties & DOMAIN_PASSWORD_COMPLEX) != 0)
1682                         && (password->data != NULL)
1683                         && (!check_password_quality((const char *) password->data)))
1684                 return SAMR_VALIDATION_STATUS_NOT_COMPLEX_ENOUGH;
1685
1686         return SAMR_VALIDATION_STATUS_SUCCESS;
1687 }
1688
1689 /*
1690  * Sets the user password using plaintext UTF16 (attribute "new_password") or
1691  * LM (attribute "lmNewHash") or NT (attribute "ntNewHash") hash. Also pass
1692  * as parameter if it's a user change or not ("userChange"). The "rejectReason"
1693  * gives some more informations if the changed failed.
1694  *
1695  * The caller should have a LDB transaction wrapping this.
1696  *
1697  * Results: NT_STATUS_OK, NT_STATUS_INTERNAL_DB_CORRUPTION,
1698  *   NT_STATUS_INVALID_PARAMETER, NT_STATUS_UNSUCCESSFUL,
1699  *   NT_STATUS_PASSWORD_RESTRICTION
1700  */
1701 NTSTATUS samdb_set_password(struct ldb_context *ctx, TALLOC_CTX *mem_ctx,
1702                             struct ldb_dn *user_dn, struct ldb_dn *domain_dn,
1703                             struct ldb_message *mod,
1704                             const DATA_BLOB *new_password,
1705                             struct samr_Password *param_lmNewHash,
1706                             struct samr_Password *param_ntNewHash,
1707                             bool user_change,
1708                             enum samPwdChangeReason *reject_reason,
1709                             struct samr_DomInfo1 **_dominfo)
1710 {
1711         const char * const user_attrs[] = { "userAccountControl",
1712                                             "lmPwdHistory",
1713                                             "ntPwdHistory",
1714                                             "dBCSPwd", "unicodePwd",
1715                                             "objectSid",
1716                                             "pwdLastSet", NULL };
1717         const char * const domain_attrs[] = { "minPwdLength", "pwdProperties",
1718                                               "pwdHistoryLength",
1719                                               "maxPwdAge", "minPwdAge", NULL };
1720         NTTIME pwdLastSet;
1721         uint32_t minPwdLength, pwdProperties, pwdHistoryLength;
1722         int64_t maxPwdAge, minPwdAge;
1723         uint32_t userAccountControl;
1724         struct samr_Password *sambaLMPwdHistory, *sambaNTPwdHistory,
1725                 *lmPwdHash, *ntPwdHash, *lmNewHash, *ntNewHash;
1726         struct samr_Password local_lmNewHash, local_ntNewHash;
1727         int sambaLMPwdHistory_len, sambaNTPwdHistory_len;
1728         struct dom_sid *domain_sid;
1729         struct ldb_message **res;
1730         bool restrictions;
1731         int count;
1732         time_t now = time(NULL);
1733         NTTIME now_nt;
1734         int i;
1735
1736         /* we need to know the time to compute password age */
1737         unix_to_nt_time(&now_nt, now);
1738
1739         /* pull all the user parameters */
1740         count = gendb_search_dn(ctx, mem_ctx, user_dn, &res, user_attrs);
1741         if (count != 1) {
1742                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
1743         }
1744         userAccountControl = samdb_result_uint(res[0], "userAccountControl", 0);
1745         sambaLMPwdHistory_len = samdb_result_hashes(mem_ctx, res[0],
1746                                          "lmPwdHistory", &sambaLMPwdHistory);
1747         sambaNTPwdHistory_len = samdb_result_hashes(mem_ctx, res[0],
1748                                          "ntPwdHistory", &sambaNTPwdHistory);
1749         lmPwdHash = samdb_result_hash(mem_ctx, res[0], "dBCSPwd");
1750         ntPwdHash = samdb_result_hash(mem_ctx, res[0], "unicodePwd");
1751         pwdLastSet = samdb_result_uint64(res[0], "pwdLastSet", 0);
1752
1753         /* Copy parameters */
1754         lmNewHash = param_lmNewHash;
1755         ntNewHash = param_ntNewHash;
1756
1757         /* Only non-trust accounts have restrictions (possibly this
1758          * test is the wrong way around, but I like to be restrictive
1759          * if possible */
1760         restrictions = !(userAccountControl & (UF_INTERDOMAIN_TRUST_ACCOUNT
1761                                                |UF_WORKSTATION_TRUST_ACCOUNT
1762                                                |UF_SERVER_TRUST_ACCOUNT));
1763
1764         if (domain_dn != NULL) {
1765                 /* pull the domain parameters */
1766                 count = gendb_search_dn(ctx, mem_ctx, domain_dn, &res,
1767                                                                 domain_attrs);
1768                 if (count != 1) {
1769                         DEBUG(2, ("samdb_set_password: Domain DN %s is invalid, for user %s\n",
1770                                   ldb_dn_get_linearized(domain_dn),
1771                                   ldb_dn_get_linearized(user_dn)));
1772                         return NT_STATUS_NO_SUCH_DOMAIN;
1773                 }
1774         } else {
1775                 /* work out the domain sid, and pull the domain from there */
1776                 domain_sid = samdb_result_sid_prefix(mem_ctx, res[0],
1777                                                                 "objectSid");
1778                 if (domain_sid == NULL) {
1779                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
1780                 }
1781
1782                 count = gendb_search(ctx, mem_ctx, NULL, &res, domain_attrs,
1783                                 "(objectSid=%s)",
1784                                 ldap_encode_ndr_dom_sid(mem_ctx, domain_sid));
1785                 if (count != 1) {
1786                         DEBUG(2, ("samdb_set_password: Could not find domain to match SID: %s, for user %s\n",
1787                                   dom_sid_string(mem_ctx, domain_sid),
1788                                   ldb_dn_get_linearized(user_dn)));
1789                         return NT_STATUS_NO_SUCH_DOMAIN;
1790                 }
1791         }
1792
1793         minPwdLength = samdb_result_uint(res[0], "minPwdLength", 0);
1794         pwdProperties = samdb_result_uint(res[0], "pwdProperties", 0);
1795         pwdHistoryLength = samdb_result_uint(res[0], "pwdHistoryLength", 0);
1796         maxPwdAge = samdb_result_int64(res[0], "maxPwdAge", 0);
1797         minPwdAge = samdb_result_int64(res[0], "minPwdAge", 0);
1798
1799         if ((userAccountControl & UF_PASSWD_NOTREQD) != 0) {
1800                 /* see [MS-ADTS] 2.2.15 */
1801                 minPwdLength = 0;
1802         }
1803
1804         if (_dominfo != NULL) {
1805                 struct samr_DomInfo1 *dominfo;
1806                 /* on failure we need to fill in the reject reasons */
1807                 dominfo = talloc(mem_ctx, struct samr_DomInfo1);
1808                 if (dominfo == NULL) {
1809                         return NT_STATUS_NO_MEMORY;
1810                 }
1811                 dominfo->min_password_length     = minPwdLength;
1812                 dominfo->password_properties     = pwdProperties;
1813                 dominfo->password_history_length = pwdHistoryLength;
1814                 dominfo->max_password_age        = maxPwdAge;
1815                 dominfo->min_password_age        = minPwdAge;
1816                 *_dominfo = dominfo;
1817         }
1818
1819         if ((restrictions != 0) && (new_password != 0)) {
1820                 char *new_pass;
1821
1822                 /* checks if the "minPwdLength" property is satisfied */
1823                 if ((restrictions != 0)
1824                         && (minPwdLength > utf16_len_n(
1825                                 new_password->data, new_password->length)/2)) {
1826                         if (reject_reason) {
1827                                 *reject_reason = SAM_PWD_CHANGE_PASSWORD_TOO_SHORT;
1828                         }
1829                         return NT_STATUS_PASSWORD_RESTRICTION;
1830                 }
1831
1832                 /* Create the NT hash */
1833                 mdfour(local_ntNewHash.hash, new_password->data,
1834                                                         new_password->length);
1835
1836                 ntNewHash = &local_ntNewHash;
1837
1838                 /* Only check complexity if we can convert it at all.  Assuming unconvertable passwords are 'strong' */
1839                 if (convert_string_talloc_convenience(mem_ctx,
1840                           lp_iconv_convenience(ldb_get_opaque(ctx, "loadparm")),
1841                           CH_UTF16, CH_UNIX,
1842                           new_password->data, new_password->length,
1843                           (void **)&new_pass, NULL, false)) {
1844
1845                         /* checks the password complexity */
1846                         if ((restrictions != 0)
1847                                 && ((pwdProperties
1848                                         & DOMAIN_PASSWORD_COMPLEX) != 0)
1849                                 && (!check_password_quality(new_pass))) {
1850                                 if (reject_reason) {
1851                                         *reject_reason = SAM_PWD_CHANGE_NOT_COMPLEX;
1852                                 }
1853                                 return NT_STATUS_PASSWORD_RESTRICTION;
1854                         }
1855
1856                         /* compute the new lm hashes (for checking history - case insenitivly!) */
1857                         if (E_deshash(new_pass, local_lmNewHash.hash)) {
1858                                 lmNewHash = &local_lmNewHash;
1859                         }
1860                 }
1861         }
1862
1863         if ((restrictions != 0) && user_change) {
1864                 /* are all password changes disallowed? */
1865                 if ((pwdProperties & DOMAIN_REFUSE_PASSWORD_CHANGE) != 0) {
1866                         if (reject_reason) {
1867                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1868                         }
1869                         return NT_STATUS_PASSWORD_RESTRICTION;
1870                 }
1871
1872                 /* can this user change the password? */
1873                 if ((userAccountControl & UF_PASSWD_CANT_CHANGE) != 0) {
1874                         if (reject_reason) {
1875                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1876                         }
1877                         return NT_STATUS_PASSWORD_RESTRICTION;
1878                 }
1879
1880                 /* Password minimum age: yes, this is a minus. The ages are in negative 100nsec units! */
1881                 if (pwdLastSet - minPwdAge > now_nt) {
1882                         if (reject_reason) {
1883                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1884                         }
1885                         return NT_STATUS_PASSWORD_RESTRICTION;
1886                 }
1887
1888                 /* check the immediately past password */
1889                 if (pwdHistoryLength > 0) {
1890                         if (lmNewHash && lmPwdHash && memcmp(lmNewHash->hash,
1891                                         lmPwdHash->hash, 16) == 0) {
1892                                 if (reject_reason) {
1893                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1894                                 }
1895                                 return NT_STATUS_PASSWORD_RESTRICTION;
1896                         }
1897                         if (ntNewHash && ntPwdHash && memcmp(ntNewHash->hash,
1898                                         ntPwdHash->hash, 16) == 0) {
1899                                 if (reject_reason) {
1900                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1901                                 }
1902                                 return NT_STATUS_PASSWORD_RESTRICTION;
1903                         }
1904                 }
1905
1906                 /* check the password history */
1907                 sambaLMPwdHistory_len = MIN(sambaLMPwdHistory_len,
1908                                                         pwdHistoryLength);
1909                 sambaNTPwdHistory_len = MIN(sambaNTPwdHistory_len,
1910                                                         pwdHistoryLength);
1911
1912                 for (i=0; lmNewHash && i<sambaLMPwdHistory_len;i++) {
1913                         if (memcmp(lmNewHash->hash, sambaLMPwdHistory[i].hash,
1914                                         16) == 0) {
1915                                 if (reject_reason) {
1916                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1917                                 }
1918                                 return NT_STATUS_PASSWORD_RESTRICTION;
1919                         }
1920                 }
1921                 for (i=0; ntNewHash && i<sambaNTPwdHistory_len;i++) {
1922                         if (memcmp(ntNewHash->hash, sambaNTPwdHistory[i].hash,
1923                                          16) == 0) {
1924                                 if (reject_reason) {
1925                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1926                                 }
1927                                 return NT_STATUS_PASSWORD_RESTRICTION;
1928                         }
1929                 }
1930         }
1931
1932 #define CHECK_RET(x) do { if (x != 0) return NT_STATUS_NO_MEMORY; } while(0)
1933
1934         /* the password is acceptable. Start forming the new fields */
1935         if (new_password != NULL) {
1936                 /* if we know the cleartext UTF16 password, then set it.
1937                  * Modules in ldb will set all the appropriate
1938                  * hashes */
1939                 CHECK_RET(ldb_msg_add_value(mod, "clearTextPassword", new_password, NULL));
1940         } else {
1941                 /* we don't have the cleartext, so set what we have of the
1942                  * hashes */
1943
1944                 if (lmNewHash) {
1945                         CHECK_RET(samdb_msg_add_hash(ctx, mem_ctx, mod, "dBCSPwd", lmNewHash));
1946                 }
1947
1948                 if (ntNewHash) {
1949                         CHECK_RET(samdb_msg_add_hash(ctx, mem_ctx, mod, "unicodePwd", ntNewHash));
1950                 }
1951         }
1952
1953         if (reject_reason) {
1954                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1955         }
1956         return NT_STATUS_OK;
1957 }
1958
1959
1960 /*
1961  * Sets the user password using plaintext UTF16 (attribute "new_password") or
1962  * LM (attribute "lmNewHash") or NT (attribute "ntNewHash") hash. Also pass
1963  * as parameter if it's a user change or not ("userChange"). The "rejectReason"
1964  * gives some more informations if the changed failed.
1965  *
1966  * This wrapper function for "samdb_set_password" takes a SID as input rather
1967  * than a user DN.
1968  *
1969  * This call encapsulates a new LDB transaction for changing the password;
1970  * therefore the user hasn't to start a new one.
1971  *
1972  * Results: NT_STATUS_OK, NT_STATUS_INTERNAL_DB_CORRUPTION,
1973  *   NT_STATUS_INVALID_PARAMETER, NT_STATUS_UNSUCCESSFUL,
1974  *   NT_STATUS_PASSWORD_RESTRICTION
1975  */
1976 NTSTATUS samdb_set_password_sid(struct ldb_context *ldb, TALLOC_CTX *mem_ctx,
1977                                 const struct dom_sid *user_sid,
1978                                 const DATA_BLOB *new_password,
1979                                 struct samr_Password *lmNewHash,
1980                                 struct samr_Password *ntNewHash,
1981                                 bool user_change,
1982                                 enum samPwdChangeReason *reject_reason,
1983                                 struct samr_DomInfo1 **_dominfo)
1984 {
1985         NTSTATUS nt_status;
1986         struct ldb_dn *user_dn;
1987         struct ldb_message *msg;
1988         int ret;
1989
1990         ret = ldb_transaction_start(ldb);
1991         if (ret != LDB_SUCCESS) {
1992                 DEBUG(1, ("Failed to start transaction: %s\n", ldb_errstring(ldb)));
1993                 return NT_STATUS_TRANSACTION_ABORTED;
1994         }
1995
1996         user_dn = samdb_search_dn(ldb, mem_ctx, NULL,
1997                                   "(&(objectSid=%s)(objectClass=user))",
1998                                   ldap_encode_ndr_dom_sid(mem_ctx, user_sid));
1999         if (!user_dn) {
2000                 ldb_transaction_cancel(ldb);
2001                 DEBUG(3, ("samdb_set_password_sid: SID %s not found in samdb, returning NO_SUCH_USER\n",
2002                           dom_sid_string(mem_ctx, user_sid)));
2003                 return NT_STATUS_NO_SUCH_USER;
2004         }
2005
2006         msg = ldb_msg_new(mem_ctx);
2007         if (msg == NULL) {
2008                 ldb_transaction_cancel(ldb);
2009                 talloc_free(user_dn);
2010                 return NT_STATUS_NO_MEMORY;
2011         }
2012
2013         msg->dn = ldb_dn_copy(msg, user_dn);
2014         if (!msg->dn) {
2015                 ldb_transaction_cancel(ldb);
2016                 talloc_free(user_dn);
2017                 talloc_free(msg);
2018                 return NT_STATUS_NO_MEMORY;
2019         }
2020
2021         nt_status = samdb_set_password(ldb, mem_ctx,
2022                                        user_dn, NULL,
2023                                        msg, new_password,
2024                                        lmNewHash, ntNewHash,
2025                                        user_change,
2026                                        reject_reason, _dominfo);
2027         if (!NT_STATUS_IS_OK(nt_status)) {
2028                 ldb_transaction_cancel(ldb);
2029                 talloc_free(user_dn);
2030                 talloc_free(msg);
2031                 return nt_status;
2032         }
2033
2034         /* modify the samdb record */
2035         ret = samdb_replace(ldb, mem_ctx, msg);
2036         if (ret != LDB_SUCCESS) {
2037                 ldb_transaction_cancel(ldb);
2038                 talloc_free(user_dn);
2039                 talloc_free(msg);
2040                 return NT_STATUS_ACCESS_DENIED;
2041         }
2042
2043         talloc_free(msg);
2044
2045         ret = ldb_transaction_commit(ldb);
2046         if (ret != LDB_SUCCESS) {
2047                 DEBUG(0,("Failed to commit transaction to change password on %s: %s\n",
2048                          ldb_dn_get_linearized(user_dn),
2049                          ldb_errstring(ldb)));
2050                 talloc_free(user_dn);
2051                 return NT_STATUS_TRANSACTION_ABORTED;
2052         }
2053
2054         talloc_free(user_dn);
2055         return NT_STATUS_OK;
2056 }
2057
2058
2059 NTSTATUS samdb_create_foreign_security_principal(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
2060                                                  struct dom_sid *sid, struct ldb_dn **ret_dn)
2061 {
2062         struct ldb_message *msg;
2063         struct ldb_dn *basedn;
2064         const char *sidstr;
2065         int ret;
2066
2067         sidstr = dom_sid_string(mem_ctx, sid);
2068         NT_STATUS_HAVE_NO_MEMORY(sidstr);
2069
2070         /* We might have to create a ForeignSecurityPrincipal, even if this user
2071          * is in our own domain */
2072
2073         msg = ldb_msg_new(mem_ctx);
2074         if (msg == NULL) {
2075                 return NT_STATUS_NO_MEMORY;
2076         }
2077
2078         /* TODO: Hmmm. This feels wrong. How do I find the base dn to
2079          * put the ForeignSecurityPrincipals? d_state->domain_dn does
2080          * not work, this is wrong for the Builtin domain, there's no
2081          * cn=For...,cn=Builtin,dc={BASEDN}.  -- vl
2082          */
2083
2084         basedn = samdb_search_dn(sam_ctx, mem_ctx, NULL,
2085                                  "(&(objectClass=container)(cn=ForeignSecurityPrincipals))");
2086
2087         if (basedn == NULL) {
2088                 DEBUG(0, ("Failed to find DN for "
2089                           "ForeignSecurityPrincipal container\n"));
2090                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
2091         }
2092
2093         /* add core elements to the ldb_message for the alias */
2094         msg->dn = ldb_dn_copy(mem_ctx, basedn);
2095         if ( ! ldb_dn_add_child_fmt(msg->dn, "CN=%s", sidstr))
2096                 return NT_STATUS_NO_MEMORY;
2097
2098         samdb_msg_add_string(sam_ctx, mem_ctx, msg,
2099                              "objectClass",
2100                              "foreignSecurityPrincipal");
2101
2102         /* create the alias */
2103         ret = ldb_add(sam_ctx, msg);
2104         if (ret != 0) {
2105                 DEBUG(0,("Failed to create foreignSecurityPrincipal "
2106                          "record %s: %s\n",
2107                          ldb_dn_get_linearized(msg->dn),
2108                          ldb_errstring(sam_ctx)));
2109                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
2110         }
2111         *ret_dn = msg->dn;
2112         return NT_STATUS_OK;
2113 }
2114
2115
2116 /*
2117   Find the DN of a domain, assuming it to be a dotted.dns name
2118 */
2119
2120 struct ldb_dn *samdb_dns_domain_to_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const char *dns_domain)
2121 {
2122         int i;
2123         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2124         const char *binary_encoded;
2125         const char **split_realm;
2126         struct ldb_dn *dn;
2127
2128         if (!tmp_ctx) {
2129                 return NULL;
2130         }
2131
2132         split_realm = (const char **)str_list_make(tmp_ctx, dns_domain, ".");
2133         if (!split_realm) {
2134                 talloc_free(tmp_ctx);
2135                 return NULL;
2136         }
2137         dn = ldb_dn_new(mem_ctx, ldb, NULL);
2138         for (i=0; split_realm[i]; i++) {
2139                 binary_encoded = ldb_binary_encode_string(tmp_ctx, split_realm[i]);
2140                 if (!ldb_dn_add_base_fmt(dn, "dc=%s", binary_encoded)) {
2141                         DEBUG(2, ("Failed to add dc=%s element to DN %s\n",
2142                                   binary_encoded, ldb_dn_get_linearized(dn)));
2143                         talloc_free(tmp_ctx);
2144                         return NULL;
2145                 }
2146         }
2147         if (!ldb_dn_validate(dn)) {
2148                 DEBUG(2, ("Failed to validated DN %s\n",
2149                           ldb_dn_get_linearized(dn)));
2150                 return NULL;
2151         }
2152         return dn;
2153 }
2154 /*
2155   Find the DN of a domain, be it the netbios or DNS name
2156 */
2157
2158 struct ldb_dn *samdb_domain_to_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx,
2159                                   const char *domain_name)
2160 {
2161         const char * const domain_ref_attrs[] = {
2162                 "ncName", NULL
2163         };
2164         const char * const domain_ref2_attrs[] = {
2165                 NULL
2166         };
2167         struct ldb_result *res_domain_ref;
2168         char *escaped_domain = ldb_binary_encode_string(mem_ctx, domain_name);
2169         /* find the domain's DN */
2170         int ret_domain = ldb_search(ldb, mem_ctx,
2171                                             &res_domain_ref,
2172                                             samdb_partitions_dn(ldb, mem_ctx),
2173                                             LDB_SCOPE_ONELEVEL,
2174                                             domain_ref_attrs,
2175                                             "(&(nETBIOSName=%s)(objectclass=crossRef))",
2176                                             escaped_domain);
2177         if (ret_domain != 0) {
2178                 return NULL;
2179         }
2180
2181         if (res_domain_ref->count == 0) {
2182                 ret_domain = ldb_search(ldb, mem_ctx,
2183                                                 &res_domain_ref,
2184                                                 samdb_dns_domain_to_dn(ldb, mem_ctx, domain_name),
2185                                                 LDB_SCOPE_BASE,
2186                                                 domain_ref2_attrs,
2187                                                 "(objectclass=domain)");
2188                 if (ret_domain != 0) {
2189                         return NULL;
2190                 }
2191
2192                 if (res_domain_ref->count == 1) {
2193                         return res_domain_ref->msgs[0]->dn;
2194                 }
2195                 return NULL;
2196         }
2197
2198         if (res_domain_ref->count > 1) {
2199                 DEBUG(0,("Found %d records matching domain [%s]\n",
2200                          ret_domain, domain_name));
2201                 return NULL;
2202         }
2203
2204         return samdb_result_dn(ldb, mem_ctx, res_domain_ref->msgs[0], "nCName", NULL);
2205
2206 }
2207
2208
2209 /*
2210   use a GUID to find a DN
2211  */
2212 int dsdb_find_dn_by_guid(struct ldb_context *ldb,
2213                          TALLOC_CTX *mem_ctx,
2214                          const char *guid_str, struct ldb_dn **dn)
2215 {
2216         int ret;
2217         struct ldb_result *res;
2218         const char *attrs[] = { NULL };
2219         struct ldb_request *search_req;
2220         char *expression;
2221         struct ldb_search_options_control *options;
2222
2223         expression = talloc_asprintf(mem_ctx, "objectGUID=%s", guid_str);
2224         if (!expression) {
2225                 DEBUG(0, (__location__ ": out of memory\n"));
2226                 return LDB_ERR_OPERATIONS_ERROR;
2227         }
2228
2229         res = talloc_zero(mem_ctx, struct ldb_result);
2230         if (!res) {
2231                 DEBUG(0, (__location__ ": out of memory\n"));
2232                 return LDB_ERR_OPERATIONS_ERROR;
2233         }
2234
2235         ret = ldb_build_search_req(&search_req, ldb, mem_ctx,
2236                                    ldb_get_default_basedn(ldb),
2237                                    LDB_SCOPE_SUBTREE,
2238                                    expression, attrs,
2239                                    NULL,
2240                                    res, ldb_search_default_callback,
2241                                    NULL);
2242         if (ret != LDB_SUCCESS) {
2243                 return ret;
2244         }
2245
2246         /* we need to cope with cross-partition links, so search for
2247            the GUID over all partitions */
2248         options = talloc(search_req, struct ldb_search_options_control);
2249         if (options == NULL) {
2250                 DEBUG(0, (__location__ ": out of memory\n"));
2251                 return LDB_ERR_OPERATIONS_ERROR;
2252         }
2253         options->search_options = LDB_SEARCH_OPTION_PHANTOM_ROOT;
2254
2255         ret = ldb_request_add_control(search_req, LDB_CONTROL_EXTENDED_DN_OID, true, NULL);
2256         if (ret != LDB_SUCCESS) {
2257                 return ret;
2258         }
2259
2260         ret = ldb_request_add_control(search_req,
2261                                       LDB_CONTROL_SEARCH_OPTIONS_OID,
2262                                       true, options);
2263         if (ret != LDB_SUCCESS) {
2264                 return ret;
2265         }
2266
2267         ret = ldb_request(ldb, search_req);
2268         if (ret != LDB_SUCCESS) {
2269                 return ret;
2270         }
2271
2272         ret = ldb_wait(search_req->handle, LDB_WAIT_ALL);
2273         if (ret != LDB_SUCCESS) {
2274                 return ret;
2275         }
2276
2277         /* this really should be exactly 1, but there is a bug in the
2278            partitions module that can return two here with the
2279            search_options control set */
2280         if (res->count < 1) {
2281                 return LDB_ERR_NO_SUCH_OBJECT;
2282         }
2283
2284         *dn = res->msgs[0]->dn;
2285
2286         return LDB_SUCCESS;
2287 }
2288
2289 /*
2290   search for attrs on one DN, allowing for deleted objects
2291  */
2292 int dsdb_search_dn_with_deleted(struct ldb_context *ldb,
2293                                 TALLOC_CTX *mem_ctx,
2294                                 struct ldb_result **_res,
2295                                 struct ldb_dn *basedn,
2296                                 const char * const *attrs)
2297 {
2298         int ret;
2299         struct ldb_request *req;
2300         TALLOC_CTX *tmp_ctx;
2301         struct ldb_result *res;
2302
2303         tmp_ctx = talloc_new(mem_ctx);
2304
2305         res = talloc_zero(tmp_ctx, struct ldb_result);
2306         if (!res) {
2307                 return LDB_ERR_OPERATIONS_ERROR;
2308         }
2309
2310         ret = ldb_build_search_req(&req, ldb, tmp_ctx,
2311                                    basedn,
2312                                    LDB_SCOPE_BASE,
2313                                    NULL,
2314                                    attrs,
2315                                    NULL,
2316                                    res,
2317                                    ldb_search_default_callback,
2318                                    NULL);
2319         if (ret != LDB_SUCCESS) {
2320                 talloc_free(tmp_ctx);
2321                 return ret;
2322         }
2323
2324         ret = ldb_request_add_control(req, LDB_CONTROL_SHOW_DELETED_OID, true, NULL);
2325         if (ret != LDB_SUCCESS) {
2326                 return ret;
2327         }
2328
2329         ret = ldb_request(ldb, req);
2330         if (ret == LDB_SUCCESS) {
2331                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2332         }
2333
2334         talloc_free(req);
2335         *_res = talloc_steal(mem_ctx, res);
2336         return ret;
2337 }
2338
2339
2340 /*
2341   use a DN to find a GUID
2342  */
2343 int dsdb_find_guid_by_dn(struct ldb_context *ldb,
2344                          struct ldb_dn *dn, struct GUID *guid)
2345 {
2346         int ret;
2347         struct ldb_result *res;
2348         const char *attrs[] = { "objectGUID", NULL };
2349         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2350
2351         ret = dsdb_search_dn_with_deleted(ldb, tmp_ctx, &res, dn, attrs);
2352         if (ret != LDB_SUCCESS) {
2353                 talloc_free(tmp_ctx);
2354                 return ret;
2355         }
2356         if (res->count < 1) {
2357                 talloc_free(tmp_ctx);
2358                 return LDB_ERR_NO_SUCH_OBJECT;
2359         }
2360         *guid = samdb_result_guid(res->msgs[0], "objectGUID");
2361         talloc_free(tmp_ctx);
2362         return LDB_SUCCESS;
2363 }
2364
2365
2366
2367 /*
2368  adds the given GUID to the given ldb_message. This value is added
2369  for the given attr_name (may be either "objectGUID" or "parentGUID").
2370  */
2371 int dsdb_msg_add_guid(struct ldb_message *msg,
2372                 struct GUID *guid,
2373                 const char *attr_name)
2374 {
2375         int ret;
2376         struct ldb_val v;
2377         NTSTATUS status;
2378         TALLOC_CTX *tmp_ctx =  talloc_init("dsdb_msg_add_guid");
2379
2380         status = GUID_to_ndr_blob(guid, tmp_ctx, &v);
2381         if (!NT_STATUS_IS_OK(status)) {
2382                 ret = LDB_ERR_OPERATIONS_ERROR;
2383                 goto done;
2384         }
2385
2386         ret = ldb_msg_add_steal_value(msg, attr_name, &v);
2387         if (ret != LDB_SUCCESS) {
2388                 DEBUG(4,(__location__ ": Failed to add %s to the message\n",
2389                                          attr_name));
2390                 goto done;
2391         }
2392
2393         ret = LDB_SUCCESS;
2394
2395 done:
2396         talloc_free(tmp_ctx);
2397         return ret;
2398
2399 }
2400
2401
2402 /*
2403   use a DN to find a SID
2404  */
2405 int dsdb_find_sid_by_dn(struct ldb_context *ldb,
2406                         struct ldb_dn *dn, struct dom_sid *sid)
2407 {
2408         int ret;
2409         struct ldb_result *res;
2410         const char *attrs[] = { "objectSID", NULL };
2411         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2412         struct dom_sid *s;
2413
2414         ZERO_STRUCTP(sid);
2415
2416         ret = dsdb_search_dn_with_deleted(ldb, tmp_ctx, &res, dn, attrs);
2417         if (ret != LDB_SUCCESS) {
2418                 talloc_free(tmp_ctx);
2419                 return ret;
2420         }
2421         if (res->count < 1) {
2422                 talloc_free(tmp_ctx);
2423                 return LDB_ERR_NO_SUCH_OBJECT;
2424         }
2425         s = samdb_result_dom_sid(tmp_ctx, res->msgs[0], "objectSID");
2426         if (s == NULL) {
2427                 talloc_free(tmp_ctx);
2428                 return LDB_ERR_NO_SUCH_OBJECT;
2429         }
2430         *sid = *s;
2431         talloc_free(tmp_ctx);
2432         return LDB_SUCCESS;
2433 }
2434
2435
2436
2437 /*
2438   load a repsFromTo blob list for a given partition GUID
2439   attr must be "repsFrom" or "repsTo"
2440  */
2441 WERROR dsdb_loadreps(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
2442                      const char *attr, struct repsFromToBlob **r, uint32_t *count)
2443 {
2444         const char *attrs[] = { attr, NULL };
2445         struct ldb_result *res = NULL;
2446         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2447         int i;
2448         struct ldb_message_element *el;
2449
2450         *r = NULL;
2451         *count = 0;
2452
2453         if (ldb_search(sam_ctx, tmp_ctx, &res, dn, LDB_SCOPE_BASE, attrs, NULL) != LDB_SUCCESS ||
2454             res->count < 1) {
2455                 DEBUG(0,("dsdb_loadreps: failed to read partition object\n"));
2456                 talloc_free(tmp_ctx);
2457                 return WERR_DS_DRA_INTERNAL_ERROR;
2458         }
2459
2460         el = ldb_msg_find_element(res->msgs[0], attr);
2461         if (el == NULL) {
2462                 /* it's OK to be empty */
2463                 talloc_free(tmp_ctx);
2464                 return WERR_OK;
2465         }
2466
2467         *count = el->num_values;
2468         *r = talloc_array(mem_ctx, struct repsFromToBlob, *count);
2469         if (*r == NULL) {
2470                 talloc_free(tmp_ctx);
2471                 return WERR_DS_DRA_INTERNAL_ERROR;
2472         }
2473
2474         for (i=0; i<(*count); i++) {
2475                 enum ndr_err_code ndr_err;
2476                 ndr_err = ndr_pull_struct_blob(&el->values[i],
2477                                                mem_ctx, lp_iconv_convenience(ldb_get_opaque(sam_ctx, "loadparm")),
2478                                                &(*r)[i],
2479                                                (ndr_pull_flags_fn_t)ndr_pull_repsFromToBlob);
2480                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
2481                         talloc_free(tmp_ctx);
2482                         return WERR_DS_DRA_INTERNAL_ERROR;
2483                 }
2484         }
2485
2486         talloc_free(tmp_ctx);
2487
2488         return WERR_OK;
2489 }
2490
2491 /*
2492   save the repsFromTo blob list for a given partition GUID
2493   attr must be "repsFrom" or "repsTo"
2494  */
2495 WERROR dsdb_savereps(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
2496                      const char *attr, struct repsFromToBlob *r, uint32_t count)
2497 {
2498         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2499         struct ldb_message *msg;
2500         struct ldb_message_element *el;
2501         int i;
2502
2503         msg = ldb_msg_new(tmp_ctx);
2504         msg->dn = dn;
2505         if (ldb_msg_add_empty(msg, attr, LDB_FLAG_MOD_REPLACE, &el) != LDB_SUCCESS) {
2506                 goto failed;
2507         }
2508
2509         el->values = talloc_array(msg, struct ldb_val, count);
2510         if (!el->values) {
2511                 goto failed;
2512         }
2513
2514         for (i=0; i<count; i++) {
2515                 struct ldb_val v;
2516                 enum ndr_err_code ndr_err;
2517
2518                 ndr_err = ndr_push_struct_blob(&v, tmp_ctx, lp_iconv_convenience(ldb_get_opaque(sam_ctx, "loadparm")),
2519                                                &r[i],
2520                                                (ndr_push_flags_fn_t)ndr_push_repsFromToBlob);
2521                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
2522                         goto failed;
2523                 }
2524
2525                 el->num_values++;
2526                 el->values[i] = v;
2527         }
2528
2529         if (ldb_modify(sam_ctx, msg) != LDB_SUCCESS) {
2530                 DEBUG(0,("Failed to store %s - %s\n", attr, ldb_errstring(sam_ctx)));
2531                 goto failed;
2532         }
2533
2534         talloc_free(tmp_ctx);
2535
2536         return WERR_OK;
2537
2538 failed:
2539         talloc_free(tmp_ctx);
2540         return WERR_DS_DRA_INTERNAL_ERROR;
2541 }
2542
2543
2544 /*
2545   load the uSNHighest attribute from the @REPLCHANGED object for a
2546   partition
2547  */
2548 int dsdb_load_partition_usn(struct ldb_context *ldb, struct ldb_dn *dn, uint64_t *uSN)
2549 {
2550         struct ldb_request *req;
2551         int ret;
2552         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2553         struct dsdb_control_current_partition *p_ctrl;
2554         struct ldb_result *res;
2555
2556         res = talloc_zero(tmp_ctx, struct ldb_result);
2557         if (!res) {
2558                 talloc_free(tmp_ctx);
2559                 return LDB_ERR_OPERATIONS_ERROR;
2560         }
2561
2562         ret = ldb_build_search_req(&req, ldb, tmp_ctx,
2563                                    ldb_dn_new(tmp_ctx, ldb, "@REPLCHANGED"),
2564                                    LDB_SCOPE_BASE,
2565                                    NULL, NULL,
2566                                    NULL,
2567                                    res, ldb_search_default_callback,
2568                                    NULL);
2569         if (ret != LDB_SUCCESS) {
2570                 talloc_free(tmp_ctx);
2571                 return ret;
2572         }
2573
2574         p_ctrl = talloc(req, struct dsdb_control_current_partition);
2575         if (p_ctrl == NULL) {
2576                 talloc_free(res);
2577                 return LDB_ERR_OPERATIONS_ERROR;
2578         }
2579         p_ctrl->version = DSDB_CONTROL_CURRENT_PARTITION_VERSION;
2580         p_ctrl->dn = dn;
2581
2582
2583         ret = ldb_request_add_control(req,
2584                                       DSDB_CONTROL_CURRENT_PARTITION_OID,
2585                                       false, p_ctrl);
2586         if (ret != LDB_SUCCESS) {
2587                 talloc_free(tmp_ctx);
2588                 return ret;
2589         }
2590
2591         /* Run the new request */
2592         ret = ldb_request(ldb, req);
2593
2594         if (ret == LDB_SUCCESS) {
2595                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2596         }
2597
2598         if (ret == LDB_ERR_NO_SUCH_OBJECT) {
2599                 /* it hasn't been created yet, which means
2600                    an implicit value of zero */
2601                 *uSN = 0;
2602                 talloc_free(tmp_ctx);
2603                 return LDB_SUCCESS;
2604         }
2605
2606         if (ret != LDB_SUCCESS) {
2607                 talloc_free(tmp_ctx);
2608                 return ret;
2609         }
2610
2611         if (res->count < 1) {
2612                 *uSN = 0;
2613         } else {
2614                 *uSN = ldb_msg_find_attr_as_uint64(res->msgs[0], "uSNHighest", 0);
2615         }
2616
2617         talloc_free(tmp_ctx);
2618
2619         return LDB_SUCCESS;
2620 }
2621
2622 /*
2623   save the uSNHighest attribute in the @REPLCHANGED object for a
2624   partition
2625  */
2626 int dsdb_save_partition_usn(struct ldb_context *ldb, struct ldb_dn *dn, uint64_t uSN)
2627 {
2628         struct ldb_request *req;
2629         struct ldb_message *msg;
2630         struct dsdb_control_current_partition *p_ctrl;
2631         int ret;
2632
2633         msg = ldb_msg_new(ldb);
2634         if (msg == NULL) {
2635                 return LDB_ERR_OPERATIONS_ERROR;
2636         }
2637
2638         msg->dn = ldb_dn_new(msg, ldb, "@REPLCHANGED");
2639         if (msg->dn == NULL) {
2640                 talloc_free(msg);
2641                 return LDB_ERR_OPERATIONS_ERROR;
2642         }
2643
2644         ret = ldb_msg_add_fmt(msg, "uSNHighest", "%llu", (unsigned long long)uSN);
2645         if (ret != LDB_SUCCESS) {
2646                 talloc_free(msg);
2647                 return ret;
2648         }
2649         msg->elements[0].flags = LDB_FLAG_MOD_REPLACE;
2650
2651
2652         p_ctrl = talloc(msg, struct dsdb_control_current_partition);
2653         if (p_ctrl == NULL) {
2654                 talloc_free(msg);
2655                 return LDB_ERR_OPERATIONS_ERROR;
2656         }
2657         p_ctrl->version = DSDB_CONTROL_CURRENT_PARTITION_VERSION;
2658         p_ctrl->dn = dn;
2659
2660         ret = ldb_build_mod_req(&req, ldb, msg,
2661                                 msg,
2662                                 NULL,
2663                                 NULL, ldb_op_default_callback,
2664                                 NULL);
2665 again:
2666         if (ret != LDB_SUCCESS) {
2667                 talloc_free(msg);
2668                 return ret;
2669         }
2670
2671         ret = ldb_request_add_control(req,
2672                                       DSDB_CONTROL_CURRENT_PARTITION_OID,
2673                                       false, p_ctrl);
2674         if (ret != LDB_SUCCESS) {
2675                 talloc_free(msg);
2676                 return ret;
2677         }
2678
2679         /* Run the new request */
2680         ret = ldb_request(ldb, req);
2681
2682         if (ret == LDB_SUCCESS) {
2683                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2684         }
2685         if (ret == LDB_ERR_NO_SUCH_OBJECT) {
2686                 ret = ldb_build_add_req(&req, ldb, msg,
2687                                         msg,
2688                                         NULL,
2689                                         NULL, ldb_op_default_callback,
2690                                         NULL);
2691                 goto again;
2692         }
2693
2694         talloc_free(msg);
2695
2696         return ret;
2697 }
2698
2699 int drsuapi_DsReplicaCursor2_compare(const struct drsuapi_DsReplicaCursor2 *c1,
2700                                                    const struct drsuapi_DsReplicaCursor2 *c2)
2701 {
2702         return GUID_compare(&c1->source_dsa_invocation_id, &c2->source_dsa_invocation_id);
2703 }
2704
2705 /*
2706   see if we are a RODC
2707
2708   TODO: This should take a sam_ctx, and lookup the right object (with
2709   a cache)
2710 */
2711 bool samdb_rodc(struct loadparm_context *lp_ctx)
2712 {
2713         return lp_parm_bool(lp_ctx, NULL, "repl", "RODC", false);
2714 }
2715
2716
2717 /*
2718   return NTDS options flags. See MS-ADTS 7.1.1.2.2.1.2.1.1
2719
2720   flags are DS_NTDS_OPTION_*
2721 */
2722 int samdb_ntds_options(struct ldb_context *ldb, uint32_t *options)
2723 {
2724         TALLOC_CTX *tmp_ctx;
2725         const char *attrs[] = { "options", NULL };
2726         int ret;
2727         struct ldb_result *res;
2728
2729         tmp_ctx = talloc_new(ldb);
2730         if (tmp_ctx == NULL) {
2731                 goto failed;
2732         }
2733
2734         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
2735         if (ret) {
2736                 goto failed;
2737         }
2738
2739         if (res->count != 1) {
2740                 goto failed;
2741         }
2742
2743         *options = samdb_result_uint(res->msgs[0], "options", 0);
2744
2745         talloc_free(tmp_ctx);
2746
2747         return LDB_SUCCESS;
2748
2749 failed:
2750         DEBUG(1,("Failed to find our own NTDS Settings objectGUID in the ldb!\n"));
2751         talloc_free(tmp_ctx);
2752         return LDB_ERR_NO_SUCH_OBJECT;
2753 }
2754
2755 /*
2756  * Function which generates a "lDAPDisplayName" attribute from a "CN" one.
2757  * Algorithm implemented according to MS-ADTS 3.1.1.2.3.4
2758  */
2759 const char *samdb_cn_to_lDAPDisplayName(TALLOC_CTX *mem_ctx, const char *cn)
2760 {
2761         char **tokens, *ret;
2762         size_t i;
2763
2764         tokens = str_list_make(mem_ctx, cn, " -_");
2765         if (tokens == NULL)
2766                 return NULL;
2767
2768         /* "tolower()" and "toupper()" should also work properly on 0x00 */
2769         tokens[0][0] = tolower(tokens[0][0]);
2770         for (i = 1; i < str_list_length((const char **)tokens); i++)
2771                 tokens[i][0] = toupper(tokens[i][0]);
2772
2773         ret = talloc_strdup(mem_ctx, tokens[0]);
2774         for (i = 1; i < str_list_length((const char **)tokens); i++)
2775                 ret = talloc_asprintf_append_buffer(ret, "%s", tokens[i]);
2776
2777         talloc_free(tokens);
2778
2779         return ret;
2780 }
2781
2782 /*
2783   return domain functional level
2784   returns DS_DOMAIN_FUNCTION_*
2785  */
2786 int dsdb_functional_level(struct ldb_context *ldb)
2787 {
2788         int *domainFunctionality =
2789                 talloc_get_type(ldb_get_opaque(ldb, "domainFunctionality"), int);
2790         if (!domainFunctionality) {
2791                 DEBUG(0,(__location__ ": WARNING: domainFunctionality not setup\n"));
2792                 return DS_DOMAIN_FUNCTION_2000;
2793         }
2794         return *domainFunctionality;
2795 }
2796
2797 /*
2798   set a GUID in an extended DN structure
2799  */
2800 int dsdb_set_extended_dn_guid(struct ldb_dn *dn, const struct GUID *guid, const char *component_name)
2801 {
2802         struct ldb_val v;
2803         NTSTATUS status;
2804         int ret;
2805
2806         status = GUID_to_ndr_blob(guid, dn, &v);
2807         if (!NT_STATUS_IS_OK(status)) {
2808                 return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
2809         }
2810
2811         ret = ldb_dn_set_extended_component(dn, component_name, &v);
2812         data_blob_free(&v);
2813         return ret;
2814 }
2815
2816 /*
2817   return a GUID from a extended DN structure
2818  */
2819 NTSTATUS dsdb_get_extended_dn_guid(struct ldb_dn *dn, struct GUID *guid, const char *component_name)
2820 {
2821         const struct ldb_val *v;
2822
2823         v = ldb_dn_get_extended_component(dn, component_name);
2824         if (v == NULL) {
2825                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
2826         }
2827
2828         return GUID_from_ndr_blob(v, guid);
2829 }
2830
2831 /*
2832   return a uint64_t from a extended DN structure
2833  */
2834 NTSTATUS dsdb_get_extended_dn_uint64(struct ldb_dn *dn, uint64_t *val, const char *component_name)
2835 {
2836         const struct ldb_val *v;
2837         char *s;
2838
2839         v = ldb_dn_get_extended_component(dn, component_name);
2840         if (v == NULL) {
2841                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
2842         }
2843         s = talloc_strndup(dn, (const char *)v->data, v->length);
2844         NT_STATUS_HAVE_NO_MEMORY(s);
2845
2846         *val = strtoull(s, NULL, 0);
2847
2848         talloc_free(s);
2849         return NT_STATUS_OK;
2850 }
2851
2852 /*
2853   return a NTTIME from a extended DN structure
2854  */
2855 NTSTATUS dsdb_get_extended_dn_nttime(struct ldb_dn *dn, NTTIME *nttime, const char *component_name)
2856 {
2857         return dsdb_get_extended_dn_uint64(dn, nttime, component_name);
2858 }
2859
2860 /*
2861   return a uint32_t from a extended DN structure
2862  */
2863 NTSTATUS dsdb_get_extended_dn_uint32(struct ldb_dn *dn, uint32_t *val, const char *component_name)
2864 {
2865         const struct ldb_val *v;
2866         char *s;
2867
2868         v = ldb_dn_get_extended_component(dn, component_name);
2869         if (v == NULL) {
2870                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
2871         }
2872
2873         s = talloc_strndup(dn, (const char *)v->data, v->length);
2874         NT_STATUS_HAVE_NO_MEMORY(s);
2875
2876         *val = strtoul(s, NULL, 0);
2877
2878         talloc_free(s);
2879         return NT_STATUS_OK;
2880 }
2881
2882 /*
2883   return true if a ldb_val containing a DN in storage form is deleted
2884  */
2885 bool dsdb_dn_is_deleted_val(struct ldb_val *val)
2886 {
2887         /* this relies on the sort order and exact format of
2888            linearized extended DNs */
2889         if (val->length >= 12 &&
2890             strncmp((const char *)val->data, "<DELETED=1>;", 12) == 0) {
2891                 return true;
2892         }
2893         return false;
2894 }
2895
2896 /*
2897   return true if a ldb_val containing a DN in storage form is
2898   in the upgraded w2k3 linked attribute format
2899  */
2900 bool dsdb_dn_is_upgraded_link_val(struct ldb_val *val)
2901 {
2902         return memmem(val->data, val->length, "<RMD_ADDTIME=", 13) != NULL;
2903 }
2904
2905 /*
2906   return a DN for a wellknown GUID
2907  */
2908 int dsdb_wellknown_dn(struct ldb_context *samdb, TALLOC_CTX *mem_ctx,
2909                       struct ldb_dn *nc_root, const char *wk_guid,
2910                       struct ldb_dn **wkguid_dn)
2911 {
2912         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2913         const char *attrs[] = { NULL };
2914         int ret;
2915         struct ldb_dn *dn;
2916         struct ldb_result *res;
2917
2918         /* construct the magic WKGUID DN */
2919         dn = ldb_dn_new_fmt(tmp_ctx, samdb, "<WKGUID=%s,%s>",
2920                             wk_guid, ldb_dn_get_linearized(nc_root));
2921         if (!wkguid_dn) {
2922                 talloc_free(tmp_ctx);
2923                 return LDB_ERR_OPERATIONS_ERROR;
2924         }
2925
2926         ret = dsdb_search_dn_with_deleted(samdb, tmp_ctx, &res, dn, attrs);
2927         if (ret != LDB_SUCCESS) {
2928                 talloc_free(tmp_ctx);
2929                 return ret;
2930         }
2931
2932         (*wkguid_dn) = talloc_steal(mem_ctx, res->msgs[0]->dn);
2933         talloc_free(tmp_ctx);
2934         return LDB_SUCCESS;
2935 }
2936
2937
2938 static int dsdb_dn_compare_ptrs(struct ldb_dn **dn1, struct ldb_dn **dn2)
2939 {
2940         return ldb_dn_compare(*dn1, *dn2);
2941 }
2942
2943 /*
2944   find a NC root given a DN within the NC
2945  */
2946 int dsdb_find_nc_root(struct ldb_context *samdb, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
2947                       struct ldb_dn **nc_root)
2948 {
2949         const char *root_attrs[] = { "namingContexts", NULL };
2950         TALLOC_CTX *tmp_ctx;
2951         int ret;
2952         struct ldb_message_element *el;
2953         struct ldb_result *root_res;
2954         int i;
2955         struct ldb_dn **nc_dns;
2956
2957         tmp_ctx = talloc_new(samdb);
2958         if (tmp_ctx == NULL) {
2959                 return LDB_ERR_OPERATIONS_ERROR;
2960         }
2961
2962         ret = ldb_search(samdb, tmp_ctx, &root_res,
2963                          ldb_dn_new(tmp_ctx, samdb, ""), LDB_SCOPE_BASE, root_attrs, NULL);
2964         if (ret) {
2965                 DEBUG(1,("Searching for namingContexts in rootDSE failed: %s\n", ldb_errstring(samdb)));
2966                 talloc_free(tmp_ctx);
2967                 return ret;
2968        }
2969
2970        el = ldb_msg_find_element(root_res->msgs[0], "namingContexts");
2971        if (!el) {
2972                DEBUG(1,("Finding namingContexts element in root_res failed: %s\n",
2973                         ldb_errstring(samdb)));
2974                talloc_free(tmp_ctx);
2975                return LDB_ERR_NO_SUCH_ATTRIBUTE;
2976        }
2977
2978        nc_dns = talloc_array(tmp_ctx, struct ldb_dn *, el->num_values);
2979        if (!nc_dns) {
2980                talloc_free(tmp_ctx);
2981                return LDB_ERR_OPERATIONS_ERROR;
2982        }
2983
2984        for (i=0; i<el->num_values; i++) {
2985                nc_dns[i] = ldb_dn_from_ldb_val(nc_dns, samdb, &el->values[i]);
2986                if (nc_dns[i] == NULL) {
2987                        talloc_free(tmp_ctx);
2988                        return LDB_ERR_OPERATIONS_ERROR;
2989                }
2990        }
2991
2992        qsort(nc_dns, el->num_values, sizeof(nc_dns[0]), (comparison_fn_t)dsdb_dn_compare_ptrs);
2993
2994        for (i=0; i<el->num_values; i++) {
2995                if (ldb_dn_compare_base(nc_dns[i], dn) == 0) {
2996                        (*nc_root) = talloc_steal(mem_ctx, nc_dns[i]);
2997                        talloc_free(tmp_ctx);
2998                        return LDB_SUCCESS;
2999                }
3000        }
3001
3002        talloc_free(tmp_ctx);
3003        return LDB_ERR_NO_SUCH_OBJECT;
3004 }
3005
3006
3007 /*
3008   find the deleted objects DN for any object, by looking for the NC
3009   root, then looking up the wellknown GUID
3010  */
3011 int dsdb_get_deleted_objects_dn(struct ldb_context *ldb,
3012                                 TALLOC_CTX *mem_ctx, struct ldb_dn *obj_dn,
3013                                 struct ldb_dn **do_dn)
3014 {
3015         struct ldb_dn *nc_root;
3016         int ret;
3017
3018         ret = dsdb_find_nc_root(ldb, mem_ctx, obj_dn, &nc_root);
3019         if (ret != LDB_SUCCESS) {
3020                 return ret;
3021         }
3022
3023         ret = dsdb_wellknown_dn(ldb, mem_ctx, nc_root, DS_GUID_DELETED_OBJECTS_CONTAINER, do_dn);
3024         talloc_free(nc_root);
3025         return ret;
3026 }
3027
3028 /*
3029   return the tombstoneLifetime, in days
3030  */
3031 int dsdb_tombstone_lifetime(struct ldb_context *ldb, uint32_t *lifetime)
3032 {
3033         struct ldb_dn *dn;
3034         dn = samdb_config_dn(ldb);
3035         if (!dn) {
3036                 return LDB_ERR_NO_SUCH_OBJECT;
3037         }
3038         dn = ldb_dn_copy(ldb, dn);
3039         if (!dn) {
3040                 return LDB_ERR_OPERATIONS_ERROR;
3041         }
3042         /* see MS-ADTS section 7.1.1.2.4.1.1. There doesn't appear to
3043          be a wellknown GUID for this */
3044         if (!ldb_dn_add_child_fmt(dn, "CN=Directory Service,CN=Windows NT")) {
3045                 talloc_free(dn);
3046                 return LDB_ERR_OPERATIONS_ERROR;
3047         }
3048
3049         *lifetime = samdb_search_uint(ldb, dn, 180, dn, "tombstoneLifetime", "objectClass=nTDSService");
3050         talloc_free(dn);
3051         return LDB_SUCCESS;
3052 }