ldb: use #include <ldb.h> for ldb
[bbaumbach/samba-autobuild/.git] / source4 / dsdb / samdb / cracknames.c
index ca87159c5889badf33194af9a43b9e61c609b679..d9224828cac5d5e0e3d3879ac0d94e71edc9aaaf 100644 (file)
@@ -6,45 +6,47 @@
 
    Copyright (C) Stefan Metzmacher 2004
    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
-   
+
    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 "includes.h"
 #include "librpc/gen_ndr/drsuapi.h"
-#include "rpc_server/common/common.h"
 #include "lib/events/events.h"
-#include "lib/ldb/include/ldb.h"
-#include "lib/ldb/include/ldb_errors.h"
+#include "rpc_server/common/common.h"
+#include <ldb.h>
+#include <ldb_errors.h>
 #include "system/kerberos.h"
 #include "auth/kerberos/kerberos.h"
 #include "libcli/ldap/ldap_ndr.h"
 #include "libcli/security/security.h"
-#include "librpc/gen_ndr/ndr_misc.h"
 #include "auth/auth.h"
-#include "util/util_ldb.h"
+#include "../lib/util/util_ldb.h"
 #include "dsdb/samdb/samdb.h"
+#include "dsdb/common/util.h"
 #include "param/param.h"
 
 static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                                   struct smb_krb5_context *smb_krb5_context,
-                                  uint32_t format_flags, uint32_t format_offered, uint32_t format_desired,
+                                  uint32_t format_flags, enum drsuapi_DsNameFormat format_offered,
+                                  enum drsuapi_DsNameFormat format_desired,
                                   struct ldb_dn *name_dn, const char *name, 
                                   const char *domain_filter, const char *result_filter, 
                                   struct drsuapi_DsNameInfo1 *info1);
 static WERROR DsCrackNameOneSyntactical(TALLOC_CTX *mem_ctx,
-                                       uint32_t format_offered, uint32_t format_desired,
+                                       enum drsuapi_DsNameFormat format_offered,
+                                       enum drsuapi_DsNameFormat format_desired,
                                        struct ldb_dn *name_dn, const char *name, 
                                        struct drsuapi_DsNameInfo1 *info1);
 
@@ -55,22 +57,22 @@ static WERROR dns_domain_from_principal(TALLOC_CTX *mem_ctx, struct smb_krb5_con
        krb5_error_code ret;
        krb5_principal principal;
        /* perhaps it's a principal with a realm, so return the right 'domain only' response */
-       char **realm;
+       const char *realm;
        ret = krb5_parse_name_flags(smb_krb5_context->krb5_context, name, 
-                                   KRB5_PRINCIPAL_PARSE_MUST_REALM, &principal);
+                                   KRB5_PRINCIPAL_PARSE_REQUIRE_REALM, &principal);
        if (ret) {
                info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
                return WERR_OK;
        }
 
        /* This isn't an allocation assignemnt, so it is free'ed with the krb5_free_principal */
-       realm = krb5_princ_realm(smb_krb5_context->krb5_context, principal);
-       
-       info1->dns_domain_name  = talloc_strdup(mem_ctx, *realm);
+       realm = krb5_principal_get_realm(smb_krb5_context->krb5_context, principal);
+
+       info1->dns_domain_name  = talloc_strdup(mem_ctx, realm);
        krb5_free_principal(smb_krb5_context->krb5_context, principal);
-       
+
        W_ERROR_HAVE_NO_MEMORY(info1->dns_domain_name);
-       
+
        info1->status = DRSUAPI_DS_NAME_STATUS_DOMAIN_ONLY;
        return WERR_OK;
 }              
