From 5c5e6599d974e0b97cd55b308081ecf801fab6e5 Mon Sep 17 00:00:00 2001
From: Johan Danielsson <joda@pdc.kth.se>
Date: Tue, 26 Aug 1997 22:29:08 +0000
Subject: [PATCH] Add _put.

git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@3171 ec53bebd-3082-4978-b11e-865c3cabbd6b
---
 lib/hdb/db.c | 50 ++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 38 insertions(+), 12 deletions(-)

diff --git a/lib/hdb/db.c b/lib/hdb/db.c
index 2e7da03d8..cf16c7892 100644
--- a/lib/hdb/db.c
+++ b/lib/hdb/db.c
@@ -189,6 +189,25 @@ DB_nextkey(krb5_context context, HDB *db, hdb_entry *entry)
     return DB_seq(context, db, entry, R_NEXT);
 }
 
+static krb5_error_code
+DB_rename(krb5_context context, HDB *db, const char *new_name)
+{
+    int ret;
+    char *old, *new;
+
+    asprintf(&old, "%s.db", db->name);
+    asprintf(&new, "%s.db", new_name);
+    ret = rename(old, new);
+    free(old);
+    free(new);
+    if(ret)
+	return errno;
+    
+    free(db->name);
+    db->name = strdup(new_name);
+    return 0;
+}
+
 static krb5_error_code
 DB__get(krb5_context context, HDB *db, krb5_data key, krb5_data *reply)
 {
@@ -215,21 +234,26 @@ DB__get(krb5_context context, HDB *db, krb5_data key, krb5_data *reply)
 }
 
 static krb5_error_code
-DB_rename(krb5_context context, HDB *db, const char *new_name)
+DB__put(krb5_context context, HDB *db, int replace, 
+	krb5_data key, krb5_data value)
 {
-    int ret;
-    char *old, *new;
+    DB *d = (DB*)db->db;
+    DBT k, v;
+    int code;
 
-    asprintf(&old, "%s.db", db->name);
-    asprintf(&new, "%s.db", new_name);
-    ret = rename(old, new);
-    free(old);
-    free(new);
-    if(ret)
+    k.data = key.data;
+    k.size = key.length;
+    v.data = value.data;
+    v.size = value.length;
+    code = db->lock(context, db, HDB_WLOCK);
+    if(code)
+	return code;
+    code = d->put(d, &k, &v, replace ? 0 : R_NOOVERWRITE);
+    db->unlock(context, db);
+    if(code < 0)
 	return errno;
-    
-    free(db->name);
-    db->name = strdup(new_name);
+    if(code == 1)
+	return HDB_ERR_EXISTS;
     return 0;
 }
 
@@ -256,6 +280,8 @@ hdb_db_open(krb5_context context, HDB **db,
     (*db)->lock = DB_lock;
     (*db)->unlock = DB_unlock;
     (*db)->rename = DB_rename;
+    (*db)->_get = DB__get;
+    (*db)->_put = DB__put;
     return 0;
 }