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