Only seed database for Debian install if we're using default config

[onak.git] / lookup.c

diff --git a/lookup.c b/lookup.c
index d5392d599d318f806b059dd1e6fc043e0bfbed53..e670d498f0a38189d9fb347254f3a6d1f5fa5023 100644 (file)
--- a/lookup.c
+++ b/lookup.c
@@ -16,6 +16,8 @@
 #include "armor.h"
 #include "charfuncs.h"
 #include "cleankey.h"
+#include "cleanup.h"
+#include "config.h"
 #include "getcgi.h"
 #include "keydb.h"
 #include "keyindex.h"
@@ -38,9 +40,9 @@ void find_keys(char *search, uint64_t keyid, bool ishex,
        int count = 0;
 
        if (ishex) {
-               count = fetch_key(keyid, &publickey, false);
+               count = config.dbbackend->fetch_key(keyid, &publickey, false);
        } else {
-               count = fetch_key_text(search, &publickey);
+               count = config.dbbackend->fetch_key_text(search, &publickey);
        }
        if (publickey != NULL) {
                if (mrhkp) {
@@ -85,6 +87,7 @@ int main(int argc, char *argv[])
        struct openpgp_publickey *publickey = NULL;
        struct openpgp_packet_list *packets = NULL;
        struct openpgp_packet_list *list_end = NULL;
+       int result;
 
        params = getcgivars(argc, argv);
        for (i = 0; params != NULL && params[i] != NULL; i += 2) {
@@ -102,7 +105,7 @@ int main(int argc, char *argv[])
                        search = params[i+1];
                        params[i+1] = NULL;
                        if (search != NULL) {
-                               keyid = strtoul(search, &end, 16);
+                               keyid = strtoull(search, &end, 16);
                                if (*search != 0 &&
                                                end != NULL &&
                                                *end == 0) {
@@ -156,12 +159,25 @@ int main(int argc, char *argv[])
        } else {
                readconfig(NULL);
                initlogthing("lookup", config.logfile);
-               initdb(true);
+               catchsignals();
+               config.dbbackend->initdb(false);
                switch (op) {
                case OP_GET:
-                       logthing(LOGTHING_NOTICE, "Getting keyid 0x%llX",
+                       if (ishex) {
+                               logthing(LOGTHING_NOTICE, 
+                                       "Getting keyid 0x%llX",
                                        keyid);
-                       if (fetch_key(keyid, &publickey, false)) {
+                               result = config.dbbackend->fetch_key(keyid,
+                                       &publickey, false);
+                       } else {
+                               logthing(LOGTHING_NOTICE, 
+                                       "Getting key(s) for search text %s",
+                                       search);
+                               result = config.dbbackend->fetch_key_text(
+                                       search,
+                                       &publickey);
+                       }
+                       if (result) {
                                puts("<pre>");
                                cleankeys(publickey);
                                flatten_publickey(publickey,
@@ -186,11 +202,13 @@ int main(int argc, char *argv[])
                                        true, mrhkp);
                        break;
                case OP_PHOTO:
-                       if (fetch_key(keyid, &publickey, false)) {
+                       if (config.dbbackend->fetch_key(keyid, &publickey,
+                                       false)) {
                                unsigned char *photo = NULL;
                                size_t         length = 0;
 
-                               if (getphoto(publickey, 0, &photo, &length)) {
+                               if (getphoto(publickey, indx, &photo,
+                                               &length)) {
                                        fwrite(photo,
                                                        1,
                                                        length,
@@ -203,13 +221,14 @@ int main(int argc, char *argv[])
                default:
                        puts("Unknown operation!");
                }
-               cleanupdb();
+               config.dbbackend->cleanupdb();
                cleanuplogthing();
                cleanupconfig();
        }
        if (!mrhkp) {
                puts("<hr>");
-               puts("Produced by onak " VERSION " by Jonathan McDowell");
+               puts("Produced by onak " PACKAGE_VERSION 
+                               " by Jonathan McDowell");
                end_html();
        }