From b7a5a70eb8d8f312325af828b47726977f4fbb79 Mon Sep 17 00:00:00 2001 From: Jonathan McDowell Date: Mon, 31 May 2004 23:48:23 +0000 Subject: [PATCH] cscvs to tla changeset 130 Author: noodles Date: 2004/05/27 03:37:58 Adding filesytem backend (dsilvers). --- keydb_fs.c | 461 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 461 insertions(+) create mode 100644 keydb_fs.c diff --git a/keydb_fs.c b/keydb_fs.c new file mode 100644 index 0000000..7e739d8 --- /dev/null +++ b/keydb_fs.c @@ -0,0 +1,461 @@ +/* + * keydb.h - Routines to store and fetch keys. + * + * Daniel Silverstone + * + * Copyright 2004 Daniel Silverstone and Project Purple + * + * $Id: keydb_fs.c,v 1.1 2004/05/27 03:37:58 noodles Exp $ + */ + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "charfuncs.h" +#include "keydb.h" +#include "keyid.h" +#include "keystructs.h" +#include "ll.h" +#include "mem.h" +#include "onak-conf.h" +#include "parsekey.h" +#include "log.h" +#include "wordlist.h" + +static int keydb_lockfile_fd = -1; +static bool keydb_lockfile_readonly; + +/*****************************************************************************/ + +/* Helper functions */ + +#define FNV_offset_basis 2166136261ul +#define FNV_mixing_prime 16777619ul + +static uint32_t calchash(uint8_t * ptr) +{ + register uint32_t h = FNV_offset_basis; + register uint32_t p = FNV_mixing_prime; + register uint32_t n = strlen(ptr); + register uint8_t *c = ptr; + while (n--) { + h *= p; + h ^= *c++; + } + return h ? h : 1; /* prevent a hash of zero happening */ +} + + +void keypath(char *buffer, uint64_t _keyid) +{ + uint64_t keyid = _keyid << 32; + snprintf(buffer, PATH_MAX, "%s/key/%02X/%02X/%08lX/%016llX", + config.db_dir, (uint8_t) ((keyid >> 56) & 0xFF), + (uint8_t) ((keyid >> 48) & 0xFF), + (uint32_t) (keyid >> 32), _keyid); +} + +void keydir(char *buffer, uint64_t _keyid) +{ + uint64_t keyid = _keyid << 32; + snprintf(buffer, PATH_MAX, "%s/key/%02X/%02X/%08lX", config.db_dir, + (uint8_t) ((keyid >> 56) & 0xFF), + (uint8_t) ((keyid >> 48) & 0xFF), + (uint32_t) (keyid >> 32)); +} + +void prove_path_to(uint64_t keyid, char *what) +{ + static char buffer[1024]; + snprintf(buffer, PATH_MAX, "%s/%s", config.db_dir, what); + mkdir(buffer, 0777); + + snprintf(buffer, PATH_MAX, "%s/%s/%02X", config.db_dir, what, + (uint8_t) ((keyid >> 24) & 0xFF)); + mkdir(buffer, 0777); + + snprintf(buffer, PATH_MAX, "%s/%s/%02X/%02X", config.db_dir, what, + (uint8_t) ((keyid >> 24) & 0xFF), + (uint8_t) ((keyid >> 16) & 0xFF)); + mkdir(buffer, 0777); + + snprintf(buffer, PATH_MAX, "%s/%s/%02X/%02X/%08lX", config.db_dir, what, + (uint8_t) ((keyid >> 24) & 0xFF), + (uint8_t) ((keyid >> 16) & 0xFF), (uint32_t) (keyid)); + mkdir(buffer, 0777); +} + +void wordpath(char *buffer, char *word, uint32_t hash, uint64_t keyid) +{ + snprintf(buffer, PATH_MAX, "%s/words/%02X/%02X/%08lX/%s/%016llX", + config.db_dir, (uint8_t) ((hash >> 24) & 0xFF), + (uint8_t) ((hash >> 16) & 0xFF), hash, word, keyid); +} + +void worddir(char *buffer, char *word, uint32_t hash) +{ + snprintf(buffer, PATH_MAX, "%s/words/%02X/%02X/%08lX/%s", config.db_dir, + (uint8_t) ((hash >> 24) & 0xFF), + (uint8_t) ((hash >> 16) & 0xFF), hash, word); +} + +/*****************************************************************************/ + +/** + * initdb - Initialize the key database. + */ +void initdb(bool readonly) +{ + char buffer[PATH_MAX]; + + keydb_lockfile_readonly = readonly; + + snprintf(buffer, PATH_MAX, "%s/.lock", config.db_dir); + + if (access(config.db_dir, R_OK | W_OK | X_OK) == -1) { + if (errno != ENOENT) { + logthing(LOGTHING_CRITICAL, + "Unable to access keydb_fs root of '%s'. (%s)", + config.db_dir, strerror(errno)); + exit(1); /* Lacking rwx on the key dir */ + } + mkdir(config.db_dir, 0777); + keydb_lockfile_fd = open(buffer, O_RDWR | O_CREAT, 0600); + } + chdir(config.db_dir); + if (keydb_lockfile_fd == -1) + keydb_lockfile_fd = open(buffer, + (keydb_lockfile_readonly) ? + O_RDONLY : O_RDWR); + if (keydb_lockfile_fd == -1) + keydb_lockfile_fd = open(buffer, O_RDWR | O_CREAT, 0600); + if (keydb_lockfile_fd == -1) { + logthing(LOGTHING_CRITICAL, + "Unable to open lockfile '%s'. (%s)", + buffer, strerror(errno)); + exit(1); /* Lacking rwx on the key dir */ + } +} + +/** + * cleanupdb - De-initialize the key database. + */ +void cleanupdb(void) +{ + /* Mmmm nothing to do here? */ + close(keydb_lockfile_fd); +} + +/** + * starttrans - Start a transaction. + */ +bool starttrans(void) +{ + struct flock lockstruct; + int remaining = 20; + lockstruct.l_type = + F_RDLCK | ((keydb_lockfile_readonly) ? 0 : F_WRLCK); + lockstruct.l_whence = SEEK_SET; + lockstruct.l_start = 0; + lockstruct.l_len = 1; + + while (fcntl(keydb_lockfile_fd, F_SETLK, &lockstruct) == -1) { + if (remaining-- == 0) + return false; /* Hope to hell that noodles DTRT */ + usleep(100); + } + return true; +} + +/** + * endtrans - End a transaction. + */ +void endtrans(void) +{ + struct flock lockstruct; + + lockstruct.l_type = F_UNLCK; + lockstruct.l_whence = SEEK_SET; + lockstruct.l_start = 0; + lockstruct.l_len = 1; + fcntl(keydb_lockfile_fd, F_SETLK, &lockstruct); +} + +/** + * fetch_key - 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. + */ +int fetch_key(uint64_t keyid, struct openpgp_publickey **publickey, + bool intrans) +{ + static char buffer[PATH_MAX]; + int ret = 0, fd; + struct openpgp_packet_list *packets = NULL; + + if (!intrans) + starttrans(); + + if ((keyid >> 32) == 0) + keyid = getfullkeyid(keyid); + + keypath(buffer, keyid); + if ((fd = open(buffer, O_RDONLY)) != -1) { + /* File is present, load it in... */ + read_openpgp_stream(file_fetchchar, &fd, &packets, 0); + parse_keys(packets, publickey); + free_packet_list(packets); + packets = NULL; + close(fd); + ret = 1; + } + + if (!intrans) + endtrans(); + return ret; +} + +/** + * 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. + * @update: If true the key exists and should be updated. + */ +int store_key(struct openpgp_publickey *publickey, bool intrans, + bool update) +{ + static char buffer[PATH_MAX]; + static char wbuffer[PATH_MAX]; + int ret = 0, fd; + struct openpgp_packet_list *packets = NULL; + struct openpgp_packet_list *list_end = NULL; + struct openpgp_publickey *next = NULL; + uint64_t keyid = get_keyid(publickey); + struct ll *wordlist = NULL, *wl = NULL; + + + if (!intrans) + starttrans(); + + prove_path_to(keyid, "key"); + keypath(buffer, keyid); + + if ((fd = + open(buffer, O_WRONLY | (update ? O_TRUNC : O_CREAT), + 0644)) != -1) { + next = publickey->next; + publickey->next = NULL; + flatten_publickey(publickey, &packets, &list_end); + publickey->next = next; + + write_openpgp_stream(file_putchar, &fd, packets); + close(fd); + free_packet_list(packets); + packets = NULL; + ret = 1; + } + + if (ret) { + wl = wordlist = makewordlistfromkey(wordlist, publickey); + while (wl) { + uint32_t hash = calchash((uint8_t *) (wl->object)); + prove_path_to(hash, "words"); + + worddir(wbuffer, wl->object, hash); + mkdir(wbuffer, 0777); + wordpath(wbuffer, wl->object, hash, keyid); + link(buffer, wbuffer); + + wl = wl->next; + } + + + + llfree(wordlist, NULL); + } + + if (!intrans) + endtrans(); + return ret; +} + +/** + * delete_key - Given a keyid delete the key from storage. + * @keyid: The keyid to delete. + * @intrans: If we're already in a transaction. + */ +int delete_key(uint64_t keyid, bool intrans) +{ + static char buffer[PATH_MAX]; + int ret; + struct openpgp_publickey *pk = NULL; + struct ll *wordlist = NULL, *wl = NULL; + + if ((keyid >> 32) == 0) + keyid = getfullkeyid(keyid); + + if (!intrans) + starttrans(); + + ret = fetch_key(keyid, &pk, true); + + if (ret) { + logthing(LOGTHING_CRITICAL, "Wordlist for key %016llX", + keyid); + wl = wordlist = makewordlistfromkey(wordlist, pk); + logthing(LOGTHING_CRITICAL, + "Wordlist for key %016llX done", keyid); + while (wl) { + uint32_t hash = calchash((uint8_t *) (wl->object)); + prove_path_to(hash, "words"); + + wordpath(buffer, wl->object, hash, keyid); + unlink(buffer); + + wl = wl->next; + } + } + + keypath(buffer, keyid); + unlink(buffer); + + if (!intrans) + endtrans(); + return 1; +} + +static struct ll *internal_get_key_by_word(char *word, struct ll *mct) +{ + struct ll *keys = NULL; + DIR *d = NULL; + char buffer[PATH_MAX]; + uint32_t hash = calchash((uint8_t *) (word)); + struct dirent *de; + + worddir(buffer, word, hash); + d = opendir(buffer); + logthing(LOGTHING_CRITICAL, "Scanning for word %s in dir %s", word, + buffer); + if (d) + do { + de = readdir(d); + if (de && de->d_name[0] != '.') { + if ((!mct) + || (llfind(mct, de->d_name, strcmp) != + NULL)) { + logthing(LOGTHING_CRITICAL, + "Found %s // %s", word, + de->d_name); + keys = + lladd(keys, + strdup(de->d_name)); + } + } + } while (de); + closedir(d); + + return keys; +} + +/* + * fetch_key_text - Trys to find the keys that contain the supplied text. + * @search: The text to search for. + * @publickey: A pointer to a structure to return the key in. + */ +int fetch_key_text(const char *search, + struct openpgp_publickey **publickey) +{ + struct ll *wordlist = NULL, *wl = NULL; + struct ll *keylist = NULL; + int addedkeys = 0; + + logthing(LOGTHING_CRITICAL, "Search was '%s'", search); + + wl = wordlist = makewordlist(wordlist, search); + + keylist = internal_get_key_by_word(wordlist->object, NULL); + + if (!keylist) { + llfree(wordlist, NULL); + return 0; + } + + wl = wl->next; + while (wl) { + struct ll *nkl = + internal_get_key_by_word(wl->object, keylist); + if (!nkl) { + llfree(wordlist, NULL); + llfree(keylist, free); + return 0; + } + llfree(keylist, free); + keylist = nkl; + wl = wl->next; + } + + llfree(wordlist, NULL); + + /* Now add the keys... */ + wl = keylist; + while (wl) { + logthing(LOGTHING_CRITICAL, "Adding key: %s", wl->object); + addedkeys += + fetch_key(strtoull(wl->object, NULL, 16), publickey, + false); + if (addedkeys >= config.maxkeys) + break; + wl = wl->next; + } + + llfree(keylist, free); + + return addedkeys; +} + +/* + * dumpdb - dump the key database + * @filenamebase: The base filename to use for the dump. + */ +int dumpdb(char *filenamebase) +{ + return 0; +} + +uint64_t getfullkeyid(uint64_t keyid) +{ + static char buffer[PATH_MAX]; + DIR *d; + struct dirent *de; + uint64_t ret = 0; + + keydir(buffer, keyid); + + d = opendir(buffer); + do { + de = readdir(d); + if (de) + if (de && de->d_name[0] != '.') { + ret = strtoull(de->d_name, NULL, 16); + } + } while (de && de->d_name[0] == '.'); + closedir(d); + return ret; +} + +/* + * Include the basic keydb routines. + */ +#define NEED_KEYID2UID 1 +#define NEED_GETKEYSIGS 1 +#include "keydb.c" -- 2.39.5