cscvs to tla changeset 137

[onak.git] / keydb_db3.c

diff --git a/keydb_db3.c b/keydb_db3.c
index d3d35c85709a8d92ad489fb340f5654e8d3c2ab8..464dcc9d445f792022c9a32efee30f1be33d20b7 100644 (file)
--- a/keydb_db3.c
+++ b/keydb_db3.c
@@ -5,7 +5,7 @@
  *
  * Copyright 2002 Project Purple
  *
  *
  * Copyright 2002 Project Purple
  *

- * $Id: keydb_db3.c,v 1.17 2003/09/28 14:54:57 noodles Exp $
+ * $Id: keydb_db3.c,v 1.26 2004/05/27 03:33:24 noodles Exp $

  */
 
 #include <assert.h>
  */
 
 #include <assert.h>


@@ -30,6 +30,7 @@
 #include "log.h"
 #include "onak-conf.h"
 #include "parsekey.h"
 #include "log.h"
 #include "onak-conf.h"
 #include "parsekey.h"

+#include "wordlist.h"

 
 /**
  *     dbenv - our database environment.
 
 /**
  *     dbenv - our database environment.


@@ -58,57 +59,11 @@ static DB_TXN *txn = NULL;
 
 DB *keydb(uint64_t keyid)
 {
 
 DB *keydb(uint64_t keyid)
 {

-       return(dbconns[keyid % numdbs]);
-}
+       uint64_t keytrun;

 
 

-/**
- *     makewordlist - Takes a string and splits it into a set of unique words.
- *     @wordlist: The current word list.
- *     @words: The string to split and add.
- *
- *     We take words and split it on non alpha numeric characters. These get
- *     added to the word list if they're not already present. If the wordlist
- *     is NULL then we start a new list, otherwise it's search for already
- *     added words. Note that words is modified in the process of scanning.
- *
- *     Returns the new word list.
- */
-struct ll *makewordlist(struct ll *wordlist, char *word)
-{
-       char *start = NULL;
-       char *end = NULL;
-
-       /*
-        * Walk through the words string, spliting on non alphanumerics and
-        * then checking if the word already exists in the list. If not then
-        * we add it.
-        */
-       end = word;
-       while (end != NULL && *end != 0) {
-               start = end;
-               while (*start != 0 && !isalnum(*start)) {
-                       start++;
-               }
-               end = start;
-               while (*end != 0 && isalnum(*end)) {
-                       *end = tolower(*end);
-                       end++;
-               }
-               if (end - start > 1) {
-                       if (*end != 0) {
-                               *end = 0;
-                               end++;
-                       }
-                       
-                       if (llfind(wordlist, start,
-                                       strcmp) == NULL) {
-                               wordlist = lladd(wordlist,
-                                               start);
-                       }
-               }
-       }
+       keytrun = keyid >> 8;

 
 

-       return wordlist;
+       return(dbconns[keytrun % numdbs]);

 }
 
 /**
 }
 
 /**


@@ -118,12 +73,13 @@ struct ll *makewordlist(struct ll *wordlist, char *word)
  *     this file are called in order to allow the DB to be initialized ready
  *     for access.
  */
  *     this file are called in order to allow the DB to be initialized ready
  *     for access.
  */

-void initdb(void)
+void initdb(bool readonly)

 {
 {

-       char  buf[1024];
-       FILE *numdb = NULL;
-       int   ret = 0;
-       int   i = 0;
+       char       buf[1024];
+       FILE      *numdb = NULL;
+       int        ret = 0;
+       int        i = 0;
+       u_int32_t  flags = 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");


@@ -132,9 +88,18 @@ void initdb(void)
                        numdbs = atoi(buf);
                }
                fclose(numdb);
                        numdbs = atoi(buf);
                }
                fclose(numdb);

-       } else {
+       } else if (!readonly) {

                logthing(LOGTHING_ERROR, "Couldn't open num_keydb: %s",
                                strerror(errno));
                logthing(LOGTHING_ERROR, "Couldn't open num_keydb: %s",
                                strerror(errno));

+               numdb = fopen(buf, "w");
+               if (numdb != NULL) {
+                       fprintf(numdb, "%d", numdbs);
+                       fclose(numdb);
+               } else {
+                       logthing(LOGTHING_ERROR,
+                               "Couldn't write num_keydb: %s",
+                               strerror(errno));
+               }

        }
 
        dbconns = malloc(sizeof (DB *) * numdbs);
        }
 
        dbconns = malloc(sizeof (DB *) * numdbs);


@@ -170,7 +135,7 @@ void initdb(void)
                        0);
        if (ret != 0) {
                logthing(LOGTHING_CRITICAL,
                        0);
        if (ret != 0) {
                logthing(LOGTHING_CRITICAL,

-                               "Erroring opening db environment: %s (%s)",
+                               "Error opening db environment: %s (%s)",

                                config.db_dir,
                                db_strerror(ret));
                exit(1);
                                config.db_dir,
                                db_strerror(ret));
                exit(1);


@@ -185,10 +150,14 @@ void initdb(void)
                }
 
                snprintf(buf, 1023, "keydb.%d.db", i);
                }
 
                snprintf(buf, 1023, "keydb.%d.db", i);

