-
Notifications
You must be signed in to change notification settings - Fork 176
/
util.go
71 lines (63 loc) · 2.23 KB
/
util.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
package protocol
import (
"fmt"
"github.com/onflow/flow-go/model/flow"
"github.com/onflow/flow-go/model/flow/filter"
)
// IsNodeAuthorizedAt returns whether the node with the given ID is a valid
// un-ejected network participant as of the given state snapshot.
func IsNodeAuthorizedAt(snapshot Snapshot, id flow.Identifier) (bool, error) {
return CheckNodeStatusAt(
snapshot,
id,
filter.HasWeight(true),
filter.Not(filter.Ejected),
)
}
// IsNodeAuthorizedWithRoleAt returns whether the node with the given ID is a valid
// un-ejected network participant with the specified role as of the given state snapshot.
// Expected errors during normal operations:
// * storage.ErrNotFound if snapshot references an unknown block
// All other errors are unexpected and potential symptoms of internal state corruption.
func IsNodeAuthorizedWithRoleAt(snapshot Snapshot, id flow.Identifier, role flow.Role) (bool, error) {
return CheckNodeStatusAt(
snapshot,
id,
filter.HasWeight(true),
filter.Not(filter.Ejected),
filter.HasRole(role),
)
}
// CheckNodeStatusAt returns whether the node with the given ID is a valid identity at the given
// state snapshot, and satisfies all checks.
// Expected errors during normal operations:
// * storage.ErrNotFound if snapshot references an unknown block
// All other errors are unexpected and potential symptoms of internal state corruption.
func CheckNodeStatusAt(snapshot Snapshot, id flow.Identifier, checks ...flow.IdentityFilter) (bool, error) {
identity, err := snapshot.Identity(id)
if IsIdentityNotFound(err) {
return false, nil
}
if err != nil {
return false, fmt.Errorf("could not retrieve node identity (id=%x): %w)", id, err)
}
for _, check := range checks {
if !check(identity) {
return false, nil
}
}
return true, nil
}
// IsSporkRootSnapshot returns whether the given snapshot is the state snapshot
// representing the initial state for a spork.
func IsSporkRootSnapshot(snapshot Snapshot) (bool, error) {
segment, err := snapshot.SealingSegment()
if err != nil {
return false, fmt.Errorf("could not get snapshot head: %w", err)
}
if len(segment.Blocks) > 1 {
// spork root snapshots uniquely have only one block in the sealing segment
return false, nil
}
return true, nil
}