From dfc00d5bd63005abf1ccde9fd4e66551dbfc86b6 Mon Sep 17 00:00:00 2001
From: Aaron Kaiser <a_kaiser@posteo.de>
Date: Mon, 22 Apr 2024 11:05:39 +0200
Subject: [PATCH] Return pubkey

---
 src/agent.jazz | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/agent.jazz b/src/agent.jazz
index 32cba63..08f13ea 100644
--- a/src/agent.jazz
+++ b/src/agent.jazz
@@ -11,7 +11,7 @@ inline fn extract_ipc_id(reg u64 sync_mem) -> reg u64 {
 inline fn generate_key(reg u64 shared_mem sync_mem private_mem key_id) -> reg u64 {
     inline int i;
     stack u8[32] key;
-    reg u64 addr outptr;
+    reg u64 addr idptr pkptr;
 
     key = #randombytes(key);
     key[0] &= 248;
@@ -27,10 +27,15 @@ inline fn generate_key(reg u64 shared_mem sync_mem private_mem key_id) -> reg u6
         (u64)[private_mem + i * 8] = key[u64 i];
     }
 
-    outptr = (u64)[sync_mem + 16];
-    outptr += shared_mem;
+    idptr = (u64)[sync_mem + 16];
+    idptr += shared_mem;
 
-    (u64)[outptr] = key_id;
+    (u64)[idptr] = key_id;
+
+    pkptr = (u64)[sync_mem + 24];
+    pkptr += shared_mem;
+
+    jade_scalarmult_curve25519_mulx_base(pkptr, private_mem);
 
     return key_id;
 }