forked from extern/smegmesh
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package ctrlserver
|
|
|
|
import (
|
|
"github.com/tim-beatham/wgmesh/pkg/conf"
|
|
"github.com/tim-beatham/wgmesh/pkg/conn"
|
|
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
|
"github.com/tim-beatham/wgmesh/pkg/rpc"
|
|
"golang.zx2c4.com/wireguard/wgctrl"
|
|
)
|
|
|
|
// NewCtrlServerParams are the params requried to create a new ctrl server
|
|
type NewCtrlServerParams struct {
|
|
Conf *conf.WgMeshConfiguration
|
|
Client *wgctrl.Client
|
|
AuthProvider rpc.AuthenticationServer
|
|
CtrlProvider rpc.MeshCtrlServerServer
|
|
SyncProvider rpc.SyncServiceServer
|
|
}
|
|
|
|
// Create a new instance of the MeshCtrlServer or error if the
|
|
// operation failed
|
|
func NewCtrlServer(params *NewCtrlServerParams) (*MeshCtrlServer, error) {
|
|
ctrlServer := new(MeshCtrlServer)
|
|
ctrlServer.MeshManager = mesh.NewMeshManager(*params.Conf, params.Client)
|
|
ctrlServer.Conf = params.Conf
|
|
|
|
connManagerParams := conn.NewConnectionManageParams{
|
|
CertificatePath: params.Conf.CertificatePath,
|
|
PrivateKey: params.Conf.PrivateKeyPath,
|
|
SkipCertVerification: params.Conf.SkipCertVerification,
|
|
}
|
|
|
|
connMgr, err := conn.NewConnectionManager(&connManagerParams)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ctrlServer.ConnectionManager = connMgr
|
|
connServerParams := conn.NewConnectionServerParams{
|
|
Conf: params.Conf,
|
|
AuthProvider: params.AuthProvider,
|
|
CtrlProvider: params.CtrlProvider,
|
|
SyncProvider: params.SyncProvider,
|
|
}
|
|
|
|
connServer, err := conn.NewConnectionServer(&connServerParams)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ctrlServer.ConnectionServer = connServer
|
|
return ctrlServer, nil
|
|
}
|
|
|
|
// Close closes the ctrl server tearing down any connections that exist
|
|
func (s *MeshCtrlServer) Close() error {
|
|
if err := s.ConnectionManager.Close(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := s.ConnectionServer.Close(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|