/* curves.c -  ECC curves regression tests

 *	Copyright (C) 2011 Free Software Foundation, Inc.

 *

 * This file is part of Libgcrypt.

 *

 * Libgcrypt 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.

 *

 * Libgcrypt 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

 */

#ifdef HAVE_CONFIG_H

#include <config.h>

#endif

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <stdarg.h>

#include "../src/gcrypt-int.h"

#define PGM "curves"

#include "t-common.h"

/* Number of curves defined in ../cipger/ecc.c */

#define N_CURVES 14

/* A real world sample public key.  */

static char const sample_key_1[] =

"(public-key\n"

" (ecdsa\n"

"  (p #00FFFFFFFF00000001000000000000000000000000FFFFFFFFFFFFFFFFFFFFFFFF#)\n"

"  (a #00FFFFFFFF00000001000000000000000000000000FFFFFFFFFFFFFFFFFFFFFFFC#)\n"

"  (b #5AC635D8AA3A93E7B3EBBD55769886BC651D06B0CC53B0F63BCE3C3E27D2604B#)\n"

"  (g #046B17D1F2E12C4247F8BCE6E563A440F277037D812DEB33A0F4A13945D898C296"

        "4FE342E2FE1A7F9B8EE7EB4A7C0F9E162BCE33576B315ECECBB6406837BF51F5#)\n"

"  (n #00FFFFFFFF00000000FFFFFFFFFFFFFFFFBCE6FAADA7179E84F3B9CAC2FC632551#)\n"

"  (h #000000000000000000000000000000000000000000000000000000000000000001#)\n"

"  (q #0442B927242237639A36CE9221B340DB1A9AB76DF2FE3E171277F6A4023DED146EE"

      "86525E38CCECFF3FB8D152CC6334F70D23A525175C1BCBDDE6E023B2228770E#)\n"

"  ))";

static char const sample_key_1_curve[] = "NIST P-256";

static unsigned int sample_key_1_nbits = 256;

static void

list_curves (void)

{

  int idx;

  const char *name;

  unsigned int nbits;

  for (idx=0; (name = gcry_pk_get_curve (NULL, idx, &nbits)); idx++)

    {

      if (verbose)

        printf ("%s - %u bits\n", name, nbits);

    }

  if (idx != N_CURVES)

    fail ("expected %d curves but got %d\n", N_CURVES, idx);

  if (gcry_pk_get_curve (NULL, -1, NULL))

    fail ("curve iteration failed\n");

}

static void

check_matching (void)

{

  gpg_error_t err;

  gcry_sexp_t key;

  const char *name;

  unsigned int nbits;

  err = gcry_sexp_new (&key, sample_key_1, 0, 1);

  if (err)

    die ("parsing s-expression string failed: %s\n", gpg_strerror (err));

  name = gcry_pk_get_curve (key, 0, &nbits);

  if (!name)

    fail ("curve name not found for sample_key_1\n");

  else if (strcmp (name, sample_key_1_curve))

    fail ("expected curve name %s but got %s for sample_key_1\n",

          sample_key_1_curve, name);

  else if (nbits != sample_key_1_nbits)

    fail ("expected curve size %u but got %u for sample_key_1\n",

          sample_key_1_nbits, nbits);

  gcry_sexp_release (key);

}

static void

check_get_params (void)

{

  gcry_sexp_t param;

  const char *name;

  param = gcry_pk_get_param (GCRY_PK_ECDSA, sample_key_1_curve);

  if (!param)

    fail ("error gerring parameters for `%s'\n", sample_key_1_curve);

  name = gcry_pk_get_curve (param, 0, NULL);

  if (!name)

    fail ("get_param: curve name not found for sample_key_1\n");

  else if (strcmp (name, sample_key_1_curve))

    fail ("get_param: expected curve name %s but got %s for sample_key_1\n",

          sample_key_1_curve, name);

  gcry_sexp_release (param);

}

int

main (int argc, char **argv)

{

  if (argc > 1 && !strcmp (argv[1], "--verbose"))

    verbose = 1;

  else if (argc > 1 && !strcmp (argv[1], "--debug"))

    verbose = debug = 1;

  if (!gcry_check_version (GCRYPT_VERSION))

    die ("version mismatch\n");

  xgcry_control (GCRYCTL_DISABLE_SECMEM, 0);

  xgcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);

  if (debug)

    xgcry_control (GCRYCTL_SET_DEBUG_FLAGS, 1u, 0);

  list_curves ();

  check_matching ();

  check_get_params ();

  return error_count ? 1 : 0;

}