2 * Copyright (c) 1997 - 2001 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
17 * 3. Neither the name of the Institute nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36 RCSID("$Id: db.c,v 1.33 2005/11/28 23:30:51 lha Exp $");
40 #if defined(HAVE_DB_185_H)
42 #elif defined(HAVE_DB_H)
46 static krb5_error_code
47 DB_close(krb5_context context, HDB *db)
49 DB *d = (DB*)db->hdb_db;
54 static krb5_error_code
55 DB_destroy(krb5_context context, HDB *db)
59 ret = hdb_clear_master_key (context, db);
65 static krb5_error_code
66 DB_lock(krb5_context context, HDB *db, int operation)
68 DB *d = (DB*)db->hdb_db;
71 return HDB_ERR_CANT_LOCK_DB;
72 return hdb_lock(fd, operation);
75 static krb5_error_code
76 DB_unlock(krb5_context context, HDB *db)
78 DB *d = (DB*)db->hdb_db;
81 return HDB_ERR_CANT_LOCK_DB;
82 return hdb_unlock(fd);
86 static krb5_error_code
87 DB_seq(krb5_context context, HDB *db,
88 unsigned flags, hdb_entry *entry, int flag)
90 DB *d = (DB*)db->hdb_db;
92 krb5_data key_data, data;
95 code = db->hdb_lock(context, db, HDB_RLOCK);
97 return HDB_ERR_DB_INUSE;
98 code = d->seq(d, &key, &value, flag);
99 db->hdb_unlock(context, db); /* XXX check value */
103 return HDB_ERR_NOENTRY;
105 key_data.data = key.data;
106 key_data.length = key.size;
107 data.data = value.data;
108 data.length = value.size;
109 if (hdb_value2entry(context, &data, entry))
110 return DB_seq(context, db, flags, entry, R_NEXT);
111 if (db->hdb_master_key_set && (flags & HDB_F_DECRYPT)) {
112 code = hdb_unseal_keys (context, db, entry);
114 hdb_free_entry (context, entry);
116 if (code == 0 && entry->principal == NULL) {
117 entry->principal = malloc(sizeof(*entry->principal));
118 if (entry->principal == NULL) {
119 krb5_set_error_string(context, "malloc: out of memory");
121 hdb_free_entry (context, entry);
123 hdb_key2principal(context, &key_data, entry->principal);
130 static krb5_error_code
131 DB_firstkey(krb5_context context, HDB *db, unsigned flags, hdb_entry *entry)
133 return DB_seq(context, db, flags, entry, R_FIRST);
137 static krb5_error_code
138 DB_nextkey(krb5_context context, HDB *db, unsigned flags, hdb_entry *entry)
140 return DB_seq(context, db, flags, entry, R_NEXT);
143 static krb5_error_code
144 DB_rename(krb5_context context, HDB *db, const char *new_name)
149 asprintf(&old, "%s.db", db->hdb_name);
150 asprintf(&new, "%s.db", new_name);
151 ret = rename(old, new);
158 db->hdb_name = strdup(new_name);
162 static krb5_error_code
163 DB__get(krb5_context context, HDB *db, krb5_data key, krb5_data *reply)
165 DB *d = (DB*)db->hdb_db;
171 code = db->hdb_lock(context, db, HDB_RLOCK);
174 code = d->get(d, &k, &v, 0);
175 db->hdb_unlock(context, db);
179 return HDB_ERR_NOENTRY;
181 krb5_data_copy(reply, v.data, v.size);
185 static krb5_error_code
186 DB__put(krb5_context context, HDB *db, int replace,
187 krb5_data key, krb5_data value)
189 DB *d = (DB*)db->hdb_db;
196 v.size = value.length;
197 code = db->hdb_lock(context, db, HDB_WLOCK);
200 code = d->put(d, &k, &v, replace ? 0 : R_NOOVERWRITE);
201 db->hdb_unlock(context, db);
205 return HDB_ERR_EXISTS;
209 static krb5_error_code
210 DB__del(krb5_context context, HDB *db, krb5_data key)
212 DB *d = (DB*)db->hdb_db;
214 krb5_error_code code;
217 code = db->hdb_lock(context, db, HDB_WLOCK);
220 code = d->del(d, &k, 0);
221 db->hdb_unlock(context, db);
223 return HDB_ERR_NOENTRY;
229 static krb5_error_code
230 DB_open(krb5_context context, HDB *db, int flags, mode_t mode)
235 asprintf(&fn, "%s.db", db->hdb_name);
237 krb5_set_error_string(context, "malloc: out of memory");
240 db->hdb_db = dbopen(fn, flags, mode, DB_BTREE, NULL);
242 /* try to open without .db extension */
243 if(db->hdb_db == NULL && errno == ENOENT)
244 db->hdb_db = dbopen(db->hdb_name, flags, mode, DB_BTREE, NULL);
245 if(db->hdb_db == NULL) {
247 krb5_set_error_string(context, "dbopen (%s): %s",
248 db->hdb_name, strerror(ret));
251 if((flags & O_ACCMODE) == O_RDONLY)
252 ret = hdb_check_db_format(context, db);
254 ret = hdb_init_db(context, db);
255 if(ret == HDB_ERR_NOENTRY) {
256 krb5_clear_error_string(context);
260 DB_close(context, db);
261 krb5_set_error_string(context, "hdb_open: failed %s database %s",
262 (flags & O_ACCMODE) == O_RDONLY ?
263 "checking format of" : "initialize",
270 hdb_db_create(krb5_context context, HDB **db,
271 const char *filename)
273 *db = calloc(1, sizeof(**db));
275 krb5_set_error_string(context, "malloc: out of memory");
279 (*db)->hdb_db = NULL;
280 (*db)->hdb_name = strdup(filename);
281 if ((*db)->hdb_name == NULL) {
282 krb5_set_error_string(context, "malloc: out of memory");
287 (*db)->hdb_master_key_set = 0;
288 (*db)->hdb_openp = 0;
289 (*db)->hdb_open = DB_open;
290 (*db)->hdb_close = DB_close;
291 (*db)->hdb_fetch = _hdb_fetch;
292 (*db)->hdb_store = _hdb_store;
293 (*db)->hdb_remove = _hdb_remove;
294 (*db)->hdb_firstkey = DB_firstkey;
295 (*db)->hdb_nextkey= DB_nextkey;
296 (*db)->hdb_lock = DB_lock;
297 (*db)->hdb_unlock = DB_unlock;
298 (*db)->hdb_rename = DB_rename;
299 (*db)->hdb__get = DB__get;
300 (*db)->hdb__put = DB__put;
301 (*db)->hdb__del = DB__del;
302 (*db)->hdb_destroy = DB_destroy;
306 #endif /* HAVE_DB1 */