cli: use new notmuch_crypto_get_context in mime-node.c

This has the affect of lazily creating the crypto contexts only when
needed.  This removes code duplication from notmuch-show and
notmuch-reply, and should speed up these functions considerably if the
crypto flags are provided but the messages don't have any
cryptographic parts.
This commit is contained in:
Jameson Graef Rollins 2012-05-26 11:45:46 -07:00 committed by David Bremner
parent b2c8fdee53
commit e04b18cf36
4 changed files with 16 additions and 49 deletions

View file

@ -150,6 +150,7 @@ _mime_node_create (mime_node_t *parent, GMimeObject *part)
{ {
mime_node_t *node = talloc_zero (parent, mime_node_t); mime_node_t *node = talloc_zero (parent, mime_node_t);
GError *err = NULL; GError *err = NULL;
notmuch_crypto_context_t *cryptoctx = NULL;
/* Set basic node properties */ /* Set basic node properties */
node->part = part; node->part = part;
@ -182,8 +183,15 @@ _mime_node_create (mime_node_t *parent, GMimeObject *part)
return NULL; return NULL;
} }
if ((GMIME_IS_MULTIPART_ENCRYPTED (part) && node->ctx->crypto->decrypt)
|| (GMIME_IS_MULTIPART_SIGNED (part) && node->ctx->crypto->verify)) {
GMimeContentType *content_type = g_mime_object_get_content_type (part);
const char *protocol = g_mime_content_type_get_parameter (content_type, "protocol");
cryptoctx = notmuch_crypto_get_context (node->ctx->crypto, protocol);
}
/* Handle PGP/MIME parts */ /* Handle PGP/MIME parts */
if (GMIME_IS_MULTIPART_ENCRYPTED (part) && node->ctx->crypto->decrypt) { if (GMIME_IS_MULTIPART_ENCRYPTED (part) && node->ctx->crypto->decrypt && cryptoctx) {
if (node->nchildren != 2) { if (node->nchildren != 2) {
/* this violates RFC 3156 section 4, so we won't bother with it. */ /* this violates RFC 3156 section 4, so we won't bother with it. */
fprintf (stderr, "Error: %d part(s) for a multipart/encrypted " fprintf (stderr, "Error: %d part(s) for a multipart/encrypted "
@ -196,10 +204,10 @@ _mime_node_create (mime_node_t *parent, GMimeObject *part)
#ifdef GMIME_ATLEAST_26 #ifdef GMIME_ATLEAST_26
GMimeDecryptResult *decrypt_result = NULL; GMimeDecryptResult *decrypt_result = NULL;
node->decrypted_child = g_mime_multipart_encrypted_decrypt node->decrypted_child = g_mime_multipart_encrypted_decrypt
(encrypteddata, node->ctx->crypto->gpgctx, &decrypt_result, &err); (encrypteddata, cryptoctx, &decrypt_result, &err);
#else #else
node->decrypted_child = g_mime_multipart_encrypted_decrypt node->decrypted_child = g_mime_multipart_encrypted_decrypt
(encrypteddata, node->ctx->crypto->gpgctx, &err); (encrypteddata, cryptoctx, &err);
#endif #endif
if (node->decrypted_child) { if (node->decrypted_child) {
node->decrypt_success = node->verify_attempted = TRUE; node->decrypt_success = node->verify_attempted = TRUE;
@ -217,7 +225,7 @@ _mime_node_create (mime_node_t *parent, GMimeObject *part)
(err ? err->message : "no error explanation given")); (err ? err->message : "no error explanation given"));
} }
} }
} else if (GMIME_IS_MULTIPART_SIGNED (part) && node->ctx->crypto->verify) { } else if (GMIME_IS_MULTIPART_SIGNED (part) && node->ctx->crypto->verify && cryptoctx) {
if (node->nchildren != 2) { if (node->nchildren != 2) {
/* this violates RFC 3156 section 5, so we won't bother with it. */ /* this violates RFC 3156 section 5, so we won't bother with it. */
fprintf (stderr, "Error: %d part(s) for a multipart/signed message " fprintf (stderr, "Error: %d part(s) for a multipart/signed message "
@ -226,7 +234,7 @@ _mime_node_create (mime_node_t *parent, GMimeObject *part)
} else { } else {
#ifdef GMIME_ATLEAST_26 #ifdef GMIME_ATLEAST_26
node->sig_list = g_mime_multipart_signed_verify node->sig_list = g_mime_multipart_signed_verify
(GMIME_MULTIPART_SIGNED (part), node->ctx->crypto->gpgctx, &err); (GMIME_MULTIPART_SIGNED (part), cryptoctx, &err);
node->verify_attempted = TRUE; node->verify_attempted = TRUE;
if (!node->sig_list) if (!node->sig_list)
@ -242,7 +250,7 @@ _mime_node_create (mime_node_t *parent, GMimeObject *part)
* In GMime 2.6, they're both non-const, so we'll be able * In GMime 2.6, they're both non-const, so we'll be able
* to clean up this asymmetry. */ * to clean up this asymmetry. */
GMimeSignatureValidity *sig_validity = g_mime_multipart_signed_verify GMimeSignatureValidity *sig_validity = g_mime_multipart_signed_verify
(GMIME_MULTIPART_SIGNED (part), node->ctx->crypto->gpgctx, &err); (GMIME_MULTIPART_SIGNED (part), cryptoctx, &err);
node->verify_attempted = TRUE; node->verify_attempted = TRUE;
node->sig_validity = sig_validity; node->sig_validity = sig_validity;
if (sig_validity) { if (sig_validity) {

View file

@ -353,7 +353,8 @@ struct mime_node {
/* Construct a new MIME node pointing to the root message part of /* Construct a new MIME node pointing to the root message part of
* message. If crypto->verify is true, signed child parts will be * message. If crypto->verify is true, signed child parts will be
* verified. If crypto->decrypt is true, encrypted child parts will be * verified. If crypto->decrypt is true, encrypted child parts will be
* decrypted. * decrypted. If crypto->gpgctx is NULL, it will be lazily
* initialized.
* *
* Return value: * Return value:
* *

View file

@ -741,25 +741,6 @@ notmuch_reply_command (void *ctx, int argc, char *argv[])
else else
reply_format_func = notmuch_reply_format_default; reply_format_func = notmuch_reply_format_default;
if (params.crypto.decrypt) {
#ifdef GMIME_ATLEAST_26
/* TODO: GMimePasswordRequestFunc */
params.crypto.gpgctx = g_mime_gpg_context_new (NULL, "gpg");
#else
GMimeSession* session = g_object_new (g_mime_session_get_type(), NULL);
params.crypto.gpgctx = g_mime_gpg_context_new (session, "gpg");
#endif
if (params.crypto.gpgctx) {
g_mime_gpg_context_set_always_trust ((GMimeGpgContext*) params.crypto.gpgctx, FALSE);
} else {
params.crypto.decrypt = FALSE;
fprintf (stderr, "Failed to construct gpg context.\n");
}
#ifndef GMIME_ATLEAST_26
g_object_unref (session);
#endif
}
config = notmuch_config_open (ctx, NULL, NULL); config = notmuch_config_open (ctx, NULL, NULL);
if (config == NULL) if (config == NULL)
return 1; return 1;

View file

@ -1056,29 +1056,6 @@ notmuch_show_command (void *ctx, unused (int argc), unused (char *argv[]))
break; break;
} }
if (params.crypto.decrypt || params.crypto.verify) {
#ifdef GMIME_ATLEAST_26
/* TODO: GMimePasswordRequestFunc */
params.crypto.gpgctx = g_mime_gpg_context_new (NULL, "gpg");
#else
GMimeSession* session = g_object_new (g_mime_session_get_type(), NULL);
params.crypto.gpgctx = g_mime_gpg_context_new (session, "gpg");
#endif
if (params.crypto.gpgctx) {
g_mime_gpg_context_set_always_trust ((GMimeGpgContext*) params.crypto.gpgctx, FALSE);
} else {
/* If we fail to create the gpgctx set the verify and
* decrypt flags to FALSE so we don't try to do any
* further verification or decryption */
params.crypto.verify = FALSE;
params.crypto.decrypt = FALSE;
fprintf (stderr, "Failed to construct gpg context.\n");
}
#ifndef GMIME_ATLEAST_26
g_object_unref (session);
#endif
}
config = notmuch_config_open (ctx, NULL, NULL); config = notmuch_config_open (ctx, NULL, NULL);
if (config == NULL) if (config == NULL)
return 1; return 1;