X-Git-Url: https://git.sommitrealweird.co.uk/onak.git/blobdiff_plain/870ceebb3ffd90a47102ff04c8f7779177573216..8e454e9e224e84113a93ad22103613be4ff72b3f:/onak.c?ds=inline

diff --git a/onak.c b/onak.c
index 87b65e8..311b44f 100644
--- a/onak.c
+++ b/onak.c
@@ -18,6 +18,7 @@
 #include "keyid.h"
 #include "keyindex.h"
 #include "keystructs.h"
+#include "log.h"
 #include "mem.h"
 #include "merge.h"
 #include "onak-conf.h"
@@ -74,7 +75,8 @@ void usage(void) {
 	puts("Usage:\n");
 	puts("\tonak [options] <command> <parameters>\n");
 	puts("\tCommands:\n");
-	puts("\tadd    - read armored OpenPGP keys from stdin and add to the keyserver");
+	puts("\tadd    - read armored OpenPGP keys from stdin and add to the"
+		" keyserver");
 	puts("\tdelete - delete a given key from the keyserver");
 	puts("\tget    - retrieves the key requested from the keyserver");
 	puts("\tindex  - search for a key and list it");
@@ -87,49 +89,77 @@ int main(int argc, char *argv[])
 	struct openpgp_packet_list	*list_end = NULL;
 	struct openpgp_publickey	*keys = NULL;
 	int				 rc = EXIT_SUCCESS;
+	int				 result = 0;
 	char				*search = NULL;
 	char				*end = NULL;
 	uint64_t			 keyid = 0;
 	bool				 ishex = false;
 	bool				 verbose = false;
+	bool				 update = false;
 	bool				 binary = false;
+	bool				 fingerprint = false;
 	int				 optchar;
 
-
-	while ((optchar = getopt(argc, argv, "bv")) != -1 ) {
+	while ((optchar = getopt(argc, argv, "bfuv")) != -1 ) {
 		switch (optchar) {
 		case 'b': 
 			binary = true;
 			break;
+		case 'f': 
+			fingerprint = true;
+			break;
+		case 'u': 
+			update = true;
+			break;
 		case 'v': 
 			verbose = true;
+			setlogthreshold(LOGTHING_INFO);
 			break;
 		}
 	}
 
+	readconfig();
+	initlogthing("onak", config.logfile);
+
 	if ((argc - optind) < 1) {
 		usage();
+	} else if (!strcmp("dump", argv[optind])) {
+		initdb();
+		dumpdb("keydump");
+		cleanupdb();
 	} else if (!strcmp("add", argv[optind])) {
 		if (binary) {
-			read_openpgp_stream(stdin_getchar, NULL, &packets);
+			result = read_openpgp_stream(stdin_getchar, NULL,
+				 &packets);
+			logthing(LOGTHING_INFO,
+					"read_openpgp_stream: %d", result);
 		} else {
 			dearmor_openpgp_stream(stdin_getchar, NULL, &packets);
 		}
 		if (packets != NULL) {
-			parse_keys(packets, &keys);
+			result = parse_keys(packets, &keys);
 			free_packet_list(packets);
 			packets = NULL;
-			if (verbose) {
-				fprintf(stderr, "Finished reading keys.\n");
-			}
-	
+			logthing(LOGTHING_INFO, "Finished reading %d keys.",
+					result);
+
 			initdb();
-			fprintf(stderr, "Got %d new keys.\n",
-					update_keys(&keys, verbose));
+			logthing(LOGTHING_NOTICE, "Got %d new keys.",
+					update_keys(&keys));
+			if (keys != NULL && update) {
+				flatten_publickey(keys,
+					&packets,
+					&list_end);
+				armor_openpgp_stream(stdout_putchar,
+					NULL,
+					packets);
+				free_packet_list(packets);
+				packets = NULL;
+			}
 			cleanupdb();
 		} else {
 			rc = 1;
-			fprintf(stderr, "No keys read.\n");
+			logthing(LOGTHING_NOTICE, "No keys read.");
 		}
 
 		if (keys != NULL) {
@@ -137,7 +167,7 @@ int main(int argc, char *argv[])
 			keys = NULL;
 		} else {
 			rc = 1;
-			fprintf(stderr, "No changes.\n");
+			logthing(LOGTHING_NOTICE, "No changes.");
 		}
 	} else if ((argc - optind) == 2) {
 		search = argv[optind+1];
@@ -151,22 +181,24 @@ int main(int argc, char *argv[])
 		}
 		initdb();
 		if (!strcmp("index", argv[optind])) {
-			find_keys(search, keyid, ishex, false, false, false);
+			find_keys(search, keyid, ishex, fingerprint,
+					false, false);
 		} else if (!strcmp("vindex", argv[optind])) {
-			find_keys(search, keyid, ishex, false, false, true);
+			find_keys(search, keyid, ishex, fingerprint,
+					false, true);
 		} else if (!strcmp("delete", argv[optind])) {
 			delete_key(getfullkeyid(keyid), false);
 		} else if (!strcmp("get", argv[optind])) {
 			if (fetch_key(keyid, &keys, false)) {
-				if (verbose) {
-					fprintf(stderr, "Got key.\n");
-				}
+				logthing(LOGTHING_INFO, "Got key.");
 				flatten_publickey(keys,
 						&packets,
 						&list_end);
 				armor_openpgp_stream(stdout_putchar,
 						NULL,
 						packets);
+				free_packet_list(packets);
+				packets = NULL;
 			} else {
 				puts("Key not found");
 			}
@@ -176,5 +208,8 @@ int main(int argc, char *argv[])
 		usage();
 	}
 
+	cleanuplogthing();
+	cleanupconfig();
+
 	return rc;
 }