]> the.earth.li Git - onak.git/blobdiff - lookup.c
Use dynamic context for all backend databases
[onak.git] / lookup.c
index 56394a5527de29890817d2b603602ef55e31a372..533433a3c1d66d9fee2e6221e095e5f1b61944ab 100644 (file)
--- a/lookup.c
+++ b/lookup.c
@@ -1,9 +1,20 @@
 /*
  * lookup.c - CGI to lookup keys.
  *
- * Jonathan McDowell <noodles@earth.li>
+ * Copyright 2002-2005,2007-2009,2011 Jonathan McDowell <noodles@earth.li>
  *
- * Copyright 2002 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, write to the Free Software Foundation, Inc., 51
+ * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  */
 
 #include <inttypes.h>
@@ -19,6 +30,7 @@
 #include "cleanup.h"
 #include "getcgi.h"
 #include "keydb.h"
+#include "keyid.h"
 #include "keyindex.h"
 #include "log.h"
 #include "mem.h"
 #define OP_INDEX   2
 #define OP_VINDEX  3
 #define OP_PHOTO   4
+#define OP_HGET    5
 
-void find_keys(char *search, uint64_t keyid, bool ishex,
-               bool fingerprint, 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, true);
+                       key_index(dbctx, publickey, verbose, fingerprint,
+                               skshash, true);
                }
                free_publickey(publickey);
        } else if (count == 0) {
@@ -71,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;
@@ -78,22 +111,29 @@ int main(int argc, char *argv[])
        int i;
        int indx = 0;
        bool fingerprint = false;
+       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;
        struct openpgp_packet_list *packets = NULL;
        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) {
                if (!strcmp(params[i], "op")) {
                        if (!strcmp(params[i+1], "get")) {
                                op = OP_GET;
+                       } else if (!strcmp(params[i+1], "hget")) {
+                               op = OP_HGET;
                        } else if (!strcmp(params[i+1], "index")) {
                                op = OP_INDEX;
                        } else if (!strcmp(params[i+1], "vindex")) {
@@ -104,7 +144,15 @@ int main(int argc, char *argv[])
                } else if (!strcmp(params[i], "search")) {
                        search = params[i+1];
                        params[i+1] = NULL;
-                       if (search != NULL) {
+                       if (search != NULL && strlen(search) == 42 &&
+                                       search[0] == '0' && search[1] == 'x') {
+                               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 &&
                                                end != NULL &&
@@ -118,6 +166,10 @@ int main(int argc, char *argv[])
                        if (!strcmp(params[i+1], "on")) {
                                fingerprint = true;
                        }
+               } else if (!strcmp(params[i], "hash")) {
+                       if (!strcmp(params[i+1], "on")) {
+                               skshash = true;
+                       }
                } else if (!strcmp(params[i], "exact")) {
                        if (!strcmp(params[i+1], "on")) {
                                exact = true;
@@ -160,14 +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:
-                       if (ishex) {
-                               result = config.dbbackend->fetch_key(keyid,
+               case OP_HGET:
+                       if (op == OP_HGET) {
+                               parse_skshash(search, &hash);
+                               result = dbctx->fetch_key_skshash(dbctx,
+                                       &hash, &publickey);
+                       } else if (ishex) {
+                               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);
                        }
@@ -193,21 +253,30 @@ int main(int argc, char *argv[])
                        }
                        break;
                case OP_INDEX:
-                       find_keys(search, keyid, ishex, fingerprint, exact,
-                                       false, mrhkp);
+                       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, exact,
-                                       true, mrhkp);
+                       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,
@@ -220,14 +289,13 @@ int main(int argc, char *argv[])
                default:
                        puts("Unknown operation!");
                }
-               config.dbbackend->cleanupdb();
+               dbctx->cleanupdb(dbctx);
                cleanuplogthing();
                cleanupconfig();
        }
        if (!mrhkp) {
                puts("<hr>");
-               puts("Produced by onak " ONAK_VERSION 
-                               " by Jonathan McDowell");
+               puts("Produced by onak " ONAK_VERSION );
                end_html();
        }