mirror of
https://github.com/luxfi/fhe.git
synced 2026-07-26 23:16:08 +00:00
Antje Worring
ac2762f412
BootstrapKey.MarshalBinary was incomplete and broken: - BRK (an interface) was gob-encoded as interface but decoded into a concrete field, so UnmarshalBinary always failed. Serialize the concrete MemBlindRotationEvaluationKeySet on both sides (symmetric). - KSK (key-switching key) and 5 of 9 test polynomials were never serialized, so a deserialized key could not bootstrap. Serialize KSK via rlwe.EvaluationKey Marshal and all 9 test polys, each with a presence flag (nil-safe). RadixCiphertext had no serialization at all; add lossless, byte-stable MarshalBinary/UnmarshalBinary mirroring the BitCiphertext pattern. Tests: BootstrapKey round-trip (was uncovered) and RadixCiphertext round-trip + byte-stability across u8/u16/u32. Full root suite green.
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
// Copyright (c) 2025, Lux Industries Inc
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package fhe
|
|
|
|
import "testing"
|
|
|
|
// TestBootstrapKey_MarshalRoundTrip proves a BootstrapKey survives a
|
|
// MarshalBinary/UnmarshalBinary round trip. The BRK field is an interface whose
|
|
// concrete type must be gob-registered; this test pins that the registration is
|
|
// in place (regression guard for the BRK decode path).
|
|
func TestBootstrapKey_MarshalRoundTrip(t *testing.T) {
|
|
params, err := NewParametersFromLiteral(PN10QP27)
|
|
if err != nil {
|
|
t.Fatalf("params: %v", err)
|
|
}
|
|
kg := NewKeyGenerator(params)
|
|
sk := kg.GenSecretKey()
|
|
bsk := kg.GenBootstrapKey(sk)
|
|
|
|
data, err := bsk.MarshalBinary()
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
if len(data) == 0 {
|
|
t.Fatal("marshal produced no bytes")
|
|
}
|
|
|
|
got := new(BootstrapKey)
|
|
if err := got.UnmarshalBinary(data); err != nil {
|
|
t.Fatalf("unmarshal: %v", err)
|
|
}
|
|
|
|
if got.BRK == nil {
|
|
t.Fatal("BRK is nil after round trip")
|
|
}
|
|
for name, p := range map[string]interface{}{
|
|
"TestPolyAND": got.TestPolyAND,
|
|
"TestPolyOR": got.TestPolyOR,
|
|
"TestPolyNAND": got.TestPolyNAND,
|
|
"TestPolyNOR": got.TestPolyNOR,
|
|
} {
|
|
if p == nil {
|
|
t.Fatalf("%s is nil after round trip", name)
|
|
}
|
|
}
|
|
}
|