@@ -80,7 +82,7 @@ static enum drsuapi_DsNameStatus LDB_lookup_spn_alias(krb5_context context, stru
                                                      const char *alias_from,
                                                      char **alias_to)
 {
-       int i;
+       unsigned int i;
        int ret;
        struct ldb_result *res;
        struct ldb_message_element *spnmappings;
@@ -99,7 +101,7 @@ static enum drsuapi_DsNameStatus LDB_lookup_spn_alias(krb5_context context, stru
        }
 
        service_dn = ldb_dn_new(tmp_ctx, ldb_ctx, "CN=Directory Service,CN=Windows NT,CN=Services");
-       if ( ! ldb_dn_add_base(service_dn, samdb_config_dn(ldb_ctx))) {
+       if ( ! ldb_dn_add_base(service_dn, ldb_get_config_basedn(ldb_ctx))) {
                return DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
        }
        service_dn_str = ldb_dn_alloc_linearized(tmp_ctx, service_dn);
@@ -121,7 +123,7 @@ static enum drsuapi_DsNameStatus LDB_lookup_spn_alias(krb5_context context, stru
                DEBUG(1, ("ldb_search: dn: %s not found", service_dn_str));
                return DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
        }
-       
+
        spnmappings = ldb_msg_find_element(res->msgs[0], "sPNMappings");
        if (!spnmappings || spnmappings->num_values == 0) {
                DEBUG(1, ("ldb_search: dn: %s no sPNMappings attribute", service_dn_str));
@@ -138,9 +140,9 @@ static enum drsuapi_DsNameStatus LDB_lookup_spn_alias(krb5_context context, stru
                        talloc_free(tmp_ctx);
                        return DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
                }
-               
+
                /* C string manipulation sucks */
-               
+
                p = strchr(mapping, '=');
                if (!p) {
                        DEBUG(1, ("ldb_search: dn: %s sPNMapping malformed: %s\n", 
@@ -177,7 +179,8 @@ static enum drsuapi_DsNameStatus LDB_lookup_spn_alias(krb5_context context, stru
 
 static WERROR DsCrackNameSPNAlias(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                                  struct smb_krb5_context *smb_krb5_context,
-                                 uint32_t format_flags, uint32_t format_offered, uint32_t format_desired,
+                                 uint32_t format_flags, enum drsuapi_DsNameFormat format_offered,
+                                 enum drsuapi_DsNameFormat format_desired,
                                  const char *name, struct drsuapi_DsNameInfo1 *info1)
 {
        WERROR wret;
@@ -187,7 +190,7 @@ static WERROR DsCrackNameSPNAlias(struct ldb_context *sam_ctx, TALLOC_CTX *mem_c
        char *new_service;
        char *new_princ;
        enum drsuapi_DsNameStatus namestatus;
-       
+
        /* parse principal */
        ret = krb5_parse_name_flags(smb_krb5_context->krb5_context, 
                                    name, KRB5_PRINCIPAL_PARSE_NO_REALM, &principal);
@@ -197,36 +200,38 @@ static WERROR DsCrackNameSPNAlias(struct ldb_context *sam_ctx, TALLOC_CTX *mem_c
                                                           ret, mem_ctx)));
                return WERR_NOMEM;
        }
-       
+
        /* grab cifs/, http/ etc */
-       
+
        /* This is checked for in callers, but be safe */
        if (principal->name.name_string.len < 2) {
                info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
+               krb5_free_principal(smb_krb5_context->krb5_context, principal);
                return WERR_OK;
        }
        service = principal->name.name_string.val[0];
        dns_name = principal->name.name_string.val[1];
-       
+
        /* MAP it */
        namestatus = LDB_lookup_spn_alias(smb_krb5_context->krb5_context, 
                                          sam_ctx, mem_ctx, 
                                          service, &new_service);
-       
+
        if (namestatus == DRSUAPI_DS_NAME_STATUS_NOT_FOUND) {
+               wret = WERR_OK;
                info1->status           = DRSUAPI_DS_NAME_STATUS_DOMAIN_ONLY;
                info1->dns_domain_name  = talloc_strdup(mem_ctx, dns_name);
                if (!info1->dns_domain_name) {
-                       krb5_free_principal(smb_krb5_context->krb5_context, principal);
-                       return WERR_NOMEM;
+                       wret = WERR_NOMEM;
                }
-               return WERR_OK;
+               krb5_free_principal(smb_krb5_context->krb5_context, principal);
+               return wret;
        } else if (namestatus != DRSUAPI_DS_NAME_STATUS_OK) {
                info1->status = namestatus;
                krb5_free_principal(smb_krb5_context->krb5_context, principal);
                return WERR_OK;
        }
-       
+
        /* ooh, very nasty playing around in the Principal... */
        free(principal->name.name_string.val[0]);
        principal->name.name_string.val[0] = strdup(new_service);
@@ -234,7 +239,7 @@ static WERROR DsCrackNameSPNAlias(struct ldb_context *sam_ctx, TALLOC_CTX *mem_c
                krb5_free_principal(smb_krb5_context->krb5_context, principal);
                return WERR_NOMEM;
        }
-       
+
        /* reform principal */
        ret = krb5_unparse_name_flags(smb_krb5_context->krb5_context, principal, 
                                      KRB5_PRINCIPAL_UNPARSE_NO_REALM, &new_princ);
@@ -243,7 +248,7 @@ static WERROR DsCrackNameSPNAlias(struct ldb_context *sam_ctx, TALLOC_CTX *mem_c
                krb5_free_principal(smb_krb5_context->krb5_context, principal);
                return WERR_NOMEM;
        }
-       
+
        wret = DsCrackNameOneName(sam_ctx, mem_ctx, format_flags, format_offered, format_desired,
                                  new_princ, info1);
        free(new_princ);
@@ -262,7 +267,8 @@ static WERROR DsCrackNameSPNAlias(struct ldb_context *sam_ctx, TALLOC_CTX *mem_c
 
 static WERROR DsCrackNameUPN(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                             struct smb_krb5_context *smb_krb5_context,
-                            uint32_t format_flags, uint32_t format_offered, uint32_t format_desired,
+                            uint32_t format_flags, enum drsuapi_DsNameFormat format_offered,
+                            enum drsuapi_DsNameFormat format_desired,
                             const char *name, struct drsuapi_DsNameInfo1 *info1)
 {
        int ldb_ret;
@@ -271,11 +277,11 @@ static WERROR DsCrackNameUPN(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
        const char *result_filter = NULL;
        krb5_error_code ret;
        krb5_principal principal;
-       char **realm;
+       const char *realm;
        char *unparsed_name_short;
        const char *domain_attrs[] = { NULL };
        struct ldb_result *domain_res = NULL;
-       
+
        /* Prevent recursion */
        if (!name) {
                info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
@@ -283,48 +289,52 @@ static WERROR DsCrackNameUPN(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
        }
 
        ret = krb5_parse_name_flags(smb_krb5_context->krb5_context, name, 
-                                   KRB5_PRINCIPAL_PARSE_MUST_REALM, &principal);
+                                   KRB5_PRINCIPAL_PARSE_REQUIRE_REALM, &principal);
        if (ret) {
                info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
                return WERR_OK;
        }
-       
-       realm = krb5_princ_realm(smb_krb5_context->krb5_context, principal);
+
+       realm = krb5_principal_get_realm(smb_krb5_context->krb5_context,
+                                        principal);
 
        ldb_ret = ldb_search(sam_ctx, mem_ctx, &domain_res,
                                     samdb_partitions_dn(sam_ctx, mem_ctx), 
                                     LDB_SCOPE_ONELEVEL,
                                     domain_attrs,
                                     "(&(&(|(&(dnsRoot=%s)(nETBIOSName=*))(nETBIOSName=%s))(objectclass=crossRef))(ncName=*))",
-                                    ldb_binary_encode_string(mem_ctx, *realm), 
-                                    ldb_binary_encode_string(mem_ctx, *realm));
+                                    ldb_binary_encode_string(mem_ctx, realm), 
+                                    ldb_binary_encode_string(mem_ctx, realm));
 
        if (ldb_ret != LDB_SUCCESS) {
                DEBUG(2, ("DsCrackNameUPN domain ref search failed: %s", ldb_errstring(sam_ctx)));
                info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
+               krb5_free_principal(smb_krb5_context->krb5_context, principal);
                return WERR_OK;
        }
-       
+
        switch (domain_res->count) {
        case 1:
                break;
        case 0:
+               krb5_free_principal(smb_krb5_context->krb5_context, principal);
                return dns_domain_from_principal(mem_ctx, smb_krb5_context, 
                                                 name, info1);
        default:
                info1->status = DRSUAPI_DS_NAME_STATUS_NOT_UNIQUE;
+               krb5_free_principal(smb_krb5_context->krb5_context, principal);
                return WERR_OK;
        }
-       
+
        ret = krb5_unparse_name_flags(smb_krb5_context->krb5_context, principal, 
                                      KRB5_PRINCIPAL_UNPARSE_NO_REALM, &unparsed_name_short);
        krb5_free_principal(smb_krb5_context->krb5_context, principal);
-               
+
        if (ret) {
                free(unparsed_name_short);
                return WERR_NOMEM;
        }
-       
+
        /* This may need to be extended for more userPrincipalName variations */
        result_filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(samAccountName=%s))", 
                                        ldb_binary_encode_string(mem_ctx, unparsed_name_short));
@@ -348,7 +358,8 @@ static WERROR DsCrackNameUPN(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
 /* Crack a single 'name', from format_offered into format_desired, returning the result in info1 */
 
 WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
-                         uint32_t format_flags, uint32_t format_offered, uint32_t format_desired,
+                         uint32_t format_flags, enum drsuapi_DsNameFormat format_offered,
+                         enum drsuapi_DsNameFormat format_desired,
                          const char *name, struct drsuapi_DsNameInfo1 *info1)
 {
        krb5_error_code ret;
@@ -374,7 +385,7 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
        switch (format_offered) {
        case DRSUAPI_DS_NAME_FORMAT_UNKNOWN:
        {
-               int i;
+               unsigned int i;
                enum drsuapi_DsNameFormat formats[] = {
                        DRSUAPI_DS_NAME_FORMAT_FQDN_1779, DRSUAPI_DS_NAME_FORMAT_USER_PRINCIPAL,
                        DRSUAPI_DS_NAME_FORMAT_NT4_ACCOUNT, DRSUAPI_DS_NAME_FORMAT_CANONICAL,
@@ -400,15 +411,15 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
        case DRSUAPI_DS_NAME_FORMAT_CANONICAL_EX:
        {
                char *str, *s, *account;
-               
+
                if (strlen(name) == 0) {
                        info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
                        return WERR_OK;
                }
-               
+
                str = talloc_strdup(mem_ctx, name);
                W_ERROR_HAVE_NO_MEMORY(str);
-               
+
                if (format_offered == DRSUAPI_DS_NAME_FORMAT_CANONICAL_EX) {
                        /* Look backwards for the \n, and replace it with / */
                        s = strrchr(str, '\n');
@@ -425,7 +436,7 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                        info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
                        return WERR_OK;
                }
-               
+
                s[0] = '\0';
                s++;
 
@@ -435,7 +446,7 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
 
                /* There may not be anything after the domain component (search for the domain itself) */
                if (s[0]) {
-                       
+
                        account = strrchr(s, '/');
                        if (!account) {
                                account = s;
@@ -453,11 +464,12 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
        case DRSUAPI_DS_NAME_FORMAT_NT4_ACCOUNT: {
                char *p;
                char *domain;
+               struct ldb_dn *dn_domain;
                const char *account = NULL;
-               
+
                domain = talloc_strdup(mem_ctx, name);
                W_ERROR_HAVE_NO_MEMORY(domain);
-               
+
                p = strchr(domain, '\\');
                if (!p) {
                        /* invalid input format */
@@ -465,21 +477,26 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                        return WERR_OK;
                }
                p[0] = '\0';
-               
+
                if (p[1]) {
                        account = &p[1];
                }
-               
+
+               /* it could be in DNS domain form */
+               dn_domain = samdb_dns_domain_to_dn(sam_ctx, mem_ctx, domain);
+               W_ERROR_HAVE_NO_MEMORY(dn_domain);
+
                domain_filter = talloc_asprintf(mem_ctx, 
-                                               "(&(&(nETBIOSName=%s)(objectclass=crossRef))(ncName=*))", 
-                                               ldb_binary_encode_string(mem_ctx, domain));
+                                               "(&(&(|(nETBIOSName=%s)(nCName=%s))(objectclass=crossRef))(ncName=*))",
+                                               ldb_binary_encode_string(mem_ctx, domain),
+                                               ldb_dn_get_linearized(dn_domain));
                W_ERROR_HAVE_NO_MEMORY(domain_filter);
                if (account) {
                        result_filter = talloc_asprintf(mem_ctx, "(sAMAccountName=%s)",
                                                        ldb_binary_encode_string(mem_ctx, account));
                        W_ERROR_HAVE_NO_MEMORY(result_filter);
                }
-               
+
                talloc_free(domain);
                break;
        }
@@ -507,7 +524,7 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                        info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
                        return WERR_OK;
                }
-                       
+
                ldap_guid = ldap_encode_ndr_GUID(mem_ctx, &guid);
                if (!ldap_guid) {
                        return WERR_NOMEM;
@@ -526,12 +543,12 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                W_ERROR_HAVE_NO_MEMORY(result_filter);
                break;
        }
-       
+
                /* A S-1234-5678 style string */
        case DRSUAPI_DS_NAME_FORMAT_SID_OR_SID_HISTORY: {
                struct dom_sid *sid = dom_sid_parse_talloc(mem_ctx, name);
                char *ldap_sid;
-                                                                           
+
                domain_filter = NULL;
                if (!sid) {
                        info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
@@ -555,19 +572,21 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                                            ldb_get_event_context(sam_ctx),
                                            (struct loadparm_context *)ldb_get_opaque(sam_ctx, "loadparm"), 
                                            &smb_krb5_context);
-               
+
                if (ret) {
                        return WERR_NOMEM;
                }
 
+               /* Ensure we reject compleate junk first */
                ret = krb5_parse_name(smb_krb5_context->krb5_context, name, &principal);
                if (ret) {
                        info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
                        return WERR_OK;
                }
-               
+
                domain_filter = NULL;
-               
+
+               /* By getting the unparsed name here, we ensure the escaping is correct (and trust the client less) */
                ret = krb5_unparse_name(smb_krb5_context->krb5_context, principal, &unparsed_name);
                if (ret) {
                        krb5_free_principal(smb_krb5_context->krb5_context, principal);
@@ -575,9 +594,11 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                }
 
                krb5_free_principal(smb_krb5_context->krb5_context, principal);
+
+               /* The ldb_binary_encode_string() here avoid LDAP filter injection attacks */
                result_filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(userPrincipalName=%s))", 
                                                ldb_binary_encode_string(mem_ctx, unparsed_name));
-               
+
                free(unparsed_name);
                W_ERROR_HAVE_NO_MEMORY(result_filter);
                break;
@@ -591,7 +612,7 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                                            ldb_get_event_context(sam_ctx),
                                            (struct loadparm_context *)ldb_get_opaque(sam_ctx, "loadparm"), 
                                            &smb_krb5_context);
-               
+
                if (ret) {
                        return WERR_NOMEM;
                }
@@ -601,18 +622,18 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                        info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
                        krb5_free_principal(smb_krb5_context->krb5_context, principal);
                        return WERR_OK;
+               } else if (ret == 0) {
+                       krb5_free_principal(smb_krb5_context->krb5_context, principal);
                }
                ret = krb5_parse_name_flags(smb_krb5_context->krb5_context, name, 
                                            KRB5_PRINCIPAL_PARSE_NO_REALM, &principal);
                if (ret) {
-                       krb5_free_principal(smb_krb5_context->krb5_context, principal);
-
                        return dns_domain_from_principal(mem_ctx, smb_krb5_context,
                                                         name, info1);
                }
 
                domain_filter = NULL;
-               
+
                ret = krb5_unparse_name_flags(smb_krb5_context->krb5_context, principal, 
                                              KRB5_PRINCIPAL_UNPARSE_NO_REALM, &unparsed_name_short);
                if (ret) {
@@ -627,6 +648,8 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                        computer_name = talloc_strndup(mem_ctx, principal->name.name_string.val[1], 
                                                      strcspn(principal->name.name_string.val[1], "."));
                        if (computer_name == NULL) {
+                               krb5_free_principal(smb_krb5_context->krb5_context, principal);
+                               free(unparsed_name_short);
                                return WERR_NOMEM;
                        }
 
@@ -640,21 +663,20 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                krb5_free_principal(smb_krb5_context->krb5_context, principal);
                free(unparsed_name_short);
                W_ERROR_HAVE_NO_MEMORY(result_filter);
-               
+
                break;
        }
        default: {
                info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
                return WERR_OK;
        }
-
        }
 
        if (format_flags & DRSUAPI_DS_NAME_FLAG_SYNTACTICAL_ONLY) {
                return DsCrackNameOneSyntactical(mem_ctx, format_offered, format_desired,
                                                 name_dn, name, info1);
        }
-       
+
        return DsCrackNameOneFilter(sam_ctx, mem_ctx, 
                                    smb_krb5_context, 
                                    format_flags, format_offered, format_desired, 
@@ -668,7 +690,8 @@ WERROR DsCrackNameOneName(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
  * database */
 
 static WERROR DsCrackNameOneSyntactical(TALLOC_CTX *mem_ctx,
-                                       uint32_t format_offered, uint32_t format_desired,
+                                       enum drsuapi_DsNameFormat format_offered,
+                                       enum drsuapi_DsNameFormat format_desired,
                                        struct ldb_dn *name_dn, const char *name, 
                                        struct drsuapi_DsNameInfo1 *info1)
 {
@@ -694,7 +717,7 @@ static WERROR DsCrackNameOneSyntactical(TALLOC_CTX *mem_ctx,
        if (!cracked) {
                return WERR_NOMEM;
        }
-       
+
        return WERR_OK; 
 }
 
@@ -707,7 +730,8 @@ static WERROR DsCrackNameOneSyntactical(TALLOC_CTX *mem_ctx,
 
 static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
                                   struct smb_krb5_context *smb_krb5_context,
-                                  uint32_t format_flags, uint32_t format_offered, uint32_t format_desired,
+                                  uint32_t format_flags, enum drsuapi_DsNameFormat format_offered,
+                                  enum drsuapi_DsNameFormat format_desired,
                                   struct ldb_dn *name_dn, const char *name, 
                                   const char *domain_filter, const char *result_filter, 
                                   struct drsuapi_DsNameInfo1 *info1)
@@ -718,7 +742,6 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
        const char * const *result_attrs;
        struct ldb_message **result_res = NULL;
        struct ldb_message *result = NULL;
-       struct ldb_dn *result_basedn = NULL;
        int i;
        char *p;
        struct ldb_dn *partitions_basedn = samdb_partitions_dn(sam_ctx, mem_ctx);
@@ -731,10 +754,10 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
 
        const char * const _domain_attrs_nt4[] = { "ncName", "dnsRoot", "nETBIOSName", NULL};
        const char * const _result_attrs_nt4[] = { "sAMAccountName", "objectSid", "objectClass", NULL};
-               
+
        const char * const _domain_attrs_guid[] = { "ncName", "dnsRoot", NULL};
        const char * const _result_attrs_guid[] = { "objectGUID", NULL};
-               
+
        const char * const _domain_attrs_display[] = { "ncName", "dnsRoot", NULL};
        const char * const _result_attrs_display[] = { "displayName", "samAccountName", NULL};
 
@@ -777,13 +800,13 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                             LDB_SCOPE_ONELEVEL,
                                             domain_attrs,
                                             "%s", domain_filter);
-               
+
                if (ldb_ret != LDB_SUCCESS) {
                        DEBUG(2, ("DsCrackNameOneFilter domain ref search failed: %s", ldb_errstring(sam_ctx)));
                        info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
                        return WERR_OK;
                }
-               
+
                switch (domain_res->count) {
                case 1:
                        break;
@@ -795,7 +818,7 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                        return WERR_OK;
                }
 
-               info1->dns_domain_name  = samdb_result_string(domain_res->msgs[0], "dnsRoot", NULL);
+               info1->dns_domain_name  = ldb_msg_find_attr_as_string(domain_res->msgs[0], "dnsRoot", NULL);
                W_ERROR_HAVE_NO_MEMORY(info1->dns_domain_name);
                info1->status           = DRSUAPI_DS_NAME_STATUS_DOMAIN_ONLY;
        } else {
@@ -806,66 +829,33 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
        if (result_filter) {
                int ret;
                struct ldb_result *res;
+               uint32_t dsdb_flags = 0;
+               struct ldb_dn *search_dn;
+
                if (domain_res) {
-                       result_basedn = samdb_result_dn(sam_ctx, mem_ctx, domain_res->msgs[0], "ncName", NULL);
-                       
-                       ret = ldb_search(sam_ctx, mem_ctx, &res,
-                                                result_basedn, LDB_SCOPE_SUBTREE, 
-                                                result_attrs, "%s", result_filter);
-                       if (ret != LDB_SUCCESS) {
-                               talloc_free(result_res);
-                               info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
-                               return WERR_OK;
-                       }
-                       ldb_ret = res->count;
-                       result_res = res->msgs;
+                       dsdb_flags = 0;
+                       search_dn = samdb_result_dn(sam_ctx, mem_ctx, domain_res->msgs[0], "ncName", NULL);
                } else {
-                       /* search with the 'phantom root' flag */
-                       struct ldb_request *req;
-
-                       res = talloc_zero(mem_ctx, struct ldb_result);
-                       W_ERROR_HAVE_NO_MEMORY(res);
-                       
-                       ret = ldb_build_search_req(&req, sam_ctx, mem_ctx,
-                                                  ldb_get_root_basedn(sam_ctx),
-                                                  LDB_SCOPE_SUBTREE,
-                                                  result_filter,
-                                                  result_attrs,
-                                                  NULL,
-                                                  res,
-                                                  ldb_search_default_callback,
-                                                  NULL);
-                       if (ret == LDB_SUCCESS) {
-                               struct ldb_search_options_control *search_options;
-                               search_options = talloc(req, struct ldb_search_options_control);
-                               W_ERROR_HAVE_NO_MEMORY(search_options);
-                               search_options->search_options = LDB_SEARCH_OPTION_PHANTOM_ROOT;
-
-                               ret = ldb_request_add_control(req, LDB_CONTROL_SEARCH_OPTIONS_OID, false, search_options);
-                       }
-                       if (ret != LDB_SUCCESS) {
-                               talloc_free(res);
-                               info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
-                               return WERR_OK;
-                       }
-                       
-                       ret = ldb_request(sam_ctx, req);
-                       
-                       if (ret == LDB_SUCCESS) {
-                               ret = ldb_wait(req->handle, LDB_WAIT_ALL);
-                       }
-                       
-                       talloc_free(req);
+                       dsdb_flags = DSDB_SEARCH_SEARCH_ALL_PARTITIONS;
+                       search_dn = ldb_get_root_basedn(sam_ctx);
+               }
 
-                       if (ret != LDB_SUCCESS) {
-                               DEBUG(2, ("DsCrackNameOneFilter phantom root search failed: %s", 
-                                         ldb_errstring(sam_ctx)));
-                               info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
-                               return WERR_OK;
-                       }
-                       ldb_ret = res->count;
-                       result_res = res->msgs;
+               /* search with the 'phantom root' flag */
+               ret = dsdb_search(sam_ctx, mem_ctx, &res,
+                                 search_dn,
+                                 LDB_SCOPE_SUBTREE,
+                                 result_attrs,
+                                 DSDB_SEARCH_SEARCH_ALL_PARTITIONS,
+                                 "%s", result_filter);
+               if (ret != LDB_SUCCESS) {
+                       DEBUG(2, ("DsCrackNameOneFilter phantom root search failed: %s",
+                                 ldb_errstring(sam_ctx)));
+                       info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
+                       return WERR_OK;
                }
+
+               ldb_ret = res->count;
+               result_res = res->msgs;
        } else if (format_offered == DRSUAPI_DS_NAME_FORMAT_FQDN_1779) {
                ldb_ret = gendb_search_dn(sam_ctx, mem_ctx, name_dn, &result_res,
                                          result_attrs);
@@ -875,7 +865,7 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                          result_attrs);
        } else {
                /* Can't happen */
-               DEBUG(0, ("LOGIC ERROR: DsCrackNameOneFilter domain ref search not availible: This can't happen..."));
+               DEBUG(0, ("LOGIC ERROR: DsCrackNameOneFilter domain ref search not available: This can't happen..."));
                info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
                return WERR_OK;
        }
@@ -891,11 +881,13 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                                   smb_krb5_context, 
                                                   format_flags, format_offered, format_desired,
                                                   name, info1);
-                       
+
                case DRSUAPI_DS_NAME_FORMAT_USER_PRINCIPAL:
                        return DsCrackNameUPN(sam_ctx, mem_ctx, smb_krb5_context, 
                                              format_flags, format_offered, format_desired,
                                              name, info1);
+               default:
+                       break;
                }
                info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
                return WERR_OK;
@@ -918,6 +910,8 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                case DRSUAPI_DS_NAME_FORMAT_CANONICAL_EX:
                                        canonical_name = ldb_dn_canonical_ex_string(mem_ctx, result_res[i]->dn);
                                        break;
+                               default:
+                                       break;
                                }
                                if (strcasecmp_m(canonical_name, name) == 0) {
                                        result = result_res[i];
@@ -941,7 +935,7 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
        if (p) {
                p[0] = '\0';
        }
-       
+
        /* here we can use result and domain_res[0] */
        switch (format_desired) {
        case DRSUAPI_DS_NAME_FORMAT_FQDN_1779: {
@@ -952,7 +946,7 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                return WERR_OK;
        }
        case DRSUAPI_DS_NAME_FORMAT_CANONICAL: {
-               info1->result_name      = samdb_result_string(result, "canonicalName", NULL);
+               info1->result_name      = ldb_msg_find_attr_as_string(result, "canonicalName", NULL);
                info1->status           = DRSUAPI_DS_NAME_STATUS_OK;
                return WERR_OK;
        }
@@ -967,7 +961,7 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
 
                const struct dom_sid *sid = samdb_result_dom_sid(mem_ctx, result, "objectSid");
                const char *_acc = "", *_dom = "";
-               
+
                if (samdb_find_attribute(sam_ctx, result, "objectClass", "domain")) {
 
                        ldb_ret = ldb_search(sam_ctx, mem_ctx, &domain_res,
@@ -975,13 +969,13 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                                     LDB_SCOPE_ONELEVEL,
                                                     domain_attrs,
                                                     "(ncName=%s)", ldb_dn_get_linearized(result->dn));
-                       
+
                        if (ldb_ret != LDB_SUCCESS) {
                                DEBUG(2, ("DsCrackNameOneFilter domain ref search failed: %s", ldb_errstring(sam_ctx)));
                                info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
                                return WERR_OK;
                        }
-                       
+
                        switch (domain_res->count) {
                        case 1:
                                break;
@@ -992,10 +986,10 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                info1->status = DRSUAPI_DS_NAME_STATUS_NOT_UNIQUE;
                                return WERR_OK;
                        }
-                       _dom = samdb_result_string(domain_res->msgs[0], "nETBIOSName", NULL);
+                       _dom = ldb_msg_find_attr_as_string(domain_res->msgs[0], "nETBIOSName", NULL);
                        W_ERROR_HAVE_NO_MEMORY(_dom);
                } else {
-                       _acc = samdb_result_string(result, "sAMAccountName", NULL);
+                       _acc = ldb_msg_find_attr_as_string(result, "sAMAccountName", NULL);
                        if (!_acc) {
                                info1->status = DRSUAPI_DS_NAME_STATUS_NO_MAPPING;
                                return WERR_OK;
@@ -1016,13 +1010,13 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                                             attrs,
                                                             "(&(objectSid=%s)(objectClass=domain))", 
                                                             ldap_encode_ndr_dom_sid(mem_ctx, dom_sid));
-                               
+
                                if (ldb_ret != LDB_SUCCESS) {
                                        DEBUG(2, ("DsCrackNameOneFilter domain search failed: %s", ldb_errstring(sam_ctx)));
                                        info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
                                        return WERR_OK;
                                }
-                               
+
                                switch (domain_res->count) {
                                case 1:
                                        break;
@@ -1039,13 +1033,13 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                                             LDB_SCOPE_ONELEVEL,
                                                             domain_attrs,
                                                             "(ncName=%s)", ldb_dn_get_linearized(domain_res->msgs[0]->dn));
-                               
+
                                if (ldb_ret != LDB_SUCCESS) {
                                        DEBUG(2, ("DsCrackNameOneFilter domain ref search failed: %s", ldb_errstring(sam_ctx)));
                                        info1->status = DRSUAPI_DS_NAME_STATUS_RESOLVE_ERROR;
                                        return WERR_OK;
                                }
-                               
+
                                switch (domain_res2->count) {
                                case 1:
                                        break;
@@ -1056,32 +1050,32 @@ static WERROR DsCrackNameOneFilter(struct ldb_context *sam_ctx, TALLOC_CTX *mem_
                                        info1->status = DRSUAPI_DS_NAME_STATUS_NOT_UNIQUE;
                                        return WERR_OK;
                                }
-                               _dom = samdb_result_string(domain_res2->msgs[0], "nETBIOSName", NULL);
+                               _dom = ldb_msg_find_attr_as_string(domain_res2->msgs[0], "nETBIOSName", NULL);
                                W_ERROR_HAVE_NO_MEMORY(_dom);
                        }
                }
 
                info1->result_name      = talloc_asprintf(mem_ctx, "%s\\%s", _dom, _acc);
                W_ERROR_HAVE_NO_MEMORY(info1->result_name);
-               
+
                info1->status           = DRSUAPI_DS_NAME_STATUS_OK;
                return WERR_OK;
        }
        case DRSUAPI_DS_NAME_FORMAT_GUID: {
                struct GUID guid;
-               
+
                guid = samdb_result_guid(result, "objectGUID");
-               
+
                info1->result_name      = GUID_string2(mem_ctx, &guid);
                W_ERROR_HAVE_NO_MEMORY(info1->result_name);
-               
+
                info1->status           = DRSUAPI_DS_NAME_STATUS_OK;
                return WERR_OK;
        }
        case DRSUAPI_DS_NAME_FORMAT_DISPLAY: {
-               info1->result_name      = samdb_result_string(result, "displayName", NULL);
+               info1->result_name      = ldb_msg_find_attr_as_string(result, "displayName", NULL);
                if (!info1->result_name) {
-                       info1->result_name      = samdb_result_string(result, "sAMAccountName", NULL);
+                       info1->result_name      = ldb_msg_find_attr_as_string(result, "sAMAccountName", NULL);
                } 
                if (!info1->result_name) {
                        info1->status = DRSUAPI_DS_NAME_STATUS_NOT_FOUND;
@@ -1136,9 +1130,9 @@ NTSTATUS crack_user_principal_name(struct ldb_context *sam_ctx,
        default:
                return NT_STATUS_UNSUCCESSFUL;
        }
-       
+
        *user_dn = ldb_dn_new(mem_ctx, sam_ctx, info1.result_name);
-       
+
        if (domain_dn) {
                werr = DsCrackNameOneName(sam_ctx, mem_ctx, 0,
                                          DRSUAPI_DS_NAME_FORMAT_CANONICAL,
@@ -1160,12 +1154,11 @@ NTSTATUS crack_user_principal_name(struct ldb_context *sam_ctx,
                default:
                        return NT_STATUS_UNSUCCESSFUL;
                }
-               
+
                *domain_dn = ldb_dn_new(mem_ctx, sam_ctx, info1.result_name);
        }
 
        return NT_STATUS_OK;
-       
 }
 
 /* Given a Service Principal Name (such as host/foo.bar.com@BAR.COM),
@@ -1199,9 +1192,9 @@ NTSTATUS crack_service_principal_name(struct ldb_context *sam_ctx,
        default:
                return NT_STATUS_UNSUCCESSFUL;
        }
-       
+
        *user_dn = ldb_dn_new(mem_ctx, sam_ctx, info1.result_name);
-       
+
        if (domain_dn) {
                werr = DsCrackNameOneName(sam_ctx, mem_ctx, 0,
                                          DRSUAPI_DS_NAME_FORMAT_CANONICAL,
@@ -1223,18 +1216,17 @@ NTSTATUS crack_service_principal_name(struct ldb_context *sam_ctx,
                default:
                        return NT_STATUS_UNSUCCESSFUL;
                }
-               
+
                *domain_dn = ldb_dn_new(mem_ctx, sam_ctx, info1.result_name);
        }
 
        return NT_STATUS_OK;
-       
 }
 
 NTSTATUS crack_name_to_nt4_name(TALLOC_CTX *mem_ctx, 
-                               struct event_context *ev_ctx, 
+                               struct tevent_context *ev_ctx, 
                                struct loadparm_context *lp_ctx,
-                               uint32_t format_offered,
+                               enum drsuapi_DsNameFormat format_offered,
                                const char *name, 
                                const char **nt4_domain, const char **nt4_account)
 {
@@ -1250,7 +1242,7 @@ NTSTATUS crack_name_to_nt4_name(TALLOC_CTX *mem_ctx,
                return NT_STATUS_OK;
        }
 
-       ldb = samdb_connect(mem_ctx, ev_ctx, lp_ctx, system_session(mem_ctx, lp_ctx));
+       ldb = samdb_connect(mem_ctx, ev_ctx, lp_ctx, system_session(lp_ctx), 0);
        if (ldb == NULL) {
                return NT_STATUS_INTERNAL_DB_CORRUPTION;
        }
@@ -1274,20 +1266,20 @@ NTSTATUS crack_name_to_nt4_name(TALLOC_CTX *mem_ctx,
        default:
                return NT_STATUS_UNSUCCESSFUL;
        }
-       
+
        *nt4_domain = talloc_strdup(mem_ctx, info1.result_name);
-       
+       if (*nt4_domain == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
        p = strchr(*nt4_domain, '\\');
        if (!p) {
                return NT_STATUS_INVALID_PARAMETER;
        }
        p[0] = '\0';
-       
-       if (p[1]) {
-               *nt4_account = talloc_strdup(mem_ctx, &p[1]);
-       }
 
-       if (!*nt4_account || !*nt4_domain) {
+       *nt4_account = talloc_strdup(mem_ctx, &p[1]);
+       if (*nt4_account == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
 
@@ -1295,13 +1287,13 @@ NTSTATUS crack_name_to_nt4_name(TALLOC_CTX *mem_ctx,
 }
 
 NTSTATUS crack_auto_name_to_nt4_name(TALLOC_CTX *mem_ctx,
-                                    struct event_context *ev_ctx, 
+                                    struct tevent_context *ev_ctx, 
                                     struct loadparm_context *lp_ctx,
                                     const char *name,
                                     const char **nt4_domain,
                                     const char **nt4_account)
 {
-       uint32_t format_offered = DRSUAPI_DS_NAME_FORMAT_UNKNOWN;
+       enum drsuapi_DsNameFormat format_offered = DRSUAPI_DS_NAME_FORMAT_UNKNOWN;
 
        /* Handle anonymous bind */
        if (!name || !*name) {
@@ -1324,3 +1316,76 @@ NTSTATUS crack_auto_name_to_nt4_name(TALLOC_CTX *mem_ctx,
 
        return crack_name_to_nt4_name(mem_ctx, ev_ctx, lp_ctx, format_offered, name, nt4_domain, nt4_account);
 }
+
+
+WERROR dcesrv_drsuapi_ListRoles(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
+                               const struct drsuapi_DsNameRequest1 *req1,
+                               struct drsuapi_DsNameCtr1 **ctr1)
+{
+       struct drsuapi_DsNameInfo1 *names;
+       uint32_t i;
+       uint32_t count = 5;/*number of fsmo role owners we are going to return*/
+
+       *ctr1 = talloc(mem_ctx, struct drsuapi_DsNameCtr1);
+       W_ERROR_HAVE_NO_MEMORY(*ctr1);
+       names = talloc_array(mem_ctx, struct drsuapi_DsNameInfo1, count);
+       W_ERROR_HAVE_NO_MEMORY(names);
+
+       for (i = 0; i < count; i++) {
+               WERROR werr;
+               struct ldb_dn *role_owner_dn, *fsmo_role_dn, *server_dn;
+               werr = dsdb_get_fsmo_role_info(mem_ctx, sam_ctx, i,
+                                              &fsmo_role_dn, &role_owner_dn);
+               if(!W_ERROR_IS_OK(werr)) {
+                       return werr;
+               }
+               server_dn = ldb_dn_copy(mem_ctx, role_owner_dn);
+               ldb_dn_remove_child_components(server_dn, 1);
+               names[i].status = DRSUAPI_DS_NAME_STATUS_OK;
+               names[i].dns_domain_name = samdb_dn_to_dnshostname(sam_ctx, mem_ctx,
+                                                                  server_dn);
+               if(!names[i].dns_domain_name) {
+                       DEBUG(4, ("list_roles: Failed to find dNSHostName for server %s",
+                                 ldb_dn_get_linearized(server_dn)));
+               }
+               names[i].result_name = talloc_strdup(mem_ctx, ldb_dn_get_linearized(role_owner_dn));
+       }
+
+       (*ctr1)->count = count;
+       (*ctr1)->array = names;
+
+       return WERR_OK;
+}
+
+WERROR dcesrv_drsuapi_CrackNamesByNameFormat(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
+                                            const struct drsuapi_DsNameRequest1 *req1,
+                                            struct drsuapi_DsNameCtr1 **ctr1)
+{
+       struct drsuapi_DsNameInfo1 *names;
+       uint32_t i, count;
+       WERROR status;
+
+       *ctr1 = talloc(mem_ctx, struct drsuapi_DsNameCtr1);
+       W_ERROR_HAVE_NO_MEMORY(*ctr1);
+
+       count = req1->count;
+       names = talloc_array(mem_ctx, struct drsuapi_DsNameInfo1, count);
+       W_ERROR_HAVE_NO_MEMORY(names);
+
+       for (i=0; i < count; i++) {
+               status = DsCrackNameOneName(sam_ctx, mem_ctx,
+                                           req1->format_flags,
+                                           req1->format_offered,
+                                           req1->format_desired,
+                                           req1->names[i].str,
+                                           &names[i]);
+               if (!W_ERROR_IS_OK(status)) {
+                       return status;
+               }
+       }
+
+       (*ctr1)->count = count;
+       (*ctr1)->array = names;
+
+       return WERR_OK;
+}