ChangeLog:
[libgcrypt.git] / tests / prime.c
1 /* prime.c - part of the Libgcrypt test suite.
2    Copyright (C) 2001, 2002, 2003, 2005 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 #ifdef HAVE_CONFIG_H
20 #include <config.h>
21 #endif
22 #include <assert.h>
23 #include <stdio.h>
24 #include <string.h>
25 #include <stdlib.h>
26
27 #include "../src/gcrypt.h"
28
29 static int verbose;
30
31 static void
32 die (const char *format, ...)
33 {
34   va_list arg_ptr;
35
36   va_start (arg_ptr, format);
37   vfprintf (stderr, format, arg_ptr);
38   va_end (arg_ptr);
39   exit (1);
40 }
41
42 static void
43 check_primes (void)
44 {
45   gcry_error_t err = GPG_ERR_NO_ERROR;
46   gcry_mpi_t *factors = NULL;
47   gcry_mpi_t prime = NULL;
48   gcry_mpi_t g;
49   unsigned int i = 0;
50   struct prime_spec
51   {
52     unsigned int prime_bits;
53     unsigned int factor_bits;
54     unsigned int flags;
55   } prime_specs[] =
56     {
57       { 1024, 100, GCRY_PRIME_FLAG_SPECIAL_FACTOR },
58       { 128, 0, 0 },
59       { 0 },
60     };
61
62   for (i = 0; prime_specs[i].prime_bits; i++)
63     {
64       err = gcry_prime_generate (&prime,
65                                  prime_specs[i].prime_bits,
66                                  prime_specs[i].factor_bits,
67                                  &factors,
68                                  NULL, NULL,
69                                  GCRY_WEAK_RANDOM,
70                                  prime_specs[i].flags);
71       assert (! err);
72       if (verbose)
73         {
74           fprintf (stderr, "test %d: p = ", i);
75           gcry_mpi_dump (prime);
76           putc ('\n', stderr);
77         }
78
79       err = gcry_prime_check (prime, 0);
80       assert (! err);
81
82       err = gcry_prime_group_generator (&g, prime, factors, NULL);
83       assert (!err);
84       gcry_prime_release_factors (factors); factors = NULL;
85       
86       if (verbose)
87         {
88           fprintf (stderr, "     %d: g = ", i);
89           gcry_mpi_dump (g);
90           putc ('\n', stderr);
91         }
92       gcry_mpi_release (g);
93
94
95       gcry_mpi_add_ui (prime, prime, 1);
96       err = gcry_prime_check (prime, 0);
97       assert (err);
98     }
99 }
100
101 int
102 main (int argc, char **argv)
103 {
104   int debug = 0;
105
106   if ((argc > 1) && (! strcmp (argv[1], "--verbose")))
107     verbose = 1;
108   else if ((argc > 1) && (! strcmp (argv[1], "--debug")))
109     verbose = debug = 1;
110
111   gcry_control (GCRYCTL_DISABLE_SECMEM, 0);
112   if (! gcry_check_version (GCRYPT_VERSION))
113     die ("version mismatch\n");
114
115   gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
116   if (debug)
117     gcry_control (GCRYCTL_SET_DEBUG_FLAGS, 1u, 0);
118
119   check_primes ();
120
121   return 0;
122 }