/* Test that routines allow reusing a source variable as destination.
Copyright 1996, 1999-2002, 2009, 2012 Free Software Foundation, Inc.
This file is part of the GNU MP Library test suite.
The GNU MP Library test suite is free software; you can redistribute it
and/or modify it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 3 of the License,
or (at your option) any later version.
The GNU MP Library test suite is distributed in the hope that it will be
useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
Public License for more details.
You should have received a copy of the GNU General Public License along with
the GNU MP Library test suite. If not, see https://www.gnu.org/licenses/. */
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "testutils.h"
#define COUNT 100
void dump3 (const char *, mpz_t, mpz_t, mpz_t);
void mpz_check_format (const mpz_t);
typedef void (*dss_func) (mpz_t, const mpz_t, const mpz_t);
typedef void (*dsi_func) (mpz_t, const mpz_t, unsigned long int);
typedef unsigned long int (*dsi_div_func) (mpz_t, const mpz_t, unsigned long int);
typedef unsigned long int (*ddsi_div_func) (mpz_t, mpz_t, const mpz_t, unsigned long int);
typedef void (*ddss_div_func) (mpz_t, mpz_t, const mpz_t, const mpz_t);
typedef void (*ds_func) (mpz_t, const mpz_t);
void
mpz_xinvert (mpz_t r, const mpz_t a, const mpz_t b)
{
int res;
res = mpz_invert (r, a, b);
if (res == 0)
mpz_set_ui (r, 0);
}
dss_func dss_funcs[] =
{
mpz_add, mpz_sub, mpz_mul,
mpz_cdiv_q, mpz_cdiv_r, mpz_fdiv_q, mpz_fdiv_r, mpz_tdiv_q, mpz_tdiv_r,
mpz_xinvert,
mpz_gcd, mpz_lcm, mpz_and, mpz_ior, mpz_xor
};
const char *dss_func_names[] =
{
"mpz_add", "mpz_sub", "mpz_mul",
"mpz_cdiv_q", "mpz_cdiv_r", "mpz_fdiv_q", "mpz_fdiv_r", "mpz_tdiv_q", "mpz_tdiv_r",
"mpz_xinvert",
"mpz_gcd", "mpz_lcm", "mpz_and", "mpz_ior", "mpz_xor"
};
char dss_func_division[] = {0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0};
dsi_func dsi_funcs[] =
{
/* Don't change order here without changing the code in main(). */
mpz_add_ui, mpz_mul_ui, mpz_sub_ui,
mpz_fdiv_q_2exp, mpz_fdiv_r_2exp,
mpz_cdiv_q_2exp, mpz_cdiv_r_2exp,
mpz_tdiv_q_2exp, mpz_tdiv_r_2exp,
mpz_mul_2exp,
mpz_pow_ui
};
const char *dsi_func_names[] =
{
"mpz_add_ui", "mpz_mul_ui", "mpz_sub_ui",
"mpz_fdiv_q_2exp", "mpz_fdiv_r_2exp",
"mpz_cdiv_q_2exp", "mpz_cdiv_r_2exp",
"mpz_tdiv_q_2exp", "mpz_tdiv_r_2exp",
"mpz_mul_2exp",
"mpz_pow_ui"
};
dsi_div_func dsi_div_funcs[] =
{
mpz_cdiv_q_ui, mpz_cdiv_r_ui,
mpz_fdiv_q_ui, mpz_fdiv_r_ui,
mpz_tdiv_q_ui, mpz_tdiv_r_ui
};
const char *dsi_div_func_names[] =
{
"mpz_cdiv_q_ui", "mpz_cdiv_r_ui",
"mpz_fdiv_q_ui", "mpz_fdiv_r_ui",
"mpz_tdiv_q_ui", "mpz_tdiv_r_ui"
};
ddsi_div_func ddsi_div_funcs[] =
{
mpz_cdiv_qr_ui,
mpz_fdiv_qr_ui,
mpz_tdiv_qr_ui
};
const char *ddsi_div_func_names[] =
{
"mpz_cdiv_qr_ui",
"mpz_fdiv_qr_ui",
"mpz_tdiv_qr_ui"
};
ddss_div_func ddss_div_funcs[] =
{
mpz_cdiv_qr,
mpz_fdiv_qr,
mpz_tdiv_qr
};
const char *ddss_div_func_names[] =
{
"mpz_cdiv_qr",
"mpz_fdiv_qr",
"mpz_tdiv_qr"
};
ds_func ds_funcs[] =
{
mpz_abs, mpz_com, mpz_neg, mpz_sqrt
};
const char *ds_func_names[] =
{
"mpz_abs", "mpz_com", "mpz_neg", "mpz_sqrt"
};
#define FAIL(class,indx,op1,op2,op3) \
do { \
class##_funcs[indx] = 0; \
dump3 (class##_func_names[indx], op1, op2, op3); \
failures++; \
} while (0)
#define FAIL2(fname,op1,op2,op3) \
do { \
dump3 (#fname, op1, op2, op3); \
failures++; \
} while (0)
void
testmain (int argc, char **argv)
{
unsigned i;
int pass, reps = COUNT;
mpz_t in1, in2, in3;
unsigned long int in2i;
mp_size_t size;
mpz_t res1, res2, res3;
mpz_t ref1, ref2, ref3;
mpz_t t;
unsigned long int r1, r2;
long failures = 0;
mpz_t bs;
unsigned long bsi, size_range;
mpz_init (bs);
mpz_init (in1);
mpz_init (in2);
mpz_init (in3);
mpz_init (ref1);
mpz_init (ref2);
mpz_init (ref3);
mpz_init (res1);
mpz_init (res2);
mpz_init (res3);
mpz_init (t);
for (pass = 1; pass <= reps; pass++)
{
mini_urandomb (bs, 32);
size_range = mpz_get_ui (bs) % 12 + 2;
mini_urandomb (bs, size_range);
size = mpz_get_ui (bs);
mini_rrandomb (in1, size);
mini_urandomb (bs, size_range);
size = mpz_get_ui (bs);
mini_rrandomb (in2, size);
mini_urandomb (bs, size_range);
size = mpz_get_ui (bs);
mini_rrandomb (in3, size);
mini_urandomb (bs, 3);
bsi = mpz_get_ui (bs);
if ((bsi & 1) != 0)
mpz_neg (in1, in1);
if ((bsi & 2) != 0)
mpz_neg (in2, in2);
if ((bsi & 4) != 0)
mpz_neg (in3, in3);
for (i = 0; i < numberof (dss_funcs); i++)
{
if (dss_funcs[i] == 0)
continue;
if (dss_func_division[i] && mpz_sgn (in2) == 0)
continue;
(dss_funcs[i]) (ref1, in1, in2);
mpz_check_format (ref1);
mpz_set (res1, in1);
(dss_funcs[i]) (res1, res1, in2);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL (dss, i, in1, in2, NULL);
mpz_set (res1, in2);
(dss_funcs[i]) (res1, in1, res1);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL (dss, i, in1, in2, NULL);
}
for (i = 0; i < numberof (ddss_div_funcs); i++)
{
if (ddss_div_funcs[i] == 0)
continue;
if (mpz_sgn (in2) == 0)
continue;
(ddss_div_funcs[i]) (ref1, ref2, in1, in2);
mpz_check_format (ref1);
mpz_check_format (ref2);
mpz_set (res1, in1);
(ddss_div_funcs[i]) (res1, res2, res1, in2);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0)
FAIL (ddss_div, i, in1, in2, NULL);
mpz_set (res2, in1);
(ddss_div_funcs[i]) (res1, res2, res2, in2);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0)
FAIL (ddss_div, i, in1, in2, NULL);
mpz_set (res1, in2);
(ddss_div_funcs[i]) (res1, res2, in1, res1);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0)
FAIL (ddss_div, i, in1, in2, NULL);
mpz_set (res2, in2);
(ddss_div_funcs[i]) (res1, res2, in1, res2);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0)
FAIL (ddss_div, i, in1, in2, NULL);
}
for (i = 0; i < numberof (ds_funcs); i++)
{
if (ds_funcs[i] == 0)
continue;
if (strcmp (ds_func_names[i], "mpz_sqrt") == 0
&& mpz_sgn (in1) < 0)
continue;
(ds_funcs[i]) (ref1, in1);
mpz_check_format (ref1);
mpz_set (res1, in1);
(ds_funcs[i]) (res1, res1);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL (ds, i, in1, in2, NULL);
}
in2i = mpz_get_ui (in2);
for (i = 0; i < numberof (dsi_funcs); i++)
{
if (dsi_funcs[i] == 0)
continue;
if (strcmp (dsi_func_names[i], "mpz_fdiv_q_2exp") == 0)
/* Limit exponent to something reasonable for the division
functions. Without this, we'd normally shift things off
the end and just generate the trivial values 1, 0, -1. */
in2i %= 0x1000;
if (strcmp (dsi_func_names[i], "mpz_mul_2exp") == 0)
/* Limit exponent more for mpz_mul_2exp to save time. */
in2i %= 0x100;
if (strcmp (dsi_func_names[i], "mpz_pow_ui") == 0)
/* Limit exponent yet more for mpz_pow_ui to save time. */
in2i %= 0x10;
(dsi_funcs[i]) (ref1, in1, in2i);
mpz_check_format (ref1);
mpz_set (res1, in1);
(dsi_funcs[i]) (res1, res1, in2i);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL (dsi, i, in1, in2, NULL);
}
if (in2i != 0) /* Don't divide by 0. */
{
for (i = 0; i < numberof (dsi_div_funcs); i++)
{
r1 = (dsi_div_funcs[i]) (ref1, in1, in2i);
mpz_check_format (ref1);
mpz_set (res1, in1);
r2 = (dsi_div_funcs[i]) (res1, res1, in2i);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0 || r1 != r2)
FAIL (dsi_div, i, in1, in2, NULL);
}
for (i = 0; i < numberof (ddsi_div_funcs); i++)
{
r1 = (ddsi_div_funcs[i]) (ref1, ref2, in1, in2i);
mpz_check_format (ref1);
mpz_set (res1, in1);
r2 = (ddsi_div_funcs[i]) (res1, res2, res1, in2i);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0 || r1 != r2)
FAIL (ddsi_div, i, in1, in2, NULL);
mpz_set (res2, in1);
(ddsi_div_funcs[i]) (res1, res2, res2, in2i);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0 || r1 != r2)
FAIL (ddsi_div, i, in1, in2, NULL);
}
}
if (mpz_sgn (in1) >= 0)
{
mpz_sqrtrem (ref1, ref2, in1);
mpz_check_format (ref1);
mpz_check_format (ref2);
mpz_set (res1, in1);
mpz_sqrtrem (res1, res2, res1);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0)
FAIL2 (mpz_sqrtrem, in1, NULL, NULL);
mpz_set (res2, in1);
mpz_sqrtrem (res1, res2, res2);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0)
FAIL2 (mpz_sqrtrem, in1, NULL, NULL);
}
if (mpz_sgn (in1) >= 0)
{
mpz_root (ref1, in1, in2i % 0x1000 + 1);
mpz_check_format (ref1);
mpz_set (res1, in1);
mpz_root (res1, res1, in2i % 0x1000 + 1);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_root, in1, in2, NULL);
}
if (mpz_sgn (in1) >= 0)
{
mpz_rootrem (ref1, ref2, in1, in2i % 0x1000 + 1);
mpz_check_format (ref1);
mpz_check_format (ref2);
mpz_set (res1, in1);
mpz_rootrem (res1, res2, res1, in2i % 0x1000 + 1);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0)
FAIL2 (mpz_rootrem, in1, in2, NULL);
mpz_set (res2, in1);
mpz_rootrem (res1, res2, res2, in2i % 0x1000 + 1);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0)
FAIL2 (mpz_rootrem, in1, in2, NULL);
}
if (pass < reps / 2) /* run fewer tests since gcdext lots of time */
{
mpz_gcdext (ref1, ref2, ref3, in1, in2);
mpz_check_format (ref1);
mpz_check_format (ref2);
mpz_check_format (ref3);
mpz_set (res1, in1);
mpz_gcdext (res1, res2, res3, res1, in2);
mpz_check_format (res1);
mpz_check_format (res2);
mpz_check_format (res3);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res2, in1);
mpz_gcdext (res1, res2, res3, res2, in2);
mpz_check_format (res1);
mpz_check_format (res2);
mpz_check_format (res3);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res3, in1);
mpz_gcdext (res1, res2, res3, res3, in2);
mpz_check_format (res1);
mpz_check_format (res2);
mpz_check_format (res3);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res1, in2);
mpz_gcdext (res1, res2, res3, in1, res1);
mpz_check_format (res1);
mpz_check_format (res2);
mpz_check_format (res3);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res2, in2);
mpz_gcdext (res1, res2, res3, in1, res2);
mpz_check_format (res1);
mpz_check_format (res2);
mpz_check_format (res3);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res3, in2);
mpz_gcdext (res1, res2, res3, in1, res3);
mpz_check_format (res1);
mpz_check_format (res2);
mpz_check_format (res3);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res1, in1);
mpz_gcdext (res1, res2, NULL, res1, in2);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res2, in1);
mpz_gcdext (res1, res2, NULL, res2, in2);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res1, in2);
mpz_gcdext (res1, res2, NULL, in1, res1);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
mpz_set (res2, in2);
mpz_gcdext (res1, res2, NULL, in1, res2);
mpz_check_format (res1);
mpz_check_format (res2);
if (mpz_cmp (ref1, res1) != 0 || mpz_cmp (ref2, res2) != 0
|| mpz_cmp (ref3, res3) != 0)
FAIL2 (mpz_gcdext, in1, in2, NULL);
}
/* Don't run mpz_powm for huge exponents or when undefined. */
if (mpz_sizeinbase (in2, 2) < 250 && mpz_sgn (in3) != 0
&& (mpz_sgn (in2) >= 0 || mpz_invert (t, in1, in3)))
{
mpz_powm (ref1, in1, in2, in3);
mpz_check_format (ref1);
mpz_set (res1, in1);
mpz_powm (res1, res1, in2, in3);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_powm, in1, in2, in3);
mpz_set (res1, in2);
mpz_powm (res1, in1, res1, in3);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_powm, in1, in2, in3);
mpz_set (res1, in3);
mpz_powm (res1, in1, in2, res1);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_powm, in1, in2, in3);
}
/* Don't run mpz_powm_ui when undefined. */
if (mpz_sgn (in3) != 0)
{
mpz_powm_ui (ref1, in1, in2i, in3);
mpz_check_format (ref1);
mpz_set (res1, in1);
mpz_powm_ui (res1, res1, in2i, in3);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_powm_ui, in1, in2, in3);
mpz_set (res1, in3);
mpz_powm_ui (res1, in1, in2i, res1);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_powm_ui, in1, in2, in3);
}
{
r1 = mpz_gcd_ui (ref1, in1, in2i);
mpz_check_format (ref1);
mpz_set (res1, in1);
r2 = mpz_gcd_ui (res1, res1, in2i);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_gcd_ui, in1, in2, NULL);
}
#if 0
if (mpz_cmp_ui (in2, 1L) > 0 && mpz_sgn (in1) != 0)
{
/* Test mpz_remove */
mpz_remove (ref1, in1, in2);
mpz_check_format (ref1);
mpz_set (res1, in1);
mpz_remove (res1, res1, in2);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_remove, in1, in2, NULL);
mpz_set (res1, in2);
mpz_remove (res1, in1, res1);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_remove, in1, in2, NULL);
}
#endif
if (mpz_sgn (in2) != 0)
{
/* Test mpz_divexact */
mpz_mul (t, in1, in2);
mpz_divexact (ref1, t, in2);
mpz_check_format (ref1);
mpz_set (res1, t);
mpz_divexact (res1, res1, in2);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_divexact, t, in2, NULL);
mpz_set (res1, in2);
mpz_divexact (res1, t, res1);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_divexact, t, in2, NULL);
}
#if 0
if (mpz_sgn (in2) > 0)
{
/* Test mpz_divexact_gcd, same as mpz_divexact */
mpz_mul (t, in1, in2);
mpz_divexact_gcd (ref1, t, in2);
mpz_check_format (ref1);
mpz_set (res1, t);
mpz_divexact_gcd (res1, res1, in2);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_divexact_gcd, t, in2, NULL);
mpz_set (res1, in2);
mpz_divexact_gcd (res1, t, res1);
mpz_check_format (res1);
if (mpz_cmp (ref1, res1) != 0)
FAIL2 (mpz_divexact_gcd, t, in2, NULL);
}
#endif
}
if (failures != 0)
{
fprintf (stderr, "mpz/reuse: %ld error%s\n", failures, "s" + (failures == 1));
exit (1);
}
mpz_clear (bs);
mpz_clear (in1);
mpz_clear (in2);
mpz_clear (in3);
mpz_clear (ref1);
mpz_clear (ref2);
mpz_clear (ref3);
mpz_clear (res1);
mpz_clear (res2);
mpz_clear (res3);
mpz_clear (t);
}
void
dump3 (const char *name, mpz_t in1, mpz_t in2, mpz_t in3)
{
printf ("failure in %s (", name);
mpz_out_str (stdout, -16, in1);
if (in2 != NULL)
{
printf (" ");
mpz_out_str (stdout, -16, in2);
}
if (in3 != NULL)
{
printf (" ");
mpz_out_str (stdout, -16, in3);
}
printf (")\n");
}
void
mpz_check_format (const mpz_t x)
{
mp_size_t n = x ->_mp_size;
if (n < 0)
n = - n;
if (n > x->_mp_alloc)
{
fprintf (stderr, "mpz_t size exceeds allocation!\n");
abort ();
}
if (n > 0 && x->_mp_d[n-1] == 0)
{
fprintf (stderr, "Unnormalized mpz_t!\n");
abort ();
}
}