mirror of
https://github.com/henrydcase/pqc.git
synced 2024-11-27 01:41:40 +00:00
0d7743d576
* Update NTRU version: https://github.com/jschanck/ntru/tree/485dde03 * Fixed ntruhrss701/clean/Makefile.Microsoft_nmake
114 lines
3.0 KiB
C
114 lines
3.0 KiB
C
#include "poly.h"
|
|
#include "verify.h"
|
|
|
|
#define POLY_R2_ADD(I,A,B,S) \
|
|
for ((I)=0; (I)<NTRU_N; (I)++) { \
|
|
(A).coeffs[(I)] ^= (B).coeffs[(I)] * (S); \
|
|
}
|
|
|
|
static void cswappoly(poly *a, poly *b, int swap) {
|
|
int i;
|
|
uint16_t t;
|
|
swap = -swap;
|
|
for (i = 0; i < NTRU_N; i++) {
|
|
t = (a->coeffs[i] ^ b->coeffs[i]) & swap;
|
|
a->coeffs[i] ^= t;
|
|
b->coeffs[i] ^= t;
|
|
}
|
|
}
|
|
|
|
static inline void poly_divx(poly *a, int s) {
|
|
int i;
|
|
|
|
for (i = 1; i < NTRU_N; i++) {
|
|
a->coeffs[i - 1] = (unsigned char) ((s * a->coeffs[i]) | (!s * a->coeffs[i - 1]));
|
|
}
|
|
a->coeffs[NTRU_N - 1] = (!s * a->coeffs[NTRU_N - 1]);
|
|
}
|
|
|
|
static inline void poly_mulx(poly *a, int s) {
|
|
int i;
|
|
|
|
for (i = 1; i < NTRU_N; i++) {
|
|
a->coeffs[NTRU_N - i] = (unsigned char) ((s * a->coeffs[NTRU_N - i - 1]) | (!s * a->coeffs[NTRU_N - i]));
|
|
}
|
|
a->coeffs[0] = (!s * a->coeffs[0]);
|
|
}
|
|
|
|
void PQCLEAN_NTRUHPS2048677_CLEAN_poly_R2_inv(poly *r, const poly *a) {
|
|
/* Schroeppel--Orman--O'Malley--Spatscheck
|
|
* "Almost Inverse" algorithm as described
|
|
* by Silverman in NTRU Tech Report #14 */
|
|
// with several modifications to make it run in constant-time
|
|
int i, j;
|
|
int k = 0;
|
|
uint16_t degf = NTRU_N - 1;
|
|
uint16_t degg = NTRU_N - 1;
|
|
int sign, t, swap;
|
|
int16_t done = 0;
|
|
poly b, f, g;
|
|
poly *c = r; // save some stack space
|
|
poly *temp_r = &f;
|
|
|
|
/* b(X) := 1 */
|
|
for (i = 1; i < NTRU_N; i++) {
|
|
b.coeffs[i] = 0;
|
|
}
|
|
b.coeffs[0] = 1;
|
|
|
|
/* c(X) := 0 */
|
|
for (i = 0; i < NTRU_N; i++) {
|
|
c->coeffs[i] = 0;
|
|
}
|
|
|
|
/* f(X) := a(X) */
|
|
for (i = 0; i < NTRU_N; i++) {
|
|
f.coeffs[i] = a->coeffs[i] & 1;
|
|
}
|
|
|
|
/* g(X) := 1 + X + X^2 + ... + X^{N-1} */
|
|
for (i = 0; i < NTRU_N; i++) {
|
|
g.coeffs[i] = 1;
|
|
}
|
|
|
|
for (j = 0; j < 2 * (NTRU_N - 1) - 1; j++) {
|
|
sign = f.coeffs[0];
|
|
swap = sign & !done & ((degf - degg) >> 15);
|
|
|
|
cswappoly(&f, &g, swap);
|
|
cswappoly(&b, c, swap);
|
|
t = (degf ^ degg) & (-swap);
|
|
degf ^= t;
|
|
degg ^= t;
|
|
|
|
POLY_R2_ADD(i, f, g, sign * (!done));
|
|
POLY_R2_ADD(i, b, (*c), sign * (!done));
|
|
|
|
poly_divx(&f, !done);
|
|
poly_mulx(c, !done);
|
|
degf -= !done;
|
|
k += !done;
|
|
|
|
done = 1 - (((uint16_t) - degf) >> 15);
|
|
}
|
|
|
|
k = k - NTRU_N * ((uint16_t)(NTRU_N - k - 1) >> 15);
|
|
|
|
/* Return X^{N-k} * b(X) */
|
|
/* This is a k-coefficient rotation. We do this by looking at the binary
|
|
representation of k, rotating for every power of 2, and performing a cmov
|
|
if the respective bit is set. */
|
|
for (i = 0; i < NTRU_N; i++) {
|
|
r->coeffs[i] = b.coeffs[i];
|
|
}
|
|
|
|
for (i = 0; i < 10; i++) {
|
|
for (j = 0; j < NTRU_N; j++) {
|
|
temp_r->coeffs[j] = r->coeffs[(j + (1 << i)) % NTRU_N];
|
|
}
|
|
PQCLEAN_NTRUHPS2048677_CLEAN_cmov((unsigned char *) & (r->coeffs),
|
|
(unsigned char *) & (temp_r->coeffs), sizeof(uint16_t) * NTRU_N, k & 1);
|
|
k >>= 1;
|
|
}
|
|
}
|