]> git.sommitrealweird.co.uk Git - onak.git/blobdiff - keydb_db4.c
Explicitly use 32 bits for keyd socket commands
[onak.git] / keydb_db4.c
index 986c49af2925f190b2d906555edaf1fc77c3aefc..6fbf9da5392798159cac6e13682ea087f154ed16 100644 (file)
@@ -1,12 +1,11 @@
 /*
  * keydb_db4.c - Routines to store and fetch keys in a DB4 database.
  *
 /*
  * keydb_db4.c - Routines to store and fetch keys in a DB4 database.
  *
- * Jonathan McDowell <noodles@earth.li>
- *
- * Copyright 2002-2004 Project Purple
+ * Copyright 2002-2008 Jonathan McDowell <noodles@earth.li>
  */
 
 #include <sys/types.h>
  */
 
 #include <sys/types.h>
+#include <sys/stat.h>
 #include <sys/uio.h>
 #include <ctype.h>
 #include <errno.h>
 #include <sys/uio.h>
 #include <ctype.h>
 #include <errno.h>
@@ -30,6 +29,8 @@
 #include "parsekey.h"
 #include "wordlist.h"
 
 #include "parsekey.h"
 #include "wordlist.h"
 
+#define DB4_UPGRADE_FILE "db_upgrade.lck"
+
 /**
  *     dbenv - our database environment.
  */
 /**
  *     dbenv - our database environment.
  */
@@ -69,6 +70,28 @@ DB *keydb(uint64_t keyid)
        return(dbconns[keytrun % numdbs]);
 }
 
        return(dbconns[keytrun % numdbs]);
 }
 
