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