source/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 "kdc.h"
  37 #include "ads.h"
  38 #include "hdb.h"
  39 #include "lib/ldb/include/ldb.h"
  40 #include "system/iconv.h"
  41
  42 enum hdb_ldb_ent_type
  43 { HDB_LDB_ENT_TYPE_CLIENT, HDB_LDB_ENT_TYPE_SERVER,
  44   HDB_LDB_ENT_TYPE_KRBTGT, HDB_LDB_ENT_TYPE_ANY };
  45
  46 static const char * const krb5_attrs[] = {
  47         "objectClass",
  48         "cn",
  49         "name",
  50         "sAMAccountName",
  51
  52         "userPrincipalName",
  53         "servicePrincipalName",
  54
  55         "userAccountControl",
  56         "sAMAccountType",
  57
  58         "objectSid",
  59         "primaryGroupID",
  60         "memberOf",
  61
  62         "unicodePWD",
  63         "lmPwdHash",
  64         "ntPwdHash",
  65
  66         "badPwdCount",
  67         "badPasswordTime",
  68         "lastLogoff",
  69         "lastLogon",
  70         "pwdLastSet",
  71         "accountExpires",
  72         "logonCount",
  73
  74         "objectGUID",
  75         "whenCreated",
  76         "whenChanged",
  77         "uSNCreated",
  78         "uSNChanged",
  79         "msDS-KeyVersionNumber",
  80         NULL
  81 };
  82
  83 static KerberosTime ldb_msg_find_krb5time_ldap_time(struct ldb_message *msg, const char *attr, KerberosTime default_val)
  84 {
  85     const char *tmp;
  86     const char *gentime;
  87     struct tm tm;
  88
  89     gentime = ldb_msg_find_string(msg, attr, NULL);
  90     if (!gentime)
  91         return default_val;
  92
  93     tmp = strptime(gentime, "%Y%m%d%H%M%SZ", &tm);
  94     if (tmp == NULL) {
  95             return default_val;
  96     }
  97
  98     return timegm(&tm);
  99 }
 100
 101 static HDBFlags uf2HDBFlags(krb5_context context, int userAccountControl, enum hdb_ldb_ent_type ent_type)
 102 {
 103         HDBFlags flags = int2HDBFlags(0);
 104
 105         krb5_warnx(context, "uf2HDBFlags: userAccountControl: %08x\n", userAccountControl);
 106
 107         /* we don't allow kadmin deletes */
 108         flags.immutable = 1;
 109
 110         /* mark the principal as invalid to start with */
 111         flags.invalid = 1;
 112
 113         flags.renewable = 1;
 114
 115         /* Account types - clear the invalid bit if it turns out to be valid */
 116         if (userAccountControl & UF_NORMAL_ACCOUNT) {
 117                 if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 118                         flags.client = 1;
 119                 }
 120                 flags.invalid = 0;
 121         }
 122
 123         if (userAccountControl & UF_INTERDOMAIN_TRUST_ACCOUNT) {
 124                 if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 125                         flags.client = 1;
 126                 }
 127                 flags.invalid = 0;
 128         }
 129         if (userAccountControl & UF_WORKSTATION_TRUST_ACCOUNT) {
 130                 if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 131                         flags.client = 1;
 132                 }
 133                 if (ent_type == HDB_LDB_ENT_TYPE_SERVER || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 134                         flags.server = 1;
 135                 }
 136                 flags.invalid = 0;
 137         }
 138         if (userAccountControl & UF_SERVER_TRUST_ACCOUNT) {
 139                 if (ent_type == HDB_LDB_ENT_TYPE_CLIENT || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 140                         flags.client = 1;
 141                 }
 142                 if (ent_type == HDB_LDB_ENT_TYPE_SERVER || ent_type == HDB_LDB_ENT_TYPE_ANY) {
 143                         flags.server = 1;
 144                 }
 145                 flags.invalid = 0;
 146         }
 147
 148         if (userAccountControl & UF_ACCOUNTDISABLE) {
 149                 flags.invalid = 1;
 150         }
 151         if (userAccountControl & UF_LOCKOUT) {
 152                 flags.invalid = 1;
 153         }
 154 /*
 155         if (userAccountControl & UF_PASSWORD_NOTREQD) {
 156                 flags.invalid = 1;
 157         }
 158 */
 159 /*
 160         if (userAccountControl & UF_PASSWORD_CANT_CHANGE) {
 161                 flags.invalid = 1;
 162         }
 163 */
 164 /*
 165         if (userAccountControl & UF_ENCRYPTED_TEXT_PASSWORD_ALLOWED) {
 166                 flags.invalid = 1;
 167         }
 168 */
 169         if (userAccountControl & UF_TEMP_DUPLICATE_ACCOUNT) {
 170                 flags.invalid = 1;
 171         }
 172
 173 /* UF_DONT_EXPIRE_PASSWD handled in LDB_message2entry() */
 174
 175 /*
 176         if (userAccountControl & UF_MNS_LOGON_ACCOUNT) {
 177                 flags.invalid = 1;
 178         }
 179 */
 180         if (userAccountControl & UF_SMARTCARD_REQUIRED) {
 181                 flags.require_hwauth = 1;
 182         }
 183         if (flags.server && (userAccountControl & UF_TRUSTED_FOR_DELEGATION)) {
 184                 flags.forwardable = 1;
 185                 flags.proxiable = 1;
 186         } else if (flags.client && (userAccountControl & UF_NOT_DELEGATED)) {
 187                 flags.forwardable = 0;
 188                 flags.proxiable = 0;
 189         } else {
 190                 flags.forwardable = 1;
 191                 flags.proxiable = 1;
 192         }
 193
 194 /*
 195         if (userAccountControl & UF_SMARTCARD_USE_DES_KEY_ONLY) {
 196                 flags.invalid = 1;
 197         }
 198 */
 199         if (userAccountControl & UF_DONT_REQUIRE_PREAUTH) {
 200                 flags.require_preauth = 0;
 201         } else {
 202                 flags.require_preauth = 1;
 203
 204         }
 205
 206         krb5_warnx(context, "uf2HDBFlags: HDBFlags: %08x\n", HDBFlags2int(flags));
 207
 208         return flags;
 209 }
 210
 211 /*
 212  * Construct an hdb_entry from a directory entry.
 213  */
 214 static krb5_error_code LDB_message2entry(krb5_context context, HDB *db,
 215                                          TALLOC_CTX *mem_ctx, krb5_const_principal principal,
 216                                          enum hdb_ldb_ent_type ent_type, struct ldb_message *realm_msg,
 217                                          struct ldb_message *msg,
 218                                          hdb_entry *ent)
 219 {
 220         const char *unicodePwd;
 221         int userAccountControl;
 222         int i;
 223         krb5_error_code ret = 0;
 224         const char *dnsdomain = ldb_msg_find_string(realm_msg, "dnsDomain", NULL);
 225         char *realm = strupper_talloc(mem_ctx, dnsdomain);
 226
 227         if (!realm) {
 228                 krb5_set_error_string(context, "talloc_strdup: out of memory");
 229                 ret = ENOMEM;
 230                 goto out;
 231         }
 232
 233         krb5_warnx(context, "LDB_message2entry:\n");
 234
 235         memset(ent, 0, sizeof(*ent));
 236
 237         userAccountControl = ldb_msg_find_int(msg, "userAccountControl", 0);
 238
 239         ent->principal = malloc(sizeof(*(ent->principal)));
 240         if (ent_type == HDB_LDB_ENT_TYPE_ANY && principal == NULL) {
 241                 const char *samAccountName = ldb_msg_find_string(msg, "samAccountName", NULL);
 242                 if (!samAccountName) {
 243                         krb5_set_error_string(context, "LDB_message2entry: no samAccountName present");
 244                         ret = ENOENT;
 245                         goto out;
 246                 }
 247                 samAccountName = ldb_msg_find_string(msg, "samAccountName", NULL);
 248                 krb5_make_principal(context, &ent->principal, realm, samAccountName, NULL);
 249         } else {
 250                 char *strdup_realm;
 251                 ret = copy_Principal(principal, ent->principal);
 252                 if (ret) {
 253                         krb5_clear_error_string(context);
 254                         goto out;
 255                 }
 256
 257                 /* While we have copied the client principal, tests
 258                  * show that Win2k3 returns the 'corrected' realm, not
 259                  * the client-specified realm.  This code attempts to
 260                  * replace the client principal's realm with the one
 261                  * we determine from our records */
 262
 263                 /* don't leak */
 264                 free(*krb5_princ_realm(context, ent->principal));
 265
 266                 /* this has to be with malloc() */
 267                 strdup_realm = strdup(realm);
 268                 if (!strdup_realm) {
 269                         ret = ENOMEM;
 270                         krb5_clear_error_string(context);
 271                         goto out;
 272                 }
 273                 krb5_princ_set_realm(context, ent->principal, &strdup_realm);
 274         }
 275
 276         ent->kvno = ldb_msg_find_int(msg, "msDS-KeyVersionNumber", 0);
 277
 278         ent->flags = uf2HDBFlags(context, userAccountControl, ent_type);
 279
 280         if (ent_type == HDB_LDB_ENT_TYPE_KRBTGT) {
 281                 ent->flags.invalid = 0;
 282                 ent->flags.server = 1;
 283         }
 284
 285         /* use 'whenCreated' */
 286         ent->created_by.time = ldb_msg_find_krb5time_ldap_time(msg, "whenCreated", 0);
 287         /* use '???' */
 288         ent->created_by.principal = NULL;
 289
 290         ent->modified_by = (Event *) malloc(sizeof(Event));
 291         if (ent->modified_by == NULL) {
 292                 krb5_set_error_string(context, "malloc: out of memory");
 293                 ret = ENOMEM;
 294                 goto out;
 295         }
 296
 297         /* use 'whenChanged' */
 298         ent->modified_by->time = ldb_msg_find_krb5time_ldap_time(msg, "whenChanged", 0);
 299         /* use '???' */
 300         ent->modified_by->principal = NULL;
 301
 302         ent->valid_start = NULL;
 303
 304         ent->valid_end = NULL;
 305         ent->pw_end = NULL;
 306
 307         ent->max_life = NULL;
 308
 309         ent->max_renew = NULL;
 310
 311         ent->generation = NULL;
 312
 313         /* create the keys and enctypes */
 314         unicodePwd = ldb_msg_find_string(msg, "unicodePwd", NULL);
 315         if (unicodePwd) {
 316                 /* Many, many thanks to lukeh@padl.com for this
 317                  * algorithm, described in his Nov 10 2004 mail to
 318                  * samba-technical@samba.org */
 319
 320                 Principal *salt_principal;
 321                 const char *user_principal_name = ldb_msg_find_string(msg, "userPrincipalName", NULL);
 322                 struct ldb_message_element *objectclasses;
 323                 struct ldb_val computer_val;
 324                 computer_val.data = discard_const_p(uint8_t,"computer");
 325                 computer_val.length = strlen((const char *)computer_val.data);
 326
 327                 objectclasses = ldb_msg_find_element(msg, "objectClass");
 328
 329                 if (objectclasses && ldb_msg_find_val(objectclasses, &computer_val)) {
 330                         /* Determine a salting principal */
 331                         char *samAccountName = talloc_strdup(mem_ctx, ldb_msg_find_string(msg, "samAccountName", NULL));
 332                         char *saltbody;
 333                         if (!samAccountName) {
 334                                 krb5_set_error_string(context, "LDB_message2entry: no samAccountName present");
 335                                 ret = ENOENT;
 336                                 goto out;
 337                         }
 338                         if (samAccountName[strlen(samAccountName)-1] == '$') {
 339                                 samAccountName[strlen(samAccountName)-1] = '\0';
 340                         }
 341                         saltbody = talloc_asprintf(mem_ctx, "%s.%s", samAccountName, dnsdomain);
 342
 343                         ret = krb5_make_principal(context, &salt_principal, realm, "host", saltbody, NULL);
 344                 } else if (user_principal_name) {
 345                         char *p;
 346                         user_principal_name = talloc_strdup(mem_ctx, user_principal_name);
 347                         if (!user_principal_name) {
 348                                 ret = ENOMEM;
 349                                 goto out;
 350                         } else {
 351                                 p = strchr(user_principal_name, '@');
 352                                 if (p) {
 353                                         p[0] = '\0';
 354                                 }
 355                                 ret = krb5_make_principal(context, &salt_principal, realm, user_principal_name, NULL);
 356                         }
 357                 } else {
 358                         const char *samAccountName = ldb_msg_find_string(msg, "samAccountName", NULL);
 359                         ret = krb5_make_principal(context, &salt_principal, realm, samAccountName, NULL);
 360                 }
 361
 362                 if (ret == 0) {
 363                         size_t num_keys = ent->keys.len;
 364                         /*
 365                          * create keys from unicodePwd
 366                          */
 367                         ret = hdb_generate_key_set_password(context, salt_principal,
 368                                                             unicodePwd,
 369                                                             &ent->keys.val, &num_keys);
 370                         ent->keys.len = num_keys;
 371                         krb5_free_principal(context, salt_principal);
 372                 }
 373
 374                 if (ret != 0) {
 375                         krb5_warnx(context, "could not generate keys from unicodePwd\n");
 376                         ent->keys.val = NULL;
 377                         ent->keys.len = 0;
 378                         goto out;
 379                 }
 380         } else {
 381                 const struct ldb_val *val;
 382                 krb5_data keyvalue;
 383
 384                 val = ldb_msg_find_ldb_val(msg, "ntPwdHash");
 385                 if (!val) {
 386                         krb5_warnx(context, "neither type of key available for this account\n");
 387                         ent->keys.val = NULL;
 388                         ent->keys.len = 0;
 389                 } else if (val->length < 16) {
 390                         ent->keys.val = NULL;
 391                         ent->keys.len = 0;
 392                         krb5_warnx(context, "ntPwdHash has invalid length: %d\n",
 393                                    (int)val->length);
 394                 } else {
 395                         ret = krb5_data_alloc (&keyvalue, 16);
 396                         if (ret) {
 397                                 krb5_set_error_string(context, "malloc: out of memory");
 398                                 ret = ENOMEM;
 399                                 goto out;
 400                         }
 401
 402                         memcpy(keyvalue.data, val->data, 16);
 403
 404                         ent->keys.val = malloc(sizeof(ent->keys.val[0]));
 405                         if (ent->keys.val == NULL) {
 406                                 krb5_data_free(&keyvalue);
 407                                 krb5_set_error_string(context, "malloc: out of memory");
 408                                 ret = ENOMEM;
 409                                 goto out;
 410                         }
 411
 412                         memset(&ent->keys.val[0], 0, sizeof(Key));
 413                         ent->keys.val[0].key.keytype = ETYPE_ARCFOUR_HMAC_MD5;
 414                         ent->keys.val[0].key.keyvalue = keyvalue;
 415
 416                         ent->keys.len = 1;
 417                 }
 418         }
 419
 420
 421         ent->etypes = malloc(sizeof(*(ent->etypes)));
 422         if (ent->etypes == NULL) {
 423                 krb5_set_error_string(context, "malloc: out of memory");
 424                 ret = ENOMEM;
 425                 goto out;
 426         }
 427         ent->etypes->len = ent->keys.len;
 428         ent->etypes->val = calloc(ent->etypes->len, sizeof(int));
 429         if (ent->etypes->val == NULL) {
 430                 krb5_set_error_string(context, "malloc: out of memory");
 431                 ret = ENOMEM;
 432                 goto out;
 433         }
 434         for (i=0; i < ent->etypes->len; i++) {
 435                 ent->etypes->val[i] = ent->keys.val[i].key.keytype;
 436         }
 437
 438 out:
 439         if (ret != 0) {
 440                 /* I don't think this frees ent itself. */
 441                 hdb_free_entry(context, ent);
 442         }
 443
 444         return ret;
 445 }
 446
 447 static krb5_error_code LDB_lookup_principal(krb5_context context, struct ldb_context *ldb_ctx,
 448                                             TALLOC_CTX *mem_ctx,
 449                                             krb5_const_principal principal,
 450                                             enum hdb_ldb_ent_type ent_type,
 451                                             const char *realm_dn,
 452                                             struct ldb_message ***pmsg)
 453 {
 454         krb5_error_code ret;
 455         int count;
 456         char *filter = NULL;
 457         const char * const *princ_attrs = krb5_attrs;
 458         char *p;
 459
 460         char *princ_str;
 461         char *princ_str_talloc;
 462         char *short_princ;
 463
 464         struct ldb_message **msg = NULL;
 465
 466         /* Structure assignment, so we don't mess with the source parameter */
 467         struct Principal princ = *principal;
 468
 469         /* Allow host/dns.name/realm@REALM, just convert into host/dns.name@REALM */
 470         if (princ.name.name_string.len == 3
 471             && StrCaseCmp(princ.name.name_string.val[2], princ.realm) == 0) {
 472                 princ.name.name_string.len = 2;
 473         }
 474
 475         ret = krb5_unparse_name(context, &princ, &princ_str);
 476
 477         if (ret != 0) {
 478                 krb5_set_error_string(context, "LDB_lookup_principal: could not parse principal");
 479                 krb5_warnx(context, "LDB_lookup_principal: could not parse principal");
 480                 return ret;
 481         }
 482
 483         princ_str_talloc = talloc_strdup(mem_ctx, princ_str);
 484         short_princ = talloc_strdup(mem_ctx, princ_str);
 485         free(princ_str);
 486         if (!short_princ || !princ_str_talloc) {
 487                 krb5_set_error_string(context, "LDB_lookup_principal: talloc_strdup() failed!");
 488                 return ENOMEM;
 489         }
 490
 491         p = strchr(short_princ, '@');
 492         if (p) {
 493                 p[0] = '\0';
 494         }
 495
 496
 497         switch (ent_type) {
 498         case HDB_LDB_ENT_TYPE_KRBTGT:
 499                 filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(samAccountName=%s))",
 500                                          KRB5_TGS_NAME);
 501                 break;
 502         case HDB_LDB_ENT_TYPE_CLIENT:
 503                 filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(|(samAccountName=%s)(userPrincipalName=%s)))",
 504                                          short_princ, princ_str_talloc);
 505                 break;
 506         case HDB_LDB_ENT_TYPE_SERVER:
 507                 filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(|(samAccountName=%s)(servicePrincipalName=%s)))",
 508                                          short_princ, short_princ);
 509                 break;
 510         case HDB_LDB_ENT_TYPE_ANY:
 511                 filter = talloc_asprintf(mem_ctx, "(&(objectClass=user)(|(|(samAccountName=%s)(servicePrincipalName=%s))(userPrincipalName=%s)))",
 512                                          short_princ, short_princ, princ_str_talloc);
 513                 break;
 514         }
 515
 516         if (!filter) {
 517                 krb5_set_error_string(context, "talloc_asprintf: out of memory");
 518                 return ENOMEM;
 519         }
 520
 521         count = ldb_search(ldb_ctx, realm_dn, LDB_SCOPE_SUBTREE, filter,
 522                            princ_attrs, &msg);
 523
 524         if (count < 1) {
 525                 krb5_warnx(context, "ldb_search: basedn: '%s' filter: '%s' failed: %d",
 526                            realm_dn, filter, count);
 527                 krb5_set_error_string(context, "ldb_search: basedn: '%s' filter: '%s' failed: %d",
 528                                       realm_dn, filter, count);
 529                 return HDB_ERR_NOENTRY;
 530         } else if (count > 1) {
 531                 talloc_free(msg);
 532                 krb5_warnx(context, "ldb_search: basedn: '%s' filter: '%s' more than 1 entry: %d",
 533                            realm_dn, filter, count);
 534                 krb5_set_error_string(context, "ldb_search: basedn: '%s' filter: '%s' more than 1 entry: %d",
 535                                       realm_dn, filter, count);
 536                 return HDB_ERR_NOENTRY;
 537         }
 538         *pmsg = talloc_steal(mem_ctx, msg);
 539         return 0;
 540 }
 541
 542 static krb5_error_code LDB_lookup_realm(krb5_context context, struct ldb_context *ldb_ctx,
 543                                         TALLOC_CTX *mem_ctx,
 544                                         const char *realm,
 545                                         struct ldb_message ***pmsg)
 546 {
 547         int count;
 548         const char *realm_dn;
 549         char *cross_ref_filter;
 550         struct ldb_message **cross_ref_msg;
 551         struct ldb_message **msg;
 552
 553         const char *cross_ref_attrs[] = {
 554                 "nCName",
 555                 NULL
 556         };
 557
 558         const char *realm_attrs[] = {
 559                 "dnsDomain",
 560                 "maxPwdAge",
 561                 NULL
 562         };
 563
 564         cross_ref_filter = talloc_asprintf(mem_ctx,
 565                                            "(&(&(|(&(dnsRoot=%s)(nETBIOSName=*))(nETBIOSName=%s))(objectclass=crossRef))(ncName=*))",
 566                                            realm, realm);
 567         if (!cross_ref_filter) {
 568                 krb5_set_error_string(context, "asprintf: out of memory");
 569                 return ENOMEM;
 570         }
 571
 572         count = ldb_search(ldb_ctx, NULL, LDB_SCOPE_SUBTREE, cross_ref_filter,
 573                            cross_ref_attrs, &cross_ref_msg);
 574
 575         if (count < 1) {
 576                 krb5_warnx(context, "ldb_search: filter: '%s' failed: %d", cross_ref_filter, count);
 577                 krb5_set_error_string(context, "ldb_search: filter: '%s' failed: %d", cross_ref_filter, count);
 578
 579                 talloc_free(cross_ref_msg);
 580                 return HDB_ERR_NOENTRY;
 581         } else if (count > 1) {
 582                 krb5_warnx(context, "ldb_search: filter: '%s' more than 1 entry: %d", cross_ref_filter, count);
 583                 krb5_set_error_string(context, "ldb_search: filter: '%s' more than 1 entry: %d", cross_ref_filter, count);
 584
 585                 talloc_free(cross_ref_msg);
 586                 return HDB_ERR_NOENTRY;
 587         }
 588
 589         realm_dn = ldb_msg_find_string(cross_ref_msg[0], "nCName", NULL);
 590
 591         count = ldb_search(ldb_ctx, realm_dn, LDB_SCOPE_BASE, "(objectClass=domain)",
 592                            realm_attrs, &msg);
 593         if (pmsg) {
 594                 *pmsg = talloc_steal(mem_ctx, msg);
 595         } else {
 596                 talloc_free(msg);
 597         }
 598
 599         if (count < 1) {
 600                 krb5_warnx(context, "ldb_search: dn: %s not found: %d", realm_dn, count);
 601                 krb5_set_error_string(context, "ldb_search: dn: %s not found: %d", realm_dn, count);
 602                 return HDB_ERR_NOENTRY;
 603         } else if (count > 1) {
 604                 krb5_warnx(context, "ldb_search: dn: '%s' more than 1 entry: %d", realm_dn, count);
 605                 krb5_set_error_string(context, "ldb_search: dn: %s more than 1 entry: %d", realm_dn, count);
 606                 return HDB_ERR_NOENTRY;
 607         }
 608
 609         return 0;
 610 }
 611
 612 static krb5_error_code LDB_lookup_spn_alias(krb5_context context, struct ldb_context *ldb_ctx,
 613                                             TALLOC_CTX *mem_ctx,
 614                                             const char *realm_dn,
 615                                             const char *alias_from,
 616                                             char **alias_to)
 617 {
 618         int i;
 619         int count;
 620         struct ldb_message **msg;
 621         struct ldb_message_element *spnmappings;
 622         char *service_dn = talloc_asprintf(mem_ctx,
 623                                            "CN=Directory Service,CN=Windows NT,CN=Services,CN=Configuration,%s",
 624                                            realm_dn);
 625         const char *directory_attrs[] = {
 626                 "sPNMappings",
 627                 NULL
 628         };
 629
 630         count = ldb_search(ldb_ctx, service_dn, LDB_SCOPE_BASE, "(objectClass=nTDSService)",
 631                            directory_attrs, &msg);
 632         talloc_steal(mem_ctx, msg);
 633
 634         if (count < 1) {
 635                 krb5_warnx(context, "ldb_search: dn: %s not found: %d", service_dn, count);
 636                 krb5_set_error_string(context, "ldb_search: dn: %s not found: %d", service_dn, count);
 637                 return HDB_ERR_NOENTRY;
 638         } else if (count > 1) {
 639                 krb5_warnx(context, "ldb_search: dn: %s found %d times!", service_dn, count);
 640                 krb5_set_error_string(context, "ldb_search: dn: %s found %d times!", service_dn, count);
 641                 return HDB_ERR_NOENTRY;
 642         }
 643
 644         spnmappings = ldb_msg_find_element(msg[0], "sPNMappings");
 645         if (!spnmappings || spnmappings->num_values == 0) {
 646                 krb5_warnx(context, "ldb_search: dn: %s no sPNMappings attribute", service_dn);
 647                 krb5_set_error_string(context, "ldb_search: dn: %s no sPNMappings attribute", service_dn);
 648         }
 649
 650         for (i = 0; i < spnmappings->num_values; i++) {
 651                 char *mapping, *p, *str;
 652                 mapping = talloc_strdup(mem_ctx,
 653                                         (const char *)spnmappings->values[i].data);
 654                 if (!mapping) {
 655                         krb5_warnx(context, "LDB_lookup_spn_alias: ldb_search: dn: %s did not have an sPNMapping", service_dn);
 656                         krb5_set_error_string(context, "LDB_lookup_spn_alias: ldb_search: dn: %s did not have an sPNMapping", service_dn);
 657                         return HDB_ERR_NOENTRY;
 658                 }
 659
 660                 /* C string manipulation sucks */
 661
 662                 p = strchr(mapping, '=');
 663                 if (!p) {
 664                         krb5_warnx(context, "ldb_search: dn: %s sPNMapping malformed: %s",
 665                                    service_dn, mapping);
 666                         krb5_set_error_string(context, "ldb_search: dn: %s sPNMapping malformed: %s",
 667                                               service_dn, mapping);
 668                 }
 669                 p[0] = '\0';
 670                 p++;
 671                 do {
 672                         str = p;
 673                         p = strchr(p, ',');
 674                         if (p) {
 675                                 p[0] = '\0';
 676                                 p++;
 677                         }
 678                         if (strcasecmp(str, alias_from) == 0) {
 679                                 *alias_to = mapping;
 680                                 return 0;
 681                         }
 682                 } while (p);
 683         }
 684         krb5_warnx(context, "LDB_lookup_spn_alias: no alias for service %s applicable", alias_from);
 685         return HDB_ERR_NOENTRY;
 686 }
 687
 688 static krb5_error_code LDB_open(krb5_context context, HDB *db, int flags, mode_t mode)
 689 {
 690         if (db->hdb_master_key_set) {
 691                 krb5_warnx(context, "LDB_open: use of a master key incompatible with LDB\n");
 692                 krb5_set_error_string(context, "LDB_open: use of a master key incompatible with LDB\n");
 693                 return HDB_ERR_NOENTRY;
 694         }
 695
 696         return 0;
 697 }
 698
 699 static krb5_error_code LDB_close(krb5_context context, HDB *db)
 700 {
 701         return 0;
 702 }
 703
 704 static krb5_error_code LDB_lock(krb5_context context, HDB *db, int operation)
 705 {
 706         return 0;
 707 }
 708
 709 static krb5_error_code LDB_unlock(krb5_context context, HDB *db)
 710 {
 711         return 0;
 712 }
 713
 714 static krb5_error_code LDB_rename(krb5_context context, HDB *db, const char *new_name)
 715 {
 716         return HDB_ERR_DB_INUSE;
 717 }
 718
 719 static krb5_error_code LDB_fetch(krb5_context context, HDB *db, unsigned flags,
 720                                  krb5_const_principal principal,
 721                                  enum hdb_ent_type ent_type,
 722                                  hdb_entry *entry)
 723 {
 724         struct ldb_message **msg = NULL;
 725         struct ldb_message **realm_msg = NULL;
 726         struct ldb_message **realm_fixed_msg = NULL;
 727         enum hdb_ldb_ent_type ldb_ent_type;
 728         krb5_error_code ret;
 729
 730         const char *realm;
 731         const char *realm_dn;
 732         TALLOC_CTX *mem_ctx = talloc_named(NULL, 0, "LDB_fetch context");
 733
 734         if (!mem_ctx) {
 735                 krb5_set_error_string(context, "LDB_fetch: talloc_named() failed!");
 736                 return ENOMEM;
 737         }
 738
 739         realm = krb5_principal_get_realm(context, principal);
 740
 741         ret = LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db,
 742                                mem_ctx, realm, &realm_msg);
 743         if (ret != 0) {
 744                 krb5_warnx(context, "LDB_fetch: could not find realm");
 745                 talloc_free(mem_ctx);
 746                 return HDB_ERR_NOENTRY;
 747         }
 748
 749         realm_dn = realm_msg[0]->dn;
 750
 751         /* Cludge, cludge cludge.  If the realm part of krbtgt/realm,
 752          * is in our db, then direct the caller at our primary
 753          * krgtgt */
 754
 755         switch (ent_type) {
 756         case HDB_ENT_TYPE_SERVER:
 757                 if (principal->name.name_string.len == 2
 758                     && (strcmp(principal->name.name_string.val[0], KRB5_TGS_NAME) == 0)
 759                     && (LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db,
 760                                          mem_ctx, principal->name.name_string.val[1], &realm_fixed_msg) == 0)) {
 761                         const char *dnsdomain = ldb_msg_find_string(realm_fixed_msg[0], "dnsDomain", NULL);
 762                         char *realm_fixed = strupper_talloc(mem_ctx, dnsdomain);
 763                         if (!realm_fixed) {
 764                                 krb5_set_error_string(context, "strupper_talloc: out of memory");
 765                                 talloc_free(mem_ctx);
 766                                 return ENOMEM;
 767                         }
 768
 769                         free(principal->name.name_string.val[1]);
 770                         principal->name.name_string.val[1] = strdup(realm_fixed);
 771                         talloc_free(realm_fixed);
 772                         if (!principal->name.name_string.val[1]) {
 773                                 krb5_set_error_string(context, "LDB_fetch: strdup() failed!");
 774                                 talloc_free(mem_ctx);
 775                                 return ENOMEM;
 776                         }
 777                         ldb_ent_type = HDB_LDB_ENT_TYPE_KRBTGT;
 778                 } else {
 779                         ldb_ent_type = HDB_LDB_ENT_TYPE_SERVER;
 780                 }
 781                 break;
 782         case HDB_ENT_TYPE_CLIENT:
 783                 ldb_ent_type = HDB_LDB_ENT_TYPE_CLIENT;
 784                 break;
 785         case HDB_ENT_TYPE_ANY:
 786                 ldb_ent_type = HDB_LDB_ENT_TYPE_ANY;
 787                 break;
 788         default:
 789                 krb5_warnx(context, "LDB_fetch: invalid ent_type specified!");
 790                 talloc_free(mem_ctx);
 791                 return HDB_ERR_NOENTRY;
 792         }
 793
 794         ret = LDB_lookup_principal(context, (struct ldb_context *)db->hdb_db,
 795                                    mem_ctx,
 796                                    principal, ldb_ent_type, realm_dn, &msg);
 797
 798         if (ret != 0) {
 799                 char *alias_from = principal->name.name_string.val[0];
 800                 char *alias_to;
 801                 Principal alias_principal;
 802
 803                 /* Try again with a servicePrincipal alias */
 804                 if (ent_type != HDB_LDB_ENT_TYPE_SERVER && ent_type != HDB_LDB_ENT_TYPE_ANY) {
 805                         talloc_free(mem_ctx);
 806                         return ret;
 807                 }
 808                 if (principal->name.name_string.len < 2) {
 809                         krb5_warnx(context, "LDB_fetch: could not find principal in DB, alias not applicable");
 810                         krb5_set_error_string(context, "LDB_fetch: could not find principal in DB, alias not applicable");
 811                         talloc_free(mem_ctx);
 812                         return ret;
 813                 }
 814
 815                 /* Look for the list of aliases */
 816                 ret = LDB_lookup_spn_alias(context,
 817                                            (struct ldb_context *)db->hdb_db, mem_ctx,
 818                                            realm_dn, alias_from,
 819                                            &alias_to);
 820                 if (ret != 0) {
 821                         talloc_free(mem_ctx);
 822                         return ret;
 823                 }
 824
 825                 ret = copy_Principal(principal, &alias_principal);
 826                 if (ret != 0) {
 827                         krb5_warnx(context, "LDB_fetch: could not copy principal");
 828                         krb5_set_error_string(context, "LDB_fetch: could not copy principal");
 829                         talloc_free(mem_ctx);
 830                         return ret;
 831                 }
 832
 833                 /* ooh, very nasty playing around in the Principal... */
 834                 free(alias_principal.name.name_string.val[0]);
 835                 alias_principal.name.name_string.val[0] = strdup(alias_to);
 836                 if (!alias_principal.name.name_string.val[0]) {
 837                         krb5_warnx(context, "LDB_fetch: strdup() failed");
 838                         krb5_set_error_string(context, "LDB_fetch: strdup() failed");
 839                         ret = ENOMEM;
 840                         talloc_free(mem_ctx);
 841                         free_Principal(&alias_principal);
 842                         return ret;
 843                 }
 844
 845                 ret = LDB_lookup_principal(context, (struct ldb_context *)db->hdb_db,
 846                                            mem_ctx,
 847                                            &alias_principal, ent_type, realm_dn, &msg);
 848                 free_Principal(&alias_principal);
 849
 850                 if (ret != 0) {
 851                         krb5_warnx(context, "LDB_fetch: could not find alias principal in DB");
 852                         krb5_set_error_string(context, "LDB_fetch: could not find alias principal in DB");
 853                         talloc_free(mem_ctx);
 854                         return ret;
 855                 }
 856
 857         }
 858         if (ret == 0) {
 859                 ret = LDB_message2entry(context, db, mem_ctx,
 860                                         principal, ldb_ent_type,
 861                                         realm_msg[0], msg[0], entry);
 862                 if (ret != 0) {
 863                         krb5_warnx(context, "LDB_fetch: message2entry failed\n");
 864                 }
 865         }
 866
 867         talloc_free(mem_ctx);
 868         return ret;
 869 }
 870
 871 static krb5_error_code LDB_store(krb5_context context, HDB *db, unsigned flags, hdb_entry *entry)
 872 {
 873         return HDB_ERR_DB_INUSE;
 874 }
 875
 876 static krb5_error_code LDB_remove(krb5_context context, HDB *db, hdb_entry *entry)
 877 {
 878         return HDB_ERR_DB_INUSE;
 879 }
 880
 881 struct hdb_ldb_seq {
 882         struct ldb_context *ctx;
 883         int index;
 884         int count;
 885         struct ldb_message **msgs;
 886         struct ldb_message **realm_msgs;
 887 };
 888
 889 static krb5_error_code LDB_seq(krb5_context context, HDB *db, unsigned flags, hdb_entry *entry)
 890 {
 891         krb5_error_code ret;
 892         struct hdb_ldb_seq *priv = (struct hdb_ldb_seq *)db->hdb_openp;
 893         TALLOC_CTX *mem_ctx;
 894         if (!priv) {
 895                 return HDB_ERR_NOENTRY;
 896         }
 897
 898         mem_ctx = talloc_named(priv, 0, "LDB_seq context");
 899
 900         if (!mem_ctx) {
 901                 krb5_set_error_string(context, "LDB_seq: talloc_named() failed!");
 902                 return ENOMEM;
 903         }
 904
 905         if (priv->index < priv->count) {
 906                 ret = LDB_message2entry(context, db, mem_ctx,
 907                                         NULL, HDB_LDB_ENT_TYPE_ANY,
 908                                         priv->realm_msgs[0], priv->msgs[priv->index++], entry);
 909         } else {
 910                 ret = HDB_ERR_NOENTRY;
 911         }
 912
 913         if (ret != 0) {
 914                 talloc_free(priv);
 915                 db->hdb_openp = NULL;
 916         } else {
 917                 talloc_free(mem_ctx);
 918         }
 919
 920         return ret;
 921 }
 922
 923 static krb5_error_code LDB_firstkey(krb5_context context, HDB *db, unsigned flags,
 924                                         hdb_entry *entry)
 925 {
 926         struct ldb_context *ldb_ctx = (struct ldb_context *)db->hdb_db;
 927         struct hdb_ldb_seq *priv = (struct hdb_ldb_seq *)db->hdb_openp;
 928         char *realm;
 929         char *realm_dn = NULL;
 930         struct ldb_message **msgs = NULL;
 931         struct ldb_message **realm_msgs = NULL;
 932         krb5_error_code ret;
 933         TALLOC_CTX *mem_ctx;
 934
 935         if (priv) {
 936                 talloc_free(priv);
 937                 db->hdb_openp = 0;
 938         }
 939
 940         priv = (struct hdb_ldb_seq *) talloc(db, struct hdb_ldb_seq);
 941         if (!priv) {
 942                 krb5_set_error_string(context, "talloc: out of memory");
 943                 return ENOMEM;
 944         }
 945
 946         priv->ctx = ldb_ctx;
 947         priv->index = 0;
 948         priv->msgs = NULL;
 949         priv->realm_msgs = NULL;
 950         priv->count = 0;
 951
 952         mem_ctx = talloc_named(priv, 0, "LDB_firstkey context");
 953
 954         if (!mem_ctx) {
 955                 krb5_set_error_string(context, "LDB_firstkey: talloc_named() failed!");
 956                 return ENOMEM;
 957         }
 958
 959         ret = krb5_get_default_realm(context, &realm);
 960         if (ret != 0) {
 961                 talloc_free(priv);
 962                 return ret;
 963         }
 964
 965         ret = LDB_lookup_realm(context, (struct ldb_context *)db->hdb_db,
 966                                mem_ctx, realm, &realm_msgs);
 967
 968         free(realm);
 969
 970         if (ret != 0) {
 971                 talloc_free(priv);
 972                 krb5_warnx(context, "LDB_firstkey: could not find realm\n");
 973                 return HDB_ERR_NOENTRY;
 974         }
 975
 976         realm_dn = realm_msgs[0]->dn;
 977
 978         priv->realm_msgs = talloc_steal(priv, realm_msgs);
 979
 980         krb5_warnx(context, "LDB_firstkey: realm ok\n");
 981
 982         priv->count = ldb_search(ldb_ctx, realm_dn,
 983                                  LDB_SCOPE_SUBTREE, "(objectClass=user)",
 984                                  krb5_attrs, &msgs);
 985
 986         priv->msgs = talloc_steal(priv, msgs);
 987
 988         if (priv->count <= 0) {
 989                 talloc_free(priv);
 990                 return HDB_ERR_NOENTRY;
 991         }
 992
 993         db->hdb_openp = priv;
 994
 995         ret = LDB_seq(context, db, flags, entry);
 996
 997         if (ret != 0) {
 998                 talloc_free(priv);
 999                 db->hdb_openp = NULL;
1000         } else {
1001                 talloc_free(mem_ctx);
1002         }
1003         return ret;
1004 }
1005
1006 static krb5_error_code LDB_nextkey(krb5_context context, HDB *db, unsigned flags,
1007                                         hdb_entry *entry)
1008 {
1009         return LDB_seq(context, db, flags, entry);
1010 }
1011
1012 static krb5_error_code LDB_destroy(krb5_context context, HDB *db)
1013 {
1014         talloc_free(db);
1015         return 0;
1016 }
1017
1018 krb5_error_code hdb_ldb_create(TALLOC_CTX *mem_ctx,
1019                                krb5_context context, struct HDB **db, const char *arg)
1020 {
1021         *db = talloc(mem_ctx, HDB);
1022         if (!*db) {
1023                 krb5_set_error_string(context, "malloc: out of memory");
1024                 return ENOMEM;
1025         }
1026
1027         (*db)->hdb_master_key_set = 0;
1028         (*db)->hdb_db = NULL;
1029
1030         /* Setup the link to LDB */
1031         (*db)->hdb_db = samdb_connect(db);
1032         if ((*db)->hdb_db == NULL) {
1033                 krb5_warnx(context, "hdb_ldb_create: samdb_connect failed!");
1034                 krb5_set_error_string(context, "samdb_connect failed!");
1035                 talloc_free(*db);
1036                 return HDB_ERR_NOENTRY;
1037         }
1038
1039         (*db)->hdb_openp = 0;
1040         (*db)->hdb_open = LDB_open;
1041         (*db)->hdb_close = LDB_close;
1042         (*db)->hdb_fetch = LDB_fetch;
1043         (*db)->hdb_store = LDB_store;
1044         (*db)->hdb_remove = LDB_remove;
1045         (*db)->hdb_firstkey = LDB_firstkey;
1046         (*db)->hdb_nextkey = LDB_nextkey;
1047         (*db)->hdb_lock = LDB_lock;
1048         (*db)->hdb_unlock = LDB_unlock;
1049         (*db)->hdb_rename = LDB_rename;
1050         /* we don't implement these, as we are not a lockable database */
1051         (*db)->hdb__get = NULL;
1052         (*db)->hdb__put = NULL;
1053         /* kadmin should not be used for deletes - use other tools instead */
1054         (*db)->hdb__del = NULL;
1055         (*db)->hdb_destroy = LDB_destroy;
1056
1057         return 0;
1058 }