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