package time_test import ( "encoding/hex" "fmt" "strings" "sync" "testing" "github.com/cloudflare/circl/sign/ed448" "github.com/iden3/go-iden3-crypto/poseidon" "github.com/libp2p/go-libp2p/core/crypto" "github.com/libp2p/go-libp2p/core/peer" "github.com/stretchr/testify/assert" "go.uber.org/zap" "source.quilibrium.com/quilibrium/monorepo/node/config" "source.quilibrium.com/quilibrium/monorepo/node/consensus/time" qcrypto "source.quilibrium.com/quilibrium/monorepo/node/crypto" "source.quilibrium.com/quilibrium/monorepo/node/keys" "source.quilibrium.com/quilibrium/monorepo/node/protobufs" "source.quilibrium.com/quilibrium/monorepo/node/store" "source.quilibrium.com/quilibrium/monorepo/node/tries" ) func generateTestProvers() ( keys.KeyManager, []peer.ID, [][]byte, [][]byte, map[string]string, *tries.RollingFrecencyCritbitTrie, ) { keyManager := keys.NewInMemoryKeyManager() peers := []peer.ID{} pubKeys := [][]byte{} privKeys := [][]byte{} addrMap := map[string]string{} for i := 0; i < 1000; i++ { keyManager.CreateSigningKey( fmt.Sprintf("test-key-%d", i), keys.KeyTypeEd448, ) k, err := keyManager.GetRawKey(fmt.Sprintf("test-key-%d", i)) if err != nil { panic(err) } privKey, err := crypto.UnmarshalEd448PrivateKey([]byte(k.PrivateKey)) if err != nil { panic(err) } privKeys = append(privKeys, []byte(k.PrivateKey)) pub := privKey.GetPublic() id, err := peer.IDFromPublicKey(pub) if err != nil { panic(err) } peers = append(peers, id) keyManager.CreateSigningKey( fmt.Sprintf("proving-key-%d", i), keys.KeyTypeEd448, ) pk, err := keyManager.GetRawKey(fmt.Sprintf("proving-key-%d", i)) if err != nil { panic(err) } pprivKey, err := crypto.UnmarshalEd448PrivateKey([]byte(pk.PrivateKey)) if err != nil { panic(err) } ppub := pprivKey.GetPublic() ppubKey, err := ppub.Raw() if err != nil { panic(err) } pubKeys = append(pubKeys, ppubKey) } proverTrie := &tries.RollingFrecencyCritbitTrie{} for i, s := range pubKeys { addr, err := poseidon.HashBytes(s) if err != nil { panic(err) } addrBytes := addr.Bytes() addrBytes = append(make([]byte, 32-len(addrBytes)), addrBytes...) proverTrie.Add(addrBytes, 0) addrMap[string(addrBytes)] = fmt.Sprintf("proving-key-%d", i) } return keyManager, peers, pubKeys, privKeys, addrMap, proverTrie } func TestDataTimeReel(t *testing.T) { logger, _ := zap.NewProduction() db := store.NewInMemKVDB() clockStore := store.NewPebbleClockStore(db, logger) prover := qcrypto.NewWesolowskiFrameProver(logger) filter := "ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff" keyManager, _, pubKeys, _, addrMap, proverTrie := generateTestProvers() // We're going to set this up by churning 100 master frames so we don't // have to zig zag on master and data frames to confirm data time reel // behaviors m := time.NewMasterTimeReel( logger, clockStore, &config.EngineConfig{ Filter: filter, GenesisSeed: strings.Repeat("00", 516), Difficulty: 10, }, prover, ) err := m.Start() assert.NoError(t, err) frame, err := m.Head() assert.NoError(t, err) frames := []*protobufs.ClockFrame{} wg := sync.WaitGroup{} wg.Add(1) frameCh := m.NewFrameCh() go func() { for i := 0; i < 100; i++ { frames = append(frames, <-frameCh) } wg.Done() }() // in order for i := int64(0); i < 100; i++ { frame, err = prover.ProveMasterClockFrame(frame, i+1, 10) assert.NoError(t, err) err := m.Insert(frame) assert.NoError(t, err) } wg.Wait() for i := 0; i < 100; i++ { assert.NotNil(t, frames[i]) assert.Equal(t, frames[i].FrameNumber, uint64(i+1)) } filterBytes, _ := hex.DecodeString(filter) // Ok, now onto the data time reel. We're going to test the following // scenarios: // 1. Z-dist optimal, in order // 2. Z-dist optimal, out of order // 3. 90% optimal, out of order // 4. Malicious majority, out of order d := time.NewDataTimeReel( filterBytes, logger, clockStore, &config.EngineConfig{ Filter: filter, GenesisSeed: strings.Repeat("00", 516), Difficulty: 10, }, prover, frames[0].Output, &qcrypto.InclusionAggregateProof{ InclusionCommitments: []*qcrypto.InclusionCommitment{}, AggregateCommitment: []byte{}, Proof: []byte{}, }, pubKeys, ) err = d.Start() assert.NoError(t, err) frame, err = d.Head() assert.NoError(t, err) dataFrames := []*protobufs.ClockFrame{} datawg := sync.WaitGroup{} datawg.Add(1) dataFrameCh := d.NewFrameCh() go func() { for i := 0; i < 40; i++ { dataFrames = append(dataFrames, <-dataFrameCh) } datawg.Done() }() // 1. z-dist optimal – proof submission is strictly master-frame evoked leader for i := int64(0); i < 10; i++ { masterSelector, err := frames[i+1].GetSelector() assert.NoError(t, err) proverSelection := proverTrie.FindNearest( masterSelector.FillBytes(make([]byte, 32)), ) optimalSigner, _ := keyManager.GetSigningKey( addrMap[string(proverSelection.External.Key)], ) frame, err = prover.ProveDataClockFrame( frame, [][]byte{}, []*protobufs.InclusionAggregateProof{}, optimalSigner, i+1, 10, ) d.Insert(frame) } // 2. z-dist optimal, out of order – proof submission is strictly master-frame // evoked leader, but arrived completely backwards insertFrames := []*protobufs.ClockFrame{} for i := int64(10); i < 20; i++ { masterSelector, err := frames[i+1].GetSelector() assert.NoError(t, err) proverSelection := proverTrie.FindNearest( masterSelector.FillBytes(make([]byte, 32)), ) optimalSigner, _ := keyManager.GetSigningKey( addrMap[string(proverSelection.External.Key)], ) frame, err = prover.ProveDataClockFrame( frame, [][]byte{}, []*protobufs.InclusionAggregateProof{}, optimalSigner, i+1, 10, ) insertFrames = append(insertFrames, frame) } for i := 9; i >= 0; i-- { err := d.Insert(insertFrames[i]) assert.NoError(t, err) } // 3. 90% optimal, out of order insertFrames = []*protobufs.ClockFrame{} for i := int64(20); i < 25; i++ { masterSelector, err := frames[i+1].GetSelector() assert.NoError(t, err) proverSelection := proverTrie.FindNearest( masterSelector.FillBytes(make([]byte, 32)), ) optimalSigner, _ := keyManager.GetSigningKey( addrMap[string(proverSelection.External.Key)], ) frame, err = prover.ProveDataClockFrame( frame, [][]byte{}, []*protobufs.InclusionAggregateProof{}, optimalSigner, i+1, 10, ) d.Insert(frame) } masterSelector, err := frames[26].GetSelector() assert.NoError(t, err) proverSelections := proverTrie.FindNearestAndApproximateNeighbors( masterSelector.FillBytes(make([]byte, 32)), ) suboptimalSigner2, _ := keyManager.GetSigningKey( addrMap[string(proverSelections[2].External.Key)], ) // What we're trying to simulate: consensus heads progressed on a slightly // less optimal prover. frame, err = prover.ProveDataClockFrame( frame, [][]byte{}, []*protobufs.InclusionAggregateProof{}, suboptimalSigner2, 26, 10, ) insertFrames = append(insertFrames, frame) for i := int64(26); i < 30; i++ { masterSelector, err := frames[i+1].GetSelector() assert.NoError(t, err) proverSelection := proverTrie.FindNearest( masterSelector.FillBytes(make([]byte, 32)), ) optimalSigner, _ := keyManager.GetSigningKey( addrMap[string(proverSelection.External.Key)], ) frame, err = prover.ProveDataClockFrame( frame, [][]byte{}, []*protobufs.InclusionAggregateProof{}, optimalSigner, i+1, 10, ) insertFrames = append(insertFrames, frame) } for i := 4; i >= 0; i-- { err := d.Insert(insertFrames[i]) assert.NoError(t, err) } // 4. Malicious majority, out of order – handle a suppressive majority and // force consensus on the lowest distance sub-tree: insertFrames = []*protobufs.ClockFrame{} conflictFrames := []*protobufs.ClockFrame{} optimalKeySet := [][]byte{} suppressedFrame := frame for i := int64(30); i < 40; i++ { masterSelector, err := frames[i+1].GetSelector() assert.NoError(t, err) proverSelections := proverTrie.FindNearestAndApproximateNeighbors( masterSelector.FillBytes(make([]byte, 32)), ) optimalSigner, _ := keyManager.GetSigningKey( addrMap[string(proverSelections[0].External.Key)], ) suboptimalSigner2, _ := keyManager.GetSigningKey( addrMap[string(proverSelections[2].External.Key)], ) optimalKeySet = append(optimalKeySet, []byte( (optimalSigner.Public()).(ed448.PublicKey), )) // What we're trying to simulate: the majority is intentionally ignoring // the most optimal signer suppressedFrame, err := prover.ProveDataClockFrame( suppressedFrame, [][]byte{}, []*protobufs.InclusionAggregateProof{}, optimalSigner, i+1, 10, ) insertFrames = append(insertFrames, suppressedFrame) frame, err = prover.ProveDataClockFrame( frame, [][]byte{}, []*protobufs.InclusionAggregateProof{}, suboptimalSigner2, i+1, 10, ) conflictFrames = append(conflictFrames, frame) } for i := 9; i >= 0; i-- { err := d.Insert(conflictFrames[i]) assert.NoError(t, err) } // Someone is honest, but clumsy: for i := 9; i >= 0; i-- { err := d.Insert(insertFrames[i]) assert.NoError(t, err) } datawg.Wait() for i := 0; i < 40; i++ { assert.NotNil(t, dataFrames[i]) if i >= 40 { assert.Equal(t, uint64(i), dataFrames[i].FrameNumber) assert.Equal( t, dataFrames[i].GetPublicKeySignatureEd448().PublicKey.KeyValue, optimalKeySet[i-30], ) } else { assert.Equal(t, uint64(i+1), dataFrames[i].FrameNumber) } } }