-
Notifications
You must be signed in to change notification settings - Fork 5
/
utils.go
86 lines (77 loc) · 2.63 KB
/
utils.go
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
package interfaces
import (
"github.com/pkg/errors"
zondpb "github.com/theQRL/qrysm/v4/proto/prysm/v1alpha1"
)
// SignedBeaconBlockHeaderFromBlock function to retrieve signed block header from block.
func SignedBeaconBlockHeaderFromBlock(block *zondpb.SignedBeaconBlock) (*zondpb.SignedBeaconBlockHeader, error) {
if block.Block == nil || block.Block.Body == nil {
return nil, errors.New("nil block")
}
bodyRoot, err := block.Block.Body.HashTreeRoot()
if err != nil {
return nil, errors.Wrap(err, "failed to get body root of block")
}
return &zondpb.SignedBeaconBlockHeader{
Header: &zondpb.BeaconBlockHeader{
Slot: block.Block.Slot,
ProposerIndex: block.Block.ProposerIndex,
ParentRoot: block.Block.ParentRoot,
StateRoot: block.Block.StateRoot,
BodyRoot: bodyRoot[:],
},
Signature: block.Signature,
}, nil
}
// SignedBeaconBlockHeaderFromBlockInterface function to retrieve signed block header from block.
func SignedBeaconBlockHeaderFromBlockInterface(sb ReadOnlySignedBeaconBlock) (*zondpb.SignedBeaconBlockHeader, error) {
b := sb.Block()
if b.IsNil() || b.Body().IsNil() {
return nil, errors.New("nil block")
}
h, err := BeaconBlockHeaderFromBlockInterface(b)
if err != nil {
return nil, errors.Wrap(err, "failed to get block header of block")
}
sig := sb.Signature()
return &zondpb.SignedBeaconBlockHeader{
Header: h,
Signature: sig[:],
}, nil
}
// BeaconBlockHeaderFromBlock function to retrieve block header from block.
func BeaconBlockHeaderFromBlock(block *zondpb.BeaconBlock) (*zondpb.BeaconBlockHeader, error) {
if block.Body == nil {
return nil, errors.New("nil block body")
}
bodyRoot, err := block.Body.HashTreeRoot()
if err != nil {
return nil, errors.Wrap(err, "failed to get body root of block")
}
return &zondpb.BeaconBlockHeader{
Slot: block.Slot,
ProposerIndex: block.ProposerIndex,
ParentRoot: block.ParentRoot,
StateRoot: block.StateRoot,
BodyRoot: bodyRoot[:],
}, nil
}
// BeaconBlockHeaderFromBlockInterface function to retrieve block header from block.
func BeaconBlockHeaderFromBlockInterface(block ReadOnlyBeaconBlock) (*zondpb.BeaconBlockHeader, error) {
if block.Body().IsNil() {
return nil, errors.New("nil block body")
}
bodyRoot, err := block.Body().HashTreeRoot()
if err != nil {
return nil, errors.Wrap(err, "failed to get body root of block")
}
parentRoot := block.ParentRoot()
stateRoot := block.StateRoot()
return &zondpb.BeaconBlockHeader{
Slot: block.Slot(),
ProposerIndex: block.ProposerIndex(),
ParentRoot: parentRoot[:],
StateRoot: stateRoot[:],
BodyRoot: bodyRoot[:],
}, nil
}