Use dynamic context for all backend databases

[onak.git] / keyd.c

diff --git a/keyd.c b/keyd.c
index 4528c96fef874963542b56eccac17896887ab493..f3966db96dcdbafde6b92b341566b1199da791a7 100644 (file)
--- a/keyd.c
+++ b/keyd.c
@@ -144,13 +144,14 @@ int sock_init(const char *sockname)
        return fd;
 }
 
-int sock_do(int fd)
+int sock_do(struct onak_dbctx *dbctx, int fd)
 {
        uint32_t cmd = KEYD_CMD_UNKNOWN;
        ssize_t  bytes = 0;
        ssize_t  count = 0;
        int      ret = 0;
        uint64_t keyid = 0;
+       uint8_t  fp[MAX_FINGERPRINT_LEN];
        char     *search = NULL;
        struct openpgp_publickey *key = NULL;
        struct openpgp_packet_list *packets = NULL;
@@ -183,7 +184,7 @@ int sock_do(int fd)
                        write(fd, &cmd, sizeof(cmd));
                        write(fd, &keyd_version, sizeof(keyd_version));
                        break;
-               case KEYD_CMD_GET:
+               case KEYD_CMD_GET_ID:
                        cmd = KEYD_REPLY_OK;
                        write(fd, &cmd, sizeof(cmd));
                        bytes = read(fd, &keyid, sizeof(keyid));
@@ -196,8 +197,9 @@ int sock_do(int fd)
                                                "Fetching 0x%" PRIX64
                                                ", result: %d",
                                                keyid,
-                                               config.dbbackend->
-                                               fetch_key(keyid, &key, false));
+                                               dbctx->fetch_key_id(dbctx,
+                                                       keyid,
+                                                       &key, false));
                                if (key != NULL) {
                                        storebuf.size = 8192;
                                        storebuf.buffer = malloc(8192);
@@ -229,7 +231,56 @@ int sock_do(int fd)
                                }
                        }
                        break;
-               case KEYD_CMD_GETTEXT:
+               case KEYD_CMD_GET_FP:
+                       cmd = KEYD_REPLY_OK;
+                       write(fd, &cmd, sizeof(cmd));
+                       read(fd, &bytes, 1);
+                       if (bytes > MAX_FINGERPRINT_LEN) {
+                               ret = 1;
+                       } else {
+                               read(fd, fp, bytes);
+                       }
+                       storebuf.offset = 0;
+                       if (ret == 0) {
+                               logthing(LOGTHING_INFO,
+                                               "Fetching by fingerprint"
+                                               ", result: %d",
+                                               dbctx->fetch_key_fp(dbctx,
+                                                       fp, bytes,
+                                                       &key, false));
+                               if (key != NULL) {
+                                       storebuf.size = 8192;
+                                       storebuf.buffer = malloc(8192);
+
+                                       flatten_publickey(key,
+                                                       &packets,
+                                                       &list_end);
+                                       write_openpgp_stream(buffer_putchar,
+                                                       &storebuf,
+                                                       packets);
+                                       logthing(LOGTHING_TRACE,
+                                                       "Sending %d bytes.",
+                                                       storebuf.offset);
+                                       write(fd, &storebuf.offset,
+                                               sizeof(storebuf.offset));
+                                       write(fd, storebuf.buffer,
+                                               storebuf.offset);
+
+                                       free(storebuf.buffer);
+                                       storebuf.buffer = NULL;
+                                       storebuf.size = storebuf.offset = 0;
+                                       free_packet_list(packets);
+                                       packets = list_end = NULL;
+                                       free_publickey(key);
+                                       key = NULL;
+                               } else {
+                                       write(fd, &storebuf.offset,
+                                               sizeof(storebuf.offset));
+                               }
+                       }
+                       break;
+
+               case KEYD_CMD_GET_TEXT:
                        cmd = KEYD_REPLY_OK;
                        write(fd, &cmd, sizeof(cmd));
                        bytes = read(fd, &count, sizeof(count));
@@ -244,8 +295,8 @@ int sock_do(int fd)
                                logthing(LOGTHING_INFO,
                                                "Fetching %s, result: %d",
                                                search,
-                                               config.dbbackend->
-                                               fetch_key_text(search, &key));
+                                               dbctx->fetch_key_text(dbctx,
+                                                       search, &key));
                                if (key != NULL) {
                                        storebuf.size = 8192;
                                        storebuf.buffer = malloc(8192);
@@ -275,6 +326,7 @@ int sock_do(int fd)
                                        write(fd, &storebuf.offset,
                                                sizeof(storebuf.offset));
                                }
