mplogic.c
上传用户:lyxiangda
上传日期:2007-01-12
资源大小:3042k
文件大小:10k
源码类别:

CA认证

开发平台:

WINDOWS

  1. /*
  2.  *  mplogic.c
  3.  *
  4.  *  Bitwise logical operations on MPI values
  5.  *
  6.  * The contents of this file are subject to the Mozilla Public
  7.  * License Version 1.1 (the "License"); you may not use this file
  8.  * except in compliance with the License. You may obtain a copy of
  9.  * the License at http://www.mozilla.org/MPL/
  10.  *
  11.  * Software distributed under the License is distributed on an "AS
  12.  * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
  13.  * implied. See the License for the specific language governing
  14.  * rights and limitations under the License.
  15.  *
  16.  * The Original Code is the MPI Arbitrary Precision Integer Arithmetic
  17.  * library.
  18.  *
  19.  * The Initial Developer of the Original Code is Michael J. Fromberger.
  20.  * Portions created by Michael J. Fromberger are 
  21.  * Copyright (C) 1998, 1999, 2000 Michael J. Fromberger. 
  22.  * All Rights Reserved.
  23.  *
  24.  * Contributor(s):
  25.  *
  26.  * Alternatively, the contents of this file may be used under the
  27.  * terms of the GNU General Public License Version 2 or later (the
  28.  * "GPL"), in which case the provisions of the GPL are applicable
  29.  * instead of those above.  If you wish to allow use of your
  30.  * version of this file only under the terms of the GPL and not to
  31.  * allow others to use your version of this file under the MPL,
  32.  * indicate your decision by deleting the provisions above and
  33.  * replace them with the notice and other provisions required by
  34.  * the GPL.  If you do not delete the provisions above, a recipient
  35.  * may use your version of this file under either the MPL or the GPL.
  36.  *
  37.  *  $Id: mplogic.c,v 1.12 2000/09/14 00:30:50 nelsonb%netscape.com Exp $
  38.  */
  39. #include "mpi-priv.h"
  40. #include "mplogic.h"
  41. /* {{{ Lookup table for population count */
  42. static unsigned char bitc[] = {
  43.    0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4, 
  44.    1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5, 
  45.    1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5, 
  46.    2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 
  47.    1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5, 
  48.    2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 
  49.    2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 
  50.    3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7, 
  51.    1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5, 
  52.    2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 
  53.    2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 
  54.    3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7, 
  55.    2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 
  56.    3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7, 
  57.    3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7, 
  58.    4, 5, 5, 6, 5, 6, 6, 7, 5, 6, 6, 7, 6, 7, 7, 8
  59. };
  60. /* }}} */
  61. /*------------------------------------------------------------------------*/
  62. /*
  63.   mpl_not(a, b)    - compute b = ~a
  64.   mpl_and(a, b, c) - compute c = a & b
  65.   mpl_or(a, b, c)  - compute c = a | b
  66.   mpl_xor(a, b, c) - compute c = a ^ b
  67.  */
  68. /* {{{ mpl_not(a, b) */
  69. mp_err mpl_not(mp_int *a, mp_int *b)
  70. {
  71.   mp_err   res;
  72.   int      ix;
  73.   ARGCHK(a != NULL && b != NULL, MP_BADARG);
  74.   if((res = mp_copy(a, b)) != MP_OKAY)
  75.     return res;
  76.   /* This relies on the fact that the digit type is unsigned */
  77.   for(ix = 0; ix < USED(b); ix++) 
  78.     DIGIT(b, ix) = ~DIGIT(b, ix);
  79.   s_mp_clamp(b);
  80.   return MP_OKAY;
  81. } /* end mpl_not() */
  82. /* }}} */
  83. /* {{{ mpl_and(a, b, c) */
  84. mp_err mpl_and(mp_int *a, mp_int *b, mp_int *c)
  85. {
  86.   mp_int  *which, *other;
  87.   mp_err   res;
  88.   int      ix;
  89.   ARGCHK(a != NULL && b != NULL && c != NULL, MP_BADARG);
  90.   if(USED(a) <= USED(b)) {
  91.     which = a;
  92.     other = b;
  93.   } else {
  94.     which = b;
  95.     other = a;
  96.   }
  97.   if((res = mp_copy(which, c)) != MP_OKAY)
  98.     return res;
  99.   for(ix = 0; ix < USED(which); ix++)
  100.     DIGIT(c, ix) &= DIGIT(other, ix);
  101.   s_mp_clamp(c);
  102.   return MP_OKAY;
  103. } /* end mpl_and() */
  104. /* }}} */
  105. /* {{{ mpl_or(a, b, c) */
  106. mp_err mpl_or(mp_int *a, mp_int *b, mp_int *c)
  107. {
  108.   mp_int  *which, *other;
  109.   mp_err   res;
  110.   int      ix;
  111.   ARGCHK(a != NULL && b != NULL && c != NULL, MP_BADARG);
  112.   if(USED(a) >= USED(b)) {
  113.     which = a;
  114.     other = b;
  115.   } else {
  116.     which = b;
  117.     other = a;
  118.   }
  119.   if((res = mp_copy(which, c)) != MP_OKAY)
  120.     return res;
  121.   for(ix = 0; ix < USED(which); ix++)
  122.     DIGIT(c, ix) |= DIGIT(other, ix);
  123.   return MP_OKAY;
  124. } /* end mpl_or() */
  125. /* }}} */
  126. /* {{{ mpl_xor(a, b, c) */
  127. mp_err mpl_xor(mp_int *a, mp_int *b, mp_int *c)
  128. {
  129.   mp_int  *which, *other;
  130.   mp_err   res;
  131.   int      ix;
  132.   ARGCHK(a != NULL && b != NULL && c != NULL, MP_BADARG);
  133.   if(USED(a) >= USED(b)) {
  134.     which = a;
  135.     other = b;
  136.   } else {
  137.     which = b;
  138.     other = a;
  139.   }
  140.   if((res = mp_copy(which, c)) != MP_OKAY)
  141.     return res;
  142.   for(ix = 0; ix < USED(which); ix++)
  143.     DIGIT(c, ix) ^= DIGIT(other, ix);
  144.   s_mp_clamp(c);
  145.   return MP_OKAY;
  146. } /* end mpl_xor() */
  147. /* }}} */
  148. /*------------------------------------------------------------------------*/
  149. /*
  150.   mpl_rsh(a, b, d)     - b = a >> d
  151.   mpl_lsh(a, b, d)     - b = a << d
  152.  */
  153. /* {{{ mpl_rsh(a, b, d) */
  154. mp_err mpl_rsh(const mp_int *a, mp_int *b, mp_digit d)
  155. {
  156.   mp_err   res;
  157.   ARGCHK(a != NULL && b != NULL, MP_BADARG);
  158.   if((res = mp_copy(a, b)) != MP_OKAY)
  159.     return res;
  160.   s_mp_div_2d(b, d);
  161.   return MP_OKAY;
  162. } /* end mpl_rsh() */
  163. /* }}} */
  164. /* {{{ mpl_lsh(a, b, d) */
  165. mp_err mpl_lsh(const mp_int *a, mp_int *b, mp_digit d)
  166. {
  167.   mp_err   res;
  168.   ARGCHK(a != NULL && b != NULL, MP_BADARG);
  169.   if((res = mp_copy(a, b)) != MP_OKAY)
  170.     return res;
  171.   return s_mp_mul_2d(b, d);
  172. } /* end mpl_lsh() */
  173. /* }}} */
  174. /*------------------------------------------------------------------------*/
  175. /*
  176.   mpl_num_set(a, num)
  177.   Count the number of set bits in the binary representation of a.
  178.   Returns MP_OKAY and sets 'num' to be the number of such bits, if
  179.   possible.  If num is NULL, the result is thrown away, but it is
  180.   not considered an error.
  181.   mpl_num_clear() does basically the same thing for clear bits.
  182.  */
  183. /* {{{ mpl_num_set(a, num) */
  184. mp_err mpl_num_set(mp_int *a, int *num)
  185. {
  186.   int            ix, db, nset = 0;
  187.   mp_digit       cur;
  188.   unsigned char  reg;
  189.   ARGCHK(a != NULL, MP_BADARG);
  190.   for(ix = 0; ix < USED(a); ix++) {
  191.     cur = DIGIT(a, ix);
  192.     
  193.     for(db = 0; db < sizeof(mp_digit); db++) {
  194.       reg = (unsigned char)(cur >> (CHAR_BIT * db));
  195.       nset += bitc[reg];
  196.     }
  197.   }
  198.   if(num)
  199.     *num = nset;
  200.   return MP_OKAY;
  201. } /* end mpl_num_set() */
  202. /* }}} */
  203. /* {{{ mpl_num_clear(a, num) */
  204. mp_err mpl_num_clear(mp_int *a, int *num)
  205. {
  206.   int            ix, db, nset = 0;
  207.   mp_digit       cur;
  208.   unsigned char  reg;
  209.   ARGCHK(a != NULL, MP_BADARG);
  210.   for(ix = 0; ix < USED(a); ix++) {
  211.     cur = DIGIT(a, ix);
  212.     
  213.     for(db = 0; db < sizeof(mp_digit); db++) {
  214.       reg = (unsigned char)(cur >> (CHAR_BIT * db));
  215.       nset += bitc[UCHAR_MAX - reg];
  216.     }
  217.   }
  218.   if(num)
  219.     *num = nset;
  220.   return MP_OKAY;
  221. } /* end mpl_num_clear() */
  222. /* }}} */
  223. /*------------------------------------------------------------------------*/
  224. /*
  225.   mpl_parity(a)
  226.   Determines the bitwise parity of the value given.  Returns MP_EVEN
  227.   if an even number of digits are set, MP_ODD if an odd number are
  228.   set.
  229.  */
  230. /* {{{ mpl_parity(a) */
  231. mp_err mpl_parity(mp_int *a)
  232. {
  233.   int      ix, par = 0;
  234.   mp_digit cur;
  235.   ARGCHK(a != NULL, MP_BADARG);
  236.   for(ix = 0; ix < USED(a); ix++) {
  237.     int   shft = (sizeof(mp_digit) * CHAR_BIT) / 2;
  238.     cur = DIGIT(a, ix);
  239.     /* Compute parity for current digit */
  240.     while(shft != 0) {
  241.       cur ^= (cur >> shft);
  242.       shft >>= 1;
  243.     }
  244.     cur &= 1;
  245.     /* XOR with running parity so far   */
  246.     par ^= cur;
  247.   }
  248.   if(par)
  249.     return MP_ODD;
  250.   else
  251.     return MP_EVEN;
  252. } /* end mpl_parity() */
  253. /* }}} */
  254. /*
  255.   mpl_set_bit
  256.   Returns MP_OKAY or some error code.
  257.   Grows a if needed to set a bit to 1.
  258.  */
  259. mp_err mpl_set_bit(mp_int *a, mp_size bitNum, mp_size value)
  260. {
  261.   mp_size      ix;
  262.   mp_err       rv;
  263.   mp_digit     mask;
  264.   ARGCHK(a != NULL, MP_BADARG);
  265.   ix = bitNum / MP_DIGIT_BIT;
  266.   if (ix + 1 > MP_USED(a)) {
  267.     rv = s_mp_pad(a, ix + 1);
  268.     if (rv != MP_OKAY)
  269.       return rv;
  270.   }
  271.   bitNum = bitNum % MP_DIGIT_BIT;
  272.   mask = (mp_digit)1 << bitNum;
  273.   if (value)
  274.     MP_DIGIT(a,ix) |= mask;
  275.   else
  276.     MP_DIGIT(a,ix) &= ~mask;
  277.   s_mp_clamp(a);
  278.   return MP_OKAY;
  279. }
  280. /*
  281.   mpl_get_bit
  282.   returns 0 or 1 or some (negative) error code.
  283.  */
  284. mp_err mpl_get_bit(const mp_int *a, mp_size bitNum)
  285. {
  286.   mp_size      bit, ix;
  287.   mp_err       rv;
  288.   ARGCHK(a != NULL, MP_BADARG);
  289.   ix = bitNum / MP_DIGIT_BIT;
  290.   ARGCHK(ix <= MP_USED(a) - 1, MP_RANGE);
  291.   bit   = bitNum % MP_DIGIT_BIT;
  292.   rv = (mp_err)(MP_DIGIT(a, ix) >> bit) & 1;
  293.   return rv;
  294. }
  295. /*
  296.   mpl_get_bits
  297.   - Extracts numBits bits from a, where the least significant extracted bit
  298.   is bit lsbNum.  Returns a negative value if error occurs.
  299.   - Because sign bit is used to indicate error, maximum number of bits to 
  300.   be returned is the lesser of (a) the number of bits in an mp_digit, or
  301.   (b) one less than the number of bits in an mp_err.
  302.   - lsbNum + numbits can be greater than the number of significant bits in
  303.   integer a, as long as bit lsbNum is in the high order digit of a.
  304.  */
  305. mp_err mpl_get_bits(const mp_int *a, mp_size lsbNum, mp_size numBits) 
  306. {
  307.   mp_size    rshift = (lsbNum % MP_DIGIT_BIT);
  308.   mp_size    lsWndx = (lsbNum / MP_DIGIT_BIT);
  309.   mp_digit * digit  = MP_DIGITS(a) + lsWndx;
  310.   mp_digit   mask   = ((1 << numBits) - 1);
  311.   ARGCHK(numBits < CHAR_BIT * sizeof mask, MP_BADARG);
  312.   ARGCHK(MP_HOWMANY(lsbNum, MP_DIGIT_BIT) <= MP_USED(a), MP_RANGE);
  313.   if ((numBits + lsbNum % MP_DIGIT_BIT <= MP_DIGIT_BIT) ||
  314.       (lsWndx + 1 >= MP_USED(a))) {
  315.     mask &= (digit[0] >> rshift);
  316.   } else {
  317.     mask &= ((digit[0] >> rshift) | (digit[1] << (MP_DIGIT_BIT - rshift)));
  318.   }
  319.   return (mp_err)mask;
  320. }
  321. /*
  322.   mpl_significant_bits
  323.   returns number of significnant bits in abs(a).
  324.   returns 1 if value is zero.
  325.  */
  326. mp_err mpl_significant_bits(const mp_int *a)
  327. {
  328.   mp_err bits  = 0;
  329.   int    ix;
  330.   ARGCHK(a != NULL, MP_BADARG);
  331.   ix = MP_USED(a);
  332.   for (ix = MP_USED(a); ix > 0; ) {
  333.     mp_digit d;
  334.     d = MP_DIGIT(a, --ix);
  335.     if (d) {
  336.       while (d) {
  337. ++bits;
  338. d >>= 1;
  339.       }
  340.       break;
  341.     }
  342.   }
  343.   bits += ix * MP_DIGIT_BIT;
  344.   if (!bits)
  345.     bits = 1;
  346.   return bits;
  347. }
  348. /*------------------------------------------------------------------------*/
  349. /* HERE THERE BE DRAGONS                                                  */