comparison libtommath/bn_mp_prime_random_ex.c @ 284:eed26cff980b

propagate from branch 'au.asn.ucc.matt.ltm.dropbear' (head 6c790cad5a7fa866ad062cb3a0c279f7ba788583) to branch 'au.asn.ucc.matt.dropbear' (head fff0894a0399405a9410ea1c6d118f342cf2aa64)
author Matt Johnston <matt@ucc.asn.au>
date Wed, 08 Mar 2006 13:23:49 +0000
parents
children 5ff8218bcee9
comparison
equal deleted inserted replaced
283:bd240aa12ba7 284:eed26cff980b
1 #include <tommath.h>
2 #ifdef BN_MP_PRIME_RANDOM_EX_C
3 /* LibTomMath, multiple-precision integer library -- Tom St Denis
4 *
5 * LibTomMath is a library that provides multiple-precision
6 * integer arithmetic as well as number theoretic functionality.
7 *
8 * The library was designed directly after the MPI library by
9 * Michael Fromberger but has been written from scratch with
10 * additional optimizations in place.
11 *
12 * The library is free for all purposes without any express
13 * guarantee it works.
14 *
15 * Tom St Denis, [email protected], http://math.libtomcrypt.org
16 */
17
18 /* makes a truly random prime of a given size (bits),
19 *
20 * Flags are as follows:
21 *
22 * LTM_PRIME_BBS - make prime congruent to 3 mod 4
23 * LTM_PRIME_SAFE - make sure (p-1)/2 is prime as well (implies LTM_PRIME_BBS)
24 * LTM_PRIME_2MSB_OFF - make the 2nd highest bit zero
25 * LTM_PRIME_2MSB_ON - make the 2nd highest bit one
26 *
27 * You have to supply a callback which fills in a buffer with random bytes. "dat" is a parameter you can
28 * have passed to the callback (e.g. a state or something). This function doesn't use "dat" itself
29 * so it can be NULL
30 *
31 */
32
33 /* This is possibly the mother of all prime generation functions, muahahahahaha! */
34 int mp_prime_random_ex(mp_int *a, int t, int size, int flags, ltm_prime_callback cb, void *dat)
35 {
36 unsigned char *tmp, maskAND, maskOR_msb, maskOR_lsb;
37 int res, err, bsize, maskOR_msb_offset;
38
39 /* sanity check the input */
40 if (size <= 1 || t <= 0) {
41 return MP_VAL;
42 }
43
44 /* LTM_PRIME_SAFE implies LTM_PRIME_BBS */
45 if (flags & LTM_PRIME_SAFE) {
46 flags |= LTM_PRIME_BBS;
47 }
48
49 /* calc the byte size */
50 bsize = (size>>3) + ((size&7)?1:0);
51
52 /* we need a buffer of bsize bytes */
53 tmp = OPT_CAST(unsigned char) XMALLOC(bsize);
54 if (tmp == NULL) {
55 return MP_MEM;
56 }
57
58 /* calc the maskAND value for the MSbyte*/
59 maskAND = ((size&7) == 0) ? 0xFF : (0xFF >> (8 - (size & 7)));
60
61 /* calc the maskOR_msb */
62 maskOR_msb = 0;
63 maskOR_msb_offset = ((size & 7) == 1) ? 1 : 0;
64 if (flags & LTM_PRIME_2MSB_ON) {
65 maskOR_msb |= 1 << ((size - 2) & 7);
66 } else if (flags & LTM_PRIME_2MSB_OFF) {
67 maskAND &= ~(1 << ((size - 2) & 7));
68 }
69
70 /* get the maskOR_lsb */
71 maskOR_lsb = 1;
72 if (flags & LTM_PRIME_BBS) {
73 maskOR_lsb |= 3;
74 }
75
76 do {
77 /* read the bytes */
78 if (cb(tmp, bsize, dat) != bsize) {
79 err = MP_VAL;
80 goto error;
81 }
82
83 /* work over the MSbyte */
84 tmp[0] &= maskAND;
85 tmp[0] |= 1 << ((size - 1) & 7);
86
87 /* mix in the maskORs */
88 tmp[maskOR_msb_offset] |= maskOR_msb;
89 tmp[bsize-1] |= maskOR_lsb;
90
91 /* read it in */
92 if ((err = mp_read_unsigned_bin(a, tmp, bsize)) != MP_OKAY) { goto error; }
93
94 /* is it prime? */
95 if ((err = mp_prime_is_prime(a, t, &res)) != MP_OKAY) { goto error; }
96 if (res == MP_NO) {
97 continue;
98 }
99
100 if (flags & LTM_PRIME_SAFE) {
101 /* see if (a-1)/2 is prime */
102 if ((err = mp_sub_d(a, 1, a)) != MP_OKAY) { goto error; }
103 if ((err = mp_div_2(a, a)) != MP_OKAY) { goto error; }
104
105 /* is it prime? */
106 if ((err = mp_prime_is_prime(a, t, &res)) != MP_OKAY) { goto error; }
107 }
108 } while (res == MP_NO);
109
110 if (flags & LTM_PRIME_SAFE) {
111 /* restore a to the original value */
112 if ((err = mp_mul_2(a, a)) != MP_OKAY) { goto error; }
113 if ((err = mp_add_d(a, 1, a)) != MP_OKAY) { goto error; }
114 }
115
116 err = MP_OKAY;
117 error:
118 XFREE(tmp);
119 return err;
120 }
121
122
123 #endif