+                               free(search);
                        }
                        break;
                case KEYD_CMD_STORE:
@@ -305,7 +357,7 @@ int sock_do(int fd)
                                                &packets,
                                                0);
                                parse_keys(packets, &key);
-                               config.dbbackend->store_key(key, false, false);
+                               dbctx->store_key(dbctx, key, false, false);
                                free_packet_list(packets);
                                packets = NULL;
                                free_publickey(key);
@@ -327,7 +379,7 @@ int sock_do(int fd)
                                                "Deleting 0x%" PRIX64
                                                ", result: %d",
                                                keyid,
-                                               config.dbbackend->delete_key(
+                                               dbctx->delete_key(dbctx,
                                                        keyid, false));
                        }
                        break;
@@ -339,7 +391,7 @@ int sock_do(int fd)
                                ret = 1;
                        }
                        if (ret == 0) {
-                               keyid = config.dbbackend->getfullkeyid(keyid);
+                               keyid = dbctx->getfullkeyid(dbctx, keyid);
                                cmd = sizeof(keyid);
                                write(fd, &cmd, sizeof(cmd));
                                write(fd, &keyid, sizeof(keyid));
@@ -348,7 +400,7 @@ int sock_do(int fd)
                case KEYD_CMD_KEYITER:
                        cmd = KEYD_REPLY_OK;
                        write(fd, &cmd, sizeof(cmd));
-                       config.dbbackend->iterate_keys(iteratefunc,
+                       dbctx->iterate_keys(dbctx, iteratefunc,
                                        &fd);
                        bytes = 0;
                        write(fd, &bytes, sizeof(bytes));
@@ -374,7 +426,7 @@ int sock_do(int fd)
                        write(fd, stats,
                                sizeof(*stats));
                        break;
-               case KEYD_CMD_GETSKSHASH:
+               case KEYD_CMD_GET_SKSHASH:
                        cmd = KEYD_REPLY_OK;
                        write(fd, &cmd, sizeof(cmd));
                        bytes = read(fd, hash.hash, sizeof(hash.hash));
@@ -386,9 +438,8 @@ int sock_do(int fd)
                                logthing(LOGTHING_INFO,
                                                "Fetching by hash"
                                                ", result: %d",
-                                               config.dbbackend->
-                                               fetch_key_skshash(&hash,
-                                                       &key));
+                                               dbctx->fetch_key_skshash(dbctx,
+                                                       &hash, &key));
                                if (key != NULL) {
                                        storebuf.size = 8192;
                                        storebuf.buffer = malloc(8192);
@@ -479,6 +530,7 @@ int main(int argc, char *argv[])
        char *configfile = NULL;
        bool foreground = false;
        int optchar;
+       struct onak_dbctx *dbctx;
 
        while ((optchar = getopt(argc, argv, "c:fh")) != -1 ) {
                switch (optchar) {
@@ -526,7 +578,7 @@ int main(int argc, char *argv[])
                maxfd = fd;
                memset(clients, -1, sizeof (clients));
 
-               config.dbbackend->initdb(false);
+               dbctx = config.dbinit(false);
 
                logthing(LOGTHING_NOTICE, "Accepting connections.");
                while (!cleanup() && select(maxfd + 1, &rfds, NULL, NULL, NULL) != -1) {
@@ -540,7 +592,7 @@ int main(int argc, char *argv[])
                                                FD_ISSET(clients[i], &rfds)) {
                                        logthing(LOGTHING_DEBUG,
                                                "Handling connection for client %d.", i);
-                                       if (sock_do(clients[i])) {
+                                       if (sock_do(dbctx, clients[i])) {
                                                sock_close(clients[i]);
                                                clients[i] = -1;
                                                logthing(LOGTHING_DEBUG,
@@ -574,7 +626,7 @@ int main(int argc, char *argv[])
                                }
                        }
                }
-               config.dbbackend->cleanupdb();
+               dbctx->cleanupdb(dbctx);
                sock_close(fd);
                unlink(sockname);
        }