Heap math: mp_add/submod_ct make work when c == d

mp_addmod_ct and mp_submod_ct expected c and d to be different pointers.
Change code to support this use case.
Fix whitespace.
This commit is contained in:
Sean Parkinson
2023-11-22 07:36:28 +10:00
parent dda72dc19c
commit 416ce54276

View File

@ -3089,10 +3089,28 @@ int mp_addmod(mp_int* a, mp_int* b, mp_int* c, mp_int* d)
int mp_submod_ct(mp_int* a, mp_int* b, mp_int* c, mp_int* d)
{
int res;
mp_int t;
mp_int* r = d;
res = mp_sub(a, b, d);
if (res == MP_OKAY && mp_isneg(d)) {
res = mp_add(d, c, d);
if (c == d) {
r = &t;
if ((res = mp_init (r)) != MP_OKAY) {
return res;
}
}
res = mp_sub (a, b, r);
if (res == MP_OKAY) {
if (mp_isneg (r)) {
res = mp_add (r, c, d);
} else if (c == d) {
res = mp_copy (r, d);
}
}
if (c == d) {
mp_clear (r);
}
return res;
@ -3102,10 +3120,28 @@ int mp_submod_ct(mp_int* a, mp_int* b, mp_int* c, mp_int* d)
int mp_addmod_ct(mp_int* a, mp_int* b, mp_int* c, mp_int* d)
{
int res;
mp_int t;
mp_int* r = d;
res = mp_add(a, b, d);
if (res == MP_OKAY && mp_cmp(d, c) != MP_LT) {
res = mp_sub(d, c, d);
if (c == d) {
r = &t;
if ((res = mp_init (r)) != MP_OKAY) {
return res;
}
}
res = mp_add (a, b, r);
if (res == MP_OKAY) {
if (mp_cmp (r, c) != MP_LT) {
res = mp_sub (r, c, d);
} else if (c == d) {
res = mp_copy (r, d);
}
}
if (c == d) {
mp_clear (r);
}
return res;