backend to read the MIT Kerberos databse file directly
This commit is contained in:
		
							
								
								
									
										743
									
								
								lib/hdb/hdb-mitdb.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										743
									
								
								lib/hdb/hdb-mitdb.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,743 @@ | |||||||
|  | /* | ||||||
|  |  * Copyright (c) 1997 - 2001 Kungliga Tekniska Högskolan | ||||||
|  |  * (Royal Institute of Technology, Stockholm, Sweden). | ||||||
|  |  * All rights reserved. | ||||||
|  |  * | ||||||
|  |  * Portions Copyright (c) 2009 Apple Inc. All rights reserved. | ||||||
|  |  * | ||||||
|  |  * Redistribution and use in source and binary forms, with or without | ||||||
|  |  * modification, are permitted provided that the following conditions | ||||||
|  |  * are met: | ||||||
|  |  * | ||||||
|  |  * 1. Redistributions of source code must retain the above copyright | ||||||
|  |  *    notice, this list of conditions and the following disclaimer. | ||||||
|  |  * | ||||||
|  |  * 2. Redistributions in binary form must reproduce the above copyright | ||||||
|  |  *    notice, this list of conditions and the following disclaimer in the | ||||||
|  |  *    documentation and/or other materials provided with the distribution. | ||||||
|  |  * | ||||||
|  |  * 3. Neither the name of the Institute nor the names of its contributors | ||||||
|  |  *    may be used to endorse or promote products derived from this software | ||||||
|  |  *    without specific prior written permission. | ||||||
|  |  * | ||||||
|  |  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND | ||||||
|  |  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | ||||||
|  |  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | ||||||
|  |  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE | ||||||
|  |  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | ||||||
|  |  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | ||||||
|  |  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | ||||||
|  |  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | ||||||
|  |  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | ||||||
|  |  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | ||||||
|  |  * SUCH DAMAGE. | ||||||
|  |  */ | ||||||
|  |  | ||||||
|  | #define KRB5_KDB_DISALLOW_POSTDATED	0x00000001 | ||||||
|  | #define KRB5_KDB_DISALLOW_FORWARDABLE	0x00000002 | ||||||
|  | #define KRB5_KDB_DISALLOW_TGT_BASED	0x00000004 | ||||||
|  | #define KRB5_KDB_DISALLOW_RENEWABLE	0x00000008 | ||||||
|  | #define KRB5_KDB_DISALLOW_PROXIABLE	0x00000010 | ||||||
|  | #define KRB5_KDB_DISALLOW_DUP_SKEY	0x00000020 | ||||||
|  | #define KRB5_KDB_DISALLOW_ALL_TIX	0x00000040 | ||||||
|  | #define KRB5_KDB_REQUIRES_PRE_AUTH	0x00000080 | ||||||
|  | #define KRB5_KDB_REQUIRES_HW_AUTH	0x00000100 | ||||||
|  | #define KRB5_KDB_REQUIRES_PWCHANGE	0x00000200 | ||||||
|  | #define KRB5_KDB_DISALLOW_SVR		0x00001000 | ||||||
|  | #define KRB5_KDB_PWCHANGE_SERVICE	0x00002000 | ||||||
|  | #define KRB5_KDB_SUPPORT_DESMD5		0x00004000 | ||||||
|  | #define KRB5_KDB_NEW_PRINC		0x00008000 | ||||||
|  |  | ||||||
|  | /* | ||||||
|  |  | ||||||
|  | key: krb5_unparse_name  + NUL | ||||||
|  |  | ||||||
|  |  16: baselength | ||||||
|  |  32: attributes | ||||||
|  |  32: max time | ||||||
|  |  32: max renewable time | ||||||
|  |  32: client expire | ||||||
|  |  32: passwd expire | ||||||
|  |  32: last successful passwd | ||||||
|  |  32: last failed attempt | ||||||
|  |  32: num of failed attempts | ||||||
|  |  16: num tl data | ||||||
|  |  16: num data data | ||||||
|  |  16: principal length | ||||||
|  |  length: principal | ||||||
|  |  for num tl data times | ||||||
|  |     16: tl data type | ||||||
|  |     16: tl data length | ||||||
|  |     length: length | ||||||
|  |  for num key data times | ||||||
|  |     16: version (num keyblocks) | ||||||
|  |     16: kvno | ||||||
|  |     for version times: | ||||||
|  |         16: type | ||||||
|  |         16: length | ||||||
|  |         length: keydata | ||||||
|  |  | ||||||
|  |  | ||||||
|  | key_data_contents[0] | ||||||
|  |  | ||||||
|  | 	int16: length | ||||||
|  | 	read-of-data: key-encrypted, key-usage 0, master-key | ||||||
|  |  | ||||||
|  | salt: | ||||||
|  |     version2 = salt in key_data->key_data_contents[1] | ||||||
|  |     else default salt. | ||||||
|  |  | ||||||
|  | */ | ||||||
|  |  | ||||||
|  | #include "hdb_locl.h" | ||||||
|  |  | ||||||
|  | #define KDB_V1_BASE_LENGTH 38 | ||||||
|  |  | ||||||
|  | #if HAVE_DB1 | ||||||
|  |  | ||||||
|  | #if defined(HAVE_DB_185_H) | ||||||
|  | #include <db_185.h> | ||||||
|  | #elif defined(HAVE_DB_H) | ||||||
|  | #include <db.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | #define CHECK(x) do { if (x) goto out; } while(0) | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_principal2key(krb5_context context, | ||||||
|  | 		  krb5_const_principal principal, | ||||||
|  | 		  krb5_data *key) | ||||||
|  | { | ||||||
|  |     krb5_error_code ret; | ||||||
|  |     char *str; | ||||||
|  |  | ||||||
|  |     ret = krb5_unparse_name(context, principal, &str); | ||||||
|  |     if (ret) | ||||||
|  | 	return ret; | ||||||
|  |     key->data = str; | ||||||
|  |     key->length = strlen(str) + 1; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | #define KRB5_KDB_SALTTYPE_NORMAL	0 | ||||||
|  | #define KRB5_KDB_SALTTYPE_V4		1 | ||||||
|  | #define KRB5_KDB_SALTTYPE_NOREALM	2 | ||||||
|  | #define KRB5_KDB_SALTTYPE_ONLYREALM	3 | ||||||
|  | #define KRB5_KDB_SALTTYPE_SPECIAL	4 | ||||||
|  | #define KRB5_KDB_SALTTYPE_AFS3		5 | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | fix_salt(krb5_context context, hdb_entry *ent, int key_num) | ||||||
|  | { | ||||||
|  |     krb5_error_code ret; | ||||||
|  |     Salt *salt = ent->keys.val[key_num].salt; | ||||||
|  |     /* fix salt type */ | ||||||
|  |     switch((int)salt->type) { | ||||||
|  |     case KRB5_KDB_SALTTYPE_NORMAL: | ||||||
|  | 	salt->type = KRB5_PADATA_PW_SALT; | ||||||
|  | 	break; | ||||||
|  |     case KRB5_KDB_SALTTYPE_V4: | ||||||
|  | 	krb5_data_free(&salt->salt); | ||||||
|  | 	salt->type = KRB5_PADATA_PW_SALT; | ||||||
|  | 	break; | ||||||
|  |     case KRB5_KDB_SALTTYPE_NOREALM: | ||||||
|  |     { | ||||||
|  | 	size_t len; | ||||||
|  | 	int i; | ||||||
|  | 	char *p; | ||||||
|  | 	 | ||||||
|  | 	len = 0; | ||||||
|  | 	for (i = 0; i < ent->principal->name.name_string.len; ++i) | ||||||
|  | 	    len += strlen(ent->principal->name.name_string.val[i]); | ||||||
|  | 	ret = krb5_data_alloc (&salt->salt, len); | ||||||
|  | 	if (ret) | ||||||
|  | 	    return ret; | ||||||
|  | 	p = salt->salt.data; | ||||||
|  | 	for (i = 0; i < ent->principal->name.name_string.len; ++i) { | ||||||
|  | 	    memcpy (p, | ||||||
|  | 		    ent->principal->name.name_string.val[i], | ||||||
|  | 		    strlen(ent->principal->name.name_string.val[i])); | ||||||
|  | 	    p += strlen(ent->principal->name.name_string.val[i]); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	salt->type = KRB5_PADATA_PW_SALT; | ||||||
|  | 	break; | ||||||
|  |     } | ||||||
|  |     case KRB5_KDB_SALTTYPE_ONLYREALM: | ||||||
|  | 	krb5_data_free(&salt->salt); | ||||||
|  | 	ret = krb5_data_copy(&salt->salt, | ||||||
|  | 			     ent->principal->realm, | ||||||
|  | 			     strlen(ent->principal->realm)); | ||||||
|  | 	if(ret) | ||||||
|  | 	    return ret; | ||||||
|  | 	salt->type = KRB5_PADATA_PW_SALT; | ||||||
|  | 	break; | ||||||
|  |     case KRB5_KDB_SALTTYPE_SPECIAL: | ||||||
|  | 	salt->type = KRB5_PADATA_PW_SALT; | ||||||
|  | 	break; | ||||||
|  |     case KRB5_KDB_SALTTYPE_AFS3: | ||||||
|  | 	krb5_data_free(&salt->salt); | ||||||
|  | 	ret = krb5_data_copy(&salt->salt, | ||||||
|  | 		       ent->principal->realm, | ||||||
|  | 		       strlen(ent->principal->realm)); | ||||||
|  | 	if(ret) | ||||||
|  | 	    return ret; | ||||||
|  | 	salt->type = KRB5_PADATA_AFS3_SALT; | ||||||
|  | 	break; | ||||||
|  |     default: | ||||||
|  | 	abort(); | ||||||
|  |     } | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_value2entry(krb5_context context, krb5_data *data, hdb_entry *entry) | ||||||
|  | { | ||||||
|  |     krb5_error_code ret; | ||||||
|  |     krb5_storage *sp; | ||||||
|  |     uint32_t u32; | ||||||
|  |     uint16_t u16, num_keys, num_tl; | ||||||
|  |     size_t i, j; | ||||||
|  |     char *p = NULL; | ||||||
|  |     int highest_kvno = 0; | ||||||
|  |  | ||||||
|  |     sp = krb5_storage_from_data(data); | ||||||
|  |     if (sp == NULL) { | ||||||
|  | 	krb5_set_error_message(context, ENOMEM, "out of memory"); | ||||||
|  | 	return ENOMEM; | ||||||
|  |     } | ||||||
|  |      | ||||||
|  |     krb5_storage_set_byteorder(sp, KRB5_STORAGE_BYTEORDER_LE); | ||||||
|  |  | ||||||
|  |     /* 16: baselength */ | ||||||
|  |     CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  |     if (u16 != KDB_V1_BASE_LENGTH) { ret = EINVAL; goto out; } | ||||||
|  |     /* 32: attributes */ | ||||||
|  |     CHECK(ret = krb5_ret_uint32(sp, &u32)); | ||||||
|  |     entry->flags.postdate =	 !(u16 & KRB5_KDB_DISALLOW_POSTDATED); | ||||||
|  |     entry->flags.forwardable =	 !(u16 & KRB5_KDB_DISALLOW_FORWARDABLE); | ||||||
|  |     entry->flags.initial =	!!(u16 & KRB5_KDB_DISALLOW_TGT_BASED); | ||||||
|  |     entry->flags.renewable =	 !(u16 & KRB5_KDB_DISALLOW_RENEWABLE); | ||||||
|  |     entry->flags.proxiable =	 !(u16 & KRB5_KDB_DISALLOW_PROXIABLE); | ||||||
|  |     /* DUP_SKEY */ | ||||||
|  |     entry->flags.invalid =	!!(u16 & KRB5_KDB_DISALLOW_ALL_TIX); | ||||||
|  |     entry->flags.require_preauth =!!(u16 & KRB5_KDB_REQUIRES_PRE_AUTH); | ||||||
|  |     entry->flags.require_hwauth =!!(u16 & KRB5_KDB_REQUIRES_HW_AUTH); | ||||||
|  |     entry->flags.server =	 !(u16 & KRB5_KDB_DISALLOW_SVR); | ||||||
|  |     entry->flags.change_pw = 	!!(u16 & KRB5_KDB_PWCHANGE_SERVICE); | ||||||
|  |     entry->flags.client =	   1; /* XXX */ | ||||||
|  |  | ||||||
|  |     /* 32: max time */ | ||||||
|  |     CHECK(ret = krb5_ret_uint32(sp, &u32)); | ||||||
|  |     if (u32) { | ||||||
|  | 	entry->max_life = malloc(sizeof(*entry->max_life)); | ||||||
|  | 	*entry->max_life = u32; | ||||||
|  |     } | ||||||
|  |     /* 32: max renewable time */ | ||||||
|  |     CHECK(ret = krb5_ret_uint32(sp, &u32)); | ||||||
|  |     if (u32) { | ||||||
|  | 	entry->max_renew = malloc(sizeof(*entry->max_renew)); | ||||||
|  | 	*entry->max_renew = u32; | ||||||
|  |     } | ||||||
|  |     /* 32: client expire */ | ||||||
|  |     CHECK(ret = krb5_ret_uint32(sp, &u32)); | ||||||
|  |     if (u32) { | ||||||
|  | 	entry->valid_end = malloc(sizeof(*entry->valid_end)); | ||||||
|  | 	*entry->valid_end = u32; | ||||||
|  |     } | ||||||
|  |     /* 32: passwd expire */ | ||||||
|  |     CHECK(ret = krb5_ret_uint32(sp, &u32)); | ||||||
|  |     if (u32) { | ||||||
|  | 	entry->pw_end = malloc(sizeof(*entry->pw_end)); | ||||||
|  | 	*entry->pw_end = u32; | ||||||
|  |     } | ||||||
|  |     /* 32: last successful passwd */ | ||||||
|  |     CHECK(ret = krb5_ret_uint32(sp, &u32)); | ||||||
|  |     /* 32: last failed attempt */ | ||||||
|  |     CHECK(ret = krb5_ret_uint32(sp, &u32)); | ||||||
|  |     /* 32: num of failed attempts */ | ||||||
|  |     CHECK(ret = krb5_ret_uint32(sp, &u32)); | ||||||
|  |     /* 16: num tl data */ | ||||||
|  |     CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  |     num_tl = u16; | ||||||
|  |     /* 16: num key data */ | ||||||
|  |     CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  |     num_keys = u16; | ||||||
|  |     /* 16: principal length */ | ||||||
|  |     CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  |     /* length: principal */ | ||||||
|  |     { | ||||||
|  | 	p = malloc(u16 + 1); | ||||||
|  | 	krb5_storage_read(sp, p, u16); | ||||||
|  | 	p[u16] = '\0'; | ||||||
|  | 	CHECK(ret = krb5_parse_name(context, p, &entry->principal)); | ||||||
|  | 	free(p); p = NULL; | ||||||
|  |     } | ||||||
|  |     /* for num tl data times | ||||||
|  |            16: tl data type | ||||||
|  |            16: tl data length | ||||||
|  |            length: length */ | ||||||
|  |     for (i = 0; i < num_tl; i++) { | ||||||
|  | 	CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  | 	CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  | 	krb5_storage_seek(sp, u16, SEEK_CUR); | ||||||
|  |     } | ||||||
|  |     /* for num key data times | ||||||
|  |        16: version (num keyblocks) | ||||||
|  |        16: kvno | ||||||
|  |        for version times: | ||||||
|  |            16: type | ||||||
|  |            16: length | ||||||
|  |            length: keydata */ | ||||||
|  |     for (i = 0; i < num_keys; i++) { | ||||||
|  | 	int keep = 0; | ||||||
|  | 	uint16_t version; | ||||||
|  | 	void *ptr; | ||||||
|  |  | ||||||
|  | 	CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  | 	version = u16; | ||||||
|  | 	CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  |  | ||||||
|  | 	if (entry->kvno < u16) { | ||||||
|  | 	    keep = 1; | ||||||
|  | 	    entry->kvno = u16; | ||||||
|  | 	    for (j = 0; j < entry->keys.len; j++) { | ||||||
|  | 		free_Key(&entry->keys.val[j]); | ||||||
|  | 		free(entry->keys.val); | ||||||
|  | 		entry->keys.len = 0; | ||||||
|  | 		entry->keys.val = NULL; | ||||||
|  | 	    } | ||||||
|  | 	} else if (entry->kvno == u16) | ||||||
|  | 	    keep = 1; | ||||||
|  |  | ||||||
|  | 	if (keep) { | ||||||
|  | 	    Key *k; | ||||||
|  |  | ||||||
|  | 	    ptr = realloc(entry->keys.val, sizeof(entry->keys.val[0]) * (entry->keys.len + 1)); | ||||||
|  | 	    if (ptr == NULL) { | ||||||
|  | 		ret = ENOMEM; | ||||||
|  | 		goto out; | ||||||
|  | 	    } | ||||||
|  | 	    entry->keys.val = ptr; | ||||||
|  |  | ||||||
|  | 	    /* k points to current Key */ | ||||||
|  | 	    k = &entry->keys.val[entry->keys.len]; | ||||||
|  |  | ||||||
|  | 	    memset(k, 0, sizeof(*k)); | ||||||
|  | 	    entry->keys.len += 1; | ||||||
|  |  | ||||||
|  | 	    entry->keys.val[i].mkvno = malloc(sizeof(*entry->keys.val[i].mkvno)); | ||||||
|  | 	    if (entry->keys.val[i].mkvno == NULL) { | ||||||
|  | 		ret = ENOMEM; | ||||||
|  | 		goto out; | ||||||
|  | 	    } | ||||||
|  | 	    *entry->keys.val[i].mkvno = 1; | ||||||
|  |  | ||||||
|  | 	    for (j = 0; j < version; j++) { | ||||||
|  | 		uint16_t type; | ||||||
|  | 		CHECK(ret = krb5_ret_uint16(sp, &type)); | ||||||
|  | 		CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  | 		if (j == 0) { /* key */ | ||||||
|  | 		    k->key.keytype = type; | ||||||
|  | 		    if (u16 < 2) { | ||||||
|  | 			ret = EINVAL; | ||||||
|  | 			goto out; | ||||||
|  | 		    } | ||||||
|  | 		    krb5_storage_seek(sp, 2, SEEK_CUR); /* skip real length */ | ||||||
|  | 		    k->key.keyvalue.length = u16 - 2; | ||||||
|  | 		    k->key.keyvalue.data = malloc(k->key.keyvalue.length); | ||||||
|  | 		    krb5_storage_read(sp, k->key.keyvalue.data, k->key.keyvalue.length); | ||||||
|  | 		} else if (j == 1) { /* salt */ | ||||||
|  | 		    k->salt = calloc(1, sizeof(*k->salt)); | ||||||
|  | 		    if (k->salt == NULL) { | ||||||
|  | 			ret = ENOMEM; | ||||||
|  | 			goto out; | ||||||
|  | 		    } | ||||||
|  | 		    k->salt->type = type; | ||||||
|  | 		    if (u16 != 0) { | ||||||
|  | 			k->salt->salt.data = malloc(u16); | ||||||
|  | 			if (k->salt->salt.data == NULL) { | ||||||
|  | 			    ret = ENOMEM; | ||||||
|  | 			    goto out; | ||||||
|  | 			} | ||||||
|  | 			k->salt->salt.length = u16; | ||||||
|  | 			krb5_storage_read(sp, k->salt->salt.data, k->salt->salt.length); | ||||||
|  | 		    }			 | ||||||
|  | 		    fix_salt(context, entry, entry->keys.len - 1); | ||||||
|  | 		} else { | ||||||
|  | 		    krb5_storage_seek(sp, u16, SEEK_CUR); | ||||||
|  | 		} | ||||||
|  | 	    } | ||||||
|  | 	} else { | ||||||
|  | 	    /* skip */ | ||||||
|  | 	    for (j = 0; j < version; j++) { | ||||||
|  | 		CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  | 		CHECK(ret = krb5_ret_uint16(sp, &u16)); | ||||||
|  | 		krb5_storage_seek(sp, u16, u16); | ||||||
|  | 	    } | ||||||
|  | 	} | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  |  out: | ||||||
|  |     if (p) | ||||||
|  | 	free(p); | ||||||
|  |     free_hdb_entry(entry); | ||||||
|  |     return ret; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | #if 0 | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_entry2value(krb5_context context, hdb_entry *entry, krb5_data *data) | ||||||
|  | { | ||||||
|  |     return EINVAL; | ||||||
|  | } | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_close(krb5_context context, HDB *db) | ||||||
|  | { | ||||||
|  |     DB *d = (DB*)db->hdb_db; | ||||||
|  |     (*d->close)(d); | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_destroy(krb5_context context, HDB *db) | ||||||
|  | { | ||||||
|  |     krb5_error_code ret; | ||||||
|  |  | ||||||
|  |     ret = hdb_clear_master_key (context, db); | ||||||
|  |     free(db->hdb_name); | ||||||
|  |     free(db); | ||||||
|  |     return ret; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_lock(krb5_context context, HDB *db, int operation) | ||||||
|  | { | ||||||
|  |     DB *d = (DB*)db->hdb_db; | ||||||
|  |     int fd = (*d->fd)(d); | ||||||
|  |     if(fd < 0) { | ||||||
|  | 	krb5_set_error_message(context, HDB_ERR_CANT_LOCK_DB, | ||||||
|  | 			       "Can't lock database: %s", db->hdb_name); | ||||||
|  | 	return HDB_ERR_CANT_LOCK_DB; | ||||||
|  |     } | ||||||
|  |     return hdb_lock(fd, operation); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_unlock(krb5_context context, HDB *db) | ||||||
|  | { | ||||||
|  |     DB *d = (DB*)db->hdb_db; | ||||||
|  |     int fd = (*d->fd)(d); | ||||||
|  |     if(fd < 0) { | ||||||
|  | 	krb5_set_error_message(context, HDB_ERR_CANT_LOCK_DB, | ||||||
|  | 			       "Can't unlock database: %s", db->hdb_name); | ||||||
|  | 	return HDB_ERR_CANT_LOCK_DB; | ||||||
|  |     } | ||||||
|  |     return hdb_unlock(fd); | ||||||
|  | } | ||||||
|  |  | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_seq(krb5_context context, HDB *db, | ||||||
|  |        unsigned flags, hdb_entry_ex *entry, int flag) | ||||||
|  | { | ||||||
|  |     DB *d = (DB*)db->hdb_db; | ||||||
|  |     DBT key, value; | ||||||
|  |     krb5_data key_data, data; | ||||||
|  |     int code; | ||||||
|  |  | ||||||
|  |     code = db->hdb_lock(context, db, HDB_RLOCK); | ||||||
|  |     if(code == -1) { | ||||||
|  | 	krb5_set_error_message(context, HDB_ERR_DB_INUSE, "Database %s in use", db->hdb_name); | ||||||
|  | 	return HDB_ERR_DB_INUSE; | ||||||
|  |     } | ||||||
|  |     code = (*d->seq)(d, &key, &value, flag); | ||||||
|  |     db->hdb_unlock(context, db); /* XXX check value */ | ||||||
|  |     if(code == -1) { | ||||||
|  | 	code = errno; | ||||||
|  | 	krb5_set_error_message(context, code, "Database %s seq error: %s", | ||||||
|  | 			       db->hdb_name, strerror(code)); | ||||||
|  | 	return code; | ||||||
|  |     } | ||||||
|  |     if(code == 1) { | ||||||
|  | 	krb5_clear_error_message(context); | ||||||
|  | 	return HDB_ERR_NOENTRY; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     key_data.data = key.data; | ||||||
|  |     key_data.length = key.size; | ||||||
|  |     data.data = value.data; | ||||||
|  |     data.length = value.size; | ||||||
|  |     memset(entry, 0, sizeof(*entry)); | ||||||
|  |  | ||||||
|  |     if (mdb_value2entry(context, &data, &entry->entry)) | ||||||
|  | 	return mdb_seq(context, db, flags, entry, R_NEXT); | ||||||
|  |  | ||||||
|  |     if (db->hdb_master_key_set && (flags & HDB_F_DECRYPT)) { | ||||||
|  | 	code = hdb_unseal_keys (context, db, &entry->entry); | ||||||
|  | 	if (code) | ||||||
|  | 	    hdb_free_entry (context, entry); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return code; | ||||||
|  | } | ||||||
|  |  | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_firstkey(krb5_context context, HDB *db, unsigned flags, hdb_entry_ex *entry) | ||||||
|  | { | ||||||
|  |     return mdb_seq(context, db, flags, entry, R_FIRST); | ||||||
|  | } | ||||||
|  |  | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_nextkey(krb5_context context, HDB *db, unsigned flags, hdb_entry_ex *entry) | ||||||
|  | { | ||||||
|  |     return mdb_seq(context, db, flags, entry, R_NEXT); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_rename(krb5_context context, HDB *db, const char *new_name) | ||||||
|  | { | ||||||
|  |     int ret; | ||||||
|  |     char *old, *new; | ||||||
|  |  | ||||||
|  |     asprintf(&old, "%s.db", db->hdb_name); | ||||||
|  |     asprintf(&new, "%s.db", new_name); | ||||||
|  |     ret = rename(old, new); | ||||||
|  |     free(old); | ||||||
|  |     free(new); | ||||||
|  |     if(ret) | ||||||
|  | 	return errno; | ||||||
|  |  | ||||||
|  |     free(db->hdb_name); | ||||||
|  |     db->hdb_name = strdup(new_name); | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb__get(krb5_context context, HDB *db, krb5_data key, krb5_data *reply) | ||||||
|  | { | ||||||
|  |     DB *d = (DB*)db->hdb_db; | ||||||
|  |     DBT k, v; | ||||||
|  |     int code; | ||||||
|  |  | ||||||
|  |     k.data = key.data; | ||||||
|  |     k.size = key.length; | ||||||
|  |     code = db->hdb_lock(context, db, HDB_RLOCK); | ||||||
|  |     if(code) | ||||||
|  | 	return code; | ||||||
|  |     code = (*d->get)(d, &k, &v, 0); | ||||||
|  |     db->hdb_unlock(context, db); | ||||||
|  |     if(code < 0) { | ||||||
|  | 	code = errno; | ||||||
|  | 	krb5_set_error_message(context, code, "Database %s get error: %s", | ||||||
|  | 			       db->hdb_name, strerror(code)); | ||||||
|  | 	return code; | ||||||
|  |     } | ||||||
|  |     if(code == 1) { | ||||||
|  | 	krb5_clear_error_message(context); | ||||||
|  | 	return HDB_ERR_NOENTRY; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     krb5_data_copy(reply, v.data, v.size); | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb__put(krb5_context context, HDB *db, int replace, | ||||||
|  | 	krb5_data key, krb5_data value) | ||||||
|  | { | ||||||
|  |     DB *d = (DB*)db->hdb_db; | ||||||
|  |     DBT k, v; | ||||||
|  |     int code; | ||||||
|  |  | ||||||
|  |     k.data = key.data; | ||||||
|  |     k.size = key.length; | ||||||
|  |     v.data = value.data; | ||||||
|  |     v.size = value.length; | ||||||
|  |     code = db->hdb_lock(context, db, HDB_WLOCK); | ||||||
|  |     if(code) | ||||||
|  | 	return code; | ||||||
|  |     code = (*d->put)(d, &k, &v, replace ? 0 : R_NOOVERWRITE); | ||||||
|  |     db->hdb_unlock(context, db); | ||||||
|  |     if(code < 0) { | ||||||
|  | 	code = errno; | ||||||
|  | 	krb5_set_error_message(context, code, "Database %s put error: %s", | ||||||
|  | 			       db->hdb_name, strerror(code)); | ||||||
|  | 	return code; | ||||||
|  |     } | ||||||
|  |     if(code == 1) { | ||||||
|  | 	krb5_clear_error_message(context); | ||||||
|  | 	return HDB_ERR_EXISTS; | ||||||
|  |     } | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb__del(krb5_context context, HDB *db, krb5_data key) | ||||||
|  | { | ||||||
|  |     DB *d = (DB*)db->hdb_db; | ||||||
|  |     DBT k; | ||||||
|  |     krb5_error_code code; | ||||||
|  |     k.data = key.data; | ||||||
|  |     k.size = key.length; | ||||||
|  |     code = db->hdb_lock(context, db, HDB_WLOCK); | ||||||
|  |     if(code) | ||||||
|  | 	return code; | ||||||
|  |     code = (*d->del)(d, &k, 0); | ||||||
|  |     db->hdb_unlock(context, db); | ||||||
|  |     if(code == 1) { | ||||||
|  | 	code = errno; | ||||||
|  | 	krb5_set_error_message(context, code, "Database %s put error: %s", | ||||||
|  | 			       db->hdb_name, strerror(code)); | ||||||
|  | 	return code; | ||||||
|  |     } | ||||||
|  |     if(code < 0) | ||||||
|  | 	return errno; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_fetch(krb5_context context, HDB *db, krb5_const_principal principal, | ||||||
|  | 	  unsigned flags, hdb_entry_ex *entry) | ||||||
|  | { | ||||||
|  |     krb5_data key, value; | ||||||
|  |     krb5_error_code code; | ||||||
|  |     char *str; | ||||||
|  |  | ||||||
|  |     code = mdb_principal2key(context, principal, &key); | ||||||
|  |     if (code) | ||||||
|  | 	return code; | ||||||
|  |     code = db->hdb__get(context, db, key, &value); | ||||||
|  |     krb5_data_free(&key); | ||||||
|  |     if(code) | ||||||
|  | 	return code; | ||||||
|  |     code = mdb_value2entry(context, &value, &entry->entry); | ||||||
|  |     krb5_data_free(&value); | ||||||
|  |     if (code) | ||||||
|  | 	return code; | ||||||
|  |  | ||||||
|  |     if (db->hdb_master_key_set && (flags & HDB_F_DECRYPT)) { | ||||||
|  | 	code = hdb_unseal_keys (context, db, &entry->entry); | ||||||
|  | 	if (code) | ||||||
|  | 	    hdb_free_entry(context, entry); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_store(krb5_context context, HDB *db, unsigned flags, hdb_entry_ex *entry) | ||||||
|  | { | ||||||
|  |     krb5_set_error_message(context, EINVAL, "can't set principal in mdb"); | ||||||
|  |     return EINVAL; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_remove(krb5_context context, HDB *db, krb5_const_principal principal) | ||||||
|  | { | ||||||
|  |     krb5_data key, value; | ||||||
|  |     krb5_error_code code; | ||||||
|  |  | ||||||
|  |     mdb_principal2key(context, principal, &key); | ||||||
|  |     code = db->hdb__del(context, db, key); | ||||||
|  |     krb5_data_free(&key); | ||||||
|  |     return code; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static krb5_error_code | ||||||
|  | mdb_open(krb5_context context, HDB *db, int flags, mode_t mode) | ||||||
|  | { | ||||||
|  |     char *fn; | ||||||
|  |     krb5_error_code ret; | ||||||
|  |  | ||||||
|  |     asprintf(&fn, "%s.db", db->hdb_name); | ||||||
|  |     if (fn == NULL) { | ||||||
|  | 	krb5_set_error_message(context, ENOMEM, "malloc: out of memory"); | ||||||
|  | 	return ENOMEM; | ||||||
|  |     } | ||||||
|  |     db->hdb_db = dbopen(fn, flags, mode, DB_BTREE, NULL); | ||||||
|  |     free(fn); | ||||||
|  |  | ||||||
|  |     if (db->hdb_db == NULL) { | ||||||
|  | 	switch (errno) { | ||||||
|  | #ifdef EFTYPE | ||||||
|  | 	case EFTYPE: | ||||||
|  | #endif | ||||||
|  | 	case EINVAL: | ||||||
|  | 	    db->hdb_db = dbopen(fn, flags, mode, DB_BTREE, NULL); | ||||||
|  | 	} | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     /* try to open without .db extension */ | ||||||
|  |     if(db->hdb_db == NULL && errno == ENOENT) | ||||||
|  | 	db->hdb_db = dbopen(db->hdb_name, flags, mode, DB_BTREE, NULL); | ||||||
|  |     if(db->hdb_db == NULL) { | ||||||
|  | 	ret = errno; | ||||||
|  | 	krb5_set_error_message(context, ret, "dbopen (%s): %s", | ||||||
|  | 			      db->hdb_name, strerror(ret)); | ||||||
|  | 	return ret; | ||||||
|  |     } | ||||||
|  |     if((flags & O_ACCMODE) == O_RDONLY) | ||||||
|  | 	ret = hdb_check_db_format(context, db); | ||||||
|  |     else | ||||||
|  | 	ret = hdb_init_db(context, db); | ||||||
|  |     if(ret == HDB_ERR_NOENTRY) { | ||||||
|  | 	krb5_clear_error_message(context); | ||||||
|  | 	return 0; | ||||||
|  |     } | ||||||
|  |     if (ret) { | ||||||
|  | 	mdb_close(context, db); | ||||||
|  | 	krb5_set_error_message(context, ret, "hdb_open: failed %s database %s", | ||||||
|  | 			      (flags & O_ACCMODE) == O_RDONLY ? | ||||||
|  | 			      "checking format of" : "initialize", | ||||||
|  | 			      db->hdb_name); | ||||||
|  |     } | ||||||
|  |     return ret; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | krb5_error_code | ||||||
|  | hdb_mdb_create(krb5_context context, HDB **db, | ||||||
|  | 	       const char *filename) | ||||||
|  | { | ||||||
|  |     *db = calloc(1, sizeof(**db)); | ||||||
|  |     if (*db == NULL) { | ||||||
|  | 	krb5_set_error_message(context, ENOMEM, "malloc: out of memory"); | ||||||
|  | 	return ENOMEM; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     (*db)->hdb_db = NULL; | ||||||
|  |     (*db)->hdb_name = strdup(filename); | ||||||
|  |     if ((*db)->hdb_name == NULL) { | ||||||
|  | 	free(*db); | ||||||
|  | 	*db = NULL; | ||||||
|  | 	krb5_set_error_message(context, ENOMEM, "malloc: out of memory"); | ||||||
|  | 	return ENOMEM; | ||||||
|  |     } | ||||||
|  |     (*db)->hdb_master_key_set = 0; | ||||||
|  |     (*db)->hdb_openp = 0; | ||||||
|  |     (*db)->hdb_capability_flags = 0; | ||||||
|  |     (*db)->hdb_open = mdb_open; | ||||||
|  |     (*db)->hdb_close = mdb_close; | ||||||
|  |     (*db)->hdb_fetch = mdb_fetch; | ||||||
|  |     (*db)->hdb_store = mdb_store; | ||||||
|  |     (*db)->hdb_remove = mdb_remove; | ||||||
|  |     (*db)->hdb_firstkey = mdb_firstkey; | ||||||
|  |     (*db)->hdb_nextkey= mdb_nextkey; | ||||||
|  |     (*db)->hdb_lock = mdb_lock; | ||||||
|  |     (*db)->hdb_unlock = mdb_unlock; | ||||||
|  |     (*db)->hdb_rename = mdb_rename; | ||||||
|  |     (*db)->hdb__get = mdb__get; | ||||||
|  |     (*db)->hdb__put = mdb__put; | ||||||
|  |     (*db)->hdb__del = mdb__del; | ||||||
|  |     (*db)->hdb_destroy = mdb_destroy; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | #endif /* HAVE_DB1 */ | ||||||
		Reference in New Issue
	
	Block a user
	 Love Hornquist Astrand
					Love Hornquist Astrand