mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-08 02:31:19 +00:00
9529c73ff1
* Initial spec rewrite * Finish adding merkle tree implementation * Last bits * Move reverse function * Add comments * Add deposit tree snapshot * Add deposit tree * Add comments + cleanup * Fixes * Add missing errors * Small fixes * Add unhandled error * Cleanup * Fix unsafe file.Close * Add missing comments * Small fixes * Address some of deepSource' compaints * Add depositCount check * Add finalizedDeposit check * Replace pointer magic with copy() * Add test for slice reversal * add back bytes method * Add package level description * Remove zerohash gen and add additional checks * Add additional comments * Small lint fixes * Forgot an error * Small fixes * Move Uint64ToBytesLittleEndian32 + test * Fix uint subtraction issue * Move mixInLength below error handling * Fix * Fix deposit root --------- Co-authored-by: rauljordan <raul@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
63 lines
1.6 KiB
Go
63 lines
1.6 KiB
Go
package depositsnapshot
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v3/encoding/bytesutil"
|
|
)
|
|
|
|
var (
|
|
// ErrZeroIndex occurs when the value of index is 0.
|
|
ErrZeroIndex = errors.New("index should be greater than 0")
|
|
)
|
|
|
|
// DepositTreeSnapshot represents the data used to create a
|
|
// deposit tree given a snapshot.
|
|
//
|
|
//nolint:unused
|
|
type DepositTreeSnapshot struct {
|
|
finalized [][32]byte
|
|
depositRoot [32]byte
|
|
depositCount uint64
|
|
executionBlock executionBlock
|
|
}
|
|
|
|
// CalculateRoot returns the root of a deposit tree snapshot.
|
|
func (ds *DepositTreeSnapshot) CalculateRoot() ([32]byte, error) {
|
|
size := ds.depositCount
|
|
index := len(ds.finalized)
|
|
root := Zerohashes[0]
|
|
for i := 0; i < DepositContractDepth; i++ {
|
|
if (size & 1) == 1 {
|
|
if index == 0 {
|
|
return [32]byte{}, ErrZeroIndex
|
|
}
|
|
index--
|
|
root = sha256.Sum256(append(ds.finalized[index][:], root[:]...))
|
|
} else {
|
|
root = sha256.Sum256(append(root[:], Zerohashes[i][:]...))
|
|
}
|
|
size >>= 1
|
|
}
|
|
return sha256.Sum256(append(root[:], bytesutil.Uint64ToBytesLittleEndian(ds.depositCount)...)), nil
|
|
}
|
|
|
|
// fromTreeParts constructs the deposit tree from pre-existing data.
|
|
//
|
|
//nolint:unused
|
|
func fromTreeParts(finalised [][32]byte, depositCount uint64, executionBlock executionBlock) (DepositTreeSnapshot, error) {
|
|
snapshot := DepositTreeSnapshot{
|
|
finalized: finalised,
|
|
depositRoot: Zerohashes[0],
|
|
depositCount: depositCount,
|
|
executionBlock: executionBlock,
|
|
}
|
|
root, err := snapshot.CalculateRoot()
|
|
if err != nil {
|
|
return snapshot, ErrInvalidSnapshotRoot
|
|
}
|
|
snapshot.depositRoot = root
|
|
return snapshot, nil
|
|
}
|