1  /* Test mpn_mod_1 variants.
       2  
       3  Copyright 2010, 2013 Free Software Foundation, Inc.
       4  
       5  This file is part of the GNU MP Library test suite.
       6  
       7  The GNU MP Library test suite is free software; you can redistribute it
       8  and/or modify it under the terms of the GNU General Public License as
       9  published by the Free Software Foundation; either version 3 of the License,
      10  or (at your option) any later version.
      11  
      12  The GNU MP Library test suite is distributed in the hope that it will be
      13  useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
      14  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
      15  Public License for more details.
      16  
      17  You should have received a copy of the GNU General Public License along with
      18  the GNU MP Library test suite.  If not, see https://www.gnu.org/licenses/.  */
      19  
      20  #include <stdio.h>
      21  #include <stdlib.h>
      22  
      23  #include "gmp-impl.h"
      24  #include "tests.h"
      25  
      26  static void
      27  check_one (mp_srcptr ap, mp_size_t n, mp_limb_t b)
      28  {
      29    mp_limb_t r_ref = refmpn_mod_1 (ap, n, b);
      30    mp_limb_t r;
      31  
      32    if (n >= 2)
      33      {
      34        mp_limb_t pre[4];
      35        mpn_mod_1_1p_cps (pre, b);
      36        r = mpn_mod_1_1p (ap, n, b << pre[1], pre);
      37        if (r != r_ref)
      38  	{
      39  	  printf ("mpn_mod_1_1p failed\n");
      40  	  goto fail;
      41  	}
      42      }
      43    if ((b & GMP_NUMB_HIGHBIT) == 0)
      44      {
      45        mp_limb_t pre[5];
      46        mpn_mod_1s_2p_cps (pre, b);
      47        r = mpn_mod_1s_2p (ap, n, b << pre[1], pre);
      48        if (r != r_ref)
      49  	{
      50  	  printf ("mpn_mod_1s_2p failed\n");
      51  	  goto fail;
      52  	}
      53      }
      54    if (b <= GMP_NUMB_MASK / 3)
      55      {
      56        mp_limb_t pre[6];
      57        mpn_mod_1s_3p_cps (pre, b);
      58        r = mpn_mod_1s_3p (ap, n, b << pre[1], pre);
      59        if (r != r_ref)
      60  	{
      61  	  printf ("mpn_mod_1s_3p failed\n");
      62  	  goto fail;
      63  	}
      64      }
      65    if (b <= GMP_NUMB_MASK / 4)
      66      {
      67        mp_limb_t pre[7];
      68        mpn_mod_1s_4p_cps (pre, b);
      69        r = mpn_mod_1s_4p (ap, n, b << pre[1], pre);
      70        if (r != r_ref)
      71  	{
      72  	  printf ("mpn_mod_1s_4p failed\n");
      73  	  goto fail;
      74  	}
      75      }
      76    r = mpn_mod_1 (ap, n, b);
      77    if (r != r_ref)
      78      {
      79        printf ("mpn_mod_1 failed\n");
      80      fail:
      81        printf ("an = %d, a: ", (int) n); mpn_dump (ap, n);
      82        printf ("b           : "); mpn_dump (&b, 1);
      83        printf ("r (expected): "); mpn_dump (&r_ref, 1);
      84        printf ("r (bad)     : "); mpn_dump (&r, 1);
      85        abort();
      86      }
      87  }
      88  
      89  int
      90  main (int argc, char **argv)
      91  {
      92    gmp_randstate_ptr rands;
      93    int i;
      94    unsigned a_bits;
      95    unsigned b_bits;
      96    mpz_t a;
      97    mpz_t b;
      98  
      99    tests_start ();
     100    rands = RANDS;
     101    mpz_init (a);
     102    mpz_init (b);
     103  
     104    for (i = 0; i < 300; i++)
     105      {
     106        mp_size_t asize;
     107        a_bits = 1 + gmp_urandomm_ui (rands, 1000);
     108        b_bits = 1 + gmp_urandomm_ui (rands, GMP_NUMB_BITS);
     109  
     110        mpz_rrandomb (a, rands, a_bits);
     111        mpz_rrandomb (b, rands, b_bits);
     112  
     113        asize = SIZ(a);
     114        if (!asize)
     115  	asize = 1;
     116        if (mpz_sgn (b) == 0)
     117  	mpz_set_ui (b, 1);
     118  
     119        check_one (PTR(a), asize, PTR(b)[0]);
     120      }
     121  
     122    mpz_clear (a);
     123    mpz_clear (b);
     124  
     125    tests_end ();
     126    return 0;
     127  }