| Austin Schuh | dace2a6 | 2020-08-18 10:56:48 -0700 | [diff] [blame] | 1 | /* Copyright 2012, 2015 Free Software Foundation, Inc. |
| 2 | |
| 3 | This file is part of the GNU MP Library test suite. |
| 4 | |
| 5 | The GNU MP Library test suite is free software; you can redistribute it |
| 6 | and/or modify it under the terms of the GNU General Public License as |
| 7 | published by the Free Software Foundation; either version 3 of the License, |
| 8 | or (at your option) any later version. |
| 9 | |
| 10 | The GNU MP Library test suite is distributed in the hope that it will be |
| 11 | useful, but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General |
| 13 | Public License for more details. |
| 14 | |
| 15 | You should have received a copy of the GNU General Public License along with |
| 16 | the 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 | |
| 29 | int |
| 30 | main (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 | } |