r25446: Merge some changes I made on the way home from SFO:
[jelmer/samba4-debian.git] / source / dsdb / samdb / samdb.c
index 0f6e57c9cf7992f633271934caa4011dd7b5a1b4..2f755d6c791fb21b8678f37ba6cfd4262440add5 100644 (file)
@@ -4,10 +4,12 @@
    interface functions for the sam database
 
    Copyright (C) Andrew Tridgell 2004
-   
+   Copyright (C) Volker Lendecke 2004
+   Copyright (C) Andrew Bartlett <abartlet@samba.org> 2006
+
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
@@ -16,8 +18,7 @@
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
 #include "librpc/gen_ndr/ndr_security.h"
 #include "lib/ldb/include/ldb.h"
 #include "lib/ldb/include/ldb_errors.h"
-#include "libcli/security/proto.h"
+#include "libcli/security/security.h"
+#include "libcli/auth/libcli_auth.h"
+#include "libcli/ldap/ldap.h"
 #include "system/time.h"
 #include "system/filesys.h"
 #include "db_wrap.h"
 #include "dsdb/samdb/samdb.h"
-#include "ads.h"
+#include "dsdb/common/flags.h"
+#include "param/param.h"
 
 /*
   connect to the SAM database
@@ -41,11 +45,13 @@ struct ldb_context *samdb_connect(TALLOC_CTX *mem_ctx,
                                  struct auth_session_info *session_info)
 {
        struct ldb_context *ldb;
-       ldb = ldb_wrap_connect(mem_ctx, lp_sam_url(), session_info,
+       ldb = ldb_wrap_connect(mem_ctx, global_loadparm, 
+                              lp_sam_url(global_loadparm), session_info,
                               NULL, 0, NULL);
        if (!ldb) {
                return NULL;
        }
+       dsdb_make_schema_global(ldb);
        return ldb;
 }
 
@@ -55,7 +61,7 @@ struct ldb_context *samdb_connect(TALLOC_CTX *mem_ctx,
 */
 int samdb_search_domain(struct ldb_context *sam_ldb,
                        TALLOC_CTX *mem_ctx, 
-                       const struct ldb_dn *basedn,
+                       struct ldb_dn *basedn,
                        struct ldb_message ***res,
                        const char * const *attrs,
                        const struct dom_sid *domain_sid,
@@ -96,7 +102,7 @@ int samdb_search_domain(struct ldb_context *sam_ldb,
 */
 const char *samdb_search_string_v(struct ldb_context *sam_ldb,
                                  TALLOC_CTX *mem_ctx,
-                                 const struct ldb_dn *basedn,
+                                 struct ldb_dn *basedn,
                                  const char *attr_name,
                                  const char *format, va_list ap) _PRINTF_ATTRIBUTE(5,0)
 {
@@ -125,7 +131,7 @@ const char *samdb_search_string_v(struct ldb_context *sam_ldb,
 */
 const char *samdb_search_string(struct ldb_context *sam_ldb,
                                TALLOC_CTX *mem_ctx,
-                               const struct ldb_dn *basedn,
+                               struct ldb_dn *basedn,
                                const char *attr_name,
                                const char *format, ...) _PRINTF_ATTRIBUTE(5,6)
 {
@@ -141,7 +147,7 @@ const char *samdb_search_string(struct ldb_context *sam_ldb,
 
 struct ldb_dn *samdb_search_dn(struct ldb_context *sam_ldb,
                               TALLOC_CTX *mem_ctx,
-                              const struct ldb_dn *basedn,
+                              struct ldb_dn *basedn,
                               const char *format, ...) _PRINTF_ATTRIBUTE(4,5)
 {
        va_list ap;
@@ -166,7 +172,7 @@ struct ldb_dn *samdb_search_dn(struct ldb_context *sam_ldb,
 */
 struct dom_sid *samdb_search_dom_sid(struct ldb_context *sam_ldb,
                                     TALLOC_CTX *mem_ctx,
-                                    const struct ldb_dn *basedn,
+                                    struct ldb_dn *basedn,
                                     const char *attr_name,
                                     const char *format, ...) _PRINTF_ATTRIBUTE(5,6)
 {
@@ -199,7 +205,7 @@ struct dom_sid *samdb_search_dom_sid(struct ldb_context *sam_ldb,
 */
 int samdb_search_count(struct ldb_context *sam_ldb,
                       TALLOC_CTX *mem_ctx,
-                      const struct ldb_dn *basedn,
+                      struct ldb_dn *basedn,
                       const char *format, ...) _PRINTF_ATTRIBUTE(4,5)
 {
        va_list ap;
@@ -221,7 +227,7 @@ int samdb_search_count(struct ldb_context *sam_ldb,
 uint_t samdb_search_uint(struct ldb_context *sam_ldb,
                         TALLOC_CTX *mem_ctx,
                         uint_t default_value,
-                        const struct ldb_dn *basedn,
+                        struct ldb_dn *basedn,
                         const char *attr_name,
                         const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
 {
@@ -249,7 +255,7 @@ uint_t samdb_search_uint(struct ldb_context *sam_ldb,
 int64_t samdb_search_int64(struct ldb_context *sam_ldb,
                           TALLOC_CTX *mem_ctx,
                           int64_t default_value,
-                          const struct ldb_dn *basedn,
+                          struct ldb_dn *basedn,
                           const char *attr_name,
                           const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
 {
@@ -277,7 +283,7 @@ int64_t samdb_search_int64(struct ldb_context *sam_ldb,
 */
 int samdb_search_string_multiple(struct ldb_context *sam_ldb,
                                 TALLOC_CTX *mem_ctx,
-                                const struct ldb_dn *basedn,
+                                struct ldb_dn *basedn,
                                 const char ***strs,
                                 const char *attr_name,
                                 const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
@@ -324,40 +330,42 @@ int samdb_search_string_multiple(struct ldb_context *sam_ldb,
 /*
   pull a uint from a result set. 
 */
-uint_t samdb_result_uint(struct ldb_message *msg, const char *attr, uint_t default_value)
+uint_t samdb_result_uint(const struct ldb_message *msg, const char *attr, uint_t default_value)
 {
-       return ldb_msg_find_uint(msg, attr, default_value);
+       return ldb_msg_find_attr_as_uint(msg, attr, default_value);
 }
 
 /*
   pull a (signed) int64 from a result set. 
 */
-int64_t samdb_result_int64(struct ldb_message *msg, const char *attr, int64_t default_value)
+int64_t samdb_result_int64(const struct ldb_message *msg, const char *attr, int64_t default_value)
 {
-       return ldb_msg_find_int64(msg, attr, default_value);
+       return ldb_msg_find_attr_as_int64(msg, attr, default_value);
 }
 
 /*
   pull a string from a result set. 
 */
-const char *samdb_result_string(struct ldb_message *msg, const char *attr, 
+const char *samdb_result_string(const struct ldb_message *msg, const char *attr, 
                                const char *default_value)
 {
-       return ldb_msg_find_string(msg, attr, default_value);
+       return ldb_msg_find_attr_as_string(msg, attr, default_value);
 }
 
-struct ldb_dn *samdb_result_dn(TALLOC_CTX *mem_ctx, struct ldb_message *msg,
+struct ldb_dn *samdb_result_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
                               const char *attr, struct ldb_dn *default_value)
 {
-       const char *string = samdb_result_string(msg, attr, NULL);
-       if (string == NULL) return default_value;
-       return ldb_dn_explode(mem_ctx, string);
+       struct ldb_dn *ret_dn = ldb_msg_find_attr_as_dn(ldb, mem_ctx, msg, attr);
+       if (!ret_dn) {
+               return default_value;
+       }
+       return ret_dn;
 }
 
 /*
   pull a rid from a objectSid in a result set. 
 */
-uint32_t samdb_result_rid_from_sid(TALLOC_CTX *mem_ctx, struct ldb_message *msg, 
+uint32_t samdb_result_rid_from_sid(TALLOC_CTX *mem_ctx, const struct ldb_message *msg, 
                                   const char *attr, uint32_t default_value)
 {
        struct dom_sid *sid;
@@ -375,7 +383,7 @@ uint32_t samdb_result_rid_from_sid(TALLOC_CTX *mem_ctx, struct ldb_message *msg,
 /*
   pull a dom_sid structure from a objectSid in a result set. 
 */
-struct dom_sid *samdb_result_dom_sid(TALLOC_CTX *mem_ctx, struct ldb_message *msg, 
+struct dom_sid *samdb_result_dom_sid(TALLOC_CTX *mem_ctx, const struct ldb_message *msg, 
                                     const char *attr)
 {
        const struct ldb_val *v;
@@ -401,7 +409,7 @@ struct dom_sid *samdb_result_dom_sid(TALLOC_CTX *mem_ctx, struct ldb_message *ms
 /*
   pull a guid structure from a objectGUID in a result set. 
 */
-struct GUID samdb_result_guid(struct ldb_message *msg, const char *attr)
+struct GUID samdb_result_guid(const struct ldb_message *msg, const char *attr)
 {
        const struct ldb_val *v;
        NTSTATUS status;
@@ -429,7 +437,7 @@ struct GUID samdb_result_guid(struct ldb_message *msg, const char *attr)
   pull a sid prefix from a objectSid in a result set. 
   this is used to find the domain sid for a user
 */
-struct dom_sid *samdb_result_sid_prefix(TALLOC_CTX *mem_ctx, struct ldb_message *msg, 
+struct dom_sid *samdb_result_sid_prefix(TALLOC_CTX *mem_ctx, const struct ldb_message *msg, 
                                        const char *attr)
 {
        struct dom_sid *sid = samdb_result_dom_sid(mem_ctx, msg, attr);
@@ -443,9 +451,7 @@ struct dom_sid *samdb_result_sid_prefix(TALLOC_CTX *mem_ctx, struct ldb_message
 */
 NTTIME samdb_result_nttime(struct ldb_message *msg, const char *attr, NTTIME default_value)
 {
-       const char *str = ldb_msg_find_string(msg, attr, NULL);
-       if (!str) return default_value;
-       return nttime_from_string(str);
+       return ldb_msg_find_attr_as_uint64(msg, attr, default_value);
 }
 
 /*
@@ -453,7 +459,7 @@ NTTIME samdb_result_nttime(struct ldb_message *msg, const char *attr, NTTIME def
 */
 uint64_t samdb_result_uint64(struct ldb_message *msg, const char *attr, uint64_t default_value)
 {
-       return ldb_msg_find_uint64(msg, attr, default_value);
+       return ldb_msg_find_attr_as_uint64(msg, attr, default_value);
 }
 
 
@@ -463,7 +469,7 @@ uint64_t samdb_result_uint64(struct ldb_message *msg, const char *attr, uint64_t
 */
 NTTIME samdb_result_allow_password_change(struct ldb_context *sam_ldb, 
                                          TALLOC_CTX *mem_ctx, 
-                                         const struct ldb_dn *domain_dn, 
+                                         struct ldb_dn *domain_dn, 
                                          struct ldb_message *msg, 
                                          const char *attr)
 {
@@ -489,7 +495,7 @@ NTTIME samdb_result_allow_password_change(struct ldb_context *sam_ldb,
 */
 NTTIME samdb_result_force_password_change(struct ldb_context *sam_ldb, 
                                          TALLOC_CTX *mem_ctx, 
-                                         const struct ldb_dn *domain_dn, 
+                                         struct ldb_dn *domain_dn, 
                                          struct ldb_message *msg)
 {
        uint64_t attr_time = samdb_result_uint64(msg, "pwdLastSet", 0);
@@ -565,7 +571,7 @@ NTSTATUS samdb_result_passwords(TALLOC_CTX *mem_ctx, struct ldb_message *msg,
        struct samr_Password *lmPwdHash, *ntPwdHash;
        if (nt_pwd) {
                int num_nt;
-               num_nt = samdb_result_hashes(mem_ctx, msg, "ntPwdHash", &ntPwdHash);
+               num_nt = samdb_result_hashes(mem_ctx, msg, "unicodePwd", &ntPwdHash);
                if (num_nt == 0) {
                        *nt_pwd = NULL;
                } else if (num_nt > 1) {
@@ -576,7 +582,7 @@ NTSTATUS samdb_result_passwords(TALLOC_CTX *mem_ctx, struct ldb_message *msg,
        }
        if (lm_pwd) {
                int num_lm;
-               num_lm = samdb_result_hashes(mem_ctx, msg, "lmPwdHash", &lmPwdHash);
+               num_lm = samdb_result_hashes(mem_ctx, msg, "dBCSPwd", &lmPwdHash);
                if (num_lm == 0) {
                        *lm_pwd = NULL;
                } else if (num_lm > 1) {
@@ -614,54 +620,120 @@ struct samr_LogonHours samdb_result_logon_hours(TALLOC_CTX *mem_ctx, struct ldb_
 */
 uint16_t samdb_result_acct_flags(struct ldb_message *msg, const char *attr)
 {
-       uint_t userAccountControl = ldb_msg_find_uint(msg, attr, 0);
+       uint_t userAccountControl = ldb_msg_find_attr_as_uint(msg, attr, 0);
        return samdb_uf2acb(userAccountControl);
 }
 
+
+/* Find an attribute, with a particular value */
+
+/* The current callers of this function expect a very specific
+ * behaviour: In particular, objectClass subclass equivilance is not
+ * wanted.  This means that we should not lookup the schema for the
+ * comparison function */
+struct ldb_message_element *samdb_find_attribute(struct ldb_context *ldb, 
+                                                const struct ldb_message *msg, 
+                                                const char *name, const char *value)
+{
+       int i;
+       struct ldb_message_element *el = ldb_msg_find_element(msg, name);
+
+       if (!el) {
+               return NULL;
+       }
+
+       for (i=0;i<el->num_values;i++) {
+               if (ldb_attr_cmp(value, (char *)el->values[i].data) == 0) {
+                       return el;
+               }
+       }
+
+       return NULL;
+}
+
+int samdb_find_or_add_value(struct ldb_context *ldb, struct ldb_message *msg, const char *name, const char *set_value)
+{
+       if (samdb_find_attribute(ldb, msg, name, set_value) == NULL) {
+               return samdb_msg_add_string(ldb, msg, msg, name, set_value);
+       }
+       return LDB_SUCCESS;
+}
+
+int samdb_find_or_add_attribute(struct ldb_context *ldb, struct ldb_message *msg, const char *name, const char *set_value)
+{
+       struct ldb_message_element *el;
+
+               el = ldb_msg_find_element(msg, name);
+       if (el) {
+               return LDB_SUCCESS;
+       }
+               
+       return samdb_msg_add_string(ldb, msg, msg, name, set_value);
+}
+
+
 /*
   copy from a template record to a message
 */
-int samdb_copy_template(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, 
-                       struct ldb_message *msg, const char *expression)
+int samdb_copy_template(struct ldb_context *ldb, 
+                       struct ldb_message *msg, const char *name,
+                       const char **errstring)
 {
-       struct ldb_message **res, *t;
+       struct ldb_result *res;
+       struct ldb_message *t;
        int ret, i, j;
-       
+       struct ldb_dn *basedn = ldb_dn_new(ldb, ldb, "cn=Templates");
 
+       *errstring = NULL;      
+
+       if (!ldb_dn_add_child_fmt(basedn, "CN=Template%s", name)) {
+               *errstring = talloc_asprintf(msg, "samdb_copy_template: ERROR: Failed to contruct DN for template '%s'", 
+                                            name);
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+       
        /* pull the template record */
-       ret = gendb_search(sam_ldb, mem_ctx, NULL, &res, NULL, "%s", expression);
-       if (ret != 1) {
-               DEBUG(1,("samdb: ERROR: template '%s' matched %d records\n", 
-                        expression, ret));
-               return -1;
+       ret = ldb_search(ldb, basedn, LDB_SCOPE_BASE, "cn=*", NULL, &res);
+       talloc_free(basedn);
+       if (ret != LDB_SUCCESS) {
+               *errstring = talloc_steal(msg, ldb_errstring(ldb));
+               return ret;
+       }
+       if (res->count != 1) {
+               *errstring = talloc_asprintf(msg, "samdb_copy_template: ERROR: template '%s' matched %d records, expected 1", 
+                                            name, 
+                                            res->count);
+               talloc_free(res);
+               return LDB_ERR_OPERATIONS_ERROR;
        }
-       t = res[0];
+       t = res->msgs[0];
 
-       for (i=0;i<t->num_elements;i++) {
+       for (i = 0; i < t->num_elements; i++) {
                struct ldb_message_element *el = &t->elements[i];
                /* some elements should not be copied from the template */
-               if (strcasecmp(el->name, "cn") == 0 ||
-                   strcasecmp(el->name, "name") == 0 ||
-                   strcasecmp(el->name, "sAMAccountName") == 0) {
+               if (ldb_attr_cmp(el->name, "cn") == 0 ||
+                   ldb_attr_cmp(el->name, "name") == 0 ||
+                   ldb_attr_cmp(el->name, "objectClass") == 0 ||
+                   ldb_attr_cmp(el->name, "sAMAccountName") == 0 ||
+                   ldb_attr_cmp(el->name, "sAMAccountName") == 0 ||
+                   ldb_attr_cmp(el->name, "distinguishedName") == 0 ||
+                   ldb_attr_cmp(el->name, "objectGUID") == 0) {
                        continue;
                }
-               for (j=0;j<el->num_values;j++) {
-                       if (strcasecmp(el->name, "objectClass") == 0 &&
-                           (strcasecmp((char *)el->values[j].data, "Template") == 0 ||
-                            strcasecmp((char *)el->values[j].data, "userTemplate") == 0 ||
-                            strcasecmp((char *)el->values[j].data, "groupTemplate") == 0 ||
-                            strcasecmp((char *)el->values[j].data, "foreignSecurityTemplate") == 0 ||
-                            strcasecmp((char *)el->values[j].data, "aliasTemplate") == 0 || 
-                            strcasecmp((char *)el->values[j].data, "trustedDomainTemplate") == 0 || 
-                            strcasecmp((char *)el->values[j].data, "secretTemplate") == 0)) {
-                               continue;
+               for (j = 0; j < el->num_values; j++) {
+                       ret = samdb_find_or_add_attribute(ldb, msg, el->name, 
+                                                         (char *)el->values[j].data);
+                       if (ret) {
+                               *errstring = talloc_asprintf(msg, "Adding attribute %s failed.", el->name);
+                               talloc_free(res);
+                               return ret;
                        }
-                       samdb_msg_add_string(sam_ldb, mem_ctx, msg, el->name, 
-                                            (char *)el->values[j].data);
                }
        }
 
-       return 0;
+       talloc_free(res);
+
+       return LDB_SUCCESS;
 }
 
 
@@ -674,7 +746,7 @@ int samdb_msg_add_string(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struc
        char *s = talloc_strdup(mem_ctx, str);
        char *a = talloc_strdup(mem_ctx, attr_name);
        if (s == NULL || a == NULL) {
-               return -1;
+               return LDB_ERR_OPERATIONS_ERROR;
        }
        return ldb_msg_add_string(msg, a, s);
 }
@@ -692,7 +764,7 @@ int samdb_msg_add_dom_sid(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, stru
        if (!NT_STATUS_IS_OK(status)) {
                return -1;
        }
-       return ldb_msg_add_value(msg, attr_name, &v);
+       return ldb_msg_add_value(msg, attr_name, &v, NULL);
 }
 
 
@@ -704,7 +776,7 @@ int samdb_msg_add_delete(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struc
 {
        /* we use an empty replace rather than a delete, as it allows for 
           samdb_replace() to be used everywhere */
-       return ldb_msg_add_empty(msg, attr_name, LDB_FLAG_MOD_REPLACE);
+       return ldb_msg_add_empty(msg, attr_name, LDB_FLAG_MOD_REPLACE, NULL);
 }
 
 /*
@@ -757,6 +829,16 @@ int samdb_msg_add_delval(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struc
        return 0;
 }
 
+/*
+  add a int element to a message
+*/
+int samdb_msg_add_int(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
+                      const char *attr_name, int v)
+{
+       const char *s = talloc_asprintf(mem_ctx, "%d", v);
+       return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, s);
+}
+
 /*
   add a uint_t element to a message
 */
@@ -799,13 +881,13 @@ int samdb_msg_add_hash(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct
                return -1;
        }
        val.length = 16;
-       return ldb_msg_add_value(msg, attr_name, &val);
+       return ldb_msg_add_value(msg, attr_name, &val, NULL);
 }
 
 /*
   add a samr_Password array to a message
 */
-int samdb_msg_add_hashes(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
+int samdb_msg_add_hashes(TALLOC_CTX *mem_ctx, struct ldb_message *msg,
                         const char *attr_name, struct samr_Password *hashes, uint_t count)
 {
        struct ldb_val val;
@@ -818,7 +900,7 @@ int samdb_msg_add_hashes(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struc
        for (i=0;i<count;i++) {
                memcpy(i*16 + (char *)val.data, hashes[i].hash, 16);
        }
-       return ldb_msg_add_value(msg, attr_name, &val);
+       return ldb_msg_add_value(msg, attr_name, &val, NULL);
 }
 
 /*
@@ -839,7 +921,7 @@ int samdb_msg_add_logon_hours(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx,
        struct ldb_val val;
        val.length = hours->units_per_week / 8;
        val.data = hours->bits;
-       return ldb_msg_add_value(msg, attr_name, &val);
+       return ldb_msg_add_value(msg, attr_name, &val, NULL);
 }
 
 /*
@@ -848,7 +930,7 @@ int samdb_msg_add_logon_hours(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx,
 int samdb_msg_add_value(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
                              const char *attr_name, const struct ldb_val *val)
 {
-       return ldb_msg_add_value(msg, attr_name, val);
+       return ldb_msg_add_value(msg, attr_name, val, NULL);
 }
 
 /*
@@ -863,7 +945,7 @@ int samdb_msg_set_value(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct
        if (el) {
                el->num_values = 0;
        }
-       return ldb_msg_add_value(msg, attr_name, val);
+       return ldb_msg_add_value(msg, attr_name, val, NULL);
 }
 
 /*
@@ -892,7 +974,7 @@ int samdb_add(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_messa
 /*
   delete a record
 */
-int samdb_delete(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, const struct ldb_dn *dn)
+int samdb_delete(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_dn *dn)
 {
        return ldb_delete(sam_ldb, dn);
 }
@@ -933,56 +1015,60 @@ struct security_descriptor *samdb_default_security_descriptor(TALLOC_CTX *mem_ct
        return sd;
 }
 
-struct ldb_dn *samdb_base_dn(TALLOC_CTX *mem_ctx) 
+struct ldb_dn *samdb_base_dn(struct ldb_context *sam_ctx) 
 {
-       TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
-       int server_role = lp_server_role();
-       const char **split_realm;
-       struct ldb_dn *dn;
-       
-       if (!tmp_ctx) {
+       return ldb_get_default_basedn(sam_ctx);
+}
+
+struct ldb_dn *samdb_config_dn(struct ldb_context *sam_ctx) 
+{
+       return ldb_get_config_basedn(sam_ctx);
+}
+
+struct ldb_dn *samdb_schema_dn(struct ldb_context *sam_ctx) 
+{
+       return ldb_get_schema_basedn(sam_ctx);
+}
+
+struct ldb_dn *samdb_root_dn(struct ldb_context *sam_ctx) 
+{
+       return ldb_get_root_basedn(sam_ctx);
+}
+
+struct ldb_dn *samdb_partitions_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
+{
+       struct ldb_dn *new_dn;
+
+       new_dn = ldb_dn_copy(mem_ctx, samdb_config_dn(sam_ctx));
+       if ( ! ldb_dn_add_child_fmt(new_dn, "CN=Partitions")) {
+               talloc_free(new_dn);
                return NULL;
        }
+       return new_dn;
+}
 
-       if ((server_role == ROLE_DOMAIN_PDC)
-           || (server_role == ROLE_DOMAIN_BDC)) {
-               int i;
-               split_realm = str_list_make(tmp_ctx, lp_realm(), ".");
-               if (!split_realm) {
-                       talloc_free(tmp_ctx);
-                       return NULL;
-               }
-               dn = NULL;
-               i = str_list_length(split_realm);
-               i--;
-               for (; i >= 0; i--) {
-                       dn = ldb_dn_build_child(tmp_ctx, "dc", split_realm[i], dn);
-                       if (!dn) {
-                               talloc_free(tmp_ctx);
-                               return NULL;
-                       }
-               }
-               return dn;
+struct ldb_dn *samdb_sites_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
+{
+       struct ldb_dn *new_dn;
+
+       new_dn = ldb_dn_copy(mem_ctx, samdb_config_dn(sam_ctx));
+       if ( ! ldb_dn_add_child_fmt(new_dn, "CN=Sites")) {
+               talloc_free(new_dn);
+               return NULL;
        }
-       return ldb_dn_string_compose(mem_ctx, NULL, "cn=%s", lp_netbios_name());
+       return new_dn;
 }
 
-
 /*
   work out the domain sid for the current open ldb
 */
 const struct dom_sid *samdb_domain_sid(struct ldb_context *ldb)
 {
-       const char *attrs[] = { "rootDomainNamingContext", NULL };
-       int ret;
-       struct ldb_result *res = NULL;
        TALLOC_CTX *tmp_ctx;
        struct dom_sid *domain_sid;
-       const char *basedn_s;
-       struct ldb_dn *basedn;
 
        /* see if we have a cached copy */
-       domain_sid = ldb_get_opaque(ldb, "cache.domain_sid");
+       domain_sid = (struct dom_sid *)ldb_get_opaque(ldb, "cache.domain_sid");
        if (domain_sid) {
                return domain_sid;
        }
@@ -992,47 +1078,951 @@ const struct dom_sid *samdb_domain_sid(struct ldb_context *ldb)
                goto failed;
        }
 
-       basedn = ldb_dn_explode(tmp_ctx, "");
-       if (basedn == NULL) {
+       /* find the domain_sid */
+       domain_sid = samdb_search_dom_sid(ldb, tmp_ctx, ldb_get_default_basedn(ldb),
+                                         "objectSid", "objectClass=domainDNS");
+       if (domain_sid == NULL) {
+               goto failed;
+       }
+
+       /* cache the domain_sid in the ldb */
+       if (ldb_set_opaque(ldb, "cache.domain_sid", domain_sid) != LDB_SUCCESS) {
+               goto failed;
+       }
+
+       talloc_steal(ldb, domain_sid);
+       talloc_free(tmp_ctx);
+
+       return domain_sid;
+
+failed:
+       DEBUG(1,("Failed to find domain_sid for open ldb\n"));
+       talloc_free(tmp_ctx);
+       return NULL;
+}
+
+/* Obtain the short name of the flexible single master operator
+ * (FSMO), such as the PDC Emulator */
+const char *samdb_result_fsmo_name(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const struct ldb_message *msg, 
+                            const char *attr)
+{
+       /* Format is cn=NTDS Settings,cn=<NETBIOS name of FSMO>,.... */
+       struct ldb_dn *fsmo_dn = ldb_msg_find_attr_as_dn(ldb, mem_ctx, msg, attr);
+       const struct ldb_val *val = ldb_dn_get_component_val(fsmo_dn, 1);
+       const char *name = ldb_dn_get_component_name(fsmo_dn, 1);
+
+       if (!name || (ldb_attr_cmp(name, "cn") != 0)) {
+               /* Ensure this matches the format.  This gives us a
+                * bit more confidence that a 'cn' value will be a
+                * ascii string */
+               return NULL;
+       }
+       if (val) {
+               return (char *)val->data;
+       }
+       return NULL;
+}
+
+/*
+  work out the ntds settings dn for the current open ldb
+*/
+struct ldb_dn *samdb_ntds_settings_dn(struct ldb_context *ldb)
+{
+       TALLOC_CTX *tmp_ctx;
+       const char *root_attrs[] = { "dsServiceName", NULL };
+       int ret;
+       struct ldb_result *root_res;
+       struct ldb_dn *settings_dn;
+       
+       /* see if we have a cached copy */
+       settings_dn = (struct ldb_dn *)ldb_get_opaque(ldb, "cache.settings_dn");
+       if (settings_dn) {
+               return settings_dn;
+       }
+
+       tmp_ctx = talloc_new(ldb);
+       if (tmp_ctx == NULL) {
                goto failed;
        }
        
-       /* find the basedn of the domain from the rootdse */
-       ret = ldb_search(ldb, basedn, LDB_SCOPE_BASE, NULL, attrs, &res);
+
+       ret = ldb_search(ldb, ldb_dn_new(tmp_ctx, ldb, ""), LDB_SCOPE_BASE, NULL, root_attrs, &root_res);
+       if (ret) {
+               DEBUG(1,("Searching for dsServiceName in rootDSE failed: %s\n", 
+                        ldb_errstring(ldb)));
+               goto failed;
+       }
+       talloc_steal(tmp_ctx, root_res);
+
+       if (root_res->count != 1) {
+               goto failed;
+       }
+
+       settings_dn = ldb_msg_find_attr_as_dn(ldb, tmp_ctx, root_res->msgs[0], "dsServiceName");
+
+       /* cache the domain_sid in the ldb */
+       if (ldb_set_opaque(ldb, "cache.settings_dn", settings_dn) != LDB_SUCCESS) {
+               goto failed;
+       }
+
+       talloc_steal(ldb, settings_dn);
+       talloc_free(tmp_ctx);
+
+       return settings_dn;
+
+failed:
+       DEBUG(1,("Failed to find our own NTDS Settings DN in the ldb!\n"));
+       talloc_free(tmp_ctx);
+       return NULL;
+}
+
+/*
+  work out the ntds settings invocationId for the current open ldb
+*/
+const struct GUID *samdb_ntds_invocation_id(struct ldb_context *ldb)
+{
+       TALLOC_CTX *tmp_ctx;
+       const char *attrs[] = { "invocationId", NULL };
+       int ret;
+       struct ldb_result *res;
+       struct GUID *invocation_id;
+       
+       /* see if we have a cached copy */
+       invocation_id = (struct GUID *)ldb_get_opaque(ldb, "cache.invocation_id");
+       if (invocation_id) {
+               return invocation_id;
+       }
+
+       tmp_ctx = talloc_new(ldb);
+       if (tmp_ctx == NULL) {
+               goto failed;
+       }
+
+       ret = ldb_search(ldb, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, NULL, attrs, &res);
+       if (ret) {
+               goto failed;
+       }
        talloc_steal(tmp_ctx, res);
-       if (ret != LDB_SUCCESS || res->count != 1) {
+
+       if (res->count != 1) {
                goto failed;
        }
 
-       basedn_s = ldb_msg_find_string(res->msgs[0], "rootDomainNamingContext", NULL);
-       if (basedn_s == NULL) {
+       invocation_id = talloc(tmp_ctx, struct GUID);
+       if (!invocation_id) {
                goto failed;
        }
 
-       basedn = ldb_dn_explode(tmp_ctx, basedn_s);
-       if (basedn == NULL) {
+       *invocation_id = samdb_result_guid(res->msgs[0], "invocationId");
+
+       /* cache the domain_sid in the ldb */
+       if (ldb_set_opaque(ldb, "cache.invocation_id", invocation_id) != LDB_SUCCESS) {
                goto failed;
        }
 
-       /* find the domain_sid */
-       domain_sid = samdb_search_dom_sid(ldb, tmp_ctx, basedn, 
-                                         "objectSid", "objectClass=domainDNS");
-       if (domain_sid == NULL) {
+       talloc_steal(ldb, invocation_id);
+       talloc_free(tmp_ctx);
+
+       return invocation_id;
+
+failed:
+       DEBUG(1,("Failed to find our own NTDS Settings invocationId in the ldb!\n"));
+       talloc_free(tmp_ctx);
+       return NULL;
+}
+
+bool samdb_set_ntds_invocation_id(struct ldb_context *ldb, const struct GUID *invocation_id_in)
+{
+       TALLOC_CTX *tmp_ctx;
+       struct GUID *invocation_id_new;
+       struct GUID *invocation_id_old;
+
+       /* see if we have a cached copy */
+       invocation_id_old = (struct GUID *)ldb_get_opaque(ldb, 
+                                                        "cache.invocation_id");
+
+       tmp_ctx = talloc_new(ldb);
+       if (tmp_ctx == NULL) {
+               goto failed;
+       }
+
+       invocation_id_new = talloc(tmp_ctx, struct GUID);
+       if (!invocation_id_new) {
                goto failed;
        }
 
+       *invocation_id_new = *invocation_id_in;
+
        /* cache the domain_sid in the ldb */
-       if (ldb_set_opaque(ldb, "cache.domain_sid", domain_sid) != LDB_SUCCESS) {
+       if (ldb_set_opaque(ldb, "cache.invocation_id", invocation_id_new) != LDB_SUCCESS) {
                goto failed;
        }
 
-       talloc_steal(ldb, domain_sid);
+       talloc_steal(ldb, invocation_id_new);
        talloc_free(tmp_ctx);
+       talloc_free(invocation_id_old);
 
-       return domain_sid;
+       return true;
 
 failed:
-       DEBUG(1,("Failed to find domain_sid for open ldb\n"));
+       DEBUG(1,("Failed to set our own cached invocationId in the ldb!\n"));
+       talloc_free(tmp_ctx);
+       return false;
+}
+
+/*
+  work out the ntds settings objectGUID for the current open ldb
+*/
+const struct GUID *samdb_ntds_objectGUID(struct ldb_context *ldb)
+{
+       TALLOC_CTX *tmp_ctx;
+       const char *attrs[] = { "objectGUID", NULL };
+       int ret;
+       struct ldb_result *res;
+       struct GUID *ntds_guid;
+       
+       /* see if we have a cached copy */
+       ntds_guid = (struct GUID *)ldb_get_opaque(ldb, "cache.ntds_guid");
+       if (ntds_guid) {
+               return ntds_guid;
+       }
+
+       tmp_ctx = talloc_new(ldb);
+       if (tmp_ctx == NULL) {
+               goto failed;
+       }
+
+       ret = ldb_search(ldb, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, NULL, attrs, &res);
+       if (ret) {
+               goto failed;
+       }
+       talloc_steal(tmp_ctx, res);
+
+       if (res->count != 1) {
+               goto failed;
+       }
+
+       ntds_guid = talloc(tmp_ctx, struct GUID);
+       if (!ntds_guid) {
+               goto failed;
+       }
+
+       *ntds_guid = samdb_result_guid(res->msgs[0], "objectGUID");
+
+       /* cache the domain_sid in the ldb */
+       if (ldb_set_opaque(ldb, "cache.ntds_guid", ntds_guid) != LDB_SUCCESS) {
+               goto failed;
+       }
+
+       talloc_steal(ldb, ntds_guid);
+       talloc_free(tmp_ctx);
+
+       return ntds_guid;
+
+failed:
+       DEBUG(1,("Failed to find our own NTDS Settings objectGUID in the ldb!\n"));
        talloc_free(tmp_ctx);
        return NULL;
 }
+
+bool samdb_set_ntds_objectGUID(struct ldb_context *ldb, const struct GUID *ntds_guid_in)
+{
+       TALLOC_CTX *tmp_ctx;
+       struct GUID *ntds_guid_new;
+       struct GUID *ntds_guid_old;
+       
+       /* see if we have a cached copy */
+       ntds_guid_old = (struct GUID *)ldb_get_opaque(ldb, "cache.ntds_guid");
+
+       tmp_ctx = talloc_new(ldb);
+       if (tmp_ctx == NULL) {
+               goto failed;
+       }
+
+       ntds_guid_new = talloc(tmp_ctx, struct GUID);
+       if (!ntds_guid_new) {
+               goto failed;
+       }
+
+       *ntds_guid_new = *ntds_guid_in;
+
+       /* cache the domain_sid in the ldb */
+       if (ldb_set_opaque(ldb, "cache.ntds_guid", ntds_guid_new) != LDB_SUCCESS) {
+               goto failed;
+       }
+
+       talloc_steal(ldb, ntds_guid_new);
+       talloc_free(tmp_ctx);
+       talloc_free(ntds_guid_old);
+
+       return true;
+
+failed:
+       DEBUG(1,("Failed to set our own cached invocationId in the ldb!\n"));
+       talloc_free(tmp_ctx);
+       return false;
+}
+
+/*
+  work out the server dn for the current open ldb
+*/
+struct ldb_dn *samdb_server_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
+{
+       return ldb_dn_get_parent(mem_ctx, samdb_ntds_settings_dn(ldb));
+}
+
+/*
+  work out the server dn for the current open ldb
+*/
+struct ldb_dn *samdb_server_site_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
+{
+       struct ldb_dn *server_dn;
+       struct ldb_dn *server_site_dn;
+
+       server_dn = samdb_server_dn(ldb, mem_ctx);
+       if (!server_dn) return NULL;
+
+       server_site_dn = ldb_dn_get_parent(mem_ctx, server_dn);
+
+       talloc_free(server_dn);
+       return server_site_dn;
+}
+
+/*
+  work out if we are the PDC for the domain of the current open ldb
+*/
+BOOL samdb_is_pdc(struct ldb_context *ldb)
+{
+       const char *dom_attrs[] = { "fSMORoleOwner", NULL };
+       int ret;
+       struct ldb_result *dom_res;
+       TALLOC_CTX *tmp_ctx;
+       BOOL is_pdc;
+       struct ldb_dn *pdc;
+
+       tmp_ctx = talloc_new(ldb);
+       if (tmp_ctx == NULL) {
+               DEBUG(1, ("talloc_new failed in samdb_is_pdc"));
+               return False;
+       }
+
+       ret = ldb_search(ldb, ldb_get_default_basedn(ldb), LDB_SCOPE_BASE, NULL, dom_attrs, &dom_res);
+       if (ret) {
+               DEBUG(1,("Searching for fSMORoleOwner in %s failed: %s\n", 
+                        ldb_dn_get_linearized(ldb_get_default_basedn(ldb)), 
+                        ldb_errstring(ldb)));
+               goto failed;
+       }
+       talloc_steal(tmp_ctx, dom_res);
+       if (dom_res->count != 1) {
+               goto failed;
+       }
+
+       pdc = ldb_msg_find_attr_as_dn(ldb, tmp_ctx, dom_res->msgs[0], "fSMORoleOwner");
+
+       if (ldb_dn_compare(samdb_ntds_settings_dn(ldb), pdc) == 0) {
+               is_pdc = True;
+       } else {
+               is_pdc = False;
+       }
+
+       talloc_free(tmp_ctx);
+
+       return is_pdc;
+
+failed:
+       DEBUG(1,("Failed to find if we are the PDC for this ldb\n"));
+       talloc_free(tmp_ctx);
+       return False;
+}
+
+
+/* Find a domain object in the parents of a particular DN.  */
+struct ldb_dn *samdb_search_for_parent_domain(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, struct ldb_dn *dn)
+{
+       TALLOC_CTX *local_ctx;
+       struct ldb_dn *sdn = dn;
+       struct ldb_result *res = NULL;
+       int ret = 0;
+       const char *attrs[] = { NULL };
+
+       local_ctx = talloc_new(mem_ctx);
+       if (local_ctx == NULL) return NULL;
+       
+       while ((sdn = ldb_dn_get_parent(local_ctx, sdn))) {
+               ret = ldb_search(ldb, sdn, LDB_SCOPE_BASE, 
+                                "(|(objectClass=domain)(objectClass=builtinDomain))", attrs, &res);
+               if (ret == LDB_SUCCESS) {
+                       talloc_steal(local_ctx, res);
+                       if (res->count == 1) {
+                               break;
+                       }
+               }
+       }
+
+       if (ret != LDB_SUCCESS || res->count != 1) {
+               talloc_free(local_ctx);
+               return NULL;
+       }
+
+       talloc_steal(mem_ctx, sdn);
+       talloc_free(local_ctx);
+
+       return sdn;
+}
+
+/*
+  check that a password is sufficiently complex
+*/
+static BOOL samdb_password_complexity_ok(const char *pass)
+{
+       return check_password_quality(pass);
+}
+
+
+
+/*
+  set the user password using plaintext, obeying any user or domain
+  password restrictions
+
+  note that this function doesn't actually store the result in the
+  database, it just fills in the "mod" structure with ldb modify
+  elements to setup the correct change when samdb_replace() is
+  called. This allows the caller to combine the change with other
+  changes (as is needed by some of the set user info levels)
+
+  The caller should probably have a transaction wrapping this
+*/
+_PUBLIC_ NTSTATUS samdb_set_password(struct ldb_context *ctx, TALLOC_CTX *mem_ctx,
+                           struct ldb_dn *user_dn,
+                           struct ldb_dn *domain_dn,
+                           struct ldb_message *mod,
+                           const char *new_pass,
+                           struct samr_Password *lmNewHash, 
+                           struct samr_Password *ntNewHash,
+                           BOOL user_change,
+                           enum samr_RejectReason *reject_reason,
+                           struct samr_DomInfo1 **_dominfo)
+{
+       const char * const user_attrs[] = { "userAccountControl", "lmPwdHistory", 
+                                           "ntPwdHistory", 
+                                           "dBCSPwd", "unicodePwd", 
+                                           "objectSid", 
+                                           "pwdLastSet", NULL };
+       const char * const domain_attrs[] = { "pwdProperties", "pwdHistoryLength", 
+                                             "maxPwdAge", "minPwdAge", 
+                                             "minPwdLength", NULL };
+       NTTIME pwdLastSet;
+       int64_t minPwdAge;
+       uint_t minPwdLength, pwdProperties, pwdHistoryLength;
+       uint_t userAccountControl;
+       struct samr_Password *sambaLMPwdHistory, *sambaNTPwdHistory, *lmPwdHash, *ntPwdHash;
+       struct samr_Password local_lmNewHash, local_ntNewHash;
+       int sambaLMPwdHistory_len, sambaNTPwdHistory_len;
+       struct dom_sid *domain_sid;
+       struct ldb_message **res;
+       BOOL restrictions;
+       int count;
+       time_t now = time(NULL);
+       NTTIME now_nt;
+       int i;
+
+       /* we need to know the time to compute password age */
+       unix_to_nt_time(&now_nt, now);
+
+       /* pull all the user parameters */
+       count = gendb_search_dn(ctx, mem_ctx, user_dn, &res, user_attrs);
+       if (count != 1) {
+               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       }
+       userAccountControl = samdb_result_uint(res[0],   "userAccountControl", 0);
+       sambaLMPwdHistory_len =   samdb_result_hashes(mem_ctx, res[0], 
+                                                "lmPwdHistory", &sambaLMPwdHistory);
+       sambaNTPwdHistory_len =   samdb_result_hashes(mem_ctx, res[0], 
+                                                "ntPwdHistory", &sambaNTPwdHistory);
+       lmPwdHash =          samdb_result_hash(mem_ctx, res[0],   "dBCSPwd");
+       ntPwdHash =          samdb_result_hash(mem_ctx, res[0],   "unicodePwd");
+       pwdLastSet =         samdb_result_uint64(res[0], "pwdLastSet", 0);
+
+       /* Only non-trust accounts have restrictions (possibly this
+        * test is the wrong way around, but I like to be restrictive
+        * if possible */
+       restrictions = !(userAccountControl & (UF_INTERDOMAIN_TRUST_ACCOUNT
+                                              |UF_WORKSTATION_TRUST_ACCOUNT
+                                              |UF_SERVER_TRUST_ACCOUNT)); 
+
+       if (domain_dn) {
+               /* pull the domain parameters */
+               count = gendb_search_dn(ctx, mem_ctx, domain_dn, &res, domain_attrs);
+               if (count != 1) {
+                       DEBUG(2, ("samdb_set_password: Domain DN %s is invalid, for user %s\n", 
+                                 ldb_dn_get_linearized(domain_dn),
+                                 ldb_dn_get_linearized(user_dn)));
+                       return NT_STATUS_NO_SUCH_DOMAIN;
+               }
+       } else {
+               /* work out the domain sid, and pull the domain from there */
+               domain_sid =         samdb_result_sid_prefix(mem_ctx, res[0], "objectSid");
+               if (domain_sid == NULL) {
+                       return NT_STATUS_INTERNAL_DB_CORRUPTION;
+               }
+
+               count = gendb_search(ctx, mem_ctx, NULL, &res, domain_attrs, 
+                                    "(objectSid=%s)", 
+                                    ldap_encode_ndr_dom_sid(mem_ctx, domain_sid));
+               if (count != 1) {
+                       DEBUG(2, ("samdb_set_password: Could not find domain to match SID: %s, for user %s\n", 
+                                 dom_sid_string(mem_ctx, domain_sid),
+                                 ldb_dn_get_linearized(user_dn)));
+                       return NT_STATUS_NO_SUCH_DOMAIN;
+               }
+       }
+
+       pwdProperties =    samdb_result_uint(res[0],   "pwdProperties", 0);
+       pwdHistoryLength = samdb_result_uint(res[0],   "pwdHistoryLength", 0);
+       minPwdLength =     samdb_result_uint(res[0],   "minPwdLength", 0);
+       minPwdAge =        samdb_result_int64(res[0],  "minPwdAge", 0);
+
+       if (_dominfo) {
+               struct samr_DomInfo1 *dominfo;
+               /* on failure we need to fill in the reject reasons */
+               dominfo = talloc(mem_ctx, struct samr_DomInfo1);
+               if (dominfo == NULL) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+               dominfo->min_password_length     = minPwdLength;
+               dominfo->password_properties     = pwdProperties;
+               dominfo->password_history_length = pwdHistoryLength;
+               dominfo->max_password_age        = minPwdAge;
+               dominfo->min_password_age        = minPwdAge;
+               *_dominfo = dominfo;
+       }
+
+       if (restrictions && new_pass) {
+
+               /* check the various password restrictions */
+               if (restrictions && minPwdLength > strlen_m(new_pass)) {
+                       if (reject_reason) {
+                               *reject_reason = SAMR_REJECT_TOO_SHORT;
+                       }
+                       return NT_STATUS_PASSWORD_RESTRICTION;
+               }
+               
+               /* possibly check password complexity */
+               if (restrictions && pwdProperties & DOMAIN_PASSWORD_COMPLEX &&
+                   !samdb_password_complexity_ok(new_pass)) {
+                       if (reject_reason) {
+                               *reject_reason = SAMR_REJECT_COMPLEXITY;
+                       }
+                       return NT_STATUS_PASSWORD_RESTRICTION;
+               }
+               
+               /* compute the new nt and lm hashes */
+               if (E_deshash(new_pass, local_lmNewHash.hash)) {
+                       lmNewHash = &local_lmNewHash;
+               }
+               if (!E_md4hash(new_pass, local_ntNewHash.hash)) {
+                       /* If we can't convert this password to UCS2, then we should not accept it */
+                       if (reject_reason) {
+                               *reject_reason = SAMR_REJECT_OTHER;
+                       }
+                       return NT_STATUS_PASSWORD_RESTRICTION;
+               }
+               ntNewHash = &local_ntNewHash;
+       }
+
+       if (user_change) {
+               /* are all password changes disallowed? */
+               if (pwdProperties & DOMAIN_REFUSE_PASSWORD_CHANGE) {
+                       if (reject_reason) {
+                               *reject_reason = SAMR_REJECT_OTHER;
+                       }
+                       return NT_STATUS_PASSWORD_RESTRICTION;
+               }
+               
+               /* can this user change password? */
+               if (userAccountControl & UF_PASSWD_CANT_CHANGE) {
+                       if (reject_reason) {
+                               *reject_reason = SAMR_REJECT_OTHER;
+                       }
+                       return NT_STATUS_PASSWORD_RESTRICTION;
+               }
+               
+               /* yes, this is a minus. The ages are in negative 100nsec units! */
+               if (pwdLastSet - minPwdAge > now_nt) {
+                       if (reject_reason) {
+                               *reject_reason = SAMR_REJECT_OTHER;
+                       }
+                       return NT_STATUS_PASSWORD_RESTRICTION;
+               }
+
+               /* check the immediately past password */
+               if (pwdHistoryLength > 0) {
+                       if (lmNewHash && lmPwdHash && memcmp(lmNewHash->hash, lmPwdHash->hash, 16) == 0) {
+                               if (reject_reason) {
+                                       *reject_reason = SAMR_REJECT_IN_HISTORY;
+                               }
+                               return NT_STATUS_PASSWORD_RESTRICTION;
+                       }
+                       if (ntNewHash && ntPwdHash && memcmp(ntNewHash->hash, ntPwdHash->hash, 16) == 0) {
+                               if (reject_reason) {
+                                       *reject_reason = SAMR_REJECT_IN_HISTORY;
+                               }
+                               return NT_STATUS_PASSWORD_RESTRICTION;
+                       }
+               }
+               
+               /* check the password history */
+               sambaLMPwdHistory_len = MIN(sambaLMPwdHistory_len, pwdHistoryLength);
+               sambaNTPwdHistory_len = MIN(sambaNTPwdHistory_len, pwdHistoryLength);
+               
+               for (i=0; lmNewHash && i<sambaLMPwdHistory_len;i++) {
+                       if (memcmp(lmNewHash->hash, sambaLMPwdHistory[i].hash, 16) == 0) {
+                               if (reject_reason) {
+                                       *reject_reason = SAMR_REJECT_IN_HISTORY;
+                               }
+                               return NT_STATUS_PASSWORD_RESTRICTION;
+                       }
+               }
+               for (i=0; ntNewHash && i<sambaNTPwdHistory_len;i++) {
+                       if (memcmp(ntNewHash->hash, sambaNTPwdHistory[i].hash, 16) == 0) {
+                               if (reject_reason) {
+                                       *reject_reason = SAMR_REJECT_IN_HISTORY;
+                               }
+                               return NT_STATUS_PASSWORD_RESTRICTION;
+                       }
+               }
+       }
+
+#define CHECK_RET(x) do { if (x != 0) return NT_STATUS_NO_MEMORY; } while(0)
+
+       /* the password is acceptable. Start forming the new fields */
+       if (new_pass) {
+               /* if we know the cleartext, then only set it.
+                * Modules in ldb will set all the appropriate
+                * hashes */
+               CHECK_RET(samdb_msg_add_string(ctx, mem_ctx, mod, 
+                                              "sambaPassword", new_pass));
+       } else {
+               /* We don't have the cleartext, so delete the old one
+                * and set what we have of the hashes */
+               CHECK_RET(samdb_msg_add_delete(ctx, mem_ctx, mod, "sambaPassword"));
+
+               if (lmNewHash) {
+                       CHECK_RET(samdb_msg_add_hash(ctx, mem_ctx, mod, "dBCSPwd", lmNewHash));
+               } else {
+                       CHECK_RET(samdb_msg_add_delete(ctx, mem_ctx, mod, "dBCSPwd"));
+               }
+               
+               if (ntNewHash) {
+                       CHECK_RET(samdb_msg_add_hash(ctx, mem_ctx, mod, "unicodePwd", ntNewHash));
+               } else {
+                       CHECK_RET(samdb_msg_add_delete(ctx, mem_ctx, mod, "unicodePwd"));
+               }
+       }
+
+       return NT_STATUS_OK;
+}
+
+
+/*
+  set the user password using plaintext, obeying any user or domain
+  password restrictions
+
+  This wrapper function takes a SID as input, rather than a user DN,
+  and actually performs the password change
+
+*/
+_PUBLIC_ NTSTATUS samdb_set_password_sid(struct ldb_context *ctx, TALLOC_CTX *mem_ctx,
+                               const struct dom_sid *user_sid,
+                               const char *new_pass,
+                               struct samr_Password *lmNewHash, 
+                               struct samr_Password *ntNewHash,
+                               BOOL user_change,
+                               enum samr_RejectReason *reject_reason,
+                               struct samr_DomInfo1 **_dominfo) 
+{
+       NTSTATUS nt_status;
+       struct ldb_dn *user_dn;
+       struct ldb_message *msg;
+       int ret;
+
+       ret = ldb_transaction_start(ctx);
+       if (ret) {
+               DEBUG(1, ("Failed to start transaction: %s\n", ldb_errstring(ctx)));
+               return NT_STATUS_TRANSACTION_ABORTED;
+       }
+
+       user_dn = samdb_search_dn(ctx, mem_ctx, NULL, 
+                                 "(&(objectSid=%s)(objectClass=user))", 
+                                 ldap_encode_ndr_dom_sid(mem_ctx, user_sid));
+       if (!user_dn) {
+               ldb_transaction_cancel(ctx);
+               DEBUG(3, ("samdb_set_password_sid: SID %s not found in samdb, returning NO_SUCH_USER\n",
+                         dom_sid_string(mem_ctx, user_sid)));
+               return NT_STATUS_NO_SUCH_USER;
+       }
+
+       msg = ldb_msg_new(mem_ctx);
+       if (msg == NULL) {
+               ldb_transaction_cancel(ctx);
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       msg->dn = ldb_dn_copy(msg, user_dn);
+       if (!msg->dn) {
+               ldb_transaction_cancel(ctx);
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       nt_status = samdb_set_password(ctx, mem_ctx,
+                                      user_dn, NULL,
+                                      msg, new_pass, 
+                                      lmNewHash, ntNewHash,
+                                      user_change, /* This is a password set, not change */
+                                      reject_reason, _dominfo);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               ldb_transaction_cancel(ctx);
+               return nt_status;
+       }
+       
+       /* modify the samdb record */
+       ret = samdb_replace(ctx, mem_ctx, msg);
+       if (ret != 0) {
+               ldb_transaction_cancel(ctx);
+               return NT_STATUS_ACCESS_DENIED;
+       }
+
+       ret = ldb_transaction_commit(ctx);
+       if (ret != 0) {
+               DEBUG(0,("Failed to commit transaction to change password on %s: %s\n",
+                        ldb_dn_get_linearized(msg->dn),
+                        ldb_errstring(ctx)));
+               return NT_STATUS_TRANSACTION_ABORTED;
+       }
+       return NT_STATUS_OK;
+}
+
+/****************************************************************************
+ Create the SID list for this user.
+****************************************************************************/
+NTSTATUS security_token_create(TALLOC_CTX *mem_ctx, 
+                              struct dom_sid *user_sid,
+                              struct dom_sid *group_sid, 
+                              int n_groupSIDs,
+                              struct dom_sid **groupSIDs, 
+                              BOOL is_authenticated,
+                              struct security_token **token)
+{
+       struct security_token *ptoken;
+       int i;
+       NTSTATUS status;
+
+       ptoken = security_token_initialise(mem_ctx);
+       NT_STATUS_HAVE_NO_MEMORY(ptoken);
+
+       ptoken->sids = talloc_array(ptoken, struct dom_sid *, n_groupSIDs + 5);
+       NT_STATUS_HAVE_NO_MEMORY(ptoken->sids);
+
+       ptoken->user_sid = talloc_reference(ptoken, user_sid);
+       ptoken->group_sid = talloc_reference(ptoken, group_sid);
+       ptoken->privilege_mask = 0;
+
+       ptoken->sids[0] = ptoken->user_sid;
+       ptoken->sids[1] = ptoken->group_sid;
+
+       /*
+        * Finally add the "standard" SIDs.
+        * The only difference between guest and "anonymous"
+        * is the addition of Authenticated_Users.
+        */
+       ptoken->sids[2] = dom_sid_parse_talloc(ptoken->sids, SID_WORLD);
+       NT_STATUS_HAVE_NO_MEMORY(ptoken->sids[2]);
+       ptoken->sids[3] = dom_sid_parse_talloc(ptoken->sids, SID_NT_NETWORK);
+       NT_STATUS_HAVE_NO_MEMORY(ptoken->sids[3]);
+       ptoken->num_sids = 4;
+
+       if (is_authenticated) {
+               ptoken->sids[4] = dom_sid_parse_talloc(ptoken->sids, SID_NT_AUTHENTICATED_USERS);
+               NT_STATUS_HAVE_NO_MEMORY(ptoken->sids[4]);
+               ptoken->num_sids++;
+       }
+
+       for (i = 0; i < n_groupSIDs; i++) {
+               size_t check_sid_idx;
+               for (check_sid_idx = 1; 
+                    check_sid_idx < ptoken->num_sids; 
+                    check_sid_idx++) {
+                       if (dom_sid_equal(ptoken->sids[check_sid_idx], groupSIDs[i])) {
+                               break;
+                       }
+               }
+
+               if (check_sid_idx == ptoken->num_sids) {
+                       ptoken->sids[ptoken->num_sids++] = talloc_reference(ptoken->sids, groupSIDs[i]);
+               }
+       }
+
+       /* setup the privilege mask for this token */
+       status = samdb_privilege_setup(ptoken);
+       if (!NT_STATUS_IS_OK(status)) {
+               talloc_free(ptoken);
+               return status;
+       }
+
+       security_token_debug(10, ptoken);
+
+       *token = ptoken;
+
+       return NT_STATUS_OK;
+}
+
+
+NTSTATUS samdb_create_foreign_security_principal(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, 
+                                                struct dom_sid *sid, struct ldb_dn **ret_dn) 
+{
+       struct ldb_message *msg;
+       struct ldb_dn *basedn;
+       const char *sidstr;
+       int ret;
+       
+       sidstr = dom_sid_string(mem_ctx, sid);
+       NT_STATUS_HAVE_NO_MEMORY(sidstr);
+       
+       /* We might have to create a ForeignSecurityPrincipal, even if this user
+        * is in our own domain */
+       
+       msg = ldb_msg_new(mem_ctx);
+       if (msg == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
+       
+       /* TODO: Hmmm. This feels wrong. How do I find the base dn to
+        * put the ForeignSecurityPrincipals? d_state->domain_dn does
+        * not work, this is wrong for the Builtin domain, there's no
+        * cn=For...,cn=Builtin,dc={BASEDN}.  -- vl
+        */
+       
+       basedn = samdb_search_dn(sam_ctx, mem_ctx, NULL,
+                                "(&(objectClass=container)(cn=ForeignSecurityPrincipals))");
+       
+       if (basedn == NULL) {
+               DEBUG(0, ("Failed to find DN for "
+                         "ForeignSecurityPrincipal container\n"));
+               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       }
+       
+       /* add core elements to the ldb_message for the alias */
+       msg->dn = ldb_dn_copy(mem_ctx, basedn);
+       if ( ! ldb_dn_add_child_fmt(msg->dn, "CN=%s", sidstr))
+               return NT_STATUS_NO_MEMORY;
+       
+       samdb_msg_add_string(sam_ctx, mem_ctx, msg,
+                            "objectClass",
+                            "foreignSecurityPrincipal");
+       
+       /* create the alias */
+       ret = samdb_add(sam_ctx, mem_ctx, msg);
+       if (ret != 0) {
+               DEBUG(0,("Failed to create foreignSecurityPrincipal "
+                        "record %s: %s\n", 
+                        ldb_dn_get_linearized(msg->dn),
+                        ldb_errstring(sam_ctx)));
+               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       }
+       *ret_dn = msg->dn;
+       return NT_STATUS_OK;
+}
+
+
+/*
+  Find the DN of a domain, assuming it to be a dotted.dns name
+*/
+
+struct ldb_dn *samdb_dns_domain_to_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const char *dns_domain) 
+{
+       int i;
+       TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
+       const char *binary_encoded;
+       const char **split_realm;
+       struct ldb_dn *dn;
+       
+       if (!tmp_ctx) {
+               return NULL;
+       }
+       
+       split_realm = str_list_make(tmp_ctx, dns_domain, ".");
+       if (!split_realm) {
+               talloc_free(tmp_ctx);
+               return NULL;
+       }
+       dn = ldb_dn_new(mem_ctx, ldb, NULL);
+       for (i=0; split_realm[i]; i++) {
+               binary_encoded = ldb_binary_encode_string(tmp_ctx, split_realm[i]);
+               if (!ldb_dn_add_base_fmt(dn, "dc=%s", binary_encoded)) {
+                       DEBUG(2, ("Failed to add dc=%s element to DN %s\n",
+                                 binary_encoded, ldb_dn_get_linearized(dn)));
+                       talloc_free(tmp_ctx);
+                       return NULL;
+               }
+       }
+       if (!ldb_dn_validate(dn)) {
+               DEBUG(2, ("Failed to validated DN %s\n",
+                         ldb_dn_get_linearized(dn)));
+               return NULL;
+       }
+       return dn;
+}
+/*
+  Find the DN of a domain, be it the netbios or DNS name 
+*/
+
+struct ldb_dn *samdb_domain_to_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, 
+                                 const char *domain_name) 
+{
+       const char * const domain_ref_attrs[] = {
+               "ncName", NULL
+       };
+       const char * const domain_ref2_attrs[] = {
+               NULL
+       };
+       struct ldb_result *res_domain_ref;
+       char *escaped_domain = ldb_binary_encode_string(mem_ctx, domain_name);
+       /* find the domain's DN */
+       int ret_domain = ldb_search_exp_fmt(ldb, mem_ctx, 
+                                           &res_domain_ref, 
+                                           samdb_partitions_dn(ldb, mem_ctx), 
+                                           LDB_SCOPE_ONELEVEL, 
+                                           domain_ref_attrs,
+                                           "(&(nETBIOSName=%s)(objectclass=crossRef))", 
+                                           escaped_domain);
+       if (ret_domain != 0) {
+               return NULL;
+       }
+       
+       if (res_domain_ref->count == 0) {
+               ret_domain = ldb_search_exp_fmt(ldb, mem_ctx, 
+                                               &res_domain_ref, 
+                                               samdb_dns_domain_to_dn(ldb, mem_ctx, domain_name),
+                                               LDB_SCOPE_BASE,
+                                               domain_ref2_attrs,
+                                               "(objectclass=domain)");
+               if (ret_domain != 0) {
+                       return NULL;
+               }
+       
+               if (res_domain_ref->count == 1) {
+                       return res_domain_ref->msgs[0]->dn;
+               }
+               return NULL;
+       }
+       
+       if (res_domain_ref->count > 1) {
+               DEBUG(0,("Found %d records matching domain [%s]\n", 
+                        ret_domain, domain_name));
+               return NULL;
+       }
+       
+       return samdb_result_dn(ldb, mem_ctx, res_domain_ref->msgs[0], "nCName", NULL);
+
+}