source4/kdc/hdb-ldb.c

   1 /*
   2  * Copyright (c) 1999-2001, 2003, PADL Software Pty Ltd.
   3  * Copyright (c) 2004, Andrew Bartlett <abartlet@samba.org>.
   4  * Copyright (c) 2004, Stefan Metzmacher <metze@samba.org>
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  *
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * 3. Neither the name of PADL Software  nor the names of its contributors
  19  *    may be used to endorse or promote products derived from this software
  20  *    without specific prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY PADL SOFTWARE AND CONTRIBUTORS ``AS IS'' AND
  23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED.  IN NO EVENT SHALL PADL SOFTWARE OR CONTRIBUTORS BE LIABLE
  26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  */
  34
  35 #include "includes.h"
  36 #include "system/time.h"
  37 #include "kdc.h"
  38 #include "ads.h"
  39 #include "hdb.h"
  40 #include "lib/ldb/include/ldb.h"
  41 #include "lib/ldb/include/ldb_errors.h"
  42 #include "librpc/gen_ndr/netlogon.h"
  43 #include "auth/auth.h"
  44 #include "auth/auth_sam.h"
  45 #include "db_wrap.h"
  46 #include "dsdb/samdb/samdb.h"
  47
  48 enum hdb_ldb_ent_type
  49 { HDB_LDB_ENT_TYPE_CLIENT, HDB_LDB_ENT_TYPE_SERVER,
  50   HDB_LDB_ENT_TYPE_KRBTGT, HDB_LDB_ENT_TYPE_ANY };
  51
  52 static const char * const krb5_attrs[] = {
  53         "objectClass",
  54         "sAMAccountName",
  55
  56         "userPrincipalName",
  57         "servicePrincipalName",
  58
  59         "krb5Key",
  60
  61         "userAccountControl",
  62
  63         "pwdLastSet",
  64         "accountExpires",
  65
  66         "whenCreated",
  67         "whenChanged",
  68
  69         "msDS-KeyVersionNumber",
  70         NULL
  71 };
  72
  73 static const char *realm_ref_attrs[] = {
  74         "nCName",
  75         "dnsRoot",
  76         NULL
  77 };
  78
  79 static KerberosTime ldb_msg_find_krb5time_ldap_time(struct ldb_message *msg, const char *attr, KerberosTime default_val)
  80 {
  81     const char *tmp;
  82     const char *gentime;
  83     struct tm tm;
  84
  85     gentime = ldb_msg_find_attr_as_string(msg, attr, NULL);
  86     if (!gentime)
  87         return default_val;
  88
  89     tmp = strptime(gentime, "%Y%m%d%H%M%SZ", &tm);
  90     if (tmp == NULL) {
  91             return default_val;
  92     }
  93
  94     return timegm(&tm);
  95 }
  96
  97 static HDBFlags uf2HDBFlags(krb5_context context, int userAccountControl, enum hdb_ldb_ent_type ent_type)
  98 {
  99         HDBFlags flags = int2HDBFlags(0);
 100
 101         /* we don't allow kadmin deletes */
 102         flags.immutable = 1;
 103
 104         /* mark the principal as invalid to start with */
 105         flags.invalid = 1;
 106
 107         flags.renewable = 1;
 108
 109         /* All accounts are servers, but this may be disabled again in the caller */
 110         flags.server = 1;
 111
 112         /* Account types - clear the invalid bit if it turns out to be valid */
 113         if (userAccountControl & UF_NORMAL_ACCOUNT) {
 114                 if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 115                         flags.client = 1;
 116                 }
 117                 flags.invalid = 0;
 118         }
 119
 120         if (userAccountControl & UF_INTERDOMAIN_TRUST_ACCOUNT) {
 121                 if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 122                         flags.client = 1;
 123                 }
 124                 flags.invalid = 0;
 125         }
 126         if (userAccountControl & UF_WORKSTATION_TRUST_ACCOUNT) {
 127                 if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 128                         flags.client = 1;
 129                 }
 130                 flags.invalid = 0;
 131         }
 132         if (userAccountControl & UF_SERVER_TRUST_ACCOUNT) {
 133                 if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 134                         flags.client = 1;
 135                 }
 136                 flags.invalid = 0;
 137         }
 138
 139         /* Not permitted to act as a client if disabled */
 140         if (userAccountControl & UF_ACCOUNTDISABLE) {
 141                 flags.client = 0;
 142         }
 143         if (userAccountControl & UF_LOCKOUT) {
 144                 flags.invalid = 1;
 145         }
 146 /*
 147         if (userAccountControl & UF_PASSWORD_NOTREQD) {
 148                 flags.invalid = 1;
 149         }
 150 */
 151 /*
 152         UF_PASSWORD_CANT_CHANGE and UF_ENCRYPTED_TEXT_PASSWORD_ALLOWED are irrelevent
 153 */
 154         if (userAccountControl & UF_TEMP_DUPLICATE_ACCOUNT) {
 155                 flags.invalid = 1;
 156         }
 157
 158 /* UF_DONT_EXPIRE_PASSWD and UF_USE_DES_KEY_ONLY handled in LDB_message2entry() */
 159
 160 /*
 161         if (userAccountControl & UF_MNS_LOGON_ACCOUNT) {
 162                 flags.invalid = 1;
 163         }
 164 */
 165         if (userAccountControl & UF_SMARTCARD_REQUIRED) {
 166                 flags.require_hwauth = 1;
 167         }
 168         if (userAccountControl & UF_TRUSTED_FOR_DELEGATION) {
 169                 flags.ok_as_delegate = 1;
 170         }
 171         if (!(userAccountControl & UF_NOT_DELEGATED)) {
 172                 flags.forwardable = 1;
 173                 flags.proxiable = 1;
 174         }
 175
 176         if (userAccountControl & UF_DONT_REQUIRE_PREAUTH) {
 177                 flags.require_preauth = 0;
 178         } else {
 179                 flags.require_preauth = 1;
 180
 181         }
 182         return flags;
 183 }
 184
 185 static int hdb_ldb_destrutor(struct hdb_ldb_private *private)
 186 {
 187     hdb_entry_ex *entry_ex = private->entry_ex;
 188     free_hdb_entry(&entry_ex->entry);
 189     return 0;
 190 }
 191
 192 static void hdb_ldb_free_entry(krb5_context context, hdb_entry_ex *entry_ex)
 193 {
 194         talloc_free(entry_ex->ctx);
 195 }
 196
 197 /*
 198  * Construct an hdb_entry from a directory entry.
 199  */
 200 static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
 201                                          TALLOC_CTX *mem_ctx, krb5_const_principal principal,
 202                                          enum hdb_ldb_ent_type ent_type,
 203                                          struct ldb_message *msg,
 204                                          struct ldb_message *realm_ref_msg,
 205                                          hdb_entry_ex *entry_ex)
 206 {
 207         unsigned int userAccountControl;
 208         int i;
 209         krb5_error_code ret = 0;
 210         krb5_boolean is_computer = FALSE;
 211         const char *dnsdomain = ldb_msg_find_attr_as_string(realm_ref_msg, "dnsRoot", NULL);
 212         char *realm = strupper_talloc(mem_ctx, dnsdomain);
 213         struct ldb_dn *domain_dn = samdb_result_dn(mem_ctx, realm_ref_msg, "nCName", ldb_dn_new(mem_ctx));
 214
 215         struct hdb_ldb_private *private;
 216         NTTIME acct_expiry;
 217         struct ldb_message_element *ldb_keys;
 218
 219         struct ldb_message_element *objectclasses;
 220         struct ldb_val computer_val;
 221         computer_val.data = discard_const_p(uint8_t,"computer");
 222         computer_val.length = strlen((const char *)computer_val.data);
 223
 224         objectclasses = ldb_msg_find_element(msg, "objectClass");
 225
 226         if (objectclasses && ldb_msg_find_val(objectclasses, &computer_val)) {
 227                 is_computer = TRUE;
 228         }
 229
 230         memset(entry_ex, 0, sizeof(*entry_ex));
 231
 232         if (!realm) {
 233                 krb5_set_error_string(context, "talloc_strdup: out of memory");
 234                 ret = ENOMEM;
 235                 goto out;
 236         }
 237
 238         private = talloc(mem_ctx, struct hdb_ldb_private);
 239         if (!private) {
 240                 ret = ENOMEM;
 241                 goto out;
 242         }
 243
 244         private->entry_ex = entry_ex;
 245
 246         talloc_set_destructor(private, hdb_ldb_destrutor);
 247
 248         entry_ex->ctx = private;
 249         entry_ex->free_entry = hdb_ldb_free_entry;
 250
 251         userAccountControl = ldb_msg_find_attr_as_uint(msg, "userAccountControl", 0);
 252
 253
 254         entry_ex->entry.principal = malloc(sizeof(*(entry_ex->entry.principal)));
 255         if (ent_type == HDB_LDB_ENT_TYPE_ANY && principal == NULL) {
 256                 const char *samAccountName = ldb_msg_find_attr_as_string(msg, "samAccountName", NULL);
 257                 if (!samAccountName) {
 258                         krb5_set_error_string(context, "LDB_message2entry: no samAccountName present");
 259                         ret = ENOENT;
 260                         goto out;
 261                 }
 262                 samAccountName = ldb_msg_find_attr_as_string(msg, "samAccountName", NULL);
 263                 krb5_make_principal(context, &entry_ex->entry.principal, realm, samAccountName, NULL);
 264         } else {
 265                 char *strdup_realm;
 266                 ret = copy_Principal(principal, entry_ex->entry.principal);
 267                 if (ret) {
 268                         krb5_clear_error_string(context);
 269                         goto out;
 270                 }
 271
 272                 /* While we have copied the client principal, tests
 273                  * show that Win2k3 returns the 'corrected' realm, not
 274                  * the client-specified realm.  This code attempts to
 275                  * replace the client principal's realm with the one
 276                  * we determine from our records */
 277
 278                 /* this has to be with malloc() */
 279                 strdup_realm = strdup(realm);
 280                 if (!strdup_realm) {
 281                         ret = ENOMEM;
 282                         krb5_clear_error_string(context);
 283                         goto out;
 284                 }
 285                 free(*krb5_princ_realm(context, entry_ex->entry.principal));
 286                 krb5_princ_set_realm(context, entry_ex->entry.principal, &strdup_realm);
 287         }
 288
 289         entry_ex->entry.kvno = ldb_msg_find_attr_as_int(msg, "msDS-KeyVersionNumber", 0);
 290
 291         entry_ex->entry.flags = uf2HDBFlags(context, userAccountControl, ent_type);
 292
 293         if (ent_type == HDB_LDB_ENT_TYPE_KRBTGT) {
 294                 entry_ex->entry.flags.invalid = 0;
 295                 entry_ex->entry.flags.server = 1;
 296                 entry_ex->entry.flags.forwardable = 1;
 297                 entry_ex->entry.flags.ok_as_delegate = 1;
 298         }
 299
 300         if (lp_parm_bool(-1, "kdc", "require spn for service", True)) {
 301                 if (!is_computer && !ldb_msg_find_attr_as_string(msg, "servicePrincipalName", NULL)) {
 302                         entry_ex->entry.flags.server = 0;
 303                 }
 304         }
 305
 306         /* use 'whenCreated' */
 307         entry_ex->entry.created_by.time = ldb_msg_find_krb5time_ldap_time(msg, "whenCreated", 0);
 308         /* use '???' */
 309         entry_ex->entry.created_by.principal = NULL;
 310
 311         entry_ex->entry.modified_by = (Event *) malloc(sizeof(Event));
 312         if (entry_ex->entry.modified_by == NULL) {
 313                 krb5_set_error_string(context, "malloc: out of memory");
 314                 ret = ENOMEM;
 315                 goto out;
 316         }
 317
 318         /* use 'whenChanged' */
 319         entry_ex->entry.modified_by->time = ldb_msg_find_krb5time_ldap_time(msg, "whenChanged", 0);
 320         /* use '???' */
 321         entry_ex->entry.modified_by->principal = NULL;
 322
 323         entry_ex->entry.valid_start = NULL;
 324
 325         acct_expiry = samdb_result_nttime(msg, "accountExpires", (NTTIME)-1);
 326         if ((acct_expiry == (NTTIME)-1) ||
 327             (acct_expiry == 0x7FFFFFFFFFFFFFFFULL)) {
 328                 entry_ex->entry.valid_end = NULL;
 329         } else {
 330                 entry_ex->entry.valid_end = malloc(sizeof(*entry_ex->entry.valid_end));
 331                 if (entry_ex->entry.valid_end == NULL) {
 332                         ret = ENOMEM;
 333                         goto out;
 334                 }
 335                 *entry_ex->entry.valid_end = nt_time_to_unix(acct_expiry);
 336         }
 337
 338         if (ent_type != HDB_LDB_ENT_TYPE_KRBTGT) {
 339                 NTTIME must_change_time
 340                         = samdb_result_force_password_change((struct ldb_context *)db->hdb_db, mem_ctx,
 341                                                              domain_dn, msg);
 342                 if (must_change_time == 0x7FFFFFFFFFFFFFFFULL) {
 343                         entry_ex->entry.pw_end = NULL;
 344                 } else {
 345                         entry_ex->entry.pw_end = malloc(sizeof(*entry_ex->entry.pw_end));
 346                         if (entry_ex->entry.pw_end == NULL) {
 347                                 ret = ENOMEM;
 348                                 goto out;
 349                         }
 350                         *entry_ex->entry.pw_end = nt_time_to_unix(must_change_time);
 351                 }
 352         } else {
 353                 entry_ex->entry.pw_end = NULL;
 354         }
 355
 356         entry_ex->entry.max_life = NULL;
 357
 358         entry_ex->entry.max_renew = NULL;
 359
 360         entry_ex->entry.generation = NULL;
 361
 362         /* Get krb5Key from the db */
 363
 364         ldb_keys = ldb_msg_find_element(msg, "krb5Key");
 365
 366         if (!ldb_keys) {
 367                 /* oh, no password.  Apparently (comment in
 368                  * hdb-ldap.c) this violates the ASN.1, but this
 369                  * allows an entry with no keys (yet). */
 370                 entry_ex->entry.keys.val = NULL;
 371                 entry_ex->entry.keys.len = 0;
 372         } else {
 373                 /* allocate space to decode into */
 374                 entry_ex->entry.keys.val = calloc(ldb_keys->num_values, sizeof(Key));
 375                 if (entry_ex->entry.keys.val == NULL) {
 376                         ret = ENOMEM;
 377                         goto out;
 378                 }
 379
 380                 entry_ex->entry.keys.len = 0;
 381
 382                 /* Decode Kerberos keys into the hdb structure */
 383                 for (i=0; i < ldb_keys->num_values; i++) {
 384                         size_t decode_len;
 385                         Key key;
 386                         ret = decode_Key(ldb_keys->values[i].data, ldb_keys->values[i].length,
 387                                          &key, &decode_len);
 388                         if (ret) {
 389                                 /* Could be bougus data in the entry, or out of memory */
 390                                 goto out;
 391                         }
 392
 393                         if (userAccountControl & UF_USE_DES_KEY_ONLY) {
 394                                 switch (key.key.keytype) {
 395                                 case KEYTYPE_DES:
 396                                         entry_ex->entry.keys.val[entry_ex->entry.keys.len] = key;
 397                                         entry_ex->entry.keys.len++;
 398                                 default:
 399                                         /* We must use DES keys only */
 400                                         break;
 401                                 }
 402                         } else {
 403                                 entry_ex->entry.keys.val[entry_ex->entry.keys.len] = key;
 404                                 entry_ex->entry.keys.len++;
 405                         }
 406                 }
 407         }
 408
 409         entry_ex->entry.etypes = malloc(sizeof(*(entry_ex->entry.etypes)));
 410         if (entry_ex->entry.etypes == NULL) {
 411                 krb5_clear_error_string(context);
 412                 ret = ENOMEM;
 413                 goto out;
 414         }
 415         entry_ex->entry.etypes->len = entry_ex->entry.keys.len;
 416         entry_ex->entry.etypes->val = calloc(entry_ex->entry.etypes->len, sizeof(int));
 417         if (entry_ex->entry.etypes->val == NULL) {
 418                 krb5_clear_error_string(context);
 419                 ret = ENOMEM;
 420                 goto out;
 421         }
 422         for (i=0; i < entry_ex->entry.etypes->len; i++) {
 423                 entry_ex->entry.etypes->val[i] = entry_ex->entry.keys.val[i].key.keytype;
 424         }
 425
 426
 427         private->msg = talloc_steal(private, msg);
 428         private->realm_ref_msg = talloc_steal(private, realm_ref_msg);
 429         private->samdb = (struct ldb_context *)db->hdb_db;
 430
 431         entry_ex->check_client_access = hdb_ldb_check_client_access;
 432         entry_ex->authz_data_tgs_req = hdb_ldb_authz_data_tgs_req;
 433         entry_ex->authz_data_as_req = hdb_ldb_authz_data_as_req;
 434
 435 out:
 436         if (ret != 0) {
 437                 /* This doesn't free ent itself, that is for the eventual caller to do */
 438                 hdb_free_entry(context, entry_ex);
 439         } else {
 440                 talloc_steal(db, entry_ex->ctx);
 441         }
 442
 443         return ret;
 444 }
 445
 446 static krb5_error_code LDB_lookup_principal(krb5_context context, struct ldb_context *ldb_ctx,
 447                                             TALLOC_CTX *mem_ctx,
 448                                             krb5_const_principal principal,
 449                                             enum hdb_ldb_ent_type ent_type,
 450                                             const struct ldb_dn *realm_dn,
 451                                             struct ldb_message ***pmsg)
 452 {
 453         krb5_error_code ret;
 454         int lret;
 455         char *filter = NULL;
 456         const char * const *princ_attrs = krb5_attrs;
 457
 458         char *short_princ;
 459         char *short_princ_talloc;
 460
 461         char *realm_dn_str;
 462
 463         struct ldb_result *res = NULL;
 464
 465         ret = krb5_unparse_name_norealm(context, principal, &short_princ);
 466
 467         if (ret != 0) {
 468                 krb5_set_error_string(context, "LDB_lookup_principal: could not parse principal");
 469                 krb5_warnx(context, "LDB_lookup_principal: could not parse principal");
 470                 return ret;
 471         }
 472
 473         short_princ_talloc = talloc_strdup(mem_ctx, short_princ);
 474         free(short_princ);
 475         if (!short_princ_talloc) {
 476                 krb5_set_error_string(context, "LDB_lookup_principal: talloc_strdup() failed!");
 477                 return ENOMEM;
 478         }
 479
 480         switch (ent_type) {
 481         case HDB_LDB_ENT_TYPE_CLIENT:
 482                 /* Can't happen */
 483                 return EINVAL;
 484         case HDB_LDB_ENT_TYPE_ANY:
 485                 /* Can't happen */
 486                 return EINVAL;
 487         case HDB_LDB_ENT_TYPE_KRBTGT:
 488                 filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(samAccountName=%s))",
 489                                          KRB5_TGS_NAME);
 490                 break;
 491         case HDB_LDB_ENT_TYPE_SERVER:
 492                 filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(samAccountName=%s))",
 493                                          short_princ_talloc);
 494                 break;
 495         }
 496
 497         if (!filter) {
 498                 krb5_set_error_string(context, "talloc_asprintf: out of memory");
 499                 return ENOMEM;
 500         }
 501
 502         lret = ldb_search(ldb_ctx, realm_dn, LDB_SCOPE_SUBTREE, filter, princ_attrs, &res);
 503
 504         realm_dn_str = ldb_dn_linearize(mem_ctx, realm_dn);
 505
 506         if (lret != LDB_SUCCESS) {
 507                 DEBUG(3, ("Failed to search for %s: %s\n", filter, ldb_errstring(ldb_ctx)));
 508                 return HDB_ERR_NOENTRY;
 509         } else if (res->count == 0 || res->count > 1) {
 510                 DEBUG(3, ("Failed find a single entry for %s: got %d\n", filter, res->count));
 511                 return HDB_ERR_NOENTRY;
 512         }
 513         talloc_steal(mem_ctx, res->msgs);
 514         *pmsg = res->msgs;
 515         talloc_free(res);
 516         return 0;
 517 }
 518
 519 static krb5_error_code LDB_lookup_realm(krb5_context context, struct ldb_context *ldb_ctx,
 520                                         TALLOC_CTX *mem_ctx,
 521                                         const char *realm,
 522                                         struct ldb_message ***pmsg)
 523 {
 524         int ret;
 525         char *cross_ref_filter;
 526         struct ldb_result *cross_ref_res;
 527         const struct ldb_dn *partitions_basedn = samdb_partitions_dn(ldb_ctx, mem_ctx);
 528
 529         cross_ref_filter = talloc_asprintf(mem_ctx,
 530                                            "(&(&(|(&(dnsRoot=%s)(nETBIOSName=*))(nETBIOSName=%s))(objectclass=crossRef))(ncName=*))",
 531                                            realm, realm);
 532         if (!cross_ref_filter) {
 533                 krb5_set_error_string(context, "asprintf: out of memory");
 534                 return ENOMEM;
 535         }
 536
 537         ret = ldb_search(ldb_ctx, partitions_basedn, LDB_SCOPE_SUBTREE, cross_ref_filter, realm_ref_attrs, &cross_ref_res);
 538
 539         if (ret != LDB_SUCCESS) {
 540                 DEBUG(3, ("Failed to search for %s: %s\n", cross_ref_filter, ldb_errstring(ldb_ctx)));
 541                 talloc_free(cross_ref_res);
 542                 return HDB_ERR_NOENTRY;
 543         } else if (cross_ref_res->count == 0 || cross_ref_res->count > 1) {
 544                 DEBUG(3, ("Failed find a single entry for %s: got %d\n", cross_ref_filter, cross_ref_res->count));
 545                 talloc_free(cross_ref_res);
 546                 return HDB_ERR_NOENTRY;
 547         }
 548
 549         if (pmsg) {
 550                 *pmsg = cross_ref_res->msgs;
 551                 talloc_steal(mem_ctx, cross_ref_res->msgs);
 552         }
 553         talloc_free(cross_ref_res);
 554
 555         return 0;
 556 }
 557
 558
 559 static krb5_error_code LDB_open(krb5_context context, HDB *db, int flags, mode_t mode)
 560 {
 561         if (db->hdb_master_key_set) {
 562                 krb5_warnx(context, "LDB_open: use of a master key incompatible with LDB\n");
 563                 krb5_set_error_string(context, "LDB_open: use of a master key incompatible with LDB\n");
 564                 return HDB_ERR_NOENTRY;
 565         }
 566
 567         return 0;
 568 }
 569
 570 static krb5_error_code LDB_close(krb5_context context, HDB *db)
 571 {
 572         return 0;
 573 }
 574
 575 static krb5_error_code LDB_lock(krb5_context context, HDB *db, int operation)
 576 {
 577         return 0;
 578 }
 579
 580 static krb5_error_code LDB_unlock(krb5_context context, HDB *db)
 581 {
 582         return 0;
 583 }
 584
 585 static krb5_error_code LDB_rename(krb5_context context, HDB *db, const char *new_name)
 586 {
 587         return HDB_ERR_DB_INUSE;
 588 }
 589
 590 static krb5_error_code LDB_fetch_client(krb5_context context, HDB *db,
 591                                         TALLOC_CTX *mem_ctx,
 592                                         krb5_const_principal principal,
 593                                         unsigned flags,
 594                                         hdb_entry_ex *entry_ex) {
 595         NTSTATUS nt_status;
 596         char *principal_string;
 597         krb5_error_code ret;
 598         struct ldb_message **msg = NULL;
 599         struct ldb_message **realm_ref_msg = NULL;
 600
 601         ret = krb5_unparse_name(context, principal, &principal_string);
 602
 603         if (ret != 0) {
 604                 return ret;
 605         }
 606
 607         nt_status = sam_get_results_principal((struct ldb_context *)db->hdb_db,
 608                                               mem_ctx, principal_string,
 609                                               &msg, &realm_ref_msg);
 610         free(principal_string);
 611         if (NT_STATUS_EQUAL(nt_status, NT_STATUS_NO_SUCH_USER)) {
 612                 return HDB_ERR_NOENTRY;
 613         } else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_NO_MEMORY)) {
 614                 return ENOMEM;
 615         } else if (!NT_STATUS_IS_OK(nt_status)) {
 616                 return EINVAL;
 617         }
 618
 619         ret = LDB_message2entry(context, db, mem_ctx,
 620                                 principal, HDB_LDB_ENT_TYPE_CLIENT,
 621                                 msg[0], realm_ref_msg[0], entry_ex);
 622         return ret;
 623 }
 624
 625 static krb5_error_code LDB_fetch_krbtgt(krb5_context context, HDB *db,
 626                                         TALLOC_CTX *mem_ctx,
 627                                         krb5_const_principal principal,
 628                                         unsigned flags,
 629                                         hdb_entry_ex *entry_ex)
 630 {
 631         krb5_error_code ret;
 632         struct ldb_message **msg = NULL;
 633         struct ldb_message **realm_ref_msg = NULL;
 634         const struct ldb_dn *realm_dn;
 635
 636         krb5_principal alloc_principal = NULL;
 637         if (principal->name.name_string.len != 2
 638             || (strcmp(principal->name.name_string.val[0], KRB5_TGS_NAME) != 0)) {
 639                 /* Not a krbtgt */
 640                 return HDB_ERR_NOENTRY;
 641         }
 642
 643         /* krbtgt case.  Either us or a trusted realm */
 644         if ((LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db,
 645                               mem_ctx, principal->name.name_string.val[1], &realm_ref_msg) == 0)) {
 646                 /* us */
 647                 /* Cludge, cludge cludge.  If the realm part of krbtgt/realm,
 648                  * is in our db, then direct the caller at our primary
 649                  * krgtgt */
 650
 651                 const char *dnsdomain = ldb_msg_find_attr_as_string(realm_ref_msg[0], "dnsRoot", NULL);
 652                 char *realm_fixed = strupper_talloc(mem_ctx, dnsdomain);
 653                 if (!realm_fixed) {
 654                         krb5_set_error_string(context, "strupper_talloc: out of memory");
 655                         return ENOMEM;
 656                 }
 657
 658                 ret = krb5_copy_principal(context, principal, &alloc_principal);
 659                 if (ret) {
 660                         return ret;
 661                 }
 662
 663                 free(alloc_principal->name.name_string.val[1]);
 664                 alloc_principal->name.name_string.val[1] = strdup(realm_fixed);
 665                 talloc_free(realm_fixed);
 666                 if (!alloc_principal->name.name_string.val[1]) {
 667                         krb5_set_error_string(context, "LDB_fetch: strdup() failed!");
 668                         return ENOMEM;
 669                 }
 670                 principal = alloc_principal;
 671                 realm_dn = samdb_result_dn(mem_ctx, realm_ref_msg[0], "nCName", NULL);
 672
 673         } else {
 674                 /* we should lookup trusted domains */
 675                 return HDB_ERR_NOENTRY;
 676         }
 677
 678         realm_dn = samdb_result_dn(mem_ctx, realm_ref_msg[0], "nCName", NULL);
 679
 680         ret = LDB_lookup_principal(context, (struct ldb_context *)db->hdb_db,
 681                                    mem_ctx,
 682                                    principal, HDB_LDB_ENT_TYPE_KRBTGT, realm_dn, &msg);
 683
 684         if (ret != 0) {
 685                 krb5_warnx(context, "LDB_fetch: could not find principal in DB");
 686                 krb5_set_error_string(context, "LDB_fetch: could not find principal in DB");
 687                 return ret;
 688         }
 689
 690         ret = LDB_message2entry(context, db, mem_ctx,
 691                                 principal, HDB_LDB_ENT_TYPE_KRBTGT,
 692                                 msg[0], realm_ref_msg[0], entry_ex);
 693         if (ret != 0) {
 694                 krb5_warnx(context, "LDB_fetch: message2entry failed");
 695         }
 696         return ret;
 697 }
 698
 699 static krb5_error_code LDB_fetch_server(krb5_context context, HDB *db,
 700                                         TALLOC_CTX *mem_ctx,
 701                                         krb5_const_principal principal,
 702                                         unsigned flags,
 703                                         hdb_entry_ex *entry_ex)
 704 {
 705         krb5_error_code ret;
 706         const char *realm;
 707         struct ldb_message **msg = NULL;
 708         struct ldb_message **realm_ref_msg = NULL;
 709         const struct ldb_dn *partitions_basedn = samdb_partitions_dn(db->hdb_db, mem_ctx);
 710         if (principal->name.name_string.len >= 2) {
 711                 /* 'normal server' case */
 712                 int ldb_ret;
 713                 NTSTATUS nt_status;
 714                 struct ldb_dn *user_dn, *domain_dn;
 715                 char *principal_string;
 716
 717                 ret = krb5_unparse_name_norealm(context, principal, &principal_string);
 718                 if (ret != 0) {
 719                         return ret;
 720                 }
 721
 722                 /* At this point we may find the host is known to be
 723                  * in a different realm, so we should generate a
 724                  * referral instead */
 725                 nt_status = crack_service_principal_name((struct ldb_context *)db->hdb_db,
 726                                                          mem_ctx, principal_string,
 727                                                          &user_dn, &domain_dn);
 728                 free(principal_string);
 729
 730                 if (!NT_STATUS_IS_OK(nt_status)) {
 731                         return HDB_ERR_NOENTRY;
 732                 }
 733
 734                 ldb_ret = gendb_search_dn((struct ldb_context *)db->hdb_db,
 735                                           mem_ctx, user_dn, &msg, krb5_attrs);
 736
 737                 if (ldb_ret != 1) {
 738                         return HDB_ERR_NOENTRY;
 739                 }
 740
 741                 ldb_ret = gendb_search((struct ldb_context *)db->hdb_db,
 742                                        mem_ctx, partitions_basedn, &realm_ref_msg, realm_ref_attrs,
 743                                        "ncName=%s", ldb_dn_linearize(mem_ctx, domain_dn));
 744
 745                 if (ldb_ret != 1) {
 746                         return HDB_ERR_NOENTRY;
 747                 }
 748
 749         } else {
 750                 const struct ldb_dn *realm_dn;
 751                 /* server as client principal case, but we must not lookup userPrincipalNames */
 752
 753                 realm = krb5_principal_get_realm(context, principal);
 754
 755                 ret = LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db,
 756                                        mem_ctx, realm, &realm_ref_msg);
 757                 if (ret != 0) {
 758                         return HDB_ERR_NOENTRY;
 759                 }
 760
 761                 realm_dn = samdb_result_dn(mem_ctx, realm_ref_msg[0], "nCName", NULL);
 762
 763                 ret = LDB_lookup_principal(context, (struct ldb_context *)db->hdb_db,
 764                                            mem_ctx,
 765                                            principal, HDB_LDB_ENT_TYPE_SERVER, realm_dn, &msg);
 766
 767                 if (ret != 0) {
 768                         return ret;
 769                 }
 770         }
 771
 772         ret = LDB_message2entry(context, db, mem_ctx,
 773                                 principal, HDB_LDB_ENT_TYPE_SERVER,
 774                                 msg[0], realm_ref_msg[0], entry_ex);
 775         if (ret != 0) {
 776                 krb5_warnx(context, "LDB_fetch: message2entry failed");
 777         }
 778
 779         return ret;
 780 }
 781
 782 static krb5_error_code LDB_fetch(krb5_context context, HDB *db,
 783                                  krb5_const_principal principal,
 784                                  unsigned flags,
 785                                  hdb_entry_ex *entry_ex)
 786 {
 787         krb5_error_code ret = HDB_ERR_NOENTRY;
 788
 789         TALLOC_CTX *mem_ctx = talloc_named(db, 0, "LDB_fetch context");
 790
 791         if (!mem_ctx) {
 792                 krb5_set_error_string(context, "LDB_fetch: talloc_named() failed!");
 793                 return ENOMEM;
 794         }
 795
 796         if (flags & HDB_F_GET_CLIENT) {
 797                 ret = LDB_fetch_client(context, db, mem_ctx, principal, flags, entry_ex);
 798                 if (ret != HDB_ERR_NOENTRY) goto done;
 799         }
 800         if (flags & HDB_F_GET_SERVER) {
 801                 ret = LDB_fetch_server(context, db, mem_ctx, principal, flags, entry_ex);
 802                 if (ret != HDB_ERR_NOENTRY) goto done;
 803                 ret = LDB_fetch_krbtgt(context, db, mem_ctx, principal, flags, entry_ex);
 804                 if (ret != HDB_ERR_NOENTRY) goto done;
 805         }
 806         if (flags & HDB_F_GET_KRBTGT) {
 807                 ret = LDB_fetch_krbtgt(context, db, mem_ctx, principal, flags, entry_ex);
 808                 if (ret != HDB_ERR_NOENTRY) goto done;
 809         }
 810
 811 done:
 812         talloc_free(mem_ctx);
 813         return ret;
 814 }
 815
 816 static krb5_error_code LDB_store(krb5_context context, HDB *db, unsigned flags, hdb_entry_ex *entry)
 817 {
 818         return HDB_ERR_DB_INUSE;
 819 }
 820
 821 static krb5_error_code LDB_remove(krb5_context context, HDB *db, krb5_const_principal principal)
 822 {
 823         return HDB_ERR_DB_INUSE;
 824 }
 825
 826 struct hdb_ldb_seq {
 827         struct ldb_context *ctx;
 828         int index;
 829         int count;
 830         struct ldb_message **msgs;
 831         struct ldb_message **realm_ref_msgs;
 832 };
 833
 834 static krb5_error_code LDB_seq(krb5_context context, HDB *db, unsigned flags, hdb_entry_ex *entry)
 835 {
 836         krb5_error_code ret;
 837         struct hdb_ldb_seq *priv = (struct hdb_ldb_seq *)db->hdb_openp;
 838         TALLOC_CTX *mem_ctx;
 839         hdb_entry_ex entry_ex;
 840         memset(&entry_ex, '\0', sizeof(entry_ex));
 841
 842         if (!priv) {
 843                 return HDB_ERR_NOENTRY;
 844         }
 845
 846         mem_ctx = talloc_named(priv, 0, "LDB_seq context");
 847
 848         if (!mem_ctx) {
 849                 krb5_set_error_string(context, "LDB_seq: talloc_named() failed!");
 850                 return ENOMEM;
 851         }
 852
 853         if (priv->index < priv->count) {
 854                 ret = LDB_message2entry(context, db, mem_ctx,
 855                                         NULL, HDB_LDB_ENT_TYPE_ANY,
 856                                         priv->msgs[priv->index++],
 857                                         priv->realm_ref_msgs[0], entry);
 858         } else {
 859                 ret = HDB_ERR_NOENTRY;
 860         }
 861
 862         if (ret != 0) {
 863                 talloc_free(priv);
 864                 db->hdb_openp = NULL;
 865         } else {
 866                 talloc_free(mem_ctx);
 867         }
 868
 869         return ret;
 870 }
 871
 872 static krb5_error_code LDB_firstkey(krb5_context context, HDB *db, unsigned flags,
 873                                         hdb_entry_ex *entry)
 874 {
 875         struct ldb_context *ldb_ctx = (struct ldb_context *)db->hdb_db;
 876         struct hdb_ldb_seq *priv = (struct hdb_ldb_seq *)db->hdb_openp;
 877         char *realm;
 878         struct ldb_dn *realm_dn = NULL;
 879         struct ldb_result *res = NULL;
 880         struct ldb_message **realm_ref_msgs = NULL;
 881         krb5_error_code ret;
 882         TALLOC_CTX *mem_ctx;
 883         int lret;
 884
 885         if (priv) {
 886                 talloc_free(priv);
 887                 db->hdb_openp = 0;
 888         }
 889
 890         priv = (struct hdb_ldb_seq *) talloc(db, struct hdb_ldb_seq);
 891         if (!priv) {
 892                 krb5_set_error_string(context, "talloc: out of memory");
 893                 return ENOMEM;
 894         }
 895
 896         priv->ctx = ldb_ctx;
 897         priv->index = 0;
 898         priv->msgs = NULL;
 899         priv->realm_ref_msgs = NULL;
 900         priv->count = 0;
 901
 902         mem_ctx = talloc_named(priv, 0, "LDB_firstkey context");
 903
 904         if (!mem_ctx) {
 905                 krb5_set_error_string(context, "LDB_firstkey: talloc_named() failed!");
 906                 return ENOMEM;
 907         }
 908
 909         ret = krb5_get_default_realm(context, &realm);
 910         if (ret != 0) {
 911                 talloc_free(priv);
 912                 return ret;
 913         }
 914
 915         ret = LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db,
 916                                mem_ctx, realm, &realm_ref_msgs);
 917
 918         free(realm);
 919
 920         if (ret != 0) {
 921                 talloc_free(priv);
 922                 krb5_warnx(context, "LDB_firstkey: could not find realm\n");
 923                 return HDB_ERR_NOENTRY;
 924         }
 925
 926         realm_dn = samdb_result_dn(mem_ctx, realm_ref_msgs[0], "nCName", NULL);
 927
 928         priv->realm_ref_msgs = talloc_steal(priv, realm_ref_msgs);
 929
 930         lret = ldb_search(ldb_ctx, realm_dn,
 931                                  LDB_SCOPE_SUBTREE, "(objectClass=user)",
 932                                  krb5_attrs, &res);
 933
 934         if (lret != LDB_SUCCESS) {
 935                 talloc_free(priv);
 936                 return HDB_ERR_NOENTRY;
 937         }
 938
 939         priv->count = res->count;
 940         priv->msgs = talloc_steal(priv, res->msgs);
 941         talloc_free(res);
 942
 943         db->hdb_openp = priv;
 944
 945         ret = LDB_seq(context, db, flags, entry);
 946
 947         if (ret != 0) {
 948                 talloc_free(priv);
 949                 db->hdb_openp = NULL;
 950         } else {
 951                 talloc_free(mem_ctx);
 952         }
 953         return ret;
 954 }
 955
 956 static krb5_error_code LDB_nextkey(krb5_context context, HDB *db, unsigned flags,
 957                                    hdb_entry_ex *entry)
 958 {
 959         return LDB_seq(context, db, flags, entry);
 960 }
 961
 962 static krb5_error_code LDB_destroy(krb5_context context, HDB *db)
 963 {
 964         talloc_free(db);
 965         return 0;
 966 }
 967
 968 /* This interface is to be called by the KDC, which is expecting Samba
 969  * calling conventions.  It is also called by a wrapper
 970  * (hdb_ldb_create) from the kpasswdd -> krb5 -> keytab_hdb -> hdb
 971  * code */
 972
 973 NTSTATUS kdc_hdb_ldb_create(TALLOC_CTX *mem_ctx,
 974                             krb5_context context, struct HDB **db, const char *arg)
 975 {
 976         NTSTATUS nt_status;
 977         struct auth_session_info *session_info;
 978         *db = talloc(mem_ctx, HDB);
 979         if (!*db) {
 980                 krb5_set_error_string(context, "malloc: out of memory");
 981                 return NT_STATUS_NO_MEMORY;
 982         }
 983
 984         (*db)->hdb_master_key_set = 0;
 985         (*db)->hdb_db = NULL;
 986
 987         nt_status = auth_system_session_info(*db, &session_info);
 988         if (!NT_STATUS_IS_OK(nt_status)) {
 989                 return nt_status;
 990         }
 991
 992         /* The idea here is very simple.  Using Kerberos to
 993          * authenticate the KDC to the LDAP server is higly likely to
 994          * be circular.
 995          *
 996          * In future we may set this up to use EXERNAL and SSL
 997          * certificates, for now it will almost certainly be NTLMSSP
 998         */
 999
1000         cli_credentials_set_kerberos_state(session_info->credentials,
1001                                            CRED_DONT_USE_KERBEROS);
1002
1003         /* Setup the link to LDB */
1004         (*db)->hdb_db = samdb_connect(*db, session_info);
1005         if ((*db)->hdb_db == NULL) {
1006                 DEBUG(1, ("hdb_ldb_create: Cannot open samdb for KDC backend!"));
1007                 return NT_STATUS_CANT_ACCESS_DOMAIN_INFO;
1008         }
1009
1010         (*db)->hdb_openp = 0;
1011         (*db)->hdb_open = LDB_open;
1012         (*db)->hdb_close = LDB_close;
1013         (*db)->hdb_fetch = LDB_fetch;
1014         (*db)->hdb_store = LDB_store;
1015         (*db)->hdb_remove = LDB_remove;
1016         (*db)->hdb_firstkey = LDB_firstkey;
1017         (*db)->hdb_nextkey = LDB_nextkey;
1018         (*db)->hdb_lock = LDB_lock;
1019         (*db)->hdb_unlock = LDB_unlock;
1020         (*db)->hdb_rename = LDB_rename;
1021         /* we don't implement these, as we are not a lockable database */
1022         (*db)->hdb__get = NULL;
1023         (*db)->hdb__put = NULL;
1024         /* kadmin should not be used for deletes - use other tools instead */
1025         (*db)->hdb__del = NULL;
1026         (*db)->hdb_destroy = LDB_destroy;
1027
1028         return NT_STATUS_OK;
1029 }
1030
1031 krb5_error_code hdb_ldb_create(krb5_context context, struct HDB **db, const char *arg)
1032 {
1033         NTSTATUS nt_status;
1034         /* Disgusting, ugly hack, but it means one less private hook */
1035         nt_status = kdc_hdb_ldb_create(context->mem_ctx, context, db, arg);
1036
1037         if (NT_STATUS_IS_OK(nt_status)) {
1038                 return 0;
1039         }
1040         return EINVAL;
1041 }