93 lines
2.5 KiB
C
93 lines
2.5 KiB
C
#include "tommath_private.h"
|
|
#ifdef S_MP_SQR_KARATSUBA_C
|
|
/* LibTomMath, multiple-precision integer library -- Tom St Denis */
|
|
/* SPDX-License-Identifier: Unlicense */
|
|
|
|
/* Karatsuba squaring, computes b = a*a using three
|
|
* half size squarings
|
|
*
|
|
* See comments of mul_karatsuba for details. It
|
|
* is essentially the same algorithm but merely
|
|
* tuned to perform recursive squarings.
|
|
*/
|
|
mp_err s_mp_sqr_karatsuba(const mp_int *a, mp_int *b)
|
|
{
|
|
mp_int x0, x1, t1, t2, x0x0, x1x1;
|
|
int B;
|
|
mp_err err;
|
|
|
|
/* min # of digits */
|
|
B = a->used;
|
|
|
|
/* now divide in two */
|
|
B = B >> 1;
|
|
|
|
/* init copy all the temps */
|
|
if ((err = mp_init_size(&x0, B)) != MP_OKAY)
|
|
goto LBL_ERR;
|
|
if ((err = mp_init_size(&x1, a->used - B)) != MP_OKAY)
|
|
goto X0;
|
|
|
|
/* init temps */
|
|
if ((err = mp_init_size(&t1, a->used * 2)) != MP_OKAY)
|
|
goto X1;
|
|
if ((err = mp_init_size(&t2, a->used * 2)) != MP_OKAY)
|
|
goto T1;
|
|
if ((err = mp_init_size(&x0x0, B * 2)) != MP_OKAY)
|
|
goto T2;
|
|
if ((err = mp_init_size(&x1x1, (a->used - B) * 2)) != MP_OKAY)
|
|
goto X0X0;
|
|
|
|
/* now shift the digits */
|
|
x0.used = B;
|
|
x1.used = a->used - B;
|
|
s_mp_copy_digs(x0.dp, a->dp, x0.used);
|
|
s_mp_copy_digs(x1.dp, a->dp + B, x1.used);
|
|
mp_clamp(&x0);
|
|
|
|
/* now calc the products x0*x0 and x1*x1 */
|
|
if ((err = mp_sqr(&x0, &x0x0)) != MP_OKAY)
|
|
goto X1X1; /* x0x0 = x0*x0 */
|
|
if ((err = mp_sqr(&x1, &x1x1)) != MP_OKAY)
|
|
goto X1X1; /* x1x1 = x1*x1 */
|
|
|
|
/* now calc (x1+x0)**2 */
|
|
if ((err = s_mp_add(&x1, &x0, &t1)) != MP_OKAY)
|
|
goto X1X1; /* t1 = x1 - x0 */
|
|
if ((err = mp_sqr(&t1, &t1)) != MP_OKAY)
|
|
goto X1X1; /* t1 = (x1 - x0) * (x1 - x0) */
|
|
|
|
/* add x0y0 */
|
|
if ((err = s_mp_add(&x0x0, &x1x1, &t2)) != MP_OKAY)
|
|
goto X1X1; /* t2 = x0x0 + x1x1 */
|
|
if ((err = s_mp_sub(&t1, &t2, &t1)) != MP_OKAY)
|
|
goto X1X1; /* t1 = (x1+x0)**2 - (x0x0 + x1x1) */
|
|
|
|
/* shift by B */
|
|
if ((err = mp_lshd(&t1, B)) != MP_OKAY)
|
|
goto X1X1; /* t1 = (x0x0 + x1x1 - (x1-x0)*(x1-x0))<<B */
|
|
if ((err = mp_lshd(&x1x1, B * 2)) != MP_OKAY)
|
|
goto X1X1; /* x1x1 = x1x1 << 2*B */
|
|
|
|
if ((err = mp_add(&x0x0, &t1, &t1)) != MP_OKAY)
|
|
goto X1X1; /* t1 = x0x0 + t1 */
|
|
if ((err = mp_add(&t1, &x1x1, b)) != MP_OKAY)
|
|
goto X1X1; /* t1 = x0x0 + t1 + x1x1 */
|
|
|
|
X1X1:
|
|
mp_clear(&x1x1);
|
|
X0X0:
|
|
mp_clear(&x0x0);
|
|
T2:
|
|
mp_clear(&t2);
|
|
T1:
|
|
mp_clear(&t1);
|
|
X1:
|
|
mp_clear(&x1);
|
|
X0:
|
|
mp_clear(&x0);
|
|
LBL_ERR:
|
|
return err;
|
|
}
|
|
#endif
|