1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
package checkpoint
import (
"crypto/ed25519"
"crypto/rand"
"encoding/base64"
"os"
"path/filepath"
"testing"
"github.com/Fuwn/plutia/internal/storage"
"github.com/Fuwn/plutia/internal/types"
)
func TestBuildAndStoreCheckpoint(t *testing.T) {
tmp := t.TempDir()
store, err := storage.OpenPebble(tmp)
if err != nil {
t.Fatalf("open pebble: %v", err)
}
defer store.Close()
pub, priv, err := ed25519.GenerateKey(rand.Reader)
if err != nil {
t.Fatalf("generate key: %v", err)
}
keyPath := filepath.Join(tmp, "mirror.key")
if err := os.WriteFile(keyPath, []byte(base64.RawURLEncoding.EncodeToString(priv)), 0o600); err != nil {
t.Fatalf("write key: %v", err)
}
mgr := NewManager(store, tmp, keyPath)
states := []types.StateV1{
{Version: 1, DID: "did:plc:a", ChainTipHash: "tip-a"},
{Version: 1, DID: "did:plc:b", ChainTipHash: "tip-b"},
}
cp, err := mgr.BuildAndStore(42, states, []string{"abc", "def"})
if err != nil {
t.Fatalf("build checkpoint: %v", err)
}
if cp.Signature == "" || cp.CheckpointHash == "" {
t.Fatalf("expected signed checkpoint")
}
payload, err := marshalCheckpointPayload(cp)
if err != nil {
t.Fatalf("marshal payload: %v", err)
}
rawSig, err := base64.RawURLEncoding.DecodeString(cp.Signature)
if err != nil {
t.Fatalf("decode signature: %v", err)
}
if !ed25519.Verify(pub, payload, rawSig) {
t.Fatalf("signature verification failed")
}
stored, ok, err := store.GetCheckpoint(cp.Sequence)
if err != nil || !ok {
t.Fatalf("stored checkpoint missing: ok=%v err=%v", ok, err)
}
if stored.CheckpointHash != cp.CheckpointHash {
t.Fatalf("checkpoint hash mismatch")
}
}
func TestCheckpointRootStability(t *testing.T) {
tmp := t.TempDir()
store, err := storage.OpenPebble(tmp)
if err != nil {
t.Fatalf("open pebble: %v", err)
}
defer store.Close()
if err := store.PutState(types.StateV1{Version: 1, DID: "did:plc:b", ChainTipHash: "tip-b"}); err != nil {
t.Fatalf("put state b: %v", err)
}
if err := store.PutState(types.StateV1{Version: 1, DID: "did:plc:a", ChainTipHash: "tip-a"}); err != nil {
t.Fatalf("put state a: %v", err)
}
_, priv, err := ed25519.GenerateKey(rand.Reader)
if err != nil {
t.Fatalf("generate key: %v", err)
}
keyPath := filepath.Join(tmp, "mirror.key")
if err := os.WriteFile(keyPath, []byte(base64.RawURLEncoding.EncodeToString(priv)), 0o600); err != nil {
t.Fatalf("write key: %v", err)
}
mgr := NewManager(store, tmp, keyPath)
cp1, err := mgr.BuildAndStoreFromStore(100, []string{"abc"})
if err != nil {
t.Fatalf("build checkpoint 1: %v", err)
}
cp2, err := mgr.BuildAndStoreFromStore(200, []string{"abc"})
if err != nil {
t.Fatalf("build checkpoint 2: %v", err)
}
if cp1.DIDMerkleRoot != cp2.DIDMerkleRoot {
t.Fatalf("did root changed for identical state: %s vs %s", cp1.DIDMerkleRoot, cp2.DIDMerkleRoot)
}
if cp1.BlockMerkleRoot != cp2.BlockMerkleRoot {
t.Fatalf("block root changed for identical block set: %s vs %s", cp1.BlockMerkleRoot, cp2.BlockMerkleRoot)
}
}
|