r12542: Move some more prototypes out to seperate headers
[samba.git] / source4 / kdc / hdb-ldb.c
index 149ebeedca778a20f8af1fa52aa902685f0ee7f8..5a3d9c25e7dc82aabf9e57b8cf082106c2d07236 100644 (file)
 #include "ads.h"
 #include "hdb.h"
 #include "lib/ldb/include/ldb.h"
+#include "lib/ldb/include/ldb_errors.h"
 #include "system/iconv.h"
+#include "librpc/gen_ndr/netlogon.h"
+#include "auth/auth.h"
+#include "dsdb/samdb/samdb.h"
 
 enum hdb_ldb_ent_type 
-{ HDB_LDB_ENT_TYPE_CLIENT, HDB_LDB_ENT_TYPE_SERVER, HDB_LDB_ENT_TYPE_KRBTGT, HDB_LDB_ENT_TYPE_ANY };
+{ HDB_LDB_ENT_TYPE_CLIENT, HDB_LDB_ENT_TYPE_SERVER, 
+  HDB_LDB_ENT_TYPE_KRBTGT, HDB_LDB_ENT_TYPE_ANY };
 
 static const char * const krb5_attrs[] = {
        "objectClass",
-       "cn",
-       "name",
        "sAMAccountName",
 
        "userPrincipalName",
        "servicePrincipalName",
 
-       "userAccountControl",
-       "sAMAccountType",
-
-       "objectSid",
-       "primaryGroupID",
-       "memberOf",
-
-       "unicodePWD",
+       "unicodePwd",
        "lmPwdHash",
        "ntPwdHash",
 
-       "badPwdCount",
-       "badPasswordTime",
-       "lastLogoff",
-       "lastLogon",
+       "userAccountControl",
+
        "pwdLastSet",
        "accountExpires",
-       "logonCount",
 
-       "objectGUID",
        "whenCreated",
        "whenChanged",
-       "uSNCreated",
-       "uSNChanged",
+
        "msDS-KeyVersionNumber",
        NULL
 };
 
