s4-dsdb Add ability to force a particular SID in the upgrade case
[idra/samba.git] / source4 / rpc_server / samr / dcesrv_samr.c
index 03acf97cab700cc817bad19bc187a7b8a3172771..7d062ffff411a66286fdfb4e2901ae830761c6e3 100644 (file)
@@ -1,4 +1,4 @@
-/* 
+/*
    Unix SMB/CIFS implementation.
 
    endpoint server for the samr pipe
@@ -6,17 +6,18 @@
    Copyright (C) Andrew Tridgell 2004
    Copyright (C) Volker Lendecke 2004
    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
-   
+   Copyright (C) Matthias Dieter Wallnöfer 2009
+
    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 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    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, see <http://www.gnu.org/licenses/>.
 */
 #include "rpc_server/common/common.h"
 #include "rpc_server/samr/dcesrv_samr.h"
 #include "system/time.h"
-#include "lib/ldb/include/ldb.h"
-#include "lib/ldb/include/ldb_errors.h"
-#include "dsdb/common/flags.h"
+#include <ldb.h>
+#include <ldb_errors.h>
+#include "../libds/common/flags.h"
 #include "dsdb/samdb/samdb.h"
+#include "dsdb/common/util.h"
 #include "libcli/ldap/ldap_ndr.h"
 #include "libcli/security/security.h"
 #include "rpc_server/samr/proto.h"
 #include "../lib/util/util_ldb.h"
 #include "param/param.h"
+#include "lib/util/tsort.h"
+#include "libds/common/flag_mapping.h"
 
 /* these query macros make samr_Query[User|Group|Alias]Info a bit easier to read */
 
 #define QUERY_STRING(msg, field, attr) \
-       info->field.string = samdb_result_string(msg, attr, "");
+       info->field.string = ldb_msg_find_attr_as_string(msg, attr, "");
 #define QUERY_UINT(msg, field, attr) \
-       info->field = samdb_result_uint(msg, attr, 0);
+       info->field = ldb_msg_find_attr_as_uint(msg, attr, 0);
 #define QUERY_RID(msg, field, attr) \
        info->field = samdb_result_rid_from_sid(mem_ctx, msg, attr, 0);
 #define QUERY_UINT64(msg, field, attr) \
-       info->field = samdb_result_uint64(msg, attr, 0);
+       info->field = ldb_msg_find_attr_as_uint64(msg, attr, 0);
 #define QUERY_APASSC(msg, field, attr) \
        info->field = samdb_result_allow_password_change(sam_ctx, mem_ctx, \
                                                         a_state->domain_state->domain_dn, msg, attr);
        struct ldb_message_element *set_el;                             \
        if (r->in.info->field.string == NULL) return NT_STATUS_INVALID_PARAMETER; \
         if (r->in.info->field.string[0] == '\0') {                     \
-               if (ldb_msg_add_empty(msg, attr, LDB_FLAG_MOD_DELETE, NULL)) { \
+               if (ldb_msg_add_empty(msg, attr, LDB_FLAG_MOD_DELETE, NULL) != LDB_SUCCESS) { \
                        return NT_STATUS_NO_MEMORY;                     \
                }                                                       \
        }                                                               \
-        if (ldb_msg_add_string(msg, attr, r->in.info->field.string) != 0) { \
+        if (ldb_msg_add_string(msg, attr, r->in.info->field.string) != LDB_SUCCESS) { \
                return NT_STATUS_NO_MEMORY;                             \
        }                                                               \
         set_el = ldb_msg_find_element(msg, attr);                      \
 
 #define SET_UINT(msg, field, attr) do {                                        \
        struct ldb_message_element *set_el;                             \
-       if (samdb_msg_add_uint(sam_ctx, mem_ctx, msg, attr, r->in.info->field) != 0) { \
+       if (samdb_msg_add_uint(sam_ctx, mem_ctx, msg, attr, r->in.info->field) != LDB_SUCCESS) { \
                return NT_STATUS_NO_MEMORY;                             \
        }                                                               \
         set_el = ldb_msg_find_element(msg, attr);                      \
        set_el->flags = LDB_FLAG_MOD_REPLACE;                           \
-} while (0)                                                            
-                                                                       
+} while (0)
+
 #define SET_INT64(msg, field, attr) do {                               \
        struct ldb_message_element *set_el;                             \
-       if (samdb_msg_add_int64(sam_ctx, mem_ctx, msg, attr, r->in.info->field) != 0) { \
+       if (samdb_msg_add_int64(sam_ctx, mem_ctx, msg, attr, r->in.info->field) != LDB_SUCCESS) { \
                return NT_STATUS_NO_MEMORY;                             \
        }                                                               \
         set_el = ldb_msg_find_element(msg, attr);                      \
        set_el->flags = LDB_FLAG_MOD_REPLACE;                           \
-} while (0)                                                            
-                                                                       
+} while (0)
+
 #define SET_UINT64(msg, field, attr) do {                              \
        struct ldb_message_element *set_el;                             \
-       if (samdb_msg_add_uint64(sam_ctx, mem_ctx, msg, attr, r->in.info->field) != 0) { \
+       if (samdb_msg_add_uint64(sam_ctx, mem_ctx, msg, attr, r->in.info->field) != LDB_SUCCESS) { \
                return NT_STATUS_NO_MEMORY;                             \
        }                                                               \
         set_el = ldb_msg_find_element(msg, attr);                      \
        set_el->flags = LDB_FLAG_MOD_REPLACE;                           \
-} while (0)                                                            
+} while (0)
 
 #define CHECK_FOR_MULTIPLES(value, flag, poss_flags)   \
        do { \
                        return NT_STATUS_INVALID_PARAMETER;             \
                }                                                       \
        } while (0)                                                     \
-       
-/* Set account flags, discarding flags that cannot be set with SAMR */                                                         
+
+/* Set account flags, discarding flags that cannot be set with SAMR */
 #define SET_AFLAGS(msg, field, attr) do {                              \
        struct ldb_message_element *set_el;                             \
        if ((r->in.info->field & (ACB_NORMAL | ACB_DOMTRUST | ACB_WSTRUST | ACB_SVRTRUST)) == 0) { \
        }                                                               \
         set_el = ldb_msg_find_element(msg, attr);                      \
        set_el->flags = LDB_FLAG_MOD_REPLACE;                           \
-} while (0)                                                            
-                                                                       
+} while (0)
+
 #define SET_LHOURS(msg, field, attr) do {                              \
        struct ldb_message_element *set_el;                             \
