]> the.earth.li Git - onak.git/blobdiff - keydb_file.c
Add ability to drop overly large packets
[onak.git] / keydb_file.c
index 1ef81b473825818123d5c633cdeb10ae672383be..bc3c578667a0a36e205b69567ea3e2dd953f43e2 100644 (file)
@@ -1,9 +1,19 @@
 /*
  * keydb.c - Routines to store and fetch keys.
  *
- * Jonathan McDowell <noodles@earth.li>
+ * Copyright 2002-2004 Jonathan McDowell <noodles@earth.li>
  *
- * Copyright 2002-2004 Project Purple
+ * This program is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the Free
+ * Software Foundation; version 2 of the License.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along with
+ * this program.  If not, see <https://www.gnu.org/licenses/>.
  */
 
 #include <sys/types.h>
@@ -11,6 +21,8 @@
 #include <dirent.h>
 #include <errno.h>
 #include <fcntl.h>
+#include <inttypes.h>
+#include <stdbool.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include "keydb.h"
 #include "keyid.h"
 #include "keystructs.h"
-#include "ll.h"
 #include "log.h"
 #include "mem.h"
+#include "onak.h"
 #include "onak-conf.h"
 #include "parsekey.h"
 
-/**
- *     initdb - Initialize the key database.
- *
- *     This is just a no-op for flat file access.
- */
-void initdb(bool readonly)
-{
-}
-
-/**
- *     cleanupdb - De-initialize the key database.
- *
- *     This is just a no-op for flat file access.
- */
-void cleanupdb(void)
-{
-}
-
 /**
  *     starttrans - Start a transaction.
  *
  *     This is just a no-op for flat file access.
  */
-bool starttrans(void)
+static bool file_starttrans(struct onak_dbctx *dbctx)
 {
        return true;
 }
@@ -59,13 +53,13 @@ bool starttrans(void)
  *
  *     This is just a no-op for flat file access.
  */
-void endtrans(void)
+static void file_endtrans(struct onak_dbctx *dbctx)
 {
        return;
 }
 
 /**
- *     fetch_key - Given a keyid fetch the key from storage.
+ *     fetch_key_id - Given a keyid fetch the key from storage.
  *     @keyid: The keyid to fetch.
  *     @publickey: A pointer to a structure to return the key in.
  *     @intrans: If we're already in a transaction.
@@ -76,14 +70,17 @@ 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_id(struct onak_dbctx *dbctx,
+               uint64_t keyid,
+               struct openpgp_publickey **publickey,
                bool intrans)
 {
+       char *db_dir = (char *) dbctx->priv;
        struct openpgp_packet_list *packets = NULL;
        char keyfile[1024];
        int fd = -1;
 
-       snprintf(keyfile, 1023, "%s/0x%llX", config.db_dir,
+       snprintf(keyfile, 1023, "%s/0x%" PRIX64, db_dir,
                        keyid & 0xFFFFFFFF);
        fd = open(keyfile, O_RDONLY); // | O_SHLOCK);
 
@@ -109,16 +106,24 @@ 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 onak_dbctx *dbctx,
+               struct openpgp_publickey *publickey, bool intrans,
+               bool update)
 {
+       char *db_dir = (char *) dbctx->priv;
        struct openpgp_packet_list *packets = NULL;
        struct openpgp_packet_list *list_end = NULL;
        struct openpgp_publickey *next = NULL;
        char keyfile[1024];
        int fd = -1;
+       uint64_t keyid;
 
-       snprintf(keyfile, 1023, "%s/0x%llX", config.db_dir,
-                       get_keyid(publickey) & 0xFFFFFFFF);
+       if (get_keyid(publickey, &keyid) != ONAK_E_OK) {
+               logthing(LOGTHING_ERROR, "Couldn't find key ID for key.");
+               return 0;
+       }
+       snprintf(keyfile, 1023, "%s/0x%" PRIX64, db_dir,
+                       keyid & 0xFFFFFFFF);
        fd = open(keyfile, O_WRONLY | O_CREAT, 0664); // | O_EXLOCK);
 
        if (fd > -1) {
@@ -144,11 +149,13 @@ 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(struct onak_dbctx *dbctx,
+               uint64_t keyid, bool intrans)
 {
+       char *db_dir = (char *) dbctx->priv;
        char keyfile[1024];
 
-       snprintf(keyfile, 1023, "%s/0x%llX", config.db_dir,
+       snprintf(keyfile, 1023, "%s/0x%" PRIX64, db_dir,
                        keyid & 0xFFFFFFFF);
 
        return unlink(keyfile);
@@ -164,7 +171,9 @@ 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(struct onak_dbctx *dbctx,
+               const char *search,
+               struct openpgp_publickey **publickey)
 {
        return 0;
 }
@@ -180,9 +189,11 @@ int fetch_key_text(const char *search, struct openpgp_publickey **publickey)
  *
  *     Returns the number of keys we iterated over.
  */