+static const char *realm_ref_attrs[] = {
+       "nCName", 
+       "dnsRoot", 
+       NULL
+};
+
 static KerberosTime ldb_msg_find_krb5time_ldap_time(struct ldb_message *msg, const char *attr, KerberosTime default_val)
 {
     const char *tmp;
@@ -109,6 +106,11 @@ static HDBFlags uf2HDBFlags(krb5_context context, int userAccountControl, enum h
        /* mark the principal as invalid to start with */
        flags.invalid = 1;
 
+       flags.renewable = 1;
+
+       /* All accounts are servers, but this may be disabled again in the caller */
+       flags.server = 1;
+
        /* Account types - clear the invalid bit if it turns out to be valid */
        if (userAccountControl & UF_NORMAL_ACCOUNT) {
                if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
@@ -127,23 +129,18 @@ static HDBFlags uf2HDBFlags(krb5_context context, int userAccountControl, enum h
                if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
                        flags.client = 1;
                }
-               if (ent_type == HDB_LDB_ENT_TYPE_SERVER || ent_type == HDB_LDB_ENT_TYPE_ANY) {
-                       flags.server = 1;
-               }
                flags.invalid = 0;
        }
        if (userAccountControl & UF_SERVER_TRUST_ACCOUNT) {
                if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
                        flags.client = 1;
                }
-               if (ent_type == HDB_LDB_ENT_TYPE_SERVER || ent_type == HDB_LDB_ENT_TYPE_ANY) {
-                       flags.server = 1;
-               }
                flags.invalid = 0;
        }
 
+       /* Not permitted to act as a client if disabled */
        if (userAccountControl & UF_ACCOUNTDISABLE) {
-               flags.invalid = 1;
+               flags.client = 0;
        }
        if (userAccountControl & UF_LOCKOUT) {
                flags.invalid = 1;
@@ -177,13 +174,10 @@ static HDBFlags uf2HDBFlags(krb5_context context, int userAccountControl, enum h
        if (userAccountControl & UF_SMARTCARD_REQUIRED) {
                flags.require_hwauth = 1;
        }
-       if (flags.server && (userAccountControl & UF_TRUSTED_FOR_DELEGATION)) {
-               flags.forwardable = 1;
-               flags.proxiable = 1;
-       } else if (flags.client && (userAccountControl & UF_NOT_DELEGATED)) {
-               flags.forwardable = 0;
-               flags.proxiable = 0;
-       } else {
+       if (userAccountControl & UF_TRUSTED_FOR_DELEGATION) {
+               flags.ok_as_delegate = 1;
+       }       
+       if (!(userAccountControl & UF_NOT_DELEGATED)) {
                flags.forwardable = 1;
                flags.proxiable = 1;
        }
@@ -205,21 +199,55 @@ static HDBFlags uf2HDBFlags(krb5_context context, int userAccountControl, enum h
        return flags;
 }
 
+static int hdb_ldb_destrutor(void *ptr)
+{
+    struct hdb_ldb_private *private = ptr;
+    hdb_entry_ex *entry_ex = private->entry_ex;
+    free_hdb_entry(&entry_ex->entry);
+    return 0;
+}
+
+static void hdb_ldb_free_entry(krb5_context context, hdb_entry_ex *entry_ex)
+{
+       talloc_free(entry_ex->ctx);
+}
+
 /*
  * Construct an hdb_entry from a directory entry.
  */
 static krb5_error_code LDB_message2entry(krb5_context context, HDB *db, 
                                         TALLOC_CTX *mem_ctx, krb5_const_principal principal,
-                                        enum hdb_ldb_ent_type ent_type, struct ldb_message *realm_msg,
+                                        enum hdb_ldb_ent_type ent_type, 
                                         struct ldb_message *msg,
-                                        hdb_entry *ent)
+                                        struct ldb_message *realm_ref_msg,
+                                        hdb_entry_ex *entry_ex)
 {
        const char *unicodePwd;
-       int userAccountControl;
+       unsigned int userAccountControl;
        int i;
        krb5_error_code ret = 0;
-       const char *dnsdomain = ldb_msg_find_string(realm_msg, "dnsDomain", NULL);
+       krb5_boolean is_computer = FALSE;
+       const char *dnsdomain = ldb_msg_find_string(realm_ref_msg, "dnsRoot", NULL);
        char *realm = strupper_talloc(mem_ctx, dnsdomain);
+       struct ldb_dn *domain_dn = samdb_result_dn(mem_ctx, realm_ref_msg, "nCName", ldb_dn_new(mem_ctx));
+
+       struct hdb_ldb_private *private;
+       NTTIME acct_expiry;
+
+       struct ldb_message_element *objectclasses;
+       struct ldb_val computer_val;
+       computer_val.data = discard_const_p(uint8_t,"computer");
+       computer_val.length = strlen((const char *)computer_val.data);
+       
+       objectclasses = ldb_msg_find_element(msg, "objectClass");
+       
+       if (objectclasses && ldb_msg_find_val(objectclasses, &computer_val)) {
+               is_computer = TRUE;
+       }
+
+       memset(entry_ex, 0, sizeof(*entry_ex));
+
+       krb5_warnx(context, "LDB_message2entry:\n");
 
        if (!realm) {
                krb5_set_error_string(context, "talloc_strdup: out of memory");
@@ -227,13 +255,23 @@ static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
                goto out;
        }
                        
-       krb5_warnx(context, "LDB_message2entry:\n");
+       private = talloc(mem_ctx, struct hdb_ldb_private);
+       if (!private) {
+               ret = ENOMEM;
+               goto out;
+       }
+
+       private->entry_ex = entry_ex;
 
-       memset(ent, 0, sizeof(*ent));
+       talloc_set_destructor(private, hdb_ldb_destrutor);
+
+       entry_ex->ctx = private;
+       entry_ex->free_entry = hdb_ldb_free_entry;
+
+       userAccountControl = ldb_msg_find_uint(msg, "userAccountControl", 0);
 
-       userAccountControl = ldb_msg_find_int(msg, "userAccountControl", 0);
        
-       ent->principal = malloc(sizeof(*(ent->principal)));
+       entry_ex->entry.principal = malloc(sizeof(*(entry_ex->entry.principal)));
        if (ent_type == HDB_LDB_ENT_TYPE_ANY && principal == NULL) {
                const char *samAccountName = ldb_msg_find_string(msg, "samAccountName", NULL);
                if (!samAccountName) {
@@ -242,10 +280,10 @@ static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
                        goto out;
                }
                samAccountName = ldb_msg_find_string(msg, "samAccountName", NULL);
-               krb5_make_principal(context, &ent->principal, realm, samAccountName, NULL);
+               krb5_make_principal(context, &entry_ex->entry.principal, realm, samAccountName, NULL);
        } else {
                char *strdup_realm;
-               ret = copy_Principal(principal, ent->principal);
+               ret = copy_Principal(principal, entry_ex->entry.principal);
                if (ret) {
                        krb5_clear_error_string(context);
                        goto out;
@@ -258,7 +296,7 @@ static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
                 * we determine from our records */
                
                /* don't leak */
-               free(*krb5_princ_realm(context, ent->principal));
+               free(*krb5_princ_realm(context, entry_ex->entry.principal));
                
                /* this has to be with malloc() */
                strdup_realm = strdup(realm);
@@ -267,45 +305,82 @@ static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
                        krb5_clear_error_string(context);
                        goto out;
                }
-               krb5_princ_set_realm(context, ent->principal, &strdup_realm);
+               krb5_princ_set_realm(context, entry_ex->entry.principal, &strdup_realm);
        }
 
-       ent->kvno = ldb_msg_find_int(msg, "msDS-KeyVersionNumber", 0);
+       entry_ex->entry.kvno = ldb_msg_find_int(msg, "msDS-KeyVersionNumber", 0);
 
-       ent->flags = uf2HDBFlags(context, userAccountControl, ent_type);
+       entry_ex->entry.flags = uf2HDBFlags(context, userAccountControl, ent_type);
 
        if (ent_type == HDB_LDB_ENT_TYPE_KRBTGT) {
-               ent->flags.invalid = 0;
-               ent->flags.server = 1;
+               entry_ex->entry.flags.invalid = 0;
+               entry_ex->entry.flags.server = 1;
+               entry_ex->entry.flags.forwardable = 1;
+               entry_ex->entry.flags.ok_as_delegate = 1;
+       }
+
+       if (lp_parm_bool(-1, "kdc", "require spn for service", True)) {
+               if (!is_computer && !ldb_msg_find_string(msg, "servicePrincipalName", NULL)) {
+                       entry_ex->entry.flags.server = 0;
+               }
        }
 
        /* use 'whenCreated' */
-       ent->created_by.time = ldb_msg_find_krb5time_ldap_time(msg, "whenCreated", 0);
+       entry_ex->entry.created_by.time = ldb_msg_find_krb5time_ldap_time(msg, "whenCreated", 0);
        /* use '???' */
-       ent->created_by.principal = NULL;
+       entry_ex->entry.created_by.principal = NULL;
 
-       ent->modified_by = (Event *) malloc(sizeof(Event));
-       if (ent->modified_by == NULL) {
+       entry_ex->entry.modified_by = (Event *) malloc(sizeof(Event));
+       if (entry_ex->entry.modified_by == NULL) {
                krb5_set_error_string(context, "malloc: out of memory");
                ret = ENOMEM;
                goto out;
        }
 
        /* use 'whenChanged' */
-       ent->modified_by->time = ldb_msg_find_krb5time_ldap_time(msg, "whenChanged", 0);
+       entry_ex->entry.modified_by->time = ldb_msg_find_krb5time_ldap_time(msg, "whenChanged", 0);
        /* use '???' */
-       ent->modified_by->principal = NULL;
+       entry_ex->entry.modified_by->principal = NULL;
 
-       ent->valid_start = NULL;
+       entry_ex->entry.valid_start = NULL;
 
-       ent->valid_end = NULL;
-       ent->pw_end = NULL;
+       acct_expiry = samdb_result_nttime(msg, "accountExpires", (NTTIME)-1);
+       if ((acct_expiry == (NTTIME)-1) ||
+           (acct_expiry == 0x7FFFFFFFFFFFFFFFULL)) {
+               entry_ex->entry.valid_end = NULL;
+       } else {
+               entry_ex->entry.valid_end = malloc(sizeof(*entry_ex->entry.valid_end));
+               if (entry_ex->entry.valid_end == NULL) {
+                       ret = ENOMEM;
+                       goto out;
+               }
+               *entry_ex->entry.valid_end = nt_time_to_unix(acct_expiry);
+       }
 
-       ent->max_life = NULL;
+       if ((ent_type != HDB_LDB_ENT_TYPE_KRBTGT) && (!(userAccountControl & UF_DONT_EXPIRE_PASSWD))) {
+               NTTIME must_change_time
+                       = samdb_result_force_password_change((struct ldb_context *)db->hdb_db, mem_ctx, 
+                                                            domain_dn, msg, 
+                                                            "pwdLastSet");
+               if (must_change_time != 0) {
+                       entry_ex->entry.pw_end = malloc(sizeof(*entry_ex->entry.pw_end));
+                       if (entry_ex->entry.pw_end == NULL) {
+                               ret = ENOMEM;
+                               goto out;
+                       }
+                       *entry_ex->entry.pw_end = nt_time_to_unix(must_change_time);
+               } else {
+                       entry_ex->entry.pw_end = NULL;
+               }
+       } else {
+               entry_ex->entry.pw_end = NULL;
+       }
+                       
+       entry_ex->entry.max_life = NULL;
 
-       ent->max_renew = NULL;
+       entry_ex->entry.max_renew = NULL;
 
-       ent->generation = NULL;
+       entry_ex->entry.generation = NULL;
 
        /* create the keys and enctypes */
        unicodePwd = ldb_msg_find_string(msg, "unicodePwd", NULL);
@@ -316,14 +391,7 @@ static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
 
                Principal *salt_principal;
                const char *user_principal_name = ldb_msg_find_string(msg, "userPrincipalName", NULL);
-               struct ldb_message_element *objectclasses;
-               struct ldb_val computer_val;
-               computer_val.data = "computer";
-               computer_val.length = strlen(computer_val.data);
-               
-               objectclasses = ldb_msg_find_element(msg, "objectClass");
-
-               if (objectclasses && ldb_msg_find_val(objectclasses, &computer_val)) {
+               if (is_computer) {
                        /* Determine a salting principal */
                        char *samAccountName = talloc_strdup(mem_ctx, ldb_msg_find_string(msg, "samAccountName", NULL));
                        char *saltbody;
@@ -357,19 +425,21 @@ static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
                }
 
                if (ret == 0) {
+                       size_t num_keys = entry_ex->entry.keys.len;
                        /*
                         * create keys from unicodePwd
                         */
                        ret = hdb_generate_key_set_password(context, salt_principal, 
-                                                   unicodePwd, 
-                                                           &ent->keys.val, &ent->keys.len);
+                                                           unicodePwd, 
+                                                           &entry_ex->entry.keys.val, &num_keys);
+                       entry_ex->entry.keys.len = num_keys;
                        krb5_free_principal(context, salt_principal);
                }
 
                if (ret != 0) {
                        krb5_warnx(context, "could not generate keys from unicodePwd\n");
-                       ent->keys.val = NULL;
-                       ent->keys.len = 0;
+                       entry_ex->entry.keys.val = NULL;
+                       entry_ex->entry.keys.len = 0;
                        goto out;
                }
        } else {
@@ -379,60 +449,72 @@ static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
                val = ldb_msg_find_ldb_val(msg, "ntPwdHash");
                if (!val) {
                        krb5_warnx(context, "neither type of key available for this account\n");
-                       ent->keys.val = NULL;
-                       ent->keys.len = 0;
+                       entry_ex->entry.keys.val = NULL;
+                       entry_ex->entry.keys.len = 0;
                } else if (val->length < 16) {
-                       ent->keys.val = NULL;
-                       ent->keys.len = 0;
-                       krb5_warnx(context, "ntPwdHash has invalid length: %d\n",val->length);
+                       entry_ex->entry.keys.val = NULL;
+                       entry_ex->entry.keys.len = 0;
+                       krb5_warnx(context, "ntPwdHash has invalid length: %d\n",
+                                  (int)val->length);
                } else {
                        ret = krb5_data_alloc (&keyvalue, 16);
                        if (ret) {
-                               krb5_set_error_string(context, "malloc: out of memory");
+                               krb5_clear_error_string(context);
                                ret = ENOMEM;
                                goto out;
                        }
 
                        memcpy(keyvalue.data, val->data, 16);
 
-                       ent->keys.val = malloc(sizeof(ent->keys.val[0]));
-                       if (ent->keys.val == NULL) {
+                       entry_ex->entry.keys.val = malloc(sizeof(entry_ex->entry.keys.val[0]));
+                       if (entry_ex->entry.keys.val == NULL) {
                                krb5_data_free(&keyvalue);
-                               krb5_set_error_string(context, "malloc: out of memory");
+                               krb5_clear_error_string(context);
                                ret = ENOMEM;
                                goto out;
                        }
                        
-                       memset(&ent->keys.val[0], 0, sizeof(Key));
-                       ent->keys.val[0].key.keytype = ETYPE_ARCFOUR_HMAC_MD5;
-                       ent->keys.val[0].key.keyvalue = keyvalue;
+                       memset(&entry_ex->entry.keys.val[0], 0, sizeof(Key));
+                       entry_ex->entry.keys.val[0].key.keytype = ETYPE_ARCFOUR_HMAC_MD5;
+                       entry_ex->entry.keys.val[0].key.keyvalue = keyvalue;
                        
-                       ent->keys.len = 1;
+                       entry_ex->entry.keys.len = 1;
                }
        }               
 
 
-       ent->etypes = malloc(sizeof(*(ent->etypes)));
-       if (ent->etypes == NULL) {
-               krb5_set_error_string(context, "malloc: out of memory");
+       entry_ex->entry.etypes = malloc(sizeof(*(entry_ex->entry.etypes)));
+       if (entry_ex->entry.etypes == NULL) {
+               krb5_clear_error_string(context);
                ret = ENOMEM;
                goto out;
        }
-       ent->etypes->len = ent->keys.len;
-       ent->etypes->val = calloc(ent->etypes->len, sizeof(int));
-       if (ent->etypes->val == NULL) {
-               krb5_set_error_string(context, "malloc: out of memory");
+       entry_ex->entry.etypes->len = entry_ex->entry.keys.len;
+       entry_ex->entry.etypes->val = calloc(entry_ex->entry.etypes->len, sizeof(int));
+       if (entry_ex->entry.etypes->val == NULL) {
+               krb5_clear_error_string(context);
                ret = ENOMEM;
                goto out;
        }
-       for (i=0; i < ent->etypes->len; i++) {
-               ent->etypes->val[i] = ent->keys.val[i].key.keytype;
+       for (i=0; i < entry_ex->entry.etypes->len; i++) {
+               entry_ex->entry.etypes->val[i] = entry_ex->entry.keys.val[i].key.keytype;
        }
 
+
+       private->msg = talloc_steal(private, msg);
+       private->realm_ref_msg = talloc_steal(private, realm_ref_msg);
+       private->samdb = (struct ldb_context *)db->hdb_db;
+       
+       entry_ex->check_client_access = hdb_ldb_check_client_access;
+       entry_ex->authz_data_tgs_req = hdb_ldb_authz_data_tgs_req;
+       entry_ex->authz_data_as_req = hdb_ldb_authz_data_as_req;
+
 out:
        if (ret != 0) {
-               /* I don't think this frees ent itself. */
-               hdb_free_entry(context, ent);
+               /* This doesn't free ent itself, that is for the eventual caller to do */
+               hdb_free_entry(context, entry_ex);
+       } else {
+               talloc_steal(db, entry_ex->ctx);
        }
 
        return ret;
@@ -442,22 +524,22 @@ static krb5_error_code LDB_lookup_principal(krb5_context context, struct ldb_con
                                            TALLOC_CTX *mem_ctx,
                                            krb5_const_principal principal,
                                            enum hdb_ldb_ent_type ent_type,
-                                           const char *realm_dn,
+                                           const struct ldb_dn *realm_dn,
                                            struct ldb_message ***pmsg)
 {
        krb5_error_code ret;
-       int count;
+       int lret;
        char *filter = NULL;
        const char * const *princ_attrs = krb5_attrs;
-       char *p;
 
-       char *princ_str;
-       char *princ_str_talloc;
        char *short_princ;
+       char *short_princ_talloc;
+
+       char *realm_dn_str;
 
-       struct ldb_message **msg;
+       struct ldb_result *res = NULL;
 
-       ret = krb5_unparse_name(context, principal, &princ_str);
+       ret = krb5_unparse_name_norealm(context, principal, &short_princ);
 
        if (ret != 0) {
                krb5_set_error_string(context, "LDB_lookup_principal: could not parse principal");
@@ -465,36 +547,27 @@ static krb5_error_code LDB_lookup_principal(krb5_context context, struct ldb_con
                return ret;
        }
 
-       princ_str_talloc = talloc_strdup(mem_ctx, princ_str);
-       short_princ = talloc_strdup(mem_ctx, princ_str);
-       free(princ_str);
-       if (!short_princ || !princ_str_talloc) {
+       short_princ_talloc = talloc_strdup(mem_ctx, short_princ);
+       free(short_princ);
+       if (!short_princ || !short_princ_talloc) {
                krb5_set_error_string(context, "LDB_lookup_principal: talloc_strdup() failed!");
                return ENOMEM;
        }
 
-       p = strchr(short_princ, '@');
-       if (p) {
-               p[0] = '\0';
-       }
-
-       
        switch (ent_type) {
+       case HDB_LDB_ENT_TYPE_CLIENT:
+               /* Can't happen */
+               return EINVAL;
+       case HDB_LDB_ENT_TYPE_ANY:
+               /* Can't happen */
+               return EINVAL;
        case HDB_LDB_ENT_TYPE_KRBTGT:
                filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(samAccountName=%s))", 
                                         KRB5_TGS_NAME);
                break;
-       case HDB_LDB_ENT_TYPE_CLIENT:
-               filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(|(samAccountName=%s)(userPrincipalName=%s)))", 
-                                        short_princ, princ_str_talloc);
-               break;
        case HDB_LDB_ENT_TYPE_SERVER:
-               filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(|(samAccountName=%s)(servicePrincipalName=%s)))", 
-                                        short_princ, short_princ);
-               break;
-       case HDB_LDB_ENT_TYPE_ANY:
-               filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(|(|(samAccountName=%s)(servicePrincipalName=%s))(userPrincipalName=%s)))", 
-                                        short_princ, short_princ, princ_str_talloc);
+               filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(samAccountName=%s))", 
+                                        short_princ_talloc);
                break;
        }
 
@@ -503,23 +576,27 @@ static krb5_error_code LDB_lookup_principal(krb5_context context, struct ldb_con
                return ENOMEM;
        }
 
-       count = ldb_search(ldb_ctx, realm_dn, LDB_SCOPE_SUBTREE, filter, 
-                          princ_attrs, &msg);
+       lret = ldb_search(ldb_ctx, realm_dn, LDB_SCOPE_SUBTREE, filter, princ_attrs, &res);
 
-       *pmsg = talloc_steal(mem_ctx, msg);
-       if (count < 1) {
-               krb5_warnx(context, "ldb_search: basedn: '%s' filter: '%s' failed: %d", 
-                          realm_dn, filter, count);
-               krb5_set_error_string(context, "ldb_search: basedn: '%s' filter: '%s' failed: %d", 
-                                     realm_dn, filter, count);
+       realm_dn_str = ldb_dn_linearize(mem_ctx, realm_dn);
+
+       if (lret != LDB_SUCCESS || res->count == 0) {
+               krb5_warnx(context, "ldb_search: basedn: '%s' filter: '%s' failed: %s", 
+                          realm_dn_str, filter, ldb_errstring(ldb_ctx));
+               krb5_set_error_string(context, "ldb_search: basedn: '%s' filter: '%s' failed: %s", 
+                                     realm_dn_str, filter, ldb_errstring(ldb_ctx));
                return HDB_ERR_NOENTRY;
-       } else if (count > 1) {
+       } else if (res->count > 1) {
                krb5_warnx(context, "ldb_search: basedn: '%s' filter: '%s' more than 1 entry: %d", 
-                          realm_dn, filter, count);
+                          realm_dn_str, filter, res->count);
                krb5_set_error_string(context, "ldb_search: basedn: '%s' filter: '%s' more than 1 entry: %d", 
-                                     realm_dn, filter, count);
+                                     realm_dn_str, filter, res->count);
+               talloc_free(res);
                return HDB_ERR_NOENTRY;
        }
