cryb-to/lib/mpi/mpi_add.c
2015-01-07 18:59:44 +00:00

127 lines
3.4 KiB
C

/*
* Copyright (c) 2014 Dag-Erling Smørgrav
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. The name of the author may not be used to endorse or promote
* products derived from this software without specific prior written
* permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include "cryb/impl.h"
#include <stdint.h>
#include <stdlib.h>
#include <cryb/mpi.h>
/*
* Store the sum of the absolutes values of A and B in X.
*/
int
mpi_add_abs(cryb_mpi *X, cryb_mpi *A, cryb_mpi *B)
{
unsigned int i;
uint32_t c;
/*
* Trivial cases: A and B are identical and / or both zero.
*/
if (A->msb == 0 && B->msb == 0)
return (0);
if (A == B) {
if (X != A && mpi_copy(X, A) != 0)
return (-1);
return (mpi_lshift(X, 1));
}
/*
* Normalize our operands: if X is identical to either A or B, we
* want it to be A. Otherwise, copy A into X. In either case,
* make sure X is large enough for the largest possible result.
*/
if (X == B) {
B = A;
A = X;
}
if (mpi_grow(X, A->msb + 1) != 0 || mpi_grow(X, B->msb + 1) != 0)
return (-1);
if (X != A)
/* this cannot fail */
mpi_copy(X, A);
/*
* From now on, we are adding B to X and A is irrelevant.
*/
/* add B into X word by word until we run out of B */
for (c = i = 0; i < (B->msb + 31) / 32; ++i) {
X->words[i] += c;
c = (X->words[i] < c);
X->words[i] += B->words[i];
c += (X->words[i] < B->words[i]);
}
/* keep propagating carry */
while (c) {
X->words[i] += c;
c = (X->words[i] < c);
++i;
}
if (X->words[i] == 0)
--i;
/* compute msb of msw */
/* XXX should use flsl() */
for (X->msb = 31; X->msb > 0; --X->msb)
if (X->words[i] & (1 << X->msb))
break;
/* add msw offset */
X->msb += i * 32 + 1;
return (0);
}
/*
* Add two numbers together.
*/
int
mpi_add(cryb_mpi *X, cryb_mpi *A, cryb_mpi *B)
{
if (A->neg && B->neg) {
if (mpi_add_abs(X, A, B) < 0)
return (-1);
X->neg = 1;
} else if (A->neg) {
if (mpi_sub_abs(X, A, B) < 0)
return (-1);
X->neg = (mpi_cmp_abs(A, B) < 0);
} else if (B->neg) {
if (mpi_sub_abs(X, A, B) < 0)
return (-1);
X->neg = (mpi_cmp_abs(A, B) > 0);
} else {
if (mpi_add_abs(X, A, B) < 0)
return (-1);
X->neg = 0;
}
return (0);
}