bignum.c 3.86 KB
Newer Older
1
/* bignum.c
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32

   Bignum operations that are missing from gmp.

   Copyright (C) 2001 Niels Möller

   This file is part of GNU Nettle.

   GNU Nettle is free software: you can redistribute it and/or
   modify it under the terms of either:

     * the GNU Lesser General Public License as published by the Free
       Software Foundation; either version 3 of the License, or (at your
       option) any later version.

   or

     * the GNU General Public License as published by the Free
       Software Foundation; either version 2 of the License, or (at your
       option) any later version.

   or both in parallel, as here.

   GNU Nettle is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   General Public License for more details.

   You should have received copies of the GNU General Public License and
   the GNU Lesser General Public License along with this program.  If
   not, see http://www.gnu.org/licenses/.
*/
33 34

#if HAVE_CONFIG_H
35
# include "config.h"
36 37 38 39 40
#endif

#include <assert.h>
#include <string.h>

41 42
#include "bignum.h"

43 44 45 46 47 48 49 50 51 52 53 54 55 56
/* Two's complement negation means that -x = ~x + 1, ~x = -(x+1),
 * and we use that x = ~~x = ~(-x-1).
 *
 * Examples:
 *
 *   x  ~x = -x+1     ~~x = x
 *  -1          0          ff
 *  -2          1          fe
 * -7f         7e          81
 * -80         7f          80
 * -81         80        ff7f
 */

/* Including extra sign bit, if needed. Also one byte for zero. */
57
size_t
58
nettle_mpz_sizeinbase_256_s(const mpz_t x)
59
{
60 61 62 63 64
  if (mpz_sgn(x) >= 0)
    return 1 + mpz_sizeinbase(x, 2) / 8;
  else
    {
      /* We'll output ~~x, so we need as many bits as for ~x */
65
      size_t size;
66 67 68 69 70 71 72 73 74
      mpz_t c;

      mpz_init(c);
      mpz_com(c, x); /* Same as c = - x - 1 = |x| + 1 */
      size = 1 + mpz_sizeinbase(c,2) / 8;
      mpz_clear(c);

      return size;
    }
75 76
}

77
size_t
78 79 80 81 82 83
nettle_mpz_sizeinbase_256_u(const mpz_t x)
{
  return (mpz_sizeinbase(x,2) + 7) / 8;
}

static void
84
nettle_mpz_to_octets(size_t length, uint8_t *s,
85
		     const mpz_t x, uint8_t sign)
86 87
{
  uint8_t *dst = s + length - 1;
88 89
  size_t size = mpz_size(x);
  size_t i;
90 91 92 93
  
  for (i = 0; i<size; i++)
    {
      mp_limb_t limb = mpz_getlimbn(x, i);
94
      size_t j;
95 96 97

      for (j = 0; length && j < sizeof(mp_limb_t); j++)
        {
98
          *dst-- = sign ^ (limb & 0xff);
99 100 101 102 103 104
          limb >>= 8;
          length--;
	}
    }
  
  if (length)
105
    memset(s, sign, length);
106 107 108
}

void
109
nettle_mpz_get_str_256(size_t length, uint8_t *s, const mpz_t x)
110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
{
  if (!length)
    {
      /* x must be zero */
      assert(!mpz_sgn(x));
      return;
    }

  if (mpz_sgn(x) >= 0)
    {
      assert(nettle_mpz_sizeinbase_256_u(x) <= length);
      nettle_mpz_to_octets(length, s, x, 0);
    }
  else
    {
      mpz_t c;
      mpz_init(c);
      mpz_com(c, x);
128

129 130 131 132 133 134 135 136
      assert(nettle_mpz_sizeinbase_256_u(c) <= length);
      nettle_mpz_to_octets(length, s, c, 0xff);

      mpz_clear(c);
    }
}

/* Converting from strings */
137

138 139
/* mpz_import was introduced in GMP-4.1 */
#define nettle_mpz_from_octets(x, length, s) \
140
   mpz_import((x), (length), 1, 1, 0, 0, (s))
141 142

void
143
nettle_mpz_set_str_256_u(mpz_t x,
144
			 size_t length, const uint8_t *s)
145
{
146
  nettle_mpz_from_octets(x, length, s);
147 148 149 150
}

void
nettle_mpz_init_set_str_256_u(mpz_t x,
151
			      size_t length, const uint8_t *s)
152
{
153 154
  mpz_init(x);
  nettle_mpz_from_octets(x, length, s);
155 156 157 158
}

void
nettle_mpz_set_str_256_s(mpz_t x,
159
			 size_t length, const uint8_t *s)
160 161
{
  if (!length)
162 163 164 165
    {
      mpz_set_ui(x, 0);
      return;
    }
166
  
167 168
  nettle_mpz_from_octets(x, length, s);

169 170
  if (s[0] & 0x80)
    {
171 172 173 174 175 176
      mpz_t t;

      mpz_init_set_ui(t, 1);
      mpz_mul_2exp(t, t, length*8);
      mpz_sub(x, x, t);
      mpz_clear(t);
177 178 179 180 181
    }
}

void
nettle_mpz_init_set_str_256_s(mpz_t x,
182
			      size_t length, const uint8_t *s)
183 184
{
  mpz_init(x);
185
  nettle_mpz_set_str_256_s(x, length, s);
186
}