1/* Test mpn_mod_1 variants.
2
3Copyright 2010, 2013 Free Software Foundation, Inc.
4
5This file is part of the GNU MP Library test suite.
6
7The GNU MP Library test suite is free software; you can redistribute it
8and/or modify it under the terms of the GNU General Public License as
9published by the Free Software Foundation; either version 3 of the License,
10or (at your option) any later version.
11
12The GNU MP Library test suite is distributed in the hope that it will be
13useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
14MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
15Public License for more details.
16
17You should have received a copy of the GNU General Public License along with
18the 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
26static void
27check_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
89int
90main (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}
128