2005-04-11 Moritz Schulte <moritz@g10code.com>
[libgcrypt.git] / tests / prime.c
1 /* prime.c - part of the Libgcrypt test suite.
2    Copyright (C) 2001, 2002, 2003 Free Software Foundation, Inc.
3
4    This program is free software; you can redistribute it and/or
5    modify it under the terms of the GNU General Public License as
6    published by the Free Software Foundation; either version 2 of the
7    License, or (at your option) any later version.
8
9    This program is distributed in the hope that it will be useful, but
10    WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12    General Public License for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program; if not, write to the Free Software
16    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
17    USA.  */
18
19 #include <assert.h>
20 #include <stdio.h>
21 #include <string.h>
22 #include <stdlib.h>
23
24 #include "../src/gcrypt.h"
25
26 static int verbose;
27
28 static void
29 die (const char *format, ...)
30 {
31   va_list arg_ptr;
32
33   va_start (arg_ptr, format);
34   vfprintf (stderr, format, arg_ptr);
35   va_end (arg_ptr);
36   exit (1);
37 }
38
39 static void
40 check_primes (void)
41 {
42   gcry_error_t err = GPG_ERR_NO_ERROR;
43   gcry_mpi_t *factors = NULL;
44   gcry_mpi_t prime = NULL;
45   gcry_mpi_t g;
46   unsigned int i = 0;
47   struct prime_spec
48   {
49     unsigned int prime_bits;
50     unsigned int factor_bits;
51     unsigned int flags;
52   } prime_specs[] =
53     {
54       { 1024, 100, GCRY_PRIME_FLAG_SPECIAL_FACTOR },
55       { 128, 0, 0 },
56       { 0 },
57     };
58
59   for (i = 0; prime_specs[i].prime_bits; i++)
60     {
61       err = gcry_prime_generate (&prime,
62                                  prime_specs[i].prime_bits,
63                                  prime_specs[i].factor_bits,
64                                  &factors,
65                                  NULL, NULL,
66                                  GCRY_WEAK_RANDOM,
67                                  prime_specs[i].flags);
68       assert (! err);
69       if (verbose)
70         {
71           fprintf (stderr, "test %d: p = ", i);
72           gcry_mpi_dump (prime);
73           putc ('\n', stderr);
74         }
75
76       err = gcry_prime_check (prime, 0);
77       assert (! err);
78
79       err = gcry_prime_group_generator (&g, prime, factors, NULL);
80       assert (!err);
81       gcry_prime_release_factors (factors); factors = NULL;
82       
83       if (verbose)
84         {
85           fprintf (stderr, "     %d: g = ", i);
86           gcry_mpi_dump (g);
87           putc ('\n', stderr);
88         }
89       gcry_mpi_release (g);
90
91
92       gcry_mpi_add_ui (prime, prime, 1);
93       err = gcry_prime_check (prime, 0);
94       assert (err);
95     }
96 }
97
98 int
99 main (int argc, char **argv)
100 {
101   int debug = 0;
102
103   if ((argc > 1) && (! strcmp (argv[1], "--verbose")))
104     verbose = 1;
105   else if ((argc > 1) && (! strcmp (argv[1], "--debug")))
106     verbose = debug = 1;
107
108   gcry_control (GCRYCTL_DISABLE_SECMEM, 0);
109   if (! gcry_check_version (GCRYPT_VERSION))
110     die ("version mismatch\n");
111
112   gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
113   if (debug)
114     gcry_control (GCRYCTL_SET_DEBUG_FLAGS, 1u, 0);
115
116   check_primes ();
117
118   return 0;
119 }