-int iterate_keys(void (*iterfunc)(void *ctx, struct openpgp_publickey *key),
+static int file_iterate_keys(struct onak_dbctx *dbctx,
+               void (*iterfunc)(void *ctx, struct openpgp_publickey *key),
                void *ctx)
 {
+       char *db_dir = (char *) dbctx->priv;
        int                         numkeys = 0;
        struct openpgp_packet_list *packets = NULL;
        struct openpgp_publickey   *key = NULL;
@@ -191,14 +202,14 @@ int iterate_keys(void (*iterfunc)(void *ctx, struct openpgp_publickey *key),
        int                         fd = -1;
        struct dirent              *curfile = NULL;
 
-       dir = opendir(config.db_dir);
+       dir = opendir(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,
+                                               db_dir,
                                                curfile->d_name);
                                fd = open(keyfile, O_RDONLY);
 
@@ -228,22 +239,6 @@ int iterate_keys(void (*iterfunc)(void *ctx, struct openpgp_publickey *key),
        return numkeys;
 }
 
-
-/**
- *     dumpdb - dump the key database
- *     @filenamebase: The base filename to use for the dump.
- *
- *     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)
-{
-       return 0;
-}
-
 /*
  * Include the basic keydb routines.
  */
@@ -251,4 +246,57 @@ int dumpdb(char *filenamebase)
 #define NEED_GETKEYSIGS 1
 #define NEED_GETFULLKEYID 1
 #define NEED_UPDATEKEYS 1
+#define NEED_GET_FP 1
 #include "keydb.c"
+
+/**
+ *     cleanupdb - De-initialize the key database.
+ *
+ *     This is just a no-op for flat file access.
+ */
+static void file_cleanupdb(struct onak_dbctx *dbctx)
+{
+       if (dbctx->priv != NULL) {
+               free(dbctx->priv);
+               dbctx->priv = NULL;
+       }
+
+       if (dbctx != NULL) {
+               free(dbctx);
+       }
+}
+
+/**
+ *     initdb - Initialize the key database.
+ *
+ *     This is just a no-op for flat file access.
+ */
+struct onak_dbctx *keydb_file_init(struct onak_db_config *dbcfg, bool readonly)
+{
+       struct onak_dbctx *dbctx;
+
+       dbctx = malloc(sizeof(struct onak_dbctx));
+       if (dbctx == NULL) {
+               return NULL;
+       }
+
+       dbctx->config = dbcfg;
+       dbctx->priv = strdup(dbcfg->location);
+
+       dbctx->cleanupdb                = file_cleanupdb;
+       dbctx->starttrans               = file_starttrans;
+       dbctx->endtrans                 = file_endtrans;
+       dbctx->fetch_key_id             = file_fetch_key_id;
+       dbctx->fetch_key_fp             = generic_fetch_key_fp;
+       dbctx->fetch_key_text           = file_fetch_key_text;
+       dbctx->store_key                = file_store_key;
+       dbctx->update_keys              = generic_update_keys;
+       dbctx->delete_key               = file_delete_key;
+       dbctx->getkeysigs               = generic_getkeysigs;
+       dbctx->cached_getkeysigs        = generic_cached_getkeysigs;
+       dbctx->keyid2uid                = generic_keyid2uid;
+       dbctx->getfullkeyid             = generic_getfullkeyid;
+       dbctx->iterate_keys             = file_iterate_keys;
+
+       return dbctx;
+}