From 8671550a5d028b0e59fe20aff5fe49ff55fdff22 Mon Sep 17 00:00:00 2001
From: Aaron Kaiser <a_kaiser@posteo.de>
Date: Tue, 23 Apr 2024 12:37:58 +0200
Subject: [PATCH] Properly scale keyid

---
 src/agent.jazz | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/agent.jazz b/src/agent.jazz
index 0d7902e..ddf4acd 100644
--- a/src/agent.jazz
+++ b/src/agent.jazz
@@ -51,13 +51,14 @@ inline fn generate_key(reg u64 shared_mem sync_mem private_mem, stack u64 key_id
 
 inline fn x25519(reg u64 shared_mem sync_mem private_mem) {
     reg u64 outptr idptr pkptr keyptr;
-    stack u64 keyid;
+    reg u64 keyid;
 
     outptr = extract_nth_ptr(sync_mem, shared_mem, 0);
     idptr = extract_nth_ptr(sync_mem, shared_mem, 1);
     pkptr = extract_nth_ptr(sync_mem, shared_mem, 2);
 
     keyid = (u64)[idptr];
+    keyid *= 8;
     keyptr = private_mem + keyid;
 
     jade_scalarmult_curve25519_amd64_mulx(outptr, keyptr, pkptr);