Peer configuration management (#69)

* feature: add config properties to the SyncResponse of the management gRpc service * fix: lint errors * chore: modify management protocol according to the review notes * fix: management proto fields sequence * feature: add proper peer configuration to be synced * chore: minor changes * feature: finalize peer config management * fix: lint errors * feature: add management server config file * refactor: extract hosts-config to a separate file * refactor: review notes applied to correct file_store usage * refactor: extract management service configuration to a file * refactor: simplify management config
2026-10-30 08:19:09 +01:00 · 2021-07-30 17:46:38 +02:00
parent c0c4c4a266
commit 2c2c1e19df
14 changed files with 690 additions and 237 deletions
@@ -0,0 +1,119 @@
+package server
+
+import (
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"net"
+	"sync"
+)
+
+type AccountManager struct {
+	Store Store
+	// mutex to synchronise account operations (e.g. generating Peer IP address inside the Network)
+	mux sync.Mutex
+}
+
+// Account represents a unique account of the system
+type Account struct {
+	Id        string
+	SetupKeys map[string]*SetupKey
+	Network   *Network
+	Peers     []*Peer
+}
+
+// SetupKey represents a pre-authorized key used to register machines (peers)
+// One key might have multiple machines
+type SetupKey struct {
+	Key string
+}
+
+// Peer represents a machine connected to the network.
+// The Peer is a Wireguard peer identified by a public key
+type Peer struct {
+	// Wireguard public key
+	Key string
+	// A setup key this peer was registered with
+	SetupKey *SetupKey
+	// IP address of the Peer
+	IP net.IP
+}
+
+// NewManager creates a new AccountManager with a provided Store
+func NewManager(store Store) *AccountManager {
+	return &AccountManager{
+		Store: store,
+		mux:   sync.Mutex{},
+	}
+}
+
+// GetPeer returns a peer from a Store
+func (manager *AccountManager) GetPeer(peerKey string) (*Peer, error) {
+	manager.mux.Lock()
+	defer manager.mux.Unlock()
+
+	peer, err := manager.Store.GetPeer(peerKey)
+	if err != nil {
+		return nil, status.Errorf(codes.NotFound, "provided peer key doesn't exists %s", peerKey)
+	}
+
+	return peer, nil
+}
+
+// GetPeersForAPeer returns a list of peers available for a given peer (key)
+// Effectively all the peers of the original peer's account except for the peer itself
+func (manager *AccountManager) GetPeersForAPeer(peerKey string) ([]*Peer, error) {
+	manager.mux.Lock()
+	defer manager.mux.Unlock()
+
+	account, err := manager.Store.GetPeerAccount(peerKey)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "Invalid peer key %s", peerKey)
+	}
+
+	var res []*Peer
+	for _, peer := range account.Peers {
+		if peer.Key != peerKey {
+			res = append(res, peer)
+		}
+	}
+
+	return res, nil
+}
+
+// AddPeer adds a new peer to the Store.
+// Each Account has a list of pre-authorised SetupKey and if no Account has a given key err wit ha code codes.Unauthenticated
+// will be returned, meaning the key is invalid
+// Each new Peer will be assigned a new next net.IP from the Account.Network and Account.Network.LastIP will be updated (IP's are not reused).
+func (manager *AccountManager) AddPeer(setupKey string, peerKey string) (*Peer, error) {
+	manager.mux.Lock()
+	defer manager.mux.Unlock()
+
+	account, err := manager.Store.GetAccountBySetupKey(setupKey)
+	if err != nil {
+		//todo
+		return nil, err
+	}
+
+	var takenIps []net.IP
+	for _, peer := range account.Peers {
+		takenIps = append(takenIps, peer.IP)
+	}
+
+	network := account.Network
+	nextIp, _ := AllocatePeerIP(network.Net, takenIps)
+
+	newPeer := &Peer{
+		Key:      peerKey,
+		SetupKey: &SetupKey{Key: setupKey},
+		IP:       nextIp,
+	}
+
+	account.Peers = append(account.Peers, newPeer)
+	err = manager.Store.SaveAccount(account)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed adding peer")
+	}
+
+	return newPeer, nil
+
+}