1/* Copyright 2012, 2015 Free Software Foundation, Inc.
2
3This file is part of the GNU MP Library test suite.
4
5The GNU MP Library test suite is free software; you can redistribute it
6and/or modify it under the terms of the GNU General Public License as
7published by the Free Software Foundation; either version 3 of the License,
8or (at your option) any later version.
9
10The GNU MP Library test suite is distributed in the hope that it will be
11useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
12MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
13Public License for more details.
14
15You should have received a copy of the GNU General Public License along with
16the GNU MP Library test suite.  If not, see https://www.gnu.org/licenses/.  */
17
18
19#include <stdlib.h>		/* for strtol */
20#include <stdio.h>		/* for printf */
21
22#include "gmp-impl.h"
23#include "longlong.h"
24#include "tests/tests.h"
25
26#define MAX_LIMBS 150
27#define COUNT 500
28
29int
30main (int argc, char **argv)
31{
32  gmp_randstate_ptr rands;
33
34  mp_ptr ap, rp, pp, app, scratch;
35  int count = COUNT;
36  unsigned i;
37  TMP_DECL;
38
39  TMP_MARK;
40
41  TESTS_REPS (count, argv, argc);
42
43  tests_start ();
44  rands = RANDS;
45
46  ap = TMP_ALLOC_LIMBS (MAX_LIMBS);
47  rp = TMP_ALLOC_LIMBS (MAX_LIMBS);
48  pp = TMP_ALLOC_LIMBS (MAX_LIMBS);
49  app = TMP_ALLOC_LIMBS (MAX_LIMBS);
50  scratch = TMP_ALLOC_LIMBS (5*MAX_LIMBS);
51
52  for (i = 0; i < count; i++)
53    {
54      mp_size_t n;
55      mp_limb_t k;
56
57      n = 1 + gmp_urandomm_ui (rands, MAX_LIMBS);
58
59      if (i & 1)
60	mpn_random2 (ap, n);
61      else
62	mpn_random (ap, n);
63
64      ap[0] |= 1;
65
66      if (i < 100)
67	k = 3 + 2*i;
68      else
69	{
70	  mpn_random (&k, 1);
71	  if (k < 3)
72	    k = 3;
73	  else
74	    k |= 1;
75	}
76      mpn_brootinv (rp, ap, n, k, scratch);
77      mpn_powlo (pp, rp, &k, 1, n, scratch);
78      mpn_mullo_n (app, ap, pp, n);
79
80      if (app[0] != 1 || !(n == 1 || mpn_zero_p (app+1, n-1)))
81	{
82	  gmp_fprintf (stderr,
83		       "mpn_brootinv returned bad result: %u limbs\n",
84		       (unsigned) n);
85	  gmp_fprintf (stderr, "k     = %Mx\n", k);
86	  gmp_fprintf (stderr, "a     = %Nx\n", ap, n);
87	  gmp_fprintf (stderr, "r     = %Nx\n", rp, n);
88	  gmp_fprintf (stderr, "r^n   = %Nx\n", pp, n);
89	  gmp_fprintf (stderr, "a r^n = %Nx\n", app, n);
90	  abort ();
91	}
92    }
93  TMP_FREE;
94  tests_end ();
95  return 0;
96}
97