Skip to content

Commit

Permalink
decrypt
Browse files Browse the repository at this point in the history
  • Loading branch information
@jlmucb
jlmucb committed Apr 21, 2024
1 parent fceb712 commit 8054e74
Show file tree
Hide file tree
Showing 3 changed files with 103 additions and 24 deletions.
14 changes: 10 additions & 4 deletions v2/include/kyber.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,11 +26,11 @@ int decompress(int q, int x, int d);

byte bit_from_ints(int bits_in_int, int bit_numb, int* pi);
byte bit_from_bytes(int bit_numb, byte* buf);
bool byte_encode(int d, int n, int* pi, int* out_len, byte* out);
bool byte_encode(int d, int n, int* pi, byte* out);
bool byte_decode(int d, int n, int in_len, byte* in, int* pi);

byte bit_from_int_vector(int bits_in_int, int bit_numb, vector<int>& v);
bool byte_encode_from_vector(int d, int n, vector<int>& v, int* out_len, byte* out);
bool byte_encode_from_vector(int d, int n, vector<int>& v, byte* out);
bool byte_decode_to_vector(int d, int n, int in_len, byte* in, vector<int>& v);

bool G(int in_len, byte* in, int bit_out_len, byte* out);
Expand Down Expand Up @@ -152,8 +152,14 @@ bool module_apply_array(module_array& A, module_vector& v, module_vector* out);
bool module_vector_is_zero(module_vector& in);
bool make_module_vector_zero(module_vector* out);
bool module_vector_equal(module_vector& in1, module_vector& in2);
bool module_vector_dot_product(module_vector& in1, module_vector& in2, coefficient_vector* out);
bool module_vector_dot_product_first_transposed(module_vector& in1, module_vector& in2, coefficient_vector* out);
bool module_vector_dot_product(module_vector& in1, module_vector& in2,
coefficient_vector* out);
bool module_vector_dot_product_first_transposed(module_vector& in1,
module_vector& in2, coefficient_vector* out);
bool ntt_module_vector_dot_product(module_vector& in1,
module_vector& in2, coefficient_vector* out);
bool ntt_module_vector_dot_product_first_transposed(module_vector& in1,
module_vector& in2, coefficient_vector* out);
void print_module_vector(module_vector& mv);

