talos/internal/app/ntpd/main.go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

package main

import (
	"flag"
	"log"

	"github.com/talos-systems/talos/internal/app/ntpd/pkg/ntp"
	"github.com/talos-systems/talos/internal/app/ntpd/pkg/reg"
	"github.com/talos-systems/talos/pkg/config"
	"github.com/talos-systems/talos/pkg/constants"
	"github.com/talos-systems/talos/pkg/grpc/factory"
	"github.com/talos-systems/talos/pkg/startup"
)

// https://access.redhat.com/solutions/39194
// Using the above as reference for setting min/max
const (
	// TODO: Once we get naming sorted we need to apply
	// for a project specific address
	// https://manage.ntppool.org/manage/vendor
	DefaultServer = "pool.ntp.org"
)

var configPath *string

func init() {
	log.SetFlags(log.Lshortfile | log.Ldate | log.Lmicroseconds | log.Ltime)

	configPath = flag.String("config", "", "the path to the config")

	flag.Parse()
}

// New instantiates a new ntp instance against a given server
// If no servers are specified, the default will be used
func main() {
	if err := startup.RandSeed(); err != nil {
		log.Fatalf("startup: %v", err)
	}

	server := DefaultServer

	content, err := config.FromFile(*configPath)
	if err != nil {
		log.Fatalf("open config: %v", err)
	}

	config, err := config.New(content)
	if err != nil {
		log.Fatalf("open config: %v", err)
	}

	// Check if ntp servers are defined
	if config.Machine().Time().Server() != "" {
		server = config.Machine().Time().Server()
	}

	n, err := ntp.NewNTPClient(
		ntp.WithServer(server),
	)
	if err != nil {
		log.Fatalf("failed to create ntp client: %v", err)
	}

	log.Println("Starting ntpd")

	errch := make(chan error)

	go func() {
		errch <- n.Daemon()
	}()

	go func() {
		errch <- factory.ListenAndServe(
			reg.NewRegistrator(n),
			factory.Network("unix"),
			factory.SocketPath(constants.TimeSocketPath),
		)
	}()

	log.Fatal(<-errch)
}