Use dynamic context for all backend databases

[onak.git] / lookup.c

diff --git a/lookup.c b/lookup.c
index 273afc225c04d567636d92bcfc50d44eaa541752..533433a3c1d66d9fee2e6221e095e5f1b61944ab 100644 (file)
--- a/lookup.c
+++ b/lookup.c
@@ -46,25 +46,30 @@
 #define OP_PHOTO   4
 #define OP_HGET    5
 
-void find_keys(char *search, uint64_t keyid, bool ishex,
-               bool fingerprint, bool skshash, bool exact, bool verbose,
-               bool mrhkp)
+void find_keys(struct onak_dbctx *dbctx,
+               char *search, uint64_t keyid, uint8_t *fp, size_t fpsize,
+               bool ishex, bool isfp, bool fingerprint, bool skshash,
+               bool exact, bool verbose, bool mrhkp)
 {
        struct openpgp_publickey *publickey = NULL;
        int count = 0;
 
        if (ishex) {
-               count = config.dbbackend->fetch_key(keyid, &publickey, false);
+               count = dbctx->fetch_key_id(dbctx, keyid, &publickey,
+                               false);
+       } else if (isfp) {
+               count = dbctx->fetch_key_fp(dbctx, fp, fpsize, &publickey,
+                               false);
        } else {
-               count = config.dbbackend->fetch_key_text(search, &publickey);
+               count = dbctx->fetch_key_text(dbctx, search, &publickey);
        }
        if (publickey != NULL) {
                if (mrhkp) {
                        printf("info:1:%d\n", count);
                        mrkey_index(publickey);
                } else {
-                       key_index(publickey, verbose, fingerprint, skshash,
-                               true);
+                       key_index(dbctx, publickey, verbose, fingerprint,
+                               skshash, true);
                }
                free_publickey(publickey);
        } else if (count == 0) {
@@ -86,6 +91,19 @@ void find_keys(char *search, uint64_t keyid, bool ishex,
        }
 }
 
+static uint8_t hex2bin(char c)
+{
+       if (c >= '0' && c <= '9') {
+               return (c - '0');
+       } else if (c >= 'a' && c <= 'f') {
+               return (c - 'a' + 10);
+       } else if (c >= 'A' && c <= 'F') {
+               return (c - 'A' + 10);
+       }
+
+       return 255;
+}
+
 int main(int argc, char *argv[])
 {
        char **params = NULL;
@@ -96,8 +114,10 @@ int main(int argc, char *argv[])
        bool skshash = false;
        bool exact = false;
        bool ishex = false;
+       bool isfp = false;
        bool mrhkp = false;
        uint64_t keyid = 0;
+       uint8_t fp[MAX_FINGERPRINT_LEN];
        char *search = NULL;
        char *end = NULL;
        struct openpgp_publickey *publickey = NULL;
@@ -105,6 +125,7 @@ int main(int argc, char *argv[])
        struct openpgp_packet_list *list_end = NULL;
        int result;
        struct skshash hash;
+       struct onak_dbctx *dbctx;
 
        params = getcgivars(argc, argv);
        for (i = 0; params != NULL && params[i] != NULL; i += 2) {
@@ -125,14 +146,12 @@ int main(int argc, char *argv[])
                        params[i+1] = NULL;
                        if (search != NULL && strlen(search) == 42 &&
                                        search[0] == '0' && search[1] == 'x') {
-                               /*
-                                * Fingerprint. Truncate to last 64 bits for
-                                * now.
-                                */
-                               keyid = strtoull(&search[26], &end, 16);
-                               if (end != NULL && *end == 0) {
-                                       ishex = true;
+                               for (i = 0; i < MAX_FINGERPRINT_LEN; i++) {
+                                       fp[i] = (hex2bin(search[2 + i * 2])
+                                                               << 4) +
+                                               hex2bin(search[3 + i * 2]);
                                }
+                               isfp = true;
                        } else if (search != NULL) {
                                keyid = strtoull(search, &end, 16);
                                if (*search != 0 &&
@@ -193,19 +212,22 @@ int main(int argc, char *argv[])
                readconfig(NULL);
                initlogthing("lookup", config.logfile);
                catchsignals();
-               config.dbbackend->initdb(false);
+               dbctx = config.dbinit(false);
                switch (op) {
                case OP_GET:
                case OP_HGET:
                        if (op == OP_HGET) {
                                parse_skshash(search, &hash);
-                               result = config.dbbackend->fetch_key_skshash(
+                               result = dbctx->fetch_key_skshash(dbctx,
                                        &hash, &publickey);
                        } else if (ishex) {
-                               result = config.dbbackend->fetch_key(keyid,
+                               result = dbctx->fetch_key_id(dbctx, keyid,
                                        &publickey, false);
+                       } else if (isfp) {
+                               result = dbctx->fetch_key_fp(dbctx, fp,
+                                       MAX_FINGERPRINT_LEN, &publickey, false);
                        } else {
-                               result = config.dbbackend->fetch_key_text(
+                               result = dbctx->fetch_key_text(dbctx,
                                        search,
                                        &publickey);
                        }
@@ -231,21 +253,30 @@ int main(int argc, char *argv[])
                        }
                        break;
                case OP_INDEX:
-                       find_keys(search, keyid, ishex, fingerprint, skshash,
+                       find_keys(dbctx, search, keyid, fp, MAX_FINGERPRINT_LEN,
+                                       ishex, isfp, fingerprint, skshash,
                                        exact, false, mrhkp);
                        break;
                case OP_VINDEX:
-                       find_keys(search, keyid, ishex, fingerprint, skshash,
+                       find_keys(dbctx, search, keyid, fp, MAX_FINGERPRINT_LEN,
+                                       ishex, isfp, fingerprint, skshash,
                                        exact, true, mrhkp);
                        break;
                case OP_PHOTO:
-                       if (config.dbbackend->fetch_key(keyid, &publickey,
-                                       false)) {
+                       if (isfp) {
+                               dbctx->fetch_key_fp(dbctx, fp,
+                                       MAX_FINGERPRINT_LEN,
+                                       &publickey, false);
+                       } else {
+                               dbctx->fetch_key_id(dbctx, keyid,
+                                       &publickey, false);
+                       }
+                       if (publickey != NULL) {
                                unsigned char *photo = NULL;
                                size_t         length = 0;
 
                                if (getphoto(publickey, indx, &photo,
-                                               &length)) {
+                                               &length) == ONAK_E_OK) {
                                        fwrite(photo,
                                                        1,
                                                        length,
@@ -258,7 +289,7 @@ int main(int argc, char *argv[])
                default:
                        puts("Unknown operation!");
                }
-               config.dbbackend->cleanupdb();
+               dbctx->cleanupdb(dbctx);
                cleanuplogthing();
                cleanupconfig();
        }