rixxc/mlkem_agent
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add encapsulate

Browse Source
This commit is contained in:
Aaron Kaiser
2024-08-27 16:49:13 +02:00
parent 099d4e4f84
commit c898843120
1 changed files with 26 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/agent.jazz
View File

@@ -1,5 +1,7 @@
from Jade require "crypto_kem/kyber_kyber768_avx2/kyber_kyber768_avx2.jazz"
param int MLKEM_SK_SIZE = 2400;
inline fn extract_nth_ptr(reg u64 sync_mem shared_mem, inline int i) -> reg u64 {
reg u64 p;
@@ -27,10 +29,10 @@ inline fn generate_keypair(reg u64 shared_mem sync_mem private_mem, stack u64 ke
coins, num_bytes = #randombytes(coins, flag);
} (num_bytes != 2*KYBER_SYMBYTES)
skptr = key_id * 2400;
skptr = key_id * MLKEM_SK_SIZE;
skptr += private_mem;
for i=0 to 8 {
for i=0 to (2*KYBER_SYMBYTES)/8 {
(u64)[skptr + i * 8] = coins[u64 i];
}
@@ -46,6 +48,27 @@ inline fn generate_keypair(reg u64 shared_mem sync_mem private_mem, stack u64 ke
return key_id;
}
inline fn encapsulate(reg u64 shared_mem sync_mem private_mem) {
inline int i;
reg u64 ssptr ctptr pkptr flag num_bytes;
stack u8[KYBER_SYMBYTES] coins;
while {
flag = 0;
coins, num_bytes = #randombytes(coins, flag);
} (num_bytes != KYBER_SYMBYTES)
for i=0 to KYBER_SYMBYTES/8 {
(u64)[ssptr + i * 8] = coins[u64 i];
}
ctptr = extract_nth_ptr(sync_mem, shared_mem, 0);
ssptr = extract_nth_ptr(sync_mem, shared_mem, 1);
pkptr = extract_nth_ptr(sync_mem, shared_mem, 2);
jade_kem_kyber_kyber768_amd64_avx2_enc_derand(ctptr, ssptr, pkptr, ssptr);
}
export fn agent_start(reg u64 shared_mem sync_mem private_mem_fd) {
stack u64 key_id;
reg u64 private_mem;
@@ -87,6 +110,7 @@ export fn agent_start(reg u64 shared_mem sync_mem private_mem_fd) {
key_id = generate_keypair(shared_mem, sync_mem, private_mem, key_id);
} else {
if (id == 1) {
encapsulate(shared_mem, sync_mem, private_mem);
} else {
if (id == 2) {
}