Expand PMF_FN_* macros.
[netbsd-mini2440.git] / dist / wpa / src / tls / libtommath.c
blob137426421c299eef8cf11f0b0a058e1ced9da35e
1 /*
2 * Minimal code for RSA support from LibTomMath 0.41
3 * http://libtom.org/
4 * http://libtom.org/files/ltm-0.41.tar.bz2
5 * This library was released in public domain by Tom St Denis.
7 * The combination in this file may not use all of the optimized algorithms
8 * from LibTomMath and may be considerable slower than the LibTomMath with its
9 * default settings. The main purpose of having this version here is to make it
10 * easier to build bignum.c wrapper without having to install and build an
11 * external library.
13 * If CONFIG_INTERNAL_LIBTOMMATH is defined, bignum.c includes this
14 * libtommath.c file instead of using the external LibTomMath library.
17 #ifndef CHAR_BIT
18 #define CHAR_BIT 8
19 #endif
21 #define BN_MP_INVMOD_C
22 #define BN_S_MP_EXPTMOD_C /* Note: #undef in tommath_superclass.h; this would
23 * require BN_MP_EXPTMOD_FAST_C instead */
24 #define BN_S_MP_MUL_DIGS_C
25 #define BN_MP_INVMOD_SLOW_C
26 #define BN_S_MP_SQR_C
27 #define BN_S_MP_MUL_HIGH_DIGS_C /* Note: #undef in tommath_superclass.h; this
28 * would require other than mp_reduce */
30 #ifdef LTM_FAST
32 /* Use faster div at the cost of about 1 kB */
33 #define BN_MP_MUL_D_C
35 /* Include faster exptmod (Montgomery) at the cost of about 2.5 kB in code */
36 #define BN_MP_EXPTMOD_FAST_C
37 #define BN_MP_MONTGOMERY_SETUP_C
38 #define BN_FAST_MP_MONTGOMERY_REDUCE_C
39 #define BN_MP_MONTGOMERY_CALC_NORMALIZATION_C
40 #define BN_MP_MUL_2_C
42 /* Include faster sqr at the cost of about 0.5 kB in code */
43 #define BN_FAST_S_MP_SQR_C
45 #else /* LTM_FAST */
47 #define BN_MP_DIV_SMALL
48 #define BN_MP_INIT_MULTI_C
49 #define BN_MP_CLEAR_MULTI_C
50 #define BN_MP_ABS_C
51 #endif /* LTM_FAST */
53 /* Current uses do not require support for negative exponent in exptmod, so we
54 * can save about 1.5 kB in leaving out invmod. */
55 #define LTM_NO_NEG_EXP
57 /* from tommath.h */
59 #ifndef MIN
60 #define MIN(x,y) ((x)<(y)?(x):(y))
61 #endif
63 #ifndef MAX
64 #define MAX(x,y) ((x)>(y)?(x):(y))
65 #endif
67 #define OPT_CAST(x)
69 typedef unsigned long mp_digit;
70 typedef u64 mp_word;
72 #define DIGIT_BIT 28
73 #define MP_28BIT
76 #define XMALLOC os_malloc
77 #define XFREE os_free
78 #define XREALLOC os_realloc
81 #define MP_MASK ((((mp_digit)1)<<((mp_digit)DIGIT_BIT))-((mp_digit)1))
83 #define MP_LT -1 /* less than */
84 #define MP_EQ 0 /* equal to */
85 #define MP_GT 1 /* greater than */
87 #define MP_ZPOS 0 /* positive integer */
88 #define MP_NEG 1 /* negative */
90 #define MP_OKAY 0 /* ok result */
91 #define MP_MEM -2 /* out of mem */
92 #define MP_VAL -3 /* invalid input */
94 #define MP_YES 1 /* yes response */
95 #define MP_NO 0 /* no response */
97 typedef int mp_err;
99 /* define this to use lower memory usage routines (exptmods mostly) */
100 #define MP_LOW_MEM
102 /* default precision */
103 #ifndef MP_PREC
104 #ifndef MP_LOW_MEM
105 #define MP_PREC 32 /* default digits of precision */
106 #else
107 #define MP_PREC 8 /* default digits of precision */
108 #endif
109 #endif
111 /* size of comba arrays, should be at least 2 * 2**(BITS_PER_WORD - BITS_PER_DIGIT*2) */
112 #define MP_WARRAY (1 << (sizeof(mp_word) * CHAR_BIT - 2 * DIGIT_BIT + 1))
114 /* the infamous mp_int structure */
115 typedef struct {
116 int used, alloc, sign;
117 mp_digit *dp;
118 } mp_int;
121 /* ---> Basic Manipulations <--- */
122 #define mp_iszero(a) (((a)->used == 0) ? MP_YES : MP_NO)
123 #define mp_iseven(a) (((a)->used > 0 && (((a)->dp[0] & 1) == 0)) ? MP_YES : MP_NO)
124 #define mp_isodd(a) (((a)->used > 0 && (((a)->dp[0] & 1) == 1)) ? MP_YES : MP_NO)
127 /* prototypes for copied functions */
128 #define s_mp_mul(a, b, c) s_mp_mul_digs(a, b, c, (a)->used + (b)->used + 1)
129 static int s_mp_exptmod(mp_int * G, mp_int * X, mp_int * P, mp_int * Y, int redmode);
130 static int s_mp_mul_digs (mp_int * a, mp_int * b, mp_int * c, int digs);
131 static int s_mp_sqr(mp_int * a, mp_int * b);
132 static int s_mp_mul_high_digs(mp_int * a, mp_int * b, mp_int * c, int digs);
134 static int fast_s_mp_mul_digs (mp_int * a, mp_int * b, mp_int * c, int digs);
136 #ifdef BN_MP_INIT_MULTI_C
137 static int mp_init_multi(mp_int *mp, ...);
138 #endif
139 #ifdef BN_MP_CLEAR_MULTI_C
140 static void mp_clear_multi(mp_int *mp, ...);
141 #endif
142 static int mp_lshd(mp_int * a, int b);
143 static void mp_set(mp_int * a, mp_digit b);
144 static void mp_clamp(mp_int * a);
145 static void mp_exch(mp_int * a, mp_int * b);
146 static void mp_rshd(mp_int * a, int b);
147 static void mp_zero(mp_int * a);
148 static int mp_mod_2d(mp_int * a, int b, mp_int * c);
149 static int mp_div_2d(mp_int * a, int b, mp_int * c, mp_int * d);
150 static int mp_init_copy(mp_int * a, mp_int * b);
151 static int mp_mul_2d(mp_int * a, int b, mp_int * c);
152 #ifndef LTM_NO_NEG_EXP
153 static int mp_div_2(mp_int * a, mp_int * b);
154 static int mp_invmod(mp_int * a, mp_int * b, mp_int * c);
155 static int mp_invmod_slow(mp_int * a, mp_int * b, mp_int * c);
156 #endif /* LTM_NO_NEG_EXP */
157 static int mp_copy(mp_int * a, mp_int * b);
158 static int mp_count_bits(mp_int * a);
159 static int mp_div(mp_int * a, mp_int * b, mp_int * c, mp_int * d);
160 static int mp_mod(mp_int * a, mp_int * b, mp_int * c);
161 static int mp_grow(mp_int * a, int size);
162 static int mp_cmp_mag(mp_int * a, mp_int * b);
163 #ifdef BN_MP_ABS_C
164 static int mp_abs(mp_int * a, mp_int * b);
165 #endif
166 static int mp_sqr(mp_int * a, mp_int * b);
167 static int mp_reduce_2k_l(mp_int *a, mp_int *n, mp_int *d);
168 static int mp_reduce_2k_setup_l(mp_int *a, mp_int *d);
169 static int mp_2expt(mp_int * a, int b);
170 static int mp_reduce_setup(mp_int * a, mp_int * b);
171 static int mp_reduce(mp_int * x, mp_int * m, mp_int * mu);
172 static int mp_init_size(mp_int * a, int size);
173 #ifdef BN_MP_EXPTMOD_FAST_C
174 static int mp_exptmod_fast (mp_int * G, mp_int * X, mp_int * P, mp_int * Y, int redmode);
175 #endif /* BN_MP_EXPTMOD_FAST_C */
176 #ifdef BN_FAST_S_MP_SQR_C
177 static int fast_s_mp_sqr (mp_int * a, mp_int * b);
178 #endif /* BN_FAST_S_MP_SQR_C */
179 #ifdef BN_MP_MUL_D_C
180 static int mp_mul_d (mp_int * a, mp_digit b, mp_int * c);
181 #endif /* BN_MP_MUL_D_C */
185 /* functions from bn_<func name>.c */
188 /* reverse an array, used for radix code */
189 static void bn_reverse (unsigned char *s, int len)
191 int ix, iy;
192 unsigned char t;
194 ix = 0;
195 iy = len - 1;
196 while (ix < iy) {
197 t = s[ix];
198 s[ix] = s[iy];
199 s[iy] = t;
200 ++ix;
201 --iy;
206 /* low level addition, based on HAC pp.594, Algorithm 14.7 */
207 static int s_mp_add (mp_int * a, mp_int * b, mp_int * c)
209 mp_int *x;
210 int olduse, res, min, max;
212 /* find sizes, we let |a| <= |b| which means we have to sort
213 * them. "x" will point to the input with the most digits
215 if (a->used > b->used) {
216 min = b->used;
217 max = a->used;
218 x = a;
219 } else {
220 min = a->used;
221 max = b->used;
222 x = b;
225 /* init result */
226 if (c->alloc < max + 1) {
227 if ((res = mp_grow (c, max + 1)) != MP_OKAY) {
228 return res;
232 /* get old used digit count and set new one */
233 olduse = c->used;
234 c->used = max + 1;
237 register mp_digit u, *tmpa, *tmpb, *tmpc;
238 register int i;
240 /* alias for digit pointers */
242 /* first input */
243 tmpa = a->dp;
245 /* second input */
246 tmpb = b->dp;
248 /* destination */
249 tmpc = c->dp;
251 /* zero the carry */
252 u = 0;
253 for (i = 0; i < min; i++) {
254 /* Compute the sum at one digit, T[i] = A[i] + B[i] + U */
255 *tmpc = *tmpa++ + *tmpb++ + u;
257 /* U = carry bit of T[i] */
258 u = *tmpc >> ((mp_digit)DIGIT_BIT);
260 /* take away carry bit from T[i] */
261 *tmpc++ &= MP_MASK;
264 /* now copy higher words if any, that is in A+B
265 * if A or B has more digits add those in
267 if (min != max) {
268 for (; i < max; i++) {
269 /* T[i] = X[i] + U */
270 *tmpc = x->dp[i] + u;
272 /* U = carry bit of T[i] */
273 u = *tmpc >> ((mp_digit)DIGIT_BIT);
275 /* take away carry bit from T[i] */
276 *tmpc++ &= MP_MASK;
280 /* add carry */
281 *tmpc++ = u;
283 /* clear digits above oldused */
284 for (i = c->used; i < olduse; i++) {
285 *tmpc++ = 0;
289 mp_clamp (c);
290 return MP_OKAY;
294 /* low level subtraction (assumes |a| > |b|), HAC pp.595 Algorithm 14.9 */
295 static int s_mp_sub (mp_int * a, mp_int * b, mp_int * c)
297 int olduse, res, min, max;
299 /* find sizes */
300 min = b->used;
301 max = a->used;
303 /* init result */
304 if (c->alloc < max) {
305 if ((res = mp_grow (c, max)) != MP_OKAY) {
306 return res;
309 olduse = c->used;
310 c->used = max;
313 register mp_digit u, *tmpa, *tmpb, *tmpc;
314 register int i;
316 /* alias for digit pointers */
317 tmpa = a->dp;
318 tmpb = b->dp;
319 tmpc = c->dp;
321 /* set carry to zero */
322 u = 0;
323 for (i = 0; i < min; i++) {
324 /* T[i] = A[i] - B[i] - U */
325 *tmpc = *tmpa++ - *tmpb++ - u;
327 /* U = carry bit of T[i]
328 * Note this saves performing an AND operation since
329 * if a carry does occur it will propagate all the way to the
330 * MSB. As a result a single shift is enough to get the carry
332 u = *tmpc >> ((mp_digit)(CHAR_BIT * sizeof (mp_digit) - 1));
334 /* Clear carry from T[i] */
335 *tmpc++ &= MP_MASK;
338 /* now copy higher words if any, e.g. if A has more digits than B */
339 for (; i < max; i++) {
340 /* T[i] = A[i] - U */
341 *tmpc = *tmpa++ - u;
343 /* U = carry bit of T[i] */
344 u = *tmpc >> ((mp_digit)(CHAR_BIT * sizeof (mp_digit) - 1));
346 /* Clear carry from T[i] */
347 *tmpc++ &= MP_MASK;
350 /* clear digits above used (since we may not have grown result above) */
351 for (i = c->used; i < olduse; i++) {
352 *tmpc++ = 0;
356 mp_clamp (c);
357 return MP_OKAY;
361 /* init a new mp_int */
362 static int mp_init (mp_int * a)
364 int i;
366 /* allocate memory required and clear it */
367 a->dp = OPT_CAST(mp_digit) XMALLOC (sizeof (mp_digit) * MP_PREC);
368 if (a->dp == NULL) {
369 return MP_MEM;
372 /* set the digits to zero */
373 for (i = 0; i < MP_PREC; i++) {
374 a->dp[i] = 0;
377 /* set the used to zero, allocated digits to the default precision
378 * and sign to positive */
379 a->used = 0;
380 a->alloc = MP_PREC;
381 a->sign = MP_ZPOS;
383 return MP_OKAY;
387 /* clear one (frees) */
388 static void mp_clear (mp_int * a)
390 int i;
392 /* only do anything if a hasn't been freed previously */
393 if (a->dp != NULL) {
394 /* first zero the digits */
395 for (i = 0; i < a->used; i++) {
396 a->dp[i] = 0;
399 /* free ram */
400 XFREE(a->dp);
402 /* reset members to make debugging easier */
403 a->dp = NULL;
404 a->alloc = a->used = 0;
405 a->sign = MP_ZPOS;
410 /* high level addition (handles signs) */
411 static int mp_add (mp_int * a, mp_int * b, mp_int * c)
413 int sa, sb, res;
415 /* get sign of both inputs */
416 sa = a->sign;
417 sb = b->sign;
419 /* handle two cases, not four */
420 if (sa == sb) {
421 /* both positive or both negative */
422 /* add their magnitudes, copy the sign */
423 c->sign = sa;
424 res = s_mp_add (a, b, c);
425 } else {
426 /* one positive, the other negative */
427 /* subtract the one with the greater magnitude from */
428 /* the one of the lesser magnitude. The result gets */
429 /* the sign of the one with the greater magnitude. */
430 if (mp_cmp_mag (a, b) == MP_LT) {
431 c->sign = sb;
432 res = s_mp_sub (b, a, c);
433 } else {
434 c->sign = sa;
435 res = s_mp_sub (a, b, c);
438 return res;
442 /* high level subtraction (handles signs) */
443 static int mp_sub (mp_int * a, mp_int * b, mp_int * c)
445 int sa, sb, res;
447 sa = a->sign;
448 sb = b->sign;
450 if (sa != sb) {
451 /* subtract a negative from a positive, OR */
452 /* subtract a positive from a negative. */
453 /* In either case, ADD their magnitudes, */
454 /* and use the sign of the first number. */
455 c->sign = sa;
456 res = s_mp_add (a, b, c);
457 } else {
458 /* subtract a positive from a positive, OR */
459 /* subtract a negative from a negative. */
460 /* First, take the difference between their */
461 /* magnitudes, then... */
462 if (mp_cmp_mag (a, b) != MP_LT) {
463 /* Copy the sign from the first */
464 c->sign = sa;
465 /* The first has a larger or equal magnitude */
466 res = s_mp_sub (a, b, c);
467 } else {
468 /* The result has the *opposite* sign from */
469 /* the first number. */
470 c->sign = (sa == MP_ZPOS) ? MP_NEG : MP_ZPOS;
471 /* The second has a larger magnitude */
472 res = s_mp_sub (b, a, c);
475 return res;
479 /* high level multiplication (handles sign) */
480 static int mp_mul (mp_int * a, mp_int * b, mp_int * c)
482 int res, neg;
483 neg = (a->sign == b->sign) ? MP_ZPOS : MP_NEG;
485 /* use Toom-Cook? */
486 #ifdef BN_MP_TOOM_MUL_C
487 if (MIN (a->used, b->used) >= TOOM_MUL_CUTOFF) {
488 res = mp_toom_mul(a, b, c);
489 } else
490 #endif
491 #ifdef BN_MP_KARATSUBA_MUL_C
492 /* use Karatsuba? */
493 if (MIN (a->used, b->used) >= KARATSUBA_MUL_CUTOFF) {
494 res = mp_karatsuba_mul (a, b, c);
495 } else
496 #endif
498 /* can we use the fast multiplier?
500 * The fast multiplier can be used if the output will
501 * have less than MP_WARRAY digits and the number of
502 * digits won't affect carry propagation
504 #ifdef BN_FAST_S_MP_MUL_DIGS_C
505 int digs = a->used + b->used + 1;
507 if ((digs < MP_WARRAY) &&
508 MIN(a->used, b->used) <=
509 (1 << ((CHAR_BIT * sizeof (mp_word)) - (2 * DIGIT_BIT)))) {
510 res = fast_s_mp_mul_digs (a, b, c, digs);
511 } else
512 #endif
513 #ifdef BN_S_MP_MUL_DIGS_C
514 res = s_mp_mul (a, b, c); /* uses s_mp_mul_digs */
515 #else
516 #error mp_mul could fail
517 res = MP_VAL;
518 #endif
521 c->sign = (c->used > 0) ? neg : MP_ZPOS;
522 return res;
526 /* d = a * b (mod c) */
527 static int mp_mulmod (mp_int * a, mp_int * b, mp_int * c, mp_int * d)
529 int res;
530 mp_int t;
532 if ((res = mp_init (&t)) != MP_OKAY) {
533 return res;
536 if ((res = mp_mul (a, b, &t)) != MP_OKAY) {
537 mp_clear (&t);
538 return res;
540 res = mp_mod (&t, c, d);
541 mp_clear (&t);
542 return res;
546 /* c = a mod b, 0 <= c < b */
547 static int mp_mod (mp_int * a, mp_int * b, mp_int * c)
549 mp_int t;
550 int res;
552 if ((res = mp_init (&t)) != MP_OKAY) {
553 return res;
556 if ((res = mp_div (a, b, NULL, &t)) != MP_OKAY) {
557 mp_clear (&t);
558 return res;
561 if (t.sign != b->sign) {
562 res = mp_add (b, &t, c);
563 } else {
564 res = MP_OKAY;
565 mp_exch (&t, c);
568 mp_clear (&t);
569 return res;
573 /* this is a shell function that calls either the normal or Montgomery
574 * exptmod functions. Originally the call to the montgomery code was
575 * embedded in the normal function but that wasted alot of stack space
576 * for nothing (since 99% of the time the Montgomery code would be called)
578 static int mp_exptmod (mp_int * G, mp_int * X, mp_int * P, mp_int * Y)
580 int dr;
582 /* modulus P must be positive */
583 if (P->sign == MP_NEG) {
584 return MP_VAL;
587 /* if exponent X is negative we have to recurse */
588 if (X->sign == MP_NEG) {
589 #ifdef LTM_NO_NEG_EXP
590 return MP_VAL;
591 #else /* LTM_NO_NEG_EXP */
592 #ifdef BN_MP_INVMOD_C
593 mp_int tmpG, tmpX;
594 int err;
596 /* first compute 1/G mod P */
597 if ((err = mp_init(&tmpG)) != MP_OKAY) {
598 return err;
600 if ((err = mp_invmod(G, P, &tmpG)) != MP_OKAY) {
601 mp_clear(&tmpG);
602 return err;
605 /* now get |X| */
606 if ((err = mp_init(&tmpX)) != MP_OKAY) {
607 mp_clear(&tmpG);
608 return err;
610 if ((err = mp_abs(X, &tmpX)) != MP_OKAY) {
611 mp_clear_multi(&tmpG, &tmpX, NULL);
612 return err;
615 /* and now compute (1/G)**|X| instead of G**X [X < 0] */
616 err = mp_exptmod(&tmpG, &tmpX, P, Y);
617 mp_clear_multi(&tmpG, &tmpX, NULL);
618 return err;
619 #else
620 #error mp_exptmod would always fail
621 /* no invmod */
622 return MP_VAL;
623 #endif
624 #endif /* LTM_NO_NEG_EXP */
627 /* modified diminished radix reduction */
628 #if defined(BN_MP_REDUCE_IS_2K_L_C) && defined(BN_MP_REDUCE_2K_L_C) && defined(BN_S_MP_EXPTMOD_C)
629 if (mp_reduce_is_2k_l(P) == MP_YES) {
630 return s_mp_exptmod(G, X, P, Y, 1);
632 #endif
634 #ifdef BN_MP_DR_IS_MODULUS_C
635 /* is it a DR modulus? */
636 dr = mp_dr_is_modulus(P);
637 #else
638 /* default to no */
639 dr = 0;
640 #endif
642 #ifdef BN_MP_REDUCE_IS_2K_C
643 /* if not, is it a unrestricted DR modulus? */
644 if (dr == 0) {
645 dr = mp_reduce_is_2k(P) << 1;
647 #endif
649 /* if the modulus is odd or dr != 0 use the montgomery method */
650 #ifdef BN_MP_EXPTMOD_FAST_C
651 if (mp_isodd (P) == 1 || dr != 0) {
652 return mp_exptmod_fast (G, X, P, Y, dr);
653 } else {
654 #endif
655 #ifdef BN_S_MP_EXPTMOD_C
656 /* otherwise use the generic Barrett reduction technique */
657 return s_mp_exptmod (G, X, P, Y, 0);
658 #else
659 #error mp_exptmod could fail
660 /* no exptmod for evens */
661 return MP_VAL;
662 #endif
663 #ifdef BN_MP_EXPTMOD_FAST_C
665 #endif
669 /* compare two ints (signed)*/
670 static int mp_cmp (mp_int * a, mp_int * b)
672 /* compare based on sign */
673 if (a->sign != b->sign) {
674 if (a->sign == MP_NEG) {
675 return MP_LT;
676 } else {
677 return MP_GT;
681 /* compare digits */
682 if (a->sign == MP_NEG) {
683 /* if negative compare opposite direction */
684 return mp_cmp_mag(b, a);
685 } else {
686 return mp_cmp_mag(a, b);
691 /* compare a digit */
692 static int mp_cmp_d(mp_int * a, mp_digit b)
694 /* compare based on sign */
695 if (a->sign == MP_NEG) {
696 return MP_LT;
699 /* compare based on magnitude */
700 if (a->used > 1) {
701 return MP_GT;
704 /* compare the only digit of a to b */
705 if (a->dp[0] > b) {
706 return MP_GT;
707 } else if (a->dp[0] < b) {
708 return MP_LT;
709 } else {
710 return MP_EQ;
715 #ifndef LTM_NO_NEG_EXP
716 /* hac 14.61, pp608 */
717 static int mp_invmod (mp_int * a, mp_int * b, mp_int * c)
719 /* b cannot be negative */
720 if (b->sign == MP_NEG || mp_iszero(b) == 1) {
721 return MP_VAL;
724 #ifdef BN_FAST_MP_INVMOD_C
725 /* if the modulus is odd we can use a faster routine instead */
726 if (mp_isodd (b) == 1) {
727 return fast_mp_invmod (a, b, c);
729 #endif
731 #ifdef BN_MP_INVMOD_SLOW_C
732 return mp_invmod_slow(a, b, c);
733 #endif
735 #ifndef BN_FAST_MP_INVMOD_C
736 #ifndef BN_MP_INVMOD_SLOW_C
737 #error mp_invmod would always fail
738 #endif
739 #endif
740 return MP_VAL;
742 #endif /* LTM_NO_NEG_EXP */
745 /* get the size for an unsigned equivalent */
746 static int mp_unsigned_bin_size (mp_int * a)
748 int size = mp_count_bits (a);
749 return (size / 8 + ((size & 7) != 0 ? 1 : 0));
753 #ifndef LTM_NO_NEG_EXP
754 /* hac 14.61, pp608 */
755 static int mp_invmod_slow (mp_int * a, mp_int * b, mp_int * c)
757 mp_int x, y, u, v, A, B, C, D;
758 int res;
760 /* b cannot be negative */
761 if (b->sign == MP_NEG || mp_iszero(b) == 1) {
762 return MP_VAL;
765 /* init temps */
766 if ((res = mp_init_multi(&x, &y, &u, &v,
767 &A, &B, &C, &D, NULL)) != MP_OKAY) {
768 return res;
771 /* x = a, y = b */
772 if ((res = mp_mod(a, b, &x)) != MP_OKAY) {
773 goto LBL_ERR;
775 if ((res = mp_copy (b, &y)) != MP_OKAY) {
776 goto LBL_ERR;
779 /* 2. [modified] if x,y are both even then return an error! */
780 if (mp_iseven (&x) == 1 && mp_iseven (&y) == 1) {
781 res = MP_VAL;
782 goto LBL_ERR;
785 /* 3. u=x, v=y, A=1, B=0, C=0,D=1 */
786 if ((res = mp_copy (&x, &u)) != MP_OKAY) {
787 goto LBL_ERR;
789 if ((res = mp_copy (&y, &v)) != MP_OKAY) {
790 goto LBL_ERR;
792 mp_set (&A, 1);
793 mp_set (&D, 1);
795 top:
796 /* 4. while u is even do */
797 while (mp_iseven (&u) == 1) {
798 /* 4.1 u = u/2 */
799 if ((res = mp_div_2 (&u, &u)) != MP_OKAY) {
800 goto LBL_ERR;
802 /* 4.2 if A or B is odd then */
803 if (mp_isodd (&A) == 1 || mp_isodd (&B) == 1) {
804 /* A = (A+y)/2, B = (B-x)/2 */
805 if ((res = mp_add (&A, &y, &A)) != MP_OKAY) {
806 goto LBL_ERR;
808 if ((res = mp_sub (&B, &x, &B)) != MP_OKAY) {
809 goto LBL_ERR;
812 /* A = A/2, B = B/2 */
813 if ((res = mp_div_2 (&A, &A)) != MP_OKAY) {
814 goto LBL_ERR;
816 if ((res = mp_div_2 (&B, &B)) != MP_OKAY) {
817 goto LBL_ERR;
821 /* 5. while v is even do */
822 while (mp_iseven (&v) == 1) {
823 /* 5.1 v = v/2 */
824 if ((res = mp_div_2 (&v, &v)) != MP_OKAY) {
825 goto LBL_ERR;
827 /* 5.2 if C or D is odd then */
828 if (mp_isodd (&C) == 1 || mp_isodd (&D) == 1) {
829 /* C = (C+y)/2, D = (D-x)/2 */
830 if ((res = mp_add (&C, &y, &C)) != MP_OKAY) {
831 goto LBL_ERR;
833 if ((res = mp_sub (&D, &x, &D)) != MP_OKAY) {
834 goto LBL_ERR;
837 /* C = C/2, D = D/2 */
838 if ((res = mp_div_2 (&C, &C)) != MP_OKAY) {
839 goto LBL_ERR;
841 if ((res = mp_div_2 (&D, &D)) != MP_OKAY) {
842 goto LBL_ERR;
846 /* 6. if u >= v then */
847 if (mp_cmp (&u, &v) != MP_LT) {
848 /* u = u - v, A = A - C, B = B - D */
849 if ((res = mp_sub (&u, &v, &u)) != MP_OKAY) {
850 goto LBL_ERR;
853 if ((res = mp_sub (&A, &C, &A)) != MP_OKAY) {
854 goto LBL_ERR;
857 if ((res = mp_sub (&B, &D, &B)) != MP_OKAY) {
858 goto LBL_ERR;
860 } else {
861 /* v - v - u, C = C - A, D = D - B */
862 if ((res = mp_sub (&v, &u, &v)) != MP_OKAY) {
863 goto LBL_ERR;
866 if ((res = mp_sub (&C, &A, &C)) != MP_OKAY) {
867 goto LBL_ERR;
870 if ((res = mp_sub (&D, &B, &D)) != MP_OKAY) {
871 goto LBL_ERR;
875 /* if not zero goto step 4 */
876 if (mp_iszero (&u) == 0)
877 goto top;
879 /* now a = C, b = D, gcd == g*v */
881 /* if v != 1 then there is no inverse */
882 if (mp_cmp_d (&v, 1) != MP_EQ) {
883 res = MP_VAL;
884 goto LBL_ERR;
887 /* if its too low */
888 while (mp_cmp_d(&C, 0) == MP_LT) {
889 if ((res = mp_add(&C, b, &C)) != MP_OKAY) {
890 goto LBL_ERR;
894 /* too big */
895 while (mp_cmp_mag(&C, b) != MP_LT) {
896 if ((res = mp_sub(&C, b, &C)) != MP_OKAY) {
897 goto LBL_ERR;
901 /* C is now the inverse */
902 mp_exch (&C, c);
903 res = MP_OKAY;
904 LBL_ERR:mp_clear_multi (&x, &y, &u, &v, &A, &B, &C, &D, NULL);
905 return res;
907 #endif /* LTM_NO_NEG_EXP */
910 /* compare maginitude of two ints (unsigned) */
911 static int mp_cmp_mag (mp_int * a, mp_int * b)
913 int n;
914 mp_digit *tmpa, *tmpb;
916 /* compare based on # of non-zero digits */
917 if (a->used > b->used) {
918 return MP_GT;
921 if (a->used < b->used) {
922 return MP_LT;
925 /* alias for a */
926 tmpa = a->dp + (a->used - 1);
928 /* alias for b */
929 tmpb = b->dp + (a->used - 1);
931 /* compare based on digits */
932 for (n = 0; n < a->used; ++n, --tmpa, --tmpb) {
933 if (*tmpa > *tmpb) {
934 return MP_GT;
937 if (*tmpa < *tmpb) {
938 return MP_LT;
941 return MP_EQ;
945 /* reads a unsigned char array, assumes the msb is stored first [big endian] */
946 static int mp_read_unsigned_bin (mp_int * a, const unsigned char *b, int c)
948 int res;
950 /* make sure there are at least two digits */
951 if (a->alloc < 2) {
952 if ((res = mp_grow(a, 2)) != MP_OKAY) {
953 return res;
957 /* zero the int */
958 mp_zero (a);
960 /* read the bytes in */
961 while (c-- > 0) {
962 if ((res = mp_mul_2d (a, 8, a)) != MP_OKAY) {
963 return res;
966 #ifndef MP_8BIT
967 a->dp[0] |= *b++;
968 a->used += 1;
969 #else
970 a->dp[0] = (*b & MP_MASK);
971 a->dp[1] |= ((*b++ >> 7U) & 1);
972 a->used += 2;
973 #endif
975 mp_clamp (a);
976 return MP_OKAY;
980 /* store in unsigned [big endian] format */
981 static int mp_to_unsigned_bin (mp_int * a, unsigned char *b)
983 int x, res;
984 mp_int t;
986 if ((res = mp_init_copy (&t, a)) != MP_OKAY) {
987 return res;
990 x = 0;
991 while (mp_iszero (&t) == 0) {
992 #ifndef MP_8BIT
993 b[x++] = (unsigned char) (t.dp[0] & 255);
994 #else
995 b[x++] = (unsigned char) (t.dp[0] | ((t.dp[1] & 0x01) << 7));
996 #endif
997 if ((res = mp_div_2d (&t, 8, &t, NULL)) != MP_OKAY) {
998 mp_clear (&t);
999 return res;
1002 bn_reverse (b, x);
1003 mp_clear (&t);
1004 return MP_OKAY;
1008 /* shift right by a certain bit count (store quotient in c, optional remainder in d) */
1009 static int mp_div_2d (mp_int * a, int b, mp_int * c, mp_int * d)
1011 mp_digit D, r, rr;
1012 int x, res;
1013 mp_int t;
1016 /* if the shift count is <= 0 then we do no work */
1017 if (b <= 0) {
1018 res = mp_copy (a, c);
1019 if (d != NULL) {
1020 mp_zero (d);
1022 return res;
1025 if ((res = mp_init (&t)) != MP_OKAY) {
1026 return res;
1029 /* get the remainder */
1030 if (d != NULL) {
1031 if ((res = mp_mod_2d (a, b, &t)) != MP_OKAY) {
1032 mp_clear (&t);
1033 return res;
1037 /* copy */
1038 if ((res = mp_copy (a, c)) != MP_OKAY) {
1039 mp_clear (&t);
1040 return res;
1043 /* shift by as many digits in the bit count */
1044 if (b >= (int)DIGIT_BIT) {
1045 mp_rshd (c, b / DIGIT_BIT);
1048 /* shift any bit count < DIGIT_BIT */
1049 D = (mp_digit) (b % DIGIT_BIT);
1050 if (D != 0) {
1051 register mp_digit *tmpc, mask, shift;
1053 /* mask */
1054 mask = (((mp_digit)1) << D) - 1;
1056 /* shift for lsb */
1057 shift = DIGIT_BIT - D;
1059 /* alias */
1060 tmpc = c->dp + (c->used - 1);
1062 /* carry */
1063 r = 0;
1064 for (x = c->used - 1; x >= 0; x--) {
1065 /* get the lower bits of this word in a temp */
1066 rr = *tmpc & mask;
1068 /* shift the current word and mix in the carry bits from the previous word */
1069 *tmpc = (*tmpc >> D) | (r << shift);
1070 --tmpc;
1072 /* set the carry to the carry bits of the current word found above */
1073 r = rr;
1076 mp_clamp (c);
1077 if (d != NULL) {
1078 mp_exch (&t, d);
1080 mp_clear (&t);
1081 return MP_OKAY;
1085 static int mp_init_copy (mp_int * a, mp_int * b)
1087 int res;
1089 if ((res = mp_init (a)) != MP_OKAY) {
1090 return res;
1092 return mp_copy (b, a);
1096 /* set to zero */
1097 static void mp_zero (mp_int * a)
1099 int n;
1100 mp_digit *tmp;
1102 a->sign = MP_ZPOS;
1103 a->used = 0;
1105 tmp = a->dp;
1106 for (n = 0; n < a->alloc; n++) {
1107 *tmp++ = 0;
1112 /* copy, b = a */
1113 static int mp_copy (mp_int * a, mp_int * b)
1115 int res, n;
1117 /* if dst == src do nothing */
1118 if (a == b) {
1119 return MP_OKAY;
1122 /* grow dest */
1123 if (b->alloc < a->used) {
1124 if ((res = mp_grow (b, a->used)) != MP_OKAY) {
1125 return res;
1129 /* zero b and copy the parameters over */
1131 register mp_digit *tmpa, *tmpb;
1133 /* pointer aliases */
1135 /* source */
1136 tmpa = a->dp;
1138 /* destination */
1139 tmpb = b->dp;
1141 /* copy all the digits */
1142 for (n = 0; n < a->used; n++) {
1143 *tmpb++ = *tmpa++;
1146 /* clear high digits */
1147 for (; n < b->used; n++) {
1148 *tmpb++ = 0;
1152 /* copy used count and sign */
1153 b->used = a->used;
1154 b->sign = a->sign;
1155 return MP_OKAY;
1159 /* shift right a certain amount of digits */
1160 static void mp_rshd (mp_int * a, int b)
1162 int x;
1164 /* if b <= 0 then ignore it */
1165 if (b <= 0) {
1166 return;
1169 /* if b > used then simply zero it and return */
1170 if (a->used <= b) {
1171 mp_zero (a);
1172 return;
1176 register mp_digit *bottom, *top;
1178 /* shift the digits down */
1180 /* bottom */
1181 bottom = a->dp;
1183 /* top [offset into digits] */
1184 top = a->dp + b;
1186 /* this is implemented as a sliding window where
1187 * the window is b-digits long and digits from
1188 * the top of the window are copied to the bottom
1190 * e.g.
1192 b-2 | b-1 | b0 | b1 | b2 | ... | bb | ---->
1193 /\ | ---->
1194 \-------------------/ ---->
1196 for (x = 0; x < (a->used - b); x++) {
1197 *bottom++ = *top++;
1200 /* zero the top digits */
1201 for (; x < a->used; x++) {
1202 *bottom++ = 0;
1206 /* remove excess digits */
1207 a->used -= b;
1211 /* swap the elements of two integers, for cases where you can't simply swap the
1212 * mp_int pointers around
1214 static void mp_exch (mp_int * a, mp_int * b)
1216 mp_int t;
1218 t = *a;
1219 *a = *b;
1220 *b = t;
1224 /* trim unused digits
1226 * This is used to ensure that leading zero digits are
1227 * trimed and the leading "used" digit will be non-zero
1228 * Typically very fast. Also fixes the sign if there
1229 * are no more leading digits
1231 static void mp_clamp (mp_int * a)
1233 /* decrease used while the most significant digit is
1234 * zero.
1236 while (a->used > 0 && a->dp[a->used - 1] == 0) {
1237 --(a->used);
1240 /* reset the sign flag if used == 0 */
1241 if (a->used == 0) {
1242 a->sign = MP_ZPOS;
1247 /* grow as required */
1248 static int mp_grow (mp_int * a, int size)
1250 int i;
1251 mp_digit *tmp;
1253 /* if the alloc size is smaller alloc more ram */
1254 if (a->alloc < size) {
1255 /* ensure there are always at least MP_PREC digits extra on top */
1256 size += (MP_PREC * 2) - (size % MP_PREC);
1258 /* reallocate the array a->dp
1260 * We store the return in a temporary variable
1261 * in case the operation failed we don't want
1262 * to overwrite the dp member of a.
1264 tmp = OPT_CAST(mp_digit) XREALLOC (a->dp, sizeof (mp_digit) * size);
1265 if (tmp == NULL) {
1266 /* reallocation failed but "a" is still valid [can be freed] */
1267 return MP_MEM;
1270 /* reallocation succeeded so set a->dp */
1271 a->dp = tmp;
1273 /* zero excess digits */
1274 i = a->alloc;
1275 a->alloc = size;
1276 for (; i < a->alloc; i++) {
1277 a->dp[i] = 0;
1280 return MP_OKAY;
1284 #ifdef BN_MP_ABS_C
1285 /* b = |a|
1287 * Simple function copies the input and fixes the sign to positive
1289 static int mp_abs (mp_int * a, mp_int * b)
1291 int res;
1293 /* copy a to b */
1294 if (a != b) {
1295 if ((res = mp_copy (a, b)) != MP_OKAY) {
1296 return res;
1300 /* force the sign of b to positive */
1301 b->sign = MP_ZPOS;
1303 return MP_OKAY;
1305 #endif
1308 /* set to a digit */
1309 static void mp_set (mp_int * a, mp_digit b)
1311 mp_zero (a);
1312 a->dp[0] = b & MP_MASK;
1313 a->used = (a->dp[0] != 0) ? 1 : 0;
1317 #ifndef LTM_NO_NEG_EXP
1318 /* b = a/2 */
1319 static int mp_div_2(mp_int * a, mp_int * b)
1321 int x, res, oldused;
1323 /* copy */
1324 if (b->alloc < a->used) {
1325 if ((res = mp_grow (b, a->used)) != MP_OKAY) {
1326 return res;
1330 oldused = b->used;
1331 b->used = a->used;
1333 register mp_digit r, rr, *tmpa, *tmpb;
1335 /* source alias */
1336 tmpa = a->dp + b->used - 1;
1338 /* dest alias */
1339 tmpb = b->dp + b->used - 1;
1341 /* carry */
1342 r = 0;
1343 for (x = b->used - 1; x >= 0; x--) {
1344 /* get the carry for the next iteration */
1345 rr = *tmpa & 1;
1347 /* shift the current digit, add in carry and store */
1348 *tmpb-- = (*tmpa-- >> 1) | (r << (DIGIT_BIT - 1));
1350 /* forward carry to next iteration */
1351 r = rr;
1354 /* zero excess digits */
1355 tmpb = b->dp + b->used;
1356 for (x = b->used; x < oldused; x++) {
1357 *tmpb++ = 0;
1360 b->sign = a->sign;
1361 mp_clamp (b);
1362 return MP_OKAY;
1364 #endif /* LTM_NO_NEG_EXP */
1367 /* shift left by a certain bit count */
1368 static int mp_mul_2d (mp_int * a, int b, mp_int * c)
1370 mp_digit d;
1371 int res;
1373 /* copy */
1374 if (a != c) {
1375 if ((res = mp_copy (a, c)) != MP_OKAY) {
1376 return res;
1380 if (c->alloc < (int)(c->used + b/DIGIT_BIT + 1)) {
1381 if ((res = mp_grow (c, c->used + b / DIGIT_BIT + 1)) != MP_OKAY) {
1382 return res;
1386 /* shift by as many digits in the bit count */
1387 if (b >= (int)DIGIT_BIT) {
1388 if ((res = mp_lshd (c, b / DIGIT_BIT)) != MP_OKAY) {
1389 return res;
1393 /* shift any bit count < DIGIT_BIT */
1394 d = (mp_digit) (b % DIGIT_BIT);
1395 if (d != 0) {
1396 register mp_digit *tmpc, shift, mask, r, rr;
1397 register int x;
1399 /* bitmask for carries */
1400 mask = (((mp_digit)1) << d) - 1;
1402 /* shift for msbs */
1403 shift = DIGIT_BIT - d;
1405 /* alias */
1406 tmpc = c->dp;
1408 /* carry */
1409 r = 0;
1410 for (x = 0; x < c->used; x++) {
1411 /* get the higher bits of the current word */
1412 rr = (*tmpc >> shift) & mask;
1414 /* shift the current word and OR in the carry */
1415 *tmpc = ((*tmpc << d) | r) & MP_MASK;
1416 ++tmpc;
1418 /* set the carry to the carry bits of the current word */
1419 r = rr;
1422 /* set final carry */
1423 if (r != 0) {
1424 c->dp[(c->used)++] = r;
1427 mp_clamp (c);
1428 return MP_OKAY;
1432 #ifdef BN_MP_INIT_MULTI_C
1433 static int mp_init_multi(mp_int *mp, ...)
1435 mp_err res = MP_OKAY; /* Assume ok until proven otherwise */
1436 int n = 0; /* Number of ok inits */
1437 mp_int* cur_arg = mp;
1438 va_list args;
1440 va_start(args, mp); /* init args to next argument from caller */
1441 while (cur_arg != NULL) {
1442 if (mp_init(cur_arg) != MP_OKAY) {
1443 /* Oops - error! Back-track and mp_clear what we already
1444 succeeded in init-ing, then return error.
1446 va_list clean_args;
1448 /* end the current list */
1449 va_end(args);
1451 /* now start cleaning up */
1452 cur_arg = mp;
1453 va_start(clean_args, mp);
1454 while (n--) {
1455 mp_clear(cur_arg);
1456 cur_arg = va_arg(clean_args, mp_int*);
1458 va_end(clean_args);
1459 res = MP_MEM;
1460 break;
1462 n++;
1463 cur_arg = va_arg(args, mp_int*);
1465 va_end(args);
1466 return res; /* Assumed ok, if error flagged above. */
1468 #endif
1471 #ifdef BN_MP_CLEAR_MULTI_C
1472 static void mp_clear_multi(mp_int *mp, ...)
1474 mp_int* next_mp = mp;
1475 va_list args;
1476 va_start(args, mp);
1477 while (next_mp != NULL) {
1478 mp_clear(next_mp);
1479 next_mp = va_arg(args, mp_int*);
1481 va_end(args);
1483 #endif
1486 /* shift left a certain amount of digits */
1487 static int mp_lshd (mp_int * a, int b)
1489 int x, res;
1491 /* if its less than zero return */
1492 if (b <= 0) {
1493 return MP_OKAY;
1496 /* grow to fit the new digits */
1497 if (a->alloc < a->used + b) {
1498 if ((res = mp_grow (a, a->used + b)) != MP_OKAY) {
1499 return res;
1504 register mp_digit *top, *bottom;
1506 /* increment the used by the shift amount then copy upwards */
1507 a->used += b;
1509 /* top */
1510 top = a->dp + a->used - 1;
1512 /* base */
1513 bottom = a->dp + a->used - 1 - b;
1515 /* much like mp_rshd this is implemented using a sliding window
1516 * except the window goes the otherway around. Copying from
1517 * the bottom to the top. see bn_mp_rshd.c for more info.
1519 for (x = a->used - 1; x >= b; x--) {
1520 *top-- = *bottom--;
1523 /* zero the lower digits */
1524 top = a->dp;
1525 for (x = 0; x < b; x++) {
1526 *top++ = 0;
1529 return MP_OKAY;
1533 /* returns the number of bits in an int */
1534 static int mp_count_bits (mp_int * a)
1536 int r;
1537 mp_digit q;
1539 /* shortcut */
1540 if (a->used == 0) {
1541 return 0;
1544 /* get number of digits and add that */
1545 r = (a->used - 1) * DIGIT_BIT;
1547 /* take the last digit and count the bits in it */
1548 q = a->dp[a->used - 1];
1549 while (q > ((mp_digit) 0)) {
1550 ++r;
1551 q >>= ((mp_digit) 1);
1553 return r;
1557 /* calc a value mod 2**b */
1558 static int mp_mod_2d (mp_int * a, int b, mp_int * c)
1560 int x, res;
1562 /* if b is <= 0 then zero the int */
1563 if (b <= 0) {
1564 mp_zero (c);
1565 return MP_OKAY;
1568 /* if the modulus is larger than the value than return */
1569 if (b >= (int) (a->used * DIGIT_BIT)) {
1570 res = mp_copy (a, c);
1571 return res;
1574 /* copy */
1575 if ((res = mp_copy (a, c)) != MP_OKAY) {
1576 return res;
1579 /* zero digits above the last digit of the modulus */
1580 for (x = (b / DIGIT_BIT) + ((b % DIGIT_BIT) == 0 ? 0 : 1); x < c->used; x++) {
1581 c->dp[x] = 0;
1583 /* clear the digit that is not completely outside/inside the modulus */
1584 c->dp[b / DIGIT_BIT] &=
1585 (mp_digit) ((((mp_digit) 1) << (((mp_digit) b) % DIGIT_BIT)) - ((mp_digit) 1));
1586 mp_clamp (c);
1587 return MP_OKAY;
1591 #ifdef BN_MP_DIV_SMALL
1593 /* slower bit-bang division... also smaller */
1594 static int mp_div(mp_int * a, mp_int * b, mp_int * c, mp_int * d)
1596 mp_int ta, tb, tq, q;
1597 int res, n, n2;
1599 /* is divisor zero ? */
1600 if (mp_iszero (b) == 1) {
1601 return MP_VAL;
1604 /* if a < b then q=0, r = a */
1605 if (mp_cmp_mag (a, b) == MP_LT) {
1606 if (d != NULL) {
1607 res = mp_copy (a, d);
1608 } else {
1609 res = MP_OKAY;
1611 if (c != NULL) {
1612 mp_zero (c);
1614 return res;
1617 /* init our temps */
1618 if ((res = mp_init_multi(&ta, &tb, &tq, &q, NULL) != MP_OKAY)) {
1619 return res;
1623 mp_set(&tq, 1);
1624 n = mp_count_bits(a) - mp_count_bits(b);
1625 if (((res = mp_abs(a, &ta)) != MP_OKAY) ||
1626 ((res = mp_abs(b, &tb)) != MP_OKAY) ||
1627 ((res = mp_mul_2d(&tb, n, &tb)) != MP_OKAY) ||
1628 ((res = mp_mul_2d(&tq, n, &tq)) != MP_OKAY)) {
1629 goto LBL_ERR;
1632 while (n-- >= 0) {
1633 if (mp_cmp(&tb, &ta) != MP_GT) {
1634 if (((res = mp_sub(&ta, &tb, &ta)) != MP_OKAY) ||
1635 ((res = mp_add(&q, &tq, &q)) != MP_OKAY)) {
1636 goto LBL_ERR;
1639 if (((res = mp_div_2d(&tb, 1, &tb, NULL)) != MP_OKAY) ||
1640 ((res = mp_div_2d(&tq, 1, &tq, NULL)) != MP_OKAY)) {
1641 goto LBL_ERR;
1645 /* now q == quotient and ta == remainder */
1646 n = a->sign;
1647 n2 = (a->sign == b->sign ? MP_ZPOS : MP_NEG);
1648 if (c != NULL) {
1649 mp_exch(c, &q);
1650 c->sign = (mp_iszero(c) == MP_YES) ? MP_ZPOS : n2;
1652 if (d != NULL) {
1653 mp_exch(d, &ta);
1654 d->sign = (mp_iszero(d) == MP_YES) ? MP_ZPOS : n;
1656 LBL_ERR:
1657 mp_clear_multi(&ta, &tb, &tq, &q, NULL);
1658 return res;
1661 #else
1663 /* integer signed division.
1664 * c*b + d == a [e.g. a/b, c=quotient, d=remainder]
1665 * HAC pp.598 Algorithm 14.20
1667 * Note that the description in HAC is horribly
1668 * incomplete. For example, it doesn't consider
1669 * the case where digits are removed from 'x' in
1670 * the inner loop. It also doesn't consider the
1671 * case that y has fewer than three digits, etc..
1673 * The overall algorithm is as described as
1674 * 14.20 from HAC but fixed to treat these cases.
1676 static int mp_div (mp_int * a, mp_int * b, mp_int * c, mp_int * d)
1678 mp_int q, x, y, t1, t2;
1679 int res, n, t, i, norm, neg;
1681 /* is divisor zero ? */
1682 if (mp_iszero (b) == 1) {
1683 return MP_VAL;
1686 /* if a < b then q=0, r = a */
1687 if (mp_cmp_mag (a, b) == MP_LT) {
1688 if (d != NULL) {
1689 res = mp_copy (a, d);
1690 } else {
1691 res = MP_OKAY;
1693 if (c != NULL) {
1694 mp_zero (c);
1696 return res;
1699 if ((res = mp_init_size (&q, a->used + 2)) != MP_OKAY) {
1700 return res;
1702 q.used = a->used + 2;
1704 if ((res = mp_init (&t1)) != MP_OKAY) {
1705 goto LBL_Q;
1708 if ((res = mp_init (&t2)) != MP_OKAY) {
1709 goto LBL_T1;
1712 if ((res = mp_init_copy (&x, a)) != MP_OKAY) {
1713 goto LBL_T2;
1716 if ((res = mp_init_copy (&y, b)) != MP_OKAY) {
1717 goto LBL_X;
1720 /* fix the sign */
1721 neg = (a->sign == b->sign) ? MP_ZPOS : MP_NEG;
1722 x.sign = y.sign = MP_ZPOS;
1724 /* normalize both x and y, ensure that y >= b/2, [b == 2**DIGIT_BIT] */
1725 norm = mp_count_bits(&y) % DIGIT_BIT;
1726 if (norm < (int)(DIGIT_BIT-1)) {
1727 norm = (DIGIT_BIT-1) - norm;
1728 if ((res = mp_mul_2d (&x, norm, &x)) != MP_OKAY) {
1729 goto LBL_Y;
1731 if ((res = mp_mul_2d (&y, norm, &y)) != MP_OKAY) {
1732 goto LBL_Y;
1734 } else {
1735 norm = 0;
1738 /* note hac does 0 based, so if used==5 then its 0,1,2,3,4, e.g. use 4 */
1739 n = x.used - 1;
1740 t = y.used - 1;
1742 /* while (x >= y*b**n-t) do { q[n-t] += 1; x -= y*b**{n-t} } */
1743 if ((res = mp_lshd (&y, n - t)) != MP_OKAY) { /* y = y*b**{n-t} */
1744 goto LBL_Y;
1747 while (mp_cmp (&x, &y) != MP_LT) {
1748 ++(q.dp[n - t]);
1749 if ((res = mp_sub (&x, &y, &x)) != MP_OKAY) {
1750 goto LBL_Y;
1754 /* reset y by shifting it back down */
1755 mp_rshd (&y, n - t);
1757 /* step 3. for i from n down to (t + 1) */
1758 for (i = n; i >= (t + 1); i--) {
1759 if (i > x.used) {
1760 continue;
1763 /* step 3.1 if xi == yt then set q{i-t-1} to b-1,
1764 * otherwise set q{i-t-1} to (xi*b + x{i-1})/yt */
1765 if (x.dp[i] == y.dp[t]) {
1766 q.dp[i - t - 1] = ((((mp_digit)1) << DIGIT_BIT) - 1);
1767 } else {
1768 mp_word tmp;
1769 tmp = ((mp_word) x.dp[i]) << ((mp_word) DIGIT_BIT);
1770 tmp |= ((mp_word) x.dp[i - 1]);
1771 tmp /= ((mp_word) y.dp[t]);
1772 if (tmp > (mp_word) MP_MASK)
1773 tmp = MP_MASK;
1774 q.dp[i - t - 1] = (mp_digit) (tmp & (mp_word) (MP_MASK));
1777 /* while (q{i-t-1} * (yt * b + y{t-1})) >
1778 xi * b**2 + xi-1 * b + xi-2
1780 do q{i-t-1} -= 1;
1782 q.dp[i - t - 1] = (q.dp[i - t - 1] + 1) & MP_MASK;
1783 do {
1784 q.dp[i - t - 1] = (q.dp[i - t - 1] - 1) & MP_MASK;
1786 /* find left hand */
1787 mp_zero (&t1);
1788 t1.dp[0] = (t - 1 < 0) ? 0 : y.dp[t - 1];
1789 t1.dp[1] = y.dp[t];
1790 t1.used = 2;
1791 if ((res = mp_mul_d (&t1, q.dp[i - t - 1], &t1)) != MP_OKAY) {
1792 goto LBL_Y;
1795 /* find right hand */
1796 t2.dp[0] = (i - 2 < 0) ? 0 : x.dp[i - 2];
1797 t2.dp[1] = (i - 1 < 0) ? 0 : x.dp[i - 1];
1798 t2.dp[2] = x.dp[i];
1799 t2.used = 3;
1800 } while (mp_cmp_mag(&t1, &t2) == MP_GT);
1802 /* step 3.3 x = x - q{i-t-1} * y * b**{i-t-1} */
1803 if ((res = mp_mul_d (&y, q.dp[i - t - 1], &t1)) != MP_OKAY) {
1804 goto LBL_Y;
1807 if ((res = mp_lshd (&t1, i - t - 1)) != MP_OKAY) {
1808 goto LBL_Y;
1811 if ((res = mp_sub (&x, &t1, &x)) != MP_OKAY) {
1812 goto LBL_Y;
1815 /* if x < 0 then { x = x + y*b**{i-t-1}; q{i-t-1} -= 1; } */
1816 if (x.sign == MP_NEG) {
1817 if ((res = mp_copy (&y, &t1)) != MP_OKAY) {
1818 goto LBL_Y;
1820 if ((res = mp_lshd (&t1, i - t - 1)) != MP_OKAY) {
1821 goto LBL_Y;
1823 if ((res = mp_add (&x, &t1, &x)) != MP_OKAY) {
1824 goto LBL_Y;
1827 q.dp[i - t - 1] = (q.dp[i - t - 1] - 1UL) & MP_MASK;
1831 /* now q is the quotient and x is the remainder
1832 * [which we have to normalize]
1835 /* get sign before writing to c */
1836 x.sign = x.used == 0 ? MP_ZPOS : a->sign;
1838 if (c != NULL) {
1839 mp_clamp (&q);
1840 mp_exch (&q, c);
1841 c->sign = neg;
1844 if (d != NULL) {
1845 mp_div_2d (&x, norm, &x, NULL);
1846 mp_exch (&x, d);
1849 res = MP_OKAY;
1851 LBL_Y:mp_clear (&y);
1852 LBL_X:mp_clear (&x);
1853 LBL_T2:mp_clear (&t2);
1854 LBL_T1:mp_clear (&t1);
1855 LBL_Q:mp_clear (&q);
1856 return res;
1859 #endif
1862 #ifdef MP_LOW_MEM
1863 #define TAB_SIZE 32
1864 #else
1865 #define TAB_SIZE 256
1866 #endif
1868 static int s_mp_exptmod (mp_int * G, mp_int * X, mp_int * P, mp_int * Y, int redmode)
1870 mp_int M[TAB_SIZE], res, mu;
1871 mp_digit buf;
1872 int err, bitbuf, bitcpy, bitcnt, mode, digidx, x, y, winsize;
1873 int (*redux)(mp_int*,mp_int*,mp_int*);
1875 /* find window size */
1876 x = mp_count_bits (X);
1877 if (x <= 7) {
1878 winsize = 2;
1879 } else if (x <= 36) {
1880 winsize = 3;
1881 } else if (x <= 140) {
1882 winsize = 4;
1883 } else if (x <= 450) {
1884 winsize = 5;
1885 } else if (x <= 1303) {
1886 winsize = 6;
1887 } else if (x <= 3529) {
1888 winsize = 7;
1889 } else {
1890 winsize = 8;
1893 #ifdef MP_LOW_MEM
1894 if (winsize > 5) {
1895 winsize = 5;
1897 #endif
1899 /* init M array */
1900 /* init first cell */
1901 if ((err = mp_init(&M[1])) != MP_OKAY) {
1902 return err;
1905 /* now init the second half of the array */
1906 for (x = 1<<(winsize-1); x < (1 << winsize); x++) {
1907 if ((err = mp_init(&M[x])) != MP_OKAY) {
1908 for (y = 1<<(winsize-1); y < x; y++) {
1909 mp_clear (&M[y]);
1911 mp_clear(&M[1]);
1912 return err;
1916 /* create mu, used for Barrett reduction */
1917 if ((err = mp_init (&mu)) != MP_OKAY) {
1918 goto LBL_M;
1921 if (redmode == 0) {
1922 if ((err = mp_reduce_setup (&mu, P)) != MP_OKAY) {
1923 goto LBL_MU;
1925 redux = mp_reduce;
1926 } else {
1927 if ((err = mp_reduce_2k_setup_l (P, &mu)) != MP_OKAY) {
1928 goto LBL_MU;
1930 redux = mp_reduce_2k_l;
1933 /* create M table
1935 * The M table contains powers of the base,
1936 * e.g. M[x] = G**x mod P
1938 * The first half of the table is not
1939 * computed though accept for M[0] and M[1]
1941 if ((err = mp_mod (G, P, &M[1])) != MP_OKAY) {
1942 goto LBL_MU;
1945 /* compute the value at M[1<<(winsize-1)] by squaring
1946 * M[1] (winsize-1) times
1948 if ((err = mp_copy (&M[1], &M[1 << (winsize - 1)])) != MP_OKAY) {
1949 goto LBL_MU;
1952 for (x = 0; x < (winsize - 1); x++) {
1953 /* square it */
1954 if ((err = mp_sqr (&M[1 << (winsize - 1)],
1955 &M[1 << (winsize - 1)])) != MP_OKAY) {
1956 goto LBL_MU;
1959 /* reduce modulo P */
1960 if ((err = redux (&M[1 << (winsize - 1)], P, &mu)) != MP_OKAY) {
1961 goto LBL_MU;
1965 /* create upper table, that is M[x] = M[x-1] * M[1] (mod P)
1966 * for x = (2**(winsize - 1) + 1) to (2**winsize - 1)
1968 for (x = (1 << (winsize - 1)) + 1; x < (1 << winsize); x++) {
1969 if ((err = mp_mul (&M[x - 1], &M[1], &M[x])) != MP_OKAY) {
1970 goto LBL_MU;
1972 if ((err = redux (&M[x], P, &mu)) != MP_OKAY) {
1973 goto LBL_MU;
1977 /* setup result */
1978 if ((err = mp_init (&res)) != MP_OKAY) {
1979 goto LBL_MU;
1981 mp_set (&res, 1);
1983 /* set initial mode and bit cnt */
1984 mode = 0;
1985 bitcnt = 1;
1986 buf = 0;
1987 digidx = X->used - 1;
1988 bitcpy = 0;
1989 bitbuf = 0;
1991 for (;;) {
1992 /* grab next digit as required */
1993 if (--bitcnt == 0) {
1994 /* if digidx == -1 we are out of digits */
1995 if (digidx == -1) {
1996 break;
1998 /* read next digit and reset the bitcnt */
1999 buf = X->dp[digidx--];
2000 bitcnt = (int) DIGIT_BIT;
2003 /* grab the next msb from the exponent */
2004 y = (buf >> (mp_digit)(DIGIT_BIT - 1)) & 1;
2005 buf <<= (mp_digit)1;
2007 /* if the bit is zero and mode == 0 then we ignore it
2008 * These represent the leading zero bits before the first 1 bit
2009 * in the exponent. Technically this opt is not required but it
2010 * does lower the # of trivial squaring/reductions used
2012 if (mode == 0 && y == 0) {
2013 continue;
2016 /* if the bit is zero and mode == 1 then we square */
2017 if (mode == 1 && y == 0) {
2018 if ((err = mp_sqr (&res, &res)) != MP_OKAY) {
2019 goto LBL_RES;
2021 if ((err = redux (&res, P, &mu)) != MP_OKAY) {
2022 goto LBL_RES;
2024 continue;
2027 /* else we add it to the window */
2028 bitbuf |= (y << (winsize - ++bitcpy));
2029 mode = 2;
2031 if (bitcpy == winsize) {
2032 /* ok window is filled so square as required and multiply */
2033 /* square first */
2034 for (x = 0; x < winsize; x++) {
2035 if ((err = mp_sqr (&res, &res)) != MP_OKAY) {
2036 goto LBL_RES;
2038 if ((err = redux (&res, P, &mu)) != MP_OKAY) {
2039 goto LBL_RES;
2043 /* then multiply */
2044 if ((err = mp_mul (&res, &M[bitbuf], &res)) != MP_OKAY) {
2045 goto LBL_RES;
2047 if ((err = redux (&res, P, &mu)) != MP_OKAY) {
2048 goto LBL_RES;
2051 /* empty window and reset */
2052 bitcpy = 0;
2053 bitbuf = 0;
2054 mode = 1;
2058 /* if bits remain then square/multiply */
2059 if (mode == 2 && bitcpy > 0) {
2060 /* square then multiply if the bit is set */
2061 for (x = 0; x < bitcpy; x++) {
2062 if ((err = mp_sqr (&res, &res)) != MP_OKAY) {
2063 goto LBL_RES;
2065 if ((err = redux (&res, P, &mu)) != MP_OKAY) {
2066 goto LBL_RES;
2069 bitbuf <<= 1;
2070 if ((bitbuf & (1 << winsize)) != 0) {
2071 /* then multiply */
2072 if ((err = mp_mul (&res, &M[1], &res)) != MP_OKAY) {
2073 goto LBL_RES;
2075 if ((err = redux (&res, P, &mu)) != MP_OKAY) {
2076 goto LBL_RES;
2082 mp_exch (&res, Y);
2083 err = MP_OKAY;
2084 LBL_RES:mp_clear (&res);
2085 LBL_MU:mp_clear (&mu);
2086 LBL_M:
2087 mp_clear(&M[1]);
2088 for (x = 1<<(winsize-1); x < (1 << winsize); x++) {
2089 mp_clear (&M[x]);
2091 return err;
2095 /* computes b = a*a */
2096 static int mp_sqr (mp_int * a, mp_int * b)
2098 int res;
2100 #ifdef BN_MP_TOOM_SQR_C
2101 /* use Toom-Cook? */
2102 if (a->used >= TOOM_SQR_CUTOFF) {
2103 res = mp_toom_sqr(a, b);
2104 /* Karatsuba? */
2105 } else
2106 #endif
2107 #ifdef BN_MP_KARATSUBA_SQR_C
2108 if (a->used >= KARATSUBA_SQR_CUTOFF) {
2109 res = mp_karatsuba_sqr (a, b);
2110 } else
2111 #endif
2113 #ifdef BN_FAST_S_MP_SQR_C
2114 /* can we use the fast comba multiplier? */
2115 if ((a->used * 2 + 1) < MP_WARRAY &&
2116 a->used <
2117 (1 << (sizeof(mp_word) * CHAR_BIT - 2*DIGIT_BIT - 1))) {
2118 res = fast_s_mp_sqr (a, b);
2119 } else
2120 #endif
2121 #ifdef BN_S_MP_SQR_C
2122 res = s_mp_sqr (a, b);
2123 #else
2124 #error mp_sqr could fail
2125 res = MP_VAL;
2126 #endif
2128 b->sign = MP_ZPOS;
2129 return res;
2133 /* reduces a modulo n where n is of the form 2**p - d
2134 This differs from reduce_2k since "d" can be larger
2135 than a single digit.
2137 static int mp_reduce_2k_l(mp_int *a, mp_int *n, mp_int *d)
2139 mp_int q;
2140 int p, res;
2142 if ((res = mp_init(&q)) != MP_OKAY) {
2143 return res;
2146 p = mp_count_bits(n);
2147 top:
2148 /* q = a/2**p, a = a mod 2**p */
2149 if ((res = mp_div_2d(a, p, &q, a)) != MP_OKAY) {
2150 goto ERR;
2153 /* q = q * d */
2154 if ((res = mp_mul(&q, d, &q)) != MP_OKAY) {
2155 goto ERR;
2158 /* a = a + q */
2159 if ((res = s_mp_add(a, &q, a)) != MP_OKAY) {
2160 goto ERR;
2163 if (mp_cmp_mag(a, n) != MP_LT) {
2164 s_mp_sub(a, n, a);
2165 goto top;
2168 ERR:
2169 mp_clear(&q);
2170 return res;
2174 /* determines the setup value */
2175 static int mp_reduce_2k_setup_l(mp_int *a, mp_int *d)
2177 int res;
2178 mp_int tmp;
2180 if ((res = mp_init(&tmp)) != MP_OKAY) {
2181 return res;
2184 if ((res = mp_2expt(&tmp, mp_count_bits(a))) != MP_OKAY) {
2185 goto ERR;
2188 if ((res = s_mp_sub(&tmp, a, d)) != MP_OKAY) {
2189 goto ERR;
2192 ERR:
2193 mp_clear(&tmp);
2194 return res;
2198 /* computes a = 2**b
2200 * Simple algorithm which zeroes the int, grows it then just sets one bit
2201 * as required.
2203 static int mp_2expt (mp_int * a, int b)
2205 int res;
2207 /* zero a as per default */
2208 mp_zero (a);
2210 /* grow a to accomodate the single bit */
2211 if ((res = mp_grow (a, b / DIGIT_BIT + 1)) != MP_OKAY) {
2212 return res;
2215 /* set the used count of where the bit will go */
2216 a->used = b / DIGIT_BIT + 1;
2218 /* put the single bit in its place */
2219 a->dp[b / DIGIT_BIT] = ((mp_digit)1) << (b % DIGIT_BIT);
2221 return MP_OKAY;
2225 /* pre-calculate the value required for Barrett reduction
2226 * For a given modulus "b" it calulates the value required in "a"
2228 static int mp_reduce_setup (mp_int * a, mp_int * b)
2230 int res;
2232 if ((res = mp_2expt (a, b->used * 2 * DIGIT_BIT)) != MP_OKAY) {
2233 return res;
2235 return mp_div (a, b, a, NULL);
2239 /* reduces x mod m, assumes 0 < x < m**2, mu is
2240 * precomputed via mp_reduce_setup.
2241 * From HAC pp.604 Algorithm 14.42
2243 static int mp_reduce (mp_int * x, mp_int * m, mp_int * mu)
2245 mp_int q;
2246 int res, um = m->used;
2248 /* q = x */
2249 if ((res = mp_init_copy (&q, x)) != MP_OKAY) {
2250 return res;
2253 /* q1 = x / b**(k-1) */
2254 mp_rshd (&q, um - 1);
2256 /* according to HAC this optimization is ok */
2257 if (((unsigned long) um) > (((mp_digit)1) << (DIGIT_BIT - 1))) {
2258 if ((res = mp_mul (&q, mu, &q)) != MP_OKAY) {
2259 goto CLEANUP;
2261 } else {
2262 #ifdef BN_S_MP_MUL_HIGH_DIGS_C
2263 if ((res = s_mp_mul_high_digs (&q, mu, &q, um)) != MP_OKAY) {
2264 goto CLEANUP;
2266 #elif defined(BN_FAST_S_MP_MUL_HIGH_DIGS_C)
2267 if ((res = fast_s_mp_mul_high_digs (&q, mu, &q, um)) != MP_OKAY) {
2268 goto CLEANUP;
2270 #else
2272 #error mp_reduce would always fail
2273 res = MP_VAL;
2274 goto CLEANUP;
2276 #endif
2279 /* q3 = q2 / b**(k+1) */
2280 mp_rshd (&q, um + 1);
2282 /* x = x mod b**(k+1), quick (no division) */
2283 if ((res = mp_mod_2d (x, DIGIT_BIT * (um + 1), x)) != MP_OKAY) {
2284 goto CLEANUP;
2287 /* q = q * m mod b**(k+1), quick (no division) */
2288 if ((res = s_mp_mul_digs (&q, m, &q, um + 1)) != MP_OKAY) {
2289 goto CLEANUP;
2292 /* x = x - q */
2293 if ((res = mp_sub (x, &q, x)) != MP_OKAY) {
2294 goto CLEANUP;
2297 /* If x < 0, add b**(k+1) to it */
2298 if (mp_cmp_d (x, 0) == MP_LT) {
2299 mp_set (&q, 1);
2300 if ((res = mp_lshd (&q, um + 1)) != MP_OKAY) {
2301 goto CLEANUP;
2303 if ((res = mp_add (x, &q, x)) != MP_OKAY) {
2304 goto CLEANUP;
2308 /* Back off if it's too big */
2309 while (mp_cmp (x, m) != MP_LT) {
2310 if ((res = s_mp_sub (x, m, x)) != MP_OKAY) {
2311 goto CLEANUP;
2315 CLEANUP:
2316 mp_clear (&q);
2318 return res;
2322 /* multiplies |a| * |b| and only computes upto digs digits of result
2323 * HAC pp. 595, Algorithm 14.12 Modified so you can control how
2324 * many digits of output are created.
2326 static int s_mp_mul_digs (mp_int * a, mp_int * b, mp_int * c, int digs)
2328 mp_int t;
2329 int res, pa, pb, ix, iy;
2330 mp_digit u;
2331 mp_word r;
2332 mp_digit tmpx, *tmpt, *tmpy;
2334 /* can we use the fast multiplier? */
2335 if (((digs) < MP_WARRAY) &&
2336 MIN (a->used, b->used) <
2337 (1 << ((CHAR_BIT * sizeof (mp_word)) - (2 * DIGIT_BIT)))) {
2338 return fast_s_mp_mul_digs (a, b, c, digs);
2341 if ((res = mp_init_size (&t, digs)) != MP_OKAY) {
2342 return res;
2344 t.used = digs;
2346 /* compute the digits of the product directly */
2347 pa = a->used;
2348 for (ix = 0; ix < pa; ix++) {
2349 /* set the carry to zero */
2350 u = 0;
2352 /* limit ourselves to making digs digits of output */
2353 pb = MIN (b->used, digs - ix);
2355 /* setup some aliases */
2356 /* copy of the digit from a used within the nested loop */
2357 tmpx = a->dp[ix];
2359 /* an alias for the destination shifted ix places */
2360 tmpt = t.dp + ix;
2362 /* an alias for the digits of b */
2363 tmpy = b->dp;
2365 /* compute the columns of the output and propagate the carry */
2366 for (iy = 0; iy < pb; iy++) {
2367 /* compute the column as a mp_word */
2368 r = ((mp_word)*tmpt) +
2369 ((mp_word)tmpx) * ((mp_word)*tmpy++) +
2370 ((mp_word) u);
2372 /* the new column is the lower part of the result */
2373 *tmpt++ = (mp_digit) (r & ((mp_word) MP_MASK));
2375 /* get the carry word from the result */
2376 u = (mp_digit) (r >> ((mp_word) DIGIT_BIT));
2378 /* set carry if it is placed below digs */
2379 if (ix + iy < digs) {
2380 *tmpt = u;
2384 mp_clamp (&t);
2385 mp_exch (&t, c);
2387 mp_clear (&t);
2388 return MP_OKAY;
2392 /* Fast (comba) multiplier
2394 * This is the fast column-array [comba] multiplier. It is
2395 * designed to compute the columns of the product first
2396 * then handle the carries afterwards. This has the effect
2397 * of making the nested loops that compute the columns very
2398 * simple and schedulable on super-scalar processors.
2400 * This has been modified to produce a variable number of
2401 * digits of output so if say only a half-product is required
2402 * you don't have to compute the upper half (a feature
2403 * required for fast Barrett reduction).
2405 * Based on Algorithm 14.12 on pp.595 of HAC.
2408 static int fast_s_mp_mul_digs (mp_int * a, mp_int * b, mp_int * c, int digs)
2410 int olduse, res, pa, ix, iz;
2411 mp_digit W[MP_WARRAY];
2412 register mp_word _W;
2414 /* grow the destination as required */
2415 if (c->alloc < digs) {
2416 if ((res = mp_grow (c, digs)) != MP_OKAY) {
2417 return res;
2421 /* number of output digits to produce */
2422 pa = MIN(digs, a->used + b->used);
2424 /* clear the carry */
2425 _W = 0;
2426 for (ix = 0; ix < pa; ix++) {
2427 int tx, ty;
2428 int iy;
2429 mp_digit *tmpx, *tmpy;
2431 /* get offsets into the two bignums */
2432 ty = MIN(b->used-1, ix);
2433 tx = ix - ty;
2435 /* setup temp aliases */
2436 tmpx = a->dp + tx;
2437 tmpy = b->dp + ty;
2439 /* this is the number of times the loop will iterrate, essentially
2440 while (tx++ < a->used && ty-- >= 0) { ... }
2442 iy = MIN(a->used-tx, ty+1);
2444 /* execute loop */
2445 for (iz = 0; iz < iy; ++iz) {
2446 _W += ((mp_word)*tmpx++)*((mp_word)*tmpy--);
2450 /* store term */
2451 W[ix] = ((mp_digit)_W) & MP_MASK;
2453 /* make next carry */
2454 _W = _W >> ((mp_word)DIGIT_BIT);
2457 /* setup dest */
2458 olduse = c->used;
2459 c->used = pa;
2462 register mp_digit *tmpc;
2463 tmpc = c->dp;
2464 for (ix = 0; ix < pa+1; ix++) {
2465 /* now extract the previous digit [below the carry] */
2466 *tmpc++ = W[ix];
2469 /* clear unused digits [that existed in the old copy of c] */
2470 for (; ix < olduse; ix++) {
2471 *tmpc++ = 0;
2474 mp_clamp (c);
2475 return MP_OKAY;
2479 /* init an mp_init for a given size */
2480 static int mp_init_size (mp_int * a, int size)
2482 int x;
2484 /* pad size so there are always extra digits */
2485 size += (MP_PREC * 2) - (size % MP_PREC);
2487 /* alloc mem */
2488 a->dp = OPT_CAST(mp_digit) XMALLOC (sizeof (mp_digit) * size);
2489 if (a->dp == NULL) {
2490 return MP_MEM;
2493 /* set the members */
2494 a->used = 0;
2495 a->alloc = size;
2496 a->sign = MP_ZPOS;
2498 /* zero the digits */
2499 for (x = 0; x < size; x++) {
2500 a->dp[x] = 0;
2503 return MP_OKAY;
2507 /* low level squaring, b = a*a, HAC pp.596-597, Algorithm 14.16 */
2508 static int s_mp_sqr (mp_int * a, mp_int * b)
2510 mp_int t;
2511 int res, ix, iy, pa;
2512 mp_word r;
2513 mp_digit u, tmpx, *tmpt;
2515 pa = a->used;
2516 if ((res = mp_init_size (&t, 2*pa + 1)) != MP_OKAY) {
2517 return res;
2520 /* default used is maximum possible size */
2521 t.used = 2*pa + 1;
2523 for (ix = 0; ix < pa; ix++) {
2524 /* first calculate the digit at 2*ix */
2525 /* calculate double precision result */
2526 r = ((mp_word) t.dp[2*ix]) +
2527 ((mp_word)a->dp[ix])*((mp_word)a->dp[ix]);
2529 /* store lower part in result */
2530 t.dp[ix+ix] = (mp_digit) (r & ((mp_word) MP_MASK));
2532 /* get the carry */
2533 u = (mp_digit)(r >> ((mp_word) DIGIT_BIT));
2535 /* left hand side of A[ix] * A[iy] */
2536 tmpx = a->dp[ix];
2538 /* alias for where to store the results */
2539 tmpt = t.dp + (2*ix + 1);
2541 for (iy = ix + 1; iy < pa; iy++) {
2542 /* first calculate the product */
2543 r = ((mp_word)tmpx) * ((mp_word)a->dp[iy]);
2545 /* now calculate the double precision result, note we use
2546 * addition instead of *2 since it's easier to optimize
2548 r = ((mp_word) *tmpt) + r + r + ((mp_word) u);
2550 /* store lower part */
2551 *tmpt++ = (mp_digit) (r & ((mp_word) MP_MASK));
2553 /* get carry */
2554 u = (mp_digit)(r >> ((mp_word) DIGIT_BIT));
2556 /* propagate upwards */
2557 while (u != ((mp_digit) 0)) {
2558 r = ((mp_word) *tmpt) + ((mp_word) u);
2559 *tmpt++ = (mp_digit) (r & ((mp_word) MP_MASK));
2560 u = (mp_digit)(r >> ((mp_word) DIGIT_BIT));
2564 mp_clamp (&t);
2565 mp_exch (&t, b);
2566 mp_clear (&t);
2567 return MP_OKAY;
2571 /* multiplies |a| * |b| and does not compute the lower digs digits
2572 * [meant to get the higher part of the product]
2574 static int s_mp_mul_high_digs (mp_int * a, mp_int * b, mp_int * c, int digs)
2576 mp_int t;
2577 int res, pa, pb, ix, iy;
2578 mp_digit u;
2579 mp_word r;
2580 mp_digit tmpx, *tmpt, *tmpy;
2582 /* can we use the fast multiplier? */
2583 #ifdef BN_FAST_S_MP_MUL_HIGH_DIGS_C
2584 if (((a->used + b->used + 1) < MP_WARRAY)
2585 && MIN (a->used, b->used) < (1 << ((CHAR_BIT * sizeof (mp_word)) - (2 * DIGIT_BIT)))) {
2586 return fast_s_mp_mul_high_digs (a, b, c, digs);
2588 #endif
2590 if ((res = mp_init_size (&t, a->used + b->used + 1)) != MP_OKAY) {
2591 return res;
2593 t.used = a->used + b->used + 1;
2595 pa = a->used;
2596 pb = b->used;
2597 for (ix = 0; ix < pa; ix++) {
2598 /* clear the carry */
2599 u = 0;
2601 /* left hand side of A[ix] * B[iy] */
2602 tmpx = a->dp[ix];
2604 /* alias to the address of where the digits will be stored */
2605 tmpt = &(t.dp[digs]);
2607 /* alias for where to read the right hand side from */
2608 tmpy = b->dp + (digs - ix);
2610 for (iy = digs - ix; iy < pb; iy++) {
2611 /* calculate the double precision result */
2612 r = ((mp_word)*tmpt) +
2613 ((mp_word)tmpx) * ((mp_word)*tmpy++) +
2614 ((mp_word) u);
2616 /* get the lower part */
2617 *tmpt++ = (mp_digit) (r & ((mp_word) MP_MASK));
2619 /* carry the carry */
2620 u = (mp_digit) (r >> ((mp_word) DIGIT_BIT));
2622 *tmpt = u;
2624 mp_clamp (&t);
2625 mp_exch (&t, c);
2626 mp_clear (&t);
2627 return MP_OKAY;
2631 #ifdef BN_MP_MONTGOMERY_SETUP_C
2632 /* setups the montgomery reduction stuff */
2633 static int
2634 mp_montgomery_setup (mp_int * n, mp_digit * rho)
2636 mp_digit x, b;
2638 /* fast inversion mod 2**k
2640 * Based on the fact that
2642 * XA = 1 (mod 2**n) => (X(2-XA)) A = 1 (mod 2**2n)
2643 * => 2*X*A - X*X*A*A = 1
2644 * => 2*(1) - (1) = 1
2646 b = n->dp[0];
2648 if ((b & 1) == 0) {
2649 return MP_VAL;
2652 x = (((b + 2) & 4) << 1) + b; /* here x*a==1 mod 2**4 */
2653 x *= 2 - b * x; /* here x*a==1 mod 2**8 */
2654 #if !defined(MP_8BIT)
2655 x *= 2 - b * x; /* here x*a==1 mod 2**16 */
2656 #endif
2657 #if defined(MP_64BIT) || !(defined(MP_8BIT) || defined(MP_16BIT))
2658 x *= 2 - b * x; /* here x*a==1 mod 2**32 */
2659 #endif
2660 #ifdef MP_64BIT
2661 x *= 2 - b * x; /* here x*a==1 mod 2**64 */
2662 #endif
2664 /* rho = -1/m mod b */
2665 *rho = (unsigned long)(((mp_word)1 << ((mp_word) DIGIT_BIT)) - x) & MP_MASK;
2667 return MP_OKAY;
2669 #endif
2672 #ifdef BN_FAST_MP_MONTGOMERY_REDUCE_C
2673 /* computes xR**-1 == x (mod N) via Montgomery Reduction
2675 * This is an optimized implementation of montgomery_reduce
2676 * which uses the comba method to quickly calculate the columns of the
2677 * reduction.
2679 * Based on Algorithm 14.32 on pp.601 of HAC.
2681 int fast_mp_montgomery_reduce (mp_int * x, mp_int * n, mp_digit rho)
2683 int ix, res, olduse;
2684 mp_word W[MP_WARRAY];
2686 /* get old used count */
2687 olduse = x->used;
2689 /* grow a as required */
2690 if (x->alloc < n->used + 1) {
2691 if ((res = mp_grow (x, n->used + 1)) != MP_OKAY) {
2692 return res;
2696 /* first we have to get the digits of the input into
2697 * an array of double precision words W[...]
2700 register mp_word *_W;
2701 register mp_digit *tmpx;
2703 /* alias for the W[] array */
2704 _W = W;
2706 /* alias for the digits of x*/
2707 tmpx = x->dp;
2709 /* copy the digits of a into W[0..a->used-1] */
2710 for (ix = 0; ix < x->used; ix++) {
2711 *_W++ = *tmpx++;
2714 /* zero the high words of W[a->used..m->used*2] */
2715 for (; ix < n->used * 2 + 1; ix++) {
2716 *_W++ = 0;
2720 /* now we proceed to zero successive digits
2721 * from the least significant upwards
2723 for (ix = 0; ix < n->used; ix++) {
2724 /* mu = ai * m' mod b
2726 * We avoid a double precision multiplication (which isn't required)
2727 * by casting the value down to a mp_digit. Note this requires
2728 * that W[ix-1] have the carry cleared (see after the inner loop)
2730 register mp_digit mu;
2731 mu = (mp_digit) (((W[ix] & MP_MASK) * rho) & MP_MASK);
2733 /* a = a + mu * m * b**i
2735 * This is computed in place and on the fly. The multiplication
2736 * by b**i is handled by offseting which columns the results
2737 * are added to.
2739 * Note the comba method normally doesn't handle carries in the
2740 * inner loop In this case we fix the carry from the previous
2741 * column since the Montgomery reduction requires digits of the
2742 * result (so far) [see above] to work. This is
2743 * handled by fixing up one carry after the inner loop. The
2744 * carry fixups are done in order so after these loops the
2745 * first m->used words of W[] have the carries fixed
2748 register int iy;
2749 register mp_digit *tmpn;
2750 register mp_word *_W;
2752 /* alias for the digits of the modulus */
2753 tmpn = n->dp;
2755 /* Alias for the columns set by an offset of ix */
2756 _W = W + ix;
2758 /* inner loop */
2759 for (iy = 0; iy < n->used; iy++) {
2760 *_W++ += ((mp_word)mu) * ((mp_word)*tmpn++);
2764 /* now fix carry for next digit, W[ix+1] */
2765 W[ix + 1] += W[ix] >> ((mp_word) DIGIT_BIT);
2768 /* now we have to propagate the carries and
2769 * shift the words downward [all those least
2770 * significant digits we zeroed].
2773 register mp_digit *tmpx;
2774 register mp_word *_W, *_W1;
2776 /* nox fix rest of carries */
2778 /* alias for current word */
2779 _W1 = W + ix;
2781 /* alias for next word, where the carry goes */
2782 _W = W + ++ix;
2784 for (; ix <= n->used * 2 + 1; ix++) {
2785 *_W++ += *_W1++ >> ((mp_word) DIGIT_BIT);
2788 /* copy out, A = A/b**n
2790 * The result is A/b**n but instead of converting from an
2791 * array of mp_word to mp_digit than calling mp_rshd
2792 * we just copy them in the right order
2795 /* alias for destination word */
2796 tmpx = x->dp;
2798 /* alias for shifted double precision result */
2799 _W = W + n->used;
2801 for (ix = 0; ix < n->used + 1; ix++) {
2802 *tmpx++ = (mp_digit)(*_W++ & ((mp_word) MP_MASK));
2805 /* zero oldused digits, if the input a was larger than
2806 * m->used+1 we'll have to clear the digits
2808 for (; ix < olduse; ix++) {
2809 *tmpx++ = 0;
2813 /* set the max used and clamp */
2814 x->used = n->used + 1;
2815 mp_clamp (x);
2817 /* if A >= m then A = A - m */
2818 if (mp_cmp_mag (x, n) != MP_LT) {
2819 return s_mp_sub (x, n, x);
2821 return MP_OKAY;
2823 #endif
2826 #ifdef BN_MP_MUL_2_C
2827 /* b = a*2 */
2828 static int mp_mul_2(mp_int * a, mp_int * b)
2830 int x, res, oldused;
2832 /* grow to accomodate result */
2833 if (b->alloc < a->used + 1) {
2834 if ((res = mp_grow (b, a->used + 1)) != MP_OKAY) {
2835 return res;
2839 oldused = b->used;
2840 b->used = a->used;
2843 register mp_digit r, rr, *tmpa, *tmpb;
2845 /* alias for source */
2846 tmpa = a->dp;
2848 /* alias for dest */
2849 tmpb = b->dp;
2851 /* carry */
2852 r = 0;
2853 for (x = 0; x < a->used; x++) {
2855 /* get what will be the *next* carry bit from the
2856 * MSB of the current digit
2858 rr = *tmpa >> ((mp_digit)(DIGIT_BIT - 1));
2860 /* now shift up this digit, add in the carry [from the previous] */
2861 *tmpb++ = ((*tmpa++ << ((mp_digit)1)) | r) & MP_MASK;
2863 /* copy the carry that would be from the source
2864 * digit into the next iteration
2866 r = rr;
2869 /* new leading digit? */
2870 if (r != 0) {
2871 /* add a MSB which is always 1 at this point */
2872 *tmpb = 1;
2873 ++(b->used);
2876 /* now zero any excess digits on the destination
2877 * that we didn't write to
2879 tmpb = b->dp + b->used;
2880 for (x = b->used; x < oldused; x++) {
2881 *tmpb++ = 0;
2884 b->sign = a->sign;
2885 return MP_OKAY;
2887 #endif
2890 #ifdef BN_MP_MONTGOMERY_CALC_NORMALIZATION_C
2892 * shifts with subtractions when the result is greater than b.
2894 * The method is slightly modified to shift B unconditionally upto just under
2895 * the leading bit of b. This saves alot of multiple precision shifting.
2897 static int mp_montgomery_calc_normalization (mp_int * a, mp_int * b)
2899 int x, bits, res;
2901 /* how many bits of last digit does b use */
2902 bits = mp_count_bits (b) % DIGIT_BIT;
2904 if (b->used > 1) {
2905 if ((res = mp_2expt (a, (b->used - 1) * DIGIT_BIT + bits - 1)) != MP_OKAY) {
2906 return res;
2908 } else {
2909 mp_set(a, 1);
2910 bits = 1;
2914 /* now compute C = A * B mod b */
2915 for (x = bits - 1; x < (int)DIGIT_BIT; x++) {
2916 if ((res = mp_mul_2 (a, a)) != MP_OKAY) {
2917 return res;
2919 if (mp_cmp_mag (a, b) != MP_LT) {
2920 if ((res = s_mp_sub (a, b, a)) != MP_OKAY) {
2921 return res;
2926 return MP_OKAY;
2928 #endif
2931 #ifdef BN_MP_EXPTMOD_FAST_C
2932 /* computes Y == G**X mod P, HAC pp.616, Algorithm 14.85
2934 * Uses a left-to-right k-ary sliding window to compute the modular exponentiation.
2935 * The value of k changes based on the size of the exponent.
2937 * Uses Montgomery or Diminished Radix reduction [whichever appropriate]
2940 static int mp_exptmod_fast (mp_int * G, mp_int * X, mp_int * P, mp_int * Y, int redmode)
2942 mp_int M[TAB_SIZE], res;
2943 mp_digit buf, mp;
2944 int err, bitbuf, bitcpy, bitcnt, mode, digidx, x, y, winsize;
2946 /* use a pointer to the reduction algorithm. This allows us to use
2947 * one of many reduction algorithms without modding the guts of
2948 * the code with if statements everywhere.
2950 int (*redux)(mp_int*,mp_int*,mp_digit);
2952 /* find window size */
2953 x = mp_count_bits (X);
2954 if (x <= 7) {
2955 winsize = 2;
2956 } else if (x <= 36) {
2957 winsize = 3;
2958 } else if (x <= 140) {
2959 winsize = 4;
2960 } else if (x <= 450) {
2961 winsize = 5;
2962 } else if (x <= 1303) {
2963 winsize = 6;
2964 } else if (x <= 3529) {
2965 winsize = 7;
2966 } else {
2967 winsize = 8;
2970 #ifdef MP_LOW_MEM
2971 if (winsize > 5) {
2972 winsize = 5;
2974 #endif
2976 /* init M array */
2977 /* init first cell */
2978 if ((err = mp_init(&M[1])) != MP_OKAY) {
2979 return err;
2982 /* now init the second half of the array */
2983 for (x = 1<<(winsize-1); x < (1 << winsize); x++) {
2984 if ((err = mp_init(&M[x])) != MP_OKAY) {
2985 for (y = 1<<(winsize-1); y < x; y++) {
2986 mp_clear (&M[y]);
2988 mp_clear(&M[1]);
2989 return err;
2993 /* determine and setup reduction code */
2994 if (redmode == 0) {
2995 #ifdef BN_MP_MONTGOMERY_SETUP_C
2996 /* now setup montgomery */
2997 if ((err = mp_montgomery_setup (P, &mp)) != MP_OKAY) {
2998 goto LBL_M;
3000 #else
3001 err = MP_VAL;
3002 goto LBL_M;
3003 #endif
3005 /* automatically pick the comba one if available (saves quite a few calls/ifs) */
3006 #ifdef BN_FAST_MP_MONTGOMERY_REDUCE_C
3007 if (((P->used * 2 + 1) < MP_WARRAY) &&
3008 P->used < (1 << ((CHAR_BIT * sizeof (mp_word)) - (2 * DIGIT_BIT)))) {
3009 redux = fast_mp_montgomery_reduce;
3010 } else
3011 #endif
3013 #ifdef BN_MP_MONTGOMERY_REDUCE_C
3014 /* use slower baseline Montgomery method */
3015 redux = mp_montgomery_reduce;
3016 #else
3017 err = MP_VAL;
3018 goto LBL_M;
3019 #endif
3021 } else if (redmode == 1) {
3022 #if defined(BN_MP_DR_SETUP_C) && defined(BN_MP_DR_REDUCE_C)
3023 /* setup DR reduction for moduli of the form B**k - b */
3024 mp_dr_setup(P, &mp);
3025 redux = mp_dr_reduce;
3026 #else
3027 err = MP_VAL;
3028 goto LBL_M;
3029 #endif
3030 } else {
3031 #if defined(BN_MP_REDUCE_2K_SETUP_C) && defined(BN_MP_REDUCE_2K_C)
3032 /* setup DR reduction for moduli of the form 2**k - b */
3033 if ((err = mp_reduce_2k_setup(P, &mp)) != MP_OKAY) {
3034 goto LBL_M;
3036 redux = mp_reduce_2k;
3037 #else
3038 err = MP_VAL;
3039 goto LBL_M;
3040 #endif
3043 /* setup result */
3044 if ((err = mp_init (&res)) != MP_OKAY) {
3045 goto LBL_M;
3048 /* create M table
3052 * The first half of the table is not computed though accept for M[0] and M[1]
3055 if (redmode == 0) {
3056 #ifdef BN_MP_MONTGOMERY_CALC_NORMALIZATION_C
3057 /* now we need R mod m */
3058 if ((err = mp_montgomery_calc_normalization (&res, P)) != MP_OKAY) {
3059 goto LBL_RES;
3061 #else
3062 err = MP_VAL;
3063 goto LBL_RES;
3064 #endif
3066 /* now set M[1] to G * R mod m */
3067 if ((err = mp_mulmod (G, &res, P, &M[1])) != MP_OKAY) {
3068 goto LBL_RES;
3070 } else {
3071 mp_set(&res, 1);
3072 if ((err = mp_mod(G, P, &M[1])) != MP_OKAY) {
3073 goto LBL_RES;
3077 /* compute the value at M[1<<(winsize-1)] by squaring M[1] (winsize-1) times */
3078 if ((err = mp_copy (&M[1], &M[1 << (winsize - 1)])) != MP_OKAY) {
3079 goto LBL_RES;
3082 for (x = 0; x < (winsize - 1); x++) {
3083 if ((err = mp_sqr (&M[1 << (winsize - 1)], &M[1 << (winsize - 1)])) != MP_OKAY) {
3084 goto LBL_RES;
3086 if ((err = redux (&M[1 << (winsize - 1)], P, mp)) != MP_OKAY) {
3087 goto LBL_RES;
3091 /* create upper table */
3092 for (x = (1 << (winsize - 1)) + 1; x < (1 << winsize); x++) {
3093 if ((err = mp_mul (&M[x - 1], &M[1], &M[x])) != MP_OKAY) {
3094 goto LBL_RES;
3096 if ((err = redux (&M[x], P, mp)) != MP_OKAY) {
3097 goto LBL_RES;
3101 /* set initial mode and bit cnt */
3102 mode = 0;
3103 bitcnt = 1;
3104 buf = 0;
3105 digidx = X->used - 1;
3106 bitcpy = 0;
3107 bitbuf = 0;
3109 for (;;) {
3110 /* grab next digit as required */
3111 if (--bitcnt == 0) {
3112 /* if digidx == -1 we are out of digits so break */
3113 if (digidx == -1) {
3114 break;
3116 /* read next digit and reset bitcnt */
3117 buf = X->dp[digidx--];
3118 bitcnt = (int)DIGIT_BIT;
3121 /* grab the next msb from the exponent */
3122 y = (mp_digit)(buf >> (DIGIT_BIT - 1)) & 1;
3123 buf <<= (mp_digit)1;
3125 /* if the bit is zero and mode == 0 then we ignore it
3126 * These represent the leading zero bits before the first 1 bit
3127 * in the exponent. Technically this opt is not required but it
3128 * does lower the # of trivial squaring/reductions used
3130 if (mode == 0 && y == 0) {
3131 continue;
3134 /* if the bit is zero and mode == 1 then we square */
3135 if (mode == 1 && y == 0) {
3136 if ((err = mp_sqr (&res, &res)) != MP_OKAY) {
3137 goto LBL_RES;
3139 if ((err = redux (&res, P, mp)) != MP_OKAY) {
3140 goto LBL_RES;
3142 continue;
3145 /* else we add it to the window */
3146 bitbuf |= (y << (winsize - ++bitcpy));
3147 mode = 2;
3149 if (bitcpy == winsize) {
3150 /* ok window is filled so square as required and multiply */
3151 /* square first */
3152 for (x = 0; x < winsize; x++) {
3153 if ((err = mp_sqr (&res, &res)) != MP_OKAY) {
3154 goto LBL_RES;
3156 if ((err = redux (&res, P, mp)) != MP_OKAY) {
3157 goto LBL_RES;
3161 /* then multiply */
3162 if ((err = mp_mul (&res, &M[bitbuf], &res)) != MP_OKAY) {
3163 goto LBL_RES;
3165 if ((err = redux (&res, P, mp)) != MP_OKAY) {
3166 goto LBL_RES;
3169 /* empty window and reset */
3170 bitcpy = 0;
3171 bitbuf = 0;
3172 mode = 1;
3176 /* if bits remain then square/multiply */
3177 if (mode == 2 && bitcpy > 0) {
3178 /* square then multiply if the bit is set */
3179 for (x = 0; x < bitcpy; x++) {
3180 if ((err = mp_sqr (&res, &res)) != MP_OKAY) {
3181 goto LBL_RES;
3183 if ((err = redux (&res, P, mp)) != MP_OKAY) {
3184 goto LBL_RES;
3187 /* get next bit of the window */
3188 bitbuf <<= 1;
3189 if ((bitbuf & (1 << winsize)) != 0) {
3190 /* then multiply */
3191 if ((err = mp_mul (&res, &M[1], &res)) != MP_OKAY) {
3192 goto LBL_RES;
3194 if ((err = redux (&res, P, mp)) != MP_OKAY) {
3195 goto LBL_RES;
3201 if (redmode == 0) {
3202 /* fixup result if Montgomery reduction is used
3203 * recall that any value in a Montgomery system is
3204 * actually multiplied by R mod n. So we have
3205 * to reduce one more time to cancel out the factor
3206 * of R.
3208 if ((err = redux(&res, P, mp)) != MP_OKAY) {
3209 goto LBL_RES;
3213 /* swap res with Y */
3214 mp_exch (&res, Y);
3215 err = MP_OKAY;
3216 LBL_RES:mp_clear (&res);
3217 LBL_M:
3218 mp_clear(&M[1]);
3219 for (x = 1<<(winsize-1); x < (1 << winsize); x++) {
3220 mp_clear (&M[x]);
3222 return err;
3224 #endif
3227 #ifdef BN_FAST_S_MP_SQR_C
3228 /* the jist of squaring...
3229 * you do like mult except the offset of the tmpx [one that
3230 * starts closer to zero] can't equal the offset of tmpy.
3231 * So basically you set up iy like before then you min it with
3232 * (ty-tx) so that it never happens. You double all those
3233 * you add in the inner loop
3235 After that loop you do the squares and add them in.
3238 static int fast_s_mp_sqr (mp_int * a, mp_int * b)
3240 int olduse, res, pa, ix, iz;
3241 mp_digit W[MP_WARRAY], *tmpx;
3242 mp_word W1;
3244 /* grow the destination as required */
3245 pa = a->used + a->used;
3246 if (b->alloc < pa) {
3247 if ((res = mp_grow (b, pa)) != MP_OKAY) {
3248 return res;
3252 /* number of output digits to produce */
3253 W1 = 0;
3254 for (ix = 0; ix < pa; ix++) {
3255 int tx, ty, iy;
3256 mp_word _W;
3257 mp_digit *tmpy;
3259 /* clear counter */
3260 _W = 0;
3262 /* get offsets into the two bignums */
3263 ty = MIN(a->used-1, ix);
3264 tx = ix - ty;
3266 /* setup temp aliases */
3267 tmpx = a->dp + tx;
3268 tmpy = a->dp + ty;
3270 /* this is the number of times the loop will iterrate, essentially
3271 while (tx++ < a->used && ty-- >= 0) { ... }
3273 iy = MIN(a->used-tx, ty+1);
3275 /* now for squaring tx can never equal ty
3276 * we halve the distance since they approach at a rate of 2x
3277 * and we have to round because odd cases need to be executed
3279 iy = MIN(iy, (ty-tx+1)>>1);
3281 /* execute loop */
3282 for (iz = 0; iz < iy; iz++) {
3283 _W += ((mp_word)*tmpx++)*((mp_word)*tmpy--);
3286 /* double the inner product and add carry */
3287 _W = _W + _W + W1;
3289 /* even columns have the square term in them */
3290 if ((ix&1) == 0) {
3291 _W += ((mp_word)a->dp[ix>>1])*((mp_word)a->dp[ix>>1]);
3294 /* store it */
3295 W[ix] = (mp_digit)(_W & MP_MASK);
3297 /* make next carry */
3298 W1 = _W >> ((mp_word)DIGIT_BIT);
3301 /* setup dest */
3302 olduse = b->used;
3303 b->used = a->used+a->used;
3306 mp_digit *tmpb;
3307 tmpb = b->dp;
3308 for (ix = 0; ix < pa; ix++) {
3309 *tmpb++ = W[ix] & MP_MASK;
3312 /* clear unused digits [that existed in the old copy of c] */
3313 for (; ix < olduse; ix++) {
3314 *tmpb++ = 0;
3317 mp_clamp (b);
3318 return MP_OKAY;
3320 #endif
3323 #ifdef BN_MP_MUL_D_C
3324 /* multiply by a digit */
3325 static int
3326 mp_mul_d (mp_int * a, mp_digit b, mp_int * c)
3328 mp_digit u, *tmpa, *tmpc;
3329 mp_word r;
3330 int ix, res, olduse;
3332 /* make sure c is big enough to hold a*b */
3333 if (c->alloc < a->used + 1) {
3334 if ((res = mp_grow (c, a->used + 1)) != MP_OKAY) {
3335 return res;
3339 /* get the original destinations used count */
3340 olduse = c->used;
3342 /* set the sign */
3343 c->sign = a->sign;
3345 /* alias for a->dp [source] */
3346 tmpa = a->dp;
3348 /* alias for c->dp [dest] */
3349 tmpc = c->dp;
3351 /* zero carry */
3352 u = 0;
3354 /* compute columns */
3355 for (ix = 0; ix < a->used; ix++) {
3356 /* compute product and carry sum for this term */
3357 r = ((mp_word) u) + ((mp_word)*tmpa++) * ((mp_word)b);
3359 /* mask off higher bits to get a single digit */
3360 *tmpc++ = (mp_digit) (r & ((mp_word) MP_MASK));
3362 /* send carry into next iteration */
3363 u = (mp_digit) (r >> ((mp_word) DIGIT_BIT));
3366 /* store final carry [if any] and increment ix offset */
3367 *tmpc++ = u;
3368 ++ix;
3370 /* now zero digits above the top */
3371 while (ix++ < olduse) {
3372 *tmpc++ = 0;
3375 /* set used count */
3376 c->used = a->used + 1;
3377 mp_clamp(c);
3379 return MP_OKAY;
3381 #endif