mleku
e0a95ca1cd
- Replaced all instances of p256k1signer with the new p8k.Signer across various modules, including event creation, policy handling, and database interactions. - Updated related test cases and benchmarks to ensure compatibility with the new signer interface. - Bumped version to v0.25.0 to reflect these significant changes and improvements in cryptographic operations.
87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package authenvelope
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"lol.mleku.dev/chk"
|
|
"next.orly.dev/pkg/interfaces/signer/p8k"
|
|
"next.orly.dev/pkg/encoders/envelopes"
|
|
"next.orly.dev/pkg/protocol/auth"
|
|
"next.orly.dev/pkg/utils"
|
|
"next.orly.dev/pkg/utils/bufpool"
|
|
)
|
|
|
|
const relayURL = "wss://example.com"
|
|
|
|
func TestAuth(t *testing.T) {
|
|
var err error
|
|
signer := p8k.MustNew()
|
|
if err = signer.Generate(); chk.E(err) {
|
|
t.Fatal(err)
|
|
}
|
|
for _ = range 1000 {
|
|
b1, b2, b3, b4 := bufpool.Get(), bufpool.Get(), bufpool.Get(), bufpool.Get()
|
|
ch := auth.GenerateChallenge()
|
|
chal := Challenge{Challenge: ch}
|
|
b1 = chal.Marshal(b1)
|
|
oChal := make([]byte, len(b1))
|
|
copy(oChal, b1)
|
|
var rem []byte
|
|
var l string
|
|
if l, b1, err = envelopes.Identify(b1); chk.E(err) {
|
|
t.Fatal(err)
|
|
}
|
|
if l != L {
|
|
t.Fatalf("invalid sentinel %s, expect %s", l, L)
|
|
}
|
|
c2 := NewChallenge()
|
|
if rem, err = c2.Unmarshal(b1); chk.E(err) {
|
|
t.Fatal(err)
|
|
}
|
|
if len(rem) != 0 {
|
|
t.Fatalf("remainder should be empty\n%s", rem)
|
|
}
|
|
if !utils.FastEqual(chal.Challenge, c2.Challenge) {
|
|
t.Fatalf(
|
|
"challenge mismatch\n%s\n%s",
|
|
chal.Challenge, c2.Challenge,
|
|
)
|
|
}
|
|
b2 = c2.Marshal(b2)
|
|
if !utils.FastEqual(oChal, b2) {
|
|
t.Fatalf("challenge mismatch\n%s\n%s", oChal, b2)
|
|
}
|
|
resp := Response{
|
|
Event: auth.CreateUnsigned(
|
|
signer.Pub(), ch,
|
|
relayURL,
|
|
),
|
|
}
|
|
if err = resp.Event.Sign(signer); chk.E(err) {
|
|
t.Fatal(err)
|
|
}
|
|
b3 = resp.Marshal(b3)
|
|
oResp := make([]byte, len(b3))
|
|
copy(oResp, b3)
|
|
if l, b3, err = envelopes.Identify(b3); chk.E(err) {
|
|
t.Fatal(err)
|
|
}
|
|
if l != L {
|
|
t.Fatalf("invalid sentinel %s, expect %s", l, L)
|
|
}
|
|
r2 := NewResponse()
|
|
if _, err = r2.Unmarshal(b3); chk.E(err) {
|
|
t.Fatal(err)
|
|
}
|
|
b4 = r2.Marshal(b4)
|
|
if !utils.FastEqual(oResp, b4) {
|
|
t.Fatalf("challenge mismatch\n%s\n%s", oResp, b4)
|
|
}
|
|
bufpool.Put(b1)
|
|
bufpool.Put(b2)
|
|
bufpool.Put(b3)
|
|
bufpool.Put(b4)
|
|
oChal, oResp = oChal[:0], oResp[:0]
|
|
}
|
|
}
|