bool ntt_module_apply_array(int g, module_array& A, module_vector& v, module_vector* out);
Expand Down
109 changes: 91 additions & 18 deletions v2/kyber/kyber.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -339,11 +339,85 @@ bool ntt_module_apply_array_transpose(int g, module_array& A, module_vector& v,

bool module_vector_dot_product(module_vector& in1, module_vector& in2,
coefficient_vector* out) {

if (!coefficient_vector_zero(out)) {
return false;
}
for (int i = 0; i < in1.dim_; i++) {
coefficient_vector t(in1.q_, in1.n_);
if (!coefficient_vector_zero(&t)) {
return false;
}
if (!coefficient_mult(*in1.c_[i], *in2.c_[i], &t)) {
return false;
}
if (!coefficient_vector_add_to(t, out)) {
return false;
}
}
return true;
}

bool module_vector_dot_product_first_transposed(module_vector& in1,
module_vector& in2, coefficient_vector* out) {

if (!coefficient_vector_zero(out)) {
return false;
}
for (int i = 0; i < in1.dim_; i++) {
coefficient_vector t(in1.q_, in1.n_);
if (!coefficient_vector_zero(&t)) {
return false;
}
if (!coefficient_mult(*in1.c_[in1.dim_ - 1 -i], *in2.c_[i], &t)) {
return false;
}
if (!coefficient_vector_add_to(t, out)) {
return false;
}
}
return true;
}

bool ntt_module_vector_dot_product(module_vector& in1, module_vector& in2,
coefficient_vector* out) {

if (!coefficient_vector_zero(out)) {
return false;
}
for (int i = 0; i < in1.dim_; i++) {
coefficient_vector t(in1.q_, in1.n_);
if (!coefficient_vector_zero(&t)) {
return false;
}
if (!multiply_ntt(17, *in1.c_[i], *in2.c_[i], &t)) {
return false;
}
if (!coefficient_vector_add_to(t, out)) {
return false;
}
}
return true;
}

bool ntt_module_vector_dot_product_first_transposed(module_vector& in1,
module_vector& in2, coefficient_vector* out) {

if (!coefficient_vector_zero(out)) {
return false;
}
for (int i = 0; i < in1.dim_; i++) {
coefficient_vector t(in1.q_, in1.n_);
if (!coefficient_vector_zero(&t)) {
return false;
}
if (!multiply_ntt(17, *in1.c_[in1.dim_ - 1 - i], *in2.c_[i], &t)) {
return false;
}
if (!coefficient_vector_add_to(t, out)) {
return false;
}
}
return true;
}

Expand Down Expand Up @@ -736,7 +810,7 @@ byte bit_from_bytes(int bit_numb, byte* buf) {
}

// encode n d-bit integers into byte array
bool byte_encode(int d, int n, int* pi, int* out_len, byte* out) {
bool byte_encode(int d, int n, int* pi, byte* out) {
int num_bits = d * n;
byte t = 0;
byte r = 0;
Expand Down Expand Up @@ -780,13 +854,14 @@ bool byte_decode(int d, int n, int in_len, byte* in, int* pi) {
}

// encode n d-bit integers into byte array
bool byte_encode_from_vector(int d, int n, vector<int>& v, int* out_len, byte* out) {
bool byte_encode_from_vector(int d, int n, vector<int>& v, byte* out) {
int num_bits = d * n;
byte t = 0;
byte r = 0;
int k = 0; // bit position in output byte
int m = 0; // current output byte number
memset(out, 0, ((d * n) + NBITSINBYTE - 1) / NBITSINBYTE);
int out_len = (num_bits + NBITSINBYTE - 1) / NBITSINBYTE;
memset(out, 0, out_len);
for (int i = 0; i < num_bits; i++) {
t = (int)bit_from_int_vector(d, i, v);
r |= t << k;
Expand Down Expand Up @@ -911,7 +986,7 @@ bool kyber_keygen(int g, kyber_parameters& p, int* ek_len, byte* ek,
memset(b_prf, 0, b_prf_len);

if (!prf(p.eta1_, 32, &parameters[32], sizeof(int), (byte*)&N,
NBITSINBYTE * 64 * p.eta1_, b_prf)) {
NBITSINBYTE * 64 * p.eta1_, b_prf)) {
printf("kyber_keygen: prf (1) failed\n");
return false;
}
Expand All @@ -926,7 +1001,7 @@ bool kyber_keygen(int g, kyber_parameters& p, int* ek_len, byte* ek,
byte b_prf[b_prf_len];
memset(b_prf, 0, b_prf_len);
if (!prf(p.eta1_, 32, &parameters[32], sizeof(int), (byte*)&N,
NBITSINBYTE * 64 * p.eta1_, b_prf)) {
NBITSINBYTE * 64 * p.eta1_, b_prf)) {
printf("kyber_keygen: prf (1) failed\n");
return false;
}
Expand Down Expand Up @@ -968,7 +1043,7 @@ bool kyber_keygen(int g, kyber_parameters& p, int* ek_len, byte* ek,

// ek := byte_encode(12) (t^) || rho
for (int i = 0; i < t_ntt.dim_; i++) {
if (!byte_encode_from_vector(12, p.n_, t_ntt.c_[i]->c_, ek_len, ek)) {
if (!byte_encode_from_vector(12, p.n_, t_ntt.c_[i]->c_, ek)) {
printf("kyber_keygen: byte_encode (2) failed\n");
return false;
}
Expand All @@ -978,7 +1053,7 @@ bool kyber_keygen(int g, kyber_parameters& p, int* ek_len, byte* ek,

// dk := byte_encode(12) (s^)
for (int i = 0; i < s_ntt.dim_; i++) {
if (!byte_encode_from_vector(12, p.n_, s_ntt.c_[i]->c_, dk_len, dk)) {
if (!byte_encode_from_vector(12, p.n_, s_ntt.c_[i]->c_, dk)) {
printf("kyber_keygen: byte_encode (1) failed\n");
return false;
}
Expand Down Expand Up @@ -1096,7 +1171,7 @@ bool kyber_encrypt(int g, kyber_parameters& p, int ek_len, byte* ek,
memset(b_prf, 0, b_prf_len);

if (!prf(p.eta1_, 32, rho, sizeof(int), (byte*)&N,
NBITSINBYTE * 64 * p.eta1_, b_prf)) {
NBITSINBYTE * 64 * p.eta1_, b_prf)) {
printf("kyber_encrypt: prf (1) failed\n");
return false;
}
Expand All @@ -1112,7 +1187,7 @@ bool kyber_encrypt(int g, kyber_parameters& p, int ek_len, byte* ek,
memset(b_prf, 0, b_prf_len);

if (!prf(p.eta2_, 32, rho, sizeof(int), (byte*)&N,
NBITSINBYTE * 64 * p.eta2_, b_prf)) {
NBITSINBYTE * 64 * p.eta2_, b_prf)) {
printf("kyber_encrypt: prf (1) failed\n");
return false;
}
Expand All @@ -1128,7 +1203,7 @@ bool kyber_encrypt(int g, kyber_parameters& p, int ek_len, byte* ek,
memset(b_prf, 0, b_prf_len);

if (!prf(p.eta2_, 32, rho, sizeof(int), (byte*)&N,
NBITSINBYTE * 64 * p.eta2_, b_prf)) {
NBITSINBYTE * 64 * p.eta2_, b_prf)) {
printf("kyber_encrypt: prf (1) failed\n");
return false;
}
Expand Down Expand Up @@ -1179,7 +1254,7 @@ bool kyber_encrypt(int g, kyber_parameters& p, int ek_len, byte* ek,
if (!coefficient_vector_zero(&mu)) {
return false;
}
if (!module_vector_dot_product(t_ntt, r_ntt, &nu_ntt)) {
if (!ntt_module_vector_dot_product(t_ntt, r_ntt, &nu_ntt)) {
return false;
}
if (!ntt_inv(g, nu_ntt, &nu)) {
Expand Down Expand Up @@ -1208,10 +1283,10 @@ bool kyber_encrypt(int g, kyber_parameters& p, int ek_len, byte* ek,
byte* pp = b_c1;
for (int i = 0; i < p.k_; i++) {
int len = (p.du_ * 256) / NBITSINBYTE;
if (!byte_encode_from_vector(p.du_, 256, compressed_u.c_[i]->c_, &len, pp)) {
if (!byte_encode_from_vector(p.du_, 256, compressed_u.c_[i]->c_, pp)) {
return false;
}
pp += len;
pp += p.du_ * 32;
}

coefficient_vector compressed_nu(p.q_, p.n_);
Expand All @@ -1220,7 +1295,7 @@ bool kyber_encrypt(int g, kyber_parameters& p, int ek_len, byte* ek,
for (int j = 0; j < compressed_nu.len_; j++) {
compressed_nu.c_[j] = compress(p.q_, nu.c_[j], p.dv_);
}
if (!byte_encode_from_vector(p.dv_, 256, compressed_nu.c_, &c2_b_len, b_c2)) {
if (!byte_encode_from_vector(p.dv_, 256, compressed_nu.c_, b_c2)) {
return false;
}

Expand Down Expand Up @@ -1293,7 +1368,7 @@ bool kyber_decrypt(int g, kyber_parameters& p, int dk_len, byte* dk,
printf("kyber_decrypt: ntt (1) failed\n");
return false;
}
if (!module_vector_dot_product_first_transposed(s_transposed, u_ntt, products.c_[i])) {
if (!ntt_module_vector_dot_product_first_transposed(s_transposed, u_ntt, products.c_[i])) {
return false;
}

Expand All @@ -1313,11 +1388,9 @@ bool kyber_decrypt(int g, kyber_parameters& p, int dk_len, byte* dk,
for (int j = 0; j < p.n_; j++) {
w.c_[0]->c_[j]= compress(p.q_, w.c_[0]->c_[j], 1);
}
#if 0
if (!byte_encode_from_vector(1, p.n_, m_vec.c_, m_len, m)) {
if (!byte_encode_from_vector(1, 256, m_vec.c_, m)) {
return false;
}
#endif
return true;
}

Expand Down
4 changes: 2 additions & 2 deletions v2/kyber/test_kyber.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ bool test_kyber1() {
int recovered_m_len = 32;
byte recovered_m[m_len];
memset(recovered_m, 0, recovered_m_len);
if (!kyber_decrypt(g, p, dk_len, dk, c_len, c, s, &recovered_m_len, recovered_m)) {
if (!kyber_decrypt(g, p, dk_len, dk, c_len, c, &recovered_m_len, recovered_m)) {
printf("Could not init kyber_decrypt\n");
return false;
}
Expand Down Expand Up @@ -247,7 +247,7 @@ bool test_kyber_support() {
if (FLAGS_print_all) {
printf("\n");
}
if (!byte_encode(dd, 4, int_in, &b_out_len, b_out)) {
if (!byte_encode(dd, 4, int_in, b_out)) {
printf("byte_encode failed\n");
return false;
}
Expand Down

0 comments on commit 8054e74

Please sign in to comment.