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   return a default security descriptor
1001 */
1002 struct security_descriptor *samdb_default_security_descriptor(TALLOC_CTX *mem_ctx)
1003 {
1004         struct security_descriptor *sd;
1005
1006         sd = security_descriptor_initialise(mem_ctx);
1007
1008         return sd;
1009 }
1010
1011 struct ldb_dn *samdb_base_dn(struct ldb_context *sam_ctx)
1012 {
1013         return ldb_get_default_basedn(sam_ctx);
1014 }
1015
1016 struct ldb_dn *samdb_config_dn(struct ldb_context *sam_ctx)
1017 {
1018         return ldb_get_config_basedn(sam_ctx);
1019 }
1020
1021 struct ldb_dn *samdb_schema_dn(struct ldb_context *sam_ctx)
1022 {
1023         return ldb_get_schema_basedn(sam_ctx);
1024 }
1025
1026 struct ldb_dn *samdb_aggregate_schema_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
1027 {
1028         struct ldb_dn *schema_dn = ldb_get_schema_basedn(sam_ctx);
1029         struct ldb_dn *aggregate_dn;
1030         if (!schema_dn) {
1031                 return NULL;
1032         }
1033
1034         aggregate_dn = ldb_dn_copy(mem_ctx, schema_dn);
1035         if (!aggregate_dn) {
1036                 return NULL;
1037         }
1038         if (!ldb_dn_add_child_fmt(aggregate_dn, "CN=Aggregate")) {
1039                 return NULL;
1040         }
1041         return aggregate_dn;
1042 }
1043
1044 struct ldb_dn *samdb_root_dn(struct ldb_context *sam_ctx)
1045 {
1046         return ldb_get_root_basedn(sam_ctx);
1047 }
1048
1049 struct ldb_dn *samdb_partitions_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
1050 {
1051         struct ldb_dn *new_dn;
1052
1053         new_dn = ldb_dn_copy(mem_ctx, samdb_config_dn(sam_ctx));
1054         if ( ! ldb_dn_add_child_fmt(new_dn, "CN=Partitions")) {
1055                 talloc_free(new_dn);
1056                 return NULL;
1057         }
1058         return new_dn;
1059 }
1060
1061 struct ldb_dn *samdb_sites_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
1062 {
1063         struct ldb_dn *new_dn;
1064
1065         new_dn = ldb_dn_copy(mem_ctx, samdb_config_dn(sam_ctx));
1066         if ( ! ldb_dn_add_child_fmt(new_dn, "CN=Sites")) {
1067                 talloc_free(new_dn);
1068                 return NULL;
1069         }
1070         return new_dn;
1071 }
1072
1073 /*
1074   work out the domain sid for the current open ldb
1075 */
1076 const struct dom_sid *samdb_domain_sid(struct ldb_context *ldb)
1077 {
1078         TALLOC_CTX *tmp_ctx;
1079         const struct dom_sid *domain_sid;
1080         const char *attrs[] = {
1081                 "objectSid",
1082                 NULL
1083         };
1084         struct ldb_result *res;
1085         int ret;
1086
1087         /* see if we have a cached copy */
1088         domain_sid = (struct dom_sid *)ldb_get_opaque(ldb, "cache.domain_sid");
1089         if (domain_sid) {
1090                 return domain_sid;
1091         }
1092
1093         tmp_ctx = talloc_new(ldb);
1094         if (tmp_ctx == NULL) {
1095                 goto failed;
1096         }
1097
1098         ret = ldb_search(ldb, tmp_ctx, &res, ldb_get_default_basedn(ldb), LDB_SCOPE_BASE, attrs, "objectSid=*");
1099
1100         if (ret != LDB_SUCCESS) {
1101                 goto failed;
1102         }
1103
1104         if (res->count != 1) {
1105                 goto failed;
1106         }
1107
1108         domain_sid = samdb_result_dom_sid(tmp_ctx, res->msgs[0], "objectSid");
1109         if (domain_sid == NULL) {
1110                 goto failed;
1111         }
1112
1113         /* cache the domain_sid in the ldb */
1114         if (ldb_set_opaque(ldb, "cache.domain_sid", discard_const_p(struct dom_sid, domain_sid)) != LDB_SUCCESS) {
1115                 goto failed;
1116         }
1117
1118         talloc_steal(ldb, domain_sid);
1119         talloc_free(tmp_ctx);
1120
1121         return domain_sid;
1122
1123 failed:
1124         DEBUG(1,("Failed to find domain_sid for open ldb\n"));
1125         talloc_free(tmp_ctx);
1126         return NULL;
1127 }
1128
1129 bool samdb_set_domain_sid(struct ldb_context *ldb, const struct dom_sid *dom_sid_in)
1130 {
1131         TALLOC_CTX *tmp_ctx;
1132         struct dom_sid *dom_sid_new;
1133         struct dom_sid *dom_sid_old;
1134
1135         /* see if we have a cached copy */
1136         dom_sid_old = talloc_get_type(ldb_get_opaque(ldb,
1137                                                      "cache.domain_sid"), struct dom_sid);
1138
1139         tmp_ctx = talloc_new(ldb);
1140         if (tmp_ctx == NULL) {
1141                 goto failed;
1142         }
1143
1144         dom_sid_new = dom_sid_dup(tmp_ctx, dom_sid_in);
1145         if (!dom_sid_new) {
1146                 goto failed;
1147         }
1148
1149         /* cache the domain_sid in the ldb */
1150         if (ldb_set_opaque(ldb, "cache.domain_sid", dom_sid_new) != LDB_SUCCESS) {
1151                 goto failed;
1152         }
1153
1154         talloc_steal(ldb, dom_sid_new);
1155         talloc_free(tmp_ctx);
1156         talloc_free(dom_sid_old);
1157
1158         return true;
1159
1160 failed:
1161         DEBUG(1,("Failed to set our own cached domain SID in the ldb!\n"));
1162         talloc_free(tmp_ctx);
1163         return false;
1164 }
1165
1166 /* Obtain the short name of the flexible single master operator
1167  * (FSMO), such as the PDC Emulator */
1168 const char *samdb_result_fsmo_name(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
1169                              const char *attr)
1170 {
1171         /* Format is cn=NTDS Settings,cn=<NETBIOS name of FSMO>,.... */
1172         struct ldb_dn *fsmo_dn = ldb_msg_find_attr_as_dn(ldb, mem_ctx, msg, attr);
1173         const struct ldb_val *val = ldb_dn_get_component_val(fsmo_dn, 1);
1174         const char *name = ldb_dn_get_component_name(fsmo_dn, 1);
1175
1176         if (!name || (ldb_attr_cmp(name, "cn") != 0)) {
1177                 /* Ensure this matches the format.  This gives us a
1178                  * bit more confidence that a 'cn' value will be a
1179                  * ascii string */
1180                 return NULL;
1181         }
1182         if (val) {
1183                 return (char *)val->data;
1184         }
1185         return NULL;
1186 }
1187
1188 /*
1189   work out the ntds settings dn for the current open ldb
1190 */
1191 struct ldb_dn *samdb_ntds_settings_dn(struct ldb_context *ldb)
1192 {
1193         TALLOC_CTX *tmp_ctx;
1194         const char *root_attrs[] = { "dsServiceName", NULL };
1195         int ret;
1196         struct ldb_result *root_res;
1197         struct ldb_dn *settings_dn;
1198
1199         /* see if we have a cached copy */
1200         settings_dn = (struct ldb_dn *)ldb_get_opaque(ldb, "cache.settings_dn");
1201         if (settings_dn) {
1202                 return settings_dn;
1203         }
1204
1205         tmp_ctx = talloc_new(ldb);
1206         if (tmp_ctx == NULL) {
1207                 goto failed;
1208         }
1209
1210         ret = ldb_search(ldb, tmp_ctx, &root_res, ldb_dn_new(tmp_ctx, ldb, ""), LDB_SCOPE_BASE, root_attrs, NULL);
1211         if (ret) {
1212                 DEBUG(1,("Searching for dsServiceName in rootDSE failed: %s\n",
1213                          ldb_errstring(ldb)));
1214                 goto failed;
1215         }
1216
1217         if (root_res->count != 1) {
1218                 goto failed;
1219         }
1220
1221         settings_dn = ldb_msg_find_attr_as_dn(ldb, tmp_ctx, root_res->msgs[0], "dsServiceName");
1222
1223         /* cache the domain_sid in the ldb */
1224         if (ldb_set_opaque(ldb, "cache.settings_dn", settings_dn) != LDB_SUCCESS) {
1225                 goto failed;
1226         }
1227
1228         talloc_steal(ldb, settings_dn);
1229         talloc_free(tmp_ctx);
1230
1231         return settings_dn;
1232
1233 failed:
1234         DEBUG(1,("Failed to find our own NTDS Settings DN in the ldb!\n"));
1235         talloc_free(tmp_ctx);
1236         return NULL;
1237 }
1238
1239 /*
1240   work out the ntds settings invocationId for the current open ldb
1241 */
1242 const struct GUID *samdb_ntds_invocation_id(struct ldb_context *ldb)
1243 {
1244         TALLOC_CTX *tmp_ctx;
1245         const char *attrs[] = { "invocationId", NULL };
1246         int ret;
1247         struct ldb_result *res;
1248         struct GUID *invocation_id;
1249
1250         /* see if we have a cached copy */
1251         invocation_id = (struct GUID *)ldb_get_opaque(ldb, "cache.invocation_id");
1252         if (invocation_id) {
1253                 return invocation_id;
1254         }
1255
1256         tmp_ctx = talloc_new(ldb);
1257         if (tmp_ctx == NULL) {
1258                 goto failed;
1259         }
1260
1261         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1262         if (ret) {
1263                 goto failed;
1264         }
1265
1266         if (res->count != 1) {
1267                 goto failed;
1268         }
1269
1270         invocation_id = talloc(tmp_ctx, struct GUID);
1271         if (!invocation_id) {
1272                 goto failed;
1273         }
1274
1275         *invocation_id = samdb_result_guid(res->msgs[0], "invocationId");
1276
1277         /* cache the domain_sid in the ldb */
1278         if (ldb_set_opaque(ldb, "cache.invocation_id", invocation_id) != LDB_SUCCESS) {
1279                 goto failed;
1280         }
1281
1282         talloc_steal(ldb, invocation_id);
1283         talloc_free(tmp_ctx);
1284
1285         return invocation_id;
1286
1287 failed:
1288         DEBUG(1,("Failed to find our own NTDS Settings invocationId in the ldb!\n"));
1289         talloc_free(tmp_ctx);
1290         return NULL;
1291 }
1292
1293 bool samdb_set_ntds_invocation_id(struct ldb_context *ldb, const struct GUID *invocation_id_in)
1294 {
1295         TALLOC_CTX *tmp_ctx;
1296         struct GUID *invocation_id_new;
1297         struct GUID *invocation_id_old;
1298
1299         /* see if we have a cached copy */
1300         invocation_id_old = (struct GUID *)ldb_get_opaque(ldb,
1301                                                          "cache.invocation_id");
1302
1303         tmp_ctx = talloc_new(ldb);
1304         if (tmp_ctx == NULL) {
1305                 goto failed;
1306         }
1307
1308         invocation_id_new = talloc(tmp_ctx, struct GUID);
1309         if (!invocation_id_new) {
1310                 goto failed;
1311         }
1312
1313         *invocation_id_new = *invocation_id_in;
1314
1315         /* cache the domain_sid in the ldb */
1316         if (ldb_set_opaque(ldb, "cache.invocation_id", invocation_id_new) != LDB_SUCCESS) {
1317                 goto failed;
1318         }
1319
1320         talloc_steal(ldb, invocation_id_new);
1321         talloc_free(tmp_ctx);
1322         talloc_free(invocation_id_old);
1323
1324         return true;
1325
1326 failed:
1327         DEBUG(1,("Failed to set our own cached invocationId in the ldb!\n"));
1328         talloc_free(tmp_ctx);
1329         return false;
1330 }
1331
1332 /*
1333   work out the ntds settings objectGUID for the current open ldb
1334 */
1335 const struct GUID *samdb_ntds_objectGUID(struct ldb_context *ldb)
1336 {
1337         TALLOC_CTX *tmp_ctx;
1338         const char *attrs[] = { "objectGUID", NULL };
1339         int ret;
1340         struct ldb_result *res;
1341         struct GUID *ntds_guid;
1342
1343         /* see if we have a cached copy */
1344         ntds_guid = (struct GUID *)ldb_get_opaque(ldb, "cache.ntds_guid");
1345         if (ntds_guid) {
1346                 return ntds_guid;
1347         }
1348
1349         tmp_ctx = talloc_new(ldb);
1350         if (tmp_ctx == NULL) {
1351                 goto failed;
1352         }
1353
1354         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1355         if (ret) {
1356                 goto failed;
1357         }
1358
1359         if (res->count != 1) {
1360                 goto failed;
1361         }
1362
1363         ntds_guid = talloc(tmp_ctx, struct GUID);
1364         if (!ntds_guid) {
1365                 goto failed;
1366         }
1367
1368         *ntds_guid = samdb_result_guid(res->msgs[0], "objectGUID");
1369
1370         /* cache the domain_sid in the ldb */
1371         if (ldb_set_opaque(ldb, "cache.ntds_guid", ntds_guid) != LDB_SUCCESS) {
1372                 goto failed;
1373         }
1374
1375         talloc_steal(ldb, ntds_guid);
1376         talloc_free(tmp_ctx);
1377
1378         return ntds_guid;
1379
1380 failed:
1381         DEBUG(1,("Failed to find our own NTDS Settings objectGUID in the ldb!\n"));
1382         talloc_free(tmp_ctx);
1383         return NULL;
1384 }
1385
1386 bool samdb_set_ntds_objectGUID(struct ldb_context *ldb, const struct GUID *ntds_guid_in)
1387 {
1388         TALLOC_CTX *tmp_ctx;
1389         struct GUID *ntds_guid_new;
1390         struct GUID *ntds_guid_old;
1391
1392         /* see if we have a cached copy */
1393         ntds_guid_old = (struct GUID *)ldb_get_opaque(ldb, "cache.ntds_guid");
1394
1395         tmp_ctx = talloc_new(ldb);
1396         if (tmp_ctx == NULL) {
1397                 goto failed;
1398         }
1399
1400         ntds_guid_new = talloc(tmp_ctx, struct GUID);
1401         if (!ntds_guid_new) {
1402                 goto failed;
1403         }
1404
1405         *ntds_guid_new = *ntds_guid_in;
1406
1407         /* cache the domain_sid in the ldb */
1408         if (ldb_set_opaque(ldb, "cache.ntds_guid", ntds_guid_new) != LDB_SUCCESS) {
1409                 goto failed;
1410         }
1411
1412         talloc_steal(ldb, ntds_guid_new);
1413         talloc_free(tmp_ctx);
1414         talloc_free(ntds_guid_old);
1415
1416         return true;
1417
1418 failed:
1419         DEBUG(1,("Failed to set our own cached invocationId in the ldb!\n"));
1420         talloc_free(tmp_ctx);
1421         return false;
1422 }
1423
1424 /*
1425   work out the server dn for the current open ldb
1426 */
1427 struct ldb_dn *samdb_server_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1428 {
1429         return ldb_dn_get_parent(mem_ctx, samdb_ntds_settings_dn(ldb));
1430 }
1431
1432 /*
1433   work out the server dn for the current open ldb
1434 */
1435 struct ldb_dn *samdb_server_site_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1436 {
1437         struct ldb_dn *server_dn;
1438         struct ldb_dn *server_site_dn;
1439
1440         server_dn = samdb_server_dn(ldb, mem_ctx);
1441         if (!server_dn) return NULL;
1442
1443         server_site_dn = ldb_dn_get_parent(mem_ctx, server_dn);
1444
1445         talloc_free(server_dn);
1446         return server_site_dn;
1447 }
1448
1449 const char *samdb_server_site_name(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1450 {
1451         const struct ldb_val *val = ldb_dn_get_rdn_val(samdb_server_site_dn(ldb, mem_ctx));
1452
1453         if (val != NULL)
1454                 return (const char *) val->data;
1455         else
1456                 return NULL;
1457 }
1458
1459 /*
1460   work out if we are the PDC for the domain of the current open ldb
1461 */
1462 bool samdb_is_pdc(struct ldb_context *ldb)
1463 {
1464         const char *dom_attrs[] = { "fSMORoleOwner", NULL };
1465         int ret;
1466         struct ldb_result *dom_res;
1467         TALLOC_CTX *tmp_ctx;
1468         bool is_pdc;
1469         struct ldb_dn *pdc;
1470
1471         tmp_ctx = talloc_new(ldb);
1472         if (tmp_ctx == NULL) {
1473                 DEBUG(1, ("talloc_new failed in samdb_is_pdc"));
1474                 return false;
1475         }
1476
1477         ret = ldb_search(ldb, tmp_ctx, &dom_res, ldb_get_default_basedn(ldb), LDB_SCOPE_BASE, dom_attrs, NULL);
1478         if (ret) {
1479                 DEBUG(1,("Searching for fSMORoleOwner in %s failed: %s\n",
1480                          ldb_dn_get_linearized(ldb_get_default_basedn(ldb)),
1481                          ldb_errstring(ldb)));
1482                 goto failed;
1483         }
1484         if (dom_res->count != 1) {
1485                 goto failed;
1486         }
1487
1488         pdc = ldb_msg_find_attr_as_dn(ldb, tmp_ctx, dom_res->msgs[0], "fSMORoleOwner");
1489
1490         if (ldb_dn_compare(samdb_ntds_settings_dn(ldb), pdc) == 0) {
1491                 is_pdc = true;
1492         } else {
1493                 is_pdc = false;
1494         }
1495
1496         talloc_free(tmp_ctx);
1497
1498         return is_pdc;
1499
1500 failed:
1501         DEBUG(1,("Failed to find if we are the PDC for this ldb\n"));
1502         talloc_free(tmp_ctx);
1503         return false;
1504 }
1505
1506 /*
1507   work out if we are a Global Catalog server for the domain of the current open ldb
1508 */
1509 bool samdb_is_gc(struct ldb_context *ldb)
1510 {
1511         const char *attrs[] = { "options", NULL };
1512         int ret, options;
1513         struct ldb_result *res;
1514         TALLOC_CTX *tmp_ctx;
1515
1516         tmp_ctx = talloc_new(ldb);
1517         if (tmp_ctx == NULL) {
1518                 DEBUG(1, ("talloc_new failed in samdb_is_pdc"));
1519                 return false;
1520         }
1521
1522         /* Query cn=ntds settings,.... */
1523         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1524         if (ret) {
1525                 talloc_free(tmp_ctx);
1526                 return false;
1527         }
1528         if (res->count != 1) {
1529                 talloc_free(tmp_ctx);
1530                 return false;
1531         }
1532
1533         options = ldb_msg_find_attr_as_int(res->msgs[0], "options", 0);
1534         talloc_free(tmp_ctx);
1535
1536         /* if options attribute has the 0x00000001 flag set, then enable the global catlog */
1537         if (options & 0x000000001) {
1538                 return true;
1539         }
1540         return false;
1541 }
1542
1543 /* Find a domain object in the parents of a particular DN.  */
1544 int samdb_search_for_parent_domain(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
1545                                    struct ldb_dn **parent_dn, const char **errstring)
1546 {
1547         TALLOC_CTX *local_ctx;
1548         struct ldb_dn *sdn = dn;
1549         struct ldb_result *res = NULL;
1550         int ret = 0;
1551         const char *attrs[] = { NULL };
1552
1553         local_ctx = talloc_new(mem_ctx);
1554         if (local_ctx == NULL) return LDB_ERR_OPERATIONS_ERROR;
1555
1556         while ((sdn = ldb_dn_get_parent(local_ctx, sdn))) {
1557                 ret = ldb_search(ldb, local_ctx, &res, sdn, LDB_SCOPE_BASE, attrs,
1558                                  "(|(objectClass=domain)(objectClass=builtinDomain))");
1559                 if (ret == LDB_SUCCESS) {
1560                         if (res->count == 1) {
1561                                 break;
1562                         }
1563                 } else {
1564                         break;
1565                 }
1566         }
1567
1568         if (ret != LDB_SUCCESS) {
1569                 *errstring = talloc_asprintf(mem_ctx, "Error searching for parent domain of %s, failed searching for %s: %s",
1570                                              ldb_dn_get_linearized(dn),
1571                                              ldb_dn_get_linearized(sdn),
1572                                              ldb_errstring(ldb));
1573                 talloc_free(local_ctx);
1574                 return ret;
1575         }
1576         if (res->count != 1) {
1577                 *errstring = talloc_asprintf(mem_ctx, "Invalid dn (%s), not child of a domain object",
1578                                              ldb_dn_get_linearized(dn));
1579                 DEBUG(0,(__location__ ": %s\n", *errstring));
1580                 talloc_free(local_ctx);
1581                 return LDB_ERR_CONSTRAINT_VIOLATION;
1582         }
1583
1584         *parent_dn = talloc_steal(mem_ctx, res->msgs[0]->dn);
1585         talloc_free(local_ctx);
1586         return ret;
1587 }
1588
1589
1590 /*
1591  * Performs checks on a user password (plaintext UNIX format - attribute
1592  * "password"). The remaining parameters have to be extracted from the domain
1593  * object in the AD.
1594  *
1595  * Result codes from "enum samr_ValidationStatus" (consider "samr.idl")
1596  */
1597 enum samr_ValidationStatus samdb_check_password(const DATA_BLOB *password,
1598                                                 const uint32_t pwdProperties,
1599                                                 const uint32_t minPwdLength)
1600 {
1601         /* checks if the "minPwdLength" property is satisfied */
1602         if (minPwdLength > password->length)
1603                 return SAMR_VALIDATION_STATUS_PWD_TOO_SHORT;
1604
1605         /* checks the password complexity */
1606         if (((pwdProperties & DOMAIN_PASSWORD_COMPLEX) != 0)
1607                         && (password->data != NULL)
1608                         && (!check_password_quality((const char *) password->data)))
1609                 return SAMR_VALIDATION_STATUS_NOT_COMPLEX_ENOUGH;
1610
1611         return SAMR_VALIDATION_STATUS_SUCCESS;
1612 }
1613
1614 /*
1615  * Sets the user password using plaintext UTF16 (attribute "new_password") or
1616  * LM (attribute "lmNewHash") or NT (attribute "ntNewHash") hash. Also pass
1617  * as parameter if it's a user change or not ("userChange"). The "rejectReason"
1618  * gives some more informations if the changed failed.
1619  *
1620  * The caller should have a LDB transaction wrapping this.
1621  *
1622  * Results: NT_STATUS_OK, NT_STATUS_INTERNAL_DB_CORRUPTION,
1623  *   NT_STATUS_INVALID_PARAMETER, NT_STATUS_UNSUCCESSFUL,
1624  *   NT_STATUS_PASSWORD_RESTRICTION
1625  */
1626 NTSTATUS samdb_set_password(struct ldb_context *ctx, TALLOC_CTX *mem_ctx,
1627                             struct ldb_dn *user_dn, struct ldb_dn *domain_dn,
1628                             struct ldb_message *mod,
1629                             const DATA_BLOB *new_password,
1630                             struct samr_Password *param_lmNewHash,
1631                             struct samr_Password *param_ntNewHash,
1632                             bool user_change,
1633                             enum samPwdChangeReason *reject_reason,
1634                             struct samr_DomInfo1 **_dominfo)
1635 {
1636         const char * const user_attrs[] = { "userAccountControl",
1637                                             "lmPwdHistory",
1638                                             "ntPwdHistory",
1639                                             "dBCSPwd", "unicodePwd",
1640                                             "objectSid",
1641                                             "pwdLastSet", NULL };
1642         const char * const domain_attrs[] = { "minPwdLength", "pwdProperties",
1643                                               "pwdHistoryLength",
1644                                               "maxPwdAge", "minPwdAge", NULL };
1645         NTTIME pwdLastSet;
1646         uint32_t minPwdLength, pwdProperties, pwdHistoryLength;
1647         int64_t maxPwdAge, minPwdAge;
1648         uint32_t userAccountControl;
1649         struct samr_Password *sambaLMPwdHistory, *sambaNTPwdHistory,
1650                 *lmPwdHash, *ntPwdHash, *lmNewHash, *ntNewHash;
1651         struct samr_Password local_lmNewHash, local_ntNewHash;
1652         int sambaLMPwdHistory_len, sambaNTPwdHistory_len;
1653         struct dom_sid *domain_sid;
1654         struct ldb_message **res;
1655         bool restrictions;
1656         int count;
1657         time_t now = time(NULL);
1658         NTTIME now_nt;
1659         int i;
1660
1661         /* we need to know the time to compute password age */
1662         unix_to_nt_time(&now_nt, now);
1663
1664         /* pull all the user parameters */
1665         count = gendb_search_dn(ctx, mem_ctx, user_dn, &res, user_attrs);
1666         if (count != 1) {
1667                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
1668         }
1669         userAccountControl = samdb_result_uint(res[0], "userAccountControl", 0);
1670         sambaLMPwdHistory_len = samdb_result_hashes(mem_ctx, res[0],
1671                                          "lmPwdHistory", &sambaLMPwdHistory);
1672         sambaNTPwdHistory_len = samdb_result_hashes(mem_ctx, res[0],
1673                                          "ntPwdHistory", &sambaNTPwdHistory);
1674         lmPwdHash = samdb_result_hash(mem_ctx, res[0], "dBCSPwd");
1675         ntPwdHash = samdb_result_hash(mem_ctx, res[0], "unicodePwd");
1676         pwdLastSet = samdb_result_uint64(res[0], "pwdLastSet", 0);
1677
1678         /* Copy parameters */
1679         lmNewHash = param_lmNewHash;
1680         ntNewHash = param_ntNewHash;
1681
1682         /* Only non-trust accounts have restrictions (possibly this
1683          * test is the wrong way around, but I like to be restrictive
1684          * if possible */
1685         restrictions = !(userAccountControl & (UF_INTERDOMAIN_TRUST_ACCOUNT
1686                                                |UF_WORKSTATION_TRUST_ACCOUNT
1687                                                |UF_SERVER_TRUST_ACCOUNT));
1688
1689         if (domain_dn != NULL) {
1690                 /* pull the domain parameters */
1691                 count = gendb_search_dn(ctx, mem_ctx, domain_dn, &res,
1692                                                                 domain_attrs);
1693                 if (count != 1) {
1694                         DEBUG(2, ("samdb_set_password: Domain DN %s is invalid, for user %s\n",
1695                                   ldb_dn_get_linearized(domain_dn),
1696                                   ldb_dn_get_linearized(user_dn)));
1697                         return NT_STATUS_NO_SUCH_DOMAIN;
1698                 }
1699         } else {
1700                 /* work out the domain sid, and pull the domain from there */
1701                 domain_sid = samdb_result_sid_prefix(mem_ctx, res[0],
1702                                                                 "objectSid");
1703                 if (domain_sid == NULL) {
1704                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
1705                 }
1706
1707                 count = gendb_search(ctx, mem_ctx, NULL, &res, domain_attrs,
1708                                 "(objectSid=%s)",
1709                                 ldap_encode_ndr_dom_sid(mem_ctx, domain_sid));
1710                 if (count != 1) {
1711                         DEBUG(2, ("samdb_set_password: Could not find domain to match SID: %s, for user %s\n",
1712                                   dom_sid_string(mem_ctx, domain_sid),
1713                                   ldb_dn_get_linearized(user_dn)));
1714                         return NT_STATUS_NO_SUCH_DOMAIN;
1715                 }
1716         }
1717
1718         minPwdLength = samdb_result_uint(res[0], "minPwdLength", 0);
1719         pwdProperties = samdb_result_uint(res[0], "pwdProperties", 0);
1720         pwdHistoryLength = samdb_result_uint(res[0], "pwdHistoryLength", 0);
1721         maxPwdAge = samdb_result_int64(res[0], "maxPwdAge", 0);
1722         minPwdAge = samdb_result_int64(res[0], "minPwdAge", 0);
1723
1724         if ((userAccountControl & UF_PASSWD_NOTREQD) != 0) {
1725                 /* see [MS-ADTS] 2.2.15 */
1726                 minPwdLength = 0;
1727         }
1728
1729         if (_dominfo != NULL) {
1730                 struct samr_DomInfo1 *dominfo;
1731                 /* on failure we need to fill in the reject reasons */
1732                 dominfo = talloc(mem_ctx, struct samr_DomInfo1);
1733                 if (dominfo == NULL) {
1734                         return NT_STATUS_NO_MEMORY;
1735                 }
1736                 dominfo->min_password_length     = minPwdLength;
1737                 dominfo->password_properties     = pwdProperties;
1738                 dominfo->password_history_length = pwdHistoryLength;
1739                 dominfo->max_password_age        = maxPwdAge;
1740                 dominfo->min_password_age        = minPwdAge;
1741                 *_dominfo = dominfo;
1742         }
1743
1744         if ((restrictions != 0) && (new_password != 0)) {
1745                 char *new_pass;
1746
1747                 /* checks if the "minPwdLength" property is satisfied */
1748                 if ((restrictions != 0)
1749                         && (minPwdLength > utf16_len_n(
1750                                 new_password->data, new_password->length)/2)) {
1751                         if (reject_reason) {
1752                                 *reject_reason = SAM_PWD_CHANGE_PASSWORD_TOO_SHORT;
1753                         }
1754                         return NT_STATUS_PASSWORD_RESTRICTION;
1755                 }
1756
1757                 /* Create the NT hash */
1758                 mdfour(local_ntNewHash.hash, new_password->data,
1759                                                         new_password->length);
1760
1761                 ntNewHash = &local_ntNewHash;
1762
1763                 /* Only check complexity if we can convert it at all.  Assuming unconvertable passwords are 'strong' */
1764                 if (convert_string_talloc_convenience(mem_ctx,
1765                           lp_iconv_convenience(ldb_get_opaque(ctx, "loadparm")),
1766                           CH_UTF16, CH_UNIX,
1767                           new_password->data, new_password->length,
1768                           (void **)&new_pass, NULL, false)) {
1769
1770                         /* checks the password complexity */
1771                         if ((restrictions != 0)
1772                                 && ((pwdProperties
1773                                         & DOMAIN_PASSWORD_COMPLEX) != 0)
1774                                 && (!check_password_quality(new_pass))) {
1775                                 if (reject_reason) {
1776                                         *reject_reason = SAM_PWD_CHANGE_NOT_COMPLEX;
1777                                 }
1778                                 return NT_STATUS_PASSWORD_RESTRICTION;
1779                         }
1780
1781                         /* compute the new lm hashes (for checking history - case insenitivly!) */
1782                         if (E_deshash(new_pass, local_lmNewHash.hash)) {
1783                                 lmNewHash = &local_lmNewHash;
1784                         }
1785                 }
1786         }
1787
1788         if ((restrictions != 0) && user_change) {
1789                 /* are all password changes disallowed? */
1790                 if ((pwdProperties & DOMAIN_REFUSE_PASSWORD_CHANGE) != 0) {
1791                         if (reject_reason) {
1792                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1793                         }
1794                         return NT_STATUS_PASSWORD_RESTRICTION;
1795                 }
1796
1797                 /* can this user change the password? */
1798                 if ((userAccountControl & UF_PASSWD_CANT_CHANGE) != 0) {
1799                         if (reject_reason) {
1800                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1801                         }
1802                         return NT_STATUS_PASSWORD_RESTRICTION;
1803                 }
1804
1805                 /* Password minimum age: yes, this is a minus. The ages are in negative 100nsec units! */
1806                 if (pwdLastSet - minPwdAge > now_nt) {
1807                         if (reject_reason) {
1808                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1809                         }
1810                         return NT_STATUS_PASSWORD_RESTRICTION;
1811                 }
1812
1813                 /* check the immediately past password */
1814                 if (pwdHistoryLength > 0) {
1815                         if (lmNewHash && lmPwdHash && memcmp(lmNewHash->hash,
1816                                         lmPwdHash->hash, 16) == 0) {
1817                                 if (reject_reason) {
1818                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1819                                 }
1820                                 return NT_STATUS_PASSWORD_RESTRICTION;
1821                         }
1822                         if (ntNewHash && ntPwdHash && memcmp(ntNewHash->hash,
1823                                         ntPwdHash->hash, 16) == 0) {
1824                                 if (reject_reason) {
1825                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1826                                 }
1827                                 return NT_STATUS_PASSWORD_RESTRICTION;
1828                         }
1829                 }
1830
1831                 /* check the password history */
1832                 sambaLMPwdHistory_len = MIN(sambaLMPwdHistory_len,
1833                                                         pwdHistoryLength);
1834                 sambaNTPwdHistory_len = MIN(sambaNTPwdHistory_len,
1835                                                         pwdHistoryLength);
1836
1837                 for (i=0; lmNewHash && i<sambaLMPwdHistory_len;i++) {
1838                         if (memcmp(lmNewHash->hash, sambaLMPwdHistory[i].hash,
1839                                         16) == 0) {
1840                                 if (reject_reason) {
1841                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1842                                 }
1843                                 return NT_STATUS_PASSWORD_RESTRICTION;
1844                         }
1845                 }
1846                 for (i=0; ntNewHash && i<sambaNTPwdHistory_len;i++) {
1847                         if (memcmp(ntNewHash->hash, sambaNTPwdHistory[i].hash,
1848                                          16) == 0) {
1849                                 if (reject_reason) {
1850                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1851                                 }
1852                                 return NT_STATUS_PASSWORD_RESTRICTION;
1853                         }
1854                 }
1855         }
1856
1857 #define CHECK_RET(x) do { if (x != 0) return NT_STATUS_NO_MEMORY; } while(0)
1858
1859         /* the password is acceptable. Start forming the new fields */
1860         if (new_password != NULL) {
1861                 /* if we know the cleartext UTF16 password, then set it.
1862                  * Modules in ldb will set all the appropriate
1863                  * hashes */
1864                 CHECK_RET(ldb_msg_add_value(mod, "clearTextPassword", new_password, NULL));
1865         } else {
1866                 /* We don't have the cleartext, so delete the old one
1867                  * and set what we have of the hashes */
1868                 CHECK_RET(samdb_msg_add_delete(ctx, mem_ctx, mod, "clearTextPassword"));
1869
1870                 if (lmNewHash) {
1871                         CHECK_RET(samdb_msg_add_hash(ctx, mem_ctx, mod, "dBCSPwd", lmNewHash));
1872                 } else {
1873                         CHECK_RET(samdb_msg_add_delete(ctx, mem_ctx, mod, "dBCSPwd"));
1874                 }
1875
1876                 if (ntNewHash) {
1877                         CHECK_RET(samdb_msg_add_hash(ctx, mem_ctx, mod, "unicodePwd", ntNewHash));
1878                 } else {
1879                         CHECK_RET(samdb_msg_add_delete(ctx, mem_ctx, mod, "unicodePwd"));
1880                 }
1881         }
1882
1883         if (reject_reason) {
1884                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1885         }
1886         return NT_STATUS_OK;
1887 }
1888
1889
1890 /*
1891  * Sets the user password using plaintext UTF16 (attribute "new_password") or
1892  * LM (attribute "lmNewHash") or NT (attribute "ntNewHash") hash. Also pass
1893  * as parameter if it's a user change or not ("userChange"). The "rejectReason"
1894  * gives some more informations if the changed failed.
1895  *
1896  * This wrapper function for "samdb_set_password" takes a SID as input rather
1897  * than a user DN.
1898  *
1899  * This call encapsulates a new LDB transaction for changing the password;
1900  * therefore the user hasn't to start a new one.
1901  *
1902  * Results: NT_STATUS_OK, NT_STATUS_INTERNAL_DB_CORRUPTION,
1903  *   NT_STATUS_INVALID_PARAMETER, NT_STATUS_UNSUCCESSFUL,
1904  *   NT_STATUS_PASSWORD_RESTRICTION
1905  */
1906 NTSTATUS samdb_set_password_sid(struct ldb_context *ldb, TALLOC_CTX *mem_ctx,
1907                                 const struct dom_sid *user_sid,
1908                                 const DATA_BLOB *new_password,
1909                                 struct samr_Password *lmNewHash,
1910                                 struct samr_Password *ntNewHash,
1911                                 bool user_change,
1912                                 enum samPwdChangeReason *reject_reason,
1913                                 struct samr_DomInfo1 **_dominfo)
1914 {
1915         NTSTATUS nt_status;
1916         struct ldb_dn *user_dn;
1917         struct ldb_message *msg;
1918         int ret;
1919
1920         ret = ldb_transaction_start(ldb);
1921         if (ret != LDB_SUCCESS) {
1922                 DEBUG(1, ("Failed to start transaction: %s\n", ldb_errstring(ldb)));
1923                 return NT_STATUS_TRANSACTION_ABORTED;
1924         }
1925
1926         user_dn = samdb_search_dn(ldb, mem_ctx, NULL,
1927                                   "(&(objectSid=%s)(objectClass=user))",
1928                                   ldap_encode_ndr_dom_sid(mem_ctx, user_sid));
1929         if (!user_dn) {
1930                 ldb_transaction_cancel(ldb);
1931                 DEBUG(3, ("samdb_set_password_sid: SID %s not found in samdb, returning NO_SUCH_USER\n",
1932                           dom_sid_string(mem_ctx, user_sid)));
1933                 return NT_STATUS_NO_SUCH_USER;
1934         }
1935
1936         msg = ldb_msg_new(mem_ctx);
1937         if (msg == NULL) {
1938                 ldb_transaction_cancel(ldb);
1939                 talloc_free(user_dn);
1940                 return NT_STATUS_NO_MEMORY;
1941         }
1942
1943         msg->dn = ldb_dn_copy(msg, user_dn);
1944         if (!msg->dn) {
1945                 ldb_transaction_cancel(ldb);
1946                 talloc_free(user_dn);
1947                 talloc_free(msg);
1948                 return NT_STATUS_NO_MEMORY;
1949         }
1950
1951         nt_status = samdb_set_password(ldb, mem_ctx,
1952                                        user_dn, NULL,
1953                                        msg, new_password,
1954                                        lmNewHash, ntNewHash,
1955                                        user_change,
1956                                        reject_reason, _dominfo);
1957         if (!NT_STATUS_IS_OK(nt_status)) {
1958                 ldb_transaction_cancel(ldb);
1959                 talloc_free(user_dn);
1960                 talloc_free(msg);
1961                 return nt_status;
1962         }
1963
1964         /* modify the samdb record */
1965         ret = samdb_replace(ldb, mem_ctx, msg);
1966         if (ret != LDB_SUCCESS) {
1967                 ldb_transaction_cancel(ldb);
1968                 talloc_free(user_dn);
1969                 talloc_free(msg);
1970                 return NT_STATUS_ACCESS_DENIED;
1971         }
1972
1973         talloc_free(msg);
1974
1975         ret = ldb_transaction_commit(ldb);
1976         if (ret != LDB_SUCCESS) {
1977                 DEBUG(0,("Failed to commit transaction to change password on %s: %s\n",
1978                          ldb_dn_get_linearized(user_dn),
1979                          ldb_errstring(ldb)));
1980                 talloc_free(user_dn);
1981                 return NT_STATUS_TRANSACTION_ABORTED;
1982         }
1983
1984         talloc_free(user_dn);
1985         return NT_STATUS_OK;
1986 }
1987
1988
1989 NTSTATUS samdb_create_foreign_security_principal(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
1990                                                  struct dom_sid *sid, struct ldb_dn **ret_dn)
1991 {
1992         struct ldb_message *msg;
1993         struct ldb_dn *basedn;
1994         const char *sidstr;
1995         int ret;
1996
1997         sidstr = dom_sid_string(mem_ctx, sid);
1998         NT_STATUS_HAVE_NO_MEMORY(sidstr);
1999
2000         /* We might have to create a ForeignSecurityPrincipal, even if this user
2001          * is in our own domain */
2002
2003         msg = ldb_msg_new(mem_ctx);
2004         if (msg == NULL) {
2005                 return NT_STATUS_NO_MEMORY;
2006         }
2007
2008         /* TODO: Hmmm. This feels wrong. How do I find the base dn to
2009          * put the ForeignSecurityPrincipals? d_state->domain_dn does
2010          * not work, this is wrong for the Builtin domain, there's no
2011          * cn=For...,cn=Builtin,dc={BASEDN}.  -- vl
2012          */
2013
2014         basedn = samdb_search_dn(sam_ctx, mem_ctx, NULL,
2015                                  "(&(objectClass=container)(cn=ForeignSecurityPrincipals))");
2016
2017         if (basedn == NULL) {
2018                 DEBUG(0, ("Failed to find DN for "
2019                           "ForeignSecurityPrincipal container\n"));
2020                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
2021         }
2022
2023         /* add core elements to the ldb_message for the alias */
2024         msg->dn = ldb_dn_copy(mem_ctx, basedn);
2025         if ( ! ldb_dn_add_child_fmt(msg->dn, "CN=%s", sidstr))
2026                 return NT_STATUS_NO_MEMORY;
2027
2028         samdb_msg_add_string(sam_ctx, mem_ctx, msg,
2029                              "objectClass",
2030                              "foreignSecurityPrincipal");
2031
2032         /* create the alias */
2033         ret = ldb_add(sam_ctx, msg);
2034         if (ret != 0) {
2035                 DEBUG(0,("Failed to create foreignSecurityPrincipal "
2036                          "record %s: %s\n",
2037                          ldb_dn_get_linearized(msg->dn),
2038                          ldb_errstring(sam_ctx)));
2039                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
2040         }
2041         *ret_dn = msg->dn;
2042         return NT_STATUS_OK;
2043 }
2044
2045
2046 /*
2047   Find the DN of a domain, assuming it to be a dotted.dns name
2048 */
2049
2050 struct ldb_dn *samdb_dns_domain_to_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const char *dns_domain)
2051 {
2052         int i;
2053         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2054         const char *binary_encoded;
2055         const char **split_realm;
2056         struct ldb_dn *dn;
2057
2058         if (!tmp_ctx) {
2059                 return NULL;
2060         }
2061
2062         split_realm = (const char **)str_list_make(tmp_ctx, dns_domain, ".");
2063         if (!split_realm) {
2064                 talloc_free(tmp_ctx);
2065                 return NULL;
2066         }
2067         dn = ldb_dn_new(mem_ctx, ldb, NULL);
2068         for (i=0; split_realm[i]; i++) {
2069                 binary_encoded = ldb_binary_encode_string(tmp_ctx, split_realm[i]);
2070                 if (!ldb_dn_add_base_fmt(dn, "dc=%s", binary_encoded)) {
2071                         DEBUG(2, ("Failed to add dc=%s element to DN %s\n",
2072                                   binary_encoded, ldb_dn_get_linearized(dn)));
2073                         talloc_free(tmp_ctx);
2074                         return NULL;
2075                 }
2076         }
2077         if (!ldb_dn_validate(dn)) {
2078                 DEBUG(2, ("Failed to validated DN %s\n",
2079                           ldb_dn_get_linearized(dn)));
2080                 return NULL;
2081         }
2082         return dn;
2083 }
2084 /*
2085   Find the DN of a domain, be it the netbios or DNS name
2086 */
2087
2088 struct ldb_dn *samdb_domain_to_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx,
2089                                   const char *domain_name)
2090 {
2091         const char * const domain_ref_attrs[] = {
2092                 "ncName", NULL
2093         };
2094         const char * const domain_ref2_attrs[] = {
2095                 NULL
2096         };
2097         struct ldb_result *res_domain_ref;
2098         char *escaped_domain = ldb_binary_encode_string(mem_ctx, domain_name);
2099         /* find the domain's DN */
2100         int ret_domain = ldb_search(ldb, mem_ctx,
2101                                             &res_domain_ref,
2102                                             samdb_partitions_dn(ldb, mem_ctx),
2103                                             LDB_SCOPE_ONELEVEL,
2104                                             domain_ref_attrs,
2105                                             "(&(nETBIOSName=%s)(objectclass=crossRef))",
2106                                             escaped_domain);
2107         if (ret_domain != 0) {
2108                 return NULL;
2109         }
2110
2111         if (res_domain_ref->count == 0) {
2112                 ret_domain = ldb_search(ldb, mem_ctx,
2113                                                 &res_domain_ref,
2114                                                 samdb_dns_domain_to_dn(ldb, mem_ctx, domain_name),
2115                                                 LDB_SCOPE_BASE,
2116                                                 domain_ref2_attrs,
2117                                                 "(objectclass=domain)");
2118                 if (ret_domain != 0) {
2119                         return NULL;
2120                 }
2121
2122                 if (res_domain_ref->count == 1) {
2123                         return res_domain_ref->msgs[0]->dn;
2124                 }
2125                 return NULL;
2126         }
2127
2128         if (res_domain_ref->count > 1) {
2129                 DEBUG(0,("Found %d records matching domain [%s]\n",
2130                          ret_domain, domain_name));
2131                 return NULL;
2132         }
2133
2134         return samdb_result_dn(ldb, mem_ctx, res_domain_ref->msgs[0], "nCName", NULL);
2135
2136 }
2137
2138
2139 /*
2140   use a GUID to find a DN
2141  */
2142 int dsdb_find_dn_by_guid(struct ldb_context *ldb,
2143                          TALLOC_CTX *mem_ctx,
2144                          const char *guid_str, struct ldb_dn **dn)
2145 {
2146         int ret;
2147         struct ldb_result *res;
2148         const char *attrs[] = { NULL };
2149         struct ldb_request *search_req;
2150         char *expression;
2151         struct ldb_search_options_control *options;
2152
2153         expression = talloc_asprintf(mem_ctx, "objectGUID=%s", guid_str);
2154         if (!expression) {
2155                 DEBUG(0, (__location__ ": out of memory\n"));
2156                 return LDB_ERR_OPERATIONS_ERROR;
2157         }
2158
2159         res = talloc_zero(mem_ctx, struct ldb_result);
2160         if (!res) {
2161                 DEBUG(0, (__location__ ": out of memory\n"));
2162                 return LDB_ERR_OPERATIONS_ERROR;
2163         }
2164
2165         ret = ldb_build_search_req(&search_req, ldb, mem_ctx,
2166                                    ldb_get_default_basedn(ldb),
2167                                    LDB_SCOPE_SUBTREE,
2168                                    expression, attrs,
2169                                    NULL,
2170                                    res, ldb_search_default_callback,
2171                                    NULL);
2172         if (ret != LDB_SUCCESS) {
2173                 return ret;
2174         }
2175
2176         /* we need to cope with cross-partition links, so search for
2177            the GUID over all partitions */
2178         options = talloc(search_req, struct ldb_search_options_control);
2179         if (options == NULL) {
2180                 DEBUG(0, (__location__ ": out of memory\n"));
2181                 return LDB_ERR_OPERATIONS_ERROR;
2182         }
2183         options->search_options = LDB_SEARCH_OPTION_PHANTOM_ROOT;
2184
2185         ret = ldb_request_add_control(search_req, LDB_CONTROL_EXTENDED_DN_OID, true, NULL);
2186         if (ret != LDB_SUCCESS) {
2187                 return ret;
2188         }
2189
2190         ret = ldb_request_add_control(search_req,
2191                                       LDB_CONTROL_SEARCH_OPTIONS_OID,
2192                                       true, options);
2193         if (ret != LDB_SUCCESS) {
2194                 return ret;
2195         }
2196
2197         ret = ldb_request(ldb, search_req);
2198         if (ret != LDB_SUCCESS) {
2199                 return ret;
2200         }
2201
2202         ret = ldb_wait(search_req->handle, LDB_WAIT_ALL);
2203         if (ret != LDB_SUCCESS) {
2204                 return ret;
2205         }
2206
2207         /* this really should be exactly 1, but there is a bug in the
2208            partitions module that can return two here with the
2209            search_options control set */
2210         if (res->count < 1) {
2211                 return LDB_ERR_NO_SUCH_OBJECT;
2212         }
2213
2214         *dn = res->msgs[0]->dn;
2215
2216         return LDB_SUCCESS;
2217 }
2218
2219 /*
2220   search for attrs on one DN, allowing for deleted objects
2221  */
2222 int dsdb_search_dn_with_deleted(struct ldb_context *ldb,
2223                                 TALLOC_CTX *mem_ctx,
2224                                 struct ldb_result **_res,
2225                                 struct ldb_dn *basedn,
2226                                 const char * const *attrs)
2227 {
2228         int ret;
2229         struct ldb_request *req;
2230         TALLOC_CTX *tmp_ctx;
2231         struct ldb_result *res;
2232
2233         tmp_ctx = talloc_new(mem_ctx);
2234
2235         res = talloc_zero(tmp_ctx, struct ldb_result);
2236         if (!res) {
2237                 return LDB_ERR_OPERATIONS_ERROR;
2238         }
2239
2240         ret = ldb_build_search_req(&req, ldb, tmp_ctx,
2241                                    basedn,
2242                                    LDB_SCOPE_BASE,
2243                                    NULL,
2244                                    attrs,
2245                                    NULL,
2246                                    res,
2247                                    ldb_search_default_callback,
2248                                    NULL);
2249         if (ret != LDB_SUCCESS) {
2250                 talloc_free(tmp_ctx);
2251                 return ret;
2252         }
2253
2254         ret = ldb_request_add_control(req, LDB_CONTROL_SHOW_DELETED_OID, true, NULL);
2255         if (ret != LDB_SUCCESS) {
2256                 return ret;
2257         }
2258
2259         ret = ldb_request(ldb, req);
2260         if (ret == LDB_SUCCESS) {
2261                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2262         }
2263
2264         talloc_free(req);
2265         *_res = talloc_steal(mem_ctx, res);
2266         return ret;
2267 }
2268
2269
2270 /*
2271   use a DN to find a GUID
2272  */
2273 int dsdb_find_guid_by_dn(struct ldb_context *ldb,
2274                          struct ldb_dn *dn, struct GUID *guid)
2275 {
2276         int ret;
2277         struct ldb_result *res;
2278         const char *attrs[] = { "objectGUID", NULL };
2279         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2280
2281         ret = dsdb_search_dn_with_deleted(ldb, tmp_ctx, &res, dn, attrs);
2282         if (ret != LDB_SUCCESS) {
2283                 talloc_free(tmp_ctx);
2284                 return ret;
2285         }
2286         if (res->count < 1) {
2287                 talloc_free(tmp_ctx);
2288                 return LDB_ERR_NO_SUCH_OBJECT;
2289         }
2290         *guid = samdb_result_guid(res->msgs[0], "objectGUID");
2291         talloc_free(tmp_ctx);
2292         return LDB_SUCCESS;
2293 }
2294
2295
2296
2297 /*
2298  adds the given GUID to the given ldb_message. This value is added
2299  for the given attr_name (may be either "objectGUID" or "parentGUID").
2300  */
2301 int dsdb_msg_add_guid(struct ldb_message *msg,
2302                 struct GUID *guid,
2303                 const char *attr_name)
2304 {
2305         int ret;
2306         struct ldb_val v;
2307         NTSTATUS status;
2308         TALLOC_CTX *tmp_ctx =  talloc_init("dsdb_msg_add_guid");
2309
2310         status = GUID_to_ndr_blob(guid, tmp_ctx, &v);
2311         if (!NT_STATUS_IS_OK(status)) {
2312                 ret = LDB_ERR_OPERATIONS_ERROR;
2313                 goto done;
2314         }
2315
2316         ret = ldb_msg_add_steal_value(msg, attr_name, &v);
2317         if (ret != LDB_SUCCESS) {
2318                 DEBUG(4,(__location__ ": Failed to add %s to the message\n",
2319                                          attr_name));
2320                 goto done;
2321         }
2322
2323         ret = LDB_SUCCESS;
2324
2325 done:
2326         talloc_free(tmp_ctx);
2327         return ret;
2328
2329 }
2330
2331
2332 /*
2333   use a DN to find a SID
2334  */
2335 int dsdb_find_sid_by_dn(struct ldb_context *ldb,
2336                         struct ldb_dn *dn, struct dom_sid *sid)
2337 {
2338         int ret;
2339         struct ldb_result *res;
2340         const char *attrs[] = { "objectSID", NULL };
2341         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2342         struct dom_sid *s;
2343
2344         ZERO_STRUCTP(sid);
2345
2346         ret = dsdb_search_dn_with_deleted(ldb, tmp_ctx, &res, dn, attrs);
2347         if (ret != LDB_SUCCESS) {
2348                 talloc_free(tmp_ctx);
2349                 return ret;
2350         }
2351         if (res->count < 1) {
2352                 talloc_free(tmp_ctx);
2353                 return LDB_ERR_NO_SUCH_OBJECT;
2354         }
2355         s = samdb_result_dom_sid(tmp_ctx, res->msgs[0], "objectSID");
2356         if (s == NULL) {
2357                 talloc_free(tmp_ctx);
2358                 return LDB_ERR_NO_SUCH_OBJECT;
2359         }
2360         *sid = *s;
2361         talloc_free(tmp_ctx);
2362         return LDB_SUCCESS;
2363 }
2364
2365
2366
2367 /*
2368   load a repsFromTo blob list for a given partition GUID
2369   attr must be "repsFrom" or "repsTo"
2370  */
2371 WERROR dsdb_loadreps(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
2372                      const char *attr, struct repsFromToBlob **r, uint32_t *count)
2373 {
2374         const char *attrs[] = { attr, NULL };
2375         struct ldb_result *res = NULL;
2376         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2377         int i;
2378         struct ldb_message_element *el;
2379
2380         *r = NULL;
2381         *count = 0;
2382
2383         if (ldb_search(sam_ctx, tmp_ctx, &res, dn, LDB_SCOPE_BASE, attrs, NULL) != LDB_SUCCESS ||
2384             res->count < 1) {
2385                 DEBUG(0,("dsdb_loadreps: failed to read partition object\n"));
2386                 talloc_free(tmp_ctx);
2387                 return WERR_DS_DRA_INTERNAL_ERROR;
2388         }
2389
2390         el = ldb_msg_find_element(res->msgs[0], attr);
2391         if (el == NULL) {
2392                 /* it's OK to be empty */
2393                 talloc_free(tmp_ctx);
2394                 return WERR_OK;
2395         }
2396
2397         *count = el->num_values;
2398         *r = talloc_array(mem_ctx, struct repsFromToBlob, *count);
2399         if (*r == NULL) {
2400                 talloc_free(tmp_ctx);
2401                 return WERR_DS_DRA_INTERNAL_ERROR;
2402         }
2403
2404         for (i=0; i<(*count); i++) {
2405                 enum ndr_err_code ndr_err;
2406                 ndr_err = ndr_pull_struct_blob(&el->values[i],
2407                                                mem_ctx, lp_iconv_convenience(ldb_get_opaque(sam_ctx, "loadparm")),
2408                                                &(*r)[i],
2409                                                (ndr_pull_flags_fn_t)ndr_pull_repsFromToBlob);
2410                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
2411                         talloc_free(tmp_ctx);
2412                         return WERR_DS_DRA_INTERNAL_ERROR;
2413                 }
2414         }
2415
2416         talloc_free(tmp_ctx);
2417
2418         return WERR_OK;
2419 }
2420
2421 /*
2422   save the repsFromTo blob list for a given partition GUID
2423   attr must be "repsFrom" or "repsTo"
2424  */
2425 WERROR dsdb_savereps(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
2426                      const char *attr, struct repsFromToBlob *r, uint32_t count)
2427 {
2428         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2429         struct ldb_message *msg;
2430         struct ldb_message_element *el;
2431         int i;
2432
2433         msg = ldb_msg_new(tmp_ctx);
2434         msg->dn = dn;
2435         if (ldb_msg_add_empty(msg, attr, LDB_FLAG_MOD_REPLACE, &el) != LDB_SUCCESS) {
2436                 goto failed;
2437         }
2438
2439         el->values = talloc_array(msg, struct ldb_val, count);
2440         if (!el->values) {
2441                 goto failed;
2442         }
2443
2444         for (i=0; i<count; i++) {
2445                 struct ldb_val v;
2446                 enum ndr_err_code ndr_err;
2447
2448                 ndr_err = ndr_push_struct_blob(&v, tmp_ctx, lp_iconv_convenience(ldb_get_opaque(sam_ctx, "loadparm")),
2449                                                &r[i],
2450                                                (ndr_push_flags_fn_t)ndr_push_repsFromToBlob);
2451                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
2452                         goto failed;
2453                 }
2454
2455                 el->num_values++;
2456                 el->values[i] = v;
2457         }
2458
2459         if (ldb_modify(sam_ctx, msg) != LDB_SUCCESS) {
2460                 DEBUG(0,("Failed to store %s - %s\n", attr, ldb_errstring(sam_ctx)));
2461                 goto failed;
2462         }
2463
2464         talloc_free(tmp_ctx);
2465
2466         return WERR_OK;
2467
2468 failed:
2469         talloc_free(tmp_ctx);
2470         return WERR_DS_DRA_INTERNAL_ERROR;
2471 }
2472
2473
2474 /*
2475   load the uSNHighest attribute from the @REPLCHANGED object for a
2476   partition
2477  */
2478 int dsdb_load_partition_usn(struct ldb_context *ldb, struct ldb_dn *dn, uint64_t *uSN)
2479 {
2480         struct ldb_request *req;
2481         int ret;
2482         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2483         struct dsdb_control_current_partition *p_ctrl;
2484         struct ldb_result *res;
2485
2486         res = talloc_zero(tmp_ctx, struct ldb_result);
2487         if (!res) {
2488                 talloc_free(tmp_ctx);
2489                 return LDB_ERR_OPERATIONS_ERROR;
2490         }
2491
2492         ret = ldb_build_search_req(&req, ldb, tmp_ctx,
2493                                    ldb_dn_new(tmp_ctx, ldb, "@REPLCHANGED"),
2494                                    LDB_SCOPE_BASE,
2495                                    NULL, NULL,
2496                                    NULL,
2497                                    res, ldb_search_default_callback,
2498                                    NULL);
2499         if (ret != LDB_SUCCESS) {
2500                 talloc_free(tmp_ctx);
2501                 return ret;
2502         }
2503
2504         p_ctrl = talloc(req, struct dsdb_control_current_partition);
2505         if (p_ctrl == NULL) {
2506                 talloc_free(res);
2507                 return LDB_ERR_OPERATIONS_ERROR;
2508         }
2509         p_ctrl->version = DSDB_CONTROL_CURRENT_PARTITION_VERSION;
2510         p_ctrl->dn = dn;
2511
2512
2513         ret = ldb_request_add_control(req,
2514                                       DSDB_CONTROL_CURRENT_PARTITION_OID,
2515                                       false, p_ctrl);
2516         if (ret != LDB_SUCCESS) {
2517                 talloc_free(tmp_ctx);
2518                 return ret;
2519         }
2520
2521         /* Run the new request */
2522         ret = ldb_request(ldb, req);
2523
2524         if (ret == LDB_SUCCESS) {
2525                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2526         }
2527
2528         if (ret == LDB_ERR_NO_SUCH_OBJECT) {
2529                 /* it hasn't been created yet, which means
2530                    an implicit value of zero */
2531                 *uSN = 0;
2532                 talloc_free(tmp_ctx);
2533                 return LDB_SUCCESS;
2534         }
2535
2536         if (ret != LDB_SUCCESS) {
2537                 talloc_free(tmp_ctx);
2538                 return ret;
2539         }
2540
2541         if (res->count < 1) {
2542                 *uSN = 0;
2543         } else {
2544                 *uSN = ldb_msg_find_attr_as_uint64(res->msgs[0], "uSNHighest", 0);
2545         }
2546
2547         talloc_free(tmp_ctx);
2548
2549         return LDB_SUCCESS;
2550 }
2551
2552 /*
2553   save the uSNHighest attribute in the @REPLCHANGED object for a
2554   partition
2555  */
2556 int dsdb_save_partition_usn(struct ldb_context *ldb, struct ldb_dn *dn, uint64_t uSN)
2557 {
2558         struct ldb_request *req;
2559         struct ldb_message *msg;
2560         struct dsdb_control_current_partition *p_ctrl;
2561         int ret;
2562
2563         msg = ldb_msg_new(ldb);
2564         if (msg == NULL) {
2565                 return LDB_ERR_OPERATIONS_ERROR;
2566         }
2567
2568         msg->dn = ldb_dn_new(msg, ldb, "@REPLCHANGED");
2569         if (msg->dn == NULL) {
2570                 talloc_free(msg);
2571                 return LDB_ERR_OPERATIONS_ERROR;
2572         }
2573
2574         ret = ldb_msg_add_fmt(msg, "uSNHighest", "%llu", (unsigned long long)uSN);
2575         if (ret != LDB_SUCCESS) {
2576                 talloc_free(msg);
2577                 return ret;
2578         }
2579         msg->elements[0].flags = LDB_FLAG_MOD_REPLACE;
2580
2581
2582         p_ctrl = talloc(msg, struct dsdb_control_current_partition);
2583         if (p_ctrl == NULL) {
2584                 talloc_free(msg);
2585                 return LDB_ERR_OPERATIONS_ERROR;
2586         }
2587         p_ctrl->version = DSDB_CONTROL_CURRENT_PARTITION_VERSION;
2588         p_ctrl->dn = dn;
2589
2590         ret = ldb_build_mod_req(&req, ldb, msg,
2591                                 msg,
2592                                 NULL,
2593                                 NULL, ldb_op_default_callback,
2594                                 NULL);
2595 again:
2596         if (ret != LDB_SUCCESS) {
2597                 talloc_free(msg);
2598                 return ret;
2599         }
2600
2601         ret = ldb_request_add_control(req,
2602                                       DSDB_CONTROL_CURRENT_PARTITION_OID,
2603                                       false, p_ctrl);
2604         if (ret != LDB_SUCCESS) {
2605                 talloc_free(msg);
2606                 return ret;
2607         }
2608
2609         /* Run the new request */
2610         ret = ldb_request(ldb, req);
2611
2612         if (ret == LDB_SUCCESS) {
2613                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2614         }
2615         if (ret == LDB_ERR_NO_SUCH_OBJECT) {
2616                 ret = ldb_build_add_req(&req, ldb, msg,
2617                                         msg,
2618                                         NULL,
2619                                         NULL, ldb_op_default_callback,
2620                                         NULL);
2621                 goto again;
2622         }
2623
2624         talloc_free(msg);
2625
2626         return ret;
2627 }
2628
2629 int drsuapi_DsReplicaCursor2_compare(const struct drsuapi_DsReplicaCursor2 *c1,
2630                                                    const struct drsuapi_DsReplicaCursor2 *c2)
2631 {
2632         return GUID_compare(&c1->source_dsa_invocation_id, &c2->source_dsa_invocation_id);
2633 }
2634
2635 /*
2636   see if we are a RODC
2637
2638   TODO: This should take a sam_ctx, and lookup the right object (with
2639   a cache)
2640 */
2641 bool samdb_rodc(struct loadparm_context *lp_ctx)
2642 {
2643         return lp_parm_bool(lp_ctx, NULL, "repl", "RODC", false);
2644 }
2645
2646
2647 /*
2648   return NTDS options flags. See MS-ADTS 7.1.1.2.2.1.2.1.1
2649
2650   flags are DS_NTDS_OPTION_*
2651 */
2652 int samdb_ntds_options(struct ldb_context *ldb, uint32_t *options)
2653 {
2654         TALLOC_CTX *tmp_ctx;
2655         const char *attrs[] = { "options", NULL };
2656         int ret;
2657         struct ldb_result *res;
2658
2659         tmp_ctx = talloc_new(ldb);
2660         if (tmp_ctx == NULL) {
2661                 goto failed;
2662         }
2663
2664         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
2665         if (ret) {
2666                 goto failed;
2667         }
2668
2669         if (res->count != 1) {
2670                 goto failed;
2671         }
2672
2673         *options = samdb_result_uint(res->msgs[0], "options", 0);
2674
2675         talloc_free(tmp_ctx);
2676
2677         return LDB_SUCCESS;
2678
2679 failed:
2680         DEBUG(1,("Failed to find our own NTDS Settings objectGUID in the ldb!\n"));
2681         talloc_free(tmp_ctx);
2682         return LDB_ERR_NO_SUCH_OBJECT;
2683 }
2684
2685 /*
2686  * Function which generates a "lDAPDisplayName" attribute from a "CN" one.
2687  * Algorithm implemented according to MS-ADTS 3.1.1.2.3.4
2688  */
2689 const char *samdb_cn_to_lDAPDisplayName(TALLOC_CTX *mem_ctx, const char *cn)
2690 {
2691         char **tokens, *ret;
2692         size_t i;
2693
2694         tokens = str_list_make(mem_ctx, cn, " -_");
2695         if (tokens == NULL)
2696                 return NULL;
2697
2698         /* "tolower()" and "toupper()" should also work properly on 0x00 */
2699         tokens[0][0] = tolower(tokens[0][0]);
2700         for (i = 1; i < str_list_length((const char **)tokens); i++)
2701                 tokens[i][0] = toupper(tokens[i][0]);
2702
2703         ret = talloc_strdup(mem_ctx, tokens[0]);
2704         for (i = 1; i < str_list_length((const char **)tokens); i++)
2705                 ret = talloc_asprintf_append_buffer(ret, "%s", tokens[i]);
2706
2707         talloc_free(tokens);
2708
2709         return ret;
2710 }
2711
2712 /*
2713   return domain functional level
2714   returns DS_DOMAIN_FUNCTION_*
2715  */
2716 int dsdb_functional_level(struct ldb_context *ldb)
2717 {
2718         int *domainFunctionality =
2719                 talloc_get_type(ldb_get_opaque(ldb, "domainFunctionality"), int);
2720         if (!domainFunctionality) {
2721                 DEBUG(0,(__location__ ": WARNING: domainFunctionality not setup\n"));
2722                 return DS_DOMAIN_FUNCTION_2000;
2723         }
2724         return *domainFunctionality;
2725 }
2726
2727 /*
2728   return a GUID from a extended DN structure
2729  */
2730 NTSTATUS dsdb_get_extended_dn_guid(struct ldb_dn *dn, struct GUID *guid)
2731 {
2732         const struct ldb_val *v;
2733
2734         v = ldb_dn_get_extended_component(dn, "GUID");
2735         if (v == NULL) {
2736                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
2737         }
2738
2739         return GUID_from_ndr_blob(v, guid);
2740 }
2741
2742 /*
2743   return true if a ldb_val containing a DN in storage form is deleted
2744  */
2745 bool dsdb_dn_is_deleted_val(struct ldb_val *val)
2746 {
2747         /* this relies on the sort order and exact format of
2748            linearized extended DNs */
2749         if (val->length >= 12 &&
2750             strncmp((const char *)val->data, "<DELETED=1>;", 12) == 0) {
2751                 return true;
2752         }
2753         return false;
2754 }
2755
2756 /*
2757   return a DN for a wellknown GUID
2758  */
2759 int dsdb_wellknown_dn(struct ldb_context *samdb, TALLOC_CTX *mem_ctx,
2760                       struct ldb_dn *nc_root, const char *wk_guid,
2761                       struct ldb_dn **wkguid_dn)
2762 {
2763         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2764         const char *attrs[] = { NULL };
2765         int ret;
2766         struct ldb_dn *dn;
2767         struct ldb_result *res;
2768
2769         /* construct the magic WKGUID DN */
2770         dn = ldb_dn_new_fmt(tmp_ctx, samdb, "<WKGUID=%s,%s>",
2771                             wk_guid, ldb_dn_get_linearized(nc_root));
2772         if (!wkguid_dn) {
2773                 talloc_free(tmp_ctx);
2774                 return LDB_ERR_OPERATIONS_ERROR;
2775         }
2776
2777         ret = dsdb_search_dn_with_deleted(samdb, tmp_ctx, &res, dn, attrs);
2778         if (ret != LDB_SUCCESS) {
2779                 talloc_free(tmp_ctx);
2780                 return ret;
2781         }
2782
2783         (*wkguid_dn) = talloc_steal(mem_ctx, res->msgs[0]->dn);
2784         talloc_free(tmp_ctx);
2785         return LDB_SUCCESS;
2786 }
2787
2788
2789 /*
2790   find a NC root given a DN within the NC
2791  */
2792 int dsdb_find_nc_root(struct ldb_context *samdb, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
2793                       struct ldb_dn **nc_root)
2794 {
2795         const char *root_attrs[] = { "namingContexts", NULL };
2796         TALLOC_CTX *tmp_ctx;
2797         int ret;
2798         struct ldb_message_element *el;
2799         struct ldb_result *root_res;
2800         int i;
2801         struct ldb_dn **nc_dns;
2802
2803         tmp_ctx = talloc_new(samdb);
2804         if (tmp_ctx == NULL) {
2805                 return LDB_ERR_OPERATIONS_ERROR;
2806         }
2807
2808         ret = ldb_search(samdb, tmp_ctx, &root_res,
2809                          ldb_dn_new(tmp_ctx, samdb, ""), LDB_SCOPE_BASE, root_attrs, NULL);
2810         if (ret) {
2811                 DEBUG(1,("Searching for namingContexts in rootDSE failed: %s\n", ldb_errstring(samdb)));
2812                 talloc_free(tmp_ctx);
2813                 return ret;
2814        }
2815
2816        el = ldb_msg_find_element(root_res->msgs[0], "namingContexts");
2817        if (!el) {
2818                DEBUG(1,("Finding namingContexts element in root_res failed: %s\n",
2819                         ldb_errstring(samdb)));
2820                talloc_free(tmp_ctx);
2821                return LDB_ERR_NO_SUCH_ATTRIBUTE;
2822        }
2823
2824        nc_dns = talloc_array(tmp_ctx, struct ldb_dn *, el->num_values);
2825        if (!nc_dns) {
2826                talloc_free(tmp_ctx);
2827                return LDB_ERR_OPERATIONS_ERROR;
2828        }
2829
2830        for (i=0; i<el->num_values; i++) {
2831                nc_dns[i] = ldb_dn_from_ldb_val(nc_dns, samdb, &el->values[i]);
2832                if (nc_dns[i] == NULL) {
2833                        talloc_free(tmp_ctx);
2834                        return LDB_ERR_OPERATIONS_ERROR;
2835                }
2836        }
2837
2838        qsort(nc_dns, el->num_values, sizeof(nc_dns[0]), (comparison_fn_t)ldb_dn_compare);
2839
2840        for (i=0; i<el->num_values; i++) {
2841                if (ldb_dn_compare_base(nc_dns[i], dn) == 0) {
2842                        (*nc_root) = talloc_steal(mem_ctx, nc_dns[i]);
2843                        talloc_free(tmp_ctx);
2844                        return LDB_SUCCESS;
2845                }
2846        }
2847
2848        talloc_free(tmp_ctx);
2849        return LDB_ERR_NO_SUCH_OBJECT;
2850 }
2851
2852
2853 /*
2854   find the deleted objects DN for any object, by looking for the NC
2855   root, then looking up the wellknown GUID
2856  */
2857 int dsdb_get_deleted_objects_dn(struct ldb_context *ldb,
2858                                 TALLOC_CTX *mem_ctx, struct ldb_dn *obj_dn,
2859                                 struct ldb_dn **do_dn)
2860 {
2861         struct ldb_dn *nc_root;
2862         int ret;
2863
2864         ret = dsdb_find_nc_root(ldb, mem_ctx, obj_dn, &nc_root);
2865         if (ret != LDB_SUCCESS) {
2866                 return ret;
2867         }
2868
2869         ret = dsdb_wellknown_dn(ldb, mem_ctx, nc_root, DS_GUID_DELETED_OBJECTS_CONTAINER, do_dn);
2870         talloc_free(nc_root);
2871         return ret;
2872 }