+/**
+ *     db4_errfunc - Direct DB errors to logfile
+ *
+ *     Basic function to take errors from the DB library and output them to
+ *     the logfile rather than stderr.
+ */
+#if (DB_VERSION_MAJOR == 4) && (DB_VERSION_MINOR < 3)
+static void db4_errfunc(const char *errpfx, const char *errmsg)
+#else
+static void db4_errfunc(const DB_ENV *edbenv, const char *errpfx,
+               const char *errmsg)
+#endif
+{
+       if (errpfx) {
+               logthing(LOGTHING_DEBUG, "db4 error: %s:%s", errpfx, errmsg);
+       } else {
+               logthing(LOGTHING_DEBUG, "db4 error: %s", errmsg);
+       }
+
+       return;
+}
+
 /**
  *     starttrans - Start a transaction.
  *
 /**
  *     starttrans - Start a transaction.
  *
@@ -155,6 +178,90 @@ static void db4_cleanupdb(void)
        }
 }
 
        }
 }
 
+/**
+ *     db4_upgradedb - Upgrade a DB4 database
+ *
+ *     Called if we discover we need to upgrade our DB4 database; ie if
+ *     we're running with a newer version of db4 than the database was
+ *     created with.
+ */
+static int db4_upgradedb(int numdb)
+{
+       DB *curdb = NULL;
+       int ret;
+       int i;
+       char buf[1024];
+       int lockfile_fd;
+       struct stat statbuf;
+
+       snprintf(buf, sizeof(buf) - 1, "%s/%s", config.db_dir,
+                       DB4_UPGRADE_FILE);
+       lockfile_fd = open(buf, O_RDWR | O_CREAT | O_EXCL, 0600);
+       if (lockfile_fd < 0) {
+               if (errno == EEXIST) {
+                       while (stat(buf, &statbuf) == 0) ;
+                       return 0;
+               } else {
+                       logthing(LOGTHING_CRITICAL, "Couldn't open database "
+                               "update lock file: %s", strerror(errno));
+                       return -1;
+               }
+       }
+       snprintf(buf, sizeof(buf) - 1, "%d", getpid());
+       write(lockfile_fd, buf, strlen(buf));
+       close(lockfile_fd);
+
+       logthing(LOGTHING_NOTICE, "Upgrading DB4 database");
+       ret = db_env_create(&dbenv, 0);
+       dbenv->set_errcall(dbenv, &db4_errfunc);
+       dbenv->remove(dbenv, config.db_dir, 0);
+       dbenv = NULL;
+       for (i = 0; i < numdb; i++) {
+               ret = db_create(&curdb, NULL, 0);
+               if (ret == 0) {
+                       snprintf(buf, sizeof(buf) - 1, "%s/keydb.%d.db",
+                               config.db_dir, i);
+                       logthing(LOGTHING_DEBUG, "Upgrading %s", buf);
+                       ret = curdb->upgrade(curdb, buf, 0);
+                       curdb->close(curdb, 0);
+               } else {
+                       logthing(LOGTHING_ERROR, "Error upgrading DB %s : %s",
+                               buf,
+                               db_strerror(ret));
+               }
+       }
+
+       ret = db_create(&curdb, NULL, 0);
+       if (ret == 0) {
+               snprintf(buf, sizeof(buf) - 1, "%s/worddb", config.db_dir);
+               logthing(LOGTHING_DEBUG, "Upgrading %s", buf);
+               ret = curdb->upgrade(curdb, buf, 0);
+               curdb->close(curdb, 0);
+       } else {
+               logthing(LOGTHING_ERROR, "Error upgrading DB %s : %s",
+                       buf,
+                       db_strerror(ret));
+       }
+
+       ret = db_create(&curdb, NULL, 0);
+       if (ret == 0) {
+               snprintf(buf, sizeof(buf) - 1, "%s/id32db", config.db_dir);
+               logthing(LOGTHING_DEBUG, "Upgrading %s", buf);
+               ret = curdb->upgrade(curdb, buf, 0);
+               curdb->close(curdb, 0);
+       } else {
+               logthing(LOGTHING_ERROR, "Error upgrading DB %s : %s",
+                       buf,
+                       db_strerror(ret));
+       }
+
+       snprintf(buf, sizeof(buf) - 1, "%s/%s", config.db_dir,
+                       DB4_UPGRADE_FILE);
+       unlink(buf);
+
+       return ret;
+}
+
 /**
  *     initdb - Initialize the key database.
  *
 /**
  *     initdb - Initialize the key database.
  *
@@ -169,6 +276,23 @@ static void db4_initdb(bool readonly)
        int        ret = 0;
        int        i = 0;
        u_int32_t  flags = 0;
        int        ret = 0;
        int        i = 0;
        u_int32_t  flags = 0;
+       struct stat statbuf;
+       int        maxlocks;
+
+       snprintf(buf, sizeof(buf) - 1, "%s/%s", config.db_dir,
+                       DB4_UPGRADE_FILE);
+       ret = stat(buf, &statbuf);
+       while ((ret == 0) || (errno != ENOENT)) {
+               if (ret != 0) {
+                       logthing(LOGTHING_CRITICAL, "Couldn't stat upgrade "
+                               "lock file: %s (%d)", strerror(errno), ret);
+                       exit(1);
+               }
+               logthing(LOGTHING_DEBUG, "DB4 upgrade in progress; waiting.");
+               sleep(5);
+               ret = stat(buf, &statbuf);
+       }
+       ret = 0;
 
        snprintf(buf, sizeof(buf) - 1, "%s/num_keydb", config.db_dir);
        numdb = fopen(buf, "r");
 
        snprintf(buf, sizeof(buf) - 1, "%s/num_keydb", config.db_dir);
        numdb = fopen(buf, "r");
@@ -191,7 +315,7 @@ static void db4_initdb(bool readonly)
                }
        }
 
                }
        }
 
-       dbconns = malloc(sizeof (DB *) * numdbs);
+       dbconns = calloc(numdbs, sizeof (DB *));
        if (dbconns == NULL) {
                logthing(LOGTHING_CRITICAL,
                                "Couldn't allocate memory for dbconns");
        if (dbconns == NULL) {
                logthing(LOGTHING_CRITICAL,
                                "Couldn't allocate memory for dbconns");
@@ -206,12 +330,24 @@ static void db4_initdb(bool readonly)
                }
        }
 
                }
        }
 
+       /*
+        * Up the number of locks we're allowed at once. We base this on
+        * the maximum number of keys we're going to return.
+        */
+       maxlocks = config.maxkeys * 16;
+       if (maxlocks < 1000) {
+               maxlocks = 1000;
+       }
+       dbenv->set_lk_max_locks(dbenv, maxlocks);
+       dbenv->set_lk_max_objects(dbenv, maxlocks);
+
        /*
         * Enable deadlock detection so that we don't block indefinitely on
         * anything. What we really want is simple 2 state locks, but I'm not
         * sure how to make the standard DB functions do that yet.
         */
        if (ret == 0) {
        /*
         * Enable deadlock detection so that we don't block indefinitely on
         * anything. What we really want is simple 2 state locks, but I'm not
         * sure how to make the standard DB functions do that yet.
         */
        if (ret == 0) {
+               dbenv->set_errcall(dbenv, &db4_errfunc);
                ret = dbenv->set_lk_detect(dbenv, DB_LOCK_DEFAULT);
                if (ret != 0) {
                        logthing(LOGTHING_CRITICAL,
                ret = dbenv->set_lk_detect(dbenv, DB_LOCK_DEFAULT);
                if (ret != 0) {
                        logthing(LOGTHING_CRITICAL,
@@ -225,6 +361,32 @@ static void db4_initdb(bool readonly)
                                DB_INIT_TXN |
                                DB_CREATE,
                                0);
                                DB_INIT_TXN |
                                DB_CREATE,
                                0);
+#ifdef DB_VERSION_MISMATCH
+               if (ret == DB_VERSION_MISMATCH) {
+                       dbenv->close(dbenv, 0);
+                       dbenv = NULL;
+                       ret = db4_upgradedb(numdbs);
+                       if (ret == 0) {
+                               ret = db_env_create(&dbenv, 0);
+                       }
+                       if (ret == 0) {
+                               dbenv->set_errcall(dbenv, &db4_errfunc);
+                               dbenv->set_lk_detect(dbenv, DB_LOCK_DEFAULT);
+                               ret = dbenv->open(dbenv, config.db_dir,
+                                       DB_INIT_LOG | DB_INIT_MPOOL |
+                                       DB_INIT_LOCK | DB_INIT_TXN |
+                                       DB_CREATE | DB_RECOVER,
+                                       0);
+
+                               if (ret == 0) {
+                                       dbenv->txn_checkpoint(dbenv,
+                                                       0,
+                                                       0,
+                                                       DB_FORCE);
+                               }
+                       }
+               }
+#endif
                if (ret != 0) {
                        logthing(LOGTHING_CRITICAL,
                                        "Error opening db environment: %s (%s)",
                if (ret != 0) {
                        logthing(LOGTHING_CRITICAL,
                                        "Error opening db environment: %s (%s)",
@@ -319,7 +481,7 @@ static void db4_initdb(bool readonly)
        }
 
        if (txn != NULL) {
        }
 
        if (txn != NULL) {
-               endtrans();
+               db4_endtrans();
        }
 
        if (ret != 0) {
        }
 
        if (ret != 0) {
@@ -332,6 +494,55 @@ static void db4_initdb(bool readonly)
        return;
 }
 
        return;
 }
 
+/**
+ *     getfullkeyid - Maps a 32bit key id to a 64bit one.
+ *     @keyid: The 32bit keyid.
+ *
+ *     This function maps a 32bit key id to the full 64bit one. It returns the
+ *     full keyid. If the key isn't found a keyid of 0 is returned.
+ */
+static uint64_t db4_getfullkeyid(uint64_t keyid)
+{
+       DBT       key, data;
+       DBC      *cursor = NULL;
+       uint32_t  shortkeyid = 0;
+       int       ret = 0;
+
+       if (keyid < 0x100000000LL) {
+               ret = id32db->cursor(id32db,
+                               txn,
+                               &cursor,
+                               0);   /* flags */
+
+               shortkeyid = keyid & 0xFFFFFFFF;
+
+               memset(&key, 0, sizeof(key));
+               memset(&data, 0, sizeof(data));
+               key.data = &shortkeyid;
+               key.size = sizeof(shortkeyid);
+               data.flags = DB_DBT_MALLOC;
+
+               ret = cursor->c_get(cursor,
+                       &key,
+                       &data,
+                       DB_SET);
+
+               if (ret == 0) {
+                       keyid = *(uint64_t *) data.data;
+
+                       if (data.data != NULL) {
+                               free(data.data);
+                               data.data = NULL;
+                       }
+               }
+
+               ret = cursor->c_close(cursor);
+               cursor = NULL;
+       }
+       
+       return keyid;
+}
+
 /**
  *     fetch_key - Given a keyid fetch the key from storage.
  *     @keyid: The keyid to fetch.
 /**
  *     fetch_key - Given a keyid fetch the key from storage.
  *     @keyid: The keyid to fetch.
@@ -354,7 +565,7 @@ static int db4_fetch_key(uint64_t keyid, struct openpgp_publickey **publickey,
        struct buffer_ctx fetchbuf;
 
        if (keyid < 0x100000000LL) {
        struct buffer_ctx fetchbuf;
 
        if (keyid < 0x100000000LL) {
-               keyid = getfullkeyid(keyid);
+               keyid = db4_getfullkeyid(keyid);
        }
 
        memset(&key, 0, sizeof(key));
        }
 
        memset(&key, 0, sizeof(key));
@@ -393,7 +604,7 @@ static int db4_fetch_key(uint64_t keyid, struct openpgp_publickey **publickey,
        }
 
        if (!intrans) {
        }
 
        if (!intrans) {
-               endtrans();
+               db4_endtrans();
        }
 
        return (numkeys);
        }
 
        return (numkeys);
@@ -426,6 +637,7 @@ static int db4_fetch_key_text(const char *search,
        struct ll *curword = NULL;
        struct keyarray keylist = { NULL, 0, 0 };
        struct keyarray newkeylist = { NULL, 0, 0 };
        struct ll *curword = NULL;
        struct keyarray keylist = { NULL, 0, 0 };
        struct keyarray newkeylist = { NULL, 0, 0 };
+       int firstpass = 1;
 
        numkeys = 0;
        searchtext = strdup(search);
 
        numkeys = 0;
        searchtext = strdup(search);
@@ -458,8 +670,12 @@ static int db4_fetch_key_text(const char *search,
                                                data.data)[i];
                        }
 
                                                data.data)[i];
                        }
 
-                       if (keylist.count == 0 ||
-                                       array_find(&keylist, keyid)) {
+                       /*
+                        * Only add the keys containing this word if this is
+                        * our first pass (ie we have no existing key list),
+                        * or the key contained a previous word.
+                        */
+                       if (firstpass || array_find(&keylist, keyid)) {
                                array_add(&newkeylist, keyid);
                        }
 
                                array_add(&newkeylist, keyid);
                        }
 
@@ -481,14 +697,19 @@ static int db4_fetch_key_text(const char *search,
                }
                ret = cursor->c_close(cursor);
                cursor = NULL;
                }
                ret = cursor->c_close(cursor);
                cursor = NULL;
-               endtrans();
+               firstpass = 0;
+               db4_endtrans();
        }
        llfree(wordlist, NULL);
        wordlist = NULL;
        }
        llfree(wordlist, NULL);
        wordlist = NULL;
+
+       if (keylist.count > config.maxkeys) {
+               keylist.count = config.maxkeys;
+       }
        
        db4_starttrans();
        for (i = 0; i < keylist.count; i++) {
        
        db4_starttrans();
        for (i = 0; i < keylist.count; i++) {
-               numkeys += fetch_key(keylist.keys[i],
+               numkeys += db4_fetch_key(keylist.keys[i],
                        publickey,
                        true);
        }
                        publickey,
                        true);
        }
@@ -496,121 +717,56 @@ static int db4_fetch_key_text(const char *search,
        free(searchtext);
        searchtext = NULL;
 
        free(searchtext);
        searchtext = NULL;
 
-       endtrans();
+       db4_endtrans();
        
        return (numkeys);
 }
 
 /**
        
        return (numkeys);
 }
 
 /**
- *     store_key - Takes a key and stores it.
- *     @publickey: A pointer to the public key to store.
+ *     delete_key - Given a keyid delete the key from storage.
+ *     @keyid: The keyid to delete.
  *     @intrans: If we're already in a transaction.
  *     @intrans: If we're already in a transaction.
- *     @update: If true the key exists and should be updated.
  *
  *
- *     Again we just use the hex representation of the keyid as the filename
- *     to store the key to. We flatten the public key to a list of OpenPGP
- *     packets and then use write_openpgp_stream() to write the stream out to
- *     the file. If update is true then we delete the old key first, otherwise
- *     we trust that it doesn't exist.
+ *     This function deletes a public key from whatever storage mechanism we
+ *     are using. Returns 0 if the key existed.
  */
  */
-static int db4_store_key(struct openpgp_publickey *publickey, bool intrans,
-               bool update)
+static int db4_delete_key(uint64_t keyid, bool intrans)
 {
 {
-       struct     openpgp_packet_list *packets = NULL;
-       struct     openpgp_packet_list *list_end = NULL;
-       struct     openpgp_publickey *next = NULL;
-       int        ret = 0;
-       int        i = 0;
-       struct     buffer_ctx storebuf;
-       DBT        key;
-       DBT        data;
-       uint64_t   keyid = 0;
+       struct openpgp_publickey *publickey = NULL;
+       DBT key, data;
+       DBC *cursor = NULL;
        uint32_t   shortkeyid = 0;
        uint64_t  *subkeyids = NULL;
        uint32_t   shortkeyid = 0;
        uint64_t  *subkeyids = NULL;
-       char     **uids = NULL;
-       char      *primary = NULL;
+       int ret = 0;
+       int i;
+       char **uids = NULL;
+       char *primary = NULL;
        unsigned char worddb_data[12];
        struct ll *wordlist = NULL;
        struct ll *curword  = NULL;
        unsigned char worddb_data[12];
        struct ll *wordlist = NULL;
        struct ll *curword  = NULL;
-       bool       deadlock = false;
-
-       keyid = get_keyid(publickey);
+       bool deadlock = false;
 
        if (!intrans) {
                db4_starttrans();
        }
 
 
        if (!intrans) {
                db4_starttrans();
        }
 
-       /*
-        * Delete the key if we already have it.
-        *
-        * TODO: Can we optimize this perhaps? Possibly when other data is
-        * involved as well? I suspect this is easiest and doesn't make a lot
-        * of difference though - the largest chunk of data is the keydata and
-        * it definitely needs updated.
-        */
-       if (update) {
-               deadlock = (delete_key(keyid, true) == -1);
-       }
-
-       /*
-        * Convert the key to a flat set of binary data.
-        */
-       if (!deadlock) {
-               next = publickey->next;
-               publickey->next = NULL;
-               flatten_publickey(publickey, &packets, &list_end);
-               publickey->next = next;
-
-               storebuf.offset = 0; 
-               storebuf.size = 8192;
-               storebuf.buffer = malloc(8192);
-       
-               write_openpgp_stream(buffer_putchar, &storebuf, packets);
-
-               /*
-                * Now we have the key data store it in the DB; the keyid is
-                * the key.
-                */
-               memset(&key, 0, sizeof(key));
-               memset(&data, 0, sizeof(data));
-               key.data = &keyid;
-               key.size = sizeof(keyid);
-               data.size = storebuf.offset;
-               data.data = storebuf.buffer;
-
-               ret = keydb(keyid)->put(keydb(keyid),
-                               txn,
-                               &key,
-                               &data,
-                               0); /* flags*/
-               if (ret != 0) {
-                       logthing(LOGTHING_ERROR,
-                                       "Problem storing key: %s",
-                                       db_strerror(ret));
-                       if (ret == DB_LOCK_DEADLOCK) {
-                               deadlock = true;
-                       }
-               }
-
-               free(storebuf.buffer);
-               storebuf.buffer = NULL;
-               storebuf.size = 0;
-               storebuf.offset = 0; 
-       
-               free_packet_list(packets);
-               packets = NULL;
-       }
+       db4_fetch_key(keyid, &publickey, true);
 
        /*
 
        /*
-        * Walk through our uids storing the words into the db with the keyid.
+        * Walk through the uids removing the words from the worddb.
         */
         */
-       if (!deadlock) {
+       if (publickey != NULL) {
                uids = keyuids(publickey, &primary);
        }
        if (uids != NULL) {
                for (i = 0; ret == 0 && uids[i] != NULL; i++) {
                        wordlist = makewordlist(wordlist, uids[i]);
                }
                uids = keyuids(publickey, &primary);
        }
        if (uids != NULL) {
                for (i = 0; ret == 0 && uids[i] != NULL; i++) {
                        wordlist = makewordlist(wordlist, uids[i]);
                }
+                               
+               ret = worddb->cursor(worddb,
+                       txn,
+                       &cursor,
+                       0);   /* flags */
 
                for (curword = wordlist; curword != NULL && !deadlock;
                                curword = curword->next) {
 
                for (curword = wordlist; curword != NULL && !deadlock;
                                curword = curword->next) {
@@ -637,20 +793,32 @@ static int db4_store_key(struct openpgp_publickey *publickey, bool intrans,
                        worddb_data[ 9] = (keyid >> 16) & 0xFF;
                        worddb_data[10] = (keyid >>  8) & 0xFF;
                        worddb_data[11] = keyid & 0xFF; 
                        worddb_data[ 9] = (keyid >> 16) & 0xFF;
                        worddb_data[10] = (keyid >>  8) & 0xFF;
                        worddb_data[11] = keyid & 0xFF; 
-                       ret = worddb->put(worddb,
-                               txn,
+
+                       ret = cursor->c_get(cursor,
                                &key,
                                &data,
                                &key,
                                &data,
-                               0);
+                               DB_GET_BOTH);
+
+                       if (ret == 0) {
+                               ret = cursor->c_del(cursor, 0);
+                               if (ret != 0) {
+                                       logthing(LOGTHING_ERROR,
+                                               "Problem deleting word: %s",
+                                               db_strerror(ret));
+                               }
+                       }
+
                        if (ret != 0) {
                                logthing(LOGTHING_ERROR,
                        if (ret != 0) {
                                logthing(LOGTHING_ERROR,
-                                       "Problem storing word: %s",
+                                       "Problem deleting word: %s",
                                        db_strerror(ret));
                                if (ret == DB_LOCK_DEADLOCK) {
                                        deadlock = true;
                                }
                        }
                }
                                        db_strerror(ret));
                                if (ret == DB_LOCK_DEADLOCK) {
                                        deadlock = true;
                                }
                        }
                }
+               ret = cursor->c_close(cursor);
+               cursor = NULL;
 
                /*
                 * Free our UID and word lists.
 
                /*
                 * Free our UID and word lists.
@@ -662,13 +830,16 @@ static int db4_store_key(struct openpgp_publickey *publickey, bool intrans,
                }
                free(uids);
                uids = NULL;
                }
                free(uids);
                uids = NULL;
+               free_publickey(publickey);
+               publickey = NULL;
        }
 
        }
 
-       /*
-        * Write the truncated 32 bit keyid so we can lookup the full id for
-        * queries.
-        */
        if (!deadlock) {
        if (!deadlock) {
+               ret = id32db->cursor(id32db,
+                       txn,
+                       &cursor,
+                       0);   /* flags */
+
                shortkeyid = keyid & 0xFFFFFFFF;
 
                memset(&key, 0, sizeof(key));
                shortkeyid = keyid & 0xFFFFFFFF;
 
                memset(&key, 0, sizeof(key));
@@ -678,22 +849,29 @@ static int db4_store_key(struct openpgp_publickey *publickey, bool intrans,
                data.data = &keyid;
                data.size = sizeof(keyid);
 
                data.data = &keyid;
                data.size = sizeof(keyid);
 
-               ret = id32db->put(id32db,
-                       txn,
+               ret = cursor->c_get(cursor,
                        &key,
                        &data,
                        &key,
                        &data,
-                       0);
+                       DB_GET_BOTH);
+
+               if (ret == 0) {
+                       ret = cursor->c_del(cursor, 0);
+                       if (ret != 0) {
+                               logthing(LOGTHING_ERROR,
+                                       "Problem deleting short keyid: %s",
+                                       db_strerror(ret));
+                       }
+               }
+
                if (ret != 0) {
                        logthing(LOGTHING_ERROR,
                if (ret != 0) {
                        logthing(LOGTHING_ERROR,
-                               "Problem storing short keyid: %s",
+                               "Problem deleting short keyid: %s",
                                db_strerror(ret));
                        if (ret == DB_LOCK_DEADLOCK) {
                                deadlock = true;
                        }
                }
                                db_strerror(ret));
                        if (ret == DB_LOCK_DEADLOCK) {
                                deadlock = true;
                        }
                }
-       }
 
 
-       if (!deadlock) {
                subkeyids = keysubkeys(publickey);
                i = 0;
                while (subkeyids != NULL && subkeyids[i] != 0) {
                subkeyids = keysubkeys(publickey);
                i = 0;
                while (subkeyids != NULL && subkeyids[i] != 0) {
@@ -706,14 +884,24 @@ static int db4_store_key(struct openpgp_publickey *publickey, bool intrans,
                        data.data = &keyid;
                        data.size = sizeof(keyid);
 
                        data.data = &keyid;
                        data.size = sizeof(keyid);
 
-                       ret = id32db->put(id32db,
-                               txn,
+                       ret = cursor->c_get(cursor,
                                &key,
                                &data,
                                &key,
                                &data,
-                               0);
+                               DB_GET_BOTH);
+
+                       if (ret == 0) {
+                               ret = cursor->c_del(cursor, 0);
+                               if (ret != 0) {
+                                       logthing(LOGTHING_ERROR,
+                                               "Problem deleting short"
+                                               " keyid: %s",
+                                               db_strerror(ret));
+                               }
+                       }
+
                        if (ret != 0) {
                                logthing(LOGTHING_ERROR,
                        if (ret != 0) {
                                logthing(LOGTHING_ERROR,
-                                       "Problem storing short keyid: %s",
+                                       "Problem deleting short keyid: %s",
                                        db_strerror(ret));
                                if (ret == DB_LOCK_DEADLOCK) {
                                        deadlock = true;
                                        db_strerror(ret));
                                if (ret == DB_LOCK_DEADLOCK) {
                                        deadlock = true;
@@ -724,60 +912,138 @@ static int db4_store_key(struct openpgp_publickey *publickey, bool intrans,
                        free(subkeyids);
                        subkeyids = NULL;
                }
                        free(subkeyids);
                        subkeyids = NULL;
                }
+
+               ret = cursor->c_close(cursor);
+               cursor = NULL;
+       }
+
+       if (!deadlock) {
+               key.data = &keyid;
+               key.size = sizeof(keyid);
+
+               keydb(keyid)->del(keydb(keyid),
+                               txn,
+                               &key,
+                               0); /* flags */
        }
 
        if (!intrans) {
        }
 
        if (!intrans) {
-               endtrans();
+               db4_endtrans();
        }
 
        }
 
-       return deadlock ? -1 : 0 ;
+       return deadlock ? (-1) : (ret == DB_NOTFOUND);
 }
 
 /**
 }
 
 /**
- *     delete_key - Given a keyid delete the key from storage.
- *     @keyid: The keyid to delete.
+ *     store_key - Takes a key and stores it.
+ *     @publickey: A pointer to the public key to store.
  *     @intrans: If we're already in a transaction.
  *     @intrans: If we're already in a transaction.
+ *     @update: If true the key exists and should be updated.
  *
  *
- *     This function deletes a public key from whatever storage mechanism we
- *     are using. Returns 0 if the key existed.
+ *     Again we just use the hex representation of the keyid as the filename
+ *     to store the key to. We flatten the public key to a list of OpenPGP
+ *     packets and then use write_openpgp_stream() to write the stream out to
+ *     the file. If update is true then we delete the old key first, otherwise
+ *     we trust that it doesn't exist.
  */
  */
-static int db4_delete_key(uint64_t keyid, bool intrans)
+static int db4_store_key(struct openpgp_publickey *publickey, bool intrans,
+               bool update)
 {
 {
-       struct openpgp_publickey *publickey = NULL;
-       DBT key, data;
-       DBC *cursor = NULL;
+       struct     openpgp_packet_list *packets = NULL;
+       struct     openpgp_packet_list *list_end = NULL;
+       struct     openpgp_publickey *next = NULL;
+       int        ret = 0;
+       int        i = 0;
+       struct     buffer_ctx storebuf;
+       DBT        key;
+       DBT        data;
+       uint64_t   keyid = 0;
        uint32_t   shortkeyid = 0;
        uint64_t  *subkeyids = NULL;
        uint32_t   shortkeyid = 0;
        uint64_t  *subkeyids = NULL;
-       int ret = 0;
-       int i;
-       char **uids = NULL;
-       char *primary = NULL;
+       char     **uids = NULL;
+       char      *primary = NULL;
        unsigned char worddb_data[12];
        struct ll *wordlist = NULL;
        struct ll *curword  = NULL;
        unsigned char worddb_data[12];
        struct ll *wordlist = NULL;
        struct ll *curword  = NULL;
-       bool deadlock = false;
+       bool       deadlock = false;
+
+       keyid = get_keyid(publickey);
 
        if (!intrans) {
                db4_starttrans();
        }
 
 
        if (!intrans) {
                db4_starttrans();
        }
 
-       fetch_key(keyid, &publickey, true);
+       /*
+        * Delete the key if we already have it.
+        *
+        * TODO: Can we optimize this perhaps? Possibly when other data is
+        * involved as well? I suspect this is easiest and doesn't make a lot
+        * of difference though - the largest chunk of data is the keydata and
+        * it definitely needs updated.
+        */
+       if (update) {
+               deadlock = (db4_delete_key(keyid, true) == -1);
+       }
 
        /*
 
        /*
-        * Walk through the uids removing the words from the worddb.
+        * Convert the key to a flat set of binary data.
         */
         */
-       if (publickey != NULL) {
+       if (!deadlock) {
+               next = publickey->next;
+               publickey->next = NULL;
+               flatten_publickey(publickey, &packets, &list_end);
+               publickey->next = next;
+
+               storebuf.offset = 0; 
+               storebuf.size = 8192;
+               storebuf.buffer = malloc(8192);
+       
+               write_openpgp_stream(buffer_putchar, &storebuf, packets);
+
+               /*
+                * Now we have the key data store it in the DB; the keyid is
+                * the key.
+                */
+               memset(&key, 0, sizeof(key));
+               memset(&data, 0, sizeof(data));
+               key.data = &keyid;
+               key.size = sizeof(keyid);
+               data.size = storebuf.offset;
+               data.data = storebuf.buffer;
+
+               ret = keydb(keyid)->put(keydb(keyid),
+                               txn,
+                               &key,
+                               &data,
+                               0); /* flags*/
+               if (ret != 0) {
+                       logthing(LOGTHING_ERROR,
+                                       "Problem storing key: %s",
+                                       db_strerror(ret));
+                       if (ret == DB_LOCK_DEADLOCK) {
+                               deadlock = true;
+                       }
+               }
+
+               free(storebuf.buffer);
+               storebuf.buffer = NULL;
+               storebuf.size = 0;
+               storebuf.offset = 0; 
+       
+               free_packet_list(packets);
+               packets = NULL;
+       }
+
+       /*
+        * Walk through our uids storing the words into the db with the keyid.
+        */
+       if (!deadlock) {
                uids = keyuids(publickey, &primary);
        }
        if (uids != NULL) {
                for (i = 0; ret == 0 && uids[i] != NULL; i++) {
                        wordlist = makewordlist(wordlist, uids[i]);
                }
                uids = keyuids(publickey, &primary);
        }
        if (uids != NULL) {
                for (i = 0; ret == 0 && uids[i] != NULL; i++) {
                        wordlist = makewordlist(wordlist, uids[i]);
                }
-                               
-               ret = worddb->cursor(worddb,
-                       txn,
-                       &cursor,
-                       0);   /* flags */
 
                for (curword = wordlist; curword != NULL && !deadlock;
                                curword = curword->next) {
 
                for (curword = wordlist; curword != NULL && !deadlock;
                                curword = curword->next) {
@@ -804,32 +1070,20 @@ static int db4_delete_key(uint64_t keyid, bool intrans)
                        worddb_data[ 9] = (keyid >> 16) & 0xFF;
                        worddb_data[10] = (keyid >>  8) & 0xFF;
                        worddb_data[11] = keyid & 0xFF; 
                        worddb_data[ 9] = (keyid >> 16) & 0xFF;
                        worddb_data[10] = (keyid >>  8) & 0xFF;
                        worddb_data[11] = keyid & 0xFF; 
-
-                       ret = cursor->c_get(cursor,
+                       ret = worddb->put(worddb,
+                               txn,
                                &key,
                                &data,
                                &key,
                                &data,
-                               DB_GET_BOTH);
-
-                       if (ret == 0) {
-                               ret = cursor->c_del(cursor, 0);
-                               if (ret != 0) {
-                                       logthing(LOGTHING_ERROR,
-                                               "Problem deleting word: %s",
-                                               db_strerror(ret));
-                               }
-                       }
-
+                               0);
                        if (ret != 0) {
                                logthing(LOGTHING_ERROR,
                        if (ret != 0) {
                                logthing(LOGTHING_ERROR,
-                                       "Problem deleting word: %s",
+                                       "Problem storing word: %s",
                                        db_strerror(ret));
                                if (ret == DB_LOCK_DEADLOCK) {
                                        deadlock = true;
                                }
                        }
                }
                                        db_strerror(ret));
                                if (ret == DB_LOCK_DEADLOCK) {
                                        deadlock = true;
                                }
                        }
                }
-               ret = cursor->c_close(cursor);
-               cursor = NULL;
 
                /*
                 * Free our UID and word lists.
 
                /*
                 * Free our UID and word lists.
@@ -841,16 +1095,13 @@ static int db4_delete_key(uint64_t keyid, bool intrans)
                }
                free(uids);
                uids = NULL;
                }
                free(uids);
                uids = NULL;
-               free_publickey(publickey);
-               publickey = NULL;
        }
 
        }
 
+       /*
+        * Write the truncated 32 bit keyid so we can lookup the full id for
+        * queries.
+        */
        if (!deadlock) {
        if (!deadlock) {
-               ret = id32db->cursor(id32db,
-                       txn,
-                       &cursor,
-                       0);   /* flags */
-
                shortkeyid = keyid & 0xFFFFFFFF;
 
                memset(&key, 0, sizeof(key));
                shortkeyid = keyid & 0xFFFFFFFF;
 
                memset(&key, 0, sizeof(key));
@@ -860,29 +1111,22 @@ static int db4_delete_key(uint64_t keyid, bool intrans)
                data.data = &keyid;
                data.size = sizeof(keyid);
 
                data.data = &keyid;
                data.size = sizeof(keyid);
 
-               ret = cursor->c_get(cursor,
+               ret = id32db->put(id32db,
+                       txn,
                        &key,
                        &data,
                        &key,
                        &data,
-                       DB_GET_BOTH);
-
-               if (ret == 0) {
-                       ret = cursor->c_del(cursor, 0);
-                       if (ret != 0) {
-                               logthing(LOGTHING_ERROR,
-                                       "Problem deleting short keyid: %s",
-                                       db_strerror(ret));
-                       }
-               }
-
+                       0);
                if (ret != 0) {
                        logthing(LOGTHING_ERROR,
                if (ret != 0) {
                        logthing(LOGTHING_ERROR,
-                               "Problem deleting short keyid: %s",
+                               "Problem storing short keyid: %s",
                                db_strerror(ret));
                        if (ret == DB_LOCK_DEADLOCK) {
                                deadlock = true;
                        }
                }
                                db_strerror(ret));
                        if (ret == DB_LOCK_DEADLOCK) {
                                deadlock = true;
                        }
                }
+       }
 
 
+       if (!deadlock) {
                subkeyids = keysubkeys(publickey);
                i = 0;
                while (subkeyids != NULL && subkeyids[i] != 0) {
                subkeyids = keysubkeys(publickey);
                i = 0;
                while (subkeyids != NULL && subkeyids[i] != 0) {
@@ -895,24 +1139,14 @@ static int db4_delete_key(uint64_t keyid, bool intrans)
                        data.data = &keyid;
                        data.size = sizeof(keyid);
 
                        data.data = &keyid;
                        data.size = sizeof(keyid);
 
-                       ret = cursor->c_get(cursor,
+                       ret = id32db->put(id32db,
+                               txn,
                                &key,
                                &data,
                                &key,
                                &data,
-                               DB_GET_BOTH);
-
-                       if (ret == 0) {
-                               ret = cursor->c_del(cursor, 0);
-                               if (ret != 0) {
-                                       logthing(LOGTHING_ERROR,
-                                               "Problem deleting short"
-                                               " keyid: %s",
-                                               db_strerror(ret));
-                               }
-                       }
-
+                               0);
                        if (ret != 0) {
                                logthing(LOGTHING_ERROR,
                        if (ret != 0) {
                                logthing(LOGTHING_ERROR,
-                                       "Problem deleting short keyid: %s",
+                                       "Problem storing short keyid: %s",
                                        db_strerror(ret));
                                if (ret == DB_LOCK_DEADLOCK) {
                                        deadlock = true;
                                        db_strerror(ret));
                                if (ret == DB_LOCK_DEADLOCK) {
                                        deadlock = true;
@@ -923,26 +1157,13 @@ static int db4_delete_key(uint64_t keyid, bool intrans)
                        free(subkeyids);
                        subkeyids = NULL;
                }
                        free(subkeyids);
                        subkeyids = NULL;
                }
-
-               ret = cursor->c_close(cursor);
-               cursor = NULL;
-       }
-
-       if (!deadlock) {
-               key.data = &keyid;
-               key.size = sizeof(keyid);
-
-               keydb(keyid)->del(keydb(keyid),
-                               txn,
-                               &key,
-                               0); /* flags */
        }
 
        if (!intrans) {
        }
 
        if (!intrans) {
-               endtrans();
+               db4_endtrans();
        }
 
        }
 
-       return deadlock ? (-1) : (ret == DB_NOTFOUND);
+       return deadlock ? -1 : 0 ;
 }
 
 /**
 }
 
 /**
@@ -1011,55 +1232,6 @@ static int db4_iterate_keys(void (*iterfunc)(void *ctx,
        return numkeys;
 }
 
        return numkeys;
 }
 
-/**
- *     getfullkeyid - Maps a 32bit key id to a 64bit one.
- *     @keyid: The 32bit keyid.
- *
- *     This function maps a 32bit key id to the full 64bit one. It returns the
- *     full keyid. If the key isn't found a keyid of 0 is returned.
- */
-static uint64_t db4_getfullkeyid(uint64_t keyid)
-{
-       DBT       key, data;
-       DBC      *cursor = NULL;
-       uint32_t  shortkeyid = 0;
-       int       ret = 0;
-
-       if (keyid < 0x100000000LL) {
-               ret = id32db->cursor(id32db,
-                               txn,
-                               &cursor,
-                               0);   /* flags */
-
-               shortkeyid = keyid & 0xFFFFFFFF;
-
-               memset(&key, 0, sizeof(key));
-               memset(&data, 0, sizeof(data));
-               key.data = &shortkeyid;
-               key.size = sizeof(shortkeyid);
-               data.flags = DB_DBT_MALLOC;
-
-               ret = cursor->c_get(cursor,
-                       &key,
-                       &data,
-                       DB_SET);
-
-               if (ret == 0) {
-                       keyid = *(uint64_t *) data.data;
-
-                       if (data.data != NULL) {
-                               free(data.data);
-                               data.data = NULL;
-                       }
-               }
-
-               ret = cursor->c_close(cursor);
-               cursor = NULL;
-       }
-       
-       return keyid;
-}
-
 /*
  * Include the basic keydb routines.
  */
 /*
  * Include the basic keydb routines.
  */