mirror of
https://github.com/henrydcase/pqc.git
synced 2024-11-23 07:59:01 +00:00
107 lines
3.1 KiB
C
107 lines
3.1 KiB
C
/*---------------------------------------------------------------------
|
||
This file has been adapted from the implementation
|
||
(available at, Public Domain https://github.com/pq-crystals/kyber)
|
||
of "CRYSTALS – Kyber: a CCA-secure module-lattice-based KEM"
|
||
by : Joppe Bos, Leo Ducas, Eike Kiltz, Tancrede Lepoint,
|
||
Vadim Lyubashevsky, John M. Schanck, Peter Schwabe & Damien stehle
|
||
----------------------------------------------------------------------*/
|
||
|
||
#include "SABER_params.h"
|
||
#include "api.h"
|
||
#include "cbd.h"
|
||
#include <stdint.h>
|
||
|
||
static uint64_t load_littleendian(const unsigned char *x, int bytes) {
|
||
int i;
|
||
uint64_t r = x[0];
|
||
for (i = 1; i < bytes; i++) {
|
||
r |= (uint64_t)x[i] << (8 * i);
|
||
}
|
||
return r;
|
||
}
|
||
|
||
|
||
void PQCLEAN_SABER_CLEAN_cbd(uint16_t *r, const unsigned char *buf) {
|
||
uint16_t Qmod_minus1 = SABER_Q - 1;
|
||
|
||
#if Saber_type == 3
|
||
uint32_t t, d, a[4], b[4];
|
||
int i, j;
|
||
|
||
for (i = 0; i < SABER_N / 4; i++) {
|
||
t = load_littleendian(buf + 3 * i, 3);
|
||
d = 0;
|
||
for (j = 0; j < 3; j++) {
|
||
d += (t >> j) & 0x249249;
|
||
}
|
||
|
||
a[0] = d & 0x7;
|
||
b[0] = (d >> 3) & 0x7;
|
||
a[1] = (d >> 6) & 0x7;
|
||
b[1] = (d >> 9) & 0x7;
|
||
a[2] = (d >> 12) & 0x7;
|
||
b[2] = (d >> 15) & 0x7;
|
||
a[3] = (d >> 18) & 0x7;
|
||
b[3] = (d >> 21);
|
||
|
||
r[4 * i + 0] = (uint16_t)(a[0] - b[0]) & Qmod_minus1;
|
||
r[4 * i + 1] = (uint16_t)(a[1] - b[1]) & Qmod_minus1;
|
||
r[4 * i + 2] = (uint16_t)(a[2] - b[2]) & Qmod_minus1;
|
||
r[4 * i + 3] = (uint16_t)(a[3] - b[3]) & Qmod_minus1;
|
||
|
||
}
|
||
#elif Saber_type == 2
|
||
uint32_t t, d, a[4], b[4];
|
||
int i, j;
|
||
|
||
for (i = 0; i < SABER_N / 4; i++) {
|
||
t = (uint32_t) load_littleendian(buf + 4 * i, 4);
|
||
d = 0;
|
||
for (j = 0; j < 4; j++) {
|
||
d += (t >> j) & 0x11111111;
|
||
}
|
||
|
||
a[0] = d & 0xf;
|
||
b[0] = (d >> 4) & 0xf;
|
||
a[1] = (d >> 8) & 0xf;
|
||
b[1] = (d >> 12) & 0xf;
|
||
a[2] = (d >> 16) & 0xf;
|
||
b[2] = (d >> 20) & 0xf;
|
||
a[3] = (d >> 24) & 0xf;
|
||
b[3] = (d >> 28);
|
||
|
||
r[4 * i + 0] = (uint16_t)(a[0] - b[0]) & Qmod_minus1;
|
||
r[4 * i + 1] = (uint16_t)(a[1] - b[1]) & Qmod_minus1;
|
||
r[4 * i + 2] = (uint16_t)(a[2] - b[2]) & Qmod_minus1;
|
||
r[4 * i + 3] = (uint16_t)(a[3] - b[3]) & Qmod_minus1;
|
||
}
|
||
#elif Saber_type == 1
|
||
uint64_t t, d, a[4], b[4];
|
||
int i, j;
|
||
|
||
for (i = 0; i < SABER_N / 4; i++) {
|
||
t = load_littleendian(buf + 5 * i, 5);
|
||
d = 0;
|
||
for (j = 0; j < 5; j++) {
|
||
d += (t >> j) & 0x0842108421UL;
|
||
}
|
||
|
||
a[0] = d & 0x1f;
|
||
b[0] = (d >> 5) & 0x1f;
|
||
a[1] = (d >> 10) & 0x1f;
|
||
b[1] = (d >> 15) & 0x1f;
|
||
a[2] = (d >> 20) & 0x1f;
|
||
b[2] = (d >> 25) & 0x1f;
|
||
a[3] = (d >> 30) & 0x1f;
|
||
b[3] = (d >> 35);
|
||
|
||
r[4 * i + 0] = (uint16_t)(a[0] - b[0]) & Qmod_minus1;
|
||
r[4 * i + 1] = (uint16_t)(a[1] - b[1]) & Qmod_minus1;
|
||
r[4 * i + 2] = (uint16_t)(a[2] - b[2]) & Qmod_minus1;
|
||
r[4 * i + 3] = (uint16_t)(a[3] - b[3]) & Qmod_minus1;
|
||
}
|
||
#else
|
||
#error "Unsupported SABER parameter."
|
||
#endif
|
||
}
|