+       talloc_steal(mem_ctx, res->msgs);
+       *pmsg = res->msgs;
+       talloc_free(res);
        return 0;
 }
 
@@ -528,22 +605,9 @@ static krb5_error_code LDB_lookup_realm(krb5_context context, struct ldb_context
                                        const char *realm,
                                        struct ldb_message ***pmsg)
 {
-       int count;
-       const char *realm_dn;
+       int ret;
        char *cross_ref_filter;
-       struct ldb_message **cross_ref_msg;
-       struct ldb_message **msg;
-
-       const char *cross_ref_attrs[] = {
-               "nCName", 
-               NULL
-       };
-
-       const char *realm_attrs[] = {
-               "dnsDomain", 
-               "maxPwdAge",
-               NULL
-       };
+       struct ldb_result *cross_ref_res;
 
        cross_ref_filter = talloc_asprintf(mem_ctx, 
                                           "(&(&(|(&(dnsRoot=%s)(nETBIOSName=*))(nETBIOSName=%s))(objectclass=crossRef))(ncName=*))",
@@ -553,121 +617,31 @@ static krb5_error_code LDB_lookup_realm(krb5_context context, struct ldb_context
                return ENOMEM;
        }
 
-       count = ldb_search(ldb_ctx, NULL, LDB_SCOPE_SUBTREE, cross_ref_filter, 
-                          cross_ref_attrs, &cross_ref_msg);
+       ret = ldb_search(ldb_ctx, NULL, LDB_SCOPE_SUBTREE, cross_ref_filter, realm_ref_attrs, &cross_ref_res);
 
-       if (count < 1) {
-               krb5_warnx(context, "ldb_search: filter: '%s' failed: %d", cross_ref_filter, count);
-               krb5_set_error_string(context, "ldb_search: filter: '%s' failed: %d", cross_ref_filter, count);
+       if (ret != LDB_SUCCESS || cross_ref_res->count == 0) {
+               krb5_warnx(context, "ldb_search: filter: '%s' failed: %s", cross_ref_filter, ldb_errstring(ldb_ctx));
+               krb5_set_error_string(context, "ldb_search: filter: '%s' failed: %s", cross_ref_filter, ldb_errstring(ldb_ctx));
 
-               talloc_free(cross_ref_msg);
+               talloc_free(cross_ref_res);
                return HDB_ERR_NOENTRY;
-       } else if (count > 1) {
-               krb5_warnx(context, "ldb_search: filter: '%s' more than 1 entry: %d", cross_ref_filter, count);
-               krb5_set_error_string(context, "ldb_search: filter: '%s' more than 1 entry: %d", cross_ref_filter, count);
+       } else if (cross_ref_res->count > 1) {
+               krb5_warnx(context, "ldb_search: filter: '%s' more than 1 entry: %d", cross_ref_filter, cross_ref_res->count);
+               krb5_set_error_string(context, "ldb_search: filter: '%s' more than 1 entry: %d", cross_ref_filter, cross_ref_res->count);
 
-               talloc_free(cross_ref_msg);
+               talloc_free(cross_ref_res);
                return HDB_ERR_NOENTRY;
        }
 
-       realm_dn = ldb_msg_find_string(cross_ref_msg[0], "nCName", NULL);
-
-       count = ldb_search(ldb_ctx, realm_dn, LDB_SCOPE_BASE, "(objectClass=domain)",
-                          realm_attrs, &msg);
        if (pmsg) {
-               *pmsg = talloc_steal(mem_ctx, msg);
+               *pmsg = talloc_steal(mem_ctx, cross_ref_res->msgs);
        } else {
-               talloc_free(msg);
-       }
-
-       if (count < 1) {
-               krb5_warnx(context, "ldb_search: dn: %s not found: %d", realm_dn, count);
-               krb5_set_error_string(context, "ldb_search: dn: %s not found: %d", realm_dn, count);
-               return HDB_ERR_NOENTRY;
-       } else if (count > 1) {
-               krb5_warnx(context, "ldb_search: dn: '%s' more than 1 entry: %d", realm_dn, count);
-               krb5_set_error_string(context, "ldb_search: dn: %s more than 1 entry: %d", realm_dn, count);
-               return HDB_ERR_NOENTRY;
+               talloc_free(cross_ref_res);
        }
 
        return 0;
 }
 
-static krb5_error_code LDB_lookup_spn_alias(krb5_context context, struct ldb_context *ldb_ctx, 
-                                           TALLOC_CTX *mem_ctx,
-                                           const char *realm_dn,
-                                           const char *alias_from,
-                                           char **alias_to)
-{
-       int i;
-       int count;
-       struct ldb_message **msg;
-       struct ldb_message_element *spnmappings;
-       char *service_dn = talloc_asprintf(mem_ctx, 
-                                          "CN=Directory Service,CN=Windows NT,CN=Services,CN=Configuration,%s", 
-                                          realm_dn);
-       const char *directory_attrs[] = {
-               "sPNMappings", 
-               NULL
-       };
-
-       count = ldb_search(ldb_ctx, service_dn, LDB_SCOPE_BASE, "(objectClass=nTDSService)",
-                          directory_attrs, &msg);
-       talloc_steal(mem_ctx, msg);
-
-       if (count < 1) {
-               krb5_warnx(context, "ldb_search: dn: %s not found: %d", service_dn, count);
-               krb5_set_error_string(context, "ldb_search: dn: %s not found: %d", service_dn, count);
-               return HDB_ERR_NOENTRY;
-       } else if (count > 1) {
-               krb5_warnx(context, "ldb_search: dn: %s found %d times!", service_dn, count);
-               krb5_set_error_string(context, "ldb_search: dn: %s found %d times!", service_dn, count);
-               return HDB_ERR_NOENTRY;
-       }
-       
-       spnmappings = ldb_msg_find_element(msg[0], "sPNMappings");
-       if (!spnmappings || spnmappings->num_values == 0) {
-               krb5_warnx(context, "ldb_search: dn: %s no sPNMappings attribute", service_dn);
-               krb5_set_error_string(context, "ldb_search: dn: %s no sPNMappings attribute", service_dn);
-       }
-
-       for (i = 0; i < spnmappings->num_values; i++) {
-               char *mapping, *p, *str;
-               mapping = talloc_strdup(mem_ctx, 
-                                       spnmappings->values[i].data);
-               if (!mapping) {
-                       krb5_warnx(context, "LDB_lookup_spn_alias: ldb_search: dn: %s did not have an sPNMapping", service_dn);
-                       krb5_set_error_string(context, "LDB_lookup_spn_alias: ldb_search: dn: %s did not have an sPNMapping", service_dn);
-                       return HDB_ERR_NOENTRY;
-               }
-               
-               /* C string manipulation sucks */
-               
-               p = strchr(mapping, '=');
-               if (!p) {
-                       krb5_warnx(context, "ldb_search: dn: %s sPNMapping malformed: %s", 
-                                  service_dn, mapping);
-                       krb5_set_error_string(context, "ldb_search: dn: %s sPNMapping malformed: %s", 
-                                             service_dn, mapping);
-               }
-               p[0] = '\0';
-               p++;
-               do {
-                       str = p;
-                       p = strchr(p, ',');
-                       if (p) {
-                               p[0] = '\0';
-                               p++;
-                       }
-                       if (strcasecmp(str, alias_from) == 0) {
-                               *alias_to = mapping;
-                               return 0;
-                       }
-               } while (p);
-       }
-       krb5_warnx(context, "LDB_lookup_spn_alias: no alias for service %s applicable", alias_from);
-       return HDB_ERR_NOENTRY;
-}
 
 static krb5_error_code LDB_open(krb5_context context, HDB *db, int flags, mode_t mode)
 {
@@ -703,125 +677,186 @@ static krb5_error_code LDB_rename(krb5_context context, HDB *db, const char *new
 static krb5_error_code LDB_fetch(krb5_context context, HDB *db, unsigned flags,
                                 krb5_const_principal principal,
                                 enum hdb_ent_type ent_type,
-                                hdb_entry *entry)
+                                hdb_entry_ex *entry_ex)
 {
        struct ldb_message **msg = NULL;
-       struct ldb_message **realm_msg = NULL;
+       struct ldb_message **realm_ref_msg = NULL;
+       struct ldb_message **realm_fixed_msg = NULL;
        enum hdb_ldb_ent_type ldb_ent_type;
        krb5_error_code ret;
 
        const char *realm;
-       const char *realm_dn;
-       TALLOC_CTX *mem_ctx = talloc_named(NULL, 0, "LDB_fetch context\n");
+       const struct ldb_dn *realm_dn;
+       TALLOC_CTX *mem_ctx = talloc_named(db, 0, "LDB_fetch context");
 
        if (!mem_ctx) {
                krb5_set_error_string(context, "LDB_fetch: talloc_named() failed!");
                return ENOMEM;
        }
 
-       realm = krb5_principal_get_realm(context, principal);
-               
-       ret = LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db, 
-                              mem_ctx, realm, &realm_msg);
-       if (ret != 0) {
-               krb5_warnx(context, "LDB_fetch: could not find realm\n");
-               talloc_free(mem_ctx);
-               return HDB_ERR_NOENTRY;
-       }
-
-       realm_dn = realm_msg[0]->dn;
-
-       /* Cludge, cludge cludge.  If the realm part of krbtgt/realm,
-        * is in our db, then direct the caller at our primary
-        * krgtgt */
-       
        switch (ent_type) {
-       case HDB_ENT_TYPE_SERVER:
-               if (principal->name.name_string.len == 2
-                   && (strcmp(principal->name.name_string.val[0], KRB5_TGS_NAME) == 0)
-                   && (LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db,
-                                        mem_ctx, principal->name.name_string.val[1], NULL) == 0)) {
-                       ldb_ent_type = HDB_LDB_ENT_TYPE_KRBTGT;
-               } else {
-                       ldb_ent_type = HDB_LDB_ENT_TYPE_SERVER;
-               }
-               break;
        case HDB_ENT_TYPE_CLIENT:
+       {
+               NTSTATUS nt_status;
+               char *principal_string;
                ldb_ent_type = HDB_LDB_ENT_TYPE_CLIENT;
-               break;
-       case HDB_ENT_TYPE_ANY:
-               ldb_ent_type = HDB_LDB_ENT_TYPE_ANY;
-               break;
-       }
 
-       ret = LDB_lookup_principal(context, (struct ldb_context *)db->hdb_db, 
-                                  mem_ctx, 
-                                  principal, ldb_ent_type, realm_dn, &msg);
-
-       if (ret != 0) {
-               char *alias_from = principal->name.name_string.val[0];
-               char *alias_to;
-               Principal alias_principal;
+               ret = krb5_unparse_name(context, principal, &principal_string);
                
-               /* Try again with a servicePrincipal alias */
-               if (ent_type != HDB_LDB_ENT_TYPE_SERVER && ent_type != HDB_LDB_ENT_TYPE_ANY) {
-                       talloc_free(mem_ctx);
-                       return ret;
-               }
-               if (principal->name.name_string.len < 2) {
-                       krb5_warnx(context, "LDB_fetch: could not find principal in DB, alias not applicable");
-                       krb5_set_error_string(context, "LDB_fetch: could not find principal in DB, alias not applicable");
-                       talloc_free(mem_ctx);
-                       return ret;
-               }
-
-               /* Look for the list of aliases */
-               ret = LDB_lookup_spn_alias(context, 
-                                          (struct ldb_context *)db->hdb_db, mem_ctx, 
-                                          realm_dn, alias_from, 
-                                          &alias_to);
                if (ret != 0) {
                        talloc_free(mem_ctx);
                        return ret;
                }
 
-               ret = copy_Principal(principal, &alias_principal);
-               if (ret != 0) {
-                       krb5_warnx(context, "LDB_fetch: could not copy principal");
-                       krb5_set_error_string(context, "LDB_fetch: could not copy principal");
+               nt_status = sam_get_results_principal((struct ldb_context *)db->hdb_db,
+                                                     mem_ctx, principal_string, 
+                                                     &msg, &realm_ref_msg);
+               free(principal_string);
+               if (NT_STATUS_EQUAL(nt_status, NT_STATUS_NO_SUCH_USER)) {
                        talloc_free(mem_ctx);
-                       return ret;
-               }
-
-               /* ooh, very nasty playing around in the Principal... */
-               free(alias_principal.name.name_string.val[0]);
-               alias_principal.name.name_string.val[0] = strdup(alias_to);
-               if (!alias_principal.name.name_string.val[0]) {
-                       krb5_warnx(context, "LDB_fetch: strdup() failed");
-                       krb5_set_error_string(context, "LDB_fetch: strdup() failed");
-                       ret = ENOMEM;
+                       return HDB_ERR_NOENTRY;
+               } else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_NO_MEMORY)) {
                        talloc_free(mem_ctx);
-                       free_Principal(&alias_principal);
-                       return ret;
+                       return ENOMEM;
+               } else if (!NT_STATUS_IS_OK(nt_status)) {
+                       talloc_free(mem_ctx);
+                       return EINVAL;
                }
 
-               ret = LDB_lookup_principal(context, (struct ldb_context *)db->hdb_db, 
-                                          mem_ctx, 
-                                          &alias_principal, ent_type, realm_dn, &msg);
-               free_Principal(&alias_principal);
+               ret = LDB_message2entry(context, db, mem_ctx, 
+                                       principal, ldb_ent_type, 
+                                       msg[0], realm_ref_msg[0], entry_ex);
 
-               if (ret != 0) {
-                       krb5_warnx(context, "LDB_fetch: could not find alias principal in DB");
-                       krb5_set_error_string(context, "LDB_fetch: could not find alias principal in DB");
+               talloc_free(mem_ctx);
+               return ret;
+       }
+       case HDB_ENT_TYPE_SERVER:
+               if (principal->name.name_string.len == 2
+                   && (strcmp(principal->name.name_string.val[0], KRB5_TGS_NAME) == 0)) {
+                       /* krbtgt case.  Either us or a trusted realm */
+                       if ((LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db,
+                                             mem_ctx, principal->name.name_string.val[1], &realm_fixed_msg) == 0)) {
+                               /* us */
+                               /* Cludge, cludge cludge.  If the realm part of krbtgt/realm,
+                                * is in our db, then direct the caller at our primary
+                                * krgtgt */
+
+                               const char *dnsdomain = ldb_msg_find_string(realm_fixed_msg[0], "dnsRoot", NULL);
+                               char *realm_fixed = strupper_talloc(mem_ctx, dnsdomain);
+                               if (!realm_fixed) {
+                                       krb5_set_error_string(context, "strupper_talloc: out of memory");
+                                       talloc_free(mem_ctx);
+                                       return ENOMEM;
+                               }
+                               
+                               free(principal->name.name_string.val[1]);
+                               principal->name.name_string.val[1] = strdup(realm_fixed);
+                               talloc_free(realm_fixed);
+                               if (!principal->name.name_string.val[1]) {
+                                       krb5_set_error_string(context, "LDB_fetch: strdup() failed!");
+                                       talloc_free(mem_ctx);
+                                       return ENOMEM;
+                               }
+                               ldb_ent_type = HDB_LDB_ENT_TYPE_KRBTGT;
+                               break;
+                       } else {
+                               /* we should lookup trusted domains */
+                               talloc_free(mem_ctx);
+                               return HDB_ERR_NOENTRY;
+                       }
+
+               } else if (principal->name.name_string.len >= 2) {
+                       /* 'normal server' case */
+                       int ldb_ret;
+                       NTSTATUS nt_status;
+                       struct ldb_dn *user_dn, *domain_dn;
+                       char *principal_string;
+                       ldb_ent_type = HDB_LDB_ENT_TYPE_SERVER;
+                       
+                       ret = krb5_unparse_name_norealm(context, principal, &principal_string);
+                       
+                       if (ret != 0) {
+                               talloc_free(mem_ctx);
+                               return ret;
+                       }
+                       
+                       /* At this point we may find the host is known to be
+                        * in a different realm, so we should generate a
+                        * referral instead */
+                       nt_status = crack_service_principal_name((struct ldb_context *)db->hdb_db,
+                                                                mem_ctx, principal_string, 
+                                                                &user_dn, &domain_dn);
+                       free(principal_string);
+                       
+                       if (!NT_STATUS_IS_OK(nt_status)) {
+                               talloc_free(mem_ctx);
+                               return HDB_ERR_NOENTRY;
+                       }
+                       
+                       ldb_ret = gendb_search_dn((struct ldb_context *)db->hdb_db,
+                                                 mem_ctx, user_dn, &msg, krb5_attrs);
+                       
+                       if (ldb_ret != 1) {
+                               talloc_free(mem_ctx);
+                               return HDB_ERR_NOENTRY;
+                       }
+                       
+                       ldb_ret = gendb_search((struct ldb_context *)db->hdb_db,
+                                              mem_ctx, NULL, &realm_ref_msg, realm_ref_attrs, 
+                                              "ncName=%s", ldb_dn_linearize(mem_ctx, domain_dn));
+                       
+                       if (ldb_ret != 1) {
+                               talloc_free(mem_ctx);
+                               return HDB_ERR_NOENTRY;
+                       }
+
+                       ret = LDB_message2entry(context, db, mem_ctx, 
+                                               principal, ldb_ent_type, 
+                                               msg[0], realm_ref_msg[0], entry_ex);
                        talloc_free(mem_ctx);
                        return ret;
+                       
+               } else {
+                       ldb_ent_type = HDB_LDB_ENT_TYPE_SERVER;
+                       /* server as client principal case, but we must not lookup userPrincipalNames */
+                       break;
                }
+       case HDB_ENT_TYPE_ANY:
+               krb5_warnx(context, "LDB_fetch: ENT_TYPE_ANY is not valid in hdb-ldb!");
+               talloc_free(mem_ctx);
+               return HDB_ERR_NOENTRY;
+       default:
+               krb5_warnx(context, "LDB_fetch: invalid ent_type specified!");
+               talloc_free(mem_ctx);
+               return HDB_ERR_NOENTRY;
+       }
+
+
+       realm = krb5_principal_get_realm(context, principal);
 
+       ret = LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db, 
+                              mem_ctx, realm, &realm_ref_msg);
+       if (ret != 0) {
+               krb5_warnx(context, "LDB_fetch: could not find realm");
+               talloc_free(mem_ctx);
+               return HDB_ERR_NOENTRY;
        }
