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