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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
// Copyright 2018 The dexon-consensus Authors
// This file is part of the dexon-consensus library.
//
// The dexon-consensus library is free software: you can redistribute it
// and/or modify it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation, either version 3 of the License,
// or (at your option) any later version.
//
// The dexon-consensus library is distributed in the hope that it will be
// useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
// General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the dexon-consensus library. If not, see
// <http://www.gnu.org/licenses/>.
package utils
import (
"context"
"fmt"
"github.com/dexon-foundation/dexon-consensus/common"
"github.com/dexon-foundation/dexon-consensus/core/types"
typesDKG "github.com/dexon-foundation/dexon-consensus/core/types/dkg"
)
type configAccessor interface {
Configuration(round uint64) *types.Config
}
// GetConfigWithPanic is a helper to access configs, and panic when config for
// that round is not ready yet.
func GetConfigWithPanic(accessor configAccessor, round uint64,
logger common.Logger) *types.Config {
if logger != nil {
logger.Debug("Calling Governance.Configuration", "round", round)
}
c := accessor.Configuration(round)
if c == nil {
panic(fmt.Errorf("configuration is not ready %v", round))
}
return c
}
type crsAccessor interface {
CRS(round uint64) common.Hash
}
// GetCRSWithPanic is a helper to access CRS, and panic when CRS for that
// round is not ready yet.
func GetCRSWithPanic(accessor crsAccessor, round uint64,
logger common.Logger) common.Hash {
if logger != nil {
logger.Debug("Calling Governance.CRS", "round", round)
}
crs := accessor.CRS(round)
if (crs == common.Hash{}) {
panic(fmt.Errorf("CRS is not ready %v", round))
}
return crs
}
// VerifyDKGComplaint verifies if its a valid DKGCompliant.
func VerifyDKGComplaint(
complaint *typesDKG.Complaint, mpk *typesDKG.MasterPublicKey) (bool, error) {
ok, err := VerifyDKGComplaintSignature(complaint)
if err != nil {
return false, err
}
if !ok {
return false, nil
}
if complaint.IsNack() {
return true, nil
}
if complaint.Round != mpk.Round {
return false, nil
}
ok, err = VerifyDKGMasterPublicKeySignature(mpk)
if err != nil {
return false, err
}
if !ok {
return false, nil
}
ok, err = mpk.PublicKeyShares.VerifyPrvShare(
typesDKG.NewID(complaint.PrivateShare.ReceiverID),
&complaint.PrivateShare.PrivateShare)
if err != nil {
return false, err
}
return ok, nil
}
// LaunchDummyReceiver launches a go routine to receive from the receive
// channel of a network module. An context is required to stop the go routine
// automatically. An optinal message handler could be provided.
func LaunchDummyReceiver(
ctx context.Context, recv <-chan interface{}, handler func(interface{})) (
context.CancelFunc, <-chan struct{}) {
var (
dummyCtx, dummyCancel = context.WithCancel(ctx)
finishedChan = make(chan struct{}, 1)
)
go func() {
defer func() {
finishedChan <- struct{}{}
}()
loop:
for {
select {
case <-dummyCtx.Done():
break loop
case v, ok := <-recv:
if !ok {
panic(fmt.Errorf(
"receive channel is closed before dummy receiver"))
}
if handler != nil {
handler(v)
}
}
}
}()
return dummyCancel, finishedChan
}
|