-       if (samdb_msg_add_logon_hours(sam_ctx, mem_ctx, msg, attr, &r->in.info->field) != 0) { \
+       if (samdb_msg_add_logon_hours(sam_ctx, mem_ctx, msg, attr, &r->in.info->field) != LDB_SUCCESS) { \
                return NT_STATUS_NO_MEMORY;                             \
        }                                                               \
         set_el = ldb_msg_find_element(msg, attr);                      \
 #define SET_PARAMETERS(msg, field, attr) do {                          \
        struct ldb_message_element *set_el;                             \
        if (r->in.info->field.length != 0) {                            \
-               if (samdb_msg_add_parameters(sam_ctx, mem_ctx, msg, attr, &r->in.info->field) != 0) { \
+               if (samdb_msg_add_parameters(sam_ctx, mem_ctx, msg, attr, &r->in.info->field) != LDB_SUCCESS) { \
                        return NT_STATUS_NO_MEMORY;                     \
                }                                                       \
                set_el = ldb_msg_find_element(msg, attr);               \
 
 
 
-/* 
-  samr_Connect 
+/*
+  samr_Connect
 
   create a connection to the SAM database
 */
@@ -164,13 +168,13 @@ static NTSTATUS dcesrv_samr_Connect(struct dcesrv_call_state *dce_call, TALLOC_C
 
        ZERO_STRUCTP(r->out.connect_handle);
 
-       c_state = talloc(dce_call->conn, struct samr_connect_state);
+       c_state = talloc(mem_ctx, struct samr_connect_state);
        if (!c_state) {
                return NT_STATUS_NO_MEMORY;
        }
 
        /* make sure the sam database is accessible */
-       c_state->sam_ctx = samdb_connect(c_state, dce_call->event_ctx, dce_call->conn->dce_ctx->lp_ctx, dce_call->conn->auth_state.session_info); 
+       c_state->sam_ctx = samdb_connect(c_state, dce_call->event_ctx, dce_call->conn->dce_ctx->lp_ctx, dce_call->conn->auth_state.session_info, 0);
        if (c_state->sam_ctx == NULL) {
                talloc_free(c_state);
                return NT_STATUS_INVALID_SYSTEM_SERVICE;
@@ -192,8 +196,8 @@ static NTSTATUS dcesrv_samr_Connect(struct dcesrv_call_state *dce_call, TALLOC_C
 }
 
 
-/* 
-  samr_Close 
+/*
+  samr_Close
 */
 static NTSTATUS dcesrv_samr_Close(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                           struct samr_Close *r)
@@ -212,8 +216,8 @@ static NTSTATUS dcesrv_samr_Close(struct dcesrv_call_state *dce_call, TALLOC_CTX
 }
 
 
-/* 
-  samr_SetSecurity 
+/*
+  samr_SetSecurity
 */
 static NTSTATUS dcesrv_samr_SetSecurity(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                 struct samr_SetSecurity *r)
@@ -222,8 +226,8 @@ static NTSTATUS dcesrv_samr_SetSecurity(struct dcesrv_call_state *dce_call, TALL
 }
 
 
-/* 
-  samr_QuerySecurity 
+/*
+  samr_QuerySecurity
 */
 static NTSTATUS dcesrv_samr_QuerySecurity(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                   struct samr_QuerySecurity *r)
@@ -248,8 +252,8 @@ static NTSTATUS dcesrv_samr_QuerySecurity(struct dcesrv_call_state *dce_call, TA
 }
 
 
-/* 
-  samr_Shutdown 
+/*
+  samr_Shutdown
 
   we refuse this operation completely. If a admin wants to shutdown samr
   in Samba then they should use the samba admin tools to disable the samr pipe
@@ -261,8 +265,8 @@ static NTSTATUS dcesrv_samr_Shutdown(struct dcesrv_call_state *dce_call, TALLOC_
 }
 
 
-/* 
-  samr_LookupDomain 
+/*
+  samr_LookupDomain
 
   this maps from a domain name to a SID
 */
@@ -290,9 +294,9 @@ static NTSTATUS dcesrv_samr_LookupDomain(struct dcesrv_call_state *dce_call, TAL
                ret = gendb_search(c_state->sam_ctx,
                                   mem_ctx, NULL, &dom_msgs, dom_attrs,
                                   "(objectClass=builtinDomain)");
-       } else if (strcasecmp_m(r->in.domain_name->string, lp_sam_name(dce_call->conn->dce_ctx->lp_ctx)) == 0) {
+       } else if (strcasecmp_m(r->in.domain_name->string, lpcfg_sam_name(dce_call->conn->dce_ctx->lp_ctx)) == 0) {
                ret = gendb_search_dn(c_state->sam_ctx,
-                                     mem_ctx, ldb_get_default_basedn(c_state->sam_ctx), 
+                                     mem_ctx, ldb_get_default_basedn(c_state->sam_ctx),
                                      &dom_msgs, dom_attrs);
        } else {
                return NT_STATUS_NO_SUCH_DOMAIN;
@@ -300,10 +304,10 @@ static NTSTATUS dcesrv_samr_LookupDomain(struct dcesrv_call_state *dce_call, TAL
        if (ret != 1) {
                return NT_STATUS_NO_SUCH_DOMAIN;
        }
-       
+
        sid = samdb_result_dom_sid(mem_ctx, dom_msgs[0],
                                   "objectSid");
-               
+
        if (sid == NULL) {
                return NT_STATUS_NO_SUCH_DOMAIN;
        }
@@ -314,8 +318,8 @@ static NTSTATUS dcesrv_samr_LookupDomain(struct dcesrv_call_state *dce_call, TAL
 }
 
 
-/* 
-  samr_EnumDomains 
+/*
+  samr_EnumDomains
 
   list the domains in the SAM
 */
@@ -325,7 +329,7 @@ static NTSTATUS dcesrv_samr_EnumDomains(struct dcesrv_call_state *dce_call, TALL
        struct samr_connect_state *c_state;
        struct dcesrv_handle *h;
        struct samr_SamArray *array;
-       int i, start_i;
+       uint32_t i, start_i;
 
        *r->out.resume_handle = 0;
        *r->out.sam = NULL;
@@ -348,7 +352,7 @@ static NTSTATUS dcesrv_samr_EnumDomains(struct dcesrv_call_state *dce_call, TALL
        if (array == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
-               
+
        array->count = 0;
        array->entries = NULL;
 
@@ -360,7 +364,7 @@ static NTSTATUS dcesrv_samr_EnumDomains(struct dcesrv_call_state *dce_call, TALL
        for (i=0;i<2-start_i;i++) {
                array->entries[i].idx = start_i + i;
                if (i == 0) {
-                       array->entries[i].name.string = lp_sam_name(dce_call->conn->dce_ctx->lp_ctx);
+                       array->entries[i].name.string = lpcfg_sam_name(dce_call->conn->dce_ctx->lp_ctx);
                } else {
                        array->entries[i].name.string = "BUILTIN";
                }
@@ -374,8 +378,8 @@ static NTSTATUS dcesrv_samr_EnumDomains(struct dcesrv_call_state *dce_call, TALL
 }
 
 
-/* 
-  samr_OpenDomain 
+/*
+  samr_OpenDomain
 */
 static NTSTATUS dcesrv_samr_OpenDomain(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                struct samr_OpenDomain *r)
@@ -397,7 +401,7 @@ static NTSTATUS dcesrv_samr_OpenDomain(struct dcesrv_call_state *dce_call, TALLO
                return NT_STATUS_INVALID_PARAMETER;
        }
 
-       d_state = talloc(c_state, struct samr_domain_state);
+       d_state = talloc(mem_ctx, struct samr_domain_state);
        if (!d_state) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -409,14 +413,14 @@ static NTSTATUS dcesrv_samr_OpenDomain(struct dcesrv_call_state *dce_call, TALLO
                d_state->domain_name = "BUILTIN";
        } else {
                d_state->builtin = false;
-               d_state->domain_name = lp_sam_name(dce_call->conn->dce_ctx->lp_ctx);
+               d_state->domain_name = lpcfg_sam_name(dce_call->conn->dce_ctx->lp_ctx);
        }
 
        ret = gendb_search(c_state->sam_ctx,
                           mem_ctx, ldb_get_default_basedn(c_state->sam_ctx), &dom_msgs, dom_attrs,
-                          "(objectSid=%s)", 
+                          "(objectSid=%s)",
                           ldap_encode_ndr_dom_sid(mem_ctx, r->in.sid));
-       
+
        if (ret == 0) {
                talloc_free(d_state);
                return NT_STATUS_NO_SUCH_DOMAIN;
@@ -430,7 +434,7 @@ static NTSTATUS dcesrv_samr_OpenDomain(struct dcesrv_call_state *dce_call, TALLO
        }
 
        d_state->domain_dn = talloc_steal(d_state, dom_msgs[0]->dn);
-       d_state->role = lp_server_role(dce_call->conn->dce_ctx->lp_ctx);
+       d_state->role = lpcfg_server_role(dce_call->conn->dce_ctx->lp_ctx);
        d_state->connect_state = talloc_reference(d_state, c_state);
        d_state->sam_ctx = c_state->sam_ctx;
        d_state->access_mask = r->in.access_mask;
@@ -442,7 +446,7 @@ static NTSTATUS dcesrv_samr_OpenDomain(struct dcesrv_call_state *dce_call, TALLO
                talloc_free(d_state);
                return NT_STATUS_NO_MEMORY;
        }
-       
+
        h_domain->data = talloc_steal(h_domain, d_state);
 
        *r->out.domain_handle = h_domain->wire_handle;
@@ -454,20 +458,20 @@ static NTSTATUS dcesrv_samr_OpenDomain(struct dcesrv_call_state *dce_call, TALLO
   return DomInfo1
 */
 static NTSTATUS dcesrv_samr_info_DomInfo1(struct samr_domain_state *state,
-                                  TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                  struct samr_DomInfo1 *info)
+                                         TALLOC_CTX *mem_ctx,
+                                         struct ldb_message **dom_msgs,
+                                         struct samr_DomInfo1 *info)
 {
        info->min_password_length =
-               samdb_result_uint(dom_msgs[0], "minPwdLength", 0);
+               ldb_msg_find_attr_as_uint(dom_msgs[0], "minPwdLength", 0);
        info->password_history_length =
-               samdb_result_uint(dom_msgs[0], "pwdHistoryLength", 0);
-       info->password_properties = 
-               samdb_result_uint(dom_msgs[0], "pwdProperties", 0);
-       info->max_password_age = 
-               samdb_result_int64(dom_msgs[0], "maxPwdAge", 0);
-       info->min_password_age = 
-               samdb_result_int64(dom_msgs[0], "minPwdAge", 0);
+               ldb_msg_find_attr_as_uint(dom_msgs[0], "pwdHistoryLength", 0);
+       info->password_properties =
+               ldb_msg_find_attr_as_uint(dom_msgs[0], "pwdProperties", 0);
+       info->max_password_age =
+               ldb_msg_find_attr_as_int64(dom_msgs[0], "maxPwdAge", 0);
+       info->min_password_age =
+               ldb_msg_find_attr_as_int64(dom_msgs[0], "minPwdAge", 0);
 
        return NT_STATUS_OK;
 }
@@ -475,31 +479,29 @@ static NTSTATUS dcesrv_samr_info_DomInfo1(struct samr_domain_state *state,
 /*
   return DomInfo2
 */
-static NTSTATUS dcesrv_samr_info_DomGeneralInformation(struct samr_domain_state *state, 
+static NTSTATUS dcesrv_samr_info_DomGeneralInformation(struct samr_domain_state *state,
                                                       TALLOC_CTX *mem_ctx,
                                                       struct ldb_message **dom_msgs,
                                                       struct samr_DomGeneralInformation *info)
 {
-       /* This pulls the NetBIOS name from the 
-          cn=NTDS Settings,cn=<NETBIOS name of PDC>,....
-          string */
-       info->primary.string = samdb_result_fsmo_name(state->sam_ctx, mem_ctx, dom_msgs[0], "fSMORoleOwner");
+       /* MS-SAMR 2.2.4.1 - ReplicaSourceNodeName: "domainReplica" attribute */
+       info->primary.string = ldb_msg_find_attr_as_string(dom_msgs[0],
+                                                          "domainReplica",
+                                                          "");
 
-       if (!info->primary.string) {
-               info->primary.string = lp_netbios_name(state->lp_ctx);
-       }
-
-       info->force_logoff_time = ldb_msg_find_attr_as_uint64(dom_msgs[0], "forceLogoff", 
+       info->force_logoff_time = ldb_msg_find_attr_as_uint64(dom_msgs[0], "forceLogoff",
                                                            0x8000000000000000LL);
 
-       info->oem_information.string = samdb_result_string(dom_msgs[0], "oEMInformation", NULL);
+       info->oem_information.string = ldb_msg_find_attr_as_string(dom_msgs[0],
+                                                                  "oEMInformation",
+                                                                  "");
        info->domain_name.string  = state->domain_name;
 
-       info->sequence_num = ldb_msg_find_attr_as_uint64(dom_msgs[0], "modifiedCount", 
+       info->sequence_num = ldb_msg_find_attr_as_uint64(dom_msgs[0], "modifiedCount",
                                                 0);
        switch (state->role) {
        case ROLE_DOMAIN_CONTROLLER:
-               /* This pulls the NetBIOS name from the 
+               /* This pulls the NetBIOS name from the
                   cn=NTDS Settings,cn=<NETBIOS name of PDC>,....
                   string */
                if (samdb_is_pdc(state->sam_ctx)) {
@@ -508,6 +510,9 @@ static NTSTATUS dcesrv_samr_info_DomGeneralInformation(struct samr_domain_state
                        info->role = SAMR_ROLE_DOMAIN_BDC;
                }
                break;
+       case ROLE_DOMAIN_PDC:
+               info->role = SAMR_ROLE_DOMAIN_PDC;
+               break;
        case ROLE_DOMAIN_MEMBER:
                info->role = SAMR_ROLE_DOMAIN_MEMBER;
                break;
@@ -516,15 +521,19 @@ static NTSTATUS dcesrv_samr_info_DomGeneralInformation(struct samr_domain_state
                break;
        }
 
-       /* No users in BUILTIN, and the LOCAL group types are only in builtin, and the global group type is never in BUILTIN */
-       info->num_users = samdb_search_count(state->sam_ctx, mem_ctx, state->domain_dn, 
+       info->num_users = samdb_search_count(state->sam_ctx, mem_ctx,
+                                            state->domain_dn,
                                             "(objectClass=user)");
-       info->num_groups = samdb_search_count(state->sam_ctx, mem_ctx, state->domain_dn,
-                                             "(&(objectClass=group)(sAMAccountType=%u))",
-                                             ATYPE_GLOBAL_GROUP);
-       info->num_aliases = samdb_search_count(state->sam_ctx, mem_ctx, state->domain_dn,
-                                              "(&(objectClass=group)(sAMAccountType=%u))",
-                                              ATYPE_LOCAL_GROUP);
+       info->num_groups = samdb_search_count(state->sam_ctx, mem_ctx,
+                                             state->domain_dn,
+                                             "(&(objectClass=group)(|(groupType=%d)(groupType=%d)))",
+                                             GTYPE_SECURITY_UNIVERSAL_GROUP,
+                                             GTYPE_SECURITY_GLOBAL_GROUP);
+       info->num_aliases = samdb_search_count(state->sam_ctx, mem_ctx,
+                                              state->domain_dn,
+                                              "(&(objectClass=group)(|(groupType=%d)(groupType=%d)))",
+                                              GTYPE_SECURITY_BUILTIN_LOCAL_GROUP,
+                                              GTYPE_SECURITY_DOMAIN_LOCAL_GROUP);
 
        return NT_STATUS_OK;
 }
@@ -533,11 +542,11 @@ static NTSTATUS dcesrv_samr_info_DomGeneralInformation(struct samr_domain_state
   return DomInfo3
 */
 static NTSTATUS dcesrv_samr_info_DomInfo3(struct samr_domain_state *state,
-                                  TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                  struct samr_DomInfo3 *info)
+                                         TALLOC_CTX *mem_ctx,
+                                         struct ldb_message **dom_msgs,
+                                         struct samr_DomInfo3 *info)
 {
-       info->force_logoff_time = ldb_msg_find_attr_as_uint64(dom_msgs[0], "forceLogoff", 
+       info->force_logoff_time = ldb_msg_find_attr_as_uint64(dom_msgs[0], "forceLogoff",
                                                      0x8000000000000000LL);
 
        return NT_STATUS_OK;
@@ -551,7 +560,9 @@ static NTSTATUS dcesrv_samr_info_DomOEMInformation(struct samr_domain_state *sta
                                    struct ldb_message **dom_msgs,
                                   struct samr_DomOEMInformation *info)
 {
-       info->oem_information.string = samdb_result_string(dom_msgs[0], "oEMInformation", NULL);
+       info->oem_information.string = ldb_msg_find_attr_as_string(dom_msgs[0],
+                                                                  "oEMInformation",
+                                                                  "");
 
        return NT_STATUS_OK;
 }
@@ -560,9 +571,9 @@ static NTSTATUS dcesrv_samr_info_DomOEMInformation(struct samr_domain_state *sta
   return DomInfo5
 */
 static NTSTATUS dcesrv_samr_info_DomInfo5(struct samr_domain_state *state,
-                                  TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                  struct samr_DomInfo5 *info)
+                                         TALLOC_CTX *mem_ctx,
+                                         struct ldb_message **dom_msgs,
+                                         struct samr_DomInfo5 *info)
 {
        info->domain_name.string  = state->domain_name;
 
@@ -573,19 +584,14 @@ static NTSTATUS dcesrv_samr_info_DomInfo5(struct samr_domain_state *state,
   return DomInfo6
 */
 static NTSTATUS dcesrv_samr_info_DomInfo6(struct samr_domain_state *state,
-                                  TALLOC_CTX *mem_ctx,
-                                  struct ldb_message **dom_msgs,
-                                  struct samr_DomInfo6 *info)
+                                         TALLOC_CTX *mem_ctx,
+                                         struct ldb_message **dom_msgs,
+                                         struct samr_DomInfo6 *info)
 {
-       /* This pulls the NetBIOS name from the 
-          cn=NTDS Settings,cn=<NETBIOS name of PDC>,....
-          string */
-       info->primary.string = samdb_result_fsmo_name(state->sam_ctx, mem_ctx, 
-                                                     dom_msgs[0], "fSMORoleOwner");
-
-       if (!info->primary.string) {
-               info->primary.string = lp_netbios_name(state->lp_ctx);
-       }
+       /* MS-SAMR 2.2.4.1 - ReplicaSourceNodeName: "domainReplica" attribute */
+       info->primary.string = ldb_msg_find_attr_as_string(dom_msgs[0],
+                                                          "domainReplica",
+                                                          "");
 
        return NT_STATUS_OK;
 }
@@ -594,14 +600,14 @@ static NTSTATUS dcesrv_samr_info_DomInfo6(struct samr_domain_state *state,
   return DomInfo7
 */
 static NTSTATUS dcesrv_samr_info_DomInfo7(struct samr_domain_state *state,
-                                  TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                  struct samr_DomInfo7 *info)
+                                         TALLOC_CTX *mem_ctx,
+                                         struct ldb_message **dom_msgs,
+                                         struct samr_DomInfo7 *info)
 {
 
        switch (state->role) {
        case ROLE_DOMAIN_CONTROLLER:
-               /* This pulls the NetBIOS name from the 
+               /* This pulls the NetBIOS name from the
                   cn=NTDS Settings,cn=<NETBIOS name of PDC>,....
                   string */
                if (samdb_is_pdc(state->sam_ctx)) {
@@ -610,6 +616,9 @@ static NTSTATUS dcesrv_samr_info_DomInfo7(struct samr_domain_state *state,
                        info->role = SAMR_ROLE_DOMAIN_BDC;
                }
                break;
+       case ROLE_DOMAIN_PDC:
+               info->role = SAMR_ROLE_DOMAIN_PDC;
+               break;
        case ROLE_DOMAIN_MEMBER:
                info->role = SAMR_ROLE_DOMAIN_MEMBER;
                break;
@@ -625,11 +634,11 @@ static NTSTATUS dcesrv_samr_info_DomInfo7(struct samr_domain_state *state,
   return DomInfo8
 */
 static NTSTATUS dcesrv_samr_info_DomInfo8(struct samr_domain_state *state,
-                                  TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                  struct samr_DomInfo8 *info)
+                                         TALLOC_CTX *mem_ctx,
+                                         struct ldb_message **dom_msgs,
+                                         struct samr_DomInfo8 *info)
 {
-       info->sequence_num = ldb_msg_find_attr_as_uint64(dom_msgs[0], "modifiedCount", 
+       info->sequence_num = ldb_msg_find_attr_as_uint64(dom_msgs[0], "modifiedCount",
                                               time(NULL));
 
        info->domain_create_time = ldb_msg_find_attr_as_uint(dom_msgs[0], "creationTime",
@@ -642,9 +651,9 @@ static NTSTATUS dcesrv_samr_info_DomInfo8(struct samr_domain_state *state,
   return DomInfo9
 */
 static NTSTATUS dcesrv_samr_info_DomInfo9(struct samr_domain_state *state,
-                                  TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                  struct samr_DomInfo9 *info)
+                                         TALLOC_CTX *mem_ctx,
+                                         struct ldb_message **dom_msgs,
+                                         struct samr_DomInfo9 *info)
 {
        info->domain_server_state = DOMAIN_SERVER_ENABLED;
 
@@ -655,17 +664,17 @@ static NTSTATUS dcesrv_samr_info_DomInfo9(struct samr_domain_state *state,
   return DomInfo11
 */
 static NTSTATUS dcesrv_samr_info_DomGeneralInformation2(struct samr_domain_state *state,
-                                   TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                   struct samr_DomGeneralInformation2 *info)
+                                                       TALLOC_CTX *mem_ctx,
+                                                       struct ldb_message **dom_msgs,
+                                                       struct samr_DomGeneralInformation2 *info)
 {
        NTSTATUS status;
        status = dcesrv_samr_info_DomGeneralInformation(state, mem_ctx, dom_msgs, &info->general);
        if (!NT_STATUS_IS_OK(status)) {
                return status;
        }
-       
-       info->lockout_duration = ldb_msg_find_attr_as_int64(dom_msgs[0], "lockoutDuration", 
+
+       info->lockout_duration = ldb_msg_find_attr_as_int64(dom_msgs[0], "lockoutDuration",
                                                    -18000000000LL);
        info->lockout_window = ldb_msg_find_attr_as_int64(dom_msgs[0], "lockOutObservationWindow",
                                                    -18000000000LL);
@@ -678,11 +687,11 @@ static NTSTATUS dcesrv_samr_info_DomGeneralInformation2(struct samr_domain_state
   return DomInfo12
 */
 static NTSTATUS dcesrv_samr_info_DomInfo12(struct samr_domain_state *state,
-                                  TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                  struct samr_DomInfo12 *info)
+                                          TALLOC_CTX *mem_ctx,
+                                          struct ldb_message **dom_msgs,
+                                          struct samr_DomInfo12 *info)
 {
-       info->lockout_duration = ldb_msg_find_attr_as_int64(dom_msgs[0], "lockoutDuration", 
+       info->lockout_duration = ldb_msg_find_attr_as_int64(dom_msgs[0], "lockoutDuration",
                                                    -18000000000LL);
        info->lockout_window = ldb_msg_find_attr_as_int64(dom_msgs[0], "lockOutObservationWindow",
                                                    -18000000000LL);
@@ -695,11 +704,11 @@ static NTSTATUS dcesrv_samr_info_DomInfo12(struct samr_domain_state *state,
   return DomInfo13
 */
 static NTSTATUS dcesrv_samr_info_DomInfo13(struct samr_domain_state *state,
-                                   TALLOC_CTX *mem_ctx,
-                                   struct ldb_message **dom_msgs,
-                                   struct samr_DomInfo13 *info)
+                                          TALLOC_CTX *mem_ctx,
+                                          struct ldb_message **dom_msgs,
+                                          struct samr_DomInfo13 *info)
 {
-       info->sequence_num = ldb_msg_find_attr_as_uint64(dom_msgs[0], "modifiedCount", 
+       info->sequence_num = ldb_msg_find_attr_as_uint64(dom_msgs[0], "modifiedCount",
                                               time(NULL));
 
        info->domain_create_time = ldb_msg_find_attr_as_uint(dom_msgs[0], "creationTime",
@@ -710,11 +719,12 @@ static NTSTATUS dcesrv_samr_info_DomInfo13(struct samr_domain_state *state,
        return NT_STATUS_OK;
 }
 
-/* 
-  samr_QueryDomainInfo 
+/*
+  samr_QueryDomainInfo
 */
-static NTSTATUS dcesrv_samr_QueryDomainInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
-                                    struct samr_QueryDomainInfo *r)
+static NTSTATUS dcesrv_samr_QueryDomainInfo(struct dcesrv_call_state *dce_call,
+                                           TALLOC_CTX *mem_ctx,
+                                           struct samr_QueryDomainInfo *r)
 {
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
@@ -722,47 +732,45 @@ static NTSTATUS dcesrv_samr_QueryDomainInfo(struct dcesrv_call_state *dce_call,
 
        struct ldb_message **dom_msgs;
        const char * const *attrs = NULL;
-       
+
        *r->out.info = NULL;
 
        DCESRV_PULL_HANDLE(h, r->in.domain_handle, SAMR_HANDLE_DOMAIN);
 
        d_state = h->data;
 
-       info = talloc(mem_ctx, union samr_DomainInfo);
-       if (!info) {
-               return NT_STATUS_NO_MEMORY;
-       }
-
        switch (r->in.level) {
-       case 1: 
+       case 1:
        {
-               static const char * const attrs2[] = { "minPwdLength", "pwdHistoryLength",
-                                                      "pwdProperties", "maxPwdAge",
-                                                      "minPwdAge", NULL };
+               static const char * const attrs2[] = { "minPwdLength",
+                                                      "pwdHistoryLength",
+                                                      "pwdProperties",
+                                                      "maxPwdAge",
+                                                      "minPwdAge",
+                                                      NULL };
                attrs = attrs2;
                break;
        }
        case 2:
        {
                static const char * const attrs2[] = {"forceLogoff",
-                                                     "oEMInformation", 
-                                                     "modifiedCount", 
-                                                     "fSMORoleOwner",
+                                                     "oEMInformation",
+                                                     "modifiedCount",
+                                                     "domainReplica",
                                                      NULL};
                attrs = attrs2;
                break;
        }
        case 3:
        {
-               static const char * const attrs2[] = {"forceLogoff", 
+               static const char * const attrs2[] = {"forceLogoff",
                                                      NULL};
                attrs = attrs2;
                break;
        }
        case 4:
        {
-               static const char * const attrs2[] = {"oEMInformation", 
+               static const char * const attrs2[] = {"oEMInformation",
                                                      NULL};
                attrs = attrs2;
                break;
@@ -774,8 +782,8 @@ static NTSTATUS dcesrv_samr_QueryDomainInfo(struct dcesrv_call_state *dce_call,
        }
        case 6:
        {
-               static const char * const attrs2[] = {"fSMORoleOwner", 
-                                                     NULL};
+               static const char * const attrs2[] = { "domainReplica",
+                                                      NULL };
                attrs = attrs2;
                break;
        }
@@ -786,43 +794,50 @@ static NTSTATUS dcesrv_samr_QueryDomainInfo(struct dcesrv_call_state *dce_call,
        }
        case 8:
        {
-               static const char * const attrs2[] = { "modifiedCount", 
-                                                      "creationTime", 
+               static const char * const attrs2[] = { "modifiedCount",
+                                                      "creationTime",
                                                       NULL };
                attrs = attrs2;
                break;
        }
        case 9:
+       {
                attrs = NULL;
-               break;          
+               break;
+       }
        case 11:
        {
-               static const char * const attrs2[] = { "oEMInformation", "forceLogoff", 
-                                                      "modifiedCount", 
-                                                      "lockoutDuration", 
-                                                      "lockOutObservationWindow", 
-                                                      "lockoutThreshold", 
+               static const char * const attrs2[] = { "oEMInformation",
+                                                      "forceLogoff",
+                                                      "modifiedCount",
+                                                      "lockoutDuration",
+                                                      "lockOutObservationWindow",
+                                                      "lockoutThreshold",
                                                       NULL};
                attrs = attrs2;
                break;
        }
        case 12:
        {
-               static const char * const attrs2[] = { "lockoutDuration", 
-                                                      "lockOutObservationWindow", 
-                                                      "lockoutThreshold", 
+               static const char * const attrs2[] = { "lockoutDuration",
+                                                      "lockOutObservationWindow",
+                                                      "lockoutThreshold",
                                                       NULL};
                attrs = attrs2;
                break;
        }
        case 13:
        {
-               static const char * const attrs2[] = { "modifiedCount", 
-                                                      "creationTime", 
+               static const char * const attrs2[] = { "modifiedCount",
+                                                      "creationTime",
                                                       NULL };
                attrs = attrs2;
                break;
        }
+       default:
+       {
+               return NT_STATUS_INVALID_INFO_CLASS;
+       }
        }
 
        /* some levels don't need a search */
@@ -830,60 +845,67 @@ static NTSTATUS dcesrv_samr_QueryDomainInfo(struct dcesrv_call_state *dce_call,
                int ret;
                ret = gendb_search_dn(d_state->sam_ctx, mem_ctx,
                                      d_state->domain_dn, &dom_msgs, attrs);
+               if (ret == 0) {
+                       return NT_STATUS_NO_SUCH_DOMAIN;
+               }
                if (ret != 1) {
                        return NT_STATUS_INTERNAL_DB_CORRUPTION;
                }
        }
 
-       *r->out.info = info;
+       /* allocate the info structure */
+       info = talloc_zero(mem_ctx, union samr_DomainInfo);
+       if (info == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-       ZERO_STRUCTP(info);
+       *r->out.info = info;
 
        switch (r->in.level) {
        case 1:
-               return dcesrv_samr_info_DomInfo1(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo1(d_state, mem_ctx, dom_msgs,
                                                 &info->info1);
        case 2:
-               return dcesrv_samr_info_DomGeneralInformation(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomGeneralInformation(d_state, mem_ctx, dom_msgs,
                                                              &info->general);
        case 3:
-               return dcesrv_samr_info_DomInfo3(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo3(d_state, mem_ctx, dom_msgs,
                                                 &info->info3);
        case 4:
-               return dcesrv_samr_info_DomOEMInformation(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomOEMInformation(d_state, mem_ctx, dom_msgs,
                                                          &info->oem);
        case 5:
-               return dcesrv_samr_info_DomInfo5(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo5(d_state, mem_ctx, dom_msgs,
                                                 &info->info5);
        case 6:
-               return dcesrv_samr_info_DomInfo6(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo6(d_state, mem_ctx, dom_msgs,
                                                 &info->info6);
        case 7:
-               return dcesrv_samr_info_DomInfo7(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo7(d_state, mem_ctx, dom_msgs,
                                                 &info->info7);
        case 8:
-               return dcesrv_samr_info_DomInfo8(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo8(d_state, mem_ctx, dom_msgs,
                                                 &info->info8);
        case 9:
-               return dcesrv_samr_info_DomInfo9(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo9(d_state, mem_ctx, dom_msgs,
                                                 &info->info9);
        case 11:
-               return dcesrv_samr_info_DomGeneralInformation2(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomGeneralInformation2(d_state, mem_ctx, dom_msgs,
                                                               &info->general2);
        case 12:
-               return dcesrv_samr_info_DomInfo12(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo12(d_state, mem_ctx, dom_msgs,
                                                  &info->info12);
        case 13:
-               return dcesrv_samr_info_DomInfo13(d_state, mem_ctx, dom_msgs, 
+               return dcesrv_samr_info_DomInfo13(d_state, mem_ctx, dom_msgs,
                                                  &info->info13);
+       default:
+               return NT_STATUS_INVALID_INFO_CLASS;
        }
-
-       return NT_STATUS_INVALID_INFO_CLASS;
 }
 
 
-/* 
-  samr_SetDomainInfo 
+/*
+  samr_SetDomainInfo
 */
 static NTSTATUS dcesrv_samr_SetDomainInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_SetDomainInfo *r)
@@ -931,7 +953,28 @@ static NTSTATUS dcesrv_samr_SetDomainInfo(struct dcesrv_call_state *dce_call, TA
                return NT_STATUS_OK;
 
        case 12:
-               
+               /*
+                * It is not possible to set lockout_duration < lockout_window.
+                * (The test is the other way around since the negative numbers
+                *  are stored...)
+                *
+                * TODO:
+                *   This check should be moved to the backend, i.e. to some
+                *   ldb module under dsdb/samdb/ldb_modules/ .
+                *
+                * This constraint is documented here for the samr rpc service:
+                * MS-SAMR 3.1.1.6 Attribute Constraints for Originating Updates
+                * http://msdn.microsoft.com/en-us/library/cc245667%28PROT.10%29.aspx
+                *
+                * And here for the ldap backend:
+                * MS-ADTS 3.1.1.5.3.2 Constraints
+                * http://msdn.microsoft.com/en-us/library/cc223462(PROT.10).aspx
+                */
+               if (r->in.info->info12.lockout_duration >
+                   r->in.info->info12.lockout_window)
+               {
+                       return NT_STATUS_INVALID_PARAMETER;
+               }
                SET_INT64  (msg, info12.lockout_duration,      "lockoutDuration");
                SET_INT64  (msg, info12.lockout_window,        "lockOutObservationWindow");
                SET_INT64  (msg, info12.lockout_threshold,     "lockoutThreshold");
@@ -944,33 +987,30 @@ static NTSTATUS dcesrv_samr_SetDomainInfo(struct dcesrv_call_state *dce_call, TA
 
        /* modify the samdb record */
        ret = ldb_modify(sam_ctx, msg);
-       if (ret != 0) {
+       if (ret != LDB_SUCCESS) {
                DEBUG(1,("Failed to modify record %s: %s\n",
                         ldb_dn_get_linearized(d_state->domain_dn),
                         ldb_errstring(sam_ctx)));
-
-               /* we really need samdb.c to return NTSTATUS */
-               return NT_STATUS_UNSUCCESSFUL;
+               return dsdb_ldb_err_to_ntstatus(ret);
        }
 
        return NT_STATUS_OK;
 }
 
-/* 
-  samr_CreateDomainGroup 
+/*
+  samr_CreateDomainGroup
 */
 static NTSTATUS dcesrv_samr_CreateDomainGroup(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                       struct samr_CreateDomainGroup *r)
 {
+       NTSTATUS status;
        struct samr_domain_state *d_state;
        struct samr_account_state *a_state;
        struct dcesrv_handle *h;
-       const char *name;
-       struct ldb_message *msg;
-       struct dom_sid *sid;
        const char *groupname;
+       struct dom_sid *group_sid;
+       struct ldb_dn *group_dn;
        struct dcesrv_handle *g_handle;
-       int ret;
 
        ZERO_STRUCTP(r->out.group_handle);
        *r->out.rid = 0;
@@ -990,71 +1030,21 @@ static NTSTATUS dcesrv_samr_CreateDomainGroup(struct dcesrv_call_state *dce_call
                return NT_STATUS_INVALID_PARAMETER;
        }
 
-       /* check if the group already exists */
-       name = samdb_search_string(d_state->sam_ctx, mem_ctx, NULL, 
-                                  "sAMAccountName",
-                                  "(&(sAMAccountName=%s)(objectclass=group))",
-                                  ldb_binary_encode_string(mem_ctx, groupname));
-       if (name != NULL) {
-               return NT_STATUS_GROUP_EXISTS;
-       }
-
-       msg = ldb_msg_new(mem_ctx);
-       if (msg == NULL) {
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       /* add core elements to the ldb_message for the user */
-       msg->dn = ldb_dn_copy(mem_ctx, d_state->domain_dn);
-       ldb_dn_add_child_fmt(msg->dn, "CN=%s,CN=Users", groupname);
-       if (!msg->dn) {
-               return NT_STATUS_NO_MEMORY;
-       }
-       samdb_msg_add_string(d_state->sam_ctx, mem_ctx, msg, "sAMAccountName", groupname);
-       samdb_msg_add_string(d_state->sam_ctx, mem_ctx, msg, "objectClass", "group");
-                            
-       /* create the group */
-       ret = ldb_add(d_state->sam_ctx, msg);
-       switch (ret) {
-       case  LDB_SUCCESS:
-               break;
-       case  LDB_ERR_ENTRY_ALREADY_EXISTS:
-               DEBUG(0,("Failed to create group record %s: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_GROUP_EXISTS;
-       case  LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS:
-               DEBUG(0,("Failed to create group record %s: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_ACCESS_DENIED;
-       default:
-               DEBUG(0,("Failed to create group record %s: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       status = dsdb_add_domain_group(d_state->sam_ctx, mem_ctx, groupname, &group_sid, &group_dn);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       a_state = talloc(d_state, struct samr_account_state);
+       a_state = talloc(mem_ctx, struct samr_account_state);
        if (!a_state) {
                return NT_STATUS_NO_MEMORY;
        }
        a_state->sam_ctx = d_state->sam_ctx;
        a_state->access_mask = r->in.access_mask;
        a_state->domain_state = talloc_reference(a_state, d_state);
-       a_state->account_dn = talloc_steal(a_state, msg->dn);
+       a_state->account_dn = talloc_steal(a_state, group_dn);
 
-       /* retrieve the sid for the group just created */
-       sid = samdb_search_dom_sid(d_state->sam_ctx, a_state,
-                                  msg->dn, "objectSid", NULL);
-       if (sid == NULL) {
-               return NT_STATUS_UNSUCCESSFUL;
-       }
-
-       a_state->account_name = talloc_strdup(a_state, groupname);
-       if (!a_state->account_name) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       a_state->account_name = talloc_steal(a_state, groupname);
 
        /* create the policy handle */
        g_handle = dcesrv_handle_new(dce_call->context, SAMR_HANDLE_GROUP);
@@ -1065,7 +1055,7 @@ static NTSTATUS dcesrv_samr_CreateDomainGroup(struct dcesrv_call_state *dce_call
        g_handle->data = talloc_steal(g_handle, a_state);
 
        *r->out.group_handle = g_handle->wire_handle;
-       *r->out.rid = sid->sub_auths[sid->num_auths-1];
+       *r->out.rid = group_sid->sub_auths[group_sid->num_auths-1];
 
        return NT_STATUS_OK;
 }
@@ -1079,8 +1069,8 @@ static int compare_SamEntry(struct samr_SamEntry *e1, struct samr_SamEntry *e2)
        return e1->idx - e2->idx;
 }
 
-/* 
-  samr_EnumDomainGroups 
+/*
+  samr_EnumDomainGroups
 */
 static NTSTATUS dcesrv_samr_EnumDomainGroups(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                      struct samr_EnumDomainGroups *r)
@@ -1088,9 +1078,10 @@ static NTSTATUS dcesrv_samr_EnumDomainGroups(struct dcesrv_call_state *dce_call,
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
        struct ldb_message **res;
-       int ldb_cnt, count, i, first;
+       int i, ldb_cnt;
+       uint32_t first, count;
        struct samr_SamEntry *entries;
-       const char * const attrs[3] = { "objectSid", "sAMAccountName", NULL };
+       const char * const attrs[] = { "objectSid", "sAMAccountName", NULL };
        struct samr_SamArray *sam;
 
        *r->out.resume_handle = 0;
@@ -1106,9 +1097,10 @@ static NTSTATUS dcesrv_samr_EnumDomainGroups(struct dcesrv_call_state *dce_call,
        ldb_cnt = samdb_search_domain(d_state->sam_ctx, mem_ctx,
                                      d_state->domain_dn, &res, attrs,
                                      d_state->domain_sid,
-                                     "(&(grouptype=%d)(objectclass=group))",
+                                     "(&(|(groupType=%d)(groupType=%d))(objectClass=group))",
+                                     GTYPE_SECURITY_UNIVERSAL_GROUP,
                                      GTYPE_SECURITY_GLOBAL_GROUP);
-       if (ldb_cnt == -1) {
+       if (ldb_cnt < 0) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
@@ -1125,26 +1117,26 @@ static NTSTATUS dcesrv_samr_EnumDomainGroups(struct dcesrv_call_state *dce_call,
 
                group_sid = samdb_result_dom_sid(mem_ctx, res[i],
                                                 "objectSid");
-               if (group_sid == NULL)
-                       continue;
+               if (group_sid == NULL) {
+                       return NT_STATUS_INTERNAL_DB_CORRUPTION;
+               }
 
                entries[count].idx =
                        group_sid->sub_auths[group_sid->num_auths-1];
                entries[count].name.string =
-                       samdb_result_string(res[i], "sAMAccountName", "");
+                       ldb_msg_find_attr_as_string(res[i], "sAMAccountName", "");
                count += 1;
        }
 
        /* sort the results by rid */
-       qsort(entries, count, sizeof(struct samr_SamEntry), 
-             (comparison_fn_t)compare_SamEntry);
+       TYPESAFE_QSORT(entries, count, compare_SamEntry);
 
        /* find the first entry to return */
        for (first=0;
             first<count && entries[first].idx <= *r->in.resume_handle;
             first++) ;
 
-       /* return the rest, limit by max_size. Note that we 
+       /* return the rest, limit by max_size. Note that we
           use the w2k3 element size value of 54 */
        *r->out.num_entries = count - first;
        *r->out.num_entries = MIN(*r->out.num_entries,
@@ -1160,6 +1152,10 @@ static NTSTATUS dcesrv_samr_EnumDomainGroups(struct dcesrv_call_state *dce_call,
 
        *r->out.sam = sam;
 
+       if (first == count) {
+               return NT_STATUS_OK;
+       }
+
        if (*r->out.num_entries < count - first) {
                *r->out.resume_handle = entries[first+*r->out.num_entries-1].idx;
                return STATUS_MORE_ENTRIES;
@@ -1169,8 +1165,8 @@ static NTSTATUS dcesrv_samr_EnumDomainGroups(struct dcesrv_call_state *dce_call,
 }
 
 
-/* 
-  samr_CreateUser2 
+/*
+  samr_CreateUser2
 
   This call uses transactions to ensure we don't get a new conflicting
   user while we are processing this, and to ensure the user either
@@ -1179,28 +1175,14 @@ static NTSTATUS dcesrv_samr_EnumDomainGroups(struct dcesrv_call_state *dce_call,
 static NTSTATUS dcesrv_samr_CreateUser2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                 struct samr_CreateUser2 *r)
 {
+       NTSTATUS status;
        struct samr_domain_state *d_state;
        struct samr_account_state *a_state;
        struct dcesrv_handle *h;
-       const char *name;
-       struct ldb_message *msg;
+       struct ldb_dn *dn;
        struct dom_sid *sid;
-       const char *account_name;
        struct dcesrv_handle *u_handle;
-       int ret;
-       const char *container, *obj_class=NULL;
-       char *cn_name;
-       int cn_name_len;
-
-       const char *attrs[] = {
-               "objectSid", 
-               "userAccountControl",
-               NULL
-       };
-
-       uint32_t user_account_control;
-
-       struct ldb_message **msgs;
+       const char *account_name;
 
        ZERO_STRUCTP(r->out.user_handle);
        *r->out.access_granted = 0;
@@ -1223,180 +1205,19 @@ static NTSTATUS dcesrv_samr_CreateUser2(struct dcesrv_call_state *dce_call, TALL
                return NT_STATUS_INVALID_PARAMETER;
        }
 
-       /*
-        * Start a transaction, so we can query and do a subsequent atomic
-        * modify
-        */
-
-       ret = ldb_transaction_start(d_state->sam_ctx);
-       if (ret != 0) {
-               DEBUG(0,("Failed to start a transaction for user creation: %s\n",
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
-       }
-
-       /* check if the user already exists */
-       name = samdb_search_string(d_state->sam_ctx, mem_ctx, NULL, 
-                                  "sAMAccountName", 
-                                  "(&(sAMAccountName=%s)(objectclass=user))", 
-                                  ldb_binary_encode_string(mem_ctx, account_name));
-       if (name != NULL) {
-               ldb_transaction_cancel(d_state->sam_ctx);
-               return NT_STATUS_USER_EXISTS;
-       }
-
-       msg = ldb_msg_new(mem_ctx);
-       if (msg == NULL) {
-               ldb_transaction_cancel(d_state->sam_ctx);
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       cn_name   = talloc_strdup(mem_ctx, account_name);
-       if (!cn_name) {
-               ldb_transaction_cancel(d_state->sam_ctx);
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       cn_name_len = strlen(cn_name);
-
-       /* This must be one of these values *only* */
-       if (r->in.acct_flags == ACB_NORMAL) {
-               container = "CN=Users";
-               obj_class = "user";
-
-       } else if (r->in.acct_flags == ACB_WSTRUST) {
-               if (cn_name[cn_name_len - 1] != '$') {
-                       ldb_transaction_cancel(d_state->sam_ctx);
-                       return NT_STATUS_FOOBAR;
-               }
-               cn_name[cn_name_len - 1] = '\0';
-               container = "CN=Computers";
-               obj_class = "computer";
-               samdb_msg_add_int(d_state->sam_ctx, mem_ctx, msg, "primaryGroupID", DOMAIN_RID_DOMAIN_MEMBERS);
-
-       } else if (r->in.acct_flags == ACB_SVRTRUST) {
-               if (cn_name[cn_name_len - 1] != '$') {
-                       ldb_transaction_cancel(d_state->sam_ctx);
-                       return NT_STATUS_FOOBAR;                
-               }
-               cn_name[cn_name_len - 1] = '\0';
-               container = "OU=Domain Controllers";
-               obj_class = "computer";
-               samdb_msg_add_int(d_state->sam_ctx, mem_ctx, msg, "primaryGroupID", DOMAIN_RID_DCS);
-       } else {
-               ldb_transaction_cancel(d_state->sam_ctx);
-               return NT_STATUS_INVALID_PARAMETER;
-       }
-
-       /* add core elements to the ldb_message for the user */
-       msg->dn = ldb_dn_copy(mem_ctx, d_state->domain_dn);
-       if ( ! ldb_dn_add_child_fmt(msg->dn, "CN=%s,%s", cn_name, container)) {
-               ldb_transaction_cancel(d_state->sam_ctx);
-               return NT_STATUS_FOOBAR;
-       }
-
-       samdb_msg_add_string(d_state->sam_ctx, mem_ctx, msg, "sAMAccountName", account_name);
-       samdb_msg_add_string(d_state->sam_ctx, mem_ctx, msg, "objectClass", obj_class);
-
-       /* create the user */
-       ret = ldb_add(d_state->sam_ctx, msg);
-       switch (ret) {
-       case LDB_SUCCESS:
-               break;
-       case LDB_ERR_ENTRY_ALREADY_EXISTS:
-               ldb_transaction_cancel(d_state->sam_ctx);
-               DEBUG(0,("Failed to create user record %s: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_USER_EXISTS;
-       case LDB_ERR_UNWILLING_TO_PERFORM:
-       case LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS:
-               ldb_transaction_cancel(d_state->sam_ctx);
-               DEBUG(0,("Failed to create user record %s: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_ACCESS_DENIED;
-       default:
-               ldb_transaction_cancel(d_state->sam_ctx);
-               DEBUG(0,("Failed to create user record %s: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       status = dsdb_add_user(d_state->sam_ctx, mem_ctx, account_name, r->in.acct_flags, NULL,
+                              &sid, &dn);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
-
-       a_state = talloc(d_state, struct samr_account_state);
+       a_state = talloc(mem_ctx, struct samr_account_state);
        if (!a_state) {
-               ldb_transaction_cancel(d_state->sam_ctx);
                return NT_STATUS_NO_MEMORY;
        }
        a_state->sam_ctx = d_state->sam_ctx;
        a_state->access_mask = r->in.access_mask;
        a_state->domain_state = talloc_reference(a_state, d_state);
-       a_state->account_dn = talloc_steal(a_state, msg->dn);
-
-       /* retrieve the sid and account control bits for the user just created */
-       ret = gendb_search_dn(d_state->sam_ctx, a_state,
-                             msg->dn, &msgs, attrs);
-
-       if (ret != 1) {
-               ldb_transaction_cancel(d_state->sam_ctx);
-               DEBUG(0,("Apparently we failed to create an account record, as %s now doesn't exist\n",
-                        ldb_dn_get_linearized(msg->dn)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
-       }
-       sid = samdb_result_dom_sid(mem_ctx, msgs[0], "objectSid");
-       if (sid == NULL) {
-               ldb_transaction_cancel(d_state->sam_ctx);
-               DEBUG(0,("Apparently we failed to get the objectSid of the just created account record %s\n",
-                        ldb_dn_get_linearized(msg->dn)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
-       }
-
-       /* Change the account control to be the correct account type.
-        * The default is for a workstation account */
-       user_account_control = samdb_result_uint(msgs[0], "userAccountControl", 0);
-       user_account_control = (user_account_control & 
-                               ~(UF_NORMAL_ACCOUNT |
-                                 UF_INTERDOMAIN_TRUST_ACCOUNT | 
-                                 UF_WORKSTATION_TRUST_ACCOUNT | 
-                                 UF_SERVER_TRUST_ACCOUNT));
-       user_account_control |= samdb_acb2uf(r->in.acct_flags);
-
-       talloc_free(msg);
-       msg = ldb_msg_new(mem_ctx);
-       if (msg == NULL) {
-               ldb_transaction_cancel(d_state->sam_ctx);
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       msg->dn = ldb_dn_copy(msg, a_state->account_dn);
-
-       if (samdb_msg_add_uint(a_state->sam_ctx, mem_ctx, msg, 
-                              "userAccountControl", 
-                              user_account_control) != 0) { 
-               ldb_transaction_cancel(d_state->sam_ctx);
-               return NT_STATUS_NO_MEMORY; 
-       }
-
-       /* modify the samdb record */
-       ret = samdb_replace(a_state->sam_ctx, mem_ctx, msg);
-       if (ret != 0) {
-               DEBUG(0,("Failed to modify account record %s to set userAccountControl: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               ldb_transaction_cancel(d_state->sam_ctx);
-
-               /* we really need samdb.c to return NTSTATUS */
-               return NT_STATUS_UNSUCCESSFUL;
-       }
-
-       ret = ldb_transaction_commit(d_state->sam_ctx);
-       if (ret != 0) {
-               DEBUG(0,("Failed to commit transaction to add and modify account record %s: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
-       }
+       a_state->account_dn = talloc_steal(a_state, dn);
 
        a_state->account_name = talloc_steal(a_state, account_name);
        if (!a_state->account_name) {
@@ -1420,8 +1241,8 @@ static NTSTATUS dcesrv_samr_CreateUser2(struct dcesrv_call_state *dce_call, TALL
 }
 
 
-/* 
-  samr_CreateUser 
+/*
+  samr_CreateUser
 */
 static NTSTATUS dcesrv_samr_CreateUser(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                struct samr_CreateUser *r)
@@ -1442,18 +1263,20 @@ static NTSTATUS dcesrv_samr_CreateUser(struct dcesrv_call_state *dce_call, TALLO
        return dcesrv_samr_CreateUser2(dce_call, mem_ctx, &r2);
 }
 
-/* 
-  samr_EnumDomainUsers 
+/*
+  samr_EnumDomainUsers
 */
 static NTSTATUS dcesrv_samr_EnumDomainUsers(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                     struct samr_EnumDomainUsers *r)
 {
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
-       struct ldb_result *res;
-       int ret, num_filtered_entries, i, first;
+       struct ldb_message **res;
+       int i, ldb_cnt;
+       uint32_t first, count;
        struct samr_SamEntry *entries;
-       const char * const attrs[] = { "objectSid", "sAMAccountName", "userAccountControl", NULL };
+       const char * const attrs[] = { "objectSid", "sAMAccountName",
+               "userAccountControl", NULL };
        struct samr_SamArray *sam;
 
        *r->out.resume_handle = 0;
@@ -1463,46 +1286,51 @@ static NTSTATUS dcesrv_samr_EnumDomainUsers(struct dcesrv_call_state *dce_call,
        DCESRV_PULL_HANDLE(h, r->in.domain_handle, SAMR_HANDLE_DOMAIN);
 
        d_state = h->data;
-       
-       /* don't have to worry about users in the builtin domain, as there are none */
-       ret = ldb_search(d_state->sam_ctx, mem_ctx, &res, d_state->domain_dn, LDB_SCOPE_SUBTREE, attrs, "objectClass=user");
 
-       if (ret != LDB_SUCCESS) {
-               DEBUG(3, ("Failed to search for Domain Users in %s: %s\n", 
-                         ldb_dn_get_linearized(d_state->domain_dn), ldb_errstring(d_state->sam_ctx)));
+       /* search for all domain users in this domain. This could possibly be
+          cached and resumed on resume_key */
+       ldb_cnt = samdb_search_domain(d_state->sam_ctx, mem_ctx,
+                                     d_state->domain_dn,
+                                     &res, attrs,
+                                     d_state->domain_sid,
+                                     "(objectClass=user)");
+       if (ldb_cnt < 0) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
        /* convert to SamEntry format */
-       entries = talloc_array(mem_ctx, struct samr_SamEntry, res->count);
+       entries = talloc_array(mem_ctx, struct samr_SamEntry, ldb_cnt);
        if (!entries) {
                return NT_STATUS_NO_MEMORY;
        }
-       num_filtered_entries = 0;
-       for (i=0;i<res->count;i++) {
+
+       count = 0;
+
+       for (i=0;i<ldb_cnt;i++) {
                /* Check if a mask has been requested */
                if (r->in.acct_flags
-                   && ((samdb_result_acct_flags(d_state->sam_ctx, mem_ctx, res->msgs[i], 
-                                                d_state->domain_dn) & r->in.acct_flags) == 0)) {
+                   && ((samdb_result_acct_flags(d_state->sam_ctx, mem_ctx,
+                                                res[i], d_state->domain_dn) & r->in.acct_flags) == 0)) {
                        continue;
                }
-               entries[num_filtered_entries].idx = samdb_result_rid_from_sid(mem_ctx, res->msgs[i], "objectSid", 0);
-               entries[num_filtered_entries].name.string = samdb_result_string(res->msgs[i], "sAMAccountName", "");
-               num_filtered_entries++;
+               entries[count].idx = samdb_result_rid_from_sid(mem_ctx, res[i],
+                                                              "objectSid", 0);
+               entries[count].name.string = ldb_msg_find_attr_as_string(res[i],
+                                                                "sAMAccountName", "");
+               count += 1;
        }
 
        /* sort the results by rid */
-       qsort(entries, num_filtered_entries, sizeof(struct samr_SamEntry), 
-             (comparison_fn_t)compare_SamEntry);
+       TYPESAFE_QSORT(entries, count, compare_SamEntry);
 
        /* find the first entry to return */
        for (first=0;
-            first<num_filtered_entries && entries[first].idx <= *r->in.resume_handle;
+            first<count && entries[first].idx <= *r->in.resume_handle;
             first++) ;
 
-       /* return the rest, limit by max_size. Note that we 
+       /* return the rest, limit by max_size. Note that we
           use the w2k3 element size value of 54 */
-       *r->out.num_entries = num_filtered_entries - first;
+       *r->out.num_entries = count - first;
        *r->out.num_entries = MIN(*r->out.num_entries,
                                 1+(r->in.max_size/SAMR_ENUM_USERS_MULTIPLIER));
 
@@ -1516,11 +1344,11 @@ static NTSTATUS dcesrv_samr_EnumDomainUsers(struct dcesrv_call_state *dce_call,
 
        *r->out.sam = sam;
 
-       if (first == num_filtered_entries) {
+       if (first == count) {
                return NT_STATUS_OK;
        }
 
-       if (*r->out.num_entries < num_filtered_entries - first) {
+       if (*r->out.num_entries < count - first) {
                *r->out.resume_handle = entries[first+*r->out.num_entries-1].idx;
                return STATUS_MORE_ENTRIES;
        }
@@ -1529,8 +1357,8 @@ static NTSTATUS dcesrv_samr_EnumDomainUsers(struct dcesrv_call_state *dce_call,
 }
 
 
-/* 
-  samr_CreateDomAlias 
+/*
+  samr_CreateDomAlias
 */
 static NTSTATUS dcesrv_samr_CreateDomAlias(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_CreateDomAlias *r)
@@ -1538,11 +1366,11 @@ static NTSTATUS dcesrv_samr_CreateDomAlias(struct dcesrv_call_state *dce_call, T
        struct samr_domain_state *d_state;
        struct samr_account_state *a_state;
        struct dcesrv_handle *h;
-       const char *alias_name, *name;
-       struct ldb_message *msg;
+       const char *alias_name;
        struct dom_sid *sid;
        struct dcesrv_handle *a_handle;
-       int ret;
+       struct ldb_dn *dn;
+       NTSTATUS status;
 
        ZERO_STRUCTP(r->out.alias_handle);
        *r->out.rid = 0;
@@ -1562,49 +1390,12 @@ static NTSTATUS dcesrv_samr_CreateDomAlias(struct dcesrv_call_state *dce_call, T
                return NT_STATUS_INVALID_PARAMETER;
        }
 
-       /* Check if alias already exists */
-       name = samdb_search_string(d_state->sam_ctx, mem_ctx, NULL,
-                                  "sAMAccountName",
-                                  "(sAMAccountName=%s)(objectclass=group))",
-                                  ldb_binary_encode_string(mem_ctx, alias_name));
-
-       if (name != NULL) {
-               return NT_STATUS_ALIAS_EXISTS;
-       }
-
-       msg = ldb_msg_new(mem_ctx);
-       if (msg == NULL) {
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       /* add core elements to the ldb_message for the alias */
-       msg->dn = ldb_dn_copy(mem_ctx, d_state->domain_dn);
-       ldb_dn_add_child_fmt(msg->dn, "CN=%s,CN=Users", alias_name);
-       if (!msg->dn) {
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       samdb_msg_add_string(d_state->sam_ctx, mem_ctx, msg, "sAMAccountName", alias_name);
-       samdb_msg_add_string(d_state->sam_ctx, mem_ctx, msg, "objectClass", "group");
-       samdb_msg_add_int(d_state->sam_ctx, mem_ctx, msg, "groupType", GTYPE_SECURITY_DOMAIN_LOCAL_GROUP);
-
-       /* create the alias */
-       ret = ldb_add(d_state->sam_ctx, msg);
-       switch (ret) {
-       case LDB_SUCCESS:
-               break;
-       case LDB_ERR_ENTRY_ALREADY_EXISTS:
-               return NT_STATUS_ALIAS_EXISTS;
-       case LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS:
-               return NT_STATUS_ACCESS_DENIED;
-       default:
-               DEBUG(0,("Failed to create alias record %s: %s\n",
-                        ldb_dn_get_linearized(msg->dn),
-                        ldb_errstring(d_state->sam_ctx)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       status = dsdb_add_domain_alias(d_state->sam_ctx, mem_ctx, alias_name, &sid, &dn);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       a_state = talloc(d_state, struct samr_account_state);
+       a_state = talloc(mem_ctx, struct samr_account_state);
        if (!a_state) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -1612,16 +1403,9 @@ static NTSTATUS dcesrv_samr_CreateDomAlias(struct dcesrv_call_state *dce_call, T
        a_state->sam_ctx = d_state->sam_ctx;
        a_state->access_mask = r->in.access_mask;
        a_state->domain_state = talloc_reference(a_state, d_state);
-       a_state->account_dn = talloc_steal(a_state, msg->dn);
+       a_state->account_dn = talloc_steal(a_state, dn);
 
-       /* retrieve the sid for the alias just created */
-       sid = samdb_search_dom_sid(d_state->sam_ctx, a_state,
-                                  msg->dn, "objectSid", NULL);
-
-       a_state->account_name = talloc_strdup(a_state, alias_name);
-       if (!a_state->account_name) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       a_state->account_name = talloc_steal(a_state, alias_name);
 
        /* create the policy handle */
        a_handle = dcesrv_handle_new(dce_call->context, SAMR_HANDLE_ALIAS);
@@ -1638,8 +1422,8 @@ static NTSTATUS dcesrv_samr_CreateDomAlias(struct dcesrv_call_state *dce_call, T
 }
 
 
-/* 
-  samr_EnumDomainAliases 
+/*
+  samr_EnumDomainAliases
 */
 static NTSTATUS dcesrv_samr_EnumDomainAliases(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_EnumDomainAliases *r)
@@ -1647,9 +1431,10 @@ static NTSTATUS dcesrv_samr_EnumDomainAliases(struct dcesrv_call_state *dce_call
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
        struct ldb_message **res;
-       int ldb_cnt, count, i, first;
+       int i, ldb_cnt;
+       uint32_t first, count;
        struct samr_SamEntry *entries;
-       const char * const attrs[3] = { "objectSid", "sAMAccountName", NULL };
+       const char * const attrs[] = { "objectSid", "sAMAccountName", NULL };
        struct samr_SamArray *sam;
 
        *r->out.resume_handle = 0;
@@ -1660,22 +1445,18 @@ static NTSTATUS dcesrv_samr_EnumDomainAliases(struct dcesrv_call_state *dce_call
 
        d_state = h->data;
 
-       /* search for all domain groups in this domain. This could possibly be
+       /* search for all domain aliases in this domain. This could possibly be
           cached and resumed based on resume_key */
-       ldb_cnt = samdb_search_domain(d_state->sam_ctx, mem_ctx,
-                                     d_state->domain_dn,
-                                     &res, attrs, 
+       ldb_cnt = samdb_search_domain(d_state->sam_ctx, mem_ctx, NULL,
+                                     &res, attrs,
                                      d_state->domain_sid,
                                      "(&(|(grouptype=%d)(grouptype=%d)))"
                                      "(objectclass=group))",
                                      GTYPE_SECURITY_BUILTIN_LOCAL_GROUP,
                                      GTYPE_SECURITY_DOMAIN_LOCAL_GROUP);
-       if (ldb_cnt == -1) {
+       if (ldb_cnt < 0) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
-       if (ldb_cnt == 0) {
-               return NT_STATUS_OK;
-       }
 
        /* convert to SamEntry format */
        entries = talloc_array(mem_ctx, struct samr_SamEntry, ldb_cnt);
@@ -1691,31 +1472,30 @@ static NTSTATUS dcesrv_samr_EnumDomainAliases(struct dcesrv_call_state *dce_call
                alias_sid = samdb_result_dom_sid(mem_ctx, res[i],
                                                 "objectSid");
 
-               if (alias_sid == NULL)
-                       continue;
+               if (alias_sid == NULL) {
+                       return NT_STATUS_INTERNAL_DB_CORRUPTION;
+               }
 
                entries[count].idx =
                        alias_sid->sub_auths[alias_sid->num_auths-1];
                entries[count].name.string =
-                       samdb_result_string(res[i], "sAMAccountName", "");
+                       ldb_msg_find_attr_as_string(res[i], "sAMAccountName", "");
                count += 1;
        }
 
        /* sort the results by rid */
-       qsort(entries, count, sizeof(struct samr_SamEntry), 
-             (comparison_fn_t)compare_SamEntry);
+       TYPESAFE_QSORT(entries, count, compare_SamEntry);
 
        /* find the first entry to return */
        for (first=0;
             first<count && entries[first].idx <= *r->in.resume_handle;
             first++) ;
 
-       if (first == count) {
-               return NT_STATUS_OK;
-       }
-
+       /* return the rest, limit by max_size. Note that we
+          use the w2k3 element size value of 54 */
        *r->out.num_entries = count - first;
-       *r->out.num_entries = MIN(*r->out.num_entries, 1000);
+       *r->out.num_entries = MIN(*r->out.num_entries,
+                                 1+(r->in.max_size/SAMR_ENUM_USERS_MULTIPLIER));
 
        sam = talloc(mem_ctx, struct samr_SamArray);
        if (!sam) {
@@ -1727,6 +1507,10 @@ static NTSTATUS dcesrv_samr_EnumDomainAliases(struct dcesrv_call_state *dce_call
 
        *r->out.sam = sam;
 
+       if (first == count) {
+               return NT_STATUS_OK;
+       }
+
        if (*r->out.num_entries < count - first) {
                *r->out.resume_handle =
                        entries[first+*r->out.num_entries-1].idx;
@@ -1737,57 +1521,61 @@ static NTSTATUS dcesrv_samr_EnumDomainAliases(struct dcesrv_call_state *dce_call
 }
 
 
-/* 
-  samr_GetAliasMembership 
+/*
+  samr_GetAliasMembership
 */
 static NTSTATUS dcesrv_samr_GetAliasMembership(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_GetAliasMembership *r)
 {
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
+       const char *filter;
+       const char * const attrs[] = { "objectSid", NULL };
        struct ldb_message **res;
-       int i, count = 0;
+       uint32_t i;
+       int count = 0;
 
        DCESRV_PULL_HANDLE(h, r->in.domain_handle, SAMR_HANDLE_DOMAIN);
 
        d_state = h->data;
 
-       if (r->in.sids->num_sids > 0) {
-               const char *filter;
-               const char * const attrs[2] = { "objectSid", NULL };
-
-               filter = talloc_asprintf(mem_ctx,
-                                        "(&(|(grouptype=%d)(grouptype=%d))"
-                                        "(objectclass=group)(|",
-                                        GTYPE_SECURITY_BUILTIN_LOCAL_GROUP,
-                                        GTYPE_SECURITY_DOMAIN_LOCAL_GROUP);
-               if (filter == NULL)
-                       return NT_STATUS_NO_MEMORY;
-
-               for (i=0; i<r->in.sids->num_sids; i++) {
-                       const char *memberdn;
+       filter = talloc_asprintf(mem_ctx,
+                                "(&(|(grouptype=%d)(grouptype=%d))"
+                                "(objectclass=group)(|",
+                                GTYPE_SECURITY_BUILTIN_LOCAL_GROUP,
+                                GTYPE_SECURITY_DOMAIN_LOCAL_GROUP);
+       if (filter == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-                       memberdn = 
-                               samdb_search_string(d_state->sam_ctx,
-                                                   mem_ctx, NULL, "distinguishedName",
-                                                   "(objectSid=%s)",
-                                                   ldap_encode_ndr_dom_sid(mem_ctx, 
-                                                                           r->in.sids->sids[i].sid));
+       for (i=0; i<r->in.sids->num_sids; i++) {
+               const char *memberdn;
 
-                       if (memberdn == NULL)
-                               continue;
+               memberdn = samdb_search_string(d_state->sam_ctx, mem_ctx, NULL,
+                                              "distinguishedName",
+                                              "(objectSid=%s)",
+                                              ldap_encode_ndr_dom_sid(mem_ctx,
+                                                                      r->in.sids->sids[i].sid));
+               if (memberdn == NULL) {
+                       continue;
+               }
 
-                       filter = talloc_asprintf(mem_ctx, "%s(member=%s)",
-                                                filter, memberdn);
-                       if (filter == NULL)
-                               return NT_STATUS_NO_MEMORY;
+               filter = talloc_asprintf(mem_ctx, "%s(member=%s)", filter,
+                                        memberdn);
+               if (filter == NULL) {
+                       return NT_STATUS_NO_MEMORY;
                }
+       }
 
-               count = samdb_search_domain(d_state->sam_ctx, mem_ctx,
-                                           d_state->domain_dn, &res, attrs,
-                                           d_state->domain_sid, "%s))", filter);
-               if (count < 0)
+       /* Find out if we had at least one valid member SID passed - otherwise
+        * just skip the search. */
+       if (strstr(filter, "member") != NULL) {
+               count = samdb_search_domain(d_state->sam_ctx, mem_ctx, NULL,
+                                           &res, attrs, d_state->domain_sid,
+                                           "%s))", filter);
+               if (count < 0) {
                        return NT_STATUS_INTERNAL_DB_CORRUPTION;
+               }
        }
 
        r->out.rids->count = 0;
@@ -1799,10 +1587,8 @@ static NTSTATUS dcesrv_samr_GetAliasMembership(struct dcesrv_call_state *dce_cal
                struct dom_sid *alias_sid;
 
                alias_sid = samdb_result_dom_sid(mem_ctx, res[i], "objectSid");
-
                if (alias_sid == NULL) {
-                       DEBUG(0, ("Could not find objectSid\n"));
-                       continue;
+                       return NT_STATUS_INTERNAL_DB_CORRUPTION;
                }
 
                r->out.rids->ids[r->out.rids->count] =
@@ -1814,15 +1600,15 @@ static NTSTATUS dcesrv_samr_GetAliasMembership(struct dcesrv_call_state *dce_cal
 }
 
 
-/* 
-  samr_LookupNames 
+/*
+  samr_LookupNames
 */
 static NTSTATUS dcesrv_samr_LookupNames(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                 struct samr_LookupNames *r)
 {
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
-       int i, num_mapped;
+       uint32_t i, num_mapped;
        NTSTATUS status = NT_STATUS_OK;
        const char * const attrs[] = { "sAMAccountType", "objectSid", NULL };
        int count;
@@ -1856,8 +1642,8 @@ static NTSTATUS dcesrv_samr_LookupNames(struct dcesrv_call_state *dce_call, TALL
                r->out.rids->ids[i] = 0;
                r->out.types->ids[i] = SID_NAME_UNKNOWN;
 
-               count = gendb_search(d_state->sam_ctx, mem_ctx, d_state->domain_dn, &res, attrs, 
-                                    "sAMAccountName=%s", 
+               count = gendb_search(d_state->sam_ctx, mem_ctx, d_state->domain_dn, &res, attrs,
+                                    "sAMAccountName=%s",
                                     ldb_binary_encode_string(mem_ctx, r->in.names[i].string));
                if (count != 1) {
                        status = STATUS_SOME_UNMAPPED;
@@ -1869,15 +1655,15 @@ static NTSTATUS dcesrv_samr_LookupNames(struct dcesrv_call_state *dce_call, TALL
                        status = STATUS_SOME_UNMAPPED;
                        continue;
                }
-               
-               atype = samdb_result_uint(res[0], "sAMAccountType", 0);
+
+               atype = ldb_msg_find_attr_as_uint(res[0], "sAMAccountType", 0);
                if (atype == 0) {
                        status = STATUS_SOME_UNMAPPED;
                        continue;
                }
 
-               rtype = samdb_atype_map(atype);
-               
+               rtype = ds_atype_map(atype);
+
                if (rtype == SID_NAME_UNKNOWN) {
                        status = STATUS_SOME_UNMAPPED;
                        continue;
@@ -1887,7 +1673,7 @@ static NTSTATUS dcesrv_samr_LookupNames(struct dcesrv_call_state *dce_call, TALL
                r->out.types->ids[i] = rtype;
                num_mapped++;
        }
-       
+
        if (num_mapped == 0) {
                return NT_STATUS_NONE_MAPPED;
        }
@@ -1895,18 +1681,18 @@ static NTSTATUS dcesrv_samr_LookupNames(struct dcesrv_call_state *dce_call, TALL
 }
 
 
-/* 
-  samr_LookupRids 
+/*
+  samr_LookupRids
 */
 static NTSTATUS dcesrv_samr_LookupRids(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_LookupRids *r)
 {
+       NTSTATUS status;
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
-       int i, total;
-       NTSTATUS status = NT_STATUS_OK;
-       struct lsa_String *names;
-       uint32_t *ids;
+       const char **names;
+       struct lsa_String *lsa_names;
+       enum lsa_SidType *ids;
 
        ZERO_STRUCTP(r->out.names);
        ZERO_STRUCTP(r->out.types);
@@ -1918,70 +1704,33 @@ static NTSTATUS dcesrv_samr_LookupRids(struct dcesrv_call_state *dce_call, TALLO
        if (r->in.num_rids == 0)
                return NT_STATUS_OK;
 
-       names = talloc_array(mem_ctx, struct lsa_String, r->in.num_rids);
-       ids = talloc_array(mem_ctx, uint32_t, r->in.num_rids);
+       lsa_names = talloc_zero_array(mem_ctx, struct lsa_String, r->in.num_rids);
+       names = talloc_zero_array(mem_ctx, const char *, r->in.num_rids);
+       ids = talloc_zero_array(mem_ctx, enum lsa_SidType, r->in.num_rids);
 
-       if ((names == NULL) || (ids == NULL))
+       if ((lsa_names == NULL) || (names == NULL) || (ids == NULL))
                return NT_STATUS_NO_MEMORY;
 
-       total = 0;
-
-       for (i=0; i<r->in.num_rids; i++) {
-               struct ldb_message **res;
-               int count;
-               const char * const attrs[] = {  "sAMAccountType",
-                                               "sAMAccountName", NULL };
-               uint32_t atype;
-               struct dom_sid *sid;
-
-               ids[i] = SID_NAME_UNKNOWN;
-
-               sid = dom_sid_add_rid(mem_ctx, d_state->domain_sid, r->in.rids[i]);
-               if (sid == NULL) {
-                       names[i].string = NULL;
-                       status = STATUS_SOME_UNMAPPED;
-                       continue;
-               }
-               
-               count = gendb_search(d_state->sam_ctx, mem_ctx,
-                                    d_state->domain_dn, &res, attrs,
-                                    "(objectSid=%s)", 
-                                    ldap_encode_ndr_dom_sid(mem_ctx, sid));
-               if (count != 1) {
-                       names[i].string = NULL;
-                       status = STATUS_SOME_UNMAPPED;
-                       continue;
-               }
-
-               names[i].string = samdb_result_string(res[0], "sAMAccountName",
-                                                     NULL);
-
-               atype = samdb_result_uint(res[0], "sAMAccountType", 0);
-               if (atype == 0) {
-                       status = STATUS_SOME_UNMAPPED;
-                       continue;
-               }
-
-               ids[i] = samdb_atype_map(atype);
-               
-               if (ids[i] == SID_NAME_UNKNOWN) {
-                       status = STATUS_SOME_UNMAPPED;
-                       continue;
-               }
-       }
-
-       r->out.names->names = names;
+       r->out.names->names = lsa_names;
        r->out.names->count = r->in.num_rids;
 
-       r->out.types->ids = ids;
+       r->out.types->ids = (uint32_t *) ids;
        r->out.types->count = r->in.num_rids;
 
+       status = dsdb_lookup_rids(d_state->sam_ctx, mem_ctx, d_state->domain_sid,
+                                 r->in.num_rids, r->in.rids, names, ids);
+       if (NT_STATUS_IS_OK(status) || NT_STATUS_EQUAL(status, NT_STATUS_NONE_MAPPED) || NT_STATUS_EQUAL(status, STATUS_SOME_UNMAPPED)) {
+               uint32_t i;
+               for (i = 0; i < r->in.num_rids; i++) {
+                       lsa_names[i].string = names[i];
+               }
+       }
        return status;
 }
 
 
-/* 
-  samr_OpenGroup 
+/*
+  samr_OpenGroup
 */
 static NTSTATUS dcesrv_samr_OpenGroup(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_OpenGroup *r)
@@ -2011,27 +1760,28 @@ static NTSTATUS dcesrv_samr_OpenGroup(struct dcesrv_call_state *dce_call, TALLOC
        /* search for the group record */
        ret = gendb_search(d_state->sam_ctx,
                           mem_ctx, d_state->domain_dn, &msgs, attrs,
-                          "(&(objectSid=%s)(objectclass=group)"
-                          "(grouptype=%d))",
+                          "(&(objectSid=%s)(objectClass=group)"
+                          "(|(groupType=%d)(groupType=%d)))",
                           ldap_encode_ndr_dom_sid(mem_ctx, sid),
+                          GTYPE_SECURITY_UNIVERSAL_GROUP,
                           GTYPE_SECURITY_GLOBAL_GROUP);
        if (ret == 0) {
                return NT_STATUS_NO_SUCH_GROUP;
        }
        if (ret != 1) {
-               DEBUG(0,("Found %d records matching sid %s\n", 
+               DEBUG(0,("Found %d records matching sid %s\n",
                         ret, dom_sid_string(mem_ctx, sid)));
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
-       groupname = samdb_result_string(msgs[0], "sAMAccountName", NULL);
+       groupname = ldb_msg_find_attr_as_string(msgs[0], "sAMAccountName", NULL);
        if (groupname == NULL) {
-               DEBUG(0,("sAMAccountName field missing for sid %s\n", 
+               DEBUG(0,("sAMAccountName field missing for sid %s\n",
                         dom_sid_string(mem_ctx, sid)));
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
-       a_state = talloc(d_state, struct samr_account_state);
+       a_state = talloc(mem_ctx, struct samr_account_state);
        if (!a_state) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -2058,16 +1808,15 @@ static NTSTATUS dcesrv_samr_OpenGroup(struct dcesrv_call_state *dce_call, TALLOC
        return NT_STATUS_OK;
 }
 
-/* 
-  samr_QueryGroupInfo 
+/*
+  samr_QueryGroupInfo
 */
 static NTSTATUS dcesrv_samr_QueryGroupInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_QueryGroupInfo *r)
 {
        struct dcesrv_handle *h;
        struct samr_account_state *a_state;
-       struct ldb_message *msg;
-       struct ldb_result *res;
+       struct ldb_message *msg, **res;
        const char * const attrs[4] = { "sAMAccountName", "description",
                                        "numMembers", NULL };
        int ret;
@@ -2078,22 +1827,17 @@ static NTSTATUS dcesrv_samr_QueryGroupInfo(struct dcesrv_call_state *dce_call, T
        DCESRV_PULL_HANDLE(h, r->in.group_handle, SAMR_HANDLE_GROUP);
 
        a_state = h->data;
-       
-       ret = ldb_search(a_state->sam_ctx, mem_ctx, &res, a_state->account_dn, LDB_SCOPE_SUBTREE, attrs, "objectClass=*");
-       
-       if (ret == LDB_ERR_NO_SUCH_OBJECT) {
+
+       /* pull all the group attributes */
+       ret = gendb_search_dn(a_state->sam_ctx, mem_ctx,
+                             a_state->account_dn, &res, attrs);
+       if (ret == 0) {
                return NT_STATUS_NO_SUCH_GROUP;
-       } else if (ret != LDB_SUCCESS) {
-               DEBUG(2, ("Error reading group info: %s\n", ldb_errstring(a_state->sam_ctx)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
-
-       if (res->count != 1) {
-               DEBUG(2, ("Error finding group info, got %d entries\n", res->count));
-               
+       if (ret != 1) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
-       msg = res->msgs[0];
+       msg = res[0];
 
        /* allocate the info structure */
        info = talloc_zero(mem_ctx, union samr_GroupInfo);
@@ -2135,8 +1879,8 @@ static NTSTATUS dcesrv_samr_QueryGroupInfo(struct dcesrv_call_state *dce_call, T
 }
 
 
-/* 
-  samr_SetGroupInfo 
+/*
+  samr_SetGroupInfo
 */
 static NTSTATUS dcesrv_samr_SetGroupInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                  struct samr_SetGroupInfo *r)
@@ -2155,7 +1899,7 @@ static NTSTATUS dcesrv_samr_SetGroupInfo(struct dcesrv_call_state *dce_call, TAL
        msg = ldb_msg_new(mem_ctx);
        if (msg == NULL) {
                return NT_STATUS_NO_MEMORY;
-       }       
+       }
 
        msg->dn = ldb_dn_copy(mem_ctx, g_state->account_dn);
        if (!msg->dn) {
@@ -2180,17 +1924,16 @@ static NTSTATUS dcesrv_samr_SetGroupInfo(struct dcesrv_call_state *dce_call, TAL
 
        /* modify the samdb record */
        ret = ldb_modify(g_state->sam_ctx, msg);
-       if (ret != 0) {
-               /* we really need samdb.c to return NTSTATUS */
-               return NT_STATUS_UNSUCCESSFUL;
+       if (ret != LDB_SUCCESS) {
+               return dsdb_ldb_err_to_ntstatus(ret);
        }
 
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_AddGroupMember 
+/*
+  samr_AddGroupMember
 */
 static NTSTATUS dcesrv_samr_AddGroupMember(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_AddGroupMember *r)
@@ -2211,24 +1954,24 @@ static NTSTATUS dcesrv_samr_AddGroupMember(struct dcesrv_call_state *dce_call, T
        d_state = a_state->domain_state;
 
        membersid = dom_sid_add_rid(mem_ctx, d_state->domain_sid, r->in.rid);
-       if (membersid == NULL)
+       if (membersid == NULL) {
                return NT_STATUS_NO_MEMORY;
+       }
 
-       /* In native mode, AD can also nest domain groups. Not sure yet
-        * whether this is also available via RPC. */
+       /* according to MS-SAMR 3.1.5.8.2 all type of accounts are accepted */
        ret = ldb_search(d_state->sam_ctx, mem_ctx, &res,
-                                d_state->domain_dn, LDB_SCOPE_SUBTREE, attrs,
-                                "(&(objectSid=%s)(objectclass=user))",
-                                ldap_encode_ndr_dom_sid(mem_ctx, membersid));
+                        d_state->domain_dn, LDB_SCOPE_SUBTREE, attrs,
+                        "(objectSid=%s)",
+                        ldap_encode_ndr_dom_sid(mem_ctx, membersid));
 
-       if (ret != 0) {
+       if (ret != LDB_SUCCESS) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
        if (res->count == 0) {
                return NT_STATUS_NO_SUCH_USER;
        }
-               
+
        if (res->count > 1) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
@@ -2245,27 +1988,28 @@ static NTSTATUS dcesrv_samr_AddGroupMember(struct dcesrv_call_state *dce_call, T
 
        mod->dn = talloc_reference(mem_ctx, a_state->account_dn);
 
-       if (samdb_msg_add_addval(d_state->sam_ctx, mem_ctx, mod, "member",
-                                memberdn) != 0)
-               return NT_STATUS_UNSUCCESSFUL;
+       ret = samdb_msg_add_addval(d_state->sam_ctx, mem_ctx, mod, "member",
+                                                               memberdn);
+       if (ret != LDB_SUCCESS) {
+               return dsdb_ldb_err_to_ntstatus(ret);
+       }
 
        ret = ldb_modify(a_state->sam_ctx, mod);
        switch (ret) {
        case LDB_SUCCESS:
                return NT_STATUS_OK;
-       case LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS:
+       case LDB_ERR_ENTRY_ALREADY_EXISTS:
                return NT_STATUS_MEMBER_IN_GROUP;
        case LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS:
                return NT_STATUS_ACCESS_DENIED;
        default:
-               return NT_STATUS_UNSUCCESSFUL;
+               return dsdb_ldb_err_to_ntstatus(ret);
        }
-
 }
 
 
-/* 
-  samr_DeleteDomainGroup 
+/*
+  samr_DeleteDomainGroup
 */
 static NTSTATUS dcesrv_samr_DeleteDomainGroup(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_DeleteDomainGroup *r)
@@ -2281,18 +2025,19 @@ static NTSTATUS dcesrv_samr_DeleteDomainGroup(struct dcesrv_call_state *dce_call
        a_state = h->data;
 
        ret = ldb_delete(a_state->sam_ctx, a_state->account_dn);
-       if (ret != 0) {
-               return NT_STATUS_UNSUCCESSFUL;
+       if (ret != LDB_SUCCESS) {
+               return dsdb_ldb_err_to_ntstatus(ret);
        }
 
+       talloc_free(h);
        ZERO_STRUCTP(r->out.group_handle);
 
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_DeleteGroupMember 
+/*
+  samr_DeleteGroupMember
 */
 static NTSTATUS dcesrv_samr_DeleteGroupMember(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_DeleteGroupMember *r)
@@ -2313,24 +2058,24 @@ static NTSTATUS dcesrv_samr_DeleteGroupMember(struct dcesrv_call_state *dce_call
        d_state = a_state->domain_state;
 
        membersid = dom_sid_add_rid(mem_ctx, d_state->domain_sid, r->in.rid);
-       if (membersid == NULL)
+       if (membersid == NULL) {
                return NT_STATUS_NO_MEMORY;
+       }
 
-       /* In native mode, AD can also nest domain groups. Not sure yet
-        * whether this is also available via RPC. */
+       /* according to MS-SAMR 3.1.5.8.2 all type of accounts are accepted */
        ret = ldb_search(d_state->sam_ctx, mem_ctx, &res,
-                                d_state->domain_dn, LDB_SCOPE_SUBTREE, attrs,
-                                "(&(objectSid=%s)(objectclass=user))",
-                                ldap_encode_ndr_dom_sid(mem_ctx, membersid));
+                        d_state->domain_dn, LDB_SCOPE_SUBTREE, attrs,
+                        "(objectSid=%s)",
+                        ldap_encode_ndr_dom_sid(mem_ctx, membersid));
 
-       if (ret != 0) {
+       if (ret != LDB_SUCCESS) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
        if (res->count == 0) {
                return NT_STATUS_NO_SUCH_USER;
        }
-               
+
        if (res->count > 1) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
@@ -2347,8 +2092,9 @@ static NTSTATUS dcesrv_samr_DeleteGroupMember(struct dcesrv_call_state *dce_call
 
        mod->dn = talloc_reference(mem_ctx, a_state->account_dn);
 
-       if (samdb_msg_add_delval(d_state->sam_ctx, mem_ctx, mod, "member",
-                                memberdn) != 0) {
+       ret = samdb_msg_add_delval(d_state->sam_ctx, mem_ctx, mod, "member",
+                                                               memberdn);
+       if (ret != LDB_SUCCESS) {
                return NT_STATUS_NO_MEMORY;
        }
 
@@ -2356,85 +2102,79 @@ static NTSTATUS dcesrv_samr_DeleteGroupMember(struct dcesrv_call_state *dce_call
        switch (ret) {
        case LDB_SUCCESS:
                return NT_STATUS_OK;
-       case LDB_ERR_NO_SUCH_ATTRIBUTE:
+       case LDB_ERR_UNWILLING_TO_PERFORM:
                return NT_STATUS_MEMBER_NOT_IN_GROUP;
        case LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS:
                return NT_STATUS_ACCESS_DENIED;
        default:
-               return NT_STATUS_UNSUCCESSFUL;
+               return dsdb_ldb_err_to_ntstatus(ret);
        }
-
 }
 
 
-/* 
-  samr_QueryGroupMember 
+/*
+  samr_QueryGroupMember
 */
 static NTSTATUS dcesrv_samr_QueryGroupMember(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                      struct samr_QueryGroupMember *r)
 {
        struct dcesrv_handle *h;
        struct samr_account_state *a_state;
-       struct ldb_message **res;
-       struct ldb_message_element *el;
-       struct samr_RidTypeArray *array;
-       const char * const attrs[2] = { "member", NULL };
-       int ret;
+       struct samr_domain_state *d_state;
+       struct samr_RidAttrArray *array;
+       unsigned int i, num_members;
+       struct dom_sid *members;
+       NTSTATUS status;
 
        DCESRV_PULL_HANDLE(h, r->in.group_handle, SAMR_HANDLE_GROUP);
 
        a_state = h->data;
+       d_state = a_state->domain_state;
 
-       /* pull the member attribute */
-       ret = gendb_search_dn(a_state->sam_ctx, mem_ctx,
-                             a_state->account_dn, &res, attrs);
-
-       if (ret != 1) {
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       status = dsdb_enum_group_mem(d_state->sam_ctx, mem_ctx,
+                                    a_state->account_dn, &members,
+                                    &num_members);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       array = talloc(mem_ctx, struct samr_RidTypeArray);
-
-       if (array == NULL)
+       array = talloc_zero(mem_ctx, struct samr_RidAttrArray);
+       if (array == NULL) {
                return NT_STATUS_NO_MEMORY;
+       }
 
-       ZERO_STRUCTP(array);
-
-       el = ldb_msg_find_element(res[0], "member");
-
-       if (el != NULL) {
-               int i;
-
-               array->count = el->num_values;
-
-               array->rids = talloc_array(mem_ctx, uint32_t,
-                                            el->num_values);
-               if (array->rids == NULL)
-                       return NT_STATUS_NO_MEMORY;
+       if (num_members == 0) {
+               *r->out.rids = array;
 
-               array->types = talloc_array(mem_ctx, uint32_t,
-                                           el->num_values);
-               if (array->types == NULL)
-                       return NT_STATUS_NO_MEMORY;
+               return NT_STATUS_OK;
+       }
 
-               for (i=0; i<el->num_values; i++) {
-                       struct ldb_message **res2;
-                       const char * const attrs2[2] = { "objectSid", NULL };
-                       ret = gendb_search_dn(a_state->sam_ctx, mem_ctx,
-                                          ldb_dn_from_ldb_val(mem_ctx, a_state->sam_ctx, &el->values[i]),
-                                          &res2, attrs2);
-                       if (ret != 1)
-                               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       array->rids = talloc_array(array, uint32_t, num_members);
+       if (array->rids == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-                       array->rids[i] =
-                               samdb_result_rid_from_sid(mem_ctx, res2[0],
-                                                         "objectSid", 0);
+       array->attributes = talloc_array(array, uint32_t, num_members);
+       if (array->attributes == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-                       if (array->rids[i] == 0)
-                               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       array->count = 0;
+       for (i=0; i<num_members; i++) {
+               if (!dom_sid_in_domain(d_state->domain_sid, &members[i])) {
+                       continue;
+               }
 
-                       array->types[i] = 7; /* RID type of some kind, not sure what the value means. */
+               status = dom_sid_split_rid(NULL, &members[i], NULL,
+                                          &array->rids[array->count]);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
                }
+
+               array->attributes[array->count] = SE_GROUP_MANDATORY |
+                                                 SE_GROUP_ENABLED_BY_DEFAULT |
+                                                 SE_GROUP_ENABLED;
+               array->count++;
        }
 
        *r->out.rids = array;
@@ -2443,8 +2183,8 @@ static NTSTATUS dcesrv_samr_QueryGroupMember(struct dcesrv_call_state *dce_call,
 }
 
 
-/* 
-  samr_SetMemberAttributesOfGroup 
+/*
+  samr_SetMemberAttributesOfGroup
 */
 static NTSTATUS dcesrv_samr_SetMemberAttributesOfGroup(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_SetMemberAttributesOfGroup *r)
@@ -2453,8 +2193,8 @@ static NTSTATUS dcesrv_samr_SetMemberAttributesOfGroup(struct dcesrv_call_state
 }
 
 
-/* 
-  samr_OpenAlias 
+/*
+  samr_OpenAlias
 */
 static NTSTATUS dcesrv_samr_OpenAlias(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_OpenAlias *r)
@@ -2481,8 +2221,7 @@ static NTSTATUS dcesrv_samr_OpenAlias(struct dcesrv_call_state *dce_call, TALLOC
                return NT_STATUS_NO_MEMORY;
 
        /* search for the group record */
-       ret = gendb_search(d_state->sam_ctx,
-                          mem_ctx, d_state->domain_dn, &msgs, attrs,
+       ret = gendb_search(d_state->sam_ctx, mem_ctx, NULL, &msgs, attrs,
                           "(&(objectSid=%s)(objectclass=group)"
                           "(|(grouptype=%d)(grouptype=%d)))",
                           ldap_encode_ndr_dom_sid(mem_ctx, sid),
@@ -2492,19 +2231,19 @@ static NTSTATUS dcesrv_samr_OpenAlias(struct dcesrv_call_state *dce_call, TALLOC
                return NT_STATUS_NO_SUCH_ALIAS;
        }
        if (ret != 1) {
-               DEBUG(0,("Found %d records matching sid %s\n", 
+               DEBUG(0,("Found %d records matching sid %s\n",
                         ret, dom_sid_string(mem_ctx, sid)));
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
-       alias_name = samdb_result_string(msgs[0], "sAMAccountName", NULL);
+       alias_name = ldb_msg_find_attr_as_string(msgs[0], "sAMAccountName", NULL);
        if (alias_name == NULL) {
-               DEBUG(0,("sAMAccountName field missing for sid %s\n", 
+               DEBUG(0,("sAMAccountName field missing for sid %s\n",
                         dom_sid_string(mem_ctx, sid)));
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
-       a_state = talloc(d_state, struct samr_account_state);
+       a_state = talloc(mem_ctx, struct samr_account_state);
        if (!a_state) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -2532,8 +2271,8 @@ static NTSTATUS dcesrv_samr_OpenAlias(struct dcesrv_call_state *dce_call, TALLOC
 }
 
 
-/* 
-  samr_QueryAliasInfo 
+/*
+  samr_QueryAliasInfo
 */
 static NTSTATUS dcesrv_samr_QueryAliasInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_QueryAliasInfo *r)
@@ -2554,7 +2293,10 @@ static NTSTATUS dcesrv_samr_QueryAliasInfo(struct dcesrv_call_state *dce_call, T
 
        /* pull all the alias attributes */
        ret = gendb_search_dn(a_state->sam_ctx, mem_ctx,
-                             a_state->account_dn ,&res, attrs);
+                             a_state->account_dn, &res, attrs);
+       if (ret == 0) {
+               return NT_STATUS_NO_SUCH_ALIAS;
+       }
        if (ret != 1) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
@@ -2589,8 +2331,8 @@ static NTSTATUS dcesrv_samr_QueryAliasInfo(struct dcesrv_call_state *dce_call, T
 }
 
 
-/* 
-  samr_SetAliasInfo 
+/*
+  samr_SetAliasInfo
 */
 static NTSTATUS dcesrv_samr_SetAliasInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_SetAliasInfo *r)
@@ -2631,17 +2373,16 @@ static NTSTATUS dcesrv_samr_SetAliasInfo(struct dcesrv_call_state *dce_call, TAL
 
        /* modify the samdb record */
        ret = ldb_modify(a_state->sam_ctx, msg);
-       if (ret != 0) {
-               /* we really need samdb.c to return NTSTATUS */
-               return NT_STATUS_UNSUCCESSFUL;
+       if (ret != LDB_SUCCESS) {
+               return dsdb_ldb_err_to_ntstatus(ret);
        }
 
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_DeleteDomAlias 
+/*
+  samr_DeleteDomAlias
 */
 static NTSTATUS dcesrv_samr_DeleteDomAlias(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_DeleteDomAlias *r)
@@ -2657,18 +2398,19 @@ static NTSTATUS dcesrv_samr_DeleteDomAlias(struct dcesrv_call_state *dce_call, T
        a_state = h->data;
 
        ret = ldb_delete(a_state->sam_ctx, a_state->account_dn);
-       if (ret != 0) {
-               return NT_STATUS_UNSUCCESSFUL;
+       if (ret != LDB_SUCCESS) {
+               return dsdb_ldb_err_to_ntstatus(ret);
        }
 
+       talloc_free(h);
        ZERO_STRUCTP(r->out.alias_handle);
 
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_AddAliasMember 
+/*
+  samr_AddAliasMember
 */
 static NTSTATUS dcesrv_samr_AddAliasMember(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_AddAliasMember *r)
@@ -2689,23 +2431,21 @@ static NTSTATUS dcesrv_samr_AddAliasMember(struct dcesrv_call_state *dce_call, T
        d_state = a_state->domain_state;
 
        ret = gendb_search(d_state->sam_ctx, mem_ctx, NULL,
-                          &msgs, attrs, "(objectsid=%s)", 
+                          &msgs, attrs, "(objectsid=%s)",
                           ldap_encode_ndr_dom_sid(mem_ctx, r->in.sid));
 
        if (ret == 1) {
                memberdn = msgs[0]->dn;
-       } else  if (ret > 1) {
-               DEBUG(0,("Found %d records matching sid %s\n", 
-                        ret, dom_sid_string(mem_ctx, r->in.sid)));
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
        } else if (ret == 0) {
-               status = samdb_create_foreign_security_principal(d_state->sam_ctx, mem_ctx, 
-                                                                r->in.sid, &memberdn);
+               status = samdb_create_foreign_security_principal(
+                       d_state->sam_ctx, mem_ctx, r->in.sid, &memberdn);
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
        } else {
-               DEBUG(0, ("samdb_search returned %d: %s\n", ret, ldb_errstring(d_state->sam_ctx)));
+               DEBUG(0,("Found %d records matching sid %s\n",
+                        ret, dom_sid_string(mem_ctx, r->in.sid)));
+               return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
        if (memberdn == NULL) {
@@ -2720,19 +2460,28 @@ static NTSTATUS dcesrv_samr_AddAliasMember(struct dcesrv_call_state *dce_call, T
 
        mod->dn = talloc_reference(mem_ctx, a_state->account_dn);
 
-       if (samdb_msg_add_addval(d_state->sam_ctx, mem_ctx, mod, "member",
-                                ldb_dn_alloc_linearized(mem_ctx, memberdn)) != 0)
-               return NT_STATUS_UNSUCCESSFUL;
-
-       if (ldb_modify(a_state->sam_ctx, mod) != 0)
-               return NT_STATUS_UNSUCCESSFUL;
+       ret = samdb_msg_add_addval(d_state->sam_ctx, mem_ctx, mod, "member",
+                                ldb_dn_alloc_linearized(mem_ctx, memberdn));
+       if (ret != LDB_SUCCESS) {
+               return dsdb_ldb_err_to_ntstatus(ret);
+       }
 
-       return NT_STATUS_OK;
+       ret = ldb_modify(a_state->sam_ctx, mod);
+       switch (ret) {
+       case LDB_SUCCESS:
+               return NT_STATUS_OK;
+       case LDB_ERR_ENTRY_ALREADY_EXISTS:
+               return NT_STATUS_MEMBER_IN_GROUP;
+       case LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS:
+               return NT_STATUS_ACCESS_DENIED;
+       default:
+               return dsdb_ldb_err_to_ntstatus(ret);
+       }
 }
 
 
-/* 
-  samr_DeleteAliasMember 
+/*
+  samr_DeleteAliasMember
 */
 static NTSTATUS dcesrv_samr_DeleteAliasMember(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_DeleteAliasMember *r)
@@ -2742,6 +2491,7 @@ static NTSTATUS dcesrv_samr_DeleteAliasMember(struct dcesrv_call_state *dce_call
        struct samr_domain_state *d_state;
        struct ldb_message *mod;
        const char *memberdn;
+       int ret;
 
        DCESRV_PULL_HANDLE(h, r->in.alias_handle, SAMR_HANDLE_ALIAS);
 
@@ -2749,11 +2499,11 @@ static NTSTATUS dcesrv_samr_DeleteAliasMember(struct dcesrv_call_state *dce_call
        d_state = a_state->domain_state;
 
        memberdn = samdb_search_string(d_state->sam_ctx, mem_ctx, NULL,
-                                      "distinguishedName", "(objectSid=%s)", 
+                                      "distinguishedName", "(objectSid=%s)",
                                       ldap_encode_ndr_dom_sid(mem_ctx, r->in.sid));
-
-       if (memberdn == NULL)
+       if (memberdn == NULL) {
                return NT_STATUS_OBJECT_NAME_NOT_FOUND;
+       }
 
        mod = ldb_msg_new(mem_ctx);
        if (mod == NULL) {
@@ -2762,19 +2512,28 @@ static NTSTATUS dcesrv_samr_DeleteAliasMember(struct dcesrv_call_state *dce_call
 
        mod->dn = talloc_reference(mem_ctx, a_state->account_dn);
 
-       if (samdb_msg_add_delval(d_state->sam_ctx, mem_ctx, mod, "member",
-                                memberdn) != 0)
-               return NT_STATUS_UNSUCCESSFUL;
-
-       if (ldb_modify(a_state->sam_ctx, mod) != 0)
-               return NT_STATUS_UNSUCCESSFUL;
+       ret = samdb_msg_add_delval(d_state->sam_ctx, mem_ctx, mod, "member",
+                                                                memberdn);
+       if (ret != LDB_SUCCESS) {
+               return dsdb_ldb_err_to_ntstatus(ret);
+       }
 
-       return NT_STATUS_OK;
+       ret = ldb_modify(a_state->sam_ctx, mod);
+       switch (ret) {
+       case LDB_SUCCESS:
+               return NT_STATUS_OK;
+       case LDB_ERR_UNWILLING_TO_PERFORM:
+               return NT_STATUS_MEMBER_NOT_IN_GROUP;
+       case LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS:
+               return NT_STATUS_ACCESS_DENIED;
+       default:
+               return dsdb_ldb_err_to_ntstatus(ret);
+       }
 }
 
 
-/* 
-  samr_GetMembersInAlias 
+/*
+  samr_GetMembersInAlias
 */
 static NTSTATUS dcesrv_samr_GetMembersInAlias(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_GetMembersInAlias *r)
@@ -2782,66 +2541,46 @@ static NTSTATUS dcesrv_samr_GetMembersInAlias(struct dcesrv_call_state *dce_call
        struct dcesrv_handle *h;
        struct samr_account_state *a_state;
        struct samr_domain_state *d_state;
-       struct ldb_message **msgs;
-       struct lsa_SidPtr *sids;
-       struct ldb_message_element *el;
-       const char * const attrs[2] = { "member", NULL};
-       int ret;
+       struct lsa_SidPtr *array;
+       unsigned int i, num_members;
+       struct dom_sid *members;
+       NTSTATUS status;
 
        DCESRV_PULL_HANDLE(h, r->in.alias_handle, SAMR_HANDLE_ALIAS);
 
        a_state = h->data;
        d_state = a_state->domain_state;
 
-       ret = gendb_search_dn(d_state->sam_ctx, mem_ctx,
-                             a_state->account_dn, &msgs, attrs);
-
-       if (ret == -1) {
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
-       } else if (ret == 0) {
-               return NT_STATUS_OBJECT_NAME_NOT_FOUND;
-       } else if (ret != 1) {
-               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       status = dsdb_enum_group_mem(d_state->sam_ctx, mem_ctx,
+                                    a_state->account_dn, &members,
+                                    &num_members);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       r->out.sids->num_sids = 0;
-       r->out.sids->sids = NULL;
+       if (num_members == 0) {
+               r->out.sids->sids = NULL;
 
-       el = ldb_msg_find_element(msgs[0], "member");
-
-       if (el != NULL) {
-               int i;
-
-               sids = talloc_array(mem_ctx, struct lsa_SidPtr,
-                                     el->num_values);
-
-               if (sids == NULL)
-                       return NT_STATUS_NO_MEMORY;
-
-               for (i=0; i<el->num_values; i++) {
-                       struct ldb_message **msgs2;
-                       const char * const attrs2[2] = { "objectSid", NULL };
-                       ret = gendb_search_dn(a_state->sam_ctx, mem_ctx,
-                                             ldb_dn_from_ldb_val(mem_ctx, a_state->sam_ctx, &el->values[i]),
-                                             &msgs2, attrs2);
-                       if (ret != 1)
-                               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+               return NT_STATUS_OK;
+       }
 
-                       sids[i].sid = samdb_result_dom_sid(mem_ctx, msgs2[0],
-                                                          "objectSid");
+       array = talloc_array(mem_ctx, struct lsa_SidPtr, num_members);
+       if (array == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-                       if (sids[i].sid == NULL)
-                               return NT_STATUS_INTERNAL_DB_CORRUPTION;
-               }
-               r->out.sids->num_sids = el->num_values;
-               r->out.sids->sids = sids;
+       for (i=0; i<num_members; i++) {
+               array[i].sid = &members[i];
        }
 
+       r->out.sids->num_sids = num_members;
+       r->out.sids->sids = array;
+
        return NT_STATUS_OK;
 }
 
-/* 
-  samr_OpenUser 
+/*
+  samr_OpenUser
 */
 static NTSTATUS dcesrv_samr_OpenUser(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                              struct samr_OpenUser *r)
@@ -2871,20 +2610,20 @@ static NTSTATUS dcesrv_samr_OpenUser(struct dcesrv_call_state *dce_call, TALLOC_
        /* search for the user record */
        ret = gendb_search(d_state->sam_ctx,
                           mem_ctx, d_state->domain_dn, &msgs, attrs,
-                          "(&(objectSid=%s)(objectclass=user))", 
+                          "(&(objectSid=%s)(objectclass=user))",
                           ldap_encode_ndr_dom_sid(mem_ctx, sid));
        if (ret == 0) {
                return NT_STATUS_NO_SUCH_USER;
        }
        if (ret != 1) {
-               DEBUG(0,("Found %d records matching sid %s\n", ret, 
+               DEBUG(0,("Found %d records matching sid %s\n", ret,
                         dom_sid_string(mem_ctx, sid)));
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
-       account_name = samdb_result_string(msgs[0], "sAMAccountName", NULL);
+       account_name = ldb_msg_find_attr_as_string(msgs[0], "sAMAccountName", NULL);
        if (account_name == NULL) {
-               DEBUG(0,("sAMAccountName field missing for sid %s\n", 
+               DEBUG(0,("sAMAccountName field missing for sid %s\n",
                         dom_sid_string(mem_ctx, sid)));
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
@@ -2918,8 +2657,8 @@ static NTSTATUS dcesrv_samr_OpenUser(struct dcesrv_call_state *dce_call, TALLOC_
 }
 
 
-/* 
-  samr_DeleteUser 
+/*
+  samr_DeleteUser
 */
 static NTSTATUS dcesrv_samr_DeleteUser(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                struct samr_DeleteUser *r)
@@ -2935,21 +2674,22 @@ static NTSTATUS dcesrv_samr_DeleteUser(struct dcesrv_call_state *dce_call, TALLO
        a_state = h->data;
 
        ret = ldb_delete(a_state->sam_ctx, a_state->account_dn);
-       if (ret != 0) {
-               DEBUG(1, ("Failed to delete user: %s: %s\n", 
-                         ldb_dn_get_linearized(a_state->account_dn), 
+       if (ret != LDB_SUCCESS) {
+               DEBUG(1, ("Failed to delete user: %s: %s\n",
+                         ldb_dn_get_linearized(a_state->account_dn),
                          ldb_errstring(a_state->sam_ctx)));
-               return NT_STATUS_UNSUCCESSFUL;
+               return dsdb_ldb_err_to_ntstatus(ret);
        }
 
+       talloc_free(h);
        ZERO_STRUCTP(r->out.user_handle);
 
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_QueryUserInfo 
+/*
+  samr_QueryUserInfo
 */
 static NTSTATUS dcesrv_samr_QueryUserInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                   struct samr_QueryUserInfo *r)
@@ -2974,15 +2714,21 @@ static NTSTATUS dcesrv_samr_QueryUserInfo(struct dcesrv_call_state *dce_call, TA
        switch (r->in.level) {
        case 1:
        {
-               static const char * const attrs2[] = {"sAMAccountName", "displayName",
-                                                     "primaryroupID", "description",
-                                                     "comment", NULL};
+               static const char * const attrs2[] = {"sAMAccountName",
+                                                     "displayName",
+                                                     "primaryroupID",
+                                                     "description",
+                                                     "comment",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 2:
        {
-               static const char * const attrs2[] = {"comment", "countryCode", "codePage", NULL};
+               static const char * const attrs2[] = {"comment",
+                                                     "countryCode",
+                                                     "codePage",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
@@ -3003,25 +2749,27 @@ static NTSTATUS dcesrv_samr_QueryUserInfo(struct dcesrv_call_state *dce_call, TA
                                                      "logonHours",
                                                      "badPwdCount",
                                                      "logonCount",
-                                                     "userAccountControl", NULL};
+                                                     "userAccountControl",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 4:
        {
-               static const char * const attrs2[] = {"logonHours", NULL};
+               static const char * const attrs2[] = {"logonHours",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 5:
        {
-               static const char * const attrs2[] = {"sAMAccountName", 
+               static const char * const attrs2[] = {"sAMAccountName",
                                                      "displayName",
                                                      "objectSid",
                                                      "primaryGroupID",
                                                      "homeDirectory",
                                                      "homeDrive",
-                                                     "scriptPath", 
+                                                     "scriptPath",
                                                      "profilePath",
                                                      "description",
                                                      "userWorkstations",
@@ -3039,73 +2787,92 @@ static NTSTATUS dcesrv_samr_QueryUserInfo(struct dcesrv_call_state *dce_call, TA
        }
        case 6:
        {
-               static const char * const attrs2[] = {"sAMAccountName", "displayName", NULL};
+               static const char * const attrs2[] = {"sAMAccountName",
+                                                     "displayName",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 7:
        {
-               static const char * const attrs2[] = {"sAMAccountName", NULL};
+               static const char * const attrs2[] = {"sAMAccountName",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 8:
        {
-               static const char * const attrs2[] = {"displayName", NULL};
+               static const char * const attrs2[] = {"displayName",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 9:
        {
-               static const char * const attrs2[] = {"primaryGroupID", NULL};
+               static const char * const attrs2[] = {"primaryGroupID",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 10:
        {
-               static const char * const attrs2[] = {"homeDirectory", "homeDrive", NULL};
+               static const char * const attrs2[] = {"homeDirectory",
+                                                     "homeDrive",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 11:
        {
-               static const char * const attrs2[] = {"scriptPath", NULL};
+               static const char * const attrs2[] = {"scriptPath",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 12:
        {
-               static const char * const attrs2[] = {"profilePath", NULL};
+               static const char * const attrs2[] = {"profilePath",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 13:
        {
-               static const char * const attrs2[] = {"description", NULL};
+               static const char * const attrs2[] = {"description",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 14:
        {
-               static const char * const attrs2[] = {"userWorkstations", NULL};
+               static const char * const attrs2[] = {"userWorkstations",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 16:
        {
-               static const char * const attrs2[] = {"userAccountControl", "pwdLastSet", NULL};
+               static const char * const attrs2[] = {"userAccountControl",
+                                                     "pwdLastSet",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
        case 17:
        {
-               static const char * const attrs2[] = {"accountExpires", NULL};
+               static const char * const attrs2[] = {"accountExpires",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
+       case 18:
+       {
+               return NT_STATUS_NOT_SUPPORTED;
+       }
        case 20:
        {
-               static const char * const attrs2[] = {"userParameters", NULL};
+               static const char * const attrs2[] = {"userParameters",
+                                                     NULL};
                attrs = attrs2;
                break;
        }
@@ -3137,11 +2904,25 @@ static NTSTATUS dcesrv_samr_QueryUserInfo(struct dcesrv_call_state *dce_call, TA
                attrs = attrs2;
                break;
        }
+       case 23:
+       case 24:
+       case 25:
+       case 26:
+       {
+               return NT_STATUS_NOT_SUPPORTED;
+       }
+       default:
+       {
+               return NT_STATUS_INVALID_INFO_CLASS;
+       }
        }
 
        /* pull all the user attributes */
        ret = gendb_search_dn(a_state->sam_ctx, mem_ctx,
-                             a_state->account_dn ,&res, attrs);
+                             a_state->account_dn, &res, attrs);
+       if (ret == 0) {
+               return NT_STATUS_NO_SUCH_USER;
+       }
        if (ret != 1) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
@@ -3285,14 +3066,19 @@ static NTSTATUS dcesrv_samr_QueryUserInfo(struct dcesrv_call_state *dce_call, TA
                QUERY_RID   (msg, info21.rid,                  "objectSid");
                QUERY_UINT  (msg, info21.primary_gid,          "primaryGroupID");
                QUERY_AFLAGS(msg, info21.acct_flags,           "userAccountControl");
-               info->info21.fields_present = 0x00FFFFFF;
+               info->info21.fields_present = 0x08FFFFFF;
                QUERY_LHOURS(msg, info21.logon_hours,          "logonHours");
                QUERY_UINT  (msg, info21.bad_password_count,   "badPwdCount");
                QUERY_UINT  (msg, info21.logon_count,          "logonCount");
+               if ((info->info21.acct_flags & ACB_PW_EXPIRED) != 0) {
+                       info->info21.password_expired = PASS_MUST_CHANGE_AT_NEXT_LOGON;
+               } else {
+                       info->info21.password_expired = PASS_DONT_CHANGE_AT_NEXT_LOGON;
+               }
                QUERY_UINT  (msg, info21.country_code,         "countryCode");
                QUERY_UINT  (msg, info21.code_page,            "codePage");
                break;
-               
+
 
        default:
                talloc_free(info);
@@ -3305,8 +3091,8 @@ static NTSTATUS dcesrv_samr_QueryUserInfo(struct dcesrv_call_state *dce_call, TA
 }
 
 
-/* 
-  samr_SetUserInfo 
+/*
+  samr_SetUserInfo
 */
 static NTSTATUS dcesrv_samr_SetUserInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                 struct samr_SetUserInfo *r)
@@ -3345,6 +3131,7 @@ static NTSTATUS dcesrv_samr_SetUserInfo(struct dcesrv_call_state *dce_call, TALL
                break;
 
        case 6:
+               SET_STRING(msg, info6.account_name,     "samAccountName");
                SET_STRING(msg, info6.full_name,        "displayName");
                break;
 
@@ -3389,88 +3176,216 @@ static NTSTATUS dcesrv_samr_SetUserInfo(struct dcesrv_call_state *dce_call, TALL
                SET_UINT64(msg, info17.acct_expiry,     "accountExpires");
                break;
 
+       case 18:
+               status = samr_set_password_buffers(dce_call,
+                                                  a_state->sam_ctx,
+                                                  a_state->account_dn,
+                                                  a_state->domain_state->domain_dn,
+                                                  mem_ctx,
+                                                  r->in.info->info18.lm_pwd_active ? r->in.info->info18.lm_pwd.hash : NULL,
+                                                  r->in.info->info18.nt_pwd_active ? r->in.info->info18.nt_pwd.hash : NULL);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+
+               if (r->in.info->info18.password_expired > 0) {
+                       struct ldb_message_element *set_el;
+                       if (samdb_msg_add_uint64(sam_ctx, mem_ctx, msg, "pwdLastSet", 0) != LDB_SUCCESS) {
+                               return NT_STATUS_NO_MEMORY;
+                       }
+                       set_el = ldb_msg_find_element(msg, "pwdLastSet");
+                       set_el->flags = LDB_FLAG_MOD_REPLACE;
+               }
+               break;
+
        case 20:
                SET_PARAMETERS(msg, info20.parameters,      "userParameters");
                break;
 
        case 21:
-#define IFSET(bit) if (bit & r->in.info->info21.fields_present)        
+               if (r->in.info->info21.fields_present == 0)
+                       return NT_STATUS_INVALID_PARAMETER;
+
+#define IFSET(bit) if (bit & r->in.info->info21.fields_present)
+               IFSET(SAMR_FIELD_LAST_LOGON)
+                       SET_UINT64(msg, info21.last_logon,     "lastLogon");
+               IFSET(SAMR_FIELD_LAST_LOGOFF)
+                       SET_UINT64(msg, info21.last_logoff,    "lastLogoff");
                IFSET(SAMR_FIELD_ACCT_EXPIRY)
-                       SET_UINT64(msg, info21.acct_expiry,    "accountExpires");       
-               IFSET(SAMR_FIELD_ACCOUNT_NAME)         
+                       SET_UINT64(msg, info21.acct_expiry,    "accountExpires");
+               IFSET(SAMR_FIELD_ACCOUNT_NAME)
                        SET_STRING(msg, info21.account_name,   "samAccountName");
-               IFSET(SAMR_FIELD_FULL_NAME) 
+               IFSET(SAMR_FIELD_FULL_NAME)
                        SET_STRING(msg, info21.full_name,      "displayName");
-               IFSET(SAMR_FIELD_DESCRIPTION)
-                       SET_STRING(msg, info21.description,    "description");
-               IFSET(SAMR_FIELD_COMMENT)
-                       SET_STRING(msg, info21.comment,        "comment");
-               IFSET(SAMR_FIELD_LOGON_SCRIPT)
-                       SET_STRING(msg, info21.logon_script,   "scriptPath");
-               IFSET(SAMR_FIELD_PROFILE_PATH)
-                       SET_STRING(msg, info21.profile_path,   "profilePath");
                IFSET(SAMR_FIELD_HOME_DIRECTORY)
                        SET_STRING(msg, info21.home_directory, "homeDirectory");
                IFSET(SAMR_FIELD_HOME_DRIVE)
                        SET_STRING(msg, info21.home_drive,     "homeDrive");
+               IFSET(SAMR_FIELD_LOGON_SCRIPT)
+                       SET_STRING(msg, info21.logon_script,   "scriptPath");
+               IFSET(SAMR_FIELD_PROFILE_PATH)
+                       SET_STRING(msg, info21.profile_path,   "profilePath");
+               IFSET(SAMR_FIELD_DESCRIPTION)
+                       SET_STRING(msg, info21.description,    "description");
                IFSET(SAMR_FIELD_WORKSTATIONS)
                        SET_STRING(msg, info21.workstations,   "userWorkstations");
-               IFSET(SAMR_FIELD_LOGON_HOURS)
-                       SET_LHOURS(msg, info21.logon_hours,    "logonHours");
+               IFSET(SAMR_FIELD_COMMENT)
+                       SET_STRING(msg, info21.comment,        "comment");
+               IFSET(SAMR_FIELD_PARAMETERS)
+                       SET_PARAMETERS(msg, info21.parameters, "userParameters");
+               IFSET(SAMR_FIELD_PRIMARY_GID)
+                       SET_UINT(msg, info21.primary_gid,      "primaryGroupID");
                IFSET(SAMR_FIELD_ACCT_FLAGS)
                        SET_AFLAGS(msg, info21.acct_flags,     "userAccountControl");
-               IFSET(SAMR_FIELD_PARAMETERS)   
-                       SET_PARAMETERS(msg, info21.parameters, "userParameters");
+               IFSET(SAMR_FIELD_LOGON_HOURS)
+                       SET_LHOURS(msg, info21.logon_hours,    "logonHours");
+               IFSET(SAMR_FIELD_BAD_PWD_COUNT)
+                       SET_UINT  (msg, info21.bad_password_count, "badPwdCount");
+               IFSET(SAMR_FIELD_NUM_LOGONS)
+                       SET_UINT  (msg, info21.logon_count,    "logonCount");
                IFSET(SAMR_FIELD_COUNTRY_CODE)
                        SET_UINT  (msg, info21.country_code,   "countryCode");
                IFSET(SAMR_FIELD_CODE_PAGE)
-                       SET_UINT  (msg, info21.code_page,      "codePage");     
+                       SET_UINT  (msg, info21.code_page,      "codePage");
+
+               /* password change fields */
+               IFSET(SAMR_FIELD_LAST_PWD_CHANGE)
+                       return NT_STATUS_ACCESS_DENIED;
+
+               IFSET((SAMR_FIELD_LM_PASSWORD_PRESENT
+                                       | SAMR_FIELD_NT_PASSWORD_PRESENT)) {
+                       uint8_t *lm_pwd_hash = NULL, *nt_pwd_hash = NULL;
+
+                       if (r->in.info->info21.lm_password_set) {
+                               if ((r->in.info->info21.lm_owf_password.length != 16)
+                                || (r->in.info->info21.lm_owf_password.size != 16)) {
+                                       return NT_STATUS_INVALID_PARAMETER;
+                               }
+
+                               lm_pwd_hash = (uint8_t *) r->in.info->info21.lm_owf_password.array;
+                       }
+                       if (r->in.info->info21.nt_password_set) {
+                               if ((r->in.info->info21.nt_owf_password.length != 16)
+                                || (r->in.info->info21.nt_owf_password.size != 16)) {
+                                       return NT_STATUS_INVALID_PARAMETER;
+                               }
+
+                               nt_pwd_hash = (uint8_t *) r->in.info->info21.nt_owf_password.array;
+                       }
+                       status = samr_set_password_buffers(dce_call,
+                                                          a_state->sam_ctx,
+                                                          a_state->account_dn,
+                                                          a_state->domain_state->domain_dn,
+                                                          mem_ctx,
+                                                          lm_pwd_hash,
+                                                          nt_pwd_hash);
+                       if (!NT_STATUS_IS_OK(status)) {
+                               return status;
+                       }
+               }
+
+
+               IFSET(SAMR_FIELD_EXPIRED_FLAG) {
+                       NTTIME t = 0;
+                       struct ldb_message_element *set_el;
+                       if (r->in.info->info21.password_expired
+                                       == PASS_DONT_CHANGE_AT_NEXT_LOGON) {
+                               unix_to_nt_time(&t, time(NULL));
+                       }
+                       if (samdb_msg_add_uint64(sam_ctx, mem_ctx, msg,
+                                                "pwdLastSet", t) != LDB_SUCCESS) {
+                               return NT_STATUS_NO_MEMORY;
+                       }
+                       set_el = ldb_msg_find_element(msg, "pwdLastSet");
+                       set_el->flags = LDB_FLAG_MOD_REPLACE;
+               }
 #undef IFSET
                break;
 
        case 23:
+               if (r->in.info->info23.info.fields_present == 0)
+                       return NT_STATUS_INVALID_PARAMETER;
+
 #define IFSET(bit) if (bit & r->in.info->info23.info.fields_present)
+               IFSET(SAMR_FIELD_LAST_LOGON)
+                       SET_UINT64(msg, info23.info.last_logon,     "lastLogon");
+               IFSET(SAMR_FIELD_LAST_LOGOFF)
+                       SET_UINT64(msg, info23.info.last_logoff,    "lastLogoff");
                IFSET(SAMR_FIELD_ACCT_EXPIRY)
-                       SET_UINT64(msg, info23.info.acct_expiry,  "accountExpires");    
-               IFSET(SAMR_FIELD_ACCOUNT_NAME)         
-                       SET_STRING(msg, info23.info.account_name, "samAccountName");
-               IFSET(SAMR_FIELD_FULL_NAME)         
-                       SET_STRING(msg, info23.info.full_name,    "displayName");
-               IFSET(SAMR_FIELD_DESCRIPTION)  
-                       SET_STRING(msg, info23.info.description,  "description");
-               IFSET(SAMR_FIELD_COMMENT)      
-                       SET_STRING(msg, info23.info.comment,      "comment");
-               IFSET(SAMR_FIELD_LOGON_SCRIPT) 
-                       SET_STRING(msg, info23.info.logon_script, "scriptPath");
-               IFSET(SAMR_FIELD_PROFILE_PATH)      
-                       SET_STRING(msg, info23.info.profile_path, "profilePath");
-               IFSET(SAMR_FIELD_WORKSTATIONS)  
-                       SET_STRING(msg, info23.info.workstations, "userWorkstations");
-               IFSET(SAMR_FIELD_LOGON_HOURS)  
-                       SET_LHOURS(msg, info23.info.logon_hours,  "logonHours");
-               IFSET(SAMR_FIELD_ACCT_FLAGS)     
-                       SET_AFLAGS(msg, info23.info.acct_flags,   "userAccountControl");
-               IFSET(SAMR_FIELD_PARAMETERS)     
+                       SET_UINT64(msg, info23.info.acct_expiry,    "accountExpires");
+               IFSET(SAMR_FIELD_ACCOUNT_NAME)
+                       SET_STRING(msg, info23.info.account_name,   "samAccountName");
+               IFSET(SAMR_FIELD_FULL_NAME)
+                       SET_STRING(msg, info23.info.full_name,      "displayName");
+               IFSET(SAMR_FIELD_HOME_DIRECTORY)
+                       SET_STRING(msg, info23.info.home_directory, "homeDirectory");
+               IFSET(SAMR_FIELD_HOME_DRIVE)
+                       SET_STRING(msg, info23.info.home_drive,     "homeDrive");
+               IFSET(SAMR_FIELD_LOGON_SCRIPT)
+                       SET_STRING(msg, info23.info.logon_script,   "scriptPath");
+               IFSET(SAMR_FIELD_PROFILE_PATH)
+                       SET_STRING(msg, info23.info.profile_path,   "profilePath");
+               IFSET(SAMR_FIELD_DESCRIPTION)
+                       SET_STRING(msg, info23.info.description,    "description");
+               IFSET(SAMR_FIELD_WORKSTATIONS)
+                       SET_STRING(msg, info23.info.workstations,   "userWorkstations");
+               IFSET(SAMR_FIELD_COMMENT)
+                       SET_STRING(msg, info23.info.comment,        "comment");
+               IFSET(SAMR_FIELD_PARAMETERS)
                        SET_PARAMETERS(msg, info23.info.parameters, "userParameters");
-               IFSET(SAMR_FIELD_COUNTRY_CODE) 
-                       SET_UINT  (msg, info23.info.country_code, "countryCode");
-               IFSET(SAMR_FIELD_CODE_PAGE)    
-                       SET_UINT  (msg, info23.info.code_page,    "codePage");
+               IFSET(SAMR_FIELD_PRIMARY_GID)
+                       SET_UINT(msg, info23.info.primary_gid,      "primaryGroupID");
+               IFSET(SAMR_FIELD_ACCT_FLAGS)
+                       SET_AFLAGS(msg, info23.info.acct_flags,     "userAccountControl");
+               IFSET(SAMR_FIELD_LOGON_HOURS)
+                       SET_LHOURS(msg, info23.info.logon_hours,    "logonHours");
+               IFSET(SAMR_FIELD_BAD_PWD_COUNT)
+                       SET_UINT  (msg, info23.info.bad_password_count, "badPwdCount");
+               IFSET(SAMR_FIELD_NUM_LOGONS)
+                       SET_UINT  (msg, info23.info.logon_count,    "logonCount");
+
+               IFSET(SAMR_FIELD_COUNTRY_CODE)
+                       SET_UINT  (msg, info23.info.country_code,   "countryCode");
+               IFSET(SAMR_FIELD_CODE_PAGE)
+                       SET_UINT  (msg, info23.info.code_page,      "codePage");
+
+               /* password change fields */
+               IFSET(SAMR_FIELD_LAST_PWD_CHANGE)
+                       return NT_STATUS_ACCESS_DENIED;
+
                IFSET(SAMR_FIELD_NT_PASSWORD_PRESENT) {
                        status = samr_set_password(dce_call,
                                                   a_state->sam_ctx,
                                                   a_state->account_dn,
                                                   a_state->domain_state->domain_dn,
-                                                  mem_ctx, msg, 
+                                                  mem_ctx,
                                                   &r->in.info->info23.password);
                } else IFSET(SAMR_FIELD_LM_PASSWORD_PRESENT) {
                        status = samr_set_password(dce_call,
                                                   a_state->sam_ctx,
                                                   a_state->account_dn,
                                                   a_state->domain_state->domain_dn,
-                                                  mem_ctx, msg, 
+                                                  mem_ctx,
                                                   &r->in.info->info23.password);
                }
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+
+               IFSET(SAMR_FIELD_EXPIRED_FLAG) {
+                       NTTIME t = 0;
+                       struct ldb_message_element *set_el;
+                       if (r->in.info->info23.info.password_expired
+                                       == PASS_DONT_CHANGE_AT_NEXT_LOGON) {
+                               unix_to_nt_time(&t, time(NULL));
+                       }
+                       if (samdb_msg_add_uint64(sam_ctx, mem_ctx, msg,
+                                                "pwdLastSet", t) != LDB_SUCCESS) {
+                               return NT_STATUS_NO_MEMORY;
+                       }
+                       set_el = ldb_msg_find_element(msg, "pwdLastSet");
+                       set_el->flags = LDB_FLAG_MOD_REPLACE;
+               }
 #undef IFSET
                break;
 
@@ -3480,53 +3395,105 @@ static NTSTATUS dcesrv_samr_SetUserInfo(struct dcesrv_call_state *dce_call, TALL
                                           a_state->sam_ctx,
                                           a_state->account_dn,
                                           a_state->domain_state->domain_dn,
-                                          mem_ctx, msg, 
+                                          mem_ctx,
                                           &r->in.info->info24.password);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+
+               if (r->in.info->info24.password_expired > 0) {
+                       struct ldb_message_element *set_el;
+                       if (samdb_msg_add_uint64(sam_ctx, mem_ctx, msg, "pwdLastSet", 0) != LDB_SUCCESS) {
+                               return NT_STATUS_NO_MEMORY;
+                       }
+                       set_el = ldb_msg_find_element(msg, "pwdLastSet");
+                       set_el->flags = LDB_FLAG_MOD_REPLACE;
+               }
                break;
 
        case 25:
+               if (r->in.info->info25.info.fields_present == 0)
+                       return NT_STATUS_INVALID_PARAMETER;
+
 #define IFSET(bit) if (bit & r->in.info->info25.info.fields_present)
+               IFSET(SAMR_FIELD_LAST_LOGON)
+                       SET_UINT64(msg, info25.info.last_logon,     "lastLogon");
+               IFSET(SAMR_FIELD_LAST_LOGOFF)
+                       SET_UINT64(msg, info25.info.last_logoff,    "lastLogoff");
                IFSET(SAMR_FIELD_ACCT_EXPIRY)
-                       SET_UINT64(msg, info25.info.acct_expiry,  "accountExpires");    
-               IFSET(SAMR_FIELD_ACCOUNT_NAME)         
-                       SET_STRING(msg, info25.info.account_name, "samAccountName");
-               IFSET(SAMR_FIELD_FULL_NAME)         
-                       SET_STRING(msg, info25.info.full_name,    "displayName");
-               IFSET(SAMR_FIELD_DESCRIPTION)  
-                       SET_STRING(msg, info25.info.description,  "description");
-               IFSET(SAMR_FIELD_COMMENT)      
-                       SET_STRING(msg, info25.info.comment,      "comment");
-               IFSET(SAMR_FIELD_LOGON_SCRIPT) 
-                       SET_STRING(msg, info25.info.logon_script, "scriptPath");
-               IFSET(SAMR_FIELD_PROFILE_PATH)      
-                       SET_STRING(msg, info25.info.profile_path, "profilePath");
-               IFSET(SAMR_FIELD_WORKSTATIONS)  
-                       SET_STRING(msg, info25.info.workstations, "userWorkstations");
-               IFSET(SAMR_FIELD_LOGON_HOURS)  
-                       SET_LHOURS(msg, info25.info.logon_hours,  "logonHours");
-               IFSET(SAMR_FIELD_ACCT_FLAGS)     
-                       SET_AFLAGS(msg, info25.info.acct_flags,   "userAccountControl");
-               IFSET(SAMR_FIELD_PARAMETERS)     
+                       SET_UINT64(msg, info25.info.acct_expiry,    "accountExpires");
+               IFSET(SAMR_FIELD_ACCOUNT_NAME)
+                       SET_STRING(msg, info25.info.account_name,   "samAccountName");
+               IFSET(SAMR_FIELD_FULL_NAME)
+                       SET_STRING(msg, info25.info.full_name,      "displayName");
+               IFSET(SAMR_FIELD_HOME_DIRECTORY)
+                       SET_STRING(msg, info25.info.home_directory, "homeDirectory");
+               IFSET(SAMR_FIELD_HOME_DRIVE)
+                       SET_STRING(msg, info25.info.home_drive,     "homeDrive");
+               IFSET(SAMR_FIELD_LOGON_SCRIPT)
+                       SET_STRING(msg, info25.info.logon_script,   "scriptPath");
+               IFSET(SAMR_FIELD_PROFILE_PATH)
+                       SET_STRING(msg, info25.info.profile_path,   "profilePath");
+               IFSET(SAMR_FIELD_DESCRIPTION)
+                       SET_STRING(msg, info25.info.description,    "description");
+               IFSET(SAMR_FIELD_WORKSTATIONS)
+                       SET_STRING(msg, info25.info.workstations,   "userWorkstations");
+               IFSET(SAMR_FIELD_COMMENT)
+                       SET_STRING(msg, info25.info.comment,        "comment");
+               IFSET(SAMR_FIELD_PARAMETERS)
                        SET_PARAMETERS(msg, info25.info.parameters, "userParameters");
-               IFSET(SAMR_FIELD_COUNTRY_CODE) 
-                       SET_UINT  (msg, info25.info.country_code, "countryCode");
-               IFSET(SAMR_FIELD_CODE_PAGE)    
-                       SET_UINT  (msg, info25.info.code_page,    "codePage");
+               IFSET(SAMR_FIELD_PRIMARY_GID)
+                       SET_UINT(msg, info25.info.primary_gid,      "primaryGroupID");
+               IFSET(SAMR_FIELD_ACCT_FLAGS)
+                       SET_AFLAGS(msg, info25.info.acct_flags,     "userAccountControl");
+               IFSET(SAMR_FIELD_LOGON_HOURS)
+                       SET_LHOURS(msg, info25.info.logon_hours,    "logonHours");
+               IFSET(SAMR_FIELD_BAD_PWD_COUNT)
+                       SET_UINT  (msg, info25.info.bad_password_count, "badPwdCount");
+               IFSET(SAMR_FIELD_NUM_LOGONS)
+                       SET_UINT  (msg, info25.info.logon_count,    "logonCount");
+               IFSET(SAMR_FIELD_COUNTRY_CODE)
+                       SET_UINT  (msg, info25.info.country_code,   "countryCode");
+               IFSET(SAMR_FIELD_CODE_PAGE)
+                       SET_UINT  (msg, info25.info.code_page,      "codePage");
+
+               /* password change fields */
+               IFSET(SAMR_FIELD_LAST_PWD_CHANGE)
+                       return NT_STATUS_ACCESS_DENIED;
+
                IFSET(SAMR_FIELD_NT_PASSWORD_PRESENT) {
                        status = samr_set_password_ex(dce_call,
                                                      a_state->sam_ctx,
                                                      a_state->account_dn,
                                                      a_state->domain_state->domain_dn,
-                                                     mem_ctx, msg, 
+                                                     mem_ctx,
                                                      &r->in.info->info25.password);
                } else IFSET(SAMR_FIELD_LM_PASSWORD_PRESENT) {
                        status = samr_set_password_ex(dce_call,
                                                      a_state->sam_ctx,
                                                      a_state->account_dn,
                                                      a_state->domain_state->domain_dn,
-                                                     mem_ctx, msg, 
+                                                     mem_ctx,
                                                      &r->in.info->info25.password);
                }
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+
+               IFSET(SAMR_FIELD_EXPIRED_FLAG) {
+                       NTTIME t = 0;
+                       struct ldb_message_element *set_el;
+                       if (r->in.info->info25.info.password_expired
+                                       == PASS_DONT_CHANGE_AT_NEXT_LOGON) {
+                               unix_to_nt_time(&t, time(NULL));
+                       }
+                       if (samdb_msg_add_uint64(sam_ctx, mem_ctx, msg,
+                                                "pwdLastSet", t) != LDB_SUCCESS) {
+                               return NT_STATUS_NO_MEMORY;
+                       }
+                       set_el = ldb_msg_find_element(msg, "pwdLastSet");
+                       set_el->flags = LDB_FLAG_MOD_REPLACE;
+               }
 #undef IFSET
                break;
 
@@ -3536,10 +3503,21 @@ static NTSTATUS dcesrv_samr_SetUserInfo(struct dcesrv_call_state *dce_call, TALL
                                              a_state->sam_ctx,
                                              a_state->account_dn,
                                              a_state->domain_state->domain_dn,
-                                             mem_ctx, msg, 
+                                             mem_ctx,
                                              &r->in.info->info26.password);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+
+               if (r->in.info->info26.password_expired > 0) {
+                       struct ldb_message_element *set_el;
+                       if (samdb_msg_add_uint64(sam_ctx, mem_ctx, msg, "pwdLastSet", 0) != LDB_SUCCESS) {
+                               return NT_STATUS_NO_MEMORY;
+                       }
+                       set_el = ldb_msg_find_element(msg, "pwdLastSet");
+                       set_el->flags = LDB_FLAG_MOD_REPLACE;
+               }
                break;
-               
 
        default:
                /* many info classes are not valid for SetUserInfo */
@@ -3551,22 +3529,23 @@ static NTSTATUS dcesrv_samr_SetUserInfo(struct dcesrv_call_state *dce_call, TALL
        }
 
        /* modify the samdb record */
-       ret = ldb_modify(a_state->sam_ctx, msg);
-       if (ret != 0) {
-               DEBUG(1,("Failed to modify record %s: %s\n",
-                        ldb_dn_get_linearized(a_state->account_dn),
-                        ldb_errstring(a_state->sam_ctx)));
-
-               /* we really need samdb.c to return NTSTATUS */
-               return NT_STATUS_UNSUCCESSFUL;
+       if (msg->num_elements > 0) {
+               ret = ldb_modify(a_state->sam_ctx, msg);
+               if (ret != LDB_SUCCESS) {
+                       DEBUG(1,("Failed to modify record %s: %s\n",
+                                ldb_dn_get_linearized(a_state->account_dn),
+                                ldb_errstring(a_state->sam_ctx)));
+
+                       return dsdb_ldb_err_to_ntstatus(ret);
+               }
        }
 
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_GetGroupsForUser 
+/*
+  samr_GetGroupsForUser
 */
 static NTSTATUS dcesrv_samr_GetGroupsForUser(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_GetGroupsForUser *r)
@@ -3577,17 +3556,19 @@ static NTSTATUS dcesrv_samr_GetGroupsForUser(struct dcesrv_call_state *dce_call,
        struct ldb_message **res;
        const char * const attrs[2] = { "objectSid", NULL };
        struct samr_RidWithAttributeArray *array;
-       int count;
+       int i, count;
 
        DCESRV_PULL_HANDLE(h, r->in.user_handle, SAMR_HANDLE_USER);
 
        a_state = h->data;
        d_state = a_state->domain_state;
 
-       count = samdb_search_domain(a_state->sam_ctx, mem_ctx, d_state->domain_dn, &res,
+       count = samdb_search_domain(a_state->sam_ctx, mem_ctx,
+                                   d_state->domain_dn, &res,
                                    attrs, d_state->domain_sid,
-                                   "(&(member=%s)(grouptype=%d)(objectclass=group))",
+                                   "(&(member=%s)(|(grouptype=%d)(grouptype=%d))(objectclass=group))",
                                    ldb_dn_get_linearized(a_state->account_dn),
+                                   GTYPE_SECURITY_UNIVERSAL_GROUP,
                                    GTYPE_SECURITY_GLOBAL_GROUP);
        if (count < 0)
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
@@ -3599,29 +3580,33 @@ static NTSTATUS dcesrv_samr_GetGroupsForUser(struct dcesrv_call_state *dce_call,
        array->count = 0;
        array->rids = NULL;
 
-       if (count > 0) {
-               int i;
-               array->rids = talloc_array(mem_ctx, struct samr_RidWithAttribute,
-                                           count);
-
-               if (array->rids == NULL)
-                       return NT_STATUS_NO_MEMORY;
+       array->rids = talloc_array(mem_ctx, struct samr_RidWithAttribute,
+                                           count + 1);
+       if (array->rids == NULL)
+               return NT_STATUS_NO_MEMORY;
 
-               for (i=0; i<count; i++) {
-                       struct dom_sid *group_sid;
+       /* Adds the primary group */
+       array->rids[0].rid = samdb_search_uint(a_state->sam_ctx, mem_ctx,
+                                              ~0, a_state->account_dn,
+                                              "primaryGroupID", NULL);
+       array->rids[0].attributes = SE_GROUP_MANDATORY
+                       | SE_GROUP_ENABLED_BY_DEFAULT | SE_GROUP_ENABLED;
+       array->count += 1;
 
-                       group_sid = samdb_result_dom_sid(mem_ctx, res[i],
-                                                        "objectSid");
-                       if (group_sid == NULL) {
-                               DEBUG(0, ("Couldn't find objectSid attrib\n"));
-                               continue;
-                       }
+       /* Adds the additional groups */
+       for (i = 0; i < count; i++) {
+               struct dom_sid *group_sid;
 
-                       array->rids[array->count].rid =
-                               group_sid->sub_auths[group_sid->num_auths-1];
-                       array->rids[array->count].attributes = SE_GROUP_MANDATORY | SE_GROUP_ENABLED_BY_DEFAULT | SE_GROUP_ENABLED;
-                       array->count += 1;
+               group_sid = samdb_result_dom_sid(mem_ctx, res[i], "objectSid");
+               if (group_sid == NULL) {
+                       return NT_STATUS_INTERNAL_DB_CORRUPTION;
                }
+
+               array->rids[i + 1].rid =
+                       group_sid->sub_auths[group_sid->num_auths-1];
+               array->rids[i + 1].attributes = SE_GROUP_MANDATORY
+                       | SE_GROUP_ENABLED_BY_DEFAULT | SE_GROUP_ENABLED;
+               array->count += 1;
        }
 
        *r->out.rids = array;
@@ -3630,23 +3615,26 @@ static NTSTATUS dcesrv_samr_GetGroupsForUser(struct dcesrv_call_state *dce_call,
 }
 
 
-/* 
-  samr_QueryDisplayInfo 
+/*
+  samr_QueryDisplayInfo
 */
 static NTSTATUS dcesrv_samr_QueryDisplayInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_QueryDisplayInfo *r)
 {
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
-       struct ldb_message **res;
-       int ldb_cnt, count, i;
-       const char * const attrs[] = { "objectSid", "sAMAccountName", "displayName",
-                                      "description", "userAccountControl", "pwdLastSet", NULL };
+       struct ldb_result *res;
+       unsigned int i;
+       uint32_t count;
+       const char * const attrs[] = { "objectSid", "sAMAccountName",
+               "displayName", "description", "userAccountControl",
+               "pwdLastSet", NULL };
        struct samr_DispEntryFull *entriesFull = NULL;
        struct samr_DispEntryFullGroup *entriesFullGroup = NULL;
        struct samr_DispEntryAscii *entriesAscii = NULL;
-       struct samr_DispEntryGeneral * entriesGeneral = NULL;
+       struct samr_DispEntryGeneral *entriesGeneral = NULL;
        const char *filter;
+       int ret;
 
        DCESRV_PULL_HANDLE(h, r->in.domain_handle, SAMR_HANDLE_DOMAIN);
 
@@ -3656,57 +3644,58 @@ static NTSTATUS dcesrv_samr_QueryDisplayInfo(struct dcesrv_call_state *dce_call,
        case 1:
        case 4:
                filter = talloc_asprintf(mem_ctx, "(&(objectclass=user)"
-                                        "(sAMAccountType=%u))",
+                                        "(sAMAccountType=%d))",
                                         ATYPE_NORMAL_ACCOUNT);
                break;
        case 2:
                filter = talloc_asprintf(mem_ctx, "(&(objectclass=user)"
-                                        "(sAMAccountType=%u))",
+                                        "(sAMAccountType=%d))",
                                         ATYPE_WORKSTATION_TRUST);
                break;
        case 3:
        case 5:
-               filter = talloc_asprintf(mem_ctx, "(&(grouptype=%d)"
-                                        "(objectclass=group))",
+               filter = talloc_asprintf(mem_ctx,
+                                        "(&(|(groupType=%d)(groupType=%d))"
+                                        "(objectClass=group))",
+                                        GTYPE_SECURITY_UNIVERSAL_GROUP,
                                         GTYPE_SECURITY_GLOBAL_GROUP);
                break;
        default:
                return NT_STATUS_INVALID_INFO_CLASS;
        }
 
-       /* search for all requested objects in this domain. This could
+       /* search for all requested objects in all domains. This could
           possibly be cached and resumed based on resume_key */
-       ldb_cnt = samdb_search_domain(d_state->sam_ctx, mem_ctx,
-                                     d_state->domain_dn, &res, attrs,
-                                     d_state->domain_sid, "%s", filter);
-       if (ldb_cnt == -1) {
+       ret = dsdb_search(d_state->sam_ctx, mem_ctx, &res, NULL,
+                         LDB_SCOPE_SUBTREE, attrs, 0, "%s", filter);
+       if (ret != LDB_SUCCESS) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
-       if (ldb_cnt == 0 || r->in.max_entries == 0) {
+       if ((res->count == 0) || (r->in.max_entries == 0)) {
                return NT_STATUS_OK;
        }
 
        switch (r->in.level) {
        case 1:
                entriesGeneral = talloc_array(mem_ctx,
-                                               struct samr_DispEntryGeneral,
-                                               ldb_cnt);
+                                             struct samr_DispEntryGeneral,
+                                             res->count);
                break;
        case 2:
                entriesFull = talloc_array(mem_ctx,
-                                            struct samr_DispEntryFull,
-                                            ldb_cnt);
+                                          struct samr_DispEntryFull,
+                                          res->count);
                break;
        case 3:
                entriesFullGroup = talloc_array(mem_ctx,
-                                            struct samr_DispEntryFullGroup,
-                                            ldb_cnt);
+                                               struct samr_DispEntryFullGroup,
+                                               res->count);
                break;
        case 4:
        case 5:
                entriesAscii = talloc_array(mem_ctx,
-                                             struct samr_DispEntryAscii,
-                                             ldb_cnt);
+                                           struct samr_DispEntryAscii,
+                                           res->count);
                break;
        }
 
@@ -3716,10 +3705,10 @@ static NTSTATUS dcesrv_samr_QueryDisplayInfo(struct dcesrv_call_state *dce_call,
 
        count = 0;
 
-       for (i=0; i<ldb_cnt; i++) {
+       for (i = 0; i < res->count; i++) {
                struct dom_sid *objectsid;
 
-               objectsid = samdb_result_dom_sid(mem_ctx, res[i],
+               objectsid = samdb_result_dom_sid(mem_ctx, res->msgs[i],
                                                 "objectSid");
                if (objectsid == NULL)
                        continue;
@@ -3727,19 +3716,22 @@ static NTSTATUS dcesrv_samr_QueryDisplayInfo(struct dcesrv_call_state *dce_call,
                switch(r->in.level) {
                case 1:
                        entriesGeneral[count].idx = count + 1;
-                       entriesGeneral[count].rid = 
+                       entriesGeneral[count].rid =
                                objectsid->sub_auths[objectsid->num_auths-1];
                        entriesGeneral[count].acct_flags =
-                               samdb_result_acct_flags(d_state->sam_ctx, mem_ctx,
-                                                       res[i], 
+                               samdb_result_acct_flags(d_state->sam_ctx,
+                                                       mem_ctx,
+                                                       res->msgs[i],
                                                        d_state->domain_dn);
                        entriesGeneral[count].account_name.string =
-                               samdb_result_string(res[i],
-                                                   "sAMAccountName", "");
+                               ldb_msg_find_attr_as_string(res->msgs[i],
+                                                           "sAMAccountName", "");
                        entriesGeneral[count].full_name.string =
-                               samdb_result_string(res[i], "displayName", "");
+                               ldb_msg_find_attr_as_string(res->msgs[i],
+                                                           "displayName", "");
                        entriesGeneral[count].description.string =
-                               samdb_result_string(res[i], "description", "");
+                               ldb_msg_find_attr_as_string(res->msgs[i],
+                                                           "description", "");
                        break;
                case 2:
                        entriesFull[count].idx = count + 1;
@@ -3748,14 +3740,16 @@ static NTSTATUS dcesrv_samr_QueryDisplayInfo(struct dcesrv_call_state *dce_call,
 
                        /* No idea why we need to or in ACB_NORMAL here, but this is what Win2k3 seems to do... */
                        entriesFull[count].acct_flags =
-                               samdb_result_acct_flags(d_state->sam_ctx, mem_ctx,
-                                                       res[i], 
+                               samdb_result_acct_flags(d_state->sam_ctx,
+                                                       mem_ctx,
+                                                       res->msgs[i],
                                                        d_state->domain_dn) | ACB_NORMAL;
                        entriesFull[count].account_name.string =
-                               samdb_result_string(res[i], "sAMAccountName",
-                                                   "");
+                               ldb_msg_find_attr_as_string(res->msgs[i],
+                                                           "sAMAccountName", "");
                        entriesFull[count].description.string =
-                               samdb_result_string(res[i], "description", "");
+                               ldb_msg_find_attr_as_string(res->msgs[i],
+                                                           "description", "");
                        break;
                case 3:
                        entriesFullGroup[count].idx = count + 1;
@@ -3765,17 +3759,18 @@ static NTSTATUS dcesrv_samr_QueryDisplayInfo(struct dcesrv_call_state *dce_call,
                        entriesFullGroup[count].acct_flags
                                = SE_GROUP_MANDATORY | SE_GROUP_ENABLED_BY_DEFAULT | SE_GROUP_ENABLED;
                        entriesFullGroup[count].account_name.string =
-                               samdb_result_string(res[i], "sAMAccountName",
-                                                   "");
+                               ldb_msg_find_attr_as_string(res->msgs[i],
+                                                           "sAMAccountName", "");
                        entriesFullGroup[count].description.string =
-                               samdb_result_string(res[i], "description", "");
+                               ldb_msg_find_attr_as_string(res->msgs[i],
+                                                           "description", "");
                        break;
                case 4:
                case 5:
                        entriesAscii[count].idx = count + 1;
                        entriesAscii[count].account_name.string =
-                               samdb_result_string(res[i], "sAMAccountName",
-                                                   "");
+                               ldb_msg_find_attr_as_string(res->msgs[i],
+                                                           "sAMAccountName", "");
                        break;
                }
 
@@ -3845,8 +3840,8 @@ static NTSTATUS dcesrv_samr_QueryDisplayInfo(struct dcesrv_call_state *dce_call,
 }
 
 
-/* 
-  samr_GetDisplayEnumerationIndex 
+/*
+  samr_GetDisplayEnumerationIndex
 */
 static NTSTATUS dcesrv_samr_GetDisplayEnumerationIndex(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_GetDisplayEnumerationIndex *r)
@@ -3855,8 +3850,8 @@ static NTSTATUS dcesrv_samr_GetDisplayEnumerationIndex(struct dcesrv_call_state
 }
 
 
-/* 
-  samr_TestPrivateFunctionsDomain 
+/*
+  samr_TestPrivateFunctionsDomain
 */
 static NTSTATUS dcesrv_samr_TestPrivateFunctionsDomain(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_TestPrivateFunctionsDomain *r)
@@ -3865,8 +3860,8 @@ static NTSTATUS dcesrv_samr_TestPrivateFunctionsDomain(struct dcesrv_call_state
 }
 
 
-/* 
-  samr_TestPrivateFunctionsUser 
+/*
+  samr_TestPrivateFunctionsUser
 */
 static NTSTATUS dcesrv_samr_TestPrivateFunctionsUser(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_TestPrivateFunctionsUser *r)
@@ -3875,8 +3870,8 @@ static NTSTATUS dcesrv_samr_TestPrivateFunctionsUser(struct dcesrv_call_state *d
 }
 
 
-/* 
-  samr_GetUserPwInfo 
+/*
+  samr_GetUserPwInfo
 */
 static NTSTATUS dcesrv_samr_GetUserPwInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                   struct samr_GetUserPwInfo *r)
@@ -3890,27 +3885,28 @@ static NTSTATUS dcesrv_samr_GetUserPwInfo(struct dcesrv_call_state *dce_call, TA
 
        a_state = h->data;
 
-       r->out.info->min_password_length = samdb_search_uint(a_state->sam_ctx, mem_ctx, 0,
-                                                            a_state->domain_state->domain_dn, "minPwdLength",
-                                                            NULL);
-       r->out.info->password_properties = samdb_search_uint(a_state->sam_ctx, mem_ctx, 0,
-                                                            a_state->account_dn,
-                                                            "pwdProperties", NULL);
+       r->out.info->min_password_length = samdb_search_uint(a_state->sam_ctx,
+               mem_ctx, 0, a_state->domain_state->domain_dn, "minPwdLength",
+               NULL);
+       r->out.info->password_properties = samdb_search_uint(a_state->sam_ctx,
+               mem_ctx, 0, a_state->account_dn, "pwdProperties", NULL);
+
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_RemoveMemberFromForeignDomain 
+/*
+  samr_RemoveMemberFromForeignDomain
 */
-static NTSTATUS dcesrv_samr_RemoveMemberFromForeignDomain(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
-                      struct samr_RemoveMemberFromForeignDomain *r)
+static NTSTATUS dcesrv_samr_RemoveMemberFromForeignDomain(struct dcesrv_call_state *dce_call,
+                                                         TALLOC_CTX *mem_ctx,
+                                                         struct samr_RemoveMemberFromForeignDomain *r)
 {
        struct dcesrv_handle *h;
        struct samr_domain_state *d_state;
        const char *memberdn;
        struct ldb_message **res;
-       const char * const attrs[3] = { "distinguishedName", "objectSid", NULL };
+       const char *no_attrs[] = { NULL };
        int i, count;
 
        DCESRV_PULL_HANDLE(h, r->in.domain_handle, SAMR_HANDLE_DOMAIN);
@@ -3918,18 +3914,15 @@ static NTSTATUS dcesrv_samr_RemoveMemberFromForeignDomain(struct dcesrv_call_sta
        d_state = h->data;
 
        memberdn = samdb_search_string(d_state->sam_ctx, mem_ctx, NULL,
-                                      "distinguishedName", "(objectSid=%s)", 
+                                      "distinguishedName", "(objectSid=%s)",
                                       ldap_encode_ndr_dom_sid(mem_ctx, r->in.sid));
        /* Nothing to do */
        if (memberdn == NULL) {
                return NT_STATUS_OK;
        }
 
-       /* TODO: Does this call only remove alias members, or does it do this
-        * for domain groups as well? */
-
        count = samdb_search_domain(d_state->sam_ctx, mem_ctx,
-                                   d_state->domain_dn, &res, attrs,
+                                   d_state->domain_dn, &res, no_attrs,
                                    d_state->domain_sid,
                                    "(&(member=%s)(objectClass=group)"
                                    "(|(groupType=%d)(groupType=%d)))",
@@ -3942,34 +3935,32 @@ static NTSTATUS dcesrv_samr_RemoveMemberFromForeignDomain(struct dcesrv_call_sta
 
        for (i=0; i<count; i++) {
                struct ldb_message *mod;
+               int ret;
 
                mod = ldb_msg_new(mem_ctx);
                if (mod == NULL) {
                        return NT_STATUS_NO_MEMORY;
                }
 
-               mod->dn = samdb_result_dn(d_state->sam_ctx, mod, res[i], "distinguishedName", NULL);
-               if (mod->dn == NULL) {
-                       talloc_free(mod);
-                       continue;
-               }
+               mod->dn = res[i]->dn;
 
                if (samdb_msg_add_delval(d_state->sam_ctx, mem_ctx, mod,
-                                        "member", memberdn) != 0)
+                                        "member", memberdn) != LDB_SUCCESS)
                        return NT_STATUS_NO_MEMORY;
 
-               if (ldb_modify(d_state->sam_ctx, mod) != 0)
-                       return NT_STATUS_UNSUCCESSFUL;
-
+               ret = ldb_modify(d_state->sam_ctx, mod);
                talloc_free(mod);
+               if (ret != LDB_SUCCESS) {
+                       return dsdb_ldb_err_to_ntstatus(ret);
+               }
        }
 
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_QueryDomainInfo2 
+/*
+  samr_QueryDomainInfo2
 
   just an alias for samr_QueryDomainInfo
 */
@@ -3985,13 +3976,13 @@ static NTSTATUS dcesrv_samr_QueryDomainInfo2(struct dcesrv_call_state *dce_call,
        r1.out.info  = r->out.info;
 
        status = dcesrv_samr_QueryDomainInfo(dce_call, mem_ctx, &r1);
-       
+
        return status;
 }
 
 
-/* 
-  samr_QueryUserInfo2 
+/*
+  samr_QueryUserInfo2
 
   just an alias for samr_QueryUserInfo
 */
@@ -4004,15 +3995,15 @@ static NTSTATUS dcesrv_samr_QueryUserInfo2(struct dcesrv_call_state *dce_call, T
        r1.in.user_handle = r->in.user_handle;
        r1.in.level  = r->in.level;
        r1.out.info  = r->out.info;
-       
+
        status = dcesrv_samr_QueryUserInfo(dce_call, mem_ctx, &r1);
 
        return status;
 }
 
 
-/* 
-  samr_QueryDisplayInfo2 
+/*
+  samr_QueryDisplayInfo2
 */
 static NTSTATUS dcesrv_samr_QueryDisplayInfo2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                                       struct samr_QueryDisplayInfo2 *r)
@@ -4035,8 +4026,8 @@ static NTSTATUS dcesrv_samr_QueryDisplayInfo2(struct dcesrv_call_state *dce_call
 }
 
 
-/* 
-  samr_GetDisplayEnumerationIndex2 
+/*
+  samr_GetDisplayEnumerationIndex2
 */
 static NTSTATUS dcesrv_samr_GetDisplayEnumerationIndex2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_GetDisplayEnumerationIndex2 *r)
@@ -4045,8 +4036,8 @@ static NTSTATUS dcesrv_samr_GetDisplayEnumerationIndex2(struct dcesrv_call_state
 }
 
 
-/* 
-  samr_QueryDisplayInfo3 
+/*
+  samr_QueryDisplayInfo3
 */
 static NTSTATUS dcesrv_samr_QueryDisplayInfo3(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_QueryDisplayInfo3 *r)
@@ -4069,8 +4060,8 @@ static NTSTATUS dcesrv_samr_QueryDisplayInfo3(struct dcesrv_call_state *dce_call
 }
 
 
-/* 
-  samr_AddMultipleMembersToAlias 
+/*
+  samr_AddMultipleMembersToAlias
 */
 static NTSTATUS dcesrv_samr_AddMultipleMembersToAlias(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_AddMultipleMembersToAlias *r)
@@ -4079,8 +4070,8 @@ static NTSTATUS dcesrv_samr_AddMultipleMembersToAlias(struct dcesrv_call_state *
 }
 
 
-/* 
-  samr_RemoveMultipleMembersFromAlias 
+/*
+  samr_RemoveMultipleMembersFromAlias
 */
 static NTSTATUS dcesrv_samr_RemoveMultipleMembersFromAlias(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_RemoveMultipleMembersFromAlias *r)
@@ -4089,12 +4080,12 @@ static NTSTATUS dcesrv_samr_RemoveMultipleMembersFromAlias(struct dcesrv_call_st
 }
 
 
-/* 
-  samr_GetDomPwInfo 
+/*
+  samr_GetDomPwInfo
 
   this fetches the default password properties for a domain
 
-  note that w2k3 completely ignores the domain name in this call, and 
+  note that w2k3 completely ignores the domain name in this call, and
   always returns the information for the servers primary domain
 */
 static NTSTATUS dcesrv_samr_GetDomPwInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
@@ -4107,34 +4098,42 @@ static NTSTATUS dcesrv_samr_GetDomPwInfo(struct dcesrv_call_state *dce_call, TAL
 
        ZERO_STRUCTP(r->out.info);
 
-       sam_ctx = samdb_connect(mem_ctx, dce_call->event_ctx, dce_call->conn->dce_ctx->lp_ctx, dce_call->conn->auth_state.session_info); 
+       sam_ctx = samdb_connect(mem_ctx, dce_call->event_ctx,
+                                        dce_call->conn->dce_ctx->lp_ctx,
+                                        dce_call->conn->auth_state.session_info, 0);
        if (sam_ctx == NULL) {
                return NT_STATUS_INVALID_SYSTEM_SERVICE;
        }
 
        /* The domain name in this call is ignored */
-       ret = gendb_search_dn(sam_ctx, 
+       ret = gendb_search_dn(sam_ctx,
                           mem_ctx, NULL, &msgs, attrs);
        if (ret <= 0) {
+               talloc_free(sam_ctx);
+
                return NT_STATUS_NO_SUCH_DOMAIN;
        }
        if (ret > 1) {
                talloc_free(msgs);
+               talloc_free(sam_ctx);
+
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
 
-       r->out.info->min_password_length = samdb_result_uint(msgs[0], "minPwdLength", 0);
-       r->out.info->password_properties = samdb_result_uint(msgs[0], "pwdProperties", 1);
+       r->out.info->min_password_length = ldb_msg_find_attr_as_uint(msgs[0],
+               "minPwdLength", 0);
+       r->out.info->password_properties = ldb_msg_find_attr_as_uint(msgs[0],
+               "pwdProperties", 1);
 
        talloc_free(msgs);
+       talloc_unlink(mem_ctx, sam_ctx);
 
-       talloc_free(sam_ctx);
        return NT_STATUS_OK;
 }
 
 
-/* 
-  samr_Connect2 
+/*
+  samr_Connect2
 */
 static NTSTATUS dcesrv_samr_Connect2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                              struct samr_Connect2 *r)
@@ -4149,8 +4148,8 @@ static NTSTATUS dcesrv_samr_Connect2(struct dcesrv_call_state *dce_call, TALLOC_
 }
 
 
-/* 
-  samr_SetUserInfo2 
+/*
+  samr_SetUserInfo2
 
   just an alias for samr_SetUserInfo
 */
@@ -4167,8 +4166,8 @@ static NTSTATUS dcesrv_samr_SetUserInfo2(struct dcesrv_call_state *dce_call, TAL
 }
 
 
-/* 
-  samr_SetBootKeyInformation 
+/*
+  samr_SetBootKeyInformation
 */
 static NTSTATUS dcesrv_samr_SetBootKeyInformation(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_SetBootKeyInformation *r)
@@ -4177,18 +4176,19 @@ static NTSTATUS dcesrv_samr_SetBootKeyInformation(struct dcesrv_call_state *dce_
 }
 
 
-/* 
-  samr_GetBootKeyInformation 
+/*
+  samr_GetBootKeyInformation
 */
 static NTSTATUS dcesrv_samr_GetBootKeyInformation(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_GetBootKeyInformation *r)
 {
-       DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
+       /* Windows Server 2008 returns this */
+       return NT_STATUS_NOT_SUPPORTED;
 }
 
 
-/* 
-  samr_Connect3 
+/*
+  samr_Connect3
 */
 static NTSTATUS dcesrv_samr_Connect3(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_Connect3 *r)
@@ -4203,8 +4203,8 @@ static NTSTATUS dcesrv_samr_Connect3(struct dcesrv_call_state *dce_call, TALLOC_
 }
 
 
-/* 
-  samr_Connect4 
+/*
+  samr_Connect4
 */
 static NTSTATUS dcesrv_samr_Connect4(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_Connect4 *r)
@@ -4219,8 +4219,8 @@ static NTSTATUS dcesrv_samr_Connect4(struct dcesrv_call_state *dce_call, TALLOC_
 }
 
 
-/* 
-  samr_Connect5 
+/*
+  samr_Connect5
 */
 static NTSTATUS dcesrv_samr_Connect5(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                              struct samr_Connect5 *r)
@@ -4242,8 +4242,8 @@ static NTSTATUS dcesrv_samr_Connect5(struct dcesrv_call_state *dce_call, TALLOC_
 }
 
 
-/* 
-  samr_RidToSid 
+/*
+  samr_RidToSid
 */
 static NTSTATUS dcesrv_samr_RidToSid(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                              struct samr_RidToSid *r)
@@ -4265,8 +4265,8 @@ static NTSTATUS dcesrv_samr_RidToSid(struct dcesrv_call_state *dce_call, TALLOC_
 }
 
 
-/* 
-  samr_SetDsrmPassword 
+/*
+  samr_SetDsrmPassword
 */
 static NTSTATUS dcesrv_samr_SetDsrmPassword(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct samr_SetDsrmPassword *r)
@@ -4275,13 +4275,58 @@ static NTSTATUS dcesrv_samr_SetDsrmPassword(struct dcesrv_call_state *dce_call,
 }
 
 
-/* 
-  samr_ValidatePassword 
+/*
+  samr_ValidatePassword
+
+  For now the call checks the password complexity (if active) and the minimum
+  password length on level 2 and 3. Level 1 is ignored for now.
 */
-static NTSTATUS dcesrv_samr_ValidatePassword(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
-                                     struct samr_ValidatePassword *r)
+static NTSTATUS dcesrv_samr_ValidatePassword(struct dcesrv_call_state *dce_call,
+                                            TALLOC_CTX *mem_ctx,
+                                            struct samr_ValidatePassword *r)
 {
-       DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
+       struct samr_GetDomPwInfo r2;
+       struct samr_PwInfo pwInfo;
+       DATA_BLOB password;
+       enum samr_ValidationStatus res;
+       NTSTATUS status;
+
+       (*r->out.rep) = talloc_zero(mem_ctx, union samr_ValidatePasswordRep);
+
+       r2.in.domain_name = NULL;
+       r2.out.info = &pwInfo;
+       status = dcesrv_samr_GetDomPwInfo(dce_call, mem_ctx, &r2);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
+
+       switch (r->in.level) {
+       case NetValidateAuthentication:
+               /* we don't support this yet */
+               return NT_STATUS_NOT_SUPPORTED;
+       break;
+       case NetValidatePasswordChange:
+               password = data_blob_const(r->in.req->req2.password.string,
+                                          r->in.req->req2.password.length);
+               res = samdb_check_password(&password,
+                                          pwInfo.password_properties,
+                                          pwInfo.min_password_length);
+               (*r->out.rep)->ctr2.status = res;
+       break;
+       case NetValidatePasswordReset:
+               password = data_blob_const(r->in.req->req3.password.string,
+                                          r->in.req->req3.password.length);
+               res = samdb_check_password(&password,
+                                          pwInfo.password_properties,
+                                          pwInfo.min_password_length);
+               (*r->out.rep)->ctr3.status = res;
+       break;
+       default:
+               return NT_STATUS_INVALID_INFO_CLASS;
+       break;
+       }
+
+       return NT_STATUS_OK;
 }