integer.h 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319
  1. /* -*- mode:c; c-file-style:"k&r"; c-basic-offset: 4; tab-width:4; indent-tabs-mode:nil; mode:auto-fill; fill-column:78; -*- */
  2. /* vim: set ts=4 sw=4 et tw=78 fo=cqt wm=0: */
  3. /* integer.h
  4. *
  5. * Copyright (C) 2006-2014 wolfSSL Inc.
  6. *
  7. * This file is part of CyaSSL.
  8. *
  9. * CyaSSL is free software; you can redistribute it and/or modify
  10. * it under the terms of the GNU General Public License as published by
  11. * the Free Software Foundation; either version 2 of the License, or
  12. * (at your option) any later version.
  13. *
  14. * CyaSSL is distributed in the hope that it will be useful,
  15. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17. * GNU General Public License for more details.
  18. *
  19. * You should have received a copy of the GNU General Public License
  20. * along with this program; if not, write to the Free Software
  21. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
  22. */
  23. /*
  24. * Based on public domain LibTomMath 0.38 by Tom St Denis, tomstdenis@iahu.ca,
  25. * http://math.libtomcrypt.com
  26. */
  27. #ifndef CTAO_CRYPT_INTEGER_H
  28. #define CTAO_CRYPT_INTEGER_H
  29. #include <stdint.h>
  30. #ifndef word32
  31. typedef uint32_t word32;
  32. #endif
  33. #ifndef byte
  34. typedef uint8_t byte;
  35. #endif
  36. /* may optionally use fast math instead, not yet supported on all platforms and
  37. may not be faster on all
  38. */
  39. #if defined(_MSC_VER) || defined(__BCPLUSPLUS__)
  40. #define WORD64_AVAILABLE
  41. #define W64LIT(x) x##ui64
  42. typedef unsigned __int64 word64;
  43. #elif defined(SIZEOF_LONG) && SIZEOF_LONG == 8
  44. #define WORD64_AVAILABLE
  45. #define W64LIT(x) x##LL
  46. typedef unsigned long word64;
  47. #elif defined(SIZEOF_LONG_LONG) && SIZEOF_LONG_LONG == 8
  48. #define WORD64_AVAILABLE
  49. #define W64LIT(x) x##LL
  50. typedef unsigned long long word64;
  51. #elif defined(__SIZEOF_LONG_LONG__) && __SIZEOF_LONG_LONG__ == 8
  52. #define WORD64_AVAILABLE
  53. #define W64LIT(x) x##LL
  54. typedef unsigned long long word64;
  55. #else
  56. #define MP_16BIT /* for mp_int, mp_word needs to be twice as big as
  57. mp_digit, no 64 bit type so make mp_digit 16 bit */
  58. #endif
  59. #include <limits.h>
  60. #ifndef MIN
  61. #define MIN(x,y) ((x)<(y)?(x):(y))
  62. #endif
  63. #ifndef MAX
  64. #define MAX(x,y) ((x)>(y)?(x):(y))
  65. #endif
  66. /* C on the other hand doesn't care */
  67. #define OPT_CAST(x)
  68. /* detect 64-bit mode if possible */
  69. #if defined(__x86_64__)
  70. #if !(defined(MP_64BIT) && defined(MP_16BIT) && defined(MP_8BIT))
  71. #define MP_64BIT
  72. #endif
  73. #endif
  74. /* if intel compiler doesn't provide 128 bit type don't turn on 64bit */
  75. #if defined(MP_64BIT) && defined(__INTEL_COMPILER) && !defined(HAVE___UINT128_T)
  76. #undef MP_64BIT
  77. #endif
  78. /* some default configurations.
  79. *
  80. * A "mp_digit" must be able to hold DIGIT_BIT + 1 bits
  81. * A "mp_word" must be able to hold 2*DIGIT_BIT + 1 bits
  82. *
  83. * At the very least a mp_digit must be able to hold 7 bits
  84. * [any size beyond that is ok provided it doesn't overflow the data type]
  85. */
  86. #ifdef MP_8BIT
  87. typedef unsigned char mp_digit;
  88. typedef unsigned short mp_word;
  89. #elif defined(MP_16BIT) || defined(NO_64BIT)
  90. typedef unsigned short mp_digit;
  91. typedef unsigned int mp_word;
  92. #elif defined(MP_64BIT)
  93. /* for GCC only on supported platforms */
  94. typedef unsigned long long mp_digit; /* 64 bit type, 128 uses mode(TI) */
  95. typedef unsigned long mp_word __attribute__ ((mode(TI)));
  96. #define DIGIT_BIT 60
  97. #else
  98. /* this is the default case, 28-bit digits */
  99. #if defined(_MSC_VER) || defined(__BORLANDC__)
  100. typedef unsigned __int64 ulong64;
  101. #else
  102. typedef unsigned long long ulong64;
  103. #endif
  104. typedef unsigned int mp_digit; /* long could be 64 now, changed TAO */
  105. typedef ulong64 mp_word;
  106. #ifdef MP_31BIT
  107. /* this is an extension that uses 31-bit digits */
  108. #define DIGIT_BIT 31
  109. #else
  110. /* default case is 28-bit digits, defines MP_28BIT as a handy test macro */
  111. #define DIGIT_BIT 28
  112. #define MP_28BIT
  113. #endif
  114. #endif
  115. /* otherwise the bits per digit is calculated automatically from the size of
  116. a mp_digit */
  117. #ifndef DIGIT_BIT
  118. #define DIGIT_BIT ((int)((CHAR_BIT * sizeof(mp_digit) - 1)))
  119. /* bits per digit */
  120. #endif
  121. #define MP_DIGIT_BIT DIGIT_BIT
  122. #define MP_MASK ((((mp_digit)1)<<((mp_digit)DIGIT_BIT))-((mp_digit)1))
  123. #define MP_DIGIT_MAX MP_MASK
  124. /* equalities */
  125. #define MP_LT -1 /* less than */
  126. #define MP_EQ 0 /* equal to */
  127. #define MP_GT 1 /* greater than */
  128. #define MP_ZPOS 0 /* positive integer */
  129. #define MP_NEG 1 /* negative */
  130. #define MP_OKAY 0 /* ok result */
  131. #define MP_MEM -2 /* out of mem */
  132. #define MP_VAL -3 /* invalid input */
  133. #define MP_RANGE MP_VAL
  134. #define MP_YES 1 /* yes response */
  135. #define MP_NO 0 /* no response */
  136. /* Primality generation flags */
  137. #define LTM_PRIME_BBS 0x0001 /* BBS style prime */
  138. #define LTM_PRIME_SAFE 0x0002 /* Safe prime (p-1)/2 == prime */
  139. #define LTM_PRIME_2MSB_ON 0x0008 /* force 2nd MSB to 1 */
  140. typedef int mp_err;
  141. /* define this to use lower memory usage routines (exptmods mostly) */
  142. #define MP_LOW_MEM
  143. /* default precision */
  144. #ifndef MP_PREC
  145. #define MP_PREC 32 /* default digits of precision */
  146. #endif
  147. /* size of comba arrays, should be at least 2 * 2**(BITS_PER_WORD -
  148. BITS_PER_DIGIT*2) */
  149. #define MP_WARRAY (1 << (sizeof(mp_word) * CHAR_BIT - 2 * DIGIT_BIT + 1))
  150. /* the infamous mp_int structure */
  151. typedef struct {
  152. int used, alloc, sign;
  153. mp_digit *dp;
  154. } mp_int;
  155. /* callback for mp_prime_random, should fill dst with random bytes and return
  156. how many read [upto len] */
  157. typedef int ltm_prime_callback(unsigned char *dst, int len, void *dat);
  158. #define USED(m) ((m)->used)
  159. #define DIGIT(m,k) ((m)->dp[(k)])
  160. #define SIGN(m) ((m)->sign)
  161. /* ---> Basic Manipulations <--- */
  162. #define mp_iszero(a) (((a)->used == 0) ? MP_YES : MP_NO)
  163. #define mp_iseven(a) \
  164. (((a)->used > 0 && (((a)->dp[0] & 1) == 0)) ? MP_YES : MP_NO)
  165. #define mp_isodd(a) \
  166. (((a)->used > 0 && (((a)->dp[0] & 1) == 1)) ? MP_YES : MP_NO)
  167. /* number of primes */
  168. #ifdef MP_8BIT
  169. #define PRIME_SIZE 31
  170. #else
  171. #define PRIME_SIZE 256
  172. #endif
  173. #define mp_prime_random(a, t, size, bbs, cb, dat) \
  174. mp_prime_random_ex(a, t, ((size) * 8) + 1, (bbs==1)?LTM_PRIME_BBS:0, cb, dat)
  175. #define mp_read_raw(mp, str, len) mp_read_signed_bin((mp), (str), (len))
  176. #define mp_raw_size(mp) mp_signed_bin_size(mp)
  177. #define mp_toraw(mp, str) mp_to_signed_bin((mp), (str))
  178. #define mp_read_mag(mp, str, len) mp_read_unsigned_bin((mp), (str), (len))
  179. #define mp_mag_size(mp) mp_unsigned_bin_size(mp)
  180. #define mp_tomag(mp, str) mp_to_unsigned_bin((mp), (str))
  181. #define mp_tobinary(M, S) mp_toradix((M), (S), 2)
  182. #define mp_tooctal(M, S) mp_toradix((M), (S), 8)
  183. #define mp_todecimal(M, S) mp_toradix((M), (S), 10)
  184. #define mp_tohex(M, S) mp_toradix((M), (S), 16)
  185. #define s_mp_mul(a, b, c) s_mp_mul_digs(a, b, c, (a)->used + (b)->used + 1)
  186. extern const char *mp_s_rmap;
  187. /* 6 functions needed by Rsa */
  188. int mp_init (mp_int * a);
  189. void mp_clear (mp_int * a);
  190. int mp_unsigned_bin_size(mp_int * a);
  191. int mp_read_unsigned_bin (mp_int * a, const unsigned char *b, int c);
  192. int mp_to_unsigned_bin (mp_int * a, unsigned char *b);
  193. int mp_exptmod (mp_int * G, mp_int * X, mp_int * P, mp_int * Y);
  194. /* end functions needed by Rsa */
  195. /* functions added to support above needed, removed TOOM and KARATSUBA */
  196. int mp_count_bits (mp_int * a);
  197. int mp_leading_bit (mp_int * a);
  198. int mp_init_copy (mp_int * a, mp_int * b);
  199. int mp_copy (mp_int * a, mp_int * b);
  200. int mp_grow (mp_int * a, int size);
  201. int mp_div_2d (mp_int * a, int b, mp_int * c, mp_int * d);
  202. void mp_zero (mp_int * a);
  203. void mp_clamp (mp_int * a);
  204. void mp_exch (mp_int * a, mp_int * b);
  205. void mp_rshd (mp_int * a, int b);
  206. void mp_rshb (mp_int * a, int b);
  207. int mp_mod_2d (mp_int * a, int b, mp_int * c);
  208. int mp_mul_2d (mp_int * a, int b, mp_int * c);
  209. int mp_lshd (mp_int * a, int b);
  210. int mp_abs (mp_int * a, mp_int * b);
  211. int mp_invmod (mp_int * a, mp_int * b, mp_int * c);
  212. int fast_mp_invmod (mp_int * a, mp_int * b, mp_int * c);
  213. int mp_invmod_slow (mp_int * a, mp_int * b, mp_int * c);
  214. int mp_cmp_mag (mp_int * a, mp_int * b);
  215. int mp_cmp (mp_int * a, mp_int * b);
  216. int mp_cmp_d(mp_int * a, mp_digit b);
  217. void mp_set (mp_int * a, mp_digit b);
  218. int mp_mod (mp_int * a, mp_int * b, mp_int * c);
  219. int mp_div(mp_int * a, mp_int * b, mp_int * c, mp_int * d);
  220. int mp_div_2(mp_int * a, mp_int * b);
  221. int mp_add (mp_int * a, mp_int * b, mp_int * c);
  222. int s_mp_add (mp_int * a, mp_int * b, mp_int * c);
  223. int s_mp_sub (mp_int * a, mp_int * b, mp_int * c);
  224. int mp_sub (mp_int * a, mp_int * b, mp_int * c);
  225. int mp_reduce_is_2k_l(mp_int *a);
  226. int mp_reduce_is_2k(mp_int *a);
  227. int mp_dr_is_modulus(mp_int *a);
  228. int mp_exptmod_fast (mp_int * G, mp_int * X, mp_int * P, mp_int * Y, int);
  229. int mp_montgomery_setup (mp_int * n, mp_digit * rho);
  230. int fast_mp_montgomery_reduce (mp_int * x, mp_int * n, mp_digit rho);
  231. int mp_montgomery_reduce (mp_int * x, mp_int * n, mp_digit rho);
  232. void mp_dr_setup(mp_int *a, mp_digit *d);
  233. int mp_dr_reduce (mp_int * x, mp_int * n, mp_digit k);
  234. int mp_reduce_2k(mp_int *a, mp_int *n, mp_digit d);
  235. int fast_s_mp_mul_high_digs (mp_int * a, mp_int * b, mp_int * c, int digs);
  236. int s_mp_mul_high_digs (mp_int * a, mp_int * b, mp_int * c, int digs);
  237. int mp_reduce_2k_setup_l(mp_int *a, mp_int *d);
  238. int mp_reduce_2k_l(mp_int *a, mp_int *n, mp_int *d);
  239. int mp_reduce (mp_int * x, mp_int * m, mp_int * mu);
  240. int mp_reduce_setup (mp_int * a, mp_int * b);
  241. int s_mp_exptmod (mp_int * G, mp_int * X, mp_int * P, mp_int * Y, int redmode);
  242. int mp_montgomery_calc_normalization (mp_int * a, mp_int * b);
  243. int s_mp_mul_digs (mp_int * a, mp_int * b, mp_int * c, int digs);
  244. int s_mp_sqr (mp_int * a, mp_int * b);
  245. int fast_s_mp_mul_digs (mp_int * a, mp_int * b, mp_int * c, int digs);
  246. int fast_s_mp_sqr (mp_int * a, mp_int * b);
  247. int mp_init_size (mp_int * a, int size);
  248. int mp_div_3 (mp_int * a, mp_int *c, mp_digit * d);
  249. int mp_mul_2(mp_int * a, mp_int * b);
  250. int mp_mul (mp_int * a, mp_int * b, mp_int * c);
  251. int mp_sqr (mp_int * a, mp_int * b);
  252. int mp_mulmod (mp_int * a, mp_int * b, mp_int * c, mp_int * d);
  253. int mp_mul_d (mp_int * a, mp_digit b, mp_int * c);
  254. int mp_2expt (mp_int * a, int b);
  255. int mp_reduce_2k_setup(mp_int *a, mp_digit *d);
  256. int mp_add_d (mp_int* a, mp_digit b, mp_int* c);
  257. int mp_set_int (mp_int * a, unsigned long b);
  258. int mp_sub_d (mp_int * a, mp_digit b, mp_int * c);
  259. /* end support added functions */
  260. /* added */
  261. int mp_init_multi(mp_int* a, mp_int* b, mp_int* c, mp_int* d, mp_int* e,
  262. mp_int* f);
  263. int mp_sqrmod(mp_int* a, mp_int* b, mp_int* c);
  264. int mp_read_radix(mp_int* a, const char* str, int radix);
  265. int mp_prime_is_prime (mp_int * a, int t, int *result);
  266. int mp_gcd (mp_int * a, mp_int * b, mp_int * c);
  267. int mp_lcm (mp_int * a, mp_int * b, mp_int * c);
  268. int mp_cnt_lsb(mp_int *a);
  269. int mp_mod_d(mp_int* a, mp_digit b, mp_digit* c);
  270. #endif /* CTAO_CRYPT_INTEGER_H */