mirror of
https://github.com/luxfi/crypto.git
synced 2026-07-27 01:54:50 +00:00
Zach Kelling
a9769373ca
- Fix bn256/gnark imports to use crypto/bitutil instead of standalone bitutil - Remove unused secp256k1 import from crypto.go - Fix mlkem test files to use correct API (GenerateKeyPair returns pub,priv,err) - Fix mlkem constant names (MLKEM512SharedKeySize not SharedSecretSize) - Restore crypto/common types as standalone (not aliases to geth/common) - Update crypto.go and keccak.go to use crypto/common instead of geth/common - Fix secp256k1 fuzz tests to use correct DecompressPubkey/CompressPubkey API - Remove stale test files and go.work files crypto package is now fully independent of geth (geth -> crypto, not reverse)
113 lines
2.0 KiB
Go
113 lines
2.0 KiB
Go
package mlkem
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkMLKEM512(b *testing.B) {
|
|
benchmarkMLKEM(b, MLKEM512)
|
|
}
|
|
|
|
func BenchmarkMLKEM768(b *testing.B) {
|
|
benchmarkMLKEM(b, MLKEM768)
|
|
}
|
|
|
|
func BenchmarkMLKEM1024(b *testing.B) {
|
|
benchmarkMLKEM(b, MLKEM1024)
|
|
}
|
|
|
|
func benchmarkMLKEM(b *testing.B, mode Mode) {
|
|
b.Run("GenerateKeyPair", func(b *testing.B) {
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
_, _, err := GenerateKeyPair(rand.Reader, mode)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
})
|
|
|
|
pub, priv, err := GenerateKeyPair(rand.Reader, mode)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
b.Run("Encapsulate", func(b *testing.B) {
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
_, _, err := pub.Encapsulate(rand.Reader)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
})
|
|
|
|
ciphertext, _, err := pub.Encapsulate(rand.Reader)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
b.Run("Decapsulate", func(b *testing.B) {
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := priv.Decapsulate(ciphertext)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
})
|
|
|
|
b.Run("Serialize", func(b *testing.B) {
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
_ = priv.Bytes()
|
|
_ = pub.Bytes()
|
|
}
|
|
})
|
|
|
|
privBytes := priv.Bytes()
|
|
pubBytes := pub.Bytes()
|
|
|
|
b.Run("Deserialize", func(b *testing.B) {
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := PrivateKeyFromBytes(privBytes, mode)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
_, err = PublicKeyFromBytes(pubBytes, mode)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
// Memory usage benchmark
|
|
func BenchmarkMLKEMMemory(b *testing.B) {
|
|
modes := []struct {
|
|
name string
|
|
mode Mode
|
|
}{
|
|
{"MLKEM512", MLKEM512},
|
|
{"MLKEM768", MLKEM768},
|
|
{"MLKEM1024", MLKEM1024},
|
|
}
|
|
|
|
for _, m := range modes {
|
|
b.Run(m.name, func(b *testing.B) {
|
|
b.ReportAllocs()
|
|
pub, priv, _ := GenerateKeyPair(rand.Reader, m.mode)
|
|
ciphertext, _, _ := pub.Encapsulate(rand.Reader)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
// Full KEM operation
|
|
_, _, _ = pub.Encapsulate(rand.Reader)
|
|
_, _ = priv.Decapsulate(ciphertext)
|
|
}
|
|
})
|
|
}
|
|
}
|