netmaker/main.go

188 lines
5.1 KiB
Go
Raw Normal View History

2021-03-26 00:17:52 +08:00
package main
import (
"context"
2021-08-10 02:56:27 +08:00
"fmt"
2021-07-21 05:18:45 +08:00
"net"
"os"
"os/signal"
2021-11-17 10:19:13 +08:00
"runtime/debug"
2021-07-21 05:18:45 +08:00
"strconv"
"sync"
"time"
2021-10-22 03:28:58 +08:00
"github.com/gravitl/netmaker/auth"
2021-07-21 05:18:45 +08:00
controller "github.com/gravitl/netmaker/controllers"
"github.com/gravitl/netmaker/database"
"github.com/gravitl/netmaker/functions"
nodepb "github.com/gravitl/netmaker/grpc"
2021-12-07 04:31:08 +08:00
"github.com/gravitl/netmaker/logger"
2021-10-22 03:28:58 +08:00
"github.com/gravitl/netmaker/logic"
2021-08-10 02:56:27 +08:00
"github.com/gravitl/netmaker/models"
2021-09-20 02:03:47 +08:00
"github.com/gravitl/netmaker/netclient/ncutils"
2021-07-21 05:18:45 +08:00
"github.com/gravitl/netmaker/servercfg"
"github.com/gravitl/netmaker/serverctl"
2021-07-21 05:18:45 +08:00
"google.golang.org/grpc"
2021-03-26 00:17:52 +08:00
)
2021-08-31 03:58:23 +08:00
// Start DB Connection and start API Request Handler
2021-03-26 00:17:52 +08:00
func main() {
2021-08-10 02:56:27 +08:00
fmt.Println(models.RetrieveLogo()) // print the logo
initialize() // initial db and grpc server
2021-11-17 10:19:13 +08:00
setGarbageCollection()
defer database.CloseDB()
2021-07-21 05:18:45 +08:00
startControllers() // start the grpc or rest endpoints
}
2021-07-29 04:40:13 +08:00
func initialize() { // Client Mode Prereq Check
2021-07-21 05:18:45 +08:00
var err error
2021-10-22 03:28:58 +08:00
2021-07-29 04:40:13 +08:00
if err = database.InitializeDatabase(); err != nil {
2021-12-11 10:09:42 +08:00
logger.FatalLog("Error connecting to database")
2021-07-29 04:40:13 +08:00
}
2021-12-07 04:31:08 +08:00
logger.Log(0, "database successfully connected")
2021-10-22 03:28:58 +08:00
var authProvider = auth.InitializeAuthProvider()
if authProvider != "" {
2021-12-29 01:07:50 +08:00
logger.Log(0, "OAuth provider,", authProvider+",", "initialized")
2021-10-22 21:47:29 +08:00
} else {
2021-12-07 04:31:08 +08:00
logger.Log(0, "no OAuth provider found or not configured, continuing without OAuth")
2021-10-22 03:28:58 +08:00
}
if servercfg.IsClientMode() != "off" {
2021-09-20 02:03:47 +08:00
output, err := ncutils.RunCmd("id -u", true)
2021-08-31 03:58:23 +08:00
if err != nil {
2021-12-07 04:31:08 +08:00
logger.FatalLog("Error running 'id -u' for prereq check. Please investigate or disable client mode.", output, err.Error())
2021-08-31 03:58:23 +08:00
}
uid, err := strconv.Atoi(string(output[:len(output)-1]))
if err != nil {
2021-12-07 04:31:08 +08:00
logger.FatalLog("Error retrieving uid from 'id -u' for prereq check. Please investigate or disable client mode.", err.Error())
2021-08-31 03:58:23 +08:00
}
if uid != 0 {
2021-12-07 04:31:08 +08:00
logger.FatalLog("To run in client mode requires root privileges. Either disable client mode or run with sudo.")
2021-08-31 03:58:23 +08:00
}
2021-09-29 02:08:16 +08:00
if err := serverctl.InitServerNetclient(); err != nil {
2021-12-07 04:31:08 +08:00
logger.FatalLog("Did not find netclient to use CLIENT_MODE")
2021-09-29 02:08:16 +08:00
}
2021-07-21 05:18:45 +08:00
}
2021-04-20 09:50:58 +08:00
2021-07-21 05:18:45 +08:00
if servercfg.IsDNSMode() {
err := functions.SetDNSDir()
2021-05-06 04:42:17 +08:00
if err != nil {
2021-12-07 04:31:08 +08:00
logger.FatalLog(err.Error())
2021-05-06 04:42:17 +08:00
}
}
2021-07-21 05:18:45 +08:00
}
2021-05-06 04:42:17 +08:00
2021-07-21 05:18:45 +08:00
func startControllers() {
var waitnetwork sync.WaitGroup
2021-05-06 04:42:17 +08:00
//Run Agent Server
if servercfg.IsAgentBackend() {
2021-07-21 05:18:45 +08:00
if !(servercfg.DisableRemoteIPCheck()) && servercfg.GetGRPCHost() == "127.0.0.1" {
2021-05-06 08:41:28 +08:00
err := servercfg.SetHost()
if err != nil {
2021-12-07 04:31:08 +08:00
logger.FatalLog("Unable to Set host. Exiting...", err.Error())
2021-05-06 08:41:28 +08:00
}
}
waitnetwork.Add(1)
2021-07-21 05:18:45 +08:00
go runGRPC(&waitnetwork)
2021-03-26 00:17:52 +08:00
}
2021-07-21 05:18:45 +08:00
if servercfg.IsDNSMode() {
2021-10-27 00:27:29 +08:00
err := logic.SetDNS()
2021-07-21 05:18:45 +08:00
if err != nil {
2021-12-07 04:31:08 +08:00
logger.Log(0, "error occurred initializing DNS: ", err.Error())
2021-07-21 05:18:45 +08:00
}
}
2021-05-06 04:42:17 +08:00
//Run Rest Server
if servercfg.IsRestBackend() {
2021-07-21 05:18:45 +08:00
if !servercfg.DisableRemoteIPCheck() && servercfg.GetAPIHost() == "127.0.0.1" {
err := servercfg.SetHost()
if err != nil {
2021-12-07 04:31:08 +08:00
logger.FatalLog("Unable to Set host. Exiting...", err.Error())
2021-07-21 05:18:45 +08:00
}
}
waitnetwork.Add(1)
2021-12-09 05:52:32 +08:00
go controller.HandleRESTRequests(&waitnetwork)
2021-03-26 00:17:52 +08:00
}
2021-12-09 05:52:32 +08:00
if !servercfg.IsAgentBackend() && !servercfg.IsRestBackend() {
2021-12-07 04:31:08 +08:00
logger.Log(0, "No Server Mode selected, so nothing is being served! Set either Agent mode (AGENT_BACKEND) or Rest mode (REST_BACKEND) to 'true'.")
2021-03-26 00:17:52 +08:00
}
2021-12-09 05:52:32 +08:00
if servercfg.IsClientMode() == "on" {
var checkintime = time.Duration(servercfg.GetServerCheckinInterval()) * time.Second
for { // best effort currently
2021-12-11 04:01:10 +08:00
var serverGroup sync.WaitGroup
serverGroup.Add(1)
go runClient(&serverGroup)
serverGroup.Wait()
2021-12-09 05:52:32 +08:00
time.Sleep(checkintime)
}
}
waitnetwork.Wait()
}
func runClient(wg *sync.WaitGroup) {
defer wg.Done()
2021-12-09 05:52:32 +08:00
go serverctl.HandleContainedClient()
2021-03-26 00:17:52 +08:00
}
2021-07-21 05:18:45 +08:00
func runGRPC(wg *sync.WaitGroup) {
2021-03-26 00:17:52 +08:00
defer wg.Done()
grpcport := servercfg.GetGRPCPort()
2021-03-26 00:17:52 +08:00
listener, err := net.Listen("tcp", ":"+grpcport)
2021-07-21 05:18:45 +08:00
// Handle errors if any
if err != nil {
2021-12-11 10:09:42 +08:00
logger.FatalLog("[netmaker] Unable to listen on port", grpcport, ": error:", err.Error())
}
2021-07-21 05:18:45 +08:00
s := grpc.NewServer(
authServerUnaryInterceptor(),
)
// Create NodeService type
srv := &controller.NodeServiceServer{}
// Register the service with the server
nodepb.RegisterNodeServiceServer(s, srv)
// Start the server in a child routine
go func() {
if err := s.Serve(listener); err != nil {
2021-12-11 10:09:42 +08:00
logger.FatalLog("Failed to serve:", err.Error())
2021-07-21 05:18:45 +08:00
}
}()
2021-12-07 04:31:08 +08:00
logger.Log(0, "Agent Server successfully started on port ", grpcport, "(gRPC)")
2021-07-21 05:18:45 +08:00
// Relay os.Interrupt to our channel (os.Interrupt = CTRL+C)
// Ignore other incoming signals
ctx, stop := signal.NotifyContext(context.TODO(), os.Interrupt)
defer stop()
2021-07-21 05:18:45 +08:00
// Block main routine until a signal is received
// As long as user doesn't press CTRL+C a message is not passed and our main routine keeps running
<-ctx.Done()
2021-07-21 05:18:45 +08:00
// After receiving CTRL+C Properly stop the server
2021-12-07 04:31:08 +08:00
logger.Log(0, "Stopping the Agent server...")
s.GracefulStop()
2021-07-21 05:18:45 +08:00
listener.Close()
2021-12-07 04:31:08 +08:00
logger.Log(0, "Agent server closed..")
logger.Log(0, "Closed DB connection.")
2021-03-26 00:17:52 +08:00
}
2021-04-06 20:24:57 +08:00
2021-03-26 00:17:52 +08:00
func authServerUnaryInterceptor() grpc.ServerOption {
return grpc.UnaryInterceptor(controller.AuthServerUnaryInterceptor)
}
2021-08-06 03:27:05 +08:00
2021-11-17 10:19:13 +08:00
func setGarbageCollection() {
_, gcset := os.LookupEnv("GOGC")
if !gcset {
debug.SetGCPercent(ncutils.DEFAULT_GC_PERCENT)
}
}