mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
package core
|
|
|
|
import (
|
|
"github.com/edgelesssys/constellation/coordinator/peer"
|
|
"github.com/edgelesssys/constellation/coordinator/storewrapper"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// GetPeers returns the stored peers if the requested version differs from the stored version.
|
|
// peers include all vpn devices namely Coordinators, Nodes and Admins.
|
|
func (c *Core) GetPeers(resourceVersion int) (int, []peer.Peer, error) {
|
|
// Most often there's nothing to do, so first check without an expensive transaction.
|
|
curVer, err := c.data().GetPeersResourceVersion()
|
|
if err != nil {
|
|
return 0, nil, err
|
|
}
|
|
if curVer == resourceVersion {
|
|
return curVer, nil, nil
|
|
}
|
|
|
|
tx, err := c.store.BeginTransaction()
|
|
if err != nil {
|
|
return 0, nil, err
|
|
}
|
|
defer tx.Rollback()
|
|
txdata := storewrapper.StoreWrapper{Store: tx}
|
|
|
|
txVer, err := txdata.GetPeersResourceVersion()
|
|
if err != nil {
|
|
return 0, nil, err
|
|
}
|
|
peers, err := txdata.GetPeers()
|
|
if err != nil {
|
|
return 0, nil, err
|
|
}
|
|
return txVer, peers, nil
|
|
}
|
|
|
|
// AddPeer adds a peer to the store and the VPN.
|
|
func (c *Core) AddPeer(peer peer.Peer) error {
|
|
if err := c.AddPeerToVPN(peer); err != nil {
|
|
return err
|
|
}
|
|
return c.AddPeerToStore(peer)
|
|
}
|
|
|
|
// AddPeerToVPN adds a peer to the the VPN.
|
|
func (c *Core) AddPeerToVPN(peer peer.Peer) error {
|
|
// don't add myself to vpn
|
|
myIP, err := c.vpn.GetInterfaceIP()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if myIP != peer.VPNIP {
|
|
if err := c.vpn.AddPeer(peer.VPNPubKey, peer.PublicIP, peer.VPNIP); err != nil {
|
|
c.zaplogger.Error("failed to add peer to VPN", zap.Error(err), zap.String("peer public_ip", peer.PublicIP), zap.String("peer vpn_ip", peer.VPNIP))
|
|
return err
|
|
}
|
|
c.zaplogger.Info("added peer to VPN", zap.String("role", peer.Role.String()), zap.String("coordinator public_ip", peer.PublicIP), zap.String("coordinator vpn_ip", peer.VPNIP))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// AddPeerToStore adds a peer to the store.
|
|
func (c *Core) AddPeerToStore(peer peer.Peer) error {
|
|
tx, err := c.store.BeginTransaction()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer tx.Rollback()
|
|
txdata := storewrapper.StoreWrapper{Store: tx}
|
|
|
|
if err := txdata.IncrementPeersResourceVersion(); err != nil {
|
|
return err
|
|
}
|
|
if err := txdata.PutPeer(peer); err != nil {
|
|
return err
|
|
}
|
|
if err := tx.Commit(); err != nil {
|
|
return err
|
|
}
|
|
c.zaplogger.Info("added peer to store", zap.String("peer public_ip", peer.PublicIP), zap.String("peer vpn_ip", peer.VPNIP))
|
|
return nil
|
|
}
|
|
|
|
// UpdatePeers synchronizes the peers known to the store and the vpn with the passed peers.
|
|
func (c *Core) UpdatePeers(peers []peer.Peer) error {
|
|
return c.vpn.UpdatePeers(peers)
|
|
}
|