"Recognize" some extra subpacket types

[onak.git] / lookup.c

diff --git a/lookup.c b/lookup.c
index 71c27c3357f32e27806fd0be668a8c5d244b7a3b..a7d9c7c4f1c7dc8baecc74446e39dfb855ef84a7 100644 (file)
--- a/lookup.c
+++ b/lookup.c
@@ -4,8 +4,6 @@
  * Jonathan McDowell <noodles@earth.li>
  *
  * Copyright 2002 Project Purple
- *
- * $Id: lookup.c,v 1.10 2003/06/07 13:45:34 noodles Exp $
  */
 
 #include <inttypes.h>
@@ -16,6 +14,10 @@
 #include <unistd.h>
 
 #include "armor.h"
+#include "charfuncs.h"
+#include "cleankey.h"
+#include "cleanup.h"
+#include "config.h"
 #include "getcgi.h"
 #include "keydb.h"
 #include "keyindex.h"
@@ -23,16 +25,13 @@
 #include "mem.h"
 #include "onak-conf.h"
 #include "parsekey.h"
+#include "photoid.h"
 
 #define OP_UNKNOWN 0
 #define OP_GET     1
 #define OP_INDEX   2
 #define OP_VINDEX  3
-
-int putnextchar(void *ctx, size_t count, unsigned char *c)
-{
-       return printf("%.*s", (int) count, c);
-}
+#define OP_PHOTO   4
 
 void find_keys(char *search, uint64_t keyid, bool ishex,
                bool fingerprint, bool exact, bool verbose, bool mrhkp)
@@ -41,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) {
@@ -77,6 +76,7 @@ int main(int argc, char *argv[])
        char **params = NULL;
        int op = OP_UNKNOWN;
        int i;
+       int indx = 0;
        bool fingerprint = false;
        bool exact = false;
        bool ishex = false;
@@ -87,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) {
@@ -97,18 +98,22 @@ int main(int argc, char *argv[])
                                op = OP_INDEX;
                        } else if (!strcmp(params[i+1], "vindex")) {
                                op = OP_VINDEX;
+                       } else if (!strcmp(params[i+1], "photo")) {
+                               op = OP_PHOTO;
                        }
                } else if (!strcmp(params[i], "search")) {
                        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) {
                                        ishex = true;
                                }
                        }
+               } else if (!strcmp(params[i], "idx")) {
+                       indx = atoi(params[i+1]);
                } else if (!strcmp(params[i], "fingerprint")) {
                        if (!strcmp(params[i+1], "on")) {
                                fingerprint = true;
@@ -141,6 +146,8 @@ int main(int argc, char *argv[])
 
        if (mrhkp) {
                puts("Content-Type: text/plain\n");
+       } else if (op == OP_PHOTO) {
+               puts("Content-Type: image/jpeg\n");
        } else {
                start_html("Lookup of key");
        }
@@ -150,21 +157,39 @@ int main(int argc, char *argv[])
        } else if (search == NULL) {
                puts("Error: No key to search for supplied.");
        } else {
-               readconfig();
+               readconfig(NULL);
                initlogthing("lookup", config.logfile);
-               initdb();
+               catchsignals();
+               config.dbbackend->initdb(true);
                switch (op) {
                case OP_GET:
-                       if (fetch_key(keyid, &publickey, false)) {
+                       if (ishex) {
+                               logthing(LOGTHING_NOTICE, 
+                                       "Getting keyid 0x%llX",
+                                       keyid);
+                               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,
                                                        &packets,
                                                        &list_end);
-                               armor_openpgp_stream(putnextchar,
+                               armor_openpgp_stream(stdout_putchar,
                                                NULL,
                                                packets);
                                puts("</pre>");
                        } else {
+                               logthing(LOGTHING_NOTICE,
+                                       "Failed to fetch key.");
                                puts("Key not found");
                        }
                        break;
@@ -176,16 +201,34 @@ int main(int argc, char *argv[])
                        find_keys(search, keyid, ishex, fingerprint, exact,
                                        true, mrhkp);
                        break;
+               case OP_PHOTO:
+                       if (config.dbbackend->fetch_key(keyid, &publickey,
+                                       false)) {
+                               unsigned char *photo = NULL;
+                               size_t         length = 0;
+
+                               if (getphoto(publickey, indx, &photo,
+                                               &length)) {
+                                       fwrite(photo,
+                                                       1,
+                                                       length,
+                                                       stdout);
+                               }
+                               free_publickey(publickey);
+                               publickey = NULL;
+                       }
+                       break;
                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();
        }