+               flags = DB_CREATE;
+               if (readonly) {
+                       flags = DB_RDONLY;
+               }

                ret = dbconns[i]->open(dbconns[i], buf,
                        NULL,
                        DB_HASH,
                ret = dbconns[i]->open(dbconns[i], buf,
                        NULL,
                        DB_HASH,

-                       DB_CREATE,
+                       flags,

                        0664);
                if (ret != 0) {
                        logthing(LOGTHING_CRITICAL,
                        0664);
                if (ret != 0) {
                        logthing(LOGTHING_CRITICAL,


@@ -207,7 +176,7 @@ void initdb(void)
        ret = worddb->set_flags(worddb, DB_DUP);
 
        ret = worddb->open(worddb, "worddb", NULL, DB_BTREE,
        ret = worddb->set_flags(worddb, DB_DUP);
 
        ret = worddb->open(worddb, "worddb", NULL, DB_BTREE,

-                       DB_CREATE,
+                       flags,

                        0664);
        if (ret != 0) {
                logthing(LOGTHING_CRITICAL,
                        0664);
        if (ret != 0) {
                logthing(LOGTHING_CRITICAL,


@@ -340,7 +309,7 @@ int fetch_key(uint64_t keyid, struct openpgp_publickey **publickey,
                fetchbuf.offset = 0;
                fetchbuf.size = data.size;
                read_openpgp_stream(buffer_fetchchar, &fetchbuf,
                fetchbuf.offset = 0;
                fetchbuf.size = data.size;
                read_openpgp_stream(buffer_fetchchar, &fetchbuf,

-                               &packets);
+                               &packets, 0);

                parse_keys(packets, publickey);
                free_packet_list(packets);
                packets = NULL;
                parse_keys(packets, publickey);
                free_packet_list(packets);
                packets = NULL;


@@ -778,41 +747,48 @@ int delete_key(uint64_t keyid, bool intrans)
  */
 int dumpdb(char *filenamebase)
 {
  */
 int dumpdb(char *filenamebase)
 {

-       DBT key, data;
-       DBC *cursor = NULL;
-       int ret = 0;
-       int fd = -1;
-       int i;
+       DBT   key, data;
+       DBC  *cursor = NULL;
+       int   ret = 0;
+       int   fd = -1;
+       int   i = 0;
+       char  filename[1024];

 
 

+       filename[1023] = 0;

        for (i = 0; i < numdbs; i++) {
        for (i = 0; i < numdbs; i++) {

-               starttrans();
-       

                ret = dbconns[i]->cursor(dbconns[i],
                        NULL,
                        &cursor,
                        0);   /* flags */
 
                ret = dbconns[i]->cursor(dbconns[i],
                        NULL,
                        &cursor,
                        0);   /* flags */
 

-               fd = open(filenamebase, O_CREAT | O_WRONLY | O_TRUNC, 0640);
-               memset(&key, 0, sizeof(key));
-               memset(&data, 0, sizeof(data));
-               ret = cursor->c_get(cursor, &key, &data, DB_NEXT);
-               while (ret == 0) {
-                       write(fd, data.data, data.size);
+               snprintf(filename, 1023, "%s.%d.pgp", filenamebase, i);
+               fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC, 0640);
+               if (fd == -1) {
+                       logthing(LOGTHING_ERROR,
+                               "Error opening keydump file (%s): %s",
+                               filename,
+                               strerror(errno));
+               } else {

                        memset(&key, 0, sizeof(key));
                        memset(&data, 0, sizeof(data));
                        ret = cursor->c_get(cursor, &key, &data, DB_NEXT);
                        memset(&key, 0, sizeof(key));
                        memset(&data, 0, sizeof(data));
                        ret = cursor->c_get(cursor, &key, &data, DB_NEXT);

-               }
-               if (ret != DB_NOTFOUND) {
-                       logthing(LOGTHING_ERROR, "Problem reading key: %s",
+                       while (ret == 0) {
+                               write(fd, data.data, data.size);
+                               memset(&key, 0, sizeof(key));
+                               memset(&data, 0, sizeof(data));
+                               ret = cursor->c_get(cursor, &key, &data,
+                                               DB_NEXT);
+                       }
+                       if (ret != DB_NOTFOUND) {
+                               logthing(LOGTHING_ERROR,
+                                       "Problem reading key: %s",

                                        db_strerror(ret));
                                        db_strerror(ret));

+                       }
+                       close(fd);

                }
 
                }
 

-               close(fd);
-

                ret = cursor->c_close(cursor);
                cursor = NULL;
                ret = cursor->c_close(cursor);
                cursor = NULL;

-       
-               endtrans();

        }
        
        return 0;
        }
        
        return 0;