X-Git-Url: https://git.sommitrealweird.co.uk/onak.git/blobdiff_plain/34b03028378025ad22c8d29f70e81109cee690c2..6a721665f41611f80cb878237b5d7fa650ccbe20:/onak.c

diff --git a/onak.c b/onak.c
index 5c07a5b..870e1bd 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,8 +75,11 @@ 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("\tdump   - dump all the keys from the keyserver to a file or"
+		" files\n\t         starting keydump*");
 	puts("\tget    - retrieves the key requested from the keyserver");
 	puts("\tindex  - search for a key and list it");
 	puts("\tvindex - search for a key and list it and its signatures");
@@ -87,6 +91,7 @@ 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;
@@ -94,44 +99,55 @@ int main(int argc, char *argv[])
 	bool				 verbose = false;
 	bool				 update = false;
 	bool				 binary = false;
+	bool				 fingerprint = false;
 	int				 optchar;
 
-
-	while ((optchar = getopt(argc, argv, "buv")) != -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,
@@ -145,7 +161,7 @@ int main(int argc, char *argv[])
 			cleanupdb();
 		} else {
 			rc = 1;
-			fprintf(stderr, "No keys read.\n");
+			logthing(LOGTHING_NOTICE, "No keys read.");
 		}
 
 		if (keys != NULL) {
@@ -153,7 +169,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];
@@ -167,19 +183,20 @@ 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);
+				free_publickey(keys);
 				armor_openpgp_stream(stdout_putchar,
 						NULL,
 						packets);
@@ -194,6 +211,7 @@ int main(int argc, char *argv[])
 		usage();
 	}
 
+	cleanuplogthing();
 	cleanupconfig();
 
 	return rc;