X-Git-Url: https://git.sommitrealweird.co.uk/onak.git/blobdiff_plain/394938c479925750ffe56c28d281945ba4003a14..9c96da2b288cebb53388ad4e30adbe9b87f67d91:/keydb_file.c?ds=inline

diff --git a/keydb_file.c b/keydb_file.c
index 4a4322b..88bb680 100644
--- a/keydb_file.c
+++ b/keydb_file.c
@@ -3,11 +3,12 @@
  *
  * Jonathan McDowell <noodles@earth.li>
  *
- * Copyright 2002 Project Purple
+ * Copyright 2002-2004 Project Purple
  */
 
 #include <sys/types.h>
 #include <sys/uio.h>
+#include <dirent.h>
 #include <errno.h>
 #include <fcntl.h>
 #include <stdio.h>
@@ -20,6 +21,7 @@
 #include "keyid.h"
 #include "keystructs.h"
 #include "ll.h"
+#include "log.h"
 #include "mem.h"
 #include "onak-conf.h"
 #include "parsekey.h"
@@ -29,7 +31,7 @@
  *
  *	This is just a no-op for flat file access.
  */
-void initdb(bool readonly)
+static void file_initdb(bool readonly)
 {
 }
 
@@ -38,7 +40,7 @@ void initdb(bool readonly)
  *
  *	This is just a no-op for flat file access.
  */
-void cleanupdb(void)
+static void file_cleanupdb(void)
 {
 }
 
@@ -47,7 +49,7 @@ void cleanupdb(void)
  *
  *	This is just a no-op for flat file access.
  */
-bool starttrans(void)
+static bool file_starttrans(void)
 {
 	return true;
 }
@@ -57,7 +59,7 @@ bool starttrans(void)
  *
  *	This is just a no-op for flat file access.
  */
-void endtrans(void)
+static void file_endtrans(void)
 {
 	return;
 }
@@ -74,7 +76,7 @@ void endtrans(void)
  *	in and then parse_keys() to parse the packets into a publickey
  *	structure.
  */
-int fetch_key(uint64_t keyid, struct openpgp_publickey **publickey,
+static int file_fetch_key(uint64_t keyid, struct openpgp_publickey **publickey,
 		bool intrans)
 {
 	struct openpgp_packet_list *packets = NULL;
@@ -107,7 +109,8 @@ int fetch_key(uint64_t keyid, struct openpgp_publickey **publickey,
  *	packets and then use write_openpgp_stream() to write the stream out to
  *	the file.
  */
-int store_key(struct openpgp_publickey *publickey, bool intrans, bool update)
+static int file_store_key(struct openpgp_publickey *publickey, bool intrans,
+		bool update)
 {
 	struct openpgp_packet_list *packets = NULL;
 	struct openpgp_packet_list *list_end = NULL;
@@ -142,7 +145,7 @@ int store_key(struct openpgp_publickey *publickey, bool intrans, bool update)
  *	This function deletes a public key from whatever storage mechanism we
  *	are using. Returns 0 if the key existed.
  */
-int delete_key(uint64_t keyid, bool intrans)
+static int file_delete_key(uint64_t keyid, bool intrans)
 {
 	char keyfile[1024];
 
@@ -162,24 +165,69 @@ int delete_key(uint64_t keyid, bool intrans)
  *
  *	TODO: Write for flat file access. Some sort of grep?
  */
-int fetch_key_text(const char *search, struct openpgp_publickey **publickey)
+static int file_fetch_key_text(const char *search,
+		struct openpgp_publickey **publickey)
 {
 	return 0;
 }
 
 /**
- *	dumpdb - dump the key database
- *	@filenamebase: The base filename to use for the dump.
+ *	iterate_keys - call a function once for each key in the db.
+ *	@iterfunc: The function to call.
+ *	@ctx: A context pointer
  *
- *	Dumps the database into one or more files, which contain pure OpenPGP
- *	that can be reimported into onak or gpg. filenamebase provides a base
- *	file name for the dump; several files may be created, all of which will
- *	begin with this string and then have a unique number and a .pgp
- *	extension.
- *          */
-int dumpdb(char *filenamebase)
+ *	Calls iterfunc once for each key in the database. ctx is passed
+ *	unaltered to iterfunc. This function is intended to aid database dumps
+ *	and statistic calculations.
+ *
+ *	Returns the number of keys we iterated over.
+ */
+static int file_iterate_keys(void (*iterfunc)(void *ctx,
+		struct openpgp_publickey *key),	void *ctx)
 {
-	return 0;
+	int                         numkeys = 0;
+	struct openpgp_packet_list *packets = NULL;
+	struct openpgp_publickey   *key = NULL;
+	DIR                        *dir;
+	char                        keyfile[1024];
+	int                         fd = -1;
+	struct dirent              *curfile = NULL;
+
+	dir = opendir(config.db_dir);
+
+	if (dir != NULL) {
+		while ((curfile = readdir(dir)) != NULL) {
+			if (curfile->d_name[0] == '0' &&
+					curfile->d_name[1] == 'x') {
+				snprintf(keyfile, 1023, "%s/%s",
+						config.db_dir,
+						curfile->d_name);
+				fd = open(keyfile, O_RDONLY);
+
+				if (fd > -1) {
+					read_openpgp_stream(file_fetchchar,
+							&fd,
+							&packets,
+							0);
+					parse_keys(packets, &key);
+
+					iterfunc(ctx, key);
+
+					free_publickey(key);
+					key = NULL;
+					free_packet_list(packets);
+					packets = NULL;
+					close(fd);
+				}
+				numkeys++;
+			}
+		}
+		
+		closedir(dir);
+		dir = NULL;
+	}
+
+	return numkeys;
 }
 
 /*
@@ -188,4 +236,22 @@ int dumpdb(char *filenamebase)
 #define NEED_KEYID2UID 1
 #define NEED_GETKEYSIGS 1
 #define NEED_GETFULLKEYID 1
+#define NEED_UPDATEKEYS 1
 #include "keydb.c"
+
+struct dbfuncs keydb_file_funcs = {
+	.initdb			= file_initdb,
+	.cleanupdb		= file_cleanupdb,
+	.starttrans		= file_starttrans,
+	.endtrans		= file_endtrans,
+	.fetch_key		= file_fetch_key,
+	.fetch_key_text		= file_fetch_key_text,
+	.store_key		= file_store_key,
+	.update_keys		= generic_update_keys,
+	.delete_key		= file_delete_key,
+	.getkeysigs		= generic_getkeysigs,
+	.cached_getkeysigs	= generic_cached_getkeysigs,
+	.keyid2uid		= generic_keyid2uid,
+	.getfullkeyid		= generic_getfullkeyid,
+	.iterate_keys		= file_iterate_keys,
+};