admin/mruby-mruby
1
0
Fork 0
mirror of https://github.com/mruby/mruby synced 2026-06-08 16:11:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

mruby-bigint (mrb_bint_powm): fixed bugs in type casting

Browse Source
This commit is contained in:
Yukihiro "Matz" Matsumoto
2024-10-24 07:21:12 +09:00
parent 0b96f7f2f7
commit f245943aed
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mrbgems/mruby-bigint/core/bigint.c
+10 -7
View File
@@ -1744,15 +1744,18 @@ mrb_bint_powm(mrb_state *mrb, mrb_value x, mrb_value exp, mrb_value mod)
mpz_t a, b, c, z;
bint_as_mpz(RBIGINT(x), &a);
if (mrb_bigint_p(mod)) {
bint_as_mpz(RBIGINT(mod), &c);
if (zero_p(&c)) mrb_int_zerodiv(mrb);
}
else {
if (mrb_integer_p(mod)) {
mrb_int m = mrb_integer(mod);
if (m == 0) mrb_int_zerodiv(mrb);
mpz_init_set_int(mrb, &c, m);
}
else {
mod = mrb_as_bint(mrb, mod);
bint_as_mpz(RBIGINT(mod), &c);
if (zero_p(&c) || uzero_p(&c)) {
mrb_int_zerodiv(mrb);
}
}
mpz_init(mrb, &z);
if (mrb_bigint_p(exp)) {
bint_as_mpz(RBIGINT(exp), &b);
@@ -1764,11 +1767,11 @@ mrb_bint_powm(mrb_state *mrb, mrb_value x, mrb_value exp, mrb_value mod)
if (e < 0) goto raise;
mpz_powm_i(mrb, &z, &a, e, &c);
}
if (!mrb_bigint_p(mod)) mpz_clear(mrb, &c);
if (mrb_integer_p(mod)) mpz_clear(mrb, &c);
return bint_norm(mrb, bint_new(mrb, &z));
raise:
if (!mrb_bigint_p(mod)) mpz_clear(mrb, &c);
if (mrb_integer_p(mod)) mpz_clear(mrb, &c);
mrb_raise(mrb, E_ARGUMENT_ERROR, "int.pow(n,m): n must be positive");
}