/* t-support.h - Helper routines for regression tests.

   Copyright (C) 2000 Werner Koch (dd9jn)

   Copyright (C) 2001, 2002, 2003, 2004 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 Lesser General Public License as

   published by the Free Software Foundation; either version 2.1 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

   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser 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 <unistd.h>

#include <errno.h>

#include <stdlib.h>

#include <locale.h>

#ifdef HAVE_W32_SYSTEM

#include <windows.h>

#endif

#include <gpgme.h>

#ifndef DIM

#define DIM(v)		     (sizeof(v)/sizeof((v)[0]))

#endif

#define fail_if_err(err)					\

  do								\

    {								\

      if (err)							\

        {							\

          fprintf (stderr, "%s:%d: %s: %s\n",			\

                   __FILE__, __LINE__, gpgme_strsource (err),	\

		   gpgme_strerror (err));			\

          exit (1);						\

        }							\

    }								\

  while (0)

#ifdef GPGRT_HAVE_MACRO_FUNCTION

void GPGRT_ATTR_NORETURN

_test (const char *expr, const char *file, int line,

       const char *func)

{

  fprintf (stderr, "Test \"%s\" in %s failed (%s:%d)\n",

           expr, func, file, line);

  exit (1);

}

# define test(expr)                                             \

  ((expr)                                                       \

   ? (void) 0                                                   \

   : _test (#expr, __FILE__, __LINE__, __FUNCTION__))

#else /*!GPGRT_HAVE_MACRO_FUNCTION*/

void

_test (const char *expr, const char *file, int line)

{

  fprintf (stderr, "Test \"%s\" failed (%s:%d)\n",

           expr, file, line);

  exit (1);

}

# define test(expr)                                             \

  ((expr)                                                       \

   ? (void) 0                                                   \

   : _test (#expr, __FILE__, __LINE__))

#endif /*!GPGRT_HAVE_MACRO_FUNCTION*/

static const char *

nonnull (const char *s)

{

  return s? s :"[none]";

}

void

print_data (gpgme_data_t dh)

{

#define BUF_SIZE 512

  char buf[BUF_SIZE + 1];

  int ret;

  ret = gpgme_data_seek (dh, 0, SEEK_SET);

  if (ret)

    fail_if_err (gpgme_err_code_from_errno (errno));

  while ((ret = gpgme_data_read (dh, buf, BUF_SIZE)) > 0)

    fwrite (buf, ret, 1, stdout);

  if (ret < 0)

    fail_if_err (gpgme_err_code_from_errno (errno));

}

gpgme_error_t

passphrase_cb (void *opaque, const char *uid_hint, const char *passphrase_info,

	       int last_was_bad, int fd)

{

  int res;

  char pass[] = "abc\n";

  int passlen = strlen (pass);

  int off = 0;

  (void)opaque;

  (void)uid_hint;

  (void)passphrase_info;

  (void)last_was_bad;

  do

    {

      res = gpgme_io_write (fd, &pass[off], passlen - off);

      if (res > 0)

	off += res;

    }

  while (res > 0 && off != passlen);

  return off == passlen ? 0 : gpgme_error_from_errno (errno);

}

char *

make_filename (const char *fname)

{

  const char *srcdir = getenv ("srcdir");

  char *buf;

  if (!srcdir)

    srcdir = ".";

  buf = malloc (strlen(srcdir) + strlen(fname) + 2);

  if (!buf)

    exit (8);

  strcpy (buf, srcdir);

  strcat (buf, "/");

  strcat (buf, fname);

  return buf;

}

void

init_gpgme (gpgme_protocol_t proto)

{

  gpgme_error_t err;

  gpgme_check_version (NULL);

  setlocale (LC_ALL, "");

  gpgme_set_locale (NULL, LC_CTYPE, setlocale (LC_CTYPE, NULL));

#ifndef HAVE_W32_SYSTEM

  gpgme_set_locale (NULL, LC_MESSAGES, setlocale (LC_MESSAGES, NULL));

#endif

  err = gpgme_engine_check_version (proto);

  fail_if_err (err);

}

void

print_import_result (gpgme_import_result_t r)

{

  gpgme_import_status_t st;

  for (st=r->imports; st; st = st->next)

    {

      printf ("  fpr: %s err: %d (%s) status:", nonnull (st->fpr),

              st->result, gpgme_strerror (st->result));

      if (st->status & GPGME_IMPORT_NEW)

        fputs (" new", stdout);

      if (st->status & GPGME_IMPORT_UID)

        fputs (" uid", stdout);

      if (st->status & GPGME_IMPORT_SIG)

        fputs (" sig", stdout);

      if (st->status & GPGME_IMPORT_SUBKEY)

        fputs (" subkey", stdout);

      if (st->status & GPGME_IMPORT_SECRET)

        fputs (" secret", stdout);

      putchar ('\n');

    }

  printf ("key import summary:\n"

          "        considered: %d\n"

          "        no user id: %d\n"

          "          imported: %d\n"

          "      imported_rsa: %d\n"

          "         unchanged: %d\n"

          "      new user ids: %d\n"

          "       new subkeys: %d\n"

          "    new signatures: %d\n"

          "   new revocations: %d\n"

          "       secret read: %d\n"

          "   secret imported: %d\n"

          "  secret unchanged: %d\n"

          "  skipped new keys: %d\n"

          "      not imported: %d\n",

          r->considered,

          r->no_user_id,

          r->imported,

          r->imported_rsa,

          r->unchanged,

          r->new_user_ids,

          r->new_sub_keys,

          r->new_signatures,

          r->new_revocations,

          r->secret_read,

          r->secret_imported,

          r->secret_unchanged,

          r->skipped_new_keys,

          r->not_imported);

}