egg/tests/test-dh.c

   1 /* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */
   2 /* test-dh.c: Test egg-dh.c
   3
   4    Copyright (C) 2009 Stefan Walter
   5
   6    The Gnome Keyring Library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Library General Public License as
   8    published by the Free Software Foundation; either version 2 of the
   9    License, or (at your option) any later version.
  10
  11    The Gnome Keyring Library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Library General Public License for more details.
  15
  16    You should have received a copy of the GNU Library General Public
  17    License along with the Gnome Library; see the file COPYING.LIB.  If not,
  18    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19    Boston, MA 02111-1307, USA.
  20
  21    Author: Stef Walter <stef@memberwebs.com>
  22 */
  23
  24 #include "config.h"
  25
  26 #include "egg/egg-dh.h"
  27 #include "egg/egg-secure-memory.h"
  28 #include "egg/egg-testing.h"
  29
  30 #include <stdlib.h>
  31 #include <stdio.h>
  32 #include <string.h>
  33
  34 #include <glib.h>
  35 #include <gcrypt.h>
  36
  37 EGG_SECURE_DEFINE_GLIB_GLOBALS ();
  38
  39 static void
  40 test_perform (void)
  41 {
  42         gcry_mpi_t p, g;
  43         gcry_mpi_t x1, X1;
  44         gcry_mpi_t x2, X2;
  45         gpointer k1, k2;
  46         gboolean ret;
  47         gsize n1, n2;
  48
  49         /* Load up the parameters */
  50         if (!egg_dh_default_params ("ietf-ike-grp-modp-768", &p, &g))
  51                 g_assert_not_reached ();
  52
  53         /* Generate secrets */
  54         ret = egg_dh_gen_pair (p, g, 0, &X1, &x1);
  55         g_assert (ret);
  56         ret = egg_dh_gen_pair (p, g, 0, &X2, &x2);
  57         g_assert (ret);
  58
  59         /* Calculate keys */
  60         k1 = egg_dh_gen_secret (X2, x1, p, &n1);
  61         g_assert (k1);
  62         k2 = egg_dh_gen_secret (X1, x2, p, &n2);
  63         g_assert (k2);
  64
  65         /* Keys must be the same */
  66         egg_assert_cmpsize (n1, ==, n2);
  67         g_assert (memcmp (k1, k2, n1) == 0);
  68
  69         gcry_mpi_release (p);
  70         gcry_mpi_release (g);
  71         gcry_mpi_release (x1);
  72         gcry_mpi_release (X1);
  73         egg_secure_free (k1);
  74         gcry_mpi_release (x2);
  75         gcry_mpi_release (X2);
  76         egg_secure_free (k2);
  77 }
  78
  79 static void
  80 test_short_pair (void)
  81 {
  82         gcry_mpi_t p, g;
  83         gcry_mpi_t x1, X1;
  84         gboolean ret;
  85
  86         /* Load up the parameters */
  87         ret = egg_dh_default_params ("ietf-ike-grp-modp-1024", &p, &g);
  88         g_assert (ret);
  89         g_assert_cmpuint (gcry_mpi_get_nbits (p), ==, 1024);
  90
  91         /* Generate secrets */
  92         ret = egg_dh_gen_pair (p, g, 512, &X1, &x1);
  93         g_assert (ret);
  94         g_assert_cmpuint (gcry_mpi_get_nbits (x1), <=, 512);
  95
  96         gcry_mpi_release (p);
  97         gcry_mpi_release (g);
  98         gcry_mpi_release (x1);
  99         gcry_mpi_release (X1);
 100 }
 101
 102 static void
 103 check_dh_default (const gchar *name, guint bits)
 104 {
 105         gboolean ret;
 106         gcry_mpi_t p, g, check;
 107         gconstpointer prime, base;
 108         gsize n_prime, n_base;
 109         gcry_error_t gcry;
 110
 111         ret = egg_dh_default_params (name, &p, &g);
 112         g_assert (ret);
 113         g_assert_cmpint (gcry_mpi_get_nbits (p), ==, bits);
 114         g_assert_cmpint (gcry_mpi_get_nbits (g), <, gcry_mpi_get_nbits (p));
 115
 116         ret = egg_dh_default_params_raw (name, &prime, &n_prime, &base, &n_base);
 117         g_assert (ret);
 118         g_assert (prime != NULL);
 119         egg_assert_cmpsize (n_prime, >, 0);
 120         g_assert (base != NULL);
 121         egg_assert_cmpsize (n_base, >, 0);
 122
 123         gcry = gcry_mpi_scan (&check, GCRYMPI_FMT_USG, prime, n_prime, NULL);
 124         g_assert (gcry == 0);
 125         g_assert (gcry_mpi_cmp (check, p) == 0);
 126         gcry_mpi_release (check);
 127
 128         gcry = gcry_mpi_scan (&check, GCRYMPI_FMT_USG, base, n_base, NULL);
 129         g_assert (gcry == 0);
 130         g_assert (gcry_mpi_cmp (check, g) == 0);
 131         gcry_mpi_release (check);
 132
 133         gcry_mpi_release (p);
 134         gcry_mpi_release (g);
 135 }
 136
 137 static void
 138 test_default_768 (void)
 139 {
 140         check_dh_default ("ietf-ike-grp-modp-768", 768);
 141 }
 142
 143 static void
 144 test_default_1024 (void)
 145 {
 146         check_dh_default ("ietf-ike-grp-modp-1024", 1024);
 147 }
 148
 149 static void
 150 test_default_1536 (void)
 151 {
 152         check_dh_default ("ietf-ike-grp-modp-1536", 1536);
 153 }
 154
 155 static void
 156 test_default_2048 (void)
 157 {
 158         check_dh_default ("ietf-ike-grp-modp-2048", 2048);
 159 }
 160
 161 static void
 162 test_default_3072 (void)
 163 {
 164         check_dh_default ("ietf-ike-grp-modp-3072", 3072);
 165 }
 166
 167 static void
 168 test_default_4096 (void)
 169 {
 170         check_dh_default ("ietf-ike-grp-modp-4096", 4096);
 171 }
 172
 173 static void
 174 test_default_8192 (void)
 175 {
 176         check_dh_default ("ietf-ike-grp-modp-8192", 8192);
 177 }
 178
 179 static void
 180 test_default_bad (void)
 181 {
 182         gboolean ret;
 183         gcry_mpi_t p, g;
 184
 185         ret = egg_dh_default_params ("bad-name", &p, &g);
 186         g_assert (!ret);
 187 }
 188
 189 int
 190 main (int argc, char **argv)
 191 {
 192         g_test_init (&argc, &argv, NULL);
 193
 194         if (!g_test_quick ()) {
 195                 g_test_add_func ("/dh/perform", test_perform);
 196                 g_test_add_func ("/dh/short_pair", test_short_pair);
 197         }
 198
 199         g_test_add_func ("/dh/default_768", test_default_768);
 200         g_test_add_func ("/dh/default_1024", test_default_1024);
 201         g_test_add_func ("/dh/default_1536", test_default_1536);
 202         g_test_add_func ("/dh/default_2048", test_default_2048);
 203         g_test_add_func ("/dh/default_3072", test_default_3072);
 204         g_test_add_func ("/dh/default_4096", test_default_4096);
 205         g_test_add_func ("/dh/default_8192", test_default_8192);
 206         g_test_add_func ("/dh/default_bad", test_default_bad);
 207
 208         return g_test_run ();
 209 }