Refactored Ed25519 to use the stdlib's KeyPair type for better readability, separate generate() and new() identity functions, new update_hash() function for identity and compilation errors fixes

This commit is contained in:
Gu://em_ 2026-07-29 18:15:54 +02:00
parent 3209ae87a1
commit 69bad53a03
3 changed files with 46 additions and 26 deletions

View file

@ -2,9 +2,13 @@
const CryptoError = @import("../crypto.zig").CryptoError;
// Constants and default implementation
const Ed25519 = @import("std").crypto.sign.Ed25519;
const secureZero = @import("std").crypto.secureZero;
// Random number generation
const rngProvider = @import("random.zig").Implementation;
// Keypair type
pub const KeyPair = Ed25519.KeyPair;
pub const secret_key_length: comptime_int = Ed25519.SecretKey.encoded_length;
pub const public_key_length: comptime_int = Ed25519.PublicKey.encoded_length;
pub const signature_length : comptime_int = Ed25519.Signature.encoded_length;
@ -12,30 +16,28 @@ pub const seed_length : comptime_int = Ed25519.Signature.encoded_length;
pub const Implementation = struct {
generateKeys: *const fn (pubkey_buffer: ?*[public_key_length]u8, privkey_buffer: *[secret_key_length]u8, rng: rngProvider) CryptoError!void,
generateKeys: *const fn (destination: *KeyPair, rng: rngProvider) CryptoError!void,
sign: *const fn (key: *const [secret_key_length]u8, data: []const u8, signature_out: *const [signature_length]u8) CryptoError!void,
verify: *const fn (key: *const [public_key_length]u8, signature: *const [signature_length]u8, data: []const u8) CryptoError!void,
};
pub fn generateKeys(pubkey_buffer: ?*[public_key_length]u8, privkey_buffer: *[secret_key_length]u8, rng: rngProvider) !void {
pub fn generateKeys (destination: *KeyPair, rng: rngProvider) !void {
var seed_buffer: [Ed25519.KeyPair.seed_length]u8 = undefined;
var seed_buffer: [KeyPair.seed_length]u8 = undefined;
defer secureZero(@TypeOf(seed_buffer), seed_buffer);
rng.generate(&seed_buffer);
const keypair: Ed25519.KeyPair = Ed25519.KeyPair.generateDeterministic(seed_buffer) catch return CryptoError.FailedKeyGeneration;
if (pubkey_buffer != null) @memcpy(pubkey_buffer.?[0..public_key_length], &keypair.public_key.bytes);
@memcpy(privkey_buffer[0..secret_key_length], &keypair.secret_key.bytes);
destination.* = KeyPair.generateDeterministic(seed_buffer) catch return CryptoError.FailedKeyGeneration;
}
pub fn sign(key: *const [secret_key_length]u8, data: []const u8, signature_out: *const [signature_length]u8) !void {
pub fn sign (key: *const [secret_key_length]u8, data: []const u8, signature_out: *const [signature_length]u8) !void {
_ = key;
_ = data;
_ = signature_out;
return CryptoError.NotImplemented;
}
pub fn verify(key: *const [public_key_length]u8, signature: *const [signature_length]u8, data: []const u8) !void {
pub fn verify (key: *const [public_key_length]u8, signature: *const [signature_length]u8, data: []const u8) !void {
// TODO maybe find a more efficient way to do that (excessive copies)
const sig = Ed25519.Signature.fromBytes(signature.*);
const pubkey = Ed25519.PublicKey.fromBytes(key.*) catch return CryptoError.InvalidKey;

View file

@ -6,6 +6,9 @@ const Ed25519 = @import("std").crypto.sign.Ed25519;
// Random number generation
const rngProvider = @import("random.zig").Implementation;
// Keypair type
pub const KeyPair = X25519.KeyPair;
pub const secret_key_length: comptime_int = X25519.secret_length;
pub const public_key_length: comptime_int = X25519.public_length;
pub const seed_length: comptime_int = X25519.seed_length;
@ -13,9 +16,9 @@ pub const seed_length: comptime_int = X25519.seed_length;
pub const Implementation = struct {
generateKeys: *const fn (pubkey_buffer: ?*[public_key_length]u8, privkey_buffer: *[secret_key_length]u8, rng: rngProvider) CryptoError!void,
fromEd25519: *const fn (ed25519_keypair: *Ed25519.Keypair, x25519_keypair: *X25519.KeyPair, rng: rngProvider) CryptoError!void,
fromEd25519: *const fn (ed25519_keypair: *Ed25519.KeyPair, x25519_keypair: *KeyPair) CryptoError!void,
// TODO
// maybe switch to a Keypair system based on the stdlib implementation
// Switch to a Keypair system based on the stdlib implementation
// generateKeys: *const fn (keypair: *X25519.KeyPair, rng: rngProvider) CryptoError!void,
};
@ -25,12 +28,12 @@ fn generateKeys (pubkey_buffer: ?*[public_key_length]u8, privkey_buffer: *[secre
rng.generate(&seed_buffer);
const keypair: X25519.KeyPair = X25519.KeyPair.generateDeterministic(seed_buffer) catch return CryptoError.FailedKeyGeneration;
if (pubkey_buffer != null) @memcpy(pubkey_buffer.?[0..public_key_length], &keypair.public_key.bytes);
@memcpy(privkey_buffer[0..secret_key_length], &keypair.secret_key.bytes);
if (pubkey_buffer != null) @memcpy(pubkey_buffer.?[0..public_key_length], &keypair.public_key);
@memcpy(privkey_buffer[0..secret_key_length], &keypair.secret_key);
}
fn fromEd25519 (ed25519_keypair: *Ed25519.Keypair, x25519_keypair: *X25519.KeyPair) CryptoError!void {
x25519_keypair.* = X25519.KeyPair.fromEd25519(ed25519_keypair);
fn fromEd25519 (ed25519_keypair: *Ed25519.KeyPair, x25519_keypair: *KeyPair) CryptoError!void {
x25519_keypair.* = X25519.KeyPair.fromEd25519(ed25519_keypair.*) catch return CryptoError.InvalidKey;
}
pub const defaultImplementation: Implementation = .{

View file

@ -21,18 +21,31 @@ pub fn resolveIdentity(comptime crypto_engine: Crypto.Engine) type {
//////// Fields
//
privKey: [Crypto.Ed25519.secret_key_length]u8,
signatureKey: [Crypto.X25519.secret_key_size]u8,
hash: []const u8,
keys: Crypto.Ed25519.KeyPair,
signatureKeys: Crypto.X25519.KeyPair,
hash: [Crypto.Sha256.hash_len]u8,
//////// Functions
//
pub fn generate() !Self {
pub fn new() !Self {
const result: Self = undefined;
crypto_engine.ed25519.generateKeys(null, &result.signatureKey, crypto_engine.random);
// TODO privKey and signature
crypto_engine.ed25519.generateKeys(result.keys, crypto_engine.random);
crypto_engine.x25519.fromEd25519(result.keys, result.signatureKeys, crypto_engine.random);
result.update_hash();
return result;
}
pub fn generate(self: *Self) !void {
crypto_engine.ed25519.generateKeys(self.keys, crypto_engine.random);
crypto_engine.x25519.fromEd25519(self.keys, self.signatureKeys, crypto_engine.random);
self.update_hash();
}
pub fn update_hash(self: *Self) void {
//WARNING PUBLIC KEY SHOULD BE TRUNCATED
// We should define the truncated hash length somewhere
crypto_engine.sha256.hash(self.keys.public_key, self.hash);
}
};
}
@ -59,10 +72,10 @@ fn fillPattern(out: []u8) void {
}
}
pub fn generateFakeKeys(pubkey_buffer: *[pubkey_size]u8, privkey_buffer: *[privkey_size]u8, rng: Crypto.Random.Implementation) Crypto.CryptoError!void {
pub fn generateFakeKeys(destination: *Crypto.Ed25519.KeyPair, rng: Crypto.Random.Implementation) Crypto.CryptoError!void {
_ = rng;
fillPattern(pubkey_buffer);
fillPattern(privkey_buffer);
fillPattern(&destination.public_key.bytes);
fillPattern(&destination.secret_key.bytes);
}
test "Basic identity generation" {
@ -72,6 +85,8 @@ test "Basic identity generation" {
};
const fake_ed25519: Crypto.Ed25519.Implementation = .{
.generateKeys = generateFakeKeys,
.sign = undefined,
.verify = undefined,
};
const crypto_engine = Crypto.resolveEngine(.{
.random = fake_rng,
@ -79,8 +94,8 @@ test "Basic identity generation" {
});
const Identity = resolveIdentity(crypto_engine);
const new_id: Identity = try Identity.generate();
const new_id: Identity = undefined;
try new_id.generate();
try expect(memeql(u8, new_id.privKey, testPattern));
try expect(memeql(u8, new_id.signatureKey, testPattern));