gpgme/gpgme/encrypt.c

/* encrypt.c -  encrypt functions
 *	Copyright (C) 2000 Werner Koch (dd9jn)
 *      Copyright (C) 2001, 2002 g10 Code GmbH
 *
 * This file is part of GPGME.
 *
 * GPGME is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * GPGME is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
 */

#include <config.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>

#include "util.h"
#include "context.h"
#include "ops.h"
#include "wait.h"

#define SKIP_TOKEN_OR_RETURN(a) do { \
    while (*(a) && *(a) != ' ') (a)++; \
    while (*(a) == ' ') (a)++; \
    if (!*(a)) \
        return; /* oops */ \
} while (0)

struct encrypt_result_s
{
  int no_valid_recipients;
  int invalid_recipients;
  GpgmeData xmlinfo;
};

void
_gpgme_release_encrypt_result (EncryptResult result)
{
  if (!result)
    return;
  gpgme_data_release (result->xmlinfo);
  xfree (result);
}

/* 
 * Parse the args and save the information 
 * in an XML structure.
 * With args of NULL the xml structure is closed.
 */
static void
append_xml_encinfo (GpgmeData *rdh, char *args)
{
  GpgmeData dh;
  char helpbuf[100];

  if (!*rdh)
    {
      if (gpgme_data_new (rdh))
	return; /* FIXME: We are ignoring out-of-core.  */
      dh = *rdh;
      _gpgme_data_append_string (dh, "<GnupgOperationInfo>\n");
    }
  else
    {
      dh = *rdh;
      _gpgme_data_append_string (dh, "  </encryption>\n");
    }

  if (!args)
    {
      /* Just close the XML containter.  */
      _gpgme_data_append_string (dh, "</GnupgOperationInfo>\n");
      return;
    }

  _gpgme_data_append_string (dh, "  <encryption>\n"
			     "    <error>\n"
			     "      <invalidRecipient/>\n");
    
  sprintf (helpbuf, "      <reason>%d</reason>\n", atoi (args));
  _gpgme_data_append_string (dh, helpbuf);
  SKIP_TOKEN_OR_RETURN (args);

  _gpgme_data_append_string (dh, "      <name>");
  _gpgme_data_append_percentstring_for_xml (dh, args);
  _gpgme_data_append_string (dh, "</name>\n"
			     "    </error>\n");
}


static void
status_handler_finish (GpgmeCtx ctx)
{
  if (ctx->result.encrypt->xmlinfo)
    {
      append_xml_encinfo (&ctx->result.encrypt->xmlinfo, NULL);
      _gpgme_set_op_info (ctx, ctx->result.encrypt->xmlinfo);
      ctx->result.encrypt->xmlinfo = NULL;
    }
  if (ctx->error)
    ; /* already set by kludge in engine-gpgsm */
  else if (ctx->result.encrypt->no_valid_recipients) 
    ctx->error = mk_error (No_Recipients);
  else if (ctx->result.encrypt->invalid_recipients) 
    ctx->error = mk_error (Invalid_Recipients);
}

void
_gpgme_encrypt_status_handler (GpgmeCtx ctx, GpgmeStatusCode code, char *args)
{
  if (ctx->error)
    {
      if (ctx->result.encrypt) /* check that we have allocated it. */
        status_handler_finish (ctx);
      return;
    }
  test_and_allocate_result (ctx, encrypt);

  switch (code)
    {
    case GPGME_STATUS_EOF:
      status_handler_finish (ctx);
      break;

    case GPGME_STATUS_INV_RECP:
      ctx->result.encrypt->invalid_recipients++;
      append_xml_encinfo (&ctx->result.encrypt->xmlinfo, args);
      break;

    case GPGME_STATUS_NO_RECP:
      ctx->result.encrypt->no_valid_recipients = 1;
      break;

    default:
      break;
    }
}


void
_gpgme_encrypt_sym_status_handler (GpgmeCtx ctx, GpgmeStatusCode code, char *args)
{
  _gpgme_passphrase_status_handler (ctx, code, args);
}


static GpgmeError
_gpgme_op_encrypt_start (GpgmeCtx ctx, int synchronous,
			 GpgmeRecipients recp, GpgmeData plain, GpgmeData ciph)
{
  GpgmeError err = 0;
  int symmetric = 0;

  /* Do some checks.  */
  if (!recp)
    symmetric = 1;
  else if (!gpgme_recipients_count (recp))
    {
      err = mk_error (No_Recipients);
      goto leave;
    }

  err = _gpgme_op_reset (ctx, synchronous);
  if (err)
    goto leave;

  if (symmetric)
    {
      err = _gpgme_passphrase_start (ctx);
      if (err)
	goto leave;
    }

  _gpgme_engine_set_status_handler (ctx->engine,
				    symmetric
				    ? _gpgme_encrypt_sym_status_handler
				    : _gpgme_encrypt_status_handler,
				    ctx);
  _gpgme_engine_set_verbosity (ctx->engine, ctx->verbosity);

  /* Check the supplied data */
  if (gpgme_data_get_type (plain) == GPGME_DATA_TYPE_NONE)
    {
      err = mk_error (No_Data);
      goto leave;
    }
  _gpgme_data_set_mode (plain, GPGME_DATA_MODE_OUT);
  if (!ciph || gpgme_data_get_type (ciph) != GPGME_DATA_TYPE_NONE)
    {
      err = mk_error (Invalid_Value);
      goto leave;
    }
  _gpgme_data_set_mode (ciph, GPGME_DATA_MODE_IN);

  err = _gpgme_engine_op_encrypt (ctx->engine, recp, plain, ciph, ctx->use_armor);


  if (!err)	/* And kick off the process.  */
    err = _gpgme_engine_start (ctx->engine, ctx);

 leave:
  if (err)
    {
      ctx->pending = 0; 
      _gpgme_engine_release (ctx->engine);
      ctx->engine = NULL;
    }
  return err;
}


GpgmeError
gpgme_op_encrypt_start (GpgmeCtx ctx, GpgmeRecipients recp, GpgmeData plain,
			GpgmeData ciph)
{
  return _gpgme_op_encrypt_start (ctx, 0, recp, plain, ciph);
}


/**
 * gpgme_op_encrypt:
 * @c: The context
 * @recp: A set of recipients 
 * @in: plaintext input
 * @out: ciphertext output
 * 
 * This function encrypts @in to @out for all recipients from
 * @recp.  Other parameters are take from the context @c.
 * The function does wait for the result.
 * 
 * Return value:  0 on success or an errorcode. 
 **/
GpgmeError
gpgme_op_encrypt (GpgmeCtx ctx, GpgmeRecipients recp,
		  GpgmeData plain, GpgmeData cipher)
{
  int err = _gpgme_op_encrypt_start (ctx, 1, recp, plain, cipher);
  if (!err)
    {
      err = _gpgme_wait_one (ctx);
      /* Old gpg versions don't return status info for invalid
	 recipients, so we simply check whether we got any output at
	 all, and if not we assume that we don't have valid
	 recipients.  */
      if (!ctx->error && gpgme_data_get_type (cipher) == GPGME_DATA_TYPE_NONE)
	ctx->error = mk_error (No_Recipients);
      err = ctx->error;
    }
  return err;
}