-       if (ret == 0) {
+
+       realm_dn = samdb_result_dn(mem_ctx, realm_ref_msg[0], "nCName", NULL);
+
+       ret = LDB_lookup_principal(context, (struct ldb_context *)db->hdb_db, 
+                                  mem_ctx, 
+                                  principal, ldb_ent_type, realm_dn, &msg);
+
+       if (ret != 0) {
+               krb5_warnx(context, "LDB_fetch: could not find principal in DB");
+               krb5_set_error_string(context, "LDB_fetch: could not find principal in DB");
+               talloc_free(mem_ctx);
+               return ret;
+       } else {
                ret = LDB_message2entry(context, db, mem_ctx, 
                                        principal, ldb_ent_type, 
-                                       realm_msg[0], msg[0], entry);
+                                       msg[0], realm_ref_msg[0], entry_ex);
                if (ret != 0) {
                        krb5_warnx(context, "LDB_fetch: message2entry failed\n");       
                }
@@ -831,12 +866,12 @@ static krb5_error_code LDB_fetch(krb5_context context, HDB *db, unsigned flags,
        return ret;
 }
 
-static krb5_error_code LDB_store(krb5_context context, HDB *db, unsigned flags, hdb_entry *entry)
+static krb5_error_code LDB_store(krb5_context context, HDB *db, unsigned flags, hdb_entry_ex *entry)
 {
        return HDB_ERR_DB_INUSE;
 }
 
-static krb5_error_code LDB_remove(krb5_context context, HDB *db, hdb_entry *entry)
+static krb5_error_code LDB_remove(krb5_context context, HDB *db, hdb_entry_ex *entry)
 {
        return HDB_ERR_DB_INUSE;
 }
@@ -846,14 +881,17 @@ struct hdb_ldb_seq {
        int index;
        int count;
        struct ldb_message **msgs;
-       struct ldb_message **realm_msgs;
+       struct ldb_message **realm_ref_msgs;
 };
 
-static krb5_error_code LDB_seq(krb5_context context, HDB *db, unsigned flags, hdb_entry *entry)
+static krb5_error_code LDB_seq(krb5_context context, HDB *db, unsigned flags, hdb_entry_ex *entry)
 {
        krb5_error_code ret;
        struct hdb_ldb_seq *priv = (struct hdb_ldb_seq *)db->hdb_openp;
        TALLOC_CTX *mem_ctx;
+       hdb_entry_ex entry_ex;
+       memset(&entry_ex, '\0', sizeof(entry_ex));
+
        if (!priv) {
                return HDB_ERR_NOENTRY;
        }
@@ -868,7 +906,8 @@ static krb5_error_code LDB_seq(krb5_context context, HDB *db, unsigned flags, hd
        if (priv->index < priv->count) {
                ret = LDB_message2entry(context, db, mem_ctx, 
                                        NULL, HDB_LDB_ENT_TYPE_ANY, 
-                                       priv->realm_msgs[0], priv->msgs[priv->index++], entry);
+                                       priv->msgs[priv->index++], 
+                                       priv->realm_ref_msgs[0], entry);
        } else {
                ret = HDB_ERR_NOENTRY;
        }
@@ -884,16 +923,17 @@ static krb5_error_code LDB_seq(krb5_context context, HDB *db, unsigned flags, hd
 }
 
 static krb5_error_code LDB_firstkey(krb5_context context, HDB *db, unsigned flags,
-                                       hdb_entry *entry)
+                                       hdb_entry_ex *entry)
 {
        struct ldb_context *ldb_ctx = (struct ldb_context *)db->hdb_db;
        struct hdb_ldb_seq *priv = (struct hdb_ldb_seq *)db->hdb_openp;
        char *realm;
-       char *realm_dn = NULL;
-       struct ldb_message **msgs = NULL;
-       struct ldb_message **realm_msgs = NULL;
+       struct ldb_dn *realm_dn = NULL;
+       struct ldb_result *res = NULL;
+       struct ldb_message **realm_ref_msgs = NULL;
        krb5_error_code ret;
        TALLOC_CTX *mem_ctx;
+       int lret;
 
        if (priv) {
                talloc_free(priv);
@@ -909,7 +949,7 @@ static krb5_error_code LDB_firstkey(krb5_context context, HDB *db, unsigned flag
        priv->ctx = ldb_ctx;
        priv->index = 0;
        priv->msgs = NULL;
-       priv->realm_msgs = NULL;
+       priv->realm_ref_msgs = NULL;
        priv->count = 0;
 
        mem_ctx = talloc_named(priv, 0, "LDB_firstkey context");
@@ -926,33 +966,34 @@ static krb5_error_code LDB_firstkey(krb5_context context, HDB *db, unsigned flag
        }
                
        ret = LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db, 
-                              mem_ctx, realm, &realm_msgs);
+                              mem_ctx, realm, &realm_ref_msgs);
 
        free(realm);
 
        if (ret != 0) {
                talloc_free(priv);
-               krb5_warnx(context, "LDB_fetch: could not find realm\n");
+               krb5_warnx(context, "LDB_firstkey: could not find realm\n");
                return HDB_ERR_NOENTRY;
        }
 
-       realm_dn = realm_msgs[0]->dn;
+       realm_dn = samdb_result_dn(mem_ctx, realm_ref_msgs[0], "nCName", NULL);
 
-       priv->realm_msgs = talloc_steal(priv, realm_msgs);
+       priv->realm_ref_msgs = talloc_steal(priv, realm_ref_msgs);
 
-       krb5_warnx(context, "LDB_lookup_principal: realm ok\n");
+       krb5_warnx(context, "LDB_firstkey: realm ok\n");
 
-       priv->count = ldb_search(ldb_ctx, realm_dn,
+       lret = ldb_search(ldb_ctx, realm_dn,
                                 LDB_SCOPE_SUBTREE, "(objectClass=user)",
-                                krb5_attrs, &msgs);
-
-       priv->msgs = talloc_steal(priv, msgs);
+                                krb5_attrs, &res);
 
-       if (priv->count <= 0) {
+       if (lret != LDB_SUCCESS) {
                talloc_free(priv);
                return HDB_ERR_NOENTRY;
        }
 
+       priv->count = res->count;
+       priv->msgs = talloc_steal(priv, res->msgs);
+
        db->hdb_openp = priv;
 
        ret = LDB_seq(context, db, flags, entry);
@@ -967,7 +1008,7 @@ static krb5_error_code LDB_firstkey(krb5_context context, HDB *db, unsigned flag
 }
 
 static krb5_error_code LDB_nextkey(krb5_context context, HDB *db, unsigned flags,
-                                       hdb_entry *entry)
+                                  hdb_entry_ex *entry)
 {
        return LDB_seq(context, db, flags, entry);
 }
@@ -978,24 +1019,44 @@ static krb5_error_code LDB_destroy(krb5_context context, HDB *db)
        return 0;
 }
 
-krb5_error_code hdb_ldb_create(krb5_context context, struct HDB **db, const char *arg)
+NTSTATUS hdb_ldb_create(TALLOC_CTX *mem_ctx, 
+                       krb5_context context, struct HDB **db, const char *arg)
 {
-       *db = talloc(NULL, HDB);
+       NTSTATUS nt_status;
+       struct auth_session_info *session_info;
+       *db = talloc(mem_ctx, HDB);
        if (!*db) {
                krb5_set_error_string(context, "malloc: out of memory");
-               return ENOMEM;
+               return NT_STATUS_NO_MEMORY;
        }
 
        (*db)->hdb_master_key_set = 0;
        (*db)->hdb_db = NULL;
-       /* in future, we could cache the connect here, but for now KISS */
 
-       (*db)->hdb_db = samdb_connect(db);
+       nt_status = auth_system_session_info(*db, &session_info);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               return nt_status;
+       }
+       
+       /* The idea here is very simple.  Using Kerberos to
+        * authenticate the KDC to the LDAP server is higly likely to
+        * be circular.
+        *
+        * In future we may set this up to use EXERNAL and SSL
+        * certificates, for now it will almost certainly be NTLMSSP
+       */
+       
+       nt_status = cli_credentials_gensec_remove_oid(session_info->credentials, 
+                                                     GENSEC_OID_KERBEROS5);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               return nt_status;
+       }
+
+       /* Setup the link to LDB */
+       (*db)->hdb_db = samdb_connect(*db, session_info);
        if ((*db)->hdb_db == NULL) {
-               krb5_warnx(context, "hdb_ldb_create: samdb_connect failed!");
-               krb5_set_error_string(context, "samdb_connect failed!");
-               talloc_free(*db);
-               return HDB_ERR_NOENTRY;
+               DEBUG(1, ("hdb_ldb_create: Cannot open samdb for KDC backend!"));
+               return NT_STATUS_CANT_ACCESS_DOMAIN_INFO;
        }
 
        (*db)->hdb_openp = 0;
@@ -1016,5 +1077,5 @@ krb5_error_code hdb_ldb_create(krb5_context context, struct HDB **db, const char
        (*db)->hdb__del = NULL;
        (*db)->hdb_destroy = LDB_destroy;
 
-       return 0;
+       return NT_STATUS_OK;
 }