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  }