forked from TrueCloudLab/neoneo-go
7589733017
And include some node-specific configurations there with backwards compatibility. Note that in the future we'll remove Ledger's fields from the ProtocolConfiguration and it'll be possible to access them in Blockchain directly (not via .Ledger). The other option tried was using two configuration types separately, but that incurs more changes to the codebase, single structure that behaves almost like the old one is better for backwards compatibility. Fixes #2676.
85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package chaindump
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/config"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/block"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
)
|
|
|
|
// DumperRestorer is an interface to get/add blocks from/to.
|
|
type DumperRestorer interface {
|
|
AddBlock(block *block.Block) error
|
|
GetBlock(hash util.Uint256) (*block.Block, error)
|
|
GetConfig() config.Blockchain
|
|
GetHeaderHash(uint32) util.Uint256
|
|
}
|
|
|
|
// Dump writes count blocks from start to the provided writer.
|
|
// Note: header needs to be written separately by a client.
|
|
func Dump(bc DumperRestorer, w *io.BinWriter, start, count uint32) error {
|
|
for i := start; i < start+count; i++ {
|
|
bh := bc.GetHeaderHash(i)
|
|
b, err := bc.GetBlock(bh)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
buf := io.NewBufBinWriter()
|
|
b.EncodeBinary(buf.BinWriter)
|
|
bytes := buf.Bytes()
|
|
w.WriteU32LE(uint32(len(bytes)))
|
|
w.WriteBytes(bytes)
|
|
if w.Err != nil {
|
|
return w.Err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Restore restores blocks from the provided reader.
|
|
// f is called after addition of every block.
|
|
func Restore(bc DumperRestorer, r *io.BinReader, skip, count uint32, f func(b *block.Block) error) error {
|
|
readBlock := func(r *io.BinReader) ([]byte, error) {
|
|
var size = r.ReadU32LE()
|
|
buf := make([]byte, size)
|
|
r.ReadBytes(buf)
|
|
return buf, r.Err
|
|
}
|
|
|
|
i := uint32(0)
|
|
for ; i < skip; i++ {
|
|
_, err := readBlock(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
stateRootInHeader := bc.GetConfig().StateRootInHeader
|
|
|
|
for ; i < skip+count; i++ {
|
|
buf, err := readBlock(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b := block.New(stateRootInHeader)
|
|
r := io.NewBinReaderFromBuf(buf)
|
|
b.DecodeBinary(r)
|
|
if r.Err != nil {
|
|
return r.Err
|
|
}
|
|
if b.Index != 0 || i != 0 || skip != 0 {
|
|
err = bc.AddBlock(b)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to add block %d: %w", i, err)
|
|
}
|
|
}
|
|
if f != nil {
|
|
if err := f(b); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|