summaryrefslogtreecommitdiff
path: root/gmp/mini-gmp/tests/t-invert.c
blob: 496e8d4f53c75a5968139cbeb0d4c4660a293bf6 (plain)
1
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
/*

Copyright 2012, Free Software Foundation, Inc.

This file is part of the GNU MP Library test suite.

The GNU MP Library test suite is free software; you can redistribute it
and/or modify it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 3 of the License,
or (at your option) any later version.

The GNU MP Library test suite 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 a copy of the GNU General Public License along with
the GNU MP Library test suite.  If not, see https://www.gnu.org/licenses/.  */

#include <limits.h>
#include <stdlib.h>
#include <stdio.h>

#include "testutils.h"

#define GMP_LIMB_BITS (sizeof(mp_limb_t) * CHAR_BIT)

#define COUNT 10000

void
testmain (int argc, char **argv)
{
  unsigned i;
  mpz_t u, m, p, t;

  mpz_init (u);
  mpz_init (m);
  mpz_init (p);
  mpz_init (t);

  for (i = 0; i < COUNT; i++)
    {
      mini_urandomb (u, GMP_LIMB_BITS);
      mpz_setbit (u, GMP_LIMB_BITS -1);

      mpz_set_ui (m, mpn_invert_limb (u->_mp_d[0]));
      mpz_setbit (m, GMP_LIMB_BITS);

      mpz_mul (p, m, u);

      mpz_set_ui (t, 0);
      mpz_setbit (t, 2* GMP_LIMB_BITS);
      mpz_sub (t, t, p);

      /* Should have 0 < B^2 - m u <= u */
      if (mpz_sgn (t) <= 0 || mpz_cmp (t, u) > 0)
	{
	  fprintf (stderr, "mpn_invert_limb failed:\n");
	  dump ("u", u);
	  dump ("m", m);
	  dump ("p", p);
	  dump ("t", t);
	  abort ();
	}
    }

  for (i = 0; i < COUNT; i++)
    {
      mini_urandomb (u, 2*GMP_LIMB_BITS);
      mpz_setbit (u, 2*GMP_LIMB_BITS -1);

      mpz_set_ui (m, mpn_invert_3by2 (u->_mp_d[1], u[0]._mp_d[0]));

      mpz_setbit (m, GMP_LIMB_BITS);

      mpz_mul (p, m, u);

      mpz_set_ui (t, 0);
      mpz_setbit (t, 3 * GMP_LIMB_BITS);
      mpz_sub (t, t, p);

      /* Should have 0 < B^3 - m u <= u */
      if (mpz_sgn (t) <= 0 || mpz_cmp (t, u) > 0)
	{
	  fprintf (stderr, "mpn_invert_3by2 failed:\n");
	  dump ("u", u);
	  dump ("m", m);
	  dump ("p", p);
	  dump ("t", t);
	  abort ();
	}
    }

  mpz_clear (u);
  mpz_clear (m);
  mpz_clear (p);
  mpz_clear (t);
}