bignum.c 4.1 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 140 141 142

#ifdef mpz_import
/* Was introduced in GMP-4.1 */
# define nettle_mpz_from_octets(x, length, s) \
   mpz_import((x), (length), 1, 1, 0, 0, (s))
#else
143 144
static void
nettle_mpz_from_octets(mpz_t x,
145
		       size_t length, const uint8_t *s)
146
{
147
  size_t i;
148

149 150
  mpz_set_ui(x, 0);

151 152
  for (i = 0; i < length; i++)
    {
153
      mpz_mul_2exp(x, x, 8);
154
      mpz_add_ui(x, x, s[i]);
155 156
    }
}
157
#endif
158 159

void
160
nettle_mpz_set_str_256_u(mpz_t x,
161
			 size_t length, const uint8_t *s)
162
{
163
  nettle_mpz_from_octets(x, length, s);
164 165 166 167
}

void
nettle_mpz_init_set_str_256_u(mpz_t x,
168
			      size_t length, const uint8_t *s)
169
{
170 171
  mpz_init(x);
  nettle_mpz_from_octets(x, length, s);
172 173 174 175
}

void
nettle_mpz_set_str_256_s(mpz_t x,
176
			 size_t length, const uint8_t *s)
177 178
{
  if (!length)
179 180 181 182
    {
      mpz_set_ui(x, 0);
      return;
    }
183
  
184 185
  nettle_mpz_from_octets(x, length, s);

186 187
  if (s[0] & 0x80)
    {
188 189 190 191 192 193
      mpz_t t;

      mpz_init_set_ui(t, 1);
      mpz_mul_2exp(t, t, length*8);
      mpz_sub(x, x, t);
      mpz_clear(t);
194 195 196 197 198
    }
}

void
nettle_mpz_init_set_str_256_s(mpz_t x,
199
			      size_t length, const uint8_t *s)
200 201
{
  mpz_init(x);
202
  nettle_mpz_set_str_256_s(x, length, s);
203
}