X-Git-Url: https://git.sommitrealweird.co.uk/onak.git/blobdiff_plain/e02c731dfbb288c736f2cd09a9b6df0507c59ddd..edae27083dd4a70b1b7b8c968949ffb7dc11c508:/add.c diff --git a/add.c b/add.c index 3af9309..3a9161f 100644 --- a/add.c +++ b/add.c @@ -12,38 +12,26 @@ #include #include "armor.h" +#include "cleankey.h" +#include "cleanup.h" +#include "charfuncs.h" #include "getcgi.h" #include "keydb.h" #include "keystructs.h" -#include "parsekey.h" +#include "log.h" +#include "mem.h" #include "merge.h" - -struct cgi_get_ctx { - char *buffer; - int offset; -}; - - -int cgi_getchar(void *ctx, size_t count, unsigned char *c) -{ - struct cgi_get_ctx *buf = NULL; - - buf = (struct cgi_get_ctx *) ctx; - - while (count-- > 0 && *c != 0) { - *c = buf->buffer[buf->offset++]; - } - - return (*c == 0); -} +#include "onak-conf.h" +#include "parsekey.h" int main(int argc, char *argv[]) { - struct openpgp_packet_list *packets = NULL; - struct openpgp_publickey *keys = NULL; - char **params = NULL; - struct cgi_get_ctx ctx; - int i; + struct openpgp_packet_list *packets = NULL; + struct openpgp_publickey *keys = NULL; + char **params = NULL; + struct buffer_ctx ctx; + int count = 0; + int i; memset(&ctx, 0, sizeof(ctx)); @@ -51,27 +39,65 @@ int main(int argc, char *argv[]) for (i = 0; params != NULL && params[i] != NULL; i += 2) { if (!strcmp(params[i], "keytext")) { ctx.buffer = params[i+1]; + ctx.size = strlen(ctx.buffer); + } else { + free(params[i+1]); } + params[i+1] = NULL; + free(params[i]); + params[i] = NULL; + } + if (params != NULL) { + free(params); + params = NULL; } - puts("Content-Type: text/html\n"); - puts("onak : Add"); + start_html("onak : Add"); if (ctx.buffer == NULL) { puts("Error: No keytext to add supplied."); + end_html(); } else { - dearmor_openpgp_stream(cgi_getchar, + readconfig(NULL); + initlogthing("add", config.logfile); + dearmor_openpgp_stream(buffer_fetchchar, &ctx, &packets); if (packets != NULL) { - parse_keys(packets, &keys); - initdb(); - printf("Got %d new keys.\n", - update_keys(&keys, false)); - cleanupdb(); + count = parse_keys(packets, &keys); + logthing(LOGTHING_NOTICE, "Received %d keys.", + count); + printf("Storing %d keys.\n", count); + end_html(); + if (stdout != NULL && fileno(stdout) != -1) { + fclose(stdout); + } + if (stderr != NULL && stderr != stdout && + fileno(stderr) != -1) { + fclose(stderr); + } + catchsignals(); + config.dbbackend->initdb(false); + + count = cleankeys(keys); + logthing(LOGTHING_INFO, "%d keys cleaned.", + count); + + count = config.dbbackend->update_keys(&keys, true); + logthing(LOGTHING_NOTICE, "Got %d new keys.", + count); + + if (keys != NULL) { + free_publickey(keys); + keys = NULL; + } + + config.dbbackend->cleanupdb(); } else { puts("No OpenPGP packets found in input."); + end_html(); } + cleanuplogthing(); + cleanupconfig(); } - puts(""); return (EXIT_SUCCESS); }