-
Notifications
You must be signed in to change notification settings - Fork 142
/
trustedNetworkManager.go
57 lines (46 loc) · 1.71 KB
/
trustedNetworkManager.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
// Copyright 2020 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0
package testutil
import (
"golang.org/x/xerrors"
"github.com/iotaledger/wasp/packages/cryptolib"
"github.com/iotaledger/wasp/packages/peering"
)
type trustedNetworkManager struct {
data map[cryptolib.PublicKeyKey]trustedNetworkDataEntry
}
type trustedNetworkDataEntry struct {
key *cryptolib.PublicKey
netID string
}
var _ peering.TrustedNetworkManager = &trustedNetworkManager{}
func NewTrustedNetworkManager() peering.TrustedNetworkManager {
return &trustedNetworkManager{
data: map[cryptolib.PublicKeyKey]trustedNetworkDataEntry{},
}
}
// IsTrustedPeer implements the peering.TrustedNetworkManager interface.
func (tnm *trustedNetworkManager) IsTrustedPeer(pubKey *cryptolib.PublicKey) error {
if _, ok := tnm.data[pubKey.AsKey()]; ok {
return nil
}
return xerrors.New("not trusted")
}
// TrustPeer implements the peering.TrustedNetworkManager interface.
func (tnm *trustedNetworkManager) TrustPeer(pubKey *cryptolib.PublicKey, netID string) (*peering.TrustedPeer, error) {
tnm.data[pubKey.AsKey()] = trustedNetworkDataEntry{pubKey, netID}
return &peering.TrustedPeer{PubKey: pubKey, NetID: netID}, nil
}
// DistrustPeer implements the peering.TrustedNetworkManager interface.
func (tnm *trustedNetworkManager) DistrustPeer(pubKey *cryptolib.PublicKey) (*peering.TrustedPeer, error) {
delete(tnm.data, pubKey.AsKey())
return nil, nil
}
// TrustedPeers implements the peering.TrustedNetworkManager interface.
func (tnm *trustedNetworkManager) TrustedPeers() ([]*peering.TrustedPeer, error) {
res := []*peering.TrustedPeer{}
for _, v := range tnm.data {
res = append(res, &peering.TrustedPeer{PubKey: v.key, NetID: v.netID})
}
return res, nil
}