52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package bundle
|
|
|
|
import (
|
|
"crypto/ed25519"
|
|
"crypto/rand"
|
|
"testing"
|
|
)
|
|
|
|
func TestBuildVerifyRoundtrip(t *testing.T) {
|
|
_, priv, err := ed25519.GenerateKey(rand.Reader)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
frag := map[string]string{
|
|
"bird.conf": "router id 192.0.2.1;\n\nprotocol device {\n}\n\nprotocol direct {\n ipv4;\n ipv6;\n}\n",
|
|
}
|
|
raw, err := BuildGzippedTar("rev-1", "sp-1", frag, priv)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
pub := priv.Public().(ed25519.PublicKey)
|
|
v, err := VerifyGzippedTar(raw, pub)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if v.Manifest.RevisionID != "rev-1" {
|
|
t.Fatalf("revision %q", v.Manifest.RevisionID)
|
|
}
|
|
if len(v.Files) != 1 {
|
|
t.Fatalf("files %d", len(v.Files))
|
|
}
|
|
}
|
|
|
|
func TestVerifyWrongKey(t *testing.T) {
|
|
_, privA, err := ed25519.GenerateKey(rand.Reader)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, privB, err := ed25519.GenerateKey(rand.Reader)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
raw, err := BuildGzippedTar("rev-x", "sp-x", map[string]string{"a.conf": "x"}, privA)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
pubB := privB.Public().(ed25519.PublicKey)
|
|
if _, err := VerifyGzippedTar(raw, pubB); err == nil {
|
|
t.Fatal("expected signature error with wrong public key")
|
|
